一、實(shí)戰(zhàn)背景

端口掃描器是網(wǎng)絡(luò)安全領(lǐng)域常用的工具，它通過(guò)對(duì)目標(biāo)主機(jī)的端口進(jìn)行掃描，檢測(cè)目標(biāo)設(shè)備是否開(kāi)放某些服務(wù)端口（如 HTTP、FTP、SSH 等）。常見(jiàn)的端口掃描場(chǎng)景包括：

• 安全審計(jì)：檢測(cè)主機(jī)是否存在未授權(quán)服務(wù)
• 漏洞掃描：識(shí)別可能存在漏洞的服務(wù)端口
• 網(wǎng)絡(luò)診斷：排查服務(wù)器是否正常工作

傳統(tǒng)的端口掃描使用串行方式，一個(gè)端口一個(gè)端口地檢查，速度非常慢。而 Go 的并發(fā)編程可以極大地提升掃描效率。

二、實(shí)戰(zhàn)目標(biāo)

我們將實(shí)現(xiàn)一個(gè)并發(fā)端口掃描器，具備以下功能：

1. 輸入目標(biāo)主機(jī) IP 和端口范圍
2. 使用并發(fā)掃描多個(gè)端口
3. 輸出每個(gè)端口的開(kāi)放狀態(tài)
4. 使用超時(shí)機(jī)制控制掃描時(shí)間，避免阻塞

三、完整代碼實(shí)現(xiàn)

1. 掃描器實(shí)現(xiàn)（scanner.go）

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è)置連接超時(shí)時(shí)間
    if err != nil {
        resultCh <- fmt.Sprintf("端口 %d: 關(guān)閉", port)
        return
    }
    conn.Close()
    resultCh <- fmt.Sprintf("端口 %d: 開(kāi)放", port)
}

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

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

    fmt.Print("請(qǐng)輸入端口范圍（例如：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)
    }
}

四、運(yùn)行方式

執(zhí)行端口掃描器

go run scanner.go

輸入示例：

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

輸出示例：

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

五、關(guān)鍵技術(shù)點(diǎn)解析

1. 使用 Goroutine 并發(fā)掃描端口

每個(gè)端口的掃描任務(wù)都是由獨(dú)立的 Goroutine 執(zhí)行的。這樣可以最大化利用 CPU 核心，實(shí)現(xiàn)端口掃描的并發(fā)化。

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

2. 使用 sync.WaitGroup 等待所有 Goroutine 完成

WaitGroup 用來(lái)等待所有掃描任務(wù)完成，確保在關(guān)閉結(jié)果通道之前所有掃描任務(wù)已經(jīng)結(jié)束。

var wg sync.WaitGroup
wg.Add(1)

3. 使用 net.DialTimeout 設(shè)置連接超時(shí)

DialTimeout 可以在指定的超時(shí)時(shí)間內(nèi)完成連接，如果超時(shí)則返回錯(cuò)誤。我們?cè)谶@里設(shè)置了 1 秒的超時(shí)。

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

4. 使用 channel 收集掃描結(jié)果

通過(guò) Channel 傳遞每個(gè)端口的掃描結(jié)果，避免了主線程和 Goroutine 之間的數(shù)據(jù)競(jìng)爭(zhēng)問(wèn)題。

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

5. 緩存掃描結(jié)果并打印

使用帶緩沖的 Channel resultCh 存儲(chǔ)掃描結(jié)果。所有任務(wù)完成后，我們關(guān)閉通道并打印掃描結(jié)果。

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

六、優(yōu)化和擴(kuò)展方向

七、小結(jié)

通過(guò)本篇案例，你學(xué)會(huì)了如何用 Go 實(shí)現(xiàn)一個(gè)并發(fā)端口掃描器，掌握了以下關(guān)鍵技術(shù)：

• 使用 Goroutine 實(shí)現(xiàn)并發(fā)任務(wù)處理
• 利用 sync.WaitGroup 等待所有并發(fā)任務(wù)完成
• 使用 net.DialTimeout 進(jìn)行超時(shí)控制，防止阻塞
• 使用 Channel 收集并傳遞任務(wù)結(jié)果

這個(gè)并發(fā)端口掃描器可以幫助你在網(wǎng)絡(luò)安全和系統(tǒng)維護(hù)中快速診斷目標(biāo)主機(jī)的開(kāi)放端口。通過(guò)優(yōu)化，您還可以將其擴(kuò)展成更完整的網(wǎng)絡(luò)掃描工具。

以上就是基于Go語(yǔ)言實(shí)現(xiàn)一個(gè)并發(fā)端口掃描器的詳細(xì)內(nèi)容，更多關(guān)于Go并發(fā)端口掃描器的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

最新評(píng)論