source: lib/format_rt.c @ 9c4b5e3

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

Added support for nonblocking events for format_rt

  • Property mode set to 100644
File size: 17.1 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 *          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
32#define _GNU_SOURCE
33
34#include "config.h"
35#include "common.h"
36#include "libtrace.h"
37#include "libtrace_int.h"
38#include "format_helper.h"
39#include "parse_cmd.h"
40#include "rt_protocol.h"
41
42#ifdef HAVE_INTTYPES_H
43#  include <inttypes.h>
44#else
45#  error "Can't find inttypes.h - this needs to be fixed"
46#endif
47
48#ifdef HAVE_STDDEF_H
49#  include <stddef.h>
50#else
51# error "Can't find stddef.h - do you define ptrdiff_t elsewhere?"
52#endif
53#include <sys/types.h>
54#include <sys/socket.h>
55#include <sys/un.h>
56#include <sys/mman.h>
57#include <sys/stat.h>
58#include <unistd.h>
59#include <assert.h>
60#include <errno.h>
61#include <netdb.h>
62#include <fcntl.h>
63#include <getopt.h>
64#include <stdio.h>
65#include <string.h>
66#include <stdlib.h>
67
68#define RT_INFO ((struct rt_format_data_t*)libtrace->format_data)
69
70int reliability = 0;
71
72char *rt_deny_reason(uint8_t reason) {
73        char *string = 0;
74
75        switch(reason) {
76                case RT_DENY_WRAPPER:
77                        string = "Rejected by TCP Wrappers";
78                        break;
79                case RT_DENY_FULL:
80                        string = "Max connections reached on server";
81                        break;
82                case RT_DENY_AUTH:
83                        string = "Authentication failed";
84                        break;
85                default:
86                        string = "Unknown reason";
87        }
88
89        return string;
90}
91
92
93struct rt_format_data_t {
94        char *hostname;
95        int port;
96        int input_fd;
97        int reliable;
98        char *pkt_buffer;
99        char *buf_current;
100        int buf_left;
101
102       
103        struct libtrace_t *dummy_erf;
104        struct libtrace_t *dummy_pcap;
105        struct libtrace_t *dummy_wag;
106};
107
108static struct libtrace_format_t rt;
109
110static int rt_connect(struct libtrace_t *libtrace) {
111        struct hostent *he;
112        struct sockaddr_in remote;
113        rt_header_t connect_msg;
114        rt_deny_conn_t deny_hdr;       
115        rt_hello_t hello_opts;
116        uint8_t reason;
117        int oldflags;
118
119       
120        if ((he=gethostbyname(RT_INFO->hostname)) == NULL) {
121                perror("gethostbyname");
122                return -1;
123        }
124        if ((RT_INFO->input_fd = socket(AF_INET, SOCK_STREAM, 0)) == -1) {
125                perror("socket");
126                return -1;
127        }
128
129        remote.sin_family = AF_INET;
130        remote.sin_port = htons(RT_INFO->port);
131        remote.sin_addr = *((struct in_addr *)he->h_addr);
132        bzero(&(remote.sin_zero), 8);
133
134        if (connect(RT_INFO->input_fd, (struct sockaddr *)&remote,
135                                sizeof(struct sockaddr)) == -1) {
136                trace_set_err(libtrace, TRACE_ERR_INIT_FAILED,
137                                "Could not connect to host %s on port %d",
138                                RT_INFO->hostname, RT_INFO->port);
139                return -1;
140        }
141
142       
143#if 0
144        oldflags = fcntl(RT_INFO->input_fd, F_GETFL, 0);
145        if (oldflags == -1) {
146                trace_set_err(libtrace, errno,
147                                "Could not get fd flags from fd %d\n",
148                                RT_INFO->input_fd);
149                return -1;
150        }
151        oldflags |= O_NONBLOCK;
152        if (fcntl(RT_INFO->input_fd, F_SETFL, oldflags) == -1) {
153                trace_set_err(libtrace, errno,
154                                "Could not set fd flags for fd %d\n",
155                                RT_INFO->input_fd);
156                return -1;
157        }
158#endif
159       
160       
161        /* We are connected, now receive message from server */
162       
163        if (recv(RT_INFO->input_fd, &connect_msg, sizeof(rt_header_t), 0) != sizeof(rt_header_t) ) {
164                trace_set_err(libtrace, TRACE_ERR_INIT_FAILED,
165                                "Could not receive connection message from %s",
166                                RT_INFO->hostname);
167                return -1;
168        }
169       
170        switch (connect_msg.type) {
171                case RT_DENY_CONN:
172                       
173                        if (recv(RT_INFO->input_fd, &deny_hdr, 
174                                                sizeof(rt_deny_conn_t),
175                                                0) != sizeof(rt_deny_conn_t)) {
176                                reason = 0;
177                        }       
178                        reason = deny_hdr.reason;
179                        trace_set_err(libtrace, TRACE_ERR_INIT_FAILED,
180                                "Connection attempt is denied: %s",
181                                rt_deny_reason(reason));       
182                        return -1;
183                case RT_HELLO:
184                        /* do something with options */
185                        if (recv(RT_INFO->input_fd, &hello_opts, 
186                                                sizeof(rt_hello_t), 0)
187                                        != sizeof(rt_hello_t)) {
188                                trace_set_err(libtrace, TRACE_ERR_INIT_FAILED,
189                                        "Failed to receive RT_HELLO options");
190                                return -1;
191                        }
192                        reliability = hello_opts.reliable;
193                       
194                        return 0;
195                default:
196                        trace_set_err(libtrace, TRACE_ERR_INIT_FAILED,
197                                        "Unknown message type received: %d",
198                                        connect_msg.type);
199                        return -1;
200        }
201        trace_set_err(libtrace, TRACE_ERR_INIT_FAILED,
202                        "Somehow you managed to reach this unreachable code");
203        return -1;
204}
205
206
207static int rt_init_input(struct libtrace_t *libtrace) {
208        char *scan;
209        char *uridata = libtrace->uridata;
210        libtrace->format_data = malloc(sizeof(struct rt_format_data_t));
211
212        RT_INFO->dummy_erf = NULL;
213        RT_INFO->dummy_pcap = NULL;
214        RT_INFO->dummy_wag = NULL;
215        RT_INFO->pkt_buffer = NULL;
216        RT_INFO->buf_current = NULL;
217        RT_INFO->buf_left = 0;
218       
219        if (strlen(uridata) == 0) {
220                RT_INFO->hostname =
221                        strdup("localhost");
222                RT_INFO->port =
223                        COLLECTOR_PORT;
224        } else {
225                if ((scan = strchr(uridata,':')) == NULL) {
226                        RT_INFO->hostname =
227                                strdup(uridata);
228                        RT_INFO->port =
229                                COLLECTOR_PORT;
230                } else {
231                        RT_INFO->hostname =
232                                (char *)strndup(uridata,
233                                                (scan - uridata));
234                        RT_INFO->port =
235                                atoi(++scan);
236                }
237        }
238
239        return rt_connect(libtrace);
240}
241       
242static int rt_start_input(struct libtrace_t *libtrace) {
243        rt_header_t start_msg;
244
245        start_msg.type = RT_START;
246        start_msg.length = sizeof(rt_start_t);
247
248       
249        /* Need to send start message to server */
250        if (send(RT_INFO->input_fd, &start_msg, sizeof(rt_header_t) +
251                                start_msg.length, 0) != sizeof(rt_header_t)) {
252                printf("Failed to send start message to server\n");
253                return -1;
254        }
255
256        return 0;
257}
258
259static int rt_fin_input(struct libtrace_t *libtrace) {
260        rt_header_t close_msg;
261
262        close_msg.type = RT_CLOSE;
263        close_msg.length = sizeof(rt_close_t);
264       
265        /* Send a close message to the server */
266        if (send(RT_INFO->input_fd, &close_msg, sizeof(rt_header_t) + 
267                                close_msg.length, 0) != sizeof(rt_header_t)
268                                + close_msg.length) {
269                printf("Failed to send close message to server\n");
270       
271        }
272        if (RT_INFO->dummy_erf) 
273                trace_destroy_dead(RT_INFO->dummy_erf);
274               
275        if (RT_INFO->dummy_pcap)
276                trace_destroy_dead(RT_INFO->dummy_pcap);
277
278        if (RT_INFO->dummy_wag)
279                trace_destroy_dead(RT_INFO->dummy_wag);
280        close(RT_INFO->input_fd);
281        free(libtrace->format_data);
282        return 0;
283}
284
285#define RT_BUF_SIZE 4000
286
287static int rt_read(struct libtrace_t *libtrace, void **buffer, size_t len, int block) {
288        int numbytes;
289        int i;
290        char *buf_ptr;
291
292        assert(len <= RT_BUF_SIZE);
293       
294        if (!RT_INFO->pkt_buffer) {
295                RT_INFO->pkt_buffer = malloc(RT_BUF_SIZE);
296                RT_INFO->buf_current = RT_INFO->pkt_buffer;
297                RT_INFO->buf_left = 0;
298        }
299
300        if (block)
301                block=0;
302        else
303                block=MSG_DONTWAIT;
304
305       
306        if (len > RT_INFO->buf_left) {
307                memcpy(RT_INFO->pkt_buffer, RT_INFO->buf_current, 
308                                RT_INFO->buf_left);
309                RT_INFO->buf_current = RT_INFO->pkt_buffer;
310
311#ifndef MSG_NOSIGNAL
312#  define MSG_NOSIGNAL 0
313#endif
314                while (len > RT_INFO->buf_left) {
315                        if ((numbytes = recv(RT_INFO->input_fd,
316                                                RT_INFO->pkt_buffer + 
317                                                RT_INFO->buf_left,
318                                                RT_BUF_SIZE-RT_INFO->buf_left,
319                                                MSG_NOSIGNAL|block)) <= 0) {
320                                if (numbytes == 0) {
321                                        trace_set_err(libtrace, TRACE_ERR_BAD_PACKET, 
322                                                        "No data received");
323                                        return -1;
324                                }
325                               
326                                if (errno == EINTR) {
327                                        /* ignore EINTR in case
328                                         * a caller is using signals
329                                         */
330                                        continue;
331                                }
332                                if (errno == EAGAIN) {
333                                        trace_set_err(libtrace,
334                                                        EAGAIN,
335                                                        "EAGAIN");
336                                        return -1;
337                                }
338                               
339                                perror("recv");
340                                trace_set_err(libtrace, TRACE_ERR_RECV_FAILED,
341                                                "Failed to read data into rt recv buffer");
342                                return -1;
343                        }
344                        /*
345                        buf_ptr = RT_INFO->pkt_buffer;
346                        for (i = 0; i < RT_BUF_SIZE ; i++) {
347                                       
348                                printf("%02x", (unsigned char)*buf_ptr);
349                                buf_ptr ++;
350                        }
351                        printf("\n");
352                        */
353                        RT_INFO->buf_left+=numbytes;
354                }
355
356        }
357        *buffer = RT_INFO->buf_current;
358        RT_INFO->buf_current += len;
359        RT_INFO->buf_left -= len;
360        assert(RT_INFO->buf_left >= 0);
361        return len;
362}
363
364
365static int rt_set_format(libtrace_t *libtrace, libtrace_packet_t *packet) 
366{
367       
368        if (packet->type >= RT_DATA_PCAP) {
369                if (!RT_INFO->dummy_pcap) {
370                        RT_INFO->dummy_pcap = trace_create_dead("pcap:-");
371                }
372                packet->trace = RT_INFO->dummy_pcap;
373                return 0;       
374        }
375
376        switch (packet->type) {
377                case RT_DATA_ERF:
378                        if (!RT_INFO->dummy_erf) {
379                                RT_INFO->dummy_erf = trace_create_dead("erf:-");
380                        }
381                        packet->trace = RT_INFO->dummy_erf;
382                        break;
383                case RT_DATA_WAG:
384                        if (!RT_INFO->dummy_wag) {
385                                RT_INFO->dummy_wag = trace_create_dead("wtf:-");
386                        }
387                        packet->trace = RT_INFO->dummy_wag;
388                        break;
389                case RT_DATA_LEGACY_ETH:
390                case RT_DATA_LEGACY_ATM:
391                case RT_DATA_LEGACY_POS:
392                        printf("Sending legacy over RT is currently not supported\n");
393                        trace_set_err(libtrace, TRACE_ERR_BAD_PACKET, "Legacy packet cannot be sent over rt");
394                        return -1;
395                default:
396                        printf("Unrecognised format: %d\n", packet->type);
397                        trace_set_err(libtrace, TRACE_ERR_BAD_PACKET, "Unrecognised packet format");
398                        return -1;
399        }
400        return 0; /* success */
401}               
402
403static void rt_set_payload(struct libtrace_packet_t *packet) {
404        dag_record_t *erfptr;
405       
406        switch (packet->type) {
407                case RT_DATA_ERF:
408                        erfptr = (dag_record_t *)packet->header;
409                       
410                        if (erfptr->flags.rxerror == 1) {
411                                packet->payload = NULL;
412                                break;
413                        }
414                        /* else drop into the default case */
415                default:
416                        packet->payload = (char *)packet->buffer +
417                                trace_get_framing_length(packet);
418                        break;
419        }
420}
421
422static int rt_send_ack(struct libtrace_t *libtrace, 
423                uint32_t seqno)  {
424       
425        static char *ack_buffer = 0;
426        char *buf_ptr;
427        int numbytes = 0;
428        int to_write = 0;
429        rt_header_t *hdr;
430        rt_ack_t *ack_hdr;
431       
432        if (!ack_buffer) {
433                ack_buffer = malloc(sizeof(rt_header_t) + sizeof(rt_ack_t));
434        }
435       
436        hdr = (rt_header_t *) ack_buffer;
437        ack_hdr = (rt_ack_t *) (ack_buffer + sizeof(rt_header_t));
438       
439        hdr->type = RT_ACK;
440        hdr->length = sizeof(rt_ack_t);
441
442        ack_hdr->sequence = seqno;
443       
444        to_write = hdr->length + sizeof(rt_header_t);
445        buf_ptr = ack_buffer;
446
447       
448        while (to_write > 0) {
449                numbytes = send(RT_INFO->input_fd, buf_ptr, to_write, 0); 
450                if (numbytes == -1) {
451                        if (errno == EINTR || errno == EAGAIN) {
452                                continue;
453                        }
454                        else {
455                                printf("Error sending ack\n");
456                                trace_set_err(libtrace, TRACE_ERR_BAD_PACKET, 
457                                                "Error sending ack");
458                                return -1;
459                        }
460                }
461                to_write = to_write - numbytes;
462                buf_ptr = buf_ptr + to_write;
463               
464        }
465
466        return 1;
467}
468
469       
470static int rt_read_packet_versatile(libtrace_t *libtrace,
471                libtrace_packet_t *packet,int blocking) {
472        rt_header_t rt_hdr;
473        rt_header_t *pkt_hdr = &rt_hdr;
474        int pkt_size = 0;
475       
476       
477        if (packet->buf_control == TRACE_CTRL_EXTERNAL || !packet->buffer) {
478                packet->buf_control = TRACE_CTRL_PACKET;
479                packet->buffer = malloc(LIBTRACE_PACKET_BUFSIZE);
480        } 
481
482
483        /* FIXME: Better error handling required */
484        if (rt_read(libtrace, (void **)&pkt_hdr, sizeof(rt_header_t),blocking) !=
485                        sizeof(rt_header_t)) {
486                return -1;
487        }
488
489        packet->type = pkt_hdr->type;
490        pkt_size = pkt_hdr->length;
491        packet->size = pkt_hdr->length;
492
493        if (packet->type >= RT_DATA_SIMPLE) {
494                if (rt_read(libtrace, &packet->buffer, pkt_size,1) != pkt_size) {
495                        printf("Error receiving packet\n");
496                        return -1;
497                }
498                packet->header = packet->buffer;
499               
500                if (rt_set_format(libtrace, packet) < 0) {
501                        return -1;
502                }
503                rt_set_payload(packet);
504
505                if (reliability > 0) {
506                       
507                        if (rt_send_ack(libtrace, pkt_hdr->sequence) 
508                                        == -1)
509                        {
510                                return -1;
511                        }
512                }
513        } else {
514                switch(packet->type) {
515                        case RT_STATUS:
516                        case RT_DUCK:
517                                if (rt_read(libtrace, &packet->buffer, 
518                                                        pkt_size,1) !=
519                                                pkt_size) {
520                                        printf("Error receiving status packet\n");
521                                        return -1;
522                                }
523                                packet->header = 0;
524                                packet->payload = packet->buffer;
525                                break;
526                        case RT_END_DATA:
527                                return 0;
528                        case RT_PAUSE_ACK:
529                                /* FIXME: Do something useful */
530                                break;
531                        case RT_OPTION:
532                                /* FIXME: Do something useful here as well */
533                                break;
534                        case RT_KEYCHANGE:
535                                break;
536                        default:
537                                printf("Bad rt type for client receipt: %d\n",
538                                        pkt_hdr->type);
539                }
540        }
541        /* Return the number of bytes read from the stream */
542        return packet->size; 
543}
544
545static int rt_read_packet(libtrace_t *libtrace,
546                libtrace_packet_t *packet) {
547        rt_read_packet_versatile(libtrace,packet,1);
548}
549
550
551static int rt_get_capture_length(const struct libtrace_packet_t *packet) {
552        switch (packet->type) {
553                case RT_DUCK:
554                        return sizeof(rt_duck_t);
555                case RT_STATUS:
556                        return sizeof(rt_status_t);
557                case RT_HELLO:
558                        return sizeof(rt_hello_t);
559                case RT_START:
560                        return sizeof(rt_start_t);
561                case RT_ACK:
562                        return sizeof(rt_ack_t);
563                case RT_END_DATA:
564                        return sizeof(rt_end_data_t);
565                case RT_CLOSE:
566                        return sizeof(rt_close_t);
567                case RT_DENY_CONN:
568                        return sizeof(rt_deny_conn_t);
569                case RT_PAUSE:
570                        return sizeof(rt_pause_t);
571                case RT_PAUSE_ACK:
572                        return sizeof(rt_pause_ack_t);
573                case RT_OPTION:
574                        return sizeof(rt_option_t);
575                case RT_KEYCHANGE:
576                        return sizeof(rt_keychange_t);
577        }
578        printf("Unknown type: %d\n", packet->type);
579        return 0;
580}
581
582static int rt_get_wire_length(const libtrace_packet_t *packet) {
583        return 0;
584}
585                       
586static int rt_get_framing_length(const libtrace_packet_t *packet) {
587        return 0;
588}
589
590static int rt_get_fd(const libtrace_t *trace) {
591        return ((struct rt_format_data_t *)trace->format_data)->input_fd;
592}
593
594struct libtrace_eventobj_t trace_event_rt(struct libtrace_t *trace, struct libtrace_packet_t *packet) {
595        struct libtrace_eventobj_t event = {0,0,0.0,0};
596        libtrace_err_t read_err;
597        int data;
598
599        assert(trace);
600        assert(packet);
601       
602        if (trace->format->get_fd) {
603                event.fd = trace->format->get_fd(trace);
604        } else {
605                event.fd = 0;
606        }
607
608        event.size = rt_read_packet_versatile(trace, packet, 0);
609        if (event.size == -1) {
610                read_err = trace_get_err(trace);
611                if (read_err.err_num == EAGAIN) {
612                        event.type = TRACE_EVENT_IOWAIT;
613                }
614                else {
615                        printf("packet error\n");
616                        event.type = TRACE_EVENT_PACKET;
617                }
618        } else if (event.size == 0) {
619                event.type = TRACE_EVENT_TERMINATE;
620               
621        }       
622        else {
623                event.type = TRACE_EVENT_PACKET;
624        }
625       
626        return event;
627}
628
629static void rt_help() {
630        printf("rt format module\n");
631        printf("Supported input URIs:\n");
632        printf("\trt:hostname:port\n");
633        printf("\trt:hostname (connects on default port)\n");
634        printf("\n");
635        printf("\te.g.: rt:localhost\n");
636        printf("\te.g.: rt:localhost:32500\n");
637        printf("\n");
638
639}
640
641
642static struct libtrace_format_t rt = {
643        "rt",
644        "$Id$",
645        TRACE_FORMAT_RT,
646        rt_init_input,                  /* init_input */
647        NULL,                           /* config_input */
648        rt_start_input,                 /* start_input */
649        NULL,                           /* init_output */
650        NULL,                           /* config_output */
651        NULL,                           /* start_output */
652        NULL,                           /* pause_output */
653        rt_fin_input,                   /* fin_input */
654        NULL,                           /* fin_output */
655        rt_read_packet,                 /* read_packet */
656        NULL,                           /* fin_packet */
657        NULL,                           /* write_packet */
658        NULL,                           /* get_link_type */
659        NULL,                           /* get_direction */
660        NULL,                           /* set_direction */
661        NULL,                           /* get_erf_timestamp */
662        NULL,                           /* get_timeval */
663        NULL,                           /* get_seconds */
664        NULL,                           /* seek_erf */
665        NULL,                           /* seek_timeval */
666        NULL,                           /* seek_seconds */
667        rt_get_capture_length,          /* get_capture_length */
668        rt_get_wire_length,                     /* get_wire_length */
669        rt_get_framing_length,          /* get_framing_length */
670        NULL,                           /* set_capture_length */
671        rt_get_fd,                      /* get_fd */
672        trace_event_rt,             /* trace_event */
673        rt_help,                        /* help */
674        NULL                            /* next pointer */
675};
676
677void __attribute__((constructor)) rt_constructor() {
678        register_format(&rt);
679}
Note: See TracBrowser for help on using the repository browser.