cmd/rpcgen/rpc_main.c

	rpc_main.c revision 8883f1c270cc8e33c18dd088e744840092b47bbb
/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */

/*
 * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */
/* Copyright (c) 1983, 1984, 1985, 1986, 1987, 1988, 1989 AT&T */
/* All Rights Reserved */
/*
 * University Copyright- Copyright (c) 1982, 1986, 1988
 * The Regents of the University of California
 * All Rights Reserved
 *
 * University Acknowledgment- Portions of this document are derived from
 * software developed by the University of California, Berkeley, and its
 * contributors.
 */

/*
 * rpc_main.c, Top level of the RPC protocol compiler.
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <strings.h>
#include <unistd.h>
#include <ctype.h>
#include <sys/types.h>
#include <sys/param.h>
#include <sys/file.h>
#include <sys/stat.h>
#include "rpc_parse.h"
#include "rpc_util.h"
#include "rpc_scan.h"


extern void write_sample_svc(definition *);
extern int write_sample_clnt(definition *);
extern void write_sample_clnt_main(void);
extern void reinitialize(void);
extern void crash(void);
extern void add_type(int, char *);
extern void add_sample_msg(void);

static void svc_output(char *, char *, int, char *);
static void clnt_output(char *, char *, int, char *);
static void c_output(char *, char *, int, char *);
static void mkfile_output(struct commandline *);
static void c_initialize(void);
static void h_output(char *, char *, int, char *);
static void s_output(int, char *[], char *, char *, int, char *, int, int);
static void l_output(char *, char *, int, char *);
static void t_output(char *, char *, int, char *);
static int do_registers(int, char *[]);
static uint_t parseargs(int, char *[], struct commandline *);
static void usage(void);
static void version_info(void);
static void options_usage(void);

#define EXTEND      1       /* alias for TRUE */
#define DONT_EXTEND 0       /* alias for FALSE */

#define SUNOS_CPP "/usr/lib/cpp"
static int cppDefined = 0;  /* explicit path for C preprocessor */


static char *cmdname;

static char *svcclosetime = "120";
static char *CPP = SUNOS_CPP;
static char CPPFLAGS[] = "-C";
static char pathbuf[MAXPATHLEN + 1];
static char *allv[] = {
    "rpcgen", "-s", "udp", "-s", "tcp",
};
static int allc = sizeof (allv)/sizeof (allv[0]);
static char *allnv[] = {
    "rpcgen", "-s", "netpath",
};
static int allnc = sizeof (allnv)/sizeof (allnv[0]);

/*
 * machinations for handling expanding argument list
 */
static void addarg(char *); /* add another argument to the list */
static void putarg(int, char *); /* put argument at specified location  */
static void clear_args(void);   /* clear argument list */
static void checkfiles(char *, char *); /* check if out file already exists */


#define ARGLISTLEN  20
#define FIXEDARGS   2

static char *arglist[ARGLISTLEN];
static int argcount = FIXEDARGS;


int nonfatalerrors; /* errors */
int inetdflag;  /* Support for inetd  is now the default */
int pmflag;     /* Support for port monitors */
int logflag;        /* Use syslog instead of fprintf for errors */
int tblflag;        /* Support for dispatch table file */
int mtflag = 0;     /* Support for MT */
int mtauto = 0;     /* Enable automatic mode */
int rflag = 1;      /* Eliminate tail recursion from structures */
#define INLINE 5
/* length at which to start doing an inline */

int inlinelen = INLINE;
/*
 * Length at which to start doing an inline. INLINE = default
 * if 0, no xdr_inline code
 */

int indefinitewait; /* If started by port monitors, hang till it wants */
int exitnow;        /* If started by port monitors, exit after the call */
int timerflag;      /* TRUE if !indefinite && !exitnow */
int newstyle;       /* newstyle of passing arguments (by value) */
int Cflag = 0;      /* ANSI C syntax */
int CCflag = 0;     /* C++ files */
static int allfiles;   /* generate all files */
int tirpcflag = 1;    /* generating code for tirpc, by default */
xdrfunc *xdrfunc_head = NULL; /* xdr function list */
xdrfunc *xdrfunc_tail = NULL; /* xdr function list */
pid_t childpid;


