From cae98e7d90e102dd9307c40e076f8cf0901e1c84 Mon Sep 17 00:00:00 2001
From: ZacharyZcR <2903735704@qq.com>
Date: Wed, 18 Dec 2024 21:55:39 +0800
Subject: [PATCH] =?UTF-8?q?refactor:=20=E9=87=8D=E6=9E=84=E6=98=A0?=
 =?UTF-8?q?=E5=B0=84?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Config/Types.go              |  24 +++++++
 Plugins/{base.go => Base.go} |  23 -------
 Plugins/Registry.go          | 127 +++++++++++++++++++++++++++++++++++
 3 files changed, 151 insertions(+), 23 deletions(-)
 create mode 100644 Config/Types.go
 rename Plugins/{base.go => Base.go} (80%)
 create mode 100644 Plugins/Registry.go

diff --git a/Config/Types.go b/Config/Types.go
new file mode 100644
index 0000000..aa6a420
--- /dev/null
+++ b/Config/Types.go
@@ -0,0 +1,24 @@
+// Config/types.go
+package Config
+
+type HostInfo struct {
+	Host    string
+	Ports   string
+	Url     string
+	Infostr []string
+}
+
+// ScanPlugin 定义扫描插件的结构
+type ScanPlugin struct {
+	Name     string                // 插件名称
+	Port     int                   // 关联的端口号，0表示特殊扫描类型
+	ScanFunc func(*HostInfo) error // 扫描函数
+}
+
+// PluginManager 管理插件注册
+var PluginManager = make(map[string]ScanPlugin)
+
+// RegisterPlugin 注册插件
+func RegisterPlugin(name string, plugin ScanPlugin) {
+	PluginManager[name] = plugin
+}
diff --git a/Plugins/base.go b/Plugins/Base.go
similarity index 80%
rename from Plugins/base.go
rename to Plugins/Base.go
index b33bc8c..7fc38a9 100644
--- a/Plugins/base.go
+++ b/Plugins/Base.go
@@ -8,29 +8,6 @@ import (
 	"net"
 )
 
-var PluginList = map[string]interface{}{
-	"21":      FtpScan,
-	"22":      SshScan,
-	"135":     Findnet,
-	"139":     NetBIOS,
-	"445":     SmbScan,
-	"1433":    MssqlScan,
-	"1521":    OracleScan,
-	"3306":    MysqlScan,
-	"3389":    RdpScan,
-	"5432":    PostgresScan,
-	"6379":    RedisScan,
-	"9000":    FcgiScan,
-	"11211":   MemcachedScan,
-	"27017":   MongodbScan,
-	"1000001": MS17010,
-	"1000002": SmbGhost,
-	"1000003": WebTitle,
-	"1000004": SmbScan2,
-	"1000005": WmiExec,
-	"1000006": LocalInfoScan,
-}
-
 func ReadBytes(conn net.Conn) (result []byte, err error) {
 	size := 4096
 	buf := make([]byte, size)
diff --git a/Plugins/Registry.go b/Plugins/Registry.go
new file mode 100644
index 0000000..d23d061
--- /dev/null
+++ b/Plugins/Registry.go
@@ -0,0 +1,127 @@
+package Plugins
+
+import "github.com/shadow1ng/fscan/Config"
+
+func init() {
+	// 注册标准端口服务扫描
+	Config.RegisterPlugin("ftp", Config.ScanPlugin{
+		Name:     "FTP",
+		Port:     21,
+		ScanFunc: FtpScan,
+	})
+
+	Config.RegisterPlugin("ssh", Config.ScanPlugin{
+		Name:     "SSH",
+		Port:     22,
+		ScanFunc: SshScan,
+	})
+
+	Config.RegisterPlugin("findnet", Config.ScanPlugin{
+		Name:     "FindNet",
+		Port:     135,
+		ScanFunc: Findnet,
+	})
+
+	Config.RegisterPlugin("netbios", Config.ScanPlugin{
+		Name:     "NetBIOS",
+		Port:     139,
+		ScanFunc: NetBIOS,
+	})
+
+	Config.RegisterPlugin("smb", Config.ScanPlugin{
+		Name:     "SMB",
+		Port:     445,
+		ScanFunc: SmbScan,
+	})
+
+	Config.RegisterPlugin("mssql", Config.ScanPlugin{
+		Name:     "MSSQL",
+		Port:     1433,
+		ScanFunc: MssqlScan,
+	})
+
+	Config.RegisterPlugin("oracle", Config.ScanPlugin{
+		Name:     "Oracle",
+		Port:     1521,
+		ScanFunc: OracleScan,
+	})
+
+	Config.RegisterPlugin("mysql", Config.ScanPlugin{
+		Name:     "MySQL",
+		Port:     3306,
+		ScanFunc: MysqlScan,
+	})
+
+	Config.RegisterPlugin("rdp", Config.ScanPlugin{
+		Name:     "RDP",
+		Port:     3389,
+		ScanFunc: RdpScan,
+	})
+
+	Config.RegisterPlugin("postgres", Config.ScanPlugin{
+		Name:     "PostgreSQL",
+		Port:     5432,
+		ScanFunc: PostgresScan,
+	})
+
+	Config.RegisterPlugin("redis", Config.ScanPlugin{
+		Name:     "Redis",
+		Port:     6379,
+		ScanFunc: RedisScan,
+	})
+
+	Config.RegisterPlugin("fcgi", Config.ScanPlugin{
+		Name:     "FastCGI",
+		Port:     9000,
+		ScanFunc: FcgiScan,
+	})
+
+	Config.RegisterPlugin("memcached", Config.ScanPlugin{
+		Name:     "Memcached",
+		Port:     11211,
+		ScanFunc: MemcachedScan,
+	})
+
+	Config.RegisterPlugin("mongodb", Config.ScanPlugin{
+		Name:     "MongoDB",
+		Port:     27017,
+		ScanFunc: MongodbScan,
+	})
+
+	// 注册特殊扫描类型
+	Config.RegisterPlugin("ms17010", Config.ScanPlugin{
+		Name:     "MS17010",
+		Port:     445,
+		ScanFunc: MS17010,
+	})
+
+	Config.RegisterPlugin("smbghost", Config.ScanPlugin{
+		Name:     "SMBGhost",
+		Port:     445,
+		ScanFunc: SmbGhost,
+	})
+
+	Config.RegisterPlugin("web", Config.ScanPlugin{
+		Name:     "WebTitle",
+		Port:     0,
+		ScanFunc: WebTitle,
+	})
+
+	Config.RegisterPlugin("smb2", Config.ScanPlugin{
+		Name:     "SMBScan2",
+		Port:     445,
+		ScanFunc: SmbScan2,
+	})
+
+	Config.RegisterPlugin("wmiexec", Config.ScanPlugin{
+		Name:     "WMIExec",
+		Port:     135,
+		ScanFunc: WmiExec,
+	})
+
+	Config.RegisterPlugin("localinfo", Config.ScanPlugin{
+		Name:     "LocalInfo",
+		Port:     0,
+		ScanFunc: LocalInfoScan,
+	})
+}