source: lib/trace.c @ 368a1ae

4.0.1-hotfixescachetimestampsdevelopdpdk-ndagetsilivelibtrace4ndag_formatpfringrc-4.0.1rc-4.0.2rc-4.0.3rc-4.0.4ringdecrementfixringperformanceringtimestampfixes
Last change on this file since 368a1ae was 368a1ae, checked in by Richard Sanger <rsangerarj@…>, 7 years ago

Fixes filtered and accepted stat counters

The accepted counter was an int instead of uint64_t.
The filtered counter was not thread safe.

This also tidies up the pread wrapper code - however includes some
changes from the next patch which changes its return so most
likely will not function correctly.

  • Property mode set to 100644
File size: 58.9 KB
Line 
1/*
2 * This file is part of libtrace
3 *
4 * Copyright (c) 2007,2008,2009,2010 The University of Waikato, Hamilton,
5 * New Zealand.
6 *
7 * Authors: Daniel Lawson
8 *          Perry Lorier
9 *          Shane Alcock
10 *         
11 * All rights reserved.
12 *
13 * This code has been developed by the University of Waikato WAND
14 * research group. For further information please see http://www.wand.net.nz/
15 *
16 * libtrace is free software; you can redistribute it and/or modify
17 * it under the terms of the GNU General Public License as published by
18 * the Free Software Foundation; either version 2 of the License, or
19 * (at your option) any later version.
20 *
21 * libtrace is distributed in the hope that it will be useful,
22 * but WITHOUT ANY WARRANTY; without even the implied warranty of
23 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
24 * GNU General Public License for more details.
25 *
26 * You should have received a copy of the GNU General Public License
27 * along with libtrace; if not, write to the Free Software
28 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
29 *
30 * $Id$
31 *
32 */
33
34
35#define _GNU_SOURCE
36#include "common.h"
37#include "config.h"
38#include <assert.h>
39#include <errno.h>
40#include <fcntl.h>
41#include <stdio.h>
42#include <stdlib.h>
43#include <string.h>
44#include <sys/stat.h>
45#include <sys/types.h>
46#ifndef WIN32
47#include <sys/socket.h>
48#endif
49#include <stdarg.h>
50#include <sys/param.h>
51
52#ifdef HAVE_LIMITS_H
53#  include <limits.h>
54#endif
55
56#ifdef HAVE_SYS_LIMITS_H
57#  include <sys/limits.h>
58#endif
59
60#ifdef HAVE_NET_IF_ARP_H
61#  include <net/if_arp.h>
62#endif
63
64#ifdef HAVE_NET_IF_H
65#  include <net/if.h>
66#endif
67
68#ifdef HAVE_NETINET_IN_H
69#  include <netinet/in.h>
70#endif
71
72#ifdef HAVE_NET_ETHERNET_H
73#  include <net/ethernet.h>
74#endif
75
76#ifdef HAVE_NETINET_IF_ETHER_H
77#  include <netinet/if_ether.h>
78#endif
79
80#include <time.h>
81#ifdef WIN32
82#include <sys/timeb.h>
83#endif
84
85#include "libtrace.h"
86#include "libtrace_int.h"
87
88#ifdef HAVE_PCAP_BPF_H
89#  include <pcap-bpf.h>
90#else
91#  ifdef HAVE_NET_BPF_H
92#    include <net/bpf.h>
93#  endif
94#endif
95
96
97#include "libtrace_int.h"
98#include "format_helper.h"
99#include "rt_protocol.h"
100
101#include <pthread.h>
102#include <signal.h>
103
104#define MAXOPTS 1024
105
106/* This file contains much of the implementation of the libtrace API itself. */
107
108static struct libtrace_format_t *formats_list = NULL;
109
110volatile int libtrace_halt = 0;
111
112/* Set once pstart is called used for backwards compatibility reasons */
113int libtrace_parallel = 0;
114
115/* strncpy is not assured to copy the final \0, so we
116 * will use our own one that does
117 */
118static void xstrncpy(char *dest, const char *src, size_t n)
119{
120        strncpy(dest,src,n);
121        dest[n]='\0';
122}
123 
124static char *xstrndup(const char *src,size_t n)
125{       
126        char *ret=(char*)malloc(n+1);
127        if (ret==NULL) {
128                fprintf(stderr,"Out of memory");
129                exit(EXIT_FAILURE);
130        }
131        xstrncpy(ret,src,n);
132        return ret;
133}
134
135
136/* call all the constructors if they haven't yet all been called */
137static void trace_init(void)
138{
139        if (!formats_list) {
140                duck_constructor();
141                erf_constructor();
142                tsh_constructor();
143                legacy_constructor();
144                atmhdr_constructor();
145                linuxnative_constructor();
146#ifdef HAVE_LIBPCAP
147                pcap_constructor();
148#endif
149                bpf_constructor();
150                pcapfile_constructor();
151                rt_constructor();
152#ifdef HAVE_DAG
153                dag_constructor();
154#endif
155#ifdef HAVE_DPDK
156        dpdk_constructor();
157#endif
158        }
159}
160
161/* Prints help information for libtrace
162 *
163 * Function prints out some basic help information regarding libtrace,
164 * and then prints out the help() function registered with each input module
165 */
166DLLEXPORT void trace_help(void) {
167        struct libtrace_format_t *tmp;
168        trace_init();
169        printf("libtrace %s\n\n",PACKAGE_VERSION);
170        printf("Following this are a list of the format modules supported in this build of libtrace\n\n");
171        for(tmp=formats_list;tmp;tmp=tmp->next) {
172                if (tmp->help)
173                        tmp->help();
174        }
175}
176
177#define URI_PROTO_LINE 16U
178
179/* Try to guess which format module is appropriate for a given trace file or
180 * device */
181static void guess_format(libtrace_t *libtrace, const char *filename)
182{
183        struct libtrace_format_t *tmp;
184       
185        /* Try and guess based on filename */
186        for(tmp = formats_list; tmp; tmp=tmp->next) {
187                if (tmp->probe_filename && tmp->probe_filename(filename)) {
188                        libtrace->format = tmp;
189                        libtrace->uridata = strdup(filename);
190                        return;
191                }
192        }
193
194        libtrace->io = wandio_create(filename);
195        if (!libtrace->io)
196                return;
197
198        /* Try and guess based on file magic */
199        for(tmp = formats_list; tmp; tmp=tmp->next) {
200                if (tmp->probe_magic && tmp->probe_magic(libtrace->io)) {
201                        libtrace->format = tmp;
202                        libtrace->uridata = strdup(filename);
203                        return;
204                }
205        }
206       
207        /* Oh well */
208        return;
209}
210
211/* Creates an input trace from a URI
212 *
213 * @params char * containing a valid libtrace URI
214 * @returns opaque pointer to a libtrace_t
215 *
216 * Some valid URI's are:
217 *  erf:/path/to/erf/file
218 *  erf:/path/to/erf/file.gz
219 *  erf:-                       (stdin)
220 *  dag:/dev/dagcard
221 *  pcapint:pcapinterface               (eg: pcapint:eth0)
222 *  pcapfile:/path/to/pcap/file
223 *  pcapfile:-
224 *  int:interface                       (eg: int:eth0) only on Linux
225 *  rt:hostname
226 *  rt:hostname:port
227 *
228 * If an error occured when attempting to open a trace, NULL is returned
229 * and an error is output to stdout.
230 */
231DLLEXPORT libtrace_t *trace_create(const char *uri) {
232        libtrace_t *libtrace = 
233                        (libtrace_t *)malloc(sizeof(libtrace_t));
234        char *scan = 0;
235        const char *uridata = 0;                 
236
237        trace_init();
238
239        assert(uri && "Passing NULL to trace_create makes me a very sad program");
240
241        if (!libtrace) {
242                /* Out of memory */
243                return NULL;
244        }
245       
246        libtrace->err.err_num = TRACE_ERR_NOERROR;
247        libtrace->format=NULL;
248       
249        libtrace->event.tdelta = 0.0;
250        libtrace->event.packet = NULL;
251        libtrace->event.psize = 0;
252        libtrace->event.trace_last_ts = 0.0;
253        libtrace->event.waiting = false;
254        libtrace->filter = NULL;
255        libtrace->snaplen = 0;
256        libtrace->started=false;
257        libtrace->uridata = NULL;
258        libtrace->io = NULL;
259        libtrace->filtered_packets = 0;
260        libtrace->accepted_packets = 0;
261       
262        /* Parallel inits */
263        ASSERT_RET(pthread_mutex_init(&libtrace->libtrace_lock, NULL), == 0);
264        ASSERT_RET(pthread_cond_init(&libtrace->perpkt_cond, NULL), == 0);
265        libtrace->state = STATE_NEW;
266        libtrace->perpkt_queue_full = false;
267        libtrace->global_blob = NULL;
268        libtrace->per_pkt = NULL;
269        libtrace->reporter = NULL;
270        libtrace->hasher = NULL;
271        libtrace_zero_ocache(&libtrace->packet_freelist);
272        libtrace_zero_thread(&libtrace->hasher_thread);
273        libtrace_zero_thread(&libtrace->reporter_thread);
274        libtrace_zero_thread(&libtrace->keepalive_thread);
275        libtrace->reporter_thread.type = THREAD_EMPTY;
276        libtrace->perpkt_thread_count = 0;
277        libtrace->perpkt_threads = NULL;
278        libtrace->tracetime = 0;
279        libtrace->first_packets.first = 0;
280        libtrace->first_packets.count = 0;
281        libtrace->first_packets.packets = NULL;
282        libtrace->dropped_packets = UINT64_MAX;
283        libtrace->received_packets = UINT64_MAX;
284        ZERO_USER_CONFIG(libtrace->config);
285
286        /* Parse the URI to determine what sort of trace we are dealing with */
287        if ((uridata = trace_parse_uri(uri, &scan)) == 0) {
288                /* Could not parse the URI nicely */
289                guess_format(libtrace,uri);
290                if (!libtrace->format) {
291                        trace_set_err(libtrace,TRACE_ERR_BAD_FORMAT,"Unable to guess format (%s)",uri);
292                        return libtrace;
293                }
294        }
295        else {
296                struct libtrace_format_t *tmp;
297
298                /* Find a format that matches the first part of the URI */
299                for (tmp=formats_list;tmp;tmp=tmp->next) {
300                        if (strlen(scan) == strlen(tmp->name) &&
301                                        strncasecmp(scan, tmp->name, strlen(scan)) == 0
302                                        ) {
303                                libtrace->format=tmp;
304                                break;
305                        }
306                }
307
308                if (libtrace->format == 0) {
309                        trace_set_err(libtrace, TRACE_ERR_BAD_FORMAT,
310                                        "Unknown format (%s)",scan);
311                        return libtrace;
312                }
313
314                libtrace->uridata = strdup(uridata);
315        }
316        /* libtrace->format now contains the type of uri
317         * libtrace->uridata contains the appropriate data for this
318         */
319       
320        /* Call the init_input function for the matching capture format */ 
321        if (libtrace->format->init_input) {
322                int err=libtrace->format->init_input(libtrace);
323                assert (err==-1 || err==0);
324                if (err==-1) {
325                        /* init_input should call trace_set_err to set
326                         * the error message
327                         */
328                        return libtrace;
329                }
330        } else {
331                trace_set_err(libtrace,TRACE_ERR_UNSUPPORTED,
332                                "Format does not support input (%s)",scan);
333                return libtrace;
334        }
335       
336        if (scan)
337                free(scan);
338        libtrace->err.err_num=TRACE_ERR_NOERROR;
339        libtrace->err.problem[0]='\0';
340        return libtrace;
341}
342
343/* Creates a "dummy" trace file that has only the format type set.
344 *
345 * @returns opaque pointer to a (sparsely initialised) libtrace_t
346 *
347 * IMPORTANT: Do not attempt to call trace_read_packet or other such functions
348 * with the dummy trace. Its intended purpose is to act as a packet->trace for
349 * libtrace_packet_t's that are not associated with a libtrace_t structure.
350 */
351DLLEXPORT libtrace_t * trace_create_dead (const char *uri) {
352        libtrace_t *libtrace = (libtrace_t *) malloc(sizeof(libtrace_t));
353        char *scan = (char *)calloc(sizeof(char),URI_PROTO_LINE);
354        char *uridata;
355        struct libtrace_format_t *tmp;
356
357        trace_init();
358       
359        libtrace->err.err_num = TRACE_ERR_NOERROR;
360
361        if((uridata = strchr(uri,':')) == NULL) {
362                xstrncpy(scan, uri, strlen(uri));
363        } else {
364                xstrncpy(scan,uri, (size_t)(uridata - uri));
365        }
366       
367        libtrace->err.err_num = TRACE_ERR_NOERROR;
368        libtrace->format=NULL;
369       
370        libtrace->event.tdelta = 0.0;
371        libtrace->event.packet = NULL;
372        libtrace->event.psize = 0;
373        libtrace->event.trace_last_ts = 0.0;
374        libtrace->filter = NULL;
375        libtrace->snaplen = 0;
376        libtrace->started=false;
377        libtrace->uridata = NULL;
378        libtrace->io = NULL;
379        libtrace->filtered_packets = 0;
380       
381        /* Parallel inits */
382        ASSERT_RET(pthread_mutex_init(&libtrace->libtrace_lock, NULL), == 0);
383        ASSERT_RET(pthread_cond_init(&libtrace->perpkt_cond, NULL), == 0);
384        libtrace->state = STATE_NEW; // TODO MAYBE DEAD
385        libtrace->perpkt_queue_full = false;
386        libtrace->global_blob = NULL;
387        libtrace->per_pkt = NULL;
388        libtrace->reporter = NULL;
389        libtrace->hasher = NULL;
390        libtrace_zero_ocache(&libtrace->packet_freelist);
391        libtrace_zero_thread(&libtrace->hasher_thread);
392        libtrace_zero_thread(&libtrace->reporter_thread);
393        libtrace_zero_thread(&libtrace->keepalive_thread);
394        libtrace->reporter_thread.type = THREAD_EMPTY;
395        libtrace->perpkt_thread_count = 0;
396        libtrace->perpkt_threads = NULL;
397        libtrace->tracetime = 0;
398        ZERO_USER_CONFIG(libtrace->config);
399       
400        for(tmp=formats_list;tmp;tmp=tmp->next) {
401                if (strlen(scan) == strlen(tmp->name) &&
402                                !strncasecmp(scan,
403                                        tmp->name,
404                                        strlen(scan))) {
405                                libtrace->format=tmp;
406                                break;
407                                }
408        }
409        if (libtrace->format == 0) {
410                trace_set_err(libtrace,TRACE_ERR_BAD_FORMAT,
411                                "Unknown format (%s)",scan);
412        }
413
414        libtrace->format_data = NULL;
415        free(scan);
416        return libtrace;
417
418}
419
420/* Creates an output trace from a URI.
421 *
422 * @param uri   the uri string describing the output format and destination
423 * @returns opaque pointer to a libtrace_output_t
424 *
425 *  If an error occured when attempting to open the output trace, NULL is
426 *  returned and trace_errno is set.
427 */
428       
429DLLEXPORT libtrace_out_t *trace_create_output(const char *uri) {
430        libtrace_out_t *libtrace = 
431                        (libtrace_out_t*)malloc(sizeof(libtrace_out_t));
432       
433        char *scan = 0;
434        const char *uridata = 0;
435        struct libtrace_format_t *tmp;
436
437        trace_init();
438
439        libtrace->err.err_num = TRACE_ERR_NOERROR;
440        strcpy(libtrace->err.problem,"Error message set\n");
441        libtrace->format = NULL;
442        libtrace->uridata = NULL;
443       
444        /* Parse the URI to determine what capture format we want to write */
445
446        if ((uridata = trace_parse_uri(uri, &scan)) == 0) {
447                trace_set_err_out(libtrace,TRACE_ERR_BAD_FORMAT,
448                                "Bad uri format (%s)",uri);
449                return libtrace;
450        }
451       
452        /* Attempt to find the format in the list of supported formats */
453        for(tmp=formats_list;tmp;tmp=tmp->next) {
454                if (strlen(scan) == strlen(tmp->name) &&
455                                !strncasecmp(scan,
456                                        tmp->name,
457                                        strlen(scan))) {
458                                libtrace->format=tmp;
459                                break;
460                                }
461        }
462        free(scan);
463
464        if (libtrace->format == NULL) {
465                trace_set_err_out(libtrace,TRACE_ERR_BAD_FORMAT,
466                                "Unknown output format (%s)",scan);
467                return libtrace;
468        }
469        libtrace->uridata = strdup(uridata);
470
471        /* libtrace->format now contains the type of uri
472         * libtrace->uridata contains the appropriate data for this
473         */
474
475        if (libtrace->format->init_output) {
476                /* 0 on success, -1 on failure */
477                switch(libtrace->format->init_output(libtrace)) {
478                        case -1: /* failure */
479                                return libtrace;
480                        case 0: /* success */
481                                break;
482                        default:
483                                assert(!"Internal error: init_output() should return -1 for failure, or 0 for success");
484                }
485        } else {
486                trace_set_err_out(libtrace,TRACE_ERR_UNSUPPORTED,
487                                "Format does not support writing (%s)",scan);
488                return libtrace;
489        }
490
491
492        libtrace->started=false;
493        return libtrace;
494}
495
496/* Start an input trace
497 * @param libtrace      the input trace to start
498 * @returns 0 on success
499 *
500 * This does the work associated with actually starting up
501 * the trace.  it may fail.
502 */
503DLLEXPORT int trace_start(libtrace_t *libtrace)
504{
505        assert(libtrace);
506        if (trace_is_err(libtrace))
507                return -1;
508        if (libtrace->format->start_input) {
509                int ret=libtrace->format->start_input(libtrace);
510                if (ret < 0) {
511                        return ret;
512                }
513        }
514
515        libtrace->started=true;
516        return 0;
517}
518
519/* Start an output trace */
520DLLEXPORT int trace_start_output(libtrace_out_t *libtrace) 
521{
522        assert(libtrace);
523        if (libtrace->format->start_output) {
524                int ret=libtrace->format->start_output(libtrace);
525                if (ret < 0) {
526                        return ret;
527                }
528        }
529
530        libtrace->started=true;
531        return 0;
532}
533
534DLLEXPORT int trace_pause(libtrace_t *libtrace)
535{
536        assert(libtrace);
537        if (!libtrace->started) {
538                trace_set_err(libtrace,TRACE_ERR_BAD_STATE, "You must call trace_start() before calling trace_pause()");
539                return -1;
540        }
541        if (libtrace->format->pause_input)
542                libtrace->format->pause_input(libtrace);
543        libtrace->started=false;
544        return 0;
545}
546
547DLLEXPORT int trace_config(libtrace_t *libtrace,
548                trace_option_t option,
549                void *value)
550{
551        int ret;
552
553        if (trace_is_err(libtrace)) {
554                return -1;
555        }
556       
557        /* If the capture format supports configuration, try using their
558         * native configuration first */
559        if (libtrace->format->config_input) {
560                ret=libtrace->format->config_input(libtrace,option,value);
561                if (ret==0)
562                        return 0;
563        }
564
565        /* If we get here, either the native configuration failed or the
566         * format did not support configuration. However, libtrace can
567         * deal with some options itself, so give that a go */
568        switch(option) {
569                case TRACE_OPTION_SNAPLEN:
570                        /* Clear the error if there was one */
571                        if (trace_is_err(libtrace)) {
572                                trace_get_err(libtrace);
573                        }
574                        if (*(int*)value<0 
575                                || *(int*)value>LIBTRACE_PACKET_BUFSIZE) {
576                                trace_set_err(libtrace,TRACE_ERR_BAD_STATE,
577                                        "Invalid snap length");
578                        }
579                        libtrace->snaplen=*(int*)value;
580                        return 0;
581                case TRACE_OPTION_FILTER:
582                        /* Clear the error if there was one */
583                        if (trace_is_err(libtrace)) {
584                                trace_get_err(libtrace);
585                        }
586                        libtrace->filter=(libtrace_filter_t *)value;
587                        return 0;
588                case TRACE_OPTION_PROMISC:
589                        if (!trace_is_err(libtrace)) {
590                                trace_set_err(libtrace,TRACE_ERR_OPTION_UNAVAIL,
591                                                "Promisc mode is not supported by this format module");
592                        }
593                        return -1;
594                case TRACE_OPTION_META_FREQ:
595                        if (!trace_is_err(libtrace)) {
596                                trace_set_err(libtrace, 
597                                                TRACE_ERR_OPTION_UNAVAIL,
598                                                "This format does not support meta-data gathering");
599                        }
600                        return -1;
601                case TRACE_OPTION_EVENT_REALTIME:
602                        if (!trace_is_err(libtrace)) {
603                                trace_set_err(libtrace, 
604                                                TRACE_ERR_OPTION_UNAVAIL,
605                                                "This format does not support realtime events");
606                        }
607                        return -1;
608                       
609        }
610        if (!trace_is_err(libtrace)) {
611                trace_set_err(libtrace,TRACE_ERR_UNKNOWN_OPTION,
612                        "Unknown option %i", option);
613        }
614        return -1;
615}
616
617DLLEXPORT int trace_config_output(libtrace_out_t *libtrace, 
618                trace_option_output_t option,
619                void *value) {
620       
621        /* Unlike the input options, libtrace does not natively support any of
622         * the output options - the format module must be able to deal with
623         * them. */
624        if (libtrace->format->config_output) {
625                return libtrace->format->config_output(libtrace, option, value);
626        }
627        return -1;
628}
629
630/* Close an input trace file, freeing up any resources it may have been using
631 *
632 */
633DLLEXPORT void trace_destroy(libtrace_t *libtrace) {
634        int i;
635        assert(libtrace);
636
637        ASSERT_RET(pthread_mutex_destroy(&libtrace->libtrace_lock), == 0);
638        ASSERT_RET(pthread_cond_destroy(&libtrace->perpkt_cond), == 0);
639
640        /* destroy any packets that are still around */
641        if (libtrace->state != STATE_NEW && libtrace->first_packets.packets) {
642                for (i = 0; i < libtrace->perpkt_thread_count; ++i) {
643                        if(libtrace->first_packets.packets[i].packet) {
644                                trace_destroy_packet(libtrace->first_packets.packets[i].packet);
645                        }
646                }
647                free(libtrace->first_packets.packets);
648                ASSERT_RET(pthread_spin_destroy(&libtrace->first_packets.lock), == 0);
649        }
650
651        if (libtrace->format) {
652                if (libtrace->started && libtrace->format->pause_input)
653                        libtrace->format->pause_input(libtrace);
654                if (libtrace->format->fin_input)
655                        libtrace->format->fin_input(libtrace);
656        }
657        /* Need to free things! */
658        if (libtrace->uridata)
659                free(libtrace->uridata);
660       
661        /* Empty any packet memory */
662        if (libtrace->state != STATE_NEW) {
663                // This has all of our packets
664                libtrace_ocache_destroy(&libtrace->packet_freelist);
665                if (libtrace->combiner.destroy)
666                        libtrace->combiner.destroy(libtrace, &libtrace->combiner);
667                free(libtrace->perpkt_threads);
668                libtrace->perpkt_threads = NULL;
669                libtrace->perpkt_thread_count = 0;
670        }
671       
672        if (libtrace->event.packet) {
673                /* Don't use trace_destroy_packet here - there is almost
674                 * certainly going to be another libtrace_packet_t that is
675                 * pointing to the buffer for this packet, so we don't want
676                 * to free it. Rather, it will get freed when the user calls
677                 * trace_destroy_packet on the libtrace_packet_t that they
678                 * own.
679                 *
680                 * All we need to do then is free our packet structure itself.
681                 */
682                 free(libtrace->event.packet);
683        }
684        free(libtrace);
685}
686
687
688DLLEXPORT void trace_destroy_dead(libtrace_t *libtrace) {
689        assert(libtrace);
690
691        ASSERT_RET(pthread_mutex_destroy(&libtrace->libtrace_lock), == 0);
692        ASSERT_RET(pthread_cond_destroy(&libtrace->perpkt_cond), == 0);
693
694        /* Don't call pause_input or fin_input, because we should never have
695         * used this trace to do any reading anyway. Do make sure we free
696         * any format_data that has been created, though. */
697        if (libtrace->format_data)
698                free(libtrace->format_data);
699        free(libtrace);
700}
701/* Close an output trace file, freeing up any resources it may have been using
702 *
703 * @param libtrace      the output trace file to be destroyed
704 */
705DLLEXPORT void trace_destroy_output(libtrace_out_t *libtrace) 
706{
707        assert(libtrace);
708        if (libtrace->format && libtrace->format->fin_output)
709                libtrace->format->fin_output(libtrace);
710        if (libtrace->uridata)
711                free(libtrace->uridata);
712        free(libtrace);
713}
714
715DLLEXPORT libtrace_packet_t *trace_create_packet(void)
716{
717        libtrace_packet_t *packet =
718                (libtrace_packet_t*)calloc((size_t)1,sizeof(libtrace_packet_t));
719
720        packet->buf_control=TRACE_CTRL_PACKET;
721        trace_clear_cache(packet);
722        return packet;
723}
724
725DLLEXPORT libtrace_packet_t *trace_copy_packet(const libtrace_packet_t *packet) {
726        libtrace_packet_t *dest = 
727                (libtrace_packet_t *)malloc(sizeof(libtrace_packet_t));
728        if (!dest) {
729                printf("Out of memory constructing packet\n");
730                abort();
731        }
732        dest->trace=packet->trace;
733        dest->buffer=malloc(65536);
734        if (!dest->buffer) {
735                printf("Out of memory allocating buffer memory\n");
736                abort();
737        }
738        dest->header=dest->buffer;
739        dest->payload=(void*)
740                ((char*)dest->buffer+trace_get_framing_length(packet));
741        dest->type=packet->type;
742        dest->buf_control=TRACE_CTRL_PACKET;
743        dest->order = packet->order;
744        dest->hash = packet->hash;
745        dest->error = packet->error;
746        /* Reset the cache - better to recalculate than try to convert
747         * the values over to the new packet */
748        trace_clear_cache(dest);       
749        /* Ooooh nasty memcpys! This is why we want to avoid copying packets
750         * as much as possible */
751        memcpy(dest->header,packet->header,trace_get_framing_length(packet));
752        memcpy(dest->payload,packet->payload,trace_get_capture_length(packet));
753
754        return dest;
755}
756
757/** Destroy a packet object
758 */
759DLLEXPORT void trace_destroy_packet(libtrace_packet_t *packet) {
760        /* Free any resources possibly associated with the packet */
761        if (libtrace_parallel && packet->trace && packet->trace->format->fin_packet) {
762                packet->trace->format->fin_packet(packet);
763        }
764       
765        if (packet->buf_control == TRACE_CTRL_PACKET && packet->buffer) {
766                free(packet->buffer);
767        }
768        packet->buf_control=(buf_control_t)'\0'; 
769                                /* A "bad" value to force an assert
770                                 * if this packet is ever reused
771                                 */
772        free(packet);
773}
774
775/**
776 * Removes any possible data stored againt the trace and releases any data.
777 * This will not destroy a reusable good malloc'd buffer (TRACE_CTRL_PACKET)
778 * use trace_destroy_packet() for those diabolical purposes.
779 */
780void trace_fin_packet(libtrace_packet_t *packet) {
781        if (packet)
782        {
783                if (packet->trace && packet->trace->format->fin_packet) {
784                        packet->trace->format->fin_packet(packet);
785                        //gettimeofday(&tv, NULL);
786                        //printf ("%d.%06d DESTROYED #%"PRIu64"\n", tv.tv_sec, tv.tv_usec, trace_packet_get(packet));
787                }
788
789                // No matter what we remove the header and link pointers
790                packet->trace = NULL;
791                packet->header = NULL;
792                packet->payload = NULL;
793
794                if (packet->buf_control != TRACE_CTRL_PACKET)
795                {
796                        packet->buffer = NULL;
797                }
798
799                trace_clear_cache(packet);
800                packet->hash = 0;
801                packet->order = 0;
802        }
803}
804
805/* Read one packet from the trace into buffer. Note that this function will
806 * block until a packet is read (or EOF is reached).
807 *
808 * @param libtrace      the libtrace opaque pointer
809 * @param packet        the packet opaque pointer
810 * @returns 0 on EOF, negative value on error
811 *
812 */
813DLLEXPORT int trace_read_packet(libtrace_t *libtrace, libtrace_packet_t *packet) {
814
815        assert(libtrace && "You called trace_read_packet() with a NULL libtrace parameter!\n");
816        if (trace_is_err(libtrace))
817                return -1;
818        if (!libtrace->started) {
819                trace_set_err(libtrace,TRACE_ERR_BAD_STATE,"You must call libtrace_start() before trace_read_packet()\n");
820                return -1;
821        }
822        if (!(packet->buf_control==TRACE_CTRL_PACKET || packet->buf_control==TRACE_CTRL_EXTERNAL)) {
823                trace_set_err(libtrace,TRACE_ERR_BAD_STATE,"Packet passed to trace_read_packet() is invalid\n");
824                return -1;
825        }
826        assert(packet);
827
828        if (libtrace->format->read_packet) {
829                do {
830                        size_t ret;
831                        int filtret;
832                        /* Finalise the packet, freeing any resources the format module
833                         * may have allocated it and zeroing all data associated with it.
834                         */
835                        trace_fin_packet(packet);
836                        /* Store the trace we are reading from into the packet opaque
837                         * structure */
838                        packet->trace = libtrace;
839                        ret=libtrace->format->read_packet(libtrace,packet);
840                        if (ret==(size_t)-1 || ret==0) {
841                                return ret;
842                        }
843                        if (libtrace->filter) {
844                                /* If the filter doesn't match, read another
845                                 * packet
846                                 */
847                                filtret = trace_apply_filter(libtrace->filter, packet);
848                                if (filtret == -1) {
849                                        /* Error compiling filter, probably */
850                                        return ~0U;
851                                }
852                               
853                                if (filtret == 0) {
854                                        ++libtrace->filtered_packets;
855                                        continue;
856                                }
857                        }
858                        if (libtrace->snaplen>0) {
859                                /* Snap the packet */
860                                trace_set_capture_length(packet,
861                                                libtrace->snaplen);
862                        }
863                        trace_packet_set_order(packet, libtrace->accepted_packets);
864                        ++libtrace->accepted_packets;
865                        return ret;
866                } while(1);
867        }
868        trace_set_err(libtrace,TRACE_ERR_UNSUPPORTED,"This format does not support reading packets\n");
869        return ~0U;
870}
871
872/* Converts the provided buffer into a libtrace packet of the given type.
873 *
874 * Unlike trace_construct_packet, the buffer is expected to begin with the
875 * appropriate capture format header for the format type that the packet is
876 * being converted to. This also allows for a packet to be converted into
877 * just about capture format that is supported by libtrace, provided the
878 * format header is present in the buffer.
879 *
880 * This function is primarily used to convert packets received via the RT
881 * protocol back into their original capture format. The RT header encapsulates
882 * the original capture format header, so after removing it the packet must
883 * have it's header and payload pointers updated and the packet format and type
884 * changed, amongst other things.
885 *
886 * Intended only for internal use at this point - this function is not
887 * available through the external libtrace API.
888 */
889int trace_prepare_packet(libtrace_t *trace, libtrace_packet_t *packet,
890                void *buffer, libtrace_rt_types_t rt_type, uint32_t flags) {
891
892        assert(packet);
893        assert(trace);
894       
895        /* XXX Proper error handling?? */
896        if (buffer == NULL)
897                return -1;
898
899        if (!(packet->buf_control==TRACE_CTRL_PACKET || packet->buf_control==TRACE_CTRL_EXTERNAL)) {
900                trace_set_err(trace,TRACE_ERR_BAD_STATE,"Packet passed to trace_read_packet() is invalid\n");
901                return -1;
902        }
903       
904        packet->trace = trace;
905       
906        /* Clear packet cache */
907        trace_clear_cache(packet);
908
909        if (trace->format->prepare_packet) {
910                return trace->format->prepare_packet(trace, packet,
911                                buffer, rt_type, flags);
912        }
913        trace_set_err(trace, TRACE_ERR_UNSUPPORTED, 
914                        "This format does not support preparing packets\n");
915        return -1;
916
917}
918
919/* Writes a packet to the specified output trace
920 *
921 * @param libtrace      describes the output format, destination, etc.
922 * @param packet        the packet to be written out
923 * @returns the number of bytes written, -1 if write failed
924 */
925DLLEXPORT int trace_write_packet(libtrace_out_t *libtrace, libtrace_packet_t *packet) {
926        assert(libtrace);
927        assert(packet);
928        /* Verify the packet is valid */
929        if (!libtrace->started) {
930                trace_set_err_out(libtrace,TRACE_ERR_BAD_STATE,
931                        "Trace is not started before trace_write_packet");
932                return -1;
933        }
934
935        if (libtrace->format->write_packet) {
936                return libtrace->format->write_packet(libtrace, packet);
937        }
938        trace_set_err_out(libtrace,TRACE_ERR_UNSUPPORTED,
939                "This format does not support writing packets");
940        return -1;
941}
942
943/* Get a pointer to the first byte of the packet payload */
944DLLEXPORT void *trace_get_packet_buffer(const libtrace_packet_t *packet,
945                libtrace_linktype_t *linktype, uint32_t *remaining) {
946        int cap_len;
947        int wire_len;
948
949        assert(packet != NULL);
950        if (linktype) *linktype = trace_get_link_type(packet);
951        if (remaining) {
952                /* I think we should choose the minimum of the capture and
953                 * wire lengths to be the "remaining" value. If the packet has
954                 * been padded to increase the capture length, we don't want
955                 * to allow subsequent protocol decoders to consider the
956                 * padding as part of the packet.
957                 *
958                 * For example, in Auck 4 there is a trace where the IP header
959                 * length is incorrect (24 bytes) followed by a 20 byte TCP
960                 * header. Total IP length is 40 bytes. As a result, the
961                 * legacyatm padding gets treated as the "missing" bytes of
962                 * the TCP header, which isn't the greatest. We're probably
963                 * better off returning an incomplete TCP header in that case.
964                 */
965               
966                cap_len = trace_get_capture_length(packet);
967                wire_len = trace_get_wire_length(packet);
968
969                assert(cap_len >= 0);
970
971                /* There is the odd corrupt packet, e.g. in IPLS II, that have
972                 * massively negative wire lens. We could assert fail here on
973                 * them, but we could at least try the capture length instead.
974                 *
975                 * You may still run into problems if you try to write that
976                 * packet, but at least reading should work OK.
977                 */
978                if (wire_len < 0)
979                        *remaining = cap_len;
980                else if (wire_len < cap_len)
981                        *remaining = wire_len;
982                else
983                        *remaining = cap_len;
984                /* *remaining = trace_get_capture_length(packet); */
985        }
986        return (void *) packet->payload;
987}
988
989
990/* Get a pointer to the first byte of the packet payload
991 *
992 * DEPRECATED - use trace_get_packet_buffer() instead */
993DLLEXPORT void *trace_get_link(const libtrace_packet_t *packet) {
994        return (void *)packet->payload;
995}
996
997/* Get the current time in DAG time format
998 * @param packet        a pointer to a libtrace_packet structure
999 * @returns a 64 bit timestamp in DAG ERF format (upper 32 bits are the seconds
1000 * past 1970-01-01, the lower 32bits are partial seconds)
1001 */ 
1002DLLEXPORT uint64_t trace_get_erf_timestamp(const libtrace_packet_t *packet) {
1003        if (packet->trace->format->get_erf_timestamp) {
1004                /* timestamp -> timestamp */
1005                return packet->trace->format->get_erf_timestamp(packet);
1006        } else if (packet->trace->format->get_timespec) {
1007                /* timespec -> timestamp */
1008                struct timespec ts;
1009                ts = packet->trace->format->get_timespec(packet);
1010                return ((((uint64_t)ts.tv_sec) << 32) +
1011                                (((uint64_t)ts.tv_nsec << 32)/1000000000));
1012        } else if (packet->trace->format->get_timeval) {
1013                /* timeval -> timestamp */
1014                struct timeval tv;
1015                tv = packet->trace->format->get_timeval(packet);
1016                return ((((uint64_t)tv.tv_sec) << 32) +
1017                                (((uint64_t)tv.tv_usec << 32)/1000000));
1018        } else if (packet->trace->format->get_seconds) {
1019                /* seconds -> timestamp */
1020                double seconds = packet->trace->format->get_seconds(packet);
1021                return (((uint64_t)seconds)<<32)
1022                          + (uint64_t)((seconds-(uint64_t)seconds)*UINT_MAX);
1023        }
1024        else {
1025                return (uint64_t)0;
1026        }
1027                     
1028}
1029
1030/* Get the current time in struct timeval
1031 * @param packet        a pointer to a libtrace_packet structure
1032 *
1033 * @returns time that this packet was seen in a struct timeval
1034 * @author Daniel Lawson
1035 * @author Perry Lorier
1036 */ 
1037DLLEXPORT struct timeval trace_get_timeval(const libtrace_packet_t *packet) {
1038        struct timeval tv;
1039        uint64_t ts = 0;
1040        if (packet->trace->format->get_timeval) {
1041                /* timeval -> timeval */
1042                tv = packet->trace->format->get_timeval(packet);
1043        } else if (packet->trace->format->get_erf_timestamp) {
1044                /* timestamp -> timeval */
1045                ts = packet->trace->format->get_erf_timestamp(packet);
1046                tv.tv_sec = ts >> 32;
1047                tv.tv_usec = ((ts&0xFFFFFFFF)*1000000)>>32;
1048                if (tv.tv_usec >= 1000000) {
1049                        tv.tv_usec -= 1000000;
1050                        tv.tv_sec += 1;
1051                }
1052        } else if (packet->trace->format->get_timespec) {
1053                struct timespec ts = packet->trace->format->get_timespec(packet);
1054                tv.tv_sec = ts.tv_sec;
1055                tv.tv_usec = ts.tv_nsec/1000;
1056        } else if (packet->trace->format->get_seconds) {
1057                /* seconds -> timeval */
1058                double seconds = packet->trace->format->get_seconds(packet);
1059                tv.tv_sec = (uint32_t)seconds;
1060                tv.tv_usec = (uint32_t)(((seconds - tv.tv_sec) * 1000000)/UINT_MAX);
1061        }
1062        else {
1063                tv.tv_sec=-1;
1064                tv.tv_usec=-1;
1065        }
1066
1067    return tv;
1068}
1069
1070DLLEXPORT struct timespec trace_get_timespec(const libtrace_packet_t *packet) {
1071        struct timespec ts;
1072
1073        if (packet->trace->format->get_timespec) {
1074                return packet->trace->format->get_timespec(packet);
1075        } else if (packet->trace->format->get_erf_timestamp) {
1076                /* timestamp -> timeval */
1077                uint64_t erfts = packet->trace->format->get_erf_timestamp(packet);
1078                ts.tv_sec = erfts >> 32;
1079                ts.tv_nsec = ((erfts&0xFFFFFFFF)*1000000000)>>32;
1080                if (ts.tv_nsec >= 1000000000) {
1081                        ts.tv_nsec -= 1000000000;
1082                        ts.tv_sec += 1;
1083                }
1084                return ts;
1085        } else if (packet->trace->format->get_timeval) {
1086                /* timeval -> timespec */
1087                struct timeval tv = packet->trace->format->get_timeval(packet);
1088                ts.tv_sec = tv.tv_sec;
1089                ts.tv_nsec = tv.tv_usec*1000;
1090                return ts;
1091        } else if (packet->trace->format->get_seconds) {
1092                /* seconds -> timespec */
1093                double seconds = packet->trace->format->get_seconds(packet);
1094                ts.tv_sec = (uint32_t)seconds;
1095                ts.tv_nsec = (long)(((seconds - ts.tv_sec) * 1000000000)/UINT_MAX);
1096                return ts;
1097        }
1098        else {
1099                ts.tv_sec=-1;
1100                ts.tv_nsec=-1;
1101                return ts;
1102        }
1103}
1104
1105
1106/* Get the current time in floating point seconds
1107 * @param packet        a pointer to a libtrace_packet structure
1108 * @returns time that this packet was seen in 64bit floating point seconds
1109 */ 
1110DLLEXPORT double trace_get_seconds(const libtrace_packet_t *packet) {
1111        double seconds = 0.0;
1112
1113        if (packet->trace->format->get_seconds) {
1114                /* seconds->seconds */
1115                seconds = packet->trace->format->get_seconds(packet);
1116        } else if (packet->trace->format->get_erf_timestamp) {
1117                /* timestamp -> seconds */
1118                uint64_t ts = 0;
1119                ts = packet->trace->format->get_erf_timestamp(packet);
1120                seconds =  (ts>>32) + ((ts & UINT_MAX)*1.0 / UINT_MAX);
1121        } else if (packet->trace->format->get_timespec) {
1122                /* timespec -> seconds */
1123                struct timespec ts;
1124                ts = packet->trace->format->get_timespec(packet);
1125                seconds = ts.tv_sec + ((ts.tv_nsec * 1.0) / 1000000000);
1126        } else if (packet->trace->format->get_timeval) {
1127                /* timeval -> seconds */
1128                struct timeval tv;
1129                tv = packet->trace->format->get_timeval(packet);
1130                seconds = tv.tv_sec + ((tv.tv_usec * 1.0) / 1000000);
1131        }
1132
1133        return seconds;
1134}
1135
1136DLLEXPORT size_t trace_get_capture_length(const libtrace_packet_t *packet) 
1137{
1138        /* Cache the capture length */
1139        if (packet->capture_length == -1) {
1140                if (!packet->trace->format->get_capture_length)
1141                        return ~0U;
1142                /* Cast away constness because this is "just" a cache */
1143                ((libtrace_packet_t*)packet)->capture_length = 
1144                        packet->trace->format->get_capture_length(packet);
1145        }
1146
1147        assert(packet->capture_length < LIBTRACE_PACKET_BUFSIZE);
1148
1149        return packet->capture_length;
1150}
1151       
1152/* Get the size of the packet as it was seen on the wire.
1153 * @param packet        a pointer to a libtrace_packet structure
1154 *
1155 * @returns the size of the packet as it was on the wire.
1156 * @note Due to the trace being a header capture, or anonymisation this may
1157 * not be the same as the Capture Len.
1158 */ 
1159DLLEXPORT size_t trace_get_wire_length(const libtrace_packet_t *packet){
1160       
1161        if (packet->wire_length == -1) {
1162                if (!packet->trace->format->get_wire_length) 
1163                        return ~0U;
1164                ((libtrace_packet_t *)packet)->wire_length = 
1165                        packet->trace->format->get_wire_length(packet);
1166        }
1167
1168        assert(packet->wire_length < LIBTRACE_PACKET_BUFSIZE);
1169        return packet->wire_length;
1170
1171}
1172
1173/* Get the length of the capture framing headers.
1174 * @param packet        the packet opaque pointer
1175 * @returns the size of the packet as it was on the wire.
1176 * @note this length corresponds to the difference between the size of a
1177 * captured packet in memory, and the captured length of the packet
1178 */ 
1179DLLEXPORT SIMPLE_FUNCTION
1180size_t trace_get_framing_length(const libtrace_packet_t *packet) {
1181        if (packet->trace->format->get_framing_length) {
1182                return packet->trace->format->get_framing_length(packet);
1183        }
1184        return ~0U;
1185}
1186
1187
1188/* Get the type of the link layer
1189 * @param packet        a pointer to a libtrace_packet structure
1190 * @returns libtrace_linktype_t
1191 */
1192DLLEXPORT libtrace_linktype_t trace_get_link_type(const libtrace_packet_t *packet ) {
1193
1194        if (packet->link_type == 0) {
1195                if (!packet->trace->format->get_link_type)
1196                        return TRACE_TYPE_UNKNOWN;
1197                ((libtrace_packet_t *)packet)->link_type =
1198                        packet->trace->format->get_link_type(packet);
1199        }
1200
1201        return packet->link_type;
1202}
1203
1204/* process a libtrace event
1205 * @param trace the libtrace opaque pointer
1206 * @param packet the libtrace_packet opaque pointer
1207 * @returns
1208 *  TRACE_EVENT_IOWAIT  Waiting on I/O on fd
1209 *  TRACE_EVENT_SLEEP   Next event in seconds
1210 *  TRACE_EVENT_PACKET  Packet arrived in buffer with size size
1211 *  TRACE_EVENT_TERMINATE Trace terminated (perhaps with an error condition)
1212 * FIXME currently keeps a copy of the packet inside the trace pointer,
1213 * which in turn is stored inside the new packet object...
1214 */
1215DLLEXPORT libtrace_eventobj_t trace_event(libtrace_t *trace, 
1216                libtrace_packet_t *packet) {
1217        libtrace_eventobj_t event = {TRACE_EVENT_IOWAIT,0,0.0,0};
1218
1219        if (!trace) {
1220                fprintf(stderr,"You called trace_event() with a NULL trace object!\n");
1221        }
1222        assert(trace);
1223        assert(packet);
1224
1225        /* Clear the packet cache */
1226        trace_clear_cache(packet);
1227       
1228        /* Store the trace we are reading from into the packet opaque
1229         * structure */
1230        packet->trace = trace;
1231
1232        if (packet->trace->format->trace_event) {
1233                /* Note: incrementing accepted, filtered etc. packet
1234                 * counters is handled by the format-specific
1235                 * function so don't increment them here.
1236                 */
1237                event=packet->trace->format->trace_event(trace,packet);
1238                }
1239        return event;
1240
1241}
1242
1243/** Setup a BPF filter based on pre-compiled byte-code.
1244 * @param bf_insns      A pointer to the start of the byte-code
1245 * @param bf_len        The number of BPF instructions
1246 * @returns             an opaque pointer to a libtrace_filter_t object
1247 * @note                The supplied byte-code is not checked for correctness.
1248 * @author              Scott Raynel
1249 */
1250DLLEXPORT libtrace_filter_t *
1251trace_create_filter_from_bytecode(void *bf_insns, unsigned int bf_len)
1252{
1253#ifndef HAVE_BPF_FILTER
1254        fprintf(stderr, "This version of libtrace does not have BPF support\n");
1255        return NULL;
1256#else
1257        struct libtrace_filter_t *filter = (struct libtrace_filter_t *)
1258                malloc(sizeof(struct libtrace_filter_t));
1259        filter->filter.bf_insns = (struct bpf_insn *)
1260                malloc(sizeof(struct bpf_insn) * bf_len);
1261       
1262        memcpy(filter->filter.bf_insns, bf_insns,
1263                        bf_len * sizeof(struct bpf_insn));
1264       
1265        filter->filter.bf_len = bf_len;
1266        filter->filterstring = NULL;
1267        filter->jitfilter = NULL;
1268        /* "flag" indicates that the filter member is valid */
1269        filter->flag = 1; 
1270       
1271        return filter;
1272#endif
1273}
1274
1275/* Create a BPF filter
1276 * @param filterstring a char * containing the bpf filter string
1277 * @returns opaque pointer pointer to a libtrace_filter_t object
1278 */
1279DLLEXPORT libtrace_filter_t *trace_create_filter(const char *filterstring) {
1280#ifdef HAVE_BPF_FILTER
1281        libtrace_filter_t *filter = (libtrace_filter_t*)
1282                                malloc(sizeof(libtrace_filter_t));
1283        filter->filterstring = strdup(filterstring);
1284        filter->jitfilter = NULL;
1285        filter->flag = 0;
1286        return filter;
1287#else
1288        fprintf(stderr,"This version of libtrace does not have bpf filter support\n");
1289        return NULL;
1290#endif
1291}
1292
1293DLLEXPORT void trace_destroy_filter(libtrace_filter_t *filter)
1294{
1295#ifdef HAVE_BPF_FILTER
1296        free(filter->filterstring);
1297        if (filter->flag)
1298                pcap_freecode(&filter->filter);
1299#ifdef HAVE_LLVM
1300        if (filter->jitfilter) 
1301                destroy_program(filter->jitfilter);
1302#endif
1303        free(filter);
1304#else
1305
1306#endif
1307}
1308
1309/* Compile a bpf filter, now we know the link type for the trace that we're
1310 * applying it to.
1311 *
1312 * @internal
1313 *
1314 * @returns -1 on error, 0 on success
1315 */
1316static int trace_bpf_compile(libtrace_filter_t *filter,
1317                const libtrace_packet_t *packet,
1318                void *linkptr, 
1319                libtrace_linktype_t linktype    ) {
1320#ifdef HAVE_BPF_FILTER
1321        /* It just so happens that the underlying libs used by pthread arn't
1322         * thread safe, namely lex/flex thingys, so single threaded compile
1323         * multi threaded running should be safe.
1324         */
1325        static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
1326        assert(filter);
1327
1328        /* If this isn't a real packet, then fail */
1329        if (!linkptr) {
1330                trace_set_err(packet->trace,
1331                                TRACE_ERR_BAD_FILTER,"Packet has no payload");
1332                return -1;
1333        }
1334       
1335        if (filter->filterstring && ! filter->flag) {
1336                pcap_t *pcap = NULL;
1337                if (linktype==(libtrace_linktype_t)-1) {
1338                        trace_set_err(packet->trace,
1339                                        TRACE_ERR_BAD_FILTER,
1340                                        "Packet has an unknown linktype");
1341                        return -1;
1342                }
1343                if (libtrace_to_pcap_dlt(linktype) == TRACE_DLT_ERROR) {
1344                        trace_set_err(packet->trace,TRACE_ERR_BAD_FILTER,
1345                                        "Unknown pcap equivalent linktype");
1346                        return -1;
1347                }
1348                assert (pthread_mutex_lock(&mutex) == 0);
1349                /* Make sure not one bet us to this */
1350                if (filter->flag) {
1351                        printf("Someone bet us to compile the filter\n");
1352                        assert (pthread_mutex_unlock(&mutex) == 0);
1353                        return 1;
1354                }
1355                pcap=(pcap_t *)pcap_open_dead(
1356                                (int)libtrace_to_pcap_dlt(linktype),
1357                                1500U);
1358                /* build filter */
1359                assert(pcap);
1360                if (pcap_compile( pcap, &filter->filter, filter->filterstring, 
1361                                        1, 0)) {
1362                        trace_set_err(packet->trace,TRACE_ERR_BAD_FILTER,
1363                                        "Unable to compile the filter \"%s\": %s", 
1364                                        filter->filterstring,
1365                                        pcap_geterr(pcap));
1366                        pcap_close(pcap);
1367                        assert (pthread_mutex_unlock(&mutex) == 0);
1368                        return -1;
1369                }
1370                pcap_close(pcap);
1371                filter->flag=1;
1372                assert (pthread_mutex_unlock(&mutex) == 0);
1373        }
1374        return 0;
1375#else
1376        assert(!"Internal bug: This should never be called when BPF not enabled");
1377        trace_set_err(packet->trace,TRACE_ERR_OPTION_UNAVAIL,
1378                                "Feature unavailable");
1379        return -1;
1380#endif
1381}
1382
1383DLLEXPORT int trace_apply_filter(libtrace_filter_t *filter,
1384                        const libtrace_packet_t *packet) {
1385#ifdef HAVE_BPF_FILTER
1386        void *linkptr = 0;
1387        uint32_t clen = 0;
1388        bool free_packet_needed = false;
1389        int ret;
1390        libtrace_linktype_t linktype;
1391        libtrace_packet_t *packet_copy = (libtrace_packet_t*)packet;
1392#ifdef HAVE_LLVM
1393        static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
1394#endif
1395
1396        assert(filter);
1397        assert(packet);
1398
1399        /* Match all non-data packets as we probably want them to pass
1400         * through to the caller */
1401        linktype = trace_get_link_type(packet);
1402
1403        if (linktype == TRACE_TYPE_NONDATA)
1404                return 1;       
1405
1406        if (libtrace_to_pcap_dlt(linktype)==TRACE_DLT_ERROR) {
1407               
1408                /* If we cannot get a suitable DLT for the packet, it may
1409                 * be because the packet is encapsulated in a link type that
1410                 * does not correspond to a DLT. Therefore, we should try
1411                 * popping off headers until we either can find a suitable
1412                 * link type or we can't do any more sensible decapsulation. */
1413               
1414                /* Copy the packet, as we don't want to trash the one we
1415                 * were passed in */
1416                packet_copy=trace_copy_packet(packet);
1417                free_packet_needed=true;
1418
1419                while (libtrace_to_pcap_dlt(linktype) == TRACE_DLT_ERROR) {
1420                        if (!demote_packet(packet_copy)) {
1421                                trace_set_err(packet->trace, 
1422                                                TRACE_ERR_NO_CONVERSION,
1423                                                "pcap does not support this format");
1424                                if (free_packet_needed) {
1425                                        trace_destroy_packet(packet_copy);
1426                                }
1427                                return -1;
1428                        }
1429                        linktype = trace_get_link_type(packet_copy);
1430                }
1431
1432        }
1433       
1434        linkptr = trace_get_packet_buffer(packet_copy,NULL,&clen);
1435        if (!linkptr) {
1436                if (free_packet_needed) {
1437                        trace_destroy_packet(packet_copy);
1438                }
1439                return 0;
1440        }
1441
1442        /* We need to compile the filter now, because before we didn't know
1443         * what the link type was
1444         */
1445        // Note internal mutex locking used here
1446        if (trace_bpf_compile(filter,packet_copy,linkptr,linktype)==-1) {
1447                if (free_packet_needed) {
1448                        trace_destroy_packet(packet_copy);
1449                }
1450                return -1;
1451        }
1452
1453        /* If we're jitting, we may need to JIT the BPF code now too */
1454#if HAVE_LLVM
1455        if (!filter->jitfilter) {
1456                ASSERT_RET(pthread_mutex_lock(&mutex), == 0);
1457                /* Again double check here like the bpf filter */
1458                if(filter->jitfilter) 
1459                        printf("Someone bet us to compile the JIT thingy\n");
1460                else
1461                /* Looking at compile_program source this appears to be thread safe
1462                 * however if this gets called twice we will leak this memory :(
1463                 * as such lock here anyways */
1464                        filter->jitfilter = compile_program(filter->filter.bf_insns, filter->filter.bf_len);
1465                ASSERT_RET(pthread_mutex_unlock(&mutex), == 0);
1466        }
1467#endif
1468
1469        assert(filter->flag);
1470        /* Now execute the filter */
1471#if HAVE_LLVM
1472        ret=filter->jitfilter->bpf_run((unsigned char *)linkptr, clen);
1473#else
1474        ret=bpf_filter(filter->filter.bf_insns,(u_char*)linkptr,(unsigned int)clen,(unsigned int)clen);
1475#endif
1476
1477        /* If we copied the packet earlier, make sure that we free it */
1478        if (free_packet_needed) {
1479                trace_destroy_packet(packet_copy);
1480        }
1481        return ret;
1482#else
1483        fprintf(stderr,"This version of libtrace does not have bpf filter support\n");
1484        return 0;
1485#endif
1486}
1487
1488/* Set the direction flag, if it has one
1489 * @param packet the packet opaque pointer
1490 * @param direction the new direction (0,1,2,3)
1491 * @returns a signed value containing the direction flag, or -1 if this is not supported
1492 */
1493DLLEXPORT libtrace_direction_t trace_set_direction(libtrace_packet_t *packet, 
1494                libtrace_direction_t direction) 
1495{
1496        assert(packet);
1497        if (packet->trace->format->set_direction) {
1498                return packet->trace->format->set_direction(packet,direction);
1499        }
1500        return (libtrace_direction_t)~0U;
1501}
1502
1503/* Get the direction flag, if it has one
1504 * @param packet a pointer to a libtrace_packet structure
1505 * @returns a signed value containing the direction flag, or -1 if this is not supported
1506 * The direction is defined as 0 for packets originating locally (ie, outbound)
1507 * and 1 for packets originating remotely (ie, inbound).
1508 * Other values are possible, which might be overloaded to mean special things
1509 * for a special trace.
1510 */
1511DLLEXPORT libtrace_direction_t trace_get_direction(const libtrace_packet_t *packet) 
1512{
1513        assert(packet);
1514        if (packet->trace->format->get_direction) {
1515                return packet->trace->format->get_direction(packet);
1516        }
1517        return (libtrace_direction_t)~0U;
1518}
1519
1520#define ROOT_SERVER(x) ((x) < 512)
1521#define ROOT_CLIENT(x) ((512 <= (x)) && ((x) < 1024))
1522#define NONROOT_SERVER(x) ((x) >= 5000)
1523#define NONROOT_CLIENT(x) ((1024 <= (x)) && ((x) < 5000))
1524#define DYNAMIC(x) ((49152 < (x)) && ((x) < 65535))
1525#define SERVER(x) ROOT_SERVER(x) || NONROOT_SERVER(x)
1526#define CLIENT(x) ROOT_CLIENT(x) || NONROOT_CLIENT(x)
1527
1528/* Attempt to deduce the 'server' port
1529 * @param protocol the IP protocol (eg, 6 or 17 for TCP or UDP)
1530 * @param source the TCP or UDP source port
1531 * @param dest the TCP or UDP destination port
1532 * @returns a hint as to which port is the server port
1533 */
1534DLLEXPORT int8_t trace_get_server_port(UNUSED uint8_t protocol, 
1535                uint16_t source, uint16_t dest) 
1536{
1537        /*
1538         * * If the ports are equal, return DEST
1539         * * Check for well-known ports in the given protocol
1540         * * Root server ports: 0 - 511
1541         * * Root client ports: 512 - 1023
1542         * * non-root client ports: 1024 - 4999
1543         * * non-root server ports: 5000+
1544         * * Check for static ranges: 1024 - 49151
1545         * * Check for dynamic ranges: 49152 - 65535
1546         * * flip a coin.
1547         */
1548       
1549        /* equal */
1550        if (source == dest)
1551                return USE_DEST;
1552
1553        /* root server port, 0 - 511 */
1554        if (ROOT_SERVER(source) && ROOT_SERVER(dest)) {
1555                if (source < dest)
1556                        return USE_SOURCE;
1557                return USE_DEST;
1558        }
1559
1560        if (ROOT_SERVER(source) && !ROOT_SERVER(dest))
1561                return USE_SOURCE;
1562        if (!ROOT_SERVER(source) && ROOT_SERVER(dest))
1563                return USE_DEST;
1564
1565        /* non-root server */
1566        if (NONROOT_SERVER(source) && NONROOT_SERVER(dest)) {
1567                if (source < dest)
1568                        return USE_SOURCE;
1569                return USE_DEST;
1570        }
1571        if (NONROOT_SERVER(source) && !NONROOT_SERVER(dest))
1572                return USE_SOURCE;
1573        if (!NONROOT_SERVER(source) && NONROOT_SERVER(dest))
1574                return USE_DEST;
1575
1576        /* root client */
1577        if (ROOT_CLIENT(source) && ROOT_CLIENT(dest)) {
1578                if (source < dest)
1579                        return USE_SOURCE;
1580                return USE_DEST;
1581        }
1582        if (ROOT_CLIENT(source) && !ROOT_CLIENT(dest)) {
1583                /* prefer root-client over nonroot-client */
1584                if (NONROOT_CLIENT(dest))
1585                        return USE_SOURCE;
1586                return USE_DEST;
1587        }
1588        if (!ROOT_CLIENT(source) && ROOT_CLIENT(dest)) {
1589                /* prefer root-client over nonroot-client */
1590                if (NONROOT_CLIENT(source))
1591                        return USE_DEST;
1592                return USE_SOURCE;
1593        }
1594       
1595        /* nonroot client */
1596        if (NONROOT_CLIENT(source) && NONROOT_CLIENT(dest)) {
1597                if (source < dest) 
1598                        return USE_SOURCE;
1599                return USE_DEST;
1600        }
1601        if (NONROOT_CLIENT(source) && !NONROOT_CLIENT(dest))
1602                return USE_DEST;
1603        if (!NONROOT_CLIENT(source) && NONROOT_CLIENT(dest))
1604                return USE_SOURCE;
1605
1606        /* dynamic range */
1607        if (DYNAMIC(source) && DYNAMIC(dest)) {
1608                if (source < dest)
1609                        return USE_SOURCE;
1610                return USE_DEST;
1611        }
1612        if (DYNAMIC(source) && !DYNAMIC(dest))
1613                return USE_DEST;
1614        if (!DYNAMIC(source) && DYNAMIC(dest))
1615                return USE_SOURCE;
1616        /*
1617        if (SERVER(source) && CLIENT(dest))
1618                return USE_SOURCE;
1619       
1620        if (SERVER(dest) && CLIENT(source))
1621                return USE_DEST;
1622        if (ROOT_SERVER(source) && !ROOT_SERVER(dest))
1623                return USE_SOURCE;
1624        if (ROOT_SERVER(dest) && !ROOT_SERVER(source))
1625                return USE_DEST;
1626        */
1627        /* failing that test... */
1628        if (source < dest) {
1629                return USE_SOURCE;
1630        } 
1631        return USE_DEST;
1632       
1633}
1634
1635/* Truncate the packet at the suggested length
1636 * @param packet        the packet opaque pointer
1637 * @param size          the new length of the packet
1638 * @returns the new size of the packet
1639 * @note size and the return size refer to the network-level payload of the
1640 * packet, and do not include any capture headers. For example, to truncate a
1641 * packet after the IP header, set size to sizeof(ethernet_header) +
1642 * sizeof(ip_header)
1643 * @note If the original network-level payload is smaller than size, then the
1644 * original size is returned and the packet is left unchanged.
1645 */
1646DLLEXPORT size_t trace_set_capture_length(libtrace_packet_t *packet, size_t size) {
1647        assert(packet);
1648
1649        if (packet->trace->format->set_capture_length) {
1650                packet->capture_length = packet->trace->format->set_capture_length(packet,size);
1651                return packet->capture_length;
1652        }
1653
1654        return ~0U;
1655}
1656
1657/* Splits a URI into two components - the format component which is seen before
1658 * the ':', and the uridata which follows the ':'.
1659 *
1660 * Returns a pointer to the URI data, but updates the format parameter to
1661 * point to a copy of the format component.
1662 */
1663
1664DLLEXPORT const char * trace_parse_uri(const char *uri, char **format) {
1665        const char *uridata = 0;
1666       
1667        if((uridata = strchr(uri,':')) == NULL) {
1668                /* Badly formed URI - needs a : */
1669                return 0;
1670        }
1671
1672        if ((unsigned)(uridata - uri) > URI_PROTO_LINE) {
1673                /* Badly formed URI - uri type is too long */
1674                return 0;
1675        }
1676
1677        /* NOTE: this is allocated memory - it should be freed by the caller
1678         * once they are done with it */
1679        *format=xstrndup(uri, (size_t)(uridata - uri));
1680
1681        /* Push uridata past the delimiter */
1682        uridata++;
1683       
1684        return uridata;
1685}
1686
1687enum base_format_t trace_get_format(libtrace_packet_t *packet) 
1688{
1689        assert(packet);
1690
1691        return packet->trace->format->type;
1692}
1693       
1694DLLEXPORT libtrace_err_t trace_get_err(libtrace_t *trace)
1695{
1696        libtrace_err_t err = trace->err;
1697        trace->err.err_num = 0; /* "OK" */
1698        trace->err.problem[0]='\0';
1699        return err;
1700}
1701
1702DLLEXPORT bool trace_is_err(libtrace_t *trace)
1703{
1704        return trace->err.err_num != 0;
1705}
1706
1707/* Prints the input error status to standard error and clears the error state */
1708DLLEXPORT void trace_perror(libtrace_t *trace,const char *msg,...)
1709{
1710        char buf[256];
1711        va_list va;
1712        va_start(va,msg);
1713        vsnprintf(buf,sizeof(buf),msg,va);
1714        va_end(va);
1715        if(trace->err.err_num) {
1716                if (trace->uridata) {
1717                        fprintf(stderr,"%s(%s): %s\n",
1718                                        buf,trace->uridata,trace->err.problem);
1719                } else {
1720                        fprintf(stderr,"%s: %s\n", buf, trace->err.problem);
1721                }
1722        } else {
1723                if (trace->uridata) {
1724                        fprintf(stderr,"%s(%s): No error\n",buf,trace->uridata);
1725                } else {
1726                        fprintf(stderr,"%s: No error\n", buf);
1727                }
1728        }
1729        trace->err.err_num = 0; /* "OK" */
1730        trace->err.problem[0]='\0';
1731}
1732
1733DLLEXPORT libtrace_err_t trace_get_err_output(libtrace_out_t *trace)
1734{
1735        libtrace_err_t err = trace->err;
1736        trace->err.err_num = TRACE_ERR_NOERROR; /* "OK" */
1737        trace->err.problem[0]='\0';
1738        return err;
1739}
1740
1741DLLEXPORT bool trace_is_err_output(libtrace_out_t *trace)
1742{
1743        return trace->err.err_num != 0;
1744}
1745
1746/* Prints the output error status to standard error and clears the error state
1747 */
1748DLLEXPORT void trace_perror_output(libtrace_out_t *trace,const char *msg,...)
1749{
1750        char buf[256];
1751        va_list va;
1752        va_start(va,msg);
1753        vsnprintf(buf,sizeof(buf),msg,va);
1754        va_end(va);
1755        if(trace->err.err_num) {
1756                fprintf(stderr,"%s(%s): %s\n",
1757                                buf,
1758                                trace->uridata?trace->uridata:"no uri",
1759                                trace->err.problem);
1760        } else {
1761                fprintf(stderr,"%s(%s): No error\n",buf,trace->uridata);
1762        }
1763        trace->err.err_num = TRACE_ERR_NOERROR; /* "OK" */
1764        trace->err.problem[0]='\0';
1765}
1766
1767DLLEXPORT int trace_seek_erf_timestamp(libtrace_t *trace, uint64_t ts)
1768{
1769        if (trace->format->seek_erf) {
1770                return trace->format->seek_erf(trace,ts);
1771        }
1772        else {
1773                if (trace->format->seek_timeval) {
1774                        struct timeval tv;
1775#if __BYTE_ORDER == __BIG_ENDIAN
1776                        tv.tv_sec = ts & 0xFFFFFFFF;
1777                        tv.tv_usec = ((ts >> 32) * 1000000) & 0xFFFFFFFF;
1778#elif __BYTE_ORDER == __LITTLE_ENDIAN
1779                        tv.tv_sec = ts >> 32;
1780                        tv.tv_usec = ((ts&0xFFFFFFFF)*1000000)>>32;
1781#else
1782#error "What on earth are you running this on?"
1783#endif
1784                        if (tv.tv_usec >= 1000000) {
1785                                tv.tv_usec -= 1000000;
1786                                tv.tv_sec += 1;
1787                        }
1788                        return trace->format->seek_timeval(trace,tv);
1789                }
1790                if (trace->format->seek_seconds) {
1791                        double seconds = 
1792                                (ts>>32) + ((ts & UINT_MAX)*1.0 / UINT_MAX);
1793                        return trace->format->seek_seconds(trace,seconds);
1794                }
1795                trace_set_err(trace,
1796                                TRACE_ERR_OPTION_UNAVAIL,
1797                                "Feature unimplemented");
1798                return -1;
1799        }
1800}
1801
1802DLLEXPORT int trace_seek_seconds(libtrace_t *trace, double seconds)
1803{
1804        if (trace->format->seek_seconds) {
1805                return trace->format->seek_seconds(trace,seconds);
1806        }
1807        else {
1808                if (trace->format->seek_timeval) {
1809                        struct timeval tv;
1810                        tv.tv_sec = (uint32_t)seconds;
1811                        tv.tv_usec = (uint32_t)(((seconds - tv.tv_sec) * 1000000)/UINT_MAX);
1812                        return trace->format->seek_timeval(trace,tv);
1813                }
1814                if (trace->format->seek_erf) {
1815                        uint64_t timestamp = 
1816                                ((uint64_t)((uint32_t)seconds) << 32) + \
1817                            (uint64_t)(( seconds - (uint32_t)seconds   ) * UINT_MAX);
1818                        return trace->format->seek_erf(trace,timestamp);
1819                }
1820                trace_set_err(trace,
1821                                TRACE_ERR_OPTION_UNAVAIL,
1822                                "Feature unimplemented");
1823                return -1;
1824        }
1825}
1826
1827DLLEXPORT int trace_seek_timeval(libtrace_t *trace, struct timeval tv)
1828{
1829        if (trace->format->seek_timeval) {
1830                return trace->format->seek_timeval(trace,tv);
1831        }
1832        else {
1833                if (trace->format->seek_erf) {
1834                        uint64_t timestamp = ((((uint64_t)tv.tv_sec) << 32) + \
1835                                (((uint64_t)tv.tv_usec * UINT_MAX)/1000000));
1836                        return trace->format->seek_erf(trace,timestamp);
1837                }
1838                if (trace->format->seek_seconds) {
1839                        double seconds = tv.tv_sec + ((tv.tv_usec * 1.0)/1000000);
1840                        return trace->format->seek_seconds(trace,seconds);
1841                }
1842                trace_set_err(trace,
1843                                TRACE_ERR_OPTION_UNAVAIL,
1844                                "Feature unimplemented");
1845                return -1;
1846        }
1847}
1848
1849/* Converts a binary ethernet MAC address into a printable string */
1850DLLEXPORT char *trace_ether_ntoa(const uint8_t *addr, char *buf)
1851{
1852        static char staticbuf[18]={0,};
1853        if (!buf)
1854                buf=staticbuf;
1855        snprintf(buf,(size_t)18,"%02x:%02x:%02x:%02x:%02x:%02x",
1856                        addr[0],addr[1],addr[2],
1857                        addr[3],addr[4],addr[5]);
1858        return buf;
1859}
1860
1861/* Converts a printable ethernet MAC address into a binary format */
1862DLLEXPORT uint8_t *trace_ether_aton(const char *buf, uint8_t *addr)
1863{
1864        uint8_t *buf2 = addr;
1865        unsigned int tmp[6];
1866        static uint8_t staticaddr[6];
1867        if (!buf2)
1868                buf2=staticaddr;
1869        sscanf(buf,"%x:%x:%x:%x:%x:%x",
1870                        &tmp[0],&tmp[1],&tmp[2],
1871                        &tmp[3],&tmp[4],&tmp[5]);
1872        buf2[0]=tmp[0]; buf2[1]=tmp[1]; buf2[2]=tmp[2];
1873        buf2[3]=tmp[3]; buf2[4]=tmp[4]; buf2[5]=tmp[5];
1874        return buf2;
1875}
1876
1877
1878/* Creates a libtrace packet from scratch using the contents of the provided
1879 * buffer as the packet payload.
1880 *
1881 * Unlike trace_prepare_packet(), the buffer should not contain any capture
1882 * format headers; instead this function will add the PCAP header to the
1883 * packet record. This also means only PCAP packets can be constructed using
1884 * this function.
1885 *
1886 */
1887DLLEXPORT
1888void trace_construct_packet(libtrace_packet_t *packet,
1889                libtrace_linktype_t linktype,
1890                const void *data,
1891                uint16_t len)
1892{
1893        size_t size;
1894        static libtrace_t *deadtrace=NULL;
1895        libtrace_pcapfile_pkt_hdr_t hdr;
1896#ifdef WIN32
1897        struct _timeb tstruct;
1898#else
1899        struct timeval tv;
1900#endif
1901
1902        /* We need a trace to attach the constructed packet to (and it needs
1903         * to be PCAP) */
1904        if (NULL == deadtrace) 
1905                deadtrace=trace_create_dead("pcapfile");
1906
1907        /* Fill in the new PCAP header */
1908#ifdef WIN32
1909        _ftime(&tstruct);
1910        hdr.ts_sec=tstruct.time;
1911        hdr.ts_usec=tstruct.millitm * 1000;
1912#else
1913        gettimeofday(&tv,NULL);
1914        hdr.ts_sec=tv.tv_sec;
1915        hdr.ts_usec=tv.tv_usec;
1916#endif
1917
1918        hdr.caplen=len;
1919        hdr.wirelen=len;
1920
1921        /* Now fill in the libtrace packet itself */
1922        packet->trace=deadtrace;
1923        size=len+sizeof(hdr);
1924        if (packet->buf_control==TRACE_CTRL_PACKET) {
1925                packet->buffer=realloc(packet->buffer,size);
1926        }
1927        else {
1928                packet->buffer=malloc(size);
1929        }
1930        packet->buf_control=TRACE_CTRL_PACKET;
1931        packet->header=packet->buffer;
1932        packet->payload=(void*)((char*)packet->buffer+sizeof(hdr));
1933       
1934        /* Ugh, memcpy - sadly necessary */
1935        memcpy(packet->header,&hdr,sizeof(hdr));
1936        memcpy(packet->payload,data,(size_t)len);
1937        packet->type=pcap_linktype_to_rt(libtrace_to_pcap_linktype(linktype));
1938
1939        trace_clear_cache(packet);
1940}
1941
1942
1943uint64_t trace_get_received_packets(libtrace_t *trace)
1944{
1945        assert(trace);
1946        uint64_t ret;
1947
1948        if (trace->format->get_received_packets) {
1949                if ((ret = trace->format->get_received_packets(trace)) != UINT64_MAX)
1950                        return ret;
1951        }
1952        // Read this cached value taken before the trace was closed
1953        return trace->received_packets;
1954}
1955
1956uint64_t trace_get_filtered_packets(libtrace_t *trace)
1957{
1958        assert(trace);
1959        int i = 0;
1960        uint64_t ret = trace->filtered_packets;
1961        for (i = 0; i < trace->perpkt_thread_count; i++) {
1962                ret += trace->perpkt_threads[i].filtered_packets;
1963        }
1964        if (trace->format->get_filtered_packets) {
1965                return trace->format->get_filtered_packets(trace)+
1966                        ret;
1967        }
1968        return ret;
1969}
1970
1971uint64_t trace_get_dropped_packets(libtrace_t *trace)
1972{
1973        assert(trace);
1974        uint64_t ret;
1975
1976        if (trace->format->get_dropped_packets) {
1977                if ((ret = trace->format->get_dropped_packets(trace)) != UINT64_MAX)
1978                        return ret;
1979        }
1980        // Read this cached value taken before the trace was closed
1981        return trace->dropped_packets;
1982}
1983
1984uint64_t trace_get_accepted_packets(libtrace_t *trace)
1985{
1986        assert(trace);
1987        int i = 0;
1988        uint64_t ret = trace->accepted_packets;
1989        for (i = 0; i < trace->perpkt_thread_count; i++) {
1990                ret += trace->perpkt_threads[i].accepted_packets;
1991        }
1992        return ret;
1993}
1994
1995void trace_clear_cache(libtrace_packet_t *packet) {
1996
1997        packet->l2_header = NULL;
1998        packet->l3_header = NULL;
1999        packet->l4_header = NULL;
2000        packet->link_type = 0;
2001        packet->l3_ethertype = 0;
2002        packet->transport_proto = 0;
2003        packet->capture_length = -1;
2004        packet->wire_length = -1;
2005        packet->payload_length = -1;
2006        packet->l2_remaining = 0;
2007        packet->l3_remaining = 0;
2008        packet->l4_remaining = 0;
2009
2010}
2011
2012void trace_interrupt(void) {
2013        libtrace_halt = 1;
2014}
2015
2016void register_format(struct libtrace_format_t *f) {
2017        assert(f->next==NULL); /* Can't register a format twice */
2018        f->next=formats_list;
2019        formats_list=f;
2020
2021        /* Now, verify that the format has at least the minimum functionality.
2022         *
2023         * This #if can be changed to a 1 to output warnings about inconsistent
2024         * functions being provided by format modules.  This generally is very
2025         * noisy, as almost all modules don't implement one or more functions
2026         * for various reasons.  This is very useful when checking a new
2027         * format module is sane.
2028         */ 
2029#if 0
2030        if (f->init_input) {
2031#define REQUIRE(x) \
2032                if (!f->x) \
2033                        fprintf(stderr,"%s: Input format should provide " #x "\n",f->name)
2034                REQUIRE(read_packet);
2035                REQUIRE(start_input);
2036                REQUIRE(fin_input);
2037                REQUIRE(get_link_type);
2038                REQUIRE(get_capture_length);
2039                REQUIRE(get_wire_length);
2040                REQUIRE(get_framing_length);
2041                REQUIRE(trace_event);
2042                if (!f->get_erf_timestamp
2043                        && !f->get_seconds
2044                        && !f->get_timeval) {
2045                        fprintf(stderr,"%s: A trace format capable of input, should provide at least one of\n"
2046"get_erf_timestamp, get_seconds or trace_timeval\n",f->name);
2047                }
2048                if (f->trace_event!=trace_event_trace) {
2049                        /* Theres nothing that a trace file could optimise with
2050                         * config_input
2051                         */
2052                        REQUIRE(pause_input);
2053                        REQUIRE(config_input);
2054                        REQUIRE(get_fd);
2055                }
2056                else {
2057                        if (f->get_fd) {
2058                                fprintf(stderr,"%s: Unnecessary get_fd\n",
2059                                                f->name);
2060                        }
2061                }
2062#undef REQUIRE
2063        }
2064        else {
2065#define REQUIRE(x) \
2066                if (f->x) \
2067                        fprintf(stderr,"%s: Non Input format shouldn't need " #x "\n",f->name)
2068                REQUIRE(read_packet);
2069                REQUIRE(start_input);
2070                REQUIRE(pause_input);
2071                REQUIRE(fin_input);
2072                REQUIRE(get_link_type);
2073                REQUIRE(get_capture_length);
2074                REQUIRE(get_wire_length);
2075                REQUIRE(get_framing_length);
2076                REQUIRE(trace_event);
2077                REQUIRE(get_seconds);
2078                REQUIRE(get_timeval);
2079                REQUIRE(get_erf_timestamp);
2080#undef REQUIRE
2081        }
2082        if (f->init_output) {
2083#define REQUIRE(x) \
2084                if (!f->x) \
2085                        fprintf(stderr,"%s: Output format should provide " #x "\n",f->name)
2086                REQUIRE(write_packet);
2087                REQUIRE(start_output);
2088                REQUIRE(config_output);
2089                REQUIRE(fin_output);
2090#undef REQUIRE
2091        }
2092        else {
2093#define REQUIRE(x) \
2094                if (f->x) \
2095                        fprintf(stderr,"%s: Non Output format shouldn't need " #x "\n",f->name)
2096                REQUIRE(write_packet);
2097                REQUIRE(start_output);
2098                REQUIRE(config_output);
2099                REQUIRE(fin_output);
2100#undef REQUIRE
2101        }
2102#endif
2103}
2104
Note: See TracBrowser for help on using the repository browser.