int
main(int argc, char *argv[])
{
    struct commandline cmd;

    (void) memset(&cmd, 0, sizeof (struct commandline));
    clear_args();
    if (!parseargs(argc, argv, &cmd))
        usage();
    /*
     * Only the client and server side stubs are likely to be customized,
     *  so in that case only, check if the outfile exists, and if so,
     *  print an error message and exit.
     */
    if (cmd.Ssflag || cmd.Scflag || cmd.makefileflag)
        checkfiles(cmd.infile, cmd.outfile);
    else
        checkfiles(cmd.infile, NULL);

    if (cmd.cflag) {
        c_output(cmd.infile, "-DRPC_XDR", DONT_EXTEND, cmd.outfile);
    } else if (cmd.hflag) {
        h_output(cmd.infile, "-DRPC_HDR", DONT_EXTEND, cmd.outfile);
    } else if (cmd.lflag) {
        l_output(cmd.infile, "-DRPC_CLNT", DONT_EXTEND, cmd.outfile);
    } else if (cmd.sflag || cmd.mflag || (cmd.nflag)) {
        s_output(argc, argv, cmd.infile, "-DRPC_SVC", DONT_EXTEND,
            cmd.outfile, cmd.mflag, cmd.nflag);
    } else if (cmd.tflag) {
        t_output(cmd.infile, "-DRPC_TBL", DONT_EXTEND, cmd.outfile);
    } else if (cmd.Ssflag) {
        svc_output(cmd.infile, "-DRPC_SERVER", DONT_EXTEND,
            cmd.outfile);
    } else if (cmd.Scflag) {
        clnt_output(cmd.infile, "-DRPC_CLIENT", DONT_EXTEND,
            cmd.outfile);
    } else if (cmd.makefileflag) {
        mkfile_output(&cmd);
    } else {
        /* the rescans are required, since cpp may effect input */
        c_output(cmd.infile, "-DRPC_XDR", EXTEND, "_xdr.c");
        reinitialize();
        h_output(cmd.infile, "-DRPC_HDR", EXTEND, ".h");
        reinitialize();
        l_output(cmd.infile, "-DRPC_CLNT", EXTEND, "_clnt.c");
        reinitialize();
        if (inetdflag || !tirpcflag)
            s_output(allc, allv, cmd.infile, "-DRPC_SVC", EXTEND,
                "_svc.c", cmd.mflag, cmd.nflag);
        else
            s_output(allnc, allnv, cmd.infile, "-DRPC_SVC",
                EXTEND, "_svc.c", cmd.mflag, cmd.nflag);
        if (tblflag) {
            reinitialize();
            t_output(cmd.infile, "-DRPC_TBL", EXTEND, "_tbl.i");
        }

        if (allfiles) {
            reinitialize();
            svc_output(cmd.infile, "-DRPC_SERVER", EXTEND,
                "_server.c");
            reinitialize();
            clnt_output(cmd.infile, "-DRPC_CLIENT", EXTEND,
                "_client.c");

        }
        if (allfiles || (cmd.makefileflag == 1)) {
            reinitialize();
            mkfile_output(&cmd);
        }

    }
    return (nonfatalerrors);
}


/*
 * add extension to filename
 */
static char *
extendfile(char *file, char *ext)
{
    char *res;
    char *p;

    res = malloc(strlen(file) + strlen(ext) + 1);
    if (res == NULL)
        abort();
    p = strrchr(file, '.');
    if (p == NULL)
        p = file + strlen(file);
    (void) strcpy(res, file);
    (void) strcpy(res + (p - file), ext);
    return (res);
}

/*
 * Open output file with given extension
 */
static void
open_output(char *infile, char *outfile)
{

    if (outfile == NULL) {
        fout = stdout;
        return;
    }

    if (infile != NULL && streq(outfile, infile)) {
        f_print(stderr,
            "%s: %s already exists.  No output generated.\n",
            cmdname, infile);
        crash();
    }
    fout = fopen(outfile, "w");
    if (fout == NULL) {
        f_print(stderr, "%s: unable to open ", cmdname);
        perror(outfile);
        crash();
    }
    record_open(outfile);

}

static void
add_warning(void)
{
    f_print(fout, "/*\n");
    f_print(fout, " * Please do not edit this file.\n");
    f_print(fout, " * It was generated using rpcgen.\n");
    f_print(fout, " */\n\n");
}

/* clear list of arguments */
static void
clear_args(void)
{
    int i;

    for (i = FIXEDARGS; i < ARGLISTLEN; i++)
        arglist[i] = NULL;
    argcount = FIXEDARGS;
}

/* make sure that a CPP exists */
static void
find_cpp(void)
{
    struct stat buf;

    if (stat(CPP, &buf) < 0)  { /* SVR4 or explicit cpp does not exist */
        if (cppDefined) {
            (void) fprintf(stderr,
                "cannot find C preprocessor: %s \n", CPP);
            crash();
        } else {    /* try the other one */
            CPP = SUNOS_CPP;
            if (stat(CPP, &buf) < 0) { /* can't find any cpp */
                (void) fprintf(stderr,
        "cannot find any C preprocessor (cpp)\n");
                crash();
            }
        }
    }
}

/*
 * Open input file with given define for C-preprocessor
 */
static void
open_input(char *infile, char *define)
{
    int pd[2];

    infilename = (infile == NULL) ? "<stdin>" : infile;
    (void) pipe(pd);
    switch (childpid = fork()) {
    case 0:
        find_cpp();
        putarg(0, CPP);
        putarg(1, CPPFLAGS);
        addarg(define);
        if (infile)
            addarg(infile);
        addarg((char *)NULL);
        (void) close(1);
        (void) dup2(pd[1], 1);
        (void) close(pd[0]);
        (void) execv(arglist[0], arglist);
        perror("execv");
        exit(1);
        /* NOTREACHED */
    case -1:
        perror("fork");
        exit(1);
    }
    (void) close(pd[1]);
    fin = fdopen(pd[0], "r");
    if (fin == NULL) {
        f_print(stderr, "%s: ", cmdname);
        perror(infilename);
        crash();
    }
}

