dbc030b2ab
on Ubuntu systems via cloud-config (LP: #1041384). - Fixed bug with user creation on Ubuntu where the default user groups are not set properly (LP: #1044044). - Fixed documentation for user creation (LP: #1044508).
102 lines
2.9 KiB
Python
102 lines
2.9 KiB
Python
# vi: ts=4 expandtab
|
|
#
|
|
# Copyright (C) 2009-2010 Canonical Ltd.
|
|
# Copyright (C) 2012 Hewlett-Packard Development Company, L.P.
|
|
#
|
|
# Author: Scott Moser <scott.moser@canonical.com>
|
|
# Author: Juerg Haefliger <juerg.haefliger@hp.com>
|
|
#
|
|
# 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/>.
|
|
|
|
from cloudinit import util
|
|
import pwd
|
|
|
|
# The ssh-import-id only seems to exist on ubuntu (for now)
|
|
# https://launchpad.net/ssh-import-id
|
|
distros = ['ubuntu']
|
|
|
|
|
|
def handle(_name, cfg, cloud, log, args):
|
|
|
|
# import for "user: XXXXX"
|
|
if len(args) != 0:
|
|
user = args[0]
|
|
ids = []
|
|
if len(args) > 1:
|
|
ids = args[1:]
|
|
|
|
import_ssh_ids(ids, user, log)
|
|
return
|
|
|
|
# import for cloudinit created users
|
|
elist = []
|
|
for user_cfg in cfg['users']:
|
|
user = None
|
|
import_ids = []
|
|
|
|
if isinstance(user_cfg, str) and user_cfg == "default":
|
|
user = cloud.distro.get_default_user()
|
|
if not user:
|
|
continue
|
|
|
|
import_ids = util.get_cfg_option_list(cfg, "ssh_import_id", [])
|
|
|
|
elif isinstance(user_cfg, dict):
|
|
user = None
|
|
import_ids = []
|
|
|
|
try:
|
|
user = user_cfg['name']
|
|
import_ids = user_cfg['ssh_import_id']
|
|
|
|
if import_ids and isinstance(import_ids, str):
|
|
import_ids = import_ids.split(',')
|
|
|
|
except:
|
|
log.debug("user %s is not configured for ssh_import" % user)
|
|
continue
|
|
|
|
if not len(import_ids):
|
|
continue
|
|
|
|
try:
|
|
import_ssh_ids(import_ids, user, log)
|
|
except Exception as exc:
|
|
util.logexc(log, "ssh-import-id failed for: %s %s" %
|
|
(user, import_ids), exc)
|
|
elist.append(exc)
|
|
|
|
if len(elist):
|
|
raise elist[0]
|
|
|
|
|
|
def import_ssh_ids(ids, user, log):
|
|
|
|
if not (user and ids):
|
|
log.debug("empty user(%s) or ids(%s). not importing", user, ids)
|
|
return
|
|
|
|
try:
|
|
_check = pwd.getpwnam(user)
|
|
except KeyError as exc:
|
|
raise exc
|
|
|
|
cmd = ["sudo", "-Hu", user, "ssh-import-id"] + ids
|
|
log.debug("Importing ssh ids for user %s.", user)
|
|
|
|
try:
|
|
util.subp(cmd, capture=False)
|
|
except util.ProcessExecutionError as exc:
|
|
util.logexc(log, "Failed to run command to import %s ssh ids", user)
|
|
raise exc
|