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