From: qydysky <qydysky@foxmail.com>
Date: Sat, 15 Jun 2024 17:58:46 +0000 (+0000)
Subject: Improve cookie.txt 储存优化
X-Git-Tag: v0.14.5~2
X-Git-Url: http://127.0.0.1:8081/?a=commitdiff_plain;h=0d8db5f64723bb2875796e1c2e357e0b87f7d099;p=bili_danmu%2F.git

Improve cookie.txt 储存优化
---

diff --git a/F/CookieCrypo.go b/F/CookieCrypo.go
index 157718e..2179c03 100644
--- a/F/CookieCrypo.go
+++ b/F/CookieCrypo.go
@@ -40,22 +40,39 @@ func CookieGet() []byte {
 				pri = d
 			}
 		} else {
-			if d, e := crypto.FileLoad(`cookie.txt`); e != nil || string(d[:6]) != `t=nol;` {
-				clog.L(`E: `, e, `cookie保存格式:`, string(d[:6]))
+			if d, e := crypto.FileLoad(`cookie.txt`); e != nil {
+				clog.L(`E: `, e, `cookie保存格式`)
 				return []byte{}
-			} else {
+			} else if string(d[:6]) == `t=nol;` {
 				return d[6:]
+			} else if string(d[:3]) == `nol` {
+				return d[3:]
+			} else {
+				clog.L(`E: `, e, `cookie保存格式:`, string(d[:6]))
+				return []byte{}
 			}
 		}
 	}
-	if source, e := crypto.FileLoad(`cookie.txt`); e != nil || string(source[:6]) != `t=pem;` {
-		clog.L(`E: `, e, `cookie保存格式:`, string(source[:6]))
-		return []byte{}
-	} else if s, e := crypto.Decrypt(source[6:], pri); e != nil {
-		clog.L(`E: `, e)
+	if d, e := crypto.FileLoad(`cookie.txt`); e != nil {
+		clog.L(`E: `, e, `cookie保存格式`)
 		return []byte{}
+	} else if string(d[:6]) == `t=pem;` {
+		if s, e := crypto.Decrypt(d[6:], pri); e != nil {
+			clog.L(`E: `, e)
+			return []byte{}
+		} else {
+			return s
+		}
+	} else if string(d[:3]) == `pem` {
+		if s, e := crypto.Decrypt(d[3:], pri); e != nil {
+			clog.L(`E: `, e)
+			return []byte{}
+		} else {
+			return s
+		}
 	} else {
-		return s
+		clog.L(`E: `, e, `cookie保存格式:`, string(d[:6]))
+		return []byte{}
 	}
 }