Go实现简单TCP扫描器
女主宣言
今天小编为大家分享一篇关于Go实现TCP扫描器的文章,如果大家对Go编写扫描器感兴趣,可以看下本篇文章。希望能对大家有所帮助。
PS:丰富的一线技术、多元化的表现形式,尽在“360云计算”,点关注哦!
Go在网络应用编程方面堪称完美。它自带的标准库也很优秀,在开发过程中可以给予我们很多帮助。
在本文中,我们将会用Go写一个简单的TCP扫描器。整个程序的代码在50行以内。在我们开始动手之前,先介绍一些理论知识。
不得不说,TCP是比我们介绍的要复杂的多,但是我们只介绍一点基础知识。TCP的握手有三个过程。首先,客户端发送一个 syn 的包,表示建立回话的开始。如果客户端收到超时,说明端口可能在防火墙后面,
第二,如果服务端应答 syn-ack 包,意味着这个端口是打开的,否则会返回 rst 包。最后,客户端需要另外发送一个 ack 包。从这时起,连接就已经建立。
我们TCP扫描器第一步先实现单个端口的测试。使用标准库中的 net.Dial 函数,该函数接收两个参数:协议和测试地址(带端口号)。
package main
import ( "fmt" "net")
func main() { _, err := net.Dial("tcp", "google.com:80") if err == nil { fmt.Println("Connection successful") } else { fmt.Println(err) }}
为了不一个一个地测试每个端口,我们将添加一个简单的循环来简化整个测试过程。
package main
import ( "fmt" "net")
func main() { for port := 80; port < 100; port++ { conn, err := net.Dial("tcp", fmt.Sprintf("google.com:%d", port)) if err == nil { conn.Close() fmt.Println("Connection successful") } else { fmt.Println(err) } }}
这种处理方式有个很大的问题,极度的慢。我们可以通过两个操作来处理一下:并行的执行及为每个连接添加超时控制。
我们来看下如何实现并行。第一步先把扫描功能拆分为一个独立函数。这样会使我们的代码看起来清晰。
func isOpen(host string, port int) bool { time.Sleep(time.Millisecond * 1) conn, err := net.Dial("tcp", fmt.Sprintf("%s:%d", host, port)) if err == nil { _ = conn.Close() return true }
return false}
我们会引入一个新的方法 WaitGroup ,详细用法信息可以参考标准库文档。在主函数中,我们可以拆分为协程去执行,然后等待执行结束。
func main() { ports := []int{}
wg := &sync.WaitGroup{} for port := 1; port < 100; port++ { wg.Add(1) go func() { opened := isOpen("google.com", port) if opened { ports = append(ports, port) } wg.Done() }() }
wg.Wait() fmt.Printf("opened ports: %v\n", ports)}
我们的代码已经执行的很快了,但是由于超时的原因,我们需要等待很久才能收到返回的错误信息。我们可以假设如果我们200毫秒内没有收到服务器的回应,就不再继续等待。
func isOpen(host string, port int, timeout time.Duration) bool { time.Sleep(time.Millisecond * 1) conn, err := net.DialTimeout("tcp", fmt.Sprintf("%s:%d", host, port), timeout) if err == nil { _ = conn.Close() return true }
return false}
func main() { ports := []int{}
wg := &sync.WaitGroup{} timeout := time.Millisecond * 200 for port := 1; port < 100; port++ { wg.Add(1) go func(p int) { opened := isOpen("google.com", p, timeout) if opened { ports = append(ports, p) } wg.Done() }(port) }
wg.Wait() fmt.Printf("opened ports: %v\n", ports)}
至此,我们就得到了一个简单的端口扫描器。但有些不好的是,不能很方便的修改域名地址以及端口号范围,我们必须要重新编译代码才可以。Go还有一个很不错的包叫做 flag 。
flag 包可以帮助我们编写命令行程序。我们可以配置每个字符串或数字。我们为主机名及要测试的端口范围和连接超时添加参数。
func main() { hostname := flag.String("hostname", "", "hostname to test") startPort := flag.Int("start-port", 80, "the port on which the scanning starts") endPort := flag.Int("end-port", 100, "the port from which the scanning ends") timeout := flag.Duration("timeout", time.Millisecond * 200, "timeout") flag.Parse()
ports := []int{}
wg := &sync.WaitGroup{} for port := *startPort; port <= *endPort; port++ { wg.Add(1) go func(p int) { opened := isOpen(*hostname, p, *timeout) if opened { ports = append(ports, p) } wg.Done() }(port) }
wg.Wait() fmt.Printf("opened ports: %v\n", ports)}
如果我们想要显示如何使用,我们可以添加一个 -h 参数,来显示使用说明。整个项目不到50行的代码,我们使用到了并行、flag 及 net 包。
唯一的问题就是,现在这个程序会有竞争条件。在只扫描少数端口时,速度比较慢,可能不会出现,但确实存在这个问题。所以我们需要使用 mutex 来修复它。
wg := &sync.WaitGroup{}mutex := &sync.Mutex{}for port := *startPort; port <= *endPort; port++ { wg.Add(1) go func(p int) { opened := isOpen(*hostname, p, *timeout) if opened { mutex.Lock() ports = append(ports, p) mutex.Unlock() } wg.Done() }(port)}
我们本次只是简单的实现端口扫描的功能。如果大家喜欢编写这种工具,可以加入自己的理解或特性。参照 nmap 等著名扫描器的实现思路,用Go来打造自己的扫描器,从而加深对网络编程的理解。
总结
以上就是本次分享的内容~
如果有什么改进建议,也可以在我们评论区留言,供大家参考学习。
360云计算
由360云平台团队打造的技术分享公众号,内容涉及数据库、大数据、微服务、容器、AIOps、IoT等众多技术领域,通过夯实的技术积累和丰富的一线实战经验,为你带来最有料的技术分享
Go实现简单TCP扫描器相关推荐
- Unity 简单TCP通信实现
参考教程:[01_C#入门到精通]新手强烈推荐:C#开发课程,一整套课程_哔哩哔哩_bilibiliP178-P187 我由于毕设需要来学网络通信,属于完全没常识的小白,网上很多现成代码却不懂概念.不 ...
- c语言扫描器,Linux C语言写的超级简单port扫描器
这个本来曾经也写过的,今天无聊复习下 再写一遍.简单的一塌糊涂,写的不咋地大家见谅哦!有空再加强 嘿嘿! #include #include #include #include #include #i ...
- 使用netty实现简单tcp服务端
文章目录 前言 服务端 启动方式 前言 最近在做的项目有一个需要对接TCP的功能,网上查了一下,决定用netty来实现. 服务端 这次的需求只需要做一个服务端,话不多说,直接上代码 pom <d ...
- python socket编程实现的简单tcp迭代server
与c/c++ socket编程对照见http://blog.csdn.net/aspnet_lyc/article/details/38946915 server: import socketPORT ...
- windows简单TCP通信 C++
尝试使用自定义struct和char发送信息.服务器根据客户端发送的不同消息进行回应. 在发送struct时先发送一个"Who"以便客户端确认. 服务器 #define WIN32 ...
- python 简单TCP通信实例 TCP断线自动重连 客户端服务端没有启动顺序
先启动客户端也可以 服务端: import socket import time#服务端HOST = '' PORT = 8888 ADDR = (HOST, PORT) BUFFSIZE = 102 ...
- 360云计算|2019年度精选文章
女主宣言 本篇整理了2019年360云计算原创优秀文章! PS:丰富的一线技术.多元化的表现形式,尽在"360云计算",点关注哦! 点击文章标题即可查看 存储&消息中间件 ...
- windows下socket开发tcp程序 简单例子
转载自:http://blog.csdn.net/trbbadboy/article/details/7408003 Windows下的简单tcp例子.代码均在VC6.0下调试过.服务器代码: //T ...
- Linux下套接字详解(四)----简单的TCP套接字应用(迭代型)
前面我们已经将了TCP/UDP的基本知识,还说了并发服务器与迭代服务器的区别,我们大致了解大多数TCP服务器是并发的,大多数UDP服务器是迭代的 ,即我们在进行数据传送的时候,往往使用服务器与客户但之 ...
最新文章
- 谷歌公布GSoC 2020 暑期代码项目名单,200个开源项目30个新增
- React项目实践系列一
- properties 资源文件读取
- xml相关php函数,PHP利用xml常用函数的详细集合示例
- javascript无限请求_JAVASCRIPT事件循环
- 聊聊spring的ioc
- List特有迭代器--ListIterator的特殊功能
- 洛谷 P2622 关灯问题II (状态压缩+BFS)
- abaqus dat文件 matlab_基于MPCCI的FLUENT与ABAQUS流固耦合分析步骤
- 斯蒂芬-如何选择约会地点大大增加约会成功机率的干货贴
- Java HashMap原理
- google人体浏览器+android,google人体浏览器
- 西安80转2000坐标系重新定义投影
- 牛刀杀鸡-开源社区API之抢楼大作战
- 回溯法解决部落冲突问题
- CSDN论坛如何能够获得积分呢?
- 助力企业设备上云,京东云上物联网产品重磅发布
- Word电子扫描仪 word文档转换为图片Pdf,Word文档扫描成Pdf,word文档加密 word转图片 word转pdf
- 4毛发渲染及着色方法
- Allegro如何录制SCR快捷键操作指导