56 lines
2.1 KiB
Text
56 lines
2.1 KiB
Text
|
#!/usr/bin/python3
|
||
|
|
#
|
||
|
|
# handle-sshpw: Code processing sshpw lines in kickstart files for the
|
||
|
|
# install environment.
|
||
|
|
#
|
||
|
|
# Copyright (C) 2012-2015 Red Hat, Inc. All rights reserved.
|
||
|
|
#
|
||
|
|
# This program is free software; you can redistribute it and/or modify
|
||
|
|
# it under the terms of the GNU General Public License as published by
|
||
|
|
# the Free Software Foundation; either version 2 of the License, or
|
||
|
|
# (at your option) any later version.
|
||
|
|
#
|
||
|
|
# 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/>.
|
||
|
|
#
|
||
|
|
# Some of this code comes from the old pyanaconda/sshd.py
|
||
|
|
#
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
from pykickstart.parser import KickstartParser
|
||
|
|
from pykickstart.version import makeVersion
|
||
|
|
from pykickstart.sections import NullSection
|
||
|
|
import pyanaconda.core.users as users
|
||
|
|
from pyanaconda.core.kickstart.version import VERSION
|
||
|
|
|
||
|
|
ksfile = '/run/install/ks.cfg'
|
||
|
|
|
||
|
|
# see if we have a file to work with
|
||
|
|
if not os.path.exists(ksfile):
|
||
|
|
sys.exit()
|
||
|
|
|
||
|
|
handler = makeVersion(VERSION)
|
||
|
|
ksparser = KickstartParser(handler, missingIncludeIsFatal=False)
|
||
|
|
ksparser.registerSection(NullSection(handler, sectionOpen="%addon"))
|
||
|
|
ksparser.readKickstart(ksfile)
|
||
|
|
|
||
|
|
userdata = ksparser.handler.sshpw.dataList()
|
||
|
|
for ud in userdata:
|
||
|
|
if users.check_user_exists(ud.username, root="/"):
|
||
|
|
if not ud.sshkey:
|
||
|
|
users.set_user_password(username=ud.username, password=ud.password,
|
||
|
|
is_crypted=ud.isCrypted, lock=ud.lock)
|
||
|
|
else:
|
||
|
|
users.create_user(username=ud.username, password=ud.password, is_crypted=ud.isCrypted,
|
||
|
|
lock=ud.lock, root="/")
|
||
|
|
|
||
|
|
if ud.sshkey:
|
||
|
|
# Setup the account so that only the sshkey can be used
|
||
|
|
users.set_user_password(username=ud.username, password="*", is_crypted=True, lock=False)
|
||
|
|
users.set_user_ssh_key(username=ud.username, key=ud.password, root="/")
|