/*      $NetBSD: getid.c,v 1.10 2014/10/27 21:46:45 christos Exp $      */
/*      from: NetBSD: getpwent.c,v 1.48 2000/10/03 03:22:26 enami Exp */
/*      from: NetBSD: getgrent.c,v 1.41 2002/01/12 23:51:30 lukem Exp */

/*
 * Copyright (c) 1987, 1988, 1989, 1993, 1994, 1995
 *      The Regents of the University of California.  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. Neither the name of the University nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

/*-
 * Copyright (c) 2002 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by Luke Mewburn of Wasabi Systems.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */

#if HAVE_NBTOOL_CONFIG_H
#include "nbtool_config.h"
#endif

#include <sys/cdefs.h>
__RCSID("$NetBSD: getid.c,v 1.10 2014/10/27 21:46:45 christos Exp $");

#include <sys/param.h>

#include <grp.h>
#include <limits.h>
#include <pwd.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <time.h>
#include <unistd.h>

#include "extern.h"

static  struct group *  gi_getgrnam(const char *);
static  struct group *  gi_getgrgid(gid_t);
static  int             gi_setgroupent(int);
static  void            gi_endgrent(void);
static  int             grstart(void);
static  int             grscan(int, gid_t, const char *);
static  int             grmatchline(int, gid_t, const char *);

static  struct passwd * gi_getpwnam(const char *);
static  struct passwd * gi_getpwuid(uid_t);
static  int             gi_setpassent(int);
static  void            gi_endpwent(void);
static  int             pwstart(void);
static  int             pwscan(int, uid_t, const char *);
static  int             pwmatchline(int, uid_t, const char *);

#define MAXGRP          200
#define MAXLINELENGTH   1024

static  FILE            *_gr_fp;
static  struct group    _gr_group;
static  int             _gr_stayopen;
static  int             _gr_filesdone;
static  FILE            *_pw_fp;
static  struct passwd   _pw_passwd;     /* password structure */
static  int             _pw_stayopen;   /* keep fd's open */
static  int             _pw_filesdone;

static  char            grfile[MAXPATHLEN];
static  char            pwfile[MAXPATHLEN];

static  char            *members[MAXGRP];
static  char            grline[MAXLINELENGTH];
static  char            pwline[MAXLINELENGTH];

int
setup_getid(const char *dir)
{
        if (dir == NULL)
                return (0);

                                /* close existing databases */
        gi_endgrent();
        gi_endpwent();

                                /* build paths to new databases */
        snprintf(grfile, sizeof(grfile), "%s/group", dir);
        snprintf(pwfile, sizeof(pwfile), "%s/master.passwd", dir);

                                /* try to open new databases */
        if (!grstart() || !pwstart())
                return (0);

                                /* switch pwcache(3) lookup functions */
        if (pwcache_groupdb(gi_setgroupent, gi_endgrent,
                            gi_getgrnam, gi_getgrgid) == -1
            || pwcache_userdb(gi_setpassent, gi_endpwent,
                            gi_getpwnam, gi_getpwuid) == -1)
                return (0);

        return (1);
}


/*
 * group lookup functions
 */

static struct group *
gi_getgrnam(const char *name)
{
        int rval;

        if (!grstart())
                return NULL;
        rval = grscan(1, 0, name);
        if (!_gr_stayopen)
                endgrent();
        return (rval) ? &_gr_group : NULL;
}

static struct group *
gi_getgrgid(gid_t gid)
{
        int rval;

        if (!grstart())
                return NULL;
        rval = grscan(1, gid, NULL);
        if (!_gr_stayopen)
                endgrent();
        return (rval) ? &_gr_group : NULL;
}

static int
gi_setgroupent(int stayopen)
{

        if (!grstart())
                return 0;
        _gr_stayopen = stayopen;
        return 1;
}

static void
gi_endgrent(void)
{

        _gr_filesdone = 0;
        if (_gr_fp) {
                (void)fclose(_gr_fp);
                _gr_fp = NULL;
        }
}

static int
grstart(void)
{

        _gr_filesdone = 0;
        if (_gr_fp) {
                rewind(_gr_fp);
                return 1;
        }
        if (grfile[0] == '\0')                  /* sanity check */
                return 0;

        _gr_fp = fopen(grfile, "r");
        if (_gr_fp != NULL)
                return 1;
        warn("Can't open `%s'", grfile);
        return 0;
}


static int
grscan(int search, gid_t gid, const char *name)
{

        if (_gr_filesdone)
                return 0;
        for (;;) {
                if (!fgets(grline, sizeof(grline), _gr_fp)) {
                        if (!search)
                                _gr_filesdone = 1;
                        return 0;
                }
                /* skip lines that are too big */
                if (!strchr(grline, '\n')) {
                        int ch;

                        while ((ch = getc(_gr_fp)) != '\n' && ch != EOF)
                                ;
                        continue;
                }
                /* skip comments */
                if (grline[0] == '#')
                        continue;
                if (grmatchline(search, gid, name))
                        return 1;
        }
        /* NOTREACHED */
}

