change buffer size for gethostname() to HOST_NAME_MAX to avoid accidently truncating...

[systembsd.git] / src / config.c

diff --git a/src/config.c b/src/config.c
index 189eb235f507788493b2e5bce8a2be97980c9078..ac9ecf468e2df72c62f6f4e2be478da8304b4491 100644 (file)
--- a/src/config.c
+++ b/src/config.c
@@ -1,3 +1,19 @@
+/*
+ * Copyright (c) 2014 Ian Sutton <ian@kremlin.cc>
+ *
+ * Permission to use, copy, modify, and distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+ * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+ */
+
 #include <unistd.h>
 #include <fcntl.h>
 #include <gio/gio.h>
@@ -5,74 +21,91 @@
 static GKeyFile *config;
 
 static const gchar *CONFIG_KEYS[] = {
-       "PrettyHostname",
-       "IconName",
-       "ChassisType"
+    "PrettyHostname",
+    "IconName",
+    "ChassisType"
 };
 
 /* NULL if key doesn't exist */
 gchar *get_option(gchar *key, gchar *group) {
 
-       if(!group)
-               group = "default";
+    if(!group)
+        group = "default";
 
-       return g_key_file_get_string(config, group, key, NULL);
+    return g_key_file_get_string(config, group, key, NULL);
 }
 
 /* false if key isn't already defined or value is invalid */
 gboolean set_option(gchar *key, gchar *value, gchar *group) {
 
-       if(!group)
-               group = "default";
+    if(!group)
+        group = "default";
 
-       if(!g_key_file_get_string(config, group, key, NULL))
-               return FALSE;
+    if(!g_key_file_get_string(config, group, key, NULL))
+        return FALSE;
 
-       //TODO safteycheck value
-       g_key_file_set_string(config, group, key, value);
-       return TRUE;
+    //TODO safteycheck value
+    g_key_file_set_string(config, group, key, value);
+    return TRUE;
 }
 
 /* initial load/check */
-
 gboolean config_init() {
 
-       static gchar *config_path;
-       int tryopen = 0;
-       /* config is already good to go */
-       if(config)
-               return TRUE;
-
-       /* does config file exist? if not, write one */
-       else if(!g_key_file_load_from_data(config, "systemd_compat.conf", &config_path, G_KEY_FILE_KEEP_COMMENTS, NULL)) {
-                       
-                       tryopen = g_open("/etc/systemd_compat.conf", O_CREAT, 644);
-                       
-                       //TODO clean this up, use g_data_dirs and g_exit
-                       /* can we open it rw? */
-                       if(!g_access("/etc/systemd_compat.conf", W_OK) && !tryopen) {
-                               g_printf("%s\n", "ERROR: cannot open systemd_compat.conf as read/write!");
-                               return FALSE;
-                       }
-                       
-                       if(tryopen) {
-                               config_path = "/etc/systemd_compat.conf";
-                               g_close(tryopen, NULL);
-                       }
-
-                       //TODO set these properly
-                       config = g_key_file_new();
-
-                       g_key_file_set_string(config, "hostnamed", "PrettyHostname", "");
-                       g_key_file_set_string(config, "hostnamed", "IconName", "Computer");     
-                       g_key_file_set_string(config, "hostnamed", "ChassisType", "laptop");
-
-                       if(!g_key_file_save_to_file(config, config_path, NULL))
-                               return FALSE;
-
-                       return TRUE;
-
-       /* it does it exist and was written to config var */
-       } else
-               return TRUE;
+    if(config)
+        return TRUE; //already init'd
+
+    config = g_key_file_new();
+    
+    const gchar *config_path;
+    GStatBuf *config_lstat;
+
+    config_path = "/etc/systemd_compat.conf";
+
+    /* does conf exist? */
+    if(g_lstat(config_path, config_lstat)) {
+
+        /* if not, can we write it */
+        if(g_access("/etc/", W_OK)) {
+            g_printf("%s\n", "no write permissions for /etc/! exiting..");
+            return FALSE;
+        }
+
+        int config_descr;
+        config_descr = g_open(config_path, O_CREAT, 644);
+
+        gchar *posix_hostname;
+        posix_hostname = g_malloc(HOST_NAME_MAX); 
+
+        gethostname(posix_hostname, HOST_NAME_MAX);
+
+        g_key_file_set_string(config, "hostnamed", "Hostname", posix_hostname);
+        g_key_file_set_string(config, "hostnamed", "PrettyHostname", "");
+        g_key_file_set_string(config, "hostnamed", "IconName", "Computer"); 
+        g_key_file_set_string(config, "hostnamed", "ChassisType", "laptop"); //TODO set these correctly
+
+        if(!g_key_file_save_to_file(config, config_path, NULL)) {
+            g_printf("failed to write config to %s!\n", config_path);
+            g_free(posix_hostname);
+            return FALSE;
+        }
+
+        g_printf("wrote config to %s\n", config_path);
+
+        g_free(posix_hostname);
+
+        return TRUE;
+
+    /* it does exist, read it */
+    } else {
+
+        if(!g_access(config_path, W_OK)) {
+            g_printf("%s\n", "no write permissions for /etc/! exiting..");
+            return FALSE;
+        } else if(g_key_file_load_from_file(config, config_path, G_KEY_FILE_KEEP_COMMENTS, NULL))
+            return TRUE;
+
+        g_printf("could not read config at %s! exiting..", config_path);
+        return FALSE;
+    }
 }