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$"); 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 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 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 244 static void 245 universal(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