add an userdata argument to register_stat_cmd_arg() and its callback to use the...
[obnox/wireshark/wip.git] / gtk / hostlist_ip.c
1 /* hostlist_ip.c   2004 Ian Schorr
2  * modified from endpoint_talkers_ip.c   2003 Ronnie Sahlberg
3  *
4  * $Id$
5  *
6  * Ethereal - Network traffic analyzer
7  * By Gerald Combs <gerald@ethereal.com>
8  * Copyright 1998 Gerald Combs
9  *
10  * This program is free software; you can redistribute it and/or
11  * modify it under the terms of the GNU General Public License
12  * as published by the Free Software Foundation; either version 2
13  * of the License, or (at your option) any later version.
14  *
15  * This program is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18  * GNU General Public License for more details.
19  *
20  * You should have received a copy of the GNU General Public License
21  * along with this program; if not, write to the Free Software
22  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
23  */
24
25 #ifdef HAVE_CONFIG_H
26 # include "config.h"
27 #endif
28
29 #include <stdio.h>
30
31 #ifdef HAVE_SYS_TYPES_H
32 # include <sys/types.h>
33 #endif
34
35 #include <gtk/gtk.h>
36 #include <string.h>
37 #include "epan/packet.h"
38 #include <epan/stat_cmd_args.h>
39 #include "../stat_menu.h"
40 #include "gui_stat_menu.h"
41 #include <epan/tap.h>
42 #include "../register.h"
43 #include "hostlist_table.h"
44 #include <epan/dissectors/packet-ip.h>
45
46
47 static int
48 ip_hostlist_packet(void *pit, packet_info *pinfo, epan_dissect_t *edt _U_, const void *vip)
49 {
50         hostlist_table *hosts=(hostlist_table *)pit;
51         const e_ip *iph=vip;
52
53         /* Take two "add" passes per packet, adding for each direction, ensures that all
54         packets are counted properly (even if address is sending to itself) 
55         XXX - this could probably be done more efficiently inside hostlist_table */
56         add_hostlist_table_data(hosts, &iph->ip_src, 0, TRUE, 1, pinfo->fd->pkt_len, SAT_NONE, PT_NONE);
57         add_hostlist_table_data(hosts, &iph->ip_dst, 0, FALSE, 1, pinfo->fd->pkt_len, SAT_NONE, PT_NONE);
58         return 1;
59 }
60
61 static void
62 gtk_ip_hostlist_init(const char *optarg, void* userdata)
63 {
64         const char *filter=NULL;
65
66         if(!strncmp(optarg,"hosts,ip,",9)){
67                 filter=optarg+9;
68         } else {
69                 filter=NULL;
70         }
71
72         init_hostlist_table(TRUE, "IPv4 Hosts", "ip", filter, ip_hostlist_packet);
73
74 }
75
76
77 static void
78 gtk_ip_hostlist_cb(GtkWidget *w _U_, gpointer d _U_)
79 {
80         gtk_ip_hostlist_init("hosts,ip",NULL);
81 }
82
83
84 void
85 register_tap_listener_ip_hostlist(void)
86 {
87         register_stat_cmd_arg("hosts,ip", gtk_ip_hostlist_init,NULL);
88
89         register_stat_menu_item("IPv4", REGISTER_STAT_GROUP_ENDPOINT_LIST,
90                 gtk_ip_hostlist_cb, NULL, NULL, NULL);
91
92         register_hostlist_table(TRUE, "IPv4", "ip", NULL /*filter*/, ip_hostlist_packet);
93 }