systemtap专家模式oneshot统计TCP半连接数量

ゞ浴缸里的玫瑰 2023-07-21 06:08 2阅读 0赞

大约不到两周前，我写了一个内核模块统计了TCP当前的半连接数量：  
[https://blog.csdn.net/dog250/article/details/105013772][https_blog.csdn.net_dog250_article_details_105013772]  
该实现采用了扫描每一个Listener的方式，从而替代了扫描所有的连接再过滤的方式，以 O ( 1 ) O(1) O(1)复杂度替代了 O ( n ) O(n) O(n)复杂度。

随后，我又换了一种方式可以实时查看当前的半连接计数值：  
[https://blog.csdn.net/dog250/article/details/105022347][https_blog.csdn.net_dog250_article_details_105022347]

再之后，一发不可收拾，开始了二进制hook手艺之旅。

回到最开始的扫描每一个Listener的方法，虽然OK，但是需要维护一个procfs的文件，以提供数据的读取，若要支持半连接统计功能，一个内核模块就必须一直处在载入状态。

能不能改成oneshot模式呢？即输出完数值，模块即卸载。

其实很容易，直接在模块的init函数里进行计算，然后输出值后，将init函数return -1即可。但这又是逆经理…这不优雅。

这里采用了stap的Guru专家模式，oneshot输出数值到控制台后，即退出。代码如下：

// dump_halfconn_num.stp
    %{ 
    #include <linux/module.h>
    #include <net/tcp.h>
    %}
    
    function dump_halfconn_num:long()
    %{ 
    	unsigned int num = 0;
    	int i;
    	struct inet_hashinfo *hashinfo = &tcp_hashinfo;
    
    	for (i = 0; i < INET_LHTABLE_SIZE; i++) { 
    		struct sock *sk;
    		struct hlist_nulls_node *node;
    		struct inet_listen_hashbucket *ilb;
    
    		ilb = &hashinfo->listening_hash[i];
    		spin_lock_bh(&ilb->lock);
    		sk_nulls_for_each(sk, node, &ilb->head) { 
    			struct inet_connection_sock *icsk = inet_csk(sk);
    			struct listen_sock *lopt;
    
    			read_lock_bh(&icsk->icsk_accept_queue.syn_wait_lock);
    			lopt = icsk->icsk_accept_queue.listen_opt;
    			if (lopt && lopt->qlen)
    				num += lopt->qlen;
    			read_unlock_bh(&icsk->icsk_accept_queue.syn_wait_lock);
    		}
    		spin_unlock_bh(&ilb->lock);
    	}
    
    	STAP_PRINTF("系统中当前的半连接数量为：%d\n", num);
    
    	STAP_RETVALUE = 0;
    %}
    
    probe begin
    { 
    	dump_halfconn_num();
    	exit(); // oneshot模式，dump完直接退出！
    }

使用方法非常简单：

[root@localhost test]# stap -g ./dump_halfconn_num.stp
    302
    [root@localhost test]#

当然了，这是因为经理今天穿了皮鞋，而且还露着白袜子。

--------------------

浙江温州皮鞋湿，下雨进水不会胖。

[https_blog.csdn.net_dog250_article_details_105013772]: https://blog.csdn.net/dog250/article/details/105013772
[https_blog.csdn.net_dog250_article_details_105022347]: https://blog.csdn.net/dog250/article/details/105022347