svc_simple.c revision 1.7
1/*
2 * Sun RPC is a product of Sun Microsystems, Inc. and is provided for
3 * unrestricted use provided that this legend is included on all tape
4 * media and as a part of the software program in whole or part.  Users
5 * may copy or modify Sun RPC without charge, but are not authorized
6 * to license or distribute it to anyone else except as part of a product or
7 * program developed by the user.
8 *
9 * SUN RPC IS PROVIDED AS IS WITH NO WARRANTIES OF ANY KIND INCLUDING THE
10 * WARRANTIES OF DESIGN, MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
11 * PURPOSE, OR ARISING FROM A COURSE OF DEALING, USAGE OR TRADE PRACTICE.
12 *
13 * Sun RPC is provided with no support and without any obligation on the
14 * part of Sun Microsystems, Inc. to assist in its use, correction,
15 * modification or enhancement.
16 *
17 * SUN MICROSYSTEMS, INC. SHALL HAVE NO LIABILITY WITH RESPECT TO THE
18 * INFRINGEMENT OF COPYRIGHTS, TRADE SECRETS OR ANY PATENTS BY SUN RPC
19 * OR ANY PART THEREOF.
20 *
21 * In no event will Sun Microsystems, Inc. be liable for any lost revenue
22 * or profits or other special, indirect and consequential damages, even if
23 * Sun has been advised of the possibility of such damages.
24 *
25 * Sun Microsystems, Inc.
26 * 2550 Garcia Avenue
27 * Mountain View, California  94043
28 */
29
30#if defined(LIBC_SCCS) && !defined(lint)
31static char *rcsid = "$OpenBSD: svc_simple.c,v 1.7 2001/03/03 06:50:28 deraadt Exp $";
32#endif /* LIBC_SCCS and not lint */
33
34/*
35 * svc_simple.c
36 * Simplified front end to rpc.
37 *
38 * Copyright (C) 1984, Sun Microsystems, Inc.
39 */
40
41#include <stdio.h>
42#include <stdlib.h>
43#include <string.h>
44#include <rpc/rpc.h>
45#include <rpc/pmap_clnt.h>
46#include <sys/socket.h>
47#include <netdb.h>
48
49static struct proglst {
50	char *(*p_progname)();
51	int  p_prognum;
52	int  p_procnum;
53	xdrproc_t p_inproc, p_outproc;
54	struct proglst *p_nxt;
55} *proglst;
56static void universal();
57static SVCXPRT *transp;
58struct proglst *pl;
59
60int
61registerrpc(prognum, versnum, procnum, progname, inproc, outproc)
62	int prognum, versnum, procnum;
63	char *(*progname)();
64	xdrproc_t inproc, outproc;
65{
66
67	if (procnum == NULLPROC) {
68		(void) fprintf(stderr,
69		    "can't reassign procedure number %u\n", NULLPROC);
70		return (-1);
71	}
72	if (transp == NULL) {
73		transp = svcudp_create(RPC_ANYSOCK);
74		if (transp == NULL) {
75			(void) fprintf(stderr, "couldn't create an rpc server\n");
76			return (-1);
77		}
78	}
79	(void) pmap_unset((u_long)prognum, (u_long)versnum);
80	if (!svc_register(transp, (u_long)prognum, (u_long)versnum,
81	    universal, IPPROTO_UDP)) {
82	    	(void) fprintf(stderr, "couldn't register prog %d vers %d\n",
83		    prognum, versnum);
84		return (-1);
85	}
86	pl = (struct proglst *)malloc(sizeof(struct proglst));
87	if (pl == NULL) {
88		(void) fprintf(stderr, "registerrpc: out of memory\n");
89		return (-1);
90	}
91	pl->p_progname = progname;
92	pl->p_prognum = prognum;
93	pl->p_procnum = procnum;
94	pl->p_inproc = inproc;
95	pl->p_outproc = outproc;
96	pl->p_nxt = proglst;
97	proglst = pl;
98	return (0);
99}
100
101static void
102universal(rqstp, transp)
103	struct svc_req *rqstp;
104	SVCXPRT *transp;
105{
106	int prog, proc;
107	char *outdata;
108	char xdrbuf[UDPMSGSIZE];
109	struct proglst *pl;
110
111	/*
112	 * enforce "procnum 0 is echo" convention
113	 */
114	if (rqstp->rq_proc == NULLPROC) {
115		if (svc_sendreply(transp, xdr_void, NULL) == FALSE) {
116			(void) fprintf(stderr, "xxx\n");
117			exit(1);
118		}
119		return;
120	}
121	prog = rqstp->rq_prog;
122	proc = rqstp->rq_proc;
123	for (pl = proglst; pl != NULL; pl = pl->p_nxt)
124		if (pl->p_prognum == prog && pl->p_procnum == proc) {
125			/* decode arguments into a CLEAN buffer */
126			memset(xdrbuf, 0, sizeof(xdrbuf)); /* required ! */
127			if (!svc_getargs(transp, pl->p_inproc, xdrbuf)) {
128				svcerr_decode(transp);
129				return;
130			}
131			outdata = (*(pl->p_progname))(xdrbuf);
132			if (outdata == NULL &&
133			    pl->p_outproc != xdr_void)
134				/* there was an error */
135				return;
136			if (!svc_sendreply(transp, pl->p_outproc, outdata)) {
137				(void) fprintf(stderr,
138				    "trouble replying to prog %d\n",
139				    pl->p_prognum);
140				exit(1);
141			}
142			/* free the decoded arguments */
143			(void)svc_freeargs(transp, pl->p_inproc, xdrbuf);
144			return;
145		}
146	(void) fprintf(stderr, "never registered prog %d\n", prog);
147	exit(1);
148}
149
150