mirror of https://github.com/qwqdanchun/fscan.git
83 lines
1.7 KiB
Go
83 lines
1.7 KiB
Go
package common
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
var Num int64
|
|
var End int64
|
|
var Results = make(chan string)
|
|
var Start = true
|
|
var LogSucTime int64
|
|
var LogErrTime int64
|
|
var WaitTime int64
|
|
var LogWG sync.WaitGroup
|
|
|
|
func LogSuccess(result string) {
|
|
LogWG.Add(1)
|
|
LogSucTime = time.Now().Unix()
|
|
if Start {
|
|
go SaveLog()
|
|
Start = false
|
|
}
|
|
Results <- result
|
|
}
|
|
|
|
func SaveLog() {
|
|
for result := range Results {
|
|
fmt.Println(result)
|
|
if IsSave {
|
|
WriteFile(result, Outputfile)
|
|
}
|
|
LogWG.Done()
|
|
}
|
|
}
|
|
|
|
func WriteFile(result string, filename string) {
|
|
var text = []byte(result + "\n")
|
|
fl, err := os.OpenFile(filename, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0777)
|
|
if err != nil {
|
|
fmt.Printf("Open %s error, %v\n", filename, err)
|
|
return
|
|
}
|
|
_, err = fl.Write(text)
|
|
fl.Close()
|
|
if err != nil {
|
|
fmt.Printf("Write %s error, %v\n", filename, err)
|
|
}
|
|
}
|
|
|
|
func LogError(errinfo interface{}) {
|
|
if WaitTime == 0 {
|
|
fmt.Println(fmt.Sprintf("已完成 %v/%v %v", End, Num, errinfo))
|
|
} else if (time.Now().Unix()-LogSucTime) > WaitTime && (time.Now().Unix()-LogErrTime) > WaitTime {
|
|
fmt.Println(fmt.Sprintf("已完成 %v/%v %v", End, Num, errinfo))
|
|
LogErrTime = time.Now().Unix()
|
|
}
|
|
}
|
|
|
|
func CheckErrs(err error) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
errs := []string{
|
|
"closed by the remote host", "too many connections",
|
|
"i/o timeout", "EOF", "A connection attempt failed",
|
|
"established connection failed", "connection attempt failed",
|
|
"Unable to read", "is not allowed to connect to this",
|
|
"no pg_hba.conf entry",
|
|
"no supported methods remain",
|
|
"No connection could be made",
|
|
}
|
|
for _, key := range errs {
|
|
if strings.Contains(strings.ToLower(err.Error()), strings.ToLower(key)) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|