mirrors/fscan
1
0
Fork 0
mirror of https://github.com/shadow1ng/fscan.git synced 2025-07-13 12:52:44 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: #439

Browse Source
This commit is contained in:
ZacharyZcR 2025-04-13 15:46:37 +08:00
parent b8cc8ab5dc
commit f79b12a23c
1 changed files with 110 additions and 47 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

157
Plugins/WebTitle.go
View File

@ -48,7 +48,7 @@ func WebTitle(info *Common.HostInfo) error {
return err return err
} }
// GOWebTitle 获取网站标题并处理URL // GOWebTitle 获取网站标题并处理URL，增强错误处理和协议切换
func GOWebTitle(info *Common.HostInfo) (err error, CheckData []WebScan.CheckDatas) { func GOWebTitle(info *Common.HostInfo) (err error, CheckData []WebScan.CheckDatas) {
Common.LogDebug(fmt.Sprintf("开始处理URL: %s", info.Url)) Common.LogDebug(fmt.Sprintf("开始处理URL: %s", info.Url))
@ -79,12 +79,35 @@ func GOWebTitle(info *Common.HostInfo) (err error, CheckData []WebScan.CheckData
} }
Common.LogDebug(fmt.Sprintf("协议检测完成后的URL: %s", info.Url)) Common.LogDebug(fmt.Sprintf("协议检测完成后的URL: %s", info.Url))
// 记录原始URL协议
originalProtocol := "http"
if strings.HasPrefix(info.Url, "https://") {
originalProtocol = "https"
}
// 第一次获取URL // 第一次获取URL
Common.LogDebug("第一次尝试访问URL") Common.LogDebug("第一次尝试访问URL")
err, result, CheckData := geturl(info, 1, CheckData) err, result, CheckData := geturl(info, 1, CheckData)
Common.LogDebug(fmt.Sprintf("第一次访问结果 - 错误: %v, 返回信息: %s", err, result)) Common.LogDebug(fmt.Sprintf("第一次访问结果 - 错误: %v, 返回信息: %s", err, result))
// 如果访问失败并且使用的是HTTPS，尝试降级到HTTP
if err != nil && !strings.Contains(err.Error(), "EOF") { if err != nil && !strings.Contains(err.Error(), "EOF") {
return if originalProtocol == "https" {
Common.LogDebug("HTTPS访问失败，尝试降级到HTTP")
// 替换协议部分
info.Url = strings.Replace(info.Url, "https://", "http://", 1)
Common.LogDebug(fmt.Sprintf("降级后的URL: %s", info.Url))
err, result, CheckData = geturl(info, 1, CheckData)
Common.LogDebug(fmt.Sprintf("HTTP降级访问结果 - 错误: %v, 返回信息: %s", err, result))
// 如果仍然失败，返回错误
if err != nil && !strings.Contains(err.Error(), "EOF") {
return
}
} else {
// 如果本来就是HTTP并且失败了，直接返回错误
return
}
} }
// 处理URL跳转 // 处理URL跳转
@ -94,32 +117,53 @@ func GOWebTitle(info *Common.HostInfo) (err error, CheckData []WebScan.CheckData
err, result, CheckData = geturl(info, 3, CheckData) err, result, CheckData = geturl(info, 3, CheckData)
Common.LogDebug(fmt.Sprintf("重定向请求结果 - 错误: %v, 返回信息: %s", err, result)) Common.LogDebug(fmt.Sprintf("重定向请求结果 - 错误: %v, 返回信息: %s", err, result))
if err != nil { if err != nil {
return // 如果重定向跟踪失败，尝试降级协议
} if strings.HasPrefix(info.Url, "https://") {
} Common.LogDebug("重定向HTTPS访问失败，尝试降级到HTTP")
info.Url = strings.Replace(info.Url, "https://", "http://", 1)
err, result, CheckData = geturl(info, 3, CheckData)
Common.LogDebug(fmt.Sprintf("重定向降级访问结果 - 错误: %v, 返回信息: %s", err, result))
}
// 处理HTTP到HTTPS的升级
if result == "https" && !strings.HasPrefix(info.Url, "https://") {
Common.LogDebug("正在升级到HTTPS")
info.Url = strings.Replace(info.Url, "http://", "https://", 1)
Common.LogDebug(fmt.Sprintf("升级后的URL: %s", info.Url))
err, result, CheckData = geturl(info, 1, CheckData)
// 处理升级后的跳转
if strings.Contains(result, "://") {
Common.LogDebug(fmt.Sprintf("HTTPS升级后发现重定向到: %s", result))
info.Url = result
err, _, CheckData = geturl(info, 3, CheckData)
if err != nil { if err != nil {
return return
} }
} }
} }
Common.LogDebug(fmt.Sprintf("GOWebTitle执行完成 - 错误: %v", err)) // 处理HTTP到HTTPS的升级提示
if err != nil { if result == "https" && !strings.HasPrefix(info.Url, "https://") {
return Common.LogDebug("正在升级到HTTPS")
info.Url = strings.Replace(info.Url, "http://", "https://", 1)
Common.LogDebug(fmt.Sprintf("升级后的URL: %s", info.Url))
err, result, CheckData = geturl(info, 1, CheckData)
Common.LogDebug(fmt.Sprintf("HTTPS升级访问结果 - 错误: %v, 返回信息: %s", err, result))
// 如果HTTPS升级后访问失败，回退到HTTP
if err != nil && !strings.Contains(err.Error(), "EOF") {
Common.LogDebug("HTTPS升级访问失败，回退到HTTP")
info.Url = strings.Replace(info.Url, "https://", "http://", 1)
err, result, CheckData = geturl(info, 1, CheckData)
Common.LogDebug(fmt.Sprintf("回退到HTTP访问结果 - 错误: %v, 返回信息: %s", err, result))
}
// 处理升级后的跳转
if strings.Contains(result, "://") {
Common.LogDebug(fmt.Sprintf("协议升级后发现重定向到: %s", result))
info.Url = result
err, _, CheckData = geturl(info, 3, CheckData)
if err != nil {
// 如果重定向跟踪失败，再次尝试降级
if strings.HasPrefix(info.Url, "https://") {
Common.LogDebug("升级后重定向HTTPS访问失败，尝试降级到HTTP")
info.Url = strings.Replace(info.Url, "https://", "http://", 1)
err, _, CheckData = geturl(info, 3, CheckData)
}
}
}
} }
Common.LogDebug(fmt.Sprintf("GOWebTitle执行完成 - 错误: %v", err))
return return
} }
@ -367,10 +411,12 @@ func gettitle(body []byte) (title string) {
return return
} }
// GetProtocol 检测目标主机的协议类型(HTTP/HTTPS) // GetProtocol 检测目标主机的协议类型(HTTP/HTTPS)，优先返回可用的协议
func GetProtocol(host string, Timeout int64) (protocol string) { func GetProtocol(host string, Timeout int64) (protocol string) {
Common.LogDebug(fmt.Sprintf("开始检测主机协议 - 主机: %s, 超时: %d秒", host, Timeout)) Common.LogDebug(fmt.Sprintf("开始检测主机协议 - 主机: %s, 超时: %d秒", host, Timeout))
protocol = "http" // 默认协议
// 默认使用http协议
protocol = "http"
timeoutDuration := time.Duration(Timeout) * time.Second timeoutDuration := time.Duration(Timeout) * time.Second
ctx, cancel := context.WithTimeout(context.Background(), timeoutDuration) ctx, cancel := context.WithTimeout(context.Background(), timeoutDuration)
@ -379,31 +425,38 @@ func GetProtocol(host string, Timeout int64) (protocol string) {
// 1. 根据标准端口快速判断协议 // 1. 根据标准端口快速判断协议
if strings.HasSuffix(host, ":80") { if strings.HasSuffix(host, ":80") {
Common.LogDebug("检测到标准HTTP端口，使用HTTP协议") Common.LogDebug("检测到标准HTTP端口，使用HTTP协议")
return return "http"
} else if strings.HasSuffix(host, ":443") { } else if strings.HasSuffix(host, ":443") {
Common.LogDebug("检测到标准HTTPS端口，使用HTTPS协议") Common.LogDebug("检测到标准HTTPS端口，使用HTTPS协议")
return "https" return "https"
} }
// 2. 并发检测HTTP和HTTPS // 2. 并发检测HTTP和HTTPS
resultChan := make(chan string, 2) type protocolResult struct {
name string
success bool
}
resultChan := make(chan protocolResult, 2)
singleTimeout := timeoutDuration / 2 // 每个协议检测的超时时间减半
// 检测HTTPS // 检测HTTPS
go func() { go func() {
Common.LogDebug("开始检测HTTPS协议")
tlsConfig := &tls.Config{ tlsConfig := &tls.Config{
InsecureSkipVerify: true, InsecureSkipVerify: true,
MinVersion: tls.VersionTLS10, MinVersion: tls.VersionTLS10,
} }
dialer := &net.Dialer{ dialer := &net.Dialer{
Timeout: timeoutDuration / 2, // 缩短单次尝试的超时时间 Timeout: singleTimeout,
} }
conn, err := tls.DialWithDialer(dialer, "tcp", host, tlsConfig) conn, err := tls.DialWithDialer(dialer, "tcp", host, tlsConfig)
if err == nil { if err == nil {
Common.LogDebug("HTTPS连接成功") Common.LogDebug("HTTPS连接成功")
conn.Close() conn.Close()
resultChan <- "https" resultChan <- protocolResult{"https", true}
return return
} }
@ -417,18 +470,21 @@ func GetProtocol(host string, Timeout int64) (protocol string) {
strings.Contains(errMsg, "x509") || strings.Contains(errMsg, "x509") ||
strings.Contains(errMsg, "secure") { strings.Contains(errMsg, "secure") {
Common.LogDebug(fmt.Sprintf("TLS握手有错误但可能是HTTPS协议: %v", err)) Common.LogDebug(fmt.Sprintf("TLS握手有错误但可能是HTTPS协议: %v", err))
resultChan <- "https" resultChan <- protocolResult{"https", true}
return return
} }
Common.LogDebug(fmt.Sprintf("HTTPS连接失败: %v", err))
} }
resultChan <- "" resultChan <- protocolResult{"https", false}
}() }()
// 检测HTTP // 检测HTTP
go func() { go func() {
Common.LogDebug("开始检测HTTP协议")
req, err := http.NewRequestWithContext(ctx, "HEAD", fmt.Sprintf("http://%s", host), nil) req, err := http.NewRequestWithContext(ctx, "HEAD", fmt.Sprintf("http://%s", host), nil)
if err != nil { if err != nil {
resultChan <- "" Common.LogDebug(fmt.Sprintf("创建HTTP请求失败: %v", err))
resultChan <- protocolResult{"http", false}
return return
} }
@ -436,61 +492,68 @@ func GetProtocol(host string, Timeout int64) (protocol string) {
Transport: &http.Transport{ Transport: &http.Transport{
TLSClientConfig: &tls.Config{InsecureSkipVerify: true}, TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
DialContext: (&net.Dialer{ DialContext: (&net.Dialer{
Timeout: timeoutDuration / 2, Timeout: singleTimeout,
}).DialContext, }).DialContext,
}, },
CheckRedirect: func(req *http.Request, via []*http.Request) error { CheckRedirect: func(req *http.Request, via []*http.Request) error {
return http.ErrUseLastResponse // 不跟随重定向 return http.ErrUseLastResponse // 不跟随重定向
}, },
Timeout: timeoutDuration / 2, Timeout: singleTimeout,
} }
resp, err := client.Do(req) resp, err := client.Do(req)
if err == nil { if err == nil {
resp.Body.Close() resp.Body.Close()
Common.LogDebug(fmt.Sprintf("HTTP连接成功，状态码: %d", resp.StatusCode)) Common.LogDebug(fmt.Sprintf("HTTP连接成功，状态码: %d", resp.StatusCode))
resultChan <- "http" resultChan <- protocolResult{"http", true}
return return
} }
Common.LogDebug(fmt.Sprintf("标准HTTP请求失败: %v，尝试原始TCP连接", err))
// 尝试原始TCP连接和简单HTTP请求 // 尝试原始TCP连接和简单HTTP请求
netConn, err := net.DialTimeout("tcp", host, timeoutDuration/2) netConn, err := net.DialTimeout("tcp", host, singleTimeout)
if err == nil { if err == nil {
defer netConn.Close() defer netConn.Close()
netConn.SetDeadline(time.Now().Add(timeoutDuration / 2)) netConn.SetDeadline(time.Now().Add(singleTimeout))
// 发送简单HTTP请求 // 发送简单HTTP请求
_, err = netConn.Write([]byte("HEAD / HTTP/1.0\r\nHost: " + host + "\r\n\r\n")) _, err = netConn.Write([]byte("HEAD / HTTP/1.0\r\nHost: " + host + "\r\n\r\n"))
if err == nil { if err == nil {
// 读取响应 // 读取响应
buf := make([]byte, 1024) buf := make([]byte, 1024)
netConn.SetDeadline(time.Now().Add(timeoutDuration / 2)) netConn.SetDeadline(time.Now().Add(singleTimeout))
n, err := netConn.Read(buf) n, err := netConn.Read(buf)
if err == nil && n > 0 { if err == nil && n > 0 {
response := string(buf[:n]) response := string(buf[:n])
if strings.Contains(response, "HTTP/") { if strings.Contains(response, "HTTP/") {
Common.LogDebug("通过原始TCP连接确认HTTP协议") Common.LogDebug("通过原始TCP连接确认HTTP协议")
resultChan <- "http" resultChan <- protocolResult{"http", true}
return return
} }
} }
} }
Common.LogDebug("原始TCP连接成功但HTTP响应无效")
} else {
Common.LogDebug(fmt.Sprintf("原始TCP连接失败: %v", err))
} }
resultChan <- "" resultChan <- protocolResult{"http", false}
}() }()
// 3. 收集结果并决定使用哪种协议 // 3. 收集结果并决定使用哪种协议
var httpsResult, httpResult string var httpsSuccess, httpSuccess bool
// 等待两个goroutine返回结果或超时 // 等待两个goroutine返回结果或超时
for i := 0; i < 2; i++ { for i := 0; i < 2; i++ {
select { select {
case result := <-resultChan: case result := <-resultChan:
if result == "https" { if result.name == "https" {
httpsResult = result httpsSuccess = result.success
} else if result == "http" { Common.LogDebug(fmt.Sprintf("HTTPS检测结果: %v", httpsSuccess))
httpResult = result } else if result.name == "http" {
httpSuccess = result.success
Common.LogDebug(fmt.Sprintf("HTTP检测结果: %v", httpSuccess))
} }
case <-ctx.Done(): case <-ctx.Done():
Common.LogDebug("协议检测超时") Common.LogDebug("协议检测超时")
@ -498,12 +561,12 @@ func GetProtocol(host string, Timeout int64) (protocol string) {
} }
} }
// 4. 决定使用哪种协议 // 4. 决定使用哪种协议 - 优先使用HTTPS，如果HTTPS不可用则使用HTTP
if httpsResult == "https" { if httpsSuccess {
Common.LogDebug("优先使用HTTPS协议") Common.LogDebug("选择使用HTTPS协议")
return "https" return "https"
} else if httpResult == "http" { } else if httpSuccess {
Common.LogDebug("使用HTTP协议") Common.LogDebug("选择使用HTTP协议")
return "http" return "http"
} }