| /* SPDX-License-Identifier: LGPL-2.1-or-later */ |
| |
| #include <getopt.h> |
| #include <stdio.h> |
| #include <stdlib.h> |
| |
| #include "alloc-util.h" |
| #include "log.h" |
| #include "main-func.h" |
| #include "pretty-print.h" |
| #include "string-util.h" |
| #include "strv.h" |
| #include "unit-name.h" |
| |
| static enum { |
| ACTION_ESCAPE, |
| ACTION_UNESCAPE, |
| ACTION_MANGLE |
| } arg_action = ACTION_ESCAPE; |
| static const char *arg_suffix = NULL; |
| static const char *arg_template = NULL; |
| static bool arg_path = false; |
| static bool arg_instance = false; |
| |
| static int help(void) { |
| _cleanup_free_ char *link = NULL; |
| int r; |
| |
| r = terminal_urlify_man("systemd-escape", "1", &link); |
| if (r < 0) |
| return log_oom(); |
| |
| printf("%s [OPTIONS...] [NAME...]\n\n" |
| "Escape strings for usage in systemd unit names.\n\n" |
| " -h --help Show this help\n" |
| " --version Show package version\n" |
| " --suffix=SUFFIX Unit suffix to append to escaped strings\n" |
| " --template=TEMPLATE Insert strings as instance into template\n" |
| " --instance With --unescape, show just the instance part\n" |
| " -u --unescape Unescape strings\n" |
| " -m --mangle Mangle strings\n" |
| " -p --path When escaping/unescaping assume the string is a path\n" |
| "\nSee the %s for details.\n", |
| program_invocation_short_name, |
| link); |
| |
| return 0; |
| } |
| |
| static int parse_argv(int argc, char *argv[]) { |
| |
| enum { |
| ARG_VERSION = 0x100, |
| ARG_SUFFIX, |
| ARG_TEMPLATE |
| }; |
| |
| static const struct option options[] = { |
| { "help", no_argument, NULL, 'h' }, |
| { "version", no_argument, NULL, ARG_VERSION }, |
| { "suffix", required_argument, NULL, ARG_SUFFIX }, |
| { "template", required_argument, NULL, ARG_TEMPLATE }, |
| { "unescape", no_argument, NULL, 'u' }, |
| { "mangle", no_argument, NULL, 'm' }, |
| { "path", no_argument, NULL, 'p' }, |
| { "instance", no_argument, NULL, 'i' }, |
| {} |
| }; |
| |
| int c; |
| |
| assert(argc >= 0); |
| assert(argv); |
| |
| while ((c = getopt_long(argc, argv, "hump", options, NULL)) >= 0) |
| |
| switch (c) { |
| |
| case 'h': |
| return help(); |
| |
| case ARG_VERSION: |
| return version(); |
| |
| case ARG_SUFFIX: { |
| UnitType t = unit_type_from_string(optarg); |
| if (t < 0) |
| return log_error_errno(t, "Invalid unit suffix type \"%s\".", optarg); |
| |
| arg_suffix = optarg; |
| break; |
| } |
| |
| case ARG_TEMPLATE: |
| if (!unit_name_is_valid(optarg, UNIT_NAME_TEMPLATE)) |
| return log_error_errno(SYNTHETIC_ERRNO(EINVAL), |
| "Template name %s is not valid.", optarg); |
| |
| arg_template = optarg; |
| break; |
| |
| case 'u': |
| arg_action = ACTION_UNESCAPE; |
| break; |
| |
| case 'm': |
| arg_action = ACTION_MANGLE; |
| break; |
| |
| case 'p': |
| arg_path = true; |
| break; |
| |
| case 'i': |
| arg_instance = true; |
| break; |
| |
| case '?': |
| return -EINVAL; |
| |
| default: |
| assert_not_reached("Unhandled option"); |
| } |
| |
| if (optind >= argc) |
| return log_error_errno(SYNTHETIC_ERRNO(EINVAL), |
| "Not enough arguments."); |
| |
| if (arg_template && arg_suffix) |
| return log_error_errno(SYNTHETIC_ERRNO(EINVAL), |
| "--suffix= and --template= may not be combined."); |
| |
| if ((arg_template || arg_suffix) && arg_action == ACTION_MANGLE) |
| return log_error_errno(SYNTHETIC_ERRNO(EINVAL), |
| "--suffix= and --template= are not compatible with --mangle."); |
| |
| if (arg_suffix && arg_action == ACTION_UNESCAPE) |
| return log_error_errno(SYNTHETIC_ERRNO(EINVAL), |
| "--suffix is not compatible with --unescape."); |
| |
| if (arg_path && !IN_SET(arg_action, ACTION_ESCAPE, ACTION_UNESCAPE)) |
| return log_error_errno(SYNTHETIC_ERRNO(EINVAL), |
| "--path may not be combined with --mangle."); |
| |
| if (arg_instance && arg_action != ACTION_UNESCAPE) |
| return log_error_errno(SYNTHETIC_ERRNO(EINVAL), |
| "--instance must be used in conjunction with --unescape."); |
| |
| if (arg_instance && arg_template) |
| return log_error_errno(SYNTHETIC_ERRNO(EINVAL), |
| "--instance may not be combined with --template."); |
| |
| return 1; |
| } |
| |
| static int run(int argc, char *argv[]) { |
| char **i; |
| int r; |
| |
| log_setup(); |
| |
| r = parse_argv(argc, argv); |
| if (r <= 0) |
| return r; |
| |
| STRV_FOREACH(i, argv + optind) { |
| _cleanup_free_ char *e = NULL; |
| |
| switch (arg_action) { |
| |
| case ACTION_ESCAPE: |
| if (arg_path) { |
| r = unit_name_path_escape(*i, &e); |
| if (r < 0) |
| return log_error_errno(r, "Failed to escape string: %m"); |
| } else { |
| e = unit_name_escape(*i); |
| if (!e) |
| return log_oom(); |
| } |
| |
| if (arg_template) { |
| char *x; |
| |
| r = unit_name_replace_instance(arg_template, e, &x); |
| if (r < 0) |
| return log_error_errno(r, "Failed to replace instance: %m"); |
| |
| free_and_replace(e, x); |
| } else if (arg_suffix) { |
| if (!strextend(&e, ".", arg_suffix)) |
| return log_oom(); |
| } |
| |
| break; |
| |
| case ACTION_UNESCAPE: { |
| _cleanup_free_ char *name = NULL; |
| |
| if (arg_template || arg_instance) { |
| _cleanup_free_ char *template = NULL; |
| |
| r = unit_name_to_instance(*i, &name); |
| if (r < 0) |
| return log_error_errno(r, "Failed to extract instance: %m"); |
| if (isempty(name)) |
| return log_error_errno(SYNTHETIC_ERRNO(EINVAL), |
| "Unit %s is missing the instance name.", *i); |
| |
| r = unit_name_template(*i, &template); |
| if (r < 0) |
| return log_error_errno(r, "Failed to extract template: %m"); |
| if (arg_template && !streq(arg_template, template)) |
| return log_error_errno(SYNTHETIC_ERRNO(EINVAL), |
| "Unit %s template %s does not match specified template %s.", |
| *i, template, arg_template); |
| } else { |
| name = strdup(*i); |
| if (!name) |
| return log_oom(); |
| } |
| |
| if (arg_path) |
| r = unit_name_path_unescape(name, &e); |
| else |
| r = unit_name_unescape(name, &e); |
| if (r < 0) |
| return log_error_errno(r, "Failed to unescape string: %m"); |
| |
| break; |
| } |
| |
| case ACTION_MANGLE: |
| r = unit_name_mangle(*i, 0, &e); |
| if (r < 0) |
| return log_error_errno(r, "Failed to mangle name: %m"); |
| |
| break; |
| } |
| |
| if (i != argv + optind) |
| fputc(' ', stdout); |
| |
| fputs(e, stdout); |
| } |
| |
| fputc('\n', stdout); |
| |
| return 0; |
| } |
| |
| DEFINE_MAIN_FUNCTION(run); |