/* valid tirpc nettypes */
static char *valid_ti_nettypes[] = {
    "netpath",
    "visible",
    "circuit_v",
    "datagram_v",
    "circuit_n",
    "datagram_n",
    "udp",
    "tcp",
    "raw",
    NULL
};

/* valid inetd nettypes */
static char *valid_i_nettypes[] = {
    "udp",
    "tcp",
    NULL
};

static int
check_nettype(char *name, char *list_to_check[])
{
    int i;
    for (i = 0; list_to_check[i] != NULL; i++) {
        if (strcmp(name, list_to_check[i]) == 0) {
            return (1);
        }
    }
    f_print(stderr, "illegal nettype :\'%s\'\n", name);
    return (0);
}

static char *
file_name(char *file, char *ext)
{
    char *temp;
    temp = extendfile(file, ext);

    if (access(temp, F_OK) != -1)
        return (temp);
    else
        return ((char *)" ");
}


static void
c_output(char *infile, char *define, int extend, char *outfile)
{
    definition *def;
    char *include;
    char *outfilename;
    long tell;

    c_initialize();
    open_input(infile, define);
    outfilename = extend ? extendfile(infile, outfile) : outfile;
    open_output(infile, outfilename);
    add_warning();
    if (infile && (include = extendfile(infile, ".h"))) {
        f_print(fout, "#include \"%s\"\n", include);
        free(include);
        /* .h file already contains rpc/rpc.h */
    } else
        f_print(fout, "#include <rpc/rpc.h>\n");
    /*
     * Include stdlib.h to support mem_alloc calls.
     */
    f_print(fout, "\n#ifndef _KERNEL\n");
    f_print(fout, "#include <stdlib.h>\n");
    f_print(fout, "#endif /* !_KERNEL */\n\n");
    tell = ftell(fout);
    while (def = get_definition()) {
        emit(def);
    }
    if (extend && tell == ftell(fout)) {
        (void) unlink(outfilename);
    }
}


static void
c_initialize(void)
{
    /*
     * add all the starting basic types.
     * We may need to add some derived types
     * if we need to generate INLINE macros.
     * These types are defined in rpc/types.h
     */
    add_type(1, "int");
    add_type(1, "long");
    add_type(1, "short");
    add_type(1, "bool");
    add_type(1, "u_int");
    add_type(1, "u_long");
    add_type(1, "u_short");
    add_type(1, "rpcprog_t");
    add_type(1, "rpcvers_t");
    add_type(1, "rpcproc_t");
    add_type(1, "rpcprot_t");
    add_type(1, "rpcport_t");
}

char rpcgen_table_dcl1[] = "struct rpcgen_table {\n";

char rpcgen_table_dcl2[] = "\txdrproc_t\txdr_arg;\n"
                "\tunsigned\tlen_arg;\n"
                "\txdrproc_t\txdr_res;\n"
                "\tunsigned\tlen_res;\n"
                "};\n";

char rpcgen_table_proc[] = "\tvoid\t*(*proc)();\n";

char rpcgen_table_proc_b[] = "\tchar\t*(*proc)();\n";


char *
generate_guard(char *pathname)
{
    char *filename, *guard, *tmp;

    filename = strrchr(pathname, '/');  /* find last component */
    filename = ((filename == 0) ? pathname : filename+1);
    guard = extendfile(filename, "_H_RPCGEN");

    /*
     * Guard must be an ANSI C identifier composed of
     * upper case letters, digits, or '_'.
     * Convert invalid characters to '_'.
     */
    for (tmp = guard; *tmp; tmp++) {
        if (!isalpha(*tmp) && !isdigit(*tmp)) {
            *tmp = '_';
            continue;
        }
        if (islower(*tmp))
            *tmp = toupper(*tmp);
    }

    /*
     * The first character must be a letter; the underscore '_'
     * counts as a letter.
     */
    if (!isalpha(guard[0]))
        guard[0] = '_';

    return (guard);
}

/*
 * Compile into an XDR header file
 */


