Skip to content

Commit

Permalink
[crmsh-4.6] Fix: bootstrap: check is_nologin more robustly (bsc#12282…
Browse files Browse the repository at this point in the history
…51) (#1556)

backport:

* #1553
  • Loading branch information
liangxin1300 authored Sep 20, 2024
2 parents dbc2fec + 2f6bfba commit e6f00fb
Show file tree
Hide file tree
Showing 2 changed files with 21 additions and 20 deletions.
21 changes: 12 additions & 9 deletions crmsh/bootstrap.py
Original file line number Diff line number Diff line change
Expand Up @@ -1003,15 +1003,18 @@ def is_nologin(user, remote=None):
"""
Check if user's shell is nologin
"""
passwd_file = "/etc/passwd"
pattern = f"{user}:.*:/.*/nologin"
if remote:
cmd = f"cat {passwd_file}|grep {pattern}"
rc, _, _ = sh.cluster_shell().get_rc_stdout_stderr_without_input(remote, cmd)
return rc == 0
else:
with open(passwd_file) as f:
return re.search(pattern, f.read()) is not None
rc, error = sh.cluster_shell().get_rc_and_error(
remote, None,
"set -e\n"
f"shell=$(getent passwd '{user}' | awk -F: '{{ print $NF }}')\n"
'[ -n "${shell}" ] && [ -f "${shell}" ] && [ -x "${shell}" ] || exit 1\n'
'case $(basename "$shell") in\n'
' nologin) exit 1 ;;\n'
' false) exit 1 ;;\n'
'esac\n'
'"${shell}" < /dev/null &>/dev/null\n'
)
return 0 != rc


def change_user_shell(user, remote=None):
Expand Down
20 changes: 9 additions & 11 deletions test/unittests/test_bootstrap.py
Original file line number Diff line number Diff line change
Expand Up @@ -443,15 +443,18 @@ def test_start_pacemaker(self, mock_installed, mock_enabled, mock_delay_start, m
mock.call(node_list, "systemctl daemon-reload"),
])

@mock.patch('crmsh.bootstrap.change_user_shell')
@mock.patch('crmsh.bootstrap.configure_ssh_key')
@mock.patch('crmsh.service_manager.ServiceManager.start_service')
def test_init_ssh(self, mock_start_service, mock_config_ssh):
def test_init_ssh(self, mock_start_service, mock_config_ssh, mock_change_user_shell):
bootstrap._context = mock.Mock(current_user="alice", user_at_node_list=[], use_ssh_agent=False)
bootstrap.init_ssh()
mock_start_service.assert_called_once_with("sshd.service", enable=True)
mock_config_ssh.assert_has_calls([
mock.call("alice")
])
mock.call("alice"),
mock.call("hacluster"),
])
mock_change_user_shell.assert_called_once_with("hacluster")

@mock.patch('crmsh.userdir.gethomedir')
def test_key_files(self, mock_gethome):
Expand All @@ -460,13 +463,6 @@ def test_key_files(self, mock_gethome):
self.assertEqual(bootstrap.key_files("root"), expected_res)
mock_gethome.assert_called_once_with("root")

@mock.patch('builtins.open')
def test_is_nologin(self, mock_open_file):
data = "hacluster:x:90:90:heartbeat processes:/var/lib/heartbeat/cores/hacluster:/sbin/nologin"
mock_open_file.return_value = mock.mock_open(read_data=data).return_value
assert bootstrap.is_nologin("hacluster") is not None
mock_open_file.assert_called_once_with("/etc/passwd")

@mock.patch('crmsh.bootstrap.confirm')
@mock.patch('logging.Logger.info')
@mock.patch('crmsh.bootstrap.is_nologin')
Expand Down Expand Up @@ -536,10 +532,12 @@ def _test_configure_ssh_key(self, mock_change_shell, mock_key_files, mock_detect

@mock.patch('crmsh.ssh_key.AuthorizedKeyManager.add')
@mock.patch('crmsh.ssh_key.KeyFileManager.ensure_key_pair_exists_for_user')
def test_configure_ssh_key(self, mock_ensure_key_pair, mock_add):
@mock.patch('crmsh.bootstrap.change_user_shell')
def test_configure_ssh_key(self, mock_change_user_shell, mock_ensure_key_pair, mock_add):
public_key = crmsh.ssh_key.InMemoryPublicKey('foo')
mock_ensure_key_pair.return_value = (True, [public_key])
bootstrap.configure_ssh_key('alice')
mock_change_user_shell.assert_called_once_with('alice')
mock_ensure_key_pair.assert_called_once_with(None, 'alice')
mock_add.assert_called_once_with(None, 'alice', public_key)

Expand Down

0 comments on commit e6f00fb

Please sign in to comment.