def test_client_playback(self):
s = flow.State()
f = tutils.tflow(resp=True)
pb = [tutils.tflow(resp=True), f]
fm = flow.FlowMaster(DummyServer(ProxyConfig()), s)
assert not fm.start_server_playback(
pb,
False,
[],
False,
False,
None,
False,
None,
False)
assert not fm.start_client_playback(pb, False)
fm.client_playback.testing = True
q = Queue.Queue()
assert not fm.state.flow_count()
fm.tick(q, 0)
assert fm.state.flow_count()
f.error = Error("error")
fm.handle_error(f)
def test_load_flows_reverse(self):
r = self._treader()
s = flow.State()
conf = ProxyConfig(mode="reverse", upstream_server=[True,True,"use-this-domain",80])
fm = flow.FlowMaster(DummyServer(conf), s)
fm.load_flows(r)
assert s._flow_list[0].request.host == "use-this-domain"
def test_err_2(self):
conf = ProxyConfig(host="invalidhost")
tutils.raises("error starting proxy server", ProxyServer, conf)
def test_err(self):
conf = ProxyConfig(port=1)
tutils.raises("error starting proxy server", ProxyServer, conf)
"""
with open('blacklist', 'r') as ff:
domains = ff.xreadlines()
for domain in domains:
domain = domain.strip().rstrip()
if domain.find('#'):
continue
blacklist.append(domain)
"""
if __name__ == '__main__':
whitelist = []
config = ProxyConfig(
host=mitm_config['host'],
port=mitm_config['port'],
# use ~/.mitmproxy/mitmproxy-ca.pem as default CA file.
cadir="~/.mitmproxy/"
)
with open('whiteLists.txt', 'r') as ff:
domains = ff.xreadlines()
for domain in domains:
domain = domain.strip().rstrip()
if domain.find('#') != -1:
continue
whitelist.append(domain)
print domain
print "will be scan."
state = flow.State()
