Added -f command for foreground operation
[unix-history] / lib / librpc / rpc / svc_tcp.c
CommitLineData
15637ed4
RG
1/* @(#)svc_tcp.c 2.2 88/08/01 4.0 RPCSRC */
2/*
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.
9 *
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.
13 *
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.
17 *
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.
21 *
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.
25 *
26 * Sun Microsystems, Inc.
27 * 2550 Garcia Avenue
28 * Mountain View, California 94043
29 */
30#if !defined(lint) && defined(SCCSIDS)
31static char sccsid[] = "@(#)svc_tcp.c 1.21 87/08/11 Copyr 1984 Sun Micro";
32#endif
33
34/*
35 * svc_tcp.c, Server side for TCP/IP based RPC.
36 *
37 * Copyright (C) 1984, Sun Microsystems, Inc.
38 *
39 * Actually implements two flavors of transporter -
40 * a tcp rendezvouser (a listner and connection establisher)
41 * and a record/tcp stream.
42 */
43
44#include <stdio.h>
45#include <rpc/rpc.h>
46#include <sys/socket.h>
47#include <errno.h>
48extern bool_t abort();
49extern errno;
50
51/*
52 * Ops vector for TCP/IP based rpc service handle
53 */
54static bool_t svctcp_recv();
55static enum xprt_stat svctcp_stat();
56static bool_t svctcp_getargs();
57static bool_t svctcp_reply();
58static bool_t svctcp_freeargs();
59static void svctcp_destroy();
60
61static struct xp_ops svctcp_op = {
62 svctcp_recv,
63 svctcp_stat,
64 svctcp_getargs,
65 svctcp_reply,
66 svctcp_freeargs,
67 svctcp_destroy
68};
69
70/*
71 * Ops vector for TCP/IP rendezvous handler
72 */
73static bool_t rendezvous_request();
74static enum xprt_stat rendezvous_stat();
75
76static struct xp_ops svctcp_rendezvous_op = {
77 rendezvous_request,
78 rendezvous_stat,
79 abort,
80 abort,
81 abort,
82 svctcp_destroy
83};
84
85static int readtcp(), writetcp();
86static SVCXPRT *makefd_xprt();
87
88struct tcp_rendezvous { /* kept in xprt->xp_p1 */
89 u_int sendsize;
90 u_int recvsize;
91};
92
93struct tcp_conn { /* kept in xprt->xp_p1 */
94 enum xprt_stat strm_stat;
95 u_long x_id;
96 XDR xdrs;
97 char verf_body[MAX_AUTH_BYTES];
98};
99
100/*
101 * Usage:
102 * xprt = svctcp_create(sock, send_buf_size, recv_buf_size);
103 *
104 * Creates, registers, and returns a (rpc) tcp based transporter.
105 * Once *xprt is initialized, it is registered as a transporter
106 * see (svc.h, xprt_register). This routine returns
107 * a NULL if a problem occurred.
108 *
109 * If sock<0 then a socket is created, else sock is used.
110 * If the socket, sock is not bound to a port then svctcp_create
111 * binds it to an arbitrary port. The routine then starts a tcp
112 * listener on the socket's associated port. In any (successful) case,
113 * xprt->xp_sock is the registered socket number and xprt->xp_port is the
114 * associated port number.
115 *
116 * Since tcp streams do buffered io similar to stdio, the caller can specify
117 * how big the send and receive buffers are via the second and third parms;
118 * 0 => use the system default.
119 */
120SVCXPRT *
121svctcp_create(sock, sendsize, recvsize)
122 register int sock;
123 u_int sendsize;
124 u_int recvsize;
125{
126 bool_t madesock = FALSE;
127 register SVCXPRT *xprt;
128 register struct tcp_rendezvous *r;
129 struct sockaddr_in addr;
130 int len = sizeof(struct sockaddr_in);
131
132 if (sock == RPC_ANYSOCK) {
133 if ((sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0) {
134 perror("svctcp_.c - udp socket creation problem");
135 return ((SVCXPRT *)NULL);
136 }
137 madesock = TRUE;
138 }
139 bzero((char *)&addr, sizeof (addr));
140 addr.sin_family = AF_INET;
141 if (bindresvport(sock, &addr)) {
142 addr.sin_port = 0;
143 (void)bind(sock, (struct sockaddr *)&addr, len);
144 }
145 if ((getsockname(sock, (struct sockaddr *)&addr, &len) != 0) ||
146 (listen(sock, 2) != 0)) {
147 perror("svctcp_.c - cannot getsockname or listen");
148 if (madesock)
149 (void)close(sock);
150 return ((SVCXPRT *)NULL);
151 }
152 r = (struct tcp_rendezvous *)mem_alloc(sizeof(*r));
153 if (r == NULL) {
154 (void) fprintf(stderr, "svctcp_create: out of memory\n");
155 return (NULL);
156 }
157 r->sendsize = sendsize;
158 r->recvsize = recvsize;
159 xprt = (SVCXPRT *)mem_alloc(sizeof(SVCXPRT));
160 if (xprt == NULL) {
161 (void) fprintf(stderr, "svctcp_create: out of memory\n");
162 return (NULL);
163 }
164 xprt->xp_p2 = NULL;
165 xprt->xp_p1 = (caddr_t)r;
166 xprt->xp_verf = _null_auth;
167 xprt->xp_ops = &svctcp_rendezvous_op;
168 xprt->xp_port = ntohs(addr.sin_port);
169 xprt->xp_sock = sock;
170 xprt_register(xprt);
171 return (xprt);
172}
173
174/*
175 * Like svtcp_create(), except the routine takes any *open* UNIX file
176 * descriptor as its first input.
177 */
178SVCXPRT *
179svcfd_create(fd, sendsize, recvsize)
180 int fd;
181 u_int sendsize;
182 u_int recvsize;
183{
184
185 return (makefd_xprt(fd, sendsize, recvsize));
186}
187
188static SVCXPRT *
189makefd_xprt(fd, sendsize, recvsize)
190 int fd;
191 u_int sendsize;
192 u_int recvsize;
193{
194 register SVCXPRT *xprt;
195 register struct tcp_conn *cd;
196
197 xprt = (SVCXPRT *)mem_alloc(sizeof(SVCXPRT));
198 if (xprt == (SVCXPRT *)NULL) {
199 (void) fprintf(stderr, "svc_tcp: makefd_xprt: out of memory\n");
200 goto done;
201 }
202 cd = (struct tcp_conn *)mem_alloc(sizeof(struct tcp_conn));
203 if (cd == (struct tcp_conn *)NULL) {
204 (void) fprintf(stderr, "svc_tcp: makefd_xprt: out of memory\n");
205 mem_free((char *) xprt, sizeof(SVCXPRT));
206 xprt = (SVCXPRT *)NULL;
207 goto done;
208 }
209 cd->strm_stat = XPRT_IDLE;
210 xdrrec_create(&(cd->xdrs), sendsize, recvsize,
211 (caddr_t)xprt, readtcp, writetcp);
212 xprt->xp_p2 = NULL;
213 xprt->xp_p1 = (caddr_t)cd;
214 xprt->xp_verf.oa_base = cd->verf_body;
215 xprt->xp_addrlen = 0;
216 xprt->xp_ops = &svctcp_op; /* truely deals with calls */
217 xprt->xp_port = 0; /* this is a connection, not a rendezvouser */
218 xprt->xp_sock = fd;
219 xprt_register(xprt);
220 done:
221 return (xprt);
222}
223
224static bool_t
225rendezvous_request(xprt)
226 register SVCXPRT *xprt;
227{
228 int sock;
229 struct tcp_rendezvous *r;
230 struct sockaddr_in addr;
231 int len;
232
233 r = (struct tcp_rendezvous *)xprt->xp_p1;
234 again:
235 len = sizeof(struct sockaddr_in);
236 if ((sock = accept(xprt->xp_sock, (struct sockaddr *)&addr,
237 &len)) < 0) {
238 if (errno == EINTR)
239 goto again;
240 return (FALSE);
241 }
242 /*
243 * make a new transporter (re-uses xprt)
244 */
245 xprt = makefd_xprt(sock, r->sendsize, r->recvsize);
246 xprt->xp_raddr = addr;
247 xprt->xp_addrlen = len;
248 return (FALSE); /* there is never an rpc msg to be processed */
249}
250
251static enum xprt_stat
252rendezvous_stat()
253{
254
255 return (XPRT_IDLE);
256}
257
258static void
259svctcp_destroy(xprt)
260 register SVCXPRT *xprt;
261{
262 register struct tcp_conn *cd = (struct tcp_conn *)xprt->xp_p1;
263
264 xprt_unregister(xprt);
265 (void)close(xprt->xp_sock);
266 if (xprt->xp_port != 0) {
267 /* a rendezvouser socket */
268 xprt->xp_port = 0;
269 } else {
270 /* an actual connection socket */
271 XDR_DESTROY(&(cd->xdrs));
272 }
273 mem_free((caddr_t)cd, sizeof(struct tcp_conn));
274 mem_free((caddr_t)xprt, sizeof(SVCXPRT));
275}
276
277/*
278 * All read operations timeout after 35 seconds.
279 * A timeout is fatal for the connection.
280 */
281static struct timeval wait_per_try = { 35, 0 };
282
283/*
284 * reads data from the tcp conection.
285 * any error is fatal and the connection is closed.
286 * (And a read of zero bytes is a half closed stream => error.)
287 */
288static int
289readtcp(xprt, buf, len)
290 register SVCXPRT *xprt;
291 caddr_t buf;
292 register int len;
293{
294 register int sock = xprt->xp_sock;
295#ifdef FD_SETSIZE
296 fd_set mask;
297 fd_set readfds;
298
299 FD_ZERO(&mask);
300 FD_SET(sock, &mask);
301#else
302 register int mask = 1 << sock;
303 int readfds;
304#endif /* def FD_SETSIZE */
305 do {
306 readfds = mask;
307 if (select(_rpc_dtablesize(), &readfds, (int*)NULL, (int*)NULL,
308 &wait_per_try) <= 0) {
309 if (errno == EINTR) {
310 continue;
311 }
312 goto fatal_err;
313 }
314#ifdef FD_SETSIZE
315 } while (!FD_ISSET(sock, &readfds));
316#else
317 } while (readfds != mask);
318#endif /* def FD_SETSIZE */
319 if ((len = read(sock, buf, len)) > 0) {
320 return (len);
321 }
322fatal_err:
323 ((struct tcp_conn *)(xprt->xp_p1))->strm_stat = XPRT_DIED;
324 return (-1);
325}
326
327/*
328 * writes data to the tcp connection.
329 * Any error is fatal and the connection is closed.
330 */
331static int
332writetcp(xprt, buf, len)
333 register SVCXPRT *xprt;
334 caddr_t buf;
335 int len;
336{
337 register int i, cnt;
338
339 for (cnt = len; cnt > 0; cnt -= i, buf += i) {
340 if ((i = write(xprt->xp_sock, buf, cnt)) < 0) {
341 ((struct tcp_conn *)(xprt->xp_p1))->strm_stat =
342 XPRT_DIED;
343 return (-1);
344 }
345 }
346 return (len);
347}
348
349static enum xprt_stat
350svctcp_stat(xprt)
351 SVCXPRT *xprt;
352{
353 register struct tcp_conn *cd =
354 (struct tcp_conn *)(xprt->xp_p1);
355
356 if (cd->strm_stat == XPRT_DIED)
357 return (XPRT_DIED);
358 if (! xdrrec_eof(&(cd->xdrs)))
359 return (XPRT_MOREREQS);
360 return (XPRT_IDLE);
361}
362
363static bool_t
364svctcp_recv(xprt, msg)
365 SVCXPRT *xprt;
366 register struct rpc_msg *msg;
367{
368 register struct tcp_conn *cd =
369 (struct tcp_conn *)(xprt->xp_p1);
370 register XDR *xdrs = &(cd->xdrs);
371
372 xdrs->x_op = XDR_DECODE;
373 (void)xdrrec_skiprecord(xdrs);
374 if (xdr_callmsg(xdrs, msg)) {
375 cd->x_id = msg->rm_xid;
376 return (TRUE);
377 }
378 return (FALSE);
379}
380
381static bool_t
382svctcp_getargs(xprt, xdr_args, args_ptr)
383 SVCXPRT *xprt;
384 xdrproc_t xdr_args;
385 caddr_t args_ptr;
386{
387
388 return ((*xdr_args)(&(((struct tcp_conn *)(xprt->xp_p1))->xdrs), args_ptr));
389}
390
391static bool_t
392svctcp_freeargs(xprt, xdr_args, args_ptr)
393 SVCXPRT *xprt;
394 xdrproc_t xdr_args;
395 caddr_t args_ptr;
396{
397 register XDR *xdrs =
398 &(((struct tcp_conn *)(xprt->xp_p1))->xdrs);
399
400 xdrs->x_op = XDR_FREE;
401 return ((*xdr_args)(xdrs, args_ptr));
402}
403
404static bool_t
405svctcp_reply(xprt, msg)
406 SVCXPRT *xprt;
407 register struct rpc_msg *msg;
408{
409 register struct tcp_conn *cd =
410 (struct tcp_conn *)(xprt->xp_p1);
411 register XDR *xdrs = &(cd->xdrs);
412 register bool_t stat;
413
414 xdrs->x_op = XDR_ENCODE;
415 msg->rm_xid = cd->x_id;
416 stat = xdr_replymsg(xdrs, msg);
417 (void)xdrrec_endofrecord(xdrs, TRUE);
418 return (stat);
419}