source: test/test-convert.c

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

Add tests for writing pcapng

  • Property mode set to 100644
File size: 7.9 KB
RevLine 
[222d8f5]1/*
2 * This file is part of libtrace
3 *
[d5a27e8]4 * Copyright (c) 2007 The University of Waikato, Hamilton, New Zealand.
[222d8f5]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 */
[8f851f3]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
[222d8f5]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>
[8f851f3]48
[222d8f5]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";
[568a341]76        if (!strcmp(type,"pcapng"))
77                return "pcapng:traces/100_packets.pcapng";
[222d8f5]78        if (!strcmp(type,"wtf"))
79                return "wtf:traces/wed.wtf";
80        if (!strcmp(type,"rtclient"))
81                return "rtclient:chasm";
[2d96bf2]82        if (!strcmp(type,"pcapfile"))
83                return "pcapfile:traces/100_packets.pcap";
[86bec15]84        if (!strcmp(type,"pcapfilens"))
85                return "pcapfile:traces/100_packetsns.pcap";
[cec8028]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";
[cd7eec7]92        if (!strcmp(type, "duck"))
93                return "duck:traces/100_packets.duck";
[1b9272f]94        if (!strcmp(type, "tsh"))
95                return "tsh:traces/10_packets.tsh.gz";
[afaf6e8]96        if (!strcmp(type, "sll1"))
97                return "pcapfile:traces/sll.pcap.gz";
98        if (!strcmp(type, "sll2"))
99                return "pcapfile:traces/100_sll.pcap";
[222d8f5]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";
[ac62ea2]108        if (!strcmp(type,"pcapfile"))
109                return "pcapfile:traces/100_packets.out.pcap";
[222d8f5]110        if (!strcmp(type,"wtf"))
111                return "wtf:traces/wed.out.wtf";
[cd7eec7]112        if (!strcmp(type,"duck"))
113                return "duck:traces/100_packets.out.duck";
[51276bd]114        if (!strcmp(type,"pcapng"))
115                return "pcapng:traces/100_packets.out.pcapng";
[222d8f5]116        return "unknown";
117}
118
[6010b54]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);
[ef5a9ab]126       
[ee2ea572]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",
[ef5a9ab]131                                tv1.tv_sec, tv1.tv_usec, 
132                                tv2.tv_sec, tv2.tv_usec);
[6010b54]133
134                return 1;
[ef5a9ab]135        }
[6010b54]136        return 0;
137
138}
[afaf6e8]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
[222d8f5]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;
[67cd8c1]173        int tcpcount = 0;
[961f687]174        libtrace_t *trace,*trace2;
[222d8f5]175        libtrace_out_t *outtrace;
[961f687]176        libtrace_packet_t *packet,*packet2;
[10cf089]177        const char *trace1name;
178        const char *trace2name;
[222d8f5]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);
[ac62ea2]191        if (trace_is_err_output(outtrace)) {
192                trace_perror_output(outtrace,"WARNING: ");
193        }
[222d8f5]194
195        trace_start(trace);
196        iferr(trace);
197        trace_start_output(outtrace);
198        iferrout(outtrace);
199       
200        packet=trace_create_packet();
201        for (;;) {
[51276bd]202
[222d8f5]203                if ((psize = trace_read_packet(trace, packet)) <0) {
204                        error = 1;
205                        break;
206                }
[51276bd]207
[222d8f5]208                if (psize == 0) {
209                        error = 0;
210                        break;
211                }
[568a341]212                if (trace_write_packet(outtrace,packet) > 0)
[51276bd]213                        if (!IS_LIBTRACE_META_PACKET(packet)) {
214                                count ++;
215                        }
[222d8f5]216                iferrout(outtrace);
217                if (count>100)
218                        break;
219        }
[f208f92]220        trace_destroy_packet(packet);
[222d8f5]221        if (error == 0) {
[961f687]222                if (count != expected) {
[222d8f5]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);
[961f687]231
232        if (error)
233                return error;
234
235        /* Now read it back in again and check it's all kosher */
[10cf089]236        trace1name = lookup_uri(argv[1]);
237        trace = trace_create(trace1name);
[961f687]238        iferr(trace);
239        trace_start(trace);
[10cf089]240        trace2name = lookup_out_uri(argv[2]);
241        trace2 = trace_create(trace2name);
[961f687]242        iferr(trace2);
243        trace_start(trace2);
244        iferr(trace2);
245        packet=trace_create_packet();
246        packet2=trace_create_packet();
247        count=0;
[67cd8c1]248        tcpcount=0;
[51276bd]249
[961f687]250        while(trace_read_packet(trace,packet)>0) {
251                int err;
[568a341]252
253                if (IS_LIBTRACE_META_PACKET(packet))
254                        continue;
255
[961f687]256                ++count;
[568a341]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
[10cf089]266                /* The capture length might be snapped down to the wire length */
[afaf6e8]267                if (length_changed(packet, packet2)) {
[51276bd]268
[10cf089]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));
[5b8f9ee]283                        abort();
[cec8028]284                }
[51276bd]285
[6010b54]286                if (time_changed(packet, packet2)) {
[ef5a9ab]287                        error = 1;
288                        break;
[6010b54]289                }
[51276bd]290
[77a38d9]291                if (trace_get_tcp(packet)) {
292                        if (!trace_get_tcp(packet2)) {
[503a58c]293                                printf("trace corrupt -- expected tcp\n");
[77a38d9]294                                error=1;
295                                break;
296                        }
[67cd8c1]297                        ++tcpcount;
[77a38d9]298                } else {
299                        if (trace_get_tcp(packet2)) {
300                                printf("trace corrupt: unexpected tcp\n");
301                                error=1;
302                                break;
303                        }
304                }
305
[961f687]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);
[f208f92]313        trace_destroy_packet(packet);
314        trace_destroy_packet(packet2);
[961f687]315
[67cd8c1]316        printf("tcpcount=%i\n",tcpcount);
317
[222d8f5]318        return error;
319}
Note: See TracBrowser for help on using the repository browser.