基于go语言的多通道线程池tcp端口扫描器--极简极简版

基于go语言的多通道线程池tcp端口扫描器

继续复习go语言，今天写了个tcp端口扫描器，用上了go的多线程、通道，正好一次性都复习了φ(゜▽゜*)♪

注： 以下设计内容参考并学习《Go黑帽子 渗透测试编程之道》

源码

直接上源码！！

package main

import (
	"fmt"
	"net"
	"sort"
)

//工人方法，参数都为通道，ports用于传入，result用于返回结果
func worker(ports, results chan int) {
    //遍历通道传过来的参数
    for p := range ports {
        address := fmt.Sprintf("scanme.nmap.org:%d", p)
        conn, err := net.Dial("tcp", address)
        if err != nil {
            //将结果通过通道传出
            results <- 0
            continue
        }
        conn.Close()
        //将结果通过通道传出
        results <- p
    }
}

func main() {
    //创建通道，定义缓冲区大小
    ports := make(chan int, 100)
    results := make(chan int)
    //定义结果切片
    var openPorts []int
    //根据缓冲区大小，创建工人池
    for i := 0; i <= cap(ports); i++ {
        go worker(ports, results)
    }

    //将端口号通过通道传入工人方法中
    go func() {
        for i := 1; i < 1024; i++ {
            ports <- i
        }
    }()

    //通过通道获取工人方法执行结果
    for i := 0; i < 1024; i++{
        port := <-results
        if port != 0 {
            openPorts = append(openPorts, port)
        }
    }

    //关闭通道
    close(ports)
    close(results)

    //按照升序排序
    sort.Ints(openPorts)
    for _, port := range openPorts {
        fmt.Printf("%d open\n", port)
    }
}

脚本开发思路

这玩意儿的思路应该是很简单的，我没有加输入，所以暂时这个脚本简陋一点，功能就是：扫描scanme.nmap.org的1-1024端口。若朋友们需要可对其进行改造。

先创建一个工人方法，这就是线程要执行的函数，函数的参数为通道，然后在主函数中创建通道，用于传递参数，根据传入参数通道的大小，创建对应的数量的工人线程，组成线程池，将要扫描的端口传入通道中，工人函数开始执行，向目标地址发起tcp连接，判断连接是否成功，以确认端口是否开放，工人函数执行结束，将返回值通过参数传递出来，主函数中获取结果，存入切片中，然后关闭通道，给切片排序，输出扫描结果。