🔧 chore: 初始化番茄小说 API 项目结构
新增项目基础框架，包含核心 API 实现、算法模块、HTTP 处理器等完整功能模块
This commit is contained in:
chenxin committed
commit
6981148727
107 files changed
+12463
No files matched your search
@@ -0,0 +1,249 @@
|
||||
package fqapi
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"compress/gzip"
|
||||
"crypto/aes"
|
||||
"crypto/cipher"
|
||||
"encoding/base64"
|
||||
"encoding/hex"
|
||||
"encoding/json"
|
||||
"fanqie/internal/common"
|
||||
"fmt"
|
||||
"math/rand"
|
||||
"regexp"
|
||||
"sort"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
// KeyInfo 密钥信息
|
||||
type KeyInfo struct {
|
||||
Key string `json:"key"`
|
||||
Ver int `json:"ver"`
|
||||
}
|
||||
|
||||
var (
|
||||
keyInfo *KeyInfo
|
||||
keyLogger = common.NewLogger("register_key", "INFO", "")
|
||||
)
|
||||
|
||||
// rHex 将数字转换为反转的16进制字符串
|
||||
func rHex(num int64) string {
|
||||
hexStr := fmt.Sprintf("%032x", num)
|
||||
result := ""
|
||||
for i := len(hexStr); i > 0; i -= 2 {
|
||||
result += hexStr[i-2 : i]
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
// createSymmetricCrypto 创建对称加密器
|
||||
func createSymmetricCrypto(algorithm string, key, iv []byte) (func([]byte) ([]byte, error), func([]byte) ([]byte, error), error) {
|
||||
if algorithm != "AES/CBC/PKCS5Padding" {
|
||||
return nil, nil, fmt.Errorf("不支持的加密算法")
|
||||
}
|
||||
|
||||
block, err := aes.NewCipher(key)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
|
||||
encrypt := func(data []byte) ([]byte, error) {
|
||||
// PKCS7填充
|
||||
padding := aes.BlockSize - len(data)%aes.BlockSize
|
||||
padText := make([]byte, padding)
|
||||
for i := range padText {
|
||||
padText[i] = byte(padding)
|
||||
}
|
||||
paddedData := append(data, padText...)
|
||||
|
||||
ciphertext := make([]byte, len(paddedData))
|
||||
mode := cipher.NewCBCEncrypter(block, iv)
|
||||
mode.CryptBlocks(ciphertext, paddedData)
|
||||
|
||||
return ciphertext, nil
|
||||
}
|
||||
|
||||
decrypt := func(data []byte) ([]byte, error) {
|
||||
mode := cipher.NewCBCDecrypter(block, iv)
|
||||
decrypted := make([]byte, len(data))
|
||||
mode.CryptBlocks(decrypted, data)
|
||||
|
||||
// 移除PKCS7填充
|
||||
padding := int(decrypted[len(decrypted)-1])
|
||||
if padding > len(decrypted) || padding > aes.BlockSize {
|
||||
return nil, fmt.Errorf("无效的填充")
|
||||
}
|
||||
|
||||
return decrypted[:len(decrypted)-padding], nil
|
||||
}
|
||||
|
||||
return encrypt, decrypt, nil
|
||||
}
|
||||
|
||||
// En AES加密
|
||||
func En(data, key, iv []byte) ([]byte, error) {
|
||||
encrypt, _, err := createSymmetricCrypto("AES/CBC/PKCS5Padding", key, iv)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return encrypt(data)
|
||||
}
|
||||
|
||||
// De AES解密
|
||||
func De(data, key, iv []byte) ([]byte, error) {
|
||||
_, decrypt, err := createSymmetricCrypto("AES/CBC/PKCS5Padding", key, iv)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return decrypt(data)
|
||||
}
|
||||
|
||||
// randomStringKey 生成随机字符串（密钥用）
|
||||
func randomStringKey(length int) string {
|
||||
const charset = "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
||||
b := make([]byte, length)
|
||||
for i := range b {
|
||||
b[i] = charset[rand.Intn(len(charset))]
|
||||
}
|
||||
return string(b)
|
||||
}
|
||||
|
||||
// GetKey 获取密钥
|
||||
func GetKey(header string) (*KeyInfo, error) {
|
||||
if header == "" {
|
||||
keyLogger.Error("请先注册设备")
|
||||
return nil, fmt.Errorf("请先注册设备")
|
||||
}
|
||||
|
||||
key, _ := base64.StdEncoding.DecodeString("rCXGfd2POMGzeiNIgo4iLg==")
|
||||
iv := []byte(randomStringKey(16))
|
||||
|
||||
// 提取device_id
|
||||
re := regexp.MustCompile(`device_id=(\d+)`)
|
||||
matches := re.FindStringSubmatch(header)
|
||||
if len(matches) < 2 {
|
||||
return nil, fmt.Errorf("无法从header中提取device_id")
|
||||
}
|
||||
|
||||
var deviceIDInt int64
|
||||
fmt.Sscanf(matches[1], "%d", &deviceIDInt)
|
||||
|
||||
// 转换device_id
|
||||
hexStr := rHex(deviceIDInt)
|
||||
data, _ := hex.DecodeString(hexStr)
|
||||
|
||||
keyLogger.Debug(fmt.Sprintf("rKey: %v", iv))
|
||||
|
||||
// 加密
|
||||
encrypted, err := En(data, key, iv)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// 组合iv和加密数据
|
||||
newArr := append(iv, encrypted...)
|
||||
content := base64.StdEncoding.EncodeToString(newArr)
|
||||
|
||||
keyLogger.Info("正在获取密钥...")
|
||||
|
||||
// 构建参数
|
||||
params := []string{
|
||||
"aid=1967",
|
||||
"app_name=novelapp",
|
||||
"channel=0",
|
||||
"device_platform=android",
|
||||
"os_version=0",
|
||||
"version_code=5.8.9.32",
|
||||
header,
|
||||
}
|
||||
|
||||
allParams := strings.Join(params, "&")
|
||||
paramList := strings.Split(allParams, "&")
|
||||
sort.Strings(paramList)
|
||||
sortedParams := strings.Join(paramList, "&")
|
||||
|
||||
// 准备请求体
|
||||
requestBody := map[string]string{
|
||||
"content": content,
|
||||
}
|
||||
|
||||
bodyJSON, _ := json.Marshal(requestBody)
|
||||
|
||||
// gzip压缩
|
||||
var gzipBuf bytes.Buffer
|
||||
gzipWriter := gzip.NewWriter(&gzipBuf)
|
||||
gzipWriter.Write(bodyJSON)
|
||||
gzipWriter.Close()
|
||||
|
||||
reqURL := "https://reading.snssdk.com/reading/crypt/registerkey?" + sortedParams
|
||||
|
||||
resp, err := GorgonRequest(reqURL, map[string]interface{}{
|
||||
"method": "POST",
|
||||
"body": gzipBuf.Bytes(),
|
||||
"headers": map[string]string{
|
||||
"User-Agent": "com.dragon.read",
|
||||
"Content-Encoding": "gzip",
|
||||
},
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
keyLogger.Error("获取密钥失败: " + err.Error())
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if resp.Status == 200 {
|
||||
var respData map[string]interface{}
|
||||
json.Unmarshal(resp.Content, &respData)
|
||||
|
||||
dataObj, ok := respData["data"].(map[string]interface{})
|
||||
if !ok {
|
||||
return nil, fmt.Errorf("响应格式错误")
|
||||
}
|
||||
|
||||
keyStr, ok := dataObj["key"].(string)
|
||||
if !ok {
|
||||
return nil, fmt.Errorf("密钥字段不存在")
|
||||
}
|
||||
|
||||
// 解码返回的密钥
|
||||
res, _ := base64.StdEncoding.DecodeString(keyStr)
|
||||
if len(res) < 32 {
|
||||
return nil, fmt.Errorf("密钥长度不足")
|
||||
}
|
||||
|
||||
data := res[16:]
|
||||
ivResp := res[0:16]
|
||||
|
||||
// 解密密钥
|
||||
decryptedKey, err := De(data, key, ivResp)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
finalKey := base64.StdEncoding.EncodeToString(decryptedKey)
|
||||
|
||||
keyVer, _ := dataObj["keyver"].(float64)
|
||||
keyInfo = &KeyInfo{
|
||||
Key: finalKey,
|
||||
Ver: int(keyVer),
|
||||
}
|
||||
|
||||
keyLogger.Info(fmt.Sprintf("密钥获取成功！"))
|
||||
|
||||
return keyInfo, nil
|
||||
}
|
||||
|
||||
return nil, fmt.Errorf("获取密钥失败")
|
||||
}
|
||||
|
||||
// GetKeyInfo 获取当前密钥信息
|
||||
func GetKeyInfo() *KeyInfo {
|
||||
return keyInfo
|
||||
}
|
||||
|
||||
func init() {
|
||||
rand.Seed(time.Now().UnixNano())
|
||||
}
|
||||
|
||||
Reference in new issue
Block a user