static int
grmatchline(int search, gid_t gid, const char *name)
{
        unsigned long   id;
        char            **m;
        char            *cp, *bp, *ep;

        /* name may be NULL if search is nonzero */

        bp = grline;
        memset(&_gr_group, 0, sizeof(_gr_group));
        _gr_group.gr_name = strsep(&bp, ":\n");
        if (search && name && strcmp(_gr_group.gr_name, name))
                return 0;
        _gr_group.gr_passwd = strsep(&bp, ":\n");
        if (!(cp = strsep(&bp, ":\n")))
                return 0;
        id = strtoul(cp, &ep, 10);
        if (id > GID_MAX || *ep != '\0')
                return 0;
        _gr_group.gr_gid = (gid_t)id;
        if (search && name == NULL && _gr_group.gr_gid != gid)
                return 0;
        cp = NULL;
        if (bp == NULL)
                return 0;
        for (_gr_group.gr_mem = m = members;; bp++) {
                if (m == &members[MAXGRP - 1])
                        break;
                if (*bp == ',') {
                        if (cp) {
                                *bp = '\0';
                                *m++ = cp;
                                cp = NULL;
                        }
                } else if (*bp == '\0' || *bp == '\n' || *bp == ' ') {
                        if (cp) {
                                *bp = '\0';
                                *m++ = cp;
                        }
                        break;
                } else if (cp == NULL)
                        cp = bp;
        }
        *m = NULL;
        return 1;
}


/*
 * user lookup functions
 */

static struct passwd *
gi_getpwnam(const char *name)
{
        int rval;

        if (!pwstart())
                return NULL;
        rval = pwscan(1, 0, name);
        if (!_pw_stayopen)
                endpwent();
        return (rval) ? &_pw_passwd : NULL;
}

static struct passwd *
gi_getpwuid(uid_t uid)
{
        int rval;

        if (!pwstart())
                return NULL;
        rval = pwscan(1, uid, NULL);
        if (!_pw_stayopen)
                endpwent();
        return (rval) ? &_pw_passwd : NULL;
}

static int
gi_setpassent(int stayopen)
{

        if (!pwstart())
                return 0;
        _pw_stayopen = stayopen;
        return 1;
}

static void
gi_endpwent(void)
{

        _pw_filesdone = 0;
        if (_pw_fp) {
                (void)fclose(_pw_fp);
                _pw_fp = NULL;
        }
}

static int
pwstart(void)
{

        _pw_filesdone = 0;
        if (_pw_fp) {
                rewind(_pw_fp);
                return 1;
        }
        if (pwfile[0] == '\0')                  /* sanity check */
                return 0;
        _pw_fp = fopen(pwfile, "r");
        if (_pw_fp != NULL)
                return 1;
        warn("Can't open `%s'", pwfile);
        return 0;
}


static int
pwscan(int search, uid_t uid, const char *name)
{

        if (_pw_filesdone)
                return 0;
        for (;;) {
                if (!fgets(pwline, sizeof(pwline), _pw_fp)) {
                        if (!search)
                                _pw_filesdone = 1;
                        return 0;
                }
                /* skip lines that are too big */
                if (!strchr(pwline, '\n')) {
                        int ch;

                        while ((ch = getc(_pw_fp)) != '\n' && ch != EOF)
                                ;
                        continue;
                }
                /* skip comments */
                if (pwline[0] == '#')
                        continue;
                if (pwmatchline(search, uid, name))
                        return 1;
        }
        /* NOTREACHED */
}

static int
pwmatchline(int search, uid_t uid, const char *name)
{
        unsigned long   id;
        char            *cp, *bp, *ep;

        /* name may be NULL if search is nonzero */

        bp = pwline;
        memset(&_pw_passwd, 0, sizeof(_pw_passwd));
        _pw_passwd.pw_name = strsep(&bp, ":\n");                /* name */
        if (search && name && strcmp(_pw_passwd.pw_name, name))
                return 0;

        _pw_passwd.pw_passwd = strsep(&bp, ":\n");              /* passwd */

        if (!(cp = strsep(&bp, ":\n")))                         /* uid */
                return 0;
        id = strtoul(cp, &ep, 10);
        if (id > UID_MAX || *ep != '\0')
                return 0;
        _pw_passwd.pw_uid = (uid_t)id;
        if (search && name == NULL && _pw_passwd.pw_uid != uid)
                return 0;

        if (!(cp = strsep(&bp, ":\n")))                         /* gid */
                return 0;
        id = strtoul(cp, &ep, 10);
        if (id > GID_MAX || *ep != '\0')
                return 0;
        _pw_passwd.pw_gid = (gid_t)id;

        if (!(ep = strsep(&bp, ":")))                           /* class */
                return 0;
        if (!(ep = strsep(&bp, ":")))                           /* change */
                return 0;
        if (!(ep = strsep(&bp, ":")))                           /* expire */
                return 0;

        if (!(_pw_passwd.pw_gecos = strsep(&bp, ":\n")))        /* gecos */
                return 0;
        if (!(_pw_passwd.pw_dir = strsep(&bp, ":\n")))          /* directory */
                return 0;
        if (!(_pw_passwd.pw_shell = strsep(&bp, ":\n")))        /* shell */
                return 0;

        if (strchr(bp, ':') != NULL)
                return 0;

        return 1;
}