Guard one-time initialization with memory barriers

[nss_nonlocal.git] / nonlocal-group.c

diff --git a/nonlocal-group.c b/nonlocal-group.c
index 8ceeb237ddac4830b442516fe55025e1e52955a0..9bbe156183ccb6db5bc2e325eb2641e866a0d1d6 100644 (file)
--- a/nonlocal-group.c
+++ b/nonlocal-group.c
@@ -24,18 +24,20 @@
  */
 
 #define _GNU_SOURCE
  */
 
 #define _GNU_SOURCE

+

 #include <sys/types.h>
 #include <sys/types.h>

-#include <unistd.h>
-#include <stdlib.h>
-#include <stdint.h>
-#include <string.h>

 #include <dlfcn.h>
 #include <dlfcn.h>

-#include <stdio.h>
-#include <syslog.h>

 #include <errno.h>
 #include <errno.h>

-#include <pwd.h>

 #include <grp.h>
 #include <nss.h>
 #include <grp.h>
 #include <nss.h>

+#include <pwd.h>
+#include <stdbool.h>
+#include <stddef.h>
+#include <stdlib.h>
+#include <string.h>
+#include <syslog.h>
+#include <unistd.h>
+

 #include "nsswitch-internal.h"
 #include "nonlocal.h"
 
 #include "nsswitch-internal.h"
 #include "nonlocal.h"
 


@@ -54,6 +56,13 @@
  */
 #define MAGIC_LOCAL_GROUPNAME "nss-local-users"
 
  */
 #define MAGIC_LOCAL_GROUPNAME "nss-local-users"
 

+/*
+ * If the MAGIC_NONLOCAL_USERNAME local user is added to a local
+ * group, then the local group will inherit the nonlocal membership of
+ * a group of the same gid.
+ */
+#define MAGIC_NONLOCAL_USERNAME "nss-nonlocal-users"
+

 
 enum nss_status
 _nss_nonlocal_getgrnam_r(const char *name, struct group *grp,
 
 enum nss_status
 _nss_nonlocal_getgrnam_r(const char *name, struct group *grp,


@@ -84,7 +93,7 @@ __nss_group_nonlocal_lookup(service_user **ni, const char *fct_name,
 
 
 enum nss_status
 
 
 enum nss_status

-check_nonlocal_gid(const char *user, gid_t gid, int *errnop)
+check_nonlocal_gid(const char *user, const char *group, gid_t gid, int *errnop)

 {
     enum nss_status status;
     struct group gbuf;
 {
     enum nss_status status;
     struct group gbuf;


@@ -99,15 +108,30 @@ check_nonlocal_gid(const char *user, gid_t gid, int *errnop)
 #include "walk_nss.h"
 #undef args
 
 #include "walk_nss.h"
 #undef args
 

-    if (status == NSS_STATUS_SUCCESS) {
-       syslog(LOG_DEBUG, "nss_nonlocal: removing local group %u (%s) from non-local user %s\n", gbuf.gr_gid, gbuf.gr_name, user);
-       free(buf);
-       status = NSS_STATUS_NOTFOUND;
-    } else if (status != NSS_STATUS_TRYAGAIN) {
-       status = NSS_STATUS_SUCCESS;
+    if (status == NSS_STATUS_TRYAGAIN)
+       return status;
+    else if (status != NSS_STATUS_SUCCESS)
+       return NSS_STATUS_SUCCESS;
+
+    if (group == NULL || strcmp(gbuf.gr_name, group) == 0) {
+       char *const *mem;
+       for (mem = gbuf.gr_mem; *mem != NULL; mem++)
+           if (strcmp(*mem, MAGIC_NONLOCAL_USERNAME) == 0) {
+               status = check_nonlocal_user(*mem, errnop);
+               if (status == NSS_STATUS_TRYAGAIN) {
+                   free(buf);
+                   return status;
+               } else if (status == NSS_STATUS_NOTFOUND) {
+                   free(buf);
+                   return NSS_STATUS_SUCCESS;
+               }
+               break;
+           }

     }
 
     }
 

-    return status;
+    syslog(LOG_DEBUG, "nss_nonlocal: removing local group %u (%s) from non-local user %s\n", gbuf.gr_gid, gbuf.gr_name, user);
+    free(buf);
+    return NSS_STATUS_NOTFOUND;

 }
 
 enum nss_status
 }
 
 enum nss_status


@@ -122,13 +146,13 @@ check_nonlocal_group(const char *user, struct group *grp, int *errnop)
     gid = strtoul(grp->gr_name, &end, 10);
     if (errno == 0 && *end == '\0' && (gid_t)gid == gid) {
        errno = old_errno;
     gid = strtoul(grp->gr_name, &end, 10);
     if (errno == 0 && *end == '\0' && (gid_t)gid == gid) {
        errno = old_errno;

-       status = check_nonlocal_gid(user, gid, errnop);
+       status = check_nonlocal_gid(user, grp->gr_name, gid, errnop);

     } else
        errno = old_errno;
     if (status != NSS_STATUS_SUCCESS)
        return status;
 
     } else
        errno = old_errno;
     if (status != NSS_STATUS_SUCCESS)
        return status;
 

-    return check_nonlocal_gid(user, grp->gr_gid, errnop);
+    return check_nonlocal_gid(user, grp->gr_name, grp->gr_gid, errnop);

 }
 
 enum nss_status
 }
 
 enum nss_status


@@ -147,6 +171,7 @@ get_local_group(const char *name, struct group *grp, char **buffer, int *errnop)
     return status;
 }
 
     return status;
 }
 

