ff73d41609
The userdatautils plugin now supports EC2 formatted scripts. It consists of multiple batch or powershell scripts enclosed into specific tags, where the batch part is executed before the powershell one. Change-Id: Ic9b2c73838c606d3eeaf039363dbdf46313379ca
65 lines
2.0 KiB
Python
65 lines
2.0 KiB
Python
# Copyright 2014 Cloudbase Solutions Srl
|
|
#
|
|
# 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.
|
|
|
|
|
|
import functools
|
|
import re
|
|
|
|
from cloudbaseinit.openstack.common import log as logging
|
|
from cloudbaseinit.plugins.common import execcmd
|
|
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
# Avoid 80+ length by using a local variable, which
|
|
# is deleted afterwards.
|
|
_compile = functools.partial(re.compile, flags=re.I)
|
|
FORMATS = (
|
|
(_compile(br'^rem\s+cmd\s'), execcmd.Shell),
|
|
(_compile(br'^#!\s*/usr/bin/env\s+python\s'), execcmd.Python),
|
|
(_compile(br'^#!'), execcmd.Bash),
|
|
(_compile(br'^#(ps1|ps1_sysnative)\s'), execcmd.PowershellSysnative),
|
|
(_compile(br'^#ps1_x86\s'), execcmd.Powershell),
|
|
(_compile(br'</?(script|powershell)>'), execcmd.EC2Config),
|
|
)
|
|
del _compile
|
|
|
|
|
|
def _get_command(data):
|
|
# Get the command which should process the given data.
|
|
for pattern, command_class in FORMATS:
|
|
if pattern.search(data):
|
|
return command_class.from_data(data)
|
|
|
|
|
|
def execute_user_data_script(user_data):
|
|
ret_val = 0
|
|
out = err = None
|
|
command = _get_command(user_data)
|
|
if not command:
|
|
LOG.warning('Unsupported user_data format')
|
|
return ret_val
|
|
|
|
try:
|
|
out, err, ret_val = command()
|
|
except Exception as exc:
|
|
LOG.warning('An error occurred during user_data execution: \'%s\'',
|
|
exc)
|
|
else:
|
|
LOG.debug('User_data stdout:\n%s', out)
|
|
LOG.debug('User_data stderr:\n%s', err)
|
|
|
|
LOG.info('User_data script ended with return code: %d', ret_val)
|
|
return ret_val
|