@@ -84,25 +84,25 @@ def get_cli_args():
8484 parser .add_argument (
8585 "--test-user-file" ,
8686 required = False ,
87- default = f"{ script_dir } /data/test_users.txt" ,
87+ default = abspath ( f"{ script_dir } /data/test_users.txt" ) ,
8888 help = "Path to the file containing test usernames" ,
8989 )
9090 parser .add_argument (
9191 "--test-password-file" ,
9292 required = False ,
93- default = f"{ script_dir } /data/test_passwords.txt" ,
93+ default = abspath ( f"{ script_dir } /data/test_passwords.txt" ) ,
9494 help = "Path to the file containing test passwords" ,
9595 )
9696 parser .add_argument (
9797 "--amsi-bypass-file" ,
9898 required = False ,
99- default = f"{ script_dir } /data/test_amsi_bypass.txt" ,
99+ default = abspath ( f"{ script_dir } /data/test_amsi_bypass.txt" ) ,
100100 help = "Path to the file containing AMSI bypasses" ,
101101 )
102102 parser .add_argument (
103103 "--test-normal-file" ,
104104 required = False ,
105- default = f"{ script_dir } /data/test_file.txt" ,
105+ default = abspath ( f"{ script_dir } /data/test_file.txt" ) ,
106106 help = "Path to file to upload/download"
107107 )
108108 parser .add_argument (
@@ -137,6 +137,8 @@ def generate_commands(args):
137137 if i + 1 in flattened_list :
138138 if line .startswith ("#" ):
139139 continue
140+ if "#" in line :
141+ line = line .split ("#" )[0 ]
140142 line = line .strip ()
141143 if args .protocols :
142144 if line .split ()[1 ] in args .protocols :
@@ -147,6 +149,8 @@ def generate_commands(args):
147149 for line in file :
148150 if line .startswith ("#" ):
149151 continue
152+ if "#" in line :
153+ line = line .split ("#" )[0 ]
150154 line = line .strip ()
151155 if args .protocols :
152156 if line .split ()[1 ] in args .protocols :
0 commit comments