使用Golang实现端口扫描工具时，通过goroutine和channel实现并发扫描，结合sync.WaitGroup和带缓冲channel控制并发数量，能高效稳定地探测开放端口。
使用Golang实现端口扫描工具时，利用其强大的并发模型可以显著提升扫描效率。Go语言通过goroutine和channel提供了轻量级的并发支持，非常适合用于网络探测类任务，比如端口扫描。下面介绍如何构建一个高效的并发端口扫描器，并分享几个关键优化技巧。
基础端口扫描逻辑
端口扫描的核心是尝试与目标IP的指定端口建立TCP连接。如果连接成功，说明端口开放。在Go中，可以使用net.DialTimeout来发起带超时的连接请求。
示例代码片段：
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
}
这个函数尝试连接指定主机和端口，成功则返回true，否则返回false。简单直接，但单次执行效率低，需引入并发。
立即学习“go语言免费学习笔记（深入）”；
使用Goroutine并发扫描端口
将每个端口的扫描任务放入独立的goroutine中，可以同时探测多个端口。配合channel收集结果，避免竞态条件。
实现思路：
- 为每个待扫描端口启动一个goroutine
- 使用无缓冲channel传递扫描结果（开放端口）
- 通过sync.WaitGroup等待所有任务完成
代码示例：
func scanPorts(host string, ports []int, timeout time.Duration) []int {
var openPorts []int
var mutex sync.Mutex
var wg sync.WaitGroup
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作为信号量来限制并发数。
优化方法：
- 创建一个容量为最大并发数的channel，作为“令牌”池
- 每个goroutine执行前先获取令牌，完成后释放
示例：
semaphore := make(chan struct{}, 100) // 最多100个并发
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)
}
这样能有效控制同时活跃的连接数，避免被系统限制或触发防火墙告警。
进一步优化建议
-
合理设置超时时间：连接超时建议设为1-3秒，太短可能误判，太长影响整体速度
-
支持端口范围配置：允许用户指定常见端口或自定义范围，减少无效扫描
-
使用更高效的协议：高级场景可考虑SYN扫描（需原始套接字权限），但TCP Connect扫描最通用
-
输出格式化：支持JSON或简洁列表输出，便于集成到其他系统
基本上就这些。Golang的并发模型让端口扫描工具既高效又易于实现。关键是合理控制并发规模，平衡速度与稳定性。不复杂但容易忽略细节。
