mirror of
https://github.com/systemd/systemd.git
synced 2024-11-01 17:51:22 +03:00
baaa35ad70
Ideally, coccinelle would strip unnecessary braces too. But I do not see any option in coccinelle for this, so instead, I edited the patch text using search&replace to remove the braces. Unfortunately this is not fully automatic, in particular it didn't deal well with if-else-if-else blocks and ifdefs, so there is an increased likelikehood be some bugs in such spots. I also removed part of the patch that coccinelle generated for udev, where we returns -1 for failure. This should be fixed independently.
259 lines
9.1 KiB
C
259 lines
9.1 KiB
C
/* SPDX-License-Identifier: LGPL-2.1+ */
|
|
|
|
#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:
|
|
|
|
if (unit_type_from_string(optarg) < 0)
|
|
return log_error_errno(SYNTHETIC_ERRNO(EINVAL),
|
|
"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_parse_environment();
|
|
log_open();
|
|
|
|
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) {
|
|
char *x;
|
|
|
|
x = strjoin(e, ".", arg_suffix);
|
|
if (!x)
|
|
return log_oom();
|
|
|
|
free_and_replace(e, x);
|
|
}
|
|
|
|
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("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("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);
|