Home | History | Annotate | Line # | Download | only in rpcgen
rpc_svcout.c revision 1.12
      1 /*	$NetBSD: rpc_svcout.c,v 1.12 2001/03/21 00:30:39 mycroft Exp $	*/
      2 /*
      3  * Sun RPC is a product of Sun Microsystems, Inc. and is provided for
      4  * unrestricted use provided that this legend is included on all tape
      5  * media and as a part of the software program in whole or part.  Users
      6  * may copy or modify Sun RPC without charge, but are not authorized
      7  * to license or distribute it to anyone else except as part of a product or
      8  * program developed by the user or with the express written consent of
      9  * Sun Microsystems, Inc.
     10  *
     11  * SUN RPC IS PROVIDED AS IS WITH NO WARRANTIES OF ANY KIND INCLUDING THE
     12  * WARRANTIES OF DESIGN, MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
     13  * PURPOSE, OR ARISING FROM A COURSE OF DEALING, USAGE OR TRADE PRACTICE.
     14  *
     15  * Sun RPC is provided with no support and without any obligation on the
     16  * part of Sun Microsystems, Inc. to assist in its use, correction,
     17  * modification or enhancement.
     18  *
     19  * SUN MICROSYSTEMS, INC. SHALL HAVE NO LIABILITY WITH RESPECT TO THE
     20  * INFRINGEMENT OF COPYRIGHTS, TRADE SECRETS OR ANY PATENTS BY SUN RPC
     21  * OR ANY PART THEREOF.
     22  *
     23  * In no event will Sun Microsystems, Inc. be liable for any lost revenue
     24  * or profits or other special, indirect and consequential damages, even if
     25  * Sun has been advised of the possibility of such damages.
     26  *
     27  * Sun Microsystems, Inc.
     28  * 2550 Garcia Avenue
     29  * Mountain View, California  94043
     30  */
     31 
     32 #include <sys/cdefs.h>
     33 #ifndef lint
     34 #if 0
     35 static char sccsid[] = "@(#)rpc_svcout.c 1.29 89/03/30 (C) 1987 SMI";
     36 #else
     37 __RCSID("$NetBSD: rpc_svcout.c,v 1.12 2001/03/21 00:30:39 mycroft Exp $");
     38 #endif
     39 #endif
     40 
     41 /*
     42  * rpc_svcout.c, Server-skeleton outputter for the RPC protocol compiler
     43  */
     44 #include <stdio.h>
     45 #include <string.h>
     46 #include "rpc_scan.h"
     47 #include "rpc_parse.h"
     48 #include "rpc_util.h"
     49 
     50 static char RQSTP[] = "rqstp";
     51 static char TRANSP[] = "transp";
     52 static char ARG[] = "argument";
     53 static char RESULT[] = "result";
     54 static char ROUTINE[] = "local";
     55 
     56 static void p_xdrfunc __P((char *, char *));
     57 static void internal_proctype __P((proc_list *));
     58 static void write_real_program __P((definition *));
     59 static void write_program __P((definition *, char *));
     60 static void printerr __P((char *, char *));
     61 static void printif __P((char *, char *, char *, char *));
     62 static void write_inetmost __P((char *));
     63 static void print_return __P((char *));
     64 static void print_pmapunset __P((char *));
     65 static void print_err_message __P((char *));
     66 static void write_timeout_func __P((void));
     67 static void write_caller_func __P((void));
     68 static void write_pm_most __P((char *, int));
     69 static void write_rpc_svc_fg __P((char *, char *));
     70 static void open_log_file __P((char *, char *));
     71 
     72 char    _errbuf[256];		/* For all messages */
     73 
     74 static void
     75 p_xdrfunc(rname, typename)
     76 	char   *rname;
     77 	char   *typename;
     78 {
     79 	if (Cflag)
     80 		f_print(fout, "\t\txdr_%s = (xdrproc_t)xdr_%s;\n", rname,
     81 		    stringfix(typename));
     82 	else
     83 		f_print(fout, "\t\txdr_%s = xdr_%s;\n", rname, stringfix(typename));
     84 }
     85 
     86 static void
     87 internal_proctype(plist)
     88 	proc_list *plist;
     89 {
     90 	f_print(fout, "static ");
     91 	ptype(plist->res_prefix, plist->res_type, 1);
     92 	f_print(fout, "*");
     93 }
     94 
     95 
     96 /*
     97  * write most of the service, that is, everything but the registrations.
     98  */
     99 void
    100 write_most(infile, netflag, nomain)
    101 	char   *infile;		/* our name */
    102 	int     netflag;
    103 	int     nomain;
    104 {
    105 	if (inetdflag || pmflag) {
    106 		char   *var_type;
    107 		var_type = (nomain ? "extern" : "static");
    108 		f_print(fout, "%s int _rpcpmstart;", var_type);
    109 		f_print(fout, "\t\t/* Started by a port monitor ? */\n");
    110 		f_print(fout, "%s int _rpcfdtype;", var_type);
    111 		f_print(fout, "\t\t/* Whether Stream or Datagram ? */\n");
    112 		if (timerflag) {
    113 			f_print(fout, "%s int _rpcsvcdirty;", var_type);
    114 			f_print(fout, "\t/* Still serving ? */\n");
    115 		}
    116 		write_svc_aux(nomain);
    117 	}
    118 	/* write out dispatcher and stubs */
    119 	write_programs(nomain ? (char *) NULL : "static");
    120 
    121 	if (nomain)
    122 		return;
    123 
    124 	f_print(fout, "\n\n");
    125 	if (Cflag)
    126 		f_print(fout, "int main( int argc, char* argv[] );\n");
    127 	f_print(fout, "\nint\n");
    128 	if (Cflag)
    129 		f_print(fout, "main( int argc, char* argv[] )\n");
    130 	else
    131 		f_print(fout, "main(argc, argv)\nint argc;\nchar *argv[];\n");
    132 	f_print(fout, "{\n");
    133 	if (inetdflag) {
    134 		write_inetmost(infile);	/* Includes call to write_rpc_svc_fg() */
    135 	} else {
    136 		if (tirpcflag) {
    137 			if (netflag) {
    138 				f_print(fout, "\tSVCXPRT *%s;\n", TRANSP);
    139 				f_print(fout, "\tstruct netconfig *nconf = NULL;\n");
    140 			}
    141 			f_print(fout, "\tpid_t pid;\n");
    142 			f_print(fout, "\tint i;\n");
    143 			f_print(fout, "\tchar mname[FMNAMESZ + 1];\n\n");
    144 			write_pm_most(infile, netflag);
    145 			f_print(fout, "\telse {\n");
    146 			write_rpc_svc_fg(infile, "\t\t");
    147 			f_print(fout, "\t}\n");
    148 		} else {
    149 			f_print(fout, "\tSVCXPRT *%s;\n", TRANSP);
    150 			f_print(fout, "\n");
    151 			print_pmapunset("\t");
    152 		}
    153 	}
    154 
    155 	if (logflag && !inetdflag) {
    156 		open_log_file(infile, "\t");
    157 	}
    158 }
    159 /*
    160  * write a registration for the given transport
    161  */
    162 void
    163 write_netid_register(transp)
    164 	char   *transp;
    165 {
    166 	list   *l;
    167 	definition *def;
    168 	version_list *vp;
    169 	char   *sp;
    170 	char    tmpbuf[32];
    171 
    172 	sp = "";
    173 	f_print(fout, "\n");
    174 	f_print(fout, "%s\tnconf = getnetconfigent(\"%s\");\n", sp, transp);
    175 	f_print(fout, "%s\tif (nconf == NULL) {\n", sp);
    176 	(void) sprintf(_errbuf, "cannot find %s netid.", transp);
    177 	sprintf(tmpbuf, "%s\t\t", sp);
    178 	print_err_message(tmpbuf);
    179 	f_print(fout, "%s\t\texit(1);\n", sp);
    180 	f_print(fout, "%s\t}\n", sp);
    181 	f_print(fout, "%s\t%s = svc_tli_create(RPC_ANYFD, nconf, 0, 0, 0);\n",
    182 	    sp, TRANSP);
    183 	f_print(fout, "%s\tif (%s == NULL) {\n", sp, TRANSP);
    184 	(void) sprintf(_errbuf, "cannot create %s service.", transp);
    185 	print_err_message(tmpbuf);
    186 	f_print(fout, "%s\t\texit(1);\n", sp);
    187 	f_print(fout, "%s\t}\n", sp);
    188 
    189 	for (l = defined; l != NULL; l = l->next) {
    190 		def = (definition *) l->val;
    191 		if (def->def_kind != DEF_PROGRAM) {
    192 			continue;
    193 		}
    194 		for (vp = def->def.pr.versions; vp != NULL; vp = vp->next) {
    195 			f_print(fout,
    196 			    "%s\t(void) rpcb_unset(%s, %s, nconf);\n",
    197 			    sp, def->def_name, vp->vers_name);
    198 			f_print(fout,
    199 			    "%s\tif (!svc_reg(%s, %s, %s, ",
    200 			    sp, TRANSP, def->def_name, vp->vers_name);
    201 			pvname(def->def_name, vp->vers_num);
    202 			f_print(fout, ", nconf)) {\n");
    203 			(void) sprintf(_errbuf, "unable to register (%s, %s, %s).",
    204 			    def->def_name, vp->vers_name, transp);
    205 			print_err_message(tmpbuf);
    206 			f_print(fout, "%s\t\texit(1);\n", sp);
    207 			f_print(fout, "%s\t}\n", sp);
    208 		}
    209 	}
    210 	f_print(fout, "%s\tfreenetconfigent(nconf);\n", sp);
    211 }
    212 /*
    213  * write a registration for the given transport for TLI
    214  */
    215 void
    216 write_nettype_register(transp)
    217 	char   *transp;
    218 {
    219 	list   *l;
    220 	definition *def;
    221 	version_list *vp;
    222 
    223 	for (l = defined; l != NULL; l = l->next) {
    224 		def = (definition *) l->val;
    225 		if (def->def_kind != DEF_PROGRAM) {
    226 			continue;
    227 		}
    228 		for (vp = def->def.pr.versions; vp != NULL; vp = vp->next) {
    229 			f_print(fout, "\tif (!svc_create(");
    230 			pvname(def->def_name, vp->vers_num);
    231 			f_print(fout, ", %s, %s, \"%s\")) {\n ",
    232 			    def->def_name, vp->vers_name, transp);
    233 			(void) sprintf(_errbuf,
    234 			    "unable to create (%s, %s) for %s.",
    235 			    def->def_name, vp->vers_name, transp);
    236 			print_err_message("\t\t");
    237 			f_print(fout, "\t\texit(1);\n");
    238 			f_print(fout, "\t}\n");
    239 		}
    240 	}
    241 }
    242 /*
    243  * write the rest of the service
    244  */
    245 void
    246 write_rest()
    247 {
    248 	f_print(fout, "\n");
    249 	if (inetdflag) {
    250 		f_print(fout, "\tif (%s == (SVCXPRT *)NULL) {\n", TRANSP);
    251 		(void) sprintf(_errbuf, "could not create a handle");
    252 		print_err_message("\t\t");
    253 		f_print(fout, "\t\texit(1);\n");
    254 		f_print(fout, "\t}\n");
    255 		if (timerflag) {
    256 			f_print(fout, "\tif (_rpcpmstart) {\n");
    257 			f_print(fout,
    258 			    "\t\t(void) signal(SIGALRM, %s closedown);\n",
    259 			    Cflag ? "(SIG_PF)" : "(void(*)())");
    260 			f_print(fout, "\t\t(void) alarm(_RPCSVC_CLOSEDOWN);\n");
    261 			f_print(fout, "\t}\n");
    262 		}
    263 	}
    264 	f_print(fout, "\tsvc_run();\n");
    265 	(void) sprintf(_errbuf, "svc_run returned");
    266 	print_err_message("\t");
    267 	f_print(fout, "\texit(1);\n");
    268 	f_print(fout, "\t/* NOTREACHED */\n");
    269 	f_print(fout, "}\n");
    270 }
    271 
    272 void
    273 write_programs(storage)
    274 	char   *storage;
    275 {
    276 	list   *l;
    277 	definition *def;
    278 
    279 	/* write out stubs for procedure  definitions */
    280 	for (l = defined; l != NULL; l = l->next) {
    281 		def = (definition *) l->val;
    282 		if (def->def_kind == DEF_PROGRAM) {
    283 			write_real_program(def);
    284 		}
    285 	}
    286 
    287 	/* write out dispatcher for each program */
    288 	for (l = defined; l != NULL; l = l->next) {
    289 		def = (definition *) l->val;
    290 		if (def->def_kind == DEF_PROGRAM) {
    291 			write_program(def, storage);
    292 		}
    293 	}
    294 
    295 
    296 }
    297 /* write out definition of internal function (e.g. _printmsg_1(...))
    298    which calls server's defintion of actual function (e.g. printmsg_1(...)).
    299    Unpacks single user argument of printmsg_1 to call-by-value format
    300    expected by printmsg_1. */
    301 static void
    302 write_real_program(def)
    303 	definition *def;
    304 {
    305 	version_list *vp;
    306 	proc_list *proc;
    307 	decl_list *l;
    308 
    309 	if (!newstyle)
    310 		return;		/* not needed for old style */
    311 	for (vp = def->def.pr.versions; vp != NULL; vp = vp->next) {
    312 		for (proc = vp->procs; proc != NULL; proc = proc->next) {
    313 			f_print(fout, "\n");
    314 			internal_proctype(proc);
    315 			f_print(fout, "\n_");
    316 			pvname(proc->proc_name, vp->vers_num);
    317 			if (Cflag) {
    318 				f_print(fout, "(");
    319 				/* arg name */
    320 				if (proc->arg_num > 1)
    321 					f_print(fout, "%s",
    322 						proc->args.argname);
    323 				else
    324 					ptype(proc->args.decls->decl.prefix,
    325 					    proc->args.decls->decl.type, 0);
    326 				f_print(fout, " *argp, struct svc_req *%s)\n",
    327 				    RQSTP);
    328 			} else {
    329 				f_print(fout, "(argp, %s)\n", RQSTP);
    330 				/* arg name */
    331 				if (proc->arg_num > 1)
    332 					f_print(fout, "\t%s *argp;\n", proc->args.argname);
    333 				else {
    334 					f_print(fout, "\t");
    335 					ptype(proc->args.decls->decl.prefix,
    336 					    proc->args.decls->decl.type, 0);
    337 					f_print(fout, " *argp;\n");
    338 				}
    339 				f_print(fout, "	struct svc_req *%s;\n", RQSTP);
    340 			}
    341 
    342 			f_print(fout, "{\n");
    343 			f_print(fout, "\treturn(");
    344 			pvname_svc(proc->proc_name, vp->vers_num);
    345 			f_print(fout, "(");
    346 			if (proc->arg_num < 2) {	/* single argument */
    347 				if (!streq(proc->args.decls->decl.type, "void"))
    348 					f_print(fout, "*argp, ");	/* non-void */
    349 			} else {
    350 				for (l = proc->args.decls; l != NULL; l = l->next)
    351 					f_print(fout, "argp->%s, ", l->decl.name);
    352 			}
    353 			f_print(fout, "%s));\n}\n", RQSTP);
    354 		}
    355 	}
    356 }
    357 
    358 static void
    359 write_program(def, storage)
    360 	definition *def;
    361 	char   *storage;
    362 {
    363 	version_list *vp;
    364 	proc_list *proc;
    365 	int     filled;
    366 
    367 	for (vp = def->def.pr.versions; vp != NULL; vp = vp->next) {
    368 		if (Cflag) {
    369 			f_print(fout, "\n");
    370 			if (storage != NULL) {
    371 				f_print(fout, "%s ", storage);
    372 			}
    373 			f_print(fout, "void ");
    374 			pvname(def->def_name, vp->vers_num);
    375 			f_print(fout, "(struct svc_req *%s, ", RQSTP);
    376 			f_print(fout, "SVCXPRT *%s);\n", TRANSP);
    377 		}
    378 		f_print(fout, "\n");
    379 		if (storage != NULL) {
    380 			f_print(fout, "%s ", storage);
    381 		}
    382 		f_print(fout, "void\n");
    383 		pvname(def->def_name, vp->vers_num);
    384 
    385 		if (Cflag) {
    386 			f_print(fout, "(struct svc_req *%s, ", RQSTP);
    387 			f_print(fout, "SVCXPRT *%s)\n", TRANSP);
    388 		} else {
    389 			f_print(fout, "(%s, %s)\n", RQSTP, TRANSP);
    390 			f_print(fout, "	struct svc_req *%s;\n", RQSTP);
    391 			f_print(fout, "	SVCXPRT *%s;\n", TRANSP);
    392 		}
    393 
    394 		f_print(fout, "{\n");
    395 
    396 		filled = 0;
    397 		f_print(fout, "\tunion {\n");
    398 		for (proc = vp->procs; proc != NULL; proc = proc->next) {
    399 			if (proc->arg_num < 2) {	/* single argument */
    400 				if (streq(proc->args.decls->decl.type,
    401 					"void")) {
    402 					continue;
    403 				}
    404 				filled = 1;
    405 				f_print(fout, "\t\t");
    406 				ptype(proc->args.decls->decl.prefix,
    407 				    proc->args.decls->decl.type, 0);
    408 				pvname(proc->proc_name, vp->vers_num);
    409 				f_print(fout, "_arg;\n");
    410 
    411 			} else {
    412 				filled = 1;
    413 				f_print(fout, "\t\t%s", proc->args.argname);
    414 				f_print(fout, " ");
    415 				pvname(proc->proc_name, vp->vers_num);
    416 				f_print(fout, "_arg;\n");
    417 			}
    418 		}
    419 		if (!filled) {
    420 			f_print(fout, "\t\tint fill;\n");
    421 		}
    422 		f_print(fout, "\t} %s;\n", ARG);
    423 		if (Mflag) {
    424 			f_print(fout, "\tunion {\n");
    425 			for (proc = vp->procs; proc != NULL; proc = proc->next) {
    426 				f_print(fout, "\t\t");
    427 				if (streq(proc->res_type, "void"))
    428 					f_print(fout, "char ");
    429 				else
    430 					ptype(proc->res_prefix, proc->res_type,
    431 					    1);
    432 				pvname(proc->proc_name, vp->vers_num);
    433 				f_print(fout, "_res;\n");
    434 			}
    435 			f_print(fout, "\t} %s;\n", RESULT);
    436 			f_print(fout, "\tbool_t retval;\n");
    437 		} else
    438 			f_print(fout, "\tchar *%s;\n", RESULT);
    439 
    440 		if (Cflag) {
    441 			f_print(fout, "\txdrproc_t xdr_%s, xdr_%s;\n", ARG, RESULT);
    442 			if (Mflag)
    443 				f_print(fout,
    444 				    "\tbool_t (*%s)(char *, void *, struct svc_req *);\n",
    445 				    ROUTINE);
    446 			else
    447 				f_print(fout,
    448 				    "\tchar *(*%s)(char *, struct svc_req *);\n",
    449 				    ROUTINE);
    450 		} else {
    451 			f_print(fout, "\tbool_t (*xdr_%s)(), (*xdr_%s)();\n", ARG, RESULT);
    452 			if (Mflag)
    453 				f_print(fout, "\tbool_t (*%s)();\n", ROUTINE);
    454 			else
    455 				f_print(fout, "\tchar *(*%s)();\n", ROUTINE);
    456 		}
    457 
    458 		f_print(fout, "\n");
    459 
    460 		if (callerflag)
    461 			f_print(fout, "\tcaller = transp;\n");	/* EVAS */
    462 		if (timerflag)
    463 			f_print(fout, "\t_rpcsvcdirty = 1;\n");
    464 		f_print(fout, "\tswitch (%s->rq_proc) {\n", RQSTP);
    465 		if (!nullproc(vp->procs)) {
    466 			f_print(fout, "\tcase NULLPROC:\n");
    467 			if (Cflag) {
    468 			  	f_print(fout,
    469 					"\t\t(void) svc_sendreply(%s, (xdrproc_t)xdr_void, (char *)NULL);\n", TRANSP);
    470 			} else {
    471 			  	f_print(fout,
    472 					"\t\t(void) svc_sendreply(%s, xdr_void, (char *)NULL);\n",
    473 					TRANSP);
    474 			}
    475 			print_return("\t\t");
    476 			f_print(fout, "\n");
    477 		}
    478 		for (proc = vp->procs; proc != NULL; proc = proc->next) {
    479 			f_print(fout, "\tcase %s:\n", proc->proc_name);
    480 			if (proc->arg_num < 2) {	/* single argument */
    481 				p_xdrfunc(ARG, proc->args.decls->decl.type);
    482 			} else {
    483 				p_xdrfunc(ARG, proc->args.argname);
    484 			}
    485 			p_xdrfunc(RESULT, proc->res_type);
    486 			if (Cflag) {
    487 				if (Mflag)
    488 					f_print(fout,
    489 					    "\t\t%s = (bool_t (*)(char *, void *, struct svc_req *))",
    490 					    ROUTINE);
    491 				else
    492 					f_print(fout,
    493 					    "\t\t%s = (char *(*)(char *, struct svc_req *))",
    494 					    ROUTINE);
    495 			} else {
    496 				if (Mflag)
    497 					f_print(fout, "\t\t%s = (bool_t (*)())", ROUTINE);
    498 				else
    499 					f_print(fout, "\t\t%s = (char *(*)())", ROUTINE);
    500 			}
    501 
    502 			if (newstyle) {	/* new style: calls internal routine */
    503 				f_print(fout, "_");
    504 			}
    505 			if (!newstyle)
    506 				pvname_svc(proc->proc_name, vp->vers_num);
    507 			else
    508 				pvname(proc->proc_name, vp->vers_num);
    509 			f_print(fout, ";\n");
    510 			f_print(fout, "\t\tbreak;\n\n");
    511 		}
    512 		f_print(fout, "\tdefault:\n");
    513 		printerr("noproc", TRANSP);
    514 		print_return("\t\t");
    515 		f_print(fout, "\t}\n");
    516 
    517 		f_print(fout, "\t(void) memset((char *)&%s, 0, sizeof (%s));\n", ARG, ARG);
    518 		printif("getargs", TRANSP, "(caddr_t)&", ARG);
    519 		printerr("decode", TRANSP);
    520 		print_return("\t\t");
    521 		f_print(fout, "\t}\n");
    522 
    523 		if (Cflag) {
    524 			if (Mflag)
    525 				f_print(fout, "\tretval = (*%s)((char *)&%s, (void *)&%s, %s);\n",
    526 				    ROUTINE, ARG, RESULT, RQSTP);
    527 			else
    528 				f_print(fout, "\t%s = (*%s)((char *)&%s, %s);\n",
    529 				    RESULT, ROUTINE, ARG, RQSTP);
    530 		} else {
    531 			if (Mflag)
    532 				f_print(fout, "\tretval = (*%s)(&%s, &%s, %s);\n",
    533 				    ROUTINE, ARG, RESULT, RQSTP);
    534 			else
    535 				f_print(fout, "\t%s = (*%s)(&%s, %s);\n",
    536 				    RESULT, ROUTINE, ARG, RQSTP);
    537 		}
    538 		if (Mflag)
    539 			f_print(fout,
    540 			    "\tif (retval > 0 && !svc_sendreply(%s, xdr_%s, (char *)&%s)) {\n",
    541 			    TRANSP, RESULT, RESULT);
    542 		else
    543 			f_print(fout,
    544 			    "\tif (%s != NULL && !svc_sendreply(%s, xdr_%s, %s)) {\n",
    545 			    RESULT, TRANSP, RESULT, RESULT);
    546 		printerr("systemerr", TRANSP);
    547 		f_print(fout, "\t}\n");
    548 
    549 		printif("freeargs", TRANSP, "(caddr_t)&", ARG);
    550 		(void) sprintf(_errbuf, "unable to free arguments");
    551 		print_err_message("\t\t");
    552 		f_print(fout, "\t\texit(1);\n");
    553 		f_print(fout, "\t}\n");
    554 
    555 		f_print(fout, "\tif (!");
    556 		pvname(def->def_name, vp->vers_num);
    557 		f_print(fout, "_freeresult");
    558 		f_print(fout, "(%s, xdr_%s, (caddr_t)&%s)) {\n", TRANSP, RESULT,
    559 		    RESULT);
    560 		(void) sprintf(_errbuf, "unable to free results");
    561 		print_err_message("\t\t");
    562 		f_print(fout, "\t\texit(1);\n");
    563 		f_print(fout, "\t}\n");
    564 
    565 		print_return("\t");
    566 		f_print(fout, "}\n");
    567 	}
    568 }
    569 
    570 static void
    571 printerr(err, transp)
    572 	char   *err;
    573 	char   *transp;
    574 {
    575 	f_print(fout, "\t\tsvcerr_%s(%s);\n", err, transp);
    576 }
    577 
    578 static void
    579 printif(proc, transp, prefix, arg)
    580 	char   *proc;
    581 	char   *transp;
    582 	char   *prefix;
    583 	char   *arg;
    584 {
    585 	f_print(fout, "\tif (!svc_%s(%s, xdr_%s, %s%s)) {\n",
    586 	    proc, transp, arg, prefix, arg);
    587 }
    588 
    589 int
    590 nullproc(proc)
    591 	proc_list *proc;
    592 {
    593 	for (; proc != NULL; proc = proc->next) {
    594 		if (streq(proc->proc_num, "0")) {
    595 			return (1);
    596 		}
    597 	}
    598 	return (0);
    599 }
    600 
    601 static void
    602 write_inetmost(infile)
    603 	char   *infile;
    604 {
    605 	f_print(fout, "\tSVCXPRT *%s = NULL;\n", TRANSP);
    606 	f_print(fout, "\tint sock;\n");
    607 	f_print(fout, "\tint proto = 0;\n");
    608 	f_print(fout, "\tstruct sockaddr_in saddr;\n");
    609 	f_print(fout, "\tint asize = sizeof (saddr);\n");
    610 	f_print(fout, "\n");
    611 	f_print(fout,
    612 	    "\tif (getsockname(0, (struct sockaddr *)&saddr, &asize) == 0) {\n");
    613 	f_print(fout, "\t\tint ssize = sizeof (int);\n\n");
    614 	f_print(fout, "\t\tif (saddr.sin_family != AF_INET)\n");
    615 	f_print(fout, "\t\t\texit(1);\n");
    616 	f_print(fout, "\t\tif (getsockopt(0, SOL_SOCKET, SO_TYPE,\n");
    617 	f_print(fout, "\t\t\t\t(char *)&_rpcfdtype, &ssize) == -1)\n");
    618 	f_print(fout, "\t\t\texit(1);\n");
    619 	f_print(fout, "\t\tsock = 0;\n");
    620 	f_print(fout, "\t\t_rpcpmstart = 1;\n");
    621 	f_print(fout, "\t\tproto = 0;\n");
    622 	open_log_file(infile, "\t\t");
    623 	f_print(fout, "\t} else {\n");
    624 	write_rpc_svc_fg(infile, "\t\t");
    625 	f_print(fout, "\t\tsock = RPC_ANYSOCK;\n");
    626 	print_pmapunset("\t\t");
    627 	f_print(fout, "\t}\n");
    628 }
    629 
    630 static void
    631 print_return(space)
    632 	char   *space;
    633 {
    634 	if (exitnow)
    635 		f_print(fout, "%sexit(0);\n", space);
    636 	else {
    637 		if (timerflag)
    638 			f_print(fout, "%s_rpcsvcdirty = 0;\n", space);
    639 		f_print(fout, "%sreturn;\n", space);
    640 	}
    641 }
    642 
    643 static void
    644 print_pmapunset(space)
    645 	char   *space;
    646 {
    647 	list   *l;
    648 	definition *def;
    649 	version_list *vp;
    650 
    651 	for (l = defined; l != NULL; l = l->next) {
    652 		def = (definition *) l->val;
    653 		if (def->def_kind == DEF_PROGRAM) {
    654 			for (vp = def->def.pr.versions; vp != NULL;
    655 			    vp = vp->next) {
    656 				f_print(fout, "%s(void) pmap_unset(%s, %s);\n",
    657 				    space, def->def_name, vp->vers_name);
    658 			}
    659 		}
    660 	}
    661 }
    662 
    663 static void
    664 print_err_message(space)
    665 	char   *space;
    666 {
    667 	if (logflag)
    668 		f_print(fout, "%ssyslog(LOG_ERR, \"%s\");\n", space, _errbuf);
    669 	else
    670 		if (inetdflag || pmflag)
    671 			f_print(fout, "%s_msgout(\"%s\");\n", space, _errbuf);
    672 		else
    673 			f_print(fout, "%sfprintf(stderr, \"%s\");\n", space, _errbuf);
    674 }
    675 /*
    676  * Write the server auxiliary function ( _msgout, timeout)
    677  */
    678 void
    679 write_svc_aux(nomain)
    680 	int     nomain;
    681 {
    682 	if (!logflag)
    683 		write_msg_out();
    684 	if (!nomain)
    685 		write_timeout_func();
    686 	if (callerflag)		/* EVAS */
    687 		write_caller_func();	/* EVAS */
    688 }
    689 /*
    690  * Write the _msgout function
    691  */
    692 void
    693 write_msg_out()
    694 {
    695 	f_print(fout, "\n");
    696 	f_print(fout, "static\n");
    697 	if (!Cflag) {
    698 		f_print(fout, "void _msgout(msg)\n");
    699 		f_print(fout, "\tchar *msg;\n");
    700 	} else {
    701 		f_print(fout, "void _msgout(char* msg)\n");
    702 	}
    703 	f_print(fout, "{\n");
    704 	f_print(fout, "#ifdef RPC_SVC_FG\n");
    705 	if (inetdflag || pmflag)
    706 		f_print(fout, "\tif (_rpcpmstart)\n");
    707 	f_print(fout, "\t\tsyslog(LOG_ERR, \"%%s\", msg);\n");
    708 	f_print(fout, "\telse\n");
    709 	f_print(fout, "\t\t(void) fprintf(stderr, \"%%s\\n\", msg);\n");
    710 	f_print(fout, "#else\n");
    711 	f_print(fout, "\tsyslog(LOG_ERR, \"%%s\", msg);\n");
    712 	f_print(fout, "#endif\n");
    713 	f_print(fout, "}\n");
    714 }
    715 /*
    716  * Write the timeout function
    717  */
    718 static void
    719 write_timeout_func()
    720 {
    721 	if (!timerflag)
    722 		return;
    723 	if (Cflag) {
    724 		f_print(fout, "\n");
    725 		f_print(fout, "static void closedown(void);\n");
    726 	}
    727 	f_print(fout, "\n");
    728 	f_print(fout, "static void\n");
    729 	f_print(fout, "closedown()\n");
    730 	f_print(fout, "{\n");
    731 	f_print(fout, "\tif (_rpcsvcdirty == 0) {\n");
    732 	f_print(fout, "\t\textern fd_set svc_fdset;\n");
    733 	f_print(fout, "\t\tstatic int size;\n");
    734 	f_print(fout, "\t\tint i, openfd;\n");
    735 	if (tirpcflag && pmflag) {
    736 		f_print(fout, "\t\tstruct t_info tinfo;\n\n");
    737 		f_print(fout, "\t\tif (!t_getinfo(0, &tinfo) && (tinfo.servtype == T_CLTS))\n");
    738 	} else {
    739 		f_print(fout, "\n\t\tif (_rpcfdtype == SOCK_DGRAM)\n");
    740 	}
    741 	f_print(fout, "\t\t\texit(0);\n");
    742 	f_print(fout, "\t\tif (size == 0) {\n");
    743 	if (tirpcflag) {
    744 		f_print(fout, "\t\t\tstruct rlimit rl;\n\n");
    745 		f_print(fout, "\t\t\trl.rlim_max = 0;\n");
    746 		f_print(fout, "\t\t\tgetrlimit(RLIMIT_NOFILE, &rl);\n");
    747 		f_print(fout, "\t\t\tif ((size = rl.rlim_max) == 0)\n");
    748 		f_print(fout, "\t\t\t\treturn;\n");
    749 	} else {
    750 		f_print(fout, "\t\t\tsize = getdtablesize();\n");
    751 	}
    752 	f_print(fout, "\t\t}\n");
    753 	f_print(fout, "\t\tfor (i = 0, openfd = 0; i < size && openfd < 2; i++)\n");
    754 	f_print(fout, "\t\t\tif (FD_ISSET(i, &svc_fdset))\n");
    755 	f_print(fout, "\t\t\t\topenfd++;\n");
    756 	f_print(fout, "\t\tif (openfd <= (_rpcpmstart?0:1))\n");
    757 	f_print(fout, "\t\t\texit(0);\n");
    758 	f_print(fout, "\t}\n");
    759 	f_print(fout, "\t(void) alarm(_RPCSVC_CLOSEDOWN);\n");
    760 	f_print(fout, "}\n");
    761 }
    762 
    763 static void
    764 write_caller_func()
    765 {				/* EVAS */
    766 #define	P(s)	f_print(fout, s);
    767 
    768 	P("\n");
    769 	P("char *svc_caller()\n");
    770 	P("{\n");
    771 	P("	struct sockaddr_in actual;\n");
    772 	P("	struct hostent *hp;\n");
    773 	P("	static struct in_addr prev;\n");
    774 	P("	static char cname[128];\n\n");
    775 
    776 	P("	actual = *svc_getcaller(caller);\n\n");
    777 
    778 	P("	if (memcmp((char *)&actual.sin_addr, (char *)&prev,\n");
    779 	P("		 sizeof(struct in_addr)) == 0)\n");
    780 	P("		return (cname);\n\n");
    781 
    782 	P("	prev = actual.sin_addr;\n\n");
    783 
    784 	P("	hp = gethostbyaddr((char *) &actual.sin_addr, sizeof(actual.sin_addr), AF_INET);\n");
    785 	P("	if (hp == NULL) {                       /* dummy one up */\n");
    786 	P("		extern char *inet_ntoa();\n");
    787 	P("		strcpy(cname, inet_ntoa(actual.sin_addr));\n");
    788 	P("	} else {\n");
    789 	P("		strcpy(cname, hp->h_name);\n");
    790 	P("	}\n\n");
    791 
    792 	P("	return (cname);\n");
    793 	P("}\n");
    794 
    795 #undef P
    796 }
    797 /*
    798  * Write the most of port monitor support
    799  */
    800 static void
    801 write_pm_most(infile, netflag)
    802 	char   *infile;
    803 	int     netflag;
    804 {
    805 	list   *l;
    806 	definition *def;
    807 	version_list *vp;
    808 
    809 	f_print(fout, "\tif (!ioctl(0, I_LOOK, mname) &&\n");
    810 	f_print(fout, "\t\t(!strcmp(mname, \"sockmod\") ||");
    811 	f_print(fout, " !strcmp(mname, \"timod\"))) {\n");
    812 	f_print(fout, "\t\tchar *netid;\n");
    813 	if (!netflag) {		/* Not included by -n option */
    814 		f_print(fout, "\t\tstruct netconfig *nconf = NULL;\n");
    815 		f_print(fout, "\t\tSVCXPRT *%s;\n", TRANSP);
    816 	}
    817 	if (timerflag)
    818 		f_print(fout, "\t\tint pmclose;\n");
    819 /* not necessary, defined in /usr/include/stdlib */
    820 /*	f_print(fout, "\t\textern char *getenv();\n");*/
    821 	f_print(fout, "\n");
    822 	f_print(fout, "\t\t_rpcpmstart = 1;\n");
    823 	if (logflag)
    824 		open_log_file(infile, "\t\t");
    825 	f_print(fout, "\t\tif ((netid = getenv(\"NLSPROVIDER\")) == NULL) {\n");
    826 	sprintf(_errbuf, "cannot get transport name");
    827 	print_err_message("\t\t\t");
    828 	f_print(fout, "\t\t} else if ((nconf = getnetconfigent(netid)) == NULL) {\n");
    829 	sprintf(_errbuf, "cannot get transport info");
    830 	print_err_message("\t\t\t");
    831 	f_print(fout, "\t\t}\n");
    832 	/*
    833 	 * A kludgy support for inetd services. Inetd only works with
    834 	 * sockmod, and RPC works only with timod, hence all this jugglery
    835 	 */
    836 	f_print(fout, "\t\tif (strcmp(mname, \"sockmod\") == 0) {\n");
    837 	f_print(fout, "\t\t\tif (ioctl(0, I_POP, 0) || ioctl(0, I_PUSH, \"timod\")) {\n");
    838 	sprintf(_errbuf, "could not get the right module");
    839 	print_err_message("\t\t\t\t");
    840 	f_print(fout, "\t\t\t\texit(1);\n");
    841 	f_print(fout, "\t\t\t}\n");
    842 	f_print(fout, "\t\t}\n");
    843 	if (timerflag)
    844 		f_print(fout, "\t\tpmclose = (t_getstate(0) != T_DATAXFER);\n");
    845 	f_print(fout, "\t\tif ((%s = svc_tli_create(0, nconf, NULL, 0, 0)) == NULL) {\n",
    846 	    TRANSP);
    847 	sprintf(_errbuf, "cannot create server handle");
    848 	print_err_message("\t\t\t");
    849 	f_print(fout, "\t\t\texit(1);\n");
    850 	f_print(fout, "\t\t}\n");
    851 	f_print(fout, "\t\tif (nconf)\n");
    852 	f_print(fout, "\t\t\tfreenetconfigent(nconf);\n");
    853 	for (l = defined; l != NULL; l = l->next) {
    854 		def = (definition *) l->val;
    855 		if (def->def_kind != DEF_PROGRAM) {
    856 			continue;
    857 		}
    858 		for (vp = def->def.pr.versions; vp != NULL; vp = vp->next) {
    859 			f_print(fout,
    860 			    "\t\tif (!svc_reg(%s, %s, %s, ",
    861 			    TRANSP, def->def_name, vp->vers_name);
    862 			pvname(def->def_name, vp->vers_num);
    863 			f_print(fout, ", 0)) {\n");
    864 			(void) sprintf(_errbuf, "unable to register (%s, %s).",
    865 			    def->def_name, vp->vers_name);
    866 			print_err_message("\t\t\t");
    867 			f_print(fout, "\t\t\texit(1);\n");
    868 			f_print(fout, "\t\t}\n");
    869 		}
    870 	}
    871 	if (timerflag) {
    872 		f_print(fout, "\t\tif (pmclose) {\n");
    873 		f_print(fout, "\t\t\t(void) signal(SIGALRM, %s closedown);\n",
    874 		    Cflag ? "(SIG_PF)" : "(void(*)())");
    875 		f_print(fout, "\t\t\t(void) alarm(_RPCSVC_CLOSEDOWN);\n");
    876 		f_print(fout, "\t\t}\n");
    877 	}
    878 	f_print(fout, "\t\tsvc_run();\n");
    879 	f_print(fout, "\t\texit(1);\n");
    880 	f_print(fout, "\t\t/* NOTREACHED */\n");
    881 	f_print(fout, "\t}\n");
    882 }
    883 /*
    884  * Support for backgrounding the server if self started.
    885  */
    886 static void
    887 write_rpc_svc_fg(infile, sp)
    888 	char   *infile;
    889 	char   *sp;
    890 {
    891 	f_print(fout, "#ifndef RPC_SVC_FG\n");
    892 	f_print(fout, "%sint size;\n", sp);
    893 	if (tirpcflag)
    894 		f_print(fout, "%sstruct rlimit rl;\n", sp);
    895 	if (inetdflag)
    896 		f_print(fout, "%sint pid, i;\n\n", sp);
    897 	f_print(fout, "%spid = fork();\n", sp);
    898 	f_print(fout, "%sif (pid < 0) {\n", sp);
    899 	f_print(fout, "%s\tperror(\"cannot fork\");\n", sp);
    900 	f_print(fout, "%s\texit(1);\n", sp);
    901 	f_print(fout, "%s}\n", sp);
    902 	f_print(fout, "%sif (pid)\n", sp);
    903 	f_print(fout, "%s\texit(0);\n", sp);
    904 	/* get number of file descriptors */
    905 	if (tirpcflag) {
    906 		f_print(fout, "%srl.rlim_max = 0;\n", sp);
    907 		f_print(fout, "%sgetrlimit(RLIMIT_NOFILE, &rl);\n", sp);
    908 		f_print(fout, "%sif ((size = rl.rlim_max) == 0)\n", sp);
    909 		f_print(fout, "%s\texit(1);\n", sp);
    910 	} else {
    911 		f_print(fout, "%ssize = getdtablesize();\n", sp);
    912 	}
    913 
    914 	f_print(fout, "%sfor (i = 0; i < size; i++)\n", sp);
    915 	f_print(fout, "%s\t(void) close(i);\n", sp);
    916 	/* Redirect stderr and stdout to console */
    917 	f_print(fout, "%si = open(\"/dev/console\", 2);\n", sp);
    918 	f_print(fout, "%s(void) dup2(i, 1);\n", sp);
    919 	f_print(fout, "%s(void) dup2(i, 2);\n", sp);
    920 	/* This removes control of the controlling terminal */
    921 	if (tirpcflag)
    922 		f_print(fout, "%ssetsid();\n", sp);
    923 	else {
    924 		f_print(fout, "%si = open(\"/dev/tty\", 2);\n", sp);
    925 		f_print(fout, "%sif (i >= 0) {\n", sp);
    926 		f_print(fout, "%s\t(void) ioctl(i, TIOCNOTTY, (char *)NULL);\n", sp);;
    927 		f_print(fout, "%s\t(void) close(i);\n", sp);
    928 		f_print(fout, "%s}\n", sp);
    929 	}
    930 	if (!logflag)
    931 		open_log_file(infile, sp);
    932 	f_print(fout, "#endif\n");
    933 	if (logflag)
    934 		open_log_file(infile, sp);
    935 }
    936 
    937 static void
    938 open_log_file(infile, sp)
    939 	char   *infile;
    940 	char   *sp;
    941 {
    942 	char   *s, *p;
    943 
    944 	s = strrchr(infile, '.');
    945 	if (s)
    946 		*s = '\0';
    947 	p = strrchr(infile, '/');
    948 	if (p)
    949 		p++;
    950 	else
    951 		p = infile;
    952 	f_print(fout, "%sopenlog(\"%s\", LOG_PID, LOG_DAEMON);\n", sp, p);
    953 	if (s)
    954 		*s = '.';
    955 }
    956 
    957 /*
    958  * write a registration for the given transport for Inetd
    959  */
    960 void
    961 write_inetd_register(transp)
    962 	char   *transp;
    963 {
    964 	list   *l;
    965 	definition *def;
    966 	version_list *vp;
    967 	char   *sp;
    968 	int     isudp;
    969 	char    tmpbuf[32];
    970 
    971 	if (inetdflag)
    972 		sp = "\t";
    973 	else
    974 		sp = "";
    975 	if (streq(transp, "udp"))
    976 		isudp = 1;
    977 	else
    978 		isudp = 0;
    979 	f_print(fout, "\n");
    980 	if (inetdflag) {
    981 		f_print(fout, "\tif ((_rpcfdtype == 0) || (_rpcfdtype == %s)) {\n",
    982 		    isudp ? "SOCK_DGRAM" : "SOCK_STREAM");
    983 	}
    984 	if (inetdflag && streq(transp, "tcp")) {
    985 		f_print(fout, "%s\tif (_rpcpmstart)\n", sp);
    986 
    987 		f_print(fout, "%s\t\t%s = svc%s_create(%s",
    988 		    sp, TRANSP, "fd", inetdflag ? "sock" : "RPC_ANYSOCK");
    989 		if (!isudp)
    990 			f_print(fout, ", 0, 0");
    991 		f_print(fout, ");\n");
    992 
    993 		f_print(fout, "%s\telse\n", sp);
    994 
    995 		f_print(fout, "%s\t\t%s = svc%s_create(%s",
    996 		    sp, TRANSP, transp, inetdflag ? "sock" : "RPC_ANYSOCK");
    997 		if (!isudp)
    998 			f_print(fout, ", 0, 0");
    999 		f_print(fout, ");\n");
   1000 
   1001 	} else {
   1002 		f_print(fout, "%s\t%s = svc%s_create(%s",
   1003 		    sp, TRANSP, transp, inetdflag ? "sock" : "RPC_ANYSOCK");
   1004 		if (!isudp)
   1005 			f_print(fout, ", 0, 0");
   1006 		f_print(fout, ");\n");
   1007 	}
   1008 	f_print(fout, "%s\tif (%s == NULL) {\n", sp, TRANSP);
   1009 	(void) sprintf(_errbuf, "cannot create %s service.", transp);
   1010 	(void) sprintf(tmpbuf, "%s\t\t", sp);
   1011 	print_err_message(tmpbuf);
   1012 	f_print(fout, "%s\t\texit(1);\n", sp);
   1013 	f_print(fout, "%s\t}\n", sp);
   1014 
   1015 	if (inetdflag) {
   1016 		f_print(fout, "%s\tif (!_rpcpmstart)\n\t", sp);
   1017 		f_print(fout, "%s\tproto = IPPROTO_%s;\n",
   1018 		    sp, isudp ? "UDP" : "TCP");
   1019 	}
   1020 	for (l = defined; l != NULL; l = l->next) {
   1021 		def = (definition *) l->val;
   1022 		if (def->def_kind != DEF_PROGRAM) {
   1023 			continue;
   1024 		}
   1025 		for (vp = def->def.pr.versions; vp != NULL; vp = vp->next) {
   1026 			f_print(fout, "%s\tif (!svc_register(%s, %s, %s, ",
   1027 			    sp, TRANSP, def->def_name, vp->vers_name);
   1028 			pvname(def->def_name, vp->vers_num);
   1029 			if (inetdflag)
   1030 				f_print(fout, ", proto)) {\n");
   1031 			else
   1032 				f_print(fout, ", IPPROTO_%s)) {\n",
   1033 				    isudp ? "UDP" : "TCP");
   1034 			(void) sprintf(_errbuf, "unable to register (%s, %s, %s).",
   1035 			    def->def_name, vp->vers_name, transp);
   1036 			print_err_message(tmpbuf);
   1037 			f_print(fout, "%s\t\texit(1);\n", sp);
   1038 			f_print(fout, "%s\t}\n", sp);
   1039 		}
   1040 	}
   1041 	if (inetdflag)
   1042 		f_print(fout, "\t}\n");
   1043 }
   1044