1 /* SPDX-License-Identifier: LGPL-2.1+ */
7 #include "alloc-util.h"
8 #include "extract-word.h"
11 #include "parse-util.h"
12 #include "proc-cmdline.h"
13 #include "process-util.h"
15 #include "string-util.h"
19 int proc_cmdline(char **ret
) {
23 /* For testing purposes it is sometimes useful to be able to override what we consider /proc/cmdline to be */
24 e
= secure_getenv("SYSTEMD_PROC_CMDLINE");
36 if (detect_container() > 0)
37 return get_process_cmdline(1, SIZE_MAX
, false, ret
);
39 return read_one_line_file("/proc/cmdline", ret
);
42 static int proc_cmdline_extract_first(const char **p
, char **ret_word
, ProcCmdlineFlags flags
) {
47 _cleanup_free_
char *word
= NULL
;
50 r
= extract_first_word(&q
, &word
, NULL
, EXTRACT_QUOTES
|EXTRACT_RELAX
);
56 /* Filter out arguments that are intended only for the initrd */
57 c
= startswith(word
, "rd.");
62 if (FLAGS_SET(flags
, PROC_CMDLINE_STRIP_RD_PREFIX
)) {
63 r
= free_and_strdup(&word
, c
);
68 } else if (FLAGS_SET(flags
, PROC_CMDLINE_RD_STRICT
) && in_initrd())
69 continue; /* And optionally filter out arguments that are intended only for the host */
72 *ret_word
= TAKE_PTR(word
);
81 int proc_cmdline_parse_given(const char *line
, proc_cmdline_parse_t parse_item
, void *data
, ProcCmdlineFlags flags
) {
87 /* The PROC_CMDLINE_VALUE_OPTIONAL flag doesn't really make sense for proc_cmdline_parse(), let's make this
89 assert(!FLAGS_SET(flags
, PROC_CMDLINE_VALUE_OPTIONAL
));
93 _cleanup_free_
char *word
= NULL
;
96 r
= proc_cmdline_extract_first(&p
, &word
, flags
);
102 value
= strchr(word
, '=');
106 r
= parse_item(word
, value
, data
);
114 int proc_cmdline_parse(proc_cmdline_parse_t parse_item
, void *data
, ProcCmdlineFlags flags
) {
115 _cleanup_free_
char *line
= NULL
;
120 r
= proc_cmdline(&line
);
124 return proc_cmdline_parse_given(line
, parse_item
, data
, flags
);
127 static bool relaxed_equal_char(char a
, char b
) {
129 (a
== '_' && b
== '-') ||
130 (a
== '-' && b
== '_');
133 char *proc_cmdline_key_startswith(const char *s
, const char *prefix
) {
137 /* Much like startswith(), but considers "-" and "_" the same */
139 for (; *prefix
!= 0; s
++, prefix
++)
140 if (!relaxed_equal_char(*s
, *prefix
))
146 bool proc_cmdline_key_streq(const char *x
, const char *y
) {
150 /* Much like streq(), but considers "-" and "_" the same */
152 for (; *x
!= 0 || *y
!= 0; x
++, y
++)
153 if (!relaxed_equal_char(*x
, *y
))
159 int proc_cmdline_get_key(const char *key
, ProcCmdlineFlags flags
, char **ret_value
) {
160 _cleanup_free_
char *line
= NULL
, *ret
= NULL
;
165 /* Looks for a specific key on the kernel command line. Supports three modes:
167 * a) The "ret_value" parameter is used. In this case a parameter beginning with the "key" string followed by
168 * "=" is searched for, and the value following it is returned in "ret_value".
170 * b) as above, but the PROC_CMDLINE_VALUE_OPTIONAL flag is set. In this case if the key is found as a separate
171 * word (i.e. not followed by "=" but instead by whitespace or the end of the command line), then this is
172 * also accepted, and "value" is returned as NULL.
174 * c) The "ret_value" parameter is NULL. In this case a search for the exact "key" parameter is performed.
176 * In all three cases, > 0 is returned if the key is found, 0 if not. */
181 if (FLAGS_SET(flags
, PROC_CMDLINE_VALUE_OPTIONAL
) && !ret_value
)
184 r
= proc_cmdline(&line
);
190 _cleanup_free_
char *word
= NULL
;
192 r
= proc_cmdline_extract_first(&p
, &word
, flags
);
201 e
= proc_cmdline_key_startswith(word
, key
);
206 r
= free_and_strdup(&ret
, e
+1);
212 } else if (*e
== 0 && FLAGS_SET(flags
, PROC_CMDLINE_VALUE_OPTIONAL
))
216 if (streq(word
, key
)) {
218 break; /* we found what we were looking for */
224 *ret_value
= TAKE_PTR(ret
);
229 int proc_cmdline_get_bool(const char *key
, bool *ret
) {
230 _cleanup_free_
char *v
= NULL
;
235 r
= proc_cmdline_get_key(key
, PROC_CMDLINE_VALUE_OPTIONAL
, &v
);
243 if (v
) { /* parameter passed */
244 r
= parse_boolean(v
);
248 } else /* no parameter passed */
254 int proc_cmdline_get_key_many_internal(ProcCmdlineFlags flags
, ...) {
255 _cleanup_free_
char *line
= NULL
;
260 /* The PROC_CMDLINE_VALUE_OPTIONAL flag doesn't really make sense for proc_cmdline_get_key_many(), let's make
262 assert(!FLAGS_SET(flags
, PROC_CMDLINE_VALUE_OPTIONAL
));
264 /* This call may clobber arguments on failure! */
266 r
= proc_cmdline(&line
);
272 _cleanup_free_
char *word
= NULL
;
274 r
= proc_cmdline_extract_first(&p
, &word
, flags
);
286 k
= va_arg(ap
, const char*);
290 assert_se(v
= va_arg(ap
, char**));
292 e
= proc_cmdline_key_startswith(word
, k
);
293 if (e
&& *e
== '=') {
294 r
= free_and_strdup(v
, e
+ 1);
310 int shall_restore_state(void) {
314 r
= proc_cmdline_get_bool("systemd.restore_state", &ret
);
318 return r
> 0 ? ret
: true;
321 static const char * const rlmap
[] = {
322 "emergency", SPECIAL_EMERGENCY_TARGET
,
323 "-b", SPECIAL_EMERGENCY_TARGET
,
324 "rescue", SPECIAL_RESCUE_TARGET
,
325 "single", SPECIAL_RESCUE_TARGET
,
326 "-s", SPECIAL_RESCUE_TARGET
,
327 "s", SPECIAL_RESCUE_TARGET
,
328 "S", SPECIAL_RESCUE_TARGET
,
329 "1", SPECIAL_RESCUE_TARGET
,
330 "2", SPECIAL_MULTI_USER_TARGET
,
331 "3", SPECIAL_MULTI_USER_TARGET
,
332 "4", SPECIAL_MULTI_USER_TARGET
,
333 "5", SPECIAL_GRAPHICAL_TARGET
,
337 static const char * const rlmap_initrd
[] = {
338 "emergency", SPECIAL_EMERGENCY_TARGET
,
339 "rescue", SPECIAL_RESCUE_TARGET
,
343 const char* runlevel_to_target(const char *word
) {
344 const char * const *rlmap_ptr
;
351 word
= startswith(word
, "rd.");
356 rlmap_ptr
= in_initrd() ? rlmap_initrd
: rlmap
;
358 for (i
= 0; rlmap_ptr
[i
]; i
+= 2)
359 if (streq(word
, rlmap_ptr
[i
]))
360 return rlmap_ptr
[i
+1];