source: test/test-convert.c @ 51276bd

develop
Last change on this file since 51276bd was 51276bd, checked in by Jacob Van Walraven <jcv9@…>, 22 months ago

Add tests for writing pcapng

  • Property mode set to 100644
File size: 7.9 KB
Line 
1/*
2 * This file is part of libtrace
3 *
4 * Copyright (c) 2007 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: test-pcap-to-erf.c,v 1.3 2006/02/27 03:41:12 perry Exp $
28 *
29 */
30#ifndef WIN32
31#  include <sys/time.h>
32#  include <netinet/in.h>
33#  include <netinet/in_systm.h>
34#  include <netinet/tcp.h>
35#  include <netinet/ip.h>
36#  include <netinet/ip_icmp.h>
37#  include <arpa/inet.h>
38#  include <sys/socket.h>
39#endif
40
41#include <stdio.h>
42#include <stdlib.h>
43#include <assert.h>
44#include <string.h>
45#include <sys/types.h>
46#include <time.h>
47#include <string.h>
48
49#include "dagformat.h"
50#include "libtrace.h"
51
52void iferr(libtrace_t *trace)
53{
54        libtrace_err_t err = trace_get_err(trace);
55        if (err.err_num==0)
56                return;
57        printf("Error: %s\n",err.problem);
58        exit(1);
59}
60
61void iferrout(libtrace_out_t *trace)
62{
63        libtrace_err_t err = trace_get_err_output(trace);
64        if (err.err_num==0)
65                return;
66        printf("Error: %s\n",err.problem);
67        exit(1);
68}
69
70char *lookup_uri(const char *type) 
71{
72        if (!strcmp(type,"erf"))
73                return "erf:traces/100_packets.erf";
74        if (!strcmp(type,"pcap"))
75                return "pcap:traces/100_packets.pcap";
76        if (!strcmp(type,"pcapng"))
77                return "pcapng:traces/100_packets.pcapng";
78        if (!strcmp(type,"wtf"))
79                return "wtf:traces/wed.wtf";
80        if (!strcmp(type,"rtclient"))
81                return "rtclient:chasm";
82        if (!strcmp(type,"pcapfile"))
83                return "pcapfile:traces/100_packets.pcap";
84        if (!strcmp(type,"pcapfilens"))
85                return "pcapfile:traces/100_packetsns.pcap";
86        if (!strcmp(type,"legacyatm"))
87                return "legacyatm:traces/legacyatm.gz";
88        if (!strcmp(type,"legacypos"))
89                return "legacypos:traces/legacypos.gz";
90        if (!strcmp(type,"legacyeth"))
91                return "legacyeth:traces/legacyeth.gz";
92        if (!strcmp(type, "duck"))
93                return "duck:traces/100_packets.duck";
94        if (!strcmp(type, "tsh"))
95                return "tsh:traces/10_packets.tsh.gz";
96        if (!strcmp(type, "sll1"))
97                return "pcapfile:traces/sll.pcap.gz";
98        if (!strcmp(type, "sll2"))
99                return "pcapfile:traces/100_sll.pcap";
100        return "unknown";
101}
102
103char *lookup_out_uri(const char *type) {
104        if (!strcmp(type,"erf"))
105                return "erf:traces/100_packets.out.erf";
106        if (!strcmp(type,"pcap"))
107                return "pcap:traces/100_packets.out.pcap";
108        if (!strcmp(type,"pcapfile"))
109                return "pcapfile:traces/100_packets.out.pcap";
110        if (!strcmp(type,"wtf"))
111                return "wtf:traces/wed.out.wtf";
112        if (!strcmp(type,"duck"))
113                return "duck:traces/100_packets.out.duck";
114        if (!strcmp(type,"pcapng"))
115                return "pcapng:traces/100_packets.out.pcapng";
116        return "unknown";
117}
118
119static int time_changed(libtrace_packet_t *packet, 
120                libtrace_packet_t *packet2) {
121
122        struct timeval tv1, tv2;
123
124        tv1 = trace_get_timeval(packet);
125        tv2 = trace_get_timeval(packet2);
126       
127        if (tv1.tv_sec != tv2.tv_sec || 
128                tv1.tv_usec - tv2.tv_usec > 1 ||
129                tv2.tv_usec - tv1.tv_usec > 1) { 
130                printf("Timestamps differ: %u.%u vs %u.%u\n",
131                                tv1.tv_sec, tv1.tv_usec, 
132                                tv2.tv_sec, tv2.tv_usec);
133
134                return 1;
135        }
136        return 0;
137
138}
139
140static int length_changed(libtrace_packet_t *packet, 
141                libtrace_packet_t *packet2) {
142
143        uint16_t cap_len_1 = trace_get_capture_length(packet);
144        uint16_t cap_len_2 = trace_get_capture_length(packet2);
145        uint16_t wlen_1 = trace_get_wire_length(packet);
146        uint16_t wlen_2 = trace_get_wire_length(packet2);
147
148        /* Special case where we demoted a packet */
149        if (trace_get_link_type(packet) == TRACE_TYPE_LINUX_SLL && 
150                        trace_get_link_type(packet2) != TRACE_TYPE_LINUX_SLL) {
151               
152                if (cap_len_2 != cap_len_1 - sizeof(libtrace_sll_header_t))
153                        return true;
154        } else {
155
156                if (cap_len_1 != cap_len_2 && cap_len_2 != wlen_2)
157                        return true;
158        }
159       
160        if (wlen_1 != wlen_2)
161                return true;
162
163        return false;
164
165}
166
167int main(int argc, char *argv[]) {
168        int psize = 0;
169        int error = 0;
170        int count = 0;
171        int level = 0;
172        int expected = 100;
173        int tcpcount = 0;
174        libtrace_t *trace,*trace2;
175        libtrace_out_t *outtrace;
176        libtrace_packet_t *packet,*packet2;
177        const char *trace1name;
178        const char *trace2name;
179
180        trace = trace_create(lookup_uri(argv[1]));
181        iferr(trace);
182
183        if (strcmp(argv[1],"rtclient")==0)
184                expected=101;
185
186        outtrace = trace_create_output(lookup_out_uri(argv[2]));
187        iferrout(outtrace);
188
189        level=0;
190        trace_config_output(outtrace,TRACE_OPTION_OUTPUT_COMPRESS,&level);
191        if (trace_is_err_output(outtrace)) {
192                trace_perror_output(outtrace,"WARNING: ");
193        }
194
195        trace_start(trace);
196        iferr(trace);
197        trace_start_output(outtrace);
198        iferrout(outtrace);
199       
200        packet=trace_create_packet();
201        for (;;) {
202
203                if ((psize = trace_read_packet(trace, packet)) <0) {
204                        error = 1;
205                        break;
206                }
207
208                if (psize == 0) {
209                        error = 0;
210                        break;
211                }
212                if (trace_write_packet(outtrace,packet) > 0)
213                        if (!IS_LIBTRACE_META_PACKET(packet)) {
214                                count ++;
215                        }
216                iferrout(outtrace);
217                if (count>100)
218                        break;
219        }
220        trace_destroy_packet(packet);
221        if (error == 0) {
222                if (count != expected) {
223                        printf("failure: %d packets expected, %d seen\n",expected,count);
224                        error = 1;
225                }
226        } else {
227                iferr(trace);
228        }
229        trace_destroy(trace);
230        trace_destroy_output(outtrace);
231
232        if (error)
233                return error;
234
235        /* Now read it back in again and check it's all kosher */
236        trace1name = lookup_uri(argv[1]);
237        trace = trace_create(trace1name);
238        iferr(trace);
239        trace_start(trace);
240        trace2name = lookup_out_uri(argv[2]);
241        trace2 = trace_create(trace2name);
242        iferr(trace2);
243        trace_start(trace2);
244        iferr(trace2);
245        packet=trace_create_packet();
246        packet2=trace_create_packet();
247        count=0;
248        tcpcount=0;
249
250        while(trace_read_packet(trace,packet)>0) {
251                int err;
252
253                if (IS_LIBTRACE_META_PACKET(packet))
254                        continue;
255
256                ++count;
257                do {
258                        if ((err=trace_read_packet(trace2,packet2))<1) {
259                                printf("premature EOF on destination, %i from %s, %i from %s\n",count,lookup_uri(argv[1]),count-1,lookup_out_uri(argv[2]));
260                                iferr(trace2);
261                                error=1;
262                                break;
263                        }
264                } while (IS_LIBTRACE_META_PACKET(packet2));
265
266                /* The capture length might be snapped down to the wire length */
267                if (length_changed(packet, packet2)) {
268
269                        printf("\t%s\t%s\n",
270                                trace1name,
271                                trace2name);
272                        printf("caplen\t%zd\t%zd\t%+zd\n",
273                                trace_get_capture_length(packet),
274                                trace_get_capture_length(packet2),
275                                trace_get_capture_length(packet2)-trace_get_capture_length(packet));
276                        printf("wirelen\t%zd\t%zd\t%+zd\n",
277                                trace_get_wire_length(packet),
278                                trace_get_wire_length(packet2),
279                                trace_get_wire_length(packet2)-trace_get_wire_length(packet));
280                        printf("link\t%d\t%d\n",
281                                trace_get_link_type(packet),
282                                trace_get_link_type(packet2));
283                        abort();
284                }
285
286                if (time_changed(packet, packet2)) {
287                        error = 1;
288                        break;
289                }
290
291                if (trace_get_tcp(packet)) {
292                        if (!trace_get_tcp(packet2)) {
293                                printf("trace corrupt -- expected tcp\n");
294                                error=1;
295                                break;
296                        }
297                        ++tcpcount;
298                } else {
299                        if (trace_get_tcp(packet2)) {
300                                printf("trace corrupt: unexpected tcp\n");
301                                error=1;
302                                break;
303                        }
304                }
305
306        }
307        if (trace_read_packet(trace2,packet2)>0) {
308                printf("Extra packets after EOF\n");
309                error=1;
310        }
311        trace_destroy(trace);
312        trace_destroy(trace2);
313        trace_destroy_packet(packet);
314        trace_destroy_packet(packet2);
315
316        printf("tcpcount=%i\n",tcpcount);
317
318        return error;
319}
Note: See TracBrowser for help on using the repository browser.