/* $Id: rpc_hout.c,v 1.4 2001/10/08 08:31:37 dm 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
 */


/*
 * rpc_hout.c, Header file outputter for the RPC protocol compiler 
 */
#include <stdio.h>
#include <ctype.h>
#include <stdlib.h>
#include "rpc_parse.h"
#include "rpc_util.h"

static void pconstdef (definition *);
static void pargdef (definition *);
static void pstructdef (definition *);
static void puniondef (definition *);
static void pprogramdef (definition *);
static void penumdef (definition *);
static void ptypedef (definition *);
static void pdefine (char *, char *);
static void puldefine (char *, char *);
static int define_printed (proc_list *, version_list *);
static int undefined2 (char *, char *);
static void parglist (proc_list *, char *);

void pxdrfuncdecl (char *, int, int);
void pdeclaration (char *, declaration *, int, char *);
static void pprogramdef (definition *);
void pprocdef (proc_list *, version_list *, char *, int, int);

/*
 * Print the C-version of an xdr definition 
 */
void
print_datadef (definition * def)
{

  if (def->def_kind == DEF_PROGRAM)	/* handle data only */
    return;

  if (def->def_kind != DEF_CONST) {
    f_print (fout, "\n");
  }
  switch (def->def_kind) {
  case DEF_STRUCT:
    pstructdef (def);
    break;
  case DEF_UNION:
    puniondef (def);
    break;
  case DEF_ENUM:
    penumdef (def);
    break;
  case DEF_TYPEDEF:
    ptypedef (def);
    break;
  case DEF_PROGRAM:
    pprogramdef (def);
    break;
  case DEF_CONST:
    pconstdef (def);
    break;
  }
  if (def->def_kind != DEF_PROGRAM && def->def_kind != DEF_CONST
      && def->def_kind != DEF_STRUCT && def->def_kind != DEF_UNION) {
    pxdrfuncdecl (def->def_name,
		  def->def_kind != DEF_TYPEDEF ||
		  !isvectordef (def->def.ty.old_type, def->def.ty.rel),
		  0);

  }
}


void
print_funcdef (definition * def)
{
  switch (def->def_kind) {
  case DEF_PROGRAM:
    f_print (fout, "\n");
    pprogramdef (def);
    break;
  default:
    break;
  }
}

void
pxdrfuncdecl (char *name, int pointerp, int structp)
{
  f_print (fout, "RPC_EXTERN bool_t xdr_%s(XDR *, %s%s%s);\n",
	   name, structp ? "struct " : "", name,
	   pointerp ? (" *") : "");
}


static void
pconstdef (definition * def)
{
  pdefine (def->def_name, def->def.co);
}

/* print out the definitions for the arguments of functions in the 
   header file 
 */
static void
pargdef (definition * def)
{
  decl_list *l;
  version_list *vers;
  char *name;
  proc_list *plist;


  for (vers = def->def.pr.versions; vers != NULL; vers = vers->next) {
    for (plist = vers->procs; plist != NULL;
	 plist = plist->next) {

      if (!newstyle || plist->arg_num < 2) {
	continue;		/* old style or single args */
      }
      name = plist->args.argname;
      f_print (fout, "struct %s {\n", name);
      for (l = plist->args.decls;
	   l != NULL; l = l->next) {
	pdeclaration (name, &l->decl, 1, ";\n");
      }
      f_print (fout, "};\n");
      f_print (fout, "typedef struct %s %s;\n", name, name);
      pxdrfuncdecl (name, 0, 0);
      f_print (fout, "\n");
    }
  }

}


static void
pstructdef (definition * def)
{
  decl_list *l;
  char *name = def->def_name;

  f_print (fout, "typedef struct %s {\n", name);
  for (l = def->def.st.decls; l != NULL; l = l->next) {
    pdeclaration (name, &l->decl, 1, ";\n");
  }
  f_print (fout, "} %s;\n", name);
  /* f_print (fout, "typedef struct %s %s;\n", name, name); */
  pxdrfuncdecl (name, 1, 0);
  if (*name != toupper (*name))
    f_print (fout, "RPC_CONSTRUCT (%c%s, %s)\n",
	     toupper (*name), name + 1, name);
}

static void
puniondef (definition * def)
{
  case_list *l;
  char *name = def->def_name;
  declaration *decl;

  f_print (fout, "typedef struct %s {\n", name);
  decl = &def->def.un.enum_decl;
  if (streq (decl->type, "bool")) {
    f_print (fout, "\tbool_t %s;\n", decl->name);
  }
  else {
    f_print (fout, "\t%s %s;\n", decl->type, decl->name);
  }
  f_print (fout, "\tunion {\n");
  for (l = def->def.un.cases; l != NULL; l = l->next) {
    if (l->contflag == 0)
      pdeclaration (name, &l->case_decl, 2, ";\n");
  }
  decl = def->def.un.default_decl;
  if (decl && !streq (decl->type, "void")) {
    pdeclaration (name, decl, 2, ";\n");
  }
  /* f_print (fout, "\t} %s_u;\n", name); */
  f_print (fout, "\t} RPC_UNION_NAME(%s);\n", name);
  f_print (fout, "} %s;\n", name);
  if (compatflag)
    f_print (fout, "#define %s_u u\n", name);
  /* f_print (fout, "typedef struct %s %s;\n", name, name); */
  pxdrfuncdecl (name, 1, 0);
  f_print (fout, "RPC_CONSTRUCT (%c%s, %s)\n",
	   toupper (*name), name + 1, name);
}

