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 } 719