1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
import os
from cloudinit import util
from cloudinit.settings import (PER_ALWAYS, PER_INSTANCE)
from cloudinit import log as logging
LOG = logging.getLogger(__name__)
CONTENT_END = "__end__"
CONTENT_START = "__begin__"
PART_CONTENT_TYPES = ["text/part-handler"]
PART_HANDLER_FN_TMPL = 'part-handler-%03d'
UNDEF_HANDLER_VERSION = 1
class PartHandler(object):
def __init__(self, frequency, version=2):
self.handler_version = version
self.frequency = frequency
def __repr__(self):
return "%s: [%s]" % (self.__class__.__name__, self.list_types())
def list_types(self):
raise NotImplementedError()
def handle_part(self, data, ctype, filename, payload, frequency):
return self._handle_part(data, ctype, filename, payload, frequency)
def _handle_part(self, data, ctype, filename, payload, frequency):
raise NotImplementedError()
def fixup_module(mod):
if not hasattr(mod, "handler_version"):
setattr(mod, "handler_version", UNDEF_HANDLER_VERSION)
if not hasattr(mod, 'list_types'):
def empty_types():
return []
setattr(mod, 'list_types', empty_types)
if not hasattr(mod, frequency):
setattr(mod, 'frequency', PER_INSTANCE)
if not hasattr(mod, 'handle_part'):
def empty_handler(data, ctype, filename, payload):
pass
setattr(mod, 'handle_part', empty_handler)
return mod
def find_module_files(root_dir):
entries = dict()
for fname in glob.glob(os.path.join(root_dir, "*.py")):
if not os.path.isfile(fname):
continue
modname = os.path.basename(fname)[0:-3]
entries[fname] = modname
return entries
def run_part(mod, data, ctype, filename, payload, frequency):
# only add the handler if the module should run
mod_freq = getattr(mod, "frequency")
if not (mod_freq == PER_ALWAYS or
(frequency == PER_INSTANCE and mod_freq == PER_INSTANCE)):
return
try:
mod_ver = getattr(mod, 'handler_version')
if mod_ver == 1:
mod.handle_part(data, ctype, filename, payload)
else:
mod.handle_part(data, ctype, filename, payload, frequency)
except:
LOG.exception("Failed calling mod %s (%s, %s, %s) with frequency %s", mod, ctype, filename, mod_ver, frequency)
def call_begin(mod, data, frequency):
run_part(mod, data, CONTENT_START, None, None, frequency)
def call_end(mod, data, frequency):
run_part(mod, data, CONTENT_END, None, None, frequency)
def walker_handle_handler(pdata, _ctype, _filename, payload):
curcount = pdata['handlercount']
modname = PART_HANDLER_FN_TMPL % (curcount)
frequency = pdata['frequency']
modfname = os.path.join(pdata['handlerdir'], "%s.py" % (modname))
# TODO: Check if path exists??
util.write_file(modfname, payload, 0600)
handlers = pdata['handlers']
try:
mod = fixup_module(importer.import_module(modname))
handlers.register(mod)
call_begin(mod, pdata['data'], frequency)
pdata['handlercount'] = curcount + 1
except:
LOG.exception("Failed at registered python file %s", modfname)
def walker_callback(pdata, ctype, filename, payload):
# data here is the part_handlers array and then the data to pass through
if ctype in PART_CONTENT_TYPES:
walker_handle_handler(pdata, ctype, filename, payload)
return
handlers = pdata['handlers']
if ctype not in handlers:
if ctype == "text/x-not-multipart":
# Extract the first line or 24 bytes for displaying in the log
start = payload.split("\n", 1)[0][:24]
if start < payload:
details = "starting '%s...'" % start.encode("string-escape")
else:
details = repr(payload)
LOG.warning("Unhandled non-multipart userdata: %s", details)
return
run_part(handlers[ctype], pdata['data'], ctype, filename, payload, pdata['frequency'])
|
