/*      $NetBSD: rpc_parse.c,v 1.23 2022/01/11 22:45:56 andvar Exp $    */
/*
* Sun RPC is a product of Sun Microsystems, Inc. and is provided for
* unrestricted use provided that this legend is included on all tape
* media and as a part of the software program in whole or part.  Users
* may copy or modify Sun RPC without charge, but are not authorized
* to license or distribute it to anyone else except as part of a product or
* program developed by the user or with the express written consent of
* Sun Microsystems, Inc.
*
* SUN RPC IS PROVIDED AS IS WITH NO WARRANTIES OF ANY KIND INCLUDING THE
* WARRANTIES OF DESIGN, MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
* PURPOSE, OR ARISING FROM A COURSE OF DEALING, USAGE OR TRADE PRACTICE.
*
* Sun RPC is provided with no support and without any obligation on the
* part of Sun Microsystems, Inc. to assist in its use, correction,
* modification or enhancement.
*
* SUN MICROSYSTEMS, INC. SHALL HAVE NO LIABILITY WITH RESPECT TO THE
* INFRINGEMENT OF COPYRIGHTS, TRADE SECRETS OR ANY PATENTS BY SUN RPC
* OR ANY PART THEREOF.
*
* In no event will Sun Microsystems, Inc. be liable for any lost revenue
* or profits or other special, indirect and consequential damages, even if
* Sun has been advised of the possibility of such damages.
*
* Sun Microsystems, Inc.
* 2550 Garcia Avenue
* Mountain View, California  94043
*/

#if HAVE_NBTOOL_CONFIG_H
#include "nbtool_config.h"
#endif

#include <sys/cdefs.h>
#if defined(__RCSID) && !defined(lint)
#if 0
static char sccsid[] = "@(#)rpc_parse.c 1.8 89/02/22 (C) 1987 SMI";
#else
__RCSID("$NetBSD: rpc_parse.c,v 1.23 2022/01/11 22:45:56 andvar Exp $");
#endif
#endif

/*
* rpc_parse.c, Parser for the RPC protocol compiler
* Copyright (C) 1987 Sun Microsystems, Inc.
*/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "rpc/types.h"
#include "rpc_scan.h"
#include "rpc_parse.h"
#include "rpc_util.h"

#define ARGNAME "arg"

static void isdefined(definition *);
static void def_struct(definition *);
static void def_program(definition *);
static void def_enum(definition *);
static void def_const(definition *);
static void def_union(definition *);
static void check_type_name(const char *, int);
static void def_typedef(definition *);
static void get_declaration(declaration *, defkind);
static void get_prog_declaration(declaration *, defkind, int);
static void get_type(const char **, const char **, defkind);
static void unsigned_dec(const char **);

/*
* return the next definition you see
*/
definition *
get_definition(void)
{
       definition *defp;
       token   tok;

       defp = ALLOC(definition);
       get_token(&tok);
       switch (tok.kind) {
       case TOK_STRUCT:
               def_struct(defp);
               break;
       case TOK_UNION:
               def_union(defp);
               break;
       case TOK_TYPEDEF:
               def_typedef(defp);
               break;
       case TOK_ENUM:
               def_enum(defp);
               break;
       case TOK_PROGRAM:
               def_program(defp);
               break;
       case TOK_CONST:
               def_const(defp);
               break;
       case TOK_EOF:
               free(defp);
               return (NULL);
       default:
               error("Expected definition keyword");
       }
       scan(TOK_SEMICOLON, &tok);
       isdefined(defp);
       return (defp);
}

static void
isdefined(definition *defp)
{
       STOREVAL(&defined, defp);
}

static void
def_struct(definition *defp)
{
       token   tok;
       declaration dec;
       decl_list *decls;
       decl_list **tailp;

       defp->def_kind = DEF_STRUCT;

       scan(TOK_IDENT, &tok);
       defp->def_name = tok.str;
       scan(TOK_LBRACE, &tok);
       tailp = &defp->def.st.decls;
       do {
               get_declaration(&dec, DEF_STRUCT);
               decls = ALLOC(decl_list);
               decls->decl = dec;
               *tailp = decls;
               tailp = &decls->next;
               scan(TOK_SEMICOLON, &tok);
               peek(&tok);
       } while (tok.kind != TOK_RBRACE);
       get_token(&tok);
       *tailp = NULL;
}