static void
h_output(char *infile, char *define, int extend, char *outfile)
{
    definition *def;
    char *outfilename;
    long tell;
    char *guard;
    list *l;
    xdrfunc *xdrfuncp;
    int i;

    open_input(infile, define);
    outfilename =  extend ? extendfile(infile, outfile) : outfile;
    open_output(infile, outfilename);
    add_warning();
    if (outfilename || infile)
        guard = generate_guard(outfilename ? outfilename: infile);
    else
        guard = "STDIN_";

    f_print(fout, "#ifndef _%s\n#define _%s\n\n", guard, guard);

    f_print(fout, "#include <rpc/rpc.h>\n");

    if (mtflag) {
        f_print(fout, "#ifndef _KERNEL\n");
        f_print(fout, "#include <synch.h>\n");
        f_print(fout, "#include <thread.h>\n");
        f_print(fout, "#endif /* !_KERNEL */\n");
    };

    /* put the C++ support */
    if (Cflag && !CCflag) {
        f_print(fout, "\n#ifdef __cplusplus\n");
        f_print(fout, "extern \"C\" {\n");
        f_print(fout, "#endif\n\n");
    }

    /* put in a typedef for quadprecision. Only with Cflag */

    /*
     * declaration of struct rpcgen_table must go before
     *  the definition of arrays like *_1_table[]
     */
    if (tblflag) {
        f_print(fout, rpcgen_table_dcl1);
        if (tirpcflag)
            f_print(fout, rpcgen_table_proc);
        else
            f_print(fout, rpcgen_table_proc_b);
        f_print(fout, rpcgen_table_dcl2);
    }

    tell = ftell(fout);

    /* print data definitions */
    while (def = get_definition())
        print_datadef(def);

    /*
     * print function declarations.
     *  Do this after data definitions because they might be used as
     *  arguments for functions
     */
    for (l = defined; l != NULL; l = l->next)
        print_funcdef(l->val);
    /* Now  print all xdr func declarations */
    if (xdrfunc_head != NULL) {
        f_print(fout, "\n/* the xdr functions */\n");

        if (CCflag) {
            f_print(fout, "\n#ifdef __cplusplus\n");
            f_print(fout, "extern \"C\" {\n");
            f_print(fout, "#endif\n");
        }

        if (!Cflag) {
            xdrfuncp = xdrfunc_head;
            while (xdrfuncp != NULL) {
                print_xdr_func_def(xdrfuncp->name,
                    xdrfuncp->pointerp, 2);
                xdrfuncp = xdrfuncp->next;
            }
        } else {
            for (i = 1; i < 3; i++) {
                if (i == 1)
                    f_print(fout,
"\n#if defined(__STDC__) || defined(__cplusplus)\n");
                else
                    f_print(fout, "\n#else /* K&R C */\n");

                xdrfuncp = xdrfunc_head;
                while (xdrfuncp != NULL) {
                    print_xdr_func_def(xdrfuncp->name,
                        xdrfuncp->pointerp, i);
                    xdrfuncp = xdrfuncp->next;
                }
            }
            f_print(fout, "\n#endif /* K&R C */\n");
        }
    }

    if (extend && tell == ftell(fout)) {
        (void) unlink(outfilename);
    }

    if (Cflag) {
        f_print(fout, "\n#ifdef __cplusplus\n");
        f_print(fout, "}\n");
        f_print(fout, "#endif\n");
    }

    f_print(fout, "\n#endif /* !_%s */\n", guard);
}

/*
 * Compile into an RPC service
 */
static void
s_output(int argc, char *argv[], char *infile, char *define, int extend,
                    char *outfile, int nomain, int netflag)
{
    char *include;
    definition *def;
    int foundprogram = 0;
    char *outfilename;

    open_input(infile, define);
    outfilename = extend ? extendfile(infile, outfile) : outfile;
    open_output(infile, outfilename);
    add_warning();
    if (infile && (include = extendfile(infile, ".h"))) {
        f_print(fout, "#include \"%s\"\n", include);
        free(include);
    } else
        f_print(fout, "#include <rpc/rpc.h>\n");

    f_print(fout, "#include <stdio.h>\n");
    f_print(fout, "#include <stdlib.h> /* getenv, exit */\n");
    f_print(fout, "#include <signal.h>\n");

    if (Cflag) {
        f_print(fout,
        "#include <rpc/pmap_clnt.h> /* for pmap_unset */\n");
        f_print(fout, "#include <string.h> /* strcmp */\n");
    }
    if (strcmp(svcclosetime, "-1") == 0)
        indefinitewait = 1;
    else if (strcmp(svcclosetime, "0") == 0)
        exitnow = 1;
    else if (inetdflag || pmflag)
        timerflag = 1;

    if (!tirpcflag && inetdflag)
        f_print(fout, "#include <sys/termios.h> /* TIOCNOTTY */\n");
    if (Cflag && (inetdflag || pmflag))
        if (tirpcflag)
            f_print(fout, "#include <unistd.h> /* setsid */\n");
    if (tirpcflag)
        f_print(fout, "#include <sys/types.h>\n");

    f_print(fout, "#include <memory.h>\n");
    f_print(fout, "#include <stropts.h>\n");
    if (inetdflag || !tirpcflag) {
        f_print(fout, "#include <sys/socket.h>\n");
        f_print(fout, "#include <netinet/in.h>\n");
        f_print(fout, "#include <rpc/svc_soc.h>\n");
    }

    if ((netflag || pmflag) && tirpcflag && !nomain)
        f_print(fout, "#include <netconfig.h>\n");
    if (tirpcflag)
        f_print(fout, "#include <sys/resource.h> /* rlimit */\n");
    if (logflag || inetdflag || pmflag)
        f_print(fout, "#include <syslog.h>\n");

    /* for ANSI-C */
    if (Cflag)
        f_print(fout,
            "\n#ifndef SIG_PF\n#define  SIG_PF void(*)\
(int)\n#endif\n");

    f_print(fout, "\n#ifdef DEBUG\n#define  RPC_SVC_FG\n#endif\n");
    if (timerflag)
        f_print(fout, "\n#define    _RPCSVC_CLOSEDOWN %s\n",
            svcclosetime);
    while (def = get_definition())
        foundprogram |= (def->def_kind == DEF_PROGRAM);
    if (extend && !foundprogram) {
        (void) unlink(outfilename);
        return;
    }
    write_most(infile, netflag, nomain);
    if (!nomain) {
        if (!do_registers(argc, argv)) {
            if (outfilename)
                (void) unlink(outfilename);
            usage();
        }
        write_rest();
    }
}

