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 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