/
niceSOFT
/
systemd
Обзор
Документация
Войти
/
niceSOFT
/
systemd
Код
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
v261
src/systemctl/systemctl-edit.c
386 строк
15 KB
Zbigniew Jędrzejewski-Szmek
tree-wide: extend verbs functions with extra per-verb data parameter
18 мар 2026, 13:28
18 мар 2026, 13:28
31ddd87
Код
Авторство
О чём код?
/* SPDX-License-Identifier: LGPL-2.1-or-later */ #include <unistd.h> #include "alloc-util.h" #include "bus-util.h" #include "edit-util.h" #include "hashmap.h" #include "label-util.h" #include "pager.h" #include "path-lookup.h" #include "path-util.h" #include "pretty-print.h" #include "string-util.h" #include "strv.h" #include "systemctl.h" #include "systemctl-daemon-reload.h" #include "systemctl-edit.h" #include "systemctl-util.h" #include "terminal-util.h" #include "unit-name.h" int verb_cat(int argc, char *argv[], uintptr_t _data, void *userdata) { _cleanup_hashmap_free_ Hashmap *cached_id_map = NULL, *cached_name_map = NULL; _cleanup_(lookup_paths_done) LookupPaths lp = {}; _cleanup_strv_free_ char **names = NULL; sd_bus *bus = NULL; bool first = true; int r, rc = 0; /* Include all units by default — i.e. continue as if the --all option was used */ if (strv_isempty(arg_states)) arg_all = true; if (arg_transport != BUS_TRANSPORT_LOCAL) return log_error_errno(SYNTHETIC_ERRNO(EINVAL), "Cannot remotely cat units."); r = lookup_paths_init_or_warn(&lp, arg_runtime_scope, 0, arg_root); if (r < 0) return r; if (install_client_side() == INSTALL_CLIENT_SIDE_NO) { r = acquire_bus(BUS_MANAGER, &bus); if (r < 0) return r; r = expand_unit_names(bus, strv_skip(argv, 1), NULL, &names, NULL); if (r < 0) return log_error_errno(r, "Failed to expand names: %m"); r = maybe_extend_with_unit_dependencies(bus, &names); if (r < 0) return r; } else { /* In client-side mode (--global, --root, etc.), just mangle names without bus interaction */ r = mangle_names("to cat", strv_skip(argv, 1), &names); if (r < 0) return r; } pager_open(arg_pager_flags); STRV_FOREACH(name, names) { _cleanup_free_ char *fragment_path = NULL; _cleanup_strv_free_ char **dropin_paths = NULL; r = unit_find_paths(bus, *name, &lp, /* force_client_side= */ !bus, &cached_id_map, &cached_name_map, &fragment_path, &dropin_paths); if (r == -ERFKILL) { printf("%s# Unit %s is masked%s.\n", ansi_highlight_magenta(), *name, ansi_normal()); continue; } if (r == -EKEYREJECTED) { printf("%s# Unit %s could not be loaded.%s\n", ansi_highlight_magenta(), *name, ansi_normal()); continue; } if (r < 0) return r; if (r == 0) { /* Skip units which have no on-disk counterpart, but propagate the error to the * user (if --force is set, eat the error, just like unit_find_paths()) */ if (!arg_force) rc = -ENOENT; continue; } if (first) first = false; else puts(""); if (bus && need_daemon_reload(bus, *name) > 0) /* ignore errors (<0), this is informational output */ fprintf(stderr, "%s# Warning: %s changed on disk, the version systemd has loaded is outdated.\n" "%s# This output shows the current version of the unit's original fragment and drop-in files.\n" "%s# If fragments or drop-ins were added or removed, they are not properly reflected in this output.\n" "%s# Run 'systemctl%s daemon-reload' to reload units.%s\n", ansi_highlight_red(), *name, ansi_highlight_red(), ansi_highlight_red(), ansi_highlight_red(), arg_runtime_scope == RUNTIME_SCOPE_SYSTEM ? "" : " --user", ansi_normal()); r = cat_files(fragment_path, dropin_paths, /* flags= */ CAT_FORMAT_HAS_SECTIONS); if (r < 0) return r; } return rc; } static int get_file_to_edit( const LookupPaths *lp, const char *name, char **ret_path) { _cleanup_free_ char *path = NULL; assert(lp); assert(name); assert(ret_path); path = path_join(lp->persistent_config, name); if (!path) return log_oom(); if (arg_runtime) { _cleanup_free_ char *run = NULL; run = path_join(lp->runtime_config, name); if (!run) return log_oom(); if (access(path, F_OK) >= 0) return log_error_errno(SYNTHETIC_ERRNO(EEXIST), "Refusing to create \"%s\" because it would be overridden by \"%s\" anyway.", run, path); *ret_path = TAKE_PTR(run); } else *ret_path = TAKE_PTR(path); return 0; } static int unit_file_create_new( EditFileContext *context, const LookupPaths *lp, const char *unit_name, const char *suffix, char * const *original_unit_paths) { _cleanup_free_ char *unit = NULL, *new_path = NULL; int r; assert(context); assert(lp); assert(unit_name); unit = strjoin(unit_name, suffix); if (!unit) return log_oom(); r = get_file_to_edit(lp, unit, &new_path); if (r < 0) return r; return edit_files_add(context, new_path, NULL, original_unit_paths); } static int unit_file_create_copy( EditFileContext *context, const LookupPaths *lp, const char *unit_name, const char *fragment_path) { _cleanup_free_ char *new_path = NULL; int r; assert(context); assert(lp); assert(fragment_path); assert(unit_name); r = get_file_to_edit(lp, unit_name, &new_path); if (r < 0) return r; if (!path_equal(fragment_path, new_path) && access(new_path, F_OK) >= 0) { char response; r = ask_char(&response, "yn", "\"%s\" already exists. Overwrite with \"%s\"? [(y)es, (n)o] ", new_path, fragment_path); if (r < 0) return r; if (response != 'y') return log_warning_errno(SYNTHETIC_ERRNO(EKEYREJECTED), "%s skipped.", unit_name); } return edit_files_add(context, new_path, fragment_path, NULL); } static int find_paths_to_edit( sd_bus *bus, EditFileContext *context, char **names) { _cleanup_hashmap_free_ Hashmap *cached_id_map = NULL, *cached_name_map = NULL; _cleanup_(lookup_paths_done) LookupPaths lp = {}; _cleanup_free_ char *drop_in_alloc = NULL, *suffix = NULL; const char *drop_in; int r; assert(context); assert(names); if (isempty(arg_drop_in)) drop_in = "override.conf"; else if (!endswith(arg_drop_in, ".conf")) { drop_in_alloc = strjoin(arg_drop_in, ".conf"); if (!drop_in_alloc) return log_oom(); drop_in = drop_in_alloc; } else drop_in = arg_drop_in; if (!filename_is_valid(drop_in)) return log_error_errno(SYNTHETIC_ERRNO(EINVAL), "Invalid drop-in file name '%s'.", drop_in); suffix = strjoin(".d/", drop_in); if (!suffix) return log_oom(); r = lookup_paths_init(&lp, arg_runtime_scope, 0, arg_root); if (r < 0) return r; STRV_FOREACH(name, names) { _cleanup_free_ char *path = NULL; _cleanup_strv_free_ char **unit_paths = NULL; r = unit_find_paths(bus, *name, &lp, /* force_client_side= */ !bus, &cached_id_map, &cached_name_map, &path, &unit_paths); if (r == -EKEYREJECTED && bus) { /* If loading of the unit failed server side complete, then the server won't tell us * the unit file path. In that case, find the file client side. */ log_debug_errno(r, "Unit '%s' was not loaded correctly, retrying client-side.", *name); r = unit_find_paths(bus, *name, &lp, /* force_client_side= */ true, &cached_id_map, &cached_name_map, &path, &unit_paths); } if (r == -ERFKILL) return log_error_errno(r, "Unit '%s' masked, cannot edit.", *name); if (r < 0) return r; /* Already logged by unit_find_paths() */ if (!path) { if (!arg_force) return log_info_errno(SYNTHETIC_ERRNO(ENOENT), "Run 'systemctl edit%s --force --full %s' to create a new unit.", arg_runtime_scope == RUNTIME_SCOPE_GLOBAL ? " --global" : arg_runtime_scope == RUNTIME_SCOPE_USER ? " --user" : "", *name); /* Create a new unit from scratch */ r = unit_file_create_new( context, &lp, *name, arg_full ? NULL : suffix, NULL); } else { _cleanup_free_ char *unit_name = NULL; r = path_extract_filename(path, &unit_name); if (r < 0) return log_error_errno(r, "Failed to extract unit name from path '%s': %m", path); /* We follow unit aliases, but we need to propagate the instance */ if (unit_name_is_valid(*name, UNIT_NAME_INSTANCE) && unit_name_is_valid(unit_name, UNIT_NAME_TEMPLATE)) { _cleanup_free_ char *instance = NULL, *tmp_name = NULL; r = unit_name_to_instance(*name, &instance); if (r < 0) return r; r = unit_name_replace_instance(unit_name, instance, &tmp_name); if (r < 0) return r; free_and_replace(unit_name, tmp_name); } if (arg_full) r = unit_file_create_copy( context, &lp, unit_name, path); else { r = strv_prepend(&unit_paths, path); if (r < 0) return log_oom(); r = unit_file_create_new( context, &lp, unit_name, suffix, unit_paths); } } if (r < 0) return r; } return 0; } int verb_edit(int argc, char *argv[], uintptr_t _data, void *userdata) { _cleanup_(edit_file_context_done) EditFileContext context = { .marker_start = DROPIN_MARKER_START, .marker_end = DROPIN_MARKER_END, .remove_parent = !arg_full, .overwrite_with_origin = true, .read_from_stdin = arg_stdin, }; _cleanup_strv_free_ char **names = NULL; sd_bus *bus = NULL; int r; if (!on_tty() && !arg_stdin) return log_error_errno(SYNTHETIC_ERRNO(EINVAL), "Cannot edit units interactively if not on a tty."); if (arg_transport != BUS_TRANSPORT_LOCAL) return log_error_errno(SYNTHETIC_ERRNO(EINVAL), "Cannot edit units remotely."); r = mac_init(); if (r < 0) return r; if (install_client_side() == INSTALL_CLIENT_SIDE_NO) { r = acquire_bus(BUS_MANAGER, &bus); if (r < 0) return r; r = expand_unit_names(bus, strv_skip(argv, 1), NULL, &names, NULL); if (r < 0) return log_error_errno(r, "Failed to expand names: %m"); } else { /* In client-side mode (--global, --root, etc.), just mangle names without bus interaction */ r = mangle_names("to edit", strv_skip(argv, 1), &names); if (r < 0) return r; } if (strv_isempty(names)) return log_error_errno(SYNTHETIC_ERRNO(ENOENT), "No units matched the specified patterns."); if (arg_stdin && arg_full && strv_length(names) != 1) return log_error_errno(SYNTHETIC_ERRNO(EINVAL), "With 'edit --stdin --full', exactly one unit for editing must be specified."); r = find_paths_to_edit(bus, &context, names); if (r < 0) return r; r = do_edit_files_and_install(&context); if (r < 0) return r; if (!arg_no_reload && install_client_side() == INSTALL_CLIENT_SIDE_NO) { r = daemon_reload(ACTION_RELOAD, /* graceful= */ false); if (r < 0) return r; } return 0; }