Golang 寫一個端口掃描器

前話

最近癡迷於Golang這個新興語言，由於它是強類型編譯型語言，能夠直接編譯成三大平臺的二進制執行文件，能夠直接運行無需其餘依賴環境。並且Golang獨特的goroutine使得多線程任務執行如new一個對象般簡單。

帶着爲學習理解Golang的好奇心情，我試着寫了個端口掃描器。

github項目連接以下， 更多的實用工具我會慢慢添加。
https://github.com/pwcong/go-tools

源碼

package main

import (
    "flag"
    "fmt"
    "net"
    "os"
    "regexp"
    "strconv"
    "strings"
    "sync"
)

var port int
var portRange string

var parallelCounts int

func 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.exe

   Usage: E:\Program Files\GoPath\bin\port-scanner.exe [Options]

   Options:

   -n int
   parallel counts (default 1)
   -p int
   port (default 80)
   -r string
   range ports. format is ~ . eg. 100~200

   $ port-scanner.exe -p 80 127.0.0.1
   port 80 is opening

   $ port-scanner.exe -r 1~100 -n 50 127.0.0.1
   port 80 is opening
   ```