70 lines
2.6 KiB
Python
70 lines
2.6 KiB
Python
# -*- coding:utf-8 -*-
|
|
#############################################################################
|
|
# Portions Copyright (c) 2020 Huawei Technologies Co.,Ltd.
|
|
# Portions Copyright (c) 2007 Agendaless Consulting and Contributors.
|
|
#
|
|
# openGauss is licensed under Mulan PSL v2.
|
|
# You can use this software according to the terms
|
|
# and conditions of the Mulan PSL v2.
|
|
# You may obtain a copy of Mulan PSL v2 at:
|
|
#
|
|
# http://license.coscl.org.cn/MulanPSL2
|
|
#
|
|
# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OF ANY KIND,
|
|
# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
|
|
# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
|
|
# See the Mulan PSL v2 for more details.
|
|
# ----------------------------------------------------------------------------
|
|
# Description : sshd config file operation.
|
|
#############################################################################
|
|
|
|
import subprocess
|
|
|
|
from gspylib.common.ErrorCode import ErrorCode
|
|
|
|
|
|
class SshdConfig:
|
|
"""
|
|
sshd config file operation
|
|
"""
|
|
|
|
def __init__(self):
|
|
pass
|
|
|
|
@staticmethod
|
|
def setKeyValueInSshd(key, value):
|
|
"""
|
|
function: Set a (key, value) pair into /etc/ssh/sshd_config,
|
|
before "Match" section.
|
|
"Match" section in sshd_config should always places in the end.
|
|
Attention: you need to remove the old (key, value)
|
|
from sshd_config manually.
|
|
input:
|
|
key: the configuration name of sshd_config
|
|
value: the configuration value(Only single line string
|
|
permitted here).
|
|
output:
|
|
void
|
|
"""
|
|
sshd_config = '/etc/ssh/sshd_config'
|
|
cmd = "grep -E '^\<Match\>' %s" % sshd_config
|
|
(status, output) = subprocess.getstatusoutput(cmd)
|
|
|
|
if status == 0:
|
|
cmd = "sed -i '/^\<Match\>.*/i %s %s' %s" % (key, value,
|
|
sshd_config)
|
|
else:
|
|
if output is not None and len(output.strip()) != 0:
|
|
raise Exception(ErrorCode.GAUSS_503["GAUSS_50321"] %
|
|
"Match section" + "Command: %s, Error: %s" %
|
|
(cmd, output))
|
|
cmd = "echo '' >> %s ; echo '%s %s' >> %s" % (sshd_config,
|
|
key, value,
|
|
sshd_config)
|
|
|
|
(status, output) = subprocess.getstatusoutput(cmd)
|
|
if status != 0:
|
|
raise Exception((ErrorCode.GAUSS_503["GAUSS_50320"] % (
|
|
key, value)) + ("Command: %s, Error: %s" % (cmd, output)))
|