source: lib/trace.c @ 4927e86

4.0.1-hotfixescachetimestampsdevelopdpdk-ndagetsilivegetfragoffhelplibtrace4ndag_formatpfringrc-4.0.1rc-4.0.2rc-4.0.3rc-4.0.4ringdecrementfixringperformanceringtimestampfixes
Last change on this file since 4927e86 was 4927e86, checked in by Perry Lorier <perry@…>, 15 years ago

Disable annoyingly verbose debugging that was accidently committed

  • Property mode set to 100644
File size: 37.3 KB
RevLine 
[a79ddbe]1/*
2 * This file is part of libtrace
3 *
4 * Copyright (c) 2004 The University of Waikato, Hamilton, New Zealand.
5 * Authors: Daniel Lawson
6 *          Perry Lorier
7 *         
8 * All rights reserved.
9 *
10 * This code has been developed by the University of Waikato WAND
11 * research group. For further information please see http://www.wand.net.nz/
12 *
13 * libtrace is free software; you can redistribute it and/or modify
14 * it under the terms of the GNU General Public License as published by
15 * the Free Software Foundation; either version 2 of the License, or
16 * (at your option) any later version.
17 *
18 * libtrace is distributed in the hope that it will be useful,
19 * but WITHOUT ANY WARRANTY; without even the implied warranty of
20 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
21 * GNU General Public License for more details.
22 *
23 * You should have received a copy of the GNU General Public License
24 * along with libtrace; if not, write to the Free Software
25 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
26 *
27 * $Id$
28 *
29 */
30
[9580a74]31
[fba4ca0]32/* @file
[2c060e3]33 *
34 * @brief Trace file processing library
35 *
36 * @author Daniel Lawson
37 * @author Perry Lorier
38 *
[68667ee]39 * @internal
[2c060e3]40 */
41#define _GNU_SOURCE
[6c05d2c]42#include "common.h"
[60fc8aa]43#include "config.h"
[2c060e3]44#include <assert.h>
45#include <errno.h>
46#include <fcntl.h>
47#include <stdio.h>
48#include <stdlib.h>
49#include <string.h>
50#include <sys/stat.h>
51#include <sys/types.h>
[3d4d52d]52#include <stdarg.h>
[e5d3718]53
54#ifdef HAVE_LIMITS_H
55#  include <limits.h>
56#endif
57
[1f3696b]58#ifdef HAVE_SYS_LIMITS_H
59#  include <sys/limits.h>
60#endif
[e5d3718]61
[808a478]62#ifdef HAVE_NET_IF_ARP_H
63#  include <net/if_arp.h>
64#endif
65
66#ifdef HAVE_NET_IF_H
67#  include <net/if.h>
68#endif
69
70#ifdef HAVE_NETINET_IN_H
71#  include <netinet/in.h>
72#endif
73
74#ifdef HAVE_NET_ETHERNET_H
75#  include <net/ethernet.h>
76#endif
77
78#ifdef HAVE_NETINET_IF_ETHER_H
79#  include <netinet/if_ether.h>
80#endif
81
[2c060e3]82#include <time.h>
[dfef05d]83
[2c060e3]84#include "libtrace.h"
85#include "fifo.h"
[9e2a109]86#include "libtrace_int.h"
[91ebc50]87#include "parse_cmd.h"
[2c060e3]88
[9580a74]89#if HAVE_PCAP_BPF_H
[6c05d2c]90#  include <pcap-bpf.h>
91#else
92#  ifdef HAVE_NET_BPF_H
93#    include <net/bpf.h>
94#  endif
95#endif
[2137b49]96
[9e2a109]97#include "libtrace_int.h"
[72bfe20]98#include "format_helper.h"
[6dbc47a]99#include "rt_protocol.h"
[2c060e3]100
[91ebc50]101#define MAXOPTS 1024
102
[c5a22ec]103
[eeab9832]104static struct libtrace_format_t *formats_list = 0;
[a6badc2]105
[6f75b9d]106/* strncpy is not assured to copy the final \0, so we
107 * will use our own one that does
108 */
109static void xstrncpy(char *dest, const char *src, size_t n)
110{
111        strncpy(dest,src,n);
112        dest[n]='\0';
113}
114 
115static char *xstrndup(const char *src,size_t n)
116{       
[67a14d4]117        char *ret=(char*)malloc(n+1);
[6f75b9d]118        xstrncpy(ret,src,n);
119        return ret;
120}
121
[9e2a109]122void register_format(struct libtrace_format_t *f) {
[eeab9832]123        assert(f->next==NULL);
124        f->next=formats_list;
125        formats_list=f;
[411f3c7]126        /* Now, verify things
127         * This #if can be changed to a 1 to output warnings about inconsistant
128         * functions being provided by format modules.  This generally is very
129         * noisy, as almost all modules don't implement one or more functions
130         * for various reasons.  This is very useful when checking a new
131         * format module is sane.
132         */ 
[4927e86]133#if 0
[411f3c7]134        if (f->init_input) {
[d5879cc]135#define REQUIRE(x) \
[411f3c7]136                if (!f->x) \
137                        fprintf(stderr,"%s: Input format should provide " #x "\n",f->name)
[d5879cc]138                REQUIRE(read_packet);
139                REQUIRE(start_input);
140                REQUIRE(fin_input);
141                REQUIRE(get_link_type);
142                REQUIRE(get_capture_length);
143                REQUIRE(get_wire_length);
144                REQUIRE(get_framing_length);
145                REQUIRE(trace_event);
[411f3c7]146                if (!f->get_erf_timestamp
147                        && !f->get_seconds
148                        && !f->get_timeval) {
[d5879cc]149                        fprintf(stderr,"%s: A trace format capable of input, should provide at least one of\n"
[411f3c7]150"get_erf_timestamp, get_seconds or trace_timeval\n",f->name);
[d5879cc]151                }
[411f3c7]152                if (f->trace_event!=trace_event_trace) {
153                        /* Theres nothing that a trace file could optimise with
154                         * config_input
155                         */
156                        REQUIRE(pause_input);
157                        REQUIRE(config_input);
[4aa4615]158                        REQUIRE(get_fd);
159                }
160                else {
[411f3c7]161                        if (f->get_fd) {
[4aa4615]162                                fprintf(stderr,"%s: Unnecessary get_fd\n",
[411f3c7]163                                                f->name);
[4aa4615]164                        }
165                }
[d5879cc]166#undef REQUIRE
167        }
168        else {
169#define REQUIRE(x) \
[411f3c7]170                if (f->x) \
171                        fprintf(stderr,"%s: Non Input format shouldn't need " #x "\n",f->name)
[d5879cc]172                REQUIRE(read_packet);
173                REQUIRE(start_input);
174                REQUIRE(pause_input);
175                REQUIRE(fin_input);
176                REQUIRE(get_link_type);
177                REQUIRE(get_capture_length);
178                REQUIRE(get_wire_length);
179                REQUIRE(get_framing_length);
180                REQUIRE(trace_event);
[4aa4615]181                REQUIRE(get_seconds);
182                REQUIRE(get_timeval);
183                REQUIRE(get_erf_timestamp);
[d5879cc]184#undef REQUIRE
185        }
[411f3c7]186        if (f->init_output) {
[d5879cc]187#define REQUIRE(x) \
[411f3c7]188                if (!f->x) \
189                        fprintf(stderr,"%s: Output format should provide " #x "\n",f->name)
[d5879cc]190                REQUIRE(write_packet);
191                REQUIRE(start_output);
192                REQUIRE(config_output);
193                REQUIRE(fin_output);
194#undef REQUIRE
195        }
196        else {
197#define REQUIRE(x) \
[411f3c7]198                if (f->x) \
199                        fprintf(stderr,"%s: Non Output format shouldn't need " #x "\n",f->name)
[d5879cc]200                REQUIRE(write_packet);
201                REQUIRE(start_output);
202                REQUIRE(config_output);
203                REQUIRE(fin_output);
204#undef REQUIRE
205        }
[4aa4615]206#endif
[4dedc28]207}
[8dd1830]208
[dd2eaee]209void erf_constructor();
210void legacy_constructor();
211void linuxnative_constructor();
212void pcap_constructor();
213void pcapfile_constructor();
214void rt_constructor();
215void wag_constructor();
216
217/* call all the constructors if they haven't yet all been called */
218void trace_init(void)
219{
220        if (!formats_list) {
221                erf_constructor();
222                legacy_constructor();
[7b2a39b]223#ifdef HAVE_NETPACKET_PACKET_H
[dd2eaee]224                linuxnative_constructor();
[7b2a39b]225#endif
[dd2eaee]226#ifdef HAVE_PCAP
227                pcap_constructor();
228#endif
229                pcapfile_constructor();
230                rt_constructor();
231                wag_constructor();
232        }
233}
234
[fba4ca0]235/* Prints help information for libtrace
[40ba6ad]236 *
237 * Function prints out some basic help information regarding libtrace,
238 * and then prints out the help() function registered with each input module
239 */
[39e141f]240DLLEXPORT void trace_help() {
[eeab9832]241        struct libtrace_format_t *tmp;
[dd2eaee]242        trace_init();
[fe43699]243        printf("libtrace %s\n",PACKAGE_VERSION);
[eeab9832]244        for(tmp=formats_list;tmp;tmp=tmp->next) {
245                if (tmp->help)
246                        tmp->help();
[40ba6ad]247        }
248}
249
[59c31da]250#define RP_BUFSIZE 65536
[2137b49]251#define URI_PROTO_LINE 16
[2c060e3]252
[fba4ca0]253/* Gets the name of the output format for a given output trace.
[4d5a9a7]254 *
255 * @params libtrace     the output trace to get the name of the format for
[fba4ca0]256 * @returns callee-owned null-terminated char* containing the output format
[4d5a9a7]257 *
258 */
[39e141f]259DLLEXPORT SIMPLE_FUNCTION
[31709e8d]260char *trace_get_output_format(const struct libtrace_out_t *libtrace) {
[4d5a9a7]261        char * format = libtrace->format->name;
[5eae97a]262
[4d5a9a7]263        return format;
264}
[a114d8b5]265
[dd2eaee]266
[fba4ca0]267/* Create a trace file from a URI
268 *
269 * @params char * containing a valid libtrace URI
[2c060e3]270 * @returns opaque pointer to a libtrace_t
271 *
272 * Valid URI's are:
[277442c]273 *  erf:/path/to/erf/file
274 *  erf:/path/to/erf/file.gz
275 *  erf:/path/to/rtclient/socket
276 *  erf:-                       (stdin)
[fba4ca0]277 *  dag:/dev/dagcard
[476df56]278 *  pcapint:pcapinterface               (eg: pcapint:eth0)
[277442c]279 *  pcap:/path/to/pcap/file
280 *  pcap:-
281 *  rtclient:hostname
282 *  rtclient:hostname:port
283 *  wag:-
284 *  wag:/path/to/wag/file
285 *  wag:/path/to/wag/file.gz
286 *  wag:/path/to/wag/socket
287 *
288 * If an error occured when attempting to open a trace, NULL is returned
289 * and an error is output to stdout.
[2c060e3]290 */
[39e141f]291DLLEXPORT struct libtrace_t *trace_create(const char *uri) {
[67a14d4]292        struct libtrace_t *libtrace = 
293                        (struct libtrace_t *)malloc(sizeof(struct libtrace_t));
[de8c420b]294        char *scan = 0;
[6f75b9d]295        const char *uridata = 0;                 
[eeab9832]296        struct libtrace_format_t *tmp;
[0c222cc]297
[dd2eaee]298        trace_init();
299
[0c222cc]300        assert(uri && "Passing NULL to trace_create makes me a very sad program");
[a114d8b5]301       
[eea0f87]302        libtrace->err.err_num = TRACE_ERR_NOERROR;
[0ea3526]303        libtrace->format=NULL;
[9e2a109]304       
[7068467]305        /* parse the URI to determine what sort of event we are dealing with */
[de8c420b]306        if ((uridata = trace_parse_uri(uri, &scan)) == 0) {
[4aa4615]307                trace_set_err(libtrace,TRACE_ERR_BAD_FORMAT,"Bad uri format (%s)",uri);
[0ea3526]308                return libtrace;
[de8c420b]309        }
310       
[9e2a109]311        libtrace->event.tdelta = 0.0;
[880aa58]312        libtrace->filter = NULL;
313        libtrace->snaplen = 0;
[d8f02df]314        libtrace->started=false;
[9e2a109]315
[eeab9832]316        for (tmp=formats_list;tmp;tmp=tmp->next) {
317                if (strlen(scan) == strlen(tmp->name) &&
318                                strncasecmp(scan, tmp->name, strlen(scan)) == 0
319                                ) {
320                        libtrace->format=tmp;
[6f75b9d]321                        break;
322                }
[9e2a109]323        }
324        if (libtrace->format == 0) {
[0ea3526]325                trace_set_err(libtrace, TRACE_ERR_BAD_FORMAT,
[880aa58]326                                "Unknown format (%s)",scan);
[eeab9832]327                return libtrace;
[9e2a109]328        }
329
330        libtrace->uridata = strdup(uridata);
[7068467]331        /* libtrace->format now contains the type of uri
332         * libtrace->uridata contains the appropriate data for this
333         */
[9e2a109]334       
335        if (libtrace->format->init_input) {
[d5879cc]336                int err=libtrace->format->init_input(libtrace);
337                assert (err==-1 || err==0);
338                if (err==-1) {
[880aa58]339                        /* init_input should call trace_set_err to set
340                         * the error message
341                         */
[0ea3526]342                        return libtrace;
[ffc8c8d]343                }
[9e2a109]344        } else {
[91db67b]345                trace_set_err(libtrace,TRACE_ERR_UNSUPPORTED,
[880aa58]346                                "Format does not support input (%s)",scan);
[0ea3526]347                return libtrace;
[9e2a109]348        }
349       
350
[b5cd711]351        libtrace->fifo = create_tracefifo(1048576);
[d8f02df]352        if (!libtrace->fifo) {
353                trace_set_err(libtrace,ENOMEM,"Could not allocate memory for fifo");
354                free(scan);
355                return libtrace;
356        }
[0ea3526]357        assert(libtrace->fifo);
[ae8196e]358        free(scan);
[ae66bde]359        libtrace->err.err_num=TRACE_ERR_NOERROR;
360        libtrace->err.problem[0]='\0';
[2c060e3]361        return libtrace;
362}
363
[fba4ca0]364/* Creates a "dummy" trace file that has only the format type set.
365 *
366 * @returns opaque pointer to a (sparsely initialised) libtrace_t
367 *
[880aa58]368 * IMPORTANT: Do not attempt to call trace_read_packet or other such functions
369 * with the dummy trace. Its intended purpose is to act as a packet->trace for
370 * libtrace_packet_t's that are not associated with a libtrace_t structure.
[fba4ca0]371 */
[39e141f]372DLLEXPORT struct libtrace_t * trace_create_dead (const char *uri) {
[67a14d4]373        struct libtrace_t *libtrace = (struct libtrace_t *)
374                                        malloc(sizeof(struct libtrace_t));
375        char *scan = (char *)calloc(sizeof(char),URI_PROTO_LINE);
[084d95a]376        char *uridata;
[eeab9832]377        struct libtrace_format_t *tmp;
[084d95a]378       
[eea0f87]379        libtrace->err.err_num = TRACE_ERR_NOERROR;
[084d95a]380
381        if((uridata = strchr(uri,':')) == NULL) {
[6f75b9d]382                xstrncpy(scan, uri, strlen(uri));
[084d95a]383        } else {
[6f75b9d]384                xstrncpy(scan,uri, (uridata - uri));
[084d95a]385        }
386       
387        libtrace->format = 0;   
388       
[eeab9832]389        for(tmp=formats_list;tmp;tmp=tmp->next) {
390                if (strlen(scan) == strlen(tmp->name) &&
[084d95a]391                                !strncasecmp(scan,
[eeab9832]392                                        tmp->name,
[084d95a]393                                        strlen(scan))) {
[eeab9832]394                                libtrace->format=tmp;
[084d95a]395                                break;
396                                }
397        }
398        if (libtrace->format == 0) {
[0ea3526]399                trace_set_err(libtrace,TRACE_ERR_BAD_FORMAT,
[880aa58]400                                "Unknown format (%s)",scan);
[084d95a]401                return 0;
402        }
[7ac9705]403
404        libtrace->format_data = NULL;
[ae8196e]405        free(scan);
[084d95a]406        return libtrace;
407
408}
[fba4ca0]409
[b69afb1]410/* Creates a trace output file from a URI.
[5eae97a]411 *
[fba4ca0]412 * @param uri   the uri string describing the output format and destination
[880aa58]413 * @returns opaque pointer to a libtrace_output_t
[fba4ca0]414 *
[880aa58]415 *  If an error occured when attempting to open the output trace, NULL is
416 *  returned and trace_errno is set.
[fba4ca0]417 */
418       
[39e141f]419DLLEXPORT libtrace_out_t *trace_create_output(const char *uri) {
[67a14d4]420        libtrace_out_t *libtrace = 
421                        (libtrace_out_t*)malloc(sizeof(struct libtrace_out_t));
[5eae97a]422       
[de8c420b]423        char *scan = 0;
[6f75b9d]424        const char *uridata = 0;
[eeab9832]425        struct libtrace_format_t *tmp;
[df2dff9]426
[dd2eaee]427        trace_init();
428
[eea0f87]429        libtrace->err.err_num = TRACE_ERR_NOERROR;
[d97bbb2]430        strcpy(libtrace->err.problem,"Error message set\n");
[eea0f87]431       
[7068467]432        /* parse the URI to determine what sort of event we are dealing with */
[df2dff9]433
[de8c420b]434        if ((uridata = trace_parse_uri(uri, &scan)) == 0) {
[eeab9832]435                trace_set_err_out(libtrace,TRACE_ERR_BAD_FORMAT,
436                                "Bad uri format (%s)",uri);
[0ea3526]437                return libtrace;
[de8c420b]438        }
439       
[eeab9832]440        libtrace->format = NULL;
441        for(tmp=formats_list;tmp;tmp=tmp->next) {
442                if (strlen(scan) == strlen(tmp->name) &&
[df2dff9]443                                !strncasecmp(scan,
[eeab9832]444                                        tmp->name,
[df2dff9]445                                        strlen(scan))) {
[eeab9832]446                                libtrace->format=tmp;
[df2dff9]447                                break;
448                                }
449        }
[eeab9832]450        if (libtrace->format == NULL) {
[0ea3526]451                trace_set_err_out(libtrace,TRACE_ERR_BAD_FORMAT,
[880aa58]452                                "Unknown output format (%s)",scan);
[eeab9832]453                return libtrace;
[df2dff9]454        }
455        libtrace->uridata = strdup(uridata);
456
457
[7068467]458        /* libtrace->format now contains the type of uri
459         * libtrace->uridata contains the appropriate data for this
460         */
[df2dff9]461
462        if (libtrace->format->init_output) {
[fe76c55]463                /* 0 on success, -1 on failure */
464                switch(libtrace->format->init_output(libtrace)) {
465                        case -1: /* failure */
466                                free(libtrace);
[eeab9832]467                                return libtrace;
[fe76c55]468                        case 0: /* success */
469                                break;
470                        default:
471                                assert(!"init_output() should return -1 for failure, or 0 for success");
[ffc8c8d]472                }
[df2dff9]473        } else {
[91db67b]474                trace_set_err_out(libtrace,TRACE_ERR_UNSUPPORTED,
[880aa58]475                                "Format does not support writing (%s)",scan);
[0ea3526]476                return libtrace;
[df2dff9]477        }
478
479
[ae8196e]480        free(scan);
[0a6638f]481        libtrace->started=false;
[91ebc50]482        return libtrace;
483}
[5eae97a]484
[37195b4]485/* Start a trace
486 * @param libtrace      the input trace to start
487 * @returns 0 on success
488 *
489 * This does the work associated with actually starting up
490 * the trace.  it may fail.
491 */
[39e141f]492DLLEXPORT int trace_start(struct libtrace_t *libtrace)
[37195b4]493{
[df338b3]494        assert(libtrace);
[37195b4]495        if (libtrace->format->start_input) {
496                int ret=libtrace->format->start_input(libtrace);
[49babe0]497                if (ret < 0) {
[37195b4]498                        return ret;
499                }
500        }
501
502        libtrace->started=true;
503        return 0;
504}
505
[39e141f]506DLLEXPORT int trace_start_output(libtrace_out_t *libtrace) 
[0a6638f]507{
508        assert(libtrace);
509        if (libtrace->format->start_output) {
510                int ret=libtrace->format->start_output(libtrace);
511                if (ret < 0) {
512                        return ret;
513                }
514        }
515
[eeab9832]516        libtrace->started=true;
[0a6638f]517        return 0;
518}
519
[39e141f]520DLLEXPORT int trace_pause(libtrace_t *libtrace)
[1fbd938]521{
522        assert(libtrace);
523        assert(libtrace->started && "BUG: Called trace_pause without calling trace_start first");
524        if (libtrace->format->pause_input)
525                libtrace->format->pause_input(libtrace);
526        libtrace->started=false;
527        return 0;
528}
529
[39e141f]530DLLEXPORT int trace_config(libtrace_t *libtrace,
[65a5900]531                trace_option_t option,
532                void *value)
533{
534        int ret;
535        if (libtrace->format->config_input) {
536                ret=libtrace->format->config_input(libtrace,option,value);
537                if (ret==0)
538                        return 0;
539        }
540        switch(option) {
541                case TRACE_OPTION_SNAPLEN:
542                        libtrace->snaplen=*(int*)value;
[67a14d4]543                        return 0;
[65a5900]544                case TRACE_OPTION_FILTER:
[67a14d4]545                        libtrace->filter=(struct libtrace_filter_t *)value;
546                        return 0;
[65a5900]547                case TRACE_OPTION_PROMISC:
[0ea3526]548                        trace_set_err(libtrace,TRACE_ERR_OPTION_UNAVAIL,
[65a5900]549                                "Promisc mode is not supported by this format module");
550                        return -1;
551        }
[67a14d4]552        trace_set_err(libtrace,TRACE_ERR_UNKNOWN_OPTION,
553                "Unknown option %i", option);
554        return -1;
[65a5900]555}
556
[fba4ca0]557/* Parses an output options string and calls the appropriate function to deal with output options.
[ef55d05]558 *
559 * @param libtrace      the output trace object to apply the options to
560 * @param options       the options string
561 * @returns -1 if option configuration failed, 0 otherwise
562 *
563 * @author Shane Alcock
564 */
[39e141f]565DLLEXPORT int trace_config_output(struct libtrace_out_t *libtrace, 
[7068467]566                trace_option_output_t option,
567                void *value) {
[9c6aa95]568        if (libtrace->format->config_output) {
[7068467]569                return libtrace->format->config_output(libtrace, option, value);
[9c6aa95]570        }
571        return -1;
[5eae97a]572}
573
[fba4ca0]574/* Close a trace file, freeing up any resources it may have been using
[2c060e3]575 *
576 */
[39e141f]577DLLEXPORT void trace_destroy(struct libtrace_t *libtrace) {
[2c060e3]578        assert(libtrace);
[ab4cb04]579        if (libtrace->started && libtrace->format->pause_input)
580                libtrace->format->pause_input(libtrace);
[4dedc28]581        libtrace->format->fin_input(libtrace);
[7068467]582        /* need to free things! */
[a114d8b5]583        free(libtrace->uridata);
[b5cd711]584        destroy_tracefifo(libtrace->fifo);
[2c060e3]585        free(libtrace);
586}
587
[084d95a]588
[39e141f]589DLLEXPORT void trace_destroy_dead(struct libtrace_t *libtrace) {
[084d95a]590        assert(libtrace);
591        free(libtrace);
592}
[fba4ca0]593/* Close an output trace file, freeing up any resources it may have been using
[5eae97a]594 *
595 * @param libtrace      the output trace file to be destroyed
596 *
597 * @author Shane Alcock
598 * */
[39e141f]599DLLEXPORT void trace_destroy_output(struct libtrace_out_t *libtrace) {
[5eae97a]600        assert(libtrace);
[1fc2f6a]601        libtrace->format->fin_output(libtrace);
[a114d8b5]602        free(libtrace->uridata);
[5eae97a]603        free(libtrace);
604}
605
[39e141f]606DLLEXPORT libtrace_packet_t *trace_create_packet() {
[67a14d4]607        libtrace_packet_t *packet = 
608                (libtrace_packet_t*)calloc(1,sizeof(libtrace_packet_t));
[c1db742]609        packet->buf_control=TRACE_CTRL_PACKET;
[14d8a63]610        return packet;
611}
612
[39e141f]613DLLEXPORT libtrace_packet_t *trace_copy_packet(const libtrace_packet_t *packet) {
[67a14d4]614        libtrace_packet_t *dest = 
615                (libtrace_packet_t *)malloc(sizeof(libtrace_packet_t));
[91c9552]616        dest->trace=packet->trace;
617        dest->buffer=malloc(
618                        trace_get_framing_length(packet)
619                        +trace_get_capture_length(packet));
620        dest->header=dest->buffer;
621        dest->payload=(void*)
622                ((char*)dest->buffer+trace_get_framing_length(packet));
623        dest->size=packet->size;
624        dest->type=packet->type;
625        dest->buf_control=TRACE_CTRL_PACKET;
626        memcpy(dest->header,packet->header,trace_get_framing_length(packet));
627        memcpy(dest->payload,packet->payload,trace_get_capture_length(packet));
[d8f02df]628
629        return dest;
[91c9552]630}
631
[14d8a63]632/** Destroy a packet object
633 *
634 * sideeffect: sets packet to NULL
635 */
[39e141f]636DLLEXPORT void trace_destroy_packet(struct libtrace_packet_t **packet) {
[91c9552]637        if ((*packet)->buf_control == TRACE_CTRL_PACKET) {
[14d8a63]638                free((*packet)->buffer);
639        }
640        free((*packet));
[752709f]641        *packet = NULL;
[14d8a63]642}       
643
[fba4ca0]644/* Read one packet from the trace into buffer
[2c060e3]645 *
[2137b49]646 * @param libtrace      the libtrace opaque pointer
647 * @param packet        the packet opaque pointer
[fba4ca0]648 * @returns 0 on EOF, negative value on error
[2c060e3]649 *
650 */
[39e141f]651DLLEXPORT int trace_read_packet(libtrace_t *libtrace, libtrace_packet_t *packet) {
[2137b49]652
[37195b4]653        assert(libtrace && "You called trace_read_packet() with a NULL libtrace parameter!\n");
654        assert(libtrace->started && "BUG: You must call libtrace_start() before trace_read_packet()\n");
655        assert(packet);
[c1db742]656        assert((packet->buf_control==TRACE_CTRL_PACKET || packet->buf_control==TRACE_CTRL_EXTERNAL)&&
[0a6638f]657                "BUG: You must allocate a packet using packet_create()");
[2137b49]658     
659        /* Store the trace we are reading from into the packet opaque
660         * structure */
661        packet->trace = libtrace;
[afd0b73]662
[4dedc28]663        if (libtrace->format->read_packet) {
[880aa58]664                do {
665                        packet->size=libtrace->format->read_packet(libtrace,packet);
[222d8f5]666                        if (packet->size==(size_t)-1 || packet->size==0) {
[880aa58]667                                return packet->size;
[222d8f5]668                        }
[880aa58]669                        if (libtrace->filter) {
670                                /* If the filter doesn't match, read another
671                                 * packet
672                                 */
673                                if (!trace_bpf_filter(libtrace->filter,packet)){
674                                        continue;
675                                }
676                        }
677                        if (libtrace->snaplen>0) {
678                                /* Snap the packet */
679                                trace_set_capture_length(packet,
680                                                libtrace->snaplen);
681                        }
682                        return packet->size;
683                } while(1);
[a22c979]684        }
[91db67b]685        trace_set_err(libtrace,TRACE_ERR_UNSUPPORTED,"This format does not support reading packets\n");
[a8a9355]686        packet->size=-1;
[9c6aa95]687        return -1;
[2c060e3]688}
689
[fba4ca0]690/* Writes a packet to the specified output
[5eae97a]691 *
692 * @param libtrace      describes the output format, destination, etc.
693 * @param packet        the packet to be written out
694 * @returns the number of bytes written, -1 if write failed
695 *
696 * @author Shane Alcock
697 * */
[39e141f]698DLLEXPORT int trace_write_packet(struct libtrace_out_t *libtrace, const struct libtrace_packet_t *packet) {
[5eae97a]699        assert(libtrace);
700        assert(packet); 
[a8a9355]701        /* Verify the packet is valid */
702        assert(packet->size<65536);
703        assert(packet->size>0);
[eeab9832]704        assert(libtrace->started);
[5eae97a]705
[1fc2f6a]706        if (libtrace->format->write_packet) {
707                return libtrace->format->write_packet(libtrace, packet);
[5eae97a]708        }
[9c6aa95]709        return -1;
[5eae97a]710}
[2c060e3]711
[39e141f]712DLLEXPORT void *trace_get_link(const struct libtrace_packet_t *packet) {
[14d8a63]713        return (void *)packet->payload;
[2c060e3]714}
715
[3b8a5ef]716/*
[eaa5529]717typedef struct legacy_framing {
718        uint64_t        ts;
719        uint32_t        crc;
720        uint32_t        header;
[3b8a5ef]721        uint32_t        data[12]; // pad to 64 bytes
[eaa5529]722} legacy_framing_t;
[3b8a5ef]723*/
[eaa5529]724
[bb09ce4]725
[fba4ca0]726/* Get the current time in DAG time format
[850630f]727 * @param packet        a pointer to a libtrace_packet structure
[2c060e3]728 * @returns a 64 bit timestamp in DAG ERF format (upper 32 bits are the seconds
729 * past 1970-01-01, the lower 32bits are partial seconds)
730 * @author Daniel Lawson
731 */ 
[39e141f]732DLLEXPORT uint64_t trace_get_erf_timestamp(const libtrace_packet_t *packet) {
[2c060e3]733        uint64_t timestamp = 0;
[ca87bcd]734        double seconds = 0.0;
[4dedc28]735        struct timeval ts;
736
[a8a9355]737        assert(packet->size>0 && packet->size<65536);
738
[4dedc28]739        if (packet->trace->format->get_erf_timestamp) {
[7068467]740                /* timestamp -> timestamp */
[4dedc28]741                timestamp = packet->trace->format->get_erf_timestamp(packet);
742        } else if (packet->trace->format->get_timeval) {
[7068467]743                /* timeval -> timestamp */
[4dedc28]744                ts = packet->trace->format->get_timeval(packet);
745                timestamp = ((((uint64_t)ts.tv_sec) << 32) + \
746                                (((uint64_t)ts.tv_usec * UINT_MAX)/1000000));
[ca87bcd]747        } else if (packet->trace->format->get_seconds) {
[7068467]748                /* seconds -> timestamp */
[ca87bcd]749                seconds = packet->trace->format->get_seconds(packet);
[1974620]750                timestamp = ((uint64_t)((uint32_t)seconds) << 32) + \
[67a14d4]751                      (uint64_t)(( seconds - (uint32_t)seconds   ) * UINT_MAX);
[ca87bcd]752        }
[4dedc28]753        return timestamp;
[2c060e3]754}
755
[fba4ca0]756/* Get the current time in struct timeval
[850630f]757 * @param packet        a pointer to a libtrace_packet structure
758 *
[2c060e3]759 * @returns time that this packet was seen in a struct timeval
760 * @author Daniel Lawson
761 * @author Perry Lorier
762 */ 
[39e141f]763DLLEXPORT struct timeval trace_get_timeval(const libtrace_packet_t *packet) {
[2c060e3]764        struct timeval tv;
[4dedc28]765        uint64_t ts = 0;
[e01bfa8]766        double seconds = 0.0;
[a8a9355]767        assert(packet->size>0 && packet->size<65536);
[4dedc28]768        if (packet->trace->format->get_timeval) {
[7068467]769                /* timeval -> timeval */
[4dedc28]770                tv = packet->trace->format->get_timeval(packet);
771        } else if (packet->trace->format->get_erf_timestamp) {
[7068467]772                /* timestamp -> timeval */
[4dedc28]773                ts = packet->trace->format->get_erf_timestamp(packet);
[8dd1830]774#if __BYTE_ORDER == __BIG_ENDIAN
[4dedc28]775                tv.tv_sec = ts & 0xFFFFFFFF;
[8dd1830]776#elif __BYTE_ORDER == __LITTLE_ENDIAN
[4dedc28]777                tv.tv_sec = ts >> 32;
[8dd1830]778#else
779#error "What on earth are you running this on?"
780#endif
[2e223eca]781                tv.tv_usec = ((ts&0xFFFFFFFF)*1000000)>>32;
[4dedc28]782                if (tv.tv_usec >= 1000000) {
783                        tv.tv_usec -= 1000000;
784                        tv.tv_sec += 1;
785                }
[ca87bcd]786        } else if (packet->trace->format->get_seconds) {
[7068467]787                /* seconds -> timeval */
[ca87bcd]788                seconds = packet->trace->format->get_seconds(packet);
789                tv.tv_sec = (uint32_t)seconds;
790                tv.tv_usec = (uint32_t)(((seconds - tv.tv_sec) * 1000000)/UINT_MAX);
[4dedc28]791        }
792
[2c060e3]793        return tv;
794}
795
[fba4ca0]796/* Get the current time in floating point seconds
[850630f]797 * @param packet        a pointer to a libtrace_packet structure
[2c060e3]798 * @returns time that this packet was seen in 64bit floating point seconds
799 * @author Perry Lorier
800 */ 
[39e141f]801DLLEXPORT double trace_get_seconds(const struct libtrace_packet_t *packet) {
[9c6aa95]802        double seconds = 0.0;
803        uint64_t ts = 0;
[ca87bcd]804        struct timeval tv;
[a8a9355]805
806        assert(packet->size>0 && packet->size<65536);
[4dedc28]807       
808        if (packet->trace->format->get_seconds) {
[7068467]809                /* seconds->seconds */
[4dedc28]810                seconds = packet->trace->format->get_seconds(packet);
811        } else if (packet->trace->format->get_erf_timestamp) {
[7068467]812                /* timestamp -> seconds */
[4dedc28]813                ts = packet->trace->format->get_erf_timestamp(packet);
814                seconds =  (ts>>32) + ((ts & UINT_MAX)*1.0 / UINT_MAX);
[ca87bcd]815        } else if (packet->trace->format->get_timeval) {
[7068467]816                /* timeval -> seconds */
[ca87bcd]817                tv = packet->trace->format->get_timeval(packet);
[b51edf5]818                seconds = tv.tv_sec + ((tv.tv_usec * 1.0) / 1000000);
[ca87bcd]819        }
820
[4dedc28]821        return seconds;
[2c060e3]822}
823
[39e141f]824DLLEXPORT size_t trace_get_capture_length(const libtrace_packet_t *packet) {
[4dedc28]825
[752709f]826        assert(packet->size<65536);
[a8a9355]827
[4dedc28]828        if (packet->trace->format->get_capture_length) {
829                return packet->trace->format->get_capture_length(packet);
[2c060e3]830        }
831        return -1;
832}
833       
[fba4ca0]834/* Get the size of the packet as it was seen on the wire.
[850630f]835 * @param packet        a pointer to a libtrace_packet structure
836 *
[2c060e3]837 * @returns the size of the packet as it was on the wire.
838 * @author Perry Lorier
839 * @author Daniel Lawson
840 * @note Due to the trace being a header capture, or anonymisation this may
841 * not be the same as the Capture Len.
842 */ 
[39e141f]843DLLEXPORT size_t trace_get_wire_length(const libtrace_packet_t *packet){
[a8a9355]844        assert(packet->size>0 && packet->size<65536);
845
[4dedc28]846        if (packet->trace->format->get_wire_length) {
847                return packet->trace->format->get_wire_length(packet);
[2c060e3]848        }
849        return -1;
850
851}
852
[7c8eacf]853/* Get the length of the capture framing headers.
854 * @param packet        the packet opaque pointer
855 * @returns the size of the packet as it was on the wire.
856 * @author Perry Lorier
857 * @author Daniel Lawson
858 * @note this length corresponds to the difference between the size of a
859 * captured packet in memory, and the captured length of the packet
860 */ 
[39e141f]861DLLEXPORT SIMPLE_FUNCTION
[d8f02df]862size_t trace_get_framing_length(const libtrace_packet_t *packet) {
[7c8eacf]863        if (packet->trace->format->get_framing_length) {
864                return packet->trace->format->get_framing_length(packet);
865        }
866        return -1;
867}
868
869
[fba4ca0]870/* Get the type of the link layer
[850630f]871 * @param packet        a pointer to a libtrace_packet structure
[2c060e3]872 * @returns libtrace_linktype_t
873 * @author Perry Lorier
874 * @author Daniel Lawson
875 */
[39e141f]876DLLEXPORT libtrace_linktype_t trace_get_link_type(const libtrace_packet_t *packet ) {
[4dedc28]877        if (packet->trace->format->get_link_type) {
878                return packet->trace->format->get_link_type(packet);
[2c060e3]879        }
[67a14d4]880        return (libtrace_linktype_t)-1;
[2c060e3]881}
882
[fba4ca0]883/* process a libtrace event
[ef36351d]884 * @param trace the libtrace opaque pointer
885 * @param packet the libtrace_packet opaque pointer
[2c060e3]886 * @returns
[68667ee]887 *  TRACE_EVENT_IOWAIT  Waiting on I/O on fd
888 *  TRACE_EVENT_SLEEP   Next event in seconds
889 *  TRACE_EVENT_PACKET  Packet arrived in buffer with size size
[7bb7dda]890 *  TRACE_EVENT_TERMINATE Trace terminated (perhaps with an error condition)
[2137b49]891 * FIXME currently keeps a copy of the packet inside the trace pointer,
892 * which in turn is stored inside the new packet object...
[277442c]893 * @author Perry Lorier
[2c060e3]894 */
[39e141f]895DLLEXPORT struct libtrace_eventobj_t trace_event(struct libtrace_t *trace, 
[ef36351d]896                struct libtrace_packet_t *packet) {
[67a14d4]897        struct libtrace_eventobj_t event = {TRACE_EVENT_IOWAIT,0,0.0,0};
[4fd12ac]898
899        if (!trace) {
900                fprintf(stderr,"You called trace_event() with a NULL trace object!\n");
901        }
902        assert(trace);
903        assert(packet);
904
905        /* Store the trace we are reading from into the packet opaque
906         * structure */
907        packet->trace = trace;
908
[72bfe20]909        if (packet->trace->format->trace_event) {
910                return packet->trace->format->trace_event(trace,packet);
911        } else {
912                return event;
[2c060e3]913        }
[72bfe20]914
[2c060e3]915}
[c681e86]916
[fba4ca0]917/* setup a BPF filter
[c681e86]918 * @param filterstring a char * containing the bpf filter string
[c5a22ec]919 * @returns opaque pointer pointer to a libtrace_filter_t object
[c681e86]920 * @author Daniel Lawson
921 */
[39e141f]922DLLEXPORT struct libtrace_filter_t *trace_bpf_setfilter(const char *filterstring) {
[1b2fd6f]923#if HAVE_BPF
[747c501]924        libtrace_filter_t *filter = (struct libtrace_filter_t*)
[67a14d4]925                                malloc(sizeof(struct libtrace_filter_t));
[c5a22ec]926        filter->filterstring = strdup(filterstring);
[df338b3]927        filter->flag = 0;
[c5a22ec]928        return filter;
[9580a74]929#else
930        fprintf(stderr,"This version of libtrace does not have bpf filter support\n");
[516a90d]931        return NULL;
[9580a74]932#endif
[c681e86]933}
934
[747c501]935DLLEXPORT void trace_destroy_bpf(libtrace_filter_t *filter)
936{
[516a90d]937#if HAVE_BPF
[747c501]938        free(filter->filterstring);
939        if (filter->flag)
940                pcap_freecode(&filter->filter);
941        free(filter);
[516a90d]942#else
943
944#endif
[747c501]945}
946
[df338b3]947/* compile a bpf filter, now we know what trace it's on
948 * @internal
949 *
950 * @returns -1 on error, 0 on success
951 */
952int trace_bpf_compile(libtrace_filter_t *filter,
953                const libtrace_packet_t *packet ) {
954#if HAVE_BPF
955        void *linkptr = 0;
956        assert(filter);
957
958        /* If this isn't a real packet, then fail */
959        linkptr = trace_get_link(packet);
960        if (!linkptr) {
[0ea3526]961                trace_set_err(packet->trace,
962                                TRACE_ERR_BAD_PACKET,"Packet has no payload");
[df338b3]963                return -1;
964        }
965       
966        if (filter->filterstring && ! filter->flag) {
967                pcap_t *pcap;
[d5879cc]968                libtrace_linktype_t linktype=trace_get_link_type(packet);
969                if (linktype==(libtrace_linktype_t)-1) {
970                        trace_set_err(packet->trace,TRACE_ERR_BAD_PACKET,
971                                        "Packet has an unknown linktype");
972                        return -1;
973                }
974                if (libtrace_to_pcap_dlt(linktype) == -1) {
975                        trace_set_err(packet->trace,TRACE_ERR_BAD_PACKET,
976                                        "Unknown pcap equivilent linktype");
977                        return -1;
978                }
[df338b3]979                pcap=(pcap_t *)pcap_open_dead(
[d5879cc]980                                libtrace_to_pcap_dlt(linktype),
[df338b3]981                                1500);
982                /* build filter */
983                if (pcap_compile( pcap, &filter->filter, filter->filterstring, 
984                                        1, 0)) {
985                        pcap_close(pcap);
[0ea3526]986                        trace_set_err(packet->trace,TRACE_ERR_BAD_PACKET,
[91c9552]987                                        "Packet has no payload");
[df338b3]988                        return -1;
989                }
990                pcap_close(pcap);
991                filter->flag=1;
992        }
993        return 0;
994#else
995        assert(!"This should never be called when BPF not enabled");
[abc8e94]996        trace_set_err(packet->trace,TRACE_ERR_OPTION_UNAVAIL,
997                                "Feature unavailable");
998        return -1;
[df338b3]999#endif
1000}
1001
[39e141f]1002DLLEXPORT int trace_bpf_filter(struct libtrace_filter_t *filter,
[58e9e02]1003                        const struct libtrace_packet_t *packet) {
[1b2fd6f]1004#if HAVE_BPF
[2137b49]1005        void *linkptr = 0;
1006        int clen = 0;
[c5a22ec]1007        assert(filter);
[2137b49]1008        assert(packet);
[b86f8d3]1009        linkptr = trace_get_link(packet);
1010        if (!linkptr) {
1011                return 0;
1012        }
[df338b3]1013
1014        /* We need to compile it now, because before we didn't know what the
1015         * link type was
1016         */
[91c9552]1017        if (trace_bpf_compile(filter,packet)==-1)
1018                return -1;
[b86f8d3]1019       
[2137b49]1020        clen = trace_get_capture_length(packet);
[c681e86]1021
[df338b3]1022        assert(filter->flag);
1023        return bpf_filter(filter->filter.bf_insns, linkptr, clen, clen);
[9580a74]1024#else
1025        fprintf(stderr,"This version of libtrace does not have bpf filter support\n");
1026        return 0;
1027#endif
[c681e86]1028}
1029
[fba4ca0]1030/* Set the direction flag, if it has one
[80a2e99]1031 * @param packet the packet opaque pointer
1032 * @param direction the new direction (0,1,2,3)
1033 * @returns a signed value containing the direction flag, or -1 if this is not supported
1034 * @author Daniel Lawson
1035 */
[39e141f]1036DLLEXPORT int8_t trace_set_direction(struct libtrace_packet_t *packet, int8_t direction) {
[9a36e6d]1037        assert(packet);
[a8a9355]1038        assert(packet->size>0 && packet->size<65536);
[4dedc28]1039        if (packet->trace->format->set_direction) {
1040                return packet->trace->format->set_direction(packet,direction);
[80a2e99]1041        }
[4dedc28]1042        return -1;
[80a2e99]1043}
1044
[fba4ca0]1045/* Get the direction flag, if it has one
[850630f]1046 * @param packet a pointer to a libtrace_packet structure
[14a9d05]1047 * @returns a signed value containing the direction flag, or -1 if this is not supported
[9075d49]1048 * The direction is defined as 0 for packets originating locally (ie, outbound)
1049 * and 1 for packets originating remotely (ie, inbound).
1050 * Other values are possible, which might be overloaded to mean special things
1051 * for a special trace.
[14a9d05]1052 * @author Daniel Lawson
1053 */
[39e141f]1054DLLEXPORT int8_t trace_get_direction(const struct libtrace_packet_t *packet) {
[2137b49]1055        assert(packet);
[a8a9355]1056        assert(packet->size>0 && packet->size<65536);
[4dedc28]1057        if (packet->trace->format->get_direction) {
1058                return packet->trace->format->get_direction(packet);
[14a9d05]1059        }
[4dedc28]1060        return -1;
[14a9d05]1061}
[c681e86]1062
[5c25d3a]1063#define ROOT_SERVER(x) ((x) < 512)
1064#define ROOT_CLIENT(x) ((512 <= (x)) && ((x) < 1024))
1065#define NONROOT_SERVER(x) ((x) >= 5000)
1066#define NONROOT_CLIENT(x) ((1024 <= (x)) && ((x) < 5000))
1067#define DYNAMIC(x) ((49152 < (x)) && ((x) < 65535))
[83e0a25]1068#define SERVER(x) ROOT_SERVER(x) || NONROOT_SERVER(x)
[b007afc]1069#define CLIENT(x) ROOT_CLIENT(x) || NONROOT_CLIENT(x)
[83e0a25]1070
[d66fbf0]1071/* Attempt to deduce the 'server' port
[1cea750]1072 * @param protocol the IP protocol (eg, 6 or 17 for TCP or UDP)
1073 * @param source the TCP or UDP source port
1074 * @param dest the TCP or UDP destination port
1075 * @returns a hint as to which port is the server port
1076 * @author Daniel Lawson
1077 */
[39e141f]1078DLLEXPORT int8_t trace_get_server_port(uint8_t protocol UNUSED, uint16_t source, uint16_t dest) {
[83e0a25]1079        /*
1080         * * If the ports are equal, return DEST
1081         * * Check for well-known ports in the given protocol
1082         * * Root server ports: 0 - 511
1083         * * Root client ports: 512 - 1023
1084         * * non-root client ports: 1024 - 4999
1085         * * non-root server ports: 5000+
1086         * * Check for static ranges: 1024 - 49151
1087         * * Check for dynamic ranges: 49152 - 65535
1088         * * flip a coin.
1089         */
[9c6aa95]1090       
[b007afc]1091        /* equal */
[9c6aa95]1092        if (source == dest)
[b007afc]1093                return USE_DEST;
1094
1095        /* root server port, 0 - 511 */
[5c25d3a]1096        if (ROOT_SERVER(source) && ROOT_SERVER(dest)) {
1097                if (source < dest)
1098                        return USE_SOURCE;
[b007afc]1099                return USE_DEST;
[5c25d3a]1100        }
[b007afc]1101
1102        if (ROOT_SERVER(source) && !ROOT_SERVER(dest))
[83e0a25]1103                return USE_SOURCE;
[b007afc]1104        if (!ROOT_SERVER(source) && ROOT_SERVER(dest))
1105                return USE_DEST;
1106
1107        /* non-root server */
[5c25d3a]1108        if (NONROOT_SERVER(source) && NONROOT_SERVER(dest)) {
1109                if (source < dest)
1110                        return USE_SOURCE;
[83e0a25]1111                return USE_DEST;
[5c25d3a]1112        }
[b007afc]1113        if (NONROOT_SERVER(source) && !NONROOT_SERVER(dest))
[83e0a25]1114                return USE_SOURCE;
[b007afc]1115        if (!NONROOT_SERVER(source) && NONROOT_SERVER(dest))
[83e0a25]1116                return USE_DEST;
[b007afc]1117
1118        /* root client */
[5c25d3a]1119        if (ROOT_CLIENT(source) && ROOT_CLIENT(dest)) {
1120                if (source < dest)
1121                        return USE_SOURCE;
[b007afc]1122                return USE_DEST;
[5c25d3a]1123        }
[f0a2f98]1124        if (ROOT_CLIENT(source) && !ROOT_CLIENT(dest)) {
1125                /* prefer root-client over nonroot-client */
1126                if (NONROOT_CLIENT(dest))
1127                        return USE_SOURCE;
[b007afc]1128                return USE_DEST;
[f0a2f98]1129        }
1130        if (!ROOT_CLIENT(source) && ROOT_CLIENT(dest)) {
1131                /* prefer root-client over nonroot-client */
1132                if (NONROOT_CLIENT(source))
1133                        return USE_DEST;
[b007afc]1134                return USE_SOURCE;
[f0a2f98]1135        }
[83e0a25]1136       
[b007afc]1137        /* nonroot client */
[5c25d3a]1138        if (NONROOT_CLIENT(source) && NONROOT_CLIENT(dest)) {
1139                if (source < dest) 
1140                        return USE_SOURCE;
[b007afc]1141                return USE_DEST;
[5c25d3a]1142        }
[b007afc]1143        if (NONROOT_CLIENT(source) && !NONROOT_CLIENT(dest))
1144                return USE_DEST;
1145        if (!NONROOT_CLIENT(source) && NONROOT_CLIENT(dest))
1146                return USE_SOURCE;
1147
[5c25d3a]1148        /* dynamic range */
[b007afc]1149        if (DYNAMIC(source) && DYNAMIC(dest))
[f9bb6c5]1150                if (source < dest)
1151                        return USE_SOURCE;
[b007afc]1152                return USE_DEST;
1153        if (DYNAMIC(source) && !DYNAMIC(dest))
1154                return USE_DEST;
1155        if (!DYNAMIC(source) && DYNAMIC(dest))
1156                return USE_SOURCE;
1157        /*
1158        if (SERVER(source) && CLIENT(dest))
1159                return USE_SOURCE;
1160       
1161        if (SERVER(dest) && CLIENT(source))
1162                return USE_DEST;
1163        if (ROOT_SERVER(source) && !ROOT_SERVER(dest))
1164                return USE_SOURCE;
1165        if (ROOT_SERVER(dest) && !ROOT_SERVER(source))
1166                return USE_DEST;
1167        */
[7068467]1168        /* failing that test... */
[83e0a25]1169        if (source < dest) {
1170                return USE_SOURCE;
1171        } 
1172        return USE_DEST;
1173       
1174}
[80a2e99]1175
[fba4ca0]1176/* Truncate the packet at the suggested length
[80a2e99]1177 * @param packet        the packet opaque pointer
[850630f]1178 * @param size          the new length of the packet
[fe7b292]1179 * @returns the new size of the packet
[c95b923]1180 * @note size and the return size refer to the network-level payload of the
1181 * packet, and do not include any capture headers. For example, to truncate a
1182 * packet after the IP header, set size to sizeof(ethernet_header) +
1183 * sizeof(ip_header)
1184 * @note If the original network-level payload is smaller than size, then the
1185 * original size is returned and the packet is left unchanged.
[80a2e99]1186 * @author Daniel Lawson
1187 */
[39e141f]1188DLLEXPORT size_t trace_set_capture_length(struct libtrace_packet_t *packet, size_t size) {
[80a2e99]1189        assert(packet);
[a8a9355]1190        assert(packet->size>0 && packet->size<65536);
[80a2e99]1191
[c95b923]1192        if (packet->trace->format->set_capture_length) {
[d3b2234]1193                int caplen=packet->trace->format->set_capture_length(packet,size);
1194                if (caplen!=-1) {
1195                        packet->size=trace_get_framing_length(packet)+caplen;
1196                }
1197                return caplen;
[80a2e99]1198        }
[fe7b292]1199
[4dedc28]1200        return -1;
[80a2e99]1201}
1202
[39e141f]1203DLLEXPORT const char * trace_parse_uri(const char *uri, char **format) {
[6f75b9d]1204        const char *uridata = 0;
[de8c420b]1205       
1206        if((uridata = strchr(uri,':')) == NULL) {
[7068467]1207                /* badly formed URI - needs a : */
[de8c420b]1208                return 0;
1209        }
1210
1211        if ((uridata - uri) > URI_PROTO_LINE) {
[7068467]1212                /* badly formed URI - uri type is too long */
[de8c420b]1213                return 0;
1214        }
1215
[6f75b9d]1216        *format=xstrndup(uri, (uridata - uri));
1217
[7068467]1218        /* push uridata past the delimiter */
[de8c420b]1219        uridata++;
[ae8196e]1220       
[de8c420b]1221        return uridata;
1222}
[6dbc47a]1223
[0ea3526]1224enum base_format_t trace_get_format(struct libtrace_packet_t *packet) 
1225{
[6dbc47a]1226        assert(packet);
1227
1228        return packet->trace->format->type;
1229}
[de8c420b]1230       
[39e141f]1231DLLEXPORT libtrace_err_t trace_get_err(libtrace_t *trace)
[880aa58]1232{
[0ea3526]1233        libtrace_err_t err = trace->err;
1234        trace->err.err_num = 0; /* "OK" */
1235        trace->err.problem[0]='\0';
1236        return err;
1237}
1238
[39e141f]1239DLLEXPORT bool trace_is_err(libtrace_t *trace)
[eeab9832]1240{
1241        return trace->err.err_num != 0;
1242}
1243
[39e141f]1244DLLEXPORT void trace_perror(libtrace_t *trace,const char *msg,...)
[eeab9832]1245{
1246        char buf[256];
1247        va_list va;
1248        va_start(va,msg);
1249        vsnprintf(buf,sizeof(buf),msg,va);
1250        va_end(va);
1251        if(trace->err.err_num) {
1252                fprintf(stderr,"%s(%s): %s\n",
1253                                buf,trace->uridata,trace->err.problem);
1254        } else {
1255                fprintf(stderr,"%s(%s): No error\n",
1256                                buf,trace->uridata);
1257        }
[114b8d6]1258        trace->err.err_num = 0; /* "OK" */
1259        trace->err.problem[0]='\0';
[eeab9832]1260}
1261
[39e141f]1262DLLEXPORT libtrace_err_t trace_get_err_output(libtrace_out_t *trace)
[0ea3526]1263{
1264        libtrace_err_t err = trace->err;
[ae66bde]1265        trace->err.err_num = TRACE_ERR_NOERROR; /* "OK" */
[0ea3526]1266        trace->err.problem[0]='\0';
1267        return err;
[880aa58]1268}
[0d57541]1269
[39e141f]1270DLLEXPORT bool trace_is_err_output(libtrace_out_t *trace)
[eeab9832]1271{
1272        return trace->err.err_num != 0;
1273}
1274
[39e141f]1275DLLEXPORT void trace_perror_output(libtrace_out_t *trace,const char *msg,...)
[eeab9832]1276{
1277        char buf[256];
1278        va_list va;
1279        va_start(va,msg);
1280        vsnprintf(buf,sizeof(buf),msg,va);
1281        va_end(va);
1282        if(trace->err.err_num) {
1283                fprintf(stderr,"%s(%s): %s\n",
1284                                buf,trace->uridata,trace->err.problem);
1285        } else {
1286                fprintf(stderr,"%s(%s): No error\n",buf,trace->uridata);
1287        }
1288}
1289
[39e141f]1290DLLEXPORT int trace_seek_erf_timestamp(libtrace_t *trace, uint64_t ts)
[0d57541]1291{
1292        if (trace->format->seek_erf) {
1293                return trace->format->seek_erf(trace,ts);
1294        }
1295        else {
[6c248a9]1296                if (trace->format->seek_timeval) {
1297                        struct timeval tv;
1298#if __BYTE_ORDER == __BIG_ENDIAN
1299                        tv.tv_sec = ts & 0xFFFFFFFF;
1300#elif __BYTE_ORDER == __LITTLE_ENDIAN
1301                        tv.tv_sec = ts >> 32;
1302#else
1303#error "What on earth are you running this on?"
1304#endif
1305                        tv.tv_usec = ((ts&0xFFFFFFFF)*1000000)>>32;
1306                        if (tv.tv_usec >= 1000000) {
1307                                tv.tv_usec -= 1000000;
1308                                tv.tv_sec += 1;
1309                        }
1310                        return trace->format->seek_timeval(trace,tv);
1311                }
1312                if (trace->format->seek_seconds) {
1313                        double seconds = 
1314                                (ts>>32) + ((ts & UINT_MAX)*1.0 / UINT_MAX);
1315                        return trace->format->seek_seconds(trace,seconds);
1316                }
1317                trace_set_err(trace,
1318                                TRACE_ERR_OPTION_UNAVAIL,
1319                                "Feature unimplemented");
1320                return -1;
1321        }
1322}
1323
[39e141f]1324DLLEXPORT int trace_seek_seconds(libtrace_t *trace, double seconds)
[6c248a9]1325{
1326        if (trace->format->seek_seconds) {
1327                return trace->format->seek_seconds(trace,seconds);
1328        }
1329        else {
1330                if (trace->format->seek_timeval) {
1331                        struct timeval tv;
1332                        tv.tv_sec = (uint32_t)seconds;
1333                        tv.tv_usec = (uint32_t)(((seconds - tv.tv_sec) * 1000000)/UINT_MAX);
1334                        return trace->format->seek_timeval(trace,tv);
1335                }
1336                if (trace->format->seek_erf) {
1337                        uint64_t timestamp = 
1338                                ((uint64_t)((uint32_t)seconds) << 32) + \
[67a14d4]1339                            (uint64_t)(( seconds - (uint32_t)seconds   ) * UINT_MAX);
[6c248a9]1340                        return trace->format->seek_erf(trace,timestamp);
1341                }
1342                trace_set_err(trace,
1343                                TRACE_ERR_OPTION_UNAVAIL,
1344                                "Feature unimplemented");
1345                return -1;
1346        }
1347}
1348
[39e141f]1349DLLEXPORT int trace_seek_timeval(libtrace_t *trace, struct timeval tv)
[6c248a9]1350{
1351        if (trace->format->seek_timeval) {
1352                return trace->format->seek_timeval(trace,tv);
1353        }
1354        else {
1355                if (trace->format->seek_erf) {
1356                        uint64_t timestamp = ((((uint64_t)tv.tv_sec) << 32) + \
1357                                (((uint64_t)tv.tv_usec * UINT_MAX)/1000000));
1358                        return trace->format->seek_erf(trace,timestamp);
1359                }
1360                if (trace->format->seek_seconds) {
[e4e1635]1361                        double seconds = tv.tv_sec + ((tv.tv_usec * 1.0)/1000000);
[6c248a9]1362                        return trace->format->seek_seconds(trace,seconds);
1363                }
[0d57541]1364                trace_set_err(trace,
1365                                TRACE_ERR_OPTION_UNAVAIL,
1366                                "Feature unimplemented");
1367                return -1;
1368        }
1369}
1370
[39e141f]1371DLLEXPORT char *trace_ether_ntoa(const uint8_t *addr, char *buf)
[752709f]1372{
1373        char *buf2 = buf;
[411f3c7]1374        char staticbuf[18]={0,};
[752709f]1375        if (!buf2)
1376                buf2=staticbuf;
[abc8e94]1377        snprintf(buf2,18,"%02x:%02x:%02x:%02x:%02x:%02x",
[752709f]1378                        addr[0],addr[1],addr[2],
1379                        addr[3],addr[4],addr[5]);
1380        return buf2;
1381}
1382
[39e141f]1383DLLEXPORT uint8_t *trace_ether_aton(const char *buf, uint8_t *addr)
[752709f]1384{
1385        uint8_t *buf2 = addr;
1386        unsigned int tmp[6];
1387        static uint8_t staticaddr[6];
1388        if (!buf2)
1389                buf2=staticaddr;
1390        sscanf(buf,"%x:%x:%x:%x:%x:%x",
1391                        &tmp[0],&tmp[1],&tmp[2],
1392                        &tmp[3],&tmp[4],&tmp[5]);
1393        buf2[0]=tmp[0]; buf2[1]=tmp[1]; buf2[2]=tmp[2];
1394        buf2[3]=tmp[3]; buf2[4]=tmp[4]; buf2[5]=tmp[5];
1395        return buf2;
1396}
[4af54d1]1397
Note: See TracBrowser for help on using the repository browser.