p34709852
/
monkey
forked from p15670423/monkey
1
0
Fork 1
Code Issues Pull Requests Projects Releases Wiki Activity

Added func to find route to victim, reorganised firewall

This commit is contained in:
daniel goldberg 2016-08-29 12:08:42 +03:00
parent 3b39ee4308
commit bdde8dfeed
2 changed files with 21 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
chaos_monkey/network/firewall.py
View File

@ -3,6 +3,11 @@ import sys
import platform import platform
def _run_netsh_cmd(command, args):
cmd = subprocess.Popen("netsh %s %s" % (command, " ".join(['%s="%s"' % (key, value) for key, value in args.items()
if value])), stdout=subprocess.PIPE)
return cmd.stdout.read().strip().lower().endswith('ok.')
class FirewallApp(object): class FirewallApp(object):
def is_enabled(self, **kwargs): def is_enabled(self, **kwargs):
return False return False
@ -26,12 +31,6 @@ class FirewallApp(object):
return return
def _run_netsh_cmd(command, args):
cmd = subprocess.Popen("netsh %s %s" % (command, " ".join(['%s="%s"' % (key, value) for key, value in args.items()
if value])), stdout=subprocess.PIPE)
return cmd.stdout.read().strip().lower().endswith('ok.')
class WinAdvFirewall(FirewallApp): class WinAdvFirewall(FirewallApp):
def __init__(self): def __init__(self):
self._rules = {} self._rules = {}
@ -93,7 +92,7 @@ class WinAdvFirewall(FirewallApp):
def close(self): def close(self):
try: try:
for rule in self._rules.keys(): for rule in self._rules.keys():
_run_netsh_cmd('advfirewall firewall delete rule', {'name': rule}) self.remove_firewall_rule({'name': rule})
except: except:
pass pass

15
chaos_monkey/network/info.py
View File

@ -5,6 +5,7 @@ import socket
import struct import struct
import psutil import psutil
import ipaddress import ipaddress
from subprocess import check_output
from random import randint from random import randint
if sys.platform == "win32": if sys.platform == "win32":
@ -112,3 +113,17 @@ def get_ips_from_interfaces():
continue continue
res.append(str(addr)) res.append(str(addr))
return res return res
if sys.platform == "win32":
def get_ip_for_connection(target_ip):
return None
else:
def get_ip_for_connection(target_ip):
try:
query_str = 'ip route get %s' % target_ip
resp = check_output(query_str.split())
substr = resp.split()
src = substr[substr.index('src')+1]
return src
except Exception:
return None