Golang 写一个端口扫描器
发布时间:2020-12-16 18:12:22 所属栏目:大数据 来源:网络整理
导读:前话 最近痴迷于Golang这个新兴语言,因为它是强类型编译型语言,可以直接编译成三大平台的二进制执行文件,可以直接运行无需其他依赖环境。而且Golang独特的goroutine使得多线程任务执行如new一个对象般简单。 带着为学习理解Golang的好奇心情,我试着写了
前话最近痴迷于Golang这个新兴语言,因为它是强类型编译型语言,可以直接编译成三大平台的二进制执行文件,可以直接运行无需其他依赖环境。而且Golang独特的goroutine使得多线程任务执行如new一个对象般简单。 带着为学习理解Golang的好奇心情,我试着写了个端口扫描器。
源码package main
import (
"flag"
"fmt"
"net"
"os"
"regexp"
"strconv"
"strings"
"sync"
)
var port int
var portRange string
var parallelCounts int
func init() {
flag.IntVar(&port,"p", 80,"port")
flag.StringVar(&portRange,"r","","range ports. format is <from>~<to>. eg. 100~200")
flag.IntVar(¶llelCounts,"n", 1,"parallel counts")
// 修改提示信息
flag.Usage = func() {
fmt.Fprintf(os.Stderr,"nUsage: %s [Options] <IP>nnOptions:nn",os.Args[0])
flag.PrintDefaults()
}
flag.Parse()
}
func printOpeningPort(port int) {
fmt.Println("port " + strconv.Itoa(port) + " is opening")
}
func checkPort(ip net.IP,port int,wg *sync.WaitGroup,parallelChan *chan int) {
defer wg.Done()
tcpAddr := net.TCPAddr{
IP: ip,Port: port,}
conn,err := net.DialTCP("tcp",nil,&tcpAddr)
if err == nil {
printOpeningPort(port)
conn.Close()
}
<-*parallelChan
}
func main() {
args := flag.Args()
if len(args) != 1 {
flag.Usage()
} else {
ip := net.ParseIP(flag.Arg(0))
// 用于协程任务控制
wg := sync.WaitGroup{}
if portRange != "" {
matched,_ := regexp.Match(`^d+~d+$`,[]byte(portRange))
if !matched {
flag.Usage()
} else {
portSecs := strings.Split(portRange,"~")
startPort,err1 := strconv.Atoi(portSecs[0])
endPort,err2 := strconv.Atoi(portSecs[1])
if err1 != nil || err2 != nil || startPort < 1 || endPort < 2 || endPort <= startPort || parallelCounts < 1 {
flag.Usage()
} else {
wg.Add(endPort - startPort + 1)
// 用于控制协程数
parallelChan := make(chan int,parallelCounts)
for i := startPort; i <= endPort; i++ {
parallelChan <- 1
go checkPort(ip,i,&wg,¶llelChan)
}
wg.Wait()
}
}
} else {
wg.Add(1)
parallelChan := make(chan int)
go func() {
parallelChan <- 1
}()
go checkPort(ip,port,¶llelChan)
wg.Wait()
}
}
}
运行结果
(编辑:李大同) 【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容! |