/*
 * generate client side stubs
 */
static void
l_output(char *infile, char *define, int extend, char *outfile)
{
    char *include;
    definition *def;
    int foundprogram = 0;
    char *outfilename;

    open_input(infile, define);
    outfilename = extend ? extendfile(infile, outfile) : outfile;
    open_output(infile, outfilename);
    add_warning();
    if (Cflag)
        f_print(fout, "#include <memory.h> /* for memset */\n");
    if (infile && (include = extendfile(infile, ".h"))) {
        f_print(fout, "#include \"%s\"\n", include);
        free(include);
    } else
        f_print(fout, "#include <rpc/rpc.h>\n");

    f_print(fout, "#ifndef _KERNEL\n");
    f_print(fout, "#include <stdio.h>\n");
    f_print(fout, "#include <stdlib.h> /* getenv, exit */\n");
    f_print(fout, "#endif /* !_KERNEL */\n");

    while (def = get_definition())
        foundprogram |= (def->def_kind == DEF_PROGRAM);
    if (extend && !foundprogram) {
        (void) unlink(outfilename);
        return;
    }
    write_stubs();
}

/*
 * generate the dispatch table
 */
static void
t_output(char *infile, char *define, int extend, char *outfile)
{
    definition *def;
    int foundprogram = 0;
    char *outfilename;

    open_input(infile, define);
    outfilename = extend ? extendfile(infile, outfile) : outfile;
    open_output(infile, outfilename);
    add_warning();
    while (def = get_definition()) {
        foundprogram |= (def->def_kind == DEF_PROGRAM);
    }
    if (extend && !foundprogram) {
        (void) unlink(outfilename);
        return;
    }
    write_tables();
}

/* sample routine for the server template */
static void
svc_output(char *infile, char *define, int extend, char *outfile)
{
    definition *def;
    char *include;
    char *outfilename;
    long tell;
    open_input(infile, define);
    outfilename = extend ? extendfile(infile, outfile) : outfile;
    checkfiles(infile, outfilename);
    /*
     * Check if outfile already exists.
     * if so, print an error message and exit
     */
    open_output(infile, outfilename);
    add_sample_msg();

    if (infile && (include = extendfile(infile, ".h"))) {
        f_print(fout, "#include \"%s\"\n", include);
        free(include);
    } else {
        f_print(fout, "#include <rpc/rpc.h>\n");
    }

    f_print(fout, "#include <stdio.h>\n");
    f_print(fout, "#include <stdlib.h> /* getenv, exit */\n");
    f_print(fout, "#include <signal.h>\n");

    tell = ftell(fout);
    while (def = get_definition())
        write_sample_svc(def);
    if (extend && tell == ftell(fout))
        (void) unlink(outfilename);
}

/* sample main routine for client */
static void
clnt_output(char *infile, char *define, int extend, char *outfile)
{
    definition *def;
    char *include;
    char *outfilename;
    long tell;
    int has_program = 0;

    open_input(infile, define);
    outfilename = extend ? extendfile(infile, outfile) : outfile;
    checkfiles(infile, outfilename);
    /*
     * Check if outfile already exists.
     * if so, print an error message and exit
     */

    open_output(infile, outfilename);
    add_sample_msg();
    if (infile && (include = extendfile(infile, ".h"))) {
        f_print(fout, "#include \"%s\"\n", include);
        free(include);
    } else
        f_print(fout, "#include <rpc/rpc.h>\n");

    f_print(fout, "#include <stdio.h>\n");
    f_print(fout, "#include <stdlib.h> /* getenv, exit */\n");

    tell = ftell(fout);
    while (def = get_definition())
        has_program += write_sample_clnt(def);

    if (has_program)
        write_sample_clnt_main();

    if (extend && tell == ftell(fout))
        (void) unlink(outfilename);
}


