Golang 写一个端口扫描器

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/pwc1996/article/details/73469850

前话

最近痴迷于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] <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.1
    port 80 is opening
    
    $ port-scanner.exe -r 1~100 -n 50 127.0.0.1
    port 80 is opening

没有更多推荐了,返回首页

私密
私密原因:
请选择设置私密原因
  • 广告
  • 抄袭
  • 版权
  • 政治
  • 色情
  • 无意义
  • 其他
其他原因:
120
出错啦
系统繁忙,请稍后再试