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