1 files changed, 75 insertions, 13 deletions

diff --git a/src/mds.c b/src/mds.c
index 07041fc..90d77ad 100644
--- a/src/mds.c
+++ b/src/mds.c
@@ -15,6 +15,7 @@
  * You should have received a copy of the GNU General Public License
  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
+#include "mds.h"
 #include "config.h"
 
 #include <sys/stat.h>
@@ -22,6 +23,8 @@
 #include <stdio.h>
 #include <unistd.h>
 #include <errno.h>
+#include <limits.h>
+#include <fcntl.h>
 
 
 /**
@@ -33,12 +36,15 @@
  */
 int main(int argc, const char** argv)
 {
-  struct stat attr;
+  char pathname[PATH_MAX];
+  unsigned int display;
+  FILE *f;
+  int fd;
   
   (void) argv;
   
   /* Sanity check the number of command line arguments. */
-  if (argc > 50)
+  if (argc > ARGC_LIMIT)
     {
       fprintf(stderr,
 	      "%s: that number of arguments is ridiculous, I will not allow it.\n",
@@ -56,6 +62,58 @@ int main(int argc, const char** argv)
     }
   
   /* Create directory for socket files, PID files and such. */
+  if (create_runtime_root_directory())
+    return 1;
+  
+  /* Determine display index. */
+  for (display = 0; display < DISPLAY_MAX; display++)
+    {
+      snprintf(pathname, sizeof(pathname) / sizeof(char), "%s/%u.pid",
+	       MDS_RUNTIME_ROOT_DIRECTORY, display);
+      
+      fd = open(pathname, O_CREAT | O_EXCL);
+      if (fd == -1)
+	{
+	  /* TODO reuse display index not no longer used */
+	  continue;
+	}
+      
+      close(fd);
+    }
+  if (display == DISPLAY_MAX)
+    {
+      fprintf(stderr,
+	      "%s: Sorry, too many displays on the system.\n",
+	      *argv);
+      return 1;
+      /* Yes, the directory could have been removed, but probably not. */
+    }
+  
+  /* TODO: Create PID file. */
+  
+  /* TODO: Save MDS_DISPLAY environment variable. */
+  
+  /* Drop privileges. They most not be propagated non-authorised components. */
+  /* setgid should not be set, but just to be safe we are restoring both user and group. */
+  if ((seteuid(getuid()) < 0) || (setegid(getgid()) < 0))
+    {
+      perror(*argv);
+      return 1;
+    }
+  
+  return 0;
+}
+
+
+/**
+ * Create directory for socket files, PID files and such
+ * 
+ * @return  Non-zero on error
+ */
+int create_runtime_root_directory(void)
+{
+  struct stat attr;
+  
   if (stat(MDS_RUNTIME_ROOT_DIRECTORY, &attr) == 0)
     {
       /* Cannot create the directory, its pathname refers to an existing. */
@@ -69,19 +127,23 @@ int main(int argc, const char** argv)
 	}
     }
   else
-    /* Directory is missing, create it. */
-    if (mkdir(MDS_RUNTIME_ROOT_DIRECTORY, 0755) < 0)
-      if (errno != EEXIST) /* Unlikely race condition. */
+    {
+      /* Directory is missing, create it. */
+      if (mkdir(MDS_RUNTIME_ROOT_DIRECTORY, S_IRWXU | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH) < 0)
 	{
-	  perror(*argv);
-	  return 1;
+	  if (errno != EEXIST) /* Unlikely race condition. */
+	    {
+	      perror(*argv);
+	      return 1;
+	    }
 	}
-  
-  /* Drop privileges. They most not be propagated non-authorised components. */
-  if (seteuid(getuid()) < 0)
-    {
-      perror(*argv);
-      return 1;
+      else
+	/* Set ownership. */
+	if (chown(MDS_RUNTIME_ROOT_DIRECTORY, ROOT_USER_UID, ROOT_GROUP_GID) < 0)
+	  {
+	    perror(*argv);
+	    return 1;
+	  }
     }
   
   return 0;