fscan/common/config_struct.go

package common

import (
	"time"

	"github.com/shadow1ng/fscan/common/config"
)

/*
config_struct.go - 配置结构体定义

简化后的结构：
- 高频字段平铺到顶层
- 子配置使用值类型（非指针）
- 删除过度分类的 AdvancedConfig
*/

// =============================================================================
// Config - 扫描器配置
// =============================================================================

// Config 扫描器完整配置 - 初始化后只读，可安全共享
type Config struct {
	// 高频访问字段 - 平铺到顶层
	Timeout           time.Duration // 通用超时
	ThreadNum         int           // 主线程数
	ThreadNumExplicit bool          // 用户显式指定了 -t
	ModuleThreadNum   int           // 模块线程数
	DisableBrute      bool          // 禁用暴力破解
	DisablePing       bool          // 禁用Ping检测
	DisableTcpProbe   bool          // 禁用TCP补充探测

	// 扫描模式
	Mode        string // 扫描模式
	LocalMode   bool   // 本地模式
	LocalPlugin string // 本地插件名
	AliveOnly   bool   // 仅存活检测
	MaxRetries  int    // 最大重试次数

	// 高级功能（从AdvancedConfig合并）
	Shellcode             string           // Shellcode
	LocalPluginsList      []string         // 本地插件列表
	DNSLog                bool             // DNSLog检测
	PersistenceTargetFile string           // 持久化目标文件
	WinPEFile             string           // WinPE文件
	PortMap               map[int][]string // 端口映射
	DefaultMap            []string         // 默认映射

	// 分组配置 - 值类型
	Credentials  CredentialConfig
	Network      NetworkConfig
	Output       OutputConfig
	POC          POCConfig
	Redis        RedisConfig
	HTTP         HTTPConfig
	LocalExploit LocalExploitConfig
	Target       TargetConfig // 扫描目标配置

	// SOCKS5代理端口配置
	Socks5ProxyPort int // SOCKS5代理端口
}

// TargetConfig 扫描目标配置
type TargetConfig struct {
	Ports        string // 端口范围字符串
	ExcludePorts string // 排除端口字符串
}

// CredentialConfig 认证相关配置
type CredentialConfig struct {
	Username      string
	Password      string
	Domain        string
	Userdict      map[string][]string
	Passwords     []string
	UserPassPairs []config.CredentialPair
	HashValues    []string
	HashBytes     [][]byte
	SSHKeyPath    string
}

// NetworkConfig 网络相关配置
type NetworkConfig struct {
	HTTPProxy       string
	Socks5Proxy     string
	Iface           string
	WebTimeout      time.Duration
	MaxRedirects    int
	PacketRateLimit int64
	MaxPacketCount  int64
	ICMPRate        float64
}

// OutputConfig 输出相关配置
type OutputConfig struct {
	File            string
	Format          string
	DisableSave     bool
	NoColor         bool
	Silent          bool
	DisableProgress bool
	ShowProgress    bool
	LogLevel        string
	Language        string
	PerfStats       bool
}

// POCConfig POC扫描相关配置
type POCConfig struct {
	PocPath  string // POC路径
	PocName  string // 指定POC名称
	Full     bool   // 完整POC扫描
	Num      int    // POC并发数
	Disabled bool   // 禁用POC扫描
}

// RedisConfig Redis利用相关配置
type RedisConfig struct {
	Disabled     bool   // 禁用Redis利用
	File         string // SSH密钥文件
	Shell        string // 反弹Shell地址
	WritePath    string // 写入路径
	WriteContent string // 写入内容
	WriteFile    string // 本地文件路径
}

// HTTPConfig HTTP请求相关配置
type HTTPConfig struct {
	Cookie    string // Cookie
	UserAgent string // User-Agent
	Accept    string // Accept头
}

// LocalExploitConfig 本地利用相关配置
type LocalExploitConfig struct {
	ReverseShellTarget  string // 反弹Shell目标
	ForwardShellPort    int    // 正向Shell端口
	KeyloggerOutputFile string // 键盘记录输出文件
	DownloadURL         string // 下载URL
	DownloadSavePath    string // 下载保存路径
}

// NewConfig 创建带默认值的Config（后备用，正常流程使用BuildConfigFromFlags）
func NewConfig() *Config {
	return &Config{
		// 高频字段 - 使用默认常量
		Timeout:         time.Duration(DefaultTimeout) * time.Second,
		ThreadNum:       DefaultThreadNum,
		ModuleThreadNum: 10,
		DisableBrute:    false,
		DisablePing:     false,
		DisableTcpProbe: false,

		// 扫描模式
		Mode:       DefaultScanMode,
		LocalMode:  false,
		AliveOnly:  false,
		MaxRetries: 3,

		// 高级功能 - 使用默认配置
		PortMap:    config.DefaultPortMap,
		DefaultMap: config.DefaultProbeMap,

		// 分组配置 - 使用默认字典
		Credentials: CredentialConfig{
			Userdict:      config.DefaultUserDict,
			Passwords:     config.DefaultPasswords,
			UserPassPairs: nil,
		},
		Network: NetworkConfig{
			WebTimeout:   time.Duration(5) * time.Second,
			MaxRedirects: 10,
			ICMPRate:     0.1,
		},
		Output: OutputConfig{
			File:         "result.txt",
			Format:       "txt",
			ShowProgress: true,
			LogLevel:     DefaultLogLevel,
			Language:     DefaultLanguage,
		},
		POC: POCConfig{
			Num: 20,
		},
		LocalExploit: LocalExploitConfig{
			ForwardShellPort: 4444,
		},
	}
}