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

减少http client初始化次数

Browse Source
This commit is contained in:
shadow1ng 2021-02-21 14:54:40 +08:00
parent 583e51d479
commit 51b8b2c0e2
1 changed files with 19 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
Plugins/webtitle.go
View File

@ -1,21 +1,19 @@
package Plugins
import (
"crypto/tls"
"fmt"
"github.com/shadow1ng/fscan/WebScan"
"github.com/shadow1ng/fscan/WebScan/lib"
"github.com/shadow1ng/fscan/common"
"io/ioutil"
"net"
"net/http"
"regexp"
"strings"
"time"
)
var CheckData []WebScan.CheckDatas
func WebTitle(info *common.HostInfo) error {
var CheckData []WebScan.CheckDatas
if info.Ports == "80" {
info.Url = fmt.Sprintf("http://%s", info.Host)
} else if info.Ports == "443" {
@ -24,18 +22,18 @@ func WebTitle(info *common.HostInfo) error {
info.Url = fmt.Sprintf("http://%s:%s", info.Host, info.Ports)
}
err, result := geturl(info, true)
err, result, CheckData := geturl(info, true, CheckData)
if err != nil {
return err
}
if result == "https" {
err, _ := geturl(info, true)
err, _, CheckData = geturl(info, true, CheckData)
if err != nil {
return err
}
}
err, _ = geturl(info, false)
err, _, CheckData = geturl(info, false, CheckData)
if err != nil {
return err
}
@ -49,41 +47,22 @@ func WebTitle(info *common.HostInfo) error {
return err
}
func geturl(info *common.HostInfo, flag bool) (err error, result string) {
func geturl(info *common.HostInfo, flag bool, CheckData []WebScan.CheckDatas) (error, string, []WebScan.CheckDatas) {
Url := info.Url
if flag == false {
Url += "/favicon.ico"
}
tr := &http.Transport{
TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
DisableKeepAlives: false,
DialContext: (&net.Dialer{
Timeout: time.Duration(info.WebTimeout) * time.Second,
KeepAlive: time.Duration(info.WebTimeout+3) * time.Second,
}).DialContext,
MaxIdleConns: 1000,
MaxIdleConnsPerHost: 1000,
IdleConnTimeout: time.Duration(info.WebTimeout+3) * time.Second,
TLSHandshakeTimeout: 5 * time.Second,
}
//u, err := url.Parse("http://127.0.0.1:8080")
//if err != nil {
// return err,result
//}
//tr.Proxy = http.ProxyURL(u)
var client = &http.Client{Timeout: time.Duration(info.WebTimeout) * time.Second, Transport: tr}
res, err := http.NewRequest("GET", Url, nil)
if err == nil {
res.Header.Add("User-agent", "Mozilla/5.0 (Windows NT 6.1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/28.0.1468.0 Safari/537.36")
res.Header.Add("Accept", "*/*")
res.Header.Add("Accept-Language", "zh-CN,zh;q=0.9")
res.Header.Add("Accept-Encoding", "gzip, deflate")
res.Header.Set("User-agent", "Mozilla/5.0 (Windows NT 6.1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/28.0.1468.0 Safari/537.36")
res.Header.Set("Accept", "*/*")
res.Header.Set("Accept-Language", "zh-CN,zh;q=0.9")
res.Header.Set("Accept-Encoding", "gzip, deflate")
if flag == true {
res.Header.Add("Cookie", "rememberMe=1")
res.Header.Set("Cookie", "rememberMe=1")
}
res.Header.Add("Connection", "close")
resp, err := client.Do(res)
res.Header.Set("Connection", "close")
resp, err := lib.Client.Do(res)
if err == nil {
defer resp.Body.Close()
var title string
@ -99,7 +78,7 @@ func geturl(info *common.HostInfo, flag bool) (err error, result string) {
title = "None"
}
if flag == true {
result = fmt.Sprintf("WebTitle:%-25v %-3v %v", Url, resp.StatusCode, title)
result := fmt.Sprintf("WebTitle:%-25v %-3v %v", Url, resp.StatusCode, title)
common.LogSuccess(result)
}
@ -107,11 +86,11 @@ func geturl(info *common.HostInfo, flag bool) (err error, result string) {
if resp.StatusCode == 400 && info.Url[:5] != "https" {
info.Url = strings.Replace(info.Url, "http://", "https://", 1)
return err, "https"
return err, "https", CheckData
}
return err, result
return err, "", CheckData
}
return err, ""
return err, "", CheckData
}
return err, ""
return err, "", CheckData
}