qiyesafe.com

golang高性能端口扫描

前言

最近有个小项目的需要,使用golang写了个端口扫描工具,不得不说golang的效率确实比python快的太多了。在使用一段时间golang之后,感觉有三个方面是优于python的:

  1. 一个方面是性能优越

  2. 第二方面是兼容性好

  3. 第三方面是可以跨平台编译成本地二进制文件,发布项目很方便。

接下来我把这个工具的源代码,以及使用方式给大家给大家分享一下。

PortScan

工具名称:PortScan

  1. 采用Go语言开发,支持从config.txt文件中读取目的ip和端口,对指定的目的服务器进行端口扫描

  2. config.txt支持配置端口列表,默认为22、36000、56000、3306

  3. 在服务器上连接目的服务器端口,仅做一次TCP三次握手

PortScan参数如下

 

参数说明:

  •  -c:用来指定config文件路径

  •  -limit:用来指定扫描端口的并发数,默认为1000

 

首先生成一个config.txt文件,内容如下:

 

ip共有255个,端口有10个,也就是说总共要进行2550次TCP连接。

 

接着使用go build PortScan.go命令,生成一个本地二进制文件PortScan.exe,然后执行命令:

PortScan.exe -c config.txt

最后的结果存储到result.txt文件中,大致用时是30s。

关键源代码

//首先从命令行中读取线程数和配置文件路径
//从配置文件中解析出ip和port
//配置文件格式为
// [ip]
// 127.0.0.1
// [port]
// 22
// 36000
// 56000
// 3306
//根据开启的线程数对指定ip和端口进行tcp连接
//如果端口开启,把ip:port按照格式返回

func Scanner(configFile string, functionid string, sendInfoFile string,limit int){

	runtime.GOMAXPROCS(runtime.NumCPU())
	data, err := ioutil.ReadFile(configFile)
	portlist := make([]string,0,10)

	if err != nil{
		log.Fatal(err)
		panic(err)
	}
	
	ip_index := bytes.Index(data,[]byte("[ip]"))

	port_index := bytes.Index(data,[]byte("[port]"))

	if ip_index < 0{

		Info.Println("文件格式有误: missing [ip]")
		return
	}

	if port_index <0{

		portlist = append(portlist,"22")
		portlist = append(portlist,"36000")
		portlist = append(portlist,"56000")
		portlist = append(portlist,"3306")

	}else{
		reg_port := regexp.MustCompile(`\d+`)
		var ports [][]byte

		if ip_index>port_index{
			ports = reg_port.FindAll(data[:ip_index],-1)

		}else{

			ports = reg_port.FindAll(data[port_index:],-1)
		}
		

		for _,v := range(ports){
			
			portlist = append(portlist,string(v))

		} 
	
	}
	

	
	reg_ip := regexp.MustCompile(`((25[0-5]|2[0-4]\d|((1\d{2})|([1-9]?\d)))\.){3}(25[0-5]|2[0-4]\d|((1\d{2})|([1-9]?\d)))`)

	ips := reg_ip.FindAll(data,-1)

	input := make(chan []byte, len(ips))

	result := make(chan string, len(ips))

	defer close(input)
	defer close(result)
	
	for _, v := range(ips){
		input <- v
	}
	//控制多少并发
	for i:=0; i<limit;i++{
		//这个时候可以启动扫描函数
		go ScanPort(portlist,input,result)
	}
   
   
	for i :=0; i< len(ips);i++{

		//将扫描的结果输出
		ip_result,ok:= <-result

		if !ok {
			break;
		}

	}
	
}

func ScanPort(portlist []string,intput chan []byte,result chan string ) {

	for{
		task,ok := <-intput

		if !ok{

			return
		}
		ip := string(task)
		Info.Println("scaning ",ip,"port",portlist)
		port_str :=""
		for i:=0; i<len(portlist); i++{
			_, err  := net.DialTimeout("tcp", ip+":"+portlist[i], time.Second*3)

			if err != nil{
				continue
			}
			port_str+=portlist[i]+" "

		}

		if len(port_str) >0{
			//说明有打开的端口
			// Info.Println(ip+":"+port_str+"open")
			result <- ip+"----"+"1"+"----"+"open port:"+port_str

		}else{

			result <- ip+"----"+"0"+"----"+"ok"
		}

	}

}

 

最后

关注公众号:七夜安全博客

  • 回复【1】:领取 Python数据分析 教程大礼包
  • 回复【2】:领取 Python Flask 全套教程
  • 回复【3】:领取 某学院 机器学习 教程
  • 回复【4】:领取 爬虫 教程
  • 回复【5】:领取 编译原理 教程 
  • 回复【6】:领取 渗透测试 教程 
  • 回复【7】:领取 人工智能数学基础 教程
本文章属于原创作品,欢迎大家转载分享,禁止修改文章的内容。尊重原创,转载请注明来自:七夜的故事 http://www.cnblogs.com/qiyeboy/
posted @ 2018-12-17 20:33  七夜的故事  阅读(8164)  评论(4编辑  收藏  举报
qiyesafe.com