Day 25 | Go 实战工具开发 (一)

package main

import (
    "fmt"
    "net"
    "time"
)

// CheckPort 探测单个端口
// 参数: host (IP或域名), port (端口号)
// 返回: bool (true 表示开放, false 表示关闭)
func CheckPort(host string, port int) bool {
    address := fmt.Sprintf("%s:%d", host, port)
    
    // 1秒超时，这对局域网或内网足够了。公网可能需要设大一点。
    conn, err := net.DialTimeout("tcp", address, 1*time.Second)
    
    if err != nil {
        // 连接失败（超时或 Refused）
        return false
    }
    
    // 重要：一旦连接成功，必须手动关闭，否则会耗尽系统句柄
    conn.Close()
    return true
}

package main

import (
    "fmt"
    "sort"
    "sync"
    "time"
)

// 把 CheckPort 函数复制到这里，或者放在同一个包下

func main() {
    host := "scanme.nmap.org" // Nmap 官方提供的测试靶机
    fmt.Printf("Scanning %s ...\n", host)
    start := time.Now()

    // 1. 初始化通道
    // jobs: 存放要扫描的端口号
    // results: 存放扫描结果（开放的端口号）
    jobs := make(chan int, 100)
    results := make(chan int, 100)
    
    var wg sync.WaitGroup

    // 2. 启动 50 个并发 Worker
    // 这些 Worker 启动后会立刻阻塞在 range jobs 上，等待任务
    for i := 0; i < 50; i++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            // 不断取任务，直到 jobs 通道被关闭
            for port := range jobs {
                if CheckPort(host, port) {
                    results <- port // 发现开放端口，发送给结果通道
                } else {
                    results <- 0    // 未开放，发送 0 表示完成
                }
            }
        }()
    }

    // 3. 生产者：分发任务 (1-1024 端口)
    // 另起一个协程去发任务，避免阻塞主线程接收结果
    go func() {
        for i := 1; i <= 1024; i++ {
            jobs <- i
        }
        close(jobs) // 发完了，记得关掉，通知 Worker 准备收工
    }()

    // 4. 收集结果
    var openPorts []int
    
    // 既然我们知道发了 1024 个任务，就肯定会收到 1024 个结果
    // 这里可以直接循环 1024 次，或者用另外的 WaitGroup 策略
    for i := 1; i <= 1024; i++ {
        port := <-results
        if port != 0 {
            openPorts = append(openPorts, port)
            fmt.Printf("✨ Port %d is open\n", port)
        }
    }

    // 5. 打印最终报告
    close(results)
    sort.Ints(openPorts) // 排序让结果更好看
    
    fmt.Printf("\n--- Scan Finished in %v ---\n", time.Since(start))
    fmt.Printf("Total %d ports open: %v\n", len(openPorts), openPorts)
}