zarp.py

#! /usr/bin/python

from os import getcwd, getuid
from sys import path, argv, exit
path.insert(0, getcwd() + '/src/')
path.insert(0, getcwd() + '/src/core/')
path.insert(0, getcwd() + '/src/modules/')
from util import get_subclass, print_menu, header, Error, Msg, debug
from commands import getoutput
import stream
import session_manager
import parse_cmd
import config
from colors import color
# module loading
from src.modules import poison, dos, scanner, services, sniffer, parameter
import importlib

class LoadedModules:
	""" Load modules
	"""
	def __init__(self):
		self.total      = 0
		self.poison     = []
		self.dos        = []
		self.sniffers   = []
		self.services   = []
		self.scanner    = []
		self.parameter  = []

	def load(self):
		""" Load modules.
		"""
		for module in poison.__all__:
			mod = getattr(importlib.import_module('src.modules.poison.%s'%module, 'poison'),module)
			self.poison.append(mod)
			self.total += 1
		for module in dos.__all__:
			mod = getattr(importlib.import_module('src.modules.dos.%s'%module, 'dos'),module)
			self.dos.append(mod)
			self.total += 1
		for module in scanner.__all__:
			mod = getattr(importlib.import_module('src.modules.scanner.%s'%module, 'scanner'),module)
			self.scanner.append(mod)
			self.total += 1
		for module in services.__all__:
			mod = getattr(importlib.import_module('src.modules.services.%s'%module, 'services'),module)
			self.services.append(mod)
			self.total += 1
		for module in sniffer.__all__:
			mod = getattr(importlib.import_module('src.modules.sniffer.%s'%module, 'sniffer'),module)
			self.sniffers.append(mod)
			self.total += 1
		for module in parameter.__all__:
			mod = getattr(importlib.import_module('src.modules.parameter.%s'%module, 'parameter'),module)
			self.parameter.append(mod)
			self.total += 1
	
def main():
	""" Zarp entry point
	"""

	# set up configuration 
	config.initialize()

	# load modules
	loader = LoadedModules()
	loader.load()
	Msg('Loaded %d modules.'%loader.total)

	# handle command line options first
	if len(argv) > 1:
		parse_cmd.parse(argv, loader)

	# menus
	main_menu =    [ 'Poisoners', 'DoS Attacks', 'Sniffers', 'Scanners',
					 'Parameter','Services','Sessions']
	
	running = True
	choice = -1
	while running:
		header()
		choice = print_menu(main_menu)      
		if choice == 0:
			# check if they've got running sessions! 
			cnt = stream.get_session_count()
			if cnt > 0:
				choice = raw_input(color.YELLOW + 'You have %d sessions running.  Are you sure? [y] '%cnt\
								+ color.END)
				if 'y' in choice.lower() or choice == '': 
					Msg('Shutting all sessions down...')
					stream.stop_session('all', -1)
					running = False 
			else:
				debug("Exiting with session count: %d"%(cnt))
				Msg("Exiting...")
				running = False
		elif choice == 1:
			while True:
				choice = print_menu([x().which for x in loader.poison])
				if choice == 0: break
				elif choice == -1: pass
				elif choice > len(loader.poison): continue
				else:
					stream.initialize(loader.poison[choice-1])
		elif choice == 2:
			while True:
				choice = print_menu([x().which for x in loader.dos])
				if choice == 0: break
				elif choice == -1: pass
				elif choice > len(loader.dos): continue
				else:
					stream.initialize(loader.dos[choice-1])
		elif choice == 3:
			while True:
				choice = print_menu([x().which for x in loader.sniffers])
				if choice == 0: break
				elif choice == -1: pass
				elif choice > len(loader.sniffers): continue
				else:
					stream.initialize(loader.sniffers[choice-1]) 
		elif choice == 4:
			while True:
				choice = print_menu([x().which for x in loader.scanner])
				if choice == 0: break
				elif choice == -1: pass
				elif choice > len(loader.scanner): continue
				else:
					stream.initialize(loader.scanner[choice-1])
		elif choice == 5:
			while True:
				choice = print_menu([x().which for x in loader.parameter])
				if choice == 0: break
				elif choice == -1: pass
				elif choice > len(loader.parameter): continue
				else:
					stream.initialize(loader.parameter[choice-1])
		elif choice == 6:
			while True:
				choice = print_menu([x().which for x in loader.services])
				if choice == 0: break
				elif choice == -1: pass
				elif choice > len(loader.services): continue
				else:
					stream.initialize(loader.services[choice-1])
		elif choice == 7:
			session_manager.menu()
		elif choice == -1:
			pass

# Application entry; dependency checks, etc.
if __name__=="__main__":
	# perm check
	if int(getuid()) > 0:
		Error('Please run as root.')
		exit(1)
	# check for forwarding
	if not getoutput('cat /proc/sys/net/ipv4/ip_forward') == '1':
		Msg('IPv4 forwarding disabled.  Enabling..')
		tmp = getoutput('sudo sh -c \'echo "1" > /proc/sys/net/ipv4/ip_forward\'')  
		if len(tmp) > 0:
			Error('Error enabling IPv4 forwarding.')
			exit(1)
	main()