mirror of
https://github.com/shadow1ng/fscan.git
synced 2025-07-14 13:22:35 +08:00
85 lines
2.1 KiB
Go
85 lines
2.1 KiB
Go
package Plugins
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"regexp"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/shadow1ng/fscan/common"
|
|
)
|
|
|
|
func WebTitle(info *common.HostInfo, ch chan int, wg *sync.WaitGroup) (err error, result string) {
|
|
info.Url = fmt.Sprintf("http://%s:%s", info.Host, info.Ports)
|
|
err, result = geturl(info)
|
|
wg.Done()
|
|
<-ch
|
|
return err, result
|
|
}
|
|
|
|
func geturl(info *common.HostInfo) (err error, result string) {
|
|
url := info.Url
|
|
var client = &http.Client{Timeout: time.Duration(info.Timeout) * time.Second}
|
|
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.Add("Connection", "close")
|
|
resp, err := client.Do(res)
|
|
if err == nil {
|
|
defer resp.Body.Close()
|
|
var title string
|
|
body, _ := ioutil.ReadAll(resp.Body)
|
|
re := regexp.MustCompile("<title>(.*)</title>")
|
|
find := re.FindAllStringSubmatch(string(body), -1)
|
|
if len(find) > 0 {
|
|
title = find[0][1]
|
|
} else {
|
|
title = "None"
|
|
}
|
|
if len(title) > 50 {
|
|
title = title[:50]
|
|
}
|
|
if resp.StatusCode == 400 && string(url[5]) != "https" {
|
|
info.Url = strings.Replace(url, "http://", "https://", 1)
|
|
return geturl(info)
|
|
} else {
|
|
result = fmt.Sprintf("WebTitle:%v %v %v", url, resp.StatusCode, title)
|
|
common.LogSuccess(result)
|
|
}
|
|
return err, result
|
|
}
|
|
}
|
|
return err, ""
|
|
}
|
|
|
|
//var client = &http.Client{
|
|
// Transport:&http.Transport{
|
|
// DialContext:(&net.Dialer{
|
|
// Timeout:time.Duration(info.Timeout)*time.Second,
|
|
// }).DialContext,
|
|
// },
|
|
// CheckRedirect:func(req *http.Request, via []*http.Request) error{
|
|
// return http.ErrUseLastResponse
|
|
// },
|
|
//}
|
|
|
|
//if info.Cookie!=""{
|
|
// res.Header.Add("Cookie",info.Cookie)
|
|
//}
|
|
//if info.Header!=""{
|
|
// var header = make(map[string]string)
|
|
// err:=json.Unmarshal([]byte(info.Header),&header)
|
|
// if err!=nil{
|
|
// Misc.CheckErr(err)
|
|
// }
|
|
// for k,v:=range header{
|
|
// res.Header.Add(k,v)
|
|
// }
|
|
//}
|