使用Golang实现端口扫描工具时,通过goroutine和channel实现并发扫描,结合sync.WaitGroup和带缓冲channel控制并发数量,能高效稳定地探测开放端口。

使用Golang实现端口扫描工具时,利用其强大的并发模型可以显著提升扫描效率。Go语言通过goroutine和channel提供了轻量级的并发支持,非常适合用于网络探测类任务,比如端口扫描。下面介绍如何构建一个高效的并发端口扫描器,并分享几个关键优化技巧。
端口扫描的核心是尝试与目标IP的指定端口建立TCP连接。如果连接成功,说明端口开放。在Go中,可以使用net.DialTimeout来发起带超时的连接请求。
示例代码片段:
<font face="Courier New">func scanPort(host string, port int, timeout time.Duration) bool {
address := fmt.Sprintf("%s:%d", host, port)
conn, err := net.DialTimeout("tcp", address, timeout)
if err != nil {
return false
}
conn.Close()
return true
}</font>这个函数尝试连接指定主机和端口,成功则返回true,否则返回false。简单直接,但单次执行效率低,需引入并发。
立即学习“go语言免费学习笔记(深入)”;
将每个端口的扫描任务放入独立的goroutine中,可以同时探测多个端口。配合channel收集结果,避免竞态条件。
实现思路:
代码示例:
<font face="Courier New">func scanPorts(host string, ports []int, timeout time.Duration) []int {
var openPorts []int
var mutex sync.Mutex
var wg sync.WaitGroup
<pre class='brush:php;toolbar:false;'>resultCh := make(chan int)
// 启动结果收集goroutine
go func() {
for port := range resultCh {
mutex.Lock()
openPorts = append(openPorts, port)
mutex.Unlock()
}
}()
for _, port := range ports {
wg.Add(1)
go func(p int) {
defer wg.Done()
if scanPort(host, p, timeout) {
resultCh <- p
}
}(port)
}
// 所有扫描完成,关闭channel
go func() {
wg.Wait()
close(resultCh)
}()
// 等待所有结果处理完毕
for range resultCh {
}
return openPorts}
虽然Go的goroutine很轻量,但一次性启动成千上万个仍可能导致文件描述符耗尽或系统负载过高。应使用带缓冲的channel作为信号量来限制并发数。
优化方法:
示例:
<font face="Courier New">semaphore := make(chan struct{}, 100) // 最多100个并发
<p>for _, port := range ports {
wg.Add(1)
go func(p int) {
defer wg.Done()
semaphore <- struct{}{} // 获取令牌
if scanPort(host, p, timeout) {
resultCh <- p
}
<-semaphore // 释放令牌
}(port)
}</font>这样能有效控制同时活跃的连接数,避免被系统限制或触发防火墙告警。
基本上就这些。Golang的并发模型让端口扫描工具既高效又易于实现。关键是合理控制并发规模,平衡速度与稳定性。不复杂但容易忽略细节。
以上就是Golang实现端口扫描工具 通过并发提高扫描效率技巧的详细内容,更多请关注php中文网其它相关文章!
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号