Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
freebsd
GitHub Repository: freebsd/freebsd-src
Path: blob/main/lib/libc/rpc/clnt_dg.c
39476 views
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
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 *
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
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 *
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
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
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
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
635
clnt_dg_abort(CLIENT *h)
636
{
637
}
638
639
static bool_t
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
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 *
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
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
842