anaconda/anaconda-40.22.3.13/utils/handle-sshpw
2024-11-14 21:39:56 -08:00

55 lines
2.1 KiB
Python
Executable file

#!/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="/")