clnt_dg.c revision 1.6 1 1.6 christos /* $NetBSD: clnt_dg.c,v 1.6 2000/12/20 23:08:51 christos Exp $ */
2 1.1 fvdl
3 1.1 fvdl /*
4 1.1 fvdl * Sun RPC is a product of Sun Microsystems, Inc. and is provided for
5 1.1 fvdl * unrestricted use provided that this legend is included on all tape
6 1.1 fvdl * media and as a part of the software program in whole or part. Users
7 1.1 fvdl * may copy or modify Sun RPC without charge, but are not authorized
8 1.1 fvdl * to license or distribute it to anyone else except as part of a product or
9 1.1 fvdl * program developed by the user.
10 1.1 fvdl *
11 1.1 fvdl * SUN RPC IS PROVIDED AS IS WITH NO WARRANTIES OF ANY KIND INCLUDING THE
12 1.1 fvdl * WARRANTIES OF DESIGN, MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
13 1.1 fvdl * PURPOSE, OR ARISING FROM A COURSE OF DEALING, USAGE OR TRADE PRACTICE.
14 1.1 fvdl *
15 1.1 fvdl * Sun RPC is provided with no support and without any obligation on the
16 1.1 fvdl * part of Sun Microsystems, Inc. to assist in its use, correction,
17 1.1 fvdl * modification or enhancement.
18 1.1 fvdl *
19 1.1 fvdl * SUN MICROSYSTEMS, INC. SHALL HAVE NO LIABILITY WITH RESPECT TO THE
20 1.1 fvdl * INFRINGEMENT OF COPYRIGHTS, TRADE SECRETS OR ANY PATENTS BY SUN RPC
21 1.1 fvdl * OR ANY PART THEREOF.
22 1.1 fvdl *
23 1.1 fvdl * In no event will Sun Microsystems, Inc. be liable for any lost revenue
24 1.1 fvdl * or profits or other special, indirect and consequential damages, even if
25 1.1 fvdl * Sun has been advised of the possibility of such damages.
26 1.1 fvdl *
27 1.1 fvdl * Sun Microsystems, Inc.
28 1.1 fvdl * 2550 Garcia Avenue
29 1.1 fvdl * Mountain View, California 94043
30 1.1 fvdl */
31 1.1 fvdl /*
32 1.1 fvdl * Copyright (c) 1986-1991 by Sun Microsystems Inc.
33 1.1 fvdl */
34 1.1 fvdl
35 1.1 fvdl /* #ident "@(#)clnt_dg.c 1.23 94/04/22 SMI" */
36 1.1 fvdl
37 1.1 fvdl #if 0
38 1.1 fvdl #if !defined(lint) && defined(SCCSIDS)
39 1.1 fvdl static char sccsid[] = "@(#)clnt_dg.c 1.19 89/03/16 Copyr 1988 Sun Micro";
40 1.1 fvdl #endif
41 1.1 fvdl #endif
42 1.1 fvdl
43 1.1 fvdl /*
44 1.1 fvdl * Implements a connectionless client side RPC.
45 1.1 fvdl */
46 1.1 fvdl
47 1.1 fvdl #include "namespace.h"
48 1.1 fvdl #include "reentrant.h"
49 1.1 fvdl #include <sys/poll.h>
50 1.1 fvdl #include <sys/types.h>
51 1.1 fvdl #include <sys/time.h>
52 1.1 fvdl #include <sys/socket.h>
53 1.1 fvdl #include <sys/ioctl.h>
54 1.1 fvdl #include <rpc/rpc.h>
55 1.1 fvdl #include <errno.h>
56 1.1 fvdl #include <stdlib.h>
57 1.2 thorpej #include <string.h>
58 1.1 fvdl #include <signal.h>
59 1.1 fvdl #include <unistd.h>
60 1.1 fvdl #include <err.h>
61 1.1 fvdl #include "rpc_com.h"
62 1.1 fvdl
63 1.1 fvdl #ifdef __weak_alias
64 1.1 fvdl __weak_alias(clnt_dg_create,_clnt_dg_create)
65 1.1 fvdl #endif
66 1.1 fvdl
67 1.1 fvdl #define RPC_MAX_BACKOFF 30 /* seconds */
68 1.1 fvdl
69 1.1 fvdl
70 1.1 fvdl static struct clnt_ops *clnt_dg_ops __P((void));
71 1.1 fvdl static bool_t time_not_ok __P((struct timeval *));
72 1.1 fvdl static enum clnt_stat clnt_dg_call __P((CLIENT *, rpcproc_t, xdrproc_t, caddr_t,
73 1.1 fvdl xdrproc_t, caddr_t, struct timeval));
74 1.1 fvdl static void clnt_dg_geterr __P((CLIENT *, struct rpc_err *));
75 1.1 fvdl static bool_t clnt_dg_freeres __P((CLIENT *, xdrproc_t, caddr_t));
76 1.1 fvdl static void clnt_dg_abort __P((CLIENT *));
77 1.1 fvdl static bool_t clnt_dg_control __P((CLIENT *, u_int, char *));
78 1.1 fvdl static void clnt_dg_destroy __P((CLIENT *));
79 1.1 fvdl static int __rpc_timeval_to_msec __P((struct timeval *));
80 1.1 fvdl
81 1.1 fvdl
82 1.1 fvdl
83 1.1 fvdl
84 1.1 fvdl /*
85 1.1 fvdl * This machinery implements per-fd locks for MT-safety. It is not
86 1.1 fvdl * sufficient to do per-CLIENT handle locks for MT-safety because a
87 1.1 fvdl * user may create more than one CLIENT handle with the same fd behind
88 1.1 fvdl * it. Therfore, we allocate an array of flags (dg_fd_locks), protected
89 1.1 fvdl * by the clnt_fd_lock mutex, and an array (dg_cv) of condition variables
90 1.1 fvdl * similarly protected. Dg_fd_lock[fd] == 1 => a call is activte on some
91 1.1 fvdl * CLIENT handle created for that fd.
92 1.1 fvdl * The current implementation holds locks across the entire RPC and reply,
93 1.1 fvdl * including retransmissions. Yes, this is silly, and as soon as this
94 1.1 fvdl * code is proven to work, this should be the first thing fixed. One step
95 1.1 fvdl * at a time.
96 1.1 fvdl */
97 1.1 fvdl static int *dg_fd_locks;
98 1.1 fvdl #ifdef __REENT
99 1.1 fvdl extern int __rpc_lock_value;
100 1.1 fvdl extern mutex_t clnt_fd_lock;
101 1.1 fvdl static cond_t *dg_cv;
102 1.1 fvdl #define release_fd_lock(fd, mask) { \
103 1.1 fvdl mutex_lock(&clnt_fd_lock); \
104 1.1 fvdl dg_fd_locks[fd] = 0; \
105 1.1 fvdl mutex_unlock(&clnt_fd_lock); \
106 1.1 fvdl thr_sigsetmask(SIG_SETMASK, &(mask), (sigset_t *) NULL); \
107 1.1 fvdl cond_signal(&dg_cv[fd]); \
108 1.1 fvdl }
109 1.1 fvdl #else
110 1.1 fvdl #define release_fd_lock(fd,mask)
111 1.1 fvdl #define __rpc_lock_value 0
112 1.1 fvdl #endif
113 1.1 fvdl
114 1.1 fvdl static const char mem_err_clnt_dg[] = "clnt_dg_create: out of memory";
115 1.1 fvdl
116 1.1 fvdl /* VARIABLES PROTECTED BY clnt_fd_lock: dg_fd_locks, dg_cv */
117 1.1 fvdl
118 1.1 fvdl /*
119 1.1 fvdl * Private data kept per client handle
120 1.1 fvdl */
121 1.1 fvdl struct cu_data {
122 1.1 fvdl int cu_fd; /* connections fd */
123 1.1 fvdl bool_t cu_closeit; /* opened by library */
124 1.1 fvdl struct sockaddr_storage cu_raddr; /* remote address */
125 1.1 fvdl int cu_rlen;
126 1.1 fvdl struct timeval cu_wait; /* retransmit interval */
127 1.1 fvdl struct timeval cu_total; /* total time for the call */
128 1.1 fvdl struct rpc_err cu_error;
129 1.1 fvdl XDR cu_outxdrs;
130 1.1 fvdl u_int cu_xdrpos;
131 1.1 fvdl u_int cu_sendsz; /* send size */
132 1.1 fvdl char *cu_outbuf;
133 1.1 fvdl u_int cu_recvsz; /* recv size */
134 1.1 fvdl struct pollfd pfdp;
135 1.1 fvdl char cu_inbuf[1];
136 1.1 fvdl };
137 1.1 fvdl
138 1.1 fvdl /*
139 1.1 fvdl * Connection less client creation returns with client handle parameters.
140 1.1 fvdl * Default options are set, which the user can change using clnt_control().
141 1.1 fvdl * fd should be open and bound.
142 1.1 fvdl * NB: The rpch->cl_auth is initialized to null authentication.
143 1.1 fvdl * Caller may wish to set this something more useful.
144 1.1 fvdl *
145 1.1 fvdl * sendsz and recvsz are the maximum allowable packet sizes that can be
146 1.1 fvdl * sent and received. Normally they are the same, but they can be
147 1.1 fvdl * changed to improve the program efficiency and buffer allocation.
148 1.1 fvdl * If they are 0, use the transport default.
149 1.1 fvdl *
150 1.1 fvdl * If svcaddr is NULL, returns NULL.
151 1.1 fvdl */
152 1.1 fvdl CLIENT *
153 1.1 fvdl clnt_dg_create(fd, svcaddr, program, version, sendsz, recvsz)
154 1.1 fvdl int fd; /* open file descriptor */
155 1.3 christos const struct netbuf *svcaddr; /* servers address */
156 1.3 christos rpcprog_t program; /* program number */
157 1.3 christos rpcvers_t version; /* version number */
158 1.1 fvdl u_int sendsz; /* buffer recv size */
159 1.1 fvdl u_int recvsz; /* buffer send size */
160 1.1 fvdl {
161 1.3 christos CLIENT *cl = NULL; /* client handle */
162 1.3 christos struct cu_data *cu = NULL; /* private data */
163 1.1 fvdl struct timeval now;
164 1.1 fvdl struct rpc_msg call_msg;
165 1.1 fvdl #ifdef __REENT
166 1.1 fvdl sigset_t mask;
167 1.1 fvdl #endif
168 1.1 fvdl sigset_t newmask;
169 1.1 fvdl struct __rpc_sockinfo si;
170 1.1 fvdl int one = 1;
171 1.1 fvdl
172 1.1 fvdl sigfillset(&newmask);
173 1.1 fvdl thr_sigsetmask(SIG_SETMASK, &newmask, &mask);
174 1.1 fvdl mutex_lock(&clnt_fd_lock);
175 1.1 fvdl if (dg_fd_locks == (int *) NULL) {
176 1.1 fvdl #ifdef __REENT
177 1.1 fvdl int cv_allocsz;
178 1.1 fvdl #endif
179 1.3 christos size_t fd_allocsz;
180 1.1 fvdl int dtbsize = __rpc_dtbsize();
181 1.1 fvdl
182 1.1 fvdl fd_allocsz = dtbsize * sizeof (int);
183 1.1 fvdl dg_fd_locks = (int *) mem_alloc(fd_allocsz);
184 1.1 fvdl if (dg_fd_locks == (int *) NULL) {
185 1.1 fvdl mutex_unlock(&clnt_fd_lock);
186 1.1 fvdl thr_sigsetmask(SIG_SETMASK, &(mask), NULL);
187 1.1 fvdl goto err1;
188 1.1 fvdl } else
189 1.1 fvdl memset(dg_fd_locks, '\0', fd_allocsz);
190 1.1 fvdl
191 1.1 fvdl #ifdef __REENT
192 1.1 fvdl cv_allocsz = dtbsize * sizeof (cond_t);
193 1.1 fvdl dg_cv = (cond_t *) mem_alloc(cv_allocsz);
194 1.1 fvdl if (dg_cv == (cond_t *) NULL) {
195 1.1 fvdl mem_free(dg_fd_locks, fd_allocsz);
196 1.1 fvdl dg_fd_locks = (int *) NULL;
197 1.1 fvdl mutex_unlock(&clnt_fd_lock);
198 1.1 fvdl thr_sigsetmask(SIG_SETMASK, &(mask), NULL);
199 1.1 fvdl goto err1;
200 1.1 fvdl } else {
201 1.1 fvdl int i;
202 1.1 fvdl
203 1.1 fvdl for (i = 0; i < dtbsize; i++)
204 1.1 fvdl cond_init(&dg_cv[i], 0, (void *) 0);
205 1.1 fvdl }
206 1.1 fvdl #endif
207 1.1 fvdl }
208 1.1 fvdl
209 1.1 fvdl mutex_unlock(&clnt_fd_lock);
210 1.1 fvdl thr_sigsetmask(SIG_SETMASK, &(mask), NULL);
211 1.1 fvdl
212 1.3 christos if (svcaddr == NULL) {
213 1.1 fvdl rpc_createerr.cf_stat = RPC_UNKNOWNADDR;
214 1.3 christos return (NULL);
215 1.1 fvdl }
216 1.1 fvdl
217 1.1 fvdl if (!__rpc_fd2sockinfo(fd, &si)) {
218 1.1 fvdl rpc_createerr.cf_stat = RPC_TLIERROR;
219 1.1 fvdl rpc_createerr.cf_error.re_errno = 0;
220 1.3 christos return (NULL);
221 1.1 fvdl }
222 1.1 fvdl /*
223 1.1 fvdl * Find the receive and the send size
224 1.1 fvdl */
225 1.1 fvdl sendsz = __rpc_get_t_size(si.si_af, si.si_proto, (int)sendsz);
226 1.1 fvdl recvsz = __rpc_get_t_size(si.si_af, si.si_proto, (int)recvsz);
227 1.1 fvdl if ((sendsz == 0) || (recvsz == 0)) {
228 1.1 fvdl rpc_createerr.cf_stat = RPC_TLIERROR; /* XXX */
229 1.1 fvdl rpc_createerr.cf_error.re_errno = 0;
230 1.3 christos return (NULL);
231 1.1 fvdl }
232 1.1 fvdl
233 1.3 christos if ((cl = mem_alloc(sizeof (CLIENT))) == NULL)
234 1.1 fvdl goto err1;
235 1.1 fvdl /*
236 1.1 fvdl * Should be multiple of 4 for XDR.
237 1.1 fvdl */
238 1.1 fvdl sendsz = ((sendsz + 3) / 4) * 4;
239 1.1 fvdl recvsz = ((recvsz + 3) / 4) * 4;
240 1.3 christos cu = mem_alloc(sizeof (*cu) + sendsz + recvsz);
241 1.3 christos if (cu == NULL)
242 1.1 fvdl goto err1;
243 1.3 christos (void) memcpy(&cu->cu_raddr, svcaddr->buf, (size_t)svcaddr->len);
244 1.1 fvdl cu->cu_rlen = svcaddr->len;
245 1.1 fvdl cu->cu_outbuf = &cu->cu_inbuf[recvsz];
246 1.1 fvdl /* Other values can also be set through clnt_control() */
247 1.1 fvdl cu->cu_wait.tv_sec = 15; /* heuristically chosen */
248 1.1 fvdl cu->cu_wait.tv_usec = 0;
249 1.1 fvdl cu->cu_total.tv_sec = -1;
250 1.1 fvdl cu->cu_total.tv_usec = -1;
251 1.1 fvdl cu->cu_sendsz = sendsz;
252 1.1 fvdl cu->cu_recvsz = recvsz;
253 1.3 christos (void) gettimeofday(&now, NULL);
254 1.3 christos call_msg.rm_xid = __RPC_GETXID(&now);
255 1.1 fvdl call_msg.rm_call.cb_prog = program;
256 1.1 fvdl call_msg.rm_call.cb_vers = version;
257 1.1 fvdl xdrmem_create(&(cu->cu_outxdrs), cu->cu_outbuf, sendsz, XDR_ENCODE);
258 1.1 fvdl if (! xdr_callhdr(&(cu->cu_outxdrs), &call_msg)) {
259 1.1 fvdl rpc_createerr.cf_stat = RPC_CANTENCODEARGS; /* XXX */
260 1.1 fvdl rpc_createerr.cf_error.re_errno = 0;
261 1.1 fvdl goto err2;
262 1.1 fvdl }
263 1.1 fvdl cu->cu_xdrpos = XDR_GETPOS(&(cu->cu_outxdrs));
264 1.1 fvdl
265 1.1 fvdl /* XXX fvdl - do we still want this? */
266 1.1 fvdl #if 0
267 1.1 fvdl (void)bindresvport_sa(fd, (struct sockaddr *)svcaddr->buf);
268 1.1 fvdl #endif
269 1.1 fvdl ioctl(fd, FIONBIO, (char *)(void *)&one);
270 1.1 fvdl
271 1.1 fvdl /*
272 1.1 fvdl * By default, closeit is always FALSE. It is users responsibility
273 1.1 fvdl * to do a close on it, else the user may use clnt_control
274 1.1 fvdl * to let clnt_destroy do it for him/her.
275 1.1 fvdl */
276 1.1 fvdl cu->cu_closeit = FALSE;
277 1.1 fvdl cu->cu_fd = fd;
278 1.1 fvdl cl->cl_ops = clnt_dg_ops();
279 1.3 christos cl->cl_private = (caddr_t)(void *)cu;
280 1.1 fvdl cl->cl_auth = authnone_create();
281 1.3 christos cl->cl_tp = NULL;
282 1.3 christos cl->cl_netid = NULL;
283 1.1 fvdl cu->pfdp.fd = cu->cu_fd;
284 1.1 fvdl cu->pfdp.events = POLLIN | POLLPRI | POLLRDNORM | POLLRDBAND;
285 1.1 fvdl return (cl);
286 1.1 fvdl err1:
287 1.1 fvdl warnx(mem_err_clnt_dg);
288 1.1 fvdl rpc_createerr.cf_stat = RPC_SYSTEMERROR;
289 1.1 fvdl rpc_createerr.cf_error.re_errno = errno;
290 1.1 fvdl err2:
291 1.1 fvdl if (cl) {
292 1.3 christos mem_free(cl, sizeof (CLIENT));
293 1.1 fvdl if (cu)
294 1.3 christos mem_free(cu, sizeof (*cu) + sendsz + recvsz);
295 1.1 fvdl }
296 1.3 christos return (NULL);
297 1.1 fvdl }
298 1.1 fvdl
299 1.1 fvdl static enum clnt_stat
300 1.1 fvdl clnt_dg_call(cl, proc, xargs, argsp, xresults, resultsp, utimeout)
301 1.3 christos CLIENT *cl; /* client handle */
302 1.1 fvdl rpcproc_t proc; /* procedure number */
303 1.1 fvdl xdrproc_t xargs; /* xdr routine for args */
304 1.1 fvdl caddr_t argsp; /* pointer to args */
305 1.1 fvdl xdrproc_t xresults; /* xdr routine for results */
306 1.1 fvdl caddr_t resultsp; /* pointer to results */
307 1.1 fvdl struct timeval utimeout; /* seconds to wait before giving up */
308 1.1 fvdl {
309 1.3 christos struct cu_data *cu = (struct cu_data *)cl->cl_private;
310 1.3 christos XDR *xdrs;
311 1.3 christos size_t outlen;
312 1.1 fvdl struct rpc_msg reply_msg;
313 1.1 fvdl XDR reply_xdrs;
314 1.1 fvdl struct timeval time_waited;
315 1.1 fvdl bool_t ok;
316 1.1 fvdl int nrefreshes = 2; /* number of times to refresh cred */
317 1.1 fvdl struct timeval timeout;
318 1.1 fvdl struct timeval retransmit_time;
319 1.1 fvdl struct timeval startime, curtime;
320 1.1 fvdl int firsttimeout = 1;
321 1.1 fvdl #ifdef __REENT
322 1.1 fvdl int dtbsize = __rpc_dtbsize();
323 1.1 fvdl sigset_t mask;
324 1.1 fvdl #endif
325 1.1 fvdl sigset_t newmask;
326 1.3 christos socklen_t fromlen, inlen;
327 1.3 christos ssize_t recvlen = 0;
328 1.1 fvdl
329 1.1 fvdl sigfillset(&newmask);
330 1.1 fvdl thr_sigsetmask(SIG_SETMASK, &newmask, &mask);
331 1.1 fvdl mutex_lock(&clnt_fd_lock);
332 1.1 fvdl while (dg_fd_locks[cu->cu_fd])
333 1.1 fvdl cond_wait(&dg_cv[cu->cu_fd], &clnt_fd_lock);
334 1.1 fvdl dg_fd_locks[cu->cu_fd] = __rpc_lock_value;
335 1.1 fvdl mutex_unlock(&clnt_fd_lock);
336 1.1 fvdl if (cu->cu_total.tv_usec == -1) {
337 1.1 fvdl timeout = utimeout; /* use supplied timeout */
338 1.1 fvdl } else {
339 1.1 fvdl timeout = cu->cu_total; /* use default timeout */
340 1.1 fvdl }
341 1.1 fvdl
342 1.1 fvdl time_waited.tv_sec = 0;
343 1.1 fvdl time_waited.tv_usec = 0;
344 1.1 fvdl retransmit_time = cu->cu_wait;
345 1.1 fvdl
346 1.1 fvdl call_again:
347 1.1 fvdl xdrs = &(cu->cu_outxdrs);
348 1.1 fvdl xdrs->x_op = XDR_ENCODE;
349 1.1 fvdl XDR_SETPOS(xdrs, cu->cu_xdrpos);
350 1.1 fvdl /*
351 1.1 fvdl * the transaction is the first thing in the out buffer
352 1.1 fvdl */
353 1.3 christos (*(u_int32_t *)(void *)(cu->cu_outbuf))++;
354 1.6 christos if ((! XDR_PUTINT32(xdrs, (int32_t *)&proc)) ||
355 1.1 fvdl (! AUTH_MARSHALL(cl->cl_auth, xdrs)) ||
356 1.1 fvdl (! (*xargs)(xdrs, argsp))) {
357 1.1 fvdl release_fd_lock(cu->cu_fd, mask);
358 1.1 fvdl return (cu->cu_error.re_status = RPC_CANTENCODEARGS);
359 1.1 fvdl }
360 1.3 christos outlen = (size_t)XDR_GETPOS(xdrs);
361 1.1 fvdl
362 1.1 fvdl send_again:
363 1.1 fvdl if (sendto(cu->cu_fd, cu->cu_outbuf, outlen, 0,
364 1.3 christos (struct sockaddr *)(void *)&cu->cu_raddr, (socklen_t)cu->cu_rlen)
365 1.1 fvdl != outlen) {
366 1.1 fvdl cu->cu_error.re_errno = errno;
367 1.1 fvdl release_fd_lock(cu->cu_fd, mask);
368 1.1 fvdl return (cu->cu_error.re_status = RPC_CANTSEND);
369 1.1 fvdl }
370 1.1 fvdl
371 1.1 fvdl /*
372 1.1 fvdl * Hack to provide rpc-based message passing
373 1.1 fvdl */
374 1.1 fvdl if (timeout.tv_sec == 0 && timeout.tv_usec == 0) {
375 1.1 fvdl release_fd_lock(cu->cu_fd, mask);
376 1.1 fvdl return (cu->cu_error.re_status = RPC_TIMEDOUT);
377 1.1 fvdl }
378 1.1 fvdl /*
379 1.1 fvdl * sub-optimal code appears here because we have
380 1.1 fvdl * some clock time to spare while the packets are in flight.
381 1.1 fvdl * (We assume that this is actually only executed once.)
382 1.1 fvdl */
383 1.1 fvdl reply_msg.acpted_rply.ar_verf = _null_auth;
384 1.1 fvdl reply_msg.acpted_rply.ar_results.where = resultsp;
385 1.1 fvdl reply_msg.acpted_rply.ar_results.proc = xresults;
386 1.1 fvdl
387 1.1 fvdl
388 1.1 fvdl for (;;) {
389 1.3 christos switch (poll(&cu->pfdp, 1,
390 1.3 christos __rpc_timeval_to_msec(&retransmit_time))) {
391 1.1 fvdl case 0:
392 1.1 fvdl time_waited.tv_sec += retransmit_time.tv_sec;
393 1.1 fvdl time_waited.tv_usec += retransmit_time.tv_usec;
394 1.1 fvdl while (time_waited.tv_usec >= 1000000) {
395 1.1 fvdl time_waited.tv_sec++;
396 1.1 fvdl time_waited.tv_usec -= 1000000;
397 1.1 fvdl }
398 1.1 fvdl /* update retransmit_time */
399 1.1 fvdl if (retransmit_time.tv_sec < RPC_MAX_BACKOFF) {
400 1.1 fvdl retransmit_time.tv_usec *= 2;
401 1.1 fvdl retransmit_time.tv_sec *= 2;
402 1.1 fvdl while (retransmit_time.tv_usec >= 1000000) {
403 1.1 fvdl retransmit_time.tv_sec++;
404 1.1 fvdl retransmit_time.tv_usec -= 1000000;
405 1.1 fvdl }
406 1.1 fvdl }
407 1.1 fvdl
408 1.1 fvdl if ((time_waited.tv_sec < timeout.tv_sec) ||
409 1.1 fvdl ((time_waited.tv_sec == timeout.tv_sec) &&
410 1.1 fvdl (time_waited.tv_usec < timeout.tv_usec)))
411 1.1 fvdl goto send_again;
412 1.1 fvdl release_fd_lock(cu->cu_fd, mask);
413 1.1 fvdl return (cu->cu_error.re_status = RPC_TIMEDOUT);
414 1.1 fvdl
415 1.1 fvdl case -1:
416 1.1 fvdl if (errno == EBADF) {
417 1.1 fvdl cu->cu_error.re_errno = errno;
418 1.1 fvdl release_fd_lock(cu->cu_fd, mask);
419 1.1 fvdl return (cu->cu_error.re_status = RPC_CANTRECV);
420 1.1 fvdl }
421 1.1 fvdl if (errno != EINTR) {
422 1.1 fvdl errno = 0; /* reset it */
423 1.1 fvdl continue;
424 1.1 fvdl }
425 1.1 fvdl /* interrupted by another signal, update time_waited */
426 1.1 fvdl if (firsttimeout) {
427 1.1 fvdl /*
428 1.1 fvdl * Could have done gettimeofday before clnt_call
429 1.1 fvdl * but that means 1 more system call per each
430 1.1 fvdl * clnt_call, so do it after first time out
431 1.1 fvdl */
432 1.1 fvdl if (gettimeofday(&startime,
433 1.1 fvdl (struct timezone *) NULL) == -1) {
434 1.1 fvdl errno = 0;
435 1.1 fvdl continue;
436 1.1 fvdl }
437 1.1 fvdl firsttimeout = 0;
438 1.1 fvdl errno = 0;
439 1.1 fvdl continue;
440 1.1 fvdl };
441 1.1 fvdl if (gettimeofday(&curtime,
442 1.1 fvdl (struct timezone *) NULL) == -1) {
443 1.1 fvdl errno = 0;
444 1.1 fvdl continue;
445 1.1 fvdl };
446 1.1 fvdl time_waited.tv_sec += curtime.tv_sec - startime.tv_sec;
447 1.1 fvdl time_waited.tv_usec += curtime.tv_usec -
448 1.1 fvdl startime.tv_usec;
449 1.1 fvdl while (time_waited.tv_usec < 0) {
450 1.1 fvdl time_waited.tv_sec--;
451 1.1 fvdl time_waited.tv_usec += 1000000;
452 1.1 fvdl };
453 1.1 fvdl while (time_waited.tv_usec >= 1000000) {
454 1.1 fvdl time_waited.tv_sec++;
455 1.1 fvdl time_waited.tv_usec -= 1000000;
456 1.1 fvdl }
457 1.1 fvdl startime.tv_sec = curtime.tv_sec;
458 1.1 fvdl startime.tv_usec = curtime.tv_usec;
459 1.1 fvdl if ((time_waited.tv_sec > timeout.tv_sec) ||
460 1.1 fvdl ((time_waited.tv_sec == timeout.tv_sec) &&
461 1.1 fvdl (time_waited.tv_usec > timeout.tv_usec))) {
462 1.1 fvdl release_fd_lock(cu->cu_fd, mask);
463 1.1 fvdl return (cu->cu_error.re_status = RPC_TIMEDOUT);
464 1.1 fvdl }
465 1.1 fvdl errno = 0; /* reset it */
466 1.1 fvdl continue;
467 1.1 fvdl };
468 1.1 fvdl
469 1.1 fvdl if (cu->pfdp.revents & POLLNVAL || (cu->pfdp.revents == 0)) {
470 1.1 fvdl cu->cu_error.re_status = RPC_CANTRECV;
471 1.1 fvdl /*
472 1.1 fvdl * Note: we're faking errno here because we
473 1.1 fvdl * previously would have expected poll() to
474 1.1 fvdl * return -1 with errno EBADF. Poll(BA_OS)
475 1.1 fvdl * returns 0 and sets the POLLNVAL revents flag
476 1.1 fvdl * instead.
477 1.1 fvdl */
478 1.1 fvdl cu->cu_error.re_errno = errno = EBADF;
479 1.1 fvdl release_fd_lock(cu->cu_fd, mask);
480 1.1 fvdl return (-1);
481 1.1 fvdl }
482 1.1 fvdl
483 1.1 fvdl /* We have some data now */
484 1.1 fvdl do {
485 1.1 fvdl if (errno == EINTR) {
486 1.1 fvdl /*
487 1.1 fvdl * Must make sure errno was not already
488 1.1 fvdl * EINTR in case recvfrom() returns -1.
489 1.1 fvdl */
490 1.1 fvdl errno = 0;
491 1.1 fvdl }
492 1.1 fvdl fromlen = sizeof (struct sockaddr_storage);
493 1.3 christos recvlen = recvfrom(cu->cu_fd, cu->cu_inbuf,
494 1.3 christos cu->cu_recvsz, 0, (struct sockaddr *)(void *)&cu->cu_raddr,
495 1.1 fvdl &fromlen);
496 1.3 christos } while (recvlen < 0 && errno == EINTR);
497 1.3 christos if (recvlen < 0) {
498 1.1 fvdl if (errno == EWOULDBLOCK)
499 1.1 fvdl continue;
500 1.1 fvdl cu->cu_error.re_errno = errno;
501 1.1 fvdl release_fd_lock(cu->cu_fd, mask);
502 1.1 fvdl return (cu->cu_error.re_status = RPC_CANTRECV);
503 1.1 fvdl }
504 1.3 christos if (recvlen < sizeof (u_int32_t))
505 1.1 fvdl continue;
506 1.1 fvdl /* see if reply transaction id matches sent id */
507 1.3 christos if (*((u_int32_t *)(void *)(cu->cu_inbuf)) !=
508 1.3 christos *((u_int32_t *)(void *)(cu->cu_outbuf)))
509 1.1 fvdl continue;
510 1.1 fvdl /* we now assume we have the proper reply */
511 1.1 fvdl break;
512 1.1 fvdl }
513 1.3 christos inlen = (socklen_t)recvlen;
514 1.1 fvdl
515 1.1 fvdl /*
516 1.1 fvdl * now decode and validate the response
517 1.1 fvdl */
518 1.1 fvdl
519 1.1 fvdl xdrmem_create(&reply_xdrs, cu->cu_inbuf, (u_int)inlen, XDR_DECODE);
520 1.1 fvdl ok = xdr_replymsg(&reply_xdrs, &reply_msg);
521 1.1 fvdl /* XDR_DESTROY(&reply_xdrs); save a few cycles on noop destroy */
522 1.1 fvdl if (ok) {
523 1.1 fvdl if ((reply_msg.rm_reply.rp_stat == MSG_ACCEPTED) &&
524 1.1 fvdl (reply_msg.acpted_rply.ar_stat == SUCCESS))
525 1.1 fvdl cu->cu_error.re_status = RPC_SUCCESS;
526 1.1 fvdl else
527 1.1 fvdl _seterr_reply(&reply_msg, &(cu->cu_error));
528 1.1 fvdl
529 1.1 fvdl if (cu->cu_error.re_status == RPC_SUCCESS) {
530 1.1 fvdl if (! AUTH_VALIDATE(cl->cl_auth,
531 1.1 fvdl &reply_msg.acpted_rply.ar_verf)) {
532 1.1 fvdl cu->cu_error.re_status = RPC_AUTHERROR;
533 1.1 fvdl cu->cu_error.re_why = AUTH_INVALIDRESP;
534 1.1 fvdl }
535 1.1 fvdl if (reply_msg.acpted_rply.ar_verf.oa_base != NULL) {
536 1.1 fvdl xdrs->x_op = XDR_FREE;
537 1.1 fvdl (void) xdr_opaque_auth(xdrs,
538 1.1 fvdl &(reply_msg.acpted_rply.ar_verf));
539 1.1 fvdl }
540 1.1 fvdl } /* end successful completion */
541 1.1 fvdl /*
542 1.1 fvdl * If unsuccesful AND error is an authentication error
543 1.1 fvdl * then refresh credentials and try again, else break
544 1.1 fvdl */
545 1.1 fvdl else if (cu->cu_error.re_status == RPC_AUTHERROR)
546 1.1 fvdl /* maybe our credentials need to be refreshed ... */
547 1.1 fvdl if (nrefreshes > 0 && AUTH_REFRESH(cl->cl_auth)) {
548 1.1 fvdl nrefreshes--;
549 1.1 fvdl goto call_again;
550 1.1 fvdl }
551 1.1 fvdl /* end of unsuccessful completion */
552 1.1 fvdl } /* end of valid reply message */
553 1.1 fvdl else {
554 1.1 fvdl cu->cu_error.re_status = RPC_CANTDECODERES;
555 1.1 fvdl
556 1.1 fvdl }
557 1.1 fvdl release_fd_lock(cu->cu_fd, mask);
558 1.1 fvdl return (cu->cu_error.re_status);
559 1.1 fvdl }
560 1.1 fvdl
561 1.1 fvdl static void
562 1.1 fvdl clnt_dg_geterr(cl, errp)
563 1.1 fvdl CLIENT *cl;
564 1.1 fvdl struct rpc_err *errp;
565 1.1 fvdl {
566 1.3 christos struct cu_data *cu = (struct cu_data *)cl->cl_private;
567 1.1 fvdl
568 1.1 fvdl *errp = cu->cu_error;
569 1.1 fvdl }
570 1.1 fvdl
571 1.1 fvdl static bool_t
572 1.1 fvdl clnt_dg_freeres(cl, xdr_res, res_ptr)
573 1.1 fvdl CLIENT *cl;
574 1.1 fvdl xdrproc_t xdr_res;
575 1.1 fvdl caddr_t res_ptr;
576 1.1 fvdl {
577 1.3 christos struct cu_data *cu = (struct cu_data *)cl->cl_private;
578 1.3 christos XDR *xdrs = &(cu->cu_outxdrs);
579 1.1 fvdl bool_t dummy;
580 1.1 fvdl #ifdef __REENT
581 1.1 fvdl sigset_t mask;
582 1.1 fvdl #endif
583 1.1 fvdl sigset_t newmask;
584 1.1 fvdl
585 1.1 fvdl sigfillset(&newmask);
586 1.1 fvdl thr_sigsetmask(SIG_SETMASK, &newmask, &mask);
587 1.1 fvdl mutex_lock(&clnt_fd_lock);
588 1.1 fvdl while (dg_fd_locks[cu->cu_fd])
589 1.1 fvdl cond_wait(&dg_cv[cu->cu_fd], &clnt_fd_lock);
590 1.1 fvdl xdrs->x_op = XDR_FREE;
591 1.1 fvdl dummy = (*xdr_res)(xdrs, res_ptr);
592 1.1 fvdl mutex_unlock(&clnt_fd_lock);
593 1.1 fvdl thr_sigsetmask(SIG_SETMASK, &mask, NULL);
594 1.1 fvdl cond_signal(&dg_cv[cu->cu_fd]);
595 1.1 fvdl return (dummy);
596 1.1 fvdl }
597 1.1 fvdl
598 1.1 fvdl /*ARGSUSED*/
599 1.1 fvdl static void
600 1.1 fvdl clnt_dg_abort(h)
601 1.1 fvdl CLIENT *h;
602 1.1 fvdl {
603 1.1 fvdl }
604 1.1 fvdl
605 1.1 fvdl static bool_t
606 1.1 fvdl clnt_dg_control(cl, request, info)
607 1.1 fvdl CLIENT *cl;
608 1.1 fvdl u_int request;
609 1.1 fvdl char *info;
610 1.1 fvdl {
611 1.3 christos struct cu_data *cu = (struct cu_data *)cl->cl_private;
612 1.1 fvdl struct netbuf *addr;
613 1.1 fvdl #ifdef __REENT
614 1.1 fvdl sigset_t mask;
615 1.1 fvdl #endif
616 1.1 fvdl sigset_t newmask;
617 1.1 fvdl
618 1.1 fvdl sigfillset(&newmask);
619 1.1 fvdl thr_sigsetmask(SIG_SETMASK, &newmask, &mask);
620 1.1 fvdl mutex_lock(&clnt_fd_lock);
621 1.1 fvdl while (dg_fd_locks[cu->cu_fd])
622 1.1 fvdl cond_wait(&dg_cv[cu->cu_fd], &clnt_fd_lock);
623 1.1 fvdl dg_fd_locks[cu->cu_fd] = __rpc_lock_value;
624 1.1 fvdl mutex_unlock(&clnt_fd_lock);
625 1.1 fvdl switch (request) {
626 1.1 fvdl case CLSET_FD_CLOSE:
627 1.1 fvdl cu->cu_closeit = TRUE;
628 1.1 fvdl release_fd_lock(cu->cu_fd, mask);
629 1.1 fvdl return (TRUE);
630 1.1 fvdl case CLSET_FD_NCLOSE:
631 1.1 fvdl cu->cu_closeit = FALSE;
632 1.1 fvdl release_fd_lock(cu->cu_fd, mask);
633 1.1 fvdl return (TRUE);
634 1.1 fvdl }
635 1.1 fvdl
636 1.1 fvdl /* for other requests which use info */
637 1.1 fvdl if (info == NULL) {
638 1.1 fvdl release_fd_lock(cu->cu_fd, mask);
639 1.1 fvdl return (FALSE);
640 1.1 fvdl }
641 1.1 fvdl switch (request) {
642 1.1 fvdl case CLSET_TIMEOUT:
643 1.3 christos if (time_not_ok((struct timeval *)(void *)info)) {
644 1.1 fvdl release_fd_lock(cu->cu_fd, mask);
645 1.1 fvdl return (FALSE);
646 1.1 fvdl }
647 1.3 christos cu->cu_total = *(struct timeval *)(void *)info;
648 1.1 fvdl break;
649 1.1 fvdl case CLGET_TIMEOUT:
650 1.3 christos *(struct timeval *)(void *)info = cu->cu_total;
651 1.1 fvdl break;
652 1.1 fvdl case CLGET_SERVER_ADDR: /* Give him the fd address */
653 1.1 fvdl /* Now obsolete. Only for backward compatibility */
654 1.3 christos (void) memcpy(info, &cu->cu_raddr, (size_t)cu->cu_rlen);
655 1.1 fvdl break;
656 1.1 fvdl case CLSET_RETRY_TIMEOUT:
657 1.3 christos if (time_not_ok((struct timeval *)(void *)info)) {
658 1.1 fvdl release_fd_lock(cu->cu_fd, mask);
659 1.1 fvdl return (FALSE);
660 1.1 fvdl }
661 1.3 christos cu->cu_wait = *(struct timeval *)(void *)info;
662 1.1 fvdl break;
663 1.1 fvdl case CLGET_RETRY_TIMEOUT:
664 1.3 christos *(struct timeval *)(void *)info = cu->cu_wait;
665 1.1 fvdl break;
666 1.1 fvdl case CLGET_FD:
667 1.3 christos *(int *)(void *)info = cu->cu_fd;
668 1.1 fvdl break;
669 1.1 fvdl case CLGET_SVC_ADDR:
670 1.3 christos addr = (struct netbuf *)(void *)info;
671 1.1 fvdl addr->buf = &cu->cu_raddr;
672 1.1 fvdl addr->len = cu->cu_rlen;
673 1.1 fvdl addr->maxlen = sizeof cu->cu_raddr;
674 1.1 fvdl break;
675 1.1 fvdl case CLSET_SVC_ADDR: /* set to new address */
676 1.3 christos addr = (struct netbuf *)(void *)info;
677 1.1 fvdl if (addr->len < sizeof cu->cu_raddr)
678 1.1 fvdl return (FALSE);
679 1.1 fvdl (void) memcpy(&cu->cu_raddr, addr->buf, addr->len);
680 1.1 fvdl cu->cu_rlen = addr->len;
681 1.1 fvdl break;
682 1.1 fvdl case CLGET_XID:
683 1.1 fvdl /*
684 1.1 fvdl * use the knowledge that xid is the
685 1.1 fvdl * first element in the call structure *.
686 1.1 fvdl * This will get the xid of the PREVIOUS call
687 1.1 fvdl */
688 1.3 christos *(u_int32_t *)(void *)info =
689 1.3 christos ntohl(*(u_int32_t *)(void *)cu->cu_outbuf);
690 1.1 fvdl break;
691 1.1 fvdl
692 1.1 fvdl case CLSET_XID:
693 1.1 fvdl /* This will set the xid of the NEXT call */
694 1.3 christos *(u_int32_t *)(void *)cu->cu_outbuf =
695 1.3 christos htonl(*(u_int32_t *)(void *)info - 1);
696 1.1 fvdl /* decrement by 1 as clnt_dg_call() increments once */
697 1.1 fvdl break;
698 1.1 fvdl
699 1.1 fvdl case CLGET_VERS:
700 1.1 fvdl /*
701 1.1 fvdl * This RELIES on the information that, in the call body,
702 1.1 fvdl * the version number field is the fifth field from the
703 1.1 fvdl * begining of the RPC header. MUST be changed if the
704 1.1 fvdl * call_struct is changed
705 1.1 fvdl */
706 1.3 christos *(u_int32_t *)(void *)info =
707 1.3 christos ntohl(*(u_int32_t *)(void *)(cu->cu_outbuf +
708 1.3 christos 4 * BYTES_PER_XDR_UNIT));
709 1.1 fvdl break;
710 1.1 fvdl
711 1.1 fvdl case CLSET_VERS:
712 1.3 christos *(u_int32_t *)(void *)(cu->cu_outbuf + 4 * BYTES_PER_XDR_UNIT)
713 1.3 christos = htonl(*(u_int32_t *)(void *)info);
714 1.1 fvdl break;
715 1.1 fvdl
716 1.1 fvdl case CLGET_PROG:
717 1.1 fvdl /*
718 1.1 fvdl * This RELIES on the information that, in the call body,
719 1.1 fvdl * the program number field is the fourth field from the
720 1.1 fvdl * begining of the RPC header. MUST be changed if the
721 1.1 fvdl * call_struct is changed
722 1.1 fvdl */
723 1.3 christos *(u_int32_t *)(void *)info =
724 1.3 christos ntohl(*(u_int32_t *)(void *)(cu->cu_outbuf +
725 1.3 christos 3 * BYTES_PER_XDR_UNIT));
726 1.1 fvdl break;
727 1.1 fvdl
728 1.1 fvdl case CLSET_PROG:
729 1.3 christos *(u_int32_t *)(void *)(cu->cu_outbuf + 3 * BYTES_PER_XDR_UNIT)
730 1.3 christos = htonl(*(u_int32_t *)(void *)info);
731 1.1 fvdl break;
732 1.1 fvdl
733 1.1 fvdl default:
734 1.1 fvdl release_fd_lock(cu->cu_fd, mask);
735 1.1 fvdl return (FALSE);
736 1.1 fvdl }
737 1.1 fvdl release_fd_lock(cu->cu_fd, mask);
738 1.1 fvdl return (TRUE);
739 1.1 fvdl }
740 1.1 fvdl
741 1.1 fvdl static void
742 1.1 fvdl clnt_dg_destroy(cl)
743 1.1 fvdl CLIENT *cl;
744 1.1 fvdl {
745 1.3 christos struct cu_data *cu = (struct cu_data *)cl->cl_private;
746 1.1 fvdl int cu_fd = cu->cu_fd;
747 1.1 fvdl #ifdef __REENT
748 1.1 fvdl sigset_t mask;
749 1.1 fvdl #endif
750 1.1 fvdl sigset_t newmask;
751 1.1 fvdl
752 1.1 fvdl sigfillset(&newmask);
753 1.1 fvdl thr_sigsetmask(SIG_SETMASK, &newmask, &mask);
754 1.1 fvdl mutex_lock(&clnt_fd_lock);
755 1.1 fvdl while (dg_fd_locks[cu_fd])
756 1.1 fvdl cond_wait(&dg_cv[cu_fd], &clnt_fd_lock);
757 1.1 fvdl if (cu->cu_closeit)
758 1.1 fvdl (void) close(cu_fd);
759 1.1 fvdl XDR_DESTROY(&(cu->cu_outxdrs));
760 1.3 christos mem_free(cu, (sizeof (*cu) + cu->cu_sendsz + cu->cu_recvsz));
761 1.1 fvdl if (cl->cl_netid && cl->cl_netid[0])
762 1.1 fvdl mem_free(cl->cl_netid, strlen(cl->cl_netid) +1);
763 1.1 fvdl if (cl->cl_tp && cl->cl_tp[0])
764 1.1 fvdl mem_free(cl->cl_tp, strlen(cl->cl_tp) +1);
765 1.3 christos mem_free(cl, sizeof (CLIENT));
766 1.1 fvdl mutex_unlock(&clnt_fd_lock);
767 1.1 fvdl thr_sigsetmask(SIG_SETMASK, &mask, NULL);
768 1.1 fvdl cond_signal(&dg_cv[cu_fd]);
769 1.1 fvdl }
770 1.1 fvdl
771 1.1 fvdl static struct clnt_ops *
772 1.1 fvdl clnt_dg_ops()
773 1.1 fvdl {
774 1.1 fvdl static struct clnt_ops ops;
775 1.1 fvdl #ifdef __REENT
776 1.1 fvdl extern mutex_t ops_lock;
777 1.1 fvdl sigset_t mask;
778 1.1 fvdl #endif
779 1.1 fvdl sigset_t newmask;
780 1.1 fvdl
781 1.1 fvdl /* VARIABLES PROTECTED BY ops_lock: ops */
782 1.1 fvdl
783 1.1 fvdl sigfillset(&newmask);
784 1.1 fvdl thr_sigsetmask(SIG_SETMASK, &newmask, &mask);
785 1.1 fvdl mutex_lock(&ops_lock);
786 1.1 fvdl if (ops.cl_call == NULL) {
787 1.1 fvdl ops.cl_call = clnt_dg_call;
788 1.1 fvdl ops.cl_abort = clnt_dg_abort;
789 1.1 fvdl ops.cl_geterr = clnt_dg_geterr;
790 1.1 fvdl ops.cl_freeres = clnt_dg_freeres;
791 1.1 fvdl ops.cl_destroy = clnt_dg_destroy;
792 1.1 fvdl ops.cl_control = clnt_dg_control;
793 1.1 fvdl }
794 1.1 fvdl mutex_unlock(&ops_lock);
795 1.1 fvdl thr_sigsetmask(SIG_SETMASK, &mask, NULL);
796 1.1 fvdl return (&ops);
797 1.1 fvdl }
798 1.1 fvdl
799 1.1 fvdl /*
800 1.1 fvdl * Make sure that the time is not garbage. -1 value is allowed.
801 1.1 fvdl */
802 1.1 fvdl static bool_t
803 1.1 fvdl time_not_ok(t)
804 1.1 fvdl struct timeval *t;
805 1.1 fvdl {
806 1.1 fvdl return (t->tv_sec < -1 || t->tv_sec > 100000000 ||
807 1.1 fvdl t->tv_usec < -1 || t->tv_usec > 1000000);
808 1.1 fvdl }
809 1.1 fvdl
810 1.1 fvdl
811 1.1 fvdl /*
812 1.1 fvdl * Convert from timevals (used by select) to milliseconds (used by poll).
813 1.1 fvdl */
814 1.1 fvdl static int
815 1.1 fvdl __rpc_timeval_to_msec(t)
816 1.1 fvdl struct timeval *t;
817 1.1 fvdl {
818 1.1 fvdl int t1, tmp;
819 1.1 fvdl
820 1.1 fvdl /*
821 1.1 fvdl * We're really returning t->tv_sec * 1000 + (t->tv_usec / 1000)
822 1.1 fvdl * but try to do so efficiently. Note: 1000 = 1024 - 16 - 8.
823 1.1 fvdl */
824 1.3 christos tmp = (int)t->tv_sec << 3;
825 1.1 fvdl t1 = -tmp;
826 1.1 fvdl t1 += t1 << 1;
827 1.1 fvdl t1 += tmp << 7;
828 1.1 fvdl if (t->tv_usec)
829 1.3 christos t1 += (int)(t->tv_usec / 1000);
830 1.1 fvdl
831 1.1 fvdl return (t1);
832 1.1 fvdl }
833