+++ /dev/null
-/*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
-
-/***
- This file is part of systemd.
-
- Copyright 2012 Lennart Poettering
-
- systemd is free software; you can redistribute it and/or modify it
- under the terms of the GNU Lesser General Public License as published by
- the Free Software Foundation; either version 2.1 of the License, or
- (at your option) any later version.
-
- systemd is distributed in the hope that it will be useful, but
- WITHOUT ANY WARRANTY; without even the implied warranty of
- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- Lesser General Public License for more details.
-
- You should have received a copy of the GNU Lesser General Public License
- along with systemd; If not, see <http://www.gnu.org/licenses/>.
-***/
-
-#include <stdlib.h>
-#include <stdbool.h>
-#include <getopt.h>
-#include <locale.h>
-#include <string.h>
-
-#include "sd-bus.h"
-
-#include "bus-util.h"
-#include "bus-error.h"
-#include "util.h"
-#include "spawn-polkit-agent.h"
-#include "build.h"
-#include "sd-id128.h"
-#include "architecture.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;
- const 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
- * assume the user doesn't know anything about pretty
- * hostnames, so let's unset the pretty hostname, and
- * just set the passed hostname as static/dynamic
- * hostname. */
-
- h = strdup(hostname);
- if (!h)
- return log_oom();
-
- hostname_cleanup(h, true);
-
- if (arg_static && streq(h, hostname))
- p = "";
- else {
- p = hostname;
- hostname = h;
- }
-
- 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:
- puts(PACKAGE_STRING);
- puts(SYSTEMD_FEATURES);
- return 0;
-
- 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_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;
-}
+++ /dev/null
-/*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
-
-/***
- This file is part of systemd.
-
- Copyright 2011 Lennart Poettering
-
- systemd is free software; you can redistribute it and/or modify it
- under the terms of the GNU Lesser General Public License as published by
- the Free Software Foundation; either version 2.1 of the License, or
- (at your option) any later version.
-
- systemd is distributed in the hope that it will be useful, but
- WITHOUT ANY WARRANTY; without even the implied warranty of
- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- Lesser General Public License for more details.
-
- You should have received a copy of the GNU Lesser General Public License
- along with systemd; If not, see <http://www.gnu.org/licenses/>.
-***/
-
-#include <errno.h>
-#include <string.h>
-#include <unistd.h>
-#include <sys/utsname.h>
-
-#include "util.h"
-#include "strv.h"
-#include "def.h"
-#include "virt.h"
-#include "env-util.h"
-#include "fileio-label.h"
-#include "bus-util.h"
-#include "event-util.h"
-#include "selinux-util.h"
-
-#define VALID_DEPLOYMENT_CHARS (DIGITS LETTERS "-.:")
-
-enum {
- PROP_HOSTNAME,
- PROP_STATIC_HOSTNAME,
- PROP_PRETTY_HOSTNAME,
- PROP_ICON_NAME,
- PROP_CHASSIS,
- PROP_DEPLOYMENT,
- PROP_LOCATION,
- PROP_KERNEL_NAME,
- PROP_KERNEL_RELEASE,
- PROP_KERNEL_VERSION,
- PROP_OS_PRETTY_NAME,
- PROP_OS_CPE_NAME,
- _PROP_MAX
-};
-
-typedef struct Context {
- char *data[_PROP_MAX];
- Hashmap *polkit_registry;
-} Context;
-
-static void context_reset(Context *c) {
- int p;
-
- assert(c);
-
- for (p = 0; p < _PROP_MAX; p++) {
- free(c->data[p]);
- c->data[p] = NULL;
- }
-}
-
-static void context_free(Context *c) {
- assert(c);
-
- context_reset(c);
- bus_verify_polkit_async_registry_free(c->polkit_registry);
-}
-
-static int context_read_data(Context *c) {
- int r;
- struct utsname u;
-
- assert(c);
-
- context_reset(c);
-
- assert_se(uname(&u) >= 0);
- c->data[PROP_KERNEL_NAME] = strdup(u.sysname);
- c->data[PROP_KERNEL_RELEASE] = strdup(u.release);
- c->data[PROP_KERNEL_VERSION] = strdup(u.version);
- if (!c->data[PROP_KERNEL_NAME] || !c->data[PROP_KERNEL_RELEASE] ||
- !c->data[PROP_KERNEL_VERSION])
- return -ENOMEM;
-
- c->data[PROP_HOSTNAME] = gethostname_malloc();
- if (!c->data[PROP_HOSTNAME])
- return -ENOMEM;
-
- r = read_one_line_file("/etc/hostname", &c->data[PROP_STATIC_HOSTNAME]);
- if (r < 0 && r != -ENOENT)
- return r;
-
- r = parse_env_file("/etc/machine-info", NEWLINE,
- "PRETTY_HOSTNAME", &c->data[PROP_PRETTY_HOSTNAME],
- "ICON_NAME", &c->data[PROP_ICON_NAME],
- "CHASSIS", &c->data[PROP_CHASSIS],
- "DEPLOYMENT", &c->data[PROP_DEPLOYMENT],
- "LOCATION", &c->data[PROP_LOCATION],
- NULL);
- if (r < 0 && r != -ENOENT)
- return r;
-
- r = parse_env_file("/etc/os-release", NEWLINE,
- "PRETTY_NAME", &c->data[PROP_OS_PRETTY_NAME],
- "CPE_NAME", &c->data[PROP_OS_CPE_NAME],
- NULL);
- if (r == -ENOENT) {
- r = parse_env_file("/usr/lib/os-release", NEWLINE,
- "PRETTY_NAME", &c->data[PROP_OS_PRETTY_NAME],
- "CPE_NAME", &c->data[PROP_OS_CPE_NAME],
- NULL);
- }
-
- if (r < 0 && r != -ENOENT)
- return r;
-
- return 0;
-}
-
-static bool valid_chassis(const char *chassis) {
- assert(chassis);
-
- return nulstr_contains(
- "vm\0"
- "container\0"
- "desktop\0"
- "laptop\0"
- "server\0"
- "tablet\0"
- "handset\0"
- "watch\0"
- "embedded\0",
- chassis);
-}
-
-static bool valid_deployment(const char *deployment) {
- assert(deployment);
-
- return in_charset(deployment, VALID_DEPLOYMENT_CHARS);
-}
-
-static const char* fallback_chassis(void) {
- int r;
- char *type;
- unsigned t;
- int v;
-
- v = detect_virtualization(NULL);
-
- if (v == VIRTUALIZATION_VM)
- return "vm";
- if (v == VIRTUALIZATION_CONTAINER)
- return "container";
-
- r = read_one_line_file("/sys/firmware/acpi/pm_profile", &type);
- if (r < 0)
- goto try_dmi;
-
- r = safe_atou(type, &t);
- free(type);
- if (r < 0)
- goto try_dmi;
-
- /* We only list the really obvious cases here as the ACPI data
- * is not really super reliable.
- *
- * See the ACPI 5.0 Spec Section 5.2.9.1 for details:
- *
- * http://www.acpi.info/DOWNLOADS/ACPIspec50.pdf
- */
-
- switch(t) {
-
- case 1:
- case 3:
- case 6:
- return "desktop";
-
- case 2:
- return "laptop";
-
- case 4:
- case 5:
- case 7:
- return "server";
-
- case 8:
- return "tablet";
- }
-
-try_dmi:
- r = read_one_line_file("/sys/class/dmi/id/chassis_type", &type);
- if (r < 0)
- return NULL;
-
- r = safe_atou(type, &t);
- free(type);
- if (r < 0)
- return NULL;
-
- /* We only list the really obvious cases here. The DMI data is
- unreliable enough, so let's not do any additional guesswork
- on top of that.
-
- See the SMBIOS Specification 2.7.1 section 7.4.1 for
- details about the values listed here:
-
- http://www.dmtf.org/sites/default/files/standards/documents/DSP0134_2.7.1.pdf
- */
-
- switch (t) {
-
- case 0x3:
- case 0x4:
- case 0x6:
- case 0x7:
- return "desktop";
-
- case 0x8:
- case 0x9:
- case 0xA:
- case 0xE:
- return "laptop";
-
- case 0xB:
- return "handset";
-
- case 0x11:
- case 0x1C:
- return "server";
- }
-
- return NULL;
-}
-
-static char* context_fallback_icon_name(Context *c) {
- const char *chassis;
-
- assert(c);
-
- if (!isempty(c->data[PROP_CHASSIS]))
- return strappend("computer-", c->data[PROP_CHASSIS]);
-
- chassis = fallback_chassis();
- if (chassis)
- return strappend("computer-", chassis);
-
- return strdup("computer");
-}
-
-
-static bool hostname_is_useful(const char *hn) {
- return !isempty(hn) && !is_localhost(hn);
-}
-
-static int context_update_kernel_hostname(Context *c) {
- const char *static_hn;
- const char *hn;
-
- assert(c);
-
- static_hn = c->data[PROP_STATIC_HOSTNAME];
-
- /* /etc/hostname with something other than "localhost"
- * has the highest preference ... */
- if (hostname_is_useful(static_hn))
- hn = static_hn;
-
- /* ... the transient host name, (ie: DHCP) comes next ... */
- else if (!isempty(c->data[PROP_HOSTNAME]))
- hn = c->data[PROP_HOSTNAME];
-
- /* ... fallback to static "localhost.*" ignored above ... */
- else if (!isempty(static_hn))
- hn = static_hn;
-
- /* ... and the ultimate fallback */
- else
- hn = "localhost";
-
- if (sethostname_idempotent(hn) < 0)
- return -errno;
-
- return 0;
-}
-
-static int context_write_data_static_hostname(Context *c) {
-
- assert(c);
-
- if (isempty(c->data[PROP_STATIC_HOSTNAME])) {
-
- if (unlink("/etc/hostname") < 0)
- return errno == ENOENT ? 0 : -errno;
-
- return 0;
- }
- return write_string_file_atomic_label("/etc/hostname", c->data[PROP_STATIC_HOSTNAME]);
-}
-
-static int context_write_data_machine_info(Context *c) {
-
- static const char * const name[_PROP_MAX] = {
- [PROP_PRETTY_HOSTNAME] = "PRETTY_HOSTNAME",
- [PROP_ICON_NAME] = "ICON_NAME",
- [PROP_CHASSIS] = "CHASSIS",
- [PROP_DEPLOYMENT] = "DEPLOYMENT",
- [PROP_LOCATION] = "LOCATION",
- };
-
- _cleanup_strv_free_ char **l = NULL;
- int r, p;
-
- assert(c);
-
- r = load_env_file(NULL, "/etc/machine-info", NULL, &l);
- if (r < 0 && r != -ENOENT)
- return r;
-
- for (p = PROP_PRETTY_HOSTNAME; p <= PROP_LOCATION; p++) {
- _cleanup_free_ char *t = NULL;
- char **u;
-
- assert(name[p]);
-
- if (isempty(c->data[p])) {
- strv_env_unset(l, name[p]);
- continue;
- }
-
- t = strjoin(name[p], "=", c->data[p], NULL);
- if (!t)
- return -ENOMEM;
-
- u = strv_env_set(l, t);
- if (!u)
- return -ENOMEM;
-
- strv_free(l);
- l = u;
- }
-
- if (strv_isempty(l)) {
- if (unlink("/etc/machine-info") < 0)
- return errno == ENOENT ? 0 : -errno;
-
- return 0;
- }
-
- return write_env_file_label("/etc/machine-info", l);
-}
-
-static int property_get_icon_name(
- sd_bus *bus,
- const char *path,
- const char *interface,
- const char *property,
- sd_bus_message *reply,
- void *userdata,
- sd_bus_error *error) {
-
- _cleanup_free_ char *n = NULL;
- Context *c = userdata;
- const char *name;
-
- if (isempty(c->data[PROP_ICON_NAME]))
- name = n = context_fallback_icon_name(c);
- else
- name = c->data[PROP_ICON_NAME];
-
- if (!name)
- return -ENOMEM;
-
- return sd_bus_message_append(reply, "s", name);
-}
-
-static int property_get_chassis(
- sd_bus *bus,
- const char *path,
- const char *interface,
- const char *property,
- sd_bus_message *reply,
- void *userdata,
- sd_bus_error *error) {
-
- Context *c = userdata;
- const char *name;
-
- if (isempty(c->data[PROP_CHASSIS]))
- name = fallback_chassis();
- else
- name = c->data[PROP_CHASSIS];
-
- return sd_bus_message_append(reply, "s", name);
-}
-
-static int method_set_hostname(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
- Context *c = userdata;
- const char *name;
- int interactive;
- char *h;
- int r;
-
- r = sd_bus_message_read(m, "sb", &name, &interactive);
- if (r < 0)
- return r;
-
- if (isempty(name))
- name = c->data[PROP_STATIC_HOSTNAME];
-
- if (isempty(name))
- name = "localhost";
-
- if (!hostname_is_valid(name))
- return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Invalid hostname '%s'", name);
-
- if (streq_ptr(name, c->data[PROP_HOSTNAME]))
- return sd_bus_reply_method_return(m, NULL);
-
- r = bus_verify_polkit_async(
- m,
- CAP_SYS_ADMIN,
- "org.freedesktop.hostname1.set-hostname",
- interactive,
- UID_INVALID,
- &c->polkit_registry,
- error);
- if (r < 0)
- return r;
- if (r == 0)
- return 1; /* No authorization for now, but the async polkit stuff will call us again when it has it */
-
- h = strdup(name);
- if (!h)
- return -ENOMEM;
-
- free(c->data[PROP_HOSTNAME]);
- c->data[PROP_HOSTNAME] = h;
-
- r = context_update_kernel_hostname(c);
- if (r < 0) {
- log_error_errno(r, "Failed to set host name: %m");
- return sd_bus_error_set_errnof(error, r, "Failed to set hostname: %s", strerror(-r));
- }
-
- log_info("Changed host name to '%s'", strna(c->data[PROP_HOSTNAME]));
-
- sd_bus_emit_properties_changed(bus, "/org/freedesktop/hostname1", "org.freedesktop.hostname1", "Hostname", NULL);
-
- return sd_bus_reply_method_return(m, NULL);
-}
-
-static int method_set_static_hostname(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
- Context *c = userdata;
- const char *name;
- int interactive;
- int r;
-
- r = sd_bus_message_read(m, "sb", &name, &interactive);
- if (r < 0)
- return r;
-
- if (isempty(name))
- name = NULL;
-
- if (streq_ptr(name, c->data[PROP_STATIC_HOSTNAME]))
- return sd_bus_reply_method_return(m, NULL);
-
- r = bus_verify_polkit_async(
- m,
- CAP_SYS_ADMIN,
- "org.freedesktop.hostname1.set-static-hostname",
- interactive,
- UID_INVALID,
- &c->polkit_registry,
- error);
- if (r < 0)
- return r;
- if (r == 0)
- return 1; /* No authorization for now, but the async polkit stuff will call us again when it has it */
-
- if (isempty(name)) {
- free(c->data[PROP_STATIC_HOSTNAME]);
- c->data[PROP_STATIC_HOSTNAME] = NULL;
- } else {
- char *h;
-
- if (!hostname_is_valid(name))
- return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Invalid static hostname '%s'", name);
-
- h = strdup(name);
- if (!h)
- return -ENOMEM;
-
- free(c->data[PROP_STATIC_HOSTNAME]);
- c->data[PROP_STATIC_HOSTNAME] = h;
- }
-
- r = context_update_kernel_hostname(c);
- if (r < 0) {
- log_error_errno(r, "Failed to set host name: %m");
- return sd_bus_error_set_errnof(error, r, "Failed to set hostname: %s", strerror(-r));
- }
-
- r = context_write_data_static_hostname(c);
- if (r < 0) {
- log_error_errno(r, "Failed to write static host name: %m");
- return sd_bus_error_set_errnof(error, r, "Failed to set static hostname: %s", strerror(-r));
- }
-
- log_info("Changed static host name to '%s'", strna(c->data[PROP_STATIC_HOSTNAME]));
-
- sd_bus_emit_properties_changed(bus, "/org/freedesktop/hostname1", "org.freedesktop.hostname1", "StaticHostname", NULL);
-
- return sd_bus_reply_method_return(m, NULL);
-}
-
-static int set_machine_info(Context *c, sd_bus *bus, sd_bus_message *m, int prop, sd_bus_message_handler_t cb, sd_bus_error *error) {
- int interactive;
- const char *name;
- int r;
-
- assert(c);
- assert(bus);
- assert(m);
-
- r = sd_bus_message_read(m, "sb", &name, &interactive);
- if (r < 0)
- return r;
-
- if (isempty(name))
- name = NULL;
-
- if (streq_ptr(name, c->data[prop]))
- return sd_bus_reply_method_return(m, NULL);
-
- /* Since the pretty hostname should always be changed at the
- * same time as the static one, use the same policy action for
- * both... */
-
- r = bus_verify_polkit_async(
- m,
- CAP_SYS_ADMIN,
- prop == PROP_PRETTY_HOSTNAME ? "org.freedesktop.hostname1.set-static-hostname" : "org.freedesktop.hostname1.set-machine-info",
- interactive,
- UID_INVALID,
- &c->polkit_registry,
- error);
- if (r < 0)
- return r;
- if (r == 0)
- return 1; /* No authorization for now, but the async polkit stuff will call us again when it has it */
-
- if (isempty(name)) {
- free(c->data[prop]);
- c->data[prop] = NULL;
- } else {
- char *h;
-
- /* The icon name might ultimately be used as file
- * name, so better be safe than sorry */
-
- if (prop == PROP_ICON_NAME && !filename_is_valid(name))
- return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Invalid icon name '%s'", name);
- if (prop == PROP_PRETTY_HOSTNAME && string_has_cc(name, NULL))
- return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Invalid pretty host name '%s'", name);
- if (prop == PROP_CHASSIS && !valid_chassis(name))
- return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Invalid chassis '%s'", name);
- if (prop == PROP_DEPLOYMENT && !valid_deployment(name))
- return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Invalid deployment '%s'", name);
- if (prop == PROP_LOCATION && string_has_cc(name, NULL))
- return sd_bus_error_setf(error, SD_BUS_ERROR_INVALID_ARGS, "Invalid location '%s'", name);
-
- h = strdup(name);
- if (!h)
- return -ENOMEM;
-
- free(c->data[prop]);
- c->data[prop] = h;
- }
-
- r = context_write_data_machine_info(c);
- if (r < 0) {
- log_error_errno(r, "Failed to write machine info: %m");
- return sd_bus_error_set_errnof(error, r, "Failed to write machine info: %s", strerror(-r));
- }
-
- log_info("Changed %s to '%s'",
- prop == PROP_PRETTY_HOSTNAME ? "pretty host name" :
- prop == PROP_DEPLOYMENT ? "deployment" :
- prop == PROP_LOCATION ? "location" :
- prop == PROP_CHASSIS ? "chassis" : "icon name", strna(c->data[prop]));
-
- sd_bus_emit_properties_changed(bus, "/org/freedesktop/hostname1", "org.freedesktop.hostname1",
- prop == PROP_PRETTY_HOSTNAME ? "PrettyHostname" :
- prop == PROP_DEPLOYMENT ? "Deployment" :
- prop == PROP_LOCATION ? "Location" :
- prop == PROP_CHASSIS ? "Chassis" : "IconName" , NULL);
-
- return sd_bus_reply_method_return(m, NULL);
-}
-
-static int method_set_pretty_hostname(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
- return set_machine_info(userdata, bus, m, PROP_PRETTY_HOSTNAME, method_set_pretty_hostname, error);
-}
-
-static int method_set_icon_name(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
- return set_machine_info(userdata, bus, m, PROP_ICON_NAME, method_set_icon_name, error);
-}
-
-static int method_set_chassis(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
- return set_machine_info(userdata, bus, m, PROP_CHASSIS, method_set_chassis, error);
-}
-
-static int method_set_deployment(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
- return set_machine_info(userdata, bus, m, PROP_DEPLOYMENT, method_set_deployment, error);
-}
-
-static int method_set_location(sd_bus *bus, sd_bus_message *m, void *userdata, sd_bus_error *error) {
- return set_machine_info(userdata, bus, m, PROP_LOCATION, method_set_location, error);
-}
-
-static const sd_bus_vtable hostname_vtable[] = {
- SD_BUS_VTABLE_START(0),
- SD_BUS_PROPERTY("Hostname", "s", NULL, offsetof(Context, data) + sizeof(char*) * PROP_HOSTNAME, SD_BUS_VTABLE_PROPERTY_EMITS_CHANGE),
- SD_BUS_PROPERTY("StaticHostname", "s", NULL, offsetof(Context, data) + sizeof(char*) * PROP_STATIC_HOSTNAME, SD_BUS_VTABLE_PROPERTY_EMITS_CHANGE),
- SD_BUS_PROPERTY("PrettyHostname", "s", NULL, offsetof(Context, data) + sizeof(char*) * PROP_PRETTY_HOSTNAME, SD_BUS_VTABLE_PROPERTY_EMITS_CHANGE),
- SD_BUS_PROPERTY("IconName", "s", property_get_icon_name, 0, SD_BUS_VTABLE_PROPERTY_EMITS_CHANGE),
- SD_BUS_PROPERTY("Chassis", "s", property_get_chassis, 0, SD_BUS_VTABLE_PROPERTY_EMITS_CHANGE),
- SD_BUS_PROPERTY("Deployment", "s", NULL, offsetof(Context, data) + sizeof(char*) * PROP_DEPLOYMENT, SD_BUS_VTABLE_PROPERTY_EMITS_CHANGE),
- SD_BUS_PROPERTY("Location", "s", NULL, offsetof(Context, data) + sizeof(char*) * PROP_LOCATION, SD_BUS_VTABLE_PROPERTY_EMITS_CHANGE),
- SD_BUS_PROPERTY("KernelName", "s", NULL, offsetof(Context, data) + sizeof(char*) * PROP_KERNEL_NAME, SD_BUS_VTABLE_PROPERTY_CONST),
- SD_BUS_PROPERTY("KernelRelease", "s", NULL, offsetof(Context, data) + sizeof(char*) * PROP_KERNEL_RELEASE, SD_BUS_VTABLE_PROPERTY_CONST),
- SD_BUS_PROPERTY("KernelVersion", "s", NULL, offsetof(Context, data) + sizeof(char*) * PROP_KERNEL_VERSION, SD_BUS_VTABLE_PROPERTY_CONST),
- SD_BUS_PROPERTY("OperatingSystemPrettyName", "s", NULL, offsetof(Context, data) + sizeof(char*) * PROP_OS_PRETTY_NAME, SD_BUS_VTABLE_PROPERTY_CONST),
- SD_BUS_PROPERTY("OperatingSystemCPEName", "s", NULL, offsetof(Context, data) + sizeof(char*) * PROP_OS_CPE_NAME, SD_BUS_VTABLE_PROPERTY_CONST),
- SD_BUS_METHOD("SetHostname", "sb", NULL, method_set_hostname, SD_BUS_VTABLE_UNPRIVILEGED),
- SD_BUS_METHOD("SetStaticHostname", "sb", NULL, method_set_static_hostname, SD_BUS_VTABLE_UNPRIVILEGED),
- SD_BUS_METHOD("SetPrettyHostname", "sb", NULL, method_set_pretty_hostname, SD_BUS_VTABLE_UNPRIVILEGED),
- SD_BUS_METHOD("SetIconName", "sb", NULL, method_set_icon_name, SD_BUS_VTABLE_UNPRIVILEGED),
- SD_BUS_METHOD("SetChassis", "sb", NULL, method_set_chassis, SD_BUS_VTABLE_UNPRIVILEGED),
- SD_BUS_METHOD("SetDeployment", "sb", NULL, method_set_deployment, SD_BUS_VTABLE_UNPRIVILEGED),
- SD_BUS_METHOD("SetLocation", "sb", NULL, method_set_location, SD_BUS_VTABLE_UNPRIVILEGED),
- SD_BUS_VTABLE_END,
-};
-
-static int connect_bus(Context *c, sd_event *event, sd_bus **_bus) {
- _cleanup_bus_close_unref_ sd_bus *bus = NULL;
- int r;
-
- assert(c);
- assert(event);
- assert(_bus);
-
- r = sd_bus_default_system(&bus);
- if (r < 0)
- return log_error_errno(r, "Failed to get system bus connection: %m");
-
- r = sd_bus_add_object_vtable(bus, NULL, "/org/freedesktop/hostname1", "org.freedesktop.hostname1", hostname_vtable, c);
- if (r < 0)
- return log_error_errno(r, "Failed to register object: %m");
-
- r = sd_bus_request_name(bus, "org.freedesktop.hostname1", 0);
- if (r < 0)
- return log_error_errno(r, "Failed to register name: %m");
-
- r = sd_bus_attach_event(bus, event, 0);
- if (r < 0)
- return log_error_errno(r, "Failed to attach bus to event loop: %m");
-
- *_bus = bus;
- bus = NULL;
-
- return 0;
-}
-
-int main(int argc, char *argv[]) {
- Context context = {};
- _cleanup_event_unref_ sd_event *event = NULL;
- _cleanup_bus_close_unref_ sd_bus *bus = NULL;
- int r;
-
- log_set_target(LOG_TARGET_AUTO);
- log_parse_environment();
- log_open();
-
- umask(0022);
- mac_selinux_init("/etc");
-
- if (argc != 1) {
- log_error("This program takes no arguments.");
- r = -EINVAL;
- goto finish;
- }
-
- if (argc != 1) {
- log_error("This program takes no arguments.");
- r = -EINVAL;
- goto finish;
- }
-
- r = sd_event_default(&event);
- if (r < 0) {
- log_error_errno(r, "Failed to allocate event loop: %m");
- goto finish;
- }
-
- sd_event_set_watchdog(event, true);
-
- r = connect_bus(&context, event, &bus);
- if (r < 0)
- goto finish;
-
- r = context_read_data(&context);
- if (r < 0) {
- log_error_errno(r, "Failed to read hostname and machine information: %m");
- goto finish;
- }
-
- r = bus_event_loop_with_idle(event, bus, "org.freedesktop.hostname1", DEFAULT_EXIT_USEC, NULL, NULL);
- if (r < 0) {
- log_error_errno(r, "Failed to run event loop: %m");
- goto finish;
- }
-
-finish:
- context_free(&context);
-
- return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS;
-}
~ianmdlvl / elogind.git / commitdiff