mirror of
https://gitlab.postmarketos.org/postmarketOS/pmbootstrap.git
synced 2025-07-23 04:25:10 +03:00
2f3f36397f
These helpers could be used to run commands as root prior to unsharing namespaces, for example to probe the binfmt module. They aren't used currently but are left in just in case... Signed-off-by: Casey Connolly <kcxt@postmarketos.org>
38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
# Copyright 2023 Anjandev Momi
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
import os
|
|
import shutil
|
|
from functools import lru_cache
|
|
|
|
|
|
@lru_cache
|
|
def which_sudo() -> str | None:
|
|
"""Return a command required to run commands as root, if any.
|
|
|
|
Find whether sudo or doas is installed for commands that require root.
|
|
Allows user to override preferred sudo with PMB_SUDO env variable.
|
|
"""
|
|
if os.getuid() == 0:
|
|
return None
|
|
|
|
supported_sudos = ["doas", "sudo"]
|
|
|
|
user_set_sudo = os.getenv("PMB_SUDO")
|
|
if user_set_sudo is not None:
|
|
if shutil.which(user_set_sudo) is None:
|
|
raise RuntimeError(
|
|
"PMB_SUDO environmental variable is set to"
|
|
f" {user_set_sudo} but pmbootstrap cannot find"
|
|
" this command on your system."
|
|
)
|
|
return user_set_sudo
|
|
|
|
for sudo in supported_sudos:
|
|
if shutil.which(sudo) is not None:
|
|
return sudo
|
|
|
|
raise RuntimeError(
|
|
"Can't find sudo or doas required to run pmbootstrap."
|
|
" Please install sudo, doas, or specify your own sudo"
|
|
" with the PMB_SUDO environmental variable."
|
|
)
|