#include <getopt.h>

#include <locale.h>

#include <stdbool.h>

#include <stdlib.h>

#include <string.h>

#include "sd-bus.h"

#include "sd-id128.h"

#include "architecture.h"

#include "bus-error.h"

#include "bus-util.h"

#include "hostname-util.h"

#include "spawn-polkit-agent.h"

#include "util.h"

static bool arg_ask_password = true;

static BusTransport arg_transport = BUS_TRANSPORT_LOCAL;

static char *arg_host = NULL;

static bool arg_transient = false;

static bool arg_pretty = false;

static bool arg_static = false;

static void polkit_agent_open_if_enabled(void) {

/* Open the polkit agent as a child process if necessary */

if (!arg_ask_password)

return;

if (arg_transport != BUS_TRANSPORT_LOCAL)

return;

polkit_agent_open();

}

typedef struct StatusInfo {

char *hostname;

char *static_hostname;

char *pretty_hostname;

char *icon_name;

char *chassis;

char *deployment;

char *location;

char *kernel_name;

char *kernel_release;

char *os_pretty_name;

char *os_cpe_name;

char *virtualization;

char *architecture;

} StatusInfo;

static void print_status_info(StatusInfo *i) {

sd_id128_t mid = {}, bid = {};

int r;

assert(i);

printf(" Static hostname: %s\n", strna(i->static_hostname));

if (!isempty(i->pretty_hostname) &&

!streq_ptr(i->pretty_hostname, i->static_hostname))

printf(" Pretty hostname: %s\n", i->pretty_hostname);

if (!isempty(i->hostname) &&

!streq_ptr(i->hostname, i->static_hostname))

printf("Transient hostname: %s\n", i->hostname);

if (!isempty(i->icon_name))

printf(" Icon name: %s\n",

strna(i->icon_name));

if (!isempty(i->chassis))

printf(" Chassis: %s\n",

strna(i->chassis));

if (!isempty(i->deployment))

printf(" Deployment: %s\n", i->deployment);

if (!isempty(i->location))

printf(" Location: %s\n", i->location);

r = sd_id128_get_machine(&mid);

if (r >= 0)

printf(" Machine ID: " SD_ID128_FORMAT_STR "\n", SD_ID128_FORMAT_VAL(mid));

r = sd_id128_get_boot(&bid);

if (r >= 0)

printf(" Boot ID: " SD_ID128_FORMAT_STR "\n", SD_ID128_FORMAT_VAL(bid));

if (!isempty(i->virtualization))

printf(" Virtualization: %s\n", i->virtualization);

if (!isempty(i->os_pretty_name))

printf(" Operating System: %s\n", i->os_pretty_name);

if (!isempty(i->os_cpe_name))

printf(" CPE OS Name: %s\n", i->os_cpe_name);

if (!isempty(i->kernel_name) && !isempty(i->kernel_release))

printf(" Kernel: %s %s\n", i->kernel_name, i->kernel_release);

if (!isempty(i->architecture))

printf(" Architecture: %s\n", i->architecture);

}

static int show_one_name(sd_bus *bus, const char* attr) {

_cleanup_bus_message_unref_ sd_bus_message *reply = NULL;

_cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;

const char *s;

int r;

r = sd_bus_get_property(

bus,

"org.freedesktop.hostname1",

"/org/freedesktop/hostname1",

"org.freedesktop.hostname1",

attr,

&error, &reply, "s");

if (r < 0) {

log_error("Could not get property: %s", bus_error_message(&error, -r));

return r;

}

r = sd_bus_message_read(reply, "s", &s);

if (r < 0)

return bus_log_parse_error(r);

printf("%s\n", s);

return 0;

}

