forked from K0lb3/UnityPy
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathCompressionHelper.py
More file actions
65 lines (47 loc) · 1.18 KB
/
Copy pathCompressionHelper.py
File metadata and controls
65 lines (47 loc) · 1.18 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
import gzip
import lzma
import struct
import brotli
import lz4.block
GZIP_MAGIC = b'\x1f\x8b'
BROTLI_MAGIC = b'brotli'
# LZMA
def decompress_lzma(data):
props, dict_size = struct.unpack("<BI", data[:5])
lc = props % 9
props = props // 9
pb = props // 5
lp = props % 5
dec = lzma.LZMADecompressor(format=lzma.FORMAT_RAW, filters=[{
"id": lzma.FILTER_LZMA1,
"dict_size": dict_size,
"lc": lc,
"lp": lp,
"pb": pb,
}])
return dec.decompress(data[5:])
def compress_lzma(data):
raise NotImplementedError("compress_lzma")
dict_size = len(data)
ec = lzma.LZMACompressor(format=lzma.FORMAT_RAW, filters=[{
"id": lzma.FILTER_LZMA1,
"dict_size": dict_size,
"lc": lc,
"lp": lp,
"pb": pb,
}])
# LZ4
def decompress_lz4(data, uncompressed_size): # LZ4M LZ4HC
return lz4.block.decompress(data, uncompressed_size)
def compress_lz4(data): # LZ4M LZ4HC
return lz4.block.compress(data, store_size=False, mode="high_compression")
# Brotli
def decompress_brotli(data):
return brotli.decompress(data)
def compress_brotli(data):
return brotli.compress(data)
# GZIP
def decompress_gzip(data):
return gzip.decompress(data)
def compress_gzip(data):
return gzip.compress(data)