source: tools/tracemerge/tracemerge.c @ 8afe034

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

Merged sam's fixes
Added unique-packets

  • Property mode set to 100644
File size: 2.8 KB
Line 
1#include <libtrace.h>
2#include <stdio.h>
3#include <stdlib.h>
4#include <inttypes.h>
5#include <stdbool.h>
6#include <getopt.h>
7
8void usage(char *argv0)
9{
10        fprintf(stderr,"Usage: %s [ -i | --set-interface ] [ -u | --unique-packets ] outputuri traceuri...\n",argv0);
11        fprintf(stderr,"\n");
12        fprintf(stderr,"Merges traces together, with -i each trace gets it's own direction/interface,\n without traces keep whatever direction/interface they have set\n");
13        exit(1);
14}
15
16int main(int argc, char *argv[])
17{
18       
19        struct libtrace_out_t *output;
20        struct libtrace_t **input;
21        struct libtrace_packet_t **packet;
22        bool *live;
23        bool set_interface=false;
24        bool unique_packets=false;
25        int i=0;
26        uint64_t last_ts=0;
27
28        while (1) {
29                int option_index;
30                struct option long_options[] = {
31                        { "set-interface",      0, 0, 'i' },
32                        { "unique-packets",     0, 0, 'u' },
33                        { NULL,                 0, 0, 0 },
34                };
35
36                int c=getopt_long(argc, argv, "iu",
37                                long_options, &option_index);
38
39                if (c==-1)
40                        break;
41
42                switch (c) {
43                        case 'i': set_interface=true; break;
44                        case 'u': unique_packets=true; break;
45                        default:
46                                fprintf(stderr,"unknown option: %c\n",c);
47                                usage(argv[0]);
48
49                }
50
51        }
52
53        if (optind+2>argc)
54                usage(argv[0]);
55
56        output=trace_create_output(argv[optind]);
57        if (trace_is_err_output(output)) {
58                trace_perror_output(output,"trace_create");
59                return 1;
60        }
61        if (trace_start_output(output)==-1) {
62                trace_perror_output(output,"trace_start");
63                return 1;
64        }
65
66        input=calloc((argc-optind),sizeof(struct libtrace_t *));
67        packet=calloc((argc-optind),sizeof(struct libtrace_packet_t *));
68        live=calloc((argc-optind),sizeof(bool));
69        for(i=0;i<argc-optind;++i) {
70                struct libtrace_t *f;
71                f=trace_create(argv[i+optind]);
72                if (trace_is_err(f)) {
73                        trace_perror(f,"trace_create");
74                        return 1;
75                }
76                if (trace_start(f)==-1) {
77                        trace_perror(f,"trace_start");
78                        return 1;
79                }
80                p=trace_create_packet();
81                input[i]=f;
82                packet[i]=p;
83                if (trace_read_packet(f,packet[i])>0)
84                        live[i]=true;
85        }
86
87        while(1) {
88                uint64_t oldest_ts=0;
89                int oldest=-1;
90                for(i=0;i<argc-2;++i) {
91                        if (!live[i] && input[i]) {
92                                int ret=trace_read_packet(input[i],packet[i]);
93                                if (ret<0) {
94                                        /* Error */
95                                        perror(argv[i+2]);
96                                        trace_destroy(input[i]);
97                                        input[i]=NULL;
98                                }
99                                else if (ret==0) {
100                                        /* EOF */
101                                        trace_destroy(input[i]);
102                                        input[i]=NULL;
103                                }
104                                else
105                                        live[i]=true;
106                        }
107                        if (live[i] && 
108                                (oldest==-1 || 
109                                 oldest_ts>trace_get_erf_timestamp(packet[i]))) {
110                                oldest=i;
111                                oldest_ts=trace_get_erf_timestamp(packet[i]);
112                        }
113                }
114                /* We have run out of packets! */
115                if (oldest==-1) {
116                        break;
117                }
118
119                live[oldest]=false;
120
121                if (set_interface)
122                        trace_set_direction(packet[oldest],oldest);
123
124                if (unique_packets && oldest_ts == last_ts)
125                        continue;
126
127                trace_write_packet(output,packet[oldest]);
128
129                last_ts=oldest_ts;
130               
131        }
132        trace_destroy_output(output);
133
134        return 0;
135}
Note: See TracBrowser for help on using the repository browser.