static void
mkfile_output(struct commandline *cmd)
{
    char *mkfilename, *clientname, *clntname, *xdrname, *hdrname;
    char *servername, *svcname, *servprogname, *clntprogname;
    char *temp;

    svcname = file_name(cmd->infile, "_svc.c");
    clntname = file_name(cmd->infile, "_clnt.c");
    xdrname = file_name(cmd->infile, "_xdr.c");
    hdrname = file_name(cmd->infile, ".h");


    if (allfiles) {
        servername = extendfile(cmd->infile, "_server.c");
        clientname = extendfile(cmd->infile, "_client.c");
    } else {
        servername = " ";
        clientname = " ";
    }
    servprogname = extendfile(cmd->infile, "_server");
    clntprogname = extendfile(cmd->infile, "_client");

    if (allfiles) {
        mkfilename = malloc(strlen("makefile.") +
            strlen(cmd->infile) + 1);
        if (mkfilename == NULL) {
            f_print(stderr, "Out of memory!\n");
            return;
        }
        temp = (char *)rindex(cmd->infile, '.');
        (void) strcpy(mkfilename, "makefile.");
        (void) strncat(mkfilename, cmd->infile,
            (temp - cmd->infile));
    } else
        mkfilename = cmd->outfile;


    checkfiles(NULL, mkfilename);
    open_output(NULL, mkfilename);

    f_print(fout, "\n# This is a template makefile generated\
        by rpcgen \n");

    f_print(fout, "\n# Parameters \n\n");

    f_print(fout, "CLIENT = %s\nSERVER = %s\n\n",
        clntprogname, servprogname);
    f_print(fout, "SOURCES_CLNT.c = \nSOURCES_CLNT.h = \n");
    f_print(fout, "SOURCES_SVC.c = \nSOURCES_SVC.h = \n");
    f_print(fout, "SOURCES.x = %s\n\n", cmd->infile);
    f_print(fout, "TARGETS_SVC.c = %s %s %s \n",
        svcname, servername, xdrname);
    f_print(fout, "TARGETS_CLNT.c = %s %s %s \n",
        clntname, clientname, xdrname);
    f_print(fout, "TARGETS = %s %s %s %s %s %s\n\n",
        hdrname, xdrname, clntname,
        svcname, clientname, servername);

    f_print(fout, "OBJECTS_CLNT = $(SOURCES_CLNT.c:%%.c=%%.o) "
        "$(TARGETS_CLNT.c:%%.c=%%.o) ");

    f_print(fout, "\nOBJECTS_SVC = $(SOURCES_SVC.c:%%.c=%%.o) "
        "$(TARGETS_SVC.c:%%.c=%%.o) ");


    f_print(fout, "\n# Compiler flags \n");
    if (mtflag)
        f_print(fout, "\nCPPFLAGS += -D_REENTRANT\n"
            "CFLAGS += -g\nLDLIBS += -lnsl\n");
    else
        f_print(fout, "\nCFLAGS += -g \nLDLIBS += -lnsl\n");
    f_print(fout, "RPCGENFLAGS = \n");

    f_print(fout, "\n# Targets \n\n");

    f_print(fout, "all : $(CLIENT) $(SERVER)\n\n");
    f_print(fout, "$(TARGETS) : $(SOURCES.x) \n");
    f_print(fout, "\trpcgen $(RPCGENFLAGS) $(SOURCES.x)\n\n");
    f_print(fout, "$(OBJECTS_CLNT) : $(SOURCES_CLNT.c) $(SOURCES_CLNT.h) \
$(TARGETS_CLNT.c) \n\n");

    f_print(fout, "$(OBJECTS_SVC) : $(SOURCES_SVC.c) $(SOURCES_SVC.h) \
$(TARGETS_SVC.c) \n\n");
    f_print(fout, "$(CLIENT) : $(OBJECTS_CLNT) \n");
    f_print(fout, "\t$(LINK.c) -o $(CLIENT) $(OBJECTS_CLNT) \
$(LDLIBS) \n\n");
    f_print(fout, "$(SERVER) : $(OBJECTS_SVC) \n");
    f_print(fout, "\t$(LINK.c) -o $(SERVER) $(OBJECTS_SVC) $(LDLIBS)\n\n ");
    f_print(fout, "clean:\n\t $(RM) core $(TARGETS) $(OBJECTS_CLNT) \
$(OBJECTS_SVC) $(CLIENT) $(SERVER)\n\n");
}


/*
 * Perform registrations for service output
 * Return 0 if failed; 1 otherwise.
 */
static int
do_registers(int argc, char *argv[])
{
    int i;

    if (inetdflag || !tirpcflag) {
        for (i = 1; i < argc; i++) {
            if (streq(argv[i], "-s")) {
                if (!check_nettype(argv[i + 1],
                    valid_i_nettypes))
                    return (0);
                write_inetd_register(argv[i + 1]);
                i++;
            }
        }
    } else {
        for (i = 1; i < argc; i++)
            if (streq(argv[i], "-s")) {
                if (!check_nettype(argv[i + 1],
                    valid_ti_nettypes))
                    return (0);
                write_nettype_register(argv[i + 1]);
                i++;
            } else if (streq(argv[i], "-n")) {
                write_netid_register(argv[i + 1]);
                i++;
            }
    }
    return (1);
}

/*
 * Add another argument to the arg list
 */
