rpc_sample.c revision 1.10 1 /* $NetBSD: rpc_sample.c,v 1.10 2004/06/20 22:20:16 jmc 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 #if HAVE_NBTOOL_CONFIG_H
33 #include "nbtool_config.h"
34 #endif
35
36 #include <sys/cdefs.h>
37 #if defined(__RCSID) && !defined(lint)
38 #if 0
39 static char sccsid[] = "@(#)rpc_sample.c 1.1 90/08/30 (C) 1987 SMI";
40 #else
41 __RCSID("$NetBSD: rpc_sample.c,v 1.10 2004/06/20 22:20:16 jmc Exp $");
42 #endif
43 #endif
44
45 /*
46 * rpc_sample.c, Sample client-server code outputter for the RPC protocol compiler
47 */
48
49 #include <stdio.h>
50 #include <string.h>
51 #include "rpc_scan.h"
52 #include "rpc_parse.h"
53 #include "rpc_util.h"
54
55 static char RQSTP[] = "rqstp";
56
57 static void write_sample_client __P((char *, version_list *));
58 static void write_sample_server __P((definition *));
59 static void return_type __P((proc_list *));
60
61 void
62 write_sample_svc(def)
63 definition *def;
64 {
65
66 if (def->def_kind != DEF_PROGRAM)
67 return;
68 write_sample_server(def);
69 }
70
71
72 int
73 write_sample_clnt(def)
74 definition *def;
75 {
76 version_list *vp;
77 int count = 0;
78
79 if (def->def_kind != DEF_PROGRAM)
80 return (0);
81 /* generate sample code for each version */
82 for (vp = def->def.pr.versions; vp != NULL; vp = vp->next) {
83 write_sample_client(def->def_name, vp);
84 ++count;
85 }
86 return (count);
87 }
88
89
90 static void
91 write_sample_client(program_name, vp)
92 char *program_name;
93 version_list *vp;
94 {
95 proc_list *proc;
96 decl_list *l;
97
98 f_print(fout, "\n\nvoid\n");
99 pvname(program_name, vp->vers_num);
100 if (Cflag)
101 f_print(fout, "(char *host)\n{\n");
102 else
103 f_print(fout, "(host)\n\tchar *host;\n{\n");
104 f_print(fout, "\tCLIENT *clnt;\n");
105
106 for (proc = vp->procs; proc != NULL; proc = proc->next) {
107 /* print out declarations for arguments */
108 if (proc->arg_num < 2 && !newstyle) {
109 f_print(fout, "\t");
110 if (streq(proc->args.decls->decl.type, "void"))
111 f_print(fout, "char "); /* cannot have "void"
112 * type */
113 else
114 ptype(proc->args.decls->decl.prefix, proc->args.decls->decl.type, 1);
115 pvname(proc->proc_name, vp->vers_num);
116 f_print(fout, "_arg;\n");
117 } else {
118 if (!streq(proc->args.decls->decl.type, "void")) {
119 for (l = proc->args.decls; l != NULL; l = l->next) {
120 f_print(fout, "\t");
121 ptype(l->decl.prefix, l->decl.type, 1);
122 pvname(proc->proc_name, vp->vers_num);
123 f_print(fout, "_%s;\n", l->decl.name);
124 /* pdeclaration(proc->args.argname, &l->decl, 1, ";\n" );*/
125 }
126 }
127 }
128 /* print out declarations for results */
129 f_print(fout, "\t");
130 if (streq(proc->res_type, "void"))
131 f_print(fout, "char"); /* cannot have "void"
132 * type */
133 else
134 ptype(proc->res_prefix, proc->res_type, 1);
135 if (!Mflag)
136 f_print(fout, "*");
137 pvname(proc->proc_name, vp->vers_num);
138 f_print(fout, "_res;\n");
139 }
140 f_print(fout, "\n");
141
142 /* generate creation of client handle */
143 f_print(fout, "\tclnt = clnt_create(host, %s, %s, \"%s\");\n",
144 program_name, vp->vers_name, tirpcflag ? "netpath" : "udp");
145 f_print(fout, "\tif (clnt == NULL) {\n");
146 f_print(fout, "\t\tclnt_pcreateerror(host);\n");
147 f_print(fout, "\t\texit(1);\n\t}\n");
148
149 /* generate calls to procedures */
150 for (proc = vp->procs; proc != NULL; proc = proc->next) {
151 if (Mflag)
152 f_print(fout, "\tif (");
153 else {
154 f_print(fout, "\t");
155 pvname(proc->proc_name, vp->vers_num);
156 f_print(fout, "_res = ");
157 }
158 pvname(proc->proc_name, vp->vers_num);
159 if (proc->arg_num < 2 && !newstyle) {
160 f_print(fout, "(");
161 if (streq(proc->args.decls->decl.type, "void")) /* cast to void* */
162 f_print(fout, "(void *)");
163 f_print(fout, "&");
164 pvname(proc->proc_name, vp->vers_num);
165 f_print(fout, "_arg, ");
166 } else {
167 if (streq(proc->args.decls->decl.type, "void")) {
168 f_print(fout, "(clnt);\n");
169 } else {
170 f_print(fout, "(");
171 for (l = proc->args.decls; l != NULL; l = l->next) {
172 pvname(proc->proc_name, vp->vers_num);
173 f_print(fout, "_%s, ", l->decl.name);
174 }
175 }
176 }
177 if (Mflag) {
178 f_print(fout, "&");
179 pvname(proc->proc_name, vp->vers_num);
180 f_print(fout, "_res, clnt) != RPC_SUCCESS)\n");
181 } else {
182 f_print(fout, "clnt);\n");
183 f_print(fout, "\tif (");
184 pvname(proc->proc_name, vp->vers_num);
185 f_print(fout, "_res == NULL)\n");
186 }
187 f_print(fout, "\t\tclnt_perror(clnt, \"call failed\");\n");
188 }
189
190 f_print(fout, "\tclnt_destroy(clnt);\n");
191 f_print(fout, "}\n");
192 }
193
194 static void
195 write_sample_server(def)
196 definition *def;
197 {
198 version_list *vp;
199 proc_list *proc;
200
201 for (vp = def->def.pr.versions; vp != NULL; vp = vp->next) {
202 for (proc = vp->procs; proc != NULL; proc = proc->next) {
203 f_print(fout, "\n");
204 if (Mflag)
205 f_print(fout, "bool_t\n");
206 else {
207 return_type(proc);
208 f_print(fout, "*\n");
209 }
210 if (Cflag)
211 pvname_svc(proc->proc_name, vp->vers_num);
212 else
213 pvname(proc->proc_name, vp->vers_num);
214 printarglist(proc, "result", RQSTP, "struct svc_req *");
215
216 f_print(fout, "{\n");
217 if (Mflag) {
218 f_print(fout, "\tbool_t retval = TRUE;\n");
219 } else {
220 f_print(fout, "\tstatic ");
221 if (streq(proc->res_type, "void"))
222 f_print(fout, "char "); /* cannot have void type */
223 else
224 return_type(proc);
225 f_print(fout, "result;\n");
226 }
227 f_print(fout,
228 "\n\t/*\n\t * insert server code here\n\t */\n\n");
229 if (Mflag) {
230 f_print(fout, "\treturn (retval);\n");
231 } else {
232 if (streq(proc->res_type, "void"))
233 f_print(fout, "\treturn ((void *)&result);\n");
234 else
235 f_print(fout, "\treturn (&result);\n");
236 }
237 f_print(fout, "}\n");
238 }
239 }
240 }
241
242 static void
243 return_type(plist)
244 proc_list *plist;
245 {
246 ptype(plist->res_prefix, plist->res_type, 1);
247 }
248
249 void
250 add_sample_msg()
251 {
252 f_print(fout, "/*\n");
253 f_print(fout, " * This is sample code generated by rpcgen.\n");
254 f_print(fout, " * These are only templates and you can use them\n");
255 f_print(fout, " * as a guideline for developing your own functions.\n");
256 f_print(fout, " */\n\n");
257 }
258
259 void
260 write_sample_clnt_main()
261 {
262 list *l;
263 definition *def;
264 version_list *vp;
265
266 f_print(fout, "\n\n");
267 if (Cflag)
268 f_print(fout, "int\nmain(int argc, char *argv[])\n{\n");
269 else
270 f_print(fout, "int\nmain(argc, argv)\n\tint argc;\n\tchar *argv[];\n{\n");
271
272 f_print(fout, "\tchar *host;");
273 f_print(fout, "\n\n\tif (argc < 2) {");
274 f_print(fout, "\n\t\tprintf(\"usage: %%s server_host\\n\", argv[0]);\n");
275 f_print(fout, "\t\texit(1);\n\t}");
276 f_print(fout, "\n\thost = argv[1];\n");
277
278 for (l = defined; l != NULL; l = l->next) {
279 def = l->val;
280 if (def->def_kind != DEF_PROGRAM) {
281 continue;
282 }
283 for (vp = def->def.pr.versions; vp != NULL; vp = vp->next) {
284 f_print(fout, "\t");
285 pvname(def->def_name, vp->vers_num);
286 f_print(fout, "(host);\n");
287 }
288 }
289 f_print(fout, "\texit(0);\n");
290 f_print(fout, "}\n");
291 }
292