From fac703d878e45b227b5e938849466fe58d180253 Mon Sep 17 00:00:00 2001 From: lns Date: Sat, 20 May 2023 17:16:31 +0200 Subject: Improved helper scripts to be more restrictive. Signed-off-by: lns --- utils/hostname2list.py | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'utils/hostname2list.py') diff --git a/utils/hostname2list.py b/utils/hostname2list.py index af0a5c349..e3facfdf0 100755 --- a/utils/hostname2list.py +++ b/utils/hostname2list.py @@ -39,8 +39,10 @@ print("""/* print("static ndpi_protocol_match "+proto.lower()+"_hostname_list[] = {") +lines = 0 with open(sys.argv[1]) as fp: for cnt, line in enumerate(fp): + lines += 1 line = line.rstrip() if(line != ""): @@ -57,3 +59,7 @@ with open(sys.argv[1]) as fp: print(" /* End */") print(" { NULL, NULL, NDPI_PROTOCOL_UNKNOWN, NDPI_PROTOCOL_CATEGORY_UNSPECIFIED, NDPI_PROTOCOL_SAFE, NDPI_PROTOCOL_DEFAULT_LEVEL }") print("};") + +if lines == 0: + sys.stderr.write(f'File {sys.argv[1]} is empty.\n') + sys.exit(1) -- cgit v1.2.3