| /* |
| * 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); |
| } |
| } |
| } |