1 /* $NetBSD: clnt_dg.c,v 1.4 2000/07/14 08:40:41 fvdl 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 #if defined(LIBC_SCCS) && !defined(lint) 35 #ident "@(#)clnt_dg.c 1.23 94/04/22 SMI" 36 static char sccsid[] = "@(#)clnt_dg.c 1.19 89/03/16 Copyr 1988 Sun Micro"; 37 #endif 38 #include <sys/cdefs.h> 39 __FBSDID("$FreeBSD$"); 40 41 /* 42 * Implements a connectionless client side RPC. 43 */ 44 45 #include "namespace.h" 46 #include "reentrant.h" 47 #include <sys/types.h> 48 #include <sys/event.h> 49 #include <sys/time.h> 50 #include <sys/socket.h> 51 #include <sys/ioctl.h> 52 #include <arpa/inet.h> 53 #include <rpc/rpc.h> 54 #include <rpc/rpcsec_gss.h> 55 #include <errno.h> 56 #include <stdlib.h> 57 #include <string.h> 58 #include <signal.h> 59 #include <unistd.h> 60 #include <err.h> 61 #include "un-namespace.h" 62 #include "rpc_com.h" 63 #include "mt_misc.h" 64 65 66 #ifdef _FREEFALL_CONFIG 67 /* 68 * Disable RPC exponential back-off for FreeBSD.org systems. 69 */ 70 #define RPC_MAX_BACKOFF 1 /* second */ 71 #else 72 #define RPC_MAX_BACKOFF 30 /* seconds */ 73 #endif 74 75 76 static struct clnt_ops *clnt_dg_ops(void); 77 static bool_t time_not_ok(struct timeval *); 78 static enum clnt_stat clnt_dg_call(CLIENT *, rpcproc_t, xdrproc_t, void *, 79 xdrproc_t, void *, struct timeval); 80 static void clnt_dg_geterr(CLIENT *, struct rpc_err *); 81 static bool_t clnt_dg_freeres(CLIENT *, xdrproc_t, void *); 82 static void clnt_dg_abort(CLIENT *); 83 static bool_t clnt_dg_control(CLIENT *, u_int, void *); 84 static void clnt_dg_destroy(CLIENT *); 85 86 87 88 89 /* 90 * This machinery implements per-fd locks for MT-safety. It is not 91 * sufficient to do per-CLIENT handle locks for MT-safety because a 92 * user may create more than one CLIENT handle with the same fd behind 93 * it. Therfore, we allocate an array of flags (dg_fd_locks), protected 94 * by the clnt_fd_lock mutex, and an array (dg_cv) of condition variables 95 * similarly protected. Dg_fd_lock[fd] == 1 => a call is activte on some 96 * CLIENT handle created for that fd. 97 * The current implementation holds locks across the entire RPC and reply, 98 * including retransmissions. Yes, this is silly, and as soon as this 99 * code is proven to work, this should be the first thing fixed. One step 100 * at a time. 101 */ 102 static int *dg_fd_locks; 103 static cond_t *dg_cv; 104 #define release_fd_lock(fd, mask) { \ 105 mutex_lock(&clnt_fd_lock); \ 106 dg_fd_locks[fd] = 0; \ 107 mutex_unlock(&clnt_fd_lock); \ 108 thr_sigsetmask(SIG_SETMASK, &(mask), NULL); \ 109 cond_signal(&dg_cv[fd]); \ 110 } 111 112 static const char mem_err_clnt_dg[] = "clnt_dg_create: out of memory"; 113 114 /* VARIABLES PROTECTED BY clnt_fd_lock: dg_fd_locks, dg_cv */ 115 116 #define MCALL_MSG_SIZE 24 117 118 /* 119 * Private data kept per client handle 120 */ 121 struct cu_data { 122 int cu_fd; /* connections fd */ 123 bool_t cu_closeit; /* opened by library */ 124 struct sockaddr_storage cu_raddr; /* remote address */ 125 int cu_rlen; 126 struct timeval cu_wait; /* retransmit interval */ 127 struct timeval cu_total; /* total time for the call */ 128 struct rpc_err cu_error; 129 XDR cu_outxdrs; 130 u_int cu_xdrpos; 131 u_int cu_sendsz; /* send size */ 132 char cu_outhdr[MCALL_MSG_SIZE]; 133 char *cu_outbuf; 134 u_int cu_recvsz; /* recv size */ 135 int cu_async; 136 int cu_connect; /* Use connect(). */ 137 int cu_connected; /* Have done connect(). */ 138 struct kevent cu_kin; 139 int cu_kq; 140 char cu_inbuf[1]; 141 }; 142 143 /* 144 * Connection less client creation returns with client handle parameters. 145 * Default options are set, which the user can change using clnt_control(). 146 * fd should be open and bound. 147 * NB: The rpch->cl_auth is initialized to null authentication. 148 * Caller may wish to set this something more useful. 149 * 150 * sendsz and recvsz are the maximum allowable packet sizes that can be 151 * sent and received. Normally they are the same, but they can be 152 * changed to improve the program efficiency and buffer allocation. 153 * If they are 0, use the transport default. 154 * 155 * If svcaddr is NULL, returns NULL. 156 */ 157 CLIENT * 158 clnt_dg_create(fd, svcaddr, program, version, sendsz, recvsz) 159 int fd; /* open file descriptor */ 160 const struct netbuf *svcaddr; /* servers address */ 161 rpcprog_t program; /* program number */ 162 rpcvers_t version; /* version number */ 163 u_int sendsz; /* buffer recv size */ 164 u_int recvsz; /* buffer send size */ 165 { 166 CLIENT *cl = NULL; /* client handle */ 167 struct cu_data *cu = NULL; /* private data */ 168 struct timeval now; 169 struct rpc_msg call_msg; 170 sigset_t mask; 171 sigset_t newmask; 172 struct __rpc_sockinfo si; 173 int one = 1; 174 175 sigfillset(&newmask); 176 thr_sigsetmask(SIG_SETMASK, &newmask, &mask); 177 mutex_lock(&clnt_fd_lock); 178 if (dg_fd_locks == (int *) NULL) { 179 int cv_allocsz; 180 size_t fd_allocsz; 181 int dtbsize = __rpc_dtbsize(); 182 183 fd_allocsz = dtbsize * sizeof (int); 184 dg_fd_locks = (int *) mem_alloc(fd_allocsz); 185 if (dg_fd_locks == (int *) NULL) { 186 mutex_unlock(&clnt_fd_lock); 187 thr_sigsetmask(SIG_SETMASK, &(mask), NULL); 188 goto err1; 189 } else 190 memset(dg_fd_locks, '\0', fd_allocsz); 191 192 cv_allocsz = dtbsize * sizeof (cond_t); 193 dg_cv = (cond_t *) mem_alloc(cv_allocsz); 194 if (dg_cv == (cond_t *) NULL) { 195 mem_free(dg_fd_locks, fd_allocsz); 196 dg_fd_locks = (int *) NULL; 197 mutex_unlock(&clnt_fd_lock); 198 thr_sigsetmask(SIG_SETMASK, &(mask), NULL); 199 goto err1; 200 } else { 201 int i; 202 203 for (i = 0; i < dtbsize; i++) 204 cond_init(&dg_cv[i], 0, (void *) 0); 205 } 206 } 207 208 mutex_unlock(&clnt_fd_lock); 209 thr_sigsetmask(SIG_SETMASK, &(mask), NULL); 210 211 if (svcaddr == NULL) { 212 rpc_createerr.cf_stat = RPC_UNKNOWNADDR; 213 return (NULL); 214 } 215 216 if (!__rpc_fd2sockinfo(fd, &si)) { 217 rpc_createerr.cf_stat = RPC_TLIERROR; 218 rpc_createerr.cf_error.re_errno = 0; 219 return (NULL); 220 } 221 /* 222 * Find the receive and the send size 223 */ 224 sendsz = __rpc_get_t_size(si.si_af, si.si_proto, (int)sendsz); 225 recvsz = __rpc_get_t_size(si.si_af, si.si_proto, (int)recvsz); 226 if ((sendsz == 0) || (recvsz == 0)) { 227 rpc_createerr.cf_stat = RPC_TLIERROR; /* XXX */ 228 rpc_createerr.cf_error.re_errno = 0; 229 return (NULL); 230 } 231 232 if ((cl = mem_alloc(sizeof (CLIENT))) == NULL) 233 goto err1; 234 /* 235 * Should be multiple of 4 for XDR. 236 */ 237 sendsz = ((sendsz + 3) / 4) * 4; 238 recvsz = ((recvsz + 3) / 4) * 4; 239 cu = mem_alloc(sizeof (*cu) + sendsz + recvsz); 240 if (cu == NULL) 241 goto err1; 242 (void) memcpy(&cu->cu_raddr, svcaddr->buf, (size_t)svcaddr->len); 243 cu->cu_rlen = svcaddr->len; 244 cu->cu_outbuf = &cu->cu_inbuf[recvsz]; 245 /* Other values can also be set through clnt_control() */ 246 cu->cu_wait.tv_sec = 15; /* heuristically chosen */ 247 cu->cu_wait.tv_usec = 0; 248 cu->cu_total.tv_sec = -1; 249 cu->cu_total.tv_usec = -1; 250 cu->cu_sendsz = sendsz; 251 cu->cu_recvsz = recvsz; 252 cu->cu_async = FALSE; 253 cu->cu_connect = FALSE; 254 cu->cu_connected = FALSE; 255 (void) gettimeofday(&now, NULL); 256 call_msg.rm_xid = __RPC_GETXID(&now); 257 call_msg.rm_call.cb_prog = program; 258 call_msg.rm_call.cb_vers = version; 259 xdrmem_create(&(cu->cu_outxdrs), cu->cu_outhdr, MCALL_MSG_SIZE, 260 XDR_ENCODE); 261 if (! xdr_callhdr(&cu->cu_outxdrs, &call_msg)) { 262 rpc_createerr.cf_stat = RPC_CANTENCODEARGS; /* XXX */ 263 rpc_createerr.cf_error.re_errno = 0; 264 goto err2; 265 } 266 cu->cu_xdrpos = XDR_GETPOS(&(cu->cu_outxdrs)); 267 XDR_DESTROY(&cu->cu_outxdrs); 268 xdrmem_create(&cu->cu_outxdrs, cu->cu_outbuf, sendsz, XDR_ENCODE); 269 270 /* XXX fvdl - do we still want this? */ 271 #if 0 272 (void)bindresvport_sa(fd, (struct sockaddr *)svcaddr->buf); 273 #endif 274 _ioctl(fd, FIONBIO, (char *)(void *)&one); 275 276 /* 277 * By default, closeit is always FALSE. It is users responsibility 278 * to do a close on it, else the user may use clnt_control 279 * to let clnt_destroy do it for him/her. 280 */ 281 cu->cu_closeit = FALSE; 282 cu->cu_fd = fd; 283 cl->cl_ops = clnt_dg_ops(); 284 cl->cl_private = (caddr_t)(void *)cu; 285 cl->cl_auth = authnone_create(); 286 cl->cl_tp = NULL; 287 cl->cl_netid = NULL; 288 cu->cu_kq = -1; 289 EV_SET(&cu->cu_kin, cu->cu_fd, EVFILT_READ, EV_ADD, 0, 0, 0); 290 return (cl); 291 err1: 292 warnx(mem_err_clnt_dg); 293 rpc_createerr.cf_stat = RPC_SYSTEMERROR; 294 rpc_createerr.cf_error.re_errno = errno; 295 err2: 296 if (cl) { 297 mem_free(cl, sizeof (CLIENT)); 298 if (cu) 299 mem_free(cu, sizeof (*cu) + sendsz + recvsz); 300 } 301 return (NULL); 302 } 303 304 static enum clnt_stat 305 clnt_dg_call(cl, proc, xargs, argsp, xresults, resultsp, utimeout) 306 CLIENT *cl; /* client handle */ 307 rpcproc_t proc; /* procedure number */ 308 xdrproc_t xargs; /* xdr routine for args */ 309 void *argsp; /* pointer to args */ 310 xdrproc_t xresults; /* xdr routine for results */ 311 void *resultsp; /* pointer to results */ 312 struct timeval utimeout; /* seconds to wait before giving up */ 313 { 314 struct cu_data *cu = (struct cu_data *)cl->cl_private; 315 XDR *xdrs; 316 size_t outlen = 0; 317 struct rpc_msg reply_msg; 318 XDR reply_xdrs; 319 bool_t ok; 320 int nrefreshes = 2; /* number of times to refresh cred */ 321 int nretries = 0; /* number of times we retransmitted */ 322 struct timeval timeout; 323 struct timeval retransmit_time; 324 struct timeval next_sendtime, starttime, time_waited, tv; 325 struct timespec ts; 326 struct kevent kv; 327 struct sockaddr *sa; 328 sigset_t mask; 329 sigset_t newmask; 330 socklen_t inlen, salen; 331 ssize_t recvlen = 0; 332 int kin_len, n, rpc_lock_value; 333 u_int32_t xid; 334 335 outlen = 0; 336 sigfillset(&newmask); 337 thr_sigsetmask(SIG_SETMASK, &newmask, &mask); 338 mutex_lock(&clnt_fd_lock); 339 while (dg_fd_locks[cu->cu_fd]) 340 cond_wait(&dg_cv[cu->cu_fd], &clnt_fd_lock); 341 if (__isthreaded) 342 rpc_lock_value = 1; 343 else 344 rpc_lock_value = 0; 345 dg_fd_locks[cu->cu_fd] = rpc_lock_value; 346 mutex_unlock(&clnt_fd_lock); 347 if (cu->cu_total.tv_usec == -1) { 348 timeout = utimeout; /* use supplied timeout */ 349 } else { 350 timeout = cu->cu_total; /* use default timeout */ 351 } 352 353 if (cu->cu_connect && !cu->cu_connected) { 354 if (_connect(cu->cu_fd, (struct sockaddr *)&cu->cu_raddr, 355 cu->cu_rlen) < 0) { 356 cu->cu_error.re_errno = errno; 357 cu->cu_error.re_status = RPC_CANTSEND; 358 goto out; 359 } 360 cu->cu_connected = 1; 361 } 362 if (cu->cu_connected) { 363 sa = NULL; 364 salen = 0; 365 } else { 366 sa = (struct sockaddr *)&cu->cu_raddr; 367 salen = cu->cu_rlen; 368 } 369 time_waited.tv_sec = 0; 370 time_waited.tv_usec = 0; 371 retransmit_time = next_sendtime = cu->cu_wait; 372 gettimeofday(&starttime, NULL); 373 374 /* Clean up in case the last call ended in a longjmp(3) call. */ 375 if (cu->cu_kq >= 0) 376 _close(cu->cu_kq); 377 if ((cu->cu_kq = kqueue()) < 0) { 378 cu->cu_error.re_errno = errno; 379 cu->cu_error.re_status = RPC_CANTSEND; 380 goto out; 381 } 382 kin_len = 1; 383 384 call_again: 385 if (cu->cu_async == TRUE && xargs == NULL) 386 goto get_reply; 387 /* 388 * the transaction is the first thing in the out buffer 389 * XXX Yes, and it's in network byte order, so we should to 390 * be careful when we increment it, shouldn't we. 391 */ 392 xid = ntohl(*(u_int32_t *)(void *)(cu->cu_outhdr)); 393 xid++; 394 *(u_int32_t *)(void *)(cu->cu_outhdr) = htonl(xid); 395 call_again_same_xid: 396 xdrs = &(cu->cu_outxdrs); 397 xdrs->x_op = XDR_ENCODE; 398 XDR_SETPOS(xdrs, 0); 399 400 if (cl->cl_auth->ah_cred.oa_flavor != RPCSEC_GSS) { 401 if ((! XDR_PUTBYTES(xdrs, cu->cu_outhdr, cu->cu_xdrpos)) || 402 (! XDR_PUTINT32(xdrs, &proc)) || 403 (! AUTH_MARSHALL(cl->cl_auth, xdrs)) || 404 (! (*xargs)(xdrs, argsp))) { 405 cu->cu_error.re_status = RPC_CANTENCODEARGS; 406 goto out; 407 } 408 } else { 409 *(uint32_t *) &cu->cu_outhdr[cu->cu_xdrpos] = htonl(proc); 410 if (!__rpc_gss_wrap(cl->cl_auth, cu->cu_outhdr, 411 cu->cu_xdrpos + sizeof(uint32_t), 412 xdrs, xargs, argsp)) { 413 cu->cu_error.re_status = RPC_CANTENCODEARGS; 414 goto out; 415 } 416 } 417 outlen = (size_t)XDR_GETPOS(xdrs); 418 419 send_again: 420 if (_sendto(cu->cu_fd, cu->cu_outbuf, outlen, 0, sa, salen) != outlen) { 421 cu->cu_error.re_errno = errno; 422 cu->cu_error.re_status = RPC_CANTSEND; 423 goto out; 424 } 425 426 /* 427 * Hack to provide rpc-based message passing 428 */ 429 if (timeout.tv_sec == 0 && timeout.tv_usec == 0) { 430 cu->cu_error.re_status = RPC_TIMEDOUT; 431 goto out; 432 } 433 434 get_reply: 435 436 /* 437 * sub-optimal code appears here because we have 438 * some clock time to spare while the packets are in flight. 439 * (We assume that this is actually only executed once.) 440 */ 441 reply_msg.acpted_rply.ar_verf = _null_auth; 442 if (cl->cl_auth->ah_cred.oa_flavor != RPCSEC_GSS) { 443 reply_msg.acpted_rply.ar_results.where = resultsp; 444 reply_msg.acpted_rply.ar_results.proc = xresults; 445 } else { 446 reply_msg.acpted_rply.ar_results.where = NULL; 447 reply_msg.acpted_rply.ar_results.proc = (xdrproc_t)xdr_void; 448 } 449 450 for (;;) { 451 /* Decide how long to wait. */ 452 if (timercmp(&next_sendtime, &timeout, <)) 453 timersub(&next_sendtime, &time_waited, &tv); 454 else 455 timersub(&timeout, &time_waited, &tv); 456 if (tv.tv_sec < 0 || tv.tv_usec < 0) 457 tv.tv_sec = tv.tv_usec = 0; 458 TIMEVAL_TO_TIMESPEC(&tv, &ts); 459 460 n = _kevent(cu->cu_kq, &cu->cu_kin, kin_len, &kv, 1, &ts); 461 /* We don't need to register the event again. */ 462 kin_len = 0; 463 464 if (n == 1) { 465 if (kv.flags & EV_ERROR) { 466 cu->cu_error.re_errno = kv.data; 467 cu->cu_error.re_status = RPC_CANTRECV; 468 goto out; 469 } 470 /* We have some data now */ 471 do { 472 recvlen = _recvfrom(cu->cu_fd, cu->cu_inbuf, 473 cu->cu_recvsz, 0, NULL, NULL); 474 } while (recvlen < 0 && errno == EINTR); 475 if (recvlen < 0 && errno != EWOULDBLOCK) { 476 cu->cu_error.re_errno = errno; 477 cu->cu_error.re_status = RPC_CANTRECV; 478 goto out; 479 } 480 if (recvlen >= sizeof(u_int32_t) && 481 (cu->cu_async == TRUE || 482 *((u_int32_t *)(void *)(cu->cu_inbuf)) == 483 *((u_int32_t *)(void *)(cu->cu_outbuf)))) { 484 /* We now assume we have the proper reply. */ 485 break; 486 } 487 } 488 if (n == -1 && errno != EINTR) { 489 cu->cu_error.re_errno = errno; 490 cu->cu_error.re_status = RPC_CANTRECV; 491 goto out; 492 } 493 gettimeofday(&tv, NULL); 494 timersub(&tv, &starttime, &time_waited); 495 496 /* Check for timeout. */ 497 if (timercmp(&time_waited, &timeout, >)) { 498 cu->cu_error.re_status = RPC_TIMEDOUT; 499 goto out; 500 } 501 502 /* Retransmit if necessary. */ 503 if (timercmp(&time_waited, &next_sendtime, >)) { 504 /* update retransmit_time */ 505 if (retransmit_time.tv_sec < RPC_MAX_BACKOFF) 506 timeradd(&retransmit_time, &retransmit_time, 507 &retransmit_time); 508 timeradd(&next_sendtime, &retransmit_time, 509 &next_sendtime); 510 nretries++; 511 512 /* 513 * When retransmitting a RPCSEC_GSS message, 514 * we must use a new sequence number (handled 515 * by __rpc_gss_wrap above). 516 */ 517 if (cl->cl_auth->ah_cred.oa_flavor != RPCSEC_GSS) 518 goto send_again; 519 else 520 goto call_again_same_xid; 521 } 522 } 523 inlen = (socklen_t)recvlen; 524 525 /* 526 * now decode and validate the response 527 */ 528 529 xdrmem_create(&reply_xdrs, cu->cu_inbuf, (u_int)recvlen, XDR_DECODE); 530 ok = xdr_replymsg(&reply_xdrs, &reply_msg); 531 /* XDR_DESTROY(&reply_xdrs); save a few cycles on noop destroy */ 532 if (ok) { 533 if ((reply_msg.rm_reply.rp_stat == MSG_ACCEPTED) && 534 (reply_msg.acpted_rply.ar_stat == SUCCESS)) 535 cu->cu_error.re_status = RPC_SUCCESS; 536 else 537 _seterr_reply(&reply_msg, &(cu->cu_error)); 538 539 if (cu->cu_error.re_status == RPC_SUCCESS) { 540 if (! AUTH_VALIDATE(cl->cl_auth, 541 &reply_msg.acpted_rply.ar_verf)) { 542 if (nretries && 543 cl->cl_auth->ah_cred.oa_flavor 544 == RPCSEC_GSS) 545 /* 546 * If we retransmitted, its 547 * possible that we will 548 * receive a reply for one of 549 * the earlier transmissions 550 * (which will use an older 551 * RPCSEC_GSS sequence 552 * number). In this case, just 553 * go back and listen for a 554 * new reply. We could keep a 555 * record of all the seq 556 * numbers we have transmitted 557 * so far so that we could 558 * accept a reply for any of 559 * them here. 560 */ 561 goto get_reply; 562 cu->cu_error.re_status = RPC_AUTHERROR; 563 cu->cu_error.re_why = AUTH_INVALIDRESP; 564 } else { 565 if (cl->cl_auth->ah_cred.oa_flavor 566 == RPCSEC_GSS) { 567 if (!__rpc_gss_unwrap(cl->cl_auth, 568 &reply_xdrs, xresults, 569 resultsp)) 570 cu->cu_error.re_status = 571 RPC_CANTDECODERES; 572 } 573 } 574 if (reply_msg.acpted_rply.ar_verf.oa_base != NULL) { 575 xdrs->x_op = XDR_FREE; 576 (void) xdr_opaque_auth(xdrs, 577 &(reply_msg.acpted_rply.ar_verf)); 578 } 579 } /* end successful completion */ 580 /* 581 * If unsuccesful AND error is an authentication error 582 * then refresh credentials and try again, else break 583 */ 584 else if (cu->cu_error.re_status == RPC_AUTHERROR) 585 /* maybe our credentials need to be refreshed ... */ 586 if (nrefreshes > 0 && 587 AUTH_REFRESH(cl->cl_auth, &reply_msg)) { 588 nrefreshes--; 589 goto call_again; 590 } 591 /* end of unsuccessful completion */ 592 } /* end of valid reply message */ 593 else { 594 cu->cu_error.re_status = RPC_CANTDECODERES; 595 596 } 597 out: 598 if (cu->cu_kq >= 0) 599 _close(cu->cu_kq); 600 cu->cu_kq = -1; 601 release_fd_lock(cu->cu_fd, mask); 602 return (cu->cu_error.re_status); 603 } 604 605 static void 606 clnt_dg_geterr(cl, errp) 607 CLIENT *cl; 608 struct rpc_err *errp; 609 { 610 struct cu_data *cu = (struct cu_data *)cl->cl_private; 611 612 *errp = cu->cu_error; 613 } 614 615 static bool_t 616 clnt_dg_freeres(cl, xdr_res, res_ptr) 617 CLIENT *cl; 618 xdrproc_t xdr_res; 619 void *res_ptr; 620 { 621 struct cu_data *cu = (struct cu_data *)cl->cl_private; 622 XDR *xdrs = &(cu->cu_outxdrs); 623 bool_t dummy; 624 sigset_t mask; 625 sigset_t newmask; 626 627 sigfillset(&newmask); 628 thr_sigsetmask(SIG_SETMASK, &newmask, &mask); 629 mutex_lock(&clnt_fd_lock); 630 while (dg_fd_locks[cu->cu_fd]) 631 cond_wait(&dg_cv[cu->cu_fd], &clnt_fd_lock); 632 xdrs->x_op = XDR_FREE; 633 dummy = (*xdr_res)(xdrs, res_ptr); 634 mutex_unlock(&clnt_fd_lock); 635 thr_sigsetmask(SIG_SETMASK, &mask, NULL); 636 cond_signal(&dg_cv[cu->cu_fd]); 637 return (dummy); 638 } 639 640 /*ARGSUSED*/ 641 static void 642 clnt_dg_abort(h) 643 CLIENT *h; 644 { 645 } 646 647 static bool_t 648 clnt_dg_control(cl, request, info) 649 CLIENT *cl; 650 u_int request; 651 void *info; 652 { 653 struct cu_data *cu = (struct cu_data *)cl->cl_private; 654 struct netbuf *addr; 655 sigset_t mask; 656 sigset_t newmask; 657 int rpc_lock_value; 658 659 sigfillset(&newmask); 660 thr_sigsetmask(SIG_SETMASK, &newmask, &mask); 661 mutex_lock(&clnt_fd_lock); 662 while (dg_fd_locks[cu->cu_fd]) 663 cond_wait(&dg_cv[cu->cu_fd], &clnt_fd_lock); 664 if (__isthreaded) 665 rpc_lock_value = 1; 666 else 667 rpc_lock_value = 0; 668 dg_fd_locks[cu->cu_fd] = rpc_lock_value; 669 mutex_unlock(&clnt_fd_lock); 670 switch (request) { 671 case CLSET_FD_CLOSE: 672 cu->cu_closeit = TRUE; 673 release_fd_lock(cu->cu_fd, mask); 674 return (TRUE); 675 case CLSET_FD_NCLOSE: 676 cu->cu_closeit = FALSE; 677 release_fd_lock(cu->cu_fd, mask); 678 return (TRUE); 679 } 680 681 /* for other requests which use info */ 682 if (info == NULL) { 683 release_fd_lock(cu->cu_fd, mask); 684 return (FALSE); 685 } 686 switch (request) { 687 case CLSET_TIMEOUT: 688 if (time_not_ok((struct timeval *)info)) { 689 release_fd_lock(cu->cu_fd, mask); 690 return (FALSE); 691 } 692 cu->cu_total = *(struct timeval *)info; 693 break; 694 case CLGET_TIMEOUT: 695 *(struct timeval *)info = cu->cu_total; 696 break; 697 case CLGET_SERVER_ADDR: /* Give him the fd address */ 698 /* Now obsolete. Only for backward compatibility */ 699 (void) memcpy(info, &cu->cu_raddr, (size_t)cu->cu_rlen); 700 break; 701 case CLSET_RETRY_TIMEOUT: 702 if (time_not_ok((struct timeval *)info)) { 703 release_fd_lock(cu->cu_fd, mask); 704 return (FALSE); 705 } 706 cu->cu_wait = *(struct timeval *)info; 707 break; 708 case CLGET_RETRY_TIMEOUT: 709 *(struct timeval *)info = cu->cu_wait; 710 break; 711 case CLGET_FD: 712 *(int *)info = cu->cu_fd; 713 break; 714 case CLGET_SVC_ADDR: 715 addr = (struct netbuf *)info; 716 addr->buf = &cu->cu_raddr; 717 addr->len = cu->cu_rlen; 718 addr->maxlen = sizeof cu->cu_raddr; 719 break; 720 case CLSET_SVC_ADDR: /* set to new address */ 721 addr = (struct netbuf *)info; 722 if (addr->len < sizeof cu->cu_raddr) { 723 release_fd_lock(cu->cu_fd, mask); 724 return (FALSE); 725 } 726 (void) memcpy(&cu->cu_raddr, addr->buf, addr->len); 727 cu->cu_rlen = addr->len; 728 break; 729 case CLGET_XID: 730 /* 731 * use the knowledge that xid is the 732 * first element in the call structure *. 733 * This will get the xid of the PREVIOUS call 734 */ 735 *(u_int32_t *)info = 736 ntohl(*(u_int32_t *)(void *)cu->cu_outhdr); 737 break; 738 739 case CLSET_XID: 740 /* This will set the xid of the NEXT call */ 741 *(u_int32_t *)(void *)cu->cu_outhdr = 742 htonl(*(u_int32_t *)info - 1); 743 /* decrement by 1 as clnt_dg_call() increments once */ 744 break; 745 746 case CLGET_VERS: 747 /* 748 * This RELIES on the information that, in the call body, 749 * the version number field is the fifth field from the 750 * begining of the RPC header. MUST be changed if the 751 * call_struct is changed 752 */ 753 *(u_int32_t *)info = 754 ntohl(*(u_int32_t *)(void *)(cu->cu_outhdr + 755 4 * BYTES_PER_XDR_UNIT)); 756 break; 757 758 case CLSET_VERS: 759 *(u_int32_t *)(void *)(cu->cu_outhdr + 4 * BYTES_PER_XDR_UNIT) 760 = htonl(*(u_int32_t *)info); 761 break; 762 763 case CLGET_PROG: 764 /* 765 * This RELIES on the information that, in the call body, 766 * the program number field is the fourth field from the 767 * begining of the RPC header. MUST be changed if the 768 * call_struct is changed 769 */ 770 *(u_int32_t *)info = 771 ntohl(*(u_int32_t *)(void *)(cu->cu_outhdr + 772 3 * BYTES_PER_XDR_UNIT)); 773 break; 774 775 case CLSET_PROG: 776 *(u_int32_t *)(void *)(cu->cu_outhdr + 3 * BYTES_PER_XDR_UNIT) 777 = htonl(*(u_int32_t *)info); 778 break; 779 case CLSET_ASYNC: 780 cu->cu_async = *(int *)info; 781 break; 782 case CLSET_CONNECT: 783 cu->cu_connect = *(int *)info; 784 break; 785 default: 786 release_fd_lock(cu->cu_fd, mask); 787 return (FALSE); 788 } 789 release_fd_lock(cu->cu_fd, mask); 790 return (TRUE); 791 } 792 793 static void 794 clnt_dg_destroy(cl) 795 CLIENT *cl; 796 { 797 struct cu_data *cu = (struct cu_data *)cl->cl_private; 798 int cu_fd = cu->cu_fd; 799 sigset_t mask; 800 sigset_t newmask; 801 802 sigfillset(&newmask); 803 thr_sigsetmask(SIG_SETMASK, &newmask, &mask); 804 mutex_lock(&clnt_fd_lock); 805 while (dg_fd_locks[cu_fd]) 806 cond_wait(&dg_cv[cu_fd], &clnt_fd_lock); 807 if (cu->cu_closeit) 808 (void)_close(cu_fd); 809 if (cu->cu_kq >= 0) 810 _close(cu->cu_kq); 811 XDR_DESTROY(&(cu->cu_outxdrs)); 812 mem_free(cu, (sizeof (*cu) + cu->cu_sendsz + cu->cu_recvsz)); 813 if (cl->cl_netid && cl->cl_netid[0]) 814 mem_free(cl->cl_netid, strlen(cl->cl_netid) +1); 815 if (cl->cl_tp && cl->cl_tp[0]) 816 mem_free(cl->cl_tp, strlen(cl->cl_tp) +1); 817 mem_free(cl, sizeof (CLIENT)); 818 mutex_unlock(&clnt_fd_lock); 819 thr_sigsetmask(SIG_SETMASK, &mask, NULL); 820 cond_signal(&dg_cv[cu_fd]); 821 } 822 823 static struct clnt_ops * 824 clnt_dg_ops() 825 { 826 static struct clnt_ops ops; 827 sigset_t mask; 828 sigset_t newmask; 829 830 /* VARIABLES PROTECTED BY ops_lock: ops */ 831 832 sigfillset(&newmask); 833 thr_sigsetmask(SIG_SETMASK, &newmask, &mask); 834 mutex_lock(&ops_lock); 835 if (ops.cl_call == NULL) { 836 ops.cl_call = clnt_dg_call; 837 ops.cl_abort = clnt_dg_abort; 838 ops.cl_geterr = clnt_dg_geterr; 839 ops.cl_freeres = clnt_dg_freeres; 840 ops.cl_destroy = clnt_dg_destroy; 841 ops.cl_control = clnt_dg_control; 842 } 843 mutex_unlock(&ops_lock); 844 thr_sigsetmask(SIG_SETMASK, &mask, NULL); 845 return (&ops); 846 } 847 848 /* 849 * Make sure that the time is not garbage. -1 value is allowed. 850 */ 851 static bool_t 852 time_not_ok(t) 853 struct timeval *t; 854 { 855 return (t->tv_sec < -1 || t->tv_sec > 100000000 || 856 t->tv_usec < -1 || t->tv_usec > 1000000); 857 } 858 859