1 /* @(#)clnt_tcp.c 2.2 88/08/01 4.0 RPCSRC */
3 * Sun RPC is a product of Sun Microsystems, Inc. and is provided for
4 * unrestricted use provided that this legend is included on all tape
5 * media and as a part of the software program in whole or part. Users
6 * may copy or modify Sun RPC without charge, but are not authorized
7 * to license or distribute it to anyone else except as part of a product or
8 * program developed by the user.
10 * SUN RPC IS PROVIDED AS IS WITH NO WARRANTIES OF ANY KIND INCLUDING THE
11 * WARRANTIES OF DESIGN, MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
12 * PURPOSE, OR ARISING FROM A COURSE OF DEALING, USAGE OR TRADE PRACTICE.
14 * Sun RPC is provided with no support and without any obligation on the
15 * part of Sun Microsystems, Inc. to assist in its use, correction,
16 * modification or enhancement.
18 * SUN MICROSYSTEMS, INC. SHALL HAVE NO LIABILITY WITH RESPECT TO THE
19 * INFRINGEMENT OF COPYRIGHTS, TRADE SECRETS OR ANY PATENTS BY SUN RPC
20 * OR ANY PART THEREOF.
22 * In no event will Sun Microsystems, Inc. be liable for any lost revenue
23 * or profits or other special, indirect and consequential damages, even if
24 * Sun has been advised of the possibility of such damages.
26 * Sun Microsystems, Inc.
28 * Mountain View, California 94043
31 static char sccsid
[] = "@(#)clnt_tcp.c 1.37 87/10/05 Copyr 1984 Sun Micro";
35 * clnt_tcp.c, Implements a TCP/IP based, client side RPC.
37 * Copyright (C) 1984, Sun Microsystems, Inc.
39 * TCP based RPC supports 'batched calls'.
40 * A sequence of calls may be batched-up in a send buffer. The rpc call
41 * return immediately to the client even though the call was not necessarily
42 * sent. The batching occurs if the results' xdr routine is NULL (0) AND
43 * the rpc timeout value is zero (see clnt.h, rpc).
45 * Clients should NOT casually batch calls that in fact return results; that is,
46 * the server side should be aware that a call is batched and not produce any
47 * return message. Batched calls that produce many result messages can
48 * deadlock (netlock) the client and the server....
50 * Now go hang yourself.
57 #include "rpc_private.h"
59 #include <sys/socket.h>
60 #include <rpc/pmap_clnt.h>
62 #define MCALL_MSG_SIZE 24
68 struct timeval ct_wait
;
69 bool_t ct_waitset
; /* wait set by clnt_control? */
70 struct sockaddr_in ct_addr
;
71 struct rpc_err ct_error
;
72 char ct_mcall
[MCALL_MSG_SIZE
]; /* marshalled callmsg */
73 u_int ct_mpos
; /* pos after marshal */
77 static int readtcp (char *, char *, int);
78 static int writetcp (char *, char *, int);
80 static enum clnt_stat
clnttcp_call (CLIENT
*, u_long
, xdrproc_t
, caddr_t
,
81 xdrproc_t
, caddr_t
, struct timeval
);
82 static void clnttcp_abort (void);
83 static void clnttcp_geterr (CLIENT
*, struct rpc_err
*);
84 static bool_t
clnttcp_freeres (CLIENT
*, xdrproc_t
, caddr_t
);
85 static bool_t
clnttcp_control (CLIENT
*, int, char *);
86 static void clnttcp_destroy (CLIENT
*);
88 static const struct clnt_ops tcp_ops
=
99 * Create a client handle for a tcp/ip connection.
100 * If *sockp<0, *sockp is set to a newly created TCP socket and it is
101 * connected to raddr. If *sockp non-negative then
102 * raddr is ignored. The rpc/tcp package does buffering
103 * similar to stdio, so the client must pick send and receive buffer sizes,];
104 * 0 => use the default.
105 * If raddr->sin_port is 0, then a binder on the remote machine is
106 * consulted for the right port number.
107 * NB: *sockp is copied into a private area.
108 * NB: It is the clients responsibility to close *sockp.
109 * NB: The rpch->cl_auth is set null authentication. Caller may wish to set this
110 * something more useful.
113 clnttcp_create (struct sockaddr_in
*raddr
, u_long prog
, u_long vers
,
114 int *sockp
, u_int sendsz
, u_int recvsz
)
118 struct rpc_msg call_msg
;
120 h
= (CLIENT
*) mem_alloc (sizeof (*h
));
121 ct
= (struct ct_data
*) mem_alloc (sizeof (*ct
));
122 if (h
== NULL
|| ct
== NULL
)
124 struct rpc_createerr
*ce
= &get_rpc_createerr ();
125 (void) fputs ("clnttcp_create: out of memory\n", stderr
);
126 ce
->cf_stat
= RPC_SYSTEMERROR
;
127 ce
->cf_error
.re_errno
= ENOMEM
;
132 * If no port number given ask the pmap for one
134 if (raddr
->sin_port
== 0)
137 if ((port
= pmap_getport (raddr
, prog
, vers
, IPPROTO_TCP
)) == 0)
139 mem_free ((caddr_t
) ct
, sizeof (struct ct_data
));
140 mem_free ((caddr_t
) h
, sizeof (CLIENT
));
141 return ((CLIENT
*) NULL
);
143 raddr
->sin_port
= htons (port
);
147 * If no socket given, open one
151 *sockp
= socket (AF_INET
, SOCK_STREAM
, IPPROTO_TCP
);
152 (void) bindresvport (*sockp
, (struct sockaddr_in
*) 0);
154 || (connect (*sockp
, (struct sockaddr
*) raddr
,
155 sizeof (*raddr
)) < 0))
157 struct rpc_createerr
*ce
= &get_rpc_createerr ();
158 ce
->cf_stat
= RPC_SYSTEMERROR
;
159 ce
->cf_error
.re_errno
= errno
;
161 (void) close (*sockp
);
164 ct
->ct_closeit
= TRUE
;
168 ct
->ct_closeit
= FALSE
;
172 * Set up private data struct
174 ct
->ct_sock
= *sockp
;
175 ct
->ct_wait
.tv_usec
= 0;
176 ct
->ct_waitset
= FALSE
;
177 ct
->ct_addr
= *raddr
;
180 * Initialize call message
182 call_msg
.rm_xid
= _create_xid ();
183 call_msg
.rm_direction
= CALL
;
184 call_msg
.rm_call
.cb_rpcvers
= RPC_MSG_VERSION
;
185 call_msg
.rm_call
.cb_prog
= prog
;
186 call_msg
.rm_call
.cb_vers
= vers
;
189 * pre-serialize the static part of the call msg and stash it away
191 xdrmem_create (&(ct
->ct_xdrs
), ct
->ct_mcall
, MCALL_MSG_SIZE
,
193 if (!xdr_callhdr (&(ct
->ct_xdrs
), &call_msg
))
197 (void) close (*sockp
);
201 ct
->ct_mpos
= XDR_GETPOS (&(ct
->ct_xdrs
));
202 XDR_DESTROY (&(ct
->ct_xdrs
));
205 * Create a client handle which uses xdrrec for serialization
206 * and authnone for authentication.
208 xdrrec_create (&(ct
->ct_xdrs
), sendsz
, recvsz
,
209 (caddr_t
) ct
, readtcp
, writetcp
);
210 h
->cl_ops
= &tcp_ops
;
211 h
->cl_private
= (caddr_t
) ct
;
212 h
->cl_auth
= authnone_create ();
217 * Something goofed, free stuff and barf
219 mem_free ((caddr_t
) ct
, sizeof (struct ct_data
));
220 mem_free ((caddr_t
) h
, sizeof (CLIENT
));
221 return ((CLIENT
*) NULL
);
223 libc_hidden_def(clnttcp_create
)
225 static enum clnt_stat
226 clnttcp_call (CLIENT
*h
, u_long proc
, xdrproc_t xdr_args
, caddr_t args_ptr
,
227 xdrproc_t xdr_results
, caddr_t results_ptr
,
228 struct timeval timeout
)
230 struct ct_data
*ct
= (struct ct_data
*) h
->cl_private
;
231 XDR
*xdrs
= &(ct
->ct_xdrs
);
232 struct rpc_msg reply_msg
;
234 u_int32_t
*msg_x_id
= (u_int32_t
*) (ct
->ct_mcall
); /* yuk */
240 ct
->ct_wait
= timeout
;
244 (xdr_results
== (xdrproc_t
) 0 && ct
->ct_wait
.tv_sec
== 0
245 && ct
->ct_wait
.tv_usec
== 0) ? FALSE
: TRUE
;
248 xdrs
->x_op
= XDR_ENCODE
;
249 ct
->ct_error
.re_status
= RPC_SUCCESS
;
250 x_id
= ntohl (--(*msg_x_id
));
251 if ((!XDR_PUTBYTES (xdrs
, ct
->ct_mcall
, ct
->ct_mpos
)) ||
252 (!XDR_PUTLONG (xdrs
, (long *) &proc
)) ||
253 (!AUTH_MARSHALL (h
->cl_auth
, xdrs
)) ||
254 (!(*xdr_args
) (xdrs
, args_ptr
)))
256 if (ct
->ct_error
.re_status
== RPC_SUCCESS
)
257 ct
->ct_error
.re_status
= RPC_CANTENCODEARGS
;
258 (void) xdrrec_endofrecord (xdrs
, TRUE
);
259 return (ct
->ct_error
.re_status
);
261 if (!xdrrec_endofrecord (xdrs
, shipnow
))
262 return ct
->ct_error
.re_status
= RPC_CANTSEND
;
266 * Hack to provide rpc-based message passing
268 if (ct
->ct_wait
.tv_sec
== 0 && ct
->ct_wait
.tv_usec
== 0)
270 return ct
->ct_error
.re_status
= RPC_TIMEDOUT
;
275 * Keep receiving until we get a valid transaction id
277 xdrs
->x_op
= XDR_DECODE
;
280 reply_msg
.acpted_rply
.ar_verf
= _null_auth
;
281 reply_msg
.acpted_rply
.ar_results
.where
= NULL
;
282 reply_msg
.acpted_rply
.ar_results
.proc
= (xdrproc_t
)xdr_void
;
283 if (!xdrrec_skiprecord (xdrs
))
284 return (ct
->ct_error
.re_status
);
285 /* now decode and validate the response header */
286 if (!xdr_replymsg (xdrs
, &reply_msg
))
288 if (ct
->ct_error
.re_status
== RPC_SUCCESS
)
290 return ct
->ct_error
.re_status
;
292 if ((u_int32_t
) reply_msg
.rm_xid
== (u_int32_t
) x_id
)
299 _seterr_reply (&reply_msg
, &(ct
->ct_error
));
300 if (ct
->ct_error
.re_status
== RPC_SUCCESS
)
302 if (!AUTH_VALIDATE (h
->cl_auth
, &reply_msg
.acpted_rply
.ar_verf
))
304 ct
->ct_error
.re_status
= RPC_AUTHERROR
;
305 ct
->ct_error
.re_why
= AUTH_INVALIDRESP
;
307 else if (!(*xdr_results
) (xdrs
, results_ptr
))
309 if (ct
->ct_error
.re_status
== RPC_SUCCESS
)
310 ct
->ct_error
.re_status
= RPC_CANTDECODERES
;
312 /* free verifier ... */
313 if (reply_msg
.acpted_rply
.ar_verf
.oa_base
!= NULL
)
315 xdrs
->x_op
= XDR_FREE
;
316 (void) xdr_opaque_auth (xdrs
, &(reply_msg
.acpted_rply
.ar_verf
));
318 } /* end successful completion */
321 /* maybe our credentials need to be refreshed ... */
322 if (refreshes
-- && AUTH_REFRESH (h
->cl_auth
))
324 } /* end of unsuccessful completion */
325 return ct
->ct_error
.re_status
;
329 clnttcp_geterr (CLIENT
*h
, struct rpc_err
*errp
)
332 (struct ct_data
*) h
->cl_private
;
334 *errp
= ct
->ct_error
;
338 clnttcp_freeres (CLIENT
*cl
, xdrproc_t xdr_res
, caddr_t res_ptr
)
340 struct ct_data
*ct
= (struct ct_data
*) cl
->cl_private
;
341 XDR
*xdrs
= &(ct
->ct_xdrs
);
343 xdrs
->x_op
= XDR_FREE
;
344 return (*xdr_res
) (xdrs
, res_ptr
);
353 clnttcp_control (CLIENT
*cl
, int request
, char *info
)
355 struct ct_data
*ct
= (struct ct_data
*) cl
->cl_private
;
361 ct
->ct_closeit
= TRUE
;
363 case CLSET_FD_NCLOSE
:
364 ct
->ct_closeit
= FALSE
;
367 ct
->ct_wait
= *(struct timeval
*) info
;
368 ct
->ct_waitset
= TRUE
;
371 *(struct timeval
*) info
= ct
->ct_wait
;
373 case CLGET_SERVER_ADDR
:
374 *(struct sockaddr_in
*) info
= ct
->ct_addr
;
377 *(int *)info
= ct
->ct_sock
;
381 * use the knowledge that xid is the
382 * first element in the call structure *.
383 * This will get the xid of the PREVIOUS call
385 *(u_long
*)info
= ntohl (*(u_long
*)ct
->ct_mcall
);
388 /* This will set the xid of the NEXT call */
389 *(u_long
*)ct
->ct_mcall
= htonl (*(u_long
*)info
- 1);
390 /* decrement by 1 as clnttcp_call() increments once */
394 * This RELIES on the information that, in the call body,
395 * the version number field is the fifth field from the
396 * begining of the RPC header. MUST be changed if the
397 * call_struct is changed
399 *(u_long
*)info
= ntohl (*(u_long
*)(ct
->ct_mcall
+
400 4 * BYTES_PER_XDR_UNIT
));
403 *(u_long
*)(ct
->ct_mcall
+ 4 * BYTES_PER_XDR_UNIT
)
404 = htonl (*(u_long
*)info
);
408 * This RELIES on the information that, in the call body,
409 * the program number field is the field from the
410 * begining of the RPC header. MUST be changed if the
411 * call_struct is changed
413 *(u_long
*)info
= ntohl(*(u_long
*)(ct
->ct_mcall
+
414 3 * BYTES_PER_XDR_UNIT
));
417 *(u_long
*)(ct
->ct_mcall
+ 3 * BYTES_PER_XDR_UNIT
)
418 = htonl(*(u_long
*)info
);
420 /* The following are only possible with TI-RPC */
421 case CLGET_RETRY_TIMEOUT
:
422 case CLSET_RETRY_TIMEOUT
:
425 case CLSET_PUSH_TIMOD
:
426 case CLSET_POP_TIMOD
:
435 clnttcp_destroy (CLIENT
*h
)
438 (struct ct_data
*) h
->cl_private
;
442 (void) close (ct
->ct_sock
);
444 XDR_DESTROY (&(ct
->ct_xdrs
));
445 mem_free ((caddr_t
) ct
, sizeof (struct ct_data
));
446 mem_free ((caddr_t
) h
, sizeof (CLIENT
));
450 * Interface between xdr serializer and tcp connection.
451 * Behaves like the system calls, read & write, but keeps some error state
452 * around for the rpc level.
455 readtcp (char *ctptr
, char *buf
, int len
)
457 struct ct_data
*ct
= (struct ct_data
*)ctptr
;
459 int milliseconds
= (ct
->ct_wait
.tv_sec
* 1000) +
460 (ct
->ct_wait
.tv_usec
/ 1000);
469 switch (poll(&fd
, 1, milliseconds
))
472 ct
->ct_error
.re_status
= RPC_TIMEDOUT
;
478 ct
->ct_error
.re_status
= RPC_CANTRECV
;
479 ct
->ct_error
.re_errno
= errno
;
484 switch (len
= read (ct
->ct_sock
, buf
, len
))
489 ct
->ct_error
.re_errno
= ECONNRESET
;
490 ct
->ct_error
.re_status
= RPC_CANTRECV
;
491 len
= -1; /* it's really an error */
495 ct
->ct_error
.re_errno
= errno
;
496 ct
->ct_error
.re_status
= RPC_CANTRECV
;
503 writetcp (char *ctptr
, char *buf
, int len
)
506 struct ct_data
*ct
= (struct ct_data
*)ctptr
;
508 for (cnt
= len
; cnt
> 0; cnt
-= i
, buf
+= i
)
510 if ((i
= write (ct
->ct_sock
, buf
, cnt
)) == -1)
512 ct
->ct_error
.re_errno
= errno
;
513 ct
->ct_error
.re_status
= RPC_CANTSEND
;