快捷導(dǎo)航

go 實(shí)現(xiàn)簡(jiǎn)易端口掃描的示例

更新時(shí)間：2021年05月22日 10:42:10 作者：codecodify

該功能實(shí)現(xiàn)原理很簡(jiǎn)單，就是發(fā)送socket連接(IP+端口)，如果能連接成功，說明目標(biāo)主機(jī)開放了某端口。當(dāng)要大量掃描端口時(shí)，就需要寫并發(fā)編程了。

我在代碼里定義了兩個(gè)通道，分別用于生產(chǎn)端口和限制連接數(shù)，如果不限制連接數(shù)，容易被對(duì)方檢測(cè)到或?qū)е聦?duì)方服務(wù)器不能正常運(yùn)行。

// 生產(chǎn)端口
var port = make(chan int, 10)

// 限制并發(fā)數(shù)
var connect = make(chan string, 5)

可以使用net庫的Dial函數(shù)做為socket客戶端，需要注意的是要設(shè)置超時(shí)時(shí)間，因?yàn)槿糁鳈C(jī)不存在，或目標(biāo)端口是關(guān)閉的，往往需要花費(fèi)數(shù)秒才返回錯(cuò)誤，這樣掃描大量端口時(shí)效率會(huì)極其低下。在go中可以使用net.Dialer結(jié)構(gòu)體設(shè)置超時(shí)時(shí)間，然后在調(diào)用Dial方法：

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

只要err不等于nil，表示目標(biāo)端口是對(duì)外開放的。
完整代碼如下：

package main

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


var wg sync.WaitGroup

func main() {
  var start, end int
  var address string
  fmt.Printf("請(qǐng)輸入目標(biāo)IP：> ")
  fmt.Scan(&address)
  fmt.Printf("請(qǐng)輸入起始端口：> ")
  fmt.Scan(&start)
  fmt.Printf("請(qǐng)輸入結(jié)束端口：> ")
  fmt.Scan(&end)
  wg.Add(end - start + 1)
  Run(address, start, end)
  wg.Wait()
  fmt.Println("執(zhí)行完畢")
}


// 生產(chǎn)端口
var port = make(chan int, 10)

// 限制并發(fā)數(shù)
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() {
    // 消費(fèi)端口
    for p := range port {
      // 往通道寫入目標(biāo)地址，超過限制并發(fā)數(shù)會(huì)阻塞
      connect <- fmt.Sprintf("%s:%d", address, p)
    }
  }()

  go Connect()


}

func Connect() {
  // 并發(fā)請(qǐng)求
  for target := range connect {
    // 設(shè)置超時(shí)時(shí)間
    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()
  }
}

這里端口生產(chǎn)通道不是必須的，只是為了演示消費(fèi)生產(chǎn)并發(fā)模型，當(dāng)然這還是最簡(jiǎn)單的。
在Run函數(shù)里我沒有關(guān)閉這兩個(gè)通道，按官方的說法是gc會(huì)回收不使用的通道，如果要手動(dòng)關(guān)閉，可以定義defer閉包進(jìn)行close。
我本地運(yùn)行結(jié)果如下：

請(qǐng)輸入目標(biāo)IP：> 127.0.0.1
請(qǐng)輸入起始端口：> 8080
請(qǐng)輸入結(jié)束端口：> 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 連接失敗
執(zhí)行完畢

這個(gè)版本比較簡(jiǎn)陋，TCP連接過程也可以簡(jiǎn)化，后續(xù)再寫另外一篇文章。因?yàn)樽罱趯W(xué)rust語言，語法內(nèi)容比較多，所以后面暫時(shí)發(fā)布編程相關(guān)的文章，提升一下語法熟練度。

以上就是go 實(shí)現(xiàn)簡(jiǎn)易端口掃描的示例的詳細(xì)內(nèi)容，更多關(guān)于go 實(shí)現(xiàn)端口掃描的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: