xref: /dragonfly/lib/libc/rpc/svc_simple.c (revision c03f08f3)
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.3 2005/11/13 12:27:04 swildner 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(int prognum, int versnum, int procnum, char *(*progname)(),
63 	    xdrproc_t inproc, xdrproc_t outproc)
64 {
65 
66 	if (procnum == NULLPROC) {
67 		fprintf(stderr,
68 		    "can't reassign procedure number %ld\n", NULLPROC);
69 		return (-1);
70 	}
71 	if (transp == 0) {
72 		transp = svcudp_create(RPC_ANYSOCK);
73 		if (transp == NULL) {
74 			fprintf(stderr, "couldn't create an rpc server\n");
75 			return (-1);
76 		}
77 	}
78 	pmap_unset((u_long)prognum, (u_long)versnum);
79 	if (!svc_register(transp, (u_long)prognum, (u_long)versnum,
80 	    universal, IPPROTO_UDP)) {
81 	    	fprintf(stderr, "couldn't register prog %d vers %d\n",
82 		    prognum, versnum);
83 		return (-1);
84 	}
85 	pl = (struct proglst *)malloc(sizeof(struct proglst));
86 	if (pl == NULL) {
87 		fprintf(stderr, "registerrpc: out of memory\n");
88 		return (-1);
89 	}
90 	pl->p_progname = progname;
91 	pl->p_prognum = prognum;
92 	pl->p_procnum = procnum;
93 	pl->p_inproc = inproc;
94 	pl->p_outproc = outproc;
95 	pl->p_nxt = proglst;
96 	proglst = pl;
97 	return (0);
98 }
99 
100 static void
101 universal(struct svc_req *rqstp, SVCXPRT *transp)
102 {
103 	int prog, proc;
104 	char *outdata;
105 	char xdrbuf[UDPMSGSIZE];
106 	struct proglst *pl;
107 
108 	/*
109 	 * enforce "procnum 0 is echo" convention
110 	 */
111 	if (rqstp->rq_proc == NULLPROC) {
112 		if (svc_sendreply(transp, xdr_void, NULL) == FALSE) {
113 			fprintf(stderr, "xxx\n");
114 			exit(1);
115 		}
116 		return;
117 	}
118 	prog = rqstp->rq_prog;
119 	proc = rqstp->rq_proc;
120 	for (pl = proglst; pl != NULL; pl = pl->p_nxt)
121 		if (pl->p_prognum == prog && pl->p_procnum == proc) {
122 			/* decode arguments into a CLEAN buffer */
123 			memset(xdrbuf, 0, sizeof(xdrbuf)); /* required ! */
124 			if (!svc_getargs(transp, pl->p_inproc, xdrbuf)) {
125 				svcerr_decode(transp);
126 				return;
127 			}
128 			outdata = (*(pl->p_progname))(xdrbuf);
129 			if (outdata == NULL && pl->p_outproc != xdr_void)
130 				/* there was an error */
131 				return;
132 			if (!svc_sendreply(transp, pl->p_outproc, outdata)) {
133 				fprintf(stderr,
134 				    "trouble replying to prog %d\n",
135 				    pl->p_prognum);
136 				exit(1);
137 			}
138 			/* free the decoded arguments */
139 			svc_freeargs(transp, pl->p_inproc, xdrbuf);
140 			return;
141 		}
142 	fprintf(stderr, "never registered prog %d\n", prog);
143 	exit(1);
144 }
145 
146