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
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
|
# Microsoft Azure Linux Agent
#
# Copyright 2018 Microsoft Corporation
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
# Requires Python 2.6+ and Openssl 1.0+
#
import os
import os.path
import time
from datetime import datetime
import azurelinuxagent.common.conf as conf
import azurelinuxagent.common.logger as logger
import azurelinuxagent.common.utils.fileutil as fileutil
import azurelinuxagent.common.utils.shellutil as shellutil # pylint: disable=W0611
from azurelinuxagent.common.event import elapsed_milliseconds, WALAEventOperation # pylint: disable=W0611
from azurelinuxagent.common.exception import ProvisionError, ProtocolError
from azurelinuxagent.common.future import ustr
from azurelinuxagent.common.protocol.util import OVF_FILE_NAME
from azurelinuxagent.common.protocol.ovfenv import OvfEnv
from azurelinuxagent.pa.provision.default import ProvisionHandler
from azurelinuxagent.pa.provision.cloudinitdetect import cloud_init_is_enabled
class CloudInitProvisionHandler(ProvisionHandler):
def __init__(self): # pylint: disable=W0235
super(CloudInitProvisionHandler, self).__init__()
def run(self):
try:
if super(CloudInitProvisionHandler, self).check_provisioned_file():
logger.info("Provisioning already completed, skipping.")
return
utc_start = datetime.utcnow()
logger.info("Running CloudInit provisioning handler")
self.wait_for_ovfenv()
self.protocol_util.get_protocol() # Trigger protocol detection
self.report_not_ready("Provisioning", "Starting")
thumbprint = self.wait_for_ssh_host_key() # pylint: disable=W0612
self.write_provisioned()
logger.info("Finished provisioning")
self.report_ready()
self.report_event("Provisioning with cloud-init succeeded ({0}s)".format(self._get_uptime_seconds()),
is_success=True,
duration=elapsed_milliseconds(utc_start))
except ProvisionError as e:
msg = "Provisioning with cloud-init failed: {0} ({1}s)".format(ustr(e), self._get_uptime_seconds())
logger.error(msg)
self.report_not_ready("ProvisioningFailed", ustr(e))
self.report_event(msg)
return
def wait_for_ovfenv(self, max_retry=1800, sleep_time=1):
"""
Wait for cloud-init to copy ovf-env.xml file from provision ISO
"""
ovf_file_path = os.path.join(conf.get_lib_dir(), OVF_FILE_NAME)
logging_interval = 10
max_logging_interval = 320
for retry in range(0, max_retry):
if os.path.isfile(ovf_file_path):
try:
ovf_env = OvfEnv(fileutil.read_file(ovf_file_path))
self.handle_provision_guest_agent(ovf_env.provision_guest_agent)
return
except ProtocolError as pe:
raise ProvisionError("OVF xml could not be parsed "
"[{0}]: {1}".format(ovf_file_path,
ustr(pe)))
else:
if retry < max_retry - 1:
if retry % logging_interval == 0:
logger.info(
"Waiting for cloud-init to copy ovf-env.xml to {0} "
"[{1} retries remaining, "
"sleeping {2}s between retries]".format(ovf_file_path,
max_retry - retry,
sleep_time))
if not cloud_init_is_enabled():
logger.warn("cloud-init does not appear to be enabled")
logging_interval = min(logging_interval * 2, max_logging_interval)
time.sleep(sleep_time)
raise ProvisionError("Giving up, ovf-env.xml was not copied to {0} "
"after {1}s".format(ovf_file_path,
max_retry * sleep_time))
def wait_for_ssh_host_key(self, max_retry=1800, sleep_time=1):
"""
Wait for cloud-init to generate ssh host key
"""
keypair_type = conf.get_ssh_host_keypair_type() # pylint: disable=W0612
path = conf.get_ssh_key_public_path()
logging_interval = 10
max_logging_interval = 320
for retry in range(0, max_retry):
if os.path.isfile(path):
logger.info("ssh host key found at: {0}".format(path))
try:
thumbprint = self.get_ssh_host_key_thumbprint(chk_err=False)
logger.info("Thumbprint obtained from : {0}".format(path))
return thumbprint
except ProvisionError:
logger.warn("Could not get thumbprint from {0}".format(path))
if retry < max_retry - 1:
if retry % logging_interval == 0:
logger.info("Waiting for ssh host key be generated at {0} "
"[{1} attempts remaining, "
"sleeping {2}s between retries]".format(path,
max_retry - retry,
sleep_time))
if not cloud_init_is_enabled():
logger.warn("cloud-init does not appear to be running")
logging_interval = min(logging_interval * 2, max_logging_interval)
time.sleep(sleep_time)
raise ProvisionError("Giving up, ssh host key was not found at {0} "
"after {1}s".format(path,
max_retry * sleep_time))
|
