src/basic/process-util.h

   1 /* SPDX-License-Identifier: LGPL-2.1+ */
   2 #pragma once
   3
   4 /***
   5   This file is part of systemd.
   6
   7   Copyright 2010 Lennart Poettering
   8 ***/
   9
  10 #include <alloca.h>
  11 #include <errno.h>
  12 #include <sched.h>
  13 #include <signal.h>
  14 #include <stdbool.h>
  15 #include <stddef.h>
  16 #include <stdio.h>
  17 #include <string.h>
  18 #include <sys/resource.h>
  19 #include <sys/types.h>
  20
  21 #include "format-util.h"
  22 //#include "ioprio.h"
  23 #include "macro.h"
  24 #include "time-util.h"
  25
  26 #define procfs_file_alloca(pid, field)                                  \
  27         ({                                                              \
  28                 pid_t _pid_ = (pid);                                    \
  29                 const char *_r_;                                        \
  30                 if (_pid_ == 0) {                                       \
  31                         _r_ = ("/proc/self/" field);                    \
  32                 } else {                                                \
  33                         _r_ = alloca(STRLEN("/proc/") + DECIMAL_STR_MAX(pid_t) + 1 + sizeof(field)); \
  34                         sprintf((char*) _r_, "/proc/"PID_FMT"/" field, _pid_);                       \
  35                 }                                                       \
  36                 _r_;                                                    \
  37         })
  38
  39 int get_process_state(pid_t pid);
  40 int get_process_comm(pid_t pid, char **name);
  41 int get_process_cmdline(pid_t pid, size_t max_length, bool comm_fallback, char **line);
  42 int get_process_exe(pid_t pid, char **name);
  43 #if 0 /// UNNEEDED by elogind
  44 int get_process_uid(pid_t pid, uid_t *uid);
  45 int get_process_gid(pid_t pid, gid_t *gid);
  46 int get_process_capeff(pid_t pid, char **capeff);
  47 int get_process_cwd(pid_t pid, char **cwd);
  48 int get_process_root(pid_t pid, char **root);
  49 int get_process_environ(pid_t pid, char **environ);
  50 int get_process_ppid(pid_t pid, pid_t *ppid);
  51 #endif // 0
  52
  53 int wait_for_terminate(pid_t pid, siginfo_t *status);
  54
  55 typedef enum WaitFlags {
  56         WAIT_LOG_ABNORMAL             = 1 << 0,
  57         WAIT_LOG_NON_ZERO_EXIT_STATUS = 1 << 1,
  58
  59         /* A shortcut for requesting the most complete logging */
  60         WAIT_LOG = WAIT_LOG_ABNORMAL|WAIT_LOG_NON_ZERO_EXIT_STATUS,
  61 } WaitFlags;
  62
  63 int wait_for_terminate_and_check(const char *name, pid_t pid, WaitFlags flags);
  64 int wait_for_terminate_with_timeout(pid_t pid, usec_t timeout);
  65 #if 0 /// UNNEEDED by elogind
  66
  67 void sigkill_wait(pid_t pid);
  68 void sigkill_waitp(pid_t *pid);
  69 #endif // 0
  70 void sigterm_wait(pid_t pid);
  71
  72 int kill_and_sigcont(pid_t pid, int sig);
  73
  74 int rename_process(const char name[]);
  75 int is_kernel_thread(pid_t pid);
  76
  77 int getenv_for_pid(pid_t pid, const char *field, char **_value);
  78
  79 bool pid_is_alive(pid_t pid);
  80 bool pid_is_unwaited(pid_t pid);
  81 #if 0 /// UNNEEDED by elogind
  82 int pid_from_same_root_fs(pid_t pid);
  83 #endif // 0
  84
  85 bool is_main_thread(void);
  86
  87 #if 0 /// UNNEEDED by elogind
  88 _noreturn_ void freeze(void);
  89
  90 bool oom_score_adjust_is_valid(int oa);
  91 #endif // 0
  92
  93 #ifndef PERSONALITY_INVALID
  94 /* personality(7) documents that 0xffffffffUL is used for querying the
  95  * current personality, hence let's use that here as error
  96  * indicator. */
  97 #define PERSONALITY_INVALID 0xffffffffLU
  98 #endif
  99
 100 #if 0 /// UNNEEDED by elogind
 101 unsigned long personality_from_string(const char *p);
 102 const char *personality_to_string(unsigned long);
 103
 104 int safe_personality(unsigned long p);
 105 int opinionated_personality(unsigned long *ret);
 106
 107 int ioprio_class_to_string_alloc(int i, char **s);
 108 int ioprio_class_from_string(const char *s);
 109
 110 const char *sigchld_code_to_string(int i) _const_;
 111 int sigchld_code_from_string(const char *s) _pure_;
 112 #endif // 0
 113
 114 int sched_policy_to_string_alloc(int i, char **s);
 115 #if 0 /// UNNEEDED by elogind
 116 int sched_policy_from_string(const char *s);
 117 #endif // 0
 118
 119 static inline pid_t PTR_TO_PID(const void *p) {
 120         return (pid_t) ((uintptr_t) p);
 121 }
 122
 123 static inline void* PID_TO_PTR(pid_t pid) {
 124         return (void*) ((uintptr_t) pid);
 125 }
 126
 127 void valgrind_summary_hack(void);
 128
 129 int pid_compare_func(const void *a, const void *b);
 130
 131 #if 0 /// UNNEEDED by elogind
 132 static inline bool nice_is_valid(int n) {
 133         return n >= PRIO_MIN && n < PRIO_MAX;
 134 }
 135 #endif // 0
 136
 137 static inline bool sched_policy_is_valid(int i) {
 138         return IN_SET(i, SCHED_OTHER, SCHED_BATCH, SCHED_IDLE, SCHED_FIFO, SCHED_RR);
 139 }
 140
 141 #if 0 /// UNNEEDED by elogind
 142 static inline bool sched_priority_is_valid(int i) {
 143         return i >= 0 && i <= sched_get_priority_max(SCHED_RR);
 144 }
 145
 146 static inline bool ioprio_class_is_valid(int i) {
 147         return IN_SET(i, IOPRIO_CLASS_NONE, IOPRIO_CLASS_RT, IOPRIO_CLASS_BE, IOPRIO_CLASS_IDLE);
 148 }
 149
 150 static inline bool ioprio_priority_is_valid(int i) {
 151         return i >= 0 && i < IOPRIO_BE_NR;
 152 }
 153 #endif // 0
 154
 155 static inline bool pid_is_valid(pid_t p) {
 156         return p > 0;
 157 }
 158
 159 static inline int sched_policy_to_string_alloc_with_check(int n, char **s) {
 160         if (!sched_policy_is_valid(n))
 161                 return -EINVAL;
 162
 163         return sched_policy_to_string_alloc(n, s);
 164 }
 165
 166 #if 0 /// UNNEEDED by elogind
 167 int ioprio_parse_priority(const char *s, int *ret);
 168 #endif // 0
 169
 170 pid_t getpid_cached(void);
 171 void reset_cached_pid(void);
 172
 173 int must_be_root(void);
 174
 175 typedef enum ForkFlags {
 176         FORK_RESET_SIGNALS = 1 << 0,
 177         FORK_CLOSE_ALL_FDS = 1 << 1,
 178         FORK_DEATHSIG      = 1 << 2,
 179         FORK_NULL_STDIO    = 1 << 3,
 180         FORK_REOPEN_LOG    = 1 << 4,
 181         FORK_LOG           = 1 << 5,
 182         FORK_WAIT          = 1 << 6,
 183         FORK_NEW_MOUNTNS   = 1 << 7,
 184         FORK_MOUNTNS_SLAVE = 1 << 8,
 185 } ForkFlags;
 186
 187 int safe_fork_full(const char *name, const int except_fds[], size_t n_except_fds, ForkFlags flags, pid_t *ret_pid);
 188
 189 static inline int safe_fork(const char *name, ForkFlags flags, pid_t *ret_pid) {
 190         return safe_fork_full(name, NULL, 0, flags, ret_pid);
 191 }
 192
 193 int fork_agent(const char *name, const int except[], size_t n_except, pid_t *pid, const char *path, ...);
 194
 195 int set_oom_score_adjust(int value);
 196
 197 #if SIZEOF_PID_T == 4
 198 /* The highest possibly (theoretic) pid_t value on this architecture. */
 199 #define PID_T_MAX ((pid_t) INT32_MAX)
 200 /* The maximum number of concurrent processes Linux allows on this architecture, as well as the highest valid PID value
 201  * the kernel will potentially assign. This reflects a value compiled into the kernel (PID_MAX_LIMIT), and sets the
 202  * upper boundary on what may be written to the /proc/sys/kernel/pid_max sysctl (but do note that the sysctl is off by
 203  * 1, since PID 0 can never exist and there can hence only be one process less than the limit would suggest). Since
 204  * these values are documented in proc(5) we feel quite confident that they are stable enough for the near future at
 205  * least to define them here too. */
 206 #define TASKS_MAX 4194303U
 207 #elif SIZEOF_PID_T == 2
 208 #define PID_T_MAX ((pid_t) INT16_MAX)
 209 #define TASKS_MAX 32767U
 210 #else
 211 #error "Unknown pid_t size"
 212 #endif
 213
 214 assert_cc(TASKS_MAX <= (unsigned long) PID_T_MAX)
 215
 216 /* Like TAKE_PTR() but for child PIDs, resetting them to 0 */
 217 #define TAKE_PID(pid)                           \
 218         ({                                      \
 219                 pid_t _pid_ = (pid);            \
 220                 (pid) = 0;                      \
 221                 _pid_;                          \
 222         })