source: lib/trace.c @ e502f76

4.0.1-hotfixescachetimestampsdevelopdpdk-ndagetsilivegetfragoffhelplibtrace4ndag_formatpfringrc-4.0.1rc-4.0.2rc-4.0.3rc-4.0.4ringdecrementfixringperformanceringtimestampfixes
Last change on this file since e502f76 was e502f76, checked in by Shane Alcock <salcock@…>, 15 years ago

Added a pause function to format_rt
Moved the rt_connect call to rt_start
Got rid of blocking in rt_read_versatile by remembering the rt header
Removed 'restrict' from trace.c as some compilers didn't like it
Fixed bugs in configure that was causing failure to detect DAG
Added a new rt type for lost connections

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