make *name prop. getters operate similarly to others
[systembsd.git] / 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 <gtk/gtk.h> */
33
34 #include "hostnamed-gen.h"
35 #include "hostnamed.h"
36
37 /* format: {
38 * (1) string to be matched against runtime machine's sysctl output.
39 * can be either the exact string or a substring contained
40 * within sysctl strings. no "guesses" here, a match should
41 * reliably indicate the chassis/icon.
42 *
43 * (2) string describing chassis type divulged by (1).
44 * must be one of "desktop", "laptop", "server",
45 * "tablet", "handset", "vm", "container" or NULL
46 * if only icon string can be ascertained. "vm" refers
47 * to operating systems running on baremetal hypervisors
48 * (hardware virtualization, like XEN) while "container"
49 * refers to OSs running on shared hypervisors like
50 * virtualbox or VMware. consider the distinction carefully
51 * as common virtualization software like KVM may share
52 * characteristics of both "vm" and "container" types.
53 *
54 * (3) string specifying icon to use. follows XDG icon spec.
55 * see http://standards.freedesktop.org/icon-naming-spec/icon-naming-spec-latest.html
56 * for allowed strings.
57 *
58 * (4) chassis precedence bit. TRUE if (2) is defined and
59 * we're certain it is the proper string. FALSE in the
60 * circumstance (2) may be the correct string, unless
61 * a match with this bit set to TRUE overrides it.
62 * if (2) is NULL, this bit is inconsequential.
63 *
64 * (5) icon precedence bit. see previous definition.
65 * } */
66 struct SYSCTL_LOOKUP_TABLE {
67 gchar *match_string;
68 gchar *chassis;
69 gchar *icon;
70 gboolean chassis_precedence;
71 gboolean icon_precedence;
72 };
73
74 GPtrArray *hostnamed_freeable;
75 Hostname1 *hostnamed_interf;
76
77 GMainLoop *hostnamed_loop;
78
79 guint bus_descriptor;
80 gboolean dbus_interface_exported; /* reliable because of gdbus operational guarantees */
81
82 gchar *HOSTNAME, *STATIC_HOSTNAME, *PRETTY_HOSTNAME;
83 gchar *CHASSIS, *ICON;
84 gchar *KERN_NAME, *KERN_RELEASE, *KERN_VERS, *OS_CPENAME;
85
86 /* TODO no specific vm or laptop icon in gnome
87 * NOTE paravirtualization on xen is only available for linuxes right now
88 * dmesg on linux systems reveals xen and virtualization method (HVM or PVM)
89 * but we will worry about those later */
90
91 /* add any sysctl strings that suggest virtualization here */
92 const struct SYSCTL_LOOKUP_TABLE chassis_indicator_table[] =
93 {
94 { "QEMU Virtual CPU", "vm", NULL, FALSE, FALSE }, /* could be QEMU running in userspace or as part of KVM */
95 { "KVM", "vm", "drive-multidisk", FALSE, FALSE },
96 { "SmartDC HVM", "vm", "drive-multidisk", TRUE, TRUE }, /* illumos-joyent kvm */
97 { "VirtualBox", "vm", "drive-multidisk", TRUE, TRUE },
98 { "VMware, Inc.", "vm", "drive-multidisk", TRUE, TRUE },
99 { "VMware Virtual Platform", "vm", "drive-multidisk", TRUE, TRUE },
100 { "Parallels", "vm", "drive-multidisk", TRUE, TRUE }, /* need verification */
101 { "Xen", "vm", "drive-multidisk", FALSE, FALSE }
102 }; /* TODO: chroots, etc. are the actual "containers", add them */
103
104 /* archs to check against when determining if machine is server */
105 const gchar *server_archs[] = {
106 "hppa",
107 "sparc",
108 "sparc64"
109 };
110
111 /* --- begin method/property/dbus signal code --- */
112
113 /* TODO the extra boolean passed to these funcs is for policykit auth */
114 /* TODO complete call with error, message, etc */
115 static gboolean
116 on_handle_set_hostname(Hostname1 *hn1_passed_interf,
117 GDBusMethodInvocation *invoc,
118 const gchar *greet,
119 gpointer data) {
120 GVariant *params;
121 gchar *proposed_hostname, *valid_hostname_buf;
122 gboolean policykit_auth, ret;
123 size_t check_length, bad_length;
124
125 bad_length = MAXHOSTNAMELEN + 1;
126 proposed_hostname = NULL;
127 ret = FALSE;
128
129 params = g_dbus_method_invocation_get_parameters(invoc);
130 g_variant_get(params, "(sb)", &proposed_hostname, &policykit_auth);
131
132 if(proposed_hostname && (valid_hostname_buf = g_hostname_to_ascii(proposed_hostname))) {
133
134 check_length = strnlen(proposed_hostname, bad_length);
135
136 if(check_length < bad_length && !sethostname(proposed_hostname, check_length))
137 ret = TRUE;
138 }
139
140 hostname1_complete_set_hostname(hn1_passed_interf, invoc);
141
142 if(proposed_hostname)
143 g_free(proposed_hostname);
144 if(valid_hostname_buf)
145 g_free(valid_hostname_buf);
146
147 return ret;
148 }
149
150 static gboolean
151 on_handle_set_static_hostname(Hostname1 *hn1_passed_interf,
152 GDBusMethodInvocation *invoc,
153 const gchar *greet,
154 gpointer data) {
155 return FALSE;
156 }
157
158 static gboolean
159 on_handle_set_pretty_hostname(Hostname1 *hn1_passed_interf,
160 GDBusMethodInvocation *invoc,
161 const gchar *greet,
162 gpointer data) {
163 return FALSE;
164 }
165
166 static gboolean
167 on_handle_set_chassis(Hostname1 *hn1_passed_interf,
168 GDBusMethodInvocation *invoc,
169 const gchar *greet,
170 gpointer data) {
171 return FALSE;
172 }
173
174 static gboolean
175 on_handle_set_icon_name(Hostname1 *hn1_passed_interf,
176 GDBusMethodInvocation *invoc,
177 const gchar *greet,
178 gpointer data) {
179 return FALSE;
180 }
181
182 /* note: all hostnamed/hostname1's properties are read-only,
183 * and do not need set_ functions, gdbus-codegen realized
184 * this from the XML and handled the to-be error of trying
185 * to set a read-only property's value
186 */
187
188 const gchar *
189 our_get_hostname() {
190
191 /*gchar *hostname_buf, *ret;
192 size_t hostname_divider;
193
194 hostname_buf = (gchar*) g_malloc0(MAXHOSTNAMELEN);
195 ret = (gchar*) g_malloc0(MAXHOSTNAMELEN);
196
197 g_ptr_array_add(hostnamed_freeable, hostname_buf);
198 g_ptr_array_add(hostnamed_freeable, ret);
199
200 if(gethostname(hostname_buf, MAXHOSTNAMELEN) || g_strcmp0(hostname_buf, "") == 0)
201 return "localhost";
202
203 hostname_divider = strcspn(hostname_buf, ".");
204
205 return strncpy(ret, hostname_buf, hostname_divider);*/
206
207 if(HOSTNAME)
208 return HOSTNAME;
209
210 return "localhost";
211 }
212
213 const gchar *
214 our_get_static_hostname() {
215
216 /*const gchar *pretty_hostname;
217 const gchar *ret;
218
219 pretty_hostname = our_get_pretty_hostname();
220
221 if(g_strcmp0(pretty_hostname, "") == 0)
222 ret = our_get_hostname();
223
224 else if((ret = g_hostname_to_ascii(pretty_hostname))) {
225
226 g_ptr_array_add(hostnamed_freeable, (gpointer)ret);
227 return ret;
228 }
229
230 return ret;*/
231
232 if(STATIC_HOSTNAME)
233 return STATIC_HOSTNAME;
234 else if(HOSTNAME)
235 return HOSTNAME;
236
237 return "localhost";
238 }
239
240 const gchar *
241 our_get_pretty_hostname() {
242
243 /*GKeyFile *config;
244 gchar *ret;
245
246 config = g_key_file_new();
247
248 if(g_key_file_load_from_file(config, "/etc/systemd_compat.conf", G_KEY_FILE_NONE, NULL)
249 && (ret = g_key_file_get_value(config, "hostnamed", "PrettyHostname", NULL))) { ret might need to be freed, docs dont specify but i am suspicious
250
251 g_key_file_unref(config);
252 return ret;
253 }
254
255 if(config)
256 g_free(config);
257
258 return "";*/
259
260 if(PRETTY_HOSTNAME)
261 return PRETTY_HOSTNAME;
262
263 return "";
264 }
265
266 const gchar *
267 our_get_chassis() {
268
269 if(CHASSIS)
270 return CHASSIS;
271
272 return "desktop"; /* this leads to the most generic beheivor in the unlikely case its returned */
273 }
274
275 const gchar *
276 our_get_icon_name() {
277
278 if(ICON)
279 return ICON;
280
281 return "";
282 }
283
284 const gchar *
285 our_get_kernel_name() {
286
287 if(KERN_NAME)
288 return KERN_NAME;
289
290 return "";
291 }
292
293 const gchar *
294 our_get_kernel_version() {
295
296 if(KERN_VERS)
297 return KERN_VERS;
298
299 return "";
300 }
301
302 const gchar *
303 our_get_kernel_release() {
304
305 if(KERN_RELEASE)
306 return KERN_RELEASE;
307
308 return "";
309 }
310
311 const gchar *
312 our_get_os_cpename() {
313
314 return "ONEDAY";
315 }
316
317 const gchar *
318 our_get_os_pretty_name() {
319
320 return "OpenBSD";
321 }
322
323 /* --- end method/property/dbus signal code, begin bus/name handlers --- */
324
325 static void hostnamed_on_bus_acquired(GDBusConnection *conn,
326 const gchar *name,
327 gpointer user_data) {
328
329 g_printf("got bus/name, exporting %s's interface...\n", name);
330
331 hostnamed_interf = hostname1_skeleton_new();
332
333 /* attach function pointers to generated struct's method handlers */
334 g_signal_connect(hostnamed_interf, "handle-set-hostname", G_CALLBACK(on_handle_set_hostname), NULL);
335 g_signal_connect(hostnamed_interf, "handle-set-static-hostname", G_CALLBACK(on_handle_set_static_hostname), NULL);
336 g_signal_connect(hostnamed_interf, "handle-set-pretty-hostname", G_CALLBACK(on_handle_set_pretty_hostname), NULL);
337 g_signal_connect(hostnamed_interf, "handle-set-chassis", G_CALLBACK(on_handle_set_chassis), NULL);
338 g_signal_connect(hostnamed_interf, "handle-set-icon-name", G_CALLBACK(on_handle_set_icon_name), NULL);
339
340 /* set our properties before export */
341 hostname1_set_hostname(hostnamed_interf, our_get_hostname());
342 hostname1_set_static_hostname(hostnamed_interf, our_get_static_hostname());
343 hostname1_set_pretty_hostname(hostnamed_interf, our_get_pretty_hostname());
344 hostname1_set_chassis(hostnamed_interf, our_get_chassis());
345 hostname1_set_icon_name(hostnamed_interf, our_get_icon_name());
346 hostname1_set_kernel_name(hostnamed_interf, our_get_kernel_name());
347 hostname1_set_kernel_version(hostnamed_interf, our_get_kernel_version());
348 hostname1_set_kernel_release(hostnamed_interf, our_get_kernel_release());
349 hostname1_set_operating_system_cpename(hostnamed_interf, our_get_os_cpename());
350 hostname1_set_operating_system_pretty_name(hostnamed_interf, our_get_os_pretty_name());
351
352 if(!g_dbus_interface_skeleton_export(G_DBUS_INTERFACE_SKELETON(hostnamed_interf),
353 conn,
354 "/org/freedesktop/hostname1",
355 NULL)) {
356
357 g_printf("failed to export %s's interface!\n", name); /* unusual edge case, TODO check errno */
358 hostnamed_mem_clean();
359
360 } else {
361
362 dbus_interface_exported = TRUE;
363 g_printf("exported %s's interface on the system bus...\n", name);
364 }
365 }
366
367 static void hostnamed_on_name_acquired(GDBusConnection *conn,
368 const gchar *name,
369 gpointer user_data) {
370
371 g_printf("success!\n");
372 }
373
374 static void hostnamed_on_name_lost(GDBusConnection *conn,
375 const gchar *name,
376 gpointer user_data) {
377
378 if(!conn) {
379
380 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);
381 hostnamed_mem_clean();
382 }
383
384 g_printf("lost name %s, exiting...\n", name);
385
386 hostnamed_mem_clean();
387 }
388
389 /* --- end bus/name handlers, begin misc unix functions --- */
390
391 /* safe call to clean and then exit
392 * this stops our GMainLoop safely before letting main() return */
393 void hostnamed_mem_clean() {
394
395 g_printf("exiting...\n");
396
397 if(dbus_interface_exported)
398 g_dbus_interface_skeleton_unexport(G_DBUS_INTERFACE_SKELETON(hostnamed_interf));
399
400 if(g_main_loop_is_running(hostnamed_loop))
401 g_main_loop_quit(hostnamed_loop);
402
403 }
404
405 /* wrapper for glib's unix signal handling; called only once if terminatating signal is raised against us */
406 gboolean unix_sig_terminate_handler(gpointer data) {
407
408 g_printf("caught SIGINT/HUP/TERM, exiting\n");
409
410 hostnamed_mem_clean();
411 return G_SOURCE_REMOVE;
412 }
413
414 void set_signal_handlers() {
415
416 /* we don't care about its descriptor, we never need to unregister these */
417 g_unix_signal_add(SIGINT, unix_sig_terminate_handler, NULL);
418 g_unix_signal_add(SIGHUP, unix_sig_terminate_handler, NULL);
419 g_unix_signal_add(SIGTERM, unix_sig_terminate_handler, NULL);
420
421 /* TODO: the "only once" guarantee only counts towards specific signals.
422 * make sure calling a SIGINT and SIGHUP doesn't cause term_handler()
423 * to be called twice */
424 }
425
426 int main() {
427
428 hostnamed_freeable = g_ptr_array_new();
429
430 /* TODO: check for valid, writable config at init. if no, complain to `make install` */
431
432 CHASSIS = ICON = OS_CPENAME = 0;
433 KERN_NAME = KERN_RELEASE = KERN_VERS = 0;
434
435 set_signal_handlers();
436
437 if(!determine_chassis_and_icon() || !set_uname_properties())
438 return 1;
439
440 hostnamed_loop = g_main_loop_new(NULL, TRUE);
441
442 bus_descriptor = g_bus_own_name(G_BUS_TYPE_SYSTEM,
443 "org.freedesktop.hostname1",
444 G_BUS_NAME_OWNER_FLAGS_NONE,
445 hostnamed_on_bus_acquired,
446 hostnamed_on_name_acquired,
447 hostnamed_on_name_lost,
448 NULL,
449 NULL);
450
451 g_main_loop_run(hostnamed_loop);
452 /* runs until single g_main_loop_quit() call is raised inside <interface>_mem_clean() */
453 g_main_loop_unref(hostnamed_loop);
454
455 /* guaranteed unownable */
456 g_bus_unown_name(bus_descriptor);
457
458 /* at this point no operations can occur with our data, it is safe to free it + its container */
459 g_ptr_array_free(hostnamed_freeable, TRUE);
460
461 return 0;
462 }
463
464 gboolean set_uname_properties() {
465
466 struct utsname un;
467
468 if(-1 == uname(&un))
469 return FALSE;
470
471 KERN_NAME = (gchar*)g_malloc0(sizeof(un.sysname));
472 g_ptr_array_add(hostnamed_freeable, KERN_NAME);
473 g_strlcpy(KERN_NAME, un.sysname, sizeof(un.sysname));
474
475 KERN_RELEASE = (gchar*)g_malloc0(sizeof(un.release));
476 g_ptr_array_add(hostnamed_freeable, KERN_RELEASE);
477 g_strlcpy(KERN_RELEASE, un.release, sizeof(un.release));
478
479 KERN_VERS = (gchar*)g_malloc0(sizeof(un.version));
480 g_ptr_array_add(hostnamed_freeable, KERN_VERS);
481 g_strlcpy(KERN_VERS, un.version, sizeof(un.version));
482
483 return TRUE;
484 }
485
486 gboolean determine_chassis_and_icon() {
487
488 const size_t bufsize = 4096;
489
490 char *hwproduct, *hwmodel, *hwvendor, *hwmachine;
491 size_t hwproduct_size, hwmodel_size, hwvendor_size, hwmachine_size;
492 int hwproduct_name[2], hwmodel_name[2], hwvendor_name[2], hwmachine_name[2];
493 unsigned int i;
494 gboolean UNSURE_CHASSIS_FLAG, UNSURE_ICON_FLAG;
495
496 hwproduct_size = hwmodel_size = hwvendor_size = hwmachine_size = bufsize;
497 UNSURE_CHASSIS_FLAG = UNSURE_ICON_FLAG = FALSE;
498 i = 0;
499
500 hwproduct = (char*)g_malloc0(4096);
501 hwmodel = (char*)g_malloc0(4096);
502 hwvendor = (char*)g_malloc0(4096);
503 hwmachine = (char*)g_malloc0(4096);
504
505 g_ptr_array_add(hostnamed_freeable, hwproduct);
506 g_ptr_array_add(hostnamed_freeable, hwmodel);
507 g_ptr_array_add(hostnamed_freeable, hwvendor);
508 g_ptr_array_add(hostnamed_freeable, hwmachine);
509
510 hwproduct_name[0] = CTL_HW;
511 hwproduct_name[1] = HW_PRODUCT;
512
513 hwmodel_name[0] = CTL_HW;
514 hwmodel_name[1] = HW_MODEL;
515
516 hwvendor_name[0] = CTL_HW;
517 hwvendor_name[1] = HW_VENDOR;
518
519 hwmachine_name[0] = CTL_HW;
520 hwmachine_name[1] = HW_MACHINE;
521
522 /* pass NULL buffer to check size first, then pass hw to be filled according to freshly-set hw_size */
523 if(-1 == sysctl(hwproduct_name, 2, NULL, &hwproduct_size, NULL, 0) || -1 == sysctl(hwproduct_name, 2, hwproduct, &hwproduct_size, NULL, 0))
524 return FALSE;
525
526 if(-1 == sysctl(hwmodel_name, 2, NULL, &hwmodel_size, NULL, 0) || -1 == sysctl(hwmodel_name, 2, hwmodel, &hwmodel_size, NULL, 0))
527 return FALSE;
528
529 if(-1 == sysctl(hwvendor_name, 2, NULL, &hwvendor_size, NULL, 0) || -1 == sysctl(hwvendor_name, 2, hwvendor, &hwvendor_size, NULL, 0))
530 return FALSE;
531
532 if(-1 == sysctl(hwmachine_name, 2, NULL, &hwmachine_size, NULL, 0) || -1 == sysctl(hwmachine_name, 2, hwmachine, &hwmachine_size, NULL, 0))
533 return FALSE;
534
535 /* TODO: test for laptop, if not, dmidecode for desktop vs. server
536 * probably move this code to vm test func and set a global after running it early, once */
537
538 for(; i < G_N_ELEMENTS(chassis_indicator_table); i++) {
539 if(strcasestr(hwproduct, chassis_indicator_table[i].match_string)
540 || strcasestr(hwmodel, chassis_indicator_table[i].match_string)
541 || strcasestr(hwvendor, chassis_indicator_table[i].match_string)) {
542
543 if(!UNSURE_CHASSIS_FLAG && chassis_indicator_table[i].chassis) {
544
545 UNSURE_CHASSIS_FLAG = chassis_indicator_table[i].chassis_precedence;
546 CHASSIS = chassis_indicator_table[i].chassis;
547 }
548
549 if(!UNSURE_ICON_FLAG && chassis_indicator_table[i].icon) {
550
551 UNSURE_ICON_FLAG = chassis_indicator_table[i].icon_precedence;
552 ICON = chassis_indicator_table[i].icon;
553 }
554 }
555 }
556
557 if(up_native_is_laptop()) {
558
559 if(!CHASSIS)
560 CHASSIS = "laptop";
561 if(!ICON)
562 ICON = "input-touchpad"; /* TODO pull an icon package that actually has the icons we're looking for */
563
564 } else if(is_server(hwmachine)) {
565
566 if(!CHASSIS)
567 CHASSIS = "server";
568 if(!ICON)
569 ICON = "uninterruptible-power-supply";
570
571 } else if(!CHASSIS || !ICON) {
572
573 if(!CHASSIS)
574 CHASSIS = "desktop";
575 if(!ICON)
576 ICON = "computer";
577 }
578
579 return (CHASSIS && ICON);
580 }
581
582 gboolean is_server(gchar *arch) {
583
584 unsigned int i;
585
586 for(; i < G_N_ELEMENTS(server_archs); i++)
587 if(strcasestr(arch, server_archs[i]))
588 return TRUE;
589
590 return FALSE;
591 }
592
593 gboolean up_native_is_laptop() {
594
595 struct apm_power_info bstate;
596 struct sensordev acpiac;
597
598 if (up_native_get_sensordev("acpiac0", &acpiac))
599 return TRUE;
600
601 if (-1 == ioctl(up_apm_get_fd(), APM_IOC_GETPOWER, &bstate))
602 g_error("ioctl on apm fd failed : %s", g_strerror(errno));
603
604 return bstate.ac_state != APM_AC_UNKNOWN;
605 }
606
607 int up_apm_get_fd() {
608
609 static int apm_fd = 0;
610
611 if(apm_fd == 0) {
612
613 g_debug("apm_fd is not initialized yet, opening");
614
615 /* open /dev/apm */
616 if((apm_fd = open("/dev/apm", O_RDONLY)) == -1) {
617 if(errno != ENXIO && errno != ENOENT)
618 g_error("cannot open device file");
619 }
620 }
621
622 return apm_fd;
623 }
624
625 gboolean up_native_get_sensordev(const char * id, struct sensordev * snsrdev) {
626
627 int devn;
628 size_t sdlen = sizeof(struct sensordev);
629 int mib[] = {CTL_HW, HW_SENSORS, 0, 0 ,0};
630
631 for (devn = 0 ; ; devn++) {
632 mib[2] = devn;
633 if(sysctl(mib, 3, snsrdev, &sdlen, NULL, 0) == -1) {
634 if(errno == ENXIO)
635 continue;
636 if(errno == ENOENT)
637 break;
638 }
639
640 if (!strcmp(snsrdev->xname, id))
641 return TRUE;
642 }
643
644 return FALSE;
645 }