程序中如何控制协程数量以优化端口扫描器
在编写一个使用协程限制端口扫描器中协程数量的 go 语言程序时,您可能会遇到扫描结果不如预期的问题。以下是问题的详细介绍和解决方案:
问题描述
使用 sync.wtgroup 限制协程数量时,在 系统上可以成功扫描出端口,但在 系统上却无法获得全部扫描结果。例如,设置 1000 个协程时,可以在 windows 上扫描出 80、443、8000 和 3389 端口,但在 linux 上只能扫描出 80 和 443 端口。
解决方案
此问题不在于 go 语言代码本身,而是 linux 系统的默认配置限制。linux 中允许同时打开的文件数量默认设置为 1024。当同时创建超过 1024 个协程时,可能会导致同时打开的文件数超过限制,从而导致套接字连接建立失败。
要解决此问题,可以采取以下方法:
- 降低协程数量:协程数量并不总是越多越好。根据实际情况调整协程数量可以提高效率。
- 修改 linux 配置:可以修改 linux 的配置限制以允许同时打开更多文件。执行以下命令以增加打开文件的限制:
ulimit -shn 10000
登录后复制
执行此命令后,应该能够使用较多的协程并扫描出全部端口。
立即学习“”;
优化后的代码
以下是经过修改的优化代码:
package main import ( "fmt" "net" "time" "sync" ) type Job struct { host string port int } type Result struct { job Job status bool } var success uint32 var fail uint32 var successMux sync.Mutex var failMux sync.Mutex var jobs = make(chan Job) var results = make(chan Result) func worker(wg *sync.WaitGroup) { for job := range jobs { _, err := net.DialTimeout("tcp", fmt.Sprintf("%s:%d", job.host, job.port), time.Millisecond*5500) if err != nil { failMux.Lock() fail++ failMux.Unlock() if job.port == 50051 { fmt.Println("I am port 50051, error:", err.Error()) } results <p>修改后的代码中,将允许同时打开的文件数增加到了 10000。这将允许使用更多的协程,并应该能够扫描出全部端口。</p>
登录后复制
以上就是Golang 端口扫描器在 Linux 系统下无法扫描出全部端口,该如何解决?的详细内容,更多请关注php中文网其它相关文章!