rpc_sample.c revision 1.6 1 /* $NetBSD: rpc_sample.c,v 1.6 2001/03/21 00:30:39 mycroft 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 #ifndef 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.6 2001/03/21 00:30:39 mycroft 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 int i;
93 decl_list *l;
94
95 f_print(fout, "\n\nvoid\n");
96 pvname(program_name, vp->vers_num);
97 if (Cflag)
98 f_print(fout, "(char *host)\n{\n");
99 else
100 f_print(fout, "(host)\n\tchar *host;\n{\n");
101 f_print(fout, "\tCLIENT *clnt;\n");
102
103 i = 0;
104 for (proc = vp->procs; proc != NULL; proc = proc->next) {
105 /* print out declarations for arguments */
106 if (proc->arg_num < 2 && !newstyle) {
107 f_print(fout, "\t");
108 if (streq(proc->args.decls->decl.type, "void"))
109 f_print(fout, "char"); /* cannot have "void"
110 * type */
111 else
112 ptype(proc->args.decls->decl.prefix, proc->args.decls->decl.type, 1);
113 pvname(proc->proc_name, vp->vers_num);
114 f_print(fout, "_arg;\n");
115 } else {
116 if (!streq(proc->args.decls->decl.type, "void")) {
117 for (l = proc->args.decls; l != NULL; l = l->next) {
118 f_print(fout, "\t");
119 ptype(l->decl.prefix, l->decl.type, 1);
120 f_print(fout, " ");
121 pvname(proc->proc_name, vp->vers_num);
122 f_print(fout, "_%s;\n", l->decl.name);
123 /* pdeclaration(proc->args.argname, &l->decl, 1, ";\n" );*/
124 }
125 }
126 }
127 /* print out declarations for results */
128 f_print(fout, "\t");
129 if (streq(proc->res_type, "void"))
130 f_print(fout, "char"); /* cannot have "void"
131 * type */
132 else
133 ptype(proc->res_prefix, proc->res_type, 1);
134 if (!Mflag)
135 f_print(fout, "*");
136 pvname(proc->proc_name, vp->vers_num);
137 f_print(fout, "_res;\n");
138 }
139 f_print(fout, "\n");
140
141 /* generate creation of client handle */
142 f_print(fout, "\tclnt = clnt_create(host, %s, %s, \"%s\");\n",
143 program_name, vp->vers_name, tirpcflag ? "netpath" : "udp");
144 f_print(fout, "\tif (clnt == NULL) {\n");
145 f_print(fout, "\t\tclnt_pcreateerror(host);\n");
146 f_print(fout, "\t\texit(1);\n\t}\n");
147
148 /* generate calls to procedures */
149 for (proc = vp->procs; proc != NULL; proc = proc->next) {
150 if (Mflag)
151 f_print(fout, "\tif (");
152 else {
153 f_print(fout, "\t");
154 pvname(proc->proc_name, vp->vers_num);
155 f_print(fout, "_res = ");
156 }
157 pvname(proc->proc_name, vp->vers_num);
158 if (proc->arg_num < 2 && !newstyle) {
159 f_print(fout, "(");
160 if (streq(proc->args.decls->decl.type, "void")) /* cast to void* */
161 f_print(fout, "(void *)");
162 f_print(fout, "&");
163 pvname(proc->proc_name, vp->vers_num);
164 f_print(fout, "_arg, ");
165 } else {
166 if (streq(proc->args.decls->decl.type, "void")) {
167 f_print(fout, "(clnt);\n");
168 } else {
169 f_print(fout, "(");
170 for (l = proc->args.decls; l != NULL; l = l->next) {
171 pvname(proc->proc_name, vp->vers_num);
172 f_print(fout, "_%s, ", l->decl.name);
173 }
174 }
175 }
176 if (Mflag) {
177 f_print(fout, "&");
178 pvname(proc->proc_name, vp->vers_num);
179 f_print(fout, "_res, clnt) != RPC_SUCCESS)\n");
180 } else {
181 f_print(fout, "clnt);\n");
182 f_print(fout, "\tif (");
183 pvname(proc->proc_name, vp->vers_num);
184 f_print(fout, "_res == NULL)\n", i);
185 }
186 f_print(fout, "\t\tclnt_perror(clnt, \"call failed:\");\n");
187 }
188
189 f_print(fout, "\tclnt_destroy(clnt);\n");
190 f_print(fout, "}\n");
191 }
192
193 static void
194 write_sample_server(def)
195 definition *def;
196 {
197 version_list *vp;
198 proc_list *proc;
199
200 for (vp = def->def.pr.versions; vp != NULL; vp = vp->next) {
201 for (proc = vp->procs; proc != NULL; proc = proc->next) {
202 f_print(fout, "\n");
203 if (Mflag)
204 f_print(fout, "bool_t\n");
205 else {
206 return_type(proc);
207 f_print(fout, "*\n");
208 }
209 if (Cflag)
210 pvname_svc(proc->proc_name, vp->vers_num);
211 else
212 pvname(proc->proc_name, vp->vers_num);
213 printarglist(proc, "result", RQSTP, "struct svc_req *");
214
215 f_print(fout, "{\n");
216 if (Mflag) {
217 f_print(fout, "\tbool_t retval;\n");
218 } else {
219 f_print(fout, "\tstatic ");
220 if (streq(proc->res_type, "void"))
221 f_print(fout, "char"); /* cannot have void type */
222 else
223 return_type(proc);
224 f_print(fout, " result;\n");
225 }
226 f_print(fout,
227 "\n\t/*\n\t * insert server code here\n\t */\n\n");
228 if (Mflag) {
229 f_print(fout, "\treturn (retval);\n");
230 } else {
231 if (streq(proc->res_type, "void"))
232 f_print(fout, "\treturn ((void *)&result);\n");
233 else
234 f_print(fout, "\treturn (&result);\n");
235 }
236 f_print(fout, "}\n");
237 }
238 }
239 }
240
241 static void
242 return_type(plist)
243 proc_list *plist;
244 {
245 ptype(plist->res_prefix, plist->res_type, 1);
246 }
247
248 void
249 add_sample_msg()
250 {
251 f_print(fout, "/*\n");
252 f_print(fout, " * This is sample code generated by rpcgen.\n");
253 f_print(fout, " * These are only templates and you can use them\n");
254 f_print(fout, " * as a guideline for developing your own functions.\n");
255 f_print(fout, " */\n\n");
256 }
257
258 void
259 write_sample_clnt_main()
260 {
261 list *l;
262 definition *def;
263 version_list *vp;
264
265 f_print(fout, "\n\n");
266 if (Cflag)
267 f_print(fout, "int\nmain(int argc, char *argv[])\n{\n");
268 else
269 f_print(fout, "int\nmain(argc, argv)\n\tint argc;\n\tchar *argv[];\n{\n");
270
271 f_print(fout, "\tchar *host;");
272 f_print(fout, "\n\n\tif (argc < 2) {");
273 f_print(fout, "\n\t\tprintf(\"usage: %%s server_host\\n\", argv[0]);\n");
274 f_print(fout, "\t\texit(1);\n\t}");
275 f_print(fout, "\n\thost = argv[1];\n");
276
277 for (l = defined; l != NULL; l = l->next) {
278 def = l->val;
279 if (def->def_kind != DEF_PROGRAM) {
280 continue;
281 }
282 for (vp = def->def.pr.versions; vp != NULL; vp = vp->next) {
283 f_print(fout, "\t");
284 pvname(def->def_name, vp->vers_num);
285 f_print(fout, "(host);\n");
286 }
287 }
288 f_print(fout, "\texit(0);\n");
289 f_print(fout, "}\n");
290 }
291