source: lib/format_dag25.c @ 52656ed

4.0.1-hotfixescachetimestampsdevelopdpdk-ndagetsilivegetfragoffhelplibtrace4ndag_formatpfringrc-4.0.1rc-4.0.2rc-4.0.3rc-4.0.4ringdecrementfixringperformanceringtimestampfixes
Last change on this file since 52656ed was 52656ed, checked in by Shane Alcock <salcock@…>, 14 years ago
  • Updated autotools to detect DAG 3.X drivers. The new detection method will also catch DAG 2.5, but this is not a problem because the API is the same so DAG 2.5 and DAG 3 can use the same code within libtrace.
  • Changed format_dag25.c to manage the dag stream manually, similar to format_dag24.c.
  • Libtrace programs using DAG 2.5 and greater should now detach the DAG stream properly, provided they are exited cleanly.
  • The snaplen can now be configured for DAG 2.5 and DAG 3.0 inputs
  • Property mode set to 100644
File size: 12.5 KB
Line 
1/*
2 * This file is part of libtrace
3 *
4 * Copyright (c) 2007 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
49#ifdef WIN32
50#  include <io.h>
51#  include <share.h>
52#  define PATH_MAX _MAX_PATH
53#  define snprintf sprintf_s
54#else
55#  include <netdb.h>
56#  ifndef PATH_MAX
57#       define PATH_MAX 4096
58#  endif
59#  include <sys/ioctl.h>
60#endif
61
62#define DATA(x) ((struct dag_format_data_t *)x->format_data)
63#define FORMAT_DATA DATA(libtrace)
64#define DUCK FORMAT_DATA->duck
65static struct libtrace_format_t dag;
66
67struct dag_format_data_t {
68        struct {
69                uint32_t last_duck;
70                uint32_t duck_freq;
71                uint32_t last_pkt;
72                libtrace_t *dummy_duck;
73        } duck;
74
75        int fd;
76        unsigned int dagstream;
77        int stream_attached;
78        uint8_t *bottom;
79        uint8_t *top;
80};
81
82static int dag_init_input(libtrace_t *libtrace) {
83        struct stat buf;
84        libtrace->format_data = (struct dag_format_data_t *)
85                malloc(sizeof(struct dag_format_data_t));
86        if (stat(libtrace->uridata, &buf) == -1) {
87                trace_set_err(libtrace,errno,"stat(%s)",libtrace->uridata);
88                return -1;
89        }
90
91        /* For now, we don't offer the ability to select the stream */
92        FORMAT_DATA->dagstream = 0;
93
94        if (S_ISCHR(buf.st_mode)) {
95                if((FORMAT_DATA->fd = dag_open(libtrace->uridata)) < 0) {
96                        trace_set_err(libtrace,errno,"Cannot open DAG %s",
97                                        libtrace->uridata);
98                        return -1;
99                }
100        } else {
101                trace_set_err(libtrace,errno,"Not a valid dag device: %s",
102                                libtrace->uridata);
103                return -1;
104        }
105
106        DUCK.last_duck = 0;
107        DUCK.duck_freq = 0;
108        DUCK.last_pkt = 0;
109        DUCK.dummy_duck = NULL;
110        FORMAT_DATA->stream_attached = 0;
111       
112        return 0;
113}
114       
115static int dag_config_input(libtrace_t *libtrace, trace_option_t option,
116                                void *data) {
117        char conf_str[4096];
118        switch(option) {
119                case TRACE_OPTION_META_FREQ:
120                        DUCK.duck_freq = *(int *)data;
121                        return 0;
122                case TRACE_OPTION_SNAPLEN:
123                        snprintf(conf_str, 4096, "varlen slen=%i", *(int *)data);
124                        if (dag_configure(FORMAT_DATA->fd, conf_str) != 0) {
125                                trace_set_err(libtrace, errno, "Failed to configure snaplen on DAG card: %s", libtrace->uridata);
126                                return -1;
127                        }
128                        return 0;
129                case TRACE_OPTION_PROMISC:
130                        /* DAG already operates in a promisc fashion */
131                        return -1;
132                case TRACE_OPTION_FILTER:
133                        return -1;
134                case TRACE_OPTION_EVENT_REALTIME:
135                        return -1;
136        }
137        return -1;
138}
139
140static int dag_start_input(libtrace_t *libtrace) {
141        struct timeval zero, nopoll;
142        uint8_t *top, *bottom;
143        uint8_t diff = 0;
144        top = bottom = NULL;
145
146        zero.tv_sec = 0;
147        zero.tv_usec = 0;
148        nopoll = zero;
149
150
151       
152        if (dag_attach_stream(FORMAT_DATA->fd, 
153                                FORMAT_DATA->dagstream, 0, 0) < 0) {
154                trace_set_err(libtrace, errno, "Cannot attach DAG stream");
155                return -1;
156        }
157
158        if (dag_start_stream(FORMAT_DATA->fd, 
159                                FORMAT_DATA->dagstream) < 0) {
160                trace_set_err(libtrace, errno, "Cannot start DAG stream");
161                return -1;
162        }
163        FORMAT_DATA->stream_attached = 1;
164        /* We don't want the dag card to do any sleeping */
165        dag_set_stream_poll(FORMAT_DATA->fd, 
166                                FORMAT_DATA->dagstream, 0, &zero, 
167                                &nopoll);
168       
169        /* Should probably flush the memory hole now */
170       
171        do {
172                top = dag_advance_stream(FORMAT_DATA->fd,
173                                        FORMAT_DATA->dagstream,
174                                        &bottom);
175                assert(top && bottom);
176                diff = top - bottom;
177                bottom -= diff;
178        } while (diff != 0);
179
180        return 0;
181}
182
183static int dag_pause_input(libtrace_t *libtrace) {
184        if (dag_stop_stream(FORMAT_DATA->fd, 
185                                FORMAT_DATA->dagstream) < 0) {
186                trace_set_err(libtrace, errno, "Could not stop DAG stream");
187                return -1;
188        }
189        if (dag_detach_stream(FORMAT_DATA->fd, 
190                                FORMAT_DATA->dagstream) < 0) {
191                trace_set_err(libtrace, errno, "Could not detach DAG stream");
192                return -1;
193        }
194        FORMAT_DATA->stream_attached = 0;
195        return 0;
196}
197
198static int dag_fin_input(libtrace_t *libtrace) {
199        if (FORMAT_DATA->stream_attached)
200                dag_pause_input(libtrace);
201        dag_close(FORMAT_DATA->fd);
202        if (DUCK.dummy_duck)
203                trace_destroy_dead(DUCK.dummy_duck);
204        free(libtrace->format_data);
205        return 0; /* success */
206}
207
208static int dag_get_duckinfo(libtrace_t *libtrace,
209                                libtrace_packet_t *packet) {
210        if (packet->buf_control == TRACE_CTRL_EXTERNAL ||
211                        !packet->buffer) {
212                packet->buffer = malloc(LIBTRACE_PACKET_BUFSIZE);
213                packet->buf_control = TRACE_CTRL_PACKET;
214                if (!packet->buffer) {
215                        trace_set_err(libtrace, errno,
216                                        "Cannot allocate packet buffer");
217                        return -1;
218                }
219        }
220
221        packet->header = 0;
222        packet->payload = packet->buffer;
223
224        /* No need to check if we can get DUCK or not - we're modern
225         * enough */
226        if ((ioctl(FORMAT_DATA->fd, DAGIOCDUCK, (duckinf_t *)packet->payload)
227                                < 0)) {
228                trace_set_err(libtrace, errno, "Error using DAGIOCDUCK");
229                return -1;
230        }
231
232        packet->type = TRACE_RT_DUCK_2_5;
233        if (!DUCK.dummy_duck)
234                DUCK.dummy_duck = trace_create_dead("rt:localhost:3434");
235        packet->trace = DUCK.dummy_duck;
236        return sizeof(duckinf_t);
237}
238
239static int dag_available(libtrace_t *libtrace) {
240        uint32_t diff = FORMAT_DATA->top - FORMAT_DATA->bottom;
241        if (diff >= dag_record_size)
242                return diff;
243        FORMAT_DATA->top = dag_advance_stream(FORMAT_DATA->fd, 
244                        FORMAT_DATA->dagstream, 
245                        &(FORMAT_DATA->bottom));
246        if (FORMAT_DATA->top == NULL) {
247                trace_set_err(libtrace, errno, "dag_advance_stream failed!");
248                return -1;
249        }
250        diff = FORMAT_DATA->top - FORMAT_DATA->bottom;
251        return diff;
252}
253
254static dag_record_t *dag_get_record(libtrace_t *libtrace) {
255        dag_record_t *erfptr = NULL;
256        uint16_t size;
257        erfptr = (dag_record_t *)FORMAT_DATA->bottom;
258        if (!erfptr)
259                return NULL;
260        size = ntohs(erfptr->rlen);
261        assert( size >= dag_record_size );
262        /* Make certain we have the full packet available */
263        if (size > (FORMAT_DATA->top - FORMAT_DATA->bottom))
264                return NULL;
265        FORMAT_DATA->bottom += size;
266        return erfptr;
267}
268
269static void dag_form_packet(dag_record_t *erfptr, libtrace_packet_t *packet) {
270        packet->buffer = erfptr;
271        packet->header = erfptr;
272        packet->type = TRACE_RT_DATA_ERF;
273        if (erfptr->flags.rxerror == 1) {
274                /* rxerror means the payload is corrupt - drop it
275                 * by tweaking rlen */
276                packet->payload = NULL;
277                erfptr->rlen = htons(erf_get_framing_length(packet));
278        } else {
279                packet->payload = (char*)packet->buffer
280                        + erf_get_framing_length(packet);
281        }
282
283}
284
285
286static int dag_read_packet(libtrace_t *libtrace, libtrace_packet_t *packet) {
287        int size = 0;
288        struct timeval tv;
289        dag_record_t *erfptr = NULL;
290        int numbytes = 0;
291       
292        if (DUCK.last_pkt - DUCK.last_duck > DUCK.duck_freq &&
293                        DUCK.duck_freq != 0) {
294                size = dag_get_duckinfo(libtrace, packet);
295                DUCK.last_duck = DUCK.last_pkt;
296                if (size != 0) {
297                        return size;
298                }
299                /* No DUCK support, so don't waste our time anymore */
300                DUCK.duck_freq = 0;
301        }
302
303        if (packet->buf_control == TRACE_CTRL_PACKET) {
304                packet->buf_control = TRACE_CTRL_EXTERNAL;
305                free(packet->buffer);
306                packet->buffer = 0;
307        }
308
309        do {
310                numbytes = dag_available(libtrace);
311                if (numbytes < 0)
312                        return numbytes;
313                if (numbytes < dag_record_size)
314                        /* Block until we see a packet */
315                        continue;
316                erfptr = dag_get_record(libtrace);
317        } while (erfptr == NULL);
318
319        dag_form_packet(erfptr, packet);
320        tv = trace_get_timeval(packet);
321        DUCK.last_pkt = tv.tv_sec;
322        return packet->payload ? htons(erfptr->rlen) : 
323                                erf_get_framing_length(packet);
324}
325
326static libtrace_eventobj_t trace_event_dag(libtrace_t *trace,
327                                        libtrace_packet_t *packet) {
328        libtrace_eventobj_t event = {0,0,0.0,0};
329        dag_record_t *erfptr = NULL;
330       
331        erfptr = dag_get_record(trace);
332        if (erfptr == NULL) {
333                /* No packet available */
334                event.type = TRACE_EVENT_SLEEP;
335                event.seconds = 0.0001;
336                return event;
337        }
338        dag_form_packet(erfptr, packet);
339        event.size = trace_get_capture_length(packet) + trace_get_framing_length(packet);
340        if (trace->filter) {
341                if (trace_apply_filter(trace->filter, packet)) {
342                        event.type = TRACE_EVENT_PACKET;
343                } else {
344                        event.type = TRACE_EVENT_SLEEP;
345                        event.seconds = 0.000001;
346                        return event;
347                }
348        } else {
349                event.type = TRACE_EVENT_PACKET;
350        }
351
352        if (trace->snaplen > 0) {
353                trace_set_capture_length(packet, trace->snaplen);
354        }
355
356        return event;
357}
358
359
360static void dag_help(void) {
361        printf("dag format module: $Revision$\n");
362        printf("Supported input URIs:\n");
363        printf("\tdag:/dev/dagn\n");
364        printf("\n");
365        printf("\te.g.: dag:/dev/dag0\n");
366        printf("\n");
367        printf("Supported output URIs:\n");
368        printf("\tnone\n");
369        printf("\n");
370}
371
372static struct libtrace_format_t dag = {
373        "dag",
374        "$Id$",
375        TRACE_FORMAT_ERF,
376        dag_init_input,                 /* init_input */
377        dag_config_input,               /* config_input */
378        dag_start_input,                /* start_input */
379        dag_pause_input,                /* pause_input */
380        NULL,                           /* init_output */
381        NULL,                           /* config_output */
382        NULL,                           /* start_output */
383        dag_fin_input,                  /* fin_input */
384        NULL,                           /* fin_output */
385        dag_read_packet,                /* read_packet */
386        NULL,                           /* fin_packet */
387        NULL,                           /* write_packet */
388        erf_get_link_type,              /* get_link_type */
389        erf_get_direction,              /* get_direction */
390        erf_set_direction,              /* set_direction */
391        erf_get_erf_timestamp,          /* get_erf_timestamp */
392        NULL,                           /* get_timeval */
393        NULL,                           /* get_seconds */
394        NULL,                           /* seek_erf */
395        NULL,                           /* seek_timeval */
396        NULL,                           /* seek_seconds */
397        erf_get_capture_length,         /* get_capture_length */
398        erf_get_wire_length,            /* get_wire_length */
399        erf_get_framing_length,         /* get_framing_length */
400        erf_set_capture_length,         /* set_capture_length */
401        NULL,                           /* get_fd */
402        trace_event_dag,                /* trace_event */
403        dag_help,                       /* help */
404        NULL                            /* next pointer */
405};
406
407void dag_constructor(void) {
408        register_format(&dag);
409}
Note: See TracBrowser for help on using the repository browser.