go语言编写端口扫描的学习笔记（一）

首先是一个基本的端口扫描，仅针对一个目标的一个端口

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的并发）
}

这样基本完成第一阶段