go语言编写端口扫描学习篇（二）

接上一篇笔记，这里我们又做了些许优化，具体我直接标注在了代码的后面，可以根据前一篇笔记比对

package main

import (
	"fmt"
	"net"
	"sort"
)

//处理工作的worker函数
func worker(ports, results chan int) {
	for p := range ports { //range是go语言系统定义的一个函数。 函数的含义是在一个数组中遍历每一个值,返回该值的下标值和此处的实际值。 假如说a[0]=10,则遍历到a[0]的时候返回值为0,10两个值
		address := fmt.Sprintf("wangehacker.cn:%d", p)
		conn, err := net.Dial("tcp", address)
		if err != nil { //这里如果端口关闭则传递一个0数值给results
			results <- 0
			continue
		}
		conn.Close()
		results <- p //端口开启则将端口号传递给results

	}
}
func main() {
	ports := make(chan int, 100)
	results := make(chan int) //创建一个单独的通道
	var openports []int

	for i := 0; i < cap(ports); i++ { //进行切片储存，方便排序
		go worker(ports, results)
	}
	go func() {
		for i := 1; i < 1024; i++ { //1024个端口号依次放入ports参数中
			ports <- 1
		}
	}()
	for i := 1; i <= 1024; i++ { //收集循环，结果不为0，也就是端口开启，则追加到切片等待排序
		port := <-results
		if port != 0 {
			openports = append(openports, port)
		}
	}
	close(ports)
	close(results)
	sort.Ints(openports)             // sort.Ints排序，openports为int类型，也就是传入为int类型用ints
	for _, port := range openports { //对结果进行一个排序
		fmt.Printf("%d open\n", port)
	}
}

有后续继续完善还会更新