diff --git a/ScappyTest.py b/ScappyTest.py index 4065fe5..feafb2e 100644 --- a/ScappyTest.py +++ b/ScappyTest.py @@ -21,13 +21,4 @@ def build_packet(x, interface): Dot11Elt(ID="TIM", info="\x00\x01\x00\x00"), iface=interface, loop=1) #sendp("I'm travelling on Ethernet", loop=1, inter=0.2) - - - - -build_packet("HellomynameisRyan","wlp2s0mon") - - - - diff --git a/args.py b/args.py new file mode 100644 index 0000000..66f6c6e --- /dev/null +++ b/args.py @@ -0,0 +1,23 @@ +import argparse + +def get_cargs(): + argumentSet = argparse.ArgumentParser() + + modeSet=argumentSet.add_mutually_exclusive_group() + modeSet.add_argument("-s", "--send", required = False, + help="string to send in packet") + modeSet.add_argument("-l", "--listen", action ="store_const", + default = None, const = "listen", required = False, + help="Mode in which to run the script") + + args = vars(argumentSet.parse_args()) + if args["send"]: + mode = "send" + message = args["send"] + elif args["listen"]: + mode = "listen" + message = None + else: + mode = "default" + message = None + return (mode, message) diff --git a/cmdargs.py b/cargexamples.py similarity index 100% rename from cmdargs.py rename to cargexamples.py diff --git a/main.py b/main.py new file mode 100644 index 0000000..eeaa4c7 --- /dev/null +++ b/main.py @@ -0,0 +1,15 @@ +import args +import sys +import ScappyTest + +mode, message = args.get_cargs() + +if mode == "send": + ScappyTest.build_packet(message, "wwlp2s0mon") + print "Packet Sent!" +elif mode == "listen": + print "Listen mode! Doesn't do anything yet" +elif mode == "default": + print "Default mode! Doesn't do anything yet" +else: + print "undefined????"