static void
pdefine (char *name, char *num)
{
  f_print (fout, "#define %s %s\n", name, num);
}

static void
puldefine (char *name, char *num)
{
  f_print (fout, "#define %s ((u_long)%s)\n", name, num);
}

static int
define_printed (proc_list * stop, version_list * start)
{
  version_list *vers;
  proc_list *proc;

  for (vers = start; vers != NULL; vers = vers->next) {
    for (proc = vers->procs; proc != NULL; proc = proc->next) {
      if (proc == stop) {
	return (0);
      }
      else if (streq (proc->proc_name, stop->proc_name)) {
	return (1);
      }
    }
  }
  abort ();
  /* NOTREACHED */
}

static void
pprogramdef (definition * def)
{
  version_list *vers;
  proc_list *proc;

#if 0
  static int puttbl;
  if (!puttbl) {
    puttbl = 1;
    f_print (fout,
	     "#ifndef _RPCGEN_TABLE_DEFINED_\n"
	     "#define _RPCGEN_TABLE_DEFINED_ 1\n"
	     "#ifndef RPCGEN_ACTION\n"
	     "#define RPCGEN_ACTION(x) 0\n"
	     "#endif /* !RPCGEN_ACTION */\n"
	     "struct rpcgen_table {\n"
	     "  char *(*proc)();\n"
	     "  xdrproc_t xdr_arg;\n"
	     "  unsigned len_arg;\n"
	     "  xdrproc_t xdr_res;\n"
	     "  unsigned len_res;\n"
	     "};\n"
	     "\n"
	     "struct rpc_program {\n"
	     "  u_int32_t progno;\n"
	     "  u_int32_t versno;\n"
	     "  const struct rpcgen_table *tbl;\n"
	     "  size_t nproc;\n"
	     "};\n"
	     "#endif /* !_RPCGEN_TABLE_DEFINED_ */\n\n");
  }
#endif

  pargdef (def);

  puldefine (def->def_name, def->def.pr.prog_num);
  for (vers = def->def.pr.versions; vers != NULL; vers = vers->next) {
    if (tblflag) {
      f_print (fout, "RPC_EXTERN const struct rpcgen_table %s_%s_table[];\n",
	       locase (def->def_name), vers->vers_num);
      f_print (fout, "RPC_EXTERN const int %s_%s_nproc;\n",
	       locase (def->def_name), vers->vers_num);
      f_print (fout, "RPC_EXTERN const struct rpc_program  %s_%s;\n",
	       locase (def->def_name), vers->vers_num);
    }
    puldefine (vers->vers_name, vers->vers_num);

    for (proc = vers->procs; proc != NULL; proc = proc->next) {
      if (!define_printed (proc, def->def.pr.versions))
	puldefine (proc->proc_name, proc->proc_num);
#if 0
      pprocdef (proc, vers, "CLIENT *", 0, 0);
      pprocdef (proc, vers, "struct svc_req *", 1, 0);
#endif
    }
  }
}

void
pprocdef (proc_list * proc, version_list * vp, char *addargtype,
	  int server_p, int mode)
{

  ptype (proc->res_prefix, proc->res_type, 1);
  f_print (fout, "*");
  if (server_p)
    pvname_svc (proc->proc_name, vp->vers_num);
  else
    pvname (proc->proc_name, vp->vers_num);

  /*
   * mode  0 == cplusplus, mode  1 = ANSI-C, mode 2 = old style C 
   */
  if (mode == 0 || mode == 1)
    parglist (proc, addargtype);
  else
    f_print (fout, "();\n");
}


/* print out argument list of procedure */
static void
parglist (proc_list * proc, char *addargtype)
{
  decl_list *dl;

  f_print (fout, "(");

  if (proc->arg_num < 2 && newstyle &&
      streq (proc->args.decls->decl.type, "void")) {
    /* 0 argument in new style:  do nothing */
  }
  else {
    for (dl = proc->args.decls; dl != NULL; dl = dl->next) {
      ptype (dl->decl.prefix, dl->decl.type, 1);
      if (!newstyle)
	f_print (fout, "*");	/* old style passes by reference */

      f_print (fout, ", ");
    }
  }

  f_print (fout, "%s);\n", addargtype);
}

