#include <signal.h>
#include <sys/param.h>
+#include <sys/sysctl.h>
#include <string.h>
#include <glib/gprintf.h>
/* --- 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,
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 *
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);
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...\n", 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");
const gchar *name,
gpointer user_data) {
- if(!conn) {
+ 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.\n", name);
- 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...\n", name);
* 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,
NULL,
NULL);
- g_main_loop_run(hostnamed_loop);
- /* runs until single g_main_loop_quit() call is raised inside <interface>_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 <interface>_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 */