猿问

golang 检查 tcp 端口打开

我需要检查远程地址是否打开了特定的 TCP 端口。为此,我选择使用 golang。到目前为止,这是我的尝试:


func raw_connect(host string, ports []string) {

  for _, port := range ports {

     timeout := time.Second

     conn, err := net.DialTimeout("tcp", host + ":" + port, timeout)

     if err != nil {

        _, err_msg := err.Error()[0], err.Error()[5:]

        fmt.Println(err_msg)

     } else {

        msg, _, err := bufio.NewReader(conn).ReadLine()

        if err != nil {

           if err == io.EOF {

              fmt.Print(host + " " + port + " - Open!\n")

           }

        } else {

           fmt.Print(host + " " + port + " - " + string(msg))

        }

        conn.Close()

     }

   }

 }

当应用程序(例如 SSH)首先返回一个字符串时,这对于 TCP 端口工作得很好,我读取它并立即打印它。


但是,当 TCP 以上的应用程序首先等待来自客户端的命令(例如 HTTP)时,就会出现超时 (if err == io.EOF子句)。


这个超时时间很长。我需要立即知道端口是否打开。


是否有更适合此目的的技术?


POPMUISE
浏览 199回答 2
2回答

桃花长相依

要检查端口,您可以检查连接是否成功。例如:func raw_connect(host string, ports []string) {    for _, port := range ports {        timeout := time.Second        conn, err := net.DialTimeout("tcp", net.JoinHostPort(host, port), timeout)        if err != nil {            fmt.Println("Connecting error:", err)        }        if conn != nil {            defer conn.Close()            fmt.Println("Opened", net.JoinHostPort(host, port))        }    }}

30秒到达战场

检查多个端口示例func tcpGather(ip string, ports []string) map[string]string {    // check emqx 1883, 8083 port    results := make(map[string]string)    for _, port := range ports {        address := net.JoinHostPort(ip, port)        // 3 second timeout        conn, err := net.DialTimeout("tcp", address, 3*time.Second)        if err != nil {            results[port] = "failed"            // todo log handler        } else {            if conn != nil {                results[port] = "success"                _ = conn.Close()            } else {                results[port] = "failed"            }        }    }    return results}
随时随地看视频慕课网APP

相关分类

Go
我要回答