mirror of
https://github.com/byt3bl33d3r/MITMf.git
synced 2024-10-17 20:50:35 -07:00
57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
import unittest
|
|
import threading
|
|
import logging
|
|
|
|
class BasicTests(unittest.TestCase):
|
|
|
|
def test_configfile(self):
|
|
from configobj import ConfigObj
|
|
config = ConfigObj('config/mitmf.conf')
|
|
|
|
def test_logger(self):
|
|
from core.logger import logger
|
|
logger.log_level = logging.DEBUG
|
|
formatter = logging.Formatter("%(asctime)s [unittest] %(message)s", datefmt="%Y-%m-%d %H:%M:%S")
|
|
log = logger().setup_logger("unittest", formatter)
|
|
|
|
def test_DNSChef(self):
|
|
from core.logger import logger
|
|
logger.log_level = logging.DEBUG
|
|
from core.servers.DNS import DNSChef
|
|
DNSChef().start()
|
|
|
|
def test_NetCreds(self):
|
|
from core.logger import logger
|
|
logger.log_level = logging.DEBUG
|
|
from core.netcreds import NetCreds
|
|
NetCreds().start('venet0:0', '172.30.96.18')
|
|
|
|
def test_SSLStrip_Proxy(self):
|
|
favicon = True
|
|
preserve_cache = True
|
|
killsessions = True
|
|
listen_port = 10000
|
|
|
|
from twisted.web import http
|
|
from twisted.internet import reactor
|
|
from core.sslstrip.CookieCleaner import CookieCleaner
|
|
from core.proxyplugins import ProxyPlugins
|
|
from core.sslstrip.StrippingProxy import StrippingProxy
|
|
from core.sslstrip.URLMonitor import URLMonitor
|
|
|
|
URLMonitor.getInstance().setFaviconSpoofing(favicon)
|
|
URLMonitor.getInstance().setCaching(preserve_cache)
|
|
CookieCleaner.getInstance().setEnabled(killsessions)
|
|
|
|
strippingFactory = http.HTTPFactory(timeout=10)
|
|
strippingFactory.protocol = StrippingProxy
|
|
|
|
reactor.listenTCP(listen_port, strippingFactory)
|
|
|
|
#ProxyPlugins().all_plugins = plugins
|
|
t = threading.Thread(name='sslstrip_test', target=reactor.run)
|
|
t.setDaemon(True)
|
|
t.start()
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main() |