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