/*      $NetBSD: rpc_util.c,v 1.18 2015/09/20 16:57:13 kamil 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_util.c 1.11 89/02/22 (C) 1987 SMI";
#else
__RCSID("$NetBSD: rpc_util.c,v 1.18 2015/09/20 16:57:13 kamil Exp $");
#endif
#endif

/*
* rpc_util.c, Utility routines for the RPC protocol compiler
*/
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <err.h>
#include <ctype.h>
#include "rpc_scan.h"
#include "rpc_parse.h"
#include "rpc_util.h"

#define ARGEXT "argument"

static void printwhere(void);

char    curline[MAXLINESIZE];   /* current read line */
char   *where = curline;        /* current point in line */
int     linenum = 0;            /* current line number */

const char *infilename;         /* input filename */

#define NFILES 7
static const char *outfiles[NFILES];    /* output file names */
int     nfiles;

FILE   *fout;                   /* file pointer of current output */
FILE   *fin;                    /* file pointer of current input */

list   *defined;                /* list of defined things */

static const char *toktostr(tok_kind);
static void printbuf(void);
static void printwhere(void);
static int findit(definition *, const char *);
static const char *fixit(const char *, const char *);
static int typedefed(definition *, const char *);

/*
* Reinitialize the world
*/
void
reinitialize(void)
{
       memset(curline, 0, MAXLINESIZE);
       where = curline;
       linenum = 0;
       defined = NULL;
}
/*
* string equality
*/
int
streq(const char *a, const char *b)
{
       return (strcmp(a, b) == 0);
}
/*
* find a value in a list
*/
definition *
findval(list *lst, const char *val, int (*cmp)(definition *, const char *))
{

       for (; lst != NULL; lst = lst->next) {
               if ((*cmp) (lst->val, val)) {
                       return (lst->val);
               }
       }
       return (NULL);
}
/*
* store a value in a list
*/
void
storeval(list **lstp, definition *val)
{
       list  **l;
       list   *lst;


       for (l = lstp; *l != NULL; l = (list **) & (*l)->next);
       lst = ALLOC(list);
       lst->val = val;
       lst->next = NULL;
       *l = lst;
}

static int
findit(definition *def, const char *type)
{
       return (streq(def->def_name, type));
}

static const char *
fixit(const char *type, const char *orig)
{
       definition *def;

       def = (definition *) FINDVAL(defined, type, findit);
       if (def == NULL || def->def_kind != DEF_TYPEDEF) {
               return (orig);
       }
       switch (def->def.ty.rel) {
       case REL_VECTOR:
               return (def->def.ty.old_type);
       case REL_ALIAS:
               return (fixit(def->def.ty.old_type, orig));
       default:
               return (orig);
       }
}

const char *
fixtype(const char *type)
{
       return (fixit(type, type));
}

const char *
stringfix(const char *type)
{
       if (streq(type, "string")) {
               return ("wrapstring");
       } else {
               return (type);
       }
}

void
ptype(const char *prefix, const char *type, int follow)
{
       if (prefix != NULL) {
               if (streq(prefix, "enum")) {
                       f_print(fout, "enum ");
               } else {
                       f_print(fout, "struct ");
               }
       }
       if (streq(type, "bool")) {
               f_print(fout, "bool_t ");
       } else
               if (streq(type, "string")) {
                       f_print(fout, "char *");
               } else {
                       f_print(fout, "%s ", follow ? fixtype(type) : type);
               }
}

static int
typedefed(definition *def, const char *type)
{
       if (def->def_kind != DEF_TYPEDEF || def->def.ty.old_prefix != NULL) {
               return (0);
       } else {
               return (streq(def->def_name, type));
       }
}

int
isvectordef(const char *type, relation rel)
{
       definition *def;

       for (;;) {
               switch (rel) {
               case REL_VECTOR:
                       return (!streq(type, "string"));
               case REL_ARRAY:
                       return (0);
               case REL_POINTER:
                       return (0);
               case REL_ALIAS:
                       def = (definition *) FINDVAL(defined, type, typedefed);
                       if (def == NULL) {
                               return (0);
                       }
                       type = def->def.ty.old_type;
                       rel = def->def.ty.rel;
               }
       }
}

char   *
locase(const char *str)
{
       char    c;
       static char buf[100];
       char   *p = buf;

       while ((c = *str++) != '\0') {
               *p++ = (c >= 'A' && c <= 'Z') ? (c - 'A' + 'a') : c;
       }
       *p = 0;
       return (buf);
}

void
pvname_svc(const char *pname, const char *vnum)
{
       f_print(fout, "%s_%s_svc", locase(pname), vnum);
}

void
pvname(const char *pname, const char *vnum)
{
       f_print(fout, "%s_%s", locase(pname), vnum);
}
/*
* print a useful (?) error message, and then die
*/
__printflike(1, 2) void
error(const char *msg, ...)
{
       va_list ap;

       printwhere();
       fprintf(stderr, "%s:%d: ", infilename, linenum);
       va_start(ap, msg);
       vfprintf(stderr, msg, ap);
       va_end(ap);
       fprintf(stderr, "\n");
       errx(EXIT_FAILURE, "Cannot recover from this error");
}
/*
* Something went wrong, unlink any files that we may have created and then
* die.
*/
void
crash(void)
{
       int     i;

       if (!docleanup)
               return;

       for (i = 0; i < nfiles; i++) {
               (void) unlink(outfiles[i]);
       }
}