static int show_all_names(sd_bus *bus) {

StatusInfo info = {};

static const struct bus_properties_map hostname_map[] = {

{ "Hostname", "s", NULL, offsetof(StatusInfo, hostname) },

{ "StaticHostname", "s", NULL, offsetof(StatusInfo, static_hostname) },

{ "PrettyHostname", "s", NULL, offsetof(StatusInfo, pretty_hostname) },

{ "IconName", "s", NULL, offsetof(StatusInfo, icon_name) },

{ "Chassis", "s", NULL, offsetof(StatusInfo, chassis) },

{ "Deployment", "s", NULL, offsetof(StatusInfo, deployment) },

{ "Location", "s", NULL, offsetof(StatusInfo, location) },

{ "KernelName", "s", NULL, offsetof(StatusInfo, kernel_name) },

{ "KernelRelease", "s", NULL, offsetof(StatusInfo, kernel_release) },

{ "OperatingSystemPrettyName", "s", NULL, offsetof(StatusInfo, os_pretty_name) },

{ "OperatingSystemCPEName", "s", NULL, offsetof(StatusInfo, os_cpe_name) },

{}

};

static const struct bus_properties_map manager_map[] = {

{ "Virtualization", "s", NULL, offsetof(StatusInfo, virtualization) },

{ "Architecture", "s", NULL, offsetof(StatusInfo, architecture) },

{}

};

int r;

r = bus_map_all_properties(bus,

"org.freedesktop.hostname1",

"/org/freedesktop/hostname1",

hostname_map,

&info);

if (r < 0)

goto fail;

bus_map_all_properties(bus,

"org.freedesktop.systemd1",

"/org/freedesktop/systemd1",

manager_map,

&info);

print_status_info(&info);

fail:

free(info.hostname);

free(info.static_hostname);

free(info.pretty_hostname);

free(info.icon_name);

free(info.chassis);

free(info.deployment);

free(info.location);

free(info.kernel_name);

free(info.kernel_release);

free(info.os_pretty_name);

free(info.os_cpe_name);

free(info.virtualization);

free(info.architecture);

return r;

}

static int show_status(sd_bus *bus, char **args, unsigned n) {

assert(args);

if (arg_pretty || arg_static || arg_transient) {

const char *attr;

if (!!arg_static + !!arg_pretty + !!arg_transient > 1) {

log_error("Cannot query more than one name type at a time");

return -EINVAL;

}

attr = arg_pretty ? "PrettyHostname" :

arg_static ? "StaticHostname" : "Hostname";

return show_one_name(bus, attr);

} else

return show_all_names(bus);

}

static int set_simple_string(sd_bus *bus, const char *method, const char *value) {

_cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;

int r = 0;

polkit_agent_open_if_enabled();

r = sd_bus_call_method(

bus,

"org.freedesktop.hostname1",

"/org/freedesktop/hostname1",

"org.freedesktop.hostname1",

method,

&error, NULL,

"sb", value, arg_ask_password);

if (r < 0)

log_error("Could not set property: %s", bus_error_message(&error, -r));

return r;

}

static int set_hostname(sd_bus *bus, char **args, unsigned n) {

_cleanup_free_ char *h = NULL;

char *hostname = args[1];

int r;

assert(args);

assert(n == 2);

if (!arg_pretty && !arg_static && !arg_transient)

arg_pretty = arg_static = arg_transient = true;

if (arg_pretty) {

const char *p;

/* If the passed hostname is already valid, then

if (arg_static && hostname_is_valid(hostname, true)) {

p = "";

/* maybe get rid of trailing dot */

hostname = hostname_cleanup(hostname);

} else {

p = h = strdup(hostname);

if (!p)

return log_oom();

hostname_cleanup(hostname);

}

r = set_simple_string(bus, "SetPrettyHostname", p);

if (r < 0)

return r;

}

if (arg_static) {

r = set_simple_string(bus, "SetStaticHostname", hostname);

if (r < 0)

return r;

}

if (arg_transient) {

r = set_simple_string(bus, "SetHostname", hostname);

if (r < 0)

return r;

}

return 0;

}

static int set_icon_name(sd_bus *bus, char **args, unsigned n) {

assert(args);

assert(n == 2);

return set_simple_string(bus, "SetIconName", args[1]);

}

static int set_chassis(sd_bus *bus, char **args, unsigned n) {

assert(args);

assert(n == 2);

return set_simple_string(bus, "SetChassis", args[1]);

}

static int set_deployment(sd_bus *bus, char **args, unsigned n) {

assert(args);

assert(n == 2);

return set_simple_string(bus, "SetDeployment", args[1]);

}

static int set_location(sd_bus *bus, char **args, unsigned n) {

assert(args);

assert(n == 2);

return set_simple_string(bus, "SetLocation", args[1]);

}

