mirror of
https://github.com/zu1k/nali.git
synced 2025-01-22 21:29:02 +08:00
76 lines
1.5 KiB
Go
76 lines
1.5 KiB
Go
package qqwry
|
||
|
||
import (
|
||
"bytes"
|
||
"compress/zlib"
|
||
"encoding/binary"
|
||
"fmt"
|
||
"io/ioutil"
|
||
"log"
|
||
|
||
"github.com/zu1k/nali/pkg/common"
|
||
)
|
||
|
||
func Download(filePath ...string) (data []byte, err error) {
|
||
fmt.Println("此方式更新的 QQWry 数据库版本过旧，请手动下载最新版纯真免费IP库: https://www.cz88.net/help")
|
||
|
||
data, err = downloadAndDecrypt()
|
||
if err != nil {
|
||
log.Printf("纯真IP库下载失败，请手动下载解压后保存到本地: %s \n", filePath)
|
||
log.Println("下载链接： https://qqwry.mirror.noc.one/qqwry.rar")
|
||
return
|
||
}
|
||
|
||
if len(filePath) == 1 {
|
||
if err := common.SaveFile(filePath[0], data); err == nil {
|
||
log.Println("已将最新的 纯真IP库 保存到本地:", filePath)
|
||
}
|
||
}
|
||
return
|
||
}
|
||
|
||
const (
|
||
mirror = "https://qqwry.mirror.noc.one/qqwry.rar"
|
||
key = "https://qqwry.mirror.noc.one/copywrite.rar"
|
||
)
|
||
|
||
func downloadAndDecrypt() (data []byte, err error) {
|
||
data, err = common.GetHttpClient().Get(mirror)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
key, err := getCopyWriteKey()
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
return unRar(data, key)
|
||
}
|
||
|
||
func unRar(data []byte, key uint32) ([]byte, error) {
|
||
for i := 0; i < 0x200; i++ {
|
||
key = key * 0x805
|
||
key++
|
||
key = key & 0xff
|
||
|
||
data[i] = byte(uint32(data[i]) ^ key)
|
||
}
|
||
|
||
reader, err := zlib.NewReader(bytes.NewReader(data))
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
return ioutil.ReadAll(reader)
|
||
}
|
||
|
||
func getCopyWriteKey() (uint32, error) {
|
||
body, err := common.GetHttpClient().Get(key)
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
|
||
return binary.LittleEndian.Uint32(body[5*4:]), nil
|
||
}
|