mirror of
https://github.com/shadow1ng/fscan.git
synced 2026-09-24 12:11:52 +08:00
refactor: 输出格式重构，重构SMB、SMB2、FTP的一些验证逻辑
This commit is contained in:
ZacharyZcR
+209
-98
@@ -1,139 +1,250 @@
|
||||
package Common
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"github.com/fatih/color"
|
||||
"io"
|
||||
"log"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"runtime"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/fatih/color"
|
||||
)
|
||||
|
||||
// 记录扫描状态的全局变量
|
||||
var (
|
||||
Num int64 // 总任务数
|
||||
End int64 // 已完成数
|
||||
Results = make(chan *string) // 结果通道
|
||||
LogSucTime int64 // 最近成功日志时间
|
||||
LogErrTime int64 // 最近错误日志时间
|
||||
WaitTime int64 // 等待时间
|
||||
Silent bool // 静默模式
|
||||
Nocolor bool // 禁用颜色
|
||||
JsonOutput bool // JSON输出
|
||||
LogWG sync.WaitGroup // 日志同步等待组
|
||||
// 全局变量
|
||||
status = &ScanStatus{lastSuccess: time.Now(), lastError: time.Now()}
|
||||
results = make(chan *LogEntry, 1000) // 使用缓冲通道
|
||||
logWG sync.WaitGroup
|
||||
|
||||
// 扫描计数
|
||||
Num int64 // 总任务数
|
||||
End int64 // 已完成任务数
|
||||
)
|
||||
|
||||
// JsonText JSON输出的结构体
|
||||
type JsonText struct {
|
||||
Type string `json:"type"` // 消息类型
|
||||
Text string `json:"text"` // 消息内容
|
||||
// 将 results 改名为 Results 使其可导出
|
||||
var (
|
||||
Results = results // 使 results 可导出
|
||||
LogWG = logWG // 使 logWG 可导出
|
||||
)
|
||||
|
||||
// ScanStatus 记录扫描状态
|
||||
type ScanStatus struct {
|
||||
mu sync.RWMutex
|
||||
total int64
|
||||
completed int64
|
||||
lastSuccess time.Time
|
||||
lastError time.Time
|
||||
}
|
||||
|
||||
// LogEntry 日志条目
|
||||
type LogEntry struct {
|
||||
Level string // "ERROR", "INFO", "SUCCESS", "DEBUG"
|
||||
Time time.Time
|
||||
Content string
|
||||
}
|
||||
|
||||
// LogLevel 定义日志等级常量
|
||||
const (
|
||||
LogLevelAll = "ALL" // 输出所有日志
|
||||
LogLevelError = "ERROR" // 错误日志
|
||||
LogLevelInfo = "INFO" // 信息日志
|
||||
LogLevelSuccess = "SUCCESS" // 成功日志
|
||||
LogLevelDebug = "DEBUG" // 调试日志
|
||||
)
|
||||
|
||||
// 定义日志颜色映射
|
||||
var logColors = map[string]color.Attribute{
|
||||
LogLevelError: color.FgRed,
|
||||
LogLevelInfo: color.FgYellow,
|
||||
LogLevelSuccess: color.FgGreen,
|
||||
LogLevelDebug: color.FgBlue,
|
||||
}
|
||||
|
||||
// bufferedFileWriter 文件写入器
|
||||
type bufferedFileWriter struct {
|
||||
file *os.File
|
||||
writer *bufio.Writer
|
||||
jsonEnc *json.Encoder
|
||||
}
|
||||
|
||||
// init 初始化日志配置
|
||||
func init() {
|
||||
log.SetOutput(io.Discard)
|
||||
LogSucTime = time.Now().Unix()
|
||||
go SaveLog()
|
||||
go processLogs()
|
||||
}
|
||||
|
||||
// formatLogMessage 格式化日志消息
|
||||
func formatLogMessage(entry *LogEntry) string {
|
||||
timeStr := entry.Time.Format("2006-01-02 15:04:05")
|
||||
return fmt.Sprintf("[%s] [%s] %s", timeStr, entry.Level, entry.Content)
|
||||
}
|
||||
|
||||
func printLog(entry *LogEntry) {
|
||||
// 根据配置的日志级别过滤
|
||||
if LogLevel != LogLevelAll && entry.Level != LogLevel {
|
||||
return
|
||||
}
|
||||
|
||||
logMsg := formatLogMessage(entry)
|
||||
if NoColor {
|
||||
fmt.Println(logMsg)
|
||||
return
|
||||
}
|
||||
|
||||
if colorAttr, ok := logColors[entry.Level]; ok {
|
||||
color.New(colorAttr).Println(logMsg)
|
||||
} else {
|
||||
fmt.Println(logMsg)
|
||||
}
|
||||
}
|
||||
|
||||
// 同样修改 LogError 和 LogInfo
|
||||
func LogError(errMsg string) {
|
||||
// 获取调用者信息
|
||||
_, file, line, ok := runtime.Caller(1)
|
||||
if !ok {
|
||||
file = "unknown"
|
||||
line = 0
|
||||
}
|
||||
// 只获取文件名
|
||||
file = filepath.Base(file)
|
||||
|
||||
// 格式化错误消息
|
||||
errorMsg := fmt.Sprintf("%s:%d - %s", file, line, errMsg)
|
||||
|
||||
select {
|
||||
case Results <- &LogEntry{
|
||||
Level: LogLevelError,
|
||||
Time: time.Now(),
|
||||
Content: errorMsg,
|
||||
}:
|
||||
logWG.Add(1)
|
||||
default:
|
||||
printLog(&LogEntry{
|
||||
Level: LogLevelError,
|
||||
Time: time.Now(),
|
||||
Content: errorMsg,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func LogInfo(msg string) {
|
||||
select {
|
||||
case Results <- &LogEntry{
|
||||
Level: LogLevelInfo,
|
||||
Time: time.Now(),
|
||||
Content: msg,
|
||||
}:
|
||||
logWG.Add(1)
|
||||
default:
|
||||
printLog(&LogEntry{
|
||||
Level: LogLevelInfo,
|
||||
Time: time.Now(),
|
||||
Content: msg,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
// LogSuccess 记录成功信息
|
||||
func LogSuccess(result string) {
|
||||
LogWG.Add(1)
|
||||
LogSucTime = time.Now().Unix()
|
||||
Results <- &result
|
||||
}
|
||||
|
||||
// SaveLog 保存日志信息
|
||||
func SaveLog() {
|
||||
for result := range Results {
|
||||
// 打印日志
|
||||
if !Silent {
|
||||
if Nocolor {
|
||||
fmt.Println(*result)
|
||||
} else {
|
||||
switch {
|
||||
case strings.HasPrefix(*result, "[+] 信息扫描"):
|
||||
color.Green(*result)
|
||||
case strings.HasPrefix(*result, "[+]"):
|
||||
color.Red(*result)
|
||||
default:
|
||||
fmt.Println(*result)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 保存到文件
|
||||
if IsSave {
|
||||
WriteFile(*result, Outputfile)
|
||||
}
|
||||
LogWG.Done()
|
||||
// 添加通道关闭检查
|
||||
select {
|
||||
case Results <- &LogEntry{
|
||||
Level: LogLevelSuccess,
|
||||
Time: time.Now(),
|
||||
Content: result,
|
||||
}:
|
||||
logWG.Add(1)
|
||||
status.mu.Lock()
|
||||
status.lastSuccess = time.Now()
|
||||
status.mu.Unlock()
|
||||
default:
|
||||
// 如果通道已关闭或已满，直接打印
|
||||
printLog(&LogEntry{
|
||||
Level: LogLevelSuccess,
|
||||
Time: time.Now(),
|
||||
Content: result,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
// WriteFile 写入文件
|
||||
func WriteFile(result string, filename string) {
|
||||
// 打开文件
|
||||
fl, err := os.OpenFile(filename, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0666)
|
||||
// JsonOutput JSON输出的结构体
|
||||
type JsonOutput struct {
|
||||
Level string `json:"level"`
|
||||
Timestamp time.Time `json:"timestamp"`
|
||||
Message string `json:"message"`
|
||||
}
|
||||
|
||||
// processLogs 处理日志信息
|
||||
func processLogs() {
|
||||
writer := newBufferedFileWriter()
|
||||
defer writer.close()
|
||||
|
||||
for entry := range results {
|
||||
if !Silent {
|
||||
printLog(entry)
|
||||
}
|
||||
|
||||
if writer != nil {
|
||||
writer.write(entry)
|
||||
}
|
||||
|
||||
logWG.Done()
|
||||
}
|
||||
}
|
||||
|
||||
func newBufferedFileWriter() *bufferedFileWriter {
|
||||
if DisableSave {
|
||||
return nil
|
||||
}
|
||||
|
||||
file, err := os.OpenFile(Outputfile, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0666)
|
||||
if err != nil {
|
||||
fmt.Printf("[-] 打开文件失败 %s: %v\n", filename, err)
|
||||
fmt.Printf("[ERROR] 打开输出文件失败 %s: %v\n", Outputfile, err)
|
||||
return nil
|
||||
}
|
||||
|
||||
writer := bufio.NewWriter(file)
|
||||
return &bufferedFileWriter{
|
||||
file: file,
|
||||
writer: writer,
|
||||
jsonEnc: json.NewEncoder(writer),
|
||||
}
|
||||
}
|
||||
|
||||
func (w *bufferedFileWriter) write(entry *LogEntry) {
|
||||
if w == nil {
|
||||
return
|
||||
}
|
||||
defer fl.Close()
|
||||
|
||||
if JsonOutput {
|
||||
// 解析JSON格式
|
||||
var scantype, text string
|
||||
if strings.HasPrefix(result, "[+]") || strings.HasPrefix(result, "[*]") || strings.HasPrefix(result, "[-]") {
|
||||
index := strings.Index(result[4:], " ")
|
||||
if index == -1 {
|
||||
scantype = "msg"
|
||||
text = result[4:]
|
||||
} else {
|
||||
scantype = result[4 : 4+index]
|
||||
text = result[4+index+1:]
|
||||
}
|
||||
} else {
|
||||
scantype = "msg"
|
||||
text = result
|
||||
if JsonFormat {
|
||||
output := JsonOutput{
|
||||
Level: entry.Level,
|
||||
Timestamp: entry.Time,
|
||||
Message: entry.Content,
|
||||
}
|
||||
|
||||
// 构造JSON对象
|
||||
jsonText := JsonText{
|
||||
Type: scantype,
|
||||
Text: text,
|
||||
if err := w.jsonEnc.Encode(output); err != nil {
|
||||
fmt.Printf("[ERROR] JSON编码失败: %v\n", err)
|
||||
}
|
||||
|
||||
// 序列化JSON
|
||||
jsonData, err := json.Marshal(jsonText)
|
||||
if err != nil {
|
||||
fmt.Printf("[-] JSON序列化失败: %v\n", err)
|
||||
jsonText = JsonText{
|
||||
Type: "msg",
|
||||
Text: result,
|
||||
}
|
||||
jsonData, _ = json.Marshal(jsonText)
|
||||
}
|
||||
jsonData = append(jsonData, []byte(",\n")...)
|
||||
_, err = fl.Write(jsonData)
|
||||
} else {
|
||||
_, err = fl.Write([]byte(result + "\n"))
|
||||
logMsg := formatLogMessage(entry) + "\n"
|
||||
if _, err := w.writer.WriteString(logMsg); err != nil {
|
||||
fmt.Printf("[ERROR] 写入文件失败: %v\n", err)
|
||||
}
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("[-] 写入文件失败 %s: %v\n", filename, err)
|
||||
}
|
||||
w.writer.Flush()
|
||||
}
|
||||
|
||||
// LogError 记录错误信息
|
||||
func LogError(errinfo interface{}) {
|
||||
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()
|
||||
func (w *bufferedFileWriter) close() {
|
||||
if w != nil {
|
||||
w.writer.Flush()
|
||||
w.file.Close()
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user