smsd/uid.c

   1 #include <sys/types.h>
   2 #include <unistd.h>
   3 #include <pwd.h>
   4 #include <grp.h>
   5 #include <stdlib.h>
   6 #include <string.h>
   7
   8 #include "uid.h"
   9
  10 /**
  11  * Sets UID.
  12  */
  13 gboolean set_uid_gid(SMSD_Parameters *params)
  14 {
  15     /* gid is set allways (either through fill_uid or by user */
  16     if (setgid(params->gid) != 0)
  17         return FALSE;
  18
  19     /* Only set group */
  20     if (params->uid == -1)
  21         return TRUE;
  22
  23     if (initgroups(params->user, params->gid) != 0)
  24         return FALSE;
  25
  26     if (setuid(params->uid) != 0)
  27         return FALSE;
  28
  29     if (seteuid(params->uid) != 0)
  30         return FALSE;
  31
  32     return TRUE;
  33 }
  34
  35 /**
  36  * Sets GID.
  37  */
  38 gboolean set_gid(int gid)
  39 {
  40     return setgid(gid) == 0;
  41 }
  42
  43 /**
  44  * Gets UID from string.
  45  */
  46 gboolean fill_uid(SMSD_Parameters *params, const char *name)
  47 {
  48     struct passwd *pwd;
  49     char *endptr;
  50     long uid;
  51
  52     pwd = getpwnam(name);
  53
  54     if (pwd == NULL) {
  55         /* Try to handle it as a number */
  56         uid = strtol(name, &endptr, 10);
  57         if (*endptr == 0 && uid > 0) {
  58             pwd = getpwuid(uid);
  59         }
  60     }
  61
  62     if (pwd != NULL) {
  63         params->uid = pwd->pw_uid;
  64         params->user = strdup(pwd->pw_name);
  65         if (params->gid == -1) {
  66             params->gid = pwd->pw_gid;
  67         }
  68         return TRUE;
  69     }
  70
  71     return FALSE;
  72 }
  73
  74 /**
  75  * Gets GID from string.
  76  */
  77 gboolean fill_gid(SMSD_Parameters *params, const char *name)
  78 {
  79     struct group *grp;
  80     char *endptr;
  81     long gid;
  82
  83     grp = getgrnam(name);
  84
  85     /* Try to handle it as a number */
  86     if (grp == NULL) {
  87         gid = strtol(name, &endptr, 10);
  88         if (*endptr == 0 && gid > 0) {
  89             grp = getgrgid(gid);
  90         }
  91     }
  92
  93     if (grp != NULL) {
  94         params->gid = grp->gr_gid;
  95         params->group = strdup(grp->gr_name);
  96         return TRUE;
  97     }
  98
  99     return FALSE;
 100 }
 101