xref: /netbsd-src/usr.bin/rpcgen/rpc_svcout.c (revision 7cc2f76925f078d01ddc9e640a98f4ccfc9f8c3b)
1 /*	$NetBSD: rpc_svcout.c,v 1.11 2000/10/11 14:46:17 is 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.11 2000/10/11 14:46:17 is 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 		f_print(fout, "\tchar *%s;\n", RESULT);
424 
425 		if (Cflag) {
426 			f_print(fout, "\txdrproc_t xdr_%s, xdr_%s;\n", ARG, RESULT);
427 			f_print(fout,
428 			    "\tchar *(*%s)(char *, struct svc_req *);\n",
429 			    ROUTINE);
430 		} else {
431 			f_print(fout, "\tbool_t (*xdr_%s)(), (*xdr_%s)();\n", ARG, RESULT);
432 			f_print(fout, "\tchar *(*%s)();\n", ROUTINE);
433 		}
434 
435 		f_print(fout, "\n");
436 
437 		if (callerflag)
438 			f_print(fout, "\tcaller = transp;\n");	/* EVAS */
439 		if (timerflag)
440 			f_print(fout, "\t_rpcsvcdirty = 1;\n");
441 		f_print(fout, "\tswitch (%s->rq_proc) {\n", RQSTP);
442 		if (!nullproc(vp->procs)) {
443 			f_print(fout, "\tcase NULLPROC:\n");
444 			if (Cflag) {
445 			  	f_print(fout,
446 					"\t\t(void) svc_sendreply(%s, (xdrproc_t) xdr_void, (char *)NULL);\n", TRANSP);
447 			} else {
448 			  	f_print(fout,
449 					"\t\t(void) svc_sendreply(%s, xdr_void, (char *)NULL);\n",
450 					TRANSP);
451 			}
452 			print_return("\t\t");
453 			f_print(fout, "\n");
454 		}
455 		for (proc = vp->procs; proc != NULL; proc = proc->next) {
456 			f_print(fout, "\tcase %s:\n", proc->proc_name);
457 			if (proc->arg_num < 2) {	/* single argument */
458 				p_xdrfunc(ARG, proc->args.decls->decl.type);
459 			} else {
460 				p_xdrfunc(ARG, proc->args.argname);
461 			}
462 			p_xdrfunc(RESULT, proc->res_type);
463 			if (Cflag)
464 				f_print(fout,
465 				    "\t\t%s = (char *(*)(char *, struct svc_req *)) ",
466 				    ROUTINE);
467 			else
468 				f_print(fout, "\t\t%s = (char *(*)()) ", ROUTINE);
469 
470 			if (newstyle) {	/* new style: calls internal routine */
471 				f_print(fout, "_");
472 			}
473 			if (!newstyle)
474 				pvname_svc(proc->proc_name, vp->vers_num);
475 			else
476 				pvname(proc->proc_name, vp->vers_num);
477 			f_print(fout, ";\n");
478 			f_print(fout, "\t\tbreak;\n\n");
479 		}
480 		f_print(fout, "\tdefault:\n");
481 		printerr("noproc", TRANSP);
482 		print_return("\t\t");
483 		f_print(fout, "\t}\n");
484 
485 		f_print(fout, "\t(void) memset((char *)&%s, 0, sizeof (%s));\n", ARG, ARG);
486 		printif("getargs", TRANSP, "(caddr_t) &", ARG);
487 		printerr("decode", TRANSP);
488 		print_return("\t\t");
489 		f_print(fout, "\t}\n");
490 
491 		if (Cflag)
492 			f_print(fout, "\t%s = (*%s)((char *)&%s, %s);\n",
493 			    RESULT, ROUTINE, ARG, RQSTP);
494 		else
495 			f_print(fout, "\t%s = (*%s)(&%s, %s);\n",
496 			    RESULT, ROUTINE, ARG, RQSTP);
497 		f_print(fout,
498 		    "\tif (%s != NULL && !svc_sendreply(%s, xdr_%s, %s)) {\n",
499 		    RESULT, TRANSP, RESULT, RESULT);
500 		printerr("systemerr", TRANSP);
501 		f_print(fout, "\t}\n");
502 
503 		printif("freeargs", TRANSP, "(caddr_t) &", ARG);
504 		(void) sprintf(_errbuf, "unable to free arguments");
505 		print_err_message("\t\t");
506 		f_print(fout, "\t\texit(1);\n");
507 		f_print(fout, "\t}\n");
508 		print_return("\t");
509 		f_print(fout, "}\n");
510 	}
511 }
512 
513 static void
514 printerr(err, transp)
515 	char   *err;
516 	char   *transp;
517 {
518 	f_print(fout, "\t\tsvcerr_%s(%s);\n", err, transp);
519 }
520 
521 static void
522 printif(proc, transp, prefix, arg)
523 	char   *proc;
524 	char   *transp;
525 	char   *prefix;
526 	char   *arg;
527 {
528 	f_print(fout, "\tif (!svc_%s(%s, xdr_%s, %s%s)) {\n",
529 	    proc, transp, arg, prefix, arg);
530 }
531 
532 int
533 nullproc(proc)
534 	proc_list *proc;
535 {
536 	for (; proc != NULL; proc = proc->next) {
537 		if (streq(proc->proc_num, "0")) {
538 			return (1);
539 		}
540 	}
541 	return (0);
542 }
543 
544 static void
545 write_inetmost(infile)
546 	char   *infile;
547 {
548 	f_print(fout, "\tSVCXPRT *%s = NULL;\n", TRANSP);
549 	f_print(fout, "\tint sock;\n");
550 	f_print(fout, "\tint proto = 0;\n");
551 	f_print(fout, "\tstruct sockaddr_in saddr;\n");
552 	f_print(fout, "\tint asize = sizeof (saddr);\n");
553 	f_print(fout, "\n");
554 	f_print(fout,
555 	    "\tif (getsockname(0, (struct sockaddr *)&saddr, &asize) == 0) {\n");
556 	f_print(fout, "\t\tint ssize = sizeof (int);\n\n");
557 	f_print(fout, "\t\tif (saddr.sin_family != AF_INET)\n");
558 	f_print(fout, "\t\t\texit(1);\n");
559 	f_print(fout, "\t\tif (getsockopt(0, SOL_SOCKET, SO_TYPE,\n");
560 	f_print(fout, "\t\t\t\t(char *)&_rpcfdtype, &ssize) == -1)\n");
561 	f_print(fout, "\t\t\texit(1);\n");
562 	f_print(fout, "\t\tsock = 0;\n");
563 	f_print(fout, "\t\t_rpcpmstart = 1;\n");
564 	f_print(fout, "\t\tproto = 0;\n");
565 	open_log_file(infile, "\t\t");
566 	f_print(fout, "\t} else {\n");
567 	write_rpc_svc_fg(infile, "\t\t");
568 	f_print(fout, "\t\tsock = RPC_ANYSOCK;\n");
569 	print_pmapunset("\t\t");
570 	f_print(fout, "\t}\n");
571 }
572 
573 static void
574 print_return(space)
575 	char   *space;
576 {
577 	if (exitnow)
578 		f_print(fout, "%sexit(0);\n", space);
579 	else {
580 		if (timerflag)
581 			f_print(fout, "%s_rpcsvcdirty = 0;\n", space);
582 		f_print(fout, "%sreturn;\n", space);
583 	}
584 }
585 
586 static void
587 print_pmapunset(space)
588 	char   *space;
589 {
590 	list   *l;
591 	definition *def;
592 	version_list *vp;
593 
594 	for (l = defined; l != NULL; l = l->next) {
595 		def = (definition *) l->val;
596 		if (def->def_kind == DEF_PROGRAM) {
597 			for (vp = def->def.pr.versions; vp != NULL;
598 			    vp = vp->next) {
599 				f_print(fout, "%s(void) pmap_unset(%s, %s);\n",
600 				    space, def->def_name, vp->vers_name);
601 			}
602 		}
603 	}
604 }
605 
606 static void
607 print_err_message(space)
608 	char   *space;
609 {
610 	if (logflag)
611 		f_print(fout, "%ssyslog(LOG_ERR, \"%s\");\n", space, _errbuf);
612 	else
613 		if (inetdflag || pmflag)
614 			f_print(fout, "%s_msgout(\"%s\");\n", space, _errbuf);
615 		else
616 			f_print(fout, "%sfprintf(stderr, \"%s\");\n", space, _errbuf);
617 }
618 /*
619  * Write the server auxiliary function ( _msgout, timeout)
620  */
621 void
622 write_svc_aux(nomain)
623 	int     nomain;
624 {
625 	if (!logflag)
626 		write_msg_out();
627 	if (!nomain)
628 		write_timeout_func();
629 	if (callerflag)		/* EVAS */
630 		write_caller_func();	/* EVAS */
631 }
632 /*
633  * Write the _msgout function
634  */
635 void
636 write_msg_out()
637 {
638 	f_print(fout, "\n");
639 	f_print(fout, "static\n");
640 	if (!Cflag) {
641 		f_print(fout, "void _msgout(msg)\n");
642 		f_print(fout, "\tchar *msg;\n");
643 	} else {
644 		f_print(fout, "void _msgout(char* msg)\n");
645 	}
646 	f_print(fout, "{\n");
647 	f_print(fout, "#ifdef RPC_SVC_FG\n");
648 	if (inetdflag || pmflag)
649 		f_print(fout, "\tif (_rpcpmstart)\n");
650 	f_print(fout, "\t\tsyslog(LOG_ERR, \"%%s\", msg);\n");
651 	f_print(fout, "\telse\n");
652 	f_print(fout, "\t\t(void) fprintf(stderr, \"%%s\\n\", msg);\n");
653 	f_print(fout, "#else\n");
654 	f_print(fout, "\tsyslog(LOG_ERR, \"%%s\", msg);\n");
655 	f_print(fout, "#endif\n");
656 	f_print(fout, "}\n");
657 }
658 /*
659  * Write the timeout function
660  */
661 static void
662 write_timeout_func()
663 {
664 	if (!timerflag)
665 		return;
666 	if (Cflag) {
667 		f_print(fout, "\n");
668 		f_print(fout, "static void closedown(void);\n");
669 	}
670 	f_print(fout, "\n");
671 	f_print(fout, "static void\n");
672 	f_print(fout, "closedown()\n");
673 	f_print(fout, "{\n");
674 	f_print(fout, "\tif (_rpcsvcdirty == 0) {\n");
675 	f_print(fout, "\t\textern fd_set svc_fdset;\n");
676 	f_print(fout, "\t\tstatic int size;\n");
677 	f_print(fout, "\t\tint i, openfd;\n");
678 	if (tirpcflag && pmflag) {
679 		f_print(fout, "\t\tstruct t_info tinfo;\n\n");
680 		f_print(fout, "\t\tif (!t_getinfo(0, &tinfo) && (tinfo.servtype == T_CLTS))\n");
681 	} else {
682 		f_print(fout, "\n\t\tif (_rpcfdtype == SOCK_DGRAM)\n");
683 	}
684 	f_print(fout, "\t\t\texit(0);\n");
685 	f_print(fout, "\t\tif (size == 0) {\n");
686 	if (tirpcflag) {
687 		f_print(fout, "\t\t\tstruct rlimit rl;\n\n");
688 		f_print(fout, "\t\t\trl.rlim_max = 0;\n");
689 		f_print(fout, "\t\t\tgetrlimit(RLIMIT_NOFILE, &rl);\n");
690 		f_print(fout, "\t\t\tif ((size = rl.rlim_max) == 0)\n");
691 		f_print(fout, "\t\t\t\treturn;\n");
692 	} else {
693 		f_print(fout, "\t\t\tsize = getdtablesize();\n");
694 	}
695 	f_print(fout, "\t\t}\n");
696 	f_print(fout, "\t\tfor (i = 0, openfd = 0; i < size && openfd < 2; i++)\n");
697 	f_print(fout, "\t\t\tif (FD_ISSET(i, &svc_fdset))\n");
698 	f_print(fout, "\t\t\t\topenfd++;\n");
699 	f_print(fout, "\t\tif (openfd <= (_rpcpmstart?0:1))\n");
700 	f_print(fout, "\t\t\texit(0);\n");
701 	f_print(fout, "\t}\n");
702 	f_print(fout, "\t(void) alarm(_RPCSVC_CLOSEDOWN);\n");
703 	f_print(fout, "}\n");
704 }
705 
706 static void
707 write_caller_func()
708 {				/* EVAS */
709 #define	P(s)	f_print(fout, s);
710 
711 	P("\n");
712 	P("char *svc_caller()\n");
713 	P("{\n");
714 	P("	struct sockaddr_in actual;\n");
715 	P("	struct hostent *hp;\n");
716 	P("	static struct in_addr prev;\n");
717 	P("	static char cname[128];\n\n");
718 
719 	P("	actual = *svc_getcaller(caller);\n\n");
720 
721 	P("	if (memcmp((char *)&actual.sin_addr, (char *)&prev,\n");
722 	P("		 sizeof(struct in_addr)) == 0)\n");
723 	P("		return (cname);\n\n");
724 
725 	P("	prev = actual.sin_addr;\n\n");
726 
727 	P("	hp = gethostbyaddr((char *) &actual.sin_addr, sizeof(actual.sin_addr), AF_INET);\n");
728 	P("	if (hp == NULL) {                       /* dummy one up */\n");
729 	P("		extern char *inet_ntoa();\n");
730 	P("		strcpy(cname, inet_ntoa(actual.sin_addr));\n");
731 	P("	} else {\n");
732 	P("		strcpy(cname, hp->h_name);\n");
733 	P("	}\n\n");
734 
735 	P("	return (cname);\n");
736 	P("}\n");
737 
738 #undef P
739 }
740 /*
741  * Write the most of port monitor support
742  */
743 static void
744 write_pm_most(infile, netflag)
745 	char   *infile;
746 	int     netflag;
747 {
748 	list   *l;
749 	definition *def;
750 	version_list *vp;
751 
752 	f_print(fout, "\tif (!ioctl(0, I_LOOK, mname) &&\n");
753 	f_print(fout, "\t\t(!strcmp(mname, \"sockmod\") ||");
754 	f_print(fout, " !strcmp(mname, \"timod\"))) {\n");
755 	f_print(fout, "\t\tchar *netid;\n");
756 	if (!netflag) {		/* Not included by -n option */
757 		f_print(fout, "\t\tstruct netconfig *nconf = NULL;\n");
758 		f_print(fout, "\t\tSVCXPRT *%s;\n", TRANSP);
759 	}
760 	if (timerflag)
761 		f_print(fout, "\t\tint pmclose;\n");
762 /* not necessary, defined in /usr/include/stdlib */
763 /*	f_print(fout, "\t\textern char *getenv();\n");*/
764 	f_print(fout, "\n");
765 	f_print(fout, "\t\t_rpcpmstart = 1;\n");
766 	if (logflag)
767 		open_log_file(infile, "\t\t");
768 	f_print(fout, "\t\tif ((netid = getenv(\"NLSPROVIDER\")) == NULL) {\n");
769 	sprintf(_errbuf, "cannot get transport name");
770 	print_err_message("\t\t\t");
771 	f_print(fout, "\t\t} else if ((nconf = getnetconfigent(netid)) == NULL) {\n");
772 	sprintf(_errbuf, "cannot get transport info");
773 	print_err_message("\t\t\t");
774 	f_print(fout, "\t\t}\n");
775 	/*
776 	 * A kludgy support for inetd services. Inetd only works with
777 	 * sockmod, and RPC works only with timod, hence all this jugglery
778 	 */
779 	f_print(fout, "\t\tif (strcmp(mname, \"sockmod\") == 0) {\n");
780 	f_print(fout, "\t\t\tif (ioctl(0, I_POP, 0) || ioctl(0, I_PUSH, \"timod\")) {\n");
781 	sprintf(_errbuf, "could not get the right module");
782 	print_err_message("\t\t\t\t");
783 	f_print(fout, "\t\t\t\texit(1);\n");
784 	f_print(fout, "\t\t\t}\n");
785 	f_print(fout, "\t\t}\n");
786 	if (timerflag)
787 		f_print(fout, "\t\tpmclose = (t_getstate(0) != T_DATAXFER);\n");
788 	f_print(fout, "\t\tif ((%s = svc_tli_create(0, nconf, NULL, 0, 0)) == NULL) {\n",
789 	    TRANSP);
790 	sprintf(_errbuf, "cannot create server handle");
791 	print_err_message("\t\t\t");
792 	f_print(fout, "\t\t\texit(1);\n");
793 	f_print(fout, "\t\t}\n");
794 	f_print(fout, "\t\tif (nconf)\n");
795 	f_print(fout, "\t\t\tfreenetconfigent(nconf);\n");
796 	for (l = defined; l != NULL; l = l->next) {
797 		def = (definition *) l->val;
798 		if (def->def_kind != DEF_PROGRAM) {
799 			continue;
800 		}
801 		for (vp = def->def.pr.versions; vp != NULL; vp = vp->next) {
802 			f_print(fout,
803 			    "\t\tif (!svc_reg(%s, %s, %s, ",
804 			    TRANSP, def->def_name, vp->vers_name);
805 			pvname(def->def_name, vp->vers_num);
806 			f_print(fout, ", 0)) {\n");
807 			(void) sprintf(_errbuf, "unable to register (%s, %s).",
808 			    def->def_name, vp->vers_name);
809 			print_err_message("\t\t\t");
810 			f_print(fout, "\t\t\texit(1);\n");
811 			f_print(fout, "\t\t}\n");
812 		}
813 	}
814 	if (timerflag) {
815 		f_print(fout, "\t\tif (pmclose) {\n");
816 		f_print(fout, "\t\t\t(void) signal(SIGALRM, %s closedown);\n",
817 		    Cflag ? "(SIG_PF)" : "(void(*)())");
818 		f_print(fout, "\t\t\t(void) alarm(_RPCSVC_CLOSEDOWN);\n");
819 		f_print(fout, "\t\t}\n");
820 	}
821 	f_print(fout, "\t\tsvc_run();\n");
822 	f_print(fout, "\t\texit(1);\n");
823 	f_print(fout, "\t\t/* NOTREACHED */\n");
824 	f_print(fout, "\t}\n");
825 }
826 /*
827  * Support for backgrounding the server if self started.
828  */
829 static void
830 write_rpc_svc_fg(infile, sp)
831 	char   *infile;
832 	char   *sp;
833 {
834 	f_print(fout, "#ifndef RPC_SVC_FG\n");
835 	f_print(fout, "%sint size;\n", sp);
836 	if (tirpcflag)
837 		f_print(fout, "%sstruct rlimit rl;\n", sp);
838 	if (inetdflag)
839 		f_print(fout, "%sint pid, i;\n\n", sp);
840 	f_print(fout, "%spid = fork();\n", sp);
841 	f_print(fout, "%sif (pid < 0) {\n", sp);
842 	f_print(fout, "%s\tperror(\"cannot fork\");\n", sp);
843 	f_print(fout, "%s\texit(1);\n", sp);
844 	f_print(fout, "%s}\n", sp);
845 	f_print(fout, "%sif (pid)\n", sp);
846 	f_print(fout, "%s\texit(0);\n", sp);
847 	/* get number of file descriptors */
848 	if (tirpcflag) {
849 		f_print(fout, "%srl.rlim_max = 0;\n", sp);
850 		f_print(fout, "%sgetrlimit(RLIMIT_NOFILE, &rl);\n", sp);
851 		f_print(fout, "%sif ((size = rl.rlim_max) == 0)\n", sp);
852 		f_print(fout, "%s\texit(1);\n", sp);
853 	} else {
854 		f_print(fout, "%ssize = getdtablesize();\n", sp);
855 	}
856 
857 	f_print(fout, "%sfor (i = 0; i < size; i++)\n", sp);
858 	f_print(fout, "%s\t(void) close(i);\n", sp);
859 	/* Redirect stderr and stdout to console */
860 	f_print(fout, "%si = open(\"/dev/console\", 2);\n", sp);
861 	f_print(fout, "%s(void) dup2(i, 1);\n", sp);
862 	f_print(fout, "%s(void) dup2(i, 2);\n", sp);
863 	/* This removes control of the controlling terminal */
864 	if (tirpcflag)
865 		f_print(fout, "%ssetsid();\n", sp);
866 	else {
867 		f_print(fout, "%si = open(\"/dev/tty\", 2);\n", sp);
868 		f_print(fout, "%sif (i >= 0) {\n", sp);
869 		f_print(fout, "%s\t(void) ioctl(i, TIOCNOTTY, (char *)NULL);\n", sp);;
870 		f_print(fout, "%s\t(void) close(i);\n", sp);
871 		f_print(fout, "%s}\n", sp);
872 	}
873 	if (!logflag)
874 		open_log_file(infile, sp);
875 	f_print(fout, "#endif\n");
876 	if (logflag)
877 		open_log_file(infile, sp);
878 }
879 
880 static void
881 open_log_file(infile, sp)
882 	char   *infile;
883 	char   *sp;
884 {
885 	char   *s, *p;
886 
887 	s = strrchr(infile, '.');
888 	if (s)
889 		*s = '\0';
890 	p = strrchr(infile, '/');
891 	if (p)
892 		p++;
893 	else
894 		p = infile;
895 	f_print(fout, "%sopenlog(\"%s\", LOG_PID, LOG_DAEMON);\n", sp, p);
896 	if (s)
897 		*s = '.';
898 }
899 
900 /*
901  * write a registration for the given transport for Inetd
902  */
903 void
904 write_inetd_register(transp)
905 	char   *transp;
906 {
907 	list   *l;
908 	definition *def;
909 	version_list *vp;
910 	char   *sp;
911 	int     isudp;
912 	char    tmpbuf[32];
913 
914 	if (inetdflag)
915 		sp = "\t";
916 	else
917 		sp = "";
918 	if (streq(transp, "udp"))
919 		isudp = 1;
920 	else
921 		isudp = 0;
922 	f_print(fout, "\n");
923 	if (inetdflag) {
924 		f_print(fout, "\tif ((_rpcfdtype == 0) || (_rpcfdtype == %s)) {\n",
925 		    isudp ? "SOCK_DGRAM" : "SOCK_STREAM");
926 	}
927 	if (inetdflag && streq(transp, "tcp")) {
928 		f_print(fout, "%s\tif (_rpcpmstart)\n", sp);
929 
930 		f_print(fout, "%s\t\t%s = svc%s_create(%s",
931 		    sp, TRANSP, "fd", inetdflag ? "sock" : "RPC_ANYSOCK");
932 		if (!isudp)
933 			f_print(fout, ", 0, 0");
934 		f_print(fout, ");\n");
935 
936 		f_print(fout, "%s\telse\n", sp);
937 
938 		f_print(fout, "%s\t\t%s = svc%s_create(%s",
939 		    sp, TRANSP, transp, inetdflag ? "sock" : "RPC_ANYSOCK");
940 		if (!isudp)
941 			f_print(fout, ", 0, 0");
942 		f_print(fout, ");\n");
943 
944 	} else {
945 		f_print(fout, "%s\t%s = svc%s_create(%s",
946 		    sp, TRANSP, transp, inetdflag ? "sock" : "RPC_ANYSOCK");
947 		if (!isudp)
948 			f_print(fout, ", 0, 0");
949 		f_print(fout, ");\n");
950 	}
951 	f_print(fout, "%s\tif (%s == NULL) {\n", sp, TRANSP);
952 	(void) sprintf(_errbuf, "cannot create %s service.", transp);
953 	(void) sprintf(tmpbuf, "%s\t\t", sp);
954 	print_err_message(tmpbuf);
955 	f_print(fout, "%s\t\texit(1);\n", sp);
956 	f_print(fout, "%s\t}\n", sp);
957 
958 	if (inetdflag) {
959 		f_print(fout, "%s\tif (!_rpcpmstart)\n\t", sp);
960 		f_print(fout, "%s\tproto = IPPROTO_%s;\n",
961 		    sp, isudp ? "UDP" : "TCP");
962 	}
963 	for (l = defined; l != NULL; l = l->next) {
964 		def = (definition *) l->val;
965 		if (def->def_kind != DEF_PROGRAM) {
966 			continue;
967 		}
968 		for (vp = def->def.pr.versions; vp != NULL; vp = vp->next) {
969 			f_print(fout, "%s\tif (!svc_register(%s, %s, %s, ",
970 			    sp, TRANSP, def->def_name, vp->vers_name);
971 			pvname(def->def_name, vp->vers_num);
972 			if (inetdflag)
973 				f_print(fout, ", proto)) {\n");
974 			else
975 				f_print(fout, ", IPPROTO_%s)) {\n",
976 				    isudp ? "UDP" : "TCP");
977 			(void) sprintf(_errbuf, "unable to register (%s, %s, %s).",
978 			    def->def_name, vp->vers_name, transp);
979 			print_err_message(tmpbuf);
980 			f_print(fout, "%s\t\texit(1);\n", sp);
981 			f_print(fout, "%s\t}\n", sp);
982 		}
983 	}
984 	if (inetdflag)
985 		f_print(fout, "\t}\n");
986 }
987