root/usr/src/cmd/ypcmd/ypxfrd_svc.c
/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License, Version 1.0 only
 * (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 2005 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 *
 */

/*
 * This source was formally rpcgen generated, but has been
 * checked in.
 */

#include "ypxfrd.h"
#include <stdio.h>
#include <stdlib.h> /* getenv, exit */
#include <signal.h>
#include <rpc/pmap_clnt.h> /* for pmap_unset */
#include <string.h> /* strcmp */
#include <unistd.h> /* setsid */
#include <sys/types.h>
#include <memory.h>
#include <stropts.h>
#include <netconfig.h>
#include <sys/resource.h> /* rlimit */
#include <syslog.h>
#include <ndbm.h>
#include "shim.h"
#include "yptol.h"

#ifndef SIG_PF
#define SIG_PF void(*)(int)
#endif

#ifdef DEBUG
#define RPC_SVC_FG
#endif

#define _RPCSVC_CLOSEDOWN 120

/*
 * Copyr 1989 Sun Micro
 * #ident       "@(#)ypxfrd.x   1.2     00/05/01 SMI"
 * This is NOT source code!
 * DO NOT EDIT THIS FILE!
 */
static int _rpcpmstart;         /* Started by a port monitor ? */

/* States a server can be in wrt request */

#define _IDLE 0
#define _SERVED 1

static int _rpcsvcstate = _IDLE;        /* Set when a request is serviced */
static int _rpcsvccount = 0;            /* Number of requests being serviced */

static void
_msgout(char *msg)
{
#ifdef RPC_SVC_FG
        if (_rpcpmstart)
                syslog(LOG_ERR, "%s", msg);
        else
                (void) fprintf(stderr, "%s\n", msg);
#else
        syslog(LOG_ERR, "%s", msg);
#endif
}

static void
closedown(int sig)
{
        if (_rpcsvcstate == _IDLE && _rpcsvccount == 0) {
                int size;
                int i, openfd = 0;

                size = svc_max_pollfd;
                for (i = 0; i < size && openfd < 2; i++)
                        if (svc_pollfd[i].fd >= 0)
                                openfd++;
                if (openfd <= 1)
                        exit(0);
        } else
                _rpcsvcstate = _IDLE;

        (void) signal(SIGALRM, (SIG_PF) closedown);
        (void) alarm(_RPCSVC_CLOSEDOWN/2);
}

static void
ypxfrd_1(struct svc_req *rqstp, register SVCXPRT *transp)
{
        union {
                hosereq getdbm_1_arg;
        } argument;
        char *result;
        xdrproc_t _xdr_argument, _xdr_result;
        char *(*local)(char *, struct svc_req *);

        _rpcsvccount++;
        switch (rqstp->rq_proc) {
        case NULLPROC:
                (void) svc_sendreply(transp,
                        (xdrproc_t)xdr_void, (char *)NULL);
                _rpcsvccount--;
                _rpcsvcstate = _SERVED;
                return;

        case getdbm:
                _xdr_argument = (xdrproc_t)xdr_hosereq;
                _xdr_result = (xdrproc_t)xdr_dbmfyl;
                local = (char *(*)(char *, struct svc_req *)) getdbm_1_svc;
                break;

        default:
                svcerr_noproc(transp);
                _rpcsvccount--;
                _rpcsvcstate = _SERVED;
                return;
        }
        (void) memset((char *)&argument, 0, sizeof (argument));
        if (!svc_getargs(transp, _xdr_argument, (caddr_t)&argument)) {
                svcerr_decode(transp);
                _rpcsvccount--;
                _rpcsvcstate = _SERVED;
                return;
        }
        result = (*local)((char *)&argument, rqstp);
        if (_xdr_result && result != NULL &&
                !svc_sendreply(transp, _xdr_result, result)) {
                svcerr_systemerr(transp);
        }
        if (!svc_freeargs(transp, _xdr_argument, (caddr_t)&argument)) {
                _msgout("unable to free arguments");
                exit(1);
        }
        _rpcsvccount--;
        _rpcsvcstate = _SERVED;
}

int
main()
{
        pid_t pid;
        int i;
        int stat;

        (void) sigset(SIGPIPE, SIG_IGN);

        /*
         * If stdin looks like a TLI endpoint, we assume
         * that we were started by a port monitor. If
         * t_getstate fails with TBADF, this is not a
         * TLI endpoint.
         */
        if (t_getstate(0) != -1 || t_errno != TBADF) {
                char *netid;
                struct netconfig *nconf = NULL;
                SVCXPRT *transp;
                int pmclose;

                _rpcpmstart = 1;
                openlog("ypxfrd", LOG_NDELAY|LOG_PID, LOG_DAEMON);

                if ((netid = getenv("NLSPROVIDER")) == NULL) {
                /* started from inetd */
                        pmclose = 1;
                } else {
                        if ((nconf = getnetconfigent(netid)) == NULL)
                                _msgout("cannot get transport info");

                        pmclose = (t_getstate(0) != T_DATAXFER);
                }
                if ((transp = svc_tli_create(0, nconf, NULL, 0, 0)) == NULL) {
                        _msgout("cannot create server handle");
                        exit(1);
                }
                if (nconf)
                        freenetconfigent(nconf);
                if (!svc_reg(transp, YPXFRD, V1, ypxfrd_1, 0)) {
                        _msgout("unable to register (YPXFRD, V1).");
                        exit(1);
                }
                if (pmclose) {
                        (void) signal(SIGALRM, (SIG_PF) closedown);
                        (void) alarm(_RPCSVC_CLOSEDOWN/2);
                }

                if (yptol_mode) {
                        stat = parseConfig(NULL, NTOL_MAP_FILE);
                        if (stat == 1) {
                                _msgout("NIS to LDAP mapping inactive.");
                        } else if (stat != 0) {
                                _msgout("Aborting after NIS to LDAP "
                                        "mapping error.");
                                exit(1);
                        }
                }

                svc_run();
                exit(1);
                /* NOTREACHED */
        }       else {
#ifndef RPC_SVC_FG
#pragma weak closefrom
                extern void closefrom();
                int size;
                struct rlimit rl;
                pid = fork();
                if (pid < 0) {
                        perror("cannot fork");
                        exit(1);
                }
                if (pid)
                        exit(0);
                closelog();
                if (closefrom != NULL)
                        closefrom(0);
                else {
                        rl.rlim_max = 0;
                        getrlimit(RLIMIT_NOFILE, &rl);
                        if ((size = rl.rlim_max) == 0)
                                exit(1);
                        for (i = 0; i < size; i++)
                                (void) close(i);
                }
                i = open("/dev/null", 2);
                (void) dup2(i, 1);
                (void) dup2(i, 2);
                openlog("ypxfrd", LOG_NDELAY|LOG_PID, LOG_DAEMON);
                setsid();
#endif
        }

        if (yptol_mode) {
                stat = parseConfig(NULL, NTOL_MAP_FILE);
                if (stat == 1) {
                        _msgout("NIS to LDAP mapping inactive.");
                } else if (stat != 0) {
                        _msgout("Aborting after NIS to LDAP mapping error.");
                        exit(1);
                }
        }

        if (!svc_create(ypxfrd_1, YPXFRD, V1, "visible")) {
                _msgout("unable to create (YPXFRD, V1) for visible.");
                exit(1);
        }

        svc_run();
        _msgout("svc_run returned");
        exit(1);
        /* NOTREACHED */
}