source: examples/protocol/protocol.c @ be54c88

4.0.1-hotfixescachetimestampsdevelopdpdk-ndagetsilivegetfragoffhelplibtrace4ndag_formatpfringrc-4.0.1rc-4.0.2rc-4.0.3rc-4.0.4ringdecrementfixringperformanceringtimestampfixes
Last change on this file since be54c88 was 781cf2c, checked in by Daniel Lawson <dlawson@…>, 17 years ago

normal / wdcap ipchksum test
getopt

  • Property mode set to 100644
File size: 3.7 KB
Line 
1/*
2 * This file is part of libtrace
3 *
4 * Copyright (c) 2004 The University of Waikato, Hamilton, New Zealand.
5 * Authors: Daniel Lawson
6 *          Perry Lorier
7 *         
8 * All rights reserved.
9 *
10 * This code has been developed by the University of Waikato WAND
11 * research group. For further information please see http://www.wand.net.nz/
12 *
13 * libtrace is free software; you can redistribute it and/or modify
14 * it under the terms of the GNU General Public License as published by
15 * the Free Software Foundation; either version 2 of the License, or
16 * (at your option) any later version.
17 *
18 * libtrace is distributed in the hope that it will be useful,
19 * but WITHOUT ANY WARRANTY; without even the implied warranty of
20 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
21 * GNU General Public License for more details.
22 *
23 * You should have received a copy of the GNU General Public License
24 * along with libtrace; if not, write to the Free Software
25 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
26 *
27 * $Id$
28 *
29 */
30
31#include <stdio.h> /* printf */
32#include <netinet/in.h> /* ntohs */
33#include <netdb.h>
34#include "dagformat.h"
35#include <getopt.h>
36
37#include <stdlib.h>
38#include <string.h>
39
40#include <errno.h>
41
42#include <linux/types.h>
43#include <linux/linkage.h>
44#define  access_ok(type,addr,size) 1
45#include <asm/checksum.h>
46#define  IN_CHKSUM(IP)  ip_fast_csum((unsigned char *)(IP), 5)
47
48#include "libtrace.h"
49
50struct libtrace_t *trace;
51struct libtrace_filter_t *filter;
52
53char *buffer[4096];
54uint64_t badchksum = 0;
55char *uri = 0;
56char *filterstring = 0;
57
58int do_cksum = 0;
59int do_w_cksum = 0;
60static void usage();
61static void parse_cmdline(int argc, char **argv);
62
63int main(int argc, char **argv) {
64
65        struct libtrace_ip *ipptr = 0;
66       
67        int status; // need to pass to rtclient_read_packet
68        int psize;
69
70        parse_cmdline(argc,argv);
71
72        // create an rtclient to hostname, on the default port
73        trace = create_trace(uri);
74        if (filterstring) {
75                filter = libtrace_bpf_setfilter(filterstring);
76        }
77
78        for (;;) {
79                if ((psize = libtrace_read_packet(trace, buffer,4096, &status)) <= 0) {
80                        // terminate
81                        break;
82                }
83                if (filter) {
84                        if (!libtrace_bpf_filter(trace, filter, buffer, 4096)) {
85                        continue;
86                        }
87                }
88                ipptr = get_ip(trace,buffer,4096);
89
90                if(do_cksum && IN_CHKSUM(ipptr)) {
91                        badchksum ++;
92                } else if (do_w_cksum && ipptr->ip_sum) {
93                        badchksum ++;
94                } else {
95                        if (ipptr) {
96                                printf("%d:%d\n",ipptr->ip_p,get_link_type(trace,buffer,4096));
97                        }
98                }
99        }
100        if (do_cksum || do_w_cksum) {
101                printf("Bad checksums seen: %llu\n",badchksum);
102        }
103        destroy_trace(trace);
104        return 0;
105}
106
107
108static void usage(char *prog) {
109        printf("usage: %s [-h] [-c | -w] [-u <uri>] [-f <filterstring>]\n",prog);
110        printf("        -h              this help message\n");
111        printf("        -c              perform ip checksum test\n");
112        printf("        -w              check WDCAPd ip checksum value\n");
113        printf("        -u uri          uri to connect to\n");
114        printf("        -f filterstring BPF filterstring to apply\n");
115        printf("\n");
116        printf(" The use of -c and -w are exclusive: -c is used for normal traces, while -w applies to traces taken from the Waikato Capture point\n");
117}
118
119static void parse_cmdline(int argc, char **argv){
120        int opt;
121        if (argc == 1) {
122                usage(argv[0]);
123                exit(0);
124        }
125       
126        while ((opt = getopt(argc,argv, "hcwu:f:")) != EOF) {
127                switch(opt) {
128                        case 'h':
129                                usage(argv[0]);
130                                exit(0);
131                        case 'c':
132                                do_cksum = 1;
133                                break;
134                        case 'w':
135                                do_w_cksum = 1;
136                                break;
137                        case 'u':
138                                uri = strdup(optarg);
139                                break;
140                        case 'f':
141                                filterstring = strdup(optarg);
142                                break;
143                        default:
144                                usage(argv[0]);
145                                exit(0);
146                }
147
148        }
149
150        if (do_cksum && do_w_cksum) {
151                usage(argv[0]);
152                exit(0);
153        }
154
155}
Note: See TracBrowser for help on using the repository browser.