1 /* $OpenBSD: rpc_clntout.c,v 1.15 2010/09/01 14:43:34 millert Exp $ */ 2 /* $NetBSD: rpc_clntout.c,v 1.4 1995/06/11 21:49:52 pk Exp $ */ 3 4 /* 5 * Copyright (c) 2010, Oracle America, Inc. 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions are 9 * met: 10 * 11 * * Redistributions of source code must retain the above copyright 12 * notice, this list of conditions and the following disclaimer. 13 * * Redistributions in binary form must reproduce the above 14 * copyright notice, this list of conditions and the following 15 * disclaimer in the documentation and/or other materials 16 * provided with the distribution. 17 * * Neither the name of the "Oracle America, Inc." nor the names of its 18 * contributors may be used to endorse or promote products derived 19 * from this software without specific prior written permission. 20 * 21 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 22 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 23 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS 24 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE 25 * COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, 26 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 27 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE 28 * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 29 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 30 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 31 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 32 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 33 */ 34 35 /* 36 * rpc_clntout.c, Client-stub outputter for the RPC protocol compiler 37 */ 38 #include <stdio.h> 39 #include <string.h> 40 #include <rpc/types.h> 41 #include "rpc_parse.h" 42 #include "rpc_util.h" 43 44 static void write_program(definition *); 45 static void printbody(proc_list *); 46 47 #define DEFAULT_TIMEOUT 25 /* in seconds */ 48 static char RESULT[] = "clnt_res"; 49 50 51 void 52 write_stubs() 53 { 54 list *l; 55 definition *def; 56 57 fprintf(fout, 58 "\n/* Default timeout can be changed using clnt_control() */\n"); 59 fprintf(fout, "static struct timeval TIMEOUT = { %d, 0 };\n", 60 DEFAULT_TIMEOUT); 61 for (l = defined; l != NULL; l = l->next) { 62 def = (definition *) l->val; 63 if (def->def_kind == DEF_PROGRAM) { 64 write_program(def); 65 } 66 } 67 } 68 69 static void 70 write_program(def) 71 definition *def; 72 { 73 version_list *vp; 74 proc_list *proc; 75 76 for (vp = def->def.pr.versions; vp != NULL; vp = vp->next) { 77 for (proc = vp->procs; proc != NULL; proc = proc->next) { 78 fprintf(fout, "\n"); 79 ptype(proc->res_prefix, proc->res_type, 1); 80 fprintf(fout, "*\n"); 81 pvname(proc->proc_name, vp->vers_num); 82 printarglist(proc, "clnt", "CLIENT *"); 83 fprintf(fout, "{\n"); 84 printbody(proc); 85 fprintf(fout, "}\n"); 86 } 87 } 88 } 89 90 /* 91 * Writes out declarations of procedure's argument list. 92 * In either ANSI C style, in one of old rpcgen style (pass by reference), 93 * or new rpcgen style (multiple arguments, pass by value); 94 */ 95 96 /* sample addargname = "clnt"; sample addargtype = "CLIENT * " */ 97 98 void printarglist(proc, addargname, addargtype) 99 proc_list *proc; 100 char *addargname, *addargtype; 101 { 102 decl_list *l; 103 104 if (!newstyle) { /* old style: always pass argument by reference */ 105 if (Cflag) { /* C++ style heading */ 106 fprintf(fout, "("); 107 ptype(proc->args.decls->decl.prefix, 108 proc->args.decls->decl.type, 1); 109 fprintf(fout, "*argp, %s%s)\n", addargtype, addargname); 110 } else { 111 fprintf(fout, "(argp, %s)\n", addargname); 112 fprintf(fout, "\t"); 113 ptype(proc->args.decls->decl.prefix, 114 proc->args.decls->decl.type, 1); 115 fprintf(fout, "*argp;\n"); 116 } 117 } else if (streq(proc->args.decls->decl.type, "void")) { 118 /* newstyle, 0 argument */ 119 if (Cflag) 120 fprintf(fout, "(%s%s)\n", addargtype, addargname); 121 else 122 fprintf(fout, "(%s)\n", addargname); 123 } else { 124 /* new style, 1 or multiple arguments */ 125 if (!Cflag) { 126 fprintf(fout, "("); 127 for (l = proc->args.decls; l != NULL; l = l->next) 128 fprintf(fout, "%s, ", l->decl.name); 129 fprintf(fout, "%s)\n", addargname); 130 for (l = proc->args.decls; l != NULL; l = l->next) 131 pdeclaration(proc->args.argname, &l->decl, 1, ";\n"); 132 } else { /* C++ style header */ 133 fprintf(fout, "("); 134 for (l = proc->args.decls; l != NULL; l = l->next) 135 pdeclaration(proc->args.argname, &l->decl, 0, ", "); 136 fprintf(fout, " %s%s)\n", addargtype, addargname); 137 } 138 } 139 140 if (!Cflag) 141 fprintf(fout, "\t%s%s;\n", addargtype, addargname); 142 } 143 144 static char * 145 ampr(char *type) 146 { 147 if (isvectordef(type, REL_ALIAS)) { 148 return (""); 149 } else { 150 return ("&"); 151 } 152 } 153 154 static void 155 printbody(proc) 156 proc_list *proc; 157 { 158 decl_list *l; 159 bool_t args2 = (proc->arg_num > 1); 160 161 /* 162 * For new style with multiple arguments, need a structure in which 163 * to stuff the arguments. 164 */ 165 if (newstyle && args2) { 166 fprintf(fout, "\t%s", proc->args.argname); 167 fprintf(fout, " arg;\n"); 168 } 169 fprintf(fout, "\tstatic "); 170 if (streq(proc->res_type, "void")) { 171 fprintf(fout, "char "); 172 } else { 173 ptype(proc->res_prefix, proc->res_type, 0); 174 } 175 fprintf(fout, "%s;\n",RESULT); 176 fprintf(fout, "\n"); 177 fprintf(fout, "\tmemset((char *)%s%s, 0, sizeof(%s));\n", 178 ampr(proc->res_type), RESULT, RESULT); 179 if (newstyle && !args2 && (streq(proc->args.decls->decl.type, "void"))) { 180 /* newstyle, 0 arguments */ 181 fprintf(fout, 182 "\tif (clnt_call(clnt, %s, xdr_void", proc->proc_name); 183 fprintf(fout, 184 ", NULL, xdr_%s, %s%s, TIMEOUT) != RPC_SUCCESS) {\n", 185 stringfix(proc->res_type), ampr(proc->res_type), RESULT); 186 } else if (newstyle && args2) { 187 /* newstyle, multiple arguments: stuff arguments into structure */ 188 for (l = proc->args.decls; l != NULL; l = l->next) { 189 fprintf(fout, "\targ.%s = %s;\n", 190 l->decl.name, l->decl.name); 191 } 192 fprintf(fout, 193 "\tif (clnt_call(clnt, %s, xdr_%s", proc->proc_name, 194 proc->args.argname); 195 fprintf(fout, 196 ", &arg, xdr_%s, %s%s, TIMEOUT) != RPC_SUCCESS) {\n", 197 stringfix(proc->res_type), 198 ampr(proc->res_type), RESULT); 199 } else { /* single argument, new or old style */ 200 fprintf(fout, 201 "\tif (clnt_call(clnt, %s, xdr_%s, %s%s, xdr_%s, " 202 "%s%s, TIMEOUT) != RPC_SUCCESS) {\n", 203 proc->proc_name, 204 stringfix(proc->args.decls->decl.type), 205 (newstyle ? "&" : ""), 206 (newstyle ? proc->args.decls->decl.name : "argp"), 207 stringfix(proc->res_type), 208 ampr(proc->res_type),RESULT); 209 } 210 fprintf(fout, "\t\treturn (NULL);\n"); 211 fprintf(fout, "\t}\n"); 212 if (streq(proc->res_type, "void")) { 213 fprintf(fout, "\treturn ((void *)%s%s);\n", 214 ampr(proc->res_type),RESULT); 215 } else { 216 fprintf(fout, "\treturn (%s%s);\n", ampr(proc->res_type),RESULT); 217 } 218 } 219 220