xref: /netbsd-src/usr.bin/rpcgen/rpc_sample.c (revision 5e4c038a45edbc7d63b7c2daa76e29f88b64a4e3)
1 /*	$NetBSD: rpc_sample.c,v 1.9 2002/01/31 19:36:49 tv 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 or with the express written consent of
9  * Sun Microsystems, Inc.
10  *
11  * SUN RPC IS PROVIDED AS IS WITH NO WARRANTIES OF ANY KIND INCLUDING THE
12  * WARRANTIES OF DESIGN, MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
13  * PURPOSE, OR ARISING FROM A COURSE OF DEALING, USAGE OR TRADE PRACTICE.
14  *
15  * Sun RPC is provided with no support and without any obligation on the
16  * part of Sun Microsystems, Inc. to assist in its use, correction,
17  * modification or enhancement.
18  *
19  * SUN MICROSYSTEMS, INC. SHALL HAVE NO LIABILITY WITH RESPECT TO THE
20  * INFRINGEMENT OF COPYRIGHTS, TRADE SECRETS OR ANY PATENTS BY SUN RPC
21  * OR ANY PART THEREOF.
22  *
23  * In no event will Sun Microsystems, Inc. be liable for any lost revenue
24  * or profits or other special, indirect and consequential damages, even if
25  * Sun has been advised of the possibility of such damages.
26  *
27  * Sun Microsystems, Inc.
28  * 2550 Garcia Avenue
29  * Mountain View, California  94043
30  */
31 
32 #include <sys/cdefs.h>
33 #if defined(__RCSID) && !defined(lint)
34 #if 0
35 static char sccsid[] = "@(#)rpc_sample.c  1.1  90/08/30  (C) 1987 SMI";
36 #else
37 __RCSID("$NetBSD: rpc_sample.c,v 1.9 2002/01/31 19:36:49 tv Exp $");
38 #endif
39 #endif
40 
41 /*
42  * rpc_sample.c, Sample client-server code outputter for the RPC protocol compiler
43  */
44 
45 #include <stdio.h>
46 #include <string.h>
47 #include "rpc_scan.h"
48 #include "rpc_parse.h"
49 #include "rpc_util.h"
50 
51 static char RQSTP[] = "rqstp";
52 
53 static void write_sample_client __P((char *, version_list *));
54 static void write_sample_server __P((definition *));
55 static void return_type __P((proc_list *));
56 
57 void
58 write_sample_svc(def)
59 	definition *def;
60 {
61 
62 	if (def->def_kind != DEF_PROGRAM)
63 		return;
64 	write_sample_server(def);
65 }
66 
67 
68 int
69 write_sample_clnt(def)
70 	definition *def;
71 {
72 	version_list *vp;
73 	int     count = 0;
74 
75 	if (def->def_kind != DEF_PROGRAM)
76 		return (0);
77 	/* generate sample code for each version */
78 	for (vp = def->def.pr.versions; vp != NULL; vp = vp->next) {
79 		write_sample_client(def->def_name, vp);
80 		++count;
81 	}
82 	return (count);
83 }
84 
85 
86 static void
87 write_sample_client(program_name, vp)
88 	char   *program_name;
89 	version_list *vp;
90 {
91 	proc_list *proc;
92 	decl_list *l;
93 
94 	f_print(fout, "\n\nvoid\n");
95 	pvname(program_name, vp->vers_num);
96 	if (Cflag)
97 		f_print(fout, "(char *host)\n{\n");
98 	else
99 		f_print(fout, "(host)\n\tchar *host;\n{\n");
100 	f_print(fout, "\tCLIENT *clnt;\n");
101 
102 	for (proc = vp->procs; proc != NULL; proc = proc->next) {
103 		/* print out declarations for arguments */
104 		if (proc->arg_num < 2 && !newstyle) {
105 			f_print(fout, "\t");
106 			if (streq(proc->args.decls->decl.type, "void"))
107 				f_print(fout, "char ");	/* cannot have "void"
108 							 * type */
109 			else
110 				ptype(proc->args.decls->decl.prefix, proc->args.decls->decl.type, 1);
111 			pvname(proc->proc_name, vp->vers_num);
112 			f_print(fout, "_arg;\n");
113 		} else {
114 			if (!streq(proc->args.decls->decl.type, "void")) {
115 				for (l = proc->args.decls; l != NULL; l = l->next) {
116 					f_print(fout, "\t");
117 					ptype(l->decl.prefix, l->decl.type, 1);
118 					pvname(proc->proc_name, vp->vers_num);
119 					f_print(fout, "_%s;\n", l->decl.name);
120 /*	  pdeclaration(proc->args.argname, &l->decl, 1, ";\n" );*/
121 				}
122 			}
123 		}
124 		/* print out declarations for results */
125 		f_print(fout, "\t");
126 		if (streq(proc->res_type, "void"))
127 			f_print(fout, "char");	/* cannot have "void"
128 						 * type */
129 		else
130 			ptype(proc->res_prefix, proc->res_type, 1);
131 		if (!Mflag)
132 			f_print(fout, "*");
133 		pvname(proc->proc_name, vp->vers_num);
134 		f_print(fout, "_res;\n");
135 	}
136 	f_print(fout, "\n");
137 
138 	/* generate creation of client handle */
139 	f_print(fout, "\tclnt = clnt_create(host, %s, %s, \"%s\");\n",
140 	    program_name, vp->vers_name, tirpcflag ? "netpath" : "udp");
141 	f_print(fout, "\tif (clnt == NULL) {\n");
142 	f_print(fout, "\t\tclnt_pcreateerror(host);\n");
143 	f_print(fout, "\t\texit(1);\n\t}\n");
144 
145 	/* generate calls to procedures */
146 	for (proc = vp->procs; proc != NULL; proc = proc->next) {
147 		if (Mflag)
148 			f_print(fout, "\tif (");
149 		else {
150 			f_print(fout, "\t");
151 			pvname(proc->proc_name, vp->vers_num);
152 			f_print(fout, "_res = ");
153 		}
154 		pvname(proc->proc_name, vp->vers_num);
155 		if (proc->arg_num < 2 && !newstyle) {
156 			f_print(fout, "(");
157 			if (streq(proc->args.decls->decl.type, "void"))	/* cast to void* */
158 				f_print(fout, "(void *)");
159 			f_print(fout, "&");
160 			pvname(proc->proc_name, vp->vers_num);
161 			f_print(fout, "_arg, ");
162 		} else {
163 			if (streq(proc->args.decls->decl.type, "void")) {
164 				f_print(fout, "(clnt);\n");
165 			} else {
166 				f_print(fout, "(");
167 				for (l = proc->args.decls; l != NULL; l = l->next) {
168 					pvname(proc->proc_name, vp->vers_num);
169 					f_print(fout, "_%s, ", l->decl.name);
170 				}
171 			}
172 		}
173 		if (Mflag) {
174 			f_print(fout, "&");
175 			pvname(proc->proc_name, vp->vers_num);
176 			f_print(fout, "_res, clnt) != RPC_SUCCESS)\n");
177 		} else {
178 			f_print(fout, "clnt);\n");
179 			f_print(fout, "\tif (");
180 			pvname(proc->proc_name, vp->vers_num);
181 			f_print(fout, "_res == NULL)\n");
182 		}
183 		f_print(fout, "\t\tclnt_perror(clnt, \"call failed\");\n");
184 	}
185 
186 	f_print(fout, "\tclnt_destroy(clnt);\n");
187 	f_print(fout, "}\n");
188 }
189 
190 static void
191 write_sample_server(def)
192 	definition *def;
193 {
194 	version_list *vp;
195 	proc_list *proc;
196 
197 	for (vp = def->def.pr.versions; vp != NULL; vp = vp->next) {
198 		for (proc = vp->procs; proc != NULL; proc = proc->next) {
199 			f_print(fout, "\n");
200 			if (Mflag)
201 				f_print(fout, "bool_t\n");
202 			else {
203 				return_type(proc);
204 				f_print(fout, "*\n");
205 			}
206 			if (Cflag)
207 				pvname_svc(proc->proc_name, vp->vers_num);
208 			else
209 				pvname(proc->proc_name, vp->vers_num);
210 			printarglist(proc, "result", RQSTP, "struct svc_req *");
211 
212 			f_print(fout, "{\n");
213 			if (Mflag) {
214 				f_print(fout, "\tbool_t retval = TRUE;\n");
215 			} else {
216 				f_print(fout, "\tstatic ");
217 				if (streq(proc->res_type, "void"))
218 					f_print(fout, "char ");	/* cannot have void type */
219 				else
220 					return_type(proc);
221 				f_print(fout, "result;\n");
222 			}
223 			f_print(fout,
224 			    "\n\t/*\n\t * insert server code here\n\t */\n\n");
225 			if (Mflag) {
226 				f_print(fout, "\treturn (retval);\n");
227 			} else {
228 				if (streq(proc->res_type, "void"))
229 					f_print(fout, "\treturn ((void *)&result);\n");
230 				else
231 					f_print(fout, "\treturn (&result);\n");
232 			}
233 			f_print(fout, "}\n");
234 		}
235 	}
236 }
237 
238 static void
239 return_type(plist)
240 	proc_list *plist;
241 {
242 	ptype(plist->res_prefix, plist->res_type, 1);
243 }
244 
245 void
246 add_sample_msg()
247 {
248 	f_print(fout, "/*\n");
249 	f_print(fout, " * This is sample code generated by rpcgen.\n");
250 	f_print(fout, " * These are only templates and you can use them\n");
251 	f_print(fout, " * as a guideline for developing your own functions.\n");
252 	f_print(fout, " */\n\n");
253 }
254 
255 void
256 write_sample_clnt_main()
257 {
258 	list   *l;
259 	definition *def;
260 	version_list *vp;
261 
262 	f_print(fout, "\n\n");
263 	if (Cflag)
264 		f_print(fout, "int\nmain(int argc, char *argv[])\n{\n");
265 	else
266 		f_print(fout, "int\nmain(argc, argv)\n\tint argc;\n\tchar *argv[];\n{\n");
267 
268 	f_print(fout, "\tchar *host;");
269 	f_print(fout, "\n\n\tif (argc < 2) {");
270 	f_print(fout, "\n\t\tprintf(\"usage: %%s server_host\\n\", argv[0]);\n");
271 	f_print(fout, "\t\texit(1);\n\t}");
272 	f_print(fout, "\n\thost = argv[1];\n");
273 
274 	for (l = defined; l != NULL; l = l->next) {
275 		def = l->val;
276 		if (def->def_kind != DEF_PROGRAM) {
277 			continue;
278 		}
279 		for (vp = def->def.pr.versions; vp != NULL; vp = vp->next) {
280 			f_print(fout, "\t");
281 			pvname(def->def_name, vp->vers_num);
282 			f_print(fout, "(host);\n");
283 		}
284 	}
285 	f_print(fout, "\texit(0);\n");
286 	f_print(fout, "}\n");
287 }
288