static void help(void) {

printf("%s [OPTIONS...] COMMAND ...\n\n"

"Query or change system hostname.\n\n"

" -h --help Show this help\n"

" --version Show package version\n"

" --no-ask-password Do not prompt for password\n"

" -H --host=[USER@]HOST Operate on remote host\n"

" -M --machine=CONTAINER Operate on local container\n"

" --transient Only set transient hostname\n"

" --static Only set static hostname\n"

" --pretty Only set pretty hostname\n\n"

"Commands:\n"

" status Show current hostname settings\n"

" set-hostname NAME Set system hostname\n"

" set-icon-name NAME Set icon name for host\n"

" set-chassis NAME Set chassis type for host\n"

" set-deployment NAME Set deployment environment for host\n"

" set-location NAME Set location for host\n"

, program_invocation_short_name);

}

static int parse_argv(int argc, char *argv[]) {

enum {

ARG_VERSION = 0x100,

ARG_NO_ASK_PASSWORD,

ARG_TRANSIENT,

ARG_STATIC,

ARG_PRETTY

};

static const struct option options[] = {

{ "help", no_argument, NULL, 'h' },

{ "version", no_argument, NULL, ARG_VERSION },

{ "transient", no_argument, NULL, ARG_TRANSIENT },

{ "static", no_argument, NULL, ARG_STATIC },

{ "pretty", no_argument, NULL, ARG_PRETTY },

{ "host", required_argument, NULL, 'H' },

{ "machine", required_argument, NULL, 'M' },

{ "no-ask-password", no_argument, NULL, ARG_NO_ASK_PASSWORD },

{}

};

int c;

assert(argc >= 0);

assert(argv);

while ((c = getopt_long(argc, argv, "hH:M:", options, NULL)) >= 0)

switch (c) {

case 'h':

help();

return 0;

case ARG_VERSION:

return version();

case 'H':

arg_transport = BUS_TRANSPORT_REMOTE;

arg_host = optarg;

break;

case 'M':

arg_transport = BUS_TRANSPORT_MACHINE;

arg_host = optarg;

break;

case ARG_TRANSIENT:

arg_transient = true;

break;

case ARG_PRETTY:

arg_pretty = true;

break;

case ARG_STATIC:

arg_static = true;

break;

case ARG_NO_ASK_PASSWORD:

arg_ask_password = false;

break;

case '?':

return -EINVAL;

default:

assert_not_reached("Unhandled option");

}

return 1;

}

static int hostnamectl_main(sd_bus *bus, int argc, char *argv[]) {

static const struct {

const char* verb;

const enum {

MORE,

LESS,

EQUAL

} argc_cmp;

const int argc;

int (* const dispatch)(sd_bus *bus, char **args, unsigned n);

} verbs[] = {

{ "status", LESS, 1, show_status },

{ "set-hostname", EQUAL, 2, set_hostname },

{ "set-icon-name", EQUAL, 2, set_icon_name },

{ "set-chassis", EQUAL, 2, set_chassis },

{ "set-deployment", EQUAL, 2, set_deployment },

{ "set-location", EQUAL, 2, set_location },

};

int left;

unsigned i;

assert(argc >= 0);

assert(argv);

left = argc - optind;

if (left <= 0)

/* Special rule: no arguments means "status" */

i = 0;

else {

if (streq(argv[optind], "help")) {

help();

return 0;

}

for (i = 0; i < ELEMENTSOF(verbs); i++)

if (streq(argv[optind], verbs[i].verb))

break;

if (i >= ELEMENTSOF(verbs)) {

log_error("Unknown operation %s", argv[optind]);

return -EINVAL;

}

}

switch (verbs[i].argc_cmp) {

case EQUAL:

if (left != verbs[i].argc) {

log_error("Invalid number of arguments.");

return -EINVAL;

}

break;

case MORE:

if (left < verbs[i].argc) {

log_error("Too few arguments.");

return -EINVAL;

}

break;

case LESS:

if (left > verbs[i].argc) {

log_error("Too many arguments.");

return -EINVAL;

}

break;

default:

assert_not_reached("Unknown comparison operator.");

}

return verbs[i].dispatch(bus, argv + optind, left);

}

int main(int argc, char *argv[]) {

_cleanup_bus_flush_close_unref_ sd_bus *bus = NULL;

int r;

setlocale(LC_ALL, "");

log_parse_environment();

log_open();

r = parse_argv(argc, argv);

if (r <= 0)

goto finish;

r = bus_open_transport(arg_transport, arg_host, false, &bus);

if (r < 0) {

log_error_errno(r, "Failed to create bus connection: %m");

goto finish;

}

r = hostnamectl_main(bus, argc, argv);

finish:

return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS;

}