lib/nss_wrapper/nss_wrapper.c

   1 /*
   2  * Copyright (C) Stefan Metzmacher 2007 <metze@samba.org>
   3  * Copyright (C) Guenther Deschner 2009 <gd@samba.org>
   4  * Copyright (C) Andreas Schneider 2013 <asn@samba.org>
   5  *
   6  * All rights reserved.
   7  *
   8  * Redistribution and use in source and binary forms, with or without
   9  * modification, are permitted provided that the following conditions
  10  * are met:
  11  *
  12  * 1. Redistributions of source code must retain the above copyright
  13  *    notice, this list of conditions and the following disclaimer.
  14  *
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in the
  17  *    documentation and/or other materials provided with the distribution.
  18  *
  19  * 3. Neither the name of the author nor the names of its contributors
  20  *    may be used to endorse or promote products derived from this software
  21  *    without specific prior written permission.
  22  *
  23  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  24  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  25  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  26  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  27  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  28  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  29  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  30  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  31  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  32  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  33  * SUCH DAMAGE.
  34  */
  35
  36 #include "config.h"
  37
  38 #include <pthread.h>
  39
  40 #include <sys/types.h>
  41 #include <sys/stat.h>
  42 #include <sys/socket.h>
  43 #include <errno.h>
  44 #include <fcntl.h>
  45 #include <stdarg.h>
  46 #include <stdbool.h>
  47 #include <stddef.h>
  48 #include <stdio.h>
  49 #include <stdint.h>
  50 #include <stdlib.h>
  51 #include <string.h>
  52 #include <unistd.h>
  53 #include <ctype.h>
  54
  55 #include <search.h>
  56 #include <assert.h>
  57
  58 /*
  59  * Defining _POSIX_PTHREAD_SEMANTICS before including pwd.h and grp.h  gives us
  60  * the posix getpwnam_r(), getpwuid_r(), getgrnam_r and getgrgid_r calls on
  61  * Solaris
  62  */
  63 #ifndef _POSIX_PTHREAD_SEMANTICS
  64 #define _POSIX_PTHREAD_SEMANTICS
  65 #endif
  66
  67 #include <pwd.h>
  68 #include <grp.h>
  69 #ifdef HAVE_SHADOW_H
  70 #include <shadow.h>
  71 #endif /* HAVE_SHADOW_H */
  72
  73 #include <netdb.h>
  74 #include <arpa/inet.h>
  75 #include <netinet/in.h>
  76
  77 #include <dlfcn.h>
  78
  79 #if defined(HAVE_NSS_H)
  80 /* Linux and BSD */
  81 #include <nss.h>
  82
  83 typedef enum nss_status NSS_STATUS;
  84 #elif defined(HAVE_NSS_COMMON_H)
  85 /* Solaris */
  86 #include <nss_common.h>
  87 #include <nss_dbdefs.h>
  88 #include <nsswitch.h>
  89
  90 typedef nss_status_t NSS_STATUS;
  91
  92 # define NSS_STATUS_SUCCESS     NSS_SUCCESS
  93 # define NSS_STATUS_NOTFOUND    NSS_NOTFOUND
  94 # define NSS_STATUS_UNAVAIL     NSS_UNAVAIL
  95 # define NSS_STATUS_TRYAGAIN    NSS_TRYAGAIN
  96 #else
  97 # error "No nsswitch support detected"
  98 #endif
  99
 100 #ifndef PTR_DIFF
 101 #define PTR_DIFF(p1, p2) ((ptrdiff_t)(((const char *)(p1)) - (const char *)(p2)))
 102 #endif
 103
 104 #ifndef _PUBLIC_
 105 #define _PUBLIC_
 106 #endif
 107
 108 #ifndef EAI_NODATA
 109 #define EAI_NODATA EAI_NONAME
 110 #endif
 111
 112 #ifndef EAI_ADDRFAMILY
 113 #define EAI_ADDRFAMILY EAI_FAMILY
 114 #endif
 115
 116 #ifndef __STRING
 117 #define __STRING(x)    #x
 118 #endif
 119
 120 #ifndef __STRINGSTRING
 121 #define __STRINGSTRING(x) __STRING(x)
 122 #endif
 123
 124 #ifndef __LINESTR__
 125 #define __LINESTR__ __STRINGSTRING(__LINE__)
 126 #endif
 127
 128 #ifndef __location__
 129 #define __location__ __FILE__ ":" __LINESTR__
 130 #endif
 131
 132 #ifndef DNS_NAME_MAX
 133 #define DNS_NAME_MAX 255
 134 #endif
 135
 136 /* GCC have printf type attribute check. */
 137 #ifdef HAVE_ATTRIBUTE_PRINTF_FORMAT
 138 #define PRINTF_ATTRIBUTE(a,b) __attribute__ ((__format__ (__printf__, a, b)))
 139 #else
 140 #define PRINTF_ATTRIBUTE(a,b)
 141 #endif /* HAVE_ATTRIBUTE_PRINTF_FORMAT */
 142
 143 #ifdef HAVE_DESTRUCTOR_ATTRIBUTE
 144 #define DESTRUCTOR_ATTRIBUTE __attribute__ ((destructor))
 145 #else
 146 #define DESTRUCTOR_ATTRIBUTE
 147 #endif /* HAVE_DESTRUCTOR_ATTRIBUTE */
 148
 149 #define ZERO_STRUCTP(x) do { if ((x) != NULL) memset((char *)(x), 0, sizeof(*(x))); } while(0)
 150
 151 #ifndef SAFE_FREE
 152 #define SAFE_FREE(x) do { if ((x) != NULL) {free(x); (x)=NULL;} } while(0)
 153 #endif
 154
 155 #ifdef HAVE_IPV6
 156 #define NWRAP_INET_ADDRSTRLEN INET6_ADDRSTRLEN
 157 #else
 158 #define NWRAP_INET_ADDRSTRLEN INET_ADDRSTRLEN
 159 #endif
 160
 161 #define NWRAP_LOCK(m) do { \
 162         pthread_mutex_lock(&( m ## _mutex)); \
 163 } while(0)
 164
 165 #define NWRAP_UNLOCK(m) do { \
 166         pthread_mutex_unlock(&( m ## _mutex)); \
 167 } while(0)
 168
 169
 170 static bool nwrap_initialized = false;
 171 static pthread_mutex_t nwrap_initialized_mutex = PTHREAD_MUTEX_INITIALIZER;
 172
 173 /* The mutex or accessing the id */
 174 static pthread_mutex_t nwrap_global_mutex = PTHREAD_MUTEX_INITIALIZER;
 175 static pthread_mutex_t nwrap_gr_global_mutex = PTHREAD_MUTEX_INITIALIZER;
 176 static pthread_mutex_t nwrap_he_global_mutex = PTHREAD_MUTEX_INITIALIZER;
 177 static pthread_mutex_t nwrap_pw_global_mutex = PTHREAD_MUTEX_INITIALIZER;
 178 static pthread_mutex_t nwrap_sp_global_mutex = PTHREAD_MUTEX_INITIALIZER;
 179
 180 /* Add new global locks here please */
 181 /* Also don't forget to add locks to
 182  * nwrap_init() function.
 183  */
 184 # define NWRAP_LOCK_ALL do { \
 185         NWRAP_LOCK(nwrap_initialized); \
 186         NWRAP_LOCK(nwrap_global); \
 187         NWRAP_LOCK(nwrap_gr_global); \
 188         NWRAP_LOCK(nwrap_he_global); \
 189         NWRAP_LOCK(nwrap_pw_global); \
 190         NWRAP_LOCK(nwrap_sp_global); \
 191 } while (0);
 192
 193 # define NWRAP_UNLOCK_ALL do {\
 194         NWRAP_UNLOCK(nwrap_sp_global); \
 195         NWRAP_UNLOCK(nwrap_pw_global); \
 196         NWRAP_UNLOCK(nwrap_he_global); \
 197         NWRAP_UNLOCK(nwrap_gr_global); \
 198         NWRAP_UNLOCK(nwrap_global); \
 199         NWRAP_UNLOCK(nwrap_initialized); \
 200 } while (0);
 201
 202 static void nwrap_thread_prepare(void)
 203 {
 204         NWRAP_LOCK_ALL;
 205 }
 206
 207 static void nwrap_thread_parent(void)
 208 {
 209         NWRAP_UNLOCK_ALL;
 210 }
 211
 212 static void nwrap_thread_child(void)
 213 {
 214         NWRAP_UNLOCK_ALL;
 215 }
 216
 217 enum nwrap_dbglvl_e {
 218         NWRAP_LOG_ERROR = 0,
 219         NWRAP_LOG_WARN,
 220         NWRAP_LOG_DEBUG,
 221         NWRAP_LOG_TRACE
 222 };
 223
 224 #ifdef NDEBUG
 225 # define NWRAP_LOG(...)
 226 #else
 227
 228 static void nwrap_log(enum nwrap_dbglvl_e dbglvl, const char *func, const char *format, ...) PRINTF_ATTRIBUTE(3, 4);
 229 # define NWRAP_LOG(dbglvl, ...) nwrap_log((dbglvl), __func__, __VA_ARGS__)
 230
 231 static void nwrap_log(enum nwrap_dbglvl_e dbglvl,
 232                       const char *func,
 233                       const char *format, ...)
 234 {
 235         char buffer[1024];
 236         va_list va;
 237         const char *d;
 238         unsigned int lvl = 0;
 239         int pid = getpid();
 240
 241         d = getenv("NSS_WRAPPER_DEBUGLEVEL");
 242         if (d != NULL) {
 243                 lvl = atoi(d);
 244         }
 245
 246         va_start(va, format);
 247         vsnprintf(buffer, sizeof(buffer), format, va);
 248         va_end(va);
 249
 250         if (lvl >= dbglvl) {
 251                 switch (dbglvl) {
 252                         case NWRAP_LOG_ERROR:
 253                                 fprintf(stderr,
 254                                         "NWRAP_ERROR(%d) - %s: %s\n",
 255                                         pid, func, buffer);
 256                                 break;
 257                         case NWRAP_LOG_WARN:
 258                                 fprintf(stderr,
 259                                         "NWRAP_WARN(%d) - %s: %s\n",
 260                                         pid, func, buffer);
 261                                 break;
 262                         case NWRAP_LOG_DEBUG:
 263                                 fprintf(stderr,
 264                                         "NWRAP_DEBUG(%d) - %s: %s\n",
 265                                         pid, func, buffer);
 266                                 break;
 267                         case NWRAP_LOG_TRACE:
 268                                 fprintf(stderr,
 269                                         "NWRAP_TRACE(%d) - %s: %s\n",
 270                                         pid, func, buffer);
 271                                 break;
 272                 }
 273         }
 274 }
 275 #endif /* NDEBUG NWRAP_LOG */
 276
 277 struct nwrap_libc_fns {
 278         struct passwd *(*_libc_getpwnam)(const char *name);
 279         int (*_libc_getpwnam_r)(const char *name, struct passwd *pwd,
 280                        char *buf, size_t buflen, struct passwd **result);
 281         struct passwd *(*_libc_getpwuid)(uid_t uid);
 282         int (*_libc_getpwuid_r)(uid_t uid, struct passwd *pwd, char *buf, size_t buflen, struct passwd **result);
 283         void (*_libc_setpwent)(void);
 284         struct passwd *(*_libc_getpwent)(void);
 285 #ifdef HAVE_SOLARIS_GETPWENT_R
 286         struct passwd *(*_libc_getpwent_r)(struct passwd *pwbuf, char *buf, size_t buflen);
 287 #else
 288         int (*_libc_getpwent_r)(struct passwd *pwbuf, char *buf, size_t buflen, struct passwd **pwbufp);
 289 #endif
 290         void (*_libc_endpwent)(void);
 291         int (*_libc_initgroups)(const char *user, gid_t gid);
 292         struct group *(*_libc_getgrnam)(const char *name);
 293         int (*_libc_getgrnam_r)(const char *name, struct group *grp, char *buf, size_t buflen, struct group **result);
 294         struct group *(*_libc_getgrgid)(gid_t gid);
 295         int (*_libc_getgrgid_r)(gid_t gid, struct group *grp, char *buf, size_t buflen, struct group **result);
 296         void (*_libc_setgrent)(void);
 297         struct group *(*_libc_getgrent)(void);
 298 #ifdef HAVE_SOLARIS_GETGRENT_R
 299         struct group *(*_libc_getgrent_r)(struct group *group, char *buf, size_t buflen);
 300 #else
 301         int (*_libc_getgrent_r)(struct group *group, char *buf, size_t buflen, struct group **result);
 302 #endif
 303         void (*_libc_endgrent)(void);
 304         int (*_libc_getgrouplist)(const char *user, gid_t group, gid_t *groups, int *ngroups);
 305
 306         void (*_libc_sethostent)(int stayopen);
 307         struct hostent *(*_libc_gethostent)(void);
 308         void (*_libc_endhostent)(void);
 309
 310         struct hostent *(*_libc_gethostbyname)(const char *name);
 311 #ifdef HAVE_GETHOSTBYNAME2 /* GNU extension */
 312         struct hostent *(*_libc_gethostbyname2)(const char *name, int af);
 313 #endif
 314         struct hostent *(*_libc_gethostbyaddr)(const void *addr, socklen_t len, int type);
 315
 316         int (*_libc_getaddrinfo)(const char *node, const char *service,
 317                                  const struct addrinfo *hints,
 318                                  struct addrinfo **res);
 319         int (*_libc_getnameinfo)(const struct sockaddr *sa, socklen_t salen,
 320                                  char *host, size_t hostlen,
 321                                  char *serv, size_t servlen,
 322                                  int flags);
 323         int (*_libc_gethostname)(char *name, size_t len);
 324 #ifdef HAVE_GETHOSTBYNAME_R
 325         int (*_libc_gethostbyname_r)(const char *name,
 326                                      struct hostent *ret,
 327                                      char *buf, size_t buflen,
 328                                      struct hostent **result, int *h_errnop);
 329 #endif
 330 #ifdef HAVE_GETHOSTBYADDR_R
 331         int (*_libc_gethostbyaddr_r)(const void *addr, socklen_t len, int type,
 332                                      struct hostent *ret,
 333                                      char *buf, size_t buflen,
 334                                      struct hostent **result, int *h_errnop);
 335 #endif
 336 };
 337
 338 struct nwrap_module_nss_fns {
 339         NSS_STATUS (*_nss_getpwnam_r)(const char *name, struct passwd *result, char *buffer,
 340                                       size_t buflen, int *errnop);
 341         NSS_STATUS (*_nss_getpwuid_r)(uid_t uid, struct passwd *result, char *buffer,
 342                                       size_t buflen, int *errnop);
 343         NSS_STATUS (*_nss_setpwent)(void);
 344         NSS_STATUS (*_nss_getpwent_r)(struct passwd *result, char *buffer,
 345                                       size_t buflen, int *errnop);
 346         NSS_STATUS (*_nss_endpwent)(void);
 347         NSS_STATUS (*_nss_initgroups)(const char *user, gid_t group, long int *start,
 348                                       long int *size, gid_t **groups, long int limit, int *errnop);
 349         NSS_STATUS (*_nss_getgrnam_r)(const char *name, struct group *result, char *buffer,
 350                                       size_t buflen, int *errnop);
 351         NSS_STATUS (*_nss_getgrgid_r)(gid_t gid, struct group *result, char *buffer,
 352                                       size_t buflen, int *errnop);
 353         NSS_STATUS (*_nss_setgrent)(void);
 354         NSS_STATUS (*_nss_getgrent_r)(struct group *result, char *buffer,
 355                                       size_t buflen, int *errnop);
 356         NSS_STATUS (*_nss_endgrent)(void);
 357 };
 358
 359 struct nwrap_backend {
 360         const char *name;
 361         const char *so_path;
 362         void *so_handle;
 363         struct nwrap_ops *ops;
 364         struct nwrap_module_nss_fns *fns;
 365 };
 366
 367 struct nwrap_ops {
 368         struct passwd * (*nw_getpwnam)(struct nwrap_backend *b,
 369                                        const char *name);
 370         int             (*nw_getpwnam_r)(struct nwrap_backend *b,
 371                                          const char *name, struct passwd *pwdst,
 372                                          char *buf, size_t buflen, struct passwd **pwdstp);
 373         struct passwd * (*nw_getpwuid)(struct nwrap_backend *b,
 374                                        uid_t uid);
 375         int             (*nw_getpwuid_r)(struct nwrap_backend *b,
 376                                          uid_t uid, struct passwd *pwdst,
 377                                          char *buf, size_t buflen, struct passwd **pwdstp);
 378         void            (*nw_setpwent)(struct nwrap_backend *b);
 379         struct passwd * (*nw_getpwent)(struct nwrap_backend *b);
 380         int             (*nw_getpwent_r)(struct nwrap_backend *b,
 381                                          struct passwd *pwdst, char *buf,
 382                                          size_t buflen, struct passwd **pwdstp);
 383         void            (*nw_endpwent)(struct nwrap_backend *b);
 384         int             (*nw_initgroups)(struct nwrap_backend *b,
 385                                          const char *user, gid_t group);
 386         struct group *  (*nw_getgrnam)(struct nwrap_backend *b,
 387                                        const char *name);
 388         int             (*nw_getgrnam_r)(struct nwrap_backend *b,
 389                                          const char *name, struct group *grdst,
 390                                          char *buf, size_t buflen, struct group **grdstp);
 391         struct group *  (*nw_getgrgid)(struct nwrap_backend *b,
 392                                        gid_t gid);
 393         int             (*nw_getgrgid_r)(struct nwrap_backend *b,
 394                                          gid_t gid, struct group *grdst,
 395                                          char *buf, size_t buflen, struct group **grdstp);
 396         void            (*nw_setgrent)(struct nwrap_backend *b);
 397         struct group *  (*nw_getgrent)(struct nwrap_backend *b);
 398         int             (*nw_getgrent_r)(struct nwrap_backend *b,
 399                                          struct group *grdst, char *buf,
 400                                          size_t buflen, struct group **grdstp);
 401         void            (*nw_endgrent)(struct nwrap_backend *b);
 402 };
 403
 404 /* Public prototypes */
 405
 406 bool nss_wrapper_enabled(void);
 407 bool nss_wrapper_shadow_enabled(void);
 408 bool nss_wrapper_hosts_enabled(void);
 409
 410 /* prototypes for files backend */
 411
 412
 413 static struct passwd *nwrap_files_getpwnam(struct nwrap_backend *b,
 414                                            const char *name);
 415 static int nwrap_files_getpwnam_r(struct nwrap_backend *b,
 416                                   const char *name, struct passwd *pwdst,
 417                                   char *buf, size_t buflen, struct passwd **pwdstp);
 418 static struct passwd *nwrap_files_getpwuid(struct nwrap_backend *b,
 419                                            uid_t uid);
 420 static int nwrap_files_getpwuid_r(struct nwrap_backend *b,
 421                                   uid_t uid, struct passwd *pwdst,
 422                                   char *buf, size_t buflen, struct passwd **pwdstp);
 423 static void nwrap_files_setpwent(struct nwrap_backend *b);
 424 static struct passwd *nwrap_files_getpwent(struct nwrap_backend *b);
 425 static int nwrap_files_getpwent_r(struct nwrap_backend *b,
 426                                   struct passwd *pwdst, char *buf,
 427                                   size_t buflen, struct passwd **pwdstp);
 428 static void nwrap_files_endpwent(struct nwrap_backend *b);
 429 static int nwrap_files_initgroups(struct nwrap_backend *b,
 430                                   const char *user, gid_t group);
 431 static struct group *nwrap_files_getgrnam(struct nwrap_backend *b,
 432                                           const char *name);
 433 static int nwrap_files_getgrnam_r(struct nwrap_backend *b,
 434                                   const char *name, struct group *grdst,
 435                                   char *buf, size_t buflen, struct group **grdstp);
 436 static struct group *nwrap_files_getgrgid(struct nwrap_backend *b,
 437                                           gid_t gid);
 438 static int nwrap_files_getgrgid_r(struct nwrap_backend *b,
 439                                   gid_t gid, struct group *grdst,
 440                                   char *buf, size_t buflen, struct group **grdstp);
 441 static void nwrap_files_setgrent(struct nwrap_backend *b);
 442 static struct group *nwrap_files_getgrent(struct nwrap_backend *b);
 443 static int nwrap_files_getgrent_r(struct nwrap_backend *b,
 444                                   struct group *grdst, char *buf,
 445                                   size_t buflen, struct group **grdstp);
 446 static void nwrap_files_endgrent(struct nwrap_backend *b);
 447
 448 /* prototypes for module backend */
 449
 450 static struct passwd *nwrap_module_getpwent(struct nwrap_backend *b);
 451 static int nwrap_module_getpwent_r(struct nwrap_backend *b,
 452                                    struct passwd *pwdst, char *buf,
 453                                    size_t buflen, struct passwd **pwdstp);
 454 static struct passwd *nwrap_module_getpwnam(struct nwrap_backend *b,
 455                                             const char *name);
 456 static int nwrap_module_getpwnam_r(struct nwrap_backend *b,
 457                                    const char *name, struct passwd *pwdst,
 458                                    char *buf, size_t buflen, struct passwd **pwdstp);
 459 static struct passwd *nwrap_module_getpwuid(struct nwrap_backend *b,
 460                                             uid_t uid);
 461 static int nwrap_module_getpwuid_r(struct nwrap_backend *b,
 462                                    uid_t uid, struct passwd *pwdst,
 463                                    char *buf, size_t buflen, struct passwd **pwdstp);
 464 static void nwrap_module_setpwent(struct nwrap_backend *b);
 465 static void nwrap_module_endpwent(struct nwrap_backend *b);
 466 static struct group *nwrap_module_getgrent(struct nwrap_backend *b);
 467 static int nwrap_module_getgrent_r(struct nwrap_backend *b,
 468                                    struct group *grdst, char *buf,
 469                                    size_t buflen, struct group **grdstp);
 470 static struct group *nwrap_module_getgrnam(struct nwrap_backend *b,
 471                                            const char *name);
 472 static int nwrap_module_getgrnam_r(struct nwrap_backend *b,
 473                                    const char *name, struct group *grdst,
 474                                    char *buf, size_t buflen, struct group **grdstp);
 475 static struct group *nwrap_module_getgrgid(struct nwrap_backend *b,
 476                                            gid_t gid);
 477 static int nwrap_module_getgrgid_r(struct nwrap_backend *b,
 478                                    gid_t gid, struct group *grdst,
 479                                    char *buf, size_t buflen, struct group **grdstp);
 480 static void nwrap_module_setgrent(struct nwrap_backend *b);
 481 static void nwrap_module_endgrent(struct nwrap_backend *b);
 482 static int nwrap_module_initgroups(struct nwrap_backend *b,
 483                                    const char *user, gid_t group);
 484
 485 struct nwrap_ops nwrap_files_ops = {
 486         .nw_getpwnam    = nwrap_files_getpwnam,
 487         .nw_getpwnam_r  = nwrap_files_getpwnam_r,
 488         .nw_getpwuid    = nwrap_files_getpwuid,
 489         .nw_getpwuid_r  = nwrap_files_getpwuid_r,
 490         .nw_setpwent    = nwrap_files_setpwent,
 491         .nw_getpwent    = nwrap_files_getpwent,
 492         .nw_getpwent_r  = nwrap_files_getpwent_r,
 493         .nw_endpwent    = nwrap_files_endpwent,
 494         .nw_initgroups  = nwrap_files_initgroups,
 495         .nw_getgrnam    = nwrap_files_getgrnam,
 496         .nw_getgrnam_r  = nwrap_files_getgrnam_r,
 497         .nw_getgrgid    = nwrap_files_getgrgid,
 498         .nw_getgrgid_r  = nwrap_files_getgrgid_r,
 499         .nw_setgrent    = nwrap_files_setgrent,
 500         .nw_getgrent    = nwrap_files_getgrent,
 501         .nw_getgrent_r  = nwrap_files_getgrent_r,
 502         .nw_endgrent    = nwrap_files_endgrent,
 503 };
 504
 505 struct nwrap_ops nwrap_module_ops = {
 506         .nw_getpwnam    = nwrap_module_getpwnam,
 507         .nw_getpwnam_r  = nwrap_module_getpwnam_r,
 508         .nw_getpwuid    = nwrap_module_getpwuid,
 509         .nw_getpwuid_r  = nwrap_module_getpwuid_r,
 510         .nw_setpwent    = nwrap_module_setpwent,
 511         .nw_getpwent    = nwrap_module_getpwent,
 512         .nw_getpwent_r  = nwrap_module_getpwent_r,
 513         .nw_endpwent    = nwrap_module_endpwent,
 514         .nw_initgroups  = nwrap_module_initgroups,
 515         .nw_getgrnam    = nwrap_module_getgrnam,
 516         .nw_getgrnam_r  = nwrap_module_getgrnam_r,
 517         .nw_getgrgid    = nwrap_module_getgrgid,
 518         .nw_getgrgid_r  = nwrap_module_getgrgid_r,
 519         .nw_setgrent    = nwrap_module_setgrent,
 520         .nw_getgrent    = nwrap_module_getgrent,
 521         .nw_getgrent_r  = nwrap_module_getgrent_r,
 522         .nw_endgrent    = nwrap_module_endgrent,
 523 };
 524
 525 struct nwrap_libc {
 526         void *handle;
 527         void *nsl_handle;
 528         void *sock_handle;
 529         struct nwrap_libc_fns *fns;
 530 };
 531
 532 struct nwrap_main {
 533         int num_backends;
 534         struct nwrap_backend *backends;
 535         struct nwrap_libc *libc;
 536 };
 537
 538 static struct nwrap_main *nwrap_main_global;
 539 static struct nwrap_main __nwrap_main_global;
 540
 541 /*
 542  * PROTOTYPES
 543  */
 544 static int nwrap_convert_he_ai(const struct hostent *he,
 545                                unsigned short port,
 546                                const struct addrinfo *hints,
 547                                struct addrinfo **pai,
 548                                bool skip_canonname);
 549
 550 /*
 551  * VECTORS
 552  */
 553
 554 #define DEFAULT_VECTOR_CAPACITY 16
 555
 556 struct nwrap_vector {
 557         void **items;
 558         size_t count;
 559         size_t capacity;
 560 };
 561
 562 /* Macro returns pointer to first element of vector->items array.
 563  *
 564  * nwrap_vector is used as a memory backend which take care of
 565  * memory allocations and other stuff like memory growing.
 566  * nwrap_vectors should not be considered as some abstract structures.
 567  * On this level, vectors are more handy than direct realloc/malloc
 568  * calls.
 569  *
 570  * nwrap_vector->items is array inside nwrap_vector which can be
 571  * directly pointed by libc structure assembled by cwrap itself.
 572  *
 573  * EXAMPLE:
 574  *
 575  * 1) struct hostent contains char **h_addr_list element.
 576  * 2) nwrap_vector holds array of pointers to addresses.
 577  *    It's easier to use vector to store results of
 578  *    file parsing etc.
 579  *
 580  * Now, pretend that cwrap assembled struct hostent and
 581  * we need to set h_addr_list to point to nwrap_vector.
 582  * Idea behind is to shield users from internal nwrap_vector
 583  * implementation.
 584  * (Yes, not fully - array terminated by NULL is needed because
 585  * it's result expected by libc function caller.)
 586  *
 587  *
 588  * CODE EXAMPLE:
 589  *
 590  * struct hostent he;
 591  * struct nwrap_vector *vector = malloc(sizeof(struct nwrap_vector));
 592  * ... don't care about failed allocation now ...
 593  *
 594  * ... fill nwrap vector ...
 595  *
 596  * struct hostent he;
 597  * he.h_addr_list = nwrap_vector_head(vector);
 598  *
 599  */
 600 #define nwrap_vector_head(vect) ((void *)((vect)->items))
 601
 602 #define nwrap_vector_foreach(item, vect, iter) \
 603         for (iter = 0, (item) = (vect).items == NULL ? NULL : (vect).items[0]; \
 604              item != NULL; \
 605              (item) = (vect).items[++iter])
 606
 607 #define nwrap_vector_is_initialized(vector) ((vector)->items != NULL)
 608
 609 static inline bool nwrap_vector_init(struct nwrap_vector *const vector)
 610 {
 611         if (vector == NULL) {
 612                 return false;
 613         }
 614
 615         /* count is initialized by ZERO_STRUCTP */
 616         ZERO_STRUCTP(vector);
 617         vector->items = malloc(sizeof(void *) * (DEFAULT_VECTOR_CAPACITY + 1));
 618         if (vector->items == NULL) {
 619                 return false;
 620         }
 621         vector->capacity = DEFAULT_VECTOR_CAPACITY;
 622         memset(vector->items, '\0', sizeof(void *) * (DEFAULT_VECTOR_CAPACITY + 1));
 623
 624         return true;
 625 }
 626
 627 static bool nwrap_vector_add_item(struct nwrap_vector *vector, void *const item)
 628 {
 629         assert (vector != NULL);
 630
 631         if (vector->items == NULL) {
 632                 nwrap_vector_init(vector);
 633         }
 634
 635         if (vector->count == vector->capacity) {
 636                 /* Items array _MUST_ be NULL terminated because it's passed
 637                  * as result to caller which expect NULL terminated array from libc.
 638                  */
 639                 void **items = realloc(vector->items, sizeof(void *) * ((vector->capacity * 2) + 1));
 640                 if (items == NULL) {
 641                         return false;
 642                 }
 643                 vector->items = items;
 644
 645                 /* Don't count ending NULL to capacity */
 646                 vector->capacity *= 2;
 647         }
 648
 649         vector->items[vector->count] = item;
 650
 651         vector->count += 1;
 652         vector->items[vector->count] = NULL;
 653
 654         return true;
 655 }
 656
 657 static bool nwrap_vector_merge(struct nwrap_vector *dst,
 658                                struct nwrap_vector *src)
 659 {
 660         void **dst_items = NULL;
 661         size_t count;
 662
 663         if (src->count == 0) {
 664                 return true;
 665         }
 666
 667         count = dst->count + src->count;
 668
 669         /* We don't need reallocation if we have enough capacity. */
 670         if (src->count > (dst->capacity - dst->count)) {
 671                 dst_items = (void **)realloc(dst->items, (count + 1) * sizeof(void *));
 672                 if (dst_items == NULL) {
 673                         return false;
 674                 }
 675                 dst->items = dst_items;
 676                 dst->capacity = count;
 677         }
 678
 679         memcpy((void *)(((long *)dst->items) + dst->count),
 680                src->items,
 681                src->count * sizeof(void *));
 682         dst->count = count;
 683
 684         return true;
 685 }
 686
 687 struct nwrap_cache {
 688         const char *path;
 689         int fd;
 690         FILE *fp;
 691         struct stat st;
 692         void *private_data;
 693
 694         struct nwrap_vector lines;
 695
 696         bool (*parse_line)(struct nwrap_cache *, char *line);
 697         void (*unload)(struct nwrap_cache *);
 698 };
 699
 700 /* passwd */
 701 struct nwrap_pw {
 702         struct nwrap_cache *cache;
 703
 704         struct passwd *list;
 705         int num;
 706         int idx;
 707 };
 708
 709 struct nwrap_cache __nwrap_cache_pw;
 710 struct nwrap_pw nwrap_pw_global;
 711
 712 static bool nwrap_pw_parse_line(struct nwrap_cache *nwrap, char *line);
 713 static void nwrap_pw_unload(struct nwrap_cache *nwrap);
 714
 715 /* shadow */
 716 #if defined(HAVE_SHADOW_H) && defined(HAVE_GETSPNAM)
 717 struct nwrap_sp {
 718         struct nwrap_cache *cache;
 719
 720         struct spwd *list;
 721         int num;
 722         int idx;
 723 };
 724
 725 struct nwrap_cache __nwrap_cache_sp;
 726 struct nwrap_sp nwrap_sp_global;
 727
 728 static bool nwrap_sp_parse_line(struct nwrap_cache *nwrap, char *line);
 729 static void nwrap_sp_unload(struct nwrap_cache *nwrap);
 730 #endif /* defined(HAVE_SHADOW_H) && defined(HAVE_GETSPNAM) */
 731
 732 /* group */
 733 struct nwrap_gr {
 734         struct nwrap_cache *cache;
 735
 736         struct group *list;
 737         int num;
 738         int idx;
 739 };
 740
 741 struct nwrap_cache __nwrap_cache_gr;
 742 struct nwrap_gr nwrap_gr_global;
 743
 744 /* hosts */
 745 static bool nwrap_he_parse_line(struct nwrap_cache *nwrap, char *line);
 746 static void nwrap_he_unload(struct nwrap_cache *nwrap);
 747
 748 struct nwrap_addrdata {
 749         unsigned char host_addr[16]; /* IPv4 or IPv6 address */
 750 };
 751
 752 static size_t max_hostents = 100;
 753
 754 struct nwrap_entdata {
 755         struct nwrap_addrdata addr;
 756         struct hostent ht;
 757
 758         struct nwrap_vector nwrap_addrdata;
 759
 760         ssize_t aliases_count;
 761 };
 762
 763 struct nwrap_entlist {
 764         struct nwrap_entlist *next;
 765         struct nwrap_entdata *ed;
 766 };
 767
 768 struct nwrap_he {
 769         struct nwrap_cache *cache;
 770
 771         struct nwrap_entdata *list;
 772         struct nwrap_vector entdata;
 773
 774         int num;
 775         int idx;
 776 };
 777
 778 static struct nwrap_cache __nwrap_cache_he;
 779 static struct nwrap_he nwrap_he_global;
 780
 781
 782 /*********************************************************
 783  * NWRAP PROTOTYPES
 784  *********************************************************/
 785
 786 static void nwrap_init(void);
 787 static bool nwrap_gr_parse_line(struct nwrap_cache *nwrap, char *line);
 788 static void nwrap_gr_unload(struct nwrap_cache *nwrap);
 789 void nwrap_destructor(void) DESTRUCTOR_ATTRIBUTE;
 790
 791 /*********************************************************
 792  * NWRAP LIBC LOADER FUNCTIONS
 793  *********************************************************/
 794
 795 enum nwrap_lib {
 796     NWRAP_LIBC,
 797     NWRAP_LIBNSL,
 798     NWRAP_LIBSOCKET,
 799 };
 800
 801 #ifndef NDEBUG
 802 static const char *nwrap_str_lib(enum nwrap_lib lib)
 803 {
 804         switch (lib) {
 805         case NWRAP_LIBC:
 806                 return "libc";
 807         case NWRAP_LIBNSL:
 808                 return "libnsl";
 809         case NWRAP_LIBSOCKET:
 810                 return "libsocket";
 811         }
 812
 813         /* Compiler would warn us about unhandled enum value if we get here */
 814         return "unknown";
 815 }
 816 #endif
 817
 818 static void *nwrap_load_lib_handle(enum nwrap_lib lib)
 819 {
 820         int flags = RTLD_LAZY;
 821         void *handle = NULL;
 822         int i;
 823
 824 #ifdef RTLD_DEEPBIND
 825         flags |= RTLD_DEEPBIND;
 826 #endif
 827
 828         switch (lib) {
 829         case NWRAP_LIBNSL:
 830 #ifdef HAVE_LIBNSL
 831                 handle = nwrap_main_global->libc->nsl_handle;
 832                 if (handle == NULL) {
 833                         for (i = 10; i >= 0; i--) {
 834                                 char soname[256] = {0};
 835
 836                                 snprintf(soname, sizeof(soname), "libnsl.so.%d", i);
 837                                 handle = dlopen(soname, flags);
 838                                 if (handle != NULL) {
 839                                         break;
 840                                 }
 841                         }
 842
 843                         nwrap_main_global->libc->nsl_handle = handle;
 844                 }
 845                 break;
 846 #endif
 847                 /* FALL TROUGH */
 848         case NWRAP_LIBSOCKET:
 849 #ifdef HAVE_LIBSOCKET
 850                 handle = nwrap_main_global->libc->sock_handle;
 851                 if (handle == NULL) {
 852                         for (i = 10; i >= 0; i--) {
 853                                 char soname[256] = {0};
 854
 855                                 snprintf(soname, sizeof(soname), "libsocket.so.%d", i);
 856                                 handle = dlopen(soname, flags);
 857                                 if (handle != NULL) {
 858                                         break;
 859                                 }
 860                         }
 861
 862                         nwrap_main_global->libc->sock_handle = handle;
 863                 }
 864                 break;
 865 #endif
 866                 /* FALL TROUGH */
 867         case NWRAP_LIBC:
 868                 handle = nwrap_main_global->libc->handle;
 869                 if (handle == NULL) {
 870                         for (i = 10; i >= 0; i--) {
 871                                 char soname[256] = {0};
 872
 873                                 snprintf(soname, sizeof(soname), "libc.so.%d", i);
 874                                 handle = dlopen(soname, flags);
 875                                 if (handle != NULL) {
 876                                         break;
 877                                 }
 878                         }
 879
 880                         nwrap_main_global->libc->handle = handle;
 881                 }
 882                 break;
 883         }
 884
 885         if (handle == NULL) {
 886 #ifdef RTLD_NEXT
 887                 handle = nwrap_main_global->libc->handle
 888                        = nwrap_main_global->libc->sock_handle
 889                        = nwrap_main_global->libc->nsl_handle
 890                        = RTLD_NEXT;
 891 #else
 892                 NWRAP_LOG(NWRAP_LOG_ERROR,
 893                           "Failed to dlopen library: %s\n",
 894                           dlerror());
 895                 exit(-1);
 896 #endif
 897         }
 898
 899         return handle;
 900 }
 901
 902 static void *_nwrap_load_lib_function(enum nwrap_lib lib, const char *fn_name)
 903 {
 904         void *handle;
 905         void *func;
 906
 907         nwrap_init();
 908
 909         handle = nwrap_load_lib_handle(lib);
 910
 911         func = dlsym(handle, fn_name);
 912         if (func == NULL) {
 913                 NWRAP_LOG(NWRAP_LOG_ERROR,
 914                                 "Failed to find %s: %s\n",
 915                                 fn_name, dlerror());
 916                 exit(-1);
 917         }
 918
 919         NWRAP_LOG(NWRAP_LOG_TRACE,
 920                         "Loaded %s from %s",
 921                         fn_name, nwrap_str_lib(lib));
 922         return func;
 923 }
 924
 925 #define nwrap_load_lib_function(lib, fn_name) \
 926         if (nwrap_main_global->libc->fns->_libc_##fn_name == NULL) { \
 927                 *(void **) (&nwrap_main_global->libc->fns->_libc_##fn_name) = \
 928                         _nwrap_load_lib_function(lib, #fn_name); \
 929         }
 930
 931 /* INTERNAL HELPER FUNCTIONS */
 932 static void nwrap_lines_unload(struct nwrap_cache *const nwrap)
 933 {
 934         size_t p;
 935         void *item;
 936         nwrap_vector_foreach(item, nwrap->lines, p) {
 937                 /* Maybe some vectors were merged ... */
 938                 SAFE_FREE(item);
 939         }
 940         SAFE_FREE(nwrap->lines.items);
 941         ZERO_STRUCTP(&nwrap->lines);
 942 }
 943
 944 /*
 945  * IMPORTANT
 946  *
 947  * Functions expeciall from libc need to be loaded individually, you can't load
 948  * all at once or gdb will segfault at startup. The same applies to valgrind and
 949  * has probably something todo with with the linker.
 950  * So we need load each function at the point it is called the first time.
 951  */
 952 static struct passwd *libc_getpwnam(const char *name)
 953 {
 954         nwrap_load_lib_function(NWRAP_LIBC, getpwnam);
 955
 956         return nwrap_main_global->libc->fns->_libc_getpwnam(name);
 957 }
 958
 959 #ifdef HAVE_GETPWNAM_R
 960 static int libc_getpwnam_r(const char *name,
 961                            struct passwd *pwd,
 962                            char *buf,
 963                            size_t buflen,
 964                            struct passwd **result)
 965 {
 966 #ifdef HAVE___POSIX_GETPWNAM_R
 967         if (nwrap_main_global->libc->fns->_libc_getpwnam_r == NULL) {
 968                 *(void **) (&nwrap_main_global->libc->fns->_libc_getpwnam_r) =
 969                         _nwrap_load_lib_function(NWRAP_LIBC, "__posix_getpwnam_r");
 970         }
 971 #else
 972         nwrap_load_lib_function(NWRAP_LIBC, getpwnam_r);
 973 #endif
 974
 975         return nwrap_main_global->libc->fns->_libc_getpwnam_r(name,
 976                                                               pwd,
 977                                                               buf,
 978                                                               buflen,
 979                                                               result);
 980 }
 981 #endif
 982
 983 static struct passwd *libc_getpwuid(uid_t uid)
 984 {
 985         nwrap_load_lib_function(NWRAP_LIBC, getpwuid);
 986
 987         return nwrap_main_global->libc->fns->_libc_getpwuid(uid);
 988 }
 989
 990 #ifdef HAVE_GETPWUID_R
 991 static int libc_getpwuid_r(uid_t uid,
 992                            struct passwd *pwd,
 993                            char *buf,
 994                            size_t buflen,
 995                            struct passwd **result)
 996 {
 997 #ifdef HAVE___POSIX_GETPWUID_R
 998         if (nwrap_main_global->libc->fns->_libc_getpwuid_r == NULL) {
 999                 *(void **) (&nwrap_main_global->libc->fns->_libc_getpwuid_r) =
1000                         _nwrap_load_lib_function(NWRAP_LIBC, "__posix_getpwuid_r");
1001         }
1002 #else
1003         nwrap_load_lib_function(NWRAP_LIBC, getpwuid_r);
1004 #endif
1005
1006         return nwrap_main_global->libc->fns->_libc_getpwuid_r(uid,
1007                                                               pwd,
1008                                                               buf,
1009                                                               buflen,
1010                                                               result);
1011 }
1012 #endif
1013
1014 static inline void str_tolower(char *dst, char *src)
1015 {
1016         register char *src_tmp = src;
1017         register char *dst_tmp = dst;
1018
1019         while (*src_tmp != '\0') {
1020                 *dst_tmp = tolower(*src_tmp);
1021                 ++src_tmp;
1022                 ++dst_tmp;
1023         }
1024 }
1025
1026 static bool str_tolower_copy(char **dst_name, const char *const src_name)
1027 {
1028         char *h_name_lower;
1029
1030         if ((dst_name == NULL) || (src_name == NULL)) {
1031                 return false;
1032         }
1033
1034         h_name_lower = strdup(src_name);
1035         if (h_name_lower == NULL) {
1036                 NWRAP_LOG(NWRAP_LOG_DEBUG, "Out of memory while strdup");
1037                 return false;
1038         }
1039
1040         str_tolower(h_name_lower, h_name_lower);
1041         *dst_name = h_name_lower;
1042         return true;
1043 }
1044
1045 static void libc_setpwent(void)
1046 {
1047         nwrap_load_lib_function(NWRAP_LIBC, setpwent);
1048
1049         nwrap_main_global->libc->fns->_libc_setpwent();
1050 }
1051
1052 static struct passwd *libc_getpwent(void)
1053 {
1054         nwrap_load_lib_function(NWRAP_LIBC, getpwent);
1055
1056         return nwrap_main_global->libc->fns->_libc_getpwent();
1057 }
1058
1059 #ifdef HAVE_SOLARIS_GETPWENT_R
1060 static struct passwd *libc_getpwent_r(struct passwd *pwdst,
1061                                       char *buf,
1062                                       int buflen)
1063 {
1064         nwrap_load_lib_function(NWRAP_LIBC, getpwent_r);
1065
1066         return nwrap_main_global->libc->fns->_libc_getpwent_r(pwdst,
1067                                                               buf,
1068                                                               buflen);
1069 }
1070 #else /* HAVE_SOLARIS_GETPWENT_R */
1071 static int libc_getpwent_r(struct passwd *pwdst,
1072                            char *buf,
1073                            size_t buflen,
1074                            struct passwd **pwdstp)
1075 {
1076         nwrap_load_lib_function(NWRAP_LIBC, getpwent_r);
1077
1078         return nwrap_main_global->libc->fns->_libc_getpwent_r(pwdst,
1079                                                               buf,
1080                                                               buflen,
1081                                                               pwdstp);
1082 }
1083 #endif /* HAVE_SOLARIS_GETPWENT_R */
1084
1085 static void libc_endpwent(void)
1086 {
1087         nwrap_load_lib_function(NWRAP_LIBC, endpwent);
1088
1089         nwrap_main_global->libc->fns->_libc_endpwent();
1090 }
1091
1092 static int libc_initgroups(const char *user, gid_t gid)
1093 {
1094         nwrap_load_lib_function(NWRAP_LIBC, initgroups);
1095
1096         return nwrap_main_global->libc->fns->_libc_initgroups(user, gid);
1097 }
1098
1099 static struct group *libc_getgrnam(const char *name)
1100 {
1101         nwrap_load_lib_function(NWRAP_LIBC, getgrnam);
1102
1103         return nwrap_main_global->libc->fns->_libc_getgrnam(name);
1104 }
1105
1106 #ifdef HAVE_GETGRNAM_R
1107 static int libc_getgrnam_r(const char *name,
1108                            struct group *grp,
1109                            char *buf,
1110                            size_t buflen,
1111                            struct group **result)
1112 {
1113 #ifdef HAVE___POSIX_GETGRNAM_R
1114         if (nwrap_main_global->libc->fns->_libc_getgrnam_r == NULL) {
1115                 *(void **) (&nwrap_main_global->libc->fns->_libc_getgrnam_r) =
1116                         _nwrap_load_lib_function(NWRAP_LIBC, "__posix_getgrnam_r");
1117         }
1118 #else
1119         nwrap_load_lib_function(NWRAP_LIBC, getgrnam_r);
1120 #endif
1121
1122         return nwrap_main_global->libc->fns->_libc_getgrnam_r(name,
1123                                                               grp,
1124                                                               buf,
1125                                                               buflen,
1126                                                               result);
1127 }
1128 #endif
1129
1130 static struct group *libc_getgrgid(gid_t gid)
1131 {
1132         nwrap_load_lib_function(NWRAP_LIBC, getgrgid);
1133
1134         return nwrap_main_global->libc->fns->_libc_getgrgid(gid);
1135 }
1136
1137 #ifdef HAVE_GETGRGID_R
1138 static int libc_getgrgid_r(gid_t gid,
1139                            struct group *grp,
1140                            char *buf,
1141                            size_t buflen,
1142                            struct group **result)
1143 {
1144 #ifdef HAVE___POSIX_GETGRGID_R
1145         if (nwrap_main_global->libc->fns->_libc_getgrgid_r == NULL) {
1146                 *(void **) (&nwrap_main_global->libc->fns->_libc_getgrgid_r) =
1147                         _nwrap_load_lib_function(NWRAP_LIBC, "__posix_getgrgid_r");
1148         }
1149 #else
1150         nwrap_load_lib_function(NWRAP_LIBC, getgrgid_r);
1151 #endif
1152
1153         return nwrap_main_global->libc->fns->_libc_getgrgid_r(gid,
1154                                                               grp,
1155                                                               buf,
1156                                                               buflen,
1157                                                               result);
1158 }
1159 #endif
1160
1161 static void libc_setgrent(void)
1162 {
1163         nwrap_load_lib_function(NWRAP_LIBC, setgrent);
1164
1165         nwrap_main_global->libc->fns->_libc_setgrent();
1166 }
1167
1168 static struct group *libc_getgrent(void)
1169 {
1170         nwrap_load_lib_function(NWRAP_LIBC, getgrent);
1171
1172         return nwrap_main_global->libc->fns->_libc_getgrent();
1173 }
1174
1175 #ifdef HAVE_GETGRENT_R
1176 #ifdef HAVE_SOLARIS_GETGRENT_R
1177 static struct group *libc_getgrent_r(struct group *group,
1178                                      char *buf,
1179                                      size_t buflen)
1180 {
1181         nwrap_load_lib_function(NWRAP_LIBC, getgrent_r);
1182
1183         return nwrap_main_global->libc->fns->_libc_getgrent_r(group,
1184                                                               buf,
1185                                                               buflen);
1186 }
1187 #else /* !HAVE_SOLARIS_GETGRENT_R */
1188 static int libc_getgrent_r(struct group *group,
1189                            char *buf,
1190                            size_t buflen,
1191                            struct group **result)
1192 {
1193         nwrap_load_lib_function(NWRAP_LIBC, getgrent_r);
1194
1195         return nwrap_main_global->libc->fns->_libc_getgrent_r(group,
1196                                                               buf,
1197                                                               buflen,
1198                                                               result);
1199 }
1200 #endif /* HAVE_SOLARIS_GETGRENT_R */
1201 #endif /* HAVE_GETGRENT_R */
1202
1203 static void libc_endgrent(void)
1204 {
1205         nwrap_load_lib_function(NWRAP_LIBC, endgrent);
1206
1207         nwrap_main_global->libc->fns->_libc_endgrent();
1208 }
1209
1210 #ifdef HAVE_GETGROUPLIST
1211 static int libc_getgrouplist(const char *user,
1212                              gid_t group,
1213                              gid_t *groups,
1214                              int *ngroups)
1215 {
1216         nwrap_load_lib_function(NWRAP_LIBC, getgrouplist);
1217
1218         return nwrap_main_global->libc->fns->_libc_getgrouplist(user,
1219                                                                 group,
1220                                                                 groups,
1221                                                                 ngroups);
1222 }
1223 #endif
1224
1225 static void libc_sethostent(int stayopen)
1226 {
1227         nwrap_load_lib_function(NWRAP_LIBNSL, sethostent);
1228
1229         nwrap_main_global->libc->fns->_libc_sethostent(stayopen);
1230 }
1231
1232 static struct hostent *libc_gethostent(void)
1233 {
1234         nwrap_load_lib_function(NWRAP_LIBNSL, gethostent);
1235
1236         return nwrap_main_global->libc->fns->_libc_gethostent();
1237 }
1238
1239 static void libc_endhostent(void)
1240 {
1241         nwrap_load_lib_function(NWRAP_LIBNSL, endhostent);
1242
1243         nwrap_main_global->libc->fns->_libc_endhostent();
1244 }
1245
1246 static struct hostent *libc_gethostbyname(const char *name)
1247 {
1248         nwrap_load_lib_function(NWRAP_LIBNSL, gethostbyname);
1249
1250         return nwrap_main_global->libc->fns->_libc_gethostbyname(name);
1251 }
1252
1253 #ifdef HAVE_GETHOSTBYNAME2 /* GNU extension */
1254 static struct hostent *libc_gethostbyname2(const char *name, int af)
1255 {
1256         nwrap_load_lib_function(NWRAP_LIBNSL, gethostbyname2);
1257
1258         return nwrap_main_global->libc->fns->_libc_gethostbyname2(name, af);
1259 }
1260 #endif
1261
1262 static struct hostent *libc_gethostbyaddr(const void *addr,
1263                                           socklen_t len,
1264                                           int type)
1265 {
1266         nwrap_load_lib_function(NWRAP_LIBNSL, gethostbyaddr);
1267
1268         return nwrap_main_global->libc->fns->_libc_gethostbyaddr(addr,
1269                                                                  len,
1270                                                                  type);
1271 }
1272
1273 static int libc_gethostname(char *name, size_t len)
1274 {
1275         nwrap_load_lib_function(NWRAP_LIBNSL, gethostname);
1276
1277         return nwrap_main_global->libc->fns->_libc_gethostname(name, len);
1278 }
1279
1280 #ifdef HAVE_GETHOSTBYNAME_R
1281 static int libc_gethostbyname_r(const char *name,
1282                                 struct hostent *ret,
1283                                 char *buf,
1284                                 size_t buflen,
1285                                 struct hostent **result,
1286                                 int *h_errnop)
1287 {
1288         nwrap_load_lib_function(NWRAP_LIBNSL, gethostbyname_r);
1289
1290         return nwrap_main_global->libc->fns->_libc_gethostbyname_r(name,
1291                                                                    ret,
1292                                                                    buf,
1293                                                                    buflen,
1294                                                                    result,
1295                                                                    h_errnop);
1296 }
1297 #endif
1298
1299 #ifdef HAVE_GETHOSTBYADDR_R
1300 static int libc_gethostbyaddr_r(const void *addr,
1301                                 socklen_t len,
1302                                 int type,
1303                                 struct hostent *ret,
1304                                 char *buf,
1305                                 size_t buflen,
1306                                 struct hostent **result,
1307                                 int *h_errnop)
1308 {
1309         nwrap_load_lib_function(NWRAP_LIBNSL, gethostbyaddr_r);
1310
1311         return nwrap_main_global->libc->fns->_libc_gethostbyaddr_r(addr,
1312                                                                    len,
1313                                                                    type,
1314                                                                    ret,
1315                                                                    buf,
1316                                                                    buflen,
1317                                                                    result,
1318                                                                    h_errnop);
1319 }
1320 #endif
1321
1322 static int libc_getaddrinfo(const char *node,
1323                             const char *service,
1324                             const struct addrinfo *hints,
1325                             struct addrinfo **res)
1326 {
1327         nwrap_load_lib_function(NWRAP_LIBSOCKET, getaddrinfo);
1328
1329         return nwrap_main_global->libc->fns->_libc_getaddrinfo(node,
1330                                                                service,
1331                                                                hints,
1332                                                                res);
1333 }
1334
1335 static int libc_getnameinfo(const struct sockaddr *sa,
1336                             socklen_t salen,
1337                             char *host,
1338                             size_t hostlen,
1339                             char *serv,
1340                             size_t servlen,
1341                             int flags)
1342 {
1343         nwrap_load_lib_function(NWRAP_LIBSOCKET, getnameinfo);
1344
1345         return nwrap_main_global->libc->fns->_libc_getnameinfo(sa,
1346                                                                salen,
1347                                                                host,
1348                                                                hostlen,
1349                                                                serv,
1350                                                                servlen,
1351                                                                flags);
1352 }
1353
1354 /*********************************************************
1355  * NWRAP NSS MODULE LOADER FUNCTIONS
1356  *********************************************************/
1357
1358 static void *nwrap_load_module_fn(struct nwrap_backend *b,
1359                                   const char *fn_name)
1360 {
1361         void *res;
1362         char *s;
1363
1364         if (!b->so_handle) {
1365                 NWRAP_LOG(NWRAP_LOG_ERROR, "No handle");
1366                 return NULL;
1367         }
1368
1369         if (asprintf(&s, "_nss_%s_%s", b->name, fn_name) == -1) {
1370                 NWRAP_LOG(NWRAP_LOG_ERROR, "Out of memory");
1371                 return NULL;
1372         }
1373
1374         res = dlsym(b->so_handle, s);
1375         if (!res) {
1376                 NWRAP_LOG(NWRAP_LOG_ERROR,
1377                           "Cannot find function %s in %s",
1378                           s, b->so_path);
1379         }
1380         SAFE_FREE(s);
1381         return res;
1382 }
1383
1384 static struct nwrap_module_nss_fns *nwrap_load_module_fns(struct nwrap_backend *b)
1385 {
1386         struct nwrap_module_nss_fns *fns;
1387
1388         if (!b->so_handle) {
1389                 return NULL;
1390         }
1391
1392         fns = (struct nwrap_module_nss_fns *)malloc(sizeof(struct nwrap_module_nss_fns));
1393         if (!fns) {
1394                 return NULL;
1395         }
1396
1397         *(void **)(&fns->_nss_getpwnam_r) =
1398                 nwrap_load_module_fn(b, "getpwnam_r");
1399         *(void **)(&fns->_nss_getpwuid_r) =
1400                 nwrap_load_module_fn(b, "getpwuid_r");
1401         *(void **)(&fns->_nss_setpwent) =
1402                 nwrap_load_module_fn(b, "setpwent");
1403         *(void **)(&fns->_nss_getpwent_r) =
1404                 nwrap_load_module_fn(b, "getpwent_r");
1405         *(void **)(&fns->_nss_endpwent) =
1406                 nwrap_load_module_fn(b, "endpwent");
1407         *(void **)(&fns->_nss_initgroups) =
1408                 nwrap_load_module_fn(b, "initgroups_dyn");
1409         *(void **)(&fns->_nss_getgrnam_r) =
1410                 nwrap_load_module_fn(b, "getgrnam_r");
1411         *(void **)(&fns->_nss_getgrgid_r)=
1412                 nwrap_load_module_fn(b, "getgrgid_r");
1413         *(void **)(&fns->_nss_setgrent) =
1414                 nwrap_load_module_fn(b, "setgrent");
1415         *(void **)(&fns->_nss_getgrent_r) =
1416                 nwrap_load_module_fn(b, "getgrent_r");
1417         *(void **)(&fns->_nss_endgrent) =
1418                 nwrap_load_module_fn(b, "endgrent");
1419
1420         return fns;
1421 }
1422
1423 static void *nwrap_load_module(const char *so_path)
1424 {
1425         void *h;
1426
1427         if (!so_path || !strlen(so_path)) {
1428                 return NULL;
1429         }
1430
1431         h = dlopen(so_path, RTLD_LAZY);
1432         if (!h) {
1433                 NWRAP_LOG(NWRAP_LOG_ERROR,
1434                           "Cannot open shared library %s",
1435                           so_path);
1436                 return NULL;
1437         }
1438
1439         return h;
1440 }
1441
1442 static bool nwrap_module_init(const char *name,
1443                               struct nwrap_ops *ops,
1444                               const char *so_path,
1445                               int *num_backends,
1446                               struct nwrap_backend **backends)
1447 {
1448         struct nwrap_backend *b;
1449
1450         *backends = (struct nwrap_backend *)realloc(*backends,
1451                 sizeof(struct nwrap_backend) * ((*num_backends) + 1));
1452         if (!*backends) {
1453                 NWRAP_LOG(NWRAP_LOG_ERROR, "Out of memory");
1454                 return false;
1455         }
1456
1457         b = &((*backends)[*num_backends]);
1458
1459         b->name = name;
1460         b->ops = ops;
1461         b->so_path = so_path;
1462
1463         if (so_path != NULL) {
1464                 b->so_handle = nwrap_load_module(so_path);
1465                 b->fns = nwrap_load_module_fns(b);
1466                 if (b->fns == NULL) {
1467                         return false;
1468                 }
1469         } else {
1470                 b->so_handle = NULL;
1471                 b->fns = NULL;
1472         }
1473
1474         (*num_backends)++;
1475
1476         return true;
1477 }
1478
1479 static void nwrap_libc_init(struct nwrap_main *r)
1480 {
1481         r->libc = malloc(sizeof(struct nwrap_libc));
1482         if (r->libc == NULL) {
1483                 printf("Failed to allocate memory for libc");
1484                 exit(-1);
1485         }
1486         ZERO_STRUCTP(r->libc);
1487
1488         r->libc->fns = malloc(sizeof(struct nwrap_libc_fns));
1489         if (r->libc->fns == NULL) {
1490                 printf("Failed to allocate memory for libc functions");
1491                 exit(-1);
1492         }
1493         ZERO_STRUCTP(r->libc->fns);
1494 }
1495
1496 static void nwrap_backend_init(struct nwrap_main *r)
1497 {
1498         const char *module_so_path = getenv("NSS_WRAPPER_MODULE_SO_PATH");
1499         const char *module_fn_name = getenv("NSS_WRAPPER_MODULE_FN_PREFIX");
1500
1501         r->num_backends = 0;
1502         r->backends = NULL;
1503
1504         if (!nwrap_module_init("files", &nwrap_files_ops, NULL,
1505                                &r->num_backends,
1506                                &r->backends)) {
1507                 NWRAP_LOG(NWRAP_LOG_ERROR,
1508                           "Failed to initialize 'files' backend");
1509                 return;
1510         }
1511
1512         if (module_so_path != NULL &&
1513             module_so_path[0] != '\0' &&
1514             module_fn_name != NULL &&
1515             module_fn_name[0] != '\0') {
1516                 if (!nwrap_module_init(module_fn_name,
1517                                        &nwrap_module_ops,
1518                                        module_so_path,
1519                                        &r->num_backends,
1520                                        &r->backends)) {
1521                         NWRAP_LOG(NWRAP_LOG_ERROR,
1522                                   "Failed to initialize '%s' backend",
1523                                   module_fn_name);
1524                         return;
1525                 }
1526         }
1527 }
1528
1529 static void nwrap_init(void)
1530 {
1531         const char *env;
1532         char *endptr;
1533         size_t max_hostents_tmp;
1534
1535         NWRAP_LOCK(nwrap_initialized);
1536         if (nwrap_initialized) {
1537                 NWRAP_UNLOCK(nwrap_initialized);
1538                 return;
1539         }
1540
1541         /*
1542          * Still holding nwrap_initialized lock here.
1543          * We don't use NWRAP_(UN)LOCK_ALL macros here because we
1544          * want to avoid overhead when other threads do their job.
1545          */
1546         NWRAP_LOCK(nwrap_global);
1547         NWRAP_LOCK(nwrap_gr_global);
1548         NWRAP_LOCK(nwrap_he_global);
1549         NWRAP_LOCK(nwrap_pw_global);
1550         NWRAP_LOCK(nwrap_sp_global);
1551
1552         nwrap_initialized = true;
1553
1554         /* Initialize pthread_atfork handlers */
1555         pthread_atfork(&nwrap_thread_prepare, &nwrap_thread_parent,
1556                        &nwrap_thread_child);
1557
1558         env = getenv("NSS_WRAPPER_MAX_HOSTENTS");
1559         if (env != NULL) {
1560                 max_hostents_tmp = (size_t)strtol(env, &endptr, 10);
1561                 if (((env != '\0') && (endptr == '\0')) ||
1562                     (max_hostents_tmp == 0)) {
1563                         NWRAP_LOG(NWRAP_LOG_DEBUG,
1564                                   "Error parsing NSS_WRAPPER_MAX_HOSTENTS "
1565                                   "value or value is too small. "
1566                                   "Using default value: %lu.",
1567                                   max_hostents);
1568                 } else {
1569                         max_hostents = max_hostents_tmp;
1570                 }
1571         }
1572         /* Initialize hash table */
1573         NWRAP_LOG(NWRAP_LOG_DEBUG,
1574                   "Initializing hash table of size %lu items.", max_hostents);
1575         if (hcreate(max_hostents) == 0) {
1576                 NWRAP_LOG(NWRAP_LOG_ERROR,
1577                           "Failed to initialize hash table");
1578                 goto done;
1579         }
1580
1581         nwrap_main_global = &__nwrap_main_global;
1582
1583         nwrap_libc_init(nwrap_main_global);
1584
1585         nwrap_backend_init(nwrap_main_global);
1586
1587         /* passwd */
1588         nwrap_pw_global.cache = &__nwrap_cache_pw;
1589
1590         nwrap_pw_global.cache->path = getenv("NSS_WRAPPER_PASSWD");
1591         nwrap_pw_global.cache->fp = NULL;
1592         nwrap_pw_global.cache->fd = -1;
1593         nwrap_pw_global.cache->private_data = &nwrap_pw_global;
1594         nwrap_pw_global.cache->parse_line = nwrap_pw_parse_line;
1595         nwrap_pw_global.cache->unload = nwrap_pw_unload;
1596
1597         /* shadow */
1598 #if defined(HAVE_SHADOW_H) && defined(HAVE_GETSPNAM)
1599         nwrap_sp_global.cache = &__nwrap_cache_sp;
1600
1601         nwrap_sp_global.cache->path = getenv("NSS_WRAPPER_SHADOW");
1602         nwrap_sp_global.cache->fp = NULL;
1603         nwrap_sp_global.cache->fd = -1;
1604         nwrap_sp_global.cache->private_data = &nwrap_sp_global;
1605         nwrap_sp_global.cache->parse_line = nwrap_sp_parse_line;
1606         nwrap_sp_global.cache->unload = nwrap_sp_unload;
1607 #endif /* defined(HAVE_SHADOW_H) && defined(HAVE_GETSPNAM) */
1608
1609         /* group */
1610         nwrap_gr_global.cache = &__nwrap_cache_gr;
1611
1612         nwrap_gr_global.cache->path = getenv("NSS_WRAPPER_GROUP");
1613         nwrap_gr_global.cache->fp = NULL;
1614         nwrap_gr_global.cache->fd = -1;
1615         nwrap_gr_global.cache->private_data = &nwrap_gr_global;
1616         nwrap_gr_global.cache->parse_line = nwrap_gr_parse_line;
1617         nwrap_gr_global.cache->unload = nwrap_gr_unload;
1618
1619         /* hosts */
1620         nwrap_he_global.cache = &__nwrap_cache_he;
1621
1622         nwrap_he_global.cache->path = getenv("NSS_WRAPPER_HOSTS");
1623         nwrap_he_global.cache->fp = NULL;
1624         nwrap_he_global.cache->fd = -1;
1625         nwrap_he_global.cache->private_data = &nwrap_he_global;
1626         nwrap_he_global.cache->parse_line = nwrap_he_parse_line;
1627         nwrap_he_global.cache->unload = nwrap_he_unload;
1628
1629 done:
1630         /* We hold all locks here so we can use NWRAP_UNLOCK_ALL. */
1631         NWRAP_UNLOCK_ALL;
1632 }
1633
1634 bool nss_wrapper_enabled(void)
1635 {
1636         nwrap_init();
1637
1638         if (nwrap_pw_global.cache->path == NULL ||
1639             nwrap_pw_global.cache->path[0] == '\0') {
1640                 return false;
1641         }
1642         if (nwrap_gr_global.cache->path == NULL ||
1643             nwrap_gr_global.cache->path[0] == '\0') {
1644                 return false;
1645         }
1646
1647         return true;
1648 }
1649
1650 #if defined(HAVE_SHADOW_H) && defined(HAVE_GETSPNAM)
1651 bool nss_wrapper_shadow_enabled(void)
1652 {
1653         nwrap_init();
1654
1655         if (nwrap_sp_global.cache->path == NULL ||
1656             nwrap_sp_global.cache->path[0] == '\0') {
1657                 return false;
1658         }
1659
1660         return true;
1661 }
1662 #endif /* defined(HAVE_SHADOW_H) && defined(HAVE_GETSPNAM) */
1663
1664 bool nss_wrapper_hosts_enabled(void)
1665 {
1666         nwrap_init();
1667
1668         if (nwrap_he_global.cache->path == NULL ||
1669             nwrap_he_global.cache->path[0] == '\0') {
1670                 return false;
1671         }
1672
1673         return true;
1674 }
1675
1676 static bool nwrap_hostname_enabled(void)
1677 {
1678         nwrap_init();
1679
1680         if (getenv("NSS_WRAPPER_HOSTNAME") == NULL) {
1681                 return false;
1682         }
1683
1684         return true;
1685 }
1686
1687 static bool nwrap_parse_file(struct nwrap_cache *nwrap)
1688 {
1689         char *line = NULL;
1690         ssize_t n;
1691         /* Unused but getline needs it */
1692         size_t len;
1693         bool ok;
1694
1695         if (nwrap->st.st_size == 0) {
1696                 NWRAP_LOG(NWRAP_LOG_DEBUG, "size == 0");
1697                 return true;
1698         }
1699
1700         /* Support for 32-bit system I guess */
1701         if (nwrap->st.st_size > INT32_MAX) {
1702                 NWRAP_LOG(NWRAP_LOG_ERROR,
1703                           "Size[%u] larger than INT32_MAX",
1704                           (unsigned)nwrap->st.st_size);
1705                 return false;
1706         }
1707
1708         rewind(nwrap->fp);
1709
1710         do {
1711                 n = getline(&line, &len, nwrap->fp);
1712                 if (n < 0) {
1713                         SAFE_FREE(line);
1714                         if (feof(nwrap->fp)) {
1715                                 break;
1716                         }
1717
1718                         NWRAP_LOG(NWRAP_LOG_ERROR,
1719                                   "Unable to read line from file: %s",
1720                                   nwrap->path);
1721                         return false;
1722                 }
1723
1724                 if (line[n - 1] == '\n') {
1725                         line[n - 1] = '\0';
1726                 }
1727
1728                 if (line[0] == '\0') {
1729                         SAFE_FREE(line);
1730                         continue;
1731                 }
1732
1733                 ok = nwrap->parse_line(nwrap, line);
1734                 if (!ok) {
1735                         NWRAP_LOG(NWRAP_LOG_ERROR,
1736                                   "Unable to parse line file: %s",
1737                                   line);
1738                         SAFE_FREE(line);
1739                         return false;
1740                 }
1741
1742                 /* Line is parsed without issues so add it to list */
1743                 ok = nwrap_vector_add_item(&(nwrap->lines), (void *const) line);
1744                 if (!ok) {
1745                         NWRAP_LOG(NWRAP_LOG_ERROR,
1746                                   "Unable to add line to vector");
1747                         return false;
1748                 }
1749
1750                 /* This forces getline to allocate new memory for line. */
1751                 line = NULL;
1752         } while (!feof(nwrap->fp));
1753
1754         return true;
1755 }
1756
1757 static void nwrap_files_cache_unload(struct nwrap_cache *nwrap)
1758 {
1759         nwrap->unload(nwrap);
1760
1761         nwrap_lines_unload(nwrap);
1762 }
1763
1764 static bool nwrap_files_cache_reload(struct nwrap_cache *nwrap)
1765 {
1766         struct stat st;
1767         int ret;
1768         bool ok;
1769         bool retried = false;
1770
1771         assert(nwrap != NULL);
1772
1773 reopen:
1774         if (nwrap->fd < 0) {
1775                 nwrap->fp = fopen(nwrap->path, "re");
1776                 if (nwrap->fp == NULL) {
1777                         nwrap->fd = -1;
1778                         NWRAP_LOG(NWRAP_LOG_ERROR,
1779                                   "Unable to open '%s' readonly %d:%s",
1780                                   nwrap->path, nwrap->fd,
1781                                   strerror(errno));
1782                         return false;
1783
1784                 }
1785                 nwrap->fd = fileno(nwrap->fp);
1786                 NWRAP_LOG(NWRAP_LOG_DEBUG, "Open '%s'", nwrap->path);
1787         }
1788
1789         ret = fstat(nwrap->fd, &st);
1790         if (ret != 0) {
1791                 NWRAP_LOG(NWRAP_LOG_ERROR,
1792                           "fstat(%s) - %d:%s",
1793                           nwrap->path,
1794                           ret,
1795                           strerror(errno));
1796                 fclose(nwrap->fp);
1797                 nwrap->fp = NULL;
1798                 nwrap->fd = -1;
1799                 return false;
1800         }
1801
1802         if (retried == false && st.st_nlink == 0) {
1803                 /* maybe someone has replaced the file... */
1804                 NWRAP_LOG(NWRAP_LOG_TRACE,
1805                           "st_nlink == 0, reopen %s",
1806                           nwrap->path);
1807                 retried = true;
1808                 memset(&nwrap->st, 0, sizeof(nwrap->st));
1809                 fclose(nwrap->fp);
1810                 nwrap->fp = NULL;
1811                 nwrap->fd = -1;
1812                 goto reopen;
1813         }
1814
1815         if (st.st_mtime == nwrap->st.st_mtime) {
1816                 NWRAP_LOG(NWRAP_LOG_TRACE,
1817                           "st_mtime[%u] hasn't changed, skip reload",
1818                           (unsigned)st.st_mtime);
1819                 return true;
1820         }
1821
1822         NWRAP_LOG(NWRAP_LOG_TRACE,
1823                   "st_mtime has changed [%u] => [%u], start reload",
1824                   (unsigned)st.st_mtime,
1825                   (unsigned)nwrap->st.st_mtime);
1826
1827         nwrap->st = st;
1828
1829         nwrap_files_cache_unload(nwrap);
1830
1831         ok = nwrap_parse_file(nwrap);
1832         if (!ok) {
1833                 NWRAP_LOG(NWRAP_LOG_ERROR, "Failed to reload %s", nwrap->path);
1834                 nwrap_files_cache_unload(nwrap);
1835                 return false;
1836         }
1837
1838         NWRAP_LOG(NWRAP_LOG_TRACE, "Reloaded %s", nwrap->path);
1839         return true;
1840 }
1841
1842 /*
1843  * the caller has to call nwrap_unload() on failure
1844  */
1845 static bool nwrap_pw_parse_line(struct nwrap_cache *nwrap, char *line)
1846 {
1847         struct nwrap_pw *nwrap_pw;
1848         char *c;
1849         char *p;
1850         char *e;
1851         struct passwd *pw;
1852         size_t list_size;
1853
1854         nwrap_pw = (struct nwrap_pw *)nwrap->private_data;
1855
1856         list_size = sizeof(*nwrap_pw->list) * (nwrap_pw->num+1);
1857         pw = (struct passwd *)realloc(nwrap_pw->list, list_size);
1858         if (!pw) {
1859                 NWRAP_LOG(NWRAP_LOG_ERROR,
1860                           "realloc(%u) failed",
1861                           (unsigned)list_size);
1862                 return false;
1863         }
1864         nwrap_pw->list = pw;
1865
1866         pw = &nwrap_pw->list[nwrap_pw->num];
1867
1868         c = line;
1869
1870         /* name */
1871         p = strchr(c, ':');
1872         if (!p) {
1873                 NWRAP_LOG(NWRAP_LOG_ERROR,
1874                           "Invalid line[%s]: '%s'",
1875                           line,
1876                           c);
1877                 return false;
1878         }
1879         *p = '\0';
1880         p++;
1881         pw->pw_name = c;
1882         c = p;
1883
1884         NWRAP_LOG(NWRAP_LOG_TRACE, "name[%s]\n", pw->pw_name);
1885
1886         /* password */
1887         p = strchr(c, ':');
1888         if (!p) {
1889                 NWRAP_LOG(NWRAP_LOG_ERROR, "Invalid line[%s]: '%s'", line, c);
1890                 return false;
1891         }
1892         *p = '\0';
1893         p++;
1894         pw->pw_passwd = c;
1895         c = p;
1896
1897         NWRAP_LOG(NWRAP_LOG_TRACE, "password[%s]\n", pw->pw_passwd);
1898
1899         /* uid */
1900         p = strchr(c, ':');
1901         if (!p) {
1902                 NWRAP_LOG(NWRAP_LOG_ERROR, "Invalid line[%s]: '%s'", line, c);
1903                 return false;
1904         }
1905         *p = '\0';
1906         p++;
1907         e = NULL;
1908         pw->pw_uid = (uid_t)strtoul(c, &e, 10);
1909         if (c == e) {
1910                 NWRAP_LOG(NWRAP_LOG_ERROR,
1911                           "Invalid line[%s]: '%s' - %s",
1912                           line, c, strerror(errno));
1913                 return false;
1914         }
1915         if (e == NULL) {
1916                 NWRAP_LOG(NWRAP_LOG_ERROR,
1917                           "Invalid line[%s]: '%s' - %s",
1918                           line, c, strerror(errno));
1919                 return false;
1920         }
1921         if (e[0] != '\0') {
1922                 NWRAP_LOG(NWRAP_LOG_ERROR,
1923                           "Invalid line[%s]: '%s' - %s",
1924                           line, c, strerror(errno));
1925                 return false;
1926         }
1927         c = p;
1928
1929         NWRAP_LOG(NWRAP_LOG_TRACE, "uid[%u]", pw->pw_uid);
1930
1931         /* gid */
1932         p = strchr(c, ':');
1933         if (!p) {
1934                 NWRAP_LOG(NWRAP_LOG_ERROR, "Invalid line[%s]: '%s'", line, c);
1935                 return false;
1936         }
1937         *p = '\0';
1938         p++;
1939         e = NULL;
1940         pw->pw_gid = (gid_t)strtoul(c, &e, 10);
1941         if (c == e) {
1942                 NWRAP_LOG(NWRAP_LOG_ERROR,
1943                           "Invalid line[%s]: '%s' - %s",
1944                           line, c, strerror(errno));
1945                 return false;
1946         }
1947         if (e == NULL) {
1948                 NWRAP_LOG(NWRAP_LOG_ERROR,
1949                           "Invalid line[%s]: '%s' - %s",
1950                           line, c, strerror(errno));
1951                 return false;
1952         }
1953         if (e[0] != '\0') {
1954                 NWRAP_LOG(NWRAP_LOG_ERROR,
1955                           "Invalid line[%s]: '%s' - %s",
1956                           line, c, strerror(errno));
1957                 return false;
1958         }
1959         c = p;
1960
1961         NWRAP_LOG(NWRAP_LOG_TRACE, "gid[%u]\n", pw->pw_gid);
1962
1963         /* gecos */
1964         p = strchr(c, ':');
1965         if (!p) {
1966                 NWRAP_LOG(NWRAP_LOG_ERROR, "invalid line[%s]: '%s'", line, c);
1967                 return false;
1968         }
1969         *p = '\0';
1970         p++;
1971         pw->pw_gecos = c;
1972         c = p;
1973
1974         NWRAP_LOG(NWRAP_LOG_TRACE, "gecos[%s]", pw->pw_gecos);
1975
1976         /* dir */
1977         p = strchr(c, ':');
1978         if (!p) {
1979                 NWRAP_LOG(NWRAP_LOG_ERROR, "'%s'", c);
1980                 return false;
1981         }
1982         *p = '\0';
1983         p++;
1984         pw->pw_dir = c;
1985         c = p;
1986
1987         NWRAP_LOG(NWRAP_LOG_TRACE, "dir[%s]", pw->pw_dir);
1988
1989         /* shell */
1990         pw->pw_shell = c;
1991         NWRAP_LOG(NWRAP_LOG_TRACE, "shell[%s]", pw->pw_shell);
1992
1993         NWRAP_LOG(NWRAP_LOG_DEBUG,
1994                   "Added user[%s:%s:%u:%u:%s:%s:%s]",
1995                   pw->pw_name, pw->pw_passwd,
1996                   pw->pw_uid, pw->pw_gid,
1997                   pw->pw_gecos, pw->pw_dir, pw->pw_shell);
1998
1999         nwrap_pw->num++;
2000         return true;
2001 }
2002
2003 static void nwrap_pw_unload(struct nwrap_cache *nwrap)
2004 {
2005         struct nwrap_pw *nwrap_pw;
2006         nwrap_pw = (struct nwrap_pw *)nwrap->private_data;
2007
2008         SAFE_FREE(nwrap_pw->list);
2009         nwrap_pw->num = 0;
2010         nwrap_pw->idx = 0;
2011 }
2012
2013 static int nwrap_pw_copy_r(const struct passwd *src, struct passwd *dst,
2014                            char *buf, size_t buflen, struct passwd **dstp)
2015 {
2016         char *first;
2017         char *last;
2018         off_t ofs;
2019
2020         first = src->pw_name;
2021
2022         last = src->pw_shell;
2023         while (*last) last++;
2024
2025         ofs = PTR_DIFF(last + 1, first);
2026
2027         if (ofs > (off_t) buflen) {
2028                 return ERANGE;
2029         }
2030
2031         memcpy(buf, first, ofs);
2032
2033         ofs = PTR_DIFF(src->pw_name, first);
2034         dst->pw_name = buf + ofs;
2035         ofs = PTR_DIFF(src->pw_passwd, first);
2036         dst->pw_passwd = buf + ofs;
2037         dst->pw_uid = src->pw_uid;
2038         dst->pw_gid = src->pw_gid;
2039         ofs = PTR_DIFF(src->pw_gecos, first);
2040         dst->pw_gecos = buf + ofs;
2041         ofs = PTR_DIFF(src->pw_dir, first);
2042         dst->pw_dir = buf + ofs;
2043         ofs = PTR_DIFF(src->pw_shell, first);
2044         dst->pw_shell = buf + ofs;
2045
2046         if (dstp) {
2047                 *dstp = dst;
2048         }
2049
2050         return 0;
2051 }
2052
2053 #if defined(HAVE_SHADOW_H) && defined(HAVE_GETSPNAM)
2054 static bool nwrap_sp_parse_line(struct nwrap_cache *nwrap, char *line)
2055 {
2056         struct nwrap_sp *nwrap_sp;
2057         struct spwd *sp;
2058         size_t list_size;
2059         char *c;
2060         char *e;
2061         char *p;
2062
2063         nwrap_sp = (struct nwrap_sp *)nwrap->private_data;
2064
2065         list_size = sizeof(*nwrap_sp->list) * (nwrap_sp->num+1);
2066         sp = (struct spwd *)realloc(nwrap_sp->list, list_size);
2067         if (sp == NULL) {
2068                 NWRAP_LOG(NWRAP_LOG_ERROR,
2069                           "realloc(%u) failed",
2070                           (unsigned)list_size);
2071                 return false;
2072         }
2073         nwrap_sp->list = sp;
2074
2075         sp = &nwrap_sp->list[nwrap_sp->num];
2076
2077         c = line;
2078
2079         /* name */
2080         p = strchr(c, ':');
2081         if (p == NULL) {
2082                 NWRAP_LOG(NWRAP_LOG_ERROR,
2083                           "name -- Invalid line[%s]: '%s'",
2084                           line,
2085                           c);
2086                 return false;
2087         }
2088         *p = '\0';
2089         p++;
2090         sp->sp_namp = c;
2091         c = p;
2092
2093         NWRAP_LOG(NWRAP_LOG_TRACE, "name[%s]\n", sp->sp_namp);
2094
2095         /* pwd */
2096         p = strchr(c, ':');
2097         if (p == NULL) {
2098                 NWRAP_LOG(NWRAP_LOG_ERROR,
2099                           "pwd -- Invalid line[%s]: '%s'",
2100                           line,
2101                           c);
2102                 return false;
2103         }
2104         *p = '\0';
2105         p++;
2106         sp->sp_pwdp = c;
2107         c = p;
2108
2109         /* lstchg (long) */
2110         if (c[0] == ':') {
2111                 sp->sp_lstchg = -1;
2112                 p++;
2113         } else {
2114                 p = strchr(c, ':');
2115                 if (p == NULL) {
2116                         NWRAP_LOG(NWRAP_LOG_ERROR,
2117                                   "lstchg -- Invalid line[%s]: '%s'",
2118                                   line,
2119                                   c);
2120                         return false;
2121                 }
2122                 *p = '\0';
2123                 p++;
2124                 sp->sp_lstchg = strtol(c, &e, 10);
2125                 if (c == e) {
2126                         NWRAP_LOG(NWRAP_LOG_ERROR,
2127                                   "lstchg -- Invalid line[%s]: '%s' - %s",
2128                                   line, c, strerror(errno));
2129                         return false;
2130                 }
2131                 if (e == NULL) {
2132                         NWRAP_LOG(NWRAP_LOG_ERROR,
2133                                   "lstchg -- Invalid line[%s]: '%s' - %s",
2134                                   line, c, strerror(errno));
2135                         return false;
2136                 }
2137                 if (e[0] != '\0') {
2138                         NWRAP_LOG(NWRAP_LOG_ERROR,
2139                                   "lstchg -- Invalid line[%s]: '%s' - %s",
2140                                   line, c, strerror(errno));
2141                         return false;
2142                 }
2143         }
2144         c = p;
2145
2146         /* min (long) */
2147         if (c[0] == ':') {
2148                 sp->sp_min = -1;
2149                 p++;
2150         } else {
2151                 p = strchr(c, ':');
2152                 if (p == NULL) {
2153                         NWRAP_LOG(NWRAP_LOG_ERROR,
2154                                   "min -- Invalid line[%s]: '%s'",
2155                                   line,
2156                                   c);
2157                         return false;
2158                 }
2159                 *p = '\0';
2160                 p++;
2161                 sp->sp_min = strtol(c, &e, 10);
2162                 if (c == e) {
2163                         NWRAP_LOG(NWRAP_LOG_ERROR,
2164                                   "min -- Invalid line[%s]: '%s' - %s",
2165                                   line, c, strerror(errno));
2166                         return false;
2167                 }
2168                 if (e == NULL) {
2169                         NWRAP_LOG(NWRAP_LOG_ERROR,
2170                                   "min -- Invalid line[%s]: '%s' - %s",
2171                                   line, c, strerror(errno));
2172                         return false;
2173                 }
2174                 if (e[0] != '\0') {
2175                         NWRAP_LOG(NWRAP_LOG_ERROR,
2176                                   "min -- Invalid line[%s]: '%s' - %s",
2177                                   line, c, strerror(errno));
2178                         return false;
2179                 }
2180         }
2181         c = p;
2182
2183         /* max (long) */
2184         if (c[0] == ':') {
2185                 sp->sp_max = -1;
2186                 p++;
2187         } else {
2188                 p = strchr(c, ':');
2189                 if (p == NULL) {
2190                         NWRAP_LOG(NWRAP_LOG_ERROR,
2191                                   "max -- Invalid line[%s]: '%s'",
2192                                   line,
2193                                   c);
2194                         return false;
2195                 }
2196                 *p = '\0';
2197                 p++;
2198                 sp->sp_max = strtol(c, &e, 10);
2199                 if (c == e) {
2200                         NWRAP_LOG(NWRAP_LOG_ERROR,
2201                                   "max -- Invalid line[%s]: '%s' - %s",
2202                                   line, c, strerror(errno));
2203                         return false;
2204                 }
2205                 if (e == NULL) {
2206                         NWRAP_LOG(NWRAP_LOG_ERROR,
2207                                   "max -- Invalid line[%s]: '%s' - %s",
2208                                   line, c, strerror(errno));
2209                         return false;
2210                 }
2211                 if (e[0] != '\0') {
2212                         NWRAP_LOG(NWRAP_LOG_ERROR,
2213                                   "max -- Invalid line[%s]: '%s' - %s",
2214                                   line, c, strerror(errno));
2215                         return false;
2216                 }
2217         }
2218         c = p;
2219
2220         /* warn (long) */
2221         if (c[0] == ':') {
2222                 sp->sp_warn = -1;
2223                 p++;
2224         } else {
2225                 p = strchr(c, ':');
2226                 if (p == NULL) {
2227                         NWRAP_LOG(NWRAP_LOG_ERROR,
2228                                   "warn -- Invalid line[%s]: '%s'",
2229                                   line,
2230                                   c);
2231                         return false;
2232                 }
2233                 *p = '\0';
2234                 p++;
2235                 sp->sp_warn = strtol(c, &e, 10);
2236                 if (c == e) {
2237                         NWRAP_LOG(NWRAP_LOG_ERROR,
2238                                   "warn -- Invalid line[%s]: '%s' - %s",
2239                                   line, c, strerror(errno));
2240                         return false;
2241                 }
2242                 if (e == NULL) {
2243                         NWRAP_LOG(NWRAP_LOG_ERROR,
2244                                   "warn -- Invalid line[%s]: '%s' - %s",
2245                                   line, c, strerror(errno));
2246                         return false;
2247                 }
2248                 if (e[0] != '\0') {
2249                         NWRAP_LOG(NWRAP_LOG_ERROR,
2250                                   "warn -- Invalid line[%s]: '%s' - %s",
2251                                   line, c, strerror(errno));
2252                         return false;
2253                 }
2254         }
2255         c = p;
2256
2257         /* inact (long) */
2258         if (c[0] == ':') {
2259                 sp->sp_inact = -1;
2260                 p++;
2261         } else {
2262                 p = strchr(c, ':');
2263                 if (p == NULL) {
2264                         NWRAP_LOG(NWRAP_LOG_ERROR,
2265                                   "inact -- Invalid line[%s]: '%s'",
2266                                   line,
2267                                   c);
2268                         return false;
2269                 }
2270                 *p = '\0';
2271                 p++;
2272                 sp->sp_inact = strtol(c, &e, 10);
2273                 if (c == e) {
2274                         NWRAP_LOG(NWRAP_LOG_ERROR,
2275                                   "inact -- Invalid line[%s]: '%s' - %s",
2276                                   line, c, strerror(errno));
2277                         return false;
2278                 }
2279                 if (e == NULL) {
2280                         NWRAP_LOG(NWRAP_LOG_ERROR,
2281                                   "inact -- Invalid line[%s]: '%s' - %s",
2282                                   line, c, strerror(errno));
2283                         return false;
2284                 }
2285                 if (e[0] != '\0') {
2286                         NWRAP_LOG(NWRAP_LOG_ERROR,
2287                                   "inact -- Invalid line[%s]: '%s' - %s",
2288                                   line, c, strerror(errno));
2289                         return false;
2290                 }
2291         }
2292         c = p;
2293
2294         /* expire (long) */
2295         if (c[0] == ':') {
2296                 sp->sp_expire = -1;
2297                 p++;
2298         } else {
2299                 p = strchr(c, ':');
2300                 if (p == NULL) {
2301                         NWRAP_LOG(NWRAP_LOG_ERROR,
2302                                   "expire -- Invalid line[%s]: '%s'",
2303                                   line,
2304                                   c);
2305                         return false;
2306                 }
2307                 *p = '\0';
2308                 p++;
2309                 sp->sp_expire = strtol(c, &e, 10);
2310                 if (c == e) {
2311                         NWRAP_LOG(NWRAP_LOG_ERROR,
2312                                   "expire -- Invalid line[%s]: '%s' - %s",
2313                                   line, c, strerror(errno));
2314                         return false;
2315                 }
2316                 if (e == NULL) {
2317                         NWRAP_LOG(NWRAP_LOG_ERROR,
2318                                   "expire -- Invalid line[%s]: '%s' - %s",
2319                                   line, c, strerror(errno));
2320                         return false;
2321                 }
2322                 if (e[0] != '\0') {
2323                         NWRAP_LOG(NWRAP_LOG_ERROR,
2324                                   "expire -- Invalid line[%s]: '%s' - %s",
2325                                   line, c, strerror(errno));
2326                         return false;
2327                 }
2328         }
2329         c = p;
2330
2331         nwrap_sp->num++;
2332         return true;
2333 }
2334
2335 static void nwrap_sp_unload(struct nwrap_cache *nwrap)
2336 {
2337         struct nwrap_sp *nwrap_sp;
2338         nwrap_sp = (struct nwrap_sp *)nwrap->private_data;
2339
2340         SAFE_FREE(nwrap_sp->list);
2341         nwrap_sp->num = 0;
2342         nwrap_sp->idx = 0;
2343 }
2344 #endif /* defined(HAVE_SHADOW_H) && defined(HAVE_GETSPNAM) */
2345
2346 /*
2347  * the caller has to call nwrap_unload() on failure
2348  */
2349 static bool nwrap_gr_parse_line(struct nwrap_cache *nwrap, char *line)
2350 {
2351         struct nwrap_gr *nwrap_gr;
2352         char *c;
2353         char *p;
2354         char *e;
2355         struct group *gr;
2356         size_t list_size;
2357         unsigned nummem;
2358
2359         nwrap_gr = (struct nwrap_gr *)nwrap->private_data;
2360
2361         list_size = sizeof(*nwrap_gr->list) * (nwrap_gr->num+1);
2362         gr = (struct group *)realloc(nwrap_gr->list, list_size);
2363         if (!gr) {
2364                 NWRAP_LOG(NWRAP_LOG_ERROR, "realloc failed");
2365                 return false;
2366         }
2367         nwrap_gr->list = gr;
2368
2369         gr = &nwrap_gr->list[nwrap_gr->num];
2370
2371         c = line;
2372
2373         /* name */
2374         p = strchr(c, ':');
2375         if (!p) {
2376                 NWRAP_LOG(NWRAP_LOG_ERROR, "Invalid line[%s]: '%s'", line, c);
2377                 return false;
2378         }
2379         *p = '\0';
2380         p++;
2381         gr->gr_name = c;
2382         c = p;
2383
2384         NWRAP_LOG(NWRAP_LOG_TRACE, "name[%s]", gr->gr_name);
2385
2386         /* password */
2387         p = strchr(c, ':');
2388         if (!p) {
2389                 NWRAP_LOG(NWRAP_LOG_ERROR, "Invalid line[%s]: '%s'", line, c);
2390                 return false;
2391         }
2392         *p = '\0';
2393         p++;
2394         gr->gr_passwd = c;
2395         c = p;
2396
2397         NWRAP_LOG(NWRAP_LOG_TRACE, "password[%s]", gr->gr_passwd);
2398
2399         /* gid */
2400         p = strchr(c, ':');
2401         if (!p) {
2402                 NWRAP_LOG(NWRAP_LOG_ERROR, "Invalid line[%s]: '%s'", line, c);
2403                 return false;
2404         }
2405         *p = '\0';
2406         p++;
2407         e = NULL;
2408         gr->gr_gid = (gid_t)strtoul(c, &e, 10);
2409         if (c == e) {
2410                 NWRAP_LOG(NWRAP_LOG_ERROR,
2411                           "Invalid line[%s]: '%s' - %s",
2412                           line, c, strerror(errno));
2413                 return false;
2414         }
2415         if (e == NULL) {
2416                 NWRAP_LOG(NWRAP_LOG_ERROR,
2417                           "Invalid line[%s]: '%s' - %s",
2418                           line, c, strerror(errno));
2419                 return false;
2420         }
2421         if (e[0] != '\0') {
2422                 NWRAP_LOG(NWRAP_LOG_ERROR,
2423                           "Invalid line[%s]: '%s' - %s",
2424                           line, c, strerror(errno));
2425                 return false;
2426         }
2427         c = p;
2428
2429         NWRAP_LOG(NWRAP_LOG_TRACE, "gid[%u]", gr->gr_gid);
2430
2431         /* members */
2432         gr->gr_mem = (char **)malloc(sizeof(char *));
2433         if (!gr->gr_mem) {
2434                 NWRAP_LOG(NWRAP_LOG_ERROR, "Out of memory");
2435                 return false;
2436         }
2437         gr->gr_mem[0] = NULL;
2438
2439         for(nummem=0; p; nummem++) {
2440                 char **m;
2441                 size_t m_size;
2442                 c = p;
2443                 p = strchr(c, ',');
2444                 if (p) {
2445                         *p = '\0';
2446                         p++;
2447                 }
2448
2449                 if (strlen(c) == 0) {
2450                         break;
2451                 }
2452
2453                 m_size = sizeof(char *) * (nummem+2);
2454                 m = (char **)realloc(gr->gr_mem, m_size);
2455                 if (!m) {
2456                         NWRAP_LOG(NWRAP_LOG_ERROR,
2457                                   "realloc(%zd) failed",
2458                                   m_size);
2459                         return false;
2460                 }
2461                 gr->gr_mem = m;
2462                 gr->gr_mem[nummem] = c;
2463                 gr->gr_mem[nummem+1] = NULL;
2464
2465                 NWRAP_LOG(NWRAP_LOG_TRACE,
2466                           "member[%u]: '%s'",
2467                           nummem, gr->gr_mem[nummem]);
2468         }
2469
2470         NWRAP_LOG(NWRAP_LOG_DEBUG,
2471                   "Added group[%s:%s:%u:] with %u members",
2472                   gr->gr_name, gr->gr_passwd, gr->gr_gid, nummem);
2473
2474         nwrap_gr->num++;
2475         return true;
2476 }
2477
2478 static void nwrap_gr_unload(struct nwrap_cache *nwrap)
2479 {
2480         int i;
2481         struct nwrap_gr *nwrap_gr;
2482         nwrap_gr = (struct nwrap_gr *)nwrap->private_data;
2483
2484         if (nwrap_gr->list) {
2485                 for (i=0; i < nwrap_gr->num; i++) {
2486                         SAFE_FREE(nwrap_gr->list[i].gr_mem);
2487                 }
2488                 SAFE_FREE(nwrap_gr->list);
2489         }
2490
2491         nwrap_gr->num = 0;
2492         nwrap_gr->idx = 0;
2493 }
2494
2495 static int nwrap_gr_copy_r(const struct group *src, struct group *dst,
2496                            char *buf, size_t buflen, struct group **dstp)
2497 {
2498         char *first;
2499         char **lastm;
2500         char *last = NULL;
2501         off_t ofsb;
2502         off_t ofsm;
2503         off_t ofs;
2504         unsigned i;
2505
2506         first = src->gr_name;
2507
2508         lastm = src->gr_mem;
2509         while (*lastm) {
2510                 last = *lastm;
2511                 lastm++;
2512         }
2513
2514         if (last == NULL) {
2515                 last = src->gr_passwd;
2516         }
2517         while (*last) last++;
2518
2519         ofsb = PTR_DIFF(last + 1, first);
2520         ofsm = PTR_DIFF(lastm + 1, src->gr_mem);
2521
2522         if ((ofsb + ofsm) > (off_t) buflen) {
2523                 return ERANGE;
2524         }
2525
2526         memcpy(buf, first, ofsb);
2527         memcpy(buf + ofsb, src->gr_mem, ofsm);
2528
2529         ofs = PTR_DIFF(src->gr_name, first);
2530         dst->gr_name = buf + ofs;
2531         ofs = PTR_DIFF(src->gr_passwd, first);
2532         dst->gr_passwd = buf + ofs;
2533         dst->gr_gid = src->gr_gid;
2534
2535         dst->gr_mem = (char **)(buf + ofsb);
2536         for (i=0; src->gr_mem[i]; i++) {
2537                 ofs = PTR_DIFF(src->gr_mem[i], first);
2538                 dst->gr_mem[i] = buf + ofs;
2539         }
2540
2541         if (dstp) {
2542                 *dstp = dst;
2543         }
2544
2545         return 0;
2546 }
2547
2548 static struct nwrap_entlist *nwrap_entlist_init(struct nwrap_entdata *ed)
2549 {
2550         struct nwrap_entlist *el;
2551
2552         if (ed == NULL) {
2553                 NWRAP_LOG(NWRAP_LOG_ERROR,
2554                           "entry is NULL, can't create list item");
2555                 return NULL;
2556         }
2557
2558         el = (struct nwrap_entlist *)malloc(sizeof(struct nwrap_entlist));
2559         if (el == NULL) {
2560                 NWRAP_LOG(NWRAP_LOG_ERROR, "malloc failed");
2561                 return NULL;
2562         }
2563
2564         el->next = NULL;
2565         el->ed = ed;
2566
2567         return el;
2568 }
2569
2570 static bool nwrap_add_ai(char *const ip_addr, struct nwrap_entdata *const ed)
2571 {
2572         ENTRY e;
2573         ENTRY *p;
2574         struct nwrap_entlist *el;
2575
2576         if (ip_addr == NULL) {
2577                 NWRAP_LOG(NWRAP_LOG_ERROR, "ip_addr NULL - can't add");
2578                 return false;
2579         }
2580
2581         el = nwrap_entlist_init(ed);
2582         if (el == NULL) {
2583                 return false;
2584         }
2585
2586         e.key = ip_addr;
2587         e.data = el;
2588
2589         p = hsearch(e, ENTER);
2590         if (p == NULL) {
2591                 NWRAP_LOG(NWRAP_LOG_ERROR, "Hash table is full");
2592                 return false;
2593         }
2594
2595         return true;
2596 }
2597
2598
2599 static bool nwrap_ed_inventarize_add_new(char *const h_name,
2600                                          struct nwrap_entdata *const ed)
2601 {
2602         ENTRY e;
2603         ENTRY *p;
2604         struct nwrap_entlist *el;
2605
2606         if (h_name == NULL) {
2607                 NWRAP_LOG(NWRAP_LOG_ERROR, "h_name NULL - can't add");
2608                 return false;
2609         }
2610
2611         el = nwrap_entlist_init(ed);
2612         if (el == NULL) {
2613                 return false;
2614         }
2615
2616         e.key = h_name;
2617         e.data = (void *)el;
2618
2619         p = hsearch(e, ENTER);
2620         if (p == NULL) {
2621                 NWRAP_LOG(NWRAP_LOG_ERROR, "Hash table is full!");
2622                 return false;
2623         }
2624
2625         return true;
2626 }
2627
2628 static bool nwrap_ed_inventarize_add_to_existing(struct nwrap_entdata *const ed,
2629                                                  struct nwrap_entlist *const el)
2630 {
2631         struct nwrap_entlist *cursor;
2632         struct nwrap_entlist *el_new;
2633
2634         if (el == NULL) {
2635                 NWRAP_LOG(NWRAP_LOG_ERROR, "list is NULL, can not add");
2636                 return false;
2637         }
2638
2639         el_new = nwrap_entlist_init(ed);
2640         if (el_new == NULL) {
2641                 return false;
2642         }
2643
2644         for (cursor = el; cursor->next != NULL; cursor = cursor->next)
2645         {
2646                 if (cursor->ed == ed) {
2647                         free(el_new);
2648                         return false;
2649                 }
2650         }
2651
2652         if (cursor->ed == ed) {
2653                 free(el_new);
2654                 return false;
2655         }
2656
2657         cursor->next = el_new;
2658         return true;
2659 }
2660
2661 static bool nwrap_ed_inventarize(char *const name,
2662                                  struct nwrap_entdata *const ed)
2663 {
2664         ENTRY e;
2665         ENTRY *p;
2666         bool ok;
2667
2668         e.key = name;
2669         e.data = NULL;
2670
2671         NWRAP_LOG(NWRAP_LOG_DEBUG, "Searching name: %s", e.key);
2672
2673         p = hsearch(e, FIND);
2674         if (p == NULL) {
2675                 NWRAP_LOG(NWRAP_LOG_DEBUG, "Name %s not found. Adding...", name);
2676                 ok = nwrap_ed_inventarize_add_new(name, ed);
2677         } else {
2678                 struct nwrap_entlist *el = (struct nwrap_entlist *)p->data;
2679
2680                 NWRAP_LOG(NWRAP_LOG_DEBUG, "Name %s found. Add record to list.", name);
2681                 ok = nwrap_ed_inventarize_add_to_existing(ed, el);
2682         }
2683
2684         return ok;
2685 }
2686
2687 static bool nwrap_add_hname(struct nwrap_entdata *const ed)
2688 {
2689         char *const h_name = (char *const)(ed->ht.h_name);
2690         unsigned i;
2691         bool ok;
2692
2693         ok = nwrap_ed_inventarize(h_name, ed);
2694         if (!ok) {
2695                 return false;
2696         }
2697
2698         if (ed->ht.h_aliases == NULL) {
2699                 return true;
2700         }
2701
2702         /* Itemize aliases */
2703         for (i = 0; ed->ht.h_aliases[i] != NULL; ++i) {
2704                 char *h_name_alias;
2705
2706                 h_name_alias = ed->ht.h_aliases[i];
2707
2708                 NWRAP_LOG(NWRAP_LOG_DEBUG, "Add alias: %s", h_name_alias);
2709
2710                 if (!nwrap_ed_inventarize(h_name_alias, ed)) {
2711                         NWRAP_LOG(NWRAP_LOG_ERROR,
2712                                   "Unable to add alias: %s", h_name_alias);
2713                         return false;
2714                 }
2715         }
2716
2717         return true;
2718 }
2719
2720 static bool nwrap_he_parse_line(struct nwrap_cache *nwrap, char *line)
2721 {
2722         struct nwrap_he *nwrap_he = (struct nwrap_he *)nwrap->private_data;
2723         bool do_aliases = true;
2724         ssize_t aliases_count = 0;
2725         char *p;
2726         char *i;
2727         char *n;
2728
2729         char *ip;
2730         bool ok;
2731
2732         struct nwrap_entdata *ed = (struct nwrap_entdata *)
2733                                    malloc(sizeof(struct nwrap_entdata));
2734         if (ed == NULL) {
2735                 NWRAP_LOG(NWRAP_LOG_ERROR,
2736                           "Unable to allocate memory for nwrap_entdata");
2737                 return false;
2738         }
2739         ZERO_STRUCTP(ed);
2740
2741         i = line;
2742
2743         /*
2744          * IP
2745          */
2746
2747         /* Walk to first char */
2748         for (p = i; *p != '.' && *p != ':' && !isxdigit((int) *p); p++) {
2749                 if (*p == '\0') {
2750                         NWRAP_LOG(NWRAP_LOG_ERROR,
2751                                   "Invalid line[%s]: '%s'",
2752                                   line, i);
2753                         free(ed);
2754                         return false;
2755                 }
2756         }
2757
2758         for (i = p; !isspace((int)*p); p++) {
2759                 if (*p == '\0') {
2760                         NWRAP_LOG(NWRAP_LOG_ERROR,
2761                                   "Invalid line[%s]: '%s'",
2762                                   line, i);
2763                         free(ed);
2764                         return false;
2765                 }
2766         }
2767
2768         *p = '\0';
2769
2770         if (inet_pton(AF_INET, i, ed->addr.host_addr)) {
2771                 ed->ht.h_addrtype = AF_INET;
2772                 ed->ht.h_length = 4;
2773 #ifdef HAVE_IPV6
2774         } else if (inet_pton(AF_INET6, i, ed->addr.host_addr)) {
2775                 ed->ht.h_addrtype = AF_INET6;
2776                 ed->ht.h_length = 16;
2777 #endif
2778         } else {
2779                 NWRAP_LOG(NWRAP_LOG_ERROR,
2780                           "Invalid line[%s]: '%s'",
2781                           line, i);
2782
2783                 free(ed);
2784                 return false;
2785         }
2786         ip = i;
2787
2788         nwrap_vector_add_item(&(ed->nwrap_addrdata),
2789                               (void *const)ed->addr.host_addr);
2790         ed->ht.h_addr_list = nwrap_vector_head(&ed->nwrap_addrdata);
2791
2792         p++;
2793
2794         /*
2795          * FQDN
2796          */
2797
2798         /* Walk to first char */
2799         for (n = p; *p != '_' && !isalnum((int) *p); p++) {
2800                 if (*p == '\0') {
2801                         NWRAP_LOG(NWRAP_LOG_ERROR,
2802                                   "Invalid line[%s]: '%s'",
2803                                   line, n);
2804
2805                         free(ed);
2806                         return false;
2807                 }
2808         }
2809
2810         for (n = p; !isspace((int)*p); p++) {
2811                 if (*p == '\0') {
2812                         do_aliases = false;
2813                         break;
2814                 }
2815         }
2816
2817         *p = '\0';
2818
2819         /* Convert to lowercase. This operate on same memory region */
2820         str_tolower(n, n);
2821         ed->ht.h_name = n;
2822
2823         /* glib's getent always dereferences he->h_aliases */
2824         ed->ht.h_aliases = malloc(sizeof(char *));
2825         if (ed->ht.h_aliases == NULL) {
2826                 free(ed);
2827                 return false;
2828         }
2829         ed->ht.h_aliases[0] = NULL;
2830
2831         /*
2832          * Aliases
2833          */
2834         while (do_aliases) {
2835                 char **aliases;
2836                 char *a;
2837
2838                 p++;
2839
2840                 /* Walk to first char */
2841                 for (a = p; *p != '_' && !isalnum((int) *p); p++) {
2842                         if (*p == '\0') {
2843                                 do_aliases = false;
2844                                 break;
2845                         }
2846                 }
2847                 /* Only trailing spaces are left */
2848                 if (!do_aliases) {
2849                         break;
2850                 }
2851
2852                 for (a = p; !isspace((int)*p); p++) {
2853                         if (*p == '\0') {
2854                                 do_aliases = false;
2855                                 break;
2856                         }
2857                 }
2858
2859                 *p = '\0';
2860
2861                 aliases = realloc(ed->ht.h_aliases, sizeof(char *) * (aliases_count + 2));
2862                 if (aliases == NULL) {
2863                         free(ed);
2864                         return false;
2865                 }
2866                 ed->ht.h_aliases = aliases;
2867
2868                 str_tolower(a, a);
2869                 aliases[aliases_count] = a;
2870                 aliases[aliases_count + 1] = NULL;
2871
2872                 aliases_count += 1;
2873         }
2874
2875         nwrap_vector_add_item(&(nwrap_he->entdata), (void *const)ed);
2876
2877         ed->aliases_count = aliases_count;
2878         /* Inventarize item */
2879         ok = nwrap_add_hname(ed);
2880         if (!ok) {
2881                 return false;
2882         }
2883
2884         ok = nwrap_add_ai(ip, ed);
2885         if (!ok) {
2886                 return false;
2887         }
2888
2889         nwrap_he->num++;
2890         return true;
2891 }
2892
2893 static void nwrap_he_unload(struct nwrap_cache *nwrap)
2894 {
2895         struct nwrap_he *nwrap_he =
2896                 (struct nwrap_he *)nwrap->private_data;
2897         struct nwrap_entdata *ed;
2898         size_t i;
2899
2900         nwrap_vector_foreach (ed, nwrap_he->entdata, i)
2901         {
2902                 SAFE_FREE(ed->nwrap_addrdata.items);
2903                 SAFE_FREE(ed->ht.h_aliases);
2904                 SAFE_FREE(ed);
2905         }
2906         SAFE_FREE(nwrap_he->entdata.items);
2907         nwrap_he->entdata.count = nwrap_he->entdata.capacity = 0;
2908
2909         nwrap_he->num = 0;
2910         nwrap_he->idx = 0;
2911 }
2912
2913
2914 /* user functions */
2915 static struct passwd *nwrap_files_getpwnam(struct nwrap_backend *b,
2916                                            const char *name)
2917 {
2918         int i;
2919         bool ok;
2920
2921         (void) b; /* unused */
2922
2923         NWRAP_LOG(NWRAP_LOG_DEBUG, "Lookup user %s in files", name);
2924
2925         ok = nwrap_files_cache_reload(nwrap_pw_global.cache);
2926         if (!ok) {
2927                 NWRAP_LOG(NWRAP_LOG_ERROR, "Error loading passwd file");
2928                 return NULL;
2929         }
2930
2931         for (i=0; i<nwrap_pw_global.num; i++) {
2932                 if (strcmp(nwrap_pw_global.list[i].pw_name, name) == 0) {
2933                         NWRAP_LOG(NWRAP_LOG_DEBUG, "user[%s] found", name);
2934                         return &nwrap_pw_global.list[i];
2935                 }
2936                 NWRAP_LOG(NWRAP_LOG_DEBUG,
2937                           "user[%s] does not match [%s]",
2938                           name,
2939                           nwrap_pw_global.list[i].pw_name);
2940         }
2941
2942         NWRAP_LOG(NWRAP_LOG_DEBUG, "user[%s] not found\n", name);
2943
2944         errno = ENOENT;
2945         return NULL;
2946 }
2947
2948 static int nwrap_files_getpwnam_r(struct nwrap_backend *b,
2949                                   const char *name, struct passwd *pwdst,
2950                                   char *buf, size_t buflen, struct passwd **pwdstp)
2951 {
2952         struct passwd *pw;
2953
2954         pw = nwrap_files_getpwnam(b, name);
2955         if (!pw) {
2956                 if (errno == 0) {
2957                         return ENOENT;
2958                 }
2959                 return errno;
2960         }
2961
2962         return nwrap_pw_copy_r(pw, pwdst, buf, buflen, pwdstp);
2963 }
2964
2965 static struct passwd *nwrap_files_getpwuid(struct nwrap_backend *b,
2966                                            uid_t uid)
2967 {
2968         int i;
2969         bool ok;
2970
2971         (void) b; /* unused */
2972
2973         ok = nwrap_files_cache_reload(nwrap_pw_global.cache);
2974         if (!ok) {
2975                 NWRAP_LOG(NWRAP_LOG_ERROR, "Error loading passwd file");
2976                 return NULL;
2977         }
2978
2979         for (i=0; i<nwrap_pw_global.num; i++) {
2980                 if (nwrap_pw_global.list[i].pw_uid == uid) {
2981                         NWRAP_LOG(NWRAP_LOG_DEBUG, "uid[%u] found", uid);
2982                         return &nwrap_pw_global.list[i];
2983                 }
2984                 NWRAP_LOG(NWRAP_LOG_DEBUG,
2985                           "uid[%u] does not match [%u]",
2986                           uid,
2987                           nwrap_pw_global.list[i].pw_uid);
2988         }
2989
2990         NWRAP_LOG(NWRAP_LOG_DEBUG, "uid[%u] not found\n", uid);
2991
2992         errno = ENOENT;
2993         return NULL;
2994 }
2995
2996 static int nwrap_files_getpwuid_r(struct nwrap_backend *b,
2997                                   uid_t uid, struct passwd *pwdst,
2998                                   char *buf, size_t buflen, struct passwd **pwdstp)
2999 {
3000         struct passwd *pw;
3001
3002         pw = nwrap_files_getpwuid(b, uid);
3003         if (!pw) {
3004                 if (errno == 0) {
3005                         return ENOENT;
3006                 }
3007                 return errno;
3008         }
3009
3010         return nwrap_pw_copy_r(pw, pwdst, buf, buflen, pwdstp);
3011 }
3012
3013 /* user enum functions */
3014 static void nwrap_files_setpwent(struct nwrap_backend *b)
3015 {
3016         (void) b; /* unused */
3017
3018         nwrap_pw_global.idx = 0;
3019 }
3020
3021 static struct passwd *nwrap_files_getpwent(struct nwrap_backend *b)
3022 {
3023         struct passwd *pw;
3024
3025         (void) b; /* unused */
3026
3027         if (nwrap_pw_global.idx == 0) {
3028                 bool ok;
3029                 ok = nwrap_files_cache_reload(nwrap_pw_global.cache);
3030                 if (!ok) {
3031                         NWRAP_LOG(NWRAP_LOG_ERROR, "Error loading passwd file");
3032                         return NULL;
3033                 }
3034         }
3035
3036         if (nwrap_pw_global.idx >= nwrap_pw_global.num) {
3037                 errno = ENOENT;
3038                 return NULL;
3039         }
3040
3041         pw = &nwrap_pw_global.list[nwrap_pw_global.idx++];
3042
3043         NWRAP_LOG(NWRAP_LOG_DEBUG,
3044                   "return user[%s] uid[%u]",
3045                   pw->pw_name, pw->pw_uid);
3046
3047         return pw;
3048 }
3049
3050 static int nwrap_files_getpwent_r(struct nwrap_backend *b,
3051                                   struct passwd *pwdst, char *buf,
3052                                   size_t buflen, struct passwd **pwdstp)
3053 {
3054         struct passwd *pw;
3055
3056         pw = nwrap_files_getpwent(b);
3057         if (!pw) {
3058                 if (errno == 0) {
3059                         return ENOENT;
3060                 }
3061                 return errno;
3062         }
3063
3064         return nwrap_pw_copy_r(pw, pwdst, buf, buflen, pwdstp);
3065 }
3066
3067 static void nwrap_files_endpwent(struct nwrap_backend *b)
3068 {
3069         (void) b; /* unused */
3070
3071         nwrap_pw_global.idx = 0;
3072 }
3073
3074 /* shadow */
3075
3076 #if defined(HAVE_SHADOW_H) && defined(HAVE_GETSPNAM)
3077
3078 #ifdef HAVE_SETSPENT
3079 static void nwrap_files_setspent(void)
3080 {
3081         nwrap_sp_global.idx = 0;
3082 }
3083
3084 static struct spwd *nwrap_files_getspent(void)
3085 {
3086         struct spwd *sp;
3087
3088         if (nwrap_sp_global.idx == 0) {
3089                 bool ok;
3090
3091                 ok = nwrap_files_cache_reload(nwrap_sp_global.cache);
3092                 if (!ok) {
3093                         NWRAP_LOG(NWRAP_LOG_ERROR, "Error loading shadow file");
3094                         return NULL;
3095                 }
3096         }
3097
3098         if (nwrap_sp_global.idx >= nwrap_sp_global.num) {
3099                 errno = ENOENT;
3100                 return NULL;
3101         }
3102
3103         sp = &nwrap_sp_global.list[nwrap_sp_global.idx++];
3104
3105         NWRAP_LOG(NWRAP_LOG_DEBUG,
3106                   "return user[%s]",
3107                   sp->sp_namp);
3108
3109         return sp;
3110 }
3111
3112 static void nwrap_files_endspent(void)
3113 {
3114         nwrap_sp_global.idx = 0;
3115 }
3116 #endif /* HAVE_SETSPENT */
3117
3118 static struct spwd *nwrap_files_getspnam(const char *name)
3119 {
3120         int i;
3121         bool ok;
3122
3123         NWRAP_LOG(NWRAP_LOG_DEBUG, "Lookup user %s in files", name);
3124
3125         ok = nwrap_files_cache_reload(nwrap_sp_global.cache);
3126         if (!ok) {
3127                 NWRAP_LOG(NWRAP_LOG_ERROR, "Error loading shadow file");
3128                 return NULL;
3129         }
3130
3131         for (i=0; i<nwrap_sp_global.num; i++) {
3132                 if (strcmp(nwrap_sp_global.list[i].sp_namp, name) == 0) {
3133                         NWRAP_LOG(NWRAP_LOG_DEBUG, "user[%s] found", name);
3134                         return &nwrap_sp_global.list[i];
3135                 }
3136                 NWRAP_LOG(NWRAP_LOG_DEBUG,
3137                           "user[%s] does not match [%s]",
3138                           name,
3139                           nwrap_sp_global.list[i].sp_namp);
3140         }
3141
3142         NWRAP_LOG(NWRAP_LOG_DEBUG, "user[%s] not found\n", name);
3143
3144         errno = ENOENT;
3145         return NULL;
3146 }
3147 #endif /* defined(HAVE_SHADOW_H) && defined(HAVE_GETSPNAM) */
3148
3149 /* misc functions */
3150 static int nwrap_files_initgroups(struct nwrap_backend *b,
3151                                   const char *user,
3152                                   gid_t group)
3153 {
3154         struct group *grp;
3155         gid_t *groups;
3156         int size = 1;
3157         int rc;
3158
3159         groups = (gid_t *)malloc(size * sizeof(gid_t));
3160         if (groups == NULL) {
3161                 NWRAP_LOG(NWRAP_LOG_ERROR, "Out of memory");
3162                 errno = ENOMEM;
3163                 return -1;
3164         }
3165         groups[0] = group;
3166
3167         nwrap_files_setgrent(b);
3168         while ((grp = nwrap_files_getgrent(b)) != NULL) {
3169                 int i = 0;
3170
3171                 NWRAP_LOG(NWRAP_LOG_DEBUG,
3172                           "Inspecting %s for group membership",
3173                           grp->gr_name);
3174
3175                 for (i=0; grp->gr_mem && grp->gr_mem[i] != NULL; i++) {
3176                         if (group != grp->gr_gid &&
3177                             (strcmp(user, grp->gr_mem[i]) == 0)) {
3178                                 NWRAP_LOG(NWRAP_LOG_DEBUG,
3179                                           "%s is member of %s",
3180                                           user,
3181                                           grp->gr_name);
3182
3183                                 groups = (gid_t *)realloc(groups,
3184                                                           (size + 1) * sizeof(gid_t));
3185                                 if (groups == NULL) {
3186                                         NWRAP_LOG(NWRAP_LOG_ERROR,
3187                                                   "Out of memory");
3188                                         errno = ENOMEM;
3189                                         return -1;
3190                                 }
3191
3192                                 groups[size] = grp->gr_gid;
3193                                 size++;
3194                         }
3195                 }
3196         }
3197
3198         nwrap_files_endgrent(b);
3199
3200         NWRAP_LOG(NWRAP_LOG_DEBUG,
3201                   "%s is member of %d groups",
3202                   user, size);
3203
3204         /* This really only works if uid_wrapper is loaded */
3205         rc = setgroups(size, groups);
3206
3207         free(groups);
3208
3209         return rc;
3210 }
3211
3212 /* group functions */
3213 static struct group *nwrap_files_getgrnam(struct nwrap_backend *b,
3214                                           const char *name)
3215 {
3216         int i;
3217         bool ok;
3218
3219         (void) b; /* unused */
3220
3221         ok = nwrap_files_cache_reload(nwrap_gr_global.cache);
3222         if (!ok) {
3223                 NWRAP_LOG(NWRAP_LOG_ERROR, "Error loading group file");
3224                 return NULL;
3225         }
3226
3227         for (i=0; i<nwrap_gr_global.num; i++) {
3228                 if (strcmp(nwrap_gr_global.list[i].gr_name, name) == 0) {
3229                         NWRAP_LOG(NWRAP_LOG_DEBUG, "group[%s] found", name);
3230                         return &nwrap_gr_global.list[i];
3231                 }
3232                 NWRAP_LOG(NWRAP_LOG_DEBUG,
3233                           "group[%s] does not match [%s]",
3234                           name,
3235                           nwrap_gr_global.list[i].gr_name);
3236         }
3237
3238         NWRAP_LOG(NWRAP_LOG_DEBUG, "group[%s] not found", name);
3239
3240         errno = ENOENT;
3241         return NULL;
3242 }
3243
3244 static int nwrap_files_getgrnam_r(struct nwrap_backend *b,
3245                                   const char *name, struct group *grdst,
3246                                   char *buf, size_t buflen, struct group **grdstp)
3247 {
3248         struct group *gr;
3249
3250         gr = nwrap_files_getgrnam(b, name);
3251         if (!gr) {
3252                 if (errno == 0) {
3253                         return ENOENT;
3254                 }
3255                 return errno;
3256         }
3257
3258         return nwrap_gr_copy_r(gr, grdst, buf, buflen, grdstp);
3259 }
3260
3261 static struct group *nwrap_files_getgrgid(struct nwrap_backend *b,
3262                                           gid_t gid)
3263 {
3264         int i;
3265         bool ok;
3266
3267         (void) b; /* unused */
3268
3269         ok = nwrap_files_cache_reload(nwrap_gr_global.cache);
3270         if (!ok) {
3271                 NWRAP_LOG(NWRAP_LOG_ERROR, "Error loading group file");
3272                 return NULL;
3273         }
3274
3275         for (i=0; i<nwrap_gr_global.num; i++) {
3276                 if (nwrap_gr_global.list[i].gr_gid == gid) {
3277                         NWRAP_LOG(NWRAP_LOG_DEBUG, "gid[%u] found", gid);
3278                         return &nwrap_gr_global.list[i];
3279                 }
3280                 NWRAP_LOG(NWRAP_LOG_DEBUG,
3281                           "gid[%u] does not match [%u]",
3282                           gid,
3283                           nwrap_gr_global.list[i].gr_gid);
3284         }
3285
3286         NWRAP_LOG(NWRAP_LOG_DEBUG, "gid[%u] not found", gid);
3287
3288         errno = ENOENT;
3289         return NULL;
3290 }
3291
3292 static int nwrap_files_getgrgid_r(struct nwrap_backend *b,
3293                                   gid_t gid, struct group *grdst,
3294                                   char *buf, size_t buflen, struct group **grdstp)
3295 {
3296         struct group *gr;
3297
3298         gr = nwrap_files_getgrgid(b, gid);
3299         if (!gr) {
3300                 if (errno == 0) {
3301                         return ENOENT;
3302                 }
3303                 return errno;
3304         }
3305
3306         return nwrap_gr_copy_r(gr, grdst, buf, buflen, grdstp);
3307 }
3308
3309 /* group enum functions */
3310 static void nwrap_files_setgrent(struct nwrap_backend *b)
3311 {
3312         (void) b; /* unused */
3313
3314         nwrap_gr_global.idx = 0;
3315 }
3316
3317 static struct group *nwrap_files_getgrent(struct nwrap_backend *b)
3318 {
3319         struct group *gr;
3320
3321         (void) b; /* unused */
3322
3323         if (nwrap_gr_global.idx == 0) {
3324                 bool ok;
3325
3326                 ok = nwrap_files_cache_reload(nwrap_gr_global.cache);
3327                 if (!ok) {
3328                         NWRAP_LOG(NWRAP_LOG_ERROR, "Error loading group file");
3329                         return NULL;
3330                 }
3331         }
3332
3333         if (nwrap_gr_global.idx >= nwrap_gr_global.num) {
3334                 errno = ENOENT;
3335                 return NULL;
3336         }
3337
3338         gr = &nwrap_gr_global.list[nwrap_gr_global.idx++];
3339
3340         NWRAP_LOG(NWRAP_LOG_DEBUG,
3341                   "return group[%s] gid[%u]",
3342                   gr->gr_name, gr->gr_gid);
3343
3344         return gr;
3345 }
3346
3347 static int nwrap_files_getgrent_r(struct nwrap_backend *b,
3348                                   struct group *grdst, char *buf,
3349                                   size_t buflen, struct group **grdstp)
3350 {
3351         struct group *gr;
3352
3353         gr = nwrap_files_getgrent(b);
3354         if (!gr) {
3355                 if (errno == 0) {
3356                         return ENOENT;
3357                 }
3358                 return errno;
3359         }
3360
3361         return nwrap_gr_copy_r(gr, grdst, buf, buflen, grdstp);
3362 }
3363
3364 static void nwrap_files_endgrent(struct nwrap_backend *b)
3365 {
3366         (void) b; /* unused */
3367
3368         nwrap_gr_global.idx = 0;
3369 }
3370
3371 /* hosts functions */
3372 static int nwrap_files_gethostbyname(const char *name, int af,
3373                                      struct hostent *result,
3374                                      struct nwrap_vector *addr_list)
3375 {
3376         struct nwrap_entlist *el;
3377         struct hostent *he;
3378         char *h_name_lower;
3379         ENTRY e;
3380         ENTRY *e_p;
3381         char canon_name[DNS_NAME_MAX] = { 0 };
3382         size_t name_len;
3383         bool he_found = false;
3384         bool ok;
3385
3386         ok = nwrap_files_cache_reload(nwrap_he_global.cache);
3387         if (!ok) {
3388                 NWRAP_LOG(NWRAP_LOG_ERROR, "error loading hosts file");
3389                 goto no_ent;
3390         }
3391
3392         name_len = strlen(name);
3393         if (name_len < sizeof(canon_name) && name[name_len - 1] == '.') {
3394                 strncpy(canon_name, name, name_len - 1);
3395                 name = canon_name;
3396         }
3397
3398         if (!str_tolower_copy(&h_name_lower, name)) {
3399                 NWRAP_LOG(NWRAP_LOG_DEBUG,
3400                           "Out of memory while converting to lower case");
3401                 goto no_ent;
3402         }
3403
3404         /* Look at hash table for element */
3405         NWRAP_LOG(NWRAP_LOG_DEBUG, "Searching for name: %s", h_name_lower);
3406         e.key = h_name_lower;
3407         e.data = NULL;
3408         e_p = hsearch(e, FIND);
3409         if (e_p == NULL) {
3410                 NWRAP_LOG(NWRAP_LOG_DEBUG, "Name %s not found.", h_name_lower);
3411                 SAFE_FREE(h_name_lower);
3412                 goto no_ent;
3413         }
3414         SAFE_FREE(h_name_lower);
3415
3416         /* Always cleanup vector and results */
3417         if (!nwrap_vector_is_initialized(addr_list)) {
3418                 if (!nwrap_vector_init(addr_list)) {
3419                         NWRAP_LOG(NWRAP_LOG_DEBUG,
3420                                   "Unable to initialize memory for addr_list vector");
3421                         goto no_ent;
3422                 }
3423         } else {
3424                 /* When vector is initialized data are valid no more.
3425                  * Quick way how to free vector is: */
3426                 addr_list->count = 0;
3427         }
3428
3429         /* Iterate through results */
3430         for (el = (struct nwrap_entlist *)e_p->data; el != NULL; el = el->next)
3431         {
3432                 he = &(el->ed->ht);
3433
3434                 /* Filter by address familiy if provided */
3435                 if (af != AF_UNSPEC && he->h_addrtype != af) {
3436                         continue;
3437                 }
3438
3439                 /*
3440                  * GLIBC HACK?
3441                  * glibc doesn't return ipv6 addresses when AF_UNSPEC is used
3442                  */
3443                 if (af == AF_UNSPEC && he->h_addrtype != AF_INET) {
3444                         continue;
3445                 }
3446
3447                 if (!he_found) {
3448                         memcpy(result, he, sizeof(struct hostent));
3449                         NWRAP_LOG(NWRAP_LOG_DEBUG,
3450                                   "Name found. Returning record for %s",
3451                                   he->h_name);
3452                         he_found = true;
3453                 }
3454                 nwrap_vector_merge(addr_list, &el->ed->nwrap_addrdata);
3455                 result->h_addr_list = nwrap_vector_head(addr_list);
3456         }
3457
3458         if (he_found) {
3459                 return 0;
3460         }
3461         NWRAP_LOG(NWRAP_LOG_DEBUG,
3462                   "Name found in database. No records matches type.");
3463
3464 no_ent:
3465         errno = ENOENT;
3466         return -1;
3467 }
3468
3469 #ifdef HAVE_GETHOSTBYNAME_R
3470 static int nwrap_gethostbyname_r(const char *name,
3471                                  struct hostent *ret,
3472                                  char *buf, size_t buflen,
3473                                  struct hostent **result, int *h_errnop)
3474 {
3475         struct nwrap_vector *addr_list = malloc(sizeof(struct nwrap_vector));
3476         int rc;
3477
3478         if (addr_list == NULL) {
3479                 NWRAP_LOG(NWRAP_LOG_ERROR,
3480                           "Unable to allocate memory for address list");
3481                 errno = ENOENT;
3482                 return -1;
3483         }
3484
3485         ZERO_STRUCTP(addr_list);
3486
3487         rc = nwrap_files_gethostbyname(name, AF_UNSPEC, ret, addr_list);
3488         if (rc == -1) {
3489                 *h_errnop = h_errno;
3490                 if (addr_list->items != NULL) {
3491                         free(addr_list->items);
3492                 }
3493                 SAFE_FREE(addr_list);
3494                 errno = ENOENT;
3495                 return -1;
3496         }
3497
3498         if (buflen < (addr_list->count * sizeof(void *))) {
3499                 SAFE_FREE(addr_list->items);
3500                 SAFE_FREE(addr_list);
3501                 return ERANGE;
3502         }
3503
3504         /* Copy all to user provided buffer and change
3505          * pointers in returned structure.
3506          * +1 is for ending NULL pointer. */
3507         memcpy(buf, addr_list->items, (addr_list->count + 1) * sizeof(void *));
3508
3509         free(addr_list->items);
3510         free(addr_list);
3511
3512         ret->h_addr_list = (char **)buf;
3513         *result = ret;
3514         return 0;
3515 }
3516
3517 int gethostbyname_r(const char *name,
3518                     struct hostent *ret,
3519                     char *buf, size_t buflen,
3520                     struct hostent **result, int *h_errnop)
3521 {
3522         if (!nss_wrapper_hosts_enabled()) {
3523                 return libc_gethostbyname_r(name,
3524                                             ret,
3525                                             buf,
3526                                             buflen,
3527                                             result,
3528                                             h_errnop);
3529         }
3530
3531         return nwrap_gethostbyname_r(name, ret, buf, buflen, result, h_errnop);
3532 }
3533 #endif
3534
3535 static int nwrap_files_getaddrinfo(const char *name,
3536                                    unsigned short port,
3537                                    const struct addrinfo *hints,
3538                                    struct addrinfo **ai,
3539                                    struct addrinfo **ai_tail)
3540 {
3541         struct nwrap_entlist *el;
3542         struct hostent *he;
3543         struct addrinfo *_ai = NULL;
3544         struct addrinfo *ai_head = NULL;
3545         struct addrinfo *ai_prev = NULL;
3546         char *h_name_lower;
3547         size_t name_len;
3548         char canon_name[DNS_NAME_MAX] = { 0 };
3549         bool skip_canonname = false;
3550         ENTRY e = { 0 };
3551         ENTRY *e_p = NULL;
3552         int rc;
3553         bool ok;
3554
3555         ok = nwrap_files_cache_reload(nwrap_he_global.cache);
3556         if (!ok) {
3557                 NWRAP_LOG(NWRAP_LOG_ERROR, "error loading hosts file");
3558                 return EAI_SYSTEM;
3559         }
3560
3561         name_len = strlen(name);
3562         if (name_len < DNS_NAME_MAX && name[name_len - 1] == '.') {
3563                 strncpy(canon_name, name, name_len - 1);
3564                 name = canon_name;
3565         }
3566
3567         if (!str_tolower_copy(&h_name_lower, name)) {
3568                 NWRAP_LOG(NWRAP_LOG_DEBUG,
3569                           "Out of memory while converting to lower case");
3570                 return EAI_MEMORY;
3571         }
3572
3573         NWRAP_LOG(NWRAP_LOG_DEBUG, "Searching for name: %s", h_name_lower);
3574         e.key = h_name_lower;
3575         e.data = NULL;
3576         e_p = hsearch(e, FIND);
3577         if (e_p == NULL) {
3578                 NWRAP_LOG(NWRAP_LOG_DEBUG, "Name %s not found.", h_name_lower);
3579                 SAFE_FREE(h_name_lower);
3580                 errno = ENOENT;
3581                 return EAI_NONAME;
3582         }
3583         NWRAP_LOG(NWRAP_LOG_DEBUG, "Name: %s found.", h_name_lower);
3584         SAFE_FREE(h_name_lower);
3585
3586         rc = EAI_NONAME;
3587         for (el = (struct nwrap_entlist *)e_p->data; el != NULL; el = el->next)
3588         {
3589                 int rc2;
3590
3591                 he = &(el->ed->ht);
3592
3593                 if (hints->ai_family != AF_UNSPEC &&
3594                     he->h_addrtype != hints->ai_family)
3595                 {
3596                         NWRAP_LOG(NWRAP_LOG_DEBUG,
3597                                   "Entry found but with wrong AF - "
3598                                   "remembering EAI_ADDRINFO.");
3599                         rc = EAI_ADDRFAMILY;
3600                         continue;
3601                 }
3602
3603                 /* Function allocates memory and returns it in ai. */
3604                 rc2 = nwrap_convert_he_ai(he,
3605                                          port,
3606                                          hints,
3607                                          &_ai,
3608                                          skip_canonname);
3609                 if (rc2 != 0) {
3610                         NWRAP_LOG(NWRAP_LOG_ERROR, "Error converting he to ai");
3611                         return rc2;
3612                 }
3613                 skip_canonname = true;
3614
3615                 if (ai_head == NULL) {
3616                         ai_head = _ai;
3617                 }
3618                 if (ai_prev != NULL) {
3619                         ai_prev->ai_next = _ai;
3620                 }
3621                 ai_prev = _ai;
3622         }
3623
3624         if (ai_head != NULL) {
3625                 rc = 0;
3626         }
3627
3628         *ai = ai_head;
3629         *ai_tail = _ai;
3630
3631         return rc;
3632 }
3633
3634 static struct hostent *nwrap_files_gethostbyaddr(const void *addr,
3635                                                  socklen_t len, int type)
3636 {
3637         struct hostent *he;
3638         char ip[NWRAP_INET_ADDRSTRLEN] = {0};
3639         struct nwrap_entdata *ed;
3640         const char *a;
3641         size_t i;
3642         bool ok;
3643
3644         (void) len; /* unused */
3645
3646         ok = nwrap_files_cache_reload(nwrap_he_global.cache);
3647         if (!ok) {
3648                 NWRAP_LOG(NWRAP_LOG_ERROR, "error loading hosts file");
3649                 return NULL;
3650         }
3651
3652         a = inet_ntop(type, addr, ip, sizeof(ip));
3653         if (a == NULL) {
3654                 errno = EINVAL;
3655                 return NULL;
3656         }
3657
3658         nwrap_vector_foreach(ed, nwrap_he_global.entdata, i)
3659         {
3660                 he = &(ed->ht);
3661                 if (he->h_addrtype != type) {
3662                         continue;
3663                 }
3664
3665                 if (memcmp(addr, he->h_addr_list[0], he->h_length) == 0) {
3666                         return he;
3667                 }
3668         }
3669
3670         errno = ENOENT;
3671         return NULL;
3672 }
3673
3674 #ifdef HAVE_GETHOSTBYADDR_R
3675 static int nwrap_gethostbyaddr_r(const void *addr, socklen_t len, int type,
3676                                  struct hostent *ret,
3677                                  char *buf, size_t buflen,
3678                                  struct hostent **result, int *h_errnop)
3679 {
3680         *result = nwrap_files_gethostbyaddr(addr, len, type);
3681         if (*result != NULL) {
3682                 memset(buf, '\0', buflen);
3683                 *ret = **result;
3684                 return 0;
3685         } else {
3686                 *h_errnop = h_errno;
3687                 return -1;
3688         }
3689 }
3690
3691 int gethostbyaddr_r(const void *addr, socklen_t len, int type,
3692                     struct hostent *ret,
3693                     char *buf, size_t buflen,
3694                     struct hostent **result, int *h_errnop)
3695 {
3696         if (!nss_wrapper_hosts_enabled()) {
3697                 return libc_gethostbyaddr_r(addr,
3698                                             len,
3699                                             type,
3700                                             ret,
3701                                             buf,
3702                                             buflen,
3703                                             result,
3704                                             h_errnop);
3705         }
3706
3707         return nwrap_gethostbyaddr_r(addr, len, type, ret, buf, buflen, result, h_errnop);
3708 }
3709 #endif
3710
3711 /* hosts enum functions */
3712 static void nwrap_files_sethostent(void)
3713 {
3714         nwrap_he_global.idx = 0;
3715 }
3716
3717 static struct hostent *nwrap_files_gethostent(void)
3718 {
3719         struct hostent *he;
3720
3721         if (nwrap_he_global.idx == 0) {
3722                 bool ok;
3723
3724                 ok = nwrap_files_cache_reload(nwrap_he_global.cache);
3725                 if (!ok) {
3726                         NWRAP_LOG(NWRAP_LOG_ERROR, "Error loading hosts file");
3727                         return NULL;
3728                 }
3729         }
3730
3731         if (nwrap_he_global.idx >= nwrap_he_global.num) {
3732                 errno = ENOENT;
3733                 return NULL;
3734         }
3735
3736         he = &((struct nwrap_entdata *)nwrap_he_global.entdata.items[nwrap_he_global.idx++])->ht;
3737
3738         NWRAP_LOG(NWRAP_LOG_DEBUG, "return hosts[%s]", he->h_name);
3739
3740         return he;
3741 }
3742
3743 static void nwrap_files_endhostent(void)
3744 {
3745         nwrap_he_global.idx = 0;
3746 }
3747
3748 /*
3749  * module backend
3750  */
3751
3752
3753 static struct passwd *nwrap_module_getpwnam(struct nwrap_backend *b,
3754                                             const char *name)
3755 {
3756         static struct passwd pwd;
3757         static char buf[1000];
3758         NSS_STATUS status;
3759
3760         if (!b->fns->_nss_getpwnam_r) {
3761                 return NULL;
3762         }
3763
3764         status = b->fns->_nss_getpwnam_r(name, &pwd, buf, sizeof(buf), &errno);
3765         if (status == NSS_STATUS_NOTFOUND) {
3766                 return NULL;
3767         }
3768         if (status != NSS_STATUS_SUCCESS) {
3769                 return NULL;
3770         }
3771
3772         return &pwd;
3773 }
3774
3775 static int nwrap_module_getpwnam_r(struct nwrap_backend *b,
3776                                    const char *name, struct passwd *pwdst,
3777                                    char *buf, size_t buflen, struct passwd **pwdstp)
3778 {
3779         int ret;
3780
3781         (void) b; /* unused */
3782         (void) pwdst; /* unused */
3783         (void) pwdstp; /* unused */
3784
3785         if (!b->fns->_nss_getpwnam_r) {
3786                 return NSS_STATUS_NOTFOUND;
3787         }
3788
3789         ret = b->fns->_nss_getpwnam_r(name, pwdst, buf, buflen, &errno);
3790         switch (ret) {
3791         case NSS_STATUS_SUCCESS:
3792                 return 0;
3793         case NSS_STATUS_NOTFOUND:
3794                 if (errno != 0) {
3795                         return errno;
3796                 }
3797                 return ENOENT;
3798         case NSS_STATUS_TRYAGAIN:
3799                 if (errno != 0) {
3800                         return errno;
3801                 }
3802                 return ERANGE;
3803         default:
3804                 if (errno != 0) {
3805                         return errno;
3806                 }
3807                 return ret;
3808         }
3809 }
3810
3811 static struct passwd *nwrap_module_getpwuid(struct nwrap_backend *b,
3812                                             uid_t uid)
3813 {
3814         static struct passwd pwd;
3815         static char buf[1000];
3816         NSS_STATUS status;
3817
3818         if (!b->fns->_nss_getpwuid_r) {
3819                 return NULL;
3820         }
3821
3822         status = b->fns->_nss_getpwuid_r(uid, &pwd, buf, sizeof(buf), &errno);
3823         if (status == NSS_STATUS_NOTFOUND) {
3824                 return NULL;
3825         }
3826         if (status != NSS_STATUS_SUCCESS) {
3827                 return NULL;
3828         }
3829         return &pwd;
3830 }
3831
3832 static int nwrap_module_getpwuid_r(struct nwrap_backend *b,
3833                                    uid_t uid, struct passwd *pwdst,
3834                                    char *buf, size_t buflen, struct passwd **pwdstp)
3835 {
3836         int ret;
3837
3838         (void) pwdstp; /* unused */
3839
3840         if (!b->fns->_nss_getpwuid_r) {
3841                 return ENOENT;
3842         }
3843
3844         ret = b->fns->_nss_getpwuid_r(uid, pwdst, buf, buflen, &errno);
3845         switch (ret) {
3846         case NSS_STATUS_SUCCESS:
3847                 return 0;
3848         case NSS_STATUS_NOTFOUND:
3849                 if (errno != 0) {
3850                         return errno;
3851                 }
3852                 return ENOENT;
3853         case NSS_STATUS_TRYAGAIN:
3854                 if (errno != 0) {
3855                         return errno;
3856                 }
3857                 return ERANGE;
3858         default:
3859                 if (errno != 0) {
3860                         return errno;
3861                 }
3862                 return ret;
3863         }
3864 }
3865
3866 static void nwrap_module_setpwent(struct nwrap_backend *b)
3867 {
3868         if (!b->fns->_nss_setpwent) {
3869                 return;
3870         }
3871
3872         b->fns->_nss_setpwent();
3873 }
3874
3875 static struct passwd *nwrap_module_getpwent(struct nwrap_backend *b)
3876 {
3877         static struct passwd pwd;
3878         static char buf[1000];
3879         NSS_STATUS status;
3880
3881         if (!b->fns->_nss_getpwent_r) {
3882                 return NULL;
3883         }
3884
3885         status = b->fns->_nss_getpwent_r(&pwd, buf, sizeof(buf), &errno);
3886         if (status == NSS_STATUS_NOTFOUND) {
3887                 return NULL;
3888         }
3889         if (status != NSS_STATUS_SUCCESS) {
3890                 return NULL;
3891         }
3892         return &pwd;
3893 }
3894
3895 static int nwrap_module_getpwent_r(struct nwrap_backend *b,
3896                                    struct passwd *pwdst, char *buf,
3897                                    size_t buflen, struct passwd **pwdstp)
3898 {
3899         int ret;
3900
3901         (void) pwdstp; /* unused */
3902
3903         if (!b->fns->_nss_getpwent_r) {
3904                 return ENOENT;
3905         }
3906
3907         ret = b->fns->_nss_getpwent_r(pwdst, buf, buflen, &errno);
3908         switch (ret) {
3909         case NSS_STATUS_SUCCESS:
3910                 return 0;
3911         case NSS_STATUS_NOTFOUND:
3912                 if (errno != 0) {
3913                         return errno;
3914                 }
3915                 return ENOENT;
3916         case NSS_STATUS_TRYAGAIN:
3917                 if (errno != 0) {
3918                         return errno;
3919                 }
3920                 return ERANGE;
3921         default:
3922                 if (errno != 0) {
3923                         return errno;
3924                 }
3925                 return ret;
3926         }
3927 }
3928
3929 static void nwrap_module_endpwent(struct nwrap_backend *b)
3930 {
3931         if (!b->fns->_nss_endpwent) {
3932                 return;
3933         }
3934
3935         b->fns->_nss_endpwent();
3936 }
3937
3938 static int nwrap_module_initgroups(struct nwrap_backend *b,
3939                                    const char *user, gid_t group)
3940 {
3941         gid_t *groups;
3942         long int start;
3943         long int size;
3944
3945         if (!b->fns->_nss_initgroups) {
3946                 return NSS_STATUS_UNAVAIL;
3947         }
3948
3949         return b->fns->_nss_initgroups(user, group, &start, &size, &groups, 0, &errno);
3950 }
3951
3952 static struct group *nwrap_module_getgrnam(struct nwrap_backend *b,
3953                                            const char *name)
3954 {
3955         static struct group grp;
3956         static char *buf;
3957         static int buflen = 1000;
3958         NSS_STATUS status;
3959
3960         if (!b->fns->_nss_getgrnam_r) {
3961                 return NULL;
3962         }
3963
3964         if (!buf) {
3965                 buf = (char *)malloc(buflen);
3966         }
3967 again:
3968         status = b->fns->_nss_getgrnam_r(name, &grp, buf, buflen, &errno);
3969         if (status == NSS_STATUS_TRYAGAIN) {
3970                 buflen *= 2;
3971                 buf = (char *)realloc(buf, buflen);
3972                 if (!buf) {
3973                         return NULL;
3974                 }
3975                 goto again;
3976         }
3977         if (status == NSS_STATUS_NOTFOUND) {
3978                 SAFE_FREE(buf);
3979                 return NULL;
3980         }
3981         if (status != NSS_STATUS_SUCCESS) {
3982                 SAFE_FREE(buf);
3983                 return NULL;
3984         }
3985         return &grp;
3986 }
3987
3988 static int nwrap_module_getgrnam_r(struct nwrap_backend *b,
3989                                    const char *name, struct group *grdst,
3990                                    char *buf, size_t buflen, struct group **grdstp)
3991 {
3992         int ret;
3993
3994         (void) grdstp; /* unused */
3995
3996         if (!b->fns->_nss_getgrnam_r) {
3997                 return ENOENT;
3998         }
3999
4000         ret = b->fns->_nss_getgrnam_r(name, grdst, buf, buflen, &errno);
4001         switch (ret) {
4002         case NSS_STATUS_SUCCESS:
4003                 return 0;
4004         case NSS_STATUS_NOTFOUND:
4005                 if (errno != 0) {
4006                         return errno;
4007                 }
4008                 return ENOENT;
4009         case NSS_STATUS_TRYAGAIN:
4010                 if (errno != 0) {
4011                         return errno;
4012                 }
4013                 return ERANGE;
4014         default:
4015                 if (errno != 0) {
4016                         return errno;
4017                 }
4018                 return ret;
4019         }
4020 }
4021
4022 static struct group *nwrap_module_getgrgid(struct nwrap_backend *b,
4023                                            gid_t gid)
4024 {
4025         static struct group grp;
4026         static char *buf;
4027         static int buflen = 1000;
4028         NSS_STATUS status;
4029
4030         if (!b->fns->_nss_getgrgid_r) {
4031                 return NULL;
4032         }
4033
4034         if (!buf) {
4035                 buf = (char *)malloc(buflen);
4036         }
4037
4038 again:
4039         status = b->fns->_nss_getgrgid_r(gid, &grp, buf, buflen, &errno);
4040         if (status == NSS_STATUS_TRYAGAIN) {
4041                 buflen *= 2;
4042                 buf = (char *)realloc(buf, buflen);
4043                 if (!buf) {
4044                         return NULL;
4045                 }
4046                 goto again;
4047         }
4048         if (status == NSS_STATUS_NOTFOUND) {
4049                 SAFE_FREE(buf);
4050                 return NULL;
4051         }
4052         if (status != NSS_STATUS_SUCCESS) {
4053                 SAFE_FREE(buf);
4054                 return NULL;
4055         }
4056         return &grp;
4057 }
4058
4059 static int nwrap_module_getgrgid_r(struct nwrap_backend *b,
4060                                    gid_t gid, struct group *grdst,
4061                                    char *buf, size_t buflen, struct group **grdstp)
4062 {
4063         int ret;
4064
4065         (void) grdstp; /* unused */
4066
4067         if (!b->fns->_nss_getgrgid_r) {
4068                 return ENOENT;
4069         }
4070
4071         ret = b->fns->_nss_getgrgid_r(gid, grdst, buf, buflen, &errno);
4072         switch (ret) {
4073         case NSS_STATUS_SUCCESS:
4074                 return 0;
4075         case NSS_STATUS_NOTFOUND:
4076                 if (errno != 0) {
4077                         return errno;
4078                 }
4079                 return ENOENT;
4080         case NSS_STATUS_TRYAGAIN:
4081                 if (errno != 0) {
4082                         return errno;
4083                 }
4084                 return ERANGE;
4085         default:
4086                 if (errno != 0) {
4087                         return errno;
4088                 }
4089                 return ret;
4090         }
4091 }
4092
4093 static void nwrap_module_setgrent(struct nwrap_backend *b)
4094 {
4095         if (!b->fns->_nss_setgrent) {
4096                 return;
4097         }
4098
4099         b->fns->_nss_setgrent();
4100 }
4101
4102 static struct group *nwrap_module_getgrent(struct nwrap_backend *b)
4103 {
4104         static struct group grp;
4105         static char *buf;
4106         static int buflen = 1024;
4107         NSS_STATUS status;
4108
4109         if (!b->fns->_nss_getgrent_r) {
4110                 return NULL;
4111         }
4112
4113         if (!buf) {
4114                 buf = (char *)malloc(buflen);
4115         }
4116
4117 again:
4118         status = b->fns->_nss_getgrent_r(&grp, buf, buflen, &errno);
4119         if (status == NSS_STATUS_TRYAGAIN) {
4120                 buflen *= 2;
4121                 buf = (char *)realloc(buf, buflen);
4122                 if (!buf) {
4123                         return NULL;
4124                 }
4125                 goto again;
4126         }
4127         if (status == NSS_STATUS_NOTFOUND) {
4128                 SAFE_FREE(buf);
4129                 return NULL;
4130         }
4131         if (status != NSS_STATUS_SUCCESS) {
4132                 SAFE_FREE(buf);
4133                 return NULL;
4134         }
4135         return &grp;
4136 }
4137
4138 static int nwrap_module_getgrent_r(struct nwrap_backend *b,
4139                                    struct group *grdst, char *buf,
4140                                    size_t buflen, struct group **grdstp)
4141 {
4142         int ret;
4143
4144         (void) grdstp; /* unused */
4145
4146         if (!b->fns->_nss_getgrent_r) {
4147                 return ENOENT;
4148         }
4149
4150         ret = b->fns->_nss_getgrent_r(grdst, buf, buflen, &errno);
4151         switch (ret) {
4152         case NSS_STATUS_SUCCESS:
4153                 return 0;
4154         case NSS_STATUS_NOTFOUND:
4155                 if (errno != 0) {
4156                         return errno;
4157                 }
4158                 return ENOENT;
4159         case NSS_STATUS_TRYAGAIN:
4160                 if (errno != 0) {
4161                         return errno;
4162                 }
4163                 return ERANGE;
4164         default:
4165                 if (errno != 0) {
4166                         return errno;
4167                 }
4168                 return ret;
4169         }
4170 }
4171
4172 static void nwrap_module_endgrent(struct nwrap_backend *b)
4173 {
4174         if (!b->fns->_nss_endgrent) {
4175                 return;
4176         }
4177
4178         b->fns->_nss_endgrent();
4179 }
4180
4181 /****************************************************************************
4182  *   GETPWNAM
4183  ***************************************************************************/
4184
4185 static struct passwd *nwrap_getpwnam(const char *name)
4186 {
4187         int i;
4188         struct passwd *pwd;
4189
4190         for (i=0; i < nwrap_main_global->num_backends; i++) {
4191                 struct nwrap_backend *b = &nwrap_main_global->backends[i];
4192                 pwd = b->ops->nw_getpwnam(b, name);
4193                 if (pwd) {
4194                         return pwd;
4195                 }
4196         }
4197
4198         return NULL;
4199 }
4200
4201 struct passwd *getpwnam(const char *name)
4202 {
4203         if (!nss_wrapper_enabled()) {
4204                 return libc_getpwnam(name);
4205         }
4206
4207         return nwrap_getpwnam(name);
4208 }
4209
4210 /****************************************************************************
4211  *   GETPWNAM_R
4212  ***************************************************************************/
4213
4214 static int nwrap_getpwnam_r(const char *name, struct passwd *pwdst,
4215                             char *buf, size_t buflen, struct passwd **pwdstp)
4216 {
4217         int i,ret;
4218
4219         for (i=0; i < nwrap_main_global->num_backends; i++) {
4220                 struct nwrap_backend *b = &nwrap_main_global->backends[i];
4221                 ret = b->ops->nw_getpwnam_r(b, name, pwdst, buf, buflen, pwdstp);
4222                 if (ret == ENOENT) {
4223                         continue;
4224                 }
4225                 return ret;
4226         }
4227
4228         return ENOENT;
4229 }
4230
4231 #ifdef HAVE_GETPWNAM_R
4232 # ifdef HAVE_SOLARIS_GETPWNAM_R
4233 int getpwnam_r(const char *name, struct passwd *pwdst,
4234                char *buf, int buflen, struct passwd **pwdstp)
4235 # else /* HAVE_SOLARIS_GETPWNAM_R */
4236 int getpwnam_r(const char *name, struct passwd *pwdst,
4237                char *buf, size_t buflen, struct passwd **pwdstp)
4238 # endif /* HAVE_SOLARIS_GETPWNAM_R */
4239 {
4240         if (!nss_wrapper_enabled()) {
4241                 return libc_getpwnam_r(name, pwdst, buf, buflen, pwdstp);
4242         }
4243
4244         return nwrap_getpwnam_r(name, pwdst, buf, buflen, pwdstp);
4245 }
4246 #endif
4247
4248 /****************************************************************************
4249  *   GETPWUID
4250  ***************************************************************************/
4251
4252 static struct passwd *nwrap_getpwuid(uid_t uid)
4253 {
4254         int i;
4255         struct passwd *pwd;
4256
4257         for (i=0; i < nwrap_main_global->num_backends; i++) {
4258                 struct nwrap_backend *b = &nwrap_main_global->backends[i];
4259                 pwd = b->ops->nw_getpwuid(b, uid);
4260                 if (pwd) {
4261                         return pwd;
4262                 }
4263         }
4264
4265         return NULL;
4266 }
4267
4268 struct passwd *getpwuid(uid_t uid)
4269 {
4270         if (!nss_wrapper_enabled()) {
4271                 return libc_getpwuid(uid);
4272         }
4273
4274         return nwrap_getpwuid(uid);
4275 }
4276
4277 /****************************************************************************
4278  *   GETPWUID_R
4279  ***************************************************************************/
4280
4281 static int nwrap_getpwuid_r(uid_t uid, struct passwd *pwdst,
4282                             char *buf, size_t buflen, struct passwd **pwdstp)
4283 {
4284         int i,ret;
4285
4286         for (i=0; i < nwrap_main_global->num_backends; i++) {
4287                 struct nwrap_backend *b = &nwrap_main_global->backends[i];
4288                 ret = b->ops->nw_getpwuid_r(b, uid, pwdst, buf, buflen, pwdstp);
4289                 if (ret == ENOENT) {
4290                         continue;
4291                 }
4292                 return ret;
4293         }
4294
4295         return ENOENT;
4296 }
4297
4298 #ifdef HAVE_SOLARIS_GETPWUID_R
4299 int getpwuid_r(uid_t uid, struct passwd *pwdst,
4300                char *buf, int buflen, struct passwd **pwdstp)
4301 #else
4302 int getpwuid_r(uid_t uid, struct passwd *pwdst,
4303                char *buf, size_t buflen, struct passwd **pwdstp)
4304 #endif
4305 {
4306         if (!nss_wrapper_enabled()) {
4307                 return libc_getpwuid_r(uid, pwdst, buf, buflen, pwdstp);
4308         }
4309
4310         return nwrap_getpwuid_r(uid, pwdst, buf, buflen, pwdstp);
4311 }
4312
4313 /****************************************************************************
4314  *   SETPWENT
4315  ***************************************************************************/
4316
4317 static void nwrap_setpwent(void)
4318 {
4319         int i;
4320
4321         for (i=0; i < nwrap_main_global->num_backends; i++) {
4322                 struct nwrap_backend *b = &nwrap_main_global->backends[i];
4323                 b->ops->nw_setpwent(b);
4324         }
4325 }
4326
4327 void setpwent(void)
4328 {
4329         if (!nss_wrapper_enabled()) {
4330                 libc_setpwent();
4331                 return;
4332         }
4333
4334         nwrap_setpwent();
4335 }
4336
4337 /****************************************************************************
4338  *   GETPWENT
4339  ***************************************************************************/
4340
4341 static struct passwd *nwrap_getpwent(void)
4342 {
4343         int i;
4344         struct passwd *pwd;
4345
4346         for (i=0; i < nwrap_main_global->num_backends; i++) {
4347                 struct nwrap_backend *b = &nwrap_main_global->backends[i];
4348                 pwd = b->ops->nw_getpwent(b);
4349                 if (pwd) {
4350                         return pwd;
4351                 }
4352         }
4353
4354         return NULL;
4355 }
4356
4357 struct passwd *getpwent(void)
4358 {
4359         if (!nss_wrapper_enabled()) {
4360                 return libc_getpwent();
4361         }
4362
4363         return nwrap_getpwent();
4364 }
4365
4366 /****************************************************************************
4367  *   GETPWENT_R
4368  ***************************************************************************/
4369
4370 static int nwrap_getpwent_r(struct passwd *pwdst, char *buf,
4371                             size_t buflen, struct passwd **pwdstp)
4372 {
4373         int i,ret;
4374
4375         for (i=0; i < nwrap_main_global->num_backends; i++) {
4376                 struct nwrap_backend *b = &nwrap_main_global->backends[i];
4377                 ret = b->ops->nw_getpwent_r(b, pwdst, buf, buflen, pwdstp);
4378                 if (ret == ENOENT) {
4379                         continue;
4380                 }
4381                 return ret;
4382         }
4383
4384         return ENOENT;
4385 }
4386
4387 #ifdef HAVE_SOLARIS_GETPWENT_R
4388 struct passwd *getpwent_r(struct passwd *pwdst, char *buf, int buflen)
4389 {
4390         struct passwd *pwdstp = NULL;
4391         int rc;
4392
4393         if (!nss_wrapper_enabled()) {
4394                 return libc_getpwent_r(pwdst, buf, buflen);
4395         }
4396         rc = nwrap_getpwent_r(pwdst, buf, buflen, &pwdstp);
4397         if (rc < 0) {
4398                 return NULL;
4399         }
4400
4401         return pwdstp;
4402 }
4403 #else /* HAVE_SOLARIS_GETPWENT_R */
4404 int getpwent_r(struct passwd *pwdst, char *buf,
4405                size_t buflen, struct passwd **pwdstp)
4406 {
4407         if (!nss_wrapper_enabled()) {
4408                 return libc_getpwent_r(pwdst, buf, buflen, pwdstp);
4409         }
4410
4411         return nwrap_getpwent_r(pwdst, buf, buflen, pwdstp);
4412 }
4413 #endif /* HAVE_SOLARIS_GETPWENT_R */
4414
4415 /****************************************************************************
4416  *   ENDPWENT
4417  ***************************************************************************/
4418
4419 static void nwrap_endpwent(void)
4420 {
4421         int i;
4422
4423         for (i=0; i < nwrap_main_global->num_backends; i++) {
4424                 struct nwrap_backend *b = &nwrap_main_global->backends[i];
4425                 b->ops->nw_endpwent(b);
4426         }
4427 }
4428
4429 void endpwent(void)
4430 {
4431         if (!nss_wrapper_enabled()) {
4432                 libc_endpwent();
4433                 return;
4434         }
4435
4436         nwrap_endpwent();
4437 }
4438
4439 /****************************************************************************
4440  *   INITGROUPS
4441  ***************************************************************************/
4442
4443 static int nwrap_initgroups(const char *user, gid_t group)
4444 {
4445         int i;
4446
4447         for (i=0; i < nwrap_main_global->num_backends; i++) {
4448                 struct nwrap_backend *b = &nwrap_main_global->backends[i];
4449                 int rc;
4450
4451                 rc = b->ops->nw_initgroups(b, user, group);
4452                 if (rc == 0) {
4453                         return 0;
4454                 }
4455         }
4456
4457         errno = ENOENT;
4458         return -1;
4459 }
4460
4461 int initgroups(const char *user, gid_t group)
4462 {
4463         if (!nss_wrapper_enabled()) {
4464                 return libc_initgroups(user, group);
4465         }
4466
4467         return nwrap_initgroups(user, group);
4468 }
4469
4470 /****************************************************************************
4471  *   GETGRNAM
4472  ***************************************************************************/
4473
4474 static struct group *nwrap_getgrnam(const char *name)
4475 {
4476         int i;
4477         struct group *grp;
4478
4479         for (i=0; i < nwrap_main_global->num_backends; i++) {
4480                 struct nwrap_backend *b = &nwrap_main_global->backends[i];
4481                 grp = b->ops->nw_getgrnam(b, name);
4482                 if (grp) {
4483                         return grp;
4484                 }
4485         }
4486
4487         return NULL;
4488 }
4489
4490 struct group *getgrnam(const char *name)
4491 {
4492         if (!nss_wrapper_enabled()) {
4493                 return libc_getgrnam(name);
4494         }
4495
4496         return nwrap_getgrnam(name);
4497 }
4498
4499 /****************************************************************************
4500  *   GETGRNAM_R
4501  ***************************************************************************/
4502
4503 static int nwrap_getgrnam_r(const char *name, struct group *grdst,
4504                             char *buf, size_t buflen, struct group **grdstp)
4505 {
4506         int i, ret;
4507
4508         for (i=0; i < nwrap_main_global->num_backends; i++) {
4509                 struct nwrap_backend *b = &nwrap_main_global->backends[i];
4510                 ret = b->ops->nw_getgrnam_r(b, name, grdst, buf, buflen, grdstp);
4511                 if (ret == ENOENT) {
4512                         continue;
4513                 }
4514                 return ret;
4515         }
4516
4517         return ENOENT;
4518 }
4519
4520 #ifdef HAVE_GETGRNAM_R
4521 # ifdef HAVE_SOLARIS_GETGRNAM_R
4522 int getgrnam_r(const char *name, struct group *grp,
4523                 char *buf, int buflen, struct group **pgrp)
4524 # else /* HAVE_SOLARIS_GETGRNAM_R */
4525 int getgrnam_r(const char *name, struct group *grp,
4526                char *buf, size_t buflen, struct group **pgrp)
4527 # endif /* HAVE_SOLARIS_GETGRNAM_R */
4528 {
4529         if (!nss_wrapper_enabled()) {
4530                 return libc_getgrnam_r(name,
4531                                        grp,
4532                                        buf,
4533                                        buflen,
4534                                        pgrp);
4535         }
4536
4537         return nwrap_getgrnam_r(name, grp, buf, buflen, pgrp);
4538 }
4539 #endif /* HAVE_GETGRNAM_R */
4540
4541 /****************************************************************************
4542  *   GETGRGID
4543  ***************************************************************************/
4544
4545 static struct group *nwrap_getgrgid(gid_t gid)
4546 {
4547         int i;
4548         struct group *grp;
4549
4550         for (i=0; i < nwrap_main_global->num_backends; i++) {
4551                 struct nwrap_backend *b = &nwrap_main_global->backends[i];
4552                 grp = b->ops->nw_getgrgid(b, gid);
4553                 if (grp) {
4554                         return grp;
4555                 }
4556         }
4557
4558         return NULL;
4559 }
4560
4561 struct group *getgrgid(gid_t gid)
4562 {
4563         if (!nss_wrapper_enabled()) {
4564                 return libc_getgrgid(gid);
4565         }
4566
4567         return nwrap_getgrgid(gid);
4568 }
4569
4570 /****************************************************************************
4571  *   GETGRGID_R
4572  ***************************************************************************/
4573
4574 static int nwrap_getgrgid_r(gid_t gid, struct group *grdst,
4575                             char *buf, size_t buflen, struct group **grdstp)
4576 {
4577         int i,ret;
4578
4579         for (i=0; i < nwrap_main_global->num_backends; i++) {
4580                 struct nwrap_backend *b = &nwrap_main_global->backends[i];
4581                 ret = b->ops->nw_getgrgid_r(b, gid, grdst, buf, buflen, grdstp);
4582                 if (ret == ENOENT) {
4583                         continue;
4584                 }
4585                 return ret;
4586         }
4587
4588         return ENOENT;
4589 }
4590
4591 #ifdef HAVE_GETGRGID_R
4592 # ifdef HAVE_SOLARIS_GETGRGID_R
4593 int getgrgid_r(gid_t gid, struct group *grdst,
4594                char *buf, int buflen, struct group **grdstp)
4595 # else /* HAVE_SOLARIS_GETGRGID_R */
4596 int getgrgid_r(gid_t gid, struct group *grdst,
4597                char *buf, size_t buflen, struct group **grdstp)
4598 # endif /* HAVE_SOLARIS_GETGRGID_R */
4599 {
4600         if (!nss_wrapper_enabled()) {
4601                 return libc_getgrgid_r(gid, grdst, buf, buflen, grdstp);
4602         }
4603
4604         return nwrap_getgrgid_r(gid, grdst, buf, buflen, grdstp);
4605 }
4606 #endif
4607
4608 /****************************************************************************
4609  *   SETGRENT
4610  ***************************************************************************/
4611
4612 static void nwrap_setgrent(void)
4613 {
4614         int i;
4615
4616         for (i=0; i < nwrap_main_global->num_backends; i++) {
4617                 struct nwrap_backend *b = &nwrap_main_global->backends[i];
4618                 b->ops->nw_setgrent(b);
4619         }
4620 }
4621
4622 #ifdef HAVE_BSD_SETGRENT
4623 int setgrent(void)
4624 #else
4625 void setgrent(void)
4626 #endif
4627 {
4628         if (!nss_wrapper_enabled()) {
4629                 libc_setgrent();
4630                 goto out;
4631         }
4632
4633         nwrap_setgrent();
4634
4635 out:
4636 #ifdef HAVE_BSD_SETGRENT
4637         return 0;
4638 #else
4639         return;
4640 #endif
4641 }
4642
4643 /****************************************************************************
4644  *   GETGRENT
4645  ***************************************************************************/
4646
4647 static struct group *nwrap_getgrent(void)
4648 {
4649         int i;
4650         struct group *grp;
4651
4652         for (i=0; i < nwrap_main_global->num_backends; i++) {
4653                 struct nwrap_backend *b = &nwrap_main_global->backends[i];
4654                 grp = b->ops->nw_getgrent(b);
4655                 if (grp) {
4656                         return grp;
4657                 }
4658         }
4659
4660         return NULL;
4661 }
4662
4663 struct group *getgrent(void)
4664 {
4665         if (!nss_wrapper_enabled()) {
4666                 return libc_getgrent();
4667         }
4668
4669         return nwrap_getgrent();
4670 }
4671
4672 /****************************************************************************
4673  *   GETGRENT_R
4674  ***************************************************************************/
4675
4676 static int nwrap_getgrent_r(struct group *grdst, char *buf,
4677                             size_t buflen, struct group **grdstp)
4678 {
4679         int i,ret;
4680
4681         for (i=0; i < nwrap_main_global->num_backends; i++) {
4682                 struct nwrap_backend *b = &nwrap_main_global->backends[i];
4683                 ret = b->ops->nw_getgrent_r(b, grdst, buf, buflen, grdstp);
4684                 if (ret == ENOENT) {
4685                         continue;
4686                 }
4687                 return ret;
4688         }
4689
4690         return ENOENT;
4691 }
4692
4693 #ifdef HAVE_SOLARIS_GETGRENT_R
4694 struct group *getgrent_r(struct group *src, char *buf, int buflen)
4695 {
4696         struct group *grdstp = NULL;
4697         int rc;
4698
4699         if (!nss_wrapper_enabled()) {
4700                 return libc_getgrent_r(src, buf, buflen);
4701         }
4702
4703         rc = nwrap_getgrent_r(src, buf, buflen, &grdstp);
4704         if (rc < 0) {
4705                 return NULL;
4706         }
4707
4708         return grdstp;
4709 }
4710 #else /* HAVE_SOLARIS_GETGRENT_R */
4711 int getgrent_r(struct group *src, char *buf,
4712                size_t buflen, struct group **grdstp)
4713 {
4714         if (!nss_wrapper_enabled()) {
4715                 return libc_getgrent_r(src, buf, buflen, grdstp);
4716         }
4717
4718         return nwrap_getgrent_r(src, buf, buflen, grdstp);
4719 }
4720 #endif /* HAVE_SOLARIS_GETGRENT_R */
4721
4722 /****************************************************************************
4723  *   ENDGRENT
4724  ***************************************************************************/
4725
4726 static void nwrap_endgrent(void)
4727 {
4728         int i;
4729
4730         for (i=0; i < nwrap_main_global->num_backends; i++) {
4731                 struct nwrap_backend *b = &nwrap_main_global->backends[i];
4732                 b->ops->nw_endgrent(b);
4733         }
4734 }
4735
4736 void endgrent(void)
4737 {
4738         if (!nss_wrapper_enabled()) {
4739                 libc_endgrent();
4740                 return;
4741         }
4742
4743         nwrap_endgrent();
4744 }
4745
4746 /****************************************************************************
4747  *   GETGROUPLIST
4748  ***************************************************************************/
4749
4750 #ifdef HAVE_GETGROUPLIST
4751 static int nwrap_getgrouplist(const char *user, gid_t group,
4752                               gid_t *groups, int *ngroups)
4753 {
4754         struct group *grp;
4755         gid_t *groups_tmp;
4756         int count = 1;
4757
4758         NWRAP_LOG(NWRAP_LOG_DEBUG, "getgrouplist called for %s", user);
4759
4760         groups_tmp = (gid_t *)malloc(count * sizeof(gid_t));
4761         if (!groups_tmp) {
4762                 NWRAP_LOG(NWRAP_LOG_ERROR, "Out of memory");
4763                 errno = ENOMEM;
4764                 return -1;
4765         }
4766         groups_tmp[0] = group;
4767
4768         nwrap_setgrent();
4769         while ((grp = nwrap_getgrent()) != NULL) {
4770                 int i = 0;
4771
4772                 NWRAP_LOG(NWRAP_LOG_DEBUG,
4773                           "Inspecting %s for group membership",
4774                           grp->gr_name);
4775
4776                 for (i=0; grp->gr_mem && grp->gr_mem[i] != NULL; i++) {
4777
4778                         if (group != grp->gr_gid &&
4779                             (strcmp(user, grp->gr_mem[i]) == 0)) {
4780
4781                                 NWRAP_LOG(NWRAP_LOG_DEBUG,
4782                                           "%s is member of %s",
4783                                           user,
4784                                           grp->gr_name);
4785
4786                                 groups_tmp = (gid_t *)realloc(groups_tmp, (count + 1) * sizeof(gid_t));
4787                                 if (!groups_tmp) {
4788                                         NWRAP_LOG(NWRAP_LOG_ERROR,
4789                                                   "Out of memory");
4790                                         errno = ENOMEM;
4791                                         return -1;
4792                                 }
4793                                 groups_tmp[count] = grp->gr_gid;
4794
4795                                 count++;
4796                         }
4797                 }
4798         }
4799
4800         nwrap_endgrent();
4801
4802         NWRAP_LOG(NWRAP_LOG_DEBUG,
4803                   "%s is member of %d groups",
4804                   user, *ngroups);
4805
4806         if (*ngroups < count) {
4807                 *ngroups = count;
4808                 free(groups_tmp);
4809                 return -1;
4810         }
4811
4812         *ngroups = count;
4813         memcpy(groups, groups_tmp, count * sizeof(gid_t));
4814         free(groups_tmp);
4815
4816         return count;
4817 }
4818
4819 int getgrouplist(const char *user, gid_t group, gid_t *groups, int *ngroups)
4820 {
4821         if (!nss_wrapper_enabled()) {
4822                 return libc_getgrouplist(user, group, groups, ngroups);
4823         }
4824
4825         return nwrap_getgrouplist(user, group, groups, ngroups);
4826 }
4827 #endif
4828
4829 /**********************************************************
4830  * SHADOW
4831  **********************************************************/
4832
4833 #if defined(HAVE_SHADOW_H) && defined(HAVE_GETSPNAM)
4834
4835 #ifdef HAVE_SETSPENT
4836 static void nwrap_setspent(void)
4837 {
4838         nwrap_files_setspent();
4839 }
4840
4841 void setspent(void)
4842 {
4843         if (!nss_wrapper_shadow_enabled()) {
4844                 return;
4845         }
4846
4847         nwrap_setspent();
4848 }
4849
4850 static struct spwd *nwrap_getspent(void)
4851 {
4852         return nwrap_files_getspent();
4853 }
4854
4855 struct spwd *getspent(void)
4856 {
4857         if (!nss_wrapper_shadow_enabled()) {
4858                 return NULL;
4859         }
4860
4861         return nwrap_getspent();
4862 }
4863
4864 static void nwrap_endspent(void)
4865 {
4866         nwrap_files_endspent();
4867 }
4868
4869 void endspent(void)
4870 {
4871         if (!nss_wrapper_shadow_enabled()) {
4872                 return;
4873         }
4874
4875         nwrap_endspent();
4876 }
4877 #endif /* HAVE_SETSPENT */
4878
4879 static struct spwd *nwrap_getspnam(const char *name)
4880 {
4881         return nwrap_files_getspnam(name);
4882 }
4883
4884 struct spwd *getspnam(const char *name)
4885 {
4886         if (!nss_wrapper_shadow_enabled()) {
4887                 return NULL;
4888         }
4889
4890         return nwrap_getspnam(name);
4891 }
4892
4893 #endif /* defined(HAVE_SHADOW_H) && defined(HAVE_GETSPNAM) */
4894
4895 /**********************************************************
4896  * NETDB
4897  **********************************************************/
4898
4899 static void nwrap_sethostent(int stayopen) {
4900         (void) stayopen; /* ignored */
4901
4902         nwrap_files_sethostent();
4903 }
4904
4905 #ifdef HAVE_SOLARIS_SETHOSTENT
4906 int sethostent(int stayopen)
4907 {
4908         if (!nss_wrapper_hosts_enabled()) {
4909                 libc_sethostent(stayopen);
4910                 return 0;
4911         }
4912
4913         nwrap_sethostent(stayopen);
4914
4915         return 0;
4916 }
4917 #else /* HAVE_SOLARIS_SETHOSTENT */
4918 void sethostent(int stayopen)
4919 {
4920         if (!nss_wrapper_hosts_enabled()) {
4921                 libc_sethostent(stayopen);
4922                 return;
4923         }
4924
4925         nwrap_sethostent(stayopen);
4926 }
4927 #endif /* HAVE_SOLARIS_SETHOSTENT */
4928
4929 static struct hostent *nwrap_gethostent(void)
4930 {
4931         return nwrap_files_gethostent();
4932 }
4933
4934 struct hostent *gethostent(void) {
4935         if (!nss_wrapper_hosts_enabled()) {
4936                 return libc_gethostent();
4937         }
4938
4939         return nwrap_gethostent();
4940 }
4941
4942 static void nwrap_endhostent(void) {
4943         nwrap_files_endhostent();
4944 }
4945
4946 #ifdef HAVE_SOLARIS_ENDHOSTENT
4947 int endhostent(void)
4948 {
4949         if (!nss_wrapper_hosts_enabled()) {
4950                 libc_endhostent();
4951                 return 0;
4952         }
4953
4954         nwrap_endhostent();
4955
4956         return 0;
4957 }
4958 #else /* HAVE_SOLARIS_ENDHOSTENT */
4959 void endhostent(void)
4960 {
4961         if (!nss_wrapper_hosts_enabled()) {
4962                 libc_endhostent();
4963                 return;
4964         }
4965
4966         nwrap_endhostent();
4967 }
4968 #endif /* HAVE_SOLARIS_ENDHOSTENT */
4969
4970 #ifdef BSD
4971 /* BSD implementation stores data in thread local storage but GLIBC does not */
4972 static __thread struct hostent user_he;
4973 static __thread struct nwrap_vector user_addrlist;
4974 #else
4975 static struct hostent user_he;
4976 static struct nwrap_vector user_addrlist;
4977 #endif /* BSD */
4978 static struct hostent *nwrap_gethostbyname(const char *name)
4979 {
4980         if (nwrap_files_gethostbyname(name, AF_UNSPEC, &user_he, &user_addrlist) == -1) {
4981                 return NULL;
4982         }
4983         return &user_he;
4984 }
4985
4986 struct hostent *gethostbyname(const char *name)
4987 {
4988         if (!nss_wrapper_hosts_enabled()) {
4989                 return libc_gethostbyname(name);
4990         }
4991
4992         return nwrap_gethostbyname(name);
4993 }
4994
4995 /* This is a GNU extension - Also can be found on BSD systems */
4996 #ifdef HAVE_GETHOSTBYNAME2
4997 #ifdef BSD
4998 /* BSD implementation stores data in  thread local storage but GLIBC not */
4999 static __thread struct hostent user_he2;
5000 static __thread struct nwrap_vector user_addrlist2;
5001 #else
5002 static struct hostent user_he2;
5003 static struct nwrap_vector user_addrlist2;
5004 #endif /* BSD */
5005 static struct hostent *nwrap_gethostbyname2(const char *name, int af)
5006 {
5007         if (nwrap_files_gethostbyname(name, af, &user_he2, &user_addrlist2) == -1) {
5008                 return NULL;
5009         }
5010         return &user_he2;
5011 }
5012
5013 struct hostent *gethostbyname2(const char *name, int af)
5014 {
5015         if (!nss_wrapper_hosts_enabled()) {
5016                 return libc_gethostbyname2(name, af);
5017         }
5018
5019         return nwrap_gethostbyname2(name, af);
5020 }
5021 #endif
5022
5023 static struct hostent *nwrap_gethostbyaddr(const void *addr,
5024                                            socklen_t len, int type)
5025 {
5026         return nwrap_files_gethostbyaddr(addr, len, type);
5027 }
5028
5029 struct hostent *gethostbyaddr(const void *addr,
5030                               socklen_t len, int type)
5031 {
5032         if (!nss_wrapper_hosts_enabled()) {
5033                 return libc_gethostbyaddr(addr, len, type);
5034         }
5035
5036         return nwrap_gethostbyaddr(addr, len, type);
5037 }
5038
5039 static const struct addrinfo default_hints =
5040 {
5041         .ai_flags = AI_ADDRCONFIG|AI_V4MAPPED,
5042         .ai_family = AF_UNSPEC,
5043         .ai_socktype = 0,
5044         .ai_protocol = 0,
5045         .ai_addrlen = 0,
5046         .ai_addr = NULL,
5047         .ai_canonname = NULL,
5048         .ai_next = NULL
5049 };
5050
5051 static int nwrap_convert_he_ai(const struct hostent *he,
5052                                unsigned short port,
5053                                const struct addrinfo *hints,
5054                                struct addrinfo **pai,
5055                                bool skip_canonname)
5056 {
5057         struct addrinfo *ai;
5058         socklen_t socklen;
5059
5060         if (he == NULL) {
5061                 return EAI_MEMORY;
5062         }
5063
5064         switch (he->h_addrtype) {
5065                 case AF_INET:
5066                         socklen = sizeof(struct sockaddr_in);
5067                         break;
5068 #ifdef HAVE_IPV6
5069                 case AF_INET6:
5070                         socklen = sizeof(struct sockaddr_in6);
5071                         break;
5072 #endif
5073                 default:
5074                         return EAI_FAMILY;
5075         }
5076
5077         ai = (struct addrinfo *)malloc(sizeof(struct addrinfo) + socklen);
5078         if (ai == NULL) {
5079                 return EAI_MEMORY;
5080         }
5081
5082         ai->ai_flags = 0;
5083         ai->ai_family = he->h_addrtype;
5084         ai->ai_socktype = hints->ai_socktype;
5085         ai->ai_protocol = hints->ai_protocol;
5086         ai->ai_canonname = NULL;
5087
5088         ai->ai_addrlen = socklen;
5089         ai->ai_addr = (void *)(ai + 1);
5090
5091 #ifdef HAVE_STRUCT_SOCKADDR_SA_LEN
5092         ai->ai_addr->sa_len = socklen;
5093 #endif
5094         ai->ai_addr->sa_family = he->h_addrtype;
5095
5096         switch (he->h_addrtype) {
5097                 case AF_INET:
5098                 {
5099                         struct sockaddr_in *sinp =
5100                                 (struct sockaddr_in *) ai->ai_addr;
5101
5102                         memset(sinp, 0, sizeof(struct sockaddr_in));
5103
5104                         sinp->sin_port = htons(port);
5105                         sinp->sin_family = AF_INET;
5106
5107                         memset (sinp->sin_zero, '\0', sizeof (sinp->sin_zero));
5108                         memcpy(&sinp->sin_addr, he->h_addr_list[0], he->h_length);
5109
5110                 }
5111                 break;
5112 #ifdef HAVE_IPV6
5113                 case AF_INET6:
5114                 {
5115                         struct sockaddr_in6 *sin6p =
5116                                 (struct sockaddr_in6 *) ai->ai_addr;
5117
5118                         memset(sin6p, 0, sizeof(struct sockaddr_in6));
5119
5120                         sin6p->sin6_port = htons(port);
5121                         sin6p->sin6_family = AF_INET6;
5122
5123                         memcpy(&sin6p->sin6_addr,
5124                                he->h_addr_list[0],
5125                                he->h_length);
5126                 }
5127                 break;
5128 #endif
5129         }
5130
5131         ai->ai_next = NULL;
5132
5133         if (he->h_name && !skip_canonname) {
5134                 ai->ai_canonname = strdup(he->h_name);
5135                 if (ai->ai_canonname == NULL) {
5136                         freeaddrinfo(ai);
5137                         return EAI_MEMORY;
5138                 }
5139         }
5140
5141         *pai = ai;
5142         return 0;
5143 }
5144
5145 static int nwrap_getaddrinfo(const char *node,
5146                              const char *service,
5147                              const struct addrinfo *hints,
5148                              struct addrinfo **res)
5149 {
5150         struct addrinfo *ai = NULL;
5151         struct addrinfo *ai_tail;
5152         unsigned short port = 0;
5153         struct {
5154                 int family;
5155                 union {
5156                         struct in_addr v4;
5157 #ifdef HAVE_IPV6
5158                         struct in6_addr v6;
5159                 } in;
5160 #endif
5161         } addr = {
5162                 .family = AF_UNSPEC,
5163         };
5164         int rc;
5165
5166         if (node == NULL && service == NULL) {
5167                 return EAI_NONAME;
5168         }
5169
5170         if (hints == NULL) {
5171                 hints = &default_hints;
5172         }
5173
5174         /* EAI_BADFLAGS
5175               hints.ai_flags   contains   invalid  flags;  or,  hints.ai_flags
5176               included AI_CANONNAME and name was NULL.
5177         */
5178         if ((hints->ai_flags & AI_CANONNAME) && (node == NULL)) {
5179                 return EAI_BADFLAGS;
5180         }
5181
5182         /* If no node has been specified, let glibc deal with it */
5183         if (node == NULL) {
5184                 int ret;
5185                 struct addrinfo *p = NULL;
5186
5187                 ret = libc_getaddrinfo(node, service, hints, &p);
5188
5189                 if (ret == 0) {
5190                         *res = p;
5191                 }
5192                 return ret;
5193         }
5194
5195         if (service != NULL && service[0] != '\0') {
5196                 const char *proto = NULL;
5197                 struct servent *s;
5198                 char *end_ptr;
5199                 long sl;
5200
5201                 errno = 0;
5202                 sl = strtol(service, &end_ptr, 10);
5203
5204                 if (*end_ptr == '\0') {
5205                         port = sl;
5206                         goto valid_port;
5207                 } else if (hints->ai_flags & AI_NUMERICSERV) {
5208                         return EAI_NONAME;
5209                 }
5210
5211                 if (hints->ai_protocol != 0) {
5212                         struct protoent *pent;
5213
5214                         pent = getprotobynumber(hints->ai_protocol);
5215                         if (pent != NULL) {
5216                                 proto = pent->p_name;
5217                         }
5218                 }
5219
5220                 s = getservbyname(service, proto);
5221                 if (s == NULL) {
5222                         return EAI_NONAME;
5223                 }
5224                 port = ntohs(s->s_port);
5225         }
5226
5227 valid_port:
5228
5229         rc = inet_pton(AF_INET, node, &addr.in.v4);
5230         if (rc == 1) {
5231                 addr.family = AF_INET;
5232         }
5233 #ifdef HAVE_IPV6
5234         if (addr.family == AF_UNSPEC) {
5235                 rc = inet_pton(AF_INET6, node, &addr.in.v6);
5236                 if (rc == 1) {
5237                         addr.family = AF_INET6;
5238                 }
5239         }
5240 #endif
5241
5242         if (addr.family == AF_UNSPEC) {
5243                if (hints->ai_flags & AI_NUMERICHOST) {
5244                         return EAI_NONAME;
5245                 }
5246         } else if ((hints->ai_family != AF_UNSPEC) &&
5247                    (hints->ai_family != addr.family))
5248         {
5249                 return EAI_ADDRFAMILY;
5250         }
5251
5252         rc = nwrap_files_getaddrinfo(node, port, hints, &ai, &ai_tail);
5253         if (rc != 0) {
5254                 int ret;
5255                 struct addrinfo *p = NULL;
5256
5257                 ret = libc_getaddrinfo(node, service, hints, &p);
5258
5259                 if (ret == 0) {
5260                         /*
5261                          * nwrap_files_getaddrinfo failed, but libc was
5262                          * successful -- use the result from libc.
5263                          */
5264                         *res = p;
5265                         return 0;
5266                 }
5267
5268                 return rc;
5269         }
5270
5271         if (ai->ai_flags == 0) {
5272                 ai->ai_flags = hints->ai_flags;
5273         }
5274         if (ai->ai_socktype == 0) {
5275                 ai->ai_socktype = SOCK_DGRAM;
5276         }
5277         if (ai->ai_protocol == 0 && ai->ai_socktype == SOCK_DGRAM) {
5278                 ai->ai_protocol = 17; /* UDP */
5279         } else if (ai->ai_protocol == 0 && ai->ai_socktype == SOCK_STREAM) {
5280                 ai->ai_protocol = 6; /* TCP */
5281         }
5282
5283         if (hints->ai_socktype == 0) {
5284                 /* Add second ai */
5285                 struct addrinfo *ai_head = ai;
5286                 struct addrinfo *ai_tmp;
5287                 struct addrinfo *ai_new_tail = ai_tail;
5288
5289                 /* Add at least one more struct */
5290                 do {
5291                         /* CHECKS! */
5292                         ai_tmp = malloc(sizeof(struct addrinfo));
5293                         memcpy(ai_tmp, ai_head, sizeof(struct addrinfo));
5294                         ai_tmp->ai_next = NULL;
5295
5296                         /* We need a deep copy or freeaddrinfo() will blow up */
5297                         if (ai_head->ai_canonname != NULL) {
5298                                 ai_tmp->ai_canonname =
5299                                         strdup(ai_head->ai_canonname);
5300                         }
5301                         /* ai_head should point inside hints. */
5302                         ai_tmp->ai_addr = ai_head->ai_addr;
5303
5304                         if (ai_head->ai_flags == 0) {
5305                                 ai_tmp->ai_flags = hints->ai_flags;
5306                         }
5307                         if (ai_head->ai_socktype == SOCK_DGRAM) {
5308                                 ai_tmp->ai_socktype = SOCK_STREAM;
5309                         } else if (ai_head->ai_socktype == SOCK_STREAM) {
5310                                 ai_tmp->ai_socktype = SOCK_DGRAM;
5311                         }
5312                         if (ai_head->ai_socktype == SOCK_DGRAM) {
5313                                 ai_tmp->ai_protocol = 17; /* UDP */
5314                         } else if (ai_head->ai_socktype == SOCK_STREAM) {
5315                                 ai_tmp->ai_protocol = 6; /* TCP */
5316                         }
5317                         ai_new_tail->ai_next = ai_tmp;
5318                         ai_new_tail = ai_tmp;
5319
5320                         if (ai_head == ai_tail) {
5321                                 break;
5322                         }
5323                         ai_head = ai_head->ai_next;
5324                 } while (1);
5325         }
5326
5327         *res = ai;
5328
5329         return 0;
5330 }
5331
5332 int getaddrinfo(const char *node, const char *service,
5333                 const struct addrinfo *hints,
5334                 struct addrinfo **res)
5335 {
5336         if (!nss_wrapper_hosts_enabled()) {
5337                 return libc_getaddrinfo(node, service, hints, res);
5338         }
5339
5340         return nwrap_getaddrinfo(node, service, hints, res);
5341 }
5342
5343 static int nwrap_getnameinfo(const struct sockaddr *sa, socklen_t salen,
5344                              char *host, size_t hostlen,
5345                              char *serv, size_t servlen,
5346                              int flags)
5347 {
5348         struct hostent *he;
5349         struct servent *service;
5350         const char *proto;
5351         const void *addr;
5352         socklen_t addrlen;
5353         uint16_t port;
5354         sa_family_t type;
5355
5356         if (sa == NULL || salen < sizeof(sa_family_t)) {
5357                 return EAI_FAMILY;
5358         }
5359
5360         if ((flags & NI_NAMEREQD) && host == NULL && serv == NULL) {
5361                 return EAI_NONAME;
5362         }
5363
5364         type = sa->sa_family;
5365         switch (type) {
5366         case AF_INET:
5367                 if (salen < sizeof(struct sockaddr_in))
5368                         return EAI_FAMILY;
5369                 addr = &((const struct sockaddr_in *)sa)->sin_addr;
5370                 addrlen = sizeof(((const struct sockaddr_in *)sa)->sin_addr);
5371                 port = ntohs(((const struct sockaddr_in *)sa)->sin_port);
5372                 break;
5373 #ifdef HAVE_IPV6
5374         case AF_INET6:
5375                 if (salen < sizeof(struct sockaddr_in6))
5376                         return EAI_FAMILY;
5377                 addr = &((const struct sockaddr_in6 *)sa)->sin6_addr;
5378                 addrlen = sizeof(((const struct sockaddr_in6 *)sa)->sin6_addr);
5379                 port = ntohs(((const struct sockaddr_in6 *)sa)->sin6_port);
5380                 break;
5381 #endif
5382         default:
5383                 return EAI_FAMILY;
5384         }
5385
5386         if (host != NULL) {
5387                 he = NULL;
5388                 if ((flags & NI_NUMERICHOST) == 0) {
5389                         he = nwrap_files_gethostbyaddr(addr, addrlen, type);
5390                         if ((flags & NI_NAMEREQD) && (he == NULL || he->h_name == NULL))
5391                                 return EAI_NONAME;
5392                 }
5393                 if (he != NULL && he->h_name != NULL) {
5394                         if (strlen(he->h_name) >= hostlen)
5395                                 return EAI_OVERFLOW;
5396                         strcpy(host, he->h_name);
5397                         if (flags & NI_NOFQDN)
5398                                 host[strcspn(host, ".")] = '\0';
5399                 } else {
5400                         if (inet_ntop(type, addr, host, hostlen) == NULL)
5401                                 return (errno == ENOSPC) ? EAI_OVERFLOW : EAI_FAIL;
5402                 }
5403         }
5404
5405         if (serv != NULL) {
5406                 service = NULL;
5407                 if ((flags & NI_NUMERICSERV) == 0) {
5408                         proto = (flags & NI_DGRAM) ? "udp" : "tcp";
5409                         service = getservbyport(htons(port), proto);
5410                 }
5411                 if (service != NULL) {
5412                         if (strlen(service->s_name) >= servlen)
5413                                 return EAI_OVERFLOW;
5414                         strcpy(serv, service->s_name);
5415                 } else {
5416                         if (snprintf(serv, servlen, "%u", port) >= (int) servlen)
5417                                 return EAI_OVERFLOW;
5418                 }
5419         }
5420
5421         return 0;
5422 }
5423
5424 #ifdef HAVE_LINUX_GETNAMEINFO
5425 int getnameinfo(const struct sockaddr *sa, socklen_t salen,
5426                 char *host, socklen_t hostlen,
5427                 char *serv, socklen_t servlen,
5428                 int flags)
5429 #elif defined(HAVE_LINUX_GETNAMEINFO_UNSIGNED)
5430 int getnameinfo(const struct sockaddr *sa, socklen_t salen,
5431                 char *host, socklen_t hostlen,
5432                 char *serv, socklen_t servlen,
5433                 unsigned int flags)
5434 #else
5435 int getnameinfo(const struct sockaddr *sa, socklen_t salen,
5436                 char *host, size_t hostlen,
5437                 char *serv, size_t servlen,
5438                 int flags)
5439 #endif
5440 {
5441         if (!nss_wrapper_hosts_enabled()) {
5442                 return libc_getnameinfo(sa, salen, host, hostlen, serv, servlen, flags);
5443         }
5444
5445         return nwrap_getnameinfo(sa, salen, host, hostlen, serv, servlen, flags);
5446 }
5447
5448 static int nwrap_gethostname(char *name, size_t len)
5449 {
5450         const char *hostname = getenv("NSS_WRAPPER_HOSTNAME");
5451
5452         if (strlen(hostname) >= len) {
5453                 errno = ENAMETOOLONG;
5454                 return -1;
5455         }
5456         snprintf(name, len, "%s", hostname);
5457
5458         return 0;
5459 }
5460
5461 #ifdef HAVE_SOLARIS_GETHOSTNAME
5462 int gethostname(char *name, int len)
5463 #else /* HAVE_SOLARIS_GETHOSTNAME */
5464 int gethostname(char *name, size_t len)
5465 #endif /* HAVE_SOLARIS_GETHOSTNAME */
5466 {
5467         if (!nwrap_hostname_enabled()) {
5468                 return libc_gethostname(name, len);
5469         }
5470
5471         return nwrap_gethostname(name, len);
5472 }
5473
5474 /****************************
5475  * DESTRUCTOR
5476  ***************************/
5477
5478 /*
5479  * This function is called when the library is unloaded and makes sure that
5480  * sockets get closed and the unix file for the socket are unlinked.
5481  */
5482 void nwrap_destructor(void)
5483 {
5484         int i;
5485
5486         NWRAP_LOCK_ALL;
5487         if (nwrap_main_global != NULL) {
5488                 struct nwrap_main *m = nwrap_main_global;
5489
5490                 /* libc */
5491                 SAFE_FREE(m->libc->fns);
5492                 if (m->libc->handle != NULL) {
5493                         dlclose(m->libc->handle);
5494                 }
5495                 if (m->libc->nsl_handle != NULL) {
5496                         dlclose(m->libc->nsl_handle);
5497                 }
5498                 if (m->libc->sock_handle != NULL) {
5499                         dlclose(m->libc->sock_handle);
5500                 }
5501                 SAFE_FREE(m->libc);
5502
5503                 /* backends */
5504                 for (i = 0; i < m->num_backends; i++) {
5505                         struct nwrap_backend *b = &(m->backends[i]);
5506
5507                         if (b->so_handle != NULL) {
5508                                 dlclose(b->so_handle);
5509                         }
5510                         SAFE_FREE(b->fns);
5511                 }
5512                 SAFE_FREE(m->backends);
5513         }
5514
5515         if (nwrap_pw_global.cache != NULL) {
5516                 struct nwrap_cache *c = nwrap_pw_global.cache;
5517
5518                 nwrap_files_cache_unload(c);
5519                 if (c->fd >= 0) {
5520                         fclose(c->fp);
5521                         c->fd = -1;
5522                 }
5523
5524                 SAFE_FREE(nwrap_pw_global.list);
5525                 nwrap_pw_global.num = 0;
5526         }
5527
5528         if (nwrap_gr_global.cache != NULL) {
5529                 struct nwrap_cache *c = nwrap_gr_global.cache;
5530
5531                 nwrap_files_cache_unload(c);
5532                 if (c->fd >= 0) {
5533                         fclose(c->fp);
5534                         c->fd = -1;
5535                 }
5536
5537                 SAFE_FREE(nwrap_gr_global.list);
5538                 nwrap_pw_global.num = 0;
5539         }
5540
5541         if (nwrap_he_global.cache != NULL) {
5542                 struct nwrap_cache *c = nwrap_he_global.cache;
5543
5544                 nwrap_files_cache_unload(c);
5545                 if (c->fd >= 0) {
5546                         fclose(c->fp);
5547                         c->fd = -1;
5548                 }
5549
5550                 nwrap_he_global.num = 0;
5551         }
5552
5553         hdestroy();
5554         NWRAP_UNLOCK_ALL;
5555 }