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