1 /* $NetBSD: clnt_vc.c,v 1.4 2000/07/14 08:40:42 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 #if defined(LIBC_SCCS) && !defined(lint) 32 static char *sccsid2 = "@(#)clnt_tcp.c 1.37 87/10/05 Copyr 1984 Sun Micro"; 33 static char *sccsid = "@(#)clnt_tcp.c 2.2 88/08/01 4.0 RPCSRC"; 34 static char sccsid3[] = "@(#)clnt_vc.c 1.19 89/03/16 Copyr 1988 Sun Micro"; 35 #endif 36 #include <sys/cdefs.h> 37 __FBSDID("$FreeBSD$"); 38 39 /* 40 * clnt_tcp.c, Implements a TCP/IP based, client side RPC. 41 * 42 * Copyright (C) 1984, Sun Microsystems, Inc. 43 * 44 * TCP based RPC supports 'batched calls'. 45 * A sequence of calls may be batched-up in a send buffer. The rpc call 46 * return immediately to the client even though the call was not necessarily 47 * sent. The batching occurs if the results' xdr routine is NULL (0) AND 48 * the rpc timeout value is zero (see clnt.h, rpc). 49 * 50 * Clients should NOT casually batch calls that in fact return results; that is, 51 * the server side should be aware that a call is batched and not produce any 52 * return message. Batched calls that produce many result messages can 53 * deadlock (netlock) the client and the server.... 54 * 55 * Now go hang yourself. 56 */ 57 58 #include "namespace.h" 59 #include "reentrant.h" 60 #include <sys/types.h> 61 #include <sys/poll.h> 62 #include <sys/syslog.h> 63 #include <sys/socket.h> 64 #include <sys/un.h> 65 #include <sys/uio.h> 66 67 #include <arpa/inet.h> 68 #include <assert.h> 69 #include <err.h> 70 #include <errno.h> 71 #include <netdb.h> 72 #include <stdio.h> 73 #include <stdlib.h> 74 #include <string.h> 75 #include <unistd.h> 76 #include <signal.h> 77 78 #include <rpc/rpc.h> 79 #include <rpc/rpcsec_gss.h> 80 #include "un-namespace.h" 81 #include "rpc_com.h" 82 #include "mt_misc.h" 83 84 #define MCALL_MSG_SIZE 24 85 86 struct cmessage { 87 struct cmsghdr cmsg; 88 struct cmsgcred cmcred; 89 }; 90 91 static enum clnt_stat clnt_vc_call(CLIENT *, rpcproc_t, xdrproc_t, void *, 92 xdrproc_t, void *, struct timeval); 93 static void clnt_vc_geterr(CLIENT *, struct rpc_err *); 94 static bool_t clnt_vc_freeres(CLIENT *, xdrproc_t, void *); 95 static void clnt_vc_abort(CLIENT *); 96 static bool_t clnt_vc_control(CLIENT *, u_int, void *); 97 static void clnt_vc_destroy(CLIENT *); 98 static struct clnt_ops *clnt_vc_ops(void); 99 static bool_t time_not_ok(struct timeval *); 100 static int read_vc(void *, void *, int); 101 static int write_vc(void *, void *, int); 102 static int __msgwrite(int, void *, size_t); 103 static int __msgread(int, void *, size_t); 104 105 struct ct_data { 106 int ct_fd; /* connection's fd */ 107 bool_t ct_closeit; /* close it on destroy */ 108 struct timeval ct_wait; /* wait interval in milliseconds */ 109 bool_t ct_waitset; /* wait set by clnt_control? */ 110 struct netbuf ct_addr; /* remote addr */ 111 struct rpc_err ct_error; 112 union { 113 char ct_mcallc[MCALL_MSG_SIZE]; /* marshalled callmsg */ 114 u_int32_t ct_mcalli; 115 } ct_u; 116 u_int ct_mpos; /* pos after marshal */ 117 XDR ct_xdrs; /* XDR stream */ 118 }; 119 120 /* 121 * This machinery implements per-fd locks for MT-safety. It is not 122 * sufficient to do per-CLIENT handle locks for MT-safety because a 123 * user may create more than one CLIENT handle with the same fd behind 124 * it. Therfore, we allocate an array of flags (vc_fd_locks), protected 125 * by the clnt_fd_lock mutex, and an array (vc_cv) of condition variables 126 * similarly protected. Vc_fd_lock[fd] == 1 => a call is activte on some 127 * CLIENT handle created for that fd. 128 * The current implementation holds locks across the entire RPC and reply. 129 * Yes, this is silly, and as soon as this code is proven to work, this 130 * should be the first thing fixed. One step at a time. 131 */ 132 static int *vc_fd_locks; 133 static cond_t *vc_cv; 134 #define release_fd_lock(fd, mask) { \ 135 mutex_lock(&clnt_fd_lock); \ 136 vc_fd_locks[fd] = 0; \ 137 mutex_unlock(&clnt_fd_lock); \ 138 thr_sigsetmask(SIG_SETMASK, &(mask), (sigset_t *) NULL); \ 139 cond_signal(&vc_cv[fd]); \ 140 } 141 142 static const char clnt_vc_errstr[] = "%s : %s"; 143 static const char clnt_vc_str[] = "clnt_vc_create"; 144 static const char clnt_read_vc_str[] = "read_vc"; 145 static const char __no_mem_str[] = "out of memory"; 146 147 /* 148 * Create a client handle for a connection. 149 * Default options are set, which the user can change using clnt_control()'s. 150 * The rpc/vc package does buffering similar to stdio, so the client 151 * must pick send and receive buffer sizes, 0 => use the default. 152 * NB: fd is copied into a private area. 153 * NB: The rpch->cl_auth is set null authentication. Caller may wish to 154 * set this something more useful. 155 * 156 * fd should be an open socket 157 */ 158 CLIENT * 159 clnt_vc_create(fd, raddr, prog, vers, sendsz, recvsz) 160 int fd; /* open file descriptor */ 161 const struct netbuf *raddr; /* servers address */ 162 const rpcprog_t prog; /* program number */ 163 const rpcvers_t vers; /* version number */ 164 u_int sendsz; /* buffer recv size */ 165 u_int recvsz; /* buffer send size */ 166 { 167 CLIENT *cl; /* client handle */ 168 struct ct_data *ct = NULL; /* client handle */ 169 struct timeval now; 170 struct rpc_msg call_msg; 171 static u_int32_t disrupt; 172 sigset_t mask; 173 sigset_t newmask; 174 struct sockaddr_storage ss; 175 socklen_t slen; 176 struct __rpc_sockinfo si; 177 178 if (disrupt == 0) 179 disrupt = (u_int32_t)(long)raddr; 180 181 cl = (CLIENT *)mem_alloc(sizeof (*cl)); 182 ct = (struct ct_data *)mem_alloc(sizeof (*ct)); 183 if ((cl == (CLIENT *)NULL) || (ct == (struct ct_data *)NULL)) { 184 (void) syslog(LOG_ERR, clnt_vc_errstr, 185 clnt_vc_str, __no_mem_str); 186 rpc_createerr.cf_stat = RPC_SYSTEMERROR; 187 rpc_createerr.cf_error.re_errno = errno; 188 goto err; 189 } 190 ct->ct_addr.buf = NULL; 191 sigfillset(&newmask); 192 thr_sigsetmask(SIG_SETMASK, &newmask, &mask); 193 mutex_lock(&clnt_fd_lock); 194 if (vc_fd_locks == (int *) NULL) { 195 int cv_allocsz, fd_allocsz; 196 int dtbsize = __rpc_dtbsize(); 197 198 fd_allocsz = dtbsize * sizeof (int); 199 vc_fd_locks = (int *) mem_alloc(fd_allocsz); 200 if (vc_fd_locks == (int *) NULL) { 201 mutex_unlock(&clnt_fd_lock); 202 thr_sigsetmask(SIG_SETMASK, &(mask), NULL); 203 goto err; 204 } else 205 memset(vc_fd_locks, '\0', fd_allocsz); 206 207 assert(vc_cv == (cond_t *) NULL); 208 cv_allocsz = dtbsize * sizeof (cond_t); 209 vc_cv = (cond_t *) mem_alloc(cv_allocsz); 210 if (vc_cv == (cond_t *) NULL) { 211 mem_free(vc_fd_locks, fd_allocsz); 212 vc_fd_locks = (int *) NULL; 213 mutex_unlock(&clnt_fd_lock); 214 thr_sigsetmask(SIG_SETMASK, &(mask), NULL); 215 goto err; 216 } else { 217 int i; 218 219 for (i = 0; i < dtbsize; i++) 220 cond_init(&vc_cv[i], 0, (void *) 0); 221 } 222 } else 223 assert(vc_cv != (cond_t *) NULL); 224 225 /* 226 * XXX - fvdl connecting while holding a mutex? 227 */ 228 slen = sizeof ss; 229 if (_getpeername(fd, (struct sockaddr *)(void *)&ss, &slen) < 0) { 230 if (errno != ENOTCONN) { 231 rpc_createerr.cf_stat = RPC_SYSTEMERROR; 232 rpc_createerr.cf_error.re_errno = errno; 233 mutex_unlock(&clnt_fd_lock); 234 thr_sigsetmask(SIG_SETMASK, &(mask), NULL); 235 goto err; 236 } 237 if (_connect(fd, (struct sockaddr *)raddr->buf, raddr->len) < 0){ 238 rpc_createerr.cf_stat = RPC_SYSTEMERROR; 239 rpc_createerr.cf_error.re_errno = errno; 240 mutex_unlock(&clnt_fd_lock); 241 thr_sigsetmask(SIG_SETMASK, &(mask), NULL); 242 goto err; 243 } 244 } 245 mutex_unlock(&clnt_fd_lock); 246 thr_sigsetmask(SIG_SETMASK, &(mask), NULL); 247 if (!__rpc_fd2sockinfo(fd, &si)) 248 goto err; 249 250 ct->ct_closeit = FALSE; 251 252 /* 253 * Set up private data struct 254 */ 255 ct->ct_fd = fd; 256 ct->ct_wait.tv_usec = 0; 257 ct->ct_waitset = FALSE; 258 ct->ct_addr.buf = malloc(raddr->maxlen); 259 if (ct->ct_addr.buf == NULL) 260 goto err; 261 memcpy(ct->ct_addr.buf, raddr->buf, raddr->len); 262 ct->ct_addr.len = raddr->len; 263 ct->ct_addr.maxlen = raddr->maxlen; 264 265 /* 266 * Initialize call message 267 */ 268 (void)gettimeofday(&now, NULL); 269 call_msg.rm_xid = ((u_int32_t)++disrupt) ^ __RPC_GETXID(&now); 270 call_msg.rm_direction = CALL; 271 call_msg.rm_call.cb_rpcvers = RPC_MSG_VERSION; 272 call_msg.rm_call.cb_prog = (u_int32_t)prog; 273 call_msg.rm_call.cb_vers = (u_int32_t)vers; 274 275 /* 276 * pre-serialize the static part of the call msg and stash it away 277 */ 278 xdrmem_create(&(ct->ct_xdrs), ct->ct_u.ct_mcallc, MCALL_MSG_SIZE, 279 XDR_ENCODE); 280 if (! xdr_callhdr(&(ct->ct_xdrs), &call_msg)) { 281 if (ct->ct_closeit) { 282 (void)_close(fd); 283 } 284 goto err; 285 } 286 ct->ct_mpos = XDR_GETPOS(&(ct->ct_xdrs)); 287 XDR_DESTROY(&(ct->ct_xdrs)); 288 assert(ct->ct_mpos + sizeof(uint32_t) <= MCALL_MSG_SIZE); 289 290 /* 291 * Create a client handle which uses xdrrec for serialization 292 * and authnone for authentication. 293 */ 294 cl->cl_ops = clnt_vc_ops(); 295 cl->cl_private = ct; 296 cl->cl_auth = authnone_create(); 297 sendsz = __rpc_get_t_size(si.si_af, si.si_proto, (int)sendsz); 298 recvsz = __rpc_get_t_size(si.si_af, si.si_proto, (int)recvsz); 299 xdrrec_create(&(ct->ct_xdrs), sendsz, recvsz, 300 cl->cl_private, read_vc, write_vc); 301 return (cl); 302 303 err: 304 if (cl) { 305 if (ct) { 306 if (ct->ct_addr.len) 307 mem_free(ct->ct_addr.buf, ct->ct_addr.len); 308 mem_free(ct, sizeof (struct ct_data)); 309 } 310 if (cl) 311 mem_free(cl, sizeof (CLIENT)); 312 } 313 return ((CLIENT *)NULL); 314 } 315 316 static enum clnt_stat 317 clnt_vc_call(cl, proc, xdr_args, args_ptr, xdr_results, results_ptr, timeout) 318 CLIENT *cl; 319 rpcproc_t proc; 320 xdrproc_t xdr_args; 321 void *args_ptr; 322 xdrproc_t xdr_results; 323 void *results_ptr; 324 struct timeval timeout; 325 { 326 struct ct_data *ct = (struct ct_data *) cl->cl_private; 327 XDR *xdrs = &(ct->ct_xdrs); 328 struct rpc_msg reply_msg; 329 u_int32_t x_id; 330 u_int32_t *msg_x_id = &ct->ct_u.ct_mcalli; /* yuk */ 331 bool_t shipnow; 332 int refreshes = 2; 333 sigset_t mask, newmask; 334 int rpc_lock_value; 335 bool_t reply_stat; 336 337 assert(cl != NULL); 338 339 sigfillset(&newmask); 340 thr_sigsetmask(SIG_SETMASK, &newmask, &mask); 341 mutex_lock(&clnt_fd_lock); 342 while (vc_fd_locks[ct->ct_fd]) 343 cond_wait(&vc_cv[ct->ct_fd], &clnt_fd_lock); 344 if (__isthreaded) 345 rpc_lock_value = 1; 346 else 347 rpc_lock_value = 0; 348 vc_fd_locks[ct->ct_fd] = rpc_lock_value; 349 mutex_unlock(&clnt_fd_lock); 350 if (!ct->ct_waitset) { 351 /* If time is not within limits, we ignore it. */ 352 if (time_not_ok(&timeout) == FALSE) 353 ct->ct_wait = timeout; 354 } 355 356 shipnow = 357 (xdr_results == NULL && timeout.tv_sec == 0 358 && timeout.tv_usec == 0) ? FALSE : TRUE; 359 360 call_again: 361 xdrs->x_op = XDR_ENCODE; 362 ct->ct_error.re_status = RPC_SUCCESS; 363 x_id = ntohl(--(*msg_x_id)); 364 365 if (cl->cl_auth->ah_cred.oa_flavor != RPCSEC_GSS) { 366 if ((! XDR_PUTBYTES(xdrs, ct->ct_u.ct_mcallc, ct->ct_mpos)) || 367 (! XDR_PUTINT32(xdrs, &proc)) || 368 (! AUTH_MARSHALL(cl->cl_auth, xdrs)) || 369 (! (*xdr_args)(xdrs, args_ptr))) { 370 if (ct->ct_error.re_status == RPC_SUCCESS) 371 ct->ct_error.re_status = RPC_CANTENCODEARGS; 372 (void)xdrrec_endofrecord(xdrs, TRUE); 373 release_fd_lock(ct->ct_fd, mask); 374 return (ct->ct_error.re_status); 375 } 376 } else { 377 *(uint32_t *) &ct->ct_u.ct_mcallc[ct->ct_mpos] = htonl(proc); 378 if (! __rpc_gss_wrap(cl->cl_auth, ct->ct_u.ct_mcallc, 379 ct->ct_mpos + sizeof(uint32_t), 380 xdrs, xdr_args, args_ptr)) { 381 if (ct->ct_error.re_status == RPC_SUCCESS) 382 ct->ct_error.re_status = RPC_CANTENCODEARGS; 383 (void)xdrrec_endofrecord(xdrs, TRUE); 384 release_fd_lock(ct->ct_fd, mask); 385 return (ct->ct_error.re_status); 386 } 387 } 388 if (! xdrrec_endofrecord(xdrs, shipnow)) { 389 release_fd_lock(ct->ct_fd, mask); 390 return (ct->ct_error.re_status = RPC_CANTSEND); 391 } 392 if (! shipnow) { 393 release_fd_lock(ct->ct_fd, mask); 394 return (RPC_SUCCESS); 395 } 396 /* 397 * Hack to provide rpc-based message passing 398 */ 399 if (timeout.tv_sec == 0 && timeout.tv_usec == 0) { 400 release_fd_lock(ct->ct_fd, mask); 401 return(ct->ct_error.re_status = RPC_TIMEDOUT); 402 } 403 404 405 /* 406 * Keep receiving until we get a valid transaction id 407 */ 408 xdrs->x_op = XDR_DECODE; 409 while (TRUE) { 410 reply_msg.acpted_rply.ar_verf = _null_auth; 411 reply_msg.acpted_rply.ar_results.where = NULL; 412 reply_msg.acpted_rply.ar_results.proc = (xdrproc_t)xdr_void; 413 if (! xdrrec_skiprecord(xdrs)) { 414 release_fd_lock(ct->ct_fd, mask); 415 return (ct->ct_error.re_status); 416 } 417 /* now decode and validate the response header */ 418 if (! xdr_replymsg(xdrs, &reply_msg)) { 419 if (ct->ct_error.re_status == RPC_SUCCESS) 420 continue; 421 release_fd_lock(ct->ct_fd, mask); 422 return (ct->ct_error.re_status); 423 } 424 if (reply_msg.rm_xid == x_id) 425 break; 426 } 427 428 /* 429 * process header 430 */ 431 _seterr_reply(&reply_msg, &(ct->ct_error)); 432 if (ct->ct_error.re_status == RPC_SUCCESS) { 433 if (! AUTH_VALIDATE(cl->cl_auth, 434 &reply_msg.acpted_rply.ar_verf)) { 435 ct->ct_error.re_status = RPC_AUTHERROR; 436 ct->ct_error.re_why = AUTH_INVALIDRESP; 437 } else { 438 if (cl->cl_auth->ah_cred.oa_flavor != RPCSEC_GSS) { 439 reply_stat = (*xdr_results)(xdrs, results_ptr); 440 } else { 441 reply_stat = __rpc_gss_unwrap(cl->cl_auth, 442 xdrs, xdr_results, results_ptr); 443 } 444 if (! reply_stat) { 445 if (ct->ct_error.re_status == RPC_SUCCESS) 446 ct->ct_error.re_status = 447 RPC_CANTDECODERES; 448 } 449 } 450 /* free verifier ... */ 451 if (reply_msg.acpted_rply.ar_verf.oa_base != NULL) { 452 xdrs->x_op = XDR_FREE; 453 (void)xdr_opaque_auth(xdrs, 454 &(reply_msg.acpted_rply.ar_verf)); 455 } 456 } /* end successful completion */ 457 else { 458 /* maybe our credentials need to be refreshed ... */ 459 if (refreshes-- && AUTH_REFRESH(cl->cl_auth, &reply_msg)) 460 goto call_again; 461 } /* end of unsuccessful completion */ 462 release_fd_lock(ct->ct_fd, mask); 463 return (ct->ct_error.re_status); 464 } 465 466 static void 467 clnt_vc_geterr(cl, errp) 468 CLIENT *cl; 469 struct rpc_err *errp; 470 { 471 struct ct_data *ct; 472 473 assert(cl != NULL); 474 assert(errp != NULL); 475 476 ct = (struct ct_data *) cl->cl_private; 477 *errp = ct->ct_error; 478 } 479 480 static bool_t 481 clnt_vc_freeres(cl, xdr_res, res_ptr) 482 CLIENT *cl; 483 xdrproc_t xdr_res; 484 void *res_ptr; 485 { 486 struct ct_data *ct; 487 XDR *xdrs; 488 bool_t dummy; 489 sigset_t mask; 490 sigset_t newmask; 491 492 assert(cl != NULL); 493 494 ct = (struct ct_data *)cl->cl_private; 495 xdrs = &(ct->ct_xdrs); 496 497 sigfillset(&newmask); 498 thr_sigsetmask(SIG_SETMASK, &newmask, &mask); 499 mutex_lock(&clnt_fd_lock); 500 while (vc_fd_locks[ct->ct_fd]) 501 cond_wait(&vc_cv[ct->ct_fd], &clnt_fd_lock); 502 xdrs->x_op = XDR_FREE; 503 dummy = (*xdr_res)(xdrs, res_ptr); 504 mutex_unlock(&clnt_fd_lock); 505 thr_sigsetmask(SIG_SETMASK, &(mask), NULL); 506 cond_signal(&vc_cv[ct->ct_fd]); 507 508 return dummy; 509 } 510 511 /*ARGSUSED*/ 512 static void 513 clnt_vc_abort(cl) 514 CLIENT *cl; 515 { 516 } 517 518 static bool_t 519 clnt_vc_control(cl, request, info) 520 CLIENT *cl; 521 u_int request; 522 void *info; 523 { 524 struct ct_data *ct; 525 void *infop = info; 526 sigset_t mask; 527 sigset_t newmask; 528 int rpc_lock_value; 529 530 assert(cl != NULL); 531 532 ct = (struct ct_data *)cl->cl_private; 533 534 sigfillset(&newmask); 535 thr_sigsetmask(SIG_SETMASK, &newmask, &mask); 536 mutex_lock(&clnt_fd_lock); 537 while (vc_fd_locks[ct->ct_fd]) 538 cond_wait(&vc_cv[ct->ct_fd], &clnt_fd_lock); 539 if (__isthreaded) 540 rpc_lock_value = 1; 541 else 542 rpc_lock_value = 0; 543 vc_fd_locks[ct->ct_fd] = rpc_lock_value; 544 mutex_unlock(&clnt_fd_lock); 545 546 switch (request) { 547 case CLSET_FD_CLOSE: 548 ct->ct_closeit = TRUE; 549 release_fd_lock(ct->ct_fd, mask); 550 return (TRUE); 551 case CLSET_FD_NCLOSE: 552 ct->ct_closeit = FALSE; 553 release_fd_lock(ct->ct_fd, mask); 554 return (TRUE); 555 default: 556 break; 557 } 558 559 /* for other requests which use info */ 560 if (info == NULL) { 561 release_fd_lock(ct->ct_fd, mask); 562 return (FALSE); 563 } 564 switch (request) { 565 case CLSET_TIMEOUT: 566 if (time_not_ok((struct timeval *)info)) { 567 release_fd_lock(ct->ct_fd, mask); 568 return (FALSE); 569 } 570 ct->ct_wait = *(struct timeval *)infop; 571 ct->ct_waitset = TRUE; 572 break; 573 case CLGET_TIMEOUT: 574 *(struct timeval *)infop = ct->ct_wait; 575 break; 576 case CLGET_SERVER_ADDR: 577 (void) memcpy(info, ct->ct_addr.buf, (size_t)ct->ct_addr.len); 578 break; 579 case CLGET_FD: 580 *(int *)info = ct->ct_fd; 581 break; 582 case CLGET_SVC_ADDR: 583 /* The caller should not free this memory area */ 584 *(struct netbuf *)info = ct->ct_addr; 585 break; 586 case CLSET_SVC_ADDR: /* set to new address */ 587 release_fd_lock(ct->ct_fd, mask); 588 return (FALSE); 589 case CLGET_XID: 590 /* 591 * use the knowledge that xid is the 592 * first element in the call structure 593 * This will get the xid of the PREVIOUS call 594 */ 595 *(u_int32_t *)info = 596 ntohl(*(u_int32_t *)(void *)&ct->ct_u.ct_mcalli); 597 break; 598 case CLSET_XID: 599 /* This will set the xid of the NEXT call */ 600 *(u_int32_t *)(void *)&ct->ct_u.ct_mcalli = 601 htonl(*((u_int32_t *)info) + 1); 602 /* increment by 1 as clnt_vc_call() decrements once */ 603 break; 604 case CLGET_VERS: 605 /* 606 * This RELIES on the information that, in the call body, 607 * the version number field is the fifth field from the 608 * begining of the RPC header. MUST be changed if the 609 * call_struct is changed 610 */ 611 *(u_int32_t *)info = 612 ntohl(*(u_int32_t *)(void *)(ct->ct_u.ct_mcallc + 613 4 * BYTES_PER_XDR_UNIT)); 614 break; 615 616 case CLSET_VERS: 617 *(u_int32_t *)(void *)(ct->ct_u.ct_mcallc + 618 4 * BYTES_PER_XDR_UNIT) = 619 htonl(*(u_int32_t *)info); 620 break; 621 622 case CLGET_PROG: 623 /* 624 * This RELIES on the information that, in the call body, 625 * the program number field is the fourth field from the 626 * begining of the RPC header. MUST be changed if the 627 * call_struct is changed 628 */ 629 *(u_int32_t *)info = 630 ntohl(*(u_int32_t *)(void *)(ct->ct_u.ct_mcallc + 631 3 * BYTES_PER_XDR_UNIT)); 632 break; 633 634 case CLSET_PROG: 635 *(u_int32_t *)(void *)(ct->ct_u.ct_mcallc + 636 3 * BYTES_PER_XDR_UNIT) = 637 htonl(*(u_int32_t *)info); 638 break; 639 640 default: 641 release_fd_lock(ct->ct_fd, mask); 642 return (FALSE); 643 } 644 release_fd_lock(ct->ct_fd, mask); 645 return (TRUE); 646 } 647 648 649 static void 650 clnt_vc_destroy(cl) 651 CLIENT *cl; 652 { 653 struct ct_data *ct = (struct ct_data *) cl->cl_private; 654 int ct_fd = ct->ct_fd; 655 sigset_t mask; 656 sigset_t newmask; 657 658 assert(cl != NULL); 659 660 ct = (struct ct_data *) cl->cl_private; 661 662 sigfillset(&newmask); 663 thr_sigsetmask(SIG_SETMASK, &newmask, &mask); 664 mutex_lock(&clnt_fd_lock); 665 while (vc_fd_locks[ct_fd]) 666 cond_wait(&vc_cv[ct_fd], &clnt_fd_lock); 667 if (ct->ct_closeit && ct->ct_fd != -1) { 668 (void)_close(ct->ct_fd); 669 } 670 XDR_DESTROY(&(ct->ct_xdrs)); 671 if (ct->ct_addr.buf) 672 free(ct->ct_addr.buf); 673 mem_free(ct, sizeof(struct ct_data)); 674 if (cl->cl_netid && cl->cl_netid[0]) 675 mem_free(cl->cl_netid, strlen(cl->cl_netid) +1); 676 if (cl->cl_tp && cl->cl_tp[0]) 677 mem_free(cl->cl_tp, strlen(cl->cl_tp) +1); 678 mem_free(cl, sizeof(CLIENT)); 679 mutex_unlock(&clnt_fd_lock); 680 thr_sigsetmask(SIG_SETMASK, &(mask), NULL); 681 cond_signal(&vc_cv[ct_fd]); 682 } 683 684 /* 685 * Interface between xdr serializer and tcp connection. 686 * Behaves like the system calls, read & write, but keeps some error state 687 * around for the rpc level. 688 */ 689 static int 690 read_vc(ctp, buf, len) 691 void *ctp; 692 void *buf; 693 int len; 694 { 695 struct sockaddr sa; 696 socklen_t sal; 697 struct ct_data *ct = (struct ct_data *)ctp; 698 struct pollfd fd; 699 int milliseconds = (int)((ct->ct_wait.tv_sec * 1000) + 700 (ct->ct_wait.tv_usec / 1000)); 701 702 if (len == 0) 703 return (0); 704 fd.fd = ct->ct_fd; 705 fd.events = POLLIN; 706 for (;;) { 707 switch (_poll(&fd, 1, milliseconds)) { 708 case 0: 709 ct->ct_error.re_status = RPC_TIMEDOUT; 710 return (-1); 711 712 case -1: 713 if (errno == EINTR) 714 continue; 715 ct->ct_error.re_status = RPC_CANTRECV; 716 ct->ct_error.re_errno = errno; 717 return (-1); 718 } 719 break; 720 } 721 722 sal = sizeof(sa); 723 if ((_getpeername(ct->ct_fd, &sa, &sal) == 0) && 724 (sa.sa_family == AF_LOCAL)) { 725 len = __msgread(ct->ct_fd, buf, (size_t)len); 726 } else { 727 len = _read(ct->ct_fd, buf, (size_t)len); 728 } 729 730 switch (len) { 731 case 0: 732 /* premature eof */ 733 ct->ct_error.re_errno = ECONNRESET; 734 ct->ct_error.re_status = RPC_CANTRECV; 735 len = -1; /* it's really an error */ 736 break; 737 738 case -1: 739 ct->ct_error.re_errno = errno; 740 ct->ct_error.re_status = RPC_CANTRECV; 741 break; 742 } 743 return (len); 744 } 745 746 static int 747 write_vc(ctp, buf, len) 748 void *ctp; 749 void *buf; 750 int len; 751 { 752 struct sockaddr sa; 753 socklen_t sal; 754 struct ct_data *ct = (struct ct_data *)ctp; 755 int i, cnt; 756 757 sal = sizeof(sa); 758 if ((_getpeername(ct->ct_fd, &sa, &sal) == 0) && 759 (sa.sa_family == AF_LOCAL)) { 760 for (cnt = len; cnt > 0; cnt -= i, buf = (char *)buf + i) { 761 if ((i = __msgwrite(ct->ct_fd, buf, 762 (size_t)cnt)) == -1) { 763 ct->ct_error.re_errno = errno; 764 ct->ct_error.re_status = RPC_CANTSEND; 765 return (-1); 766 } 767 } 768 } else { 769 for (cnt = len; cnt > 0; cnt -= i, buf = (char *)buf + i) { 770 if ((i = _write(ct->ct_fd, buf, (size_t)cnt)) == -1) { 771 ct->ct_error.re_errno = errno; 772 ct->ct_error.re_status = RPC_CANTSEND; 773 return (-1); 774 } 775 } 776 } 777 return (len); 778 } 779 780 static struct clnt_ops * 781 clnt_vc_ops() 782 { 783 static struct clnt_ops ops; 784 sigset_t mask, newmask; 785 786 /* VARIABLES PROTECTED BY ops_lock: ops */ 787 788 sigfillset(&newmask); 789 thr_sigsetmask(SIG_SETMASK, &newmask, &mask); 790 mutex_lock(&ops_lock); 791 if (ops.cl_call == NULL) { 792 ops.cl_call = clnt_vc_call; 793 ops.cl_abort = clnt_vc_abort; 794 ops.cl_geterr = clnt_vc_geterr; 795 ops.cl_freeres = clnt_vc_freeres; 796 ops.cl_destroy = clnt_vc_destroy; 797 ops.cl_control = clnt_vc_control; 798 } 799 mutex_unlock(&ops_lock); 800 thr_sigsetmask(SIG_SETMASK, &(mask), NULL); 801 return (&ops); 802 } 803 804 /* 805 * Make sure that the time is not garbage. -1 value is disallowed. 806 * Note this is different from time_not_ok in clnt_dg.c 807 */ 808 static bool_t 809 time_not_ok(t) 810 struct timeval *t; 811 { 812 return (t->tv_sec <= -1 || t->tv_sec > 100000000 || 813 t->tv_usec <= -1 || t->tv_usec > 1000000); 814 } 815 816 static int 817 __msgread(sock, buf, cnt) 818 int sock; 819 void *buf; 820 size_t cnt; 821 { 822 struct iovec iov[1]; 823 struct msghdr msg; 824 union { 825 struct cmsghdr cmsg; 826 char control[CMSG_SPACE(sizeof(struct cmsgcred))]; 827 } cm; 828 829 bzero((char *)&cm, sizeof(cm)); 830 iov[0].iov_base = buf; 831 iov[0].iov_len = cnt; 832 833 msg.msg_iov = iov; 834 msg.msg_iovlen = 1; 835 msg.msg_name = NULL; 836 msg.msg_namelen = 0; 837 msg.msg_control = (caddr_t)&cm; 838 msg.msg_controllen = CMSG_SPACE(sizeof(struct cmsgcred)); 839 msg.msg_flags = 0; 840 841 return(_recvmsg(sock, &msg, 0)); 842 } 843 844 static int 845 __msgwrite(sock, buf, cnt) 846 int sock; 847 void *buf; 848 size_t cnt; 849 { 850 struct iovec iov[1]; 851 struct msghdr msg; 852 union { 853 struct cmsghdr cmsg; 854 char control[CMSG_SPACE(sizeof(struct cmsgcred))]; 855 } cm; 856 857 bzero((char *)&cm, sizeof(cm)); 858 iov[0].iov_base = buf; 859 iov[0].iov_len = cnt; 860 861 cm.cmsg.cmsg_type = SCM_CREDS; 862 cm.cmsg.cmsg_level = SOL_SOCKET; 863 cm.cmsg.cmsg_len = CMSG_LEN(sizeof(struct cmsgcred)); 864 865 msg.msg_iov = iov; 866 msg.msg_iovlen = 1; 867 msg.msg_name = NULL; 868 msg.msg_namelen = 0; 869 msg.msg_control = (caddr_t)&cm; 870 msg.msg_controllen = CMSG_SPACE(sizeof(struct cmsgcred)); 871 msg.msg_flags = 0; 872 873 return(_sendmsg(sock, &msg, 0)); 874 } 875