基于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连接,判断连接是否成功,以确认端口是否开放,工人函数执行结束,将返回值通过参数传递出来,主函数中获取结果,存入切片中,然后关闭通道,给切片排序,输出扫描结果。