Merge pull request #158 from AbelLin1214/master

修复#155 提到的must login错误

Author: keac

README.md

@@ -161,6 +161,13 @@ Cookie信息请勿直接 `document.cookie`，可能因为http-only 选项无法
 
 ![image-20230722194839975](./README/image-20230722194839975.png)
 
+
+**TYC auth_token**
+
+配置COOKIE后配置auth_token
+
+![image-20250215132223242](./README/image-20250215132223242.jpg)
+
 其他Cookie请自行参考获取
 
 

README/image-20250215132223242.jpg

@@ -2,11 +2,12 @@ package common
 
 import (
 	"fmt"
+	"path/filepath"
+	"time"
+
 	"github.com/tidwall/gjson"
 	"github.com/wgpsec/ENScan/common/utils"
 	"go.mongodb.org/mongo-driver/bson/primitive"
-	"path/filepath"
-	"time"
 )
 
 var (
@@ -143,6 +144,7 @@ type ENConfig struct {
 		KuaiCha    string `yaml:"kuaicha"`
 		Tianyancha string `yaml:"tianyancha"`
 		Tycid      string `yaml:"tycid"`
+		AuthToken  string `yaml:"auth_token"`
 		QiMai      string `yaml:"qimai"`
 	}
 	App struct {

common/config.go

@@ -3,15 +3,16 @@ package tianyancha
 import (
 	"crypto/tls"
 	"fmt"
+	"regexp"
+	"strings"
+	"time"
+
 	"github.com/antchfx/htmlquery"
 	"github.com/imroc/req/v3"
 	"github.com/robertkrimen/otto"
 	"github.com/wgpsec/ENScan/common"
 	"github.com/wgpsec/ENScan/common/gologger"
 	"golang.org/x/net/html"
-	"regexp"
-	"strings"
-	"time"
 )
 
 func getENMap() map[string]*common.EnsGo {
@@ -144,7 +145,7 @@ func GetReq(url string, data string, options *common.ENOptions) string {
 		clientR.SetHeader("Content-Type", "application/json")
 		//client.Header.Del("Cookie")
 		clientR.SetHeader("X-Tycid", options.ENConfig.Cookies.Tycid)
-		//client.Header.Set("X-Auth-Token", "")
+		clientR.SetHeader("X-Auth-Token", options.ENConfig.Cookies.AuthToken)
 	}
 	//加延迟1S
 	//强制延时1s