xref: /dragonfly/lib/libc/rpc/svc_simple.c (revision 2cd2d2b5)
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  * @(#)svc_simple.c 1.18 87/08/11 Copyr 1984 Sun Micro
30  * @(#)svc_simple.c	2.2 88/08/01 4.0 RPCSRC
31  * $FreeBSD: src/lib/libc/rpc/svc_simple.c,v 1.9 1999/08/28 00:00:50 peter Exp $
32  * $DragonFly: src/lib/libc/rpc/svc_simple.c,v 1.2 2003/06/17 04:26:45 dillon Exp $
33  */
34 
35 /*
36  * svc_simple.c
37  * Simplified front end to rpc.
38  *
39  * Copyright (C) 1984, Sun Microsystems, Inc.
40  */
41 
42 #include <stdio.h>
43 #include <stdlib.h>
44 #include <string.h>
45 #include <rpc/rpc.h>
46 #include <rpc/pmap_clnt.h>
47 #include <sys/socket.h>
48 #include <netdb.h>
49 
50 static struct proglst {
51 	char *(*p_progname)();
52 	int  p_prognum;
53 	int  p_procnum;
54 	xdrproc_t p_inproc, p_outproc;
55 	struct proglst *p_nxt;
56 } *proglst;
57 static void universal();
58 static SVCXPRT *transp;
59 struct proglst *pl;
60 
61 int
62 registerrpc(prognum, versnum, procnum, progname, inproc, outproc)
63 	int prognum, versnum, procnum;
64 	char *(*progname)();
65 	xdrproc_t inproc, outproc;
66 {
67 
68 	if (procnum == NULLPROC) {
69 		(void) fprintf(stderr,
70 		    "can't reassign procedure number %ld\n", NULLPROC);
71 		return (-1);
72 	}
73 	if (transp == 0) {
74 		transp = svcudp_create(RPC_ANYSOCK);
75 		if (transp == NULL) {
76 			(void) fprintf(stderr, "couldn't create an rpc server\n");
77 			return (-1);
78 		}
79 	}
80 	(void) pmap_unset((u_long)prognum, (u_long)versnum);
81 	if (!svc_register(transp, (u_long)prognum, (u_long)versnum,
82 	    universal, IPPROTO_UDP)) {
83 	    	(void) fprintf(stderr, "couldn't register prog %d vers %d\n",
84 		    prognum, versnum);
85 		return (-1);
86 	}
87 	pl = (struct proglst *)malloc(sizeof(struct proglst));
88 	if (pl == NULL) {
89 		(void) fprintf(stderr, "registerrpc: out of memory\n");
90 		return (-1);
91 	}
92 	pl->p_progname = progname;
93 	pl->p_prognum = prognum;
94 	pl->p_procnum = procnum;
95 	pl->p_inproc = inproc;
96 	pl->p_outproc = outproc;
97 	pl->p_nxt = proglst;
98 	proglst = pl;
99 	return (0);
100 }
101 
102 static void
103 universal(rqstp, transp)
104 	struct svc_req *rqstp;
105 	SVCXPRT *transp;
106 {
107 	int prog, proc;
108 	char *outdata;
109 	char xdrbuf[UDPMSGSIZE];
110 	struct proglst *pl;
111 
112 	/*
113 	 * enforce "procnum 0 is echo" convention
114 	 */
115 	if (rqstp->rq_proc == NULLPROC) {
116 		if (svc_sendreply(transp, xdr_void, NULL) == FALSE) {
117 			(void) fprintf(stderr, "xxx\n");
118 			exit(1);
119 		}
120 		return;
121 	}
122 	prog = rqstp->rq_prog;
123 	proc = rqstp->rq_proc;
124 	for (pl = proglst; pl != NULL; pl = pl->p_nxt)
125 		if (pl->p_prognum == prog && pl->p_procnum == proc) {
126 			/* decode arguments into a CLEAN buffer */
127 			memset(xdrbuf, 0, sizeof(xdrbuf)); /* required ! */
128 			if (!svc_getargs(transp, pl->p_inproc, xdrbuf)) {
129 				svcerr_decode(transp);
130 				return;
131 			}
132 			outdata = (*(pl->p_progname))(xdrbuf);
133 			if (outdata == NULL && 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