35 lines
933 B
Python
35 lines
933 B
Python
# Copyright (C) 2023 Canonical Ltd.
|
||
# Author: Benjamin Drung <benjamin.drung@canonical.com>
|
||
# SPDX-License-Identifier: GPL-2.0-or-later
|
||
|
||
"""Functions around users and groups."""
|
||
|
||
import dataclasses
|
||
import os
|
||
|
||
|
||
@dataclasses.dataclass()
|
||
class UserGroupID:
|
||
"""Pair of user and group ID."""
|
||
|
||
uid: int
|
||
gid: int
|
||
|
||
def is_root(self) -> bool:
|
||
"""Check if the user or group ID is root."""
|
||
return self.uid == 0 or self.gid == 0
|
||
|
||
@classmethod
|
||
def from_systemd_coredump(cls, coredump):
|
||
"""Extract user and group from systemd-coredump dictionary."""
|
||
uid = coredump.get("COREDUMP_UID")
|
||
assert isinstance(uid, int)
|
||
gid = coredump.get("COREDUMP_GID")
|
||
assert isinstance(gid, int)
|
||
return cls(uid, gid)
|
||
|
||
|
||
def get_process_user_and_group() -> UserGroupID:
|
||
"""Return the current process’s real user and group."""
|
||
return UserGroupID(os.getuid(), os.getgid())
|