/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */
/*
 * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <unistd.h>
#include <netconfig.h>
#include <netdir.h>
#include <rpc/rpc.h>
#include <sys/file.h>
#include <sys/param.h>
#include "ypxfrd.h"
#include <ndbm.h>
#include <rpcsvc/yp_prot.h>
#include <rpcsvc/nis.h>
#include <strings.h>

#include <sys/isa_defs.h>       /* for ENDIAN defines */

#if defined(_LITTLE_ENDIAN)
#define DOSWAB 1
#endif

static struct timeval TIMEOUT = {25, 0};
static  DBM     *db;

extern bool secure_map;
extern void logprintf(char *, ...);

/* delete the dbm file with name file */
static int
dbm_deletefile(file)
char *file;
{
        char    pag1[MAXPATHLEN];
        char    dir1[MAXPATHLEN];
        int err;
        strcpy(pag1, file);
        strcat(pag1, ".pag");
        strcpy(dir1, file);
        strcat(dir1, ".dir");
        err = 0;
        if (unlink(pag1) < 0) {
                perror("unlinkpag");
                err = -1;
        }

        if (unlink(dir1) < 0) {
                perror("unlinkdir");
                return (-1);
        }
        return (err);
}

/* xdr just the .pag file of a dbm file */
static  bool_t
xdr_pages(xdrs)
        XDR     *xdrs;
{
        static struct pag res;
        struct pag      *PAG;
#ifdef DOSWAB
        short   *s;
        int             i;
#endif
        bool_t  more;
        bool_t  goteof;
        off64_t where;
        int     true = 1;

        goteof = FALSE;
        if (!xdr_pag(xdrs, &res))
                return (FALSE);
        PAG = &res;
        while (true) {
                if (PAG->status == OK) {
#ifdef DOSWAB
                s = (short *)PAG->pag_u.ok.blkdat;
                s[0] = ntohs(s[0]);
                for (i = 1; i <= s[0]; i++)
                        s[i] = ntohs(s[i]);
#endif
                        errno = 0;
                        where = (((off64_t)PAG->pag_u.ok.blkno) * PBLKSIZ);
                        (void) lseek64(db->dbm_pagf, where, L_SET);
                        if (errno != 0) {
                                perror("seek");
                                exit(-1);
                        }
                        if (write(db->dbm_pagf,
                                PAG->pag_u.ok.blkdat, PBLKSIZ) < 0) {
                                perror("write");
                                exit(-1);
                        }
                } else if (PAG->status == GETDBM_ERROR) {
                        (void) printf("clnt call getpag GETDBM_ERROR\n");
                        exit(-1);
                } else if (PAG->status == GETDBM_EOF)
                        goteof = TRUE;
                if (!xdr_bool(xdrs, &more))
                        return (FALSE);
                if (more == FALSE)
                        return (goteof);
                if (!xdr_pag(xdrs, &res))
                        return (FALSE);
        }
        /*NOTREACHED*/
        return (TRUE);
}
/* xdr  just the .dir part of a dbm file */
static  bool_t
xdr_dirs(xdrs)
        XDR     *xdrs;
{
        static  struct dir res;
        struct  dir     *DIR;
        bool_t  more;
        bool_t  goteof;
        off64_t where;
        int     true = 1;

        goteof = FALSE;
        if (!xdr_dir(xdrs, &res))
                return (FALSE);
        DIR = &res;
        while (true) {
                if (DIR->status == OK) {
                        errno = 0;
                        where = (((off64_t)DIR->dir_u.ok.blkno) * DBLKSIZ);
                        (void) lseek64(db->dbm_dirf, where, L_SET);
                        if (errno != 0) {
                                perror("seek");
                                exit(-1);
                        }
                        if (write(db->dbm_dirf,
                                DIR->dir_u.ok.blkdat, DBLKSIZ) < 0) {
                                perror("write");
                                exit(-1);
                        }
                } else if (DIR->status == GETDBM_ERROR) {
                        (void) printf("clnt call getdir GETDBM_ERROR\n");
                        exit(-1);
                } else if (DIR->status == GETDBM_EOF)
                        goteof = TRUE;
                if (!xdr_bool(xdrs, &more))
                        return (FALSE);
                if (more == FALSE)
                        return (goteof);
                if (!xdr_dir(xdrs, &res))
                        return (FALSE);
        }
        /*NOTREACHED*/
        return (TRUE);
}

