/*      $NetBSD: pmap_svc.c,v 1.10 2021/03/07 00:23:06 christos Exp $   */
/*      $FreeBSD: head/usr.sbin/rpcbind/pmap_svc.c 258564 2013-11-25 16:44:02Z hrs $ */

/*-
* Copyright (c) 2009, Sun Microsystems, Inc.
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
* - Redistributions of source code must retain the above copyright notice,
*   this list of conditions and the following disclaimer.
* - 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.
* - Neither the name of Sun Microsystems, Inc. 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 COPYRIGHT HOLDERS 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 COPYRIGHT HOLDER 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) 1984 - 1991 by Sun Microsystems, Inc.
*/

/* #ident       "@(#)pmap_svc.c 1.14    93/07/05 SMI" */

#if 0
#ifndef lint
static  char sccsid[] = "@(#)pmap_svc.c 1.23 89/04/05 Copyr 1984 Sun Micro";
#endif
#endif

/*
* pmap_svc.c
* The server procedure for the version 2 portmapper.
* All the portmapper related interface from the portmap side.
*/

#ifdef PORTMAP
#include <sys/types.h>
#include <sys/socket.h>
#include <stdio.h>
#include <rpc/rpc.h>
#include <rpc/pmap_prot.h>
#include <rpc/rpcb_prot.h>
#ifdef RPCBIND_DEBUG
#include <stdlib.h>
#endif
#include "rpcbind.h"

static struct pmaplist *find_service_pmap(rpcprog_t, rpcvers_t, rpcprot_t);
static bool_t pmapproc_change(struct svc_req *, SVCXPRT *, u_long);
static bool_t pmapproc_getport(struct svc_req *, SVCXPRT *);
static bool_t pmapproc_dump(struct svc_req *, SVCXPRT *);

/*
* Called for all the version 2 inquiries.
*/
void
pmap_service(struct svc_req *rqstp, SVCXPRT *xprt)
{
       rpcbs_procinfo(RPCBVERS_2_STAT, rqstp->rq_proc);
       switch (rqstp->rq_proc) {
       case PMAPPROC_NULL:
               /*
                * Null proc call
                */
#ifdef RPCBIND_DEBUG
               if (debugging)
                       fprintf(stderr, "PMAPPROC_NULL\n");
#endif
               check_access(xprt, rqstp->rq_proc, NULL, PMAPVERS);
               if ((!svc_sendreply(xprt, (xdrproc_t) xdr_void, NULL)) &&
                       debugging) {
                       if (doabort) {
                               rpcbind_abort();
                       }
               }
               break;

       case PMAPPROC_SET:
               /*
                * Set a program, version to port mapping
                */
               pmapproc_change(rqstp, xprt, rqstp->rq_proc);
               break;

       case PMAPPROC_UNSET:
               /*
                * Remove a program, version to port mapping.
                */
               pmapproc_change(rqstp, xprt, rqstp->rq_proc);
               break;

       case PMAPPROC_GETPORT:
               /*
                * Lookup the mapping for a program, version and return its
                * port number.
                */
               pmapproc_getport(rqstp, xprt);
               break;

       case PMAPPROC_DUMP:
               /*
                * Return the current set of mapped program, version
                */
#ifdef RPCBIND_DEBUG
               if (debugging)
                       fprintf(stderr, "PMAPPROC_DUMP\n");
#endif
               pmapproc_dump(rqstp, xprt);
               break;

       case PMAPPROC_CALLIT:
               /*
                * Calls a procedure on the local machine. If the requested
                * procedure is not registered this procedure does not return
                * error information!!
                * This procedure is only supported on rpc/udp and calls via
                * rpc/udp. It passes null authentication parameters.
                */
               rpcbproc_callit_com(rqstp, xprt, PMAPPROC_CALLIT, PMAPVERS);
               break;

       default:
               svcerr_noproc(xprt);
               break;
       }
}