void
record_open(const char *file)
{
       if (nfiles < NFILES) {
               outfiles[nfiles++] = file;
       } else {
               errx(EXIT_FAILURE, "Too many files!");
       }
}

/*
* error, token encountered was not the expected one
*/
void
expected1(tok_kind exp1)
{
       error("Expected '%s'", toktostr(exp1));
}
/*
* error, token encountered was not one of two expected ones
*/
void
expected2(tok_kind exp1, tok_kind exp2)
{
       error("Expected '%s' or '%s'",
           toktostr(exp1),
           toktostr(exp2));
}
/*
* error, token encountered was not one of 3 expected ones
*/
void
expected3(tok_kind exp1, tok_kind exp2, tok_kind exp3)
{
       error("Expected '%s', '%s', or '%s'",
           toktostr(exp1),
           toktostr(exp2),
           toktostr(exp3));
}

void
tabify(FILE *f, int tab)
{
       while (tab--) {
               (void) fputc('\t', f);
       }
}


static token tokstrings[] = {
       {TOK_IDENT, "identifier"},
       {TOK_CONST, "const"},
       {TOK_RPAREN, ")"},
       {TOK_LPAREN, "("},
       {TOK_RBRACE, "}"},
       {TOK_LBRACE, "{"},
       {TOK_LBRACKET, "["},
       {TOK_RBRACKET, "]"},
       {TOK_STAR, "*"},
       {TOK_COMMA, ","},
       {TOK_EQUAL, "="},
       {TOK_COLON, ":"},
       {TOK_SEMICOLON, ";"},
       {TOK_UNION, "union"},
       {TOK_STRUCT, "struct"},
       {TOK_SWITCH, "switch"},
       {TOK_CASE, "case"},
       {TOK_DEFAULT, "default"},
       {TOK_ENUM, "enum"},
       {TOK_TYPEDEF, "typedef"},
       {TOK_INT, "int"},
       {TOK_SHORT, "short"},
       {TOK_LONG, "long"},
       {TOK_UNSIGNED, "unsigned"},
       {TOK_DOUBLE, "double"},
       {TOK_FLOAT, "float"},
       {TOK_CHAR, "char"},
       {TOK_STRING, "string"},
       {TOK_OPAQUE, "opaque"},
       {TOK_BOOL, "bool"},
       {TOK_VOID, "void"},
       {TOK_PROGRAM, "program"},
       {TOK_VERSION, "version"},
       {TOK_EOF, "??????"}
};

static const char *
toktostr(tok_kind kind)
{
       token  *sp;

       for (sp = tokstrings; sp->kind != TOK_EOF && sp->kind != kind; sp++);
       return (sp->str);
}

static void
printbuf(void)
{
       char    c;
       int     i;
       int     cnt;

#define TABSIZE 4

       for (i = 0; (c = curline[i]) != '\0'; i++) {
               if (c == '\t') {
                       cnt = 8 - (i % TABSIZE);
                       c = ' ';
               } else {
                       cnt = 1;
               }
               while (cnt--) {
                       (void) fputc(c, stderr);
               }
       }
}

static void
printwhere(void)
{
       int     i;
       char    c;
       int     cnt;

       printbuf();
       for (i = 0; i < where - curline; i++) {
               c = curline[i];
               if (c == '\t') {
                       cnt = 8 - (i % TABSIZE);
               } else {
                       cnt = 1;
               }
               while (cnt--) {
                       (void) fputc('^', stderr);
               }
       }
       (void) fputc('\n', stderr);
}

char   *
make_argname(const char *pname, const char *vname)
{
       char   *name;
       size_t len;

       len = strlen(pname) + strlen(vname) + strlen(ARGEXT) + 3;
       name = malloc(len);
       if (!name) {
               err(EXIT_FAILURE, "malloc");
       }
       snprintf(name, len, "%s_%s_%s", locase(pname), vname, ARGEXT);
       return (name);
}

bas_type *typ_list_h;
bas_type *typ_list_t;

void
add_type(int len, const char *type)
{
       bas_type *ptr;

       if ((ptr = malloc(sizeof(bas_type))) == NULL) {
               err(EXIT_FAILURE, "malloc");
       }
       ptr->name = type;
       ptr->length = len;
       ptr->next = NULL;
       if (typ_list_t == NULL) {
               typ_list_t = ptr;
               typ_list_h = ptr;
       } else {
               typ_list_t->next = ptr;
               typ_list_t = ptr;
       }
}

bas_type *
find_type(const char *type)
{
       bas_type *ptr;

       ptr = typ_list_h;


       while (ptr != NULL) {
               if (strcmp(ptr->name, type) == 0)
                       return (ptr);
               else
                       ptr = ptr->next;
       }
       return (NULL);
}