首先是一个基本的端口扫描,仅针对一个目标的一个端口
package main
import (
"fmt"
"net"
)
func main() {
_, err := net.Dial("tcp", "scan.nmap.org:80")
if err == nil {
fmt.Println("连接成功!")
} else {
fmt.Println("端口没有开启!")
}
}
这里的代码我们做到了写死一个目标,然后写死了一个端口进行检测,下一步我们就是做一定的优化,由此我得出了下面的代码。
package main
import (
"fmt"
"net"
)
func main() {
for i := 1; i <= 1024; i++ {
address := fmt.Sprintf("wangehacker.cn:%d", i)
conn, err := net.Dial("tcp", address)
// if err == nil {
// fmt.Println("连接成功!", i)
// }
if err != nil {
continue //端口已关闭或者已经过滤
}
conn.Close()
fmt.Printf("%d 开启\n", i)
}
}
我们在conn参数上用close来关闭链接,当然,我们代码的确可以检测出来,但是速度真的很慢。
因此我们就要使用go的并发
package main
import (
"fmt"
"net"
)
func main() {
for i := 1; i <= 1024; i++ {
go func(j int) {
address := fmt.Sprintf("wangehacker.cn:%d", j)
conn, err := net.Dial("tcp", address)
// if err == nil {
// fmt.Println("连接成功!", i)
//
if err != nil {
return
}
conn.Close()
fmt.Printf("%d 开启\n", j)
}(i)
}
}
这里我们实现了并发效果,代码会在完成后立刻退出,这会导致代码比代码与目标端口之间数据包交换还要快,我们就无法得到结果,于是我们需要利用sync中的waitGroup控制。waitGroup是一种结构体类型。
package main
import (
"fmt"
"net"
"sync"
)
func main() {
var wg sync.WaitGroup
for i := 1; i <= 1024; i++ {
wg.Add(2) //给一个计数器,到0前不进一步执行
go func(j int) {
defer wg.Done() //将计数器减一
address := fmt.Sprintf("wangehacker.cn:%d", j)
conn, err := net.Dial("tcp", address)
// if err == nil {
// fmt.Println("连接成功!", i)
//
if err != nil {
return
}
conn.Close()
fmt.Printf("%d 开启\n", j)
}(i)
}
wg.Wait() //组织在其中条用它的goroutine执行(go的并发)
}
这样基本完成第一阶段