/*
* returns the item with the given program, version number. If that version
* number is not found, it returns the item with that program number, so that
* the port number is now returned to the caller. The caller when makes a
* call to this program, version number, the call will fail and it will
* return with PROGVERS_MISMATCH. The user can then determine the highest
* and the lowest version number for this program using clnt_geterr() and
* use those program version numbers.
*/
static struct pmaplist *
find_service_pmap(rpcprog_t prog, rpcvers_t vers, rpcprot_t prot)
{
       register struct pmaplist *hit = NULL;
       register struct pmaplist *pml;

       for (pml = list_pml; pml != NULL; pml = pml->pml_next) {
               if ((pml->pml_map.pm_prog != prog) ||
                       (pml->pml_map.pm_prot != prot))
                       continue;
               hit = pml;
               if (pml->pml_map.pm_vers == vers)
                       break;
       }
       return (hit);
}

static bool_t
pmapproc_change(struct svc_req *rqstp __unused, SVCXPRT *xprt, unsigned long op)
{
       struct pmap reg;
       RPCB rpcbreg;
       long ans;
       struct sockcred *sc;
       char uidbuf[32];

       if (!svc_getargs(xprt, (xdrproc_t) xdr_pmap, (char *)&reg)) {
               svcerr_decode(xprt);
               return (FALSE);
       }

#ifdef RPCBIND_DEBUG
       if (debugging)
               fprintf(stderr, "%s request for (%lu, %lu) : ",
                   op == PMAPPROC_SET ? "PMAP_SET" : "PMAP_UNSET",
                   reg.pm_prog, reg.pm_vers);
#endif

       if (!check_access(xprt, op, &reg, PMAPVERS)) {
               svcerr_weakauth(xprt);
               return FALSE;
       }

       (void)svc_getcaller(xprt);
       sc = __svc_getcallercreds(xprt);

       /*
        * Can't use getpwnam here. We might end up calling ourselves
        * and looping.
        */
       if (sc == NULL)
               rpcbreg.r_owner = __UNCONST(rpcbind_unknown);
       else if (sc->sc_uid == 0)
               rpcbreg.r_owner = __UNCONST(rpcbind_superuser);
       else {
               /* r_owner will be strdup-ed later */
               snprintf(uidbuf, sizeof uidbuf, "%d", sc->sc_uid);
               rpcbreg.r_owner = uidbuf;
       }

       rpcbreg.r_prog = reg.pm_prog;
       rpcbreg.r_vers = reg.pm_vers;

       if (op == PMAPPROC_SET) {
               char buf[32];

               snprintf(buf, sizeof(buf), "0.0.0.0.%d.%d",
                   (int)((reg.pm_port >> 8) & 0xff),
                   (int)(reg.pm_port & 0xff));
               rpcbreg.r_addr = buf;
               if (reg.pm_prot == IPPROTO_UDP) {
                       rpcbreg.r_netid = __UNCONST(udptrans);
               } else if (reg.pm_prot == IPPROTO_TCP) {
                       rpcbreg.r_netid = __UNCONST(tcptrans);
               } else {
                       ans = FALSE;
                       goto done_change;
               }
               ans = map_set(&rpcbreg, rpcbreg.r_owner);
       } else if (op == PMAPPROC_UNSET) {
               bool_t ans1, ans2;

               rpcbreg.r_addr = NULL;
               rpcbreg.r_netid = __UNCONST(tcptrans);
               ans1 = map_unset(&rpcbreg, rpcbreg.r_owner);
               rpcbreg.r_netid = __UNCONST(udptrans);
               ans2 = map_unset(&rpcbreg, rpcbreg.r_owner);
               ans = ans1 || ans2;
       } else {
               ans = FALSE;
       }
done_change:
       if ((!svc_sendreply(xprt, (xdrproc_t) xdr_long, (caddr_t) &ans)) &&
           debugging) {
               fprintf(stderr, "portmap: svc_sendreply\n");
               if (doabort) {
                       rpcbind_abort();
               }
       }
#ifdef RPCBIND_DEBUG
       if (debugging)
               fprintf(stderr, "%s\n", ans == TRUE ? "succeeded" : "failed");
#endif
       if (op == PMAPPROC_SET)
               rpcbs_set(RPCBVERS_2_STAT, ans);
       else
               rpcbs_unset(RPCBVERS_2_STAT, ans);
       return (TRUE);
}

