diff options
Diffstat (limited to 'tools/mock-meta.py')
| -rwxr-xr-x | tools/mock-meta.py | 301 |
1 files changed, 164 insertions, 137 deletions
diff --git a/tools/mock-meta.py b/tools/mock-meta.py index 9dd067b9..4ac1ea4f 100755 --- a/tools/mock-meta.py +++ b/tools/mock-meta.py @@ -36,74 +36,74 @@ except ImportError: from http import client as hclient -log = logging.getLogger('meta-server') +log = logging.getLogger("meta-server") EC2_VERSIONS = [ - '1.0', - '2007-01-19', - '2007-03-01', - '2007-08-29', - '2007-10-10', - '2007-12-15', - '2008-02-01', - '2008-09-01', - '2009-04-04', + "1.0", + "2007-01-19", + "2007-03-01", + "2007-08-29", + "2007-10-10", + "2007-12-15", + "2008-02-01", + "2008-09-01", + "2009-04-04", ] BLOCK_DEVS = [ - 'ami', - 'ephemeral0', - 'root', + "ami", + "ephemeral0", + "root", ] -DEV_PREFIX = 'v' # This seems to vary alot depending on images... +DEV_PREFIX = "v" # This seems to vary alot depending on images... DEV_MAPPINGS = { - 'ephemeral0': '%sda2' % (DEV_PREFIX), - 'root': '/dev/%sda1' % (DEV_PREFIX), - 'ami': '%sda1' % (DEV_PREFIX), - 'swap': '%sda3' % (DEV_PREFIX), + "ephemeral0": "%sda2" % (DEV_PREFIX), + "root": "/dev/%sda1" % (DEV_PREFIX), + "ami": "%sda1" % (DEV_PREFIX), + "swap": "%sda3" % (DEV_PREFIX), } META_CAPABILITIES = [ - 'aki-id', - 'ami-id', - 'ami-launch-index', - 'ami-manifest-path', - 'ari-id', - 'block-device-mapping/', - 'hostname', - 'instance-action', - 'instance-id', - 'instance-type', - 'local-hostname', - 'local-ipv4', - 'placement/', - 'product-codes', - 'public-hostname', - 'public-ipv4', - 'public-keys/', - 'reservation-id', - 'security-groups' + "aki-id", + "ami-id", + "ami-launch-index", + "ami-manifest-path", + "ari-id", + "block-device-mapping/", + "hostname", + "instance-action", + "instance-id", + "instance-type", + "local-hostname", + "local-ipv4", + "placement/", + "product-codes", + "public-hostname", + "public-ipv4", + "public-keys/", + "reservation-id", + "security-groups", ] PUB_KEYS = { - 'brickies': [ - ('ssh-rsa ' - 'AAAAB3NzaC1yc2EAAAABIwAAAQEA3I7VUf2l5gSn5uavROsc5HRDpZdQueUq5ozemN' - 'Sj8T7enqKHOEaFoU2VoPgGEWC9RyzSQVeyD6s7APMcE82EtmW4skVEgEGSbDc1pvxz' - 'xtchBj78hJP6Cf5TCMFSXw+Fz5rF1dR23QDbN1mkHs7adr8GW4kSWqU7Q7NDwfIrJJ' - 'tO7Hi42GyXtvEONHbiRPOe8stqUly7MvUoN+5kfjBM8Qqpfl2+FNhTYWpMfYdPUnE7' - 'u536WqzFmsaqJctz3gBxH9Ex7dFtrxR4qiqEr9Qtlu3xGn7Bw07/+i1D+ey3ONkZLN' - '+LQ714cgj8fRS4Hj29SCmXp5Kt5/82cD/VN3NtHw== brickies'), - '', + "brickies": [ + "ssh-rsa " + "AAAAB3NzaC1yc2EAAAABIwAAAQEA3I7VUf2l5gSn5uavROsc5HRDpZdQueUq5ozemN" + "Sj8T7enqKHOEaFoU2VoPgGEWC9RyzSQVeyD6s7APMcE82EtmW4skVEgEGSbDc1pvxz" + "xtchBj78hJP6Cf5TCMFSXw+Fz5rF1dR23QDbN1mkHs7adr8GW4kSWqU7Q7NDwfIrJJ" + "tO7Hi42GyXtvEONHbiRPOe8stqUly7MvUoN+5kfjBM8Qqpfl2+FNhTYWpMfYdPUnE7" + "u536WqzFmsaqJctz3gBxH9Ex7dFtrxR4qiqEr9Qtlu3xGn7Bw07/+i1D+ey3ONkZLN" + "+LQ714cgj8fRS4Hj29SCmXp5Kt5/82cD/VN3NtHw== brickies", + "", ], } INSTANCE_TYPES = [ - 'm1.large', - 'm1.medium', - 'm1.small', - 'm1.xlarge', + "m1.large", + "m1.medium", + "m1.small", + "m1.xlarge", ] AVAILABILITY_ZONES = [ @@ -111,13 +111,13 @@ AVAILABILITY_ZONES = [ "us-east-1b", "us-east-1c", "us-east-1d", - 'eu-west-1a', - 'eu-west-1b', - 'us-west-1', + "eu-west-1a", + "eu-west-1b", + "us-west-1", ] PLACEMENT_CAPABILITIES = { - 'availability-zone': AVAILABILITY_ZONES, + "availability-zone": AVAILABILITY_ZONES, } NOT_IMPL_RESPONSE = json.dumps({}) @@ -130,12 +130,14 @@ class WebException(Exception): def yamlify(data): - formatted = yaml.dump(data, - line_break="\n", - indent=4, - explicit_start=True, - explicit_end=True, - default_flow_style=False) + formatted = yaml.dump( + data, + line_break="\n", + indent=4, + explicit_start=True, + explicit_end=True, + default_flow_style=False, + ) return formatted @@ -164,7 +166,7 @@ ID_CHARS = [c for c in (string.ascii_uppercase + string.digits)] def id_generator(size=6, lower=False): - txt = ''.join(random.choice(ID_CHARS) for x in range(size)) + txt = "".join(random.choice(ID_CHARS) for x in range(size)) if lower: return txt.lower() else: @@ -176,14 +178,14 @@ def get_ssh_keys(): keys.update(PUB_KEYS) # Nice helper to add in the 'running' users key (if they have one) - key_pth = os.path.expanduser('~/.ssh/id_rsa.pub') + key_pth = os.path.expanduser("~/.ssh/id_rsa.pub") if not os.path.isfile(key_pth): - key_pth = os.path.expanduser('~/.ssh/id_dsa.pub') + key_pth = os.path.expanduser("~/.ssh/id_dsa.pub") if os.path.isfile(key_pth): - with open(key_pth, 'rb') as fh: + with open(key_pth, "rb") as fh: contents = fh.read() - keys[os.getlogin()] = [contents, ''] + keys[os.getlogin()] = [contents, ""] return keys @@ -193,7 +195,6 @@ class HTTPServerV6(HTTPServer): class MetaDataHandler(object): - def __init__(self, opts): self.opts = opts self.instances = {} @@ -206,17 +207,17 @@ class MetaDataHandler(object): return "\n".join(caps) action = params[0] action = action.lower() - if action == 'instance-id': - return 'i-%s' % (id_generator(lower=True)) - elif action == 'ami-launch-index': + if action == "instance-id": + return "i-%s" % (id_generator(lower=True)) + elif action == "ami-launch-index": return "%s" % random.choice([0, 1, 2, 3]) - elif action == 'aki-id': - return 'aki-%s' % (id_generator(lower=True)) - elif action == 'ami-id': - return 'ami-%s' % (id_generator(lower=True)) - elif action == 'ari-id': - return 'ari-%s' % (id_generator(lower=True)) - elif action == 'block-device-mapping': + elif action == "aki-id": + return "aki-%s" % (id_generator(lower=True)) + elif action == "ami-id": + return "ami-%s" % (id_generator(lower=True)) + elif action == "ari-id": + return "ari-%s" % (id_generator(lower=True)) + elif action == "block-device-mapping": nparams = params[1:] if not nparams: return "\n".join(BLOCK_DEVS) @@ -226,23 +227,23 @@ class MetaDataHandler(object): return "\n".join(sorted(list(DEV_MAPPINGS.keys()))) else: return str(subvalue) - elif action in ['hostname', 'local-hostname', 'public-hostname']: + elif action in ["hostname", "local-hostname", "public-hostname"]: # Just echo back there own hostname that they called in on.. return "%s" % (who) - elif action == 'instance-type': + elif action == "instance-type": return random.choice(INSTANCE_TYPES) - elif action == 'ami-manifest-path': - return 'my-amis/spamd-image.manifest.xml' - elif action == 'security-groups': - return 'default' - elif action in ['local-ipv4', 'public-ipv4']: + elif action == "ami-manifest-path": + return "my-amis/spamd-image.manifest.xml" + elif action == "security-groups": + return "default" + elif action in ["local-ipv4", "public-ipv4"]: # Just echo back there own ip that they called in on... - return "%s" % (kwargs.get('client_ip', '10.0.0.1')) - elif action == 'reservation-id': + return "%s" % (kwargs.get("client_ip", "10.0.0.1")) + elif action == "reservation-id": return "r-%s" % (id_generator(lower=True)) - elif action == 'product-codes': + elif action == "product-codes": return "%s" % (id_generator(size=8)) - elif action == 'public-keys': + elif action == "public-keys": nparams = params[1:] # This is a weird kludge, why amazon why!!! # public-keys is messed up, list of /latest/meta-data/public-keys/ @@ -267,51 +268,55 @@ class MetaDataHandler(object): hclient.NOT_FOUND, "Unknown key id %r" % mybe_key ) from e # Extract the possible sub-params - result = traverse(nparams[1:], { - "openssh-key": "\n".join(avail_keys[key_name]), - }) + result = traverse( + nparams[1:], + { + "openssh-key": "\n".join(avail_keys[key_name]), + }, + ) if isinstance(result, (dict)): # TODO(harlowja): This might not be right?? result = "\n".join(sorted(result.keys())) if not result: - result = '' + result = "" return result else: contents = [] for (i, key_id) in enumerate(key_ids): contents.append("%s=%s" % (i, key_id)) return "\n".join(contents) - elif action == 'placement': + elif action == "placement": nparams = params[1:] if not nparams: pcaps = sorted(PLACEMENT_CAPABILITIES.keys()) return "\n".join(pcaps) else: pentry = nparams[0].strip().lower() - if pentry == 'availability-zone': + if pentry == "availability-zone": zones = PLACEMENT_CAPABILITIES[pentry] return "%s" % random.choice(zones) else: - return "%s" % (PLACEMENT_CAPABILITIES.get(pentry, '')) + return "%s" % (PLACEMENT_CAPABILITIES.get(pentry, "")) else: - log.warning(("Did not implement action %s, " - "returning empty response: %r"), - action, NOT_IMPL_RESPONSE) + log.warning( + "Did not implement action %s, returning empty response: %r", + action, + NOT_IMPL_RESPONSE, + ) return NOT_IMPL_RESPONSE class UserDataHandler(object): - def __init__(self, opts): self.opts = opts def _get_user_blob(self, **kwargs): blob = None - if self.opts['user_data_file'] is not None: - blob = self.opts['user_data_file'] + if self.opts["user_data_file"] is not None: + blob = self.opts["user_data_file"] if not blob: blob_mp = { - 'hostname': kwargs.get('who', 'localhost'), + "hostname": kwargs.get("who", "localhost"), } lines = [ "#cloud-config", @@ -334,9 +339,8 @@ user_fetcher = None class Ec2Handler(BaseHTTPRequestHandler): - def _get_versions(self): - versions = ['latest'] + EC2_VERSIONS + versions = ["latest"] + EC2_VERSIONS versions = sorted(versions) return "\n".join(versions) @@ -347,33 +351,35 @@ class Ec2Handler(BaseHTTPRequestHandler): def _find_method(self, path): # Puke! (globals) func_mapping = { - 'user-data': user_fetcher.get_data, - 'meta-data': meta_fetcher.get_data, + "user-data": user_fetcher.get_data, + "meta-data": meta_fetcher.get_data, } - segments = [piece for piece in path.split('/') if len(piece)] + segments = [piece for piece in path.split("/") if len(piece)] log.info("Received segments %s", segments) if not segments: return self._get_versions date = segments[0].strip().lower() if date not in self._get_versions(): - raise WebException(hclient.BAD_REQUEST, - "Unknown version format %r" % date) + raise WebException( + hclient.BAD_REQUEST, "Unknown version format %r" % date + ) if len(segments) < 2: raise WebException(hclient.BAD_REQUEST, "No action provided") look_name = segments[1].lower() if look_name not in func_mapping: - raise WebException(hclient.BAD_REQUEST, - "Unknown requested data %r" % look_name) + raise WebException( + hclient.BAD_REQUEST, "Unknown requested data %r" % look_name + ) base_func = func_mapping[look_name] who = self.address_string() ip_from = self.client_address[0] if who == ip_from: # Nothing resolved, so just use 'localhost' - who = 'localhost' + who = "localhost" kwargs = { - 'params': list(segments[2:]), - 'who': who, - 'client_ip': ip_from, + "params": list(segments[2:]), + "who": who, + "client_ip": ip_from, } return functools.partial(base_func, **kwargs) @@ -384,12 +390,13 @@ class Ec2Handler(BaseHTTPRequestHandler): func = self._find_method(self.path) data = func() if not data: - data = '' + data = "" self.send_response(hclient.OK) self.send_header("Content-Type", "binary/octet-stream") self.send_header("Content-Length", len(data)) - log.info("Sending data (len=%s):\n%s", len(data), - format_text(data)) + log.info( + "Sending data (len=%s):\n%s", len(data), format_text(data) + ) self.end_headers() self.wfile.write(data.encode()) except RuntimeError as e: @@ -407,7 +414,7 @@ class Ec2Handler(BaseHTTPRequestHandler): self._do_response() -def setup_logging(log_level, fmt='%(levelname)s: @%(name)s : %(message)s'): +def setup_logging(log_level, fmt="%(levelname)s: @%(name)s : %(message)s"): root_logger = logging.getLogger() console_logger = logging.StreamHandler(sys.stdout) console_logger.setFormatter(logging.Formatter(fmt)) @@ -417,27 +424,47 @@ def setup_logging(log_level, fmt='%(levelname)s: @%(name)s : %(message)s'): def extract_opts(): parser = argparse.ArgumentParser() - parser.add_argument("-p", "--port", dest="port", action="store", type=int, - default=80, metavar="PORT", - help=("port from which to serve traffic" - " (default: %default)")) - parser.add_argument("-a", "--addr", dest="address", action="store", - type=str, default='::', metavar="ADDRESS", - help=("address from which to serve traffic" - " (default: %default)")) - parser.add_argument("-f", '--user-data-file', dest='user_data_file', - action='store', metavar='FILE', - help=("user data filename to serve back to" - "incoming requests")) - parser.add_argument('extra', nargs='*') + parser.add_argument( + "-p", + "--port", + dest="port", + action="store", + type=int, + default=80, + metavar="PORT", + help="port from which to serve traffic (default: %default)", + ) + parser.add_argument( + "-a", + "--addr", + dest="address", + action="store", + type=str, + default="::", + metavar="ADDRESS", + help="address from which to serve traffic (default: %default)", + ) + parser.add_argument( + "-f", + "--user-data-file", + dest="user_data_file", + action="store", + metavar="FILE", + help="user data filename to serve back toincoming requests", + ) + parser.add_argument("extra", nargs="*") args = parser.parse_args() - out = {'port': args.port, 'address': args.address, 'extra': args.extra, - 'user_data_file': None} + out = { + "port": args.port, + "address": args.address, + "extra": args.extra, + "user_data_file": None, + } if args.user_data_file: if not os.path.isfile(args.user_data_file): parser.error("Option -f specified a non-existent file") - with open(args.user_data_file, 'rb') as fh: - out['user_data_file'] = fh.read() + with open(args.user_data_file, "rb") as fh: + out["user_data_file"] = fh.read() return out @@ -455,14 +482,14 @@ def run_server(): setup_logging(logging.DEBUG) setup_fetchers(opts) log.info("CLI opts: %s", opts) - server_address = (opts['address'], opts['port']) + server_address = (opts["address"], opts["port"]) server = HTTPServerV6(server_address, Ec2Handler) sa = server.socket.getsockname() log.info("Serving ec2 metadata on %s using port %s ...", sa[0], sa[1]) server.serve_forever() -if __name__ == '__main__': +if __name__ == "__main__": run_server() # vi: ts=4 expandtab |