1 /* $NetBSD: svc_simple.c,v 1.33 2013/03/11 20:19:29 tron Exp $ */ 2 3 /* 4 * Copyright (c) 2010, Oracle America, Inc. 5 * 6 * Redistribution and use in source and binary forms, with or without 7 * modification, are permitted provided that the following conditions are 8 * met: 9 * 10 * * Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * * Redistributions in binary form must reproduce the above 13 * copyright notice, this list of conditions and the following 14 * disclaimer in the documentation and/or other materials 15 * provided with the distribution. 16 * * Neither the name of the "Oracle America, 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 21 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 22 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS 23 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE 24 * COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, 25 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE 27 * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 28 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 29 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 30 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 31 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 32 */ 33 /* 34 * Copyright (c) 1986-1991 by Sun Microsystems Inc. 35 */ 36 37 /* #pragma ident "@(#)svc_simple.c 1.18 94/04/24 SMI" */ 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 <sys/cdefs.h> 52 #if defined(LIBC_SCCS) && !defined(lint) 53 __RCSID("$NetBSD: svc_simple.c,v 1.33 2013/03/11 20:19:29 tron Exp $"); 54 #endif 55 56 #include "namespace.h" 57 #include "reentrant.h" 58 #include <sys/types.h> 59 #include <rpc/rpc.h> 60 #include <rpc/nettype.h> 61 #include <assert.h> 62 #include <err.h> 63 #include <stdio.h> 64 #include <stdlib.h> 65 #include <string.h> 66 67 #include "rpc_internal.h" 68 69 #ifdef __weak_alias 70 __weak_alias(rpc_reg,_rpc_reg) 71 #endif 72 73 static void universal(struct svc_req *, SVCXPRT *); 74 75 static struct proglst { 76 char *(*p_progname)(char *); 77 rpcprog_t p_prognum; 78 rpcvers_t p_versnum; 79 rpcproc_t p_procnum; 80 SVCXPRT *p_transp; 81 char *p_netid; 82 char *p_xdrbuf; 83 int p_recvsz; 84 xdrproc_t p_inproc, p_outproc; 85 struct proglst *p_nxt; 86 } *proglst; 87 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( 106 rpcprog_t prognum, /* program number */ 107 rpcvers_t versnum, /* version number */ 108 rpcproc_t procnum, /* procedure number */ 109 char *(*progname)(char *), /* Server routine */ 110 xdrproc_t inproc, /* in XDR procedure */ 111 xdrproc_t outproc, /* out XDR procedure */ 112 char *nettype) /* nettype */ 113 { 114 struct netconfig *nconf; 115 int done = FALSE; 116 void *handle; 117 #ifdef _REENTRANT 118 extern mutex_t proglst_lock; 119 #endif 120 121 if (procnum == NULLPROC) { 122 warnx("%s: can't reassign procedure number %u", __func__, 123 NULLPROC); 124 return (-1); 125 } 126 127 if (nettype == NULL) 128 nettype = __UNCONST("netpath"); /* The default behavior */ 129 if ((handle = __rpc_setconf(nettype)) == NULL) { 130 warnx("%s: %s", __func__, __reg_err1); 131 return (-1); 132 } 133 /* VARIABLES PROTECTED BY proglst_lock: proglst */ 134 mutex_lock(&proglst_lock); 135 while ((nconf = __rpc_getconf(handle)) != NULL) { 136 struct proglst *pl; 137 SVCXPRT *svcxprt; 138 int madenow; 139 u_int recvsz; 140 char *xdrbuf; 141 char *netid; 142 143 madenow = FALSE; 144 svcxprt = NULL; 145 recvsz = 0; 146 xdrbuf = NULL; 147 netid = NULL; 148 for (pl = proglst; pl; pl = pl->p_nxt) 149 if (strcmp(pl->p_netid, nconf->nc_netid) == 0) { 150 svcxprt = pl->p_transp; 151 xdrbuf = pl->p_xdrbuf; 152 recvsz = pl->p_recvsz; 153 netid = pl->p_netid; 154 break; 155 } 156 157 if (svcxprt == NULL) { 158 struct __rpc_sockinfo si; 159 160 svcxprt = svc_tli_create(RPC_ANYFD, nconf, NULL, 0, 0); 161 if (svcxprt == NULL) 162 continue; 163 if (!__rpc_fd2sockinfo(svcxprt->xp_fd, &si)) { 164 warnx("%s: %s", __func__, __reg_err2); 165 SVC_DESTROY(svcxprt); 166 continue; 167 } 168 recvsz = __rpc_get_t_size(si.si_af, si.si_proto, 0); 169 if (recvsz == 0) { 170 warnx("%s: %s", __func__, __reg_err3); 171 SVC_DESTROY(svcxprt); 172 continue; 173 } 174 if (((xdrbuf = mem_alloc((size_t)recvsz)) == NULL) || 175 ((netid = strdup(nconf->nc_netid)) == NULL)) { 176 warnx("%s: %s", __func__, __no_mem_str); 177 if (xdrbuf != NULL) 178 free(xdrbuf); 179 if (netid != NULL) 180 free(netid); 181 SVC_DESTROY(svcxprt); 182 break; 183 } 184 madenow = TRUE; 185 } 186 /* 187 * Check if this (program, version, netid) had already been 188 * registered. The check may save a few RPC calls to rpcbind 189 */ 190 for (pl = proglst; pl; pl = pl->p_nxt) 191 if ((pl->p_prognum == prognum) && 192 (pl->p_versnum == versnum) && 193 (strcmp(pl->p_netid, netid) == 0)) 194 break; 195 if (pl == NULL) { /* Not yet */ 196 (void) rpcb_unset(prognum, versnum, nconf); 197 } else { 198 /* so that svc_reg does not call rpcb_set() */ 199 nconf = NULL; 200 } 201 202 if (!svc_reg(svcxprt, prognum, versnum, universal, nconf)) { 203 warnx("%s: couldn't register prog %u vers %u for %s", 204 __func__, (unsigned)prognum, 205 (unsigned)versnum, netid); 206 if (madenow) { 207 SVC_DESTROY(svcxprt); 208 free(xdrbuf); 209 free(netid); 210 } 211 continue; 212 } 213 214 pl = malloc(sizeof(*pl)); 215 if (pl == NULL) { 216 warn("%s: %s", __func__, __no_mem_str); 217 if (madenow) { 218 SVC_DESTROY(svcxprt); 219 free(xdrbuf); 220 free(netid); 221 } 222 break; 223 } 224 pl->p_progname = progname; 225 pl->p_prognum = prognum; 226 pl->p_versnum = versnum; 227 pl->p_procnum = procnum; 228 pl->p_inproc = inproc; 229 pl->p_outproc = outproc; 230 pl->p_transp = svcxprt; 231 pl->p_xdrbuf = xdrbuf; 232 pl->p_recvsz = recvsz; 233 pl->p_netid = netid; 234 pl->p_nxt = proglst; 235 proglst = pl; 236 done = TRUE; 237 } 238 __rpc_endconf(handle); 239 mutex_unlock(&proglst_lock); 240 241 if (done == FALSE) { 242 warnx("%s: can't find suitable transport for %s", 243 __func__, nettype); 244 return (-1); 245 } 246 return (0); 247 } 248 249 /* 250 * The universal handler for the services registered using registerrpc. 251 * It handles both the connectionless and the connection oriented cases. 252 */ 253 254 static void 255 universal(struct svc_req *rqstp, SVCXPRT *transp) 256 { 257 rpcprog_t prog; 258 rpcvers_t vers; 259 rpcproc_t proc; 260 char *outdata; 261 char *xdrbuf; 262 struct proglst *pl; 263 #ifdef _REENTRANT 264 extern mutex_t proglst_lock; 265 #endif 266 267 _DIAGASSERT(rqstp != NULL); 268 _DIAGASSERT(transp != NULL); 269 270 /* 271 * enforce "procnum 0 is echo" convention 272 */ 273 if (rqstp->rq_proc == NULLPROC) { 274 if (svc_sendreply(transp, (xdrproc_t) xdr_void, NULL) == 275 FALSE) { 276 warnx("%s: svc_sendreply failed", __func__); 277 } 278 return; 279 } 280 prog = rqstp->rq_prog; 281 vers = rqstp->rq_vers; 282 proc = rqstp->rq_proc; 283 mutex_lock(&proglst_lock); 284 for (pl = proglst; pl; pl = pl->p_nxt) 285 if (pl->p_prognum == prog && pl->p_procnum == proc && 286 pl->p_versnum == vers && 287 (strcmp(pl->p_netid, transp->xp_netid) == 0)) { 288 /* decode arguments into a CLEAN buffer */ 289 xdrbuf = pl->p_xdrbuf; 290 /* Zero the arguments: reqd ! */ 291 (void) memset(xdrbuf, 0, (size_t)pl->p_recvsz); 292 /* 293 * Assuming that sizeof (xdrbuf) would be enough 294 * for the arguments; if not then the program 295 * may bomb. BEWARE! 296 */ 297 if (!svc_getargs(transp, pl->p_inproc, xdrbuf)) { 298 svcerr_decode(transp); 299 mutex_unlock(&proglst_lock); 300 return; 301 } 302 outdata = (*(pl->p_progname))(xdrbuf); 303 if (outdata == NULL && 304 pl->p_outproc != (xdrproc_t) xdr_void){ 305 /* there was an error */ 306 mutex_unlock(&proglst_lock); 307 return; 308 } 309 if (!svc_sendreply(transp, pl->p_outproc, outdata)) { 310 warnx("%s: trouble replying to prog %u vers %u", 311 __func__, (unsigned)prog, (unsigned)vers); 312 mutex_unlock(&proglst_lock); 313 return; 314 } 315 /* free the decoded arguments */ 316 (void) svc_freeargs(transp, pl->p_inproc, xdrbuf); 317 mutex_unlock(&proglst_lock); 318 return; 319 } 320 mutex_unlock(&proglst_lock); 321 /* This should never happen */ 322 warnx("%s: never registered prog %u vers %u", __func__, 323 (unsigned)prog, (unsigned)vers); 324 return; 325 } 326