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