src/interfaces/hostnamed/hostnamed.c

   1 /*
   2  * Copyright (c) 2014 Ian Sutton <ian@kremlin.cc>
   3  *
   4  * Permission to use, copy, modify, and distribute this software for any
   5  * purpose with or without fee is hereby granted, provided that the above
   6  * copyright notice and this permission notice appear in all copies.
   7  *
   8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
   9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15  */
  16
  17 #include <unistd.h>
  18 #include <limits.h>
  19 #include <signal.h>
  20 #include <string.h>
  21
  22 #include <sys/param.h>
  23 #include <sys/sysctl.h>
  24 #include <sys/sensors.h>
  25 #include <sys/ioctl.h>
  26 #include <sys/utsname.h>
  27
  28 #include <machine/apmvar.h>
  29
  30 #include <glib/gprintf.h>
  31 #include <glib-unix.h>
  32 #include <polkit/polkit.h>
  33
  34 #include "hostnamed-gen.h"
  35 #include "hostnamed.h"
  36
  37 #include "../../polkit-auth.h"
  38
  39 /* format: {
  40  *           (1) string to be matched against runtime machine's sysctl output.
  41  *               can be either the exact string or a substring contained
  42  *               within sysctl strings. no "guesses" here, a match should
  43  *               reliably indicate the chassis/icon.
  44  *
  45  *           (2) string describing chassis type divulged by (1).
  46  *               must be one of "desktop", "laptop", "server",
  47  *               "tablet", "handset", "vm", "container" or NULL
  48  *               if only icon string can be ascertained. "vm" refers
  49  *               to operating systems running on baremetal hypervisors
  50  *               (hardware virtualization, like XEN) while "container"
  51  *               refers to OSs running on shared hypervisors like
  52  *               virtualbox or VMware. consider the distinction carefully
  53  *               as common virtualization software like KVM may share
  54  *               characteristics of both "vm" and "container" types.
  55  *
  56  *           (3) string specifying icon to use. follows XDG icon spec.
  57  *               see http://standards.freedesktop.org/icon-naming-spec/icon-naming-spec-latest.html
  58  *               for allowed strings.
  59  *
  60  *           (4) chassis precedence bit. TRUE if (2) is defined and
  61  *               we're certain it is the proper string. FALSE in the
  62  *               circumstance (2) may be the correct string, unless
  63  *               a match with this bit set to TRUE overrides it.
  64  *               if (2) is NULL, this bit is inconsequential.
  65  *
  66  *           (5) icon precedence bit. see previous definition.
  67  *         }                                               */
  68 struct SYSCTL_LOOKUP_TABLE {
  69     gchar    *match_string;
  70     gchar    *chassis;
  71     gchar    *icon;
  72     gboolean chassis_precedence;
  73     gboolean icon_precedence;
  74 };
  75
  76 GPtrArray *hostnamed_freeable;
  77 Hostname1 *hostnamed_interf;
  78
  79 GMainLoop *hostnamed_loop;
  80
  81 guint bus_descriptor;
  82 gboolean dbus_interface_exported; /* reliable because of gdbus operational guarantees */
  83
  84 gchar *HOSTNAME, *STATIC_HOSTNAME, *PRETTY_HOSTNAME;
  85 gchar *CHASSIS, *ICON;
  86 gchar *KERN_NAME, *KERN_RELEASE, *KERN_VERS, *OS_CPENAME;
  87
  88 /* TODO no specific vm or laptop icon in gnome
  89  * NOTE paravirtualization on xen is only available for linuxes right now
  90  * dmesg on linux systems reveals xen and virtualization method (HVM or PVM)
  91  * but we will worry about those later */
  92
  93 /* add any sysctl strings that suggest virtualization here */
  94 const struct SYSCTL_LOOKUP_TABLE chassis_indicator_table[] =
  95 {
  96     { "QEMU Virtual CPU",        "vm",        NULL,              FALSE, FALSE }, /* could be QEMU running in userspace or as part of KVM */
  97     { "KVM",                     "vm",        "drive-multidisk", FALSE, FALSE },
  98     { "SmartDC HVM",             "vm",        "drive-multidisk", TRUE,  TRUE  }, /* illumos-joyent kvm */
  99     { "VirtualBox",              "vm",        "drive-multidisk", TRUE,  TRUE  },
 100     { "VMware, Inc.",            "vm",        "drive-multidisk", TRUE,  TRUE  },
 101     { "VMware Virtual Platform", "vm",        "drive-multidisk", TRUE,  TRUE  },
 102     { "Parallels",               "vm",        "drive-multidisk", TRUE,  TRUE  }, /* need verification */
 103     { "Xen",                     "vm",        "drive-multidisk", FALSE, FALSE }
 104 }; /* TODO: chroots, etc. are the actual "containers", add them */
 105
 106 /* archs to check against when determining if machine is server */
 107 const gchar *server_archs[] = {
 108     "hppa",
 109     "sparc",
 110     "sparc64"
 111 };
 112
 113 /* --- begin method/property/dbus signal code --- */
 114
 115 /* TODO free some strings here */
 116 static gboolean
 117 on_handle_set_hostname(Hostname1 *hn1_passed_interf,
 118                        GDBusMethodInvocation *invoc,
 119                        const gchar *greet,
 120                        gpointer data) {
 121     GVariant *params;
 122     gchar *proposed_hostname, *valid_hostname_buf;
 123     const gchar *bus_name;
 124     gboolean policykit_auth, ret, try_to_set;
 125     size_t check_length;
 126     check_auth_result is_authed;
 127
 128     proposed_hostname = NULL;
 129     ret = try_to_set = FALSE;
 130
 131     params = g_dbus_method_invocation_get_parameters(invoc);
 132     g_variant_get(params, "(sb)", &proposed_hostname, &policykit_auth);
 133     bus_name = g_dbus_method_invocation_get_sender(invoc);
 134
 135     /* verify caller has correct permissions via polkit */
 136     is_authed = polkit_try_auth(bus_name, "org.freedesktop.hostname1.set-hostname", policykit_auth);
 137
 138     switch(is_authed) {
 139
 140         case AUTHORIZED_NATIVELY:
 141         case AUTHORIZED_BY_PROMPT:
 142             try_to_set = TRUE;
 143             break;
 144
 145         case UNAUTHORIZED_NATIVELY:
 146         case UNAUTHORIZED_FAILED_PROMPT:
 147             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.EACCES", "Insufficient permissions to set hostname.");
 148             break;
 149
 150         case ERROR_BADBUS:
 151             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.EFAULT", "Provided bus name is invalid.");
 152             break;
 153
 154         case ERROR_BADACTION:
 155             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.EFAULT", "Provided action ID is invalid.");
 156             break;
 157
 158         case ERROR_GENERIC:
 159         default:
 160             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.ECANCELED", "Failed to set hostname for unknown reason.");
 161             break;
 162     }
 163
 164     /* verify passed hostname's validity */
 165     if(try_to_set && proposed_hostname && (valid_hostname_buf = g_hostname_to_ascii(proposed_hostname))) {
 166
 167         check_length = strnlen(valid_hostname_buf, MAXHOSTNAMELEN + 1);
 168
 169         if(check_length > MAXHOSTNAMELEN) {
 170
 171             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.ENAMETOOLONG", "Hostname string exceeded maximum length.");
 172             g_free(valid_hostname_buf);
 173
 174         } else if(sethostname(proposed_hostname, check_length)) {
 175
 176             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.ECANCELED", "Failed to set hostname for unknown reason.");
 177             g_free(valid_hostname_buf);
 178
 179         } else {
 180
 181             HOSTNAME = valid_hostname_buf;
 182             hostname1_set_hostname(hn1_passed_interf, HOSTNAME);
 183             g_ptr_array_add(hostnamed_freeable, valid_hostname_buf);
 184             ret = TRUE;
 185             hostname1_complete_set_hostname(hn1_passed_interf, invoc);
 186         }
 187     }
 188
 189     return ret;
 190 }
 191
 192 static gboolean
 193 on_handle_set_static_hostname(Hostname1 *hn1_passed_interf,
 194                               GDBusMethodInvocation *invoc,
 195                               const gchar *greet,
 196                               gpointer data) {
 197
 198     GVariant *params;
 199     gchar *proposed_static_hostname, *valid_static_hostname_buf;
 200     const gchar *bus_name;
 201     gboolean policykit_auth, ret, try_to_set;
 202     size_t check_length;
 203     check_auth_result is_authed;
 204
 205     proposed_static_hostname = NULL;
 206     ret = try_to_set = FALSE;
 207
 208     params = g_dbus_method_invocation_get_parameters(invoc);
 209     g_variant_get(params, "(sb)", &proposed_static_hostname, &policykit_auth);
 210     bus_name = g_dbus_method_invocation_get_sender(invoc);
 211
 212     /* verify caller has correct permissions via polkit */
 213     is_authed = polkit_try_auth(bus_name, "org.freedesktop.hostname1.set-static-hostname", policykit_auth);
 214
 215     switch(is_authed) {
 216
 217         case AUTHORIZED_NATIVELY:
 218         case AUTHORIZED_BY_PROMPT:
 219             try_to_set = TRUE;
 220             break;
 221
 222         case UNAUTHORIZED_NATIVELY:
 223         case UNAUTHORIZED_FAILED_PROMPT:
 224             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.EACCES", "Insufficient permissions to set static hostname.");
 225             break;
 226
 227         case ERROR_BADBUS:
 228             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.EFAULT", "Provided bus name is invalid.");
 229             break;
 230
 231         case ERROR_BADACTION:
 232             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.EFAULT", "Provided action ID is invalid.");
 233             break;
 234
 235         case ERROR_GENERIC:
 236         default:
 237             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.ECANCELED", "Failed to set static hostname for unknown reason.");
 238             break;
 239     }
 240
 241     /* verify passed hostname's validity */
 242     if(try_to_set && proposed_static_hostname && (valid_static_hostname_buf = g_hostname_to_ascii(proposed_static_hostname))) {
 243
 244         check_length = strnlen(valid_static_hostname_buf, MAXHOSTNAMELEN + 1);
 245
 246         if(check_length > MAXHOSTNAMELEN) {
 247
 248             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.ENAMETOOLONG", "Static hostname string exceeded maximum length.");
 249             g_free(valid_static_hostname_buf);
 250
 251         } else if(!(STATIC_HOSTNAME = valid_static_hostname_buf)) {
 252
 253             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.ECANCELED", "Failed to set static hostname for unknown reason.");
 254             g_free(valid_static_hostname_buf);
 255
 256         } else {
 257
 258             g_strdelimit(STATIC_HOSTNAME, " ", '-');
 259             hostname1_set_static_hostname(hn1_passed_interf, STATIC_HOSTNAME);
 260             g_ptr_array_add(hostnamed_freeable, valid_static_hostname_buf);
 261             ret = (!sethostname(valid_static_hostname_buf, MAXHOSTNAMELEN)) ? TRUE : FALSE; /* TODO set /etc/myname, guarantee domain or substitue .home.network" */
 262             hostname1_complete_set_static_hostname(hn1_passed_interf, invoc);
 263         }
 264     }
 265
 266     return ret;
 267 }
 268
 269 static gboolean
 270 on_handle_set_pretty_hostname(Hostname1 *hn1_passed_interf,
 271                               GDBusMethodInvocation *invoc,
 272                               const gchar *greet,
 273                               gpointer data) {
 274
 275     GVariant *params;
 276     gchar *proposed_pretty_hostname, *valid_pretty_hostname_buf, *computed_static_hostname;
 277     const gchar *bus_name;
 278     gboolean policykit_auth, ret, try_to_set;
 279     size_t check_length;
 280     check_auth_result is_authed;
 281     GKeyFile *config;
 282
 283     config = g_key_file_new();
 284     proposed_pretty_hostname = NULL;
 285     ret = try_to_set = FALSE;
 286
 287     params = g_dbus_method_invocation_get_parameters(invoc);
 288     g_variant_get(params, "(sb)", &proposed_pretty_hostname, &policykit_auth);
 289     bus_name = g_dbus_method_invocation_get_sender(invoc);
 290
 291     /* verify caller has correct permissions via polkit */
 292     is_authed = polkit_try_auth(bus_name, "org.freedesktop.hostname1.set-pretty-hostname", policykit_auth);
 293
 294     switch(is_authed) {
 295
 296         case AUTHORIZED_NATIVELY:
 297         case AUTHORIZED_BY_PROMPT:
 298             try_to_set = TRUE;
 299             break;
 300
 301         case UNAUTHORIZED_NATIVELY:
 302         case UNAUTHORIZED_FAILED_PROMPT:
 303             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.EACCES", "Insufficient permissions to set pretty hostname.");
 304             break;
 305
 306         case ERROR_BADBUS:
 307             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.EFAULT", "Provided bus name is invalid.");
 308             break;
 309
 310         case ERROR_BADACTION:
 311             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.EFAULT", "Provided action ID is invalid.");
 312             break;
 313
 314         case ERROR_GENERIC:
 315         default:
 316             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.ECANCELED", "Failed to set pretty hostname for unknown reason.");
 317             break;
 318     }
 319
 320     /* verify passed hostname's validity */
 321     if(try_to_set && proposed_pretty_hostname && (valid_pretty_hostname_buf = g_locale_to_utf8(proposed_pretty_hostname, -1, 0, 0, NULL))) {
 322
 323         check_length = strnlen(valid_pretty_hostname_buf, MAXHOSTNAMELEN + 1);
 324
 325         if(check_length > MAXHOSTNAMELEN) {
 326
 327             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.ENAMETOOLONG", "Static hostname string exceeded maximum length.");
 328             g_free(valid_pretty_hostname_buf);
 329
 330         } else if(!(PRETTY_HOSTNAME = valid_pretty_hostname_buf)) {
 331
 332             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.ECANCELED", "Failed to set pretty hostname for unknown reason.");
 333             g_free(valid_pretty_hostname_buf);
 334
 335         } else {
 336
 337             hostname1_set_pretty_hostname(hn1_passed_interf, PRETTY_HOSTNAME);
 338             g_ptr_array_add(hostnamed_freeable, valid_pretty_hostname_buf);
 339             hostname1_complete_set_pretty_hostname(hn1_passed_interf, invoc);
 340             ret = TRUE;
 341
 342             if(g_key_file_load_from_file(config, "/etc/machine-info", G_KEY_FILE_NONE, NULL)) {
 343
 344                 g_key_file_set_string(config, "hostnamed", "PRETTY_HOSTNAME", valid_pretty_hostname_buf);
 345
 346                 if((computed_static_hostname = g_hostname_to_ascii(PRETTY_HOSTNAME))) {
 347
 348                     g_strdelimit(computed_static_hostname, " ", '-');
 349                     hostname1_set_static_hostname(hn1_passed_interf, computed_static_hostname);
 350                     STATIC_HOSTNAME = computed_static_hostname;
 351                     g_ptr_array_add(hostnamed_freeable, computed_static_hostname);
 352                     g_key_file_set_string(config, "hostnamed", "StaticHostname", computed_static_hostname);
 353
 354                 } else
 355                     g_free(computed_static_hostname);
 356             }
 357
 358         }
 359     }
 360
 361     g_key_file_save_to_file(config, "/etc/machine-info", NULL);
 362     g_key_file_unref(config);
 363
 364     return ret;
 365 }
 366
 367 static gboolean
 368 on_handle_set_chassis(Hostname1 *hn1_passed_interf,
 369                       GDBusMethodInvocation *invoc,
 370                       const gchar *greet,
 371                       gpointer data) {
 372
 373     GVariant *params;
 374     gchar *proposed_chassis_name, *valid_chassis_name_buf;
 375     const gchar *bus_name;
 376     gboolean policykit_auth, ret, try_to_set;
 377     check_auth_result is_authed;
 378     GKeyFile *config;
 379
 380     config = g_key_file_new();
 381     proposed_chassis_name = NULL;
 382     ret = try_to_set = FALSE;
 383     valid_chassis_name_buf = (gchar *)g_malloc0(8192);
 384
 385     params = g_dbus_method_invocation_get_parameters(invoc);
 386     g_variant_get(params, "(sb)", &proposed_chassis_name, &policykit_auth);
 387     bus_name = g_dbus_method_invocation_get_sender(invoc);
 388
 389     g_strlcpy(valid_chassis_name_buf, proposed_chassis_name, (gsize)64);
 390
 391     /* verify caller has correct permissions via polkit */
 392     is_authed = polkit_try_auth(bus_name, "org.freedesktop.hostname1.set-chassis", policykit_auth);
 393
 394     switch(is_authed) {
 395
 396         case AUTHORIZED_NATIVELY:
 397         case AUTHORIZED_BY_PROMPT:
 398             try_to_set = TRUE;
 399             break;
 400
 401         case UNAUTHORIZED_NATIVELY:
 402         case UNAUTHORIZED_FAILED_PROMPT:
 403             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.EACCES", "Insufficient permissions to set chassis type.");
 404             break;
 405
 406         case ERROR_BADBUS:
 407             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.EFAULT", "Provided bus name is invalid.");
 408             break;
 409
 410         case ERROR_BADACTION:
 411             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.EFAULT", "Provided action ID is invalid.");
 412             break;
 413
 414         case ERROR_GENERIC:
 415         default:
 416             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.ECANCELED", "Failed to set chassis type for unknown reason.");
 417             break;
 418     }
 419
 420     /* verify passed chassis type's validity */
 421     if(try_to_set && proposed_chassis_name) {
 422
 423         if(!is_valid_chassis_type(proposed_chassis_name)) {
 424
 425             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.ECANCELED", "Chassis type must be 'desktop', 'laptop', 'server', 'tablet', 'handset', 'vm', or 'container'.");
 426             g_free(valid_chassis_name_buf);
 427
 428         } else if(!(CHASSIS = valid_chassis_name_buf)) {
 429
 430             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.ECANCELED", "Failed to set chassis type for unknown reason.");
 431             g_free(valid_chassis_name_buf);
 432
 433         } else {
 434
 435             hostname1_set_chassis(hn1_passed_interf, CHASSIS);
 436             g_ptr_array_add(hostnamed_freeable, valid_chassis_name_buf);
 437             hostname1_complete_set_chassis(hn1_passed_interf, invoc);
 438
 439             if(g_key_file_load_from_file(config, "/etc/machine-info", G_KEY_FILE_NONE, NULL)) {
 440
 441                 ret = TRUE;
 442                 g_key_file_set_string(config, "hostnamed", "ChassisType", valid_chassis_name_buf);
 443
 444             }
 445         }
 446     }
 447
 448     g_key_file_save_to_file(config, "/etc/machine-info", NULL);
 449     g_key_file_unref(config);
 450
 451     return ret;
 452 }
 453
 454 static gboolean
 455 on_handle_set_icon_name(Hostname1 *hn1_passed_interf,
 456                         GDBusMethodInvocation *invoc,
 457                         const gchar *greet,
 458                         gpointer data) {
 459
 460     GVariant *params;
 461     gchar *proposed_icon_name, *valid_icon_name_buf;
 462     const gchar *bus_name;
 463     gboolean policykit_auth, ret, try_to_set;
 464     check_auth_result is_authed;
 465     GKeyFile *config;
 466
 467     config = g_key_file_new();
 468     proposed_icon_name = NULL;
 469     ret = try_to_set = FALSE;
 470
 471     params = g_dbus_method_invocation_get_parameters(invoc);
 472     g_variant_get(params, "(sb)", &proposed_icon_name, &policykit_auth);
 473     bus_name = g_dbus_method_invocation_get_sender(invoc);
 474
 475     /* verify caller has correct permissions via polkit */
 476     is_authed = polkit_try_auth(bus_name, "org.freedesktop.hostname1.set-icon-name", policykit_auth);
 477
 478     switch(is_authed) {
 479
 480         case AUTHORIZED_NATIVELY:
 481         case AUTHORIZED_BY_PROMPT:
 482             try_to_set = TRUE;
 483             break;
 484
 485         case UNAUTHORIZED_NATIVELY:
 486         case UNAUTHORIZED_FAILED_PROMPT:
 487             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.EACCES", "Insufficient permissions to set icon name.");
 488             break;
 489
 490         case ERROR_BADBUS:
 491             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.EFAULT", "Provided bus name is invalid.");
 492             break;
 493
 494         case ERROR_BADACTION:
 495             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.EFAULT", "Provided action ID is invalid.");
 496             break;
 497
 498         case ERROR_GENERIC:
 499         default:
 500             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.ECANCELED", "Failed to set icon name for unknown reason.");
 501             break;
 502     }
 503
 504     /* verify passed chassis type's validity */
 505     if(try_to_set && proposed_icon_name) {
 506
 507         g_strlcpy(valid_icon_name_buf, proposed_icon_name, (gsize)64);
 508
 509         if(!(ICON = valid_icon_name_buf)) {
 510
 511             g_dbus_method_invocation_return_dbus_error(invoc, "org.freedesktop.hostname1.Error.ECANCELED", "Failed to set icon name for unknown reason.");
 512             g_free(valid_icon_name_buf);
 513
 514         } else {
 515
 516             hostname1_set_icon_name(hn1_passed_interf, ICON);
 517             g_ptr_array_add(hostnamed_freeable, valid_icon_name_buf);
 518             hostname1_complete_set_icon_name(hn1_passed_interf, invoc);
 519
 520             if(g_key_file_load_from_file(config, "/etc/machine-info", G_KEY_FILE_NONE, NULL)) {
 521
 522                 ret = TRUE;
 523                 g_key_file_set_string(config, "hostnamed", "IconName", valid_icon_name_buf);
 524
 525             }
 526         }
 527     }
 528
 529     g_key_file_save_to_file(config, "/etc/machine-info", NULL);
 530     g_key_file_unref(config);
 531
 532     return ret;
 533 }
 534
 535 /* note: all hostnamed/hostname1's properties are read-only,
 536  * and do not need set_ functions, gdbus-codegen realized
 537  * this from the XML and handled the to-be error of trying
 538  * to set a read-only property's value
 539  */
 540
 541 const gchar *
 542 our_get_hostname() {
 543
 544     gchar *hostname_buf;
 545     hostname_buf = (gchar *)g_malloc0(MAXHOSTNAMELEN);
 546
 547     if(gethostname(hostname_buf, MAXHOSTNAMELEN))
 548         return "localhost.home.network"; /* TODO bomb out here probably */
 549
 550     else if(!g_strcmp0(HOSTNAME, hostname_buf)) {
 551
 552         g_free(hostname_buf);
 553         return HOSTNAME;
 554     }
 555
 556     g_ptr_array_add(hostnamed_freeable, hostname_buf);
 557     HOSTNAME = hostname_buf;
 558     hostname1_set_hostname(hostnamed_interf, HOSTNAME);
 559
 560     return HOSTNAME;
 561 }
 562
 563 const gchar *
 564 our_get_static_hostname() {
 565
 566     if(STATIC_HOSTNAME && g_strcmp0(STATIC_HOSTNAME, ""))
 567         return STATIC_HOSTNAME;
 568     else if(HOSTNAME)
 569         return HOSTNAME;
 570
 571     return "localhost.home.network";
 572 }
 573
 574 const gchar *
 575 our_get_pretty_hostname() {
 576
 577     if(PRETTY_HOSTNAME)
 578         return PRETTY_HOSTNAME;
 579
 580     return "";
 581 }
 582
 583 const gchar *
 584 our_get_chassis() {
 585
 586    if(CHASSIS)
 587         return CHASSIS;
 588
 589     return "desktop"; /* this leads to the most generic beheivor in the unlikely case its returned */
 590 }
 591
 592 const gchar *
 593 our_get_icon_name() {
 594
 595     if(ICON)
 596         return ICON;
 597
 598     return "";
 599 }
 600
 601 const gchar *
 602 our_get_kernel_name() {
 603
 604     if(KERN_NAME)
 605         return KERN_NAME;
 606
 607     return "";
 608 }
 609
 610 const gchar *
 611 our_get_kernel_version() {
 612
 613     if(KERN_VERS)
 614         return KERN_VERS;
 615
 616     return "";
 617 }
 618
 619 const gchar *
 620 our_get_kernel_release() {
 621
 622     if(KERN_RELEASE)
 623         return KERN_RELEASE;
 624
 625     return "";
 626 }
 627
 628 const gchar *
 629 our_get_os_cpename() {
 630
 631     /* XXX needs to parse /etc/os-release (fallback to /usr/local/lib/os-release) */
 632     return "";
 633 }
 634
 635 const gchar *
 636 our_get_os_pretty_name() {
 637
 638     return "OpenBSD";
 639 }
 640
 641 /* --- end method/property/dbus signal code, begin bus/name handlers --- */
 642
 643 static void hostnamed_on_bus_acquired(GDBusConnection *conn,
 644                                       const gchar *name,
 645                                       gpointer user_data) {
 646
 647     g_printf("got bus/name, exporting %s's interface...\n", name);
 648
 649     hostnamed_interf = hostname1_skeleton_new();
 650
 651     /* attach function pointers to generated struct's method handlers */
 652     g_signal_connect(hostnamed_interf, "handle-set-hostname", G_CALLBACK(on_handle_set_hostname), NULL);
 653     g_signal_connect(hostnamed_interf, "handle-set-static-hostname", G_CALLBACK(on_handle_set_static_hostname), NULL);
 654     g_signal_connect(hostnamed_interf, "handle-set-pretty-hostname", G_CALLBACK(on_handle_set_pretty_hostname), NULL);
 655     g_signal_connect(hostnamed_interf, "handle-set-chassis", G_CALLBACK(on_handle_set_chassis), NULL);
 656     g_signal_connect(hostnamed_interf, "handle-set-icon-name", G_CALLBACK(on_handle_set_icon_name), NULL);
 657
 658     /* set our properties before export */
 659     hostname1_set_hostname(hostnamed_interf, our_get_hostname());
 660     hostname1_set_static_hostname(hostnamed_interf, our_get_static_hostname());
 661     hostname1_set_pretty_hostname(hostnamed_interf, our_get_pretty_hostname());
 662     hostname1_set_chassis(hostnamed_interf, our_get_chassis());
 663     hostname1_set_icon_name(hostnamed_interf, our_get_icon_name());
 664     hostname1_set_kernel_name(hostnamed_interf, our_get_kernel_name());
 665     hostname1_set_kernel_version(hostnamed_interf, our_get_kernel_version());
 666     hostname1_set_kernel_release(hostnamed_interf, our_get_kernel_release());
 667     hostname1_set_operating_system_cpename(hostnamed_interf, our_get_os_cpename());
 668     hostname1_set_operating_system_pretty_name(hostnamed_interf, our_get_os_pretty_name());
 669
 670     if(!g_dbus_interface_skeleton_export(G_DBUS_INTERFACE_SKELETON(hostnamed_interf),
 671                                          conn,
 672                                          "/org/freedesktop/hostname1",
 673                                          NULL)) {
 674
 675         g_printf("failed to export %s's interface!\n", name); /* unusual edge case, TODO check errno */
 676         hostnamed_mem_clean();
 677
 678     } else {
 679
 680         dbus_interface_exported = TRUE;
 681         g_printf("exported %s's interface on the system bus...\n", name);
 682     }
 683 }
 684
 685 static void hostnamed_on_name_acquired(GDBusConnection *conn,
 686                                        const gchar *name,
 687                                        gpointer user_data) {
 688
 689     g_printf("success!\n");
 690 }
 691
 692 static void hostnamed_on_name_lost(GDBusConnection *conn,
 693                                    const gchar *name,
 694                                    gpointer user_data) {
 695
 696     if(!conn) {
 697
 698         g_printf("failed to connect to the system bus while trying to acquire name '%s': either dbus-daemon isn't running or we don't have permission to push names and/or their interfaces to it.\n", name);
 699         hostnamed_mem_clean();
 700     }
 701
 702     g_printf("lost name %s, exiting...\n", name);
 703
 704     hostnamed_mem_clean();
 705 }
 706
 707 /* --- end bus/name handlers, begin misc unix functions --- */
 708
 709 /* safe call to clean and then exit
 710  * this stops our GMainLoop safely before letting main() return */
 711 void hostnamed_mem_clean() {
 712
 713     g_printf("exiting...\n");
 714
 715     if(dbus_interface_exported)
 716         g_dbus_interface_skeleton_unexport(G_DBUS_INTERFACE_SKELETON(hostnamed_interf));
 717
 718     if(g_main_loop_is_running(hostnamed_loop))
 719         g_main_loop_quit(hostnamed_loop);
 720
 721 }
 722
 723 /* wrapper for glib's unix signal handling; called only once if terminatating signal is raised against us */
 724 gboolean unix_sig_terminate_handler(gpointer data) {
 725
 726     g_printf("caught SIGINT/HUP/TERM, exiting\n");
 727
 728     hostnamed_mem_clean();
 729     return G_SOURCE_REMOVE;
 730 }
 731
 732 void set_signal_handlers() {
 733
 734     /* we don't care about its descriptor, we never need to unregister these */
 735     g_unix_signal_add(SIGINT,  unix_sig_terminate_handler, NULL);
 736     g_unix_signal_add(SIGHUP,  unix_sig_terminate_handler, NULL);
 737     g_unix_signal_add(SIGTERM, unix_sig_terminate_handler, NULL);
 738
 739     /* TODO: the "only once" guarantee only counts towards specific signals.
 740      *       make sure calling a SIGINT and SIGHUP doesn't cause term_handler()
 741      *       to be called twice */
 742 }
 743
 744 int main() {
 745
 746     hostnamed_freeable = g_ptr_array_new();
 747
 748     /* TODO: check for valid, writable config at init. if no, complain to `make install` */
 749
 750     CHASSIS = ICON = OS_CPENAME = 0;
 751     KERN_NAME = KERN_RELEASE = KERN_VERS = 0;
 752     HOSTNAME = STATIC_HOSTNAME = PRETTY_HOSTNAME = NULL;
 753
 754     set_signal_handlers();
 755
 756     if(!determine_chassis_and_icon() || !set_uname_properties() || !set_names())
 757         return 1;
 758
 759     hostnamed_loop = g_main_loop_new(NULL, TRUE);
 760
 761     bus_descriptor = g_bus_own_name(G_BUS_TYPE_SYSTEM,
 762                                    "org.freedesktop.hostname1",
 763                                    G_BUS_NAME_OWNER_FLAGS_NONE,
 764                                    hostnamed_on_bus_acquired,
 765                                    hostnamed_on_name_acquired,
 766                                    hostnamed_on_name_lost,
 767                                    NULL,
 768                                    NULL);
 769
 770     g_main_loop_run(hostnamed_loop);
 771     /* runs until single g_main_loop_quit() call is raised inside <interface>_mem_clean() */
 772     g_main_loop_unref(hostnamed_loop);
 773
 774     /* guaranteed unownable */
 775     g_bus_unown_name(bus_descriptor);
 776
 777     /* at this point no operations can occur with our data, it is safe to free it + its container */
 778     g_ptr_array_free(hostnamed_freeable, TRUE);
 779
 780     return 0;
 781 }
 782
 783 gboolean set_names() {
 784
 785     /* (1) set up */
 786     gchar *hostname_buf, *static_hostname_buf, *pretty_hostname_buf;
 787     GKeyFile *config;
 788     size_t hostname_divider;
 789
 790     hostname_buf        = (gchar*) g_malloc0(MAXHOSTNAMELEN);
 791     static_hostname_buf = (gchar*) g_malloc0(4096);
 792     pretty_hostname_buf = (gchar*) g_malloc0(4096);
 793
 794     config = g_key_file_new();
 795
 796     g_ptr_array_add(hostnamed_freeable, hostname_buf);
 797     g_ptr_array_add(hostnamed_freeable, static_hostname_buf);
 798     g_ptr_array_add(hostnamed_freeable, pretty_hostname_buf);
 799
 800     /* (2) set HOSTNAME */
 801     if(gethostname(hostname_buf, MAXHOSTNAMELEN) || !g_strcmp0(hostname_buf, ""))
 802         HOSTNAME = "localhost";
 803
 804     HOSTNAME = hostname_buf;
 805
 806     /* this bit gets you the /etc/myname style hostname
 807     hostname_divider = strcspn(hostname_buf, ".");
 808     strncpy(ret, hostname_buf, hostname_divider); */
 809
 810     /* (3) set PRETTY_HOSTNAME */
 811     if(g_key_file_load_from_file(config, "/etc/machine-info", G_KEY_FILE_NONE, NULL)
 812         && (pretty_hostname_buf = g_key_file_get_value(config, "hostnamed", "PRETTY_HOSTNAME", NULL)))
 813         PRETTY_HOSTNAME = pretty_hostname_buf;
 814     else
 815         PRETTY_HOSTNAME = "";
 816
 817     if(config)
 818         g_key_file_unref(config);
 819
 820     /* (4) set STATIC_HOSTNAME */
 821     if(!g_strcmp0(PRETTY_HOSTNAME, ""))
 822         STATIC_HOSTNAME = HOSTNAME;
 823
 824     else if((static_hostname_buf = g_hostname_to_ascii(PRETTY_HOSTNAME)))
 825         STATIC_HOSTNAME = static_hostname_buf;
 826
 827     return (HOSTNAME && STATIC_HOSTNAME && PRETTY_HOSTNAME) ? TRUE : FALSE;
 828
 829 }
 830
 831 gboolean set_uname_properties() {
 832
 833     struct utsname un;
 834
 835     if(-1 == uname(&un))
 836         return FALSE;
 837
 838     KERN_NAME = (gchar*)g_malloc0(sizeof(un.sysname));
 839     g_ptr_array_add(hostnamed_freeable, KERN_NAME);
 840     g_strlcpy(KERN_NAME, un.sysname, sizeof(un.sysname));
 841
 842     KERN_RELEASE = (gchar*)g_malloc0(sizeof(un.release));
 843     g_ptr_array_add(hostnamed_freeable, KERN_RELEASE);
 844     g_strlcpy(KERN_RELEASE, un.release, sizeof(un.release));
 845
 846     KERN_VERS = (gchar*)g_malloc0(sizeof(un.version));
 847     g_ptr_array_add(hostnamed_freeable, KERN_VERS);
 848     g_strlcpy(KERN_VERS, un.version, sizeof(un.version));
 849
 850     return TRUE;
 851 }
 852
 853 gboolean determine_chassis_and_icon() {
 854
 855     const size_t bufsize = 4096;
 856
 857     char *hwproduct, *hwmodel, *hwvendor, *hwmachine;
 858     size_t hwproduct_size, hwmodel_size, hwvendor_size, hwmachine_size;
 859     int hwproduct_name[2], hwmodel_name[2], hwvendor_name[2], hwmachine_name[2];
 860     unsigned int i;
 861     gboolean UNSURE_CHASSIS_FLAG, UNSURE_ICON_FLAG;
 862
 863     hwproduct_size = hwmodel_size = hwvendor_size = hwmachine_size = bufsize;
 864     UNSURE_CHASSIS_FLAG = UNSURE_ICON_FLAG = FALSE;
 865     i = 0;
 866
 867     hwproduct = (char*)g_malloc0(4096);
 868     hwmodel   = (char*)g_malloc0(4096);
 869     hwvendor  = (char*)g_malloc0(4096);
 870     hwmachine = (char*)g_malloc0(4096);
 871
 872     g_ptr_array_add(hostnamed_freeable, hwproduct);
 873     g_ptr_array_add(hostnamed_freeable, hwmodel);
 874     g_ptr_array_add(hostnamed_freeable, hwvendor);
 875     g_ptr_array_add(hostnamed_freeable, hwmachine);
 876
 877     hwproduct_name[0] = CTL_HW;
 878     hwproduct_name[1] = HW_PRODUCT;
 879
 880     hwmodel_name[0] = CTL_HW;
 881     hwmodel_name[1] = HW_MODEL;
 882
 883     hwvendor_name[0] = CTL_HW;
 884     hwvendor_name[1] = HW_VENDOR;
 885
 886     hwmachine_name[0] = CTL_HW;
 887     hwmachine_name[1] = HW_MACHINE;
 888
 889     /* pass NULL buffer to check size first, then pass hw to be filled according to freshly-set hw_size */
 890     if(-1 == sysctl(hwproduct_name, 2, NULL, &hwproduct_size, NULL, 0) || -1 == sysctl(hwproduct_name, 2, hwproduct, &hwproduct_size, NULL, 0))
 891         return FALSE;
 892
 893     if(-1 == sysctl(hwmodel_name, 2, NULL, &hwmodel_size, NULL, 0) || -1 == sysctl(hwmodel_name, 2, hwmodel, &hwmodel_size, NULL, 0))
 894         return FALSE;
 895
 896     if(-1 == sysctl(hwvendor_name, 2, NULL, &hwvendor_size, NULL, 0) || -1 == sysctl(hwvendor_name, 2, hwvendor, &hwvendor_size, NULL, 0))
 897         return FALSE;
 898
 899     if(-1 == sysctl(hwmachine_name, 2, NULL, &hwmachine_size, NULL, 0) || -1 == sysctl(hwmachine_name, 2, hwmachine, &hwmachine_size, NULL, 0))
 900         return FALSE;
 901
 902     /* TODO: test for laptop, if not, dmidecode for desktop vs. server
 903      *       probably move this code to vm test func and set a global after running it early, once */
 904
 905     for(; i < G_N_ELEMENTS(chassis_indicator_table); i++) {
 906         if(strcasestr(hwproduct,    chassis_indicator_table[i].match_string)
 907             || strcasestr(hwmodel,  chassis_indicator_table[i].match_string)
 908             || strcasestr(hwvendor, chassis_indicator_table[i].match_string)) {
 909
 910             if(!UNSURE_CHASSIS_FLAG && chassis_indicator_table[i].chassis) {
 911
 912                 UNSURE_CHASSIS_FLAG = chassis_indicator_table[i].chassis_precedence;
 913                 CHASSIS = chassis_indicator_table[i].chassis;
 914             }
 915
 916             if(!UNSURE_ICON_FLAG && chassis_indicator_table[i].icon) {
 917
 918                 UNSURE_ICON_FLAG = chassis_indicator_table[i].icon_precedence;
 919                 ICON = chassis_indicator_table[i].icon;
 920             }
 921         }
 922     }
 923
 924     if(up_native_is_laptop()) {
 925
 926         if(!CHASSIS)
 927             CHASSIS = "laptop";
 928         if(!ICON)
 929             ICON = "input-touchpad"; /* TODO pull an icon package that actually has the icons we're looking for */
 930
 931     } else if(is_server(hwmachine)) {
 932
 933         if(!CHASSIS)
 934             CHASSIS = "server";
 935         if(!ICON)
 936             ICON = "uninterruptible-power-supply";
 937
 938     } else if(!CHASSIS || !ICON) {
 939
 940         if(!CHASSIS)
 941             CHASSIS = "desktop";
 942         if(!ICON)
 943             ICON = "computer";
 944     }
 945
 946     return (CHASSIS && ICON);
 947 }
 948
 949 gboolean is_server(gchar *arch) {
 950
 951     unsigned int i;
 952
 953     for(; i < G_N_ELEMENTS(server_archs); i++)
 954         if(strcasestr(arch, server_archs[i]))
 955             return TRUE;
 956
 957     return FALSE;
 958 }
 959
 960 gboolean up_native_is_laptop() {
 961
 962     struct apm_power_info bstate;
 963     struct sensordev acpiac;
 964
 965     if (up_native_get_sensordev("acpiac0", &acpiac))
 966         return TRUE;
 967
 968     if (-1 == ioctl(up_apm_get_fd(), APM_IOC_GETPOWER, &bstate))
 969         g_error("ioctl on apm fd failed : %s", g_strerror(errno));
 970
 971     return bstate.ac_state != APM_AC_UNKNOWN;
 972 }
 973
 974 int up_apm_get_fd() {
 975
 976     static int apm_fd = 0;
 977
 978     if(apm_fd == 0) {
 979
 980         g_debug("apm_fd is not initialized yet, opening");
 981
 982         /* open /dev/apm */
 983         if((apm_fd = open("/dev/apm", O_RDONLY)) == -1) {
 984             if(errno != ENXIO && errno != ENOENT)
 985                 g_error("cannot open device file");
 986         }
 987     }
 988
 989     return apm_fd;
 990 }
 991
 992 gboolean up_native_get_sensordev(const char * id, struct sensordev * snsrdev) {
 993
 994     int devn;
 995     size_t sdlen = sizeof(struct sensordev);
 996     int mib[] = {CTL_HW, HW_SENSORS, 0, 0 ,0};
 997
 998     for (devn = 0 ; ; devn++) {
 999         mib[2] = devn;
1000         if(sysctl(mib, 3, snsrdev, &sdlen, NULL, 0) == -1) {
1001             if(errno == ENXIO)
1002                 continue;
1003             if(errno == ENOENT)
1004                 break;
1005         }
1006
1007         if (!strcmp(snsrdev->xname, id))
1008             return TRUE;
1009     }
1010
1011     return FALSE;
1012 }
1013
1014 static gboolean is_valid_chassis_type(gchar *test) {
1015
1016     if(!g_strcmp0(test, "desktop") ||
1017        !g_strcmp0(test, "laptop") ||
1018        !g_strcmp0(test, "server") ||
1019        !g_strcmp0(test, "tablet") ||
1020        !g_strcmp0(test, "handset") ||
1021        !g_strcmp0(test, "vm") ||
1022        !g_strcmp0(test, "container") ||
1023        !g_strcmp0(test, ""))
1024         return TRUE;
1025
1026     return FALSE;
1027 }