|  | /* | 
|  | * rpc_clntout.c, Client-stub outputter for the RPC protocol compiler | 
|  | * Copyright (c) 2010, Oracle America, Inc. | 
|  | * | 
|  | * Redistribution and use in source and binary forms, with or without | 
|  | * modification, are permitted provided that the following conditions are | 
|  | * met: | 
|  | * | 
|  | *     * Redistributions of source code must retain the above copyright | 
|  | *       notice, this list of conditions and the following disclaimer. | 
|  | *     * Redistributions in binary form must reproduce the above | 
|  | *       copyright notice, this list of conditions and the following | 
|  | *       disclaimer in the documentation and/or other materials | 
|  | *       provided with the distribution. | 
|  | *     * Neither the name of the "Oracle America, Inc." nor the names of its | 
|  | *       contributors may be used to endorse or promote products derived | 
|  | *       from this software without specific prior written permission. | 
|  | * | 
|  | *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS | 
|  | *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT | 
|  | *   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS | 
|  | *   FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE | 
|  | *   COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, | 
|  | *   INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL | 
|  | *   DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE | 
|  | *   GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS | 
|  | *   INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, | 
|  | *   WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING | 
|  | *   NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | 
|  | *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | 
|  | */ | 
|  | #include <stdio.h> | 
|  | #include <string.h> | 
|  | #include "rpc/types.h" | 
|  | #include "rpc_parse.h" | 
|  | #include "rpc_util.h" | 
|  | #include "proto.h" | 
|  |  | 
|  | #define DEFAULT_TIMEOUT 25	/* in seconds */ | 
|  | static const char RESULT[] = "clnt_res"; | 
|  |  | 
|  | static void write_program (definition * def); | 
|  | static void printbody (proc_list * proc); | 
|  | static const char *ampr (const char *type); | 
|  | static void printbody (proc_list * proc); | 
|  |  | 
|  |  | 
|  | void | 
|  | write_stubs (void) | 
|  | { | 
|  | list *l; | 
|  | definition *def; | 
|  |  | 
|  | fprintf (fout, | 
|  | "\n/* Default timeout can be changed using clnt_control() */\n"); | 
|  | fprintf (fout, "static struct timeval TIMEOUT = { %d, 0 };\n", | 
|  | DEFAULT_TIMEOUT); | 
|  | for (l = defined; l != NULL; l = l->next) | 
|  | { | 
|  | def = (definition *) l->val; | 
|  | if (def->def_kind == DEF_PROGRAM) | 
|  | { | 
|  | write_program (def); | 
|  | } | 
|  | } | 
|  | } | 
|  |  | 
|  | static void | 
|  | write_program (definition * def) | 
|  | { | 
|  | version_list *vp; | 
|  | proc_list *proc; | 
|  |  | 
|  | for (vp = def->def.pr.versions; vp != NULL; vp = vp->next) | 
|  | { | 
|  | for (proc = vp->procs; proc != NULL; proc = proc->next) | 
|  | { | 
|  | fprintf (fout, "\n"); | 
|  | if (mtflag == 0) | 
|  | { | 
|  | ptype (proc->res_prefix, proc->res_type, 1); | 
|  | fprintf (fout, "*\n"); | 
|  | pvname (proc->proc_name, vp->vers_num); | 
|  | printarglist (proc, RESULT, "clnt", "CLIENT *"); | 
|  | } | 
|  | else | 
|  | { | 
|  | fprintf (fout, "enum clnt_stat \n"); | 
|  | pvname (proc->proc_name, vp->vers_num); | 
|  | printarglist (proc, RESULT, "clnt", "CLIENT *"); | 
|  | } | 
|  | fprintf (fout, "{\n"); | 
|  | printbody (proc); | 
|  | fprintf (fout, "}\n"); | 
|  | } | 
|  | } | 
|  | } | 
|  |  | 
|  | /* Writes out declarations of procedure's argument list. | 
|  | In either ANSI C style, in one of old rpcgen style (pass by reference), | 
|  | or new rpcgen style (multiple arguments, pass by value); | 
|  | */ | 
|  |  | 
|  | /* sample addargname = "clnt"; sample addargtype = "CLIENT * " */ | 
|  |  | 
|  | void | 
|  | printarglist (proc_list * proc,  const char *result, | 
|  | const char *addargname, const char *addargtype) | 
|  | { | 
|  |  | 
|  | decl_list *l; | 
|  |  | 
|  | if (!newstyle) | 
|  | { /* old style: always pass argument by reference */ | 
|  | if (Cflag) | 
|  | {			/* C++ style heading */ | 
|  | fprintf (fout, "("); | 
|  | ptype (proc->args.decls->decl.prefix, | 
|  | proc->args.decls->decl.type, 1); | 
|  |  | 
|  | if (mtflag) | 
|  | {/* Generate result field */ | 
|  | fprintf (fout, "*argp, "); | 
|  | ptype(proc->res_prefix, proc->res_type, 1); | 
|  | fprintf (fout, "*%s, %s%s)\n", result, addargtype, addargname); | 
|  | } | 
|  | else | 
|  | fprintf (fout, "*argp, %s%s)\n", addargtype, addargname); | 
|  | } | 
|  | else | 
|  | { | 
|  | if (!mtflag) | 
|  | fprintf (fout, "(argp, %s)\n", addargname); | 
|  | else | 
|  | fprintf (fout, "(argp, %s, %s)\n", result, addargname); | 
|  | fprintf (fout, "\t"); | 
|  | ptype (proc->args.decls->decl.prefix, | 
|  | proc->args.decls->decl.type, 1); | 
|  | fprintf (fout, "*argp;\n"); | 
|  | if (mtflag) | 
|  | { | 
|  | fprintf (fout, "\t"); | 
|  | ptype (proc->res_prefix, proc->res_type, 1); | 
|  | fprintf (fout, "*%s;\n", result); | 
|  | } | 
|  | } | 
|  | } | 
|  | else if (streq (proc->args.decls->decl.type, "void")) | 
|  | { | 
|  | /* newstyle, 0 argument */ | 
|  | if (mtflag) | 
|  | { | 
|  | fprintf (fout, "("); | 
|  | if (Cflag) | 
|  | { | 
|  | ptype(proc->res_prefix, proc->res_type, 1); | 
|  | fprintf (fout, "*%s, %s%s)\n", result, addargtype, addargname); | 
|  | } | 
|  | else | 
|  | fprintf (fout, "(%s)\n", addargname); | 
|  | } | 
|  | else if (Cflag) | 
|  | fprintf (fout, "(%s%s)\n", addargtype, addargname); | 
|  | else | 
|  | fprintf (fout, "(%s)\n", addargname); | 
|  | } | 
|  | else | 
|  | { | 
|  | /* new style, 1 or multiple arguments */ | 
|  | if (!Cflag) | 
|  | { | 
|  | fprintf (fout, "("); | 
|  | for (l = proc->args.decls; l != NULL; l = l->next) | 
|  | fprintf (fout, "%s, ", l->decl.name); | 
|  | if (mtflag) | 
|  | fprintf (fout, "%s, ", result); | 
|  | fprintf (fout, "%s)\n", addargname); | 
|  | for (l = proc->args.decls; l != NULL; l = l->next) | 
|  | { | 
|  | pdeclaration (proc->args.argname, &l->decl, 1, ";\n"); | 
|  | } | 
|  | if (mtflag) | 
|  | { | 
|  | fprintf (fout, "\t"); | 
|  | ptype (proc->res_prefix, proc->res_type, 1); | 
|  | fprintf (fout, "*%s;\n", result); | 
|  | } | 
|  | } | 
|  | else | 
|  | {			/* C++ style header */ | 
|  | fprintf (fout, "("); | 
|  | for (l = proc->args.decls; l != NULL; l = l->next) | 
|  | { | 
|  | pdeclaration (proc->args.argname, &l->decl, 0, ", "); | 
|  | } | 
|  | if (mtflag) | 
|  | { | 
|  | ptype (proc->res_prefix, proc->res_type, 1); | 
|  | fprintf (fout, "*%s, ", result); | 
|  | } | 
|  | fprintf (fout, " %s%s)\n", addargtype, addargname); | 
|  | } | 
|  | } | 
|  |  | 
|  | if (!Cflag) | 
|  | fprintf (fout, "\t%s%s;\n", addargtype, addargname); | 
|  | } | 
|  |  | 
|  |  | 
|  | static | 
|  | const char * | 
|  | ampr (const char *type) | 
|  | { | 
|  | if (isvectordef (type, REL_ALIAS)) | 
|  | { | 
|  | return ""; | 
|  | } | 
|  | else | 
|  | { | 
|  | return "&"; | 
|  | } | 
|  | } | 
|  |  | 
|  | static void | 
|  | printbody (proc_list * proc) | 
|  | { | 
|  | decl_list *l; | 
|  | bool_t args2 = (proc->arg_num > 1); | 
|  | /*  int i; */ | 
|  |  | 
|  | /* For new style with multiple arguments, need a structure in which | 
|  | to stuff the arguments. */ | 
|  | if (newstyle && args2) | 
|  | { | 
|  | fprintf (fout, "\t%s", proc->args.argname); | 
|  | fprintf (fout, " arg;\n"); | 
|  | } | 
|  | if (!mtflag) | 
|  | { | 
|  | fprintf (fout, "\tstatic "); | 
|  | if (streq (proc->res_type, "void")) | 
|  | { | 
|  | fprintf (fout, "char "); | 
|  | } | 
|  | else | 
|  | { | 
|  | ptype (proc->res_prefix, proc->res_type, 0); | 
|  | } | 
|  | fprintf (fout, "%s;\n", RESULT); | 
|  | fprintf (fout, "\n"); | 
|  | fprintf (fout, "\tmemset((char *)%s%s, 0, sizeof(%s));\n", | 
|  | ampr (proc->res_type), RESULT, RESULT); | 
|  | } | 
|  | if (newstyle && !args2 && (streq (proc->args.decls->decl.type, "void"))) | 
|  | { | 
|  | /* newstyle, 0 arguments */ | 
|  | if (mtflag) | 
|  | fprintf (fout, "\t return "); | 
|  | else | 
|  | fprintf (fout, "\t if "); | 
|  | fprintf (fout, | 
|  | "(clnt_call (clnt, %s, (xdrproc_t) xdr_void, ", proc->proc_name); | 
|  |  | 
|  | fprintf (fout, | 
|  | "(caddr_t) NULL,\n\t\t(xdrproc_t) xdr_%s, (caddr_t) %s%s,", | 
|  | stringfix(proc->res_type), (mtflag)?"":ampr(proc->res_type), | 
|  | RESULT); | 
|  | if (mtflag) | 
|  | fprintf (fout, "\n\t\tTIMEOUT));\n\n"); | 
|  | else | 
|  | fprintf (fout, "\n\t\tTIMEOUT) != RPC_SUCCESS) {\n"); | 
|  | } | 
|  | else if (newstyle && args2) | 
|  | { | 
|  | /* newstyle, multiple arguments:  stuff arguments into structure */ | 
|  | for (l = proc->args.decls; l != NULL; l = l->next) | 
|  | { | 
|  | fprintf (fout, "\targ.%s = %s;\n", | 
|  | l->decl.name, l->decl.name); | 
|  | } | 
|  | if (mtflag) | 
|  | fprintf (fout, "\treturn "); | 
|  | else | 
|  | fprintf (fout, "\tif "); | 
|  |  | 
|  | fprintf (fout, | 
|  | "(clnt_call (clnt, %s, (xdrproc_t) xdr_%s", proc->proc_name, | 
|  | proc->args.argname); | 
|  | fprintf (fout, | 
|  | ", (caddr_t) &arg,\n\t\t(xdrproc_t) xdr_%s, (caddr_t) %s%s,", | 
|  | stringfix(proc->res_type), (mtflag)?"":ampr(proc->res_type), | 
|  | RESULT); | 
|  | if (mtflag) | 
|  | fprintf (fout, "\n\t\tTIMEOUT));\n"); | 
|  | else | 
|  | fprintf (fout, "\n\t\tTIMEOUT) != RPC_SUCCESS) {\n"); | 
|  | } | 
|  | else | 
|  | {				/* single argument, new or old style */ | 
|  | if (!mtflag) | 
|  | fprintf (fout, | 
|  | "\tif (clnt_call (clnt, %s,\n\t\t(xdrproc_t) xdr_%s, (caddr_t) %s%s,\n\t\t(xdrproc_t) xdr_%s, (caddr_t) %s%s,\n\t\tTIMEOUT) != RPC_SUCCESS) {\n", | 
|  | proc->proc_name, | 
|  | stringfix (proc->args.decls->decl.type), | 
|  | (newstyle ? "&" : ""), | 
|  | (newstyle ? proc->args.decls->decl.name : "argp"), | 
|  | stringfix (proc->res_type), ampr (proc->res_type), | 
|  | RESULT); | 
|  | else | 
|  | fprintf(fout, | 
|  | "\treturn (clnt_call(clnt, %s,\n\t\t(xdrproc_t) xdr_%s, (caddr_t) %s%s,\n\t\t(xdrproc_t) xdr_%s, (caddr_t) %s%s,\n\t\tTIMEOUT));\n", | 
|  | proc->proc_name, | 
|  | stringfix (proc->args.decls->decl.type), | 
|  | (newstyle ? "&" : ""), | 
|  | (newstyle ? proc->args.decls->decl.name : "argp"), | 
|  | stringfix (proc->res_type), "", | 
|  | RESULT); | 
|  | } | 
|  | if (!mtflag) | 
|  | { | 
|  | fprintf (fout, "\t\treturn (NULL);\n"); | 
|  | fprintf (fout, "\t}\n"); | 
|  | if (streq (proc->res_type, "void")) | 
|  | { | 
|  | fprintf (fout, "\treturn ((void *)%s%s);\n", | 
|  | ampr (proc->res_type), RESULT); | 
|  | } | 
|  | else | 
|  | { | 
|  | fprintf (fout, "\treturn (%s%s);\n", ampr (proc->res_type), RESULT); | 
|  | } | 
|  | } | 
|  | } |