Removed away include of system headers in comstack.h. Instead

[yaz-moved-to-github.git] / src / unix.c

diff --git a/src/unix.c b/src/unix.c
index 2c1c979..da80f53 100644 (file)
--- a/src/unix.c
+++ b/src/unix.c
@@ -1,27 +1,46 @@
 /*
- * Copyright (c) 1995-2003, Index Data
+ * Copyright (C) 1995-2005, Index Data ApS
  * See the file LICENSE for details.
  *
- * $Id: unix.c,v 1.1 2003-10-27 12:21:36 adam Exp $
+ * $Id: unix.c,v 1.9 2005-01-16 21:51:50 adam Exp $
  * UNIX socket COMSTACK. By Morten Bøgeskov.
  */
+/**
+ * \file unix.c
+ * \brief Implements UNIX domain socket COMSTACK
+ */
+
 #ifndef WIN32
 
 #include <stdio.h>
 #include <string.h>
 #include <stdlib.h>
-#include <unistd.h>
 #include <errno.h>
+#if HAVE_UNISTD_H
+#include <unistd.h>
+#endif
 #include <fcntl.h>
 #include <signal.h>
 
+#include <grp.h>
+#if HAVE_PWD_H
+#include <pwd.h>
+#endif
+#if HAVE_SYS_TYPES_H
+#include <sys/types.h>
+#endif
+
+#if HAVE_SYS_SOCKET_H
 #include <sys/socket.h>
+#endif
+#if HAVE_SYS_STAT_H
 #include <sys/stat.h>
+#endif
+#if HAVE_SYS_UN_H
 #include <sys/un.h>
+#endif
 
-#include <yaz/comstack.h>
 #include <yaz/unix.h>
-#include <yaz/log.h>
 #include <yaz/nmem.h>
 
 #ifndef YAZ_SOCKLEN_T
@@ -65,6 +84,9 @@ typedef struct unix_state
     int towrite;  /* to verify against user input */
     int (*complete)(const unsigned char *buf, int len); /* length/comple. */
     struct sockaddr_un addr;  /* returned by cs_straddr */
+    int uid;
+    int gid;
+    int umask;
     char buf[128]; /* returned by cs_addrstr */
 } unix_state;
 
@@ -163,11 +185,102 @@ static int unix_strtoaddr_ex(const char *str, struct sockaddr_un *add)
 static void *unix_straddr(COMSTACK h, const char *str)
 {
     unix_state *sp = (unix_state *)h->cprivate;
+    char * s = strdup(str);
+    char * f = s;
+    const char * file = NULL;
+    char * eol;
+
+    sp->uid = sp->gid = sp->umask = -1;
+
+    if ((eol = strchr(s, ',')))
+    {
+       do
+       {
+           if ((eol = strchr(s, ',')))
+               *eol++ = '\0';
+           if (sp->uid  == -1 && strncmp(s, "user=",  5) == 0)
+           {
+               char * arg = s + 5;
+               if (strspn(arg, "0123456789") == strlen(arg))
+               {
+                   sp->uid = atoi(arg);
+               }
+               else
+               {
+                   struct passwd * pw = getpwnam(arg);
+                   if(pw == NULL)
+                   {
+                       printf("No such user\n");
+                       free(f);
+                       return 0;
+                   }
+                   sp->uid = pw->pw_uid;
+               }
+           }
+           else if (sp->gid == -1 && strncmp(s, "group=", 6) == 0)
+           {
+               char * arg = s + 6;
+               if (strspn(arg, "0123456789") == strlen(arg))
+               {
+                   sp->gid = atoi(arg);
+               }
+               else
+               {
+                   struct group * gr = getgrnam(arg);
+                   if (gr == NULL)
+                   {
+                       printf("No such group\n");
+                       free(f);
+                       return 0;
+                   }
+                   sp->gid = gr->gr_gid;
+               }
+           }
+           else if (sp->umask == -1 && strncmp(s, "umask=", 6) == 0)
+           {
+               char * end;
+               char * arg = s + 6;
+               
+               sp->umask = strtol(arg, &end, 8);
+               if (errno == EINVAL ||
+                   *end)
+               {
+                   printf("Invalid umask\n");
+                   free(f);
+                   return 0;
+               }
+           }
+           else if (file == NULL && strncmp(s, "file=", 5) == 0)
+           {
+               char * arg = s + 5;
+               file = arg;
+           }
+           else
+           {
+               printf("invalid or double argument: %s\n", s);
+               free(f);
+               return 0;
+           }
+       } while((s = eol));
+    }
+    else
+    {
+       file = str;
+    }
+    if(! file)
+    {
+       errno = EINVAL;
+       return 0;
+    }
 
     TRC(fprintf(stderr, "unix_straddr: %s\n", str ? str : "NULL"));
 
-    if (!unix_strtoaddr_ex (str, &sp->addr))
+    if (!unix_strtoaddr_ex (file, &sp->addr))
+    {
+       free(f);
        return 0;
+    }
+    free(f);
     return &sp->addr;
 }
 
@@ -245,11 +358,9 @@ static int unix_rcvconnect(COMSTACK h)
     return 0;
 }
 
-#define CERTF "ztest.pem"
-#define KEYF "ztest.pem"
-
 static int unix_bind(COMSTACK h, void *address, int mode)
 {
+    unix_state *sp = (unix_state *)h->cprivate;
     struct sockaddr *addr = (struct sockaddr *)address;
     const char * path = ((struct sockaddr_un *)addr)->sun_path;
     struct stat stat_buf;
@@ -291,7 +402,8 @@ static int unix_bind(COMSTACK h, void *address, int mode)
        h->cerrno = CSYSERR;
        return -1;
     }
-    chmod(path, 0777);
+    chown(path, sp->uid, sp->gid);
+    chmod(path, sp->umask != -1 ? sp->umask : 0666);
     if (mode == CS_SERVER && listen(h->iofile, 3) < 0)
     {
        h->cerrno = CSYSERR;
@@ -468,7 +580,7 @@ static int unix_get(COMSTACK h, char **buf, int *bufsize)
                return -1;
        }
        else if (!res)
-           return 0;
+           return hasread;
        hasread += res;
     }
     TRC (fprintf (stderr, "  Out of read loop with hasread=%d, berlen=%d\n",
@@ -598,4 +710,4 @@ static int unix_set_blocking(COMSTACK p, int blocking)
     p->blocking = blocking;
     return 1;
 }
-#endif
+#endif /* WIN32 */