sunrpc/rpc_main.c

   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 or with the express written consent of
   8  * Sun Microsystems, Inc.
   9  *
  10  * SUN RPC IS PROVIDED AS IS WITH NO WARRANTIES OF ANY KIND INCLUDING THE
  11  * WARRANTIES OF DESIGN, MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
  12  * PURPOSE, OR ARISING FROM A COURSE OF DEALING, USAGE OR TRADE PRACTICE.
  13  *
  14  * Sun RPC is provided with no support and without any obligation on the
  15  * part of Sun Microsystems, Inc. to assist in its use, correction,
  16  * modification or enhancement.
  17  *
  18  * SUN MICROSYSTEMS, INC. SHALL HAVE NO LIABILITY WITH RESPECT TO THE
  19  * INFRINGEMENT OF COPYRIGHTS, TRADE SECRETS OR ANY PATENTS BY SUN RPC
  20  * OR ANY PART THEREOF.
  21  *
  22  * In no event will Sun Microsystems, Inc. be liable for any lost revenue
  23  * or profits or other special, indirect and consequential damages, even if
  24  * Sun has been advised of the possibility of such damages.
  25  *
  26  * Sun Microsystems, Inc.
  27  * 2550 Garcia Avenue
  28  * Mountain View, California  94043
  29  */
  30
  31 /*
  32  * From @(#)rpc_main.c 1.30 89/03/30 (C) 1987 SMI;
  33  */
  34 char main_rcsid[] =
  35   "$Id$";
  36
  37 /*
  38  * rpc_main.c, Top level of the RPC protocol compiler.
  39  */
  40
  41 #include <stdio.h>
  42 #include <string.h>
  43 #include <unistd.h>
  44 #include <ctype.h>
  45 #include <sys/types.h>
  46 #include <sys/param.h>
  47 #include <sys/file.h>
  48 #include <sys/stat.h>
  49 #include "rpc_parse.h"
  50 #include "rpc_util.h"
  51 #include "rpc_scan.h"
  52 #include "proto.h"
  53
  54
  55 #define EXTEND  1               /* alias for TRUE */
  56 #define DONT_EXTEND     0               /* alias for FALSE */
  57
  58 #define SVR4_CPP "/usr/ccs/lib/cpp"
  59 #define SUNOS_CPP "/lib/cpp"
  60 static int cppDefined = 0;          /* explicit path for C preprocessor */
  61
  62 struct commandline {
  63         int cflag;              /* xdr C routines */
  64         int hflag;              /* header file */
  65         int lflag;              /* client side stubs */
  66         int mflag;              /* server side stubs */
  67         int nflag;              /* netid flag */
  68         int sflag;              /* server stubs for the given transport */
  69         int tflag;              /* dispatch Table file */
  70         int Ssflag;             /* produce server sample code */
  71         int Scflag;             /* produce client sample code */
  72         const char *infile;     /* input module name */
  73         const char *outfile;    /* output module name */
  74 };
  75
  76
  77 static const char *cmdname;
  78
  79 static const char *svcclosetime = "120";
  80 static const char *CPP = SVR4_CPP;
  81 static char CPPFLAGS[] = "-C";
  82 static char pathbuf[MAXPATHLEN + 1];
  83 static const char *allv[] = {
  84         "rpcgen", "-s", "udp", "-s", "tcp",
  85 };
  86 static int allc = sizeof(allv)/sizeof(allv[0]);
  87 static const char *allnv[] = {
  88         "rpcgen", "-s", "netpath",
  89 };
  90 static int allnc = sizeof(allnv)/sizeof(allnv[0]);
  91
  92 /*
  93  * machinations for handling expanding argument list
  94  */
  95 static void addarg(const char *);       /* add another argument to the list */
  96 static void putarg(int, const char *); /* put argument at specified location */
  97 static void clear_args(void);           /* clear argument list */
  98 static void checkfiles(const char *, const char *);
  99                                         /* check if out file already exists */
 100
 101 static void clear_args(void);
 102 static char *extendfile(const char *file, const char *ext);
 103 static void open_output(const char *infile, const char *outfile);
 104 static void add_warning(void);
 105 static void clear_args(void);
 106 static void find_cpp(void);
 107 static void open_input(const char *infile, const char *define);
 108 static int check_nettype(const char *name, const char *list_to_check[]);
 109 static void c_output(const char *infile, const char *define,
 110                      int extend, const char *outfile);
 111 static void h_output(const char *infile, const char *define,
 112                      int extend, const char *outfile);
 113 static void s_output(int argc, const char *argv[], const char *infile,
 114                      const char *define, int extend,
 115                      const char *outfile, int nomain, int netflag);
 116 static void l_output(const char *infile, const char *define,
 117                      int extend, const char *outfile);
 118 static void t_output(const char *infile, const char *define,
 119                      int extend, const char *outfile);
 120 static void svc_output(const char *infile, const char *define,
 121                        int extend, const char *outfile);
 122 static void clnt_output(const char *infile, const char *define,
 123                         int extend, const char *outfile);
 124 static int do_registers(int argc, const char *argv[]);
 125 static void addarg(const char *cp);
 126 static void putarg(int whereto, const char *cp);
 127 static void checkfiles(const char *infile, const char *outfile);
 128 static int parseargs(int argc, const char *argv[], struct commandline *cmd);
 129 static void usage(void);
 130 static void options_usage(void);
 131 static void c_initialize(void);
 132 static char *generate_guard(const char *pathname);
 133
 134
 135 #define ARGLISTLEN      20
 136 #define FIXEDARGS         2
 137
 138 static const char *arglist[ARGLISTLEN];
 139 static int argcount = FIXEDARGS;
 140
 141
 142 int nonfatalerrors;     /* errors */
 143 int inetdflag/* = 1*/;  /* Support for inetd */ /* is now the default */
 144 int pmflag;             /* Support for port monitors */
 145 int logflag;            /* Use syslog instead of fprintf for errors */
 146 int tblflag;            /* Support for dispatch table file */
 147
 148 #define INLINE 3
 149 /*length at which to start doing an inline */
 150
 151 int inlineflag=INLINE; /* length at which to start doing an inline. 3 = default
 152                 if 0, no xdr_inline code */
 153
 154 int indefinitewait;     /* If started by port monitors, hang till it wants */
 155 int exitnow;            /* If started by port monitors, exit after the call */
 156 int timerflag;          /* TRUE if !indefinite && !exitnow */
 157 int newstyle;           /* newstyle of passing arguments (by value) */
 158 #ifdef __GNU_LIBRARY__
 159 int Cflag = 1 ;         /* ANSI C syntax */
 160 #else
 161 int Cflag = 0 ;         /* ANSI C syntax */
 162 #endif
 163 static int allfiles;   /* generate all files */
 164 #ifdef __GNU_LIBRARY__
 165 int tirpcflag = 0;       /* generating code for tirpc, by default */
 166 #else
 167 int tirpcflag = 1;       /* generating code for tirpc, by default */
 168 #endif
 169
 170 int
 171 main(int argc, const char *argv[])
 172 {
 173         struct commandline cmd;
 174
 175         (void) memset((char *)&cmd, 0, sizeof (struct commandline));
 176         clear_args();
 177         if (!parseargs(argc, argv, &cmd))
 178                 usage();
 179
 180         if (cmd.cflag || cmd.hflag || cmd.lflag || cmd.tflag || cmd.sflag ||
 181                 cmd.mflag || cmd.nflag || cmd.Ssflag || cmd.Scflag ) {
 182           checkfiles(cmd.infile, cmd.outfile);
 183         }
 184         else
 185           checkfiles(cmd.infile,NULL);
 186
 187         if (cmd.cflag) {
 188                 c_output(cmd.infile, "-DRPC_XDR", DONT_EXTEND, cmd.outfile);
 189         } else if (cmd.hflag) {
 190                 h_output(cmd.infile, "-DRPC_HDR", DONT_EXTEND, cmd.outfile);
 191         } else if (cmd.lflag) {
 192                 l_output(cmd.infile, "-DRPC_CLNT", DONT_EXTEND, cmd.outfile);
 193         } else if (cmd.sflag || cmd.mflag || (cmd.nflag)) {
 194                 s_output(argc, argv, cmd.infile, "-DRPC_SVC", DONT_EXTEND,
 195                          cmd.outfile, cmd.mflag, cmd.nflag);
 196         } else if (cmd.tflag) {
 197                 t_output(cmd.infile, "-DRPC_TBL", DONT_EXTEND, cmd.outfile);
 198         } else if  (cmd.Ssflag) {
 199                   svc_output(cmd.infile, "-DRPC_SERVER", DONT_EXTEND, cmd.outfile);
 200         } else if (cmd.Scflag) {
 201                   clnt_output(cmd.infile, "-DRPC_CLIENT", DONT_EXTEND, cmd.outfile);
 202         } else {
 203                 /* the rescans are required, since cpp may effect input */
 204                 c_output(cmd.infile, "-DRPC_XDR", EXTEND, "_xdr.c");
 205                 reinitialize();
 206                 h_output(cmd.infile, "-DRPC_HDR", EXTEND, ".h");
 207                 reinitialize();
 208                 l_output(cmd.infile, "-DRPC_CLNT", EXTEND, "_clnt.c");
 209                 reinitialize();
 210                 if (inetdflag || !tirpcflag )
 211                         s_output(allc, allv, cmd.infile, "-DRPC_SVC", EXTEND,
 212                                  "_svc.c", cmd.mflag, cmd.nflag);
 213                 else
 214                         s_output(allnc, allnv, cmd.infile, "-DRPC_SVC",
 215                                  EXTEND, "_svc.c", cmd.mflag, cmd.nflag);
 216                 if (tblflag) {
 217                         reinitialize();
 218                         t_output(cmd.infile, "-DRPC_TBL", EXTEND, "_tbl.i");
 219                 }
 220                 if (allfiles) {
 221                   reinitialize();
 222                   svc_output(cmd.infile, "-DRPC_SERVER", EXTEND, "_server.c");
 223                 }
 224                 if (allfiles) {
 225                   reinitialize();
 226                   clnt_output(cmd.infile, "-DRPC_CLIENT", EXTEND, "_client.c");
 227                 }
 228         }
 229         exit(nonfatalerrors);
 230         /* NOTREACHED */
 231 }
 232
 233 /*
 234  * add extension to filename
 235  */
 236 static char *
 237 extendfile(const char *file, const char *ext)
 238 {
 239         char *res;
 240         const char *p;
 241
 242         res = alloc(strlen(file) + strlen(ext) + 1);
 243         if (res == NULL) {
 244                 abort();
 245         }
 246         p = strrchr(file, '.');
 247         if (p == NULL) {
 248                 p = file + strlen(file);
 249         }
 250         (void) strcpy(res, file);
 251         (void) strcpy(res + (p - file), ext);
 252         return (res);
 253 }
 254
 255 /*
 256  * Open output file with given extension
 257  */
 258 static void
 259 open_output(const char *infile, const char *outfile)
 260 {
 261
 262         if (outfile == NULL) {
 263                 fout = stdout;
 264                 return;
 265         }
 266
 267         if (infile != NULL && streq(outfile, infile)) {
 268                 f_print(stderr, "%s: output would overwrite %s\n", cmdname,
 269                         infile);
 270                 crash();
 271         }
 272         fout = fopen(outfile, "w");
 273         if (fout == NULL) {
 274                 f_print(stderr, "%s: unable to open ", cmdname);
 275                 perror(outfile);
 276                 crash();
 277         }
 278         record_open(outfile);
 279 }
 280
 281 static void
 282 add_warning(void)
 283 {
 284         f_print(fout, "/*\n");
 285         f_print(fout, " * Please do not edit this file.\n");
 286         f_print(fout, " * It was generated using rpcgen.\n");
 287         f_print(fout, " */\n\n");
 288 }
 289
 290 static void
 291 add_stdheaders(void)
 292 {
 293         f_print(fout, "#include <rpc/types.h>\n");
 294         f_print(fout, "#include <rpc/xdr.h>\n\n");
 295 }
 296
 297 /* clear list of arguments */
 298 static void clear_args(void)
 299 {
 300   int i;
 301   for( i=FIXEDARGS; i<ARGLISTLEN; i++ )
 302     arglist[i] = NULL;
 303   argcount = FIXEDARGS;
 304 }
 305
 306 /* make sure that a CPP exists */
 307 static void find_cpp(void)
 308 {
 309   struct stat buf;
 310
 311   if (stat(CPP, &buf) < 0 )  {  /* SVR4 or explicit cpp does not exist */
 312     if (cppDefined) {
 313       fprintf( stderr, "cannot find C preprocessor: %s \n", CPP );
 314       crash();
 315     } else {                    /* try the other one */
 316       CPP = SUNOS_CPP;
 317       if( stat( CPP, &buf ) < 0 ) { /* can't find any cpp */
 318         fprintf( stderr, "cannot find any C preprocessor (cpp)\n" );
 319         crash();
 320       }
 321     }
 322   }
 323 }
 324
 325 /*
 326  * Open input file with given define for C-preprocessor
 327  */
 328 static void
 329 open_input(const char *infile, const char *define)
 330 {
 331         int pd[2];
 332
 333         infilename = (infile == NULL) ? "<stdin>" : infile;
 334         (void) pipe(pd);
 335         switch (fork()) {
 336         case 0:
 337                 find_cpp();
 338                 putarg(0, CPP);
 339                 putarg(1, CPPFLAGS);
 340                 addarg(define);
 341                 addarg(infile);
 342                 addarg((char *)NULL);
 343                 (void) close(1);
 344                 (void) dup2(pd[1], 1);
 345                 (void) close(pd[0]);
 346                 execv(arglist[0], (char **)arglist);
 347                 perror("execv");
 348                 exit(1);
 349         case -1:
 350                 perror("fork");
 351                 exit(1);
 352         }
 353         (void) close(pd[1]);
 354         fin = fdopen(pd[0], "r");
 355         if (fin == NULL) {
 356                 f_print(stderr, "%s: ", cmdname);
 357                 perror(infilename);
 358                 crash();
 359         }
 360 }
 361
 362 /* valid tirpc nettypes */
 363 static const char *valid_ti_nettypes[] =
 364 {
 365   "netpath",
 366   "visible",
 367   "circuit_v",
 368   "datagram_v",
 369   "circuit_n",
 370   "datagram_n",
 371   "udp",
 372   "tcp",
 373   "raw",
 374   NULL
 375   };
 376
 377 /* valid inetd nettypes */
 378 static const char *valid_i_nettypes[] =
 379 {
 380   "udp",
 381   "tcp",
 382   NULL
 383 };
 384
 385 static int check_nettype(const char *name, const char *list_to_check[]) {
 386   int i;
 387   for( i = 0; list_to_check[i] != NULL; i++ ) {
 388           if( strcmp( name, list_to_check[i] ) == 0 ) {
 389             return 1;
 390           }
 391   }
 392   f_print( stderr, "illegal nettype :\'%s\'\n", name );
 393   return 0;
 394 }
 395
 396 /*
 397  * Compile into an XDR routine output file
 398  */
 399
 400 static void
 401 c_output(const char *infile, const char *define, int extend,
 402          const char *outfile)
 403 {
 404         definition *def;
 405         char *include;
 406         const char *outfilename;
 407         long tell;
 408
 409         c_initialize();
 410         open_input(infile, define);
 411         outfilename = extend ? extendfile(infile, outfile) : outfile;
 412         open_output(infile, outfilename);
 413         add_warning();
 414         add_stdheaders();
 415         if (infile && (include = extendfile(infile, ".h"))) {
 416                 f_print(fout, "#include \"%s\"\n", include);
 417                 free(include);
 418                 /* .h file already contains rpc/rpc.h */
 419         } else
 420           f_print(fout, "#include <rpc/rpc.h>\n");
 421         tell = ftell(fout);
 422         while ((def = get_definition())!=NULL) {
 423                 emit(def);
 424         }
 425         if (extend && tell == ftell(fout)) {
 426                 (void) unlink(outfilename);
 427         }
 428 }
 429
 430 void
 431 c_initialize(void)
 432 {
 433
 434   /* add all the starting basic types */
 435
 436   add_type(1,"int");
 437   add_type(1,"long");
 438   add_type(1,"short");
 439   add_type(1,"bool");
 440
 441   add_type(1,"u_int");
 442   add_type(1,"u_long");
 443   add_type(1,"u_short");
 444
 445 }
 446
 447 char rpcgen_table_dcl[] = "struct rpcgen_table {\n\
 448         char    *(*proc)();\n\
 449         xdrproc_t       xdr_arg;\n\
 450         unsigned        len_arg;\n\
 451         xdrproc_t       xdr_res;\n\
 452         unsigned        len_res;\n\
 453 };\n";
 454
 455
 456 static char *generate_guard(const char *pathname) {
 457         const char *filename;
 458         char *guard, *tmp;
 459
 460         filename = strrchr(pathname, '/' );  /* find last component */
 461         filename = ((filename == NULL) ? pathname : filename+1);
 462         guard = strdup(filename);
 463         /* convert to upper case */
 464         tmp = guard;
 465         while (*tmp) {
 466                 if (islower(*tmp))
 467                         *tmp = toupper(*tmp);
 468                 tmp++;
 469         }
 470
 471         guard = extendfile(guard, "_H_RPCGEN");
 472         return guard;
 473 }
 474
 475 /*
 476  * Compile into an XDR header file
 477  */
 478
 479
 480 static void
 481 h_output(const char *infile, const char *define, int extend,
 482          const char *outfile)
 483 {
 484         definition *def;
 485         const char *ifilename;
 486         const char *outfilename;
 487         long tell;
 488         char *guard;
 489         list *l;
 490
 491         open_input(infile, define);
 492         outfilename =  extend ? extendfile(infile, outfile) : outfile;
 493         open_output(infile, outfilename);
 494         add_warning();
 495         ifilename = (infile == NULL) ? "STDIN" : infile;
 496         guard = generate_guard(  outfilename ? outfilename: ifilename );
 497
 498         f_print(fout,"#ifndef _%s\n#define _%s\n\n", guard,
 499                 guard);
 500
 501         f_print(fout, "#include <rpc/rpc.h>\n\n");
 502
 503         tell = ftell(fout);
 504         /* print data definitions */
 505         while ((def = get_definition())!=NULL) {
 506                 print_datadef(def);
 507         }
 508
 509         /* print function declarations.
 510            Do this after data definitions because they might be used as
 511            arguments for functions */
 512         for (l = defined; l != NULL; l = l->next) {
 513                 print_funcdef(l->val);
 514         }
 515         if (extend && tell == ftell(fout)) {
 516                 (void) unlink(outfilename);
 517         } else if (tblflag) {
 518                 f_print(fout, rpcgen_table_dcl);
 519         }
 520         f_print(fout, "\n#endif /* !_%s */\n", guard);
 521 }
 522
 523 /*
 524  * Compile into an RPC service
 525  */
 526 static void
 527 s_output(int argc, const char *argv[], const char *infile, const char *define,
 528          int extend, const char *outfile, int nomain, int netflag)
 529 {
 530         char *include;
 531         definition *def;
 532         int foundprogram = 0;
 533         const char *outfilename;
 534
 535         open_input(infile, define);
 536         outfilename = extend ? extendfile(infile, outfile) : outfile;
 537         open_output(infile, outfilename);
 538         add_warning();
 539         if (infile && (include = extendfile(infile, ".h"))) {
 540                 f_print(fout, "#include \"%s\"\n", include);
 541                 free(include);
 542         } else
 543           f_print(fout, "#include <rpc/rpc.h>\n");
 544
 545         f_print(fout, "#include <stdio.h>\n");
 546         f_print(fout, "#include <stdlib.h>/* getenv, exit */\n");
 547         if (Cflag) {
 548                 f_print (fout, "#include <rpc/pmap_clnt.h> /* for pmap_unset */\n");
 549                 f_print (fout, "#include <string.h> /* strcmp */ \n");
 550         }
 551         if (strcmp(svcclosetime, "-1") == 0)
 552                 indefinitewait = 1;
 553         else if (strcmp(svcclosetime, "0") == 0)
 554                 exitnow = 1;
 555         else if (inetdflag || pmflag) {
 556                 f_print(fout, "#include <signal.h>\n");
 557           timerflag = 1;
 558         }
 559
 560         if( !tirpcflag && inetdflag )
 561 #ifdef __GNU_LIBRARY__
 562           f_print(fout, "#include <sys/ioctl.h> /* ioctl, TIOCNOTTY */\n");
 563 #else
 564           f_print(fout, "#include <sys/ttycom.h>/* TIOCNOTTY */\n");
 565 #endif
 566         if( Cflag && (inetdflag || pmflag ) ) {
 567 #ifdef __GNU_LIBRARY__
 568           f_print(fout, "#include <sys/types.h> /* open */\n");
 569           f_print(fout, "#include <sys/stat.h> /* open */\n");
 570           f_print(fout, "#include <fcntl.h> /* open */\n");
 571           f_print(fout, "#include <unistd.h> /* getdtablesize */\n");
 572 #else
 573           f_print(fout, "#ifdef __cplusplus\n");
 574           f_print(fout, "#include <sysent.h> /* getdtablesize, open */\n");
 575           f_print(fout, "#endif /* __cplusplus */\n");
 576 #endif
 577           if( tirpcflag )
 578             f_print(fout, "#include <unistd.h> /* setsid */\n");
 579         }
 580         if( tirpcflag )
 581           f_print(fout, "#include <sys/types.h>\n");
 582
 583         f_print(fout, "#include <memory.h>\n");
 584 #ifndef __GNU_LIBRARY__
 585         f_print(fout, "#include <stropts.h>\n");
 586 #endif
 587         if (inetdflag || !tirpcflag ) {
 588                 f_print(fout, "#include <sys/socket.h>\n");
 589                 f_print(fout, "#include <netinet/in.h>\n");
 590         }
 591
 592         if ( (netflag || pmflag) && tirpcflag ) {
 593                 f_print(fout, "#include <netconfig.h>\n");
 594         }
 595         if (/*timerflag &&*/ tirpcflag)
 596                 f_print(fout, "#include <sys/resource.h> /* rlimit */\n");
 597         if (logflag || inetdflag || pmflag) {
 598 #ifdef __GNU_LIBRARY__
 599                 f_print(fout, "#include <syslog.h>\n");
 600 #else
 601                 f_print(fout, "#ifdef SYSLOG\n");
 602                 f_print(fout, "#include <syslog.h>\n");
 603                 f_print(fout, "#else\n");
 604                 f_print(fout, "#define LOG_ERR 1\n");
 605                 f_print(fout, "#define openlog(a, b, c)\n");
 606                 f_print(fout, "#endif\n");
 607 #endif
 608         }
 609
 610         /* for ANSI-C */
 611         f_print(fout, "\n#ifdef __STDC__\n#define SIG_PF void(*)(int)\n#endif\n");
 612
 613 #ifndef __GNU_LIBRARY__
 614         f_print(fout, "\n#ifdef DEBUG\n#define RPC_SVC_FG\n#endif\n");
 615 #endif
 616         if (timerflag)
 617                 f_print(fout, "\n#define _RPCSVC_CLOSEDOWN %s\n", svcclosetime);
 618         while ((def = get_definition())!=NULL) {
 619                 foundprogram |= (def->def_kind == DEF_PROGRAM);
 620         }
 621         if (extend && !foundprogram) {
 622                 (void) unlink(outfilename);
 623                 return;
 624         }
 625         write_most(infile, netflag, nomain);
 626         if (!nomain) {
 627                 if( !do_registers(argc, argv) ) {
 628                   if (outfilename)
 629                     (void) unlink(outfilename);
 630                   usage();
 631                 }
 632                 write_rest();
 633         }
 634 }
 635
 636 /*
 637  * generate client side stubs
 638  */
 639 static void
 640 l_output(const char *infile, const char *define, int extend,
 641          const char *outfile)
 642 {
 643         char *include;
 644         definition *def;
 645         int foundprogram = 0;
 646         const char *outfilename;
 647
 648         open_input(infile, define);
 649         outfilename = extend ? extendfile(infile, outfile) : outfile;
 650         open_output(infile, outfilename);
 651         add_warning();
 652         if (Cflag)
 653           f_print (fout, "#include <memory.h> /* for memset */\n");
 654         if (infile && (include = extendfile(infile, ".h"))) {
 655                 f_print(fout, "#include \"%s\"\n", include);
 656                 free(include);
 657         } else
 658           f_print(fout, "#include <rpc/rpc.h>\n");
 659         while ((def = get_definition())!=NULL) {
 660                 foundprogram |= (def->def_kind == DEF_PROGRAM);
 661         }
 662         if (extend && !foundprogram) {
 663                 (void) unlink(outfilename);
 664                 return;
 665         }
 666         write_stubs();
 667 }
 668
 669 /*
 670  * generate the dispatch table
 671  */
 672 static void
 673 t_output(const char *infile, const char *define, int extend,
 674          const char *outfile)
 675 {
 676         definition *def;
 677         int foundprogram = 0;
 678         const char *outfilename;
 679
 680         open_input(infile, define);
 681         outfilename = extend ? extendfile(infile, outfile) : outfile;
 682         open_output(infile, outfilename);
 683         add_warning();
 684         while ((def = get_definition())!=NULL) {
 685                 foundprogram |= (def->def_kind == DEF_PROGRAM);
 686         }
 687         if (extend && !foundprogram) {
 688                 (void) unlink(outfilename);
 689                 return;
 690         }
 691         write_tables();
 692 }
 693
 694 /* sample routine for the server template */
 695 static void
 696 svc_output(const char *infile, const char *define, int extend,
 697            const char *outfile)
 698 {
 699   definition *def;
 700   char *include;
 701   const char *outfilename;
 702   long tell;
 703
 704   open_input(infile, define);
 705   outfilename = extend ? extendfile(infile, outfile) : outfile;
 706   checkfiles(infile,outfilename); /*check if outfile already exists.
 707                                   if so, print an error message and exit*/
 708   open_output(infile, outfilename);
 709   add_sample_msg();
 710
 711   if (infile && (include = extendfile(infile, ".h"))) {
 712     f_print(fout, "#include \"%s\"\n", include);
 713     free(include);
 714   } else
 715     f_print(fout, "#include <rpc/rpc.h>\n");
 716
 717   tell = ftell(fout);
 718   while ((def = get_definition())!=NULL) {
 719           write_sample_svc(def);
 720   }
 721   if (extend && tell == ftell(fout)) {
 722           (void) unlink(outfilename);
 723   }
 724 }
 725
 726
 727 /* sample main routine for client */
 728 static void
 729 clnt_output(const char *infile, const char *define, int extend,
 730             const char *outfile)
 731 {
 732   definition *def;
 733   char *include;
 734   const char *outfilename;
 735   long tell;
 736   int has_program = 0;
 737
 738   open_input(infile, define);
 739   outfilename = extend ? extendfile(infile, outfile) : outfile;
 740   checkfiles(infile,outfilename); /*check if outfile already exists.
 741                                   if so, print an error message and exit*/
 742
 743   open_output(infile, outfilename);
 744   add_sample_msg();
 745   if (infile && (include = extendfile(infile, ".h"))) {
 746     f_print(fout, "#include \"%s\"\n", include);
 747     free(include);
 748   } else
 749     f_print(fout, "#include <rpc/rpc.h>\n");
 750   tell = ftell(fout);
 751   while ((def = get_definition())!=NULL) {
 752     has_program += write_sample_clnt(def);
 753   }
 754
 755   if( has_program )
 756     write_sample_clnt_main();
 757
 758   if (extend && tell == ftell(fout)) {
 759     (void) unlink(outfilename);
 760   }
 761 }
 762
 763 /*
 764  * Perform registrations for service output
 765  * Return 0 if failed; 1 otherwise.
 766  */
 767 static int do_registers(int argc, const char *argv[])
 768 {
 769         int i;
 770
 771         if ( inetdflag || !tirpcflag) {
 772                 for (i = 1; i < argc; i++) {
 773                         if (streq(argv[i], "-s")) {
 774                                 if(!check_nettype( argv[i + 1], valid_i_nettypes ))
 775                                   return 0;
 776                                 write_inetd_register(argv[i + 1]);
 777                                 i++;
 778                         }
 779                 }
 780         } else {
 781                 for (i = 1; i < argc; i++)
 782                         if (streq(argv[i], "-s")) {
 783                                 if(!check_nettype( argv[i + 1], valid_ti_nettypes ))
 784                                   return 0;
 785                                 write_nettype_register(argv[i + 1]);
 786                                 i++;
 787                         } else if (streq(argv[i], "-n")) {
 788                                 write_netid_register(argv[i + 1]);
 789                                 i++;
 790                         }
 791         }
 792         return 1;
 793 }
 794
 795 /*
 796  * Add another argument to the arg list
 797  */
 798 static void
 799 addarg(const char *cp)
 800 {
 801         if (argcount >= ARGLISTLEN) {
 802                 f_print(stderr, "rpcgen: too many defines\n");
 803                 crash();
 804                 /*NOTREACHED*/
 805         }
 806         arglist[argcount++] = cp;
 807
 808 }
 809
 810 static void
 811 putarg(int whereto, const char *cp)
 812 {
 813         if (whereto >= ARGLISTLEN) {
 814                 f_print(stderr, "rpcgen: arglist coding error\n");
 815                 crash();
 816                 /*NOTREACHED*/
 817         }
 818         arglist[whereto] = cp;
 819 }
 820
 821 /*
 822  * if input file is stdin and an output file is specified then complain
 823  * if the file already exists. Otherwise the file may get overwritten
 824  * If input file does not exist, exit with an error
 825  */
 826
 827 static void
 828 checkfiles(const char *infile, const char *outfile)
 829 {
 830
 831   struct stat buf;
 832
 833   if(infile)                    /* infile ! = NULL */
 834     if(stat(infile,&buf) < 0)
 835       {
 836         perror(infile);
 837         crash();
 838       };
 839   if (outfile) {
 840     if (stat(outfile, &buf) < 0)
 841       return;                   /* file does not exist */
 842     else {
 843       f_print(stderr,
 844               "file '%s' already exists and may be overwritten\n", outfile);
 845       crash();
 846     }
 847   }
 848 }
 849
 850 /*
 851  * Parse command line arguments
 852  */
 853 static int
 854 parseargs(int argc, const char *argv[], struct commandline *cmd)
 855 {
 856         int i;
 857         int j;
 858         int c;
 859         char flag[(1 << 8 * sizeof(char))];
 860         int nflags;
 861
 862         cmdname = argv[0];
 863         cmd->infile = cmd->outfile = NULL;
 864         if (argc < 2) {
 865                 return (0);
 866         }
 867         allfiles = 0;
 868         flag['c'] = 0;
 869         flag['h'] = 0;
 870         flag['l'] = 0;
 871         flag['m'] = 0;
 872         flag['o'] = 0;
 873         flag['s'] = 0;
 874         flag['n'] = 0;
 875         flag['t'] = 0;
 876         flag['S'] = 0;
 877         flag['C'] = 0;
 878         for (i = 1; i < argc; i++) {
 879                 if (argv[i][0] != '-') {
 880                         if (cmd->infile) {
 881                                 f_print( stderr, "Cannot specify more than one input file!\n");
 882
 883                                 return (0);
 884                         }
 885                         cmd->infile = argv[i];
 886                 } else {
 887                         for (j = 1; argv[i][j] != 0; j++) {
 888                                 c = argv[i][j];
 889                                 switch (c) {
 890                                 case 'a':
 891                                         allfiles = 1;
 892                                         break;
 893                                 case 'c':
 894                                 case 'h':
 895                                 case 'l':
 896                                 case 'm':
 897                                 case 't':
 898                                         if (flag[c]) {
 899                                                 return (0);
 900                                         }
 901                                         flag[c] = 1;
 902                                         break;
 903                                 case 'S':
 904                                         /* sample flag: Ss or Sc.
 905                                            Ss means set flag['S'];
 906                                            Sc means set flag['C']; */
 907                                         c = argv[i][++j];  /* get next char */
 908                                         if( c == 's' )
 909                                           c = 'S';
 910                                         else if( c == 'c' )
 911                                           c = 'C';
 912                                         else
 913                                           return( 0 );
 914
 915                                         if (flag[c]) {
 916                                                 return (0);
 917                                         }
 918                                         flag[c] = 1;
 919                                         break;
 920                                 case 'C':  /* ANSI C syntax */
 921                                         Cflag = 1;
 922                                         break;
 923
 924 #ifdef __GNU_LIBRARY__
 925                                 case 'k':  /* K&R C syntax */
 926                                         Cflag = 0;
 927                                         break;
 928
 929 #endif
 930                                 case 'b':  /* turn TIRPC flag off for
 931                                             generating backward compatible
 932                                             */
 933                                         tirpcflag = 0;
 934                                         break;
 935
 936 #ifdef __GNU_LIBRARY__
 937                                 case '5':  /* turn TIRPC flag on for
 938                                             generating SysVr4 compatible
 939                                             */
 940                                         tirpcflag = 1;
 941                                         break;
 942
 943 #endif
 944                                 case 'I':
 945                                         inetdflag = 1;
 946                                         break;
 947                                 case 'N':
 948                                         newstyle = 1;
 949                                         break;
 950                                 case 'L':
 951                                         logflag = 1;
 952                                         break;
 953                                 case 'K':
 954                                         if (++i == argc) {
 955                                                 return (0);
 956                                         }
 957                                         svcclosetime = argv[i];
 958                                         goto nextarg;
 959                                 case 'T':
 960                                         tblflag = 1;
 961                                         break;
 962                                 case 'i' :
 963                                         if (++i == argc) {
 964                                                 return (0);
 965                                         }
 966                                         inlineflag = atoi(argv[i]);
 967                                         goto nextarg;
 968                                 case 'n':
 969                                 case 'o':
 970                                 case 's':
 971                                         if (argv[i][j - 1] != '-' ||
 972                                             argv[i][j + 1] != 0) {
 973                                                 return (0);
 974                                         }
 975                                         flag[c] = 1;
 976                                         if (++i == argc) {
 977                                                 return (0);
 978                                         }
 979                                         if (c == 's') {
 980                                                 if (!streq(argv[i], "udp") &&
 981                                                     !streq(argv[i], "tcp")) {
 982                                                         return (0);
 983                                                 }
 984                                         } else if (c == 'o') {
 985                                                 if (cmd->outfile) {
 986                                                         return (0);
 987                                                 }
 988                                                 cmd->outfile = argv[i];
 989                                         }
 990                                         goto nextarg;
 991                                 case 'D':
 992                                         if (argv[i][j - 1] != '-') {
 993                                                 return (0);
 994                                         }
 995                                         (void) addarg(argv[i]);
 996                                         goto nextarg;
 997                                 case 'Y':
 998                                         if (++i == argc) {
 999                                                 return (0);
1000                                         }
1001                                         (void) strcpy(pathbuf, argv[i]);
1002                                         (void) strcat(pathbuf, "/cpp");
1003                                         CPP = pathbuf;
1004                                         cppDefined = 1;
1005                                         goto nextarg;
1006
1007
1008
1009                                 default:
1010                                         return (0);
1011                                 }
1012                         }
1013         nextarg:
1014                         ;
1015                 }
1016         }
1017
1018         cmd->cflag = flag['c'];
1019         cmd->hflag = flag['h'];
1020         cmd->lflag = flag['l'];
1021         cmd->mflag = flag['m'];
1022         cmd->nflag = flag['n'];
1023         cmd->sflag = flag['s'];
1024         cmd->tflag = flag['t'];
1025         cmd->Ssflag = flag['S'];
1026         cmd->Scflag = flag['C'];
1027
1028         if( tirpcflag ) {
1029           pmflag = inetdflag ? 0 : 1;     /* pmflag or inetdflag is always TRUE */
1030           if( (inetdflag && cmd->nflag)) { /* netid not allowed with inetdflag */
1031             f_print(stderr, "Cannot use netid flag with inetd flag!\n");
1032             return (0);
1033           }
1034         } else {  /* 4.1 mode */
1035           pmflag = 0;               /* set pmflag only in tirpcmode */
1036 #ifndef __GNU_LIBRARY__
1037           inetdflag = 1;            /* inetdflag is TRUE by default */
1038 #endif
1039           if( cmd->nflag ) {          /* netid needs TIRPC */
1040             f_print( stderr, "Cannot use netid flag without TIRPC!\n");
1041             return( 0 );
1042           }
1043         }
1044
1045         if( newstyle && ( tblflag || cmd->tflag) ) {
1046           f_print( stderr, "Cannot use table flags with newstyle!\n");
1047           return( 0 );
1048         }
1049
1050         /* check no conflicts with file generation flags */
1051         nflags = cmd->cflag + cmd->hflag + cmd->lflag + cmd->mflag +
1052                 cmd->sflag + cmd->nflag + cmd->tflag + cmd->Ssflag + cmd->Scflag;
1053
1054         if (nflags == 0) {
1055                 if (cmd->outfile != NULL || cmd->infile == NULL) {
1056                         return (0);
1057                 }
1058         } else if (nflags > 1) {
1059                 f_print( stderr, "Cannot have more than one file generation flag!\n");
1060                 return (0);
1061         }
1062         return (1);
1063 }
1064
1065 static void
1066 usage(void)
1067 {
1068         f_print(stderr, "usage:  %s infile\n", cmdname);
1069         f_print(stderr, "\t%s [-a][-b][-C][-Dname[=value]] -i size  [-I [-K seconds]] [-L][-N][-T] infile\n",
1070                         cmdname);
1071         f_print(stderr, "\t%s [-c | -h | -l | -m | -t | -Sc | -Ss] [-o outfile] [infile]\n",
1072                         cmdname);
1073         f_print(stderr, "\t%s [-s nettype]* [-o outfile] [infile]\n", cmdname);
1074         f_print(stderr, "\t%s [-n netid]* [-o outfile] [infile]\n", cmdname);
1075         options_usage();
1076         exit(1);
1077 }
1078
1079 static void
1080 options_usage(void)
1081 {
1082         f_print(stderr, "options:\n");
1083         f_print(stderr, "-a\t\tgenerate all files, including samples\n");
1084         f_print(stderr, "-b\t\tbackward compatibility mode (generates code for SunOS 4.1)\n");
1085         f_print(stderr, "-c\t\tgenerate XDR routines\n");
1086         f_print(stderr, "-C\t\tANSI C mode\n");
1087         f_print(stderr, "-Dname[=value]\tdefine a symbol (same as #define)\n");
1088         f_print(stderr, "-h\t\tgenerate header file\n");
1089         f_print(stderr, "-i size\t\tsize at which to start generating inline code\n");
1090         f_print(stderr, "-I\t\tgenerate code for inetd support in server (for SunOS 4.1)\n");
1091         f_print(stderr, "-K seconds\tserver exits after K seconds of inactivity\n");
1092         f_print(stderr, "-l\t\tgenerate client side stubs\n");
1093         f_print(stderr, "-L\t\tserver errors will be printed to syslog\n");
1094         f_print(stderr, "-m\t\tgenerate server side stubs\n");
1095         f_print(stderr, "-n netid\tgenerate server code that supports named netid\n");
1096         f_print(stderr, "-N\t\tsupports multiple arguments and call-by-value\n");
1097         f_print(stderr, "-o outfile\tname of the output file\n");
1098         f_print(stderr, "-s nettype\tgenerate server code that supports named nettype\n");
1099         f_print(stderr, "-Sc\t\tgenerate sample client code that uses remote procedures\n");
1100         f_print(stderr, "-Ss\t\tgenerate sample server code that defines remote procedures\n");
1101         f_print(stderr, "-t\t\tgenerate RPC dispatch table\n");
1102         f_print(stderr, "-T\t\tgenerate code to support RPC dispatch tables\n");
1103         f_print(stderr, "-Y path\t\tdirectory name to find C preprocessor (cpp)\n");
1104
1105         exit(1);
1106 }