static void
def_program(definition *defp)
{
       token   tok;
       declaration dec;
       decl_list *decls;
       decl_list **tailp;
       version_list *vlist;
       version_list **vtailp;
       proc_list *plist;
       proc_list **ptailp;
       int     num_args;
       bool_t  isvoid = FALSE; /* whether first argument is void */
       defp->def_kind = DEF_PROGRAM;
       scan(TOK_IDENT, &tok);
       defp->def_name = tok.str;
       scan(TOK_LBRACE, &tok);
       vtailp = &defp->def.pr.versions;
       tailp = &defp->def.st.decls;
       scan(TOK_VERSION, &tok);
       do {
               scan(TOK_IDENT, &tok);
               vlist = ALLOC(version_list);
               vlist->vers_name = tok.str;
               scan(TOK_LBRACE, &tok);
               ptailp = &vlist->procs;
               do {
                       /* get result type */
                       plist = ALLOC(proc_list);
                       get_type(&plist->res_prefix, &plist->res_type,
                           DEF_PROGRAM);
                       if (streq(plist->res_type, "opaque")) {
                               error("Illegal result type");
                       }
                       scan(TOK_IDENT, &tok);
                       plist->proc_name = tok.str;
                       scan(TOK_LPAREN, &tok);
                       /* get args - first one */
                       num_args = 1;
                       isvoid = FALSE;
                       /* type of DEF_PROGRAM in the first
                        * get_prog_declaration and DEF_STRUCT in the next
                        * allows void as argument if it is the only argument */
                       get_prog_declaration(&dec, DEF_PROGRAM, num_args);
                       if (streq(dec.type, "void"))
                               isvoid = TRUE;
                       decls = ALLOC(decl_list);
                       plist->args.decls = decls;
                       decls->decl = dec;
                       tailp = &decls->next;
                       /* get args */
                       while (peekscan(TOK_COMMA, &tok)) {
                               num_args++;
                               get_prog_declaration(&dec, DEF_STRUCT,
                                   num_args);
                               decls = ALLOC(decl_list);
                               decls->decl = dec;
                               *tailp = decls;
                               if (streq(dec.type, "void"))
                                       isvoid = TRUE;
                               tailp = &decls->next;
                       }
                       /* multiple arguments are only allowed in newstyle */
                       if (!newstyle && num_args > 1) {
                               error("Only one argument is allowed");
                       }
                       if (isvoid && num_args > 1) {
                               error("Illegal use of void in program definition");
                       }
                       *tailp = NULL;
                       scan(TOK_RPAREN, &tok);
                       scan(TOK_EQUAL, &tok);
                       scan_num(&tok);
                       scan(TOK_SEMICOLON, &tok);
                       plist->proc_num = tok.str;
                       plist->arg_num = num_args;
                       *ptailp = plist;
                       ptailp = &plist->next;
                       peek(&tok);
               } while (tok.kind != TOK_RBRACE);
               *ptailp = NULL;
               *vtailp = vlist;
               vtailp = &vlist->next;
               scan(TOK_RBRACE, &tok);
               scan(TOK_EQUAL, &tok);
               scan_num(&tok);
               vlist->vers_num = tok.str;
               /* make the argument structure name for each arg */
               for (plist = vlist->procs; plist != NULL;
                   plist = plist->next) {
                       plist->args.argname = make_argname(plist->proc_name,
                           vlist->vers_num);
                       /* free the memory ?? */
               }
               scan(TOK_SEMICOLON, &tok);
               scan2(TOK_VERSION, TOK_RBRACE, &tok);
       } while (tok.kind == TOK_VERSION);
       scan(TOK_EQUAL, &tok);
       scan_num(&tok);
       defp->def.pr.prog_num = tok.str;
       *vtailp = NULL;
}


static void
def_enum(definition *defp)
{
       token   tok;
       enumval_list *elist;
       enumval_list **tailp;

       defp->def_kind = DEF_ENUM;
       scan(TOK_IDENT, &tok);
       defp->def_name = tok.str;
       scan(TOK_LBRACE, &tok);
       tailp = &defp->def.en.vals;
       do {
               scan(TOK_IDENT, &tok);
               elist = ALLOC(enumval_list);
               elist->name = tok.str;
               elist->assignment = NULL;
               scan3(TOK_COMMA, TOK_RBRACE, TOK_EQUAL, &tok);
               if (tok.kind == TOK_EQUAL) {
                       scan_num(&tok);
                       elist->assignment = tok.str;
                       scan2(TOK_COMMA, TOK_RBRACE, &tok);
               }
               *tailp = elist;
               tailp = &elist->next;
       } while (tok.kind != TOK_RBRACE);
       *tailp = NULL;
}

static void
def_const(definition *defp)
{
       token   tok;

       defp->def_kind = DEF_CONST;
       scan(TOK_IDENT, &tok);
       defp->def_name = tok.str;
       scan(TOK_EQUAL, &tok);
       scan2(TOK_IDENT, TOK_STRCONST, &tok);
       defp->def.co = tok.str;
}

