159 lines
4.3 KiB
Python
159 lines
4.3 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Serve cloud init files
|
|
"""
|
|
|
|
import configparser
|
|
import os
|
|
import socket
|
|
import sys
|
|
from ipaddress import AddressValueError, IPv4Address, IPv6Address
|
|
|
|
import yaml
|
|
import cherrypy
|
|
from cherrypy.lib.static import serve_file
|
|
|
|
|
|
PATH = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
CONFIG = configparser.ConfigParser()
|
|
CONFIG.read(os.path.join(PATH, "config.ini"))
|
|
|
|
USER_DATA_FILENAME = CONFIG["app"].get("user_data", "user-data")
|
|
META_DATA_FILENAME = CONFIG["app"].get("meta_data", "meta-data")
|
|
REDIRECT_FILENAME = CONFIG["app"].get("redirect", "redirect")
|
|
|
|
|
|
class CloudInitApp:
|
|
"""
|
|
Serve cloud init files
|
|
"""
|
|
|
|
def __init__(self):
|
|
self.remoteip = None
|
|
self.hostinfo = ('localhost', )
|
|
|
|
def _can_ip_be_proxy(self):
|
|
self.remoteip = cherrypy.request.remote.ip
|
|
try:
|
|
ipobj = IPv4Address(self.remoteip)
|
|
except AddressValueError:
|
|
try:
|
|
ipobj = IPv6Address(self.remoteip)
|
|
except AddressValueError:
|
|
return False
|
|
return not ipobj.is_global
|
|
|
|
def _init_ip(self):
|
|
"""
|
|
Get remote IP
|
|
"""
|
|
if self._can_ip_be_proxy():
|
|
try:
|
|
self.remoteip = cherrypy.request.headers.get(
|
|
'X-Real-Ip',
|
|
cherrypy.request.remote.ip
|
|
)
|
|
except KeyError:
|
|
pass
|
|
|
|
try:
|
|
self.hostinfo = socket.gethostbyaddr(self.remoteip)
|
|
except socket.herror:
|
|
pass
|
|
|
|
def _redirect_if_needed(self):
|
|
filepath = os.path.join(PATH, "data", self.hostinfo[0],
|
|
REDIRECT_FILENAME)
|
|
if os.path.exists(filepath):
|
|
try:
|
|
with open(filepath) as redirect:
|
|
content = redirect.read().splitlines()
|
|
raise cherrypy.HTTPRedirect(content[0], 301)
|
|
except IOError:
|
|
return False
|
|
return False
|
|
|
|
@cherrypy.expose
|
|
def user_data(self):
|
|
"""
|
|
Serves a static file
|
|
"""
|
|
self._init_ip()
|
|
self._redirect_if_needed()
|
|
filepath = os.path.join(PATH, "data", self.hostinfo[0],
|
|
USER_DATA_FILENAME)
|
|
if not os.path.exists(filepath):
|
|
filepath = os.path.join(PATH, "data", USER_DATA_FILENAME)
|
|
return serve_file(filepath, "text/yaml", "attachment")
|
|
|
|
@cherrypy.expose
|
|
def meta_data(self):
|
|
"""
|
|
Return meta-data in YAML
|
|
"""
|
|
self._init_ip()
|
|
self._redirect_if_needed()
|
|
hostname = self.hostinfo[0]
|
|
data = {
|
|
"instance-id": hostname.split(".")[0],
|
|
"local-hostname": hostname
|
|
}
|
|
|
|
filepath = os.path.join(PATH, "data", hostname, META_DATA_FILENAME)
|
|
if os.path.exists(filepath):
|
|
with open(filepath, "r") as metadata:
|
|
for line in metadata.readlines():
|
|
linesplit = list(map(lambda k: k.strip(), line.split(":")))
|
|
data[linesplit[0]] = linesplit[1]
|
|
|
|
cherrypy.response.headers['Content-Type'] = \
|
|
'text/yaml'
|
|
cherrypy.response.headers['Content-Disposition'] = \
|
|
'attachment; filename="user-data"'
|
|
return yaml.dump(data)
|
|
|
|
@cherrypy.expose
|
|
def vendor_data(self):
|
|
"""
|
|
Return empty vendor-data
|
|
"""
|
|
return ""
|
|
|
|
@cherrypy.expose
|
|
def finished(self, data):
|
|
"""
|
|
Saves additional meta-data
|
|
|
|
:param data: meta-data to be added
|
|
"""
|
|
self._init_ip()
|
|
folder = os.path.join(PATH, "data", self.hostinfo[0])
|
|
if not os.path.exists(folder):
|
|
os.makedirs(folder)
|
|
|
|
with open(os.path.join(folder, META_DATA_FILENAME), "w") as fin:
|
|
fin.write(data)
|
|
|
|
|
|
ROOT = CloudInitApp()
|
|
|
|
if __name__ == "__main__":
|
|
cherrypy.server.socket_host = \
|
|
CONFIG["server"].get("server_host", "127.0.0.1")
|
|
cherrypy.server.socket_port = \
|
|
CONFIG["server"].getint("server_port", 8081)
|
|
ENGINE = cherrypy.engine
|
|
|
|
cherrypy.tree.mount(ROOT)
|
|
if hasattr(ENGINE, "signal_handler"):
|
|
ENGINE.signal_handler.subscribe()
|
|
if hasattr(ENGINE, "console_control_handler"):
|
|
ENGINE.console_control_handler.subscribe()
|
|
try:
|
|
ENGINE.start()
|
|
except Exception:
|
|
sys.exit(1)
|
|
else:
|
|
ENGINE.block()
|