1/*	$NetBSD: svc_simple.c,v 1.20 2000/07/06 03:10:35 christos Exp $	*/
2
3/*-
4 * SPDX-License-Identifier: BSD-3-Clause
5 *
6 * Copyright (c) 2009, Sun Microsystems, Inc.
7 * All rights reserved.
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions are met:
11 * - Redistributions of source code must retain the above copyright notice,
12 *   this list of conditions and the following disclaimer.
13 * - Redistributions in binary form must reproduce the above copyright notice,
14 *   this list of conditions and the following disclaimer in the documentation
15 *   and/or other materials provided with the distribution.
16 * - Neither the name of Sun Microsystems, Inc. nor the names of its
17 *   contributors may be used to endorse or promote products derived
18 *   from this software without specific prior written permission.
19 *
20 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
21 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
24 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
25 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
26 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
27 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
28 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
29 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
30 * POSSIBILITY OF SUCH DAMAGE.
31 */
32/*
33 * Copyright (c) 1986-1991 by Sun Microsystems Inc.
34 */
35
36/* #pragma ident	"@(#)svc_simple.c	1.18	94/04/24 SMI" */
37#include <sys/cdefs.h>
38__FBSDID("$FreeBSD$");
39
40/*
41 * svc_simple.c
42 * Simplified front end to rpc.
43 */
44
45/*
46 * This interface creates a virtual listener for all the services
47 * started through rpc_reg(). It listens on the same endpoint for
48 * all the services and then executes the corresponding service
49 * for the given prognum and procnum.
50 */
51
52#include "namespace.h"
53#include "reentrant.h"
54#include <sys/types.h>
55#include <rpc/rpc.h>
56#include <rpc/nettype.h>
57#include <stdio.h>
58#include <stdlib.h>
59#include <string.h>
60#include <err.h>
61#include "un-namespace.h"
62
63#include "rpc_com.h"
64#include "mt_misc.h"
65
66static void universal(struct svc_req *, SVCXPRT *);
67
68static struct proglst {
69	char *(*p_progname)(char *);
70	rpcprog_t p_prognum;
71	rpcvers_t p_versnum;
72	rpcproc_t p_procnum;
73	SVCXPRT *p_transp;
74	char *p_netid;
75	char *p_xdrbuf;
76	int p_recvsz;
77	xdrproc_t p_inproc, p_outproc;
78	struct proglst *p_nxt;
79} *proglst;
80
81static const char rpc_reg_err[] = "%s: %s";
82static const char rpc_reg_msg[] = "rpc_reg: ";
83static const char __reg_err1[] = "can't find appropriate transport";
84static const char __reg_err2[] = "can't get protocol info";
85static const char __reg_err3[] = "unsupported transport size";
86static const char __no_mem_str[] = "out of memory";
87
88/*
89 * For simplified, easy to use kind of rpc interfaces.
90 * nettype indicates the type of transport on which the service will be
91 * listening. Used for conservation of the system resource. Only one
92 * handle is created for all the services (actually one of each netid)
93 * and same xdrbuf is used for same netid. The size of the arguments
94 * is also limited by the recvsize for that transport, even if it is
95 * a COTS transport. This may be wrong, but for cases like these, they
96 * should not use the simplified interfaces like this.
97 *
98 * prognum - program number
99 * versnum - version number
100 * procnum - procedure number
101 * progname - Server routine
102 * inproc, outproc - in/out XDR procedures
103 * nettype - nettype
104 */
105int
106rpc_reg(rpcprog_t prognum, rpcvers_t versnum, rpcproc_t procnum,
107    char *(*progname)(char *), xdrproc_t inproc, xdrproc_t outproc,
108    char *nettype)
109{
110	struct netconfig *nconf;
111	int done = FALSE;
112	void *handle;
113
114
115	if (procnum == NULLPROC) {
116		warnx("%s can't reassign procedure number %u", rpc_reg_msg,
117			NULLPROC);
118		return (-1);
119	}
120
121	if (nettype == NULL)
122		nettype = "netpath";		/* The default behavior */
123	if ((handle = __rpc_setconf(nettype)) == NULL) {
124		warnx(rpc_reg_err, rpc_reg_msg, __reg_err1);
125		return (-1);
126	}
127/* VARIABLES PROTECTED BY proglst_lock: proglst */
128	mutex_lock(&proglst_lock);
129	while ((nconf = __rpc_getconf(handle)) != NULL) {
130		struct proglst *pl;
131		SVCXPRT *svcxprt;
132		int madenow;
133		u_int recvsz;
134		char *xdrbuf;
135		char *netid;
136
137		madenow = FALSE;
138		svcxprt = NULL;
139		recvsz = 0;
140		xdrbuf = netid = NULL;
141		for (pl = proglst; pl; pl = pl->p_nxt) {
142			if (strcmp(pl->p_netid, nconf->nc_netid) == 0) {
143				svcxprt = pl->p_transp;
144				xdrbuf = pl->p_xdrbuf;
145				recvsz = pl->p_recvsz;
146				netid = pl->p_netid;
147				break;
148			}
149		}
150
151		if (svcxprt == NULL) {
152			struct __rpc_sockinfo si;
153
154			svcxprt = svc_tli_create(RPC_ANYFD, nconf, NULL, 0, 0);
155			if (svcxprt == NULL)
156				continue;
157			if (!__rpc_fd2sockinfo(svcxprt->xp_fd, &si)) {
158				warnx(rpc_reg_err, rpc_reg_msg, __reg_err2);
159				SVC_DESTROY(svcxprt);
160				continue;
161			}
162			recvsz = __rpc_get_t_size(si.si_af, si.si_proto, 0);
163			if (recvsz == 0) {
164				warnx(rpc_reg_err, rpc_reg_msg, __reg_err3);
165				SVC_DESTROY(svcxprt);
166				continue;
167			}
168			if (((xdrbuf = malloc((unsigned)recvsz)) == NULL) ||
169				((netid = strdup(nconf->nc_netid)) == NULL)) {
170				warnx(rpc_reg_err, rpc_reg_msg, __no_mem_str);
171				free(xdrbuf);
172				free(netid);
173				SVC_DESTROY(svcxprt);
174				break;
175			}
176			madenow = TRUE;
177		}
178		/*
179		 * Check if this (program, version, netid) had already been
180		 * registered.  The check may save a few RPC calls to rpcbind
181		 */
182		for (pl = proglst; pl; pl = pl->p_nxt)
183			if ((pl->p_prognum == prognum) &&
184				(pl->p_versnum == versnum) &&
185				(strcmp(pl->p_netid, netid) == 0))
186				break;
187		if (pl == NULL) { /* Not yet */
188			(void) rpcb_unset(prognum, versnum, nconf);
189		} else {
190			/* so that svc_reg does not call rpcb_set() */
191			nconf = NULL;
192		}
193
194		if (!svc_reg(svcxprt, prognum, versnum, universal, nconf)) {
195			warnx("%s couldn't register prog %u vers %u for %s",
196				rpc_reg_msg, (unsigned)prognum,
197				(unsigned)versnum, netid);
198			if (madenow) {
199				SVC_DESTROY(svcxprt);
200				free(xdrbuf);
201				free(netid);
202			}
203			continue;
204		}
205
206		pl = malloc(sizeof (struct proglst));
207		if (pl == NULL) {
208			warnx(rpc_reg_err, rpc_reg_msg, __no_mem_str);
209			if (madenow) {
210				SVC_DESTROY(svcxprt);
211				free(xdrbuf);
212				free(netid);
213			}
214			break;
215		}
216		pl->p_progname = progname;
217		pl->p_prognum = prognum;
218		pl->p_versnum = versnum;
219		pl->p_procnum = procnum;
220		pl->p_inproc = inproc;
221		pl->p_outproc = outproc;
222		pl->p_transp = svcxprt;
223		pl->p_xdrbuf = xdrbuf;
224		pl->p_recvsz = recvsz;
225		pl->p_netid = netid;
226		pl->p_nxt = proglst;
227		proglst = pl;
228		done = TRUE;
229	}
230	__rpc_endconf(handle);
231	mutex_unlock(&proglst_lock);
232
233	if (done == FALSE) {
234		warnx("%s can't find suitable transport for %s",
235			rpc_reg_msg, nettype);
236		return (-1);
237	}
238	return (0);
239}
240
241/*
242 * The universal handler for the services registered using registerrpc.
243 * It handles both the connectionless and the connection oriented cases.
244 */
245
246static void
247universal(struct svc_req *rqstp, SVCXPRT *transp)
248{
249	rpcprog_t prog;
250	rpcvers_t vers;
251	rpcproc_t proc;
252	char *outdata;
253	char *xdrbuf;
254	struct proglst *pl;
255
256	/*
257	 * enforce "procnum 0 is echo" convention
258	 */
259	if (rqstp->rq_proc == NULLPROC) {
260		if (svc_sendreply(transp, (xdrproc_t) xdr_void, NULL) ==
261		    FALSE) {
262			warnx("svc_sendreply failed");
263		}
264		return;
265	}
266	prog = rqstp->rq_prog;
267	vers = rqstp->rq_vers;
268	proc = rqstp->rq_proc;
269	mutex_lock(&proglst_lock);
270	for (pl = proglst; pl; pl = pl->p_nxt)
271		if (pl->p_prognum == prog && pl->p_procnum == proc &&
272			pl->p_versnum == vers &&
273			(strcmp(pl->p_netid, transp->xp_netid) == 0)) {
274			/* decode arguments into a CLEAN buffer */
275			xdrbuf = pl->p_xdrbuf;
276			/* Zero the arguments: reqd ! */
277			(void) memset(xdrbuf, 0, (size_t)pl->p_recvsz);
278			/*
279			 * Assuming that sizeof (xdrbuf) would be enough
280			 * for the arguments; if not then the program
281			 * may bomb. BEWARE!
282			 */
283			if (!svc_getargs(transp, pl->p_inproc, xdrbuf)) {
284				svcerr_decode(transp);
285				mutex_unlock(&proglst_lock);
286				return;
287			}
288			outdata = (*(pl->p_progname))(xdrbuf);
289			if (outdata == NULL &&
290				pl->p_outproc != (xdrproc_t) xdr_void){
291				/* there was an error */
292				mutex_unlock(&proglst_lock);
293				return;
294			}
295			if (!svc_sendreply(transp, pl->p_outproc, outdata)) {
296				warnx(
297			"rpc: rpc_reg trouble replying to prog %u vers %u",
298				(unsigned)prog, (unsigned)vers);
299				mutex_unlock(&proglst_lock);
300				return;
301			}
302			/* free the decoded arguments */
303			(void)svc_freeargs(transp, pl->p_inproc, xdrbuf);
304			mutex_unlock(&proglst_lock);
305			return;
306		}
307	mutex_unlock(&proglst_lock);
308	/* This should never happen */
309	warnx("rpc: rpc_reg: never registered prog %u vers %u",
310		(unsigned)prog, (unsigned)vers);
311	return;
312}
313