static void
def_union(definition *defp)
{
       token   tok;
       declaration dec;
       case_list *cases;
       case_list **tailp;

       defp->def_kind = DEF_UNION;
       scan(TOK_IDENT, &tok);
       defp->def_name = tok.str;
       scan(TOK_SWITCH, &tok);
       scan(TOK_LPAREN, &tok);
       get_declaration(&dec, DEF_UNION);
       defp->def.un.enum_decl = dec;
       tailp = &defp->def.un.cases;
       scan(TOK_RPAREN, &tok);
       scan(TOK_LBRACE, &tok);
       scan(TOK_CASE, &tok);
       while (tok.kind == TOK_CASE) {
               scan2(TOK_IDENT, TOK_CHARCONST, &tok);
               cases = ALLOC(case_list);
               cases->case_name = tok.str;
               scan(TOK_COLON, &tok);
               /* now peek at next token */
               if (peekscan(TOK_CASE, &tok)) {

                       do {
                               scan2(TOK_IDENT, TOK_CHARCONST, &tok);
                               cases->contflag = 1;    /* continued case
                                                        * statement */
                               *tailp = cases;
                               tailp = &cases->next;
                               cases = ALLOC(case_list);
                               cases->case_name = tok.str;
                               scan(TOK_COLON, &tok);

                       } while (peekscan(TOK_CASE, &tok));
               }
               get_declaration(&dec, DEF_UNION);
               cases->case_decl = dec;
               cases->contflag = 0;    /* no continued case statement */
               *tailp = cases;
               tailp = &cases->next;
               scan(TOK_SEMICOLON, &tok);

               scan3(TOK_CASE, TOK_DEFAULT, TOK_RBRACE, &tok);
       }
       *tailp = NULL;
       if (tok.kind == TOK_DEFAULT) {
               scan(TOK_COLON, &tok);
               get_declaration(&dec, DEF_UNION);
               defp->def.un.default_decl = ALLOC(declaration);
               *defp->def.un.default_decl = dec;
               scan(TOK_SEMICOLON, &tok);
               scan(TOK_RBRACE, &tok);
       } else {
               defp->def.un.default_decl = NULL;
       }
}

static const char *const reserved_words[] = {
       "array",
       "bytes",
       "destroy",
       "free",
       "getpos",
       "inline",
       "pointer",
       "reference",
       "setpos",
       "sizeof",
       "union",
       "vector",
       NULL
};

static const char *const reserved_types[] = {
       "opaque",
       "string",
       NULL
};
/* check that the given name is not one that would eventually result in
  xdr routines that would conflict with internal XDR routines. */
static void
check_type_name(const char *name, int new_type)
{
       int     i;

       for (i = 0; reserved_words[i] != NULL; i++) {
               if (strcmp(name, reserved_words[i]) == 0) {
                       error("Illegal (reserved) name '%s' in type definition", name);
               }
       }
       if (new_type) {
               for (i = 0; reserved_types[i] != NULL; i++) {
                       if (strcmp(name, reserved_types[i]) == 0) {
                               error("Illegal (reserved) name '%s' in type definition", name);
                       }
               }
       }
}

static void
def_typedef(definition *defp)
{
       declaration dec;

       defp->def_kind = DEF_TYPEDEF;
       get_declaration(&dec, DEF_TYPEDEF);
       defp->def_name = dec.name;
       check_type_name(dec.name, 1);
       defp->def.ty.old_prefix = dec.prefix;
       defp->def.ty.old_type = dec.type;
       defp->def.ty.rel = dec.rel;
       defp->def.ty.array_max = dec.array_max;
}

static void
get_declaration(declaration *dec, defkind dkind)
{
       token   tok;

       get_type(&dec->prefix, &dec->type, dkind);
       dec->rel = REL_ALIAS;
       if (streq(dec->type, "void")) {
               return;
       }
       check_type_name(dec->type, 0);

       scan2(TOK_STAR, TOK_IDENT, &tok);
       if (tok.kind == TOK_STAR) {
               dec->rel = REL_POINTER;
               scan(TOK_IDENT, &tok);
       }
       dec->name = tok.str;
       if (peekscan(TOK_LBRACKET, &tok)) {
               if (dec->rel == REL_POINTER) {
                       error("No array-of-pointer declarations -- use typedef");
               }
               dec->rel = REL_VECTOR;
               scan_num(&tok);
               dec->array_max = tok.str;
               scan(TOK_RBRACKET, &tok);
       } else
               if (peekscan(TOK_LANGLE, &tok)) {
                       if (dec->rel == REL_POINTER) {
                               error("No array-of-pointer declarations -- use typedef");
                       }
                       dec->rel = REL_ARRAY;
                       if (peekscan(TOK_RANGLE, &tok)) {
                               dec->array_max = "(unsigned int)~0";
                               /* unspecified size, use * max */
                       } else {
                               scan_num(&tok);
                               dec->array_max = tok.str;
                               scan(TOK_RANGLE, &tok);
                       }
               }
       if (streq(dec->type, "opaque")) {
               if (dec->rel != REL_ARRAY && dec->rel != REL_VECTOR) {
                       error("Array declaration expected");
               }
       } else
               if (streq(dec->type, "string")) {
                       if (dec->rel != REL_ARRAY) {
                               error("Variable-length array declaration expected");
                       }
               }
}

