c5ebf8deec
Most of the Windows-specific plugins were able to work on other platforms as well, as long as osutils provides the relevant functionalities. This restructuring is an effort for helping future additions of other platforms. Change-Id: I21a35eb6eb8e2439cfdf861c59afc51c3618a779
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
|