source: lib/format_rt.c @ 85a79b0

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

Fixed lots of warnings

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