fscan/Common/common.go

package common

/*
common.go - 简化的统一入口

移除所有向后兼容层，提供清晰的模块化接口。
直接导出各子模块的核心功能，避免代码债务。
*/

import (
	"context"
	"crypto/tls"
	"fmt"
	"net"
	"sync"
	"time"

	"github.com/shadow1ng/fscan/common/core"
	"github.com/shadow1ng/fscan/common/logging"
	"github.com/shadow1ng/fscan/common/output"
)

// =============================================================================
// 核心类型导出 - 直接从core模块导出
// =============================================================================

type HostInfo = core.HostInfo
type ScanPlugin = core.ScanPlugin

// 插件类型常量
const (
	PluginTypeService = core.PluginTypeService
	PluginTypeWeb     = core.PluginTypeWeb
	PluginTypeLocal   = core.PluginTypeLocal
	PluginTypeBrute   = core.PluginTypeBrute
	PluginTypePoc     = core.PluginTypePoc
	PluginTypeScan    = core.PluginTypeScan
)

// 全局插件管理器
var PluginManager = core.LegacyPluginManager

// =============================================================================
// 核心功能导出 - 直接调用对应模块
// =============================================================================

// 插件系统
func RegisterPlugin(name string, plugin ScanPlugin) {
	if err := core.RegisterPlugin(name, plugin); err != nil {
		LogError("Failed to register plugin " + name + ": " + err.Error())
	}
}

func GetGlobalPluginManager() *core.PluginManager {
	return core.GetGlobalPluginManager()
}

// =============================================================================
// 日志系统简化接口
// =============================================================================

var globalLogger *logging.Logger
var loggerMutex sync.Mutex

func getGlobalLogger() *logging.Logger {
	loggerMutex.Lock()
	defer loggerMutex.Unlock()
	
	if globalLogger == nil {
		level := getLogLevelFromString(LogLevel)
		config := &logging.LoggerConfig{
			Level: level, 
			EnableColor: !NoColor, 
			SlowOutput: SlowLogOutput,
			ShowProgress: true, 
			StartTime: StartTime,
		}
		globalLogger = logging.NewLogger(config)
		if ProgressBar != nil {
			globalLogger.SetProgressBar(ProgressBar)
		}
		globalLogger.SetOutputMutex(&OutputMutex)
		
		// 设置协调输出函数，使用LogWithProgress
		globalLogger.SetCoordinatedOutput(LogWithProgress)
	}
	return globalLogger
}

func getLogLevelFromString(levelStr string) logging.LogLevel {
	switch levelStr {
	case "all", "ALL":
		return logging.LevelAll
	case "error", "ERROR":
		return logging.LevelError
	case "base", "BASE":
		return logging.LevelBase
	case "info", "INFO":
		return logging.LevelInfo
	case "success", "SUCCESS":
		return logging.LevelSuccess
	case "debug", "DEBUG":
		return logging.LevelDebug
	case "info,success":
		return logging.LevelInfoSuccess
	case "base,info,success", "BASE_INFO_SUCCESS":
		return logging.LevelBaseInfoSuccess
	default:
		return logging.LevelInfoSuccess
	}
}

// 日志函数
func InitLogger() { 
	loggerMutex.Lock()
	globalLogger = nil
	loggerMutex.Unlock()
	getGlobalLogger().Initialize() 
}

func LogDebug(msg string)       { getGlobalLogger().Debug(msg) }
func LogBase(msg string)        { getGlobalLogger().Base(msg) }
func LogInfo(msg string)        { getGlobalLogger().Info(msg) }
func LogSuccess(result string)  { getGlobalLogger().Success(result) }
func LogError(errMsg string)    { getGlobalLogger().Error(errMsg) }

// =============================================================================
// 输出系统简化接口
// =============================================================================

var ResultOutput *output.Manager

func InitOutput() error {
	if Outputfile == "" {
		return fmt.Errorf("output file not specified")
	}
	
	var format output.OutputFormat
	switch OutputFormat {
	case "txt":
		format = output.FormatTXT
	case "json":
		format = output.FormatJSON
	case "csv":
		format = output.FormatCSV
	default:
		return fmt.Errorf("invalid output format: %s", OutputFormat)
	}

	config := output.DefaultManagerConfig(Outputfile, format)
	manager, err := output.NewManager(config)
	if err != nil {
		return err
	}
	ResultOutput = manager
	output.SetGlobalManager(manager)
	return nil
}

func CloseOutput() error {
	if ResultOutput == nil {
		return nil
	}
	return ResultOutput.Close()
}

func SaveResult(result *output.ScanResult) error {
	if ResultOutput == nil {
		return fmt.Errorf("output not initialized")
	}
	return ResultOutput.SaveResult(result)
}

// =============================================================================
// 网络连接辅助函数
// =============================================================================

// WrapperTcpWithTimeout TCP连接包装器，带超时
func WrapperTcpWithTimeout(network, address string, timeout time.Duration) (net.Conn, error) {
	return net.DialTimeout(network, address, timeout)
}

// WrapperTcpWithContext TCP连接包装器，带上下文
func WrapperTcpWithContext(ctx context.Context, network, address string) (net.Conn, error) {
	var d net.Dialer
	return d.DialContext(ctx, network, address)
}

// WrapperTlsWithContext TLS连接包装器，带上下文
func WrapperTlsWithContext(ctx context.Context, network, address string, config *tls.Config) (net.Conn, error) {
	d := &tls.Dialer{Config: config}
	return d.DialContext(ctx, network, address)
}

// =============================================================================
// 错误处理辅助函数
// =============================================================================

// CheckErrs 检查单个错误 - 简化版本
func CheckErrs(err error) error {
	return err
}