/* ARGSUSED */
static bool_t
pmapproc_getport(struct svc_req *rqstp __unused, SVCXPRT *xprt)
{
       struct pmap reg;
       long lport;
       int port = 0;
       struct pmaplist *fnd;
#ifdef RPCBIND_DEBUG
       char *uaddr;
#endif

       if (!svc_getargs(xprt, (xdrproc_t) xdr_pmap, (char *)&reg)) {
               svcerr_decode(xprt);
               return (FALSE);
       }

       if (!check_access(xprt, PMAPPROC_GETPORT, &reg, PMAPVERS)) {
               svcerr_weakauth(xprt);
               return FALSE;
       }

#ifdef RPCBIND_DEBUG
       if (debugging) {
               uaddr =  taddr2uaddr(rpcbind_get_conf(xprt->xp_netid),
                           svc_getrpccaller(xprt));
               fprintf(stderr, "PMAP_GETPORT req for (%lu, %lu, %s) from %s :",
                       reg.pm_prog, reg.pm_vers,
                       reg.pm_prot == IPPROTO_UDP ? "udp" : "tcp", uaddr);
               free(uaddr);
       }
#endif
       fnd = find_service_pmap(reg.pm_prog, reg.pm_vers, reg.pm_prot);
       if (fnd) {
               char serveuaddr[32];
               int h1, h2, h3, h4, p1, p2;
               const char *netid, *ua;

               if (reg.pm_prot == IPPROTO_UDP) {
                       ua = udp_uaddr;
                       netid = udptrans;
               } else {
                       ua = tcp_uaddr; /* To get the len */
                       netid = tcptrans;
               }
               if (ua == NULL) {
                       goto sendreply;
               }
               if (sscanf(ua, "%d.%d.%d.%d.%d.%d", &h1, &h2, &h3,
                               &h4, &p1, &p2) == 6) {
                       p1 = (fnd->pml_map.pm_port >> 8) & 0xff;
                       p2 = (fnd->pml_map.pm_port) & 0xff;
                       snprintf(serveuaddr, sizeof(serveuaddr),
                           "%d.%d.%d.%d.%d.%d", h1, h2, h3, h4, p1, p2);
                       if (is_bound(netid, serveuaddr)) {
                               port = fnd->pml_map.pm_port;
                       } else { /* this service is dead; delete it */
                               delete_prog(reg.pm_prog);
                       }
               }
       }
sendreply:
       lport = port;
       if ((!svc_sendreply(xprt, (xdrproc_t) xdr_long, (caddr_t)&lport)) &&
                       debugging) {
               (void) fprintf(stderr, "portmap: svc_sendreply\n");
               if (doabort) {
                       rpcbind_abort();
               }
       }
#ifdef RPCBIND_DEBUG
       if (debugging)
               fprintf(stderr, "port = %d\n", port);
#endif
       rpcbs_getaddr(RPCBVERS_2_STAT, reg.pm_prog, reg.pm_vers,
               reg.pm_prot == IPPROTO_UDP ? udptrans : tcptrans,
               port ? udptrans : "");

       return (TRUE);
}

/* ARGSUSED */
static bool_t
pmapproc_dump(struct svc_req *rqstp __unused, SVCXPRT *xprt)
{
       if (!svc_getargs(xprt, (xdrproc_t)xdr_void, NULL)) {
               svcerr_decode(xprt);
               return (FALSE);
       }

       if (!check_access(xprt, PMAPPROC_DUMP, NULL, PMAPVERS)) {
               svcerr_weakauth(xprt);
               return FALSE;
       }

       if ((!svc_sendreply(xprt, (xdrproc_t) xdr_pmaplist_ptr,
                       (caddr_t)&list_pml)) && debugging) {
               if (debugging)
                       (void) fprintf(stderr, "portmap: svc_sendreply\n");
               if (doabort) {
                       rpcbind_abort();
               }
       }
       return (TRUE);
}

#endif /* PORTMAP */