source: lib/trace.c @ 3a169d7

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

trace_write_packet isn't const -- it can promote/demote packets

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