class Main:
def __init__(self, src, dst, dports, iface):
self.src = src
self.dst = dst
self.dports = dports
self.iface = iface
self.inp_queue = Queue()
self.outp_queue = Queue()
self.res_queue = Queue()
self.preceiver = PacketReceiver(iface = self.iface, queue = self.inp_queue)
self.psender = PacketSender(iface = self.iface, queue = self.outp_queue)
self.scanner = Scanner(self.inp_queue, self.outp_queue, self.res_queue, self.src, self.dst, self.dports)
def start(self):
self.preceiver.start()
self.psender.start()
self.scanner.start()
class TestPacketSender(unittest.TestCase):
def mk_testp(self):
return Ether()/IP(dst='127.0.0.1')/TCP()
def do_test(self, dry_run):
self.outp_queue = Queue()
self.psender = PacketSender(iface = 'lo', queue = self.outp_queue, dry_run = dry_run)
self.psender.start()
for i in range(1000):
p = self.mk_testp()
self.outp_queue.put(p)
time.sleep(5)
# def test_dryrun(self):
# self.do_test(True)
def test_wetrun(self):
self.do_test(False)
