golang读取chrome/edge浏览器本地cookies数据库

cflying

去爬个资料，登录验证太麻烦，不如本地浏览器登录后读取cookies直接爬来的方便，本来打算网上找一个现成的代码来读取浏览器cookies数据库，结果度娘半天竟然没有（网上全都是python的，偶尔有个读保存账号和密码的，结果N年前的解密方式还没更新），要不就只有去仓库下，感觉有点小题大做了，所以不如拿python的改一个单文件的，这样使用起来一个文件带到哪都方便。

1. 
   
2. package main
   
3. 
   
4. import (
   
5.         "database/sql"
   
6.         "fmt"
   
7.         "os"
   
8.         "io/ioutil"
   
9.         "encoding/json"
   
10.         "encoding/base64"
    
11.         "syscall"
    
12.         "unsafe"
    
13.         "crypto/aes"
    
14.         "crypto/cipher"
    
15.         _ "github.com/mattn/go-sqlite3"
    
16. )
    
17. 
    
18. 
    
19. type DATA_BLOB struct {
    
20.         cbData uint32
    
21.         pbData *byte
    
22. }
    
23. func NewBlob(d []byte) *DATA_BLOB {
    
24.         if len(d)==0 {
    
25.                 return &DATA_BLOB{}
    
26.         }
    
27.         b:=&DATA_BLOB{
    
28.         pbData:&d[0],
    
29.         cbData:uint32(len(d)),
    
30.         }
    
31.         
    
32.         return b
    
33. }
    
34. func (b *DATA_BLOB) ToByteArray() []byte {
    
35.         d:=make([]byte, b.cbData)
    
36.         copy(d, (*[1 << 30]byte)(unsafe.Pointer(b.pbData))[:])
    
37.         return d
    
38. }
    
39. 
    
40. func dpapi_decrypt(encrypted []byte)([]byte, error){
    
41.         dllcrypt32:=syscall.NewLazyDLL("Crypt32.dll")
    
42.         dllkernel32:=syscall.NewLazyDLL("Kernel32.dll")
    
43. 
    
44.         procDecryptData:=dllcrypt32.NewProc("CryptUnprotectData")
    
45.         procLocalFree:=dllkernel32.NewProc("LocalFree")
    
46.         var outblob DATA_BLOB
    
47.         r,_,err:=procDecryptData.Call(uintptr(unsafe.Pointer(NewBlob(encrypted))), 0, 0, 0, 0,0x1, uintptr(unsafe.Pointer(&outblob)))
    
48.         if r==0 {
    
49.                 return nil, err        
    
50.         }
    
51.         defer procLocalFree.Call(uintptr(unsafe.Pointer(outblob.pbData)))
    
52.         return outblob.ToByteArray(), err
    
53. }
    
54. 
    
55. 
    
56. func aesGCMDecrypt(data []byte, key []byte, iv []byte) []byte {
    
57.         block,_:=aes.NewCipher(key)
    
58.         aesGcm,_:=cipher.NewGCM(block)
    
59.         plaintext,_:=aesGcm.Open(nil,iv,data, nil)
    
60.         return plaintext
    
61. }
    
62. 
    
63. func aes_decrypt(encrypted_txt string)([]byte,){
    
64.         key_file,_:=os.OpenFile(os.Getenv("LOCALAPPDATA")+ "\\Microsoft\\Edge\\User Data\\Local State",os.O_RDONLY,os.ModePerm)
    
65.         defer key_file.Close()
    
66.         key_data,_:=ioutil.ReadAll(key_file)
    
67. 
    
68.         dynamic:=make(map[string]interface{})
    
69.         json.Unmarshal([]byte(string(key_data)),&dynamic)
    
70.         encrypted_key,_:=base64.StdEncoding.DecodeString(dynamic["os_crypt"].(map[string]interface{})["encrypted_key"].(string))
    
71.         encrypted_key=encrypted_key[5:]//去掉DPAPI后的加密key
    
72.         key,_:=dpapi_decrypt(encrypted_key)//解密后的key
    
73.         nonce:=encrypted_txt[3:15]//AES随机12位vi
    
74.         decryptor:=aesGCMDecrypt([]byte(encrypted_txt[15:]),key,[]byte(nonce))
    
75.         return decryptor
    
76. }
    
77. 
    
78. /*前缀处理*/
    
79. func chrome_decrypt (encrypted_txt string)[]byte{
    
80.         var decrypted_txt []byte
    
81.         if encrypted_txt[:3]=="v10"{
    
82.                 decrypted_txt=aes_decrypt(encrypted_txt)
    
83.         }
    
84.         if encrypted_txt[:4]=="DPAP"{
    
85.                 decrypted_txt,_=dpapi_decrypt([]byte(encrypted_txt))
    
86.         }
    
87.         return decrypted_txt
    
88. }
    
89. 
    
90. 
    
91. func get_cookies_from_chrome(domain string)([]map[string]string){
    
92.         query:=`SELECT name, encrypted_value as value,host_key as domain, path FROM cookies where host_key like `+`"%`+domain+`%"`
    
93.         fmt.Println(query)
    
94.         var file string
    
95.         if _,err:=os.Stat(os.Getenv("LOCALAPPDATA")+"\\Microsoft\\Edge\\User Data\\Default\\Network\\Cookies");err==nil{
    
96.                 file=os.Getenv("LOCALAPPDATA")+"\\Microsoft\\Edge\\User Data\\Default\\Network\\Cookies"
    
97.         }else if _,err:=os.Stat(os.Getenv("LOCALAPPDATA")+"\\Microsoft\\Edge\\User Data\\Profile 3\\Network\\Cookies");err==nil{
    
98.                 file=os.Getenv("LOCALAPPDATA")+"\\Microsoft\\Edge\\User Data\\Profile 3\\Network\\Cookies"
    
99.         }
    
100.        
     
101.         fmt.Println(file)
     
102.         db,_:= sql.Open("sqlite3", file)
     
103.         defer db.Close()
     
104.         rows,_:= db.Query(query)
     
105.         defer rows.Close()
     
106.         var cookies []map[string]string
     
107.         for rows.Next() {
     
108.         var name,value,domain,path string
     
109.         rows.Scan(&name, &value, &domain, &path)        
     
110.         if name != "" && value != ""{
     
111.                 cc_data_tmp:=make(map[string]string)
     
112.                 cc_data_tmp["name"]=name
     
113.                 cc_data_tmp["vlaue"]=string(chrome_decrypt(value))
     
114.                 cc_data_tmp["doamin"]=domain
     
115.                 cc_data_tmp["path"]=path
     
116.                 cookies=append(cookies,cc_data_tmp)
     
117.         }               
     
118. }
     
119. 
     
120.        
     
121.         return cookies
     
122. }
     
123. 
     
124. func main() {
     
125.         //fmt.Println(get_cookies_from_chrome(".hao123.com")[0]["vlaue"])
     
126.         for _,v:=range get_cookies_from_chrome(".hao123.com"){
     
127.                 fmt.Println(v)
     
128.         }
     
129. 
     
130. }
     

复制代码

saku0819

一看就是个高手