source: lib/trace.c @ cab58c5

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

Move RT_* to TRACE_RT_* and put it in libtrace.h

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