xref: /netbsd-src/lib/libc/rpc/svc_simple.c (revision 23c8222edbfb0f0932d88a8351d3a0cf817dfb9e)
1 /*	$NetBSD: svc_simple.c,v 1.26 2003/09/09 03:56:40 itojun Exp $	*/
2 
3 /*
4  * Sun RPC is a product of Sun Microsystems, Inc. and is provided for
5  * unrestricted use provided that this legend is included on all tape
6  * media and as a part of the software program in whole or part.  Users
7  * may copy or modify Sun RPC without charge, but are not authorized
8  * to license or distribute it to anyone else except as part of a product or
9  * program developed by the user.
10  *
11  * SUN RPC IS PROVIDED AS IS WITH NO WARRANTIES OF ANY KIND INCLUDING THE
12  * WARRANTIES OF DESIGN, MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
13  * PURPOSE, OR ARISING FROM A COURSE OF DEALING, USAGE OR TRADE PRACTICE.
14  *
15  * Sun RPC is provided with no support and without any obligation on the
16  * part of Sun Microsystems, Inc. to assist in its use, correction,
17  * modification or enhancement.
18  *
19  * SUN MICROSYSTEMS, INC. SHALL HAVE NO LIABILITY WITH RESPECT TO THE
20  * INFRINGEMENT OF COPYRIGHTS, TRADE SECRETS OR ANY PATENTS BY SUN RPC
21  * OR ANY PART THEREOF.
22  *
23  * In no event will Sun Microsystems, Inc. be liable for any lost revenue
24  * or profits or other special, indirect and consequential damages, even if
25  * Sun has been advised of the possibility of such damages.
26  *
27  * Sun Microsystems, Inc.
28  * 2550 Garcia Avenue
29  * Mountain View, California  94043
30  */
31 /*
32  * Copyright (c) 1986-1991 by Sun Microsystems Inc.
33  */
34 
35 /* #pragma ident	"@(#)svc_simple.c	1.18	94/04/24 SMI" */
36 
37 /*
38  * svc_simple.c
39  * Simplified front end to rpc.
40  */
41 
42 /*
43  * This interface creates a virtual listener for all the services
44  * started thru rpc_reg(). It listens on the same endpoint for
45  * all the services and then executes the corresponding service
46  * for the given prognum and procnum.
47  */
48 
49 #include <sys/cdefs.h>
50 #if defined(LIBC_SCCS) && !defined(lint)
51 __RCSID("$NetBSD: svc_simple.c,v 1.26 2003/09/09 03:56:40 itojun Exp $");
52 #endif
53 
54 #include "namespace.h"
55 #include "reentrant.h"
56 #include <sys/types.h>
57 #include <rpc/rpc.h>
58 #include <rpc/nettype.h>
59 #include <assert.h>
60 #include <err.h>
61 #include <stdio.h>
62 #include <stdlib.h>
63 #include <string.h>
64 
65 #include "rpc_internal.h"
66 
67 #ifdef __weak_alias
68 __weak_alias(rpc_reg,_rpc_reg)
69 #endif
70 
71 static void universal __P((struct svc_req *, SVCXPRT *));
72 
73 static struct proglst {
74 	char *(*p_progname) __P((char *));
75 	rpcprog_t p_prognum;
76 	rpcvers_t p_versnum;
77 	rpcproc_t p_procnum;
78 	SVCXPRT *p_transp;
79 	char *p_netid;
80 	char *p_xdrbuf;
81 	int p_recvsz;
82 	xdrproc_t p_inproc, p_outproc;
83 	struct proglst *p_nxt;
84 } *proglst;
85 
86 static const char rpc_reg_err[] = "%s: %s";
87 static const char rpc_reg_msg[] = "rpc_reg: ";
88 static const char __reg_err1[] = "can't find appropriate transport";
89 static const char __reg_err2[] = "can't get protocol info";
90 static const char __reg_err3[] = "unsupported transport size";
91 static const char __no_mem_str[] = "out of memory";
92 
93 /*
94  * For simplified, easy to use kind of rpc interfaces.
95  * nettype indicates the type of transport on which the service will be
96  * listening. Used for conservation of the system resource. Only one
97  * handle is created for all the services (actually one of each netid)
98  * and same xdrbuf is used for same netid. The size of the arguments
99  * is also limited by the recvsize for that transport, even if it is
100  * a COTS transport. This may be wrong, but for cases like these, they
101  * should not use the simplified interfaces like this.
102  */
103 
104 int
105 rpc_reg(prognum, versnum, procnum, progname, inproc, outproc, nettype)
106 	rpcprog_t prognum;			/* program number */
107 	rpcvers_t versnum;			/* version number */
108 	rpcproc_t procnum;			/* procedure number */
109 	char *(*progname) __P((char *)); /* Server routine */
110 	xdrproc_t inproc, outproc;	/* in/out XDR procedures */
111 	char *nettype;			/* nettype */
112 {
113 	struct netconfig *nconf;
114 	int done = FALSE;
115 	void *handle;
116 #ifdef _REENTRANT
117 	extern mutex_t proglst_lock;
118 #endif
119 
120 	if (procnum == NULLPROC) {
121 		warnx("%s can't reassign procedure number %u", rpc_reg_msg,
122 			NULLPROC);
123 		return (-1);
124 	}
125 
126 	if (nettype == NULL)
127 		nettype = "netpath";		/* The default behavior */
128 	if ((handle = __rpc_setconf(nettype)) == NULL) {
129 		warnx(rpc_reg_err, rpc_reg_msg, __reg_err1);
130 		return (-1);
131 	}
132 /* VARIABLES PROTECTED BY proglst_lock: proglst */
133 	mutex_lock(&proglst_lock);
134 	while ((nconf = __rpc_getconf(handle)) != NULL) {
135 		struct proglst *pl;
136 		SVCXPRT *svcxprt;
137 		int madenow;
138 		u_int recvsz;
139 		char *xdrbuf;
140 		char *netid;
141 
142 		madenow = FALSE;
143 		svcxprt = NULL;
144 		for (pl = proglst; pl; pl = pl->p_nxt)
145 			if (strcmp(pl->p_netid, nconf->nc_netid) == 0) {
146 				svcxprt = pl->p_transp;
147 				xdrbuf = pl->p_xdrbuf;
148 				recvsz = pl->p_recvsz;
149 				netid = pl->p_netid;
150 				break;
151 			}
152 
153 		if (svcxprt == NULL) {
154 			struct __rpc_sockinfo si;
155 
156 			svcxprt = svc_tli_create(RPC_ANYFD, nconf, NULL, 0, 0);
157 			if (svcxprt == NULL)
158 				continue;
159 			if (!__rpc_fd2sockinfo(svcxprt->xp_fd, &si)) {
160 				warnx(rpc_reg_err, rpc_reg_msg, __reg_err2);
161 				SVC_DESTROY(svcxprt);
162 				continue;
163 			}
164 			recvsz = __rpc_get_t_size(si.si_af, si.si_proto, 0);
165 			if (recvsz == 0) {
166 				warnx(rpc_reg_err, rpc_reg_msg, __reg_err3);
167 				SVC_DESTROY(svcxprt);
168 				continue;
169 			}
170 			if (((xdrbuf = malloc((unsigned)recvsz)) == NULL) ||
171 				((netid = strdup(nconf->nc_netid)) == NULL)) {
172 				warnx(rpc_reg_err, rpc_reg_msg, __no_mem_str);
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 cant 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 
246 static void
247 universal(rqstp, transp)
248 	struct svc_req *rqstp;
249 	SVCXPRT *transp;
250 {
251 	rpcprog_t prog;
252 	rpcvers_t vers;
253 	rpcproc_t proc;
254 	char *outdata;
255 	char *xdrbuf;
256 	struct proglst *pl;
257 #ifdef _REENTRANT
258 	extern mutex_t proglst_lock;
259 #endif
260 
261 	_DIAGASSERT(rqstp != NULL);
262 	_DIAGASSERT(transp != NULL);
263 
264 	/*
265 	 * enforce "procnum 0 is echo" convention
266 	 */
267 	if (rqstp->rq_proc == NULLPROC) {
268 		if (svc_sendreply(transp, (xdrproc_t) xdr_void, NULL) ==
269 		    FALSE) {
270 			warnx("svc_sendreply failed");
271 		}
272 		return;
273 	}
274 	prog = rqstp->rq_prog;
275 	vers = rqstp->rq_vers;
276 	proc = rqstp->rq_proc;
277 	mutex_lock(&proglst_lock);
278 	for (pl = proglst; pl; pl = pl->p_nxt)
279 		if (pl->p_prognum == prog && pl->p_procnum == proc &&
280 			pl->p_versnum == vers &&
281 			(strcmp(pl->p_netid, transp->xp_netid) == 0)) {
282 			/* decode arguments into a CLEAN buffer */
283 			xdrbuf = pl->p_xdrbuf;
284 			/* Zero the arguments: reqd ! */
285 			(void) memset(xdrbuf, 0, (size_t)pl->p_recvsz);
286 			/*
287 			 * Assuming that sizeof (xdrbuf) would be enough
288 			 * for the arguments; if not then the program
289 			 * may bomb. BEWARE!
290 			 */
291 			if (!svc_getargs(transp, pl->p_inproc, xdrbuf)) {
292 				svcerr_decode(transp);
293 				mutex_unlock(&proglst_lock);
294 				return;
295 			}
296 			outdata = (*(pl->p_progname))(xdrbuf);
297 			if (outdata == NULL &&
298 				pl->p_outproc != (xdrproc_t) xdr_void){
299 				/* there was an error */
300 				mutex_unlock(&proglst_lock);
301 				return;
302 			}
303 			if (!svc_sendreply(transp, pl->p_outproc, outdata)) {
304 				warnx(
305 			"rpc: rpc_reg trouble replying to prog %u vers %u",
306 				(unsigned)prog, (unsigned)vers);
307 				mutex_unlock(&proglst_lock);
308 				return;
309 			}
310 			/* free the decoded arguments */
311 			(void) svc_freeargs(transp, pl->p_inproc, xdrbuf);
312 			mutex_unlock(&proglst_lock);
313 			return;
314 		}
315 	mutex_unlock(&proglst_lock);
316 	/* This should never happen */
317 	warnx("rpc: rpc_reg: never registered prog %u vers %u",
318 		(unsigned)prog, (unsigned)vers);
319 	return;
320 }
321