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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
| import math
import binascii
def baseDecode(inputString, filePath, basePath):
dist = {char: index for index, char in enumerate(inputString)}
bitLength = math.ceil(math.log2(len(inputString)))
with open(filePath, "r", encoding="utf-8") as file:
encryptedText = file.read().strip()
encryptedNumbers = [dist[char] for char in encryptedText]
encryptedBinary = ''.join([bin(num)[2:].zfill(bitLength) for num in encryptedNumbers])
byteArray = bytearray(int(encryptedBinary[i:i+8], 2) for i in range(0, len(encryptedBinary), 8))
return encryptedBinary, byteArray
def guessFileType(byteArray):
magicNumber = binascii.hexlify(byteArray[:8]).decode('utf-8')
fileTypes = {
"504b0304": "zip",
"52617221": "rar",
"377abcaf": "7z",
"03f30d0a": "pyc",
"ffd8ffe0": "jpg",
"ffd8ffe1": "jpg",
"ffd8ffe2": "jpg",
"89504e47": "png",
"47494638": "gif",
"1f8b0800": "gz",
"38425053": "psd",
"49492a00": "tif",
"424d": "bmp",
"41433130": "dwg",
"7b5c7274": "rtf",
"3c3f786d6c": "xml",
"68746d6c3e": "html",
"44656c69766572792d646174653a": "eml",
"cfad12fec5fd746f": "dbx",
"2142444e": "pst",
"d0cf11e0": "doc/xls",
"5374616e64617264204a": "mdb",
"ff575043": "wpd",
"252150532d41646f6265": "eps/ps",
"255044462d312e": "pdf",
"ac9ebd8f": "qdf",
"e3828596": "pwl",
"57415645": "wav",
"41564920": "avi",
"2e7261fd": "ram",
"2e524d46": "rm",
"000001ba": "mpg",
"6d6f6f76": "mov",
"3026b2758e66cf11": "asf",
"4d546864": "mid",
"00000018667479704d3441": "m4a",
}
maxMagicLen = max(len(key) for key in fileTypes.keys())
magicNumber = magicNumber[:maxMagicLen]
for key, value in fileTypes.items():
if magicNumber.startswith(key):
return value
return None
def saveFile(fileExtension, byteArray, encryptedBinary, basePath):
if fileExtension is None:
binaryOutputPath = f"{basePath}binary.txt"
with open(binaryOutputPath, "w", encoding="utf-8") as file:
file.write(encryptedBinary)
print(f"无法识别文件类型，已自动保存二进制字符串为文本文件: {binaryOutputPath}")
else:
outputFilePath = f"{basePath}decrypted_file.{fileExtension}"
with open(outputFilePath, "wb") as file:
file.write(byteArray)
print(f"识别到类型为: {fileExtension}")
print(f"文件已自动保存为: {outputFilePath}")
saveBinary = input("是否要保存二进制字符串到文本文件？(yes/no): ").strip().lower()
if saveBinary == 'yes' or saveBinary == 'y':
binaryOutputPath = f"{basePath}binary.txt"
with open(binaryOutputPath, "w", encoding="utf-8") as file:
file.write(encryptedBinary)
print(f"二进制字符串已保存为: {binaryOutputPath}")
def main():
basePath = "C:/Users/67300/Desktop/Misc/base8/"
filePath = f"{basePath}new-base.txt"
inputString = "BaseCTF!"
encryptedBinary, byteArray = baseDecode(inputString, filePath, basePath)
fileExtension = guessFileType(byteArray)
saveFile(fileExtension, byteArray, encryptedBinary, basePath)
if __name__ == "__main__":
main()
|
