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