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

Update log.go

Browse Source
This commit is contained in:
影舞者 2023-11-13 11:41:26 +08:00 committed by GitHub
parent 5d154ce6a1
commit 989389fd52
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 20 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
common/log.go
View File

@ -2,12 +2,11 @@ package common
import (
"fmt"
"github.com/fatih/color"
"os"
"strings"
"sync"
"time"
"github.com/fatih/color"
)
var Num int64
@ -18,8 +17,8 @@ var LogSucTime int64
var LogErrTime int64
var WaitTime int64
var Silent bool
var Nocolor bool
var LogWG sync.WaitGroup
var Outputfile string
func init() {
LogSucTime = time.Now().Unix()
@ -35,17 +34,21 @@ func LogSuccess(result string) {
func SaveLog() {
for result := range Results {
if !Silent {
if strings.Contains(*result, "[+]") {
color.Green(*result)
} else if strings.Contains(*result, "[*]") {
color.Cyan(*result)
if Nocolor {
fmt.Println(*result)
} else {
if strings.HasPrefix(*result, "[+] InfoScan") {
color.Green(*result)
} else if strings.HasPrefix(*result, "[+]") {
color.Red(*result)
} else {
fmt.Println(*result)
}
}
}
if IsSave {
WriteFile(*result, Outputfile)
}
LogWG.Done()
}
}
@ -57,23 +60,19 @@ func WriteFile(result string, filename string) {
fmt.Printf("Open %s error, %v\n", filename, err)
return
}
defer func() {
_ = fl.Close()
}()
if _, err := fl.Write(text); err != nil {
_, 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 || (time.Now().Unix()-LogSucTime) > WaitTime && (time.Now().Unix()-LogErrTime) > WaitTime {
color.Red(fmt.Sprintf("Completed %v/%v %v \n", End, Num, errinfo))
if WaitTime != 0 {
LogErrTime = time.Now().Unix()
}
if WaitTime == 0 {
fmt.Printf("已完成 %v/%v %v \n", End, Num, errinfo)
} else if (time.Now().Unix()-LogSucTime) > WaitTime && (time.Now().Unix()-LogErrTime) > WaitTime {
fmt.Printf("已完成 %v/%v %v \n", End, Num, errinfo)
LogErrTime = time.Now().Unix()
}
}