Skip to content
Extraits de code Groupes Projets
Valider cb0066ad rédigé par François De Keersmaeker's avatar François De Keersmaeker
Parcourir les fichiers

Arg packet number: can specify multiple packets

parent 47480cfc
Aucune branche associée trouvée
Aucune étiquette associée trouvée
Aucune requête de fusion associée trouvée
...@@ -36,8 +36,8 @@ if __name__ == "__main__": ...@@ -36,8 +36,8 @@ if __name__ == "__main__":
parser.add_argument("-d", "--dry-run", action="store_true", help="Dry run: do not write output PCAP file.") parser.add_argument("-d", "--dry-run", action="store_true", help="Dry run: do not write output PCAP file.")
# Optional flag: -r / --random-range # Optional flag: -r / --random-range
parser.add_argument("-r", "--random-range", type=int, default=0, help="Upper bound for random range.") parser.add_argument("-r", "--random-range", type=int, default=0, help="Upper bound for random range.")
# Optional flag: -n # Optional flag: -n / --packet-number
parser.add_argument("-n", "--packet-number", type=int, help="Index of the packet to edit.") parser.add_argument("-n", "--packet-number", type=int, action="append", help="Index of the packet to edit, starting form 1. Can be specifed multiple times.")
# Parse arguments # Parse arguments
args = parser.parse_args() args = parser.parse_args()
...@@ -64,17 +64,18 @@ if __name__ == "__main__": ...@@ -64,17 +64,18 @@ if __name__ == "__main__":
writer.writeheader() writer.writeheader()
if args.packet_number is not None: if args.packet_number is not None:
# Edit one specific packet # Edit specific packets
packet = packets[args.packet_number - 1] # -1 because packet numbers start at 1 for i in args.packet_number:
try: packet = packets[i - 1] # -1 because packet numbers start at 1
my_packet = Packet.init_packet(packet, args.packet_number) try:
except ValueError: my_packet = Packet.init_packet(packet, i)
# No supported protocol found in packet, skip it except ValueError:
pass # No supported protocol found in packet, skip it
else: pass
d = my_packet.tweak() else:
if d is not None: d = my_packet.tweak()
writer.writerow(d) if d is not None:
writer.writerow(d)
else: else:
# Randomly edit packets # Randomly edit packets
......
0% Chargement en cours ou .
You are about to add 0 people to the discussion. Proceed with caution.
Terminez d'abord l'édition de ce message.
Veuillez vous inscrire ou vous pour commenter