Changeset 4389bd1 for examples

Timestamp:

11/05/04 16:58:44 (16 years ago)

Author:

Daniel Lawson <dlawson@…>

Branches:

4.0.1-hotfixes, cachetimestamps, develop, dpdk-ndag, etsilive, getfragoff, help, libtrace4, master, ndag_format, pfring, rc-4.0.1, rc-4.0.2, rc-4.0.3, rc-4.0.4, ringdecrementfix, ringperformance, ringtimestampfixes

Children:

533d7c1

Parents:

a6d38b6

Message:

changes in the 'capture' program

Location:

examples/capture

Files:

• Makefile (modified) (1 diff)
• capture.c (modified) (8 diffs)

examples/capture/Makefile

@@ -7 +7 @@
 CFLAGS += $(INCLUDE)
 libdir = -L$(PREFIX)/lib
-LDLIBS = $(libdir) -ltrace
+LDLIBS = $(libdir) -ltrace -lzl
 
-BINS = capture
+BINS = capture capture-gzip
 .PHONY: all clean distclean install depend
 

examples/capture/capture.c

@@ -40 +40 @@
 #include <sys/time.h>
 #include <stdint.h>
+#include "zlib.h"
 #include "libtrace.h"
 #include "dagformat.h"
@@ -66 +67 @@
 }
 
+#define BUFSIZE 65536
+
+
 void secondreport() {
 
@@ -71 +75 @@
 
         if (hdrcount >= 10) {
-                printf("Byte count:     Packet count:           Loss count\n");
+                fprintf(stderr,"Byte count:     Packet count:           Loss count\n");
                 hdrcount = 0;
         }
         hdrcount++;
-        printf("\t\t%d\t\t%d\t\t%d \n", 
+        fprintf(stderr,"\t\t%d\t\t%d\t\t%d \n", 
                         counter[BYTES],
                         counter[PACKETS],
@@ -88 +92 @@
         char *uri = 0;
         char *filename = 0;
-        FILE *fout = 0;
+        gzFile *fout = 0;
         int psize = 0;
+        int offset = 0;
         struct sigaction sigact;
         dag_record_t *erfptr = 0;
@@ -95 +100 @@
 
         struct itimerval itv;
+        void *buffer, *buffer2;
+
+
+        buffer = malloc(BUFSIZE);
+        buffer2 = buffer;
 
         /* 
@@ -122 +132 @@
                 uri = strdup(argv[1]);
                 filename = strdup(argv[2]);
-                fout = fopen(filename,"w");
+                fout = gzopen(filename,"ab");
         }
         
@@ -148 +158 @@
                 }
                 
-                if (filename) 
-                        fwrite(erfptr,psize,1,fout);
+                if (filename) {
+                        if (offset + psize > BUFSIZE) {
+                                //gzwrite(fout,erfptr,psize);
+                                fprintf(stderr,"writing to disk\n");
+                                gzwrite(fout,buffer,offset);
+                                offset = 0;
+                        }
+                        
+                        memcpy(buffer + offset,erfptr,psize);
+                        offset += psize;
+
+
+                }
 
 
@@ -155 +176 @@
 
         trace_destroy(trace);
-        fclose(fout);
+        gzclose(fout);
         return 0;
 }