X-Git-Url: http://git.onelab.eu/?p=fprobe-ulog.git;a=blobdiff_plain;f=src%2Ffprobe-ulog.c;h=e939dabf7464fd6b7795c69913c563c9f4347f5f;hp=c77c6382fdaa0d7b876519b50a9cb505d8602427;hb=ba0903e394f2aff6a29d19576f99dfab4c92ae69;hpb=581cfdb6ad60733e425725b4ffefa057dea5abd9

diff --git a/src/fprobe-ulog.c b/src/fprobe-ulog.c
index c77c638..e939dab 100644
--- a/src/fprobe-ulog.c
+++ b/src/fprobe-ulog.c
@@ -1505,7 +1505,7 @@ bad_lhost:
 	/* Process collectors parameters. Brrrr... :-[ */
 
 	npeers = argc - optind;
-	if (npeers > 1) {
+	if (npeers >= 1) {
 		/* Send to remote Netflow collector */
 		if (!(peers = malloc(npeers * sizeof(struct peer)))) goto err_malloc;
 		for (i = optind, npeers = 0; i < argc; i++, npeers++) {
@@ -1570,10 +1570,8 @@ bad_collector:
 	}
 	else if (parms[fflag].count) {
 		// log into a file
-		if (!(peers = malloc(npeers * sizeof(struct peer)))) goto err_malloc;
-		if (!(peers[npeers].fname = malloc(strnlen(parms[fflag].arg,MAX_PATH_LEN)))) goto err_malloc;
-		strncpy(peers[npeers].fname, parms[fflag].arg, MAX_PATH_LEN);
-
+		if (!(peers = malloc(sizeof(struct peer)))) goto err_malloc;
+		if (!(peers[npeers].fname = strndup(parms[fflag].arg, MAX_PATH_LEN))) goto err_malloc;
 		peers[npeers].write_fd = START_DATA_FD;
 		peers[npeers].type = PEER_FILE;
 		peers[npeers].seq = 0;