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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
|
# -*- coding:'utf-8' -*-
import time,traceback,sqlite3,os,json,base64,sys
from cryptography.hazmat.backends import default_backend
from cryptography.hazmat.primitives.ciphers import Cipher
from cryptography.hazmat.primitives.ciphers.algorithms import AES
from cryptography.hazmat.primitives.ciphers.modes import CBC
from cryptography.hazmat.primitives.hashes import SHA1
from cryptography.hazmat.primitives.kdf.pbkdf2 import PBKDF2HMAC
from cryptography.hazmat.primitives.ciphers.aead import AESGCM
# pip install cryptography
# pip install pypiwin32 (需要管理员权限)
def decryptString(key,data):
iv = data[3:15]
cipherbytes = data[15:]
aesgcm=AESGCM(key)
plainbytes=aesgcm.decrypt(iv, cipherbytes, None)
plaintext=plainbytes.decode('utf-8')
return plaintext
def getKey():
LocalState = ''
if (os.path.exists(os.environ['LOCALAPPDATA']+r"\Google\Chrome\User Data\Local State")):
LocalState = os.environ['LOCALAPPDATA']+r"\Google\Chrome\User Data\Local State"
elif (os.path.exists(os.environ['LOCALAPPDATA']+r"\Google\Chrome Dev\User Data\Local State")):
LocalState = os.environ['LOCALAPPDATA']+r"\Google\Chrome Dev\User Data\Local State"
else:
raise Exception('not found Cookie file!')
with open(LocalState, 'r', encoding='utf-8') as f:
base64_encrypted_key = json.load(f)['os_crypt']['encrypted_key']
# from win32crypt import CryptUnprotectData
import win32crypt
encrypted_key_with_header=base64.b64decode(base64_encrypted_key)
encrypted_key=encrypted_key_with_header[5:]
key = win32crypt.CryptUnprotectData(encrypted_key,None,None,None,0)[1]
# mykey = base64.b64encode(key)
return key
def getChromeCookie(hostsUrl):
cookiepath = ''
if (os.path.exists(os.environ['LOCALAPPDATA']+r"\Google\Chrome\User Data\Default\Network\Cookies")):
cookiepath = os.environ['LOCALAPPDATA']+r"\Google\Chrome\User Data\Default\Network\Cookies"
elif (os.path.exists(os.environ['LOCALAPPDATA']+r"\Google\Chrome\User Data\Default\Cookies")):
cookiepath = os.environ['LOCALAPPDATA']+r"\Google\Chrome\User Data\Default\Cookies"
elif (os.path.exists(os.environ['LOCALAPPDATA']+r"\Google\Chrome Dev\User Data\Default\Cookies")):
cookiepath = os.environ['LOCALAPPDATA']+r"\Google\Chrome Dev\User Data\Default\Cookies"
else:
raise Exception('not found Cookie file!')
sql = f"select name,encrypted_value from cookies where host_key = '{hostsUrl}'"
try:
conn = sqlite3.connect(cookiepath)
conn.text_factory = bytes
res=conn.execute(sql).fetchall()
conn.close()
except Exception as e:
print(e)
key = getKey()
# row = res.pop()
# cookieList.append( f'{str(row[0],encoding = "utf-8")}={decryptString(key, row[1])}')
cookies = {}
for item in res:
cookies[item[0].decode("utf-8")] = decryptString(key, item[1])
# print(cookies)
return cookies
def clean(decrypted: bytes) -> str:
last = decrypted[-1]
if isinstance(last, int):
return decrypted[:-last].decode("utf8")
return decrypted[: -ord(last)].decode("utf8")
def chrome_decrypt(encrypted_value: bytes, key: bytes, init_vector: bytes):
encrypted_value = encrypted_value[3:]
cipher = Cipher(algorithm=AES(key), mode=CBC(init_vector), backend=default_backend())
decryptor = cipher.decryptor()
decrypted = decryptor.update(encrypted_value) + decryptor.finalize()
return clean(decrypted)
def chrome_cookies(url: str, browser: str = "Chrome",):
import keyring
file_path = ''
if (os.path.exists(os.path.expanduser('~/Library/Application Support/Google/Chrome/Default/Extension Cookies'))):
file_path = os.path.expanduser('~/Library/Application Support/Google/Chrome/Default/Extension Cookies')
elif (os.path.exists(os.path.expanduser('~/Library/Application Support/Chromium/Default/Cookies'))):
file_path = os.path.expanduser('~/Library/Application Support/Chromium/Default/Cookies')
elif (os.path.exists(os.path.expanduser('~/Library/Application Support/Google/Chrome Dev/Default/Cookies'))):
file_path = os.path.expanduser('~/Library/Application Support/Google/Chrome Dev/Default/Cookies')
else:
raise Exception('not found Cookie file!')
config = {
"my_pass": keyring.get_password(
"{} Safe Storage".format(browser), browser
),
"iterations": 1003,
"cookie_file": file_path,
"init_vector": b" " * 16,
"length": 16,
"salt": b"saltysalt"
}
if isinstance(config["my_pass"], str):
config["my_pass"] = config["my_pass"].encode("utf8")
kdf = PBKDF2HMAC(
algorithm=SHA1(),
backend=default_backend(),
iterations=config["iterations"],
length=config["length"],
salt=config["salt"],
)
enc_key = kdf.derive(config["my_pass"])
try:
conn = sqlite3.connect(config["cookie_file"])
except sqlite3.OperationalError:
print("Unable to connect to cookie_file at: {}\n".format(config["cookie_file"]))
raise
sql = ("select name, encrypted_value from cookies where host_key like ?")
cookies = ''
for item in conn.execute(sql, (url,)).fetchall():
cookies = chrome_decrypt(item[1], key=enc_key, init_vector=config["init_vector"])
conn.rollback()
return cookies
def getCookie(url):
# mac
if sys.platform == "darwin":
return chrome_cookies(url)
# windows
elif sys.platform == 'win32':
return getChromeCookie(url)
print(json.dumps(getCookie('.bilibili.com')))
|