cc_seed_random.py 3.12 KB
Newer Older
Joshua Harlow's avatar
Joshua Harlow committed
1 2 3
# vi: ts=4 expandtab
#
#    Copyright (C) 2013 Yahoo! Inc.
4
#    Copyright (C) 2014 Canonical, Ltd
Joshua Harlow's avatar
Joshua Harlow committed
5 6
#
#    Author: Joshua Harlow <harlowja@yahoo-inc.com>
7 8
#    Author: Dustin Kirkland <kirkland@ubuntu.com>
#    Author: Scott Moser <scott.moser@canonical.com>
Joshua Harlow's avatar
Joshua Harlow committed
9 10 11 12 13 14 15 16 17 18 19 20 21
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU General Public License version 3, as
#    published by the Free Software Foundation.
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU General Public License for more details.
#
#    You should have received a copy of the GNU General Public License
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.

22
import base64
23
import os
24

Scott Moser's avatar
Scott Moser committed
25
from six import BytesIO
26

27
from cloudinit import log as logging
28
from cloudinit.settings import PER_INSTANCE
29
from cloudinit import util
Joshua Harlow's avatar
Joshua Harlow committed
30 31

frequency = PER_INSTANCE
32
LOG = logging.getLogger(__name__)
33 34 35 36


def _decode(data, encoding=None):
    if not data:
Scott Moser's avatar
Scott Moser committed
37
        return b''
38
    if not encoding or encoding.lower() in ['raw']:
Scott Moser's avatar
Scott Moser committed
39
        return util.encode_text(data)
40
    elif encoding.lower() in ['base64', 'b64']:
Scott Moser's avatar
Scott Moser committed
41
        return base64.b64decode(data)
42
    elif encoding.lower() in ['gzip', 'gz']:
Scott Moser's avatar
Scott Moser committed
43
        return util.decomp_gzip(data, quiet=False, decode=None)
44 45
    else:
        raise IOError("Unknown random_seed encoding: %s" % (encoding))
Joshua Harlow's avatar
Joshua Harlow committed
46 47


48 49 50 51 52
def handle_random_seed_command(command, required, env=None):
    if not command and required:
        raise ValueError("no command found but required=true")
    elif not command:
        LOG.debug("no command provided")
53 54
        return

55 56 57 58 59 60 61
    cmd = command[0]
    if not util.which(cmd):
        if required:
            raise ValueError("command '%s' not found but required=true", cmd)
        else:
            LOG.debug("command '%s' not found for seed_command", cmd)
            return
62
    util.subp(command, env=env, capture=False)
63 64 65 66 67


def handle(name, cfg, cloud, log, _args):
    mycfg = cfg.get('random_seed', {})
    seed_path = mycfg.get('file', '/dev/urandom')
Scott Moser's avatar
Scott Moser committed
68
    seed_data = mycfg.get('data', b'')
69

Scott Moser's avatar
Scott Moser committed
70
    seed_buf = BytesIO()
71 72
    if seed_data:
        seed_buf.write(_decode(seed_data, encoding=mycfg.get('encoding')))
73

74 75
    # 'random_seed' is set up by Azure datasource, and comes already in
    # openstack meta_data.json
76 77
    metadata = cloud.datasource.metadata
    if metadata and 'random_seed' in metadata:
Scott Moser's avatar
Scott Moser committed
78
        seed_buf.write(util.encode_text(metadata['random_seed']))
79 80 81

    seed_data = seed_buf.getvalue()
    if len(seed_data):
82
        log.debug("%s: adding %s bytes of random seed entropy to %s", name,
83 84
                  len(seed_data), seed_path)
        util.append_file(seed_path, seed_data)
85

86
    command = mycfg.get('command', None)
87 88 89 90 91 92 93 94
    req = mycfg.get('command_required', False)
    try:
        env = os.environ.copy()
        env['RANDOM_SEED_FILE'] = seed_path
        handle_random_seed_command(command=command, required=req, env=env)
    except ValueError as e:
        log.warn("handling random command [%s] failed: %s", command, e)
        raise e