Changeset 538

Timestamp:

12/09/05 11:03:51 (7 years ago)

Author:

spa1

Message:

Separated wag device and wag trace into two separate uris - wag and wtf
respectively. Fixed wag_read to work correctly with wag drivers.
Updated anything that reads from the wag header as it is now stored
in network byte order.

File:

• branches/libtrace-2_0_25/lib/format_wag.c (modified) (15 diffs)

branches/libtrace-2_0_25/lib/format_wag.c

@@ -77 +77 @@
 
 static struct libtrace_format_t wag;
+static struct libtrace_format_t wag_trace;
 
 #define CONNINFO libtrace->format_data->conn_info
@@ -126 +127 @@
 static int wag_init_input(struct libtrace_t *libtrace) {
         struct stat buf;
-        struct sockaddr_un unix_sock;
+        //struct sockaddr_un unix_sock;
         libtrace->format_data = (struct libtrace_format_data_t *) 
                 calloc(1,sizeof(struct libtrace_format_data_t));
         CONNINFO.path = libtrace->uridata;
+        
+        if (stat(CONNINFO.path,&buf) == -1 ) {
+                perror("stat");
+                return 0;
+        }
+        if (S_ISCHR(buf.st_mode)) {
+                libtrace->sourcetype = DEVICE;
+                                
+                INPUT.fd = open(CONNINFO.path, O_RDONLY);
+
+        } else {
+                fprintf(stderr, "%s is not a valid char device, exiting\n",
+                                CONNINFO.path);
+                return 0;
+                
+        }
+        return 1;
+}
+
+static int wtf_init_input(struct libtrace_t *libtrace) {
+        struct stat buf;
+        struct sockaddr_un unix_sock;
+
+        libtrace->format_data = (struct libtrace_format_data_t *)
+                calloc(1,sizeof(struct libtrace_format_data_t));
+        CONNINFO.path = libtrace->uridata;
+
         if (!strncmp(CONNINFO.path,"-",1)) {
                 // STDIN
@@ -136 +164 @@
 
         } else {
+
+
+                // Do we need this socket stuff at all??
+                // If we do, put it into wag_init_input as it uses
+                // INPUT.fd
+
+                /*
                 if (stat(CONNINFO.path,&buf) == -1 ) {
                         perror("stat");
@@ -161 +196 @@
                         }
                 } else { 
+                */
                         // TRACE
                         libtrace->sourcetype = TRACE;
@@ -172 +208 @@
                                         O_LARGEFILE), "r");
 
-                }
+                //}
         }
         return 1;
 }
 
-static int wag_init_output(struct libtrace_out_t *libtrace) {
+
+static int wtf_init_output(struct libtrace_out_t *libtrace) {
         char *filemode = 0;
         libtrace->format_data = (struct libtrace_format_data_out_t *)
@@ -192 +229 @@
                                         libtrace->uridata,
                                         O_CREAT | O_LARGEFILE | O_WRONLY,
-                                        S_IRUSR | S_IWUSR), filemode);
+                                        S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP), filemode);
         }
 
@@ -198 +235 @@
 }
 
