Categraf 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见链接

推荐阅读

 您阅读本篇文章共花了：