static void
addarg(char *cp)
{
    if (argcount >= ARGLISTLEN) {
        f_print(stderr, "rpcgen: too many defines\n");
        crash();
        /*NOTREACHED*/
    }
    arglist[argcount++] = cp;
}

static void
putarg(int where, char *cp)
{
    if (where >= ARGLISTLEN) {
        f_print(stderr, "rpcgen: arglist coding error\n");
        crash();
        /*NOTREACHED*/
    }
    arglist[where] = cp;
}

/*
 * if input file is stdin and an output file is specified then complain
 * if the file already exists. Otherwise the file may get overwritten
 * If input file does not exist, exit with an error
 */
static void
checkfiles(char *infile, char *outfile)
{
    struct stat buf;

    if (infile) {       /* infile ! = NULL */
        if (stat(infile, &buf) < 0) {
            perror(infile);
            crash();
        }
    }
    if (outfile) {
        if (stat(outfile, &buf) < 0)
            return; /* file does not exist */
        f_print(stderr,
            "file '%s' already exists and may be overwritten\n",
            outfile);
        crash();
    }
}

/*
 * Parse command line arguments
 */
static uint_t
parseargs(int argc, char *argv[], struct commandline *cmd)
{
    int i;
    int j;
    char c, ch;
    char flag[(1 << 8 * sizeof (char))];
    int nflags;

    cmdname = argv[0];
    cmd->infile = cmd->outfile = NULL;
    if (argc < 2)
        return (0);
    allfiles = 0;
    flag['c'] = 0;
    flag['h'] = 0;
    flag['l'] = 0;
    flag['m'] = 0;
    flag['o'] = 0;
    flag['s'] = 0;
    flag['n'] = 0;
    flag['t'] = 0;
    flag['S'] = 0;
    flag['C'] = 0;
    flag['M'] = 0;

    for (i = 1; i < argc; i++) {
        if (argv[i][0] != '-') {
            if (cmd->infile) {
                f_print(stderr,
    "Cannot specify more than one input file.\n");

                return (0);
            }
            cmd->infile = argv[i];
        } else {
            for (j = 1; argv[i][j] != 0; j++) {
                c = argv[i][j];
                switch (c) {
                case 'a':
                    allfiles = 1;
                    break;
                case 'c':
                case 'h':
                case 'l':
                case 'm':
                case 't':
                    if (flag[c])
                        return (0);
                    flag[c] = 1;
                    break;
                case 'S':
                    /*
                     * sample flag: Ss or Sc.
                     *  Ss means set flag['S'];
                     *  Sc means set flag['C'];
                     *  Sm means set flag['M'];
                     */
                    ch = argv[i][++j]; /* get next char */
                    if (ch == 's')
                        ch = 'S';
                    else if (ch == 'c')
                        ch = 'C';
                    else if (ch == 'm')
                        ch = 'M';
                    else
                        return (0);

                    if (flag[ch])
                        return (0);
                    flag[ch] = 1;
                    break;
                case 'C': /* ANSI C syntax */
                    Cflag = 1;
                    ch = argv[i][j+1]; /* get next char */

                    if (ch != 'C')
                        break;
                    CCflag = 1;
                    break;
                case 'b':
                    /*
                     *  Turn TIRPC flag off for
                     *  generating backward compatible
                     *  code
                     */
                    tirpcflag = 0;
                    break;

                case 'I':
                    inetdflag = 1;
                    break;
                case 'N':
                    newstyle = 1;
                    break;
                case 'L':
                    logflag = 1;
                    break;
                case 'K':
                    if (++i == argc)
                        return (0);
                    svcclosetime = argv[i];
                    goto nextarg;
                case 'T':
                    tblflag = 1;
                    break;
                case 'A':
                    mtauto = 1;
                    /* FALLTHRU */
                case 'M':
                    mtflag = 1;
                    break;
                case 'i' :
                    if (++i == argc)
                        return (0);
                    inlinelen = atoi(argv[i]);
                    goto nextarg;
                case 'n':
                case 'o':
                case 's':
                    if (argv[i][j - 1] != '-' ||
                        argv[i][j + 1] != 0)
                        return (0);
                    flag[c] = 1;
                    if (++i == argc)
                        return (0);
                    if (c == 'o') {
                        if (cmd->outfile)
                            return (0);
                        cmd->outfile = argv[i];
                    }
                    goto nextarg;
                case 'D':
                    if (argv[i][j - 1] != '-')
                        return (0);
                    (void) addarg(argv[i]);
                    goto nextarg;
                case 'v':
                    version_info();
                    return (0);
                case 'Y':
                    if (++i == argc)
                        return (0);
                    (void) strcpy(pathbuf, argv[i]);
                    (void) strcat(pathbuf, "/cpp");
                    CPP = pathbuf;
                    cppDefined = 1;
                    goto nextarg;
                case 'r':
                    rflag = !rflag;
                    break;
                default:
                    return (0);
                }
            }
        nextarg:
            ;
        }
    }

    cmd->cflag = flag['c'];
    cmd->hflag = flag['h'];
    cmd->lflag = flag['l'];
    cmd->mflag = flag['m'];
    cmd->nflag = flag['n'];
    cmd->sflag = flag['s'];
    cmd->tflag = flag['t'];
    cmd->Ssflag = flag['S'];
    cmd->Scflag = flag['C'];
    cmd->makefileflag = flag['M'];

    if (tirpcflag) {
        if (inetdflag) {
            f_print(stderr,
                "Cannot use -I flag without -b flag.\n");
            return (0);
        }
        pmflag = 1;
    } else {        /* 4.1 mode */
        pmflag = 0; /* set pmflag only in tirpcmode */
        inetdflag = 1;  /* inetdflag is TRUE by default */
        if (cmd->nflag) { /* netid needs TIRPC */
            f_print(stderr,
                "Cannot use netid flag without TIRPC.\n");
            return (0);
        }
    }

    if (newstyle && (tblflag || cmd->tflag)) {
        f_print(stderr, "Cannot use table flags with newstyle.\n");
        return (0);
    }

    /* check no conflicts with file generation flags */
    nflags = cmd->cflag + cmd->hflag + cmd->lflag + cmd->mflag +
        cmd->sflag + cmd->nflag + cmd->tflag + cmd->Ssflag +
        cmd->Scflag + cmd->makefileflag;

    if (nflags == 0) {
        if (cmd->outfile != NULL || cmd->infile == NULL)
            return (0);
    } else if (cmd->infile == NULL &&
        (cmd->Ssflag || cmd->Scflag || cmd->makefileflag)) {
        f_print(stderr, "\"infile\" is required for template"
            " generation flags.\n");
        return (0);
    }
    if (nflags > 1) {
        f_print(stderr,
            "Cannot have more than one file generation flag.\n");
        return (0);
    }
    return (1);
}