+static bool grent_initialized = false;

 static service_user *grent_startp, *grent_nip;
 static void *grent_fct_start;
 static union {
 static service_user *grent_startp, *grent_nip;
 static void *grent_fct_start;
 static union {


@@ -171,9 +196,12 @@ _nss_nonlocal_setgrent(int stayopen)
     if (status != NSS_STATUS_SUCCESS)
        return status;
 
     if (status != NSS_STATUS_SUCCESS)
        return status;
 

-    if (grent_fct_start == NULL)
+    if (!grent_initialized) {

        __nss_group_nonlocal_lookup(&grent_startp, grent_fct_name,
                                    &grent_fct_start);
        __nss_group_nonlocal_lookup(&grent_startp, grent_fct_name,
                                    &grent_fct_start);

+       __sync_synchronize();
+       grent_initialized = true;
+    }

     grent_nip = grent_startp;
     grent_fct.ptr = grent_fct_start;
     return NSS_STATUS_SUCCESS;
     grent_nip = grent_startp;
     grent_fct.ptr = grent_fct_start;
     return NSS_STATUS_SUCCESS;


@@ -391,7 +419,7 @@ _nss_nonlocal_initgroups_dyn(const char *user, gid_t group, long int *start,
 
                if (status == NSS_STATUS_SUCCESS) {
                    nonlocal_errno = *errnop;
 
                if (status == NSS_STATUS_SUCCESS) {
                    nonlocal_errno = *errnop;

-                   status = check_nonlocal_gid(user, pwbuf.pw_gid,
+                   status = check_nonlocal_gid(user, NULL, pwbuf.pw_gid,

                                                &nonlocal_errno);
                    free(buf);
                }
                                                &nonlocal_errno);
                    free(buf);
                }


@@ -434,7 +462,8 @@ _nss_nonlocal_initgroups_dyn(const char *user, gid_t group, long int *start,
        if (i < out)
            continue;
 
        if (i < out)
            continue;
 

-       status = check_nonlocal_gid(user, (*groupsp)[in], &nonlocal_errno);
+       status = check_nonlocal_gid(user, NULL, (*groupsp)[in],
+                                   &nonlocal_errno);

        if (status == NSS_STATUS_SUCCESS) {
            (*groupsp)[out++] = (*groupsp)[in];
        } else if (status == NSS_STATUS_TRYAGAIN) {
        if (status == NSS_STATUS_SUCCESS) {
            (*groupsp)[out++] = (*groupsp)[in];
        } else if (status == NSS_STATUS_TRYAGAIN) {