xref: /freebsd/usr.bin/rpcgen/rpc_main.c (revision d6b92ffa)
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 = nitems(allv);
86 static const char *allnv[] = {
87 	"rpcgen", "-s", "netpath",
88 };
89 static int allnc = nitems(allnv);
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
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 *
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
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
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
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
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
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
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 *
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
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
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 *
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 == NULL) ? 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
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
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
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
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
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
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 
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
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
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
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
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
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
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
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