| skipped 442 lines |
| 443 | 443 | | """ |
| 444 | 444 | | Kicks off the engine with the given arguments. |
| 445 | 445 | | """ |
| 446 | | - | try: |
| 447 | | - | nat_config = {} |
| 448 | | - | if args.get("sender_ip") and args.get("routing_ip") and args.get("forward_ip"): |
| 449 | | - | nat_config = {"sender_ip" : args["sender_ip"], |
| 450 | | - | "routing_ip" : args["routing_ip"], |
| 451 | | - | "forward_ip" : args["forward_ip"]} |
| | 446 | + | nat_config = {} |
| | 447 | + | if args.get("sender_ip") and args.get("routing_ip") and args.get("forward_ip"): |
| | 448 | + | nat_config = {"sender_ip": args["sender_ip"], |
| | 449 | + | "routing_ip": args["routing_ip"], |
| | 450 | + | "forward_ip": args["forward_ip"]} |
| 452 | 451 | | |
| 453 | | - | eng = Engine(args["server_port"], |
| 454 | | - | args["strategy"], |
| 455 | | - | environment_id=args["environment_id"], |
| 456 | | - | server_side=args["server_side"], |
| 457 | | - | output_directory=args["output_directory"], |
| 458 | | - | forwarder=nat_config, |
| 459 | | - | log_level=args["log"], |
| 460 | | - | in_queue_num=args["in_queue_num"], |
| 461 | | - | out_queue_num=args["out_queue_num"], |
| 462 | | - | save_seen_packets=args["no_save_packets"], |
| 463 | | - | demo_mode=args["demo_mode"]) |
| | 452 | + | with Engine(args["server_port"], |
| | 453 | + | args["strategy"], |
| | 454 | + | environment_id=args["environment_id"], |
| | 455 | + | server_side=args["server_side"], |
| | 456 | + | output_directory=args["output_directory"], |
| | 457 | + | forwarder=nat_config, |
| | 458 | + | log_level=args["log"], |
| | 459 | + | in_queue_num=args["in_queue_num"], |
| | 460 | + | out_queue_num=args["out_queue_num"], |
| | 461 | + | save_seen_packets=args["no_save_packets"], |
| | 462 | + | demo_mode=args["demo_mode"]) as eng: |
| 464 | 463 | | eng.initialize_nfqueue() |
| 465 | | - | while True: |
| 466 | | - | time.sleep(0.5) |
| 467 | | - | finally: |
| 468 | | - | eng.shutdown_nfqueue() |
| | 464 | + | |
| | 465 | + | threading.Event().wait() # Wait forever |
| 469 | 466 | | |
| 470 | 467 | | |
| 471 | 468 | | if __name__ == "__main__": |
| skipped 2 lines |
