UDP端口探活的那些细节

一 背景

商业客户反馈用categraf的net_response插件配置了udp探测, 遇到报错了，如图 

udp是无连接的，无法用建立连接的形式判断端口。 插件最初的设计是需要配置udp的发送字符，并且配置期望返回的字符串，

[[instances]]
targets = [
      "127.0.0.1:161",
]

protocol = "udp"

## string sent to the server
  send = "hello"
## expected string in answer
  expect = "hello"

通过返回字符与期望字符是否相等，来判断端口是否连通。用户随即发了另一张图 ，用ncat 来探测端口是ok的 

ncat 探测逻辑

先看下 ncat的udp探测逻辑

/*
 * udptest()
 * Do a few writes to see if the UDP port is there.
 * Fails once PF state table is full.
 */
int
udptest(int s)
{
    int i, t;

    if ((write(s, "X", 1) != 1) ||
        ((write(s, "X", 1) != 1) && (errno == ECONNREFUSED)))
        return -1;

    /* Give the remote host some time to reply. */
    for (i = 0, t = (timeout == -1) ? UDP_SCAN_TIMEOUT : (timeout / 1000);
         i < t; i++) {
        sleep(1);
        if ((write(s, "X", 1) != 1) && (errno == ECONNREFUSED))
            return -1;
    }
    return 1;
}

先理一下代码片段的探测逻辑，先向目标写入一个X，观察是否有ECONNREFUSED, 如果有，则表示端口没有打开； 如果没有ECONNREFUSED，>则看一下timeout是否设置，没有设置，则for循环3次（UDP_SCAN_TIMEOUT），如果设置了timeout， 则for循环timeout的次数（以秒计 ）。再看下for循环里面，依然是每次写入一个 X ，观察是否有 ECONNREFUSED。

简单来说，就是向探测目标发送一个X，观察是否有connection refused，没有的话表明目标端口是打开的（即使对端没有返回任何内容导致超时 ）。

实现

看完这个逻辑就简单了，我们可以用go照着实现。网络上的udp port scanner 除了发送X , 还有发送0的, 也有根据已知端口，按照协议发送数>据的。简单和通用起见，还是按照ncat的逻辑来。

       msg := []byte("X")
        t := math.Max(float64(time.Duration(ins.ReadTimeout)/time.Second), 3)
        for i := 0; i < int(t); i++ {
            time.Sleep(1 * time.Second)
            _, err = conn.Write(msg)
            if err != nil && config.Config.DebugMode {
                log.Printf("E! write udp failed, address: %s, error: %s", address, err)
            }
            if err != nil && strings.Contains(err.Error(), "refused") {
                fields["result_code"] = ConnectionFailed
                return tags, fields, nil
            }
    }

完整PR见链接