262 lines
8.4 KiB
C
262 lines
8.4 KiB
C
/* SPDX-License-Identifier: LGPL-2.1-or-later */
|
|
|
|
#include <getopt.h>
|
|
#include <stdio.h>
|
|
|
|
#include "alloc-util.h"
|
|
#include "gpt.h"
|
|
#include "id128-print.h"
|
|
#include "main-func.h"
|
|
#include "pretty-print.h"
|
|
#include "strv.h"
|
|
#include "format-table.h"
|
|
#include "terminal-util.h"
|
|
#include "util.h"
|
|
#include "verbs.h"
|
|
|
|
static Id128PrettyPrintMode arg_mode = ID128_PRINT_ID128;
|
|
static sd_id128_t arg_app = {};
|
|
|
|
static int verb_new(int argc, char **argv, void *userdata) {
|
|
return id128_print_new(arg_mode);
|
|
}
|
|
|
|
static int verb_machine_id(int argc, char **argv, void *userdata) {
|
|
sd_id128_t id;
|
|
int r;
|
|
|
|
if (sd_id128_is_null(arg_app))
|
|
r = sd_id128_get_machine(&id);
|
|
else
|
|
r = sd_id128_get_machine_app_specific(arg_app, &id);
|
|
if (r < 0)
|
|
return log_error_errno(r, "Failed to get %smachine-ID: %m",
|
|
sd_id128_is_null(arg_app) ? "" : "app-specific ");
|
|
|
|
return id128_pretty_print(id, arg_mode);
|
|
}
|
|
|
|
static int verb_boot_id(int argc, char **argv, void *userdata) {
|
|
sd_id128_t id;
|
|
int r;
|
|
|
|
if (sd_id128_is_null(arg_app))
|
|
r = sd_id128_get_boot(&id);
|
|
else
|
|
r = sd_id128_get_boot_app_specific(arg_app, &id);
|
|
if (r < 0)
|
|
return log_error_errno(r, "Failed to get %sboot-ID: %m",
|
|
sd_id128_is_null(arg_app) ? "" : "app-specific ");
|
|
|
|
return id128_pretty_print(id, arg_mode);
|
|
}
|
|
|
|
static int verb_invocation_id(int argc, char **argv, void *userdata) {
|
|
sd_id128_t id;
|
|
int r;
|
|
|
|
if (!sd_id128_is_null(arg_app))
|
|
return log_error_errno(SYNTHETIC_ERRNO(EINVAL),
|
|
"Verb \"invocation-id\" cannot be combined with --app-specific=.");
|
|
|
|
r = sd_id128_get_invocation(&id);
|
|
if (r < 0)
|
|
return log_error_errno(r, "Failed to get invocation-ID: %m");
|
|
|
|
return id128_pretty_print(id, arg_mode);
|
|
}
|
|
|
|
static int show_one(Table **table, const char *name, sd_id128_t uuid, bool first) {
|
|
int r;
|
|
|
|
if (arg_mode == ID128_PRINT_PRETTY) {
|
|
_cleanup_free_ char *id = NULL;
|
|
|
|
id = strreplace(name, "-", "_");
|
|
if (!id)
|
|
return log_oom();
|
|
|
|
ascii_strupper(id);
|
|
|
|
r = id128_pretty_print_sample(id, uuid);
|
|
if (r < 0)
|
|
return r;
|
|
if (!first)
|
|
puts("");
|
|
return 0;
|
|
|
|
} else {
|
|
if (!*table) {
|
|
*table = table_new("name", "id");
|
|
if (!*table)
|
|
return log_oom();
|
|
table_set_width(*table, 0);
|
|
}
|
|
|
|
return table_add_many(*table,
|
|
TABLE_STRING, name,
|
|
arg_mode == ID128_PRINT_ID128 ? TABLE_ID128 : TABLE_UUID,
|
|
uuid);
|
|
}
|
|
}
|
|
|
|
static int verb_show(int argc, char **argv, void *userdata) {
|
|
_cleanup_(table_unrefp) Table *table = NULL;
|
|
char **p;
|
|
int r;
|
|
|
|
argv = strv_skip(argv, 1);
|
|
if (strv_isempty(argv))
|
|
for (const GptPartitionType *e = gpt_partition_type_table; e->name; e++) {
|
|
r = show_one(&table, e->name, e->uuid, e == gpt_partition_type_table);
|
|
if (r < 0)
|
|
return r;
|
|
}
|
|
else
|
|
STRV_FOREACH(p, argv) {
|
|
sd_id128_t uuid;
|
|
bool have_uuid;
|
|
const char *id;
|
|
|
|
/* Check if the argument is an actual UUID first */
|
|
have_uuid = sd_id128_from_string(*p, &uuid) >= 0;
|
|
|
|
if (have_uuid)
|
|
id = gpt_partition_type_uuid_to_string(uuid) ?: "XYZ";
|
|
else {
|
|
r = gpt_partition_type_uuid_from_string(*p, &uuid);
|
|
if (r < 0)
|
|
return log_error_errno(r, "Unknown identifier \"%s\".", *p);
|
|
|
|
id = *p;
|
|
}
|
|
|
|
r = show_one(&table, id, uuid, p == argv);
|
|
if (r < 0)
|
|
return r;
|
|
}
|
|
|
|
if (table) {
|
|
r = table_print(table, NULL);
|
|
if (r < 0)
|
|
return table_log_print_error(r);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int help(void) {
|
|
_cleanup_free_ char *link = NULL;
|
|
int r;
|
|
|
|
r = terminal_urlify_man("systemd-id128", "1", &link);
|
|
if (r < 0)
|
|
return log_oom();
|
|
|
|
printf("%s [OPTIONS...] COMMAND\n\n"
|
|
"%sGenerate and print 128bit identifiers.%s\n"
|
|
"\nCommands:\n"
|
|
" new Generate a new ID\n"
|
|
" machine-id Print the ID of current machine\n"
|
|
" boot-id Print the ID of current boot\n"
|
|
" invocation-id Print the ID of current invocation\n"
|
|
" show [NAME] Print one or more well-known IDs\n"
|
|
" help Show this help\n"
|
|
"\nOptions:\n"
|
|
" -h --help Show this help\n"
|
|
" -p --pretty Generate samples of program code\n"
|
|
" -a --app-specific=ID Generate app-specific IDs\n"
|
|
" -u --uuid Output in UUID format\n"
|
|
"\nSee the %s for details.\n"
|
|
, program_invocation_short_name
|
|
, ansi_highlight(), ansi_normal()
|
|
, link
|
|
);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int verb_help(int argc, char **argv, void *userdata) {
|
|
return help();
|
|
}
|
|
|
|
static int parse_argv(int argc, char *argv[]) {
|
|
enum {
|
|
ARG_VERSION = 0x100,
|
|
};
|
|
|
|
static const struct option options[] = {
|
|
{ "help", no_argument, NULL, 'h' },
|
|
{ "version", no_argument, NULL, ARG_VERSION },
|
|
{ "pretty", no_argument, NULL, 'p' },
|
|
{ "app-specific", required_argument, NULL, 'a' },
|
|
{ "uuid", no_argument, NULL, 'u' },
|
|
{},
|
|
};
|
|
|
|
int c, r;
|
|
|
|
assert(argc >= 0);
|
|
assert(argv);
|
|
|
|
while ((c = getopt_long(argc, argv, "hpa:u", options, NULL)) >= 0)
|
|
switch (c) {
|
|
|
|
case 'h':
|
|
return help();
|
|
|
|
case ARG_VERSION:
|
|
return version();
|
|
|
|
case 'p':
|
|
arg_mode = ID128_PRINT_PRETTY;
|
|
break;
|
|
|
|
case 'a':
|
|
r = sd_id128_from_string(optarg, &arg_app);
|
|
if (r < 0)
|
|
return log_error_errno(r, "Failed to parse \"%s\" as application-ID: %m", optarg);
|
|
break;
|
|
|
|
case 'u':
|
|
arg_mode = ID128_PRINT_UUID;
|
|
break;
|
|
|
|
case '?':
|
|
return -EINVAL;
|
|
|
|
default:
|
|
assert_not_reached("Unhandled option");
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
static int id128_main(int argc, char *argv[]) {
|
|
static const Verb verbs[] = {
|
|
{ "new", VERB_ANY, 1, 0, verb_new },
|
|
{ "machine-id", VERB_ANY, 1, 0, verb_machine_id },
|
|
{ "boot-id", VERB_ANY, 1, 0, verb_boot_id },
|
|
{ "invocation-id", VERB_ANY, 1, 0, verb_invocation_id },
|
|
{ "show", VERB_ANY, VERB_ANY, 0, verb_show },
|
|
{ "help", VERB_ANY, VERB_ANY, 0, verb_help },
|
|
{}
|
|
};
|
|
|
|
return dispatch_verb(argc, argv, verbs, NULL);
|
|
}
|
|
|
|
static int run(int argc, char *argv[]) {
|
|
int r;
|
|
|
|
log_setup_cli();
|
|
|
|
r = parse_argv(argc, argv);
|
|
if (r <= 0)
|
|
return r;
|
|
|
|
return id128_main(argc, argv);
|
|
}
|
|
|
|
DEFINE_MAIN_FUNCTION(run);
|