xref: /freebsd/lib/libc/rpc/svc_simple.c (revision b52b9d56d4e96089873a75f9e29062eec19fabba)
1 /*	$NetBSD: svc_simple.c,v 1.20 2000/07/06 03:10:35 christos 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 #include <sys/cdefs.h>
37 __FBSDID("$FreeBSD$");
38 
39 /*
40  * svc_simple.c
41  * Simplified front end to rpc.
42  */
43 
44 /*
45  * This interface creates a virtual listener for all the services
46  * started thru rpc_reg(). It listens on the same endpoint for
47  * all the services and then executes the corresponding service
48  * for the given prognum and procnum.
49  */
50 
51 #include "namespace.h"
52 #include "reentrant.h"
53 #include <sys/types.h>
54 #include <rpc/rpc.h>
55 #include <rpc/nettype.h>
56 #include <stdio.h>
57 #include <stdlib.h>
58 #include <string.h>
59 #include <err.h>
60 #include "un-namespace.h"
61 
62 #include "rpc_com.h"
63 
64 static void universal(struct svc_req *, SVCXPRT *);
65 
66 static 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 
79 static const char rpc_reg_err[] = "%s: %s";
80 static const char rpc_reg_msg[] = "rpc_reg: ";
81 static const char __reg_err1[] = "can't find appropriate transport";
82 static const char __reg_err2[] = "can't get protocol info";
83 static const char __reg_err3[] = "unsupported transport size";
84 static 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 
97 int
98 rpc_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 	extern mutex_t proglst_lock;
110 
111 
112 
113 	if (procnum == NULLPROC) {
114 		warnx("%s can't reassign procedure number %u", rpc_reg_msg,
115 			NULLPROC);
116 		return (-1);
117 	}
118 
119 	if (nettype == NULL)
120 		nettype = "netpath";		/* The default behavior */
121 	if ((handle = __rpc_setconf(nettype)) == NULL) {
122 		warnx(rpc_reg_err, rpc_reg_msg, __reg_err1);
123 		return (-1);
124 	}
125 /* VARIABLES PROTECTED BY proglst_lock: proglst */
126 	mutex_lock(&proglst_lock);
127 	while ((nconf = __rpc_getconf(handle)) != NULL) {
128 		struct proglst *pl;
129 		SVCXPRT *svcxprt;
130 		int madenow;
131 		u_int recvsz;
132 		char *xdrbuf;
133 		char *netid;
134 
135 		madenow = FALSE;
136 		svcxprt = NULL;
137 		recvsz = 0;
138 		xdrbuf = netid = NULL;
139 		for (pl = proglst; pl; pl = pl->p_nxt) {
140 			if (strcmp(pl->p_netid, nconf->nc_netid) == 0) {
141 				svcxprt = pl->p_transp;
142 				xdrbuf = pl->p_xdrbuf;
143 				recvsz = pl->p_recvsz;
144 				netid = pl->p_netid;
145 				break;
146 			}
147 		}
148 
149 		if (svcxprt == NULL) {
150 			struct __rpc_sockinfo si;
151 
152 			svcxprt = svc_tli_create(RPC_ANYFD, nconf, NULL, 0, 0);
153 			if (svcxprt == NULL)
154 				continue;
155 			if (!__rpc_fd2sockinfo(svcxprt->xp_fd, &si)) {
156 				warnx(rpc_reg_err, rpc_reg_msg, __reg_err2);
157 				SVC_DESTROY(svcxprt);
158 				continue;
159 			}
160 			recvsz = __rpc_get_t_size(si.si_af, si.si_proto, 0);
161 			if (recvsz == 0) {
162 				warnx(rpc_reg_err, rpc_reg_msg, __reg_err3);
163 				SVC_DESTROY(svcxprt);
164 				continue;
165 			}
166 			if (((xdrbuf = malloc((unsigned)recvsz)) == NULL) ||
167 				((netid = strdup(nconf->nc_netid)) == NULL)) {
168 				warnx(rpc_reg_err, rpc_reg_msg, __no_mem_str);
169 				SVC_DESTROY(svcxprt);
170 				break;
171 			}
172 			madenow = TRUE;
173 		}
174 		/*
175 		 * Check if this (program, version, netid) had already been
176 		 * registered.  The check may save a few RPC calls to rpcbind
177 		 */
178 		for (pl = proglst; pl; pl = pl->p_nxt)
179 			if ((pl->p_prognum == prognum) &&
180 				(pl->p_versnum == versnum) &&
181 				(strcmp(pl->p_netid, netid) == 0))
182 				break;
183 		if (pl == NULL) { /* Not yet */
184 			(void) rpcb_unset(prognum, versnum, nconf);
185 		} else {
186 			/* so that svc_reg does not call rpcb_set() */
187 			nconf = NULL;
188 		}
189 
190 		if (!svc_reg(svcxprt, prognum, versnum, universal, nconf)) {
191 			warnx("%s couldn't register prog %u vers %u for %s",
192 				rpc_reg_msg, (unsigned)prognum,
193 				(unsigned)versnum, netid);
194 			if (madenow) {
195 				SVC_DESTROY(svcxprt);
196 				free(xdrbuf);
197 				free(netid);
198 			}
199 			continue;
200 		}
201 
202 		pl = malloc(sizeof (struct proglst));
203 		if (pl == NULL) {
204 			warnx(rpc_reg_err, rpc_reg_msg, __no_mem_str);
205 			if (madenow) {
206 				SVC_DESTROY(svcxprt);
207 				free(xdrbuf);
208 				free(netid);
209 			}
210 			break;
211 		}
212 		pl->p_progname = progname;
213 		pl->p_prognum = prognum;
214 		pl->p_versnum = versnum;
215 		pl->p_procnum = procnum;
216 		pl->p_inproc = inproc;
217 		pl->p_outproc = outproc;
218 		pl->p_transp = svcxprt;
219 		pl->p_xdrbuf = xdrbuf;
220 		pl->p_recvsz = recvsz;
221 		pl->p_netid = netid;
222 		pl->p_nxt = proglst;
223 		proglst = pl;
224 		done = TRUE;
225 	}
226 	__rpc_endconf(handle);
227 	mutex_unlock(&proglst_lock);
228 
229 	if (done == FALSE) {
230 		warnx("%s cant find suitable transport for %s",
231 			rpc_reg_msg, nettype);
232 		return (-1);
233 	}
234 	return (0);
235 }
236 
237 /*
238  * The universal handler for the services registered using registerrpc.
239  * It handles both the connectionless and the connection oriented cases.
240  */
241 
242 static void
243 universal(rqstp, transp)
244 	struct svc_req *rqstp;
245 	SVCXPRT *transp;
246 {
247 	rpcprog_t prog;
248 	rpcvers_t vers;
249 	rpcproc_t proc;
250 	char *outdata;
251 	char *xdrbuf;
252 	struct proglst *pl;
253 	extern mutex_t proglst_lock;
254 
255 	/*
256 	 * enforce "procnum 0 is echo" convention
257 	 */
258 	if (rqstp->rq_proc == NULLPROC) {
259 		if (svc_sendreply(transp, (xdrproc_t) xdr_void, NULL) ==
260 		    FALSE) {
261 			warnx("svc_sendreply failed");
262 		}
263 		return;
264 	}
265 	prog = rqstp->rq_prog;
266 	vers = rqstp->rq_vers;
267 	proc = rqstp->rq_proc;
268 	mutex_lock(&proglst_lock);
269 	for (pl = proglst; pl; pl = pl->p_nxt)
270 		if (pl->p_prognum == prog && pl->p_procnum == proc &&
271 			pl->p_versnum == vers &&
272 			(strcmp(pl->p_netid, transp->xp_netid) == 0)) {
273 			/* decode arguments into a CLEAN buffer */
274 			xdrbuf = pl->p_xdrbuf;
275 			/* Zero the arguments: reqd ! */
276 			(void) memset(xdrbuf, 0, sizeof (pl->p_recvsz));
277 			/*
278 			 * Assuming that sizeof (xdrbuf) would be enough
279 			 * for the arguments; if not then the program
280 			 * may bomb. BEWARE!
281 			 */
282 			if (!svc_getargs(transp, pl->p_inproc, xdrbuf)) {
283 				svcerr_decode(transp);
284 				mutex_unlock(&proglst_lock);
285 				return;
286 			}
287 			outdata = (*(pl->p_progname))(xdrbuf);
288 			if (outdata == NULL &&
289 				pl->p_outproc != (xdrproc_t) xdr_void){
290 				/* there was an error */
291 				mutex_unlock(&proglst_lock);
292 				return;
293 			}
294 			if (!svc_sendreply(transp, pl->p_outproc, outdata)) {
295 				warnx(
296 			"rpc: rpc_reg trouble replying to prog %u vers %u",
297 				(unsigned)prog, (unsigned)vers);
298 				mutex_unlock(&proglst_lock);
299 				return;
300 			}
301 			/* free the decoded arguments */
302 			(void)svc_freeargs(transp, pl->p_inproc, xdrbuf);
303 			mutex_unlock(&proglst_lock);
304 			return;
305 		}
306 	mutex_unlock(&proglst_lock);
307 	/* This should never happen */
308 	warnx("rpc: rpc_reg: never registered prog %u vers %u",
309 		(unsigned)prog, (unsigned)vers);
310 	return;
311 }
312