优化端口探测结果输出

Author: echo0d

src/develop/Go/ImprovingGo/2_fscanPro.md

@@ -1,6 +1,6 @@
 # 2. Go练习-fscan
 
-## 2.1. 存活主机顺序输出
+## 2.1. 优化存活主机探测结果输出
 
 Plugins目录下的icmp.go文件中CheckLive函数对应着探测存活主机功能，堆栈：
 
@@ -73,3 +73,56 @@ chanHosts := make(chan string, len(hostslist))
 
 这回应该没啥问题了，就是不知道效率被影响的程度，毕竟多解析了一遍IP，等用的时候再说吧
 
+## 2.2. 优化端口探测结果输出
+
+fscan 端口扫描函数调用堆栈
+
+```
+Plugins.PortScan (portscan.go:48) github.com/shadow1ng/fscan/Plugins
+Plugins.Scan (scanner.go:40) github.com/shadow1ng/fscan/Plugins
+main.main (main.go:15) main
+runtime.main (proc.go:250) runtime
+```
+
+![image-20240819095429072](img/2_fscanPro/image-20240819095429072.png)
+
+原本的探测结果在`PortConnect()`函数。一边扫描一边就输出，这样太乱了。但是也有对应的接收结果的 `results` channel, 那直接删除掉原本的输出，处理 `AliveAddress` 结果即可 修改代码如下
+
+![image-20240819095521796](img/2_fscanPro/image-20240819095521796.png)
+
+在`PortConnect()`函数中`AliveAddress` 返回之前，处理一遍结果。
+
+```go
+func PortScan(hostslist []string, ports string, timeout int64) []string {
+    //
+	// 中间省略
+    // 
+	wg.Wait()
+	close(Addrs)
+	close(results)
+	MapIPToPorts(AliveAddress)
+	return AliveAddress
+}
+
+func MapIPToPorts(ipPortList []string) {
+	ipToPorts := make(map[string][]int)
+	// 遍历列表，解析IP和端口，并将端口添加到对应IP的列表中
+	for _, ipPort := range ipPortList {
+		parts := strings.Split(ipPort, ":")
+		ip := parts[0]
+		port, _ := strconv.Atoi(parts[1])
+		ipToPorts[ip] = append(ipToPorts[ip], port)
+	}
+
+	for ip, ports := range ipToPorts {
+		sort.Ints(ports)
+		result := fmt.Sprintf(" %s: %v", ip, ports)
+		common.LogSuccess(result)
+	}
+}
+```
+
+结果大概这样
+
+![image-20240819101048478](img/2_fscanPro/image-20240819101048478.png)
+