X-Git-Url: http://git.onelab.eu/?p=fprobe-ulog.git;a=blobdiff_plain;f=src%2Ffprobe-ulog.c;h=8c66067b2ba862ca5bcd5ef05fc51c1b3d075e12;hp=5df381bba52209151000d7b75a362f604a369957;hb=164b27f0c908f203af585fcd999033b62ba54a22;hpb=abc715123b4da7f0f60e89245c38f46c0fe8a7cc

diff --git a/src/fprobe-ulog.c b/src/fprobe-ulog.c
index 5df381b..8c66067 100644
--- a/src/fprobe-ulog.c
+++ b/src/fprobe-ulog.c
@@ -6,10 +6,13 @@
 
 	$Id: fprobe-ulog.c,v 1.1.2.4 2005/01/30 09:06:19 sla Exp $
 
-	7/11/2007	Sapan Bhatia <sapanb@cs.princeton.edu> 
+			Sapan Bhatia <sapanb@cs.princeton.edu> 
 			
-		Added data collection (-f) functionality, xid support in the header and log file
-		rotation.
+	7/11/2007	Added data collection (-f) functionality, xid support in the header and log file
+			rotation.
+
+	15/11/2007	Added check to make sure fprobe doesn't overflow the disk. Also added a test facility.
+
 */
 
 #include <common.h>
@@ -31,7 +34,7 @@
 
 /* statfs() */
 
-#include <sys/statfs.h>
+#include <sys/vfs.h>
 
 #include <libipulog/libipulog.h>
 struct ipulog_handle {
@@ -92,6 +95,9 @@ struct ipulog_handle {
 #include <hash.h>
 #include <mem.h>
 
+#define PIDFILE "/var/log/fprobe-ulog.pid"
+#define STD_NETFLOW_PDU
+
 enum {
 	aflag,
 	Bflag,
@@ -422,19 +428,21 @@ unsigned get_log_fd(char *fname, int cur_fd) {
 	cur_uptime = getuptime_minutes(&now);
 
 
-	if (fstatfs(cur_fd, &statfs) && cur_fd!=START_VALUE) {
-		my_log(LOG_ERR, "PANIC! Can't stat disk to calculate free blocks");
-	}
-	else {
-		if (min_free && (statfs.f_bfree < min_free)) 
-			switch(cur_epoch) {
-				case 0: /* Uh oh. Our first file filled up all of the free space. Just bail out. */
-					my_log(LOG_ERR, "The first epoch filled up all the free space on disk. Bailing out.");
-					exit(1);
-				default:
-					my_log(LOG_INFO, "Disk almost full. I'm going to drop data. Max epochs = %d\n",cur_epoch);
-					cur_epoch = -1;
-			}
+	if (cur_fd!=START_VALUE) {
+		if (fstatfs(cur_fd, &statfs) == -1) {
+			my_log(LOG_ERR, "PANIC! Can't stat disk to calculate free blocks");
+		}
+		else {
+			if (min_free && (statfs.f_bavail < min_free)) 
+				switch(cur_epoch) {
+					case 0: /* Uh oh. Our first file filled up all of the free space. Just bail out. */
+						my_log(LOG_ERR, "The first epoch filled up all the free space on disk. Bailing out.");
+						exit(1);
+					default:
+						my_log(LOG_INFO, "Disk almost full (%u free blocks). I'm going to drop data. Max epochs = %d\n",statfs.f_bavail,cur_epoch);
+						cur_epoch = -1;
+				}
+		}
 	}
 
 	/* Epoch length in minutes */
@@ -811,7 +819,9 @@ void *emit_thread()
 			p = fill(netflow->HeaderFields, netflow->HeaderFormat, 0, &emit_packet);
 			size = netflow->HeaderSize + emit_count * netflow->FlowSize;
 			/* Netflow PDUs need to be padded to 1464 bytes - Sapan */
+#ifdef STD_NETFLOW_PDU
 			if (size < NETFLOW_PDU_SIZE) size = NETFLOW_PDU_SIZE;
+#endif
 			peer_rot_cur = 0;
 			for (i = 0; i < npeers; i++) {
 				if (peers[i].type == PEER_FILE) {
@@ -1244,6 +1254,44 @@ void *cap_thread()
 	return 0;
 }
 
+/* Copied out of CoDemux */
+
+static int init_daemon() {
+  pid_t pid;
+  FILE *pidfile;
+
+  pidfile = fopen(PIDFILE, "w");
+  if (pidfile == NULL) {
+    my_log(LOG_ERR, "%s creation failed\n", PIDFILE);
+  }
+
+  if ((pid = fork()) < 0) {
+    fclose(pidfile);
+    my_log(LOG_ERR, "Could not fork!\n");
+    return(-1);
+  }
+  else if (pid != 0) {
+    /* i'm the parent, writing down the child pid  */
+    fprintf(pidfile, "%u\n", pid);
+    fclose(pidfile);
+    exit(0);
+  }
+
+  /* close the pid file */
+  fclose(pidfile);
+
+  /* routines for any daemon process
+     1. create a new session 
+     2. change directory to the root
+     3. change the file creation permission 
+  */
+  setsid();
+  chdir("/usr/local/fprobe");
+  umask(0);
+
+  return(0);
+}
+
 int main(int argc, char **argv)
 {
 	char errpbuf[512];
@@ -1492,6 +1540,9 @@ bad_collector:
 	/* Daemonize (if log destination stdout-free) */
 
 	my_log_open(ident, verbosity, log_dest);
+
+	init_daemon();
+
 	if (!(log_dest & 2)) {
 		/* Crash-proofing - Sapan*/
 		while (1) {