Simplify get user home code

Previously there was a function for this (get_user_home: *sigh*). Now we
just check if the getenv for HOME returns a null pointer and save double
copying into a buffer.

1 files changed, 6 insertions, 17 deletions

diff --git a/src/main.c b/src/main.c
index 9640ed8..4201592 100644
--- a/src/main.c
+++ b/src/main.c
@@ -59,19 +59,6 @@ long days_to_seconds(int days) {
 
 
 /**
- * TODO
- */
-int get_user_home(char* out) {
-  if(getenv("HOME") != NULL) {
-    strcpy(out, getenv("HOME"));
-    return 0;
-  } else {
-    return 1;
-  }
-}
-
-
-/**
  * Writes the current timestamp to the specified lastlogin file
  * 
  * @param path Path to the lastlogin file
@@ -152,7 +139,6 @@ int exec_cmd(char* cmd) {
  * Ye olde main function
  */
 int main(int argc, char* argv[]) {
-  char home[128];
   int days;
   char* cmd;
   // Expected path to lastlogin file
@@ -167,10 +153,13 @@ int main(int argc, char* argv[]) {
     return 1;
   }
 
-
   // Compose path to lastlogin file
-  get_user_home(home);
-  strcpy(lastlogin_path, home);
+  if(getenv("HOME")) {
+    strcpy(lastlogin_path, getenv("HOME"));
+  } else {
+    printf("ERROR: HOME is unset\n");
+    return 1;
+  }
   strcat(lastlogin_path, "/.lastlogin");
 
   // Parse program arguments