static void
usage(void)
{
    f_print(stderr, "%s  (%d.%d)\n", cmdname, RPCGEN_MAJOR, RPCGEN_MINOR);
    f_print(stderr, "usage:  %s infile\n", cmdname);
    f_print(stderr, "\t%s [-abCLNTMA] [-Dname[=value]] [-i size]"
        " [-I [-K seconds]] [-Y path] infile\n", cmdname);
    f_print(stderr, "\t%s [-c | -h | -l | -m | -t | -Sc | -Ss | -Sm]"
        " [-o outfile] [infile]\n", cmdname);
    f_print(stderr, "\t%s [-s nettype]* [-o outfile] [infile]\n", cmdname);
    f_print(stderr, "\t%s [-n netid]* [-o outfile] [infile]\n", cmdname);
    options_usage();
    exit(1);
}

static void
version_info(void)
{
    f_print(stderr, "%s %d.%d\n", cmdname, RPCGEN_MAJOR, RPCGEN_MINOR);
    exit(1);
}

static void
options_usage(void)
{
    /* BEGIN CSTYLED */
    f_print(stderr, "options:\n");
    f_print(stderr, "-a\t\tgenerate all files, including samples\n");
    f_print(stderr, "-A\t\tgenerate code to enable automatic MT mode\n");
    f_print(stderr, "-b\t\tbackward compatibility mode (generates code"
            " for SunOS 4.X)\n");
    f_print(stderr, "-c\t\tgenerate XDR routines\n");
    f_print(stderr, "-C\t\tANSI C mode\n");
    f_print(stderr, "-Dname[=value]\tdefine a symbol (same as #define)\n");
    f_print(stderr, "-h\t\tgenerate header file\n");
    f_print(stderr, "-i size\t\tsize at which to start generating"
            " inline code\n");
    f_print(stderr, "-I\t\tgenerate code for inetd support in server"
            " (for SunOS 4.X)\n");
    f_print(stderr, "-K seconds\tserver exits after K seconds of"
            " inactivity\n");
    f_print(stderr, "-l\t\tgenerate client side stubs\n");
    f_print(stderr, "-L\t\tserver errors will be printed to syslog\n");
    f_print(stderr, "-m\t\tgenerate server side stubs\n");
    f_print(stderr, "-M\t\tgenerate MT-safe code\n");
    f_print(stderr, "-n netid\tgenerate server code that supports"
            " named netid\n");
    f_print(stderr, "-N\t\tsupports multiple arguments and"
            " call-by-value\n");
    f_print(stderr, "-o outfile\tname of the output file\n");
    f_print(stderr, "-s nettype\tgenerate server code that supports named"
            " nettype\n");
    f_print(stderr, "-Sc\t\tgenerate sample client code that uses remote"
            " procedures\n");
    f_print(stderr, "-Ss\t\tgenerate sample server code that defines"
            " remote procedures\n");
    f_print(stderr, "-Sm \t\tgenerate makefile template \n");

    f_print(stderr, "-t\t\tgenerate RPC dispatch table\n");
    f_print(stderr, "-T\t\tgenerate code to support RPC dispatch tables\n");
    f_print(stderr, "-v\t\tprint version information and exit\n");
    f_print(stderr, "-Y path\t\tpath where cpp is found\n");
    /* END CSTYLED */
    exit(1);
}