[Go语言] go 实现简易端口扫描的示例

// 生产端口
var port = make(chan int, 10)
// 限制并发数
var connect = make(chan string, 5)

d := net.Dialer{Timeout: time.Second}
dial, err := d.Dial("tcp", target)

package main
import (
  "fmt"
  "net"
  "sync"
  "time"
)

var wg sync.WaitGroup
func main() {
  var start, end int
  var address string
  fmt.Printf("请输入目标IP：> ")
  fmt.Scan(&address)
  fmt.Printf("请输入起始端口：> ")
  fmt.Scan(&start)
  fmt.Printf("请输入结束端口：> ")
  fmt.Scan(&end)
  wg.Add(end - start + 1)
  Run(address, start, end)
  wg.Wait()
  fmt.Println("执行完毕")
}

// 生产端口
var port = make(chan int, 10)
// 限制并发数
var connect = make(chan string, 5)
func Run(address string,  start, end int) {
  go func() {
  for i := start; i <= end; i++ {
    port <- i
  }
  }()
  go func() {
  // 消费端口
  for p := range port {
    // 往通道写入目标地址，超过限制并发数会阻塞
    connect <- fmt.Sprintf("%s:%d", address, p)
  }
  }()
  go Connect()

}
func Connect() {
  // 并发请求
  for target := range connect {
  // 设置超时时间
  d := net.Dialer{Timeout: time.Second}
  dial, err := d.Dial("tcp", target)
  if err == nil {
    fmt.Printf("%s 连接成功\n", target)
    dial.Close()
  }else{
    fmt.Printf("%s 连接失败\n", target)
  }
  wg.Done()
  }
}

请输入目标IP：> 127.0.0.1
请输入起始端口：> 8080
请输入结束端口：> 8094
127.0.0.1:8080 连接成功
127.0.0.1:8081 连接成功
127.0.0.1:8082 连接成功
127.0.0.1:8083 连接成功
127.0.0.1:8084 连接成功
127.0.0.1:8085 连接成功
127.0.0.1:8086 连接成功
127.0.0.1:8087 连接成功
127.0.0.1:8088 连接成功
127.0.0.1:8089 连接成功
127.0.0.1:8090 连接成功
127.0.0.1:8091 连接成功
127.0.0.1:8092 连接成功
127.0.0.1:8093 连接失败
127.0.0.1:8094 连接失败
执行完毕