我在代码里定义了两个通道,分别用于生产端口和限制连接数,如果不限制连接数,容易被对方检测到或导致对方服务器不能正常运行。
// 生产端口 var port = make(chan int, 10) // 限制并发数 var connect = make(chan string, 5)
可以使用net库的dial函数做为socket客户端,需要注意的是要设置超时时间,因为若主机不存在,或目标端口是关闭的,往往需要花费数秒才返回错误,这样扫描大量端口时效率会极其低下。在go中可以使用net.dialer结构体设置超时时间,然后在调用dial方法:
d := net.dialer{timeout: time.second} dial, err := d.dial("tcp", target)
只要err不等于nil,表示目标端口是对外开放的。
完整代码如下:
package main import ( "fmt" "net" "sync" "time" ) var wg sync.waitgroup func main() { var start, end int var address string fmt.printf("请输入目标ip:> ") fmt.scan(&address) fmt.printf("请输入起始端口:> ") fmt.scan(&start) fmt.printf("请输入结束端口:> ") fmt.scan(&end) wg.add(end - start + 1) run(address, start, end) wg.wait() fmt.println("执行完毕") } // 生产端口 var port = make(chan int, 10) // 限制并发数 var connect = make(chan string, 5) func run(address string, start, end int) { go func() { for i := start; i <= end; i++ { port <- i } }() go func() { // 消费端口 for p := range port { // 往通道写入目标地址,超过限制并发数会阻塞 connect <- fmt.sprintf("%s:%d", address, p) } }() go connect() } func connect() { // 并发请求 for target := range connect { // 设置超时时间 d := net.dialer{timeout: time.second} dial, err := d.dial("tcp", target) if err == nil { fmt.printf("%s 连接成功\n", target) dial.close() }else{ fmt.printf("%s 连接失败\n", target) } wg.done() } }
这里端口生产通道不是必须的,只是为了演示消费生产并发模型,当然这还是最简单的。
在run函数里我没有关闭这两个通道,按官方的说法是gc会回收不使用的通道,如果要手动关闭,可以定义defer闭包进行close。
我本地运行结果如下:
请输入目标ip:> 127.0.0.1 请输入起始端口:> 8080 请输入结束端口:> 8094 127.0.0.1:8080 连接成功 127.0.0.1:8081 连接成功 127.0.0.1:8082 连接成功 127.0.0.1:8083 连接成功 127.0.0.1:8084 连接成功 127.0.0.1:8085 连接成功 127.0.0.1:8086 连接成功 127.0.0.1:8087 连接成功 127.0.0.1:8088 连接成功 127.0.0.1:8089 连接成功 127.0.0.1:8090 连接成功 127.0.0.1:8091 连接成功 127.0.0.1:8092 连接成功 127.0.0.1:8093 连接失败 127.0.0.1:8094 连接失败 执行完毕
这个版本比较简陋,tcp连接过程也可以简化,后续再写另外一篇文章。因为最近在学rust语言,语法内容比较多,所以后面暂时发布编程相关的文章,提升一下语法熟练度。
以上就是go 实现简易端口扫描的示例的详细内容,更多关于go 实现端口扫描的资料请关注www.887551.com其它相关文章!