1/*
2 * Sun RPC is a product of Sun Microsystems, Inc. and is provided for
3 * unrestricted use provided that this legend is included on all tape
4 * media and as a part of the software program in whole or part.  Users
5 * may copy or modify Sun RPC without charge, but are not authorized
6 * to license or distribute it to anyone else except as part of a product or
7 * program developed by the user.
8 *
9 * SUN RPC IS PROVIDED AS IS WITH NO WARRANTIES OF ANY KIND INCLUDING THE
10 * WARRANTIES OF DESIGN, MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
11 * PURPOSE, OR ARISING FROM A COURSE OF DEALING, USAGE OR TRADE PRACTICE.
12 *
13 * Sun RPC is provided with no support and without any obligation on the
14 * part of Sun Microsystems, Inc. to assist in its use, correction,
15 * modification or enhancement.
16 *
17 * SUN MICROSYSTEMS, INC. SHALL HAVE NO LIABILITY WITH RESPECT TO THE
18 * INFRINGEMENT OF COPYRIGHTS, TRADE SECRETS OR ANY PATENTS BY SUN RPC
19 * OR ANY PART THEREOF.
20 *
21 * In no event will Sun Microsystems, Inc. be liable for any lost revenue
22 * or profits or other special, indirect and consequential damages, even if
23 * Sun has been advised of the possibility of such damages.
24 *
25 * Sun Microsystems, Inc.
26 * 2550 Garcia Avenue
27 * Mountain View, California  94043
28 */
29
30#if 0
31#ifndef lint
32#ident	"@(#)rpc_cout.c	1.14	93/07/05 SMI"
33static char sccsid[] = "@(#)rpc_cout.c 1.13 89/02/22 (C) 1987 SMI";
34#endif
35#endif
36
37#include <sys/cdefs.h>
38__FBSDID("$FreeBSD: stable/11/usr.bin/rpcgen/rpc_cout.c 312385 2017-01-18 15:22:54Z pfg $");
39
40/*
41 * rpc_cout.c, XDR routine outputter for the RPC protocol compiler
42 * Copyright (C) 1987, Sun Microsystems, Inc.
43 */
44#include <ctype.h>
45#include <stdio.h>
46#include <string.h>
47#include "rpc_parse.h"
48#include "rpc_scan.h"
49#include "rpc_util.h"
50
51static void print_header( definition * );
52static void print_trailer( void );
53static void print_stat( int , declaration * );
54static void emit_enum( definition * );
55static void emit_program( definition * );
56static void emit_union( definition * );
57static void emit_struct( definition * );
58static void emit_typedef( definition * );
59static void emit_inline( int, declaration *, int );
60static void emit_single_in_line( int, declaration *, int, relation );
61
62/*
63 * Emit the C-routine for the given definition
64 */
65void
66emit(definition *def)
67{
68	if (def->def_kind == DEF_CONST) {
69		return;
70	}
71	if (def->def_kind == DEF_PROGRAM) {
72		emit_program(def);
73		return;
74	}
75	if (def->def_kind == DEF_TYPEDEF) {
76		/*
77		 * now we need to handle declarations like
78		 * struct typedef foo foo;
79		 * since we dont want this to be expanded into 2 calls to xdr_foo
80		 */
81
82		if (strcmp(def->def.ty.old_type, def->def_name) == 0)
83			return;
84	}
85	print_header(def);
86	switch (def->def_kind) {
87	case DEF_UNION:
88		emit_union(def);
89		break;
90	case DEF_ENUM:
91		emit_enum(def);
92		break;
93	case DEF_STRUCT:
94		emit_struct(def);
95		break;
96	case DEF_TYPEDEF:
97		emit_typedef(def);
98		break;
99		/* DEF_CONST and DEF_PROGRAM have already been handled */
100	default:
101		break;
102	}
103	print_trailer();
104}
105
106static int
107findtype(definition *def, const char *type)
108{
109
110	if (def->def_kind == DEF_PROGRAM || def->def_kind == DEF_CONST) {
111		return (0);
112	} else {
113		return (streq(def->def_name, type));
114	}
115}
116
117static int
118undefined(const char *type)
119{
120	definition *def;
121
122	def = (definition *) FINDVAL(defined, type, findtype);
123	return (def == NULL);
124}
125
126
127static void
128print_generic_header(const char *procname, int pointerp)
129{
130	f_print(fout, "\n");
131	f_print(fout, "bool_t\n");
132	f_print(fout, "xdr_%s(", procname);
133	f_print(fout, "XDR *xdrs, ");
134	f_print(fout, "%s ", procname);
135	if (pointerp)
136		f_print(fout, "*");
137	f_print(fout, "objp)\n{\n\n");
138}
139
140static void
141print_header(definition *def)
142{
143	print_generic_header(def->def_name,
144			    def->def_kind != DEF_TYPEDEF ||
145			    !isvectordef(def->def.ty.old_type,
146					 def->def.ty.rel));
147	/* Now add Inline support */
148
149	if (inline_size == 0)
150		return;
151	/* May cause lint to complain. but  ... */
152	f_print(fout, "\tregister long *buf;\n\n");
153}
154
155static void
156print_prog_header(proc_list *plist)
157{
158	print_generic_header(plist->args.argname, 1);
159}
160
161static void
162print_trailer(void)
163{
164	f_print(fout, "\treturn (TRUE);\n");
165	f_print(fout, "}\n");
166}
167
168
169static void
170print_ifopen(int indent, const char *name)
171{
172	tabify(fout, indent);
173	f_print(fout, "if (!xdr_%s(xdrs", name);
174}
175
176static void
177print_ifarg(const char *arg)
178{
179	f_print(fout, ", %s", arg);
180}
181
182static void
183print_ifsizeof(int indent, const char *prefix, const char *type)
184{
185	if (indent) {
186		f_print(fout, ",\n");
187		tabify(fout, indent);
188	} else  {
189		f_print(fout, ", ");
190	}
191	if (streq(type, "bool")) {
192		f_print(fout, "sizeof (bool_t), (xdrproc_t) xdr_bool");
193	} else {
194		f_print(fout, "sizeof (");
195		if (undefined(type) && prefix) {
196			f_print(fout, "%s ", prefix);
197		}
198		f_print(fout, "%s), (xdrproc_t) xdr_%s", type, type);
199	}
200}
201
202static void
203print_ifclose(int indent, int brace)
204{
205	f_print(fout, "))\n");
206	tabify(fout, indent);
207	f_print(fout, "\treturn (FALSE);\n");
208	if (brace)
209		f_print(fout, "\t}\n");
210}
211
212static void
213print_ifstat(int indent, const char *prefix, const char *type, relation rel,
214    const char *amax, const char *objname, const char *name)
215{
216	const char *alt = NULL;
217	int brace = 0;
218
219	switch (rel) {
220	case REL_POINTER:
221		brace = 1;
222		f_print(fout, "\t{\n");
223		f_print(fout, "\t%s **pp = %s;\n", type, objname);
224		print_ifopen(indent, "pointer");
225		print_ifarg("(char **)");
226		f_print(fout, "pp");
227		print_ifsizeof(0, prefix, type);
228		break;
229	case REL_VECTOR:
230		if (streq(type, "string")) {
231			alt = "string";
232		} else if (streq(type, "opaque")) {
233			alt = "opaque";
234		}
235		if (alt) {
236			print_ifopen(indent, alt);
237			print_ifarg(objname);
238		} else {
239			print_ifopen(indent, "vector");
240			print_ifarg("(char *)");
241			f_print(fout, "%s", objname);
242		}
243		print_ifarg(amax);
244		if (!alt) {
245			print_ifsizeof(indent + 1, prefix, type);
246		}
247		break;
248	case REL_ARRAY:
249		if (streq(type, "string")) {
250			alt = "string";
251		} else if (streq(type, "opaque")) {
252			alt = "bytes";
253		}
254		if (streq(type, "string")) {
255			print_ifopen(indent, alt);
256			print_ifarg(objname);
257		} else {
258			if (alt) {
259				print_ifopen(indent, alt);
260			} else {
261				print_ifopen(indent, "array");
262			}
263			print_ifarg("(char **)");
264			if (*objname == '&') {
265				f_print(fout, "%s.%s_val, (u_int *) %s.%s_len",
266					objname, name, objname, name);
267			} else {
268				f_print(fout,
269					"&%s->%s_val, (u_int *) &%s->%s_len",
270					objname, name, objname, name);
271			}
272		}
273		print_ifarg(amax);
274		if (!alt) {
275			print_ifsizeof(indent + 1, prefix, type);
276		}
277		break;
278	case REL_ALIAS:
279		print_ifopen(indent, type);
280		print_ifarg(objname);
281		break;
282	}
283	print_ifclose(indent, brace);
284}
285
286/* ARGSUSED */
287static void
288emit_enum(definition *def __unused)
289{
290	print_ifopen(1, "enum");
291	print_ifarg("(enum_t *)objp");
292	print_ifclose(1, 0);
293}
294
295static void
296emit_program(definition *def)
297{
298	decl_list *dl;
299	version_list *vlist;
300	proc_list *plist;
301
302	for (vlist = def->def.pr.versions; vlist != NULL; vlist = vlist->next)
303		for (plist = vlist->procs; plist != NULL; plist = plist->next) {
304			if (!newstyle || plist->arg_num < 2)
305				continue; /* old style, or single argument */
306			print_prog_header(plist);
307			for (dl = plist->args.decls; dl != NULL;
308			     dl = dl->next)
309				print_stat(1, &dl->decl);
310			print_trailer();
311		}
312}
313
314
315static void
316emit_union(definition *def)
317{
318	declaration *dflt;
319	case_list *cl;
320	declaration *cs;
321	char *object;
322	const char *vecformat = "objp->%s_u.%s";
323	const char *format = "&objp->%s_u.%s";
324
325	print_stat(1, &def->def.un.enum_decl);
326	f_print(fout, "\tswitch (objp->%s) {\n", def->def.un.enum_decl.name);
327	for (cl = def->def.un.cases; cl != NULL; cl = cl->next) {
328
329		f_print(fout, "\tcase %s:\n", cl->case_name);
330		if (cl->contflag == 1) /* a continued case statement */
331			continue;
332		cs = &cl->case_decl;
333		if (!streq(cs->type, "void")) {
334			object = xmalloc(strlen(def->def_name) +
335			                 strlen(format) + strlen(cs->name) + 1);
336			if (isvectordef (cs->type, cs->rel)) {
337				s_print(object, vecformat, def->def_name,
338					cs->name);
339			} else {
340				s_print(object, format, def->def_name,
341					cs->name);
342			}
343			print_ifstat(2, cs->prefix, cs->type, cs->rel,
344				     cs->array_max, object, cs->name);
345			free(object);
346		}
347		f_print(fout, "\t\tbreak;\n");
348	}
349	dflt = def->def.un.default_decl;
350	if (dflt != NULL) {
351		if (!streq(dflt->type, "void")) {
352			f_print(fout, "\tdefault:\n");
353			object = xmalloc(strlen(def->def_name) +
354			                 strlen(format) + strlen(dflt->name) + 1);
355			if (isvectordef (dflt->type, dflt->rel)) {
356				s_print(object, vecformat, def->def_name,
357					dflt->name);
358			} else {
359				s_print(object, format, def->def_name,
360					dflt->name);
361			}
362
363			print_ifstat(2, dflt->prefix, dflt->type, dflt->rel,
364				    dflt->array_max, object, dflt->name);
365			free(object);
366			f_print(fout, "\t\tbreak;\n");
367		} else {
368			f_print(fout, "\tdefault:\n");
369			f_print(fout, "\t\tbreak;\n");
370		}
371	} else {
372		f_print(fout, "\tdefault:\n");
373		f_print(fout, "\t\treturn (FALSE);\n");
374	}
375
376	f_print(fout, "\t}\n");
377}
378
379static void
380inline_struct(definition *def, int flag)
381{
382	decl_list *dl;
383	int i, size;
384	decl_list *cur, *psav;
385	bas_type *ptr;
386	char *sizestr;
387	const char *plus;
388	char ptemp[256];
389	int indent = 1;
390
391	cur = NULL;
392	if (flag == PUT)
393		f_print(fout, "\n\tif (xdrs->x_op == XDR_ENCODE) {\n");
394	else
395		f_print(fout, "\t\treturn (TRUE);\n\t} else if (xdrs->x_op == XDR_DECODE) {\n");
396
397	i = 0;
398	size = 0;
399	sizestr = NULL;
400	for (dl = def->def.st.decls; dl != NULL; dl = dl->next) { /* xxx */
401		/* now walk down the list and check for basic types */
402		if ((dl->decl.prefix == NULL) &&
403		    ((ptr = find_type(dl->decl.type)) != NULL) &&
404		    ((dl->decl.rel == REL_ALIAS) ||
405		     (dl->decl.rel == REL_VECTOR))){
406			if (i == 0)
407				cur = dl;
408			i++;
409
410			if (dl->decl.rel == REL_ALIAS)
411				size += ptr->length;
412			else {
413				/* this code is required to handle arrays */
414				if (sizestr == NULL)
415					plus = "";
416				else
417					plus = " + ";
418
419				if (ptr->length != 1)
420					s_print(ptemp, "%s%s * %d",
421						plus, dl->decl.array_max,
422						ptr->length);
423				else
424					s_print(ptemp, "%s%s", plus,
425						dl->decl.array_max);
426
427				/* now concatenate to sizestr !!!! */
428				if (sizestr == NULL) {
429					sizestr = xstrdup(ptemp);
430				}
431				else{
432					sizestr = xrealloc(sizestr,
433							  strlen(sizestr)
434							  +strlen(ptemp)+1);
435					sizestr = strcat(sizestr, ptemp);
436					/* build up length of array */
437				}
438			}
439		} else {
440			if (i > 0) {
441				if (sizestr == NULL && size < inline_size){
442					/*
443					 * don't expand into inline code
444					 * if size < inline_size
445					 */
446					while (cur != dl){
447						print_stat(indent + 1, &cur->decl);
448						cur = cur->next;
449					}
450				} else {
451					/* were already looking at a xdr_inlineable structure */
452					tabify(fout, indent + 1);
453					if (sizestr == NULL)
454						f_print(fout, "buf = XDR_INLINE(xdrs, %d * BYTES_PER_XDR_UNIT);",
455							size);
456					else {
457						if (size == 0)
458							f_print(fout,
459								"buf = XDR_INLINE(xdrs, (%s) * BYTES_PER_XDR_UNIT);",
460								sizestr);
461						else
462							f_print(fout,
463								"buf = XDR_INLINE(xdrs, (%d + (%s)) * BYTES_PER_XDR_UNIT);",
464								size, sizestr);
465
466					}
467					f_print(fout, "\n");
468					tabify(fout, indent + 1);
469					f_print(fout,
470						"if (buf == NULL) {\n");
471
472					psav = cur;
473					while (cur != dl){
474						print_stat(indent + 2, &cur->decl);
475						cur = cur->next;
476					}
477
478					f_print(fout, "\n\t\t} else {\n");
479
480					cur = psav;
481					while (cur != dl){
482						emit_inline(indent + 2, &cur->decl, flag);
483						cur = cur->next;
484					}
485
486					tabify(fout, indent + 1);
487					f_print(fout, "}\n");
488				}
489			}
490			size = 0;
491			i = 0;
492			free(sizestr);
493			sizestr = NULL;
494			print_stat(indent + 1, &dl->decl);
495		}
496	}
497
498	if (i > 0) {
499		if (sizestr == NULL && size < inline_size){
500			/* don't expand into inline code if size < inline_size */
501			while (cur != dl){
502				print_stat(indent + 1, &cur->decl);
503				cur = cur->next;
504			}
505		} else {
506			/* were already looking at a xdr_inlineable structure */
507			if (sizestr == NULL)
508				f_print(fout, "\t\tbuf = XDR_INLINE(xdrs, %d * BYTES_PER_XDR_UNIT);",
509					size);
510			else
511				if (size == 0)
512					f_print(fout,
513						"\t\tbuf = XDR_INLINE(xdrs, (%s) * BYTES_PER_XDR_UNIT);",
514						sizestr);
515				else
516					f_print(fout,
517						"\t\tbuf = XDR_INLINE(xdrs, (%d + (%s)) * BYTES_PER_XDR_UNIT);",
518						size, sizestr);
519
520			f_print(fout, "\n\t\tif (buf == NULL) {\n");
521			psav = cur;
522			while (cur != NULL){
523				print_stat(indent + 2, &cur->decl);
524				cur = cur->next;
525			}
526			f_print(fout, "\t\t} else {\n");
527
528			cur = psav;
529			while (cur != dl){
530				emit_inline(indent + 2, &cur->decl, flag);
531				cur = cur->next;
532			}
533			f_print(fout, "\t\t}\n");
534		}
535	}
536}
537
538static void
539emit_struct(definition *def)
540{
541	decl_list *dl;
542	int j, size, flag;
543	bas_type *ptr;
544	int can_inline;
545
546	if (inline_size == 0) {
547		/* No xdr_inlining at all */
548		for (dl = def->def.st.decls; dl != NULL; dl = dl->next)
549			print_stat(1, &dl->decl);
550		return;
551	}
552
553	for (dl = def->def.st.decls; dl != NULL; dl = dl->next)
554		if (dl->decl.rel == REL_VECTOR &&
555		    strcmp(dl->decl.type, "opaque") != 0){
556			f_print(fout, "\tint i;\n");
557			break;
558		}
559
560	size = 0;
561	can_inline = 0;
562	/*
563	 * Make a first pass and see if inling is possible.
564	 */
565	for (dl = def->def.st.decls; dl != NULL; dl = dl->next)
566		if ((dl->decl.prefix == NULL) &&
567		    ((ptr = find_type(dl->decl.type)) != NULL) &&
568		    ((dl->decl.rel == REL_ALIAS)||
569		     (dl->decl.rel == REL_VECTOR))){
570			if (dl->decl.rel == REL_ALIAS)
571				size += ptr->length;
572			else {
573				can_inline = 1;
574				break; /* can be inlined */
575			}
576		} else {
577			if (size >= inline_size){
578				can_inline = 1;
579				break; /* can be inlined */
580			}
581			size = 0;
582		}
583	if (size >= inline_size)
584		can_inline = 1;
585
586	if (can_inline == 0){	/* can not inline, drop back to old mode */
587		for (dl = def->def.st.decls; dl != NULL; dl = dl->next)
588			print_stat(1, &dl->decl);
589		return;
590	}
591
592	flag = PUT;
593	for (j = 0; j < 2; j++){
594		inline_struct(def, flag);
595		if (flag == PUT)
596			flag = GET;
597	}
598
599	f_print(fout, "\t\treturn (TRUE);\n\t}\n\n");
600
601	/* now take care of XDR_FREE case */
602
603	for (dl = def->def.st.decls; dl != NULL; dl = dl->next)
604		print_stat(1, &dl->decl);
605
606}
607
608static void
609emit_typedef(definition *def)
610{
611	const char *prefix = def->def.ty.old_prefix;
612	const char *type = def->def.ty.old_type;
613	const char *amax = def->def.ty.array_max;
614	relation rel = def->def.ty.rel;
615
616	print_ifstat(1, prefix, type, rel, amax, "objp", def->def_name);
617}
618
619static void
620print_stat(int indent, declaration *dec)
621{
622	const char *prefix = dec->prefix;
623	const char *type = dec->type;
624	const char *amax = dec->array_max;
625	relation rel = dec->rel;
626	char name[256];
627
628	if (isvectordef(type, rel)) {
629		s_print(name, "objp->%s", dec->name);
630	} else {
631		s_print(name, "&objp->%s", dec->name);
632	}
633	print_ifstat(indent, prefix, type, rel, amax, name, dec->name);
634}
635
636
637char *upcase(const char *);
638
639static void
640emit_inline(int indent, declaration *decl, int flag)
641{
642	switch (decl->rel) {
643	case  REL_ALIAS :
644		emit_single_in_line(indent, decl, flag, REL_ALIAS);
645		break;
646	case REL_VECTOR :
647		tabify(fout, indent);
648		f_print(fout, "{\n");
649		tabify(fout, indent + 1);
650		f_print(fout, "%s *genp;\n\n", decl->type);
651		tabify(fout, indent + 1);
652		f_print(fout,
653			"for (i = 0, genp = objp->%s;\n", decl->name);
654		tabify(fout, indent + 2);
655		f_print(fout, "i < %s; i++) {\n", decl->array_max);
656		emit_single_in_line(indent + 2, decl, flag, REL_VECTOR);
657		tabify(fout, indent + 1);
658		f_print(fout, "}\n");
659		tabify(fout, indent);
660		f_print(fout, "}\n");
661		break;
662	default:
663		break;
664	}
665}
666
667static void
668emit_single_in_line(int indent, declaration *decl, int flag, relation rel)
669{
670	char *upp_case;
671
672	tabify(fout, indent);
673	if (flag == PUT)
674		f_print(fout, "IXDR_PUT_");
675	else
676		if (rel == REL_ALIAS)
677			f_print(fout, "objp->%s = IXDR_GET_", decl->name);
678		else
679			f_print(fout, "*genp++ = IXDR_GET_");
680
681	upp_case = upcase(decl->type);
682
683	/* hack	 - XX */
684	if (strcmp(upp_case, "INT") == 0)
685	{
686		free(upp_case);
687		upp_case = strdup("LONG");
688	}
689
690	if (strcmp(upp_case, "U_INT") == 0)
691	{
692		free(upp_case);
693		upp_case = strdup("U_LONG");
694	}
695	if (flag == PUT)
696		if (rel == REL_ALIAS)
697			f_print(fout,
698				"%s(buf, objp->%s);\n", upp_case, decl->name);
699		else
700			f_print(fout, "%s(buf, *genp++);\n", upp_case);
701
702	else
703		f_print(fout, "%s(buf);\n", upp_case);
704	free(upp_case);
705}
706
707char *
708upcase(const char *str)
709{
710	char *ptr, *hptr;
711
712	ptr = (char *)xmalloc(strlen(str)+1);
713
714	hptr = ptr;
715	while (*str != '\0')
716		*ptr++ = toupper(*str++);
717
718	*ptr = '\0';
719	return (hptr);
720}
721