package main

import (
    "fmt"
    "net"
    "strconv"
    "sync"
    "time"
)

func scanPort(ip string, port int, wg *sync.WaitGroup, resultCh chan<- string) {
    defer wg.Done()

    address := fmt.Sprintf("%s:%d", ip, port)
    conn, err := net.DialTimeout("tcp", address, 1*time.Second) // 設(shè)置連接超時時間
    if err != nil {
        resultCh <- fmt.Sprintf("端口 %d: 關(guān)閉", port)
        return
    }
    conn.Close()
    resultCh <- fmt.Sprintf("端口 %d: 開放", port)
}

func main() {
    var wg sync.WaitGroup
    resultCh := make(chan string, 100) // 創(chuàng)建帶緩沖的 Channel，緩沖區(qū)大小為 100

    fmt.Print("請輸入目標(biāo) IP 地址: ")
    var ip string
    fmt.Scanln(&ip)

    fmt.Print("請輸入端口范圍（例如：80 100）: ")
    var startPort, endPort int
    fmt.Scanln(&startPort, &endPort)

    // 掃描指定范圍的端口
    for port := startPort; port <= endPort; port++ {
        wg.Add(1)
        go scanPort(ip, port, &wg, resultCh)
    }

    // 等待所有掃描任務(wù)完成并關(guān)閉 Channel
    go func() {
        wg.Wait()
        close(resultCh)
    }()

    // 打印掃描結(jié)果
    fmt.Println("\n掃描結(jié)果：")
    for result := range resultCh {
        fmt.Println(result)
    }
}

go run scanner.go

請輸入目標(biāo) IP 地址: 192.168.1.1
請輸入端口范圍（例如：80 100）: 80 90

掃描結(jié)果：
端口 80: 開放
端口 81: 關(guān)閉
端口 82: 關(guān)閉
端口 83: 關(guān)閉
端口 84: 開放
端口 85: 關(guān)閉
端口 86: 開放
端口 87: 關(guān)閉
端口 88: 關(guān)閉
端口 89: 開放
端口 90: 關(guān)閉

go scanPort(ip, port, &wg, resultCh)

var wg sync.WaitGroup
wg.Add(1)

conn, err := net.DialTimeout("tcp", address, 1*time.Second)

resultCh <- fmt.Sprintf("端口 %d: 開放", port)

go func() {
    wg.Wait()
    close(resultCh)
}()