-static int wag_config_output(struct libtrace_out_t *libtrace, int argc, char *argv[]) {
+static int wtf_config_output(struct libtrace_out_t *libtrace, int argc, char *argv[]) {
 #if HAVE_ZLIB
         int opt;
@@ -227 +264 @@
 
 static int wag_fin_input(struct libtrace_t *libtrace) {
+        close(INPUT.fd);
+        return 0;
+}
+
+static int wtf_fin_input(struct libtrace_t *libtrace) {
         LIBTRACE_CLOSE(INPUT.file);
         return 0;
 }
 
-static int wag_fin_output(struct libtrace_out_t *libtrace) {
+static int wtf_fin_output(struct libtrace_out_t *libtrace) {
         LIBTRACE_CLOSE(OUTPUT.file);
         return 0;
@@ -238 +280 @@
 static int wag_read(struct libtrace_t *libtrace, void *buffer, size_t len) {
         int numbytes;
+        int framesize;
         assert(libtrace);
 
@@ -243 +286 @@
                 buffer = malloc(len);
 
-        while(1) {
-                switch(libtrace->sourcetype) {
-                        case DEVICE:
-                                if ((numbytes=read(INPUT.fd, 
-                                                                buffer, 
-                                                                len)) == -1) {
-                                        perror("read");
-                                        return -1;
-                                }
-                                break;
-                        default:
-                                if ((numbytes=LIBTRACE_READ(INPUT.file,
-                                                                buffer,
-                                                                len)) == -1) {
-                                        perror("libtrace_read");
-                                        return -1;
-                                }
-                }
-                break;
-        }
-        return numbytes;
-
+        // read in wag_frame_hdr
+        if ((numbytes = read(INPUT.fd, 
+                        buffer,
+                        sizeof(struct wag_frame_hdr))) != sizeof(struct wag_frame_hdr)) {
+                return -1;
+        }
+         
+        framesize = ntohs(((struct wag_frame_hdr *)buffer)->size);
+
+        if (framesize > len) {
+                return -1;
+        }
+
+        // read in remainder of packet
+        if((numbytes = read(INPUT.fd,
+                        buffer + sizeof(struct wag_frame_hdr),
+                        framesize - sizeof(struct wag_frame_hdr))) != 
+                        (framesize - sizeof(struct wag_frame_hdr))) {
+                
+                return -1;
+          
+        }
+
+        return framesize;
 }
 
@@ -270 +315 @@
 static int wag_read_packet(struct libtrace_t *libtrace, struct libtrace_packet_t *packet) {
         int numbytes;
+        
+        char buf[RP_BUFSIZE];
+        
+        
+        packet->trace = libtrace;
+        
+        if ((numbytes = wag_read(libtrace, buf, RP_BUFSIZE)) <= 0) {
+            
+                return numbytes;
+        }
+
+        memcpy(packet->buffer, buf, numbytes);
+        
+        packet->status.type = RT_DATA;
+        packet->status.message = 0;
+        packet->size = numbytes;
+        return numbytes;
+}
+
+static int wtf_read_packet(struct libtrace_t *libtrace, struct libtrace_packet_t *packet) {
+        int numbytes;
+        void *buffer = packet->buffer;
+        void *buffer2 = packet->buffer;
+        int framesize;
         int size;
-        char buf[RP_BUFSIZE];
-        int read_required = 0;
-
-        void *buffer = 0;
-
-        packet->trace = libtrace;
-        buffer = packet->buffer;
-        
-
-        do {
-                if (tracefifo_out_available(libtrace->fifo) == 0 || read_required) {
-                        if ((numbytes = wag_read(libtrace,buf,RP_BUFSIZE)) <= 0) {
-                                return numbytes;
-                        }
-                        assert(libtrace->fifo);
-                        tracefifo_write(libtrace->fifo,buf,numbytes);
-                        read_required = 0;
-                }
-                // read in wag_frame_hdr
-                if ((numbytes = tracefifo_out_read(libtrace->fifo, 
-                                                buffer,
-                                                sizeof(struct wag_frame_hdr)))
-                                == 0 ) {
-                        tracefifo_out_reset(libtrace->fifo);
-                        read_required = 1;
-                        continue;
-                }
-                
-                size = ntohs(((struct wag_frame_hdr *)buffer)->size);
-
-                // wag isn't in network byte order yet
-                //size = htons(size);
-                //printf("%d %d\n",size,htons(size));
-
-                // read in full packet
-                if((numbytes = tracefifo_out_read(libtrace->fifo,buffer,size)) == 0) {
-                        tracefifo_out_reset(libtrace->fifo);
-                        read_required = 1;
-                        continue;
-                }
-
-                // have the whole packet
-                tracefifo_out_update(libtrace->fifo,size);
-                tracefifo_ack_update(libtrace->fifo,size);
-
-                packet->status.type = RT_DATA;
-                packet->status.message = 0;
-                packet->size = numbytes;
-                return numbytes;
-        } while(1);
-}
-
-static int wag_write_packet(struct libtrace_out_t *libtrace, const struct libtrace_packet_t *packet) {
+        
+        if ((numbytes = LIBTRACE_READ(INPUT.file, buffer, sizeof(struct wag_frame_hdr))) == -1) {
+                perror("libtrace_read");
+                return -1;
+        }
+
+        if (numbytes == 0) {
+                return 0;
+        }
+
+        framesize = ntohs(((struct wag_frame_hdr *)buffer)->size);
+        buffer2 = buffer + sizeof(struct wag_frame_hdr);
+        size = framesize - sizeof(struct wag_frame_hdr);
+        assert(size < LIBTRACE_PACKET_BUFSIZE);
+
+        
+        if ((numbytes=LIBTRACE_READ(INPUT.file, buffer2, size)) != size) {
+                perror("libtrace read");
+                return -1;
+        }
+
+        packet->status.type = RT_DATA;
+        packet->status.message = 0;
+        packet->size = framesize;
+        return framesize;
+        
+}                               
+        
+static int wtf_write_packet(struct libtrace_out_t *libtrace, const struct libtrace_packet_t *packet) {
         int numbytes =0 ;
-        if (packet->trace->format != &wag) {
-                fprintf(stderr,"Cannot convert from wag to %s format yet\n",
+        if (packet->trace->format != &wag_trace) {
+                fprintf(stderr,"Cannot convert from wag trace format to %s format yet\n",
                                 packet->trace->format->name);
                 return -1;
@@ -358 +403 @@
         struct wag_data_frame *wagptr = (struct wag_data_frame *)packet->buffer;
         uint64_t timestamp = 0;
-        timestamp = (((uint64_t)wagptr->ts.secs) << 32) + wagptr->ts.subsecs;
-        //timestamp |= (uint64_t)wagptr->ts.secs<<32;
-        //timestamp = ((timestamp%44000000)*(UINT_MAX/44000000)) 
-        //      | ((timestamp/44000000)<<32);
+        timestamp = ((uint64_t)(ntohl(wagptr->ts.secs)) << 32) | (uint64_t)(ntohl(wagptr->ts.subsecs));
         return timestamp;
 }
@@ -397 +439 @@
         printf("Supported input URIs:\n");
         printf("\twag:/dev/wagn\n");
-        printf("\twag:/path/to/trace.wag\n");
-        printf("\twag:/path/to/trace.wag.gz\n");
         printf("\n");
         printf("\te.g.: wag:/dev/wag0\n");
-        printf("\te.g.: wag:/tmp/trace.wag.gz\n");
         printf("\n");
         printf("Supported output URIs:\n");
-        printf("\tnone\n");
+        printf("\tNone\n");
+        printf("\n");
+}
+
+static void wtf_help() {
+        printf("wag trace format module: $Revision$\n");
+        printf("Supported input URIs:\n");
+        printf("\twtf:/path/to/trace.wag\n");
+        printf("\twtf:/path/to/trace.wag.gz\n");
+        printf("\n");
+        printf("\te.g.: wtf:/tmp/trace.wag.gz\n");
+        printf("\n");
+        printf("Supported output URIs:\n");
+        printf("\twtf:/path/to/trace.wag\n");
+        printf("\twtf:/path/to/trace.wag.gz\n");
+        printf("\n");
+        printf("\te.g.: wtf:/tmp/trace.wag.gz\n");
         printf("\n");
 }
@@ -411 +466 @@
         "wag",
         "$Id$",
-        "wag",
+        "wtf",
         wag_init_input,                 /* init_input */        
-        wag_init_output,                /* init_output */
-        wag_config_output,              /* config_output */
+        NULL,                           /* init_output */
+        NULL,                           /* config_output */
         wag_fin_input,                  /* fin_input */
-        wag_fin_output,                 /* fin_output */
+        NULL,                           /* fin_output */
         wag_read_packet,                /* read_packet */
-        wag_write_packet,               /* write_packet */
+        NULL,                           /* write_packet */
         wag_get_link,                   /* get_link */
         wag_get_link_type,              /* get_link_type */
@@ -435 +490 @@
 };
 
+/* wtf stands for Wag Trace Format */
+
+static struct libtrace_format_t wag_trace = {
+        "wtf",
+        "$Id$",
+        "wtf",
+        wtf_init_input,                 /* init_input */
+        wtf_init_output,                /* init_output */
+        wtf_config_output,              /* config_output */
+        wtf_fin_input,                  /* fin_input */
+        wtf_fin_output,                 /* fin_output */
+        wtf_read_packet,                /* read_packet */
+        wtf_write_packet,               /* write_packet */
+        wag_get_link,                   /* get_link */
+        wag_get_link_type,              /* get_link_type */
+        wag_get_direction,              /* get_direction */
+        NULL,                           /* set_direction */
+        wag_get_erf_timestamp,          /* get_erf_timestamp */
+        NULL,                           /* get_timeval */
+        NULL,                           /* get_seconds */
+        wag_get_capture_length,         /* get_capture_length */
+        wag_get_wire_length,            /* get_wire_length */
+        wag_get_framing_length,         /* get_framing_length */
+        NULL,                           /* set_capture_length */
+        wag_get_fd,                     /* get_fd */
+        wag_event_trace,                /* trace_event */
+        wtf_help                        /* help */
+};
+
+
 void __attribute__((constructor)) wag_constructor() {
         register_format(&wag);
-}
+        register_format(&wag_trace);
+}