X-Git-Url: https://uglyman.kremlin.cc/gitweb/gitweb.cgi?a=blobdiff_plain;f=src%2Finterfaces%2Fhostnamed%2Fhostnamed.c;h=075a7f3f9ea70cbbfeda205f5282195e45f0f699;hb=962fad6dfccd01b88036754d968aa13798a142bc;hp=979df6e78da437ed16d09f7b034ef61137804ab8;hpb=3eb1ef91aacdff3ee2dda5973d844e2875297415;p=systembsd.git diff --git a/src/interfaces/hostnamed/hostnamed.c b/src/interfaces/hostnamed/hostnamed.c index 979df6e..075a7f3 100644 --- a/src/interfaces/hostnamed/hostnamed.c +++ b/src/interfaces/hostnamed/hostnamed.c @@ -19,6 +19,7 @@ #include #include +#include #include #include @@ -37,6 +38,16 @@ gboolean dbus_interface_exported; /* reliable because of gdbus operational guara /* --- begin method/property/dbus signal code --- */ +/* add any sysctl strings that suggest virtualization here */ +const gchar* vmstring_list[] = { + "QEMU Virtual CPU", + "SmartDC HVM", + "KVM", + "VirtualBox" +}; + +static gboolean is_vm; + static gboolean on_handle_set_hostname(Hostname1 *hn1_passed_interf, GDBusMethodInvocation *invoc, @@ -86,38 +97,91 @@ on_handle_set_icon_name(Hostname1 *hn1_passed_interf, const gchar * our_get_hostname() { - gchar *hostname_buf, *ret; - size_t hostname_divider; + gchar *hostname_buf, *ret; + size_t hostname_divider; + + hostname_buf = (gchar*) g_malloc0(MAXHOSTNAMELEN); + ret = (gchar*) g_malloc0(MAXHOSTNAMELEN); - hostname_buf = (gchar*) g_malloc0(MAXHOSTNAMELEN); /* todo check & free */ - ret = (gchar*) g_malloc0(MAXHOSTNAMELEN); - g_ptr_array_add(hostnamed_freeable, hostname_buf); - g_ptr_array_add(hostnamed_freeable, ret); + g_ptr_array_add(hostnamed_freeable, hostname_buf); + g_ptr_array_add(hostnamed_freeable, ret); - if(gethostname(hostname_buf, MAXHOSTNAMELEN)) - return ""; + if(gethostname(hostname_buf, MAXHOSTNAMELEN) || g_strcmp0(hostname_buf, "") == 0) + return "localhost"; - hostname_divider = strcspn(hostname_buf, "."); + hostname_divider = strcspn(hostname_buf, "."); - return strncpy(ret, hostname_buf, hostname_divider); + return strncpy(ret, hostname_buf, hostname_divider); } const gchar * our_get_static_hostname() { - return "TODO"; + const gchar *pretty_hostname; + const gchar *ret; + + pretty_hostname = our_get_pretty_hostname(); + + if(g_strcmp0(pretty_hostname, "") == 0) + ret = our_get_hostname(); + + else if((ret = g_hostname_to_ascii(pretty_hostname))) { + + g_ptr_array_add(hostnamed_freeable, (gpointer)ret); + return ret; + } + + return ret; } const gchar * our_get_pretty_hostname() { - return "TODO"; + GKeyFile *config; + gchar *ret; + + config = g_key_file_new(); + + if(g_key_file_load_from_file(config, "/etc/systemd_compat.conf", G_KEY_FILE_NONE, NULL) + && (ret = g_key_file_get_value(config, "hostnamed", "PrettyHostname", NULL))) { /* ret might need to be freed, docs dont specify but i am suspicious */ + + g_key_file_unref(config); + return ret; + } + + if(config) + g_free(config); + + return ""; } const gchar * our_get_chassis() { - return "TODO"; + char *hwproduct, *hwmodel; + size_t hwproduct_size, hwmodel_size; + int hwproduct_name[2], hwmodel_name[2]; + + hwproduct_name[0] = CTL_HW; + hwproduct_name[1] = HW_PRODUCT; + + hwmodel_name[0] = CTL_HW; + hwmodel_name[1] = HW_MODEL; + + /* pass NULL buffer to check size first, then pass hw to be filled according to freshly-set hw_size */ + if(sysctl(&hwproduct_name, 2, NULL, &hwproduct_size, NULL, 0) || sysctl(&hwproduct_name, 2, hwproduct, &hwproduct_size, NULL, 0)) + return "desktop"; /* TODO error properly here */ + + if(sysctl(&hwmodel_name, 2, NULL, &hwmodel_size, NULL, 0) || sysctl(&hwmodel_name, 2, hwmodel, &hwmodel_size, NULL, 0)) + return "desktop"; /* TODO error properly here */ + + if(test_against_known_vm_strings(hwproduct) || test_against_known_vm_strings(hwmodel)) + return "vm"; /*TODO differentiate between VMs (hardware virt, seperate kernel) and containers (paravirt, shared kernel) + + /* TODO: test for laptop, if not, dmidecode for desktop vs. server + * probably move this code to vm test func and set a global after running it early, once */ + + return "desktop"; } const gchar * @@ -164,7 +228,7 @@ static void hostnamed_on_bus_acquired(GDBusConnection *conn, g_printf("got bus/name, exporting %s's interface...\n", name); - hostnamed_interf = hostname1_skeleton_new(); + hostnamed_interf = hostname1_skeleton_new(); /* attach function pointers to generated struct's method handlers */ g_signal_connect(hostnamed_interf, "handle-set-hostname", G_CALLBACK(on_handle_set_hostname), NULL); @@ -191,36 +255,33 @@ static void hostnamed_on_bus_acquired(GDBusConnection *conn, NULL)) { g_printf("failed to export %s's interface!\n", name); /* unusual edge case, TODO check errno */ - hostnamed_mem_clean(); + hostnamed_mem_clean(); } else { - dbus_interface_exported = TRUE; - g_printf("exported %s's interface on the system bus...", name); - } - + dbus_interface_exported = TRUE; + g_printf("exported %s's interface on the system bus...\n", name); + } } static void hostnamed_on_name_acquired(GDBusConnection *conn, - const gchar *name, + const gchar *name, gpointer user_data) { g_printf("success!\n"); - } static void hostnamed_on_name_lost(GDBusConnection *conn, const gchar *name, gpointer user_data) { - if(!conn) { - - 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", name); + if(!conn) { - hostnamed_mem_clean(); - } + 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); + hostnamed_mem_clean(); + } - g_printf("lost name %s, exiting...", name); + g_printf("lost name %s, exiting...\n", name); hostnamed_mem_clean(); } @@ -228,44 +289,48 @@ static void hostnamed_on_name_lost(GDBusConnection *conn, /* --- end bus/name handlers, begin misc unix functions --- */ /* safe call to clean and then exit - * this stops our GMainLoop safely before letting main() return */ + * this stops our GMainLoop safely before letting main() return */ void hostnamed_mem_clean() { - g_printf("exiting...\n"); + g_printf("exiting...\n"); - if(dbus_interface_exported) - g_dbus_interface_skeleton_unexport(G_DBUS_INTERFACE_SKELETON(hostnamed_interf)); + if(dbus_interface_exported) + g_dbus_interface_skeleton_unexport(G_DBUS_INTERFACE_SKELETON(hostnamed_interf)); - if(g_main_loop_is_running(hostnamed_loop)) - g_main_loop_quit(hostnamed_loop); + if(g_main_loop_is_running(hostnamed_loop)) + g_main_loop_quit(hostnamed_loop); } /* wrapper for glib's unix signal handling; called only once if terminatating signal is raised against us */ gboolean unix_sig_terminate_handler(gpointer data) { - g_printf("caught SIGINT/HUP/TERM, exiting\n"); + g_printf("caught SIGINT/HUP/TERM, exiting\n"); - hostnamed_mem_clean(); - return G_SOURCE_REMOVE; + hostnamed_mem_clean(); + return G_SOURCE_REMOVE; } void set_signal_handlers() { - /* we don't care about its descriptor, we never need to unregister these */ - g_unix_signal_add(SIGINT, unix_sig_terminate_handler, NULL); - g_unix_signal_add(SIGHUP, unix_sig_terminate_handler, NULL); - g_unix_signal_add(SIGTERM, unix_sig_terminate_handler, NULL); + /* we don't care about its descriptor, we never need to unregister these */ + g_unix_signal_add(SIGINT, unix_sig_terminate_handler, NULL); + g_unix_signal_add(SIGHUP, unix_sig_terminate_handler, NULL); + g_unix_signal_add(SIGTERM, unix_sig_terminate_handler, NULL); + + /* TODO: the "only once" guarantee only counts towards specific signals. + * make sure calling a SIGINT and SIGHUP doesn't cause term_handler() + * to be called twice */ } int main() { - - set_signal_handlers(); + + set_signal_handlers(); - hostnamed_loop = g_main_loop_new(NULL, TRUE); - hostnamed_freeable = g_ptr_array_new(); + hostnamed_loop = g_main_loop_new(NULL, TRUE); + hostnamed_freeable = g_ptr_array_new(); - bus_descriptor = g_bus_own_name(G_BUS_TYPE_SYSTEM, + bus_descriptor = g_bus_own_name(G_BUS_TYPE_SYSTEM, "org.freedesktop.hostname1", G_BUS_NAME_OWNER_FLAGS_NONE, hostnamed_on_bus_acquired, @@ -274,17 +339,31 @@ int main() { NULL, NULL); - g_main_loop_run(hostnamed_loop); - /* runs until single g_main_loop_quit() call is raised inside _mem_clean() */ - g_main_loop_unref(hostnamed_loop); + g_main_loop_run(hostnamed_loop); + /* runs until single g_main_loop_quit() call is raised inside _mem_clean() */ + g_main_loop_unref(hostnamed_loop); + + /* guaranteed unownable */ + g_bus_unown_name(bus_descriptor); + + /* at this point no operations can occur with our data, it is safe to free it + its container */ + g_ptr_array_free(hostnamed_freeable, TRUE); + + return 0; +} + +gboolean test_against_known_vm_strings(gchar *sysctl_string) { + + unsigned int i; - /* guaranteed unownable */ - g_bus_unown_name(bus_descriptor); + if(is_vm) + return TRUE; - /* at this point no operations can occur with our data, it is safe to free it + its container */ - g_ptr_array_free(hostnamed_freeable, TRUE); + for(; i < G_N_ELEMENTS(vmstring_list); i++) + if(strcasestr(sysctl_string, vmstring_list[i])) + return (is_vm = TRUE) ? TRUE : FALSE; - return 0; + return FALSE; } /* TODO figure out DMI variables on obsd */