请教strcut设计的问题

brantc

小弟有个程序是抓包的，有一个链表是根据source ip和dest ip进行统计有多少包被抓到。
所以我的包的数据结构是这样的：

1. struct node {
   
2.   u_char s_ip[16];        //source IP
   
3.   u_char d_ip[16];        //destination IP
   
4.   u_int32_t package_amount;   //总共抓到的包数量，会被累加
   
5.   struct node *next;
   
6. };
   
7. struct node *head;      
   
8. struct node *search;      //(1)标识找到的位置 (2)用于开辟新节点的变量
   
9. struct node *tail;           //尾指针始终指向最后一个节点

复制代码

但是呢，因为TCP和UDP是有端口的，我上面的结构就无法体现，而且我要在打印统计时把端口也打印出来，于是我就把上面的node修改成：

1. struct node {
   
2.   u_char s_ip[16];        //source IP
   
3.   u_char d_ip[16];        //destination IP
   
4.   u_int32_t s_port;      //source port
   
5.   u_int32_t d_port;      //destination port
   
6.   u_int32_t package_amount;   //总共抓到的包数量，会被累加
   
7.   struct node *next;
   
8. };
   
9. struct node *tu_head;         // tu表示tcp和udp
   
10. struct node *tu_search;      //(1)标识找到的位置 (2)用于开辟新节点的变量
    
11. struct node *tu_tail;           //尾指针始终指向最后一个节点
    

复制代码

但是呢，我的程序只做了抓TCP和UDP，我还想在遇到不认识的协议时，仍然能输出该s_ip和d_ip的统计数量，当然此时就没有端口的信息，这时最简单的方法就是再增加一个struct，例如：

1. struct unknown_node {
   
2.   u_char s_ip[16];        //source IP
   
3.   u_char d_ip[16];        //destination IP
   
4.   u_int32_t package_amount;   //总共抓到的包数量，会被累加
   
5.   struct node *next;
   
6. };

复制代码

同理，就需要增加一个相关struct的链表处理，如下：

1. struct unknown_node *unknown_head;      
   
2. struct unknown_node *unknown_search;      //(1)标识找到的位置 (2)用于开辟新节点的变量
   
3. struct unknown_node *unknown_tail;           //尾指针始终指向最后一个节点

复制代码

这样就很麻烦了。
因为我首先要判断是不是TCP或者UDP，如果是，用一个函数进行struct node 类型的tu_head/tu_search/tu_tail的处理；
如果不是，我就需要另一个函数进行struct unknown_node类型的unknown_head/unknown_search/unknown_tail的处理。

有没有什么办法，可以让我只用一个函数就处理两种struct的链表的处理（C++的用法参考，比如用什么继承呀之类的，就可以在函数用父类的类型，不用管传进来的参数的类型。而C语言就必须传进的参数和函数声明的参数一致（用强制转换相当不好））？

或者，有没有办法让两种strcut变成一种（这样我原来的处理链表的程序和程序流程就可以都不用大改）？比如在struct里面用#ifdef等控制语句控制或者用union类型？我只是大概有点印象好像见过，但是具体不知道怎么操作。

请指教！
谢谢！

brantc

好嘛，没人回我。
我帖stackoverflow去了，有大佬已经回答了。
链接付这里，供大家参考：http://stackoverflow.com/questio ... odate-more-elements

mightofcode

继续用原来那个node啊
加个int说明是那种包,不就可以满足你的需求了么

就是有点冗余数据

wk_g7

我喜欢这样做

u_int32_t package_amount;   //总共抓到的包数量，会被累加
u_int32_t package_tcp_amount;   //总共抓到的tcp or udp包数量，会被累加

struct node {
  u_char s_ip[16];        //source IP
  u_char d_ip[16];        //destination IP
  u_int32_t s_port;      //source port
  u_int32_t d_port;      //destination port

  int protocol_type;
  struct node *next;
  struct node * next_tcp;
  struct node * next_not_tcp;
};

struct node *head;      
struct node *search;      //(1)标识找到的位置 (2)用于开辟新节点的变量

struct node *tail;           //尾指针始终指向最后一个节点

brantc

回复 3# mightofcode


    嗯，就是不想有冗余的。

brantc

回复 4# wk_g7


    感谢回复。
你的办法不错，但是还是会有冗余的数据。
看起来，如果只用一个结构体，就始终会有冗余的。。。。
小弟最后还是用的一个结构体
不是TCP或者UDP，port就置-1。。。。自己都觉得太傻了 。。。
空了再参考下其他的类似问题的好设计吧