/*
 * xdr a dbm file from ypxfrd
 * note that if the client or server do not support ndbm
 * we may not use this optional protocol
 */

int
xdr_myfyl(xdrs, objp)
        XDR *xdrs;
        int *objp;
{
        if (!xdr_answer(xdrs, (answer *)objp))
                return (FALSE);

        if (*objp != OK)
                return (TRUE);

        if (!xdr_pages(xdrs))
                return (FALSE);

        if (!xdr_dirs(xdrs))
                return (FALSE);

        return (TRUE);
}

int
ypxfrd_getdbm(tempmap, master, domain, map)
        char *tempmap;
        char *master;
        char *domain;
        char *map;
{
        hosereq rmap;
        CLIENT  *clnt;
        int             res;
        int     recvsiz = 24 * 1024;
        struct netconfig *nconf;
        int fd;
        struct netbuf *svcaddr;
        struct t_bind *tbind;
        char *netid[] = { "tcp6", "tcp" };
        int i, lastnetid = (sizeof (netid)/sizeof (netid[0])) - 1;

        for (i = 0; i <= lastnetid; i++) {
                if ((nconf = getnetconfigent(netid[i])) == NULL) {
                        if (i != lastnetid)
                                continue;
                        logprintf("ypxfr: tcp transport not supported\n");
                        return (-1);
                }
                if ((fd = t_open(nconf->nc_device, O_RDWR, NULL)) == -1) {
                        freenetconfigent(nconf);
                        if (i != lastnetid)
                                continue;
                        logprintf("ypxfr: TLI problems\n");
                        return (-1);
                }
                if (secure_map == TRUE) {
                        if (netdir_options(nconf, ND_SET_RESERVEDPORT, fd,
                                        NULL) == -1) {
                                (void) close(fd);
                                freenetconfigent(nconf);
                                if (i != lastnetid)
                                        continue;
                                logprintf(
                        "ypxfr: cannot bind to reserved port for %s\n%s\n",
                                        netid[i], netdir_sperror());
                                return (-1);
                        }
                }

                /* LINTED pointer alignment */
                if ((tbind = (struct t_bind *)t_alloc(fd, T_BIND, T_ADDR)) ==
                        NULL) {
                        (void) close(fd);
                        freenetconfigent(nconf);
                        if (i != lastnetid)
                                continue;
                        logprintf("ypxfr: TLI problems\n");
                        return (-1);
                }
                svcaddr = &(tbind->addr);
                if (rpcb_getaddr(YPXFRD, 1, nconf, svcaddr, master)
                        == FALSE) {
                        (void) t_free((char *)tbind, T_BIND);
                        (void) close(fd);
                        freenetconfigent(nconf);
                        if (i != lastnetid)
                                continue;
                        logprintf("ypxfr: couldnot get %s address\n", master);
                        return (-1);
                }
                if ((clnt = __nis_clnt_create(fd, nconf, 0, svcaddr, 0,
                                                YPXFRD, 1, recvsiz, 0)) == 0) {
                        (void) t_free((char *)tbind, T_BIND);
                        (void) close(fd);
                        freenetconfigent(nconf);
                        if (i != lastnetid)
                                continue;
                        clnt_pcreateerror(
                                "ypxfr (get_map) - TCP channel create failure");
                        return (-1);
                }
                (void) t_free((char *)tbind, T_BIND);
                break;
        }
        (void) CLNT_CONTROL(clnt, CLSET_FD_CLOSE, (char *)NULL);

        rmap.map = map;
        rmap.domain = domain;
        (void) memset((char *)&res, 0, sizeof (res));
        db = dbm_open(tempmap, O_RDWR + O_CREAT + O_TRUNC, 0777);
        if (db == NULL) {
                logprintf("dbm_open failed %s\n", tempmap);
                perror(tempmap);
                return (-2);
        }

        if (clnt_call(clnt, getdbm, xdr_hosereq, (char *)&rmap, xdr_myfyl,
                (char *)&res, TIMEOUT) != RPC_SUCCESS) {
                logprintf("clnt call to ypxfrd getdbm failed.\n");
                clnt_perror(clnt, "getdbm");
                (void) dbm_deletefile(tempmap);
                return (-3);
        }
        if (res != OK) {
                logprintf("clnt call %s ypxfrd getdbm NOTOK %s %s code=%d\n",
                        master, domain, map, res);
                (void) dbm_deletefile(tempmap);
                return (-4);
        }
        return (0);

}