static void
get_prog_declaration(declaration *dec, defkind dkind, int num /* arg number */)
{
       token   tok;
       char    name[255];      /* argument name */

       if (dkind == DEF_PROGRAM) {
               peek(&tok);
               if (tok.kind == TOK_RPAREN) {   /* no arguments */
                       dec->rel = REL_ALIAS;
                       dec->type = "void";
                       dec->prefix = NULL;
                       dec->name = NULL;
                       return;
               }
       }
       get_type(&dec->prefix, &dec->type, dkind);
       dec->rel = REL_ALIAS;
       if (peekscan(TOK_IDENT, &tok))  /* optional name of argument */
               strcpy(name, tok.str);
       else
               sprintf(name, "%s%d", ARGNAME, num);    /* default name of
                                                        * argument */

       dec->name = strdup(name);

       if (streq(dec->type, "void")) {
               return;
       }
       if (streq(dec->type, "opaque")) {
               error("Opaque -- illegal argument type");
       }
       if (peekscan(TOK_STAR, &tok)) {
               if (streq(dec->type, "string")) {
                       error("Pointer to string not allowed in program arguments\n");
               }
               dec->rel = REL_POINTER;
               if (peekscan(TOK_IDENT, &tok))  /* optional name of argument */
                       dec->name = strdup(tok.str);
       }
       if (peekscan(TOK_LANGLE, &tok)) {
               if (!streq(dec->type, "string")) {
                       error("Arrays cannot be declared as arguments to procedures -- use typedef");
               }
               dec->rel = REL_ARRAY;
               if (peekscan(TOK_RANGLE, &tok)) {
                       dec->array_max = "(unsigned int)~0";
                       /* unspecified size, use max */
               } else {
                       scan_num(&tok);
                       dec->array_max = tok.str;
                       scan(TOK_RANGLE, &tok);
               }
       }
       if (streq(dec->type, "string")) {
               if (dec->rel != REL_ARRAY) {    /* .x specifies just string as
                                                * type of argument - make it
                                                * string<> */
                       dec->rel = REL_ARRAY;
                       dec->array_max = "(unsigned int)~0";
                       /* unspecified size, use max */
               }
       }
}



static void
get_type(const char **prefixp, const char **typep, defkind dkind)
{
       token   tok;

       *prefixp = NULL;
       get_token(&tok);
       switch (tok.kind) {
       case TOK_IDENT:
               *typep = tok.str;
               break;
       case TOK_STRUCT:
       case TOK_ENUM:
       case TOK_UNION:
               *prefixp = tok.str;
               scan(TOK_IDENT, &tok);
               *typep = tok.str;
               break;
       case TOK_UNSIGNED:
               unsigned_dec(typep);
               break;
       case TOK_SHORT:
               *typep = "short";
               (void) peekscan(TOK_INT, &tok);
               break;
       case TOK_LONG:
               *typep = "long";
               (void) peekscan(TOK_INT, &tok);
               break;
       case TOK_HYPER:
               *typep = "longlong_t";
               (void) peekscan(TOK_INT, &tok);
               break;
       case TOK_VOID:
               if (dkind != DEF_UNION && dkind != DEF_PROGRAM) {
                       error("Void is allowed only inside union and program definitions with one argument");
               }
               *typep = tok.str;
               break;
       case TOK_STRING:
       case TOK_OPAQUE:
       case TOK_CHAR:
       case TOK_INT:
       case TOK_FLOAT:
       case TOK_DOUBLE:
       case TOK_BOOL:
       case TOK_QUAD:
               *typep = tok.str;
               break;
       default:
               error("Type specifier expected");
       }
}

static void
unsigned_dec(const char **typep)
{
       token   tok;

       peek(&tok);
       switch (tok.kind) {
       case TOK_CHAR:
               get_token(&tok);
               *typep = "u_char";
               break;
       case TOK_SHORT:
               get_token(&tok);
               *typep = "u_short";
               (void) peekscan(TOK_INT, &tok);
               break;
       case TOK_LONG:
               get_token(&tok);
               *typep = "u_long";
               (void) peekscan(TOK_INT, &tok);
               break;
       case TOK_HYPER:
               get_token(&tok);
               *typep = "u_longlong_t";
               (void) peekscan(TOK_INT, &tok);
               break;
       case TOK_INT:
               get_token(&tok);
               *typep = "u_int";
               break;
       default:
               *typep = "u_int";
               break;
       }
}