Golang 寫一個連接埠掃描器

這是一個建立於 的文章，其中的資訊可能已經有所發展或是發生改變。

前話

最近癡迷於Golang這個新興語言，因為它是強型別編譯型語言，可以直接編譯成三大平台的二進位執行檔案，可以直接運行無需其他依賴環境。而且Golang獨特的goroutine使得多線程任務執行如new一個對象般簡單。

帶著為學習理解Golang的好奇心情，我試著寫了個連接埠掃描器。

github項目連結如下， 更多的工具 + 生產力我會慢慢添加。
https://github.com/pwcong/go-tools

源碼

package mainimport (    "flag"    "fmt"    "net"    "os"    "regexp"    "strconv"    "strings"    "sync")var port intvar portRange stringvar parallelCounts intfunc init() {    flag.IntVar(&port, "p", 80, "port")    flag.StringVar(&portRange, "r", "", "range ports. format is <from>~<to>. eg. 100~200")    flag.IntVar(&parallelCounts, "n", 1, "parallel counts")    // 修改提示資訊    flag.Usage = func() {        fmt.Fprintf(os.Stderr, "\nUsage: %s [Options] <IP>\n\nOptions:\n\n", os.Args[0])        flag.PrintDefaults()    }    flag.Parse()}func printOpeningPort(port int) {    fmt.Println("port " + strconv.Itoa(port) + " is opening")}func checkPort(ip net.IP, port int, wg *sync.WaitGroup, parallelChan *chan int) {    defer wg.Done()    tcpAddr := net.TCPAddr{        IP:   ip,        Port: port,    }    conn, err := net.DialTCP("tcp", nil, &tcpAddr)    if err == nil {        printOpeningPort(port)        conn.Close()    }    <-*parallelChan}func main() {    args := flag.Args()    if len(args) != 1 {        flag.Usage()    } else {        ip := net.ParseIP(flag.Arg(0))        // 用於協程任務控制        wg := sync.WaitGroup{}        if portRange != "" {            matched, _ := regexp.Match(`^\d+~\d+$`, []byte(portRange))            if !matched {                flag.Usage()            } else {                portSecs := strings.Split(portRange, "~")                startPort, err1 := strconv.Atoi(portSecs[0])                endPort, err2 := strconv.Atoi(portSecs[1])                if err1 != nil || err2 != nil || startPort < 1 || endPort < 2 || endPort <= startPort || parallelCounts < 1 {                    flag.Usage()                } else {                    wg.Add(endPort - startPort + 1)                    // 用於控制協程數                    parallelChan := make(chan int, parallelCounts)                    for i := startPort; i <= endPort; i++ {                        parallelChan <- 1                        go checkPort(ip, i, &wg, &parallelChan)                    }                    wg.Wait()                }            }        } else {            wg.Add(1)            parallelChan := make(chan int)            go func() {                parallelChan <- 1            }()            go checkPort(ip, port, &wg, &parallelChan)            wg.Wait()        }    }}

運行結果

1. 執行 go build ./main.go， 產生二進位檔案

2. 運行二進位檔案，結果如下：

   $ port-scanner.exeUsage: E:\Program Files\GoPath\bin\port-scanner.exe [Options] <IP>Options:  -n int        parallel counts (default 1)  -p int        port (default 80)  -r string        range ports. format is <from>~<to>. eg. 100~200$ port-scanner.exe -p 80 127.0.0.1port 80 is opening$ port-scanner.exe -r 1~100 -n 50 127.0.0.1port 80 is opening