This is to test the waters.
[obnox/wireshark/wip.git] / tap-iousers.c
index 695e3ef1d5ec663cdf97a5d74f63134173f14589..c030f469a7f0cc4741f4eb3d528d29b473b83da6 100644 (file)
@@ -1,10 +1,10 @@
 /* tap-iousers.c
  * iostat   2003 Ronnie Sahlberg
  *
- * $Id: tap-iousers.c,v 1.12 2003/08/25 00:44:20 guy Exp $
+ * $Id$
  *
- * Ethereal - Network traffic analyzer
- * By Gerald Combs <gerald@ethereal.com>
+ * Wireshark - Network traffic analyzer
+ * By Gerald Combs <gerald@wireshark.org>
  * Copyright 1998 Gerald Combs
  * 
  * This program is free software; you can redistribute it and/or
 #include <string.h>
 #include <epan/packet_info.h>
 #include <epan/packet.h>
-#include <epan/resolv.h>
-#include "tap.h"
+#include <epan/emem.h>
+#include <epan/addr_resolv.h>
+#include <epan/tap.h>
+#include <epan/conversation.h>
+#include <epan/stat_cmd_args.h>
 #include "register.h"
-#include "packet-ip.h"
-#include "packet-ipx.h"
-#include "packet-tcp.h"
-#include "packet-udp.h"
-#include "packet-eth.h"
-#include "packet-tr.h"
-#include "packet-fc.h"
-#include <string.h>
+#include <epan/dissectors/packet-ip.h>
+#include <epan/dissectors/packet-ipx.h>
+#include <epan/dissectors/packet-tcp.h>
+#include <epan/dissectors/packet-udp.h>
+#include <epan/dissectors/packet-eth.h>
+#include <epan/dissectors/packet-sctp.h>
+#include <epan/dissectors/packet-tr.h>
+#include <epan/dissectors/packet-scsi.h>
+#include <epan/dissectors/packet-fc.h>
+#include <epan/dissectors/packet-fddi.h>
 
 typedef struct _io_users_t {
-       char *type;
+       const char *type;
        char *filter;
        struct _io_users_item_t *items;
 } io_users_t;
@@ -67,20 +72,30 @@ typedef struct _io_users_item_t {
 
 
 static int
-iousers_udpip_packet(io_users_t *iu, packet_info *pinfo, epan_dissect_t *edt _U_, void *vudph)
+iousers_udpip_packet(void *arg, packet_info *pinfo, epan_dissect_t *edt _U_, const void *vudph)
 {
-       e_udphdr *udph=vudph;
+       io_users_t *iu=arg;
+       const e_udphdr *udph=vudph;
        char name1[256],name2[256];
        io_users_item_t *iui;
        int direction=0;
 
-       if(CMP_ADDRESS(&udph->ip_src, &udph->ip_dst)>0){
-               snprintf(name1,256,"%s:%s",address_to_str(&udph->ip_src),get_udp_port(udph->uh_sport));
-               snprintf(name2,256,"%s:%s",address_to_str(&udph->ip_dst),get_udp_port(udph->uh_dport));
+       if(udph->uh_sport>udph->uh_dport){
+               direction=0;
+               g_snprintf(name1,256,"%s:%s",address_to_str(&udph->ip_src),get_udp_port(udph->uh_sport));
+               g_snprintf(name2,256,"%s:%s",address_to_str(&udph->ip_dst),get_udp_port(udph->uh_dport));
+       } else if(udph->uh_sport<udph->uh_dport){
+               direction=1;
+               g_snprintf(name2,256,"%s:%s",address_to_str(&udph->ip_src),get_udp_port(udph->uh_sport));
+               g_snprintf(name1,256,"%s:%s",address_to_str(&udph->ip_dst),get_udp_port(udph->uh_dport));
+       } else if(CMP_ADDRESS(&udph->ip_src, &udph->ip_dst)>0){
+               direction=0;
+               g_snprintf(name1,256,"%s:%s",address_to_str(&udph->ip_src),get_udp_port(udph->uh_sport));
+               g_snprintf(name2,256,"%s:%s",address_to_str(&udph->ip_dst),get_udp_port(udph->uh_dport));
        } else {
                direction=1;
-               snprintf(name2,256,"%s:%s",address_to_str(&udph->ip_src),get_udp_port(udph->uh_sport));
-               snprintf(name1,256,"%s:%s",address_to_str(&udph->ip_dst),get_udp_port(udph->uh_dport));
+               g_snprintf(name2,256,"%s:%s",address_to_str(&udph->ip_src),get_udp_port(udph->uh_sport));
+               g_snprintf(name1,256,"%s:%s",address_to_str(&udph->ip_dst),get_udp_port(udph->uh_dport));
        }
 
        for(iui=iu->items;iui;iui=iui->next){
@@ -117,20 +132,89 @@ iousers_udpip_packet(io_users_t *iu, packet_info *pinfo, epan_dissect_t *edt _U_
 
 
 static int
-iousers_tcpip_packet(io_users_t *iu, packet_info *pinfo, epan_dissect_t *edt _U_, void *vtcph)
+iousers_sctp_packet(void *arg, packet_info *pinfo, epan_dissect_t *edt _U_, const void *vsctp)
 {
-       struct tcpheader *tcph=vtcph;
+       io_users_t *iu=arg;
+       const struct _sctp_info* sctph = vsctp;
+       char name1[256],name2[256], s_sport[10], s_dport[10];
+       io_users_item_t *iui;
+       int direction=0;
+
+       g_snprintf(s_sport, sizeof s_sport, "%d",sctph->sport);
+       g_snprintf(s_dport, sizeof s_dport, "%d",sctph->dport);
+       
+       if(sctph->sport > sctph->dport) {
+               direction=0;
+               g_snprintf(name1,256,"%s:%s",address_to_str(&sctph->ip_src),s_sport);   
+               g_snprintf(name2,256,"%s:%s",address_to_str(&sctph->ip_dst),s_dport);   
+       } else if(sctph->sport < sctph->dport) {
+               direction=1;
+               g_snprintf(name1,256,"%s:%s",address_to_str(&sctph->ip_src),s_sport);   
+               g_snprintf(name2,256,"%s:%s",address_to_str(&sctph->ip_dst),s_dport);   
+       } else {
+               direction=0;
+               g_snprintf(name1,256,"%s:%s",address_to_str(&sctph->ip_src),s_sport);   
+               g_snprintf(name2,256,"%s:%s",address_to_str(&sctph->ip_dst),s_dport);   
+       }
+
+       for(iui=iu->items;iui;iui=iui->next){
+               if((!strcmp(iui->name1, name1))
+                && (!strcmp(iui->name2, name2)) ){
+                       break;
+               }
+       }
+
+       if(!iui){
+                iui=g_malloc(sizeof(io_users_item_t));
+                iui->next=iu->items;
+                iu->items=iui;
+/*              iui->addr1=NULL;*/
+                iui->name1=strdup(name1);
+/*              iui->addr2=NULL;*/
+                iui->name2=strdup(name2);
+                iui->frames1=0;
+                iui->frames2=0;
+                iui->bytes1=0;
+                iui->bytes2=0;
+        }
+
+       if(direction){
+                iui->frames1++;
+                iui->bytes1+=pinfo->fd->pkt_len;
+        } else {
+                iui->frames2++;
+                iui->bytes2+=pinfo->fd->pkt_len;
+        }
+
+        return 1;
+}
+
+
+static int
+iousers_tcpip_packet(void *arg, packet_info *pinfo, epan_dissect_t *edt _U_, const void *vtcph)
+{
+       io_users_t *iu=arg;
+       const struct tcpheader *tcph=vtcph;
        char name1[256],name2[256];
        io_users_item_t *iui;
        int direction=0;
 
-       if(CMP_ADDRESS(&tcph->ip_src, &tcph->ip_dst)>0){
-               snprintf(name1,256,"%s:%s",address_to_str(&tcph->ip_src),get_tcp_port(tcph->th_sport));
-               snprintf(name2,256,"%s:%s",address_to_str(&tcph->ip_dst),get_tcp_port(tcph->th_dport));
+       if(tcph->th_sport>tcph->th_dport){
+               direction=0;
+               g_snprintf(name1,256,"%s:%s",address_to_str(&tcph->ip_src),get_tcp_port(tcph->th_sport));
+               g_snprintf(name2,256,"%s:%s",address_to_str(&tcph->ip_dst),get_tcp_port(tcph->th_dport));
+       } else if(tcph->th_sport<tcph->th_dport){
+               direction=1;
+               g_snprintf(name2,256,"%s:%s",address_to_str(&tcph->ip_src),get_tcp_port(tcph->th_sport));
+               g_snprintf(name1,256,"%s:%s",address_to_str(&tcph->ip_dst),get_tcp_port(tcph->th_dport));
+       } else if(CMP_ADDRESS(&tcph->ip_src, &tcph->ip_dst)>0){
+               direction=0;
+               g_snprintf(name1,256,"%s:%s",address_to_str(&tcph->ip_src),get_tcp_port(tcph->th_sport));
+               g_snprintf(name2,256,"%s:%s",address_to_str(&tcph->ip_dst),get_tcp_port(tcph->th_dport));
        } else {
                direction=1;
-               snprintf(name2,256,"%s:%s",address_to_str(&tcph->ip_src),get_tcp_port(tcph->th_sport));
-               snprintf(name1,256,"%s:%s",address_to_str(&tcph->ip_dst),get_tcp_port(tcph->th_dport));
+               g_snprintf(name2,256,"%s:%s",address_to_str(&tcph->ip_src),get_tcp_port(tcph->th_sport));
+               g_snprintf(name1,256,"%s:%s",address_to_str(&tcph->ip_dst),get_tcp_port(tcph->th_dport));
        }
 
        for(iui=iu->items;iui;iui=iui->next){
@@ -167,10 +251,11 @@ iousers_tcpip_packet(io_users_t *iu, packet_info *pinfo, epan_dissect_t *edt _U_
 
 
 static int
-iousers_ip_packet(io_users_t *iu, packet_info *pinfo, epan_dissect_t *edt _U_, void *vip)
+iousers_ip_packet(void *arg, packet_info *pinfo, epan_dissect_t *edt _U_, const void *vip)
 {
-       e_ip *iph=vip;
-       address *addr1, *addr2;
+       io_users_t *iu=arg;
+       const e_ip *iph=vip;
+       const address *addr1, *addr2;
        io_users_item_t *iui;
 
        if(CMP_ADDRESS(&iph->ip_src, &iph->ip_dst)>0){
@@ -214,10 +299,11 @@ iousers_ip_packet(io_users_t *iu, packet_info *pinfo, epan_dissect_t *edt _U_, v
 }
 
 static int
-iousers_ipx_packet(io_users_t *iu, packet_info *pinfo, epan_dissect_t *edt _U_, void *vipx)
+iousers_ipx_packet(void *arg, packet_info *pinfo, epan_dissect_t *edt _U_, const void *vipx)
 {
-       ipxhdr_t *ipxh=vipx;
-       address *addr1, *addr2;
+       io_users_t *iu=arg;
+       const ipxhdr_t *ipxh=vipx;
+       const address *addr1, *addr2;
        io_users_item_t *iui;
 
        if(CMP_ADDRESS(&ipxh->ipx_src, &ipxh->ipx_dst)>0){
@@ -261,10 +347,11 @@ iousers_ipx_packet(io_users_t *iu, packet_info *pinfo, epan_dissect_t *edt _U_,
 }
 
 static int
-iousers_fc_packet(io_users_t *iu, packet_info *pinfo, epan_dissect_t *edt _U_, void *vfc)
+iousers_fc_packet(void *arg, packet_info *pinfo, epan_dissect_t *edt _U_, const void *vfc)
 {
-       fc_hdr *fchdr=vfc;
-       address *addr1, *addr2;
+       io_users_t *iu=arg;
+       const fc_hdr *fchdr=vfc;
+       const address *addr1, *addr2;
        io_users_item_t *iui;
 
        if(CMP_ADDRESS(&fchdr->s_id, &fchdr->d_id)<0){
@@ -308,10 +395,11 @@ iousers_fc_packet(io_users_t *iu, packet_info *pinfo, epan_dissect_t *edt _U_, v
 }
 
 static int
-iousers_eth_packet(io_users_t *iu, packet_info *pinfo, epan_dissect_t *edt _U_, void *veth)
+iousers_eth_packet(void *arg, packet_info *pinfo, epan_dissect_t *edt _U_, const void *veth)
 {
-       eth_hdr *ehdr=veth;
-       address *addr1, *addr2;
+       io_users_t *iu=arg;
+       const eth_hdr *ehdr=veth;
+       const address *addr1, *addr2;
        io_users_item_t *iui;
 
        if(CMP_ADDRESS(&ehdr->src, &ehdr->dst)<0){
@@ -355,10 +443,59 @@ iousers_eth_packet(io_users_t *iu, packet_info *pinfo, epan_dissect_t *edt _U_,
 }
 
 static int
-iousers_tr_packet(io_users_t *iu, packet_info *pinfo, epan_dissect_t *edt _U_, void *vtr)
+iousers_fddi_packet(void *arg, packet_info *pinfo, epan_dissect_t *edt _U_, const void *veth)
 {
-       tr_hdr *trhdr=vtr;
-       address *addr1, *addr2;
+       io_users_t *iu=arg;
+       const fddi_hdr *ehdr=veth;
+       const address *addr1, *addr2;
+       io_users_item_t *iui;
+
+       if(CMP_ADDRESS(&ehdr->src, &ehdr->dst)<0){
+               addr1=&ehdr->src;
+               addr2=&ehdr->dst;
+       } else {
+               addr2=&ehdr->src;
+               addr1=&ehdr->dst;
+       }
+
+       for(iui=iu->items;iui;iui=iui->next){
+               if((!CMP_ADDRESS(&iui->addr1, addr1))
+               &&(!CMP_ADDRESS(&iui->addr2, addr2)) ){
+                       break;
+               }
+       }
+
+       if(!iui){
+               iui=g_malloc(sizeof(io_users_item_t));
+               iui->next=iu->items;
+               iu->items=iui;
+               COPY_ADDRESS(&iui->addr1, addr1);
+               iui->name1=strdup(address_to_str(addr1));
+               COPY_ADDRESS(&iui->addr2, addr2);
+               iui->name2=strdup(address_to_str(addr2));
+               iui->frames1=0;
+               iui->frames2=0;
+               iui->bytes1=0;
+               iui->bytes2=0;
+       }
+
+       if(!CMP_ADDRESS(&ehdr->dst,&iui->addr1)){
+               iui->frames1++;
+               iui->bytes1+=pinfo->fd->pkt_len;
+       } else {
+               iui->frames2++;
+               iui->bytes2+=pinfo->fd->pkt_len;
+       }
+
+       return 1;
+}
+
+static int
+iousers_tr_packet(void *arg, packet_info *pinfo, epan_dissect_t *edt _U_, const void *vtr)
+{
+       io_users_t *iu=arg;
+       const tr_hdr *trhdr=vtr;
+       const address *addr1, *addr2;
        io_users_item_t *iui;
 
        if(CMP_ADDRESS(&trhdr->src, &trhdr->dst)<0){
@@ -402,13 +539,14 @@ iousers_tr_packet(io_users_t *iu, packet_info *pinfo, epan_dissect_t *edt _U_, v
 }
 
 static void
-iousers_draw(io_users_t *iu)
+iousers_draw(void *arg)
 {
+       io_users_t *iu = arg;
        io_users_item_t *iui;
        guint32 last_frames, max_frames;
 
        printf("================================================================================\n");
-       printf("%s Talkers\n",iu->type);
+       printf("%s Conversations\n",iu->type);
        printf("Filter:%s\n",iu->filter?iu->filter:"<No Filter>");
        printf("                                               |       <-      | |       ->      | |     Total     |\n");
        printf("                                               | Frames  Bytes | | Frames  Bytes | | Frames  Bytes |\n");
@@ -444,84 +582,104 @@ iousers_draw(io_users_t *iu)
 }
 
 void
-iousers_init(char *optarg)
+iousers_init(const char *optarg, void* userdata _U_)
 {
-       char *filter=NULL;
-       char *tap_type, *tap_type_name;
-       static int (*packet_func)(io_users_t *, packet_info *, epan_dissect_t *, void *);
+       const char *filter=NULL;
+       const char *tap_type, *tap_type_name;
+       tap_packet_cb packet_func;
        io_users_t *iu=NULL;
        GString *error_string;
 
-       if(!strncmp(optarg,"talkers,eth",11)){
-               if(optarg[11]==','){
-                       filter=optarg+12;
+       if(!strncmp(optarg,"conv,eth",8)){
+               if(optarg[8]==','){
+                       filter=optarg+9;
                } else {
                        filter=NULL;
                }
                tap_type="eth";
                tap_type_name="Ethernet";
                packet_func=iousers_eth_packet;
-       } else if(!strncmp(optarg,"talkers,fc",10)){
-               if(optarg[10]==','){
-                       filter=optarg+11;
+       } else if(!strncmp(optarg,"conv,fc",7)){
+               if(optarg[7]==','){
+                       filter=optarg+8;
                } else {
                        filter=NULL;
                }
                tap_type="fc";
                tap_type_name="Fibre Channel";
                packet_func=iousers_fc_packet;
-       } else if(!strncmp(optarg,"talkers,tcp",11)){
-               if(optarg[11]==','){
-                       filter=optarg+12;
+       } else if(!strncmp(optarg,"conv,fddi",9)){
+               if(optarg[9]==','){
+                       filter=optarg+10;
+               } else {
+                       filter=NULL;
+               }
+               tap_type="fddi";
+               tap_type_name="FDDI";
+               packet_func=iousers_fddi_packet;
+       } else if(!strncmp(optarg,"conv,tcp",8)){
+               if(optarg[8]==','){
+                       filter=optarg+9;
                } else {
                        filter=NULL;
                }
                tap_type="tcp";
                tap_type_name="TCP";
                packet_func=iousers_tcpip_packet;
-       } else if(!strncmp(optarg,"talkers,udp",11)){
-               if(optarg[11]==','){
-                       filter=optarg+12;
+       } else if(!strncmp(optarg,"conv,udp",8)){
+               if(optarg[8]==','){
+                       filter=optarg+9;
                } else {
                        filter=NULL;
                }
                tap_type="udp";
                tap_type_name="UDP";
                packet_func=iousers_udpip_packet;
-       } else if(!strncmp(optarg,"talkers,tr",10)){
-               if(optarg[10]==','){
-                       filter=optarg+11;
+       } else if(!strncmp(optarg,"conv,tr",7)){
+               if(optarg[7]==','){
+                       filter=optarg+8;
                } else {
                        filter=NULL;
                }
                tap_type="tr";
                tap_type_name="Token Ring";
                packet_func=iousers_tr_packet;
-       } else if(!strncmp(optarg,"talkers,ipx",11)){
-               if(optarg[11]==','){
-                       filter=optarg+12;
+       } else if(!strncmp(optarg,"conv,ipx",8)){
+               if(optarg[8]==','){
+                       filter=optarg+9;
                } else {
                        filter=NULL;
                }
                tap_type="ipx";
                tap_type_name="IPX";
                packet_func=iousers_ipx_packet;
-       } else if(!strncmp(optarg,"talkers,ip",10)){
-               if(optarg[10]==','){
-                       filter=optarg+11;
+       } else if(!strncmp(optarg,"conv,ip",7)){
+               if(optarg[7]==','){
+                       filter=optarg+8;
                } else {
                        filter=NULL;
                }
                tap_type="ip";
                tap_type_name="IPv4";
                packet_func=iousers_ip_packet;
+       } else if(!strncmp(optarg,"conv,sctp",9)) {
+               if(optarg[9]==','){
+                               filter=optarg+10;
+               } else {
+                        filter=NULL;
+                }
+               tap_type="sctp";
+               tap_type_name="SCTP";
+               packet_func=iousers_sctp_packet;
        } else {
-               fprintf(stderr, "tethereal: invalid \"-z talkers,<type>[,<filter>]\" argument\n");
+               fprintf(stderr, "tshark: invalid \"-z conv,<type>[,<filter>]\" argument\n");
                fprintf(stderr,"   <type> must be one of\n");
                fprintf(stderr,"      \"eth\"\n");
                fprintf(stderr,"      \"fc\"\n");
+               fprintf(stderr,"      \"fddi\"\n");
                fprintf(stderr,"      \"ip\"\n");
                fprintf(stderr,"      \"ipx\"\n");
+               fprintf(stderr,"      \"sctp\"\n");
                fprintf(stderr,"      \"tcp\"\n");
                fprintf(stderr,"      \"tr\"\n");
                fprintf(stderr,"      \"udp\"\n");
@@ -538,13 +696,13 @@ iousers_init(char *optarg)
                iu->filter=NULL;
        }
 
-       error_string=register_tap_listener(tap_type, iu, filter, NULL, (void*)packet_func, (void*)iousers_draw);
+       error_string=register_tap_listener(tap_type, iu, filter, NULL, packet_func, iousers_draw);
        if(error_string){
                if(iu->items){
                        g_free(iu->items);
                }
                g_free(iu);
-               fprintf(stderr, "tethereal: Couldn't register talkers tap: %s\n",
+               fprintf(stderr, "tshark: Couldn't register conversations tap: %s\n",
                    error_string->str);
                g_string_free(error_string, TRUE);
                exit(1);
@@ -555,6 +713,5 @@ iousers_init(char *optarg)
 void
 register_tap_listener_iousers(void)
 {
-       register_ethereal_tap("talkers,", iousers_init);
+       register_stat_cmd_arg("conv,", iousers_init, NULL);
 }
-