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