netsniff-ng: Allow to specify compiled BPF from stdin

Allow read compiled BPF instructions from stdin by via '-f -' option.

Signed-off-by: Vadim Kochan <vadim4j@gmail.com>
Signed-off-by: Tobias Klauser <tklauser@distanz.ch>

1 files changed, 7 insertions, 2 deletions

diff --git a/bpf.c b/bpf.c
index 2d769ec..66d1b6c 100644
--- a/bpf.c
+++ b/bpf.c
@@ -720,7 +720,11 @@ void bpf_parse_rules(char *rulefile, struct sock_fprog *bpf, uint32_t link_type)
 		return;
 	}
 
-	fp = fopen(rulefile, "r");
+	if (!strcmp(rulefile, "-"))
+		fp = stdin;
+	else
+		fp = fopen(rulefile, "r");
+
 	if (!fp) {
 		bpf_try_compile(rulefile, bpf, link_type);
 		return;
@@ -753,7 +757,8 @@ void bpf_parse_rules(char *rulefile, struct sock_fprog *bpf, uint32_t link_type)
 		fmemset(buff, 0, sizeof(buff));
 	}
 
-	fclose(fp);
+	if (fp != stdin)
+		fclose(fp);
 
 	if (unlikely(__bpf_validate(bpf) == 0))
 		panic("This is not a valid BPF program!\n");