Changeset 166 for branches/renetcolSMP/src/renetcolSMP4.c

Timestamp:

07/23/14 15:16:06 (8 years ago)

Author:

andreu

Message:

RECORDFLOW option

Files:

• branches/renetcolSMP/src/renetcolSMP4.c (modified) (8 diffs)

branches/renetcolSMP/src/renetcolSMP4.c

@@ -37 +37 @@
 FILE *fIP;                                 /* pointer on the IP output file */
 FILE *fMPLS;                              /* pointer on the MPLS output file */
+
+#if defined(RECORDFLOW)
+FILE *fRECORDF1;
+FILE *fRECORDF2;
+char    *bufRF1 = NULL;
+char    *bufRF2 = NULL;
+char *nameRF[128];
+char *pathRF[256];
+#endif
 
 struct sigaction myAction; /* handler for renetcolSender communication */
@@ -118 +127 @@
     fprintf(stderr, "Couldn't open PCAP file %s: %s\n", myPcapFileName, errBuffer);
     exit(1);
+  }
+#endif
+
+#if defined(RECORDFLOW)
+  strncpy(pathRF, RECORD_FILE_PATH,sizeof(RECORD_FILE_PATH));
+  strncpy(nameRF, SMP4_RECORD_FILE_1,sizeof(SMP4_RECORD_FILE_1));
+  errno = 0;
+  prepend_timestamp ( nameRF, sizeof nameRF );
+  if ( errno != ERANGE )
+    printf ( "Timestamped: %s\n", nameRF );
+  else {
+      /* Handle the error */
+  }
+  strncat(pathRF,nameRF, sizeof(nameRF));
+  if ((fRECORDF1 = fopen(pathRF, "w")) == NULL) {
+    syslog(LOG_ERR,"Error in %s opening : %s", pathRF, strerror(errno));
+  } else {
+    fprintf(stderr,"fRECORDF1 %d",fRECORDF1);
+    bufRF1 = (char *)malloc (262144);
+    setvbuf (fRECORDF1, bufRF1, _IOFBF, 262144);
   }
 #endif
@@ -377 +406 @@
 }
 
+char *prepend_timestamp ( char s[], size_t capacity )
+{
+    const char fmt[] = "%y-%m-%d_%H-%M";
+    size_t n = sizeof ( fmt );
+    size_t len = strlen ( s );
+    if ( len + n < capacity ) {
+        time_t init = time ( NULL );
+        memmove ( s + n, s, len + 1 );
+        strftime ( s, n, fmt, localtime ( &init ) );
+        s[n - 1] = '_';
+    }
+    else
+        errno = ERANGE;
+
+    return s;
+}
+
+
 /*
  * send his PID to another program via IPC message queue
@@ -551 +598 @@
   int dd = 0;
   FILE *TPLFILE;
-
+#if defined(RECORDFLOW)
+  FILE *currentRECORDFLOWFILE;
+#endif
   now = time((time_t *)NULL);
   tmPtr = localtime(&now);
@@ -557 +606 @@
   lastMPLSOutput = now;
   lastIPOutput = now;
+#if defined(RECORDFLOW)  
+currentRECORDFLOWFILE = fRECORDF1;
+#endif
   do {
 #ifdef DEBUG
@@ -566 +618 @@
         shmForAgg->currentTable = 1;
         shmForAgg->secondTable = 0;
+#if defined(RECORDFLOW)
+        /*free(bufRF1);*/
+        fflush(fRECORDF1);
+        if (fclose(fRECORDF1) == NULL){
+          fprintf(stderr,"error fclose : %s",strerror(errno));
+        }
+        strncpy(pathRF, RECORD_FILE_PATH,sizeof(RECORD_FILE_PATH));
+        strncpy(nameRF, SMP4_RECORD_FILE_2,sizeof(SMP4_RECORD_FILE_2));
+        errno = 0;
+        prepend_timestamp ( nameRF, sizeof nameRF );
+        if ( errno != ERANGE )
+          printf ( "Timestamped: %s\n", nameRF );
+        else {
+          /* Handle the error */
+        }
+        strncat(pathRF, nameRF, sizeof(nameRF));
+
+        if ((fRECORDF2 = fopen(pathRF, "w")) == NULL) {
+          syslog(LOG_ERR,"Error in %s opening : %s", pathRF, strerror(errno));
+        } else {
+          bufRF2 = (char *)malloc (262144);
+          setvbuf (fRECORDF2, bufRF2, _IOFBF, 262144);
+        }
+        currentRECORDFLOWFILE = fRECORDF2;
+#endif
       }else{
         shmForAgg->currentTable = 0;
         shmForAgg->secondTable = 1;
+#if defined(RECORDFLOW)
+        /*free(bufRF2);*/
+        fflush(fRECORDF2);
+        fclose(fRECORDF2);
+        strncpy(pathRF, RECORD_FILE_PATH,sizeof(RECORD_FILE_PATH));
+        strncpy(nameRF, SMP4_RECORD_FILE_1,sizeof(SMP4_RECORD_FILE_1));
+        errno = 0;
+        prepend_timestamp ( nameRF, sizeof nameRF );
+        if ( errno != ERANGE )
+          printf ( "Timestamped: %s\n", nameRF );
+        else {
+          /* Handle the error */
+        }
+        strncat(pathRF, nameRF, sizeof(nameRF));
+        if ((fRECORDF1 = fopen(pathRF, "w")) == NULL) {
+          syslog(LOG_ERR,"Error in %s opening : %s", pathRF, strerror(errno));
+        } else {
+          bufRF1 = (char *)malloc (262144);
+          setvbuf (fRECORDF1, bufRF1, _IOFBF, 262144);
+        }
+        currentRECORDFLOWFILE = fRECORDF1;
+#endif
       }
       shmForAgg->readed = 1;
@@ -700 +799 @@
           }
         } else if (currentFlowsetId > 255) {
+#if defined(RECORDFLOW)
+          if ( (shift = checkDataFlowSet(shift,
+                                         &myPtrs,
+                                         myQueue,
+                                         shmForAgg->prefixV4Tab[shmForAgg->currentTable],
+                                         (size_t) shmForAgg->v4PrefixNb,
+                                         shmForAgg->prefixV4SubnetTab[shmForAgg->currentTable],
+                                         (size_t) shmForAgg->v4SubnetNb,
+           currentRECORDFLOWFILE) ) < 0 ) {
+#else
           if ( (shift = checkDataFlowSet(shift,
                                          &myPtrs,
@@ -707 +816 @@
                                          shmForAgg->prefixV4SubnetTab[shmForAgg->currentTable],
                                          (size_t) shmForAgg->v4SubnetNb) ) < 0 ) {
+#endif
 #ifdef DEBUG
             fprintf (stderr, "wd> ");