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语言编写端口扫描的学习笔记(一)_第1张图片因此我们就要使用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的并发)
}

这样基本完成第一阶段
go语言编写端口扫描的学习笔记(一)_第2张图片

你可能感兴趣的:(go黑帽子程序编程,学习,golang)