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
30
31 #if 0
32 #ifndef lint
33 #ident "@(#)rpc_main.c 1.21 94/04/25 SMI"
34 static char sccsid[] = "@(#)rpc_main.c 1.30 89/03/30 (C) 1987 SMI";
35 #endif
36 #endif
37
38 #include <sys/cdefs.h>
39 __FBSDID("$FreeBSD$");
40
41 /*
42 * rpc_main.c, Top level of the RPC protocol compiler.
43 * Copyright (C) 1987, Sun Microsystems, Inc.
44 */
45
46 #include <err.h>
47 #include <ctype.h>
48 #include <stdio.h>
49 #include <string.h>
50 #include <unistd.h>
51 #include <sys/types.h>
52 #include <sys/param.h>
53 #include <sys/file.h>
54 #include <sys/stat.h>
55 #include "rpc_parse.h"
56 #include "rpc_scan.h"
57 #include "rpc_util.h"
58
59 static void c_output(const char *, const char *, int, const char *);
60 static void h_output(const char *, const char *, int, const char *, int);
61 static void l_output(const char *, const char *, int, const char *);
62 static void t_output(const char *, const char *, int, const char *);
63 static void clnt_output(const char *, const char *, int, const char * );
64 static char *generate_guard(const char *);
65 static void c_initialize(void);
66
67 static void usage(void);
68 static void options_usage(void);
69 static int do_registers(int, const char **);
70 static int parseargs(int, const char **, struct commandline *);
71 static void svc_output(const char *, const char *, int, const char *);
72 static void mkfile_output(struct commandline *);
73 static void s_output(int, const char **, const char *, const char *, int, const char *, int, int);
74
75 #define EXTEND 1 /* alias for TRUE */
76 #define DONT_EXTEND 0 /* alias for FALSE */
77
78 static const char *svcclosetime = "120";
79 static const char *CPP = NULL;
80 static const char CPPFLAGS[] = "-C";
81 static char pathbuf[MAXPATHLEN + 1];
82 static const char *allv[] = {
83 "rpcgen", "-s", "udp", "-s", "tcp",
84 };
85 static int allc = sizeof (allv)/sizeof (allv[0]);
86 static const char *allnv[] = {
87 "rpcgen", "-s", "netpath",
88 };
89 static int allnc = sizeof (allnv)/sizeof (allnv[0]);
90
91 /*
92 * machinations for handling expanding argument list
93 */
94 static void addarg(const char *); /* add another argument to the list */
95 static void insarg(int, const char *); /* insert arg at specified location */
96 static void clear_args(void); /* clear argument list */
97 static void checkfiles(const char *, const char *);
98 /* check if out file already exists */
99
100
101
102 #define ARGLISTLEN 20
103 #define FIXEDARGS 0
104
105 static char *arglist[ARGLISTLEN];
106 static int argcount = FIXEDARGS;
107
108
109 int nonfatalerrors; /* errors */
110 int inetdflag = 0; /* Support for inetd is disabled by default, use -I */
111 int pmflag = 0; /* Support for port monitors is disabled by default */
112 int tirpc_socket = 1; /* TI-RPC on socket, no TLI library */
113 int logflag; /* Use syslog instead of fprintf for errors */
114 int tblflag; /* Support for dispatch table file */
115 int mtflag = 0; /* Support for MT */
116
117 #define INLINE 0
118 /* length at which to start doing an inline */
119
120 int inline_size = INLINE;
121 /*
122 * Length at which to start doing an inline. INLINE = default
123 * if 0, no xdr_inline code
124 */
125
126 int indefinitewait; /* If started by port monitors, hang till it wants */
127 int exitnow; /* If started by port monitors, exit after the call */
128 int timerflag; /* TRUE if !indefinite && !exitnow */
129 int newstyle; /* newstyle of passing arguments (by value) */
130 int CCflag = 0; /* C++ files */
131 static int allfiles; /* generate all files */
132 int tirpcflag = 1; /* generating code for tirpc, by default */
133 xdrfunc *xdrfunc_head = NULL; /* xdr function list */
134 xdrfunc *xdrfunc_tail = NULL; /* xdr function list */
135 pid_t childpid;
136
137
138 int
main(int argc,const char * argv[])139 main(int argc, const char *argv[])
140 {
141 struct commandline cmd;
142
143 (void) memset((char *)&cmd, 0, sizeof (struct commandline));
144 clear_args();
145 if (!parseargs(argc, argv, &cmd))
146 usage();
147 /*
148 * Only the client and server side stubs are likely to be customized,
149 * so in that case only, check if the outfile exists, and if so,
150 * print an error message and exit.
151 */
152 if (cmd.Ssflag || cmd.Scflag || cmd.makefileflag) {
153 checkfiles(cmd.infile, cmd.outfile);
154 }
155 else
156 checkfiles(cmd.infile, NULL);
157
158 if (cmd.cflag) {
159 c_output(cmd.infile, "-DRPC_XDR", DONT_EXTEND, cmd.outfile);
160 } else if (cmd.hflag) {
161 h_output(cmd.infile, "-DRPC_HDR", DONT_EXTEND, cmd.outfile,
162 cmd.hflag);
163 } else if (cmd.lflag) {
164 l_output(cmd.infile, "-DRPC_CLNT", DONT_EXTEND, cmd.outfile);
165 } else if (cmd.sflag || cmd.mflag || (cmd.nflag)) {
166 s_output(argc, argv, cmd.infile, "-DRPC_SVC", DONT_EXTEND,
167 cmd.outfile, cmd.mflag, cmd.nflag);
168 } else if (cmd.tflag) {
169 t_output(cmd.infile, "-DRPC_TBL", DONT_EXTEND, cmd.outfile);
170 } else if (cmd.Ssflag) {
171 svc_output(cmd.infile, "-DRPC_SERVER", DONT_EXTEND,
172 cmd.outfile);
173 } else if (cmd.Scflag) {
174 clnt_output(cmd.infile, "-DRPC_CLIENT", DONT_EXTEND,
175 cmd.outfile);
176 } else if (cmd.makefileflag) {
177 mkfile_output(&cmd);
178 } else {
179 /* the rescans are required, since cpp may effect input */
180 c_output(cmd.infile, "-DRPC_XDR", EXTEND, "_xdr.c");
181 reinitialize();
182 h_output(cmd.infile, "-DRPC_HDR", EXTEND, ".h", cmd.hflag);
183 reinitialize();
184 l_output(cmd.infile, "-DRPC_CLNT", EXTEND, "_clnt.c");
185 reinitialize();
186 if (inetdflag || !tirpcflag)
187 s_output(allc, allv, cmd.infile, "-DRPC_SVC", EXTEND,
188 "_svc.c", cmd.mflag, cmd.nflag);
189 else
190 s_output(allnc, allnv, cmd.infile, "-DRPC_SVC",
191 EXTEND, "_svc.c", cmd.mflag, cmd.nflag);
192 if (tblflag) {
193 reinitialize();
194 t_output(cmd.infile, "-DRPC_TBL", EXTEND, "_tbl.i");
195 }
196
197 if (allfiles) {
198 reinitialize();
199 svc_output(cmd.infile, "-DRPC_SERVER", EXTEND,
200 "_server.c");
201 reinitialize();
202 clnt_output(cmd.infile, "-DRPC_CLIENT", EXTEND,
203 "_client.c");
204
205 }
206 if (allfiles || (cmd.makefileflag == 1)){
207 reinitialize();
208 mkfile_output(&cmd);
209 }
210
211 }
212 exit(nonfatalerrors);
213 /* NOTREACHED */
214 }
215
216
217 /*
218 * add extension to filename
219 */
220 static char *
extendfile(const char * path,const char * ext)221 extendfile(const char *path, const char *ext)
222 {
223 char *res;
224 const char *p;
225 const char *file;
226
227 if ((file = strrchr(path, '/')) == NULL)
228 file = path;
229 else
230 file++;
231 res = xmalloc(strlen(file) + strlen(ext) + 1);
232 p = strrchr(file, '.');
233 if (p == NULL) {
234 p = file + strlen(file);
235 }
236 (void) strcpy(res, file);
237 (void) strcpy(res + (p - file), ext);
238 return (res);
239 }
240
241 /*
242 * Open output file with given extension
243 */
244 static void
open_output(const char * infile,const char * outfile)245 open_output(const char *infile, const char *outfile)
246 {
247
248 if (outfile == NULL) {
249 fout = stdout;
250 return;
251 }
252
253 if (infile != NULL && streq(outfile, infile)) {
254 warnx("%s already exists. No output generated", infile);
255 crash();
256 }
257 fout = fopen(outfile, "w");
258 if (fout == NULL) {
259 warn("unable to open %s", outfile);
260 crash();
261 }
262 record_open(outfile);
263
264 return;
265 }
266
267 static void
add_warning(void)268 add_warning(void)
269 {
270 f_print(fout, "/*\n");
271 f_print(fout, " * Please do not edit this file.\n");
272 f_print(fout, " * It was generated using rpcgen.\n");
273 f_print(fout, " */\n\n");
274 }
275
276 /* clear list of arguments */
277 static void
clear_args(void)278 clear_args(void)
279 {
280 int i;
281 for (i = FIXEDARGS; i < ARGLISTLEN; i++)
282 arglist[i] = NULL;
283 argcount = FIXEDARGS;
284 }
285
286 /* prepend C-preprocessor and flags before arguments */
287 static void
prepend_cpp(void)288 prepend_cpp(void)
289 {
290 int idx = 1;
291 const char *var;
292 char *dupvar, *s, *t;
293
294 if (CPP != NULL)
295 insarg(0, CPP);
296 else if ((var = getenv("RPCGEN_CPP")) == NULL)
297 insarg(0, "/usr/bin/cpp");
298 else {
299 /* Parse command line in a rudimentary way */
300 dupvar = xstrdup(var);
301 for (s = dupvar, idx = 0; (t = strsep(&s, " \t")) != NULL; ) {
302 if (t[0])
303 insarg(idx++, t);
304 }
305 free(dupvar);
306 }
307
308 insarg(idx, CPPFLAGS);
309 }
310
311 /*
312 * Open input file with given define for C-preprocessor
313 */
314 static void
open_input(const char * infile,const char * define)315 open_input(const char *infile, const char *define)
316 {
317 int pd[2];
318
319 infilename = (infile == NULL) ? "<stdin>" : infile;
320 (void) pipe(pd);
321 switch (childpid = fork()) {
322 case 0:
323 prepend_cpp();
324 addarg(define);
325 if (infile)
326 addarg(infile);
327 addarg((char *)NULL);
328 (void) close(1);
329 (void) dup2(pd[1], 1);
330 (void) close(pd[0]);
331 execvp(arglist[0], arglist);
332 err(1, "execvp %s", arglist[0]);
333 case -1:
334 err(1, "fork");
335 }
336 (void) close(pd[1]);
337 fin = fdopen(pd[0], "r");
338 if (fin == NULL) {
339 warn("%s", infilename);
340 crash();
341 }
342 }
343
344 /* valid tirpc nettypes */
345 static const char *valid_ti_nettypes[] =
346 {
347 "netpath",
348 "visible",
349 "circuit_v",
350 "datagram_v",
351 "circuit_n",
352 "datagram_n",
353 "udp",
354 "tcp",
355 "raw",
356 NULL
357 };
358
359 /* valid inetd nettypes */
360 static const char *valid_i_nettypes[] =
361 {
362 "udp",
363 "tcp",
364 NULL
365 };
366
367 static int
check_nettype(const char * name,const char * list_to_check[])368 check_nettype(const char *name, const char *list_to_check[])
369 {
370 int i;
371 for (i = 0; list_to_check[i] != NULL; i++) {
372 if (strcmp(name, list_to_check[i]) == 0) {
373 return (1);
374 }
375 }
376 warnx("illegal nettype :\'%s\'", name);
377 return (0);
378 }
379
380 static const char *
file_name(const char * file,const char * ext)381 file_name(const char *file, const char *ext)
382 {
383 char *temp;
384 temp = extendfile(file, ext);
385
386 if (access(temp, F_OK) != -1)
387 return (temp);
388 else
389 return (" ");
390
391 }
392
393
394 static void
c_output(const char * infile,const char * define,int extend,const char * outfile)395 c_output(const char *infile, const char *define, int extend, const char *outfile)
396 {
397 definition *def;
398 char *include;
399 const char *outfilename;
400 long tell;
401
402 c_initialize();
403 open_input(infile, define);
404 outfilename = extend ? extendfile(infile, outfile) : outfile;
405 open_output(infile, outfilename);
406 add_warning();
407 if (infile && (include = extendfile(infile, ".h"))) {
408 f_print(fout, "#include \"%s\"\n", include);
409 free(include);
410 /* .h file already contains rpc/rpc.h */
411 } else
412 f_print(fout, "#include <rpc/rpc.h>\n");
413 tell = ftell(fout);
414 while ( (def = get_definition()) ) {
415 emit(def);
416 }
417 if (extend && tell == ftell(fout)) {
418 (void) unlink(outfilename);
419 }
420 }
421
422
423 void
c_initialize(void)424 c_initialize(void)
425 {
426
427 /* add all the starting basic types */
428 add_type(1, "int");
429 add_type(1, "long");
430 add_type(1, "short");
431 add_type(1, "bool");
432 add_type(1, "u_int");
433 add_type(1, "u_long");
434 add_type(1, "u_short");
435
436 }
437
438 static const char rpcgen_table_dcl[] = "struct rpcgen_table {\n\
439 char *(*proc)(); \n\
440 xdrproc_t xdr_arg; \n\
441 unsigned len_arg; \n\
442 xdrproc_t xdr_res; \n\
443 unsigned len_res; \n\
444 }; \n";
445
446
447 char *
generate_guard(const char * pathname)448 generate_guard(const char *pathname)
449 {
450 const char *filename;
451 char *guard, *tmp, *stopat;
452
453 filename = strrchr(pathname, '/'); /* find last component */
454 filename = ((filename == 0) ? pathname : filename+1);
455 guard = xstrdup(filename);
456 stopat = strrchr(guard, '.');
457
458 /*
459 * Convert to a valid C macro name and make it upper case.
460 * Map macro unfriendly characterss to '_'.
461 */
462 for (tmp = guard; *tmp != '\000'; ++tmp) {
463 if (islower(*tmp))
464 *tmp = toupper(*tmp);
465 else if (isupper(*tmp) || *tmp == '_')
466 /* OK for C */;
467 else if (tmp == guard)
468 *tmp = '_';
469 else if (isdigit(*tmp))
470 /* OK for all but first character */;
471 else if (tmp == stopat) {
472 *tmp = '\0';
473 break;
474 } else
475 *tmp = '_';
476 }
477 /*
478 * Can't have a '_' in front, because it'll end up being "__".
479 * "__" macros shoudln't be used. So, remove all of the
480 * '_' characters from the front.
481 */
482 if (*guard == '_') {
483 for (tmp = guard; *tmp == '_'; ++tmp)
484 ;
485 strcpy(guard, tmp);
486 }
487 guard = extendfile(guard, "_H_RPCGEN");
488 return (guard);
489 }
490
491 /*
492 * Compile into an XDR header file
493 */
494
495
496 static void
h_output(const char * infile,const char * define,int extend,const char * outfile,int headeronly)497 h_output(const char *infile, const char *define, int extend, const char *outfile, int headeronly)
498 {
499 definition *def;
500 const char *outfilename;
501 long tell;
502 const char *guard;
503 list *l;
504 xdrfunc *xdrfuncp;
505
506 open_input(infile, define);
507 outfilename = extend ? extendfile(infile, outfile) : outfile;
508 open_output(infile, outfilename);
509 add_warning();
510 if (outfilename || infile){
511 guard = generate_guard(outfilename ? outfilename: infile);
512 } else
513 guard = "STDIN_";
514
515 f_print(fout, "#ifndef _%s\n#define _%s\n\n", guard,
516 guard);
517
518 f_print(fout, "#include <rpc/rpc.h>\n");
519
520 if (mtflag)
521 f_print(fout, "#include <pthread.h>\n");
522
523 /* put the C++ support */
524 if (!CCflag) {
525 f_print(fout, "\n#ifdef __cplusplus\n");
526 f_print(fout, "extern \"C\" {\n");
527 f_print(fout, "#endif\n\n");
528 }
529
530 /* put in a typedef for quadprecision. Only with Cflag */
531
532 tell = ftell(fout);
533
534 /* print data definitions */
535 while ( (def = get_definition()) ) {
536 print_datadef(def, headeronly);
537 }
538
539 /*
540 * print function declarations.
541 * Do this after data definitions because they might be used as
542 * arguments for functions
543 */
544 for (l = defined; l != NULL; l = l->next) {
545 print_funcdef(l->val, headeronly);
546 }
547 /* Now print all xdr func declarations */
548 if (xdrfunc_head != NULL){
549
550 f_print(fout,
551 "\n/* the xdr functions */\n");
552
553 if (CCflag){
554 f_print(fout, "\n#ifdef __cplusplus\n");
555 f_print(fout, "extern \"C\" {\n");
556 f_print(fout, "#endif\n");
557 }
558
559 xdrfuncp = xdrfunc_head;
560 while (xdrfuncp != NULL){
561 print_xdr_func_def(xdrfuncp->name, xdrfuncp->pointerp);
562 xdrfuncp = xdrfuncp->next;
563 }
564 }
565
566 if (extend && tell == ftell(fout)) {
567 (void) unlink(outfilename);
568 } else if (tblflag) {
569 f_print(fout, rpcgen_table_dcl);
570 }
571
572 f_print(fout, "\n#ifdef __cplusplus\n");
573 f_print(fout, "}\n");
574 f_print(fout, "#endif\n");
575
576 f_print(fout, "\n#endif /* !_%s */\n", guard);
577 }
578
579 /*
580 * Compile into an RPC service
581 */
582 static void
s_output(int argc,const char * argv[],const char * infile,const char * define,int extend,const char * outfile,int nomain,int netflag)583 s_output(int argc, const char *argv[], const char *infile, const char *define,
584 int extend, const char *outfile, int nomain, int netflag)
585 {
586 char *include;
587 definition *def;
588 int foundprogram = 0;
589 const char *outfilename;
590
591 open_input(infile, define);
592 outfilename = extend ? extendfile(infile, outfile) : outfile;
593 open_output(infile, outfilename);
594 add_warning();
595 if (infile && (include = extendfile(infile, ".h"))) {
596 f_print(fout, "#include \"%s\"\n", include);
597 free(include);
598 } else
599 f_print(fout, "#include <rpc/rpc.h>\n");
600
601 f_print(fout, "#include <stdio.h>\n");
602 f_print(fout, "#include <stdlib.h> /* getenv, exit */\n");
603 f_print (fout, "#include <rpc/pmap_clnt.h> /* for pmap_unset */\n");
604 f_print (fout, "#include <string.h> /* strcmp */\n");
605 if (tirpcflag)
606 f_print(fout, "#include <rpc/rpc_com.h>\n");
607 if (strcmp(svcclosetime, "-1") == 0)
608 indefinitewait = 1;
609 else if (strcmp(svcclosetime, "0") == 0)
610 exitnow = 1;
611 else if (inetdflag || pmflag) {
612 f_print(fout, "#include <signal.h>\n");
613 timerflag = 1;
614 }
615
616 if (!tirpcflag && inetdflag)
617 f_print(fout, "#include <sys/ttycom.h> /* TIOCNOTTY */\n");
618 if (inetdflag || pmflag) {
619 f_print(fout, "#ifdef __cplusplus\n");
620 f_print(fout,
621 "#include <sys/sysent.h> /* getdtablesize, open */\n");
622 f_print(fout, "#endif /* __cplusplus */\n");
623 }
624 if (tirpcflag) {
625 f_print(fout, "#include <fcntl.h> /* open */\n");
626 f_print(fout, "#include <unistd.h> /* fork / setsid */\n");
627 f_print(fout, "#include <sys/types.h>\n");
628 }
629
630 f_print(fout, "#include <string.h>\n");
631 if (inetdflag || !tirpcflag) {
632 f_print(fout, "#include <sys/socket.h>\n");
633 f_print(fout, "#include <netinet/in.h>\n");
634 }
635
636 if ((netflag || pmflag) && tirpcflag && !nomain) {
637 f_print(fout, "#include <netconfig.h>\n");
638 }
639 if (tirpcflag)
640 f_print(fout, "#include <sys/resource.h> /* rlimit */\n");
641 if (logflag || inetdflag || pmflag || tirpcflag)
642 f_print(fout, "#include <syslog.h>\n");
643
644 f_print(fout, "\n#ifdef DEBUG\n#define RPC_SVC_FG\n#endif\n");
645 if (timerflag)
646 f_print(fout, "\n#define _RPCSVC_CLOSEDOWN %s\n",
647 svcclosetime);
648 while ( (def = get_definition()) ) {
649 foundprogram |= (def->def_kind == DEF_PROGRAM);
650 }
651 if (extend && !foundprogram) {
652 (void) unlink(outfilename);
653 return;
654 }
655 write_most(infile, netflag, nomain);
656 if (!nomain) {
657 if (!do_registers(argc, argv)) {
658 if (outfilename)
659 (void) unlink(outfilename);
660 usage();
661 }
662 write_rest();
663 }
664 }
665
666 /*
667 * generate client side stubs
668 */
669 static void
l_output(const char * infile,const char * define,int extend,const char * outfile)670 l_output(const char *infile, const char *define, int extend, const char *outfile)
671 {
672 char *include;
673 definition *def;
674 int foundprogram = 0;
675 const char *outfilename;
676
677 open_input(infile, define);
678 outfilename = extend ? extendfile(infile, outfile) : outfile;
679 open_output(infile, outfilename);
680 add_warning();
681 f_print (fout, "#include <string.h> /* for memset */\n");
682 if (infile && (include = extendfile(infile, ".h"))) {
683 f_print(fout, "#include \"%s\"\n", include);
684 free(include);
685 } else
686 f_print(fout, "#include <rpc/rpc.h>\n");
687 while ( (def = get_definition()) ) {
688 foundprogram |= (def->def_kind == DEF_PROGRAM);
689 }
690 if (extend && !foundprogram) {
691 (void) unlink(outfilename);
692 return;
693 }
694 write_stubs();
695 }
696
697 /*
698 * generate the dispatch table
699 */
700 static void
t_output(const char * infile,const char * define,int extend,const char * outfile)701 t_output(const char *infile, const char *define, int extend, const char *outfile)
702 {
703 definition *def;
704 int foundprogram = 0;
705 const char *outfilename;
706
707 open_input(infile, define);
708 outfilename = extend ? extendfile(infile, outfile) : outfile;
709 open_output(infile, outfilename);
710 add_warning();
711 while ( (def = get_definition()) ) {
712 foundprogram |= (def->def_kind == DEF_PROGRAM);
713 }
714 if (extend && !foundprogram) {
715 (void) unlink(outfilename);
716 return;
717 }
718 write_tables();
719 }
720
721 /* sample routine for the server template */
722 static void
svc_output(const char * infile,const char * define,int extend,const char * outfile)723 svc_output(const char *infile, const char *define, int extend, const char *outfile)
724 {
725 definition *def;
726 char *include;
727 const char *outfilename;
728 long tell;
729 open_input(infile, define);
730 outfilename = extend ? extendfile(infile, outfile) : outfile;
731 checkfiles(infile, outfilename);
732 /*
733 * Check if outfile already exists.
734 * if so, print an error message and exit
735 */
736 open_output(infile, outfilename);
737 add_sample_msg();
738
739 if (infile && (include = extendfile(infile, ".h"))) {
740 f_print(fout, "#include \"%s\"\n", include);
741 free(include);
742 } else
743 f_print(fout, "#include <rpc/rpc.h>\n");
744
745 tell = ftell(fout);
746 while ( (def = get_definition()) ) {
747 write_sample_svc(def);
748 }
749 if (extend && tell == ftell(fout)) {
750 (void) unlink(outfilename);
751 }
752 }
753
754 /* sample main routine for client */
755 static void
clnt_output(const char * infile,const char * define,int extend,const char * outfile)756 clnt_output(const char *infile, const char *define, int extend, const char *outfile)
757 {
758 definition *def;
759 char *include;
760 const char *outfilename;
761 long tell;
762 int has_program = 0;
763
764 open_input(infile, define);
765 outfilename = extend ? extendfile(infile, outfile) : outfile;
766 checkfiles(infile, outfilename);
767 /*
768 * Check if outfile already exists.
769 * if so, print an error message and exit
770 */
771
772 open_output(infile, outfilename);
773 add_sample_msg();
774 if (infile && (include = extendfile(infile, ".h"))) {
775 f_print(fout, "#include \"%s\"\n", include);
776 free(include);
777 } else
778 f_print(fout, "#include <rpc/rpc.h>\n");
779 f_print(fout, "#include <stdio.h>\n");
780 f_print(fout, "#include <stdlib.h>\n");
781 tell = ftell(fout);
782 while ( (def = get_definition()) ) {
783 has_program += write_sample_clnt(def);
784 }
785
786 if (has_program)
787 write_sample_clnt_main();
788
789 if (extend && tell == ftell(fout)) {
790 (void) unlink(outfilename);
791 }
792 }
793
794
mkfile_output(struct commandline * cmd)795 static void mkfile_output(struct commandline *cmd)
796 {
797 const char *mkfilename, *clientname, *clntname, *xdrname, *hdrname;
798 const char *servername, *svcname, *servprogname, *clntprogname;
799 char *temp, *mkftemp;
800
801 svcname = file_name(cmd->infile, "_svc.c");
802 clntname = file_name(cmd->infile, "_clnt.c");
803 xdrname = file_name(cmd->infile, "_xdr.c");
804 hdrname = file_name(cmd->infile, ".h");
805
806
807 if (allfiles){
808 servername = extendfile(cmd->infile, "_server.c");
809 clientname = extendfile(cmd->infile, "_client.c");
810 }else{
811 servername = " ";
812 clientname = " ";
813 }
814 servprogname = extendfile(cmd->infile, "_server");
815 clntprogname = extendfile(cmd->infile, "_client");
816
817 if (allfiles){
818 mkftemp = xmalloc(strlen("makefile.") +
819 strlen(cmd->infile) + 1);
820 temp = strrchr(cmd->infile, '.');
821 strcpy(mkftemp, "makefile.");
822 (void) strncat(mkftemp, cmd->infile,
823 (temp - cmd->infile));
824 mkfilename = mkftemp;
825 } else
826 mkfilename = cmd->outfile;
827
828
829 checkfiles(NULL, mkfilename);
830 open_output(NULL, mkfilename);
831
832 f_print(fout, "\n# This is a template makefile generated\
833 by rpcgen \n");
834
835 f_print(fout, "\n# Parameters \n\n");
836
837 f_print(fout, "CLIENT = %s\nSERVER = %s\n\n",
838 clntprogname, servprogname);
839 f_print(fout, "SOURCES_CLNT.c = \nSOURCES_CLNT.h = \n");
840 f_print(fout, "SOURCES_SVC.c = \nSOURCES_SVC.h = \n");
841 f_print(fout, "SOURCES.x = %s\n\n", cmd->infile);
842 f_print(fout, "TARGETS_SVC.c = %s %s %s \n",
843 svcname, servername, xdrname);
844 f_print(fout, "TARGETS_CLNT.c = %s %s %s \n",
845 clntname, clientname, xdrname);
846 f_print(fout, "TARGETS = %s %s %s %s %s %s\n\n",
847 hdrname, xdrname, clntname,
848 svcname, clientname, servername);
849
850 f_print(fout, "OBJECTS_CLNT = $(SOURCES_CLNT.c:%%.c=%%.o) \
851 $(TARGETS_CLNT.c:%%.c=%%.o) ");
852
853 f_print(fout, "\nOBJECTS_SVC = $(SOURCES_SVC.c:%%.c=%%.o) \
854 $(TARGETS_SVC.c:%%.c=%%.o) ");
855
856
857 f_print(fout, "\n# Compiler flags \n");
858 if (mtflag)
859 f_print(fout, "\nCFLAGS += -D_REENTRANT -D_THEAD_SAFE \nLDLIBS += -pthread\n");
860
861 f_print(fout, "RPCGENFLAGS = \n");
862
863 f_print(fout, "\n# Targets \n\n");
864
865 f_print(fout, "all : $(CLIENT) $(SERVER)\n\n");
866 f_print(fout, "$(TARGETS) : $(SOURCES.x) \n");
867 f_print(fout, "\trpcgen $(RPCGENFLAGS) $(SOURCES.x)\n\n");
868 if (allfiles) {
869 f_print(fout, "\trpcgen -Sc $(RPCGENFLAGS) $(SOURCES.x) -o %s\n\n", clientname);
870 f_print(fout, "\trpcgen -Ss $(RPCGENFLAGS) $(SOURCES.x) -o %s\n\n", servername);
871 }
872 f_print(fout, "$(OBJECTS_CLNT) : $(SOURCES_CLNT.c) $(SOURCES_CLNT.h) \
873 $(TARGETS_CLNT.c) \n\n");
874
875 f_print(fout, "$(OBJECTS_SVC) : $(SOURCES_SVC.c) $(SOURCES_SVC.h) \
876 $(TARGETS_SVC.c) \n\n");
877 f_print(fout, "$(CLIENT) : $(OBJECTS_CLNT) \n");
878 f_print(fout, "\t$(CC) -o $(CLIENT) $(OBJECTS_CLNT) \
879 $(LDLIBS) \n\n");
880 f_print(fout, "$(SERVER) : $(OBJECTS_SVC) \n");
881 f_print(fout, "\t$(CC) -o $(SERVER) $(OBJECTS_SVC) $(LDLIBS)\n\n");
882 f_print(fout, "clean:\n\t rm -f core $(TARGETS) $(OBJECTS_CLNT) \
883 $(OBJECTS_SVC) $(CLIENT) $(SERVER)\n\n");
884 }
885
886
887
888 /*
889 * Perform registrations for service output
890 * Return 0 if failed; 1 otherwise.
891 */
892 static int
do_registers(int argc,const char * argv[])893 do_registers(int argc, const char *argv[])
894 {
895 int i;
896
897 if (inetdflag || !tirpcflag) {
898 for (i = 1; i < argc; i++) {
899 if (streq(argv[i], "-s")) {
900 if (!check_nettype(argv[i + 1],
901 valid_i_nettypes))
902 return (0);
903 write_inetd_register(argv[i + 1]);
904 i++;
905 }
906 }
907 } else {
908 for (i = 1; i < argc; i++)
909 if (streq(argv[i], "-s")) {
910 if (!check_nettype(argv[i + 1],
911 valid_ti_nettypes))
912 return (0);
913 write_nettype_register(argv[i + 1]);
914 i++;
915 } else if (streq(argv[i], "-n")) {
916 write_netid_register(argv[i + 1]);
917 i++;
918 }
919 }
920 return (1);
921 }
922
923 /*
924 * Add another argument to the arg list
925 */
926 static void
addarg(const char * cp)927 addarg(const char *cp)
928 {
929 if (argcount >= ARGLISTLEN) {
930 warnx("too many defines");
931 crash();
932 /*NOTREACHED*/
933 }
934 if (cp != NULL)
935 arglist[argcount++] = xstrdup(cp);
936 else
937 arglist[argcount++] = NULL;
938
939 }
940
941 /*
942 * Insert an argument at the specified location
943 */
944 static void
insarg(int place,const char * cp)945 insarg(int place, const char *cp)
946 {
947 int i;
948
949 if (argcount >= ARGLISTLEN) {
950 warnx("too many defines");
951 crash();
952 /*NOTREACHED*/
953 }
954
955 /* Move up existing arguments */
956 for (i = argcount - 1; i >= place; i--)
957 arglist[i + 1] = arglist[i];
958
959 arglist[place] = xstrdup(cp);
960 argcount++;
961 }
962
963 /*
964 * if input file is stdin and an output file is specified then complain
965 * if the file already exists. Otherwise the file may get overwritten
966 * If input file does not exist, exit with an error
967 */
968
969 static void
checkfiles(const char * infile,const char * outfile)970 checkfiles(const char *infile, const char *outfile)
971 {
972
973 struct stat buf;
974
975 if (infile) /* infile ! = NULL */
976 if (stat(infile, &buf) < 0)
977 {
978 warn("%s", infile);
979 crash();
980 };
981 if (outfile) {
982 if (stat(outfile, &buf) < 0)
983 return; /* file does not exist */
984 else {
985 warnx("file '%s' already exists and may be overwritten", outfile);
986 crash();
987 }
988 }
989 }
990
991 /*
992 * Parse command line arguments
993 */
994 static int
parseargs(int argc,const char * argv[],struct commandline * cmd)995 parseargs(int argc, const char *argv[], struct commandline *cmd)
996 {
997 int i;
998 int j;
999 char c, ch;
1000 char flag[(1 << 8 * sizeof (char))];
1001 int nflags;
1002
1003 cmd->infile = cmd->outfile = NULL;
1004 if (argc < 2) {
1005 return (0);
1006 }
1007 allfiles = 0;
1008 flag['c'] = 0;
1009 flag['h'] = 0;
1010 flag['l'] = 0;
1011 flag['m'] = 0;
1012 flag['o'] = 0;
1013 flag['s'] = 0;
1014 flag['n'] = 0;
1015 flag['t'] = 0;
1016 flag['S'] = 0;
1017 flag['C'] = 0;
1018 flag['M'] = 0;
1019
1020 for (i = 1; i < argc; i++) {
1021 if (argv[i][0] != '-') {
1022 if (cmd->infile) {
1023 warnx("cannot specify more than one input file");
1024 return (0);
1025 }
1026 cmd->infile = argv[i];
1027 } else {
1028 for (j = 1; argv[i][j] != 0; j++) {
1029 c = argv[i][j];
1030 switch (c) {
1031 case 'a':
1032 allfiles = 1;
1033 break;
1034 case 'c':
1035 case 'h':
1036 case 'l':
1037 case 'm':
1038 case 't':
1039 if (flag[(int)c]) {
1040 return (0);
1041 }
1042 flag[(int)c] = 1;
1043 break;
1044 case 'S':
1045 /*
1046 * sample flag: Ss or Sc.
1047 * Ss means set flag['S'];
1048 * Sc means set flag['C'];
1049 * Sm means set flag['M'];
1050 */
1051 ch = argv[i][++j]; /* get next char */
1052 if (ch == 's')
1053 ch = 'S';
1054 else if (ch == 'c')
1055 ch = 'C';
1056 else if (ch == 'm')
1057 ch = 'M';
1058 else
1059 return (0);
1060
1061 if (flag[(int)ch]) {
1062 return (0);
1063 }
1064 flag[(int)ch] = 1;
1065 break;
1066 case 'C': /* ANSI C syntax */
1067 ch = argv[i][j+1]; /* get next char */
1068
1069 if (ch != 'C')
1070 break;
1071 CCflag = 1;
1072 break;
1073 case 'b':
1074 /*
1075 * Turn TIRPC flag off for
1076 * generating backward compatible
1077 * code
1078 */
1079 tirpcflag = 0;
1080 break;
1081
1082 case 'I':
1083 inetdflag = 1;
1084 break;
1085 case 'N':
1086 newstyle = 1;
1087 break;
1088 case 'L':
1089 logflag = 1;
1090 break;
1091 case 'P':
1092 pmflag = 1;
1093 break;
1094 case 'K':
1095 if (++i == argc) {
1096 return (0);
1097 }
1098 svcclosetime = argv[i];
1099 goto nextarg;
1100 case 'T':
1101 tblflag = 1;
1102 break;
1103 case 'M':
1104 mtflag = 1;
1105 break;
1106 case 'i' :
1107 if (++i == argc) {
1108 return (0);
1109 }
1110 inline_size = atoi(argv[i]);
1111 goto nextarg;
1112 case 'n':
1113 case 'o':
1114 case 's':
1115 if (argv[i][j - 1] != '-' ||
1116 argv[i][j + 1] != 0) {
1117 return (0);
1118 }
1119 flag[(int)c] = 1;
1120 if (++i == argc) {
1121 return (0);
1122 }
1123 if (c == 'o') {
1124 if (cmd->outfile) {
1125 return (0);
1126 }
1127 cmd->outfile = argv[i];
1128 }
1129 goto nextarg;
1130 case 'D':
1131 if (argv[i][j - 1] != '-') {
1132 return (0);
1133 }
1134 (void) addarg(argv[i]);
1135 goto nextarg;
1136 case 'Y':
1137 if (++i == argc) {
1138 return (0);
1139 }
1140 if (strlcpy(pathbuf, argv[i],
1141 sizeof(pathbuf)) >= sizeof(pathbuf)
1142 || strlcat(pathbuf, "/cpp",
1143 sizeof(pathbuf)) >=
1144 sizeof(pathbuf)) {
1145 warnx("argument too long");
1146 return (0);
1147 }
1148 CPP = pathbuf;
1149 goto nextarg;
1150
1151
1152
1153 default:
1154 return (0);
1155 }
1156 }
1157 nextarg:
1158 ;
1159 }
1160 }
1161
1162 cmd->cflag = flag['c'];
1163 cmd->hflag = flag['h'];
1164 cmd->lflag = flag['l'];
1165 cmd->mflag = flag['m'];
1166 cmd->nflag = flag['n'];
1167 cmd->sflag = flag['s'];
1168 cmd->tflag = flag['t'];
1169 cmd->Ssflag = flag['S'];
1170 cmd->Scflag = flag['C'];
1171 cmd->makefileflag = flag['M'];
1172
1173 if (tirpcflag) {
1174 if (inetdflag)
1175 pmflag = 0;
1176 if ((inetdflag && cmd->nflag)) {
1177 /* netid not allowed with inetdflag */
1178 warnx("cannot use netid flag with inetd flag");
1179 return (0);
1180 }
1181 } else { /* 4.1 mode */
1182 pmflag = 0; /* set pmflag only in tirpcmode */
1183 if (cmd->nflag) { /* netid needs TIRPC */
1184 warnx("cannot use netid flag without TIRPC");
1185 return (0);
1186 }
1187 }
1188
1189 if (newstyle && (tblflag || cmd->tflag)) {
1190 warnx("cannot use table flags with newstyle");
1191 return (0);
1192 }
1193
1194 /* check no conflicts with file generation flags */
1195 nflags = cmd->cflag + cmd->hflag + cmd->lflag + cmd->mflag +
1196 cmd->sflag + cmd->nflag + cmd->tflag + cmd->Ssflag +
1197 cmd->Scflag + cmd->makefileflag;
1198
1199 if (nflags == 0) {
1200 if (cmd->outfile != NULL || cmd->infile == NULL) {
1201 return (0);
1202 }
1203 } else if (cmd->infile == NULL &&
1204 (cmd->Ssflag || cmd->Scflag || cmd->makefileflag)) {
1205 warnx("\"infile\" is required for template generation flags");
1206 return (0);
1207 } if (nflags > 1) {
1208 warnx("cannot have more than one file generation flag");
1209 return (0);
1210 }
1211 return (1);
1212 }
1213
1214 static void
usage(void)1215 usage(void)
1216 {
1217 f_print(stderr, "%s\n%s\n%s\n%s\n%s\n",
1218 "usage: rpcgen infile",
1219 " rpcgen [-abCLNTM] [-Dname[=value]] [-i size]\
1220 [-I -P [-K seconds]] [-Y path] infile",
1221 " rpcgen [-c | -h | -l | -m | -t | -Sc | -Ss | -Sm]\
1222 [-o outfile] [infile]",
1223 " rpcgen [-s nettype]* [-o outfile] [infile]",
1224 " rpcgen [-n netid]* [-o outfile] [infile]");
1225 options_usage();
1226 exit(1);
1227 }
1228
1229 static void
options_usage(void)1230 options_usage(void)
1231 {
1232 f_print(stderr, "options:\n");
1233 f_print(stderr, "-a\t\tgenerate all files, including samples\n");
1234 f_print(stderr, "-b\t\tbackward compatibility mode (generates code \
1235 for FreeBSD 4.X)\n");
1236 f_print(stderr, "-c\t\tgenerate XDR routines\n");
1237 f_print(stderr, "-C\t\tANSI C mode\n");
1238 f_print(stderr, "-Dname[=value]\tdefine a symbol (same as #define)\n");
1239 f_print(stderr, "-h\t\tgenerate header file\n");
1240 f_print(stderr, "-i size\t\tsize at which to start generating\
1241 inline code\n");
1242 f_print(stderr, "-I\t\tgenerate code for inetd support in server\n");
1243 f_print(stderr, "-K seconds\tserver exits after K seconds of\
1244 inactivity\n");
1245 f_print(stderr, "-l\t\tgenerate client side stubs\n");
1246 f_print(stderr, "-L\t\tserver errors will be printed to syslog\n");
1247 f_print(stderr, "-m\t\tgenerate server side stubs\n");
1248 f_print(stderr, "-M\t\tgenerate MT-safe code\n");
1249 f_print(stderr, "-n netid\tgenerate server code that supports\
1250 named netid\n");
1251 f_print(stderr, "-N\t\tsupports multiple arguments and\
1252 call-by-value\n");
1253 f_print(stderr, "-o outfile\tname of the output file\n");
1254 f_print(stderr, "-P\t\tgenerate code for port monitoring support in server\n");
1255 f_print(stderr, "-s nettype\tgenerate server code that supports named\
1256 nettype\n");
1257 f_print(stderr, "-Sc\t\tgenerate sample client code that uses remote\
1258 procedures\n");
1259 f_print(stderr, "-Ss\t\tgenerate sample server code that defines\
1260 remote procedures\n");
1261 f_print(stderr, "-Sm \t\tgenerate makefile template \n");
1262
1263 f_print(stderr, "-t\t\tgenerate RPC dispatch table\n");
1264 f_print(stderr, "-T\t\tgenerate code to support RPC dispatch tables\n");
1265 f_print(stderr, "-Y path\t\tpath where cpp is found\n");
1266 exit(1);
1267 }
1268