source: lib/format_dag24.c @ f3f3558

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

Ready for release

  • Property mode set to 100644
File size: 12.6 KB
Line 
1/*
2 * This file is part of libtrace
3 *
4 * Copyright (c) 2007,2008 The University of Waikato, Hamilton, New Zealand.
5 * Authors: Daniel Lawson
6 *          Perry Lorier
7 *          Shane Alcock
8 *
9 * All rights reserved.
10 *
11 * This code has been developed by the University of Waikato WAND
12 * research group. For further information please see http://www.wand.net.nz/
13 *
14 * libtrace is free software; you can redistribute it and/or modify
15 * it under the terms of the GNU General Public License as published by
16 * the Free Software Foundation; either version 2 of the License, or
17 * (at your option) any later version.
18 *
19 * libtrace is distributed in the hope that it will be useful,
20 * but WITHOUT ANY WARRANTY; without even the implied warranty of
21 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
22 * GNU General Public License for more details.
23 *
24 * You should have received a copy of the GNU General Public License
25 * along with libtrace; if not, write to the Free Software
26 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
27 *
28 * $Id$
29 *
30 */
31#define _GNU_SOURCE
32
33#include "config.h"
34#include "common.h"
35#include "libtrace.h"
36#include "libtrace_int.h"
37#include "format_helper.h"
38#include "format_erf.h"
39
40#include <assert.h>
41#include <errno.h>
42#include <fcntl.h>
43#include <stdio.h>
44#include <string.h>
45#include <stdlib.h>
46
47#include <sys/mman.h>
48#ifdef WIN32
49#  include <io.h>
50#  include <share.h>
51#  define PATH_MAX _MAX_PATH
52#  define snprintf sprintf_s
53#else
54#  include <netdb.h>
55#  ifndef PATH_MAX
56#       define PATH_MAX 4096
57#  endif
58#  include <sys/ioctl.h>
59#endif
60
61
62static struct libtrace_format_t dag;
63
64#define DATA(x) ((struct dag_format_data_t *)x->format_data)
65#define DUCK DATA(libtrace)->duck
66#define FORMAT_DATA DATA(libtrace)
67
68struct dag_format_data_t {
69        struct {
70                uint32_t last_duck;
71                uint32_t duck_freq;
72                uint32_t last_pkt;
73                libtrace_t *dummy_duck;
74        } duck; 
75       
76        int fd;
77        void *buf;
78        uint32_t diff;
79        uint32_t offset;
80        uint32_t bottom;
81        uint32_t top;
82        uint64_t drops;
83};
84
85static int dag_available(libtrace_t *libtrace) {
86
87        if (FORMAT_DATA->diff > 0)
88                return FORMAT_DATA->diff;
89
90        FORMAT_DATA->bottom = FORMAT_DATA->top;
91        FORMAT_DATA->top = dag_offset(
92                        FORMAT_DATA->fd,
93                        &(FORMAT_DATA->bottom),
94                        DAGF_NONBLOCK);
95        FORMAT_DATA->diff = FORMAT_DATA->top - FORMAT_DATA->bottom;
96        FORMAT_DATA->offset = 0;
97        return FORMAT_DATA->diff;
98}
99
100static int dag_init_input(libtrace_t *libtrace) {
101        struct stat buf;
102        libtrace->format_data = (struct dag_format_data_t *)
103                malloc(sizeof(struct dag_format_data_t));
104        if (stat(libtrace->uridata, &buf) == -1) {
105                trace_set_err(libtrace,errno,"stat(%s)",libtrace->uridata);
106                return -1;
107        }
108
109        FORMAT_DATA->top = 0;
110        FORMAT_DATA->bottom = 0;
111        if (S_ISCHR(buf.st_mode)) {
112                /* DEVICE */
113                if((FORMAT_DATA->fd = dag_open(libtrace->uridata)) < 0) {
114                        trace_set_err(libtrace,errno,"Cannot open DAG %s",
115                                        libtrace->uridata);
116                        return -1;
117                }
118                if((FORMAT_DATA->buf = (void *)dag_mmap(FORMAT_DATA->fd)) == MAP_FAILED) {
119                        trace_set_err(libtrace,errno,"Cannot mmap DAG %s",
120                                        libtrace->uridata);
121                        return -1;
122                }
123        } else {
124                trace_set_err(libtrace,errno,"Not a valid dag device: %s",
125                                libtrace->uridata);
126                return -1;
127        }
128
129        DUCK.last_duck = 0;
130        DUCK.duck_freq = 0;
131        DUCK.last_pkt = 0;
132        DUCK.dummy_duck = NULL;
133        FORMAT_DATA->drops = 0;
134
135        return 0;
136}
137
138static int dag_config_input(libtrace_t *libtrace, trace_option_t option,
139                                void *data) {
140        switch(option) {
141                case TRACE_OPTION_META_FREQ:
142                        DUCK.duck_freq = *(int *)data;
143                        return 0;
144                case TRACE_OPTION_SNAPLEN:
145                        /* Surely we can set this?? Fall through for now*/
146                        return -1;
147                case TRACE_OPTION_PROMISC:
148                        /* DAG already operates in a promisc fashion */
149                        return -1;
150                case TRACE_OPTION_FILTER:
151                        return -1;
152                case TRACE_OPTION_EVENT_REALTIME:
153                        return -1;
154        }
155        return -1;
156}
157
158static int dag_start_input(libtrace_t *libtrace) {     
159        if(dag_start(FORMAT_DATA->fd) < 0) {
160                trace_set_err(libtrace,errno,"Cannot start DAG %s",
161                                libtrace->uridata);
162                return -1;
163        }
164
165        /* Flush the memory hole */
166        while(dag_available(libtrace) != 0)
167                FORMAT_DATA->diff = 0;
168        FORMAT_DATA->drops = 0;
169        return 0;
170}
171
172static int dag_pause_input(libtrace_t *libtrace) {
173        dag_stop(FORMAT_DATA->fd);
174        return 0;
175}
176
177static int dag_fin_input(libtrace_t *libtrace) {
178        dag_close(FORMAT_DATA->fd);
179        if (DUCK.dummy_duck)
180                trace_destroy_dead(DUCK.dummy_duck);
181        free(libtrace->format_data);
182        return 0; /* success */
183}
184
185static int dag_get_duckinfo(libtrace_t *libtrace,
186                                libtrace_packet_t *packet) {
187        dag_inf lt_dag_inf;
188
189        if (packet->buf_control == TRACE_CTRL_EXTERNAL ||
190                        !packet->buffer) {
191                packet->buffer = malloc(LIBTRACE_PACKET_BUFSIZE);
192                packet->buf_control = TRACE_CTRL_PACKET;
193                if (!packet->buffer) {
194                        trace_set_err(libtrace, errno,
195                                        "Cannot allocate packet buffer");
196                        return -1;
197                }
198        }
199
200        packet->header = 0;
201        packet->payload = packet->buffer;
202
203        if ((ioctl(FORMAT_DATA->fd, DAG_IOINF, &lt_dag_inf) < 0)) {
204                trace_set_err(libtrace, errno,
205                                "Error using DAG_IOINF");
206                return -1;
207        }
208        if (!IsDUCK(&lt_dag_inf)) {
209                printf("WARNING: %s does not have modern clock support - No DUCK information will be gathered\n", libtrace->uridata);
210                return 0;
211        }
212
213        if ((ioctl(FORMAT_DATA->fd, DAG_IOGETDUCK, (duck_inf *)packet->payload)
214                                < 0)) {
215                trace_set_err(libtrace, errno, "Error using DAG_IOGETDUCK");
216                return -1;
217        }
218
219        packet->type = TRACE_RT_DUCK_2_4;
220        if (!DUCK.dummy_duck)
221                DUCK.dummy_duck = trace_create_dead("duck:dummy");
222        packet->trace = DUCK.dummy_duck;
223        return sizeof(duck_inf);
224}
225
226
227static dag_record_t *dag_get_record(libtrace_t *libtrace) {
228        dag_record_t *erfptr = NULL;
229        uint16_t size;
230        erfptr = (dag_record_t *) ((char *)FORMAT_DATA->buf + 
231                        (FORMAT_DATA->bottom + FORMAT_DATA->offset));
232
233        if (!erfptr)
234                return NULL;
235        size = ntohs(erfptr->rlen);
236        assert( size >= dag_record_size );
237        FORMAT_DATA->offset += size;
238        FORMAT_DATA->diff -= size;
239        return erfptr;
240}
241
242static void dag_form_packet(dag_record_t *erfptr, libtrace_packet_t *packet) {
243        packet->buffer = erfptr;
244        packet->header = erfptr;
245        packet->type = TRACE_RT_DATA_ERF;
246        if (erfptr->flags.rxerror == 1) {
247                /* rxerror means the payload is corrupt - drop it
248                 * by tweaking rlen */
249                packet->payload = NULL;
250                erfptr->rlen = htons(erf_get_framing_length(packet));
251        } else {
252                packet->payload = (char*)packet->buffer
253                        + erf_get_framing_length(packet);
254        }
255
256}
257
258static int dag_read_packet(libtrace_t *libtrace, libtrace_packet_t *packet) {
259        int numbytes;
260        int size = 0;
261        struct timeval tv;
262        dag_record_t *erfptr = NULL;
263
264        if (DUCK.last_pkt - DUCK.last_duck > DUCK.duck_freq &&
265                        DUCK.duck_freq != 0) {
266                size = dag_get_duckinfo(libtrace, packet);
267                DUCK.last_duck = DUCK.last_pkt;
268                if (size != 0) {
269                        return size;
270                }
271                /* No DUCK support, so don't waste our time anymore */
272                DUCK.duck_freq = 0;
273        }
274
275        if (packet->buf_control == TRACE_CTRL_PACKET) {
276                packet->buf_control = TRACE_CTRL_EXTERNAL;
277                free(packet->buffer);
278                packet->buffer = 0;
279        }
280
281
282        do {
283                numbytes = dag_available(libtrace);
284                if (numbytes < 0)
285                        return numbytes;
286                if (numbytes == 0)
287                        continue;
288                erfptr = dag_get_record(libtrace);
289        } while (erfptr == NULL);
290        dag_form_packet(erfptr, packet);
291        tv = trace_get_timeval(packet);
292        DUCK.last_pkt = tv.tv_sec;
293        DATA(libtrace)->drops += ntohs(erfptr->lctr);
294        return packet->payload ? htons(erfptr->rlen) : erf_get_framing_length(packet);
295}
296
297static libtrace_eventobj_t trace_event_dag(libtrace_t *trace,
298                                        libtrace_packet_t *packet) {
299        libtrace_eventobj_t event = {0,0,0.0,0};
300        int data;
301
302        data = dag_available(trace);
303        if (data > 0) {
304                event.size = dag_read_packet(trace,packet);
305                //DATA(trace)->dag.diff -= event.size;
306                if (trace->filter) {
307                        if (trace_apply_filter(trace->filter, packet)) {
308                                event.type = TRACE_EVENT_PACKET;
309                        } else {
310                                event.type = TRACE_EVENT_SLEEP;
311                                event.seconds = 0.000001;
312                                return event;
313                        }
314                } else {
315                        event.type = TRACE_EVENT_PACKET;
316                }
317                if (trace->snaplen > 0) {
318                        trace_set_capture_length(packet, trace->snaplen);
319                }
320
321                return event;
322        }
323        assert(data == 0);
324        event.type = TRACE_EVENT_SLEEP;
325        event.seconds = 0.0001;
326        return event;
327}
328
329static uint64_t dag24_get_dropped_packets(libtrace_t *trace)
330{
331        return DATA(trace)->drops;
332}
333
334static void dag_help(void) {
335        printf("dag format module: $Revision$\n");
336        printf("Supported input URIs:\n");
337        printf("\tdag:/dev/dagn\n");
338        printf("\n");
339        printf("\te.g.: dag:/dev/dag0\n");
340        printf("\n");
341        printf("Supported output URIs:\n");
342        printf("\tnone\n");
343        printf("\n");
344}
345
346static struct libtrace_format_t dag = {
347        "dag",
348        "$Id$",
349        TRACE_FORMAT_ERF,
350        dag_init_input,                 /* init_input */
351        dag_config_input,               /* config_input */
352        dag_start_input,                /* start_input */
353        dag_pause_input,                /* pause_input */
354        NULL,                           /* init_output */
355        NULL,                           /* config_output */
356        NULL,                           /* start_output */
357        dag_fin_input,                  /* fin_input */
358        NULL,                           /* fin_output */
359        dag_read_packet,                /* read_packet */
360        NULL,                           /* fin_packet */
361        NULL,                           /* write_packet */
362        erf_get_link_type,              /* get_link_type */
363        erf_get_direction,              /* get_direction */
364        erf_set_direction,              /* set_direction */
365        erf_get_erf_timestamp,          /* get_erf_timestamp */
366        NULL,                           /* get_timeval */
367        NULL,                           /* get_seconds */
368        NULL,                           /* seek_erf */
369        NULL,                           /* seek_timeval */
370        NULL,                           /* seek_seconds */
371        erf_get_capture_length,         /* get_capture_length */
372        erf_get_wire_length,            /* get_wire_length */
373        erf_get_framing_length,         /* get_framing_length */
374        erf_set_capture_length,         /* set_capture_length */
375        NULL,                           /* get_received_packets */
376        NULL,                           /* get_filtered_packets */
377        dag24_get_dropped_packets,      /* get_dropped_packets */
378        NULL,                           /* get_captured_packets */
379        NULL,                           /* get_fd */
380        trace_event_dag,                /* trace_event */
381        dag_help,                       /* help */
382        NULL                            /* next pointer */
383};
384
385void dag_constructor(void) {
386        register_format(&dag);
387}
Note: See TracBrowser for help on using the repository browser.