/* SPDX-License-Identifier: LGPL-2.1+ */ #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);