static void
penumdef (definition * def)
{
  char *name = def->def_name;
  enumval_list *l;
  char *last = NULL;
  int count = 0;

  f_print (fout, "enum %s {\n", name);
  for (l = def->def.en.vals; l != NULL; l = l->next) {
    f_print (fout, "\t%s", l->name);
    if (l->assignment) {
      f_print (fout, " = %s", l->assignment);
      last = l->assignment;
      count = 1;
    }
    else {
      if (last == NULL) {
	f_print (fout, " = %d", count++);
      }
      else {
	f_print (fout, " = %s + %d", last, count++);
      }
    }
    if (l->next)
      f_print (fout, ",\n");
    else
      f_print (fout, "\n");
  }
  f_print (fout, "};\n");
  f_print (fout, "typedef enum %s %s;\n", name, name);
}

static void
ptypedef (definition * def)
{
  char *name = def->def_name;
  char *old = def->def.ty.old_type;
  char prefix[8];		/* enough to contain "struct ", including NUL */
  relation rel = def->def.ty.rel;


  if (!streq (name, old)) {
    if (streq (old, "string")) {
      old = "char";
      rel = REL_POINTER;
    }
    else if (streq (old, "opaque")) {
      old = "char";
    }
    else if (streq (old, "bool")) {
      old = "bool_t";
    }
    if (undefined2 (old, name) && def->def.ty.old_prefix) {
      s_print (prefix, "%s ", def->def.ty.old_prefix);
    }
    else {
      prefix[0] = 0;
    }
    f_print (fout, "typedef ");
    switch (rel) {
    case REL_ARRAY:
      f_print (fout, "struct {\n");
#if 0
      f_print (fout, "\tu_int %s_len;\n", name);
      f_print (fout, "\t%s%s *%s_val;\n", prefix, old, name);
#else
      f_print (fout, "\tu_int len;\n");
      if (compatflag)
	f_print (fout, "#ifndef %s_len\n#define %s_len len\n#endif\n",
		 name, name);
      f_print (fout, "\t%s%s *val;\n", prefix, old);
      if (compatflag)
	f_print (fout, "#ifndef %s_val\n#define %s_val val\n#endif\n",
		 name, name);
#endif
      f_print (fout, "} %s", name);
      break;
    case REL_POINTER:
      f_print (fout, "%s%s *%s", prefix, old, name);
      break;
    case REL_VECTOR:
      f_print (fout, "%s%s %s[%s]", prefix, old, name,
	       def->def.ty.array_max);
      break;
    case REL_ALIAS:
      f_print (fout, "%s%s %s", prefix, old, name);
      break;
    }
    f_print (fout, ";\n");
  }
}

void
pdeclaration (char *name, declaration * dec, int tab, char *separator)
{
  char buf[8];			/* enough to hold "struct ", include NUL */
  char *prefix;
  char *type;

  if (streq (dec->type, "void")) {
    return;
  }
  tabify (fout, tab);
  if (streq (dec->type, name) && !dec->prefix) {
    f_print (fout, "struct ");
  }
  if (streq (dec->type, "string")) {
    f_print (fout, "char *%s", dec->name);
  }
  else {
    prefix = "";
    if (streq (dec->type, "bool")) {
      type = "bool_t";
    }
    else if (streq (dec->type, "opaque")) {
      type = "char";
    }
    else {
      if (dec->prefix) {
	s_print (buf, "%s ", dec->prefix);
	prefix = buf;
      }
      type = dec->type;
    }
    switch (dec->rel) {
    case REL_ALIAS:
      f_print (fout, "%s%s %s", prefix, type, dec->name);
      break;
    case REL_VECTOR:
      f_print (fout, "%s%s %s[%s]", prefix, type, dec->name,
	       dec->array_max);
      break;
    case REL_POINTER:
      f_print (fout, "%s%s *%s", prefix, type, dec->name);
      break;
    case REL_ARRAY:
      f_print (fout, "struct {\n");
      tabify (fout, tab);
      /* f_print (fout, "\tu_int %s_len;\n", dec->name); */
      f_print (fout, "\tu_int len;\n");
      if (compatflag)
	f_print (fout, "#undef %s_len\n#define %s_len len\n",
		 dec->name, dec->name);
      tabify (fout, tab);
      /* f_print (fout, "\t%s%s *%s_val;\n", prefix, type, dec->name); */
      f_print (fout, "\t%s%s *val;\n", prefix, type);
      if (compatflag)
	f_print (fout, "#undef %s_val\n#define %s_val val\n",
		 dec->name, dec->name);
      tabify (fout, tab);
      f_print (fout, "} %s", dec->name);
      break;
    }
  }
  f_print (fout, separator);
}

static int
undefined2 (char *type, char *stop)
{
  list *l;
  definition *def;

  for (l = defined; l != NULL; l = l->next) {
    def = (definition *) l->val;
    if (def->def_kind != DEF_PROGRAM) {
      if (streq (def->def_name, stop)) {
	return (1);
      }
      else if (streq (def->def_name, type)) {
	return (0);
      }
    }
  }
  return (1);
}