1 /*
   2  * CDDL HEADER START
   3  *
   4  * The contents of this file are subject to the terms of the
   5  * Common Development and Distribution License, Version 1.0 only
   6  * (the "License").  You may not use this file except in compliance
   7  * with the License.
   8  *
   9  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
  10  * or http://www.opensolaris.org/os/licensing.
  11  * See the License for the specific language governing permissions
  12  * and limitations under the License.
  13  *
  14  * When distributing Covered Code, include this CDDL HEADER in each
  15  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
  16  * If applicable, add the following below this CDDL HEADER, with the
  17  * fields enclosed by brackets "[]" replaced with your own identifying
  18  * information: Portions Copyright [yyyy] [name of copyright owner]
  19  *
  20  * CDDL HEADER END
  21  */
  22 
  23 /*
  24  * Copyright 2005 Sun Microsystems, Inc.  All rights reserved.
  25  * Use is subject to license terms.
  26  */
  27 
  28 /*
  29  * Copyright (c) 2012 by Delphix. All rights reserved.
  30  */
  31 
  32 #include <sys/types.h>
  33 #include <sys/sysmacros.h>
  34 #include <sys/isa_defs.h>
  35 
  36 #include <strings.h>
  37 #include <stdlib.h>
  38 #include <setjmp.h>
  39 #include <assert.h>
  40 #include <errno.h>
  41 
  42 #include <dt_impl.h>
  43 #include <dt_grammar.h>
  44 #include <dt_parser.h>
  45 #include <dt_provider.h>
  46 
  47 static void dt_cg_node(dt_node_t *, dt_irlist_t *, dt_regset_t *);
  48 
  49 static dt_irnode_t *
  50 dt_cg_node_alloc(uint_t label, dif_instr_t instr)
  51 {
  52         dt_irnode_t *dip = malloc(sizeof (dt_irnode_t));
  53 
  54         if (dip == NULL)
  55                 longjmp(yypcb->pcb_jmpbuf, EDT_NOMEM);
  56 
  57         dip->di_label = label;
  58         dip->di_instr = instr;
  59         dip->di_extern = NULL;
  60         dip->di_next = NULL;
  61 
  62         return (dip);
  63 }
  64 
  65 /*
  66  * Code generator wrapper function for ctf_member_info.  If we are given a
  67  * reference to a forward declaration tag, search the entire type space for
  68  * the actual definition and then call ctf_member_info on the result.
  69  */
  70 static ctf_file_t *
  71 dt_cg_membinfo(ctf_file_t *fp, ctf_id_t type, const char *s, ctf_membinfo_t *mp)
  72 {
  73         while (ctf_type_kind(fp, type) == CTF_K_FORWARD) {
  74                 char n[DT_TYPE_NAMELEN];
  75                 dtrace_typeinfo_t dtt;
  76 
  77                 if (ctf_type_name(fp, type, n, sizeof (n)) == NULL ||
  78                     dt_type_lookup(n, &dtt) == -1 || (
  79                     dtt.dtt_ctfp == fp && dtt.dtt_type == type))
  80                         break; /* unable to improve our position */
  81 
  82                 fp = dtt.dtt_ctfp;
  83                 type = ctf_type_resolve(fp, dtt.dtt_type);
  84         }
  85 
  86         if (ctf_member_info(fp, type, s, mp) == CTF_ERR)
  87                 return (NULL); /* ctf_errno is set for us */
  88 
  89         return (fp);
  90 }
  91 
  92 static void
  93 dt_cg_xsetx(dt_irlist_t *dlp, dt_ident_t *idp, uint_t lbl, int reg, uint64_t x)
  94 {
  95         int flag = idp != NULL ? DT_INT_PRIVATE : DT_INT_SHARED;
  96         int intoff = dt_inttab_insert(yypcb->pcb_inttab, x, flag);
  97         dif_instr_t instr = DIF_INSTR_SETX((uint_t)intoff, reg);
  98 
  99         if (intoff == -1)
 100                 longjmp(yypcb->pcb_jmpbuf, EDT_NOMEM);
 101 
 102         if (intoff > DIF_INTOFF_MAX)
 103                 longjmp(yypcb->pcb_jmpbuf, EDT_INT2BIG);
 104 
 105         dt_irlist_append(dlp, dt_cg_node_alloc(lbl, instr));
 106 
 107         if (idp != NULL)
 108                 dlp->dl_last->di_extern = idp;
 109 }
 110 
 111 static void
 112 dt_cg_setx(dt_irlist_t *dlp, int reg, uint64_t x)
 113 {
 114         dt_cg_xsetx(dlp, NULL, DT_LBL_NONE, reg, x);
 115 }
 116 
 117 /*
 118  * When loading bit-fields, we want to convert a byte count in the range
 119  * 1-8 to the closest power of 2 (e.g. 3->4, 5->8, etc).  The clp2() function
 120  * is a clever implementation from "Hacker's Delight" by Henry Warren, Jr.
 121  */
 122 static size_t
 123 clp2(size_t x)
 124 {
 125         x--;
 126 
 127         x |= (x >> 1);
 128         x |= (x >> 2);
 129         x |= (x >> 4);
 130         x |= (x >> 8);
 131         x |= (x >> 16);
 132 
 133         return (x + 1);
 134 }
 135 
 136 /*
 137  * Lookup the correct load opcode to use for the specified node and CTF type.
 138  * We determine the size and convert it to a 3-bit index.  Our lookup table
 139  * is constructed to use a 5-bit index, consisting of the 3-bit size 0-7, a
 140  * bit for the sign, and a bit for userland address.  For example, a 4-byte
 141  * signed load from userland would be at the following table index:
 142  * user=1 sign=1 size=4 => binary index 11011 = decimal index 27
 143  */
 144 static uint_t
 145 dt_cg_load(dt_node_t *dnp, ctf_file_t *ctfp, ctf_id_t type)
 146 {
 147         static const uint_t ops[] = {
 148                 DIF_OP_LDUB,    DIF_OP_LDUH,    0,      DIF_OP_LDUW,
 149                 0,              0,              0,      DIF_OP_LDX,
 150                 DIF_OP_LDSB,    DIF_OP_LDSH,    0,      DIF_OP_LDSW,
 151                 0,              0,              0,      DIF_OP_LDX,
 152                 DIF_OP_ULDUB,   DIF_OP_ULDUH,   0,      DIF_OP_ULDUW,
 153                 0,              0,              0,      DIF_OP_ULDX,
 154                 DIF_OP_ULDSB,   DIF_OP_ULDSH,   0,      DIF_OP_ULDSW,
 155                 0,              0,              0,      DIF_OP_ULDX,
 156         };
 157 
 158         ctf_encoding_t e;
 159         ssize_t size;
 160 
 161         /*
 162          * If we're loading a bit-field, the size of our load is found by
 163          * rounding cte_bits up to a byte boundary and then finding the
 164          * nearest power of two to this value (see clp2(), above).
 165          */
 166         if ((dnp->dn_flags & DT_NF_BITFIELD) &&
 167             ctf_type_encoding(ctfp, type, &e) != CTF_ERR)
 168                 size = clp2(P2ROUNDUP(e.cte_bits, NBBY) / NBBY);
 169         else
 170                 size = ctf_type_size(ctfp, type);
 171 
 172         if (size < 1 || size > 8 || (size & (size - 1)) != 0) {
 173                 xyerror(D_UNKNOWN, "internal error -- cg cannot load "
 174                     "size %ld when passed by value\n", (long)size);
 175         }
 176 
 177         size--; /* convert size to 3-bit index */
 178 
 179         if (dnp->dn_flags & DT_NF_SIGNED)
 180                 size |= 0x08;
 181         if (dnp->dn_flags & DT_NF_USERLAND)
 182                 size |= 0x10;
 183 
 184         return (ops[size]);
 185 }
 186 
 187 static void
 188 dt_cg_ptrsize(dt_node_t *dnp, dt_irlist_t *dlp, dt_regset_t *drp,
 189     uint_t op, int dreg)
 190 {
 191         ctf_file_t *ctfp = dnp->dn_ctfp;
 192         ctf_arinfo_t r;
 193         dif_instr_t instr;
 194         ctf_id_t type;
 195         uint_t kind;
 196         ssize_t size;
 197         int sreg;
 198 
 199         type = ctf_type_resolve(ctfp, dnp->dn_type);
 200         kind = ctf_type_kind(ctfp, type);
 201         assert(kind == CTF_K_POINTER || kind == CTF_K_ARRAY);
 202 
 203         if (kind == CTF_K_ARRAY) {
 204                 if (ctf_array_info(ctfp, type, &r) != 0) {
 205                         yypcb->pcb_hdl->dt_ctferr = ctf_errno(ctfp);
 206                         longjmp(yypcb->pcb_jmpbuf, EDT_CTF);
 207                 }
 208                 type = r.ctr_contents;
 209         } else
 210                 type = ctf_type_reference(ctfp, type);
 211 
 212         if ((size = ctf_type_size(ctfp, type)) == 1)
 213                 return; /* multiply or divide by one can be omitted */
 214 
 215         sreg = dt_regset_alloc(drp);
 216         dt_cg_setx(dlp, sreg, size);
 217         instr = DIF_INSTR_FMT(op, dreg, sreg, dreg);
 218         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 219         dt_regset_free(drp, sreg);
 220 }
 221 
 222 /*
 223  * If the result of a "." or "->" operation is a bit-field, we use this routine
 224  * to generate an epilogue to the load instruction that extracts the value.  In
 225  * the diagrams below the "ld??" is the load instruction that is generated to
 226  * load the containing word that is generating prior to calling this function.
 227  *
 228  * Epilogue for unsigned fields:        Epilogue for signed fields:
 229  *
 230  * ldu? [r1], r1                        lds? [r1], r1
 231  * setx USHIFT, r2                      setx 64 - SSHIFT, r2
 232  * srl  r1, r2, r1                      sll  r1, r2, r1
 233  * setx (1 << bits) - 1, r2               setx 64 - bits, r2
 234  * and  r1, r2, r1                      sra  r1, r2, r1
 235  *
 236  * The *SHIFT constants above changes value depending on the endian-ness of our
 237  * target architecture.  Refer to the comments below for more details.
 238  */
 239 static void
 240 dt_cg_field_get(dt_node_t *dnp, dt_irlist_t *dlp, dt_regset_t *drp,
 241     ctf_file_t *fp, const ctf_membinfo_t *mp)
 242 {
 243         ctf_encoding_t e;
 244         dif_instr_t instr;
 245         uint64_t shift;
 246         int r1, r2;
 247 
 248         if (ctf_type_encoding(fp, mp->ctm_type, &e) != 0 || e.cte_bits > 64) {
 249                 xyerror(D_UNKNOWN, "cg: bad field: off %lu type <%ld> "
 250                     "bits %u\n", mp->ctm_offset, mp->ctm_type, e.cte_bits);
 251         }
 252 
 253         assert(dnp->dn_op == DT_TOK_PTR || dnp->dn_op == DT_TOK_DOT);
 254         r1 = dnp->dn_left->dn_reg;
 255         r2 = dt_regset_alloc(drp);
 256 
 257         /*
 258          * On little-endian architectures, ctm_offset counts from the right so
 259          * ctm_offset % NBBY itself is the amount we want to shift right to
 260          * move the value bits to the little end of the register to mask them.
 261          * On big-endian architectures, ctm_offset counts from the left so we
 262          * must subtract (ctm_offset % NBBY + cte_bits) from the size in bits
 263          * we used for the load.  The size of our load in turn is found by
 264          * rounding cte_bits up to a byte boundary and then finding the
 265          * nearest power of two to this value (see clp2(), above).  These
 266          * properties are used to compute shift as USHIFT or SSHIFT, below.
 267          */
 268         if (dnp->dn_flags & DT_NF_SIGNED) {
 269 #ifdef _BIG_ENDIAN
 270                 shift = clp2(P2ROUNDUP(e.cte_bits, NBBY) / NBBY) * NBBY -
 271                     mp->ctm_offset % NBBY;
 272 #else
 273                 shift = mp->ctm_offset % NBBY + e.cte_bits;
 274 #endif
 275                 dt_cg_setx(dlp, r2, 64 - shift);
 276                 instr = DIF_INSTR_FMT(DIF_OP_SLL, r1, r2, r1);
 277                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 278 
 279                 dt_cg_setx(dlp, r2, 64 - e.cte_bits);
 280                 instr = DIF_INSTR_FMT(DIF_OP_SRA, r1, r2, r1);
 281                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 282         } else {
 283 #ifdef _BIG_ENDIAN
 284                 shift = clp2(P2ROUNDUP(e.cte_bits, NBBY) / NBBY) * NBBY -
 285                     (mp->ctm_offset % NBBY + e.cte_bits);
 286 #else
 287                 shift = mp->ctm_offset % NBBY;
 288 #endif
 289                 dt_cg_setx(dlp, r2, shift);
 290                 instr = DIF_INSTR_FMT(DIF_OP_SRL, r1, r2, r1);
 291                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 292 
 293                 dt_cg_setx(dlp, r2, (1ULL << e.cte_bits) - 1);
 294                 instr = DIF_INSTR_FMT(DIF_OP_AND, r1, r2, r1);
 295                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 296         }
 297 
 298         dt_regset_free(drp, r2);
 299 }
 300 
 301 /*
 302  * If the destination of a store operation is a bit-field, we use this routine
 303  * to generate a prologue to the store instruction that loads the surrounding
 304  * bits, clears the destination field, and ORs in the new value of the field.
 305  * In the diagram below the "st?" is the store instruction that is generated to
 306  * store the containing word that is generating after calling this function.
 307  *
 308  * ld   [dst->dn_reg], r1
 309  * setx ~(((1 << cte_bits) - 1) << (ctm_offset % NBBY)), r2
 310  * and  r1, r2, r1
 311  *
 312  * setx (1 << cte_bits) - 1, r2
 313  * and  src->dn_reg, r2, r2
 314  * setx ctm_offset % NBBY, r3
 315  * sll  r2, r3, r2
 316  *
 317  * or   r1, r2, r1
 318  * st?  r1, [dst->dn_reg]
 319  *
 320  * This routine allocates a new register to hold the value to be stored and
 321  * returns it.  The caller is responsible for freeing this register later.
 322  */
 323 static int
 324 dt_cg_field_set(dt_node_t *src, dt_irlist_t *dlp,
 325     dt_regset_t *drp, dt_node_t *dst)
 326 {
 327         uint64_t cmask, fmask, shift;
 328         dif_instr_t instr;
 329         int r1, r2, r3;
 330 
 331         ctf_membinfo_t m;
 332         ctf_encoding_t e;
 333         ctf_file_t *fp, *ofp;
 334         ctf_id_t type;
 335 
 336         assert(dst->dn_op == DT_TOK_PTR || dst->dn_op == DT_TOK_DOT);
 337         assert(dst->dn_right->dn_kind == DT_NODE_IDENT);
 338 
 339         fp = dst->dn_left->dn_ctfp;
 340         type = ctf_type_resolve(fp, dst->dn_left->dn_type);
 341 
 342         if (dst->dn_op == DT_TOK_PTR) {
 343                 type = ctf_type_reference(fp, type);
 344                 type = ctf_type_resolve(fp, type);
 345         }
 346 
 347         if ((fp = dt_cg_membinfo(ofp = fp, type,
 348             dst->dn_right->dn_string, &m)) == NULL) {
 349                 yypcb->pcb_hdl->dt_ctferr = ctf_errno(ofp);
 350                 longjmp(yypcb->pcb_jmpbuf, EDT_CTF);
 351         }
 352 
 353         if (ctf_type_encoding(fp, m.ctm_type, &e) != 0 || e.cte_bits > 64) {
 354                 xyerror(D_UNKNOWN, "cg: bad field: off %lu type <%ld> "
 355                     "bits %u\n", m.ctm_offset, m.ctm_type, e.cte_bits);
 356         }
 357 
 358         r1 = dt_regset_alloc(drp);
 359         r2 = dt_regset_alloc(drp);
 360         r3 = dt_regset_alloc(drp);
 361 
 362         /*
 363          * Compute shifts and masks.  We need to compute "shift" as the amount
 364          * we need to shift left to position our field in the containing word.
 365          * Refer to the comments in dt_cg_field_get(), above, for more info.
 366          * We then compute fmask as the mask that truncates the value in the
 367          * input register to width cte_bits, and cmask as the mask used to
 368          * pass through the containing bits and zero the field bits.
 369          */
 370 #ifdef _BIG_ENDIAN
 371         shift = clp2(P2ROUNDUP(e.cte_bits, NBBY) / NBBY) * NBBY -
 372             (m.ctm_offset % NBBY + e.cte_bits);
 373 #else
 374         shift = m.ctm_offset % NBBY;
 375 #endif
 376         fmask = (1ULL << e.cte_bits) - 1;
 377         cmask = ~(fmask << shift);
 378 
 379         instr = DIF_INSTR_LOAD(
 380             dt_cg_load(dst, fp, m.ctm_type), dst->dn_reg, r1);
 381         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 382 
 383         dt_cg_setx(dlp, r2, cmask);
 384         instr = DIF_INSTR_FMT(DIF_OP_AND, r1, r2, r1);
 385         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 386 
 387         dt_cg_setx(dlp, r2, fmask);
 388         instr = DIF_INSTR_FMT(DIF_OP_AND, src->dn_reg, r2, r2);
 389         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 390 
 391         dt_cg_setx(dlp, r3, shift);
 392         instr = DIF_INSTR_FMT(DIF_OP_SLL, r2, r3, r2);
 393         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 394 
 395         instr = DIF_INSTR_FMT(DIF_OP_OR, r1, r2, r1);
 396         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 397 
 398         dt_regset_free(drp, r3);
 399         dt_regset_free(drp, r2);
 400 
 401         return (r1);
 402 }
 403 
 404 static void
 405 dt_cg_store(dt_node_t *src, dt_irlist_t *dlp, dt_regset_t *drp, dt_node_t *dst)
 406 {
 407         ctf_encoding_t e;
 408         dif_instr_t instr;
 409         size_t size;
 410         int reg;
 411 
 412         /*
 413          * If we're loading a bit-field, the size of our store is found by
 414          * rounding dst's cte_bits up to a byte boundary and then finding the
 415          * nearest power of two to this value (see clp2(), above).
 416          */
 417         if ((dst->dn_flags & DT_NF_BITFIELD) &&
 418             ctf_type_encoding(dst->dn_ctfp, dst->dn_type, &e) != CTF_ERR)
 419                 size = clp2(P2ROUNDUP(e.cte_bits, NBBY) / NBBY);
 420         else
 421                 size = dt_node_type_size(src);
 422 
 423         if (src->dn_flags & DT_NF_REF) {
 424                 reg = dt_regset_alloc(drp);
 425                 dt_cg_setx(dlp, reg, size);
 426                 instr = DIF_INSTR_COPYS(src->dn_reg, reg, dst->dn_reg);
 427                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 428                 dt_regset_free(drp, reg);
 429         } else {
 430                 if (dst->dn_flags & DT_NF_BITFIELD)
 431                         reg = dt_cg_field_set(src, dlp, drp, dst);
 432                 else
 433                         reg = src->dn_reg;
 434 
 435                 switch (size) {
 436                 case 1:
 437                         instr = DIF_INSTR_STORE(DIF_OP_STB, reg, dst->dn_reg);
 438                         break;
 439                 case 2:
 440                         instr = DIF_INSTR_STORE(DIF_OP_STH, reg, dst->dn_reg);
 441                         break;
 442                 case 4:
 443                         instr = DIF_INSTR_STORE(DIF_OP_STW, reg, dst->dn_reg);
 444                         break;
 445                 case 8:
 446                         instr = DIF_INSTR_STORE(DIF_OP_STX, reg, dst->dn_reg);
 447                         break;
 448                 default:
 449                         xyerror(D_UNKNOWN, "internal error -- cg cannot store "
 450                             "size %lu when passed by value\n", (ulong_t)size);
 451                 }
 452                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 453 
 454                 if (dst->dn_flags & DT_NF_BITFIELD)
 455                         dt_regset_free(drp, reg);
 456         }
 457 }
 458 
 459 /*
 460  * Generate code for a typecast or for argument promotion from the type of the
 461  * actual to the type of the formal.  We need to generate code for casts when
 462  * a scalar type is being narrowed or changing signed-ness.  We first shift the
 463  * desired bits high (losing excess bits if narrowing) and then shift them down
 464  * using logical shift (unsigned result) or arithmetic shift (signed result).
 465  */
 466 static void
 467 dt_cg_typecast(const dt_node_t *src, const dt_node_t *dst,
 468     dt_irlist_t *dlp, dt_regset_t *drp)
 469 {
 470         size_t srcsize = dt_node_type_size(src);
 471         size_t dstsize = dt_node_type_size(dst);
 472 
 473         dif_instr_t instr;
 474         int rg;
 475 
 476         if (!dt_node_is_scalar(dst))
 477                 return; /* not a scalar */
 478         if (dstsize == srcsize &&
 479             ((src->dn_flags ^ dst->dn_flags) & DT_NF_SIGNED) == 0)
 480                 return; /* not narrowing or changing signed-ness */
 481         if (dstsize > srcsize && (src->dn_flags & DT_NF_SIGNED) == 0)
 482                 return; /* nothing to do in this case */
 483 
 484         rg = dt_regset_alloc(drp);
 485 
 486         if (dstsize > srcsize) {
 487                 int n = sizeof (uint64_t) * NBBY - srcsize * NBBY;
 488                 int s = (dstsize - srcsize) * NBBY;
 489 
 490                 dt_cg_setx(dlp, rg, n);
 491 
 492                 instr = DIF_INSTR_FMT(DIF_OP_SLL, src->dn_reg, rg, dst->dn_reg);
 493                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 494 
 495                 if ((dst->dn_flags & DT_NF_SIGNED) || n == s) {
 496                         instr = DIF_INSTR_FMT(DIF_OP_SRA,
 497                             dst->dn_reg, rg, dst->dn_reg);
 498                         dt_irlist_append(dlp,
 499                             dt_cg_node_alloc(DT_LBL_NONE, instr));
 500                 } else {
 501                         dt_cg_setx(dlp, rg, s);
 502                         instr = DIF_INSTR_FMT(DIF_OP_SRA,
 503                             dst->dn_reg, rg, dst->dn_reg);
 504                         dt_irlist_append(dlp,
 505                             dt_cg_node_alloc(DT_LBL_NONE, instr));
 506                         dt_cg_setx(dlp, rg, n - s);
 507                         instr = DIF_INSTR_FMT(DIF_OP_SRL,
 508                             dst->dn_reg, rg, dst->dn_reg);
 509                         dt_irlist_append(dlp,
 510                             dt_cg_node_alloc(DT_LBL_NONE, instr));
 511                 }
 512         } else if (dstsize != sizeof (uint64_t)) {
 513                 int n = sizeof (uint64_t) * NBBY - dstsize * NBBY;
 514 
 515                 dt_cg_setx(dlp, rg, n);
 516 
 517                 instr = DIF_INSTR_FMT(DIF_OP_SLL, src->dn_reg, rg, dst->dn_reg);
 518                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 519 
 520                 instr = DIF_INSTR_FMT((dst->dn_flags & DT_NF_SIGNED) ?
 521                     DIF_OP_SRA : DIF_OP_SRL, dst->dn_reg, rg, dst->dn_reg);
 522                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 523         }
 524 
 525         dt_regset_free(drp, rg);
 526 }
 527 
 528 /*
 529  * Generate code to push the specified argument list on to the tuple stack.
 530  * We use this routine for handling subroutine calls and associative arrays.
 531  * We must first generate code for all subexpressions before loading the stack
 532  * because any subexpression could itself require the use of the tuple stack.
 533  * This holds a number of registers equal to the number of arguments, but this
 534  * is not a huge problem because the number of arguments can't exceed the
 535  * number of tuple register stack elements anyway.  At most one extra register
 536  * is required (either by dt_cg_typecast() or for dtdt_size, below).  This
 537  * implies that a DIF implementation should offer a number of general purpose
 538  * registers at least one greater than the number of tuple registers.
 539  */
 540 static void
 541 dt_cg_arglist(dt_ident_t *idp, dt_node_t *args,
 542     dt_irlist_t *dlp, dt_regset_t *drp)
 543 {
 544         const dt_idsig_t *isp = idp->di_data;
 545         dt_node_t *dnp;
 546         int i = 0;
 547 
 548         for (dnp = args; dnp != NULL; dnp = dnp->dn_list)
 549                 dt_cg_node(dnp, dlp, drp);
 550 
 551         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, DIF_INSTR_FLUSHTS));
 552 
 553         for (dnp = args; dnp != NULL; dnp = dnp->dn_list, i++) {
 554                 dtrace_diftype_t t;
 555                 dif_instr_t instr;
 556                 uint_t op;
 557                 int reg;
 558 
 559                 dt_node_diftype(yypcb->pcb_hdl, dnp, &t);
 560 
 561                 isp->dis_args[i].dn_reg = dnp->dn_reg; /* re-use register */
 562                 dt_cg_typecast(dnp, &isp->dis_args[i], dlp, drp);
 563                 isp->dis_args[i].dn_reg = -1;
 564 
 565                 if (t.dtdt_flags & DIF_TF_BYREF) {
 566                         op = DIF_OP_PUSHTR;
 567                         if (t.dtdt_size != 0) {
 568                                 reg = dt_regset_alloc(drp);
 569                                 dt_cg_setx(dlp, reg, t.dtdt_size);
 570                         } else {
 571                                 reg = DIF_REG_R0;
 572                         }
 573                 } else {
 574                         op = DIF_OP_PUSHTV;
 575                         reg = DIF_REG_R0;
 576                 }
 577 
 578                 instr = DIF_INSTR_PUSHTS(op, t.dtdt_kind, reg, dnp->dn_reg);
 579                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 580                 dt_regset_free(drp, dnp->dn_reg);
 581 
 582                 if (reg != DIF_REG_R0)
 583                         dt_regset_free(drp, reg);
 584         }
 585 
 586         if (i > yypcb->pcb_hdl->dt_conf.dtc_diftupregs)
 587                 longjmp(yypcb->pcb_jmpbuf, EDT_NOTUPREG);
 588 }
 589 
 590 static void
 591 dt_cg_arithmetic_op(dt_node_t *dnp, dt_irlist_t *dlp,
 592     dt_regset_t *drp, uint_t op)
 593 {
 594         int is_ptr_op = (dnp->dn_op == DT_TOK_ADD || dnp->dn_op == DT_TOK_SUB ||
 595             dnp->dn_op == DT_TOK_ADD_EQ || dnp->dn_op == DT_TOK_SUB_EQ);
 596 
 597         int lp_is_ptr = dt_node_is_pointer(dnp->dn_left);
 598         int rp_is_ptr = dt_node_is_pointer(dnp->dn_right);
 599 
 600         dif_instr_t instr;
 601 
 602         if (lp_is_ptr && rp_is_ptr) {
 603                 assert(dnp->dn_op == DT_TOK_SUB);
 604                 is_ptr_op = 0;
 605         }
 606 
 607         dt_cg_node(dnp->dn_left, dlp, drp);
 608         if (is_ptr_op && rp_is_ptr)
 609                 dt_cg_ptrsize(dnp, dlp, drp, DIF_OP_MUL, dnp->dn_left->dn_reg);
 610 
 611         dt_cg_node(dnp->dn_right, dlp, drp);
 612         if (is_ptr_op && lp_is_ptr)
 613                 dt_cg_ptrsize(dnp, dlp, drp, DIF_OP_MUL, dnp->dn_right->dn_reg);
 614 
 615         instr = DIF_INSTR_FMT(op, dnp->dn_left->dn_reg,
 616             dnp->dn_right->dn_reg, dnp->dn_left->dn_reg);
 617 
 618         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 619         dt_regset_free(drp, dnp->dn_right->dn_reg);
 620         dnp->dn_reg = dnp->dn_left->dn_reg;
 621 
 622         if (lp_is_ptr && rp_is_ptr)
 623                 dt_cg_ptrsize(dnp->dn_right,
 624                     dlp, drp, DIF_OP_UDIV, dnp->dn_reg);
 625 }
 626 
 627 static uint_t
 628 dt_cg_stvar(const dt_ident_t *idp)
 629 {
 630         static const uint_t aops[] = { DIF_OP_STGAA, DIF_OP_STTAA, DIF_OP_NOP };
 631         static const uint_t sops[] = { DIF_OP_STGS, DIF_OP_STTS, DIF_OP_STLS };
 632 
 633         uint_t i = (((idp->di_flags & DT_IDFLG_LOCAL) != 0) << 1) |
 634             ((idp->di_flags & DT_IDFLG_TLS) != 0);
 635 
 636         return (idp->di_kind == DT_IDENT_ARRAY ? aops[i] : sops[i]);
 637 }
 638 
 639 static void
 640 dt_cg_prearith_op(dt_node_t *dnp, dt_irlist_t *dlp, dt_regset_t *drp, uint_t op)
 641 {
 642         ctf_file_t *ctfp = dnp->dn_ctfp;
 643         dif_instr_t instr;
 644         ctf_id_t type;
 645         ssize_t size = 1;
 646         int reg;
 647 
 648         if (dt_node_is_pointer(dnp)) {
 649                 type = ctf_type_resolve(ctfp, dnp->dn_type);
 650                 assert(ctf_type_kind(ctfp, type) == CTF_K_POINTER);
 651                 size = ctf_type_size(ctfp, ctf_type_reference(ctfp, type));
 652         }
 653 
 654         dt_cg_node(dnp->dn_child, dlp, drp);
 655         dnp->dn_reg = dnp->dn_child->dn_reg;
 656 
 657         reg = dt_regset_alloc(drp);
 658         dt_cg_setx(dlp, reg, size);
 659 
 660         instr = DIF_INSTR_FMT(op, dnp->dn_reg, reg, dnp->dn_reg);
 661         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 662         dt_regset_free(drp, reg);
 663 
 664         /*
 665          * If we are modifying a variable, generate an stv instruction from
 666          * the variable specified by the identifier.  If we are storing to a
 667          * memory address, generate code again for the left-hand side using
 668          * DT_NF_REF to get the address, and then generate a store to it.
 669          * In both paths, we store the value in dnp->dn_reg (the new value).
 670          */
 671         if (dnp->dn_child->dn_kind == DT_NODE_VAR) {
 672                 dt_ident_t *idp = dt_ident_resolve(dnp->dn_child->dn_ident);
 673 
 674                 idp->di_flags |= DT_IDFLG_DIFW;
 675                 instr = DIF_INSTR_STV(dt_cg_stvar(idp),
 676                     idp->di_id, dnp->dn_reg);
 677                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 678         } else {
 679                 uint_t rbit = dnp->dn_child->dn_flags & DT_NF_REF;
 680 
 681                 assert(dnp->dn_child->dn_flags & DT_NF_WRITABLE);
 682                 assert(dnp->dn_child->dn_flags & DT_NF_LVALUE);
 683 
 684                 dnp->dn_child->dn_flags |= DT_NF_REF; /* force pass-by-ref */
 685                 dt_cg_node(dnp->dn_child, dlp, drp);
 686 
 687                 dt_cg_store(dnp, dlp, drp, dnp->dn_child);
 688                 dt_regset_free(drp, dnp->dn_child->dn_reg);
 689 
 690                 dnp->dn_left->dn_flags &= ~DT_NF_REF;
 691                 dnp->dn_left->dn_flags |= rbit;
 692         }
 693 }
 694 
 695 static void
 696 dt_cg_postarith_op(dt_node_t *dnp, dt_irlist_t *dlp,
 697     dt_regset_t *drp, uint_t op)
 698 {
 699         ctf_file_t *ctfp = dnp->dn_ctfp;
 700         dif_instr_t instr;
 701         ctf_id_t type;
 702         ssize_t size = 1;
 703         int nreg;
 704 
 705         if (dt_node_is_pointer(dnp)) {
 706                 type = ctf_type_resolve(ctfp, dnp->dn_type);
 707                 assert(ctf_type_kind(ctfp, type) == CTF_K_POINTER);
 708                 size = ctf_type_size(ctfp, ctf_type_reference(ctfp, type));
 709         }
 710 
 711         dt_cg_node(dnp->dn_child, dlp, drp);
 712         dnp->dn_reg = dnp->dn_child->dn_reg;
 713 
 714         nreg = dt_regset_alloc(drp);
 715         dt_cg_setx(dlp, nreg, size);
 716         instr = DIF_INSTR_FMT(op, dnp->dn_reg, nreg, nreg);
 717         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 718 
 719         /*
 720          * If we are modifying a variable, generate an stv instruction from
 721          * the variable specified by the identifier.  If we are storing to a
 722          * memory address, generate code again for the left-hand side using
 723          * DT_NF_REF to get the address, and then generate a store to it.
 724          * In both paths, we store the value from 'nreg' (the new value).
 725          */
 726         if (dnp->dn_child->dn_kind == DT_NODE_VAR) {
 727                 dt_ident_t *idp = dt_ident_resolve(dnp->dn_child->dn_ident);
 728 
 729                 idp->di_flags |= DT_IDFLG_DIFW;
 730                 instr = DIF_INSTR_STV(dt_cg_stvar(idp), idp->di_id, nreg);
 731                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 732         } else {
 733                 uint_t rbit = dnp->dn_child->dn_flags & DT_NF_REF;
 734                 int oreg = dnp->dn_reg;
 735 
 736                 assert(dnp->dn_child->dn_flags & DT_NF_WRITABLE);
 737                 assert(dnp->dn_child->dn_flags & DT_NF_LVALUE);
 738 
 739                 dnp->dn_child->dn_flags |= DT_NF_REF; /* force pass-by-ref */
 740                 dt_cg_node(dnp->dn_child, dlp, drp);
 741 
 742                 dnp->dn_reg = nreg;
 743                 dt_cg_store(dnp, dlp, drp, dnp->dn_child);
 744                 dnp->dn_reg = oreg;
 745 
 746                 dt_regset_free(drp, dnp->dn_child->dn_reg);
 747                 dnp->dn_left->dn_flags &= ~DT_NF_REF;
 748                 dnp->dn_left->dn_flags |= rbit;
 749         }
 750 
 751         dt_regset_free(drp, nreg);
 752 }
 753 
 754 /*
 755  * Determine if we should perform signed or unsigned comparison for an OP2.
 756  * If both operands are of arithmetic type, perform the usual arithmetic
 757  * conversions to determine the common real type for comparison [ISOC 6.5.8.3].
 758  */
 759 static int
 760 dt_cg_compare_signed(dt_node_t *dnp)
 761 {
 762         dt_node_t dn;
 763 
 764         if (dt_node_is_string(dnp->dn_left) ||
 765             dt_node_is_string(dnp->dn_right))
 766                 return (1); /* strings always compare signed */
 767         else if (!dt_node_is_arith(dnp->dn_left) ||
 768             !dt_node_is_arith(dnp->dn_right))
 769                 return (0); /* non-arithmetic types always compare unsigned */
 770 
 771         bzero(&dn, sizeof (dn));
 772         dt_node_promote(dnp->dn_left, dnp->dn_right, &dn);
 773         return (dn.dn_flags & DT_NF_SIGNED);
 774 }
 775 
 776 static void
 777 dt_cg_compare_op(dt_node_t *dnp, dt_irlist_t *dlp, dt_regset_t *drp, uint_t op)
 778 {
 779         uint_t lbl_true = dt_irlist_label(dlp);
 780         uint_t lbl_post = dt_irlist_label(dlp);
 781 
 782         dif_instr_t instr;
 783         uint_t opc;
 784 
 785         dt_cg_node(dnp->dn_left, dlp, drp);
 786         dt_cg_node(dnp->dn_right, dlp, drp);
 787 
 788         if (dt_node_is_string(dnp->dn_left) || dt_node_is_string(dnp->dn_right))
 789                 opc = DIF_OP_SCMP;
 790         else
 791                 opc = DIF_OP_CMP;
 792 
 793         instr = DIF_INSTR_CMP(opc, dnp->dn_left->dn_reg, dnp->dn_right->dn_reg);
 794         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 795         dt_regset_free(drp, dnp->dn_right->dn_reg);
 796         dnp->dn_reg = dnp->dn_left->dn_reg;
 797 
 798         instr = DIF_INSTR_BRANCH(op, lbl_true);
 799         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 800 
 801         instr = DIF_INSTR_MOV(DIF_REG_R0, dnp->dn_reg);
 802         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 803 
 804         instr = DIF_INSTR_BRANCH(DIF_OP_BA, lbl_post);
 805         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 806 
 807         dt_cg_xsetx(dlp, NULL, lbl_true, dnp->dn_reg, 1);
 808         dt_irlist_append(dlp, dt_cg_node_alloc(lbl_post, DIF_INSTR_NOP));
 809 }
 810 
 811 /*
 812  * Code generation for the ternary op requires some trickery with the assembler
 813  * in order to conserve registers.  We generate code for dn_expr and dn_left
 814  * and free their registers so they do not have be consumed across codegen for
 815  * dn_right.  We insert a dummy MOV at the end of dn_left into the destination
 816  * register, which is not yet known because we haven't done dn_right yet, and
 817  * save the pointer to this instruction node.  We then generate code for
 818  * dn_right and use its register as our output.  Finally, we reach back and
 819  * patch the instruction for dn_left to move its output into this register.
 820  */
 821 static void
 822 dt_cg_ternary_op(dt_node_t *dnp, dt_irlist_t *dlp, dt_regset_t *drp)
 823 {
 824         uint_t lbl_false = dt_irlist_label(dlp);
 825         uint_t lbl_post = dt_irlist_label(dlp);
 826 
 827         dif_instr_t instr;
 828         dt_irnode_t *dip;
 829 
 830         dt_cg_node(dnp->dn_expr, dlp, drp);
 831         instr = DIF_INSTR_TST(dnp->dn_expr->dn_reg);
 832         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 833         dt_regset_free(drp, dnp->dn_expr->dn_reg);
 834 
 835         instr = DIF_INSTR_BRANCH(DIF_OP_BE, lbl_false);
 836         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 837 
 838         dt_cg_node(dnp->dn_left, dlp, drp);
 839         instr = DIF_INSTR_MOV(dnp->dn_left->dn_reg, DIF_REG_R0);
 840         dip = dt_cg_node_alloc(DT_LBL_NONE, instr); /* save dip for below */
 841         dt_irlist_append(dlp, dip);
 842         dt_regset_free(drp, dnp->dn_left->dn_reg);
 843 
 844         instr = DIF_INSTR_BRANCH(DIF_OP_BA, lbl_post);
 845         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 846 
 847         dt_irlist_append(dlp, dt_cg_node_alloc(lbl_false, DIF_INSTR_NOP));
 848         dt_cg_node(dnp->dn_right, dlp, drp);
 849         dnp->dn_reg = dnp->dn_right->dn_reg;
 850 
 851         /*
 852          * Now that dn_reg is assigned, reach back and patch the correct MOV
 853          * instruction into the tail of dn_left.  We know dn_reg was unused
 854          * at that point because otherwise dn_right couldn't have allocated it.
 855          */
 856         dip->di_instr = DIF_INSTR_MOV(dnp->dn_left->dn_reg, dnp->dn_reg);
 857         dt_irlist_append(dlp, dt_cg_node_alloc(lbl_post, DIF_INSTR_NOP));
 858 }
 859 
 860 static void
 861 dt_cg_logical_and(dt_node_t *dnp, dt_irlist_t *dlp, dt_regset_t *drp)
 862 {
 863         uint_t lbl_false = dt_irlist_label(dlp);
 864         uint_t lbl_post = dt_irlist_label(dlp);
 865 
 866         dif_instr_t instr;
 867 
 868         dt_cg_node(dnp->dn_left, dlp, drp);
 869         instr = DIF_INSTR_TST(dnp->dn_left->dn_reg);
 870         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 871         dt_regset_free(drp, dnp->dn_left->dn_reg);
 872 
 873         instr = DIF_INSTR_BRANCH(DIF_OP_BE, lbl_false);
 874         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 875 
 876         dt_cg_node(dnp->dn_right, dlp, drp);
 877         instr = DIF_INSTR_TST(dnp->dn_right->dn_reg);
 878         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 879         dnp->dn_reg = dnp->dn_right->dn_reg;
 880 
 881         instr = DIF_INSTR_BRANCH(DIF_OP_BE, lbl_false);
 882         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 883 
 884         dt_cg_setx(dlp, dnp->dn_reg, 1);
 885 
 886         instr = DIF_INSTR_BRANCH(DIF_OP_BA, lbl_post);
 887         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 888 
 889         instr = DIF_INSTR_MOV(DIF_REG_R0, dnp->dn_reg);
 890         dt_irlist_append(dlp, dt_cg_node_alloc(lbl_false, instr));
 891 
 892         dt_irlist_append(dlp, dt_cg_node_alloc(lbl_post, DIF_INSTR_NOP));
 893 }
 894 
 895 static void
 896 dt_cg_logical_xor(dt_node_t *dnp, dt_irlist_t *dlp, dt_regset_t *drp)
 897 {
 898         uint_t lbl_next = dt_irlist_label(dlp);
 899         uint_t lbl_tail = dt_irlist_label(dlp);
 900 
 901         dif_instr_t instr;
 902 
 903         dt_cg_node(dnp->dn_left, dlp, drp);
 904         instr = DIF_INSTR_TST(dnp->dn_left->dn_reg);
 905         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 906 
 907         instr = DIF_INSTR_BRANCH(DIF_OP_BE, lbl_next);
 908         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 909         dt_cg_setx(dlp, dnp->dn_left->dn_reg, 1);
 910 
 911         dt_irlist_append(dlp, dt_cg_node_alloc(lbl_next, DIF_INSTR_NOP));
 912         dt_cg_node(dnp->dn_right, dlp, drp);
 913 
 914         instr = DIF_INSTR_TST(dnp->dn_right->dn_reg);
 915         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 916 
 917         instr = DIF_INSTR_BRANCH(DIF_OP_BE, lbl_tail);
 918         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 919         dt_cg_setx(dlp, dnp->dn_right->dn_reg, 1);
 920 
 921         instr = DIF_INSTR_FMT(DIF_OP_XOR, dnp->dn_left->dn_reg,
 922             dnp->dn_right->dn_reg, dnp->dn_left->dn_reg);
 923 
 924         dt_irlist_append(dlp, dt_cg_node_alloc(lbl_tail, instr));
 925 
 926         dt_regset_free(drp, dnp->dn_right->dn_reg);
 927         dnp->dn_reg = dnp->dn_left->dn_reg;
 928 }
 929 
 930 static void
 931 dt_cg_logical_or(dt_node_t *dnp, dt_irlist_t *dlp, dt_regset_t *drp)
 932 {
 933         uint_t lbl_true = dt_irlist_label(dlp);
 934         uint_t lbl_false = dt_irlist_label(dlp);
 935         uint_t lbl_post = dt_irlist_label(dlp);
 936 
 937         dif_instr_t instr;
 938 
 939         dt_cg_node(dnp->dn_left, dlp, drp);
 940         instr = DIF_INSTR_TST(dnp->dn_left->dn_reg);
 941         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 942         dt_regset_free(drp, dnp->dn_left->dn_reg);
 943 
 944         instr = DIF_INSTR_BRANCH(DIF_OP_BNE, lbl_true);
 945         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 946 
 947         dt_cg_node(dnp->dn_right, dlp, drp);
 948         instr = DIF_INSTR_TST(dnp->dn_right->dn_reg);
 949         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 950         dnp->dn_reg = dnp->dn_right->dn_reg;
 951 
 952         instr = DIF_INSTR_BRANCH(DIF_OP_BE, lbl_false);
 953         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 954 
 955         dt_cg_xsetx(dlp, NULL, lbl_true, dnp->dn_reg, 1);
 956 
 957         instr = DIF_INSTR_BRANCH(DIF_OP_BA, lbl_post);
 958         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 959 
 960         instr = DIF_INSTR_MOV(DIF_REG_R0, dnp->dn_reg);
 961         dt_irlist_append(dlp, dt_cg_node_alloc(lbl_false, instr));
 962 
 963         dt_irlist_append(dlp, dt_cg_node_alloc(lbl_post, DIF_INSTR_NOP));
 964 }
 965 
 966 static void
 967 dt_cg_logical_neg(dt_node_t *dnp, dt_irlist_t *dlp, dt_regset_t *drp)
 968 {
 969         uint_t lbl_zero = dt_irlist_label(dlp);
 970         uint_t lbl_post = dt_irlist_label(dlp);
 971 
 972         dif_instr_t instr;
 973 
 974         dt_cg_node(dnp->dn_child, dlp, drp);
 975         dnp->dn_reg = dnp->dn_child->dn_reg;
 976 
 977         instr = DIF_INSTR_TST(dnp->dn_reg);
 978         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 979 
 980         instr = DIF_INSTR_BRANCH(DIF_OP_BE, lbl_zero);
 981         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 982 
 983         instr = DIF_INSTR_MOV(DIF_REG_R0, dnp->dn_reg);
 984         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 985 
 986         instr = DIF_INSTR_BRANCH(DIF_OP_BA, lbl_post);
 987         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
 988 
 989         dt_cg_xsetx(dlp, NULL, lbl_zero, dnp->dn_reg, 1);
 990         dt_irlist_append(dlp, dt_cg_node_alloc(lbl_post, DIF_INSTR_NOP));
 991 }
 992 
 993 static void
 994 dt_cg_asgn_op(dt_node_t *dnp, dt_irlist_t *dlp, dt_regset_t *drp)
 995 {
 996         dif_instr_t instr;
 997         dt_ident_t *idp;
 998 
 999         /*
1000          * If we are performing a structure assignment of a translated type,
1001          * we must instantiate all members and create a snapshot of the object
1002          * in scratch space.  We allocs a chunk of memory, generate code for
1003          * each member, and then set dnp->dn_reg to the scratch object address.
1004          */
1005         if ((idp = dt_node_resolve(dnp->dn_right, DT_IDENT_XLSOU)) != NULL) {
1006                 ctf_membinfo_t ctm;
1007                 dt_xlator_t *dxp = idp->di_data;
1008                 dt_node_t *mnp, dn, mn;
1009                 int r1, r2;
1010 
1011                 /*
1012                  * Create two fake dt_node_t's representing operator "." and a
1013                  * right-hand identifier child node.  These will be repeatedly
1014                  * modified according to each instantiated member so that we
1015                  * can pass them to dt_cg_store() and effect a member store.
1016                  */
1017                 bzero(&dn, sizeof (dt_node_t));
1018                 dn.dn_kind = DT_NODE_OP2;
1019                 dn.dn_op = DT_TOK_DOT;
1020                 dn.dn_left = dnp;
1021                 dn.dn_right = &mn;
1022 
1023                 bzero(&mn, sizeof (dt_node_t));
1024                 mn.dn_kind = DT_NODE_IDENT;
1025                 mn.dn_op = DT_TOK_IDENT;
1026 
1027                 /*
1028                  * Allocate a register for our scratch data pointer.  First we
1029                  * set it to the size of our data structure, and then replace
1030                  * it with the result of an allocs of the specified size.
1031                  */
1032                 r1 = dt_regset_alloc(drp);
1033                 dt_cg_setx(dlp, r1,
1034                     ctf_type_size(dxp->dx_dst_ctfp, dxp->dx_dst_base));
1035 
1036                 instr = DIF_INSTR_ALLOCS(r1, r1);
1037                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
1038 
1039                 /*
1040                  * When dt_cg_asgn_op() is called, we have already generated
1041                  * code for dnp->dn_right, which is the translator input.  We
1042                  * now associate this register with the translator's input
1043                  * identifier so it can be referenced during our member loop.
1044                  */
1045                 dxp->dx_ident->di_flags |= DT_IDFLG_CGREG;
1046                 dxp->dx_ident->di_id = dnp->dn_right->dn_reg;
1047 
1048                 for (mnp = dxp->dx_members; mnp != NULL; mnp = mnp->dn_list) {
1049                         /*
1050                          * Generate code for the translator member expression,
1051                          * and then cast the result to the member type.
1052                          */
1053                         dt_cg_node(mnp->dn_membexpr, dlp, drp);
1054                         mnp->dn_reg = mnp->dn_membexpr->dn_reg;
1055                         dt_cg_typecast(mnp->dn_membexpr, mnp, dlp, drp);
1056 
1057                         /*
1058                          * Ask CTF for the offset of the member so we can store
1059                          * to the appropriate offset.  This call has already
1060                          * been done once by the parser, so it should succeed.
1061                          */
1062                         if (ctf_member_info(dxp->dx_dst_ctfp, dxp->dx_dst_base,
1063                             mnp->dn_membname, &ctm) == CTF_ERR) {
1064                                 yypcb->pcb_hdl->dt_ctferr =
1065                                     ctf_errno(dxp->dx_dst_ctfp);
1066                                 longjmp(yypcb->pcb_jmpbuf, EDT_CTF);
1067                         }
1068 
1069                         /*
1070                          * If the destination member is at offset 0, store the
1071                          * result directly to r1 (the scratch buffer address).
1072                          * Otherwise allocate another temporary for the offset
1073                          * and add r1 to it before storing the result.
1074                          */
1075                         if (ctm.ctm_offset != 0) {
1076                                 r2 = dt_regset_alloc(drp);
1077 
1078                                 /*
1079                                  * Add the member offset rounded down to the
1080                                  * nearest byte.  If the offset was not aligned
1081                                  * on a byte boundary, this member is a bit-
1082                                  * field and dt_cg_store() will handle masking.
1083                                  */
1084                                 dt_cg_setx(dlp, r2, ctm.ctm_offset / NBBY);
1085                                 instr = DIF_INSTR_FMT(DIF_OP_ADD, r1, r2, r2);
1086                                 dt_irlist_append(dlp,
1087                                     dt_cg_node_alloc(DT_LBL_NONE, instr));
1088 
1089                                 dt_node_type_propagate(mnp, &dn);
1090                                 dn.dn_right->dn_string = mnp->dn_membname;
1091                                 dn.dn_reg = r2;
1092 
1093                                 dt_cg_store(mnp, dlp, drp, &dn);
1094                                 dt_regset_free(drp, r2);
1095 
1096                         } else {
1097                                 dt_node_type_propagate(mnp, &dn);
1098                                 dn.dn_right->dn_string = mnp->dn_membname;
1099                                 dn.dn_reg = r1;
1100 
1101                                 dt_cg_store(mnp, dlp, drp, &dn);
1102                         }
1103 
1104                         dt_regset_free(drp, mnp->dn_reg);
1105                 }
1106 
1107                 dxp->dx_ident->di_flags &= ~DT_IDFLG_CGREG;
1108                 dxp->dx_ident->di_id = 0;
1109 
1110                 if (dnp->dn_right->dn_reg != -1)
1111                         dt_regset_free(drp, dnp->dn_right->dn_reg);
1112 
1113                 assert(dnp->dn_reg == dnp->dn_right->dn_reg);
1114                 dnp->dn_reg = r1;
1115         }
1116 
1117         /*
1118          * If we are storing to a variable, generate an stv instruction from
1119          * the variable specified by the identifier.  If we are storing to a
1120          * memory address, generate code again for the left-hand side using
1121          * DT_NF_REF to get the address, and then generate a store to it.
1122          * In both paths, we assume dnp->dn_reg already has the new value.
1123          */
1124         if (dnp->dn_left->dn_kind == DT_NODE_VAR) {
1125                 idp = dt_ident_resolve(dnp->dn_left->dn_ident);
1126 
1127                 if (idp->di_kind == DT_IDENT_ARRAY)
1128                         dt_cg_arglist(idp, dnp->dn_left->dn_args, dlp, drp);
1129 
1130                 idp->di_flags |= DT_IDFLG_DIFW;
1131                 instr = DIF_INSTR_STV(dt_cg_stvar(idp),
1132                     idp->di_id, dnp->dn_reg);
1133                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
1134         } else {
1135                 uint_t rbit = dnp->dn_left->dn_flags & DT_NF_REF;
1136 
1137                 assert(dnp->dn_left->dn_flags & DT_NF_WRITABLE);
1138                 assert(dnp->dn_left->dn_flags & DT_NF_LVALUE);
1139 
1140                 dnp->dn_left->dn_flags |= DT_NF_REF; /* force pass-by-ref */
1141 
1142                 dt_cg_node(dnp->dn_left, dlp, drp);
1143                 dt_cg_store(dnp, dlp, drp, dnp->dn_left);
1144                 dt_regset_free(drp, dnp->dn_left->dn_reg);
1145 
1146                 dnp->dn_left->dn_flags &= ~DT_NF_REF;
1147                 dnp->dn_left->dn_flags |= rbit;
1148         }
1149 }
1150 
1151 static void
1152 dt_cg_assoc_op(dt_node_t *dnp, dt_irlist_t *dlp, dt_regset_t *drp)
1153 {
1154         dif_instr_t instr;
1155         uint_t op;
1156 
1157         assert(dnp->dn_kind == DT_NODE_VAR);
1158         assert(!(dnp->dn_ident->di_flags & DT_IDFLG_LOCAL));
1159         assert(dnp->dn_args != NULL);
1160 
1161         dt_cg_arglist(dnp->dn_ident, dnp->dn_args, dlp, drp);
1162 
1163         dnp->dn_reg = dt_regset_alloc(drp);
1164 
1165         if (dnp->dn_ident->di_flags & DT_IDFLG_TLS)
1166                 op = DIF_OP_LDTAA;
1167         else
1168                 op = DIF_OP_LDGAA;
1169 
1170         dnp->dn_ident->di_flags |= DT_IDFLG_DIFR;
1171         instr = DIF_INSTR_LDV(op, dnp->dn_ident->di_id, dnp->dn_reg);
1172         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
1173 
1174         /*
1175          * If the associative array is a pass-by-reference type, then we are
1176          * loading its value as a pointer to either load or store through it.
1177          * The array element in question may not have been faulted in yet, in
1178          * which case DIF_OP_LD*AA will return zero.  We append an epilogue
1179          * of instructions similar to the following:
1180          *
1181          *        ld?aa  id, %r1        ! base ld?aa instruction above
1182          *        tst    %r1            ! start of epilogue
1183          *   +--- bne    label
1184          *   |    setx   size, %r1
1185          *   |    allocs %r1, %r1
1186          *   |    st?aa  id, %r1
1187          *   |    ld?aa  id, %r1
1188          *   v
1189          * label: < rest of code >
1190          *
1191          * The idea is that we allocs a zero-filled chunk of scratch space and
1192          * do a DIF_OP_ST*AA to fault in and initialize the array element, and
1193          * then reload it to get the faulted-in address of the new variable
1194          * storage.  This isn't cheap, but pass-by-ref associative array values
1195          * are (thus far) uncommon and the allocs cost only occurs once.  If
1196          * this path becomes important to DTrace users, we can improve things
1197          * by adding a new DIF opcode to fault in associative array elements.
1198          */
1199         if (dnp->dn_flags & DT_NF_REF) {
1200                 uint_t stvop = op == DIF_OP_LDTAA ? DIF_OP_STTAA : DIF_OP_STGAA;
1201                 uint_t label = dt_irlist_label(dlp);
1202 
1203                 instr = DIF_INSTR_TST(dnp->dn_reg);
1204                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
1205 
1206                 instr = DIF_INSTR_BRANCH(DIF_OP_BNE, label);
1207                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
1208 
1209                 dt_cg_setx(dlp, dnp->dn_reg, dt_node_type_size(dnp));
1210                 instr = DIF_INSTR_ALLOCS(dnp->dn_reg, dnp->dn_reg);
1211                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
1212 
1213                 dnp->dn_ident->di_flags |= DT_IDFLG_DIFW;
1214                 instr = DIF_INSTR_STV(stvop, dnp->dn_ident->di_id, dnp->dn_reg);
1215                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
1216 
1217                 instr = DIF_INSTR_LDV(op, dnp->dn_ident->di_id, dnp->dn_reg);
1218                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
1219 
1220                 dt_irlist_append(dlp, dt_cg_node_alloc(label, DIF_INSTR_NOP));
1221         }
1222 }
1223 
1224 static void
1225 dt_cg_array_op(dt_node_t *dnp, dt_irlist_t *dlp, dt_regset_t *drp)
1226 {
1227         dt_probe_t *prp = yypcb->pcb_probe;
1228         uintmax_t saved = dnp->dn_args->dn_value;
1229         dt_ident_t *idp = dnp->dn_ident;
1230 
1231         dif_instr_t instr;
1232         uint_t op;
1233         size_t size;
1234         int reg, n;
1235 
1236         assert(dnp->dn_kind == DT_NODE_VAR);
1237         assert(!(idp->di_flags & DT_IDFLG_LOCAL));
1238 
1239         assert(dnp->dn_args->dn_kind == DT_NODE_INT);
1240         assert(dnp->dn_args->dn_list == NULL);
1241 
1242         /*
1243          * If this is a reference in the args[] array, temporarily modify the
1244          * array index according to the static argument mapping (if any),
1245          * unless the argument reference is provided by a dynamic translator.
1246          * If we're using a dynamic translator for args[], then just set dn_reg
1247          * to an invalid reg and return: DIF_OP_XLARG will fetch the arg later.
1248          */
1249         if (idp->di_id == DIF_VAR_ARGS) {
1250                 if ((idp->di_kind == DT_IDENT_XLPTR ||
1251                     idp->di_kind == DT_IDENT_XLSOU) &&
1252                     dt_xlator_dynamic(idp->di_data)) {
1253                         dnp->dn_reg = -1;
1254                         return;
1255                 }
1256                 dnp->dn_args->dn_value = prp->pr_mapping[saved];
1257         }
1258 
1259         dt_cg_node(dnp->dn_args, dlp, drp);
1260         dnp->dn_args->dn_value = saved;
1261 
1262         dnp->dn_reg = dnp->dn_args->dn_reg;
1263 
1264         if (idp->di_flags & DT_IDFLG_TLS)
1265                 op = DIF_OP_LDTA;
1266         else
1267                 op = DIF_OP_LDGA;
1268 
1269         idp->di_flags |= DT_IDFLG_DIFR;
1270 
1271         instr = DIF_INSTR_LDA(op, idp->di_id,
1272             dnp->dn_args->dn_reg, dnp->dn_reg);
1273 
1274         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
1275 
1276         /*
1277          * If this is a reference to the args[] array, we need to take the
1278          * additional step of explicitly eliminating any bits larger than the
1279          * type size: the DIF interpreter in the kernel will always give us
1280          * the raw (64-bit) argument value, and any bits larger than the type
1281          * size may be junk.  As a practical matter, this arises only on 64-bit
1282          * architectures and only when the argument index is larger than the
1283          * number of arguments passed directly to DTrace: if a 8-, 16- or
1284          * 32-bit argument must be retrieved from the stack, it is possible
1285          * (and it some cases, likely) that the upper bits will be garbage.
1286          */
1287         if (idp->di_id != DIF_VAR_ARGS || !dt_node_is_scalar(dnp))
1288                 return;
1289 
1290         if ((size = dt_node_type_size(dnp)) == sizeof (uint64_t))
1291                 return;
1292 
1293         reg = dt_regset_alloc(drp);
1294         assert(size < sizeof (uint64_t));
1295         n = sizeof (uint64_t) * NBBY - size * NBBY;
1296 
1297         dt_cg_setx(dlp, reg, n);
1298 
1299         instr = DIF_INSTR_FMT(DIF_OP_SLL, dnp->dn_reg, reg, dnp->dn_reg);
1300         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
1301 
1302         instr = DIF_INSTR_FMT((dnp->dn_flags & DT_NF_SIGNED) ?
1303             DIF_OP_SRA : DIF_OP_SRL, dnp->dn_reg, reg, dnp->dn_reg);
1304 
1305         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
1306         dt_regset_free(drp, reg);
1307 }
1308 
1309 /*
1310  * Generate code for an inlined variable reference.  Inlines can be used to
1311  * define either scalar or associative array substitutions.  For scalars, we
1312  * simply generate code for the parse tree saved in the identifier's din_root,
1313  * and then cast the resulting expression to the inline's declaration type.
1314  * For arrays, we take the input parameter subtrees from dnp->dn_args and
1315  * temporarily store them in the din_root of each din_argv[i] identifier,
1316  * which are themselves inlines and were set up for us by the parser.  The
1317  * result is that any reference to the inlined parameter inside the top-level
1318  * din_root will turn into a recursive call to dt_cg_inline() for a scalar
1319  * inline whose din_root will refer to the subtree pointed to by the argument.
1320  */
1321 static void
1322 dt_cg_inline(dt_node_t *dnp, dt_irlist_t *dlp, dt_regset_t *drp)
1323 {
1324         dt_ident_t *idp = dnp->dn_ident;
1325         dt_idnode_t *inp = idp->di_iarg;
1326 
1327         dt_idnode_t *pinp;
1328         dt_node_t *pnp;
1329         int i;
1330 
1331         assert(idp->di_flags & DT_IDFLG_INLINE);
1332         assert(idp->di_ops == &dt_idops_inline);
1333 
1334         if (idp->di_kind == DT_IDENT_ARRAY) {
1335                 for (i = 0, pnp = dnp->dn_args;
1336                     pnp != NULL; pnp = pnp->dn_list, i++) {
1337                         if (inp->din_argv[i] != NULL) {
1338                                 pinp = inp->din_argv[i]->di_iarg;
1339                                 pinp->din_root = pnp;
1340                         }
1341                 }
1342         }
1343 
1344         dt_cg_node(inp->din_root, dlp, drp);
1345         dnp->dn_reg = inp->din_root->dn_reg;
1346         dt_cg_typecast(inp->din_root, dnp, dlp, drp);
1347 
1348         if (idp->di_kind == DT_IDENT_ARRAY) {
1349                 for (i = 0; i < inp->din_argc; i++) {
1350                         pinp = inp->din_argv[i]->di_iarg;
1351                         pinp->din_root = NULL;
1352                 }
1353         }
1354 }
1355 
1356 typedef struct dt_xlmemb {
1357         dt_ident_t *dtxl_idp;           /* translated ident */
1358         dt_irlist_t *dtxl_dlp;          /* instruction list */
1359         dt_regset_t *dtxl_drp;          /* register set */
1360         int dtxl_sreg;                  /* location of the translation input */
1361         int dtxl_dreg;                  /* location of our allocated buffer */
1362 } dt_xlmemb_t;
1363 
1364 /*ARGSUSED*/
1365 static int
1366 dt_cg_xlate_member(const char *name, ctf_id_t type, ulong_t off, void *arg)
1367 {
1368         dt_xlmemb_t *dx = arg;
1369         dt_ident_t *idp = dx->dtxl_idp;
1370         dt_irlist_t *dlp = dx->dtxl_dlp;
1371         dt_regset_t *drp = dx->dtxl_drp;
1372 
1373         dt_node_t *mnp;
1374         dt_xlator_t *dxp;
1375 
1376         int reg, treg;
1377         uint32_t instr;
1378         size_t size;
1379 
1380         /* Generate code for the translation. */
1381         dxp = idp->di_data;
1382         mnp = dt_xlator_member(dxp, name);
1383 
1384         /* If there's no translator for the given member, skip it. */
1385         if (mnp == NULL)
1386                 return (0);
1387 
1388         dxp->dx_ident->di_flags |= DT_IDFLG_CGREG;
1389         dxp->dx_ident->di_id = dx->dtxl_sreg;
1390 
1391         dt_cg_node(mnp->dn_membexpr, dlp, drp);
1392 
1393         dxp->dx_ident->di_flags &= ~DT_IDFLG_CGREG;
1394         dxp->dx_ident->di_id = 0;
1395 
1396         treg = mnp->dn_membexpr->dn_reg;
1397 
1398         /* Compute the offset into our buffer and store the result there. */
1399         reg = dt_regset_alloc(drp);
1400 
1401         dt_cg_setx(dlp, reg, off / NBBY);
1402         instr = DIF_INSTR_FMT(DIF_OP_ADD, dx->dtxl_dreg, reg, reg);
1403         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
1404 
1405         size = ctf_type_size(mnp->dn_membexpr->dn_ctfp,
1406             mnp->dn_membexpr->dn_type);
1407         if (dt_node_is_scalar(mnp->dn_membexpr)) {
1408                 /*
1409                  * Copying scalars is simple.
1410                  */
1411                 switch (size) {
1412                 case 1:
1413                         instr = DIF_INSTR_STORE(DIF_OP_STB, treg, reg);
1414                         break;
1415                 case 2:
1416                         instr = DIF_INSTR_STORE(DIF_OP_STH, treg, reg);
1417                         break;
1418                 case 4:
1419                         instr = DIF_INSTR_STORE(DIF_OP_STW, treg, reg);
1420                         break;
1421                 case 8:
1422                         instr = DIF_INSTR_STORE(DIF_OP_STX, treg, reg);
1423                         break;
1424                 default:
1425                         xyerror(D_UNKNOWN, "internal error -- unexpected "
1426                             "size: %lu\n", (ulong_t)size);
1427                 }
1428 
1429                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
1430 
1431         } else if (dt_node_is_string(mnp->dn_membexpr)) {
1432                 int szreg;
1433 
1434                 /*
1435                  * Use the copys instruction for strings.
1436                  */
1437                 szreg = dt_regset_alloc(drp);
1438                 dt_cg_setx(dlp, szreg, size);
1439                 instr = DIF_INSTR_COPYS(treg, szreg, reg);
1440                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
1441                 dt_regset_free(drp, szreg);
1442         } else {
1443                 int szreg;
1444 
1445                 /*
1446                  * If it's anything else then we'll just bcopy it.
1447                  */
1448                 szreg = dt_regset_alloc(drp);
1449                 dt_cg_setx(dlp, szreg, size);
1450                 dt_irlist_append(dlp,
1451                     dt_cg_node_alloc(DT_LBL_NONE, DIF_INSTR_FLUSHTS));
1452                 instr = DIF_INSTR_PUSHTS(DIF_OP_PUSHTV, DIF_TYPE_CTF,
1453                     DIF_REG_R0, treg);
1454                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
1455                 instr = DIF_INSTR_PUSHTS(DIF_OP_PUSHTV, DIF_TYPE_CTF,
1456                     DIF_REG_R0, reg);
1457                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
1458                 instr = DIF_INSTR_PUSHTS(DIF_OP_PUSHTV, DIF_TYPE_CTF,
1459                     DIF_REG_R0, szreg);
1460                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
1461                 instr = DIF_INSTR_CALL(DIF_SUBR_BCOPY, szreg);
1462                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
1463                 dt_regset_free(drp, szreg);
1464         }
1465 
1466         dt_regset_free(drp, reg);
1467         dt_regset_free(drp, treg);
1468 
1469         return (0);
1470 }
1471 
1472 /*
1473  * If we're expanding a translated type, we create an appropriately sized
1474  * buffer with alloca() and then translate each member into it.
1475  */
1476 static int
1477 dt_cg_xlate_expand(dt_node_t *dnp, dt_ident_t *idp, dt_irlist_t *dlp,
1478     dt_regset_t *drp)
1479 {
1480         dt_xlmemb_t dlm;
1481         uint32_t instr;
1482         int dreg;
1483         size_t size;
1484 
1485         dreg = dt_regset_alloc(drp);
1486         size = ctf_type_size(dnp->dn_ident->di_ctfp, dnp->dn_ident->di_type);
1487 
1488         /* Call alloca() to create the buffer. */
1489         dt_cg_setx(dlp, dreg, size);
1490 
1491         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, DIF_INSTR_FLUSHTS));
1492 
1493         instr = DIF_INSTR_PUSHTS(DIF_OP_PUSHTV, DIF_TYPE_CTF, DIF_REG_R0, dreg);
1494         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
1495 
1496         instr = DIF_INSTR_CALL(DIF_SUBR_ALLOCA, dreg);
1497         dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
1498 
1499         /* Generate the translation for each member. */
1500         dlm.dtxl_idp = idp;
1501         dlm.dtxl_dlp = dlp;
1502         dlm.dtxl_drp = drp;
1503         dlm.dtxl_sreg = dnp->dn_reg;
1504         dlm.dtxl_dreg = dreg;
1505         (void) ctf_member_iter(dnp->dn_ident->di_ctfp,
1506             dnp->dn_ident->di_type, dt_cg_xlate_member,
1507             &dlm);
1508 
1509         return (dreg);
1510 }
1511 
1512 static void
1513 dt_cg_node(dt_node_t *dnp, dt_irlist_t *dlp, dt_regset_t *drp)
1514 {
1515         ctf_file_t *ctfp = dnp->dn_ctfp;
1516         ctf_file_t *octfp;
1517         ctf_membinfo_t m;
1518         ctf_id_t type;
1519 
1520         dif_instr_t instr;
1521         dt_ident_t *idp;
1522         ssize_t stroff;
1523         uint_t op;
1524 
1525         switch (dnp->dn_op) {
1526         case DT_TOK_COMMA:
1527                 dt_cg_node(dnp->dn_left, dlp, drp);
1528                 dt_regset_free(drp, dnp->dn_left->dn_reg);
1529                 dt_cg_node(dnp->dn_right, dlp, drp);
1530                 dnp->dn_reg = dnp->dn_right->dn_reg;
1531                 break;
1532 
1533         case DT_TOK_ASGN:
1534                 dt_cg_node(dnp->dn_right, dlp, drp);
1535                 dnp->dn_reg = dnp->dn_right->dn_reg;
1536                 dt_cg_asgn_op(dnp, dlp, drp);
1537                 break;
1538 
1539         case DT_TOK_ADD_EQ:
1540                 dt_cg_arithmetic_op(dnp, dlp, drp, DIF_OP_ADD);
1541                 dt_cg_asgn_op(dnp, dlp, drp);
1542                 break;
1543 
1544         case DT_TOK_SUB_EQ:
1545                 dt_cg_arithmetic_op(dnp, dlp, drp, DIF_OP_SUB);
1546                 dt_cg_asgn_op(dnp, dlp, drp);
1547                 break;
1548 
1549         case DT_TOK_MUL_EQ:
1550                 dt_cg_arithmetic_op(dnp, dlp, drp, DIF_OP_MUL);
1551                 dt_cg_asgn_op(dnp, dlp, drp);
1552                 break;
1553 
1554         case DT_TOK_DIV_EQ:
1555                 dt_cg_arithmetic_op(dnp, dlp, drp,
1556                     (dnp->dn_flags & DT_NF_SIGNED) ? DIF_OP_SDIV : DIF_OP_UDIV);
1557                 dt_cg_asgn_op(dnp, dlp, drp);
1558                 break;
1559 
1560         case DT_TOK_MOD_EQ:
1561                 dt_cg_arithmetic_op(dnp, dlp, drp,
1562                     (dnp->dn_flags & DT_NF_SIGNED) ? DIF_OP_SREM : DIF_OP_UREM);
1563                 dt_cg_asgn_op(dnp, dlp, drp);
1564                 break;
1565 
1566         case DT_TOK_AND_EQ:
1567                 dt_cg_arithmetic_op(dnp, dlp, drp, DIF_OP_AND);
1568                 dt_cg_asgn_op(dnp, dlp, drp);
1569                 break;
1570 
1571         case DT_TOK_XOR_EQ:
1572                 dt_cg_arithmetic_op(dnp, dlp, drp, DIF_OP_XOR);
1573                 dt_cg_asgn_op(dnp, dlp, drp);
1574                 break;
1575 
1576         case DT_TOK_OR_EQ:
1577                 dt_cg_arithmetic_op(dnp, dlp, drp, DIF_OP_OR);
1578                 dt_cg_asgn_op(dnp, dlp, drp);
1579                 break;
1580 
1581         case DT_TOK_LSH_EQ:
1582                 dt_cg_arithmetic_op(dnp, dlp, drp, DIF_OP_SLL);
1583                 dt_cg_asgn_op(dnp, dlp, drp);
1584                 break;
1585 
1586         case DT_TOK_RSH_EQ:
1587                 dt_cg_arithmetic_op(dnp, dlp, drp,
1588                     (dnp->dn_flags & DT_NF_SIGNED) ? DIF_OP_SRA : DIF_OP_SRL);
1589                 dt_cg_asgn_op(dnp, dlp, drp);
1590                 break;
1591 
1592         case DT_TOK_QUESTION:
1593                 dt_cg_ternary_op(dnp, dlp, drp);
1594                 break;
1595 
1596         case DT_TOK_LOR:
1597                 dt_cg_logical_or(dnp, dlp, drp);
1598                 break;
1599 
1600         case DT_TOK_LXOR:
1601                 dt_cg_logical_xor(dnp, dlp, drp);
1602                 break;
1603 
1604         case DT_TOK_LAND:
1605                 dt_cg_logical_and(dnp, dlp, drp);
1606                 break;
1607 
1608         case DT_TOK_BOR:
1609                 dt_cg_arithmetic_op(dnp, dlp, drp, DIF_OP_OR);
1610                 break;
1611 
1612         case DT_TOK_XOR:
1613                 dt_cg_arithmetic_op(dnp, dlp, drp, DIF_OP_XOR);
1614                 break;
1615 
1616         case DT_TOK_BAND:
1617                 dt_cg_arithmetic_op(dnp, dlp, drp, DIF_OP_AND);
1618                 break;
1619 
1620         case DT_TOK_EQU:
1621                 dt_cg_compare_op(dnp, dlp, drp, DIF_OP_BE);
1622                 break;
1623 
1624         case DT_TOK_NEQ:
1625                 dt_cg_compare_op(dnp, dlp, drp, DIF_OP_BNE);
1626                 break;
1627 
1628         case DT_TOK_LT:
1629                 dt_cg_compare_op(dnp, dlp, drp,
1630                     dt_cg_compare_signed(dnp) ? DIF_OP_BL : DIF_OP_BLU);
1631                 break;
1632 
1633         case DT_TOK_LE:
1634                 dt_cg_compare_op(dnp, dlp, drp,
1635                     dt_cg_compare_signed(dnp) ? DIF_OP_BLE : DIF_OP_BLEU);
1636                 break;
1637 
1638         case DT_TOK_GT:
1639                 dt_cg_compare_op(dnp, dlp, drp,
1640                     dt_cg_compare_signed(dnp) ? DIF_OP_BG : DIF_OP_BGU);
1641                 break;
1642 
1643         case DT_TOK_GE:
1644                 dt_cg_compare_op(dnp, dlp, drp,
1645                     dt_cg_compare_signed(dnp) ? DIF_OP_BGE : DIF_OP_BGEU);
1646                 break;
1647 
1648         case DT_TOK_LSH:
1649                 dt_cg_arithmetic_op(dnp, dlp, drp, DIF_OP_SLL);
1650                 break;
1651 
1652         case DT_TOK_RSH:
1653                 dt_cg_arithmetic_op(dnp, dlp, drp,
1654                     (dnp->dn_flags & DT_NF_SIGNED) ? DIF_OP_SRA : DIF_OP_SRL);
1655                 break;
1656 
1657         case DT_TOK_ADD:
1658                 dt_cg_arithmetic_op(dnp, dlp, drp, DIF_OP_ADD);
1659                 break;
1660 
1661         case DT_TOK_SUB:
1662                 dt_cg_arithmetic_op(dnp, dlp, drp, DIF_OP_SUB);
1663                 break;
1664 
1665         case DT_TOK_MUL:
1666                 dt_cg_arithmetic_op(dnp, dlp, drp, DIF_OP_MUL);
1667                 break;
1668 
1669         case DT_TOK_DIV:
1670                 dt_cg_arithmetic_op(dnp, dlp, drp,
1671                     (dnp->dn_flags & DT_NF_SIGNED) ? DIF_OP_SDIV : DIF_OP_UDIV);
1672                 break;
1673 
1674         case DT_TOK_MOD:
1675                 dt_cg_arithmetic_op(dnp, dlp, drp,
1676                     (dnp->dn_flags & DT_NF_SIGNED) ? DIF_OP_SREM : DIF_OP_UREM);
1677                 break;
1678 
1679         case DT_TOK_LNEG:
1680                 dt_cg_logical_neg(dnp, dlp, drp);
1681                 break;
1682 
1683         case DT_TOK_BNEG:
1684                 dt_cg_node(dnp->dn_child, dlp, drp);
1685                 dnp->dn_reg = dnp->dn_child->dn_reg;
1686                 instr = DIF_INSTR_NOT(dnp->dn_reg, dnp->dn_reg);
1687                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
1688                 break;
1689 
1690         case DT_TOK_PREINC:
1691                 dt_cg_prearith_op(dnp, dlp, drp, DIF_OP_ADD);
1692                 break;
1693 
1694         case DT_TOK_POSTINC:
1695                 dt_cg_postarith_op(dnp, dlp, drp, DIF_OP_ADD);
1696                 break;
1697 
1698         case DT_TOK_PREDEC:
1699                 dt_cg_prearith_op(dnp, dlp, drp, DIF_OP_SUB);
1700                 break;
1701 
1702         case DT_TOK_POSTDEC:
1703                 dt_cg_postarith_op(dnp, dlp, drp, DIF_OP_SUB);
1704                 break;
1705 
1706         case DT_TOK_IPOS:
1707                 dt_cg_node(dnp->dn_child, dlp, drp);
1708                 dnp->dn_reg = dnp->dn_child->dn_reg;
1709                 break;
1710 
1711         case DT_TOK_INEG:
1712                 dt_cg_node(dnp->dn_child, dlp, drp);
1713                 dnp->dn_reg = dnp->dn_child->dn_reg;
1714 
1715                 instr = DIF_INSTR_FMT(DIF_OP_SUB, DIF_REG_R0,
1716                     dnp->dn_reg, dnp->dn_reg);
1717 
1718                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
1719                 break;
1720 
1721         case DT_TOK_DEREF:
1722                 dt_cg_node(dnp->dn_child, dlp, drp);
1723                 dnp->dn_reg = dnp->dn_child->dn_reg;
1724 
1725                 if (dt_node_is_dynamic(dnp->dn_child)) {
1726                         int reg;
1727                         idp = dt_node_resolve(dnp->dn_child, DT_IDENT_XLPTR);
1728                         assert(idp != NULL);
1729                         reg = dt_cg_xlate_expand(dnp, idp, dlp, drp);
1730 
1731                         dt_regset_free(drp, dnp->dn_child->dn_reg);
1732                         dnp->dn_reg = reg;
1733 
1734                 } else if (!(dnp->dn_flags & DT_NF_REF)) {
1735                         uint_t ubit = dnp->dn_flags & DT_NF_USERLAND;
1736 
1737                         /*
1738                          * Save and restore DT_NF_USERLAND across dt_cg_load():
1739                          * we need the sign bit from dnp and the user bit from
1740                          * dnp->dn_child in order to get the proper opcode.
1741                          */
1742                         dnp->dn_flags |=
1743                             (dnp->dn_child->dn_flags & DT_NF_USERLAND);
1744 
1745                         instr = DIF_INSTR_LOAD(dt_cg_load(dnp, ctfp,
1746                             dnp->dn_type), dnp->dn_reg, dnp->dn_reg);
1747 
1748                         dnp->dn_flags &= ~DT_NF_USERLAND;
1749                         dnp->dn_flags |= ubit;
1750 
1751                         dt_irlist_append(dlp,
1752                             dt_cg_node_alloc(DT_LBL_NONE, instr));
1753                 }
1754                 break;
1755 
1756         case DT_TOK_ADDROF: {
1757                 uint_t rbit = dnp->dn_child->dn_flags & DT_NF_REF;
1758 
1759                 dnp->dn_child->dn_flags |= DT_NF_REF; /* force pass-by-ref */
1760                 dt_cg_node(dnp->dn_child, dlp, drp);
1761                 dnp->dn_reg = dnp->dn_child->dn_reg;
1762 
1763                 dnp->dn_child->dn_flags &= ~DT_NF_REF;
1764                 dnp->dn_child->dn_flags |= rbit;
1765                 break;
1766         }
1767 
1768         case DT_TOK_SIZEOF: {
1769                 size_t size = dt_node_sizeof(dnp->dn_child);
1770                 dnp->dn_reg = dt_regset_alloc(drp);
1771                 assert(size != 0);
1772                 dt_cg_setx(dlp, dnp->dn_reg, size);
1773                 break;
1774         }
1775 
1776         case DT_TOK_STRINGOF:
1777                 dt_cg_node(dnp->dn_child, dlp, drp);
1778                 dnp->dn_reg = dnp->dn_child->dn_reg;
1779                 break;
1780 
1781         case DT_TOK_XLATE:
1782                 /*
1783                  * An xlate operator appears in either an XLATOR, indicating a
1784                  * reference to a dynamic translator, or an OP2, indicating
1785                  * use of the xlate operator in the user's program.  For the
1786                  * dynamic case, generate an xlate opcode with a reference to
1787                  * the corresponding member, pre-computed for us in dn_members.
1788                  */
1789                 if (dnp->dn_kind == DT_NODE_XLATOR) {
1790                         dt_xlator_t *dxp = dnp->dn_xlator;
1791 
1792                         assert(dxp->dx_ident->di_flags & DT_IDFLG_CGREG);
1793                         assert(dxp->dx_ident->di_id != 0);
1794 
1795                         dnp->dn_reg = dt_regset_alloc(drp);
1796 
1797                         if (dxp->dx_arg == -1) {
1798                                 instr = DIF_INSTR_MOV(
1799                                     dxp->dx_ident->di_id, dnp->dn_reg);
1800                                 dt_irlist_append(dlp,
1801                                     dt_cg_node_alloc(DT_LBL_NONE, instr));
1802                                 op = DIF_OP_XLATE;
1803                         } else
1804                                 op = DIF_OP_XLARG;
1805 
1806                         instr = DIF_INSTR_XLATE(op, 0, dnp->dn_reg);
1807                         dt_irlist_append(dlp,
1808                             dt_cg_node_alloc(DT_LBL_NONE, instr));
1809 
1810                         dlp->dl_last->di_extern = dnp->dn_xmember;
1811                         break;
1812                 }
1813 
1814                 assert(dnp->dn_kind == DT_NODE_OP2);
1815                 dt_cg_node(dnp->dn_right, dlp, drp);
1816                 dnp->dn_reg = dnp->dn_right->dn_reg;
1817                 break;
1818 
1819         case DT_TOK_LPAR:
1820                 dt_cg_node(dnp->dn_right, dlp, drp);
1821                 dnp->dn_reg = dnp->dn_right->dn_reg;
1822                 dt_cg_typecast(dnp->dn_right, dnp, dlp, drp);
1823                 break;
1824 
1825         case DT_TOK_PTR:
1826         case DT_TOK_DOT:
1827                 assert(dnp->dn_right->dn_kind == DT_NODE_IDENT);
1828                 dt_cg_node(dnp->dn_left, dlp, drp);
1829 
1830                 /*
1831                  * If the left-hand side of PTR or DOT is a dynamic variable,
1832                  * we expect it to be the output of a D translator.   In this
1833                  * case, we look up the parse tree corresponding to the member
1834                  * that is being accessed and run the code generator over it.
1835                  * We then cast the result as if by the assignment operator.
1836                  */
1837                 if ((idp = dt_node_resolve(
1838                     dnp->dn_left, DT_IDENT_XLSOU)) != NULL ||
1839                     (idp = dt_node_resolve(
1840                     dnp->dn_left, DT_IDENT_XLPTR)) != NULL) {
1841 
1842                         dt_xlator_t *dxp;
1843                         dt_node_t *mnp;
1844 
1845                         dxp = idp->di_data;
1846                         mnp = dt_xlator_member(dxp, dnp->dn_right->dn_string);
1847                         assert(mnp != NULL);
1848 
1849                         dxp->dx_ident->di_flags |= DT_IDFLG_CGREG;
1850                         dxp->dx_ident->di_id = dnp->dn_left->dn_reg;
1851 
1852                         dt_cg_node(mnp->dn_membexpr, dlp, drp);
1853                         dnp->dn_reg = mnp->dn_membexpr->dn_reg;
1854                         dt_cg_typecast(mnp->dn_membexpr, dnp, dlp, drp);
1855 
1856                         dxp->dx_ident->di_flags &= ~DT_IDFLG_CGREG;
1857                         dxp->dx_ident->di_id = 0;
1858 
1859                         if (dnp->dn_left->dn_reg != -1)
1860                                 dt_regset_free(drp, dnp->dn_left->dn_reg);
1861                         break;
1862                 }
1863 
1864                 ctfp = dnp->dn_left->dn_ctfp;
1865                 type = ctf_type_resolve(ctfp, dnp->dn_left->dn_type);
1866 
1867                 if (dnp->dn_op == DT_TOK_PTR) {
1868                         type = ctf_type_reference(ctfp, type);
1869                         type = ctf_type_resolve(ctfp, type);
1870                 }
1871 
1872                 if ((ctfp = dt_cg_membinfo(octfp = ctfp, type,
1873                     dnp->dn_right->dn_string, &m)) == NULL) {
1874                         yypcb->pcb_hdl->dt_ctferr = ctf_errno(octfp);
1875                         longjmp(yypcb->pcb_jmpbuf, EDT_CTF);
1876                 }
1877 
1878                 if (m.ctm_offset != 0) {
1879                         int reg;
1880 
1881                         reg = dt_regset_alloc(drp);
1882 
1883                         /*
1884                          * If the offset is not aligned on a byte boundary, it
1885                          * is a bit-field member and we will extract the value
1886                          * bits below after we generate the appropriate load.
1887                          */
1888                         dt_cg_setx(dlp, reg, m.ctm_offset / NBBY);
1889 
1890                         instr = DIF_INSTR_FMT(DIF_OP_ADD,
1891                             dnp->dn_left->dn_reg, reg, dnp->dn_left->dn_reg);
1892 
1893                         dt_irlist_append(dlp,
1894                             dt_cg_node_alloc(DT_LBL_NONE, instr));
1895                         dt_regset_free(drp, reg);
1896                 }
1897 
1898                 if (!(dnp->dn_flags & DT_NF_REF)) {
1899                         uint_t ubit = dnp->dn_flags & DT_NF_USERLAND;
1900 
1901                         /*
1902                          * Save and restore DT_NF_USERLAND across dt_cg_load():
1903                          * we need the sign bit from dnp and the user bit from
1904                          * dnp->dn_left in order to get the proper opcode.
1905                          */
1906                         dnp->dn_flags |=
1907                             (dnp->dn_left->dn_flags & DT_NF_USERLAND);
1908 
1909                         instr = DIF_INSTR_LOAD(dt_cg_load(dnp,
1910                             ctfp, m.ctm_type), dnp->dn_left->dn_reg,
1911                             dnp->dn_left->dn_reg);
1912 
1913                         dnp->dn_flags &= ~DT_NF_USERLAND;
1914                         dnp->dn_flags |= ubit;
1915 
1916                         dt_irlist_append(dlp,
1917                             dt_cg_node_alloc(DT_LBL_NONE, instr));
1918 
1919                         if (dnp->dn_flags & DT_NF_BITFIELD)
1920                                 dt_cg_field_get(dnp, dlp, drp, ctfp, &m);
1921                 }
1922 
1923                 dnp->dn_reg = dnp->dn_left->dn_reg;
1924                 break;
1925 
1926         case DT_TOK_STRING:
1927                 dnp->dn_reg = dt_regset_alloc(drp);
1928 
1929                 assert(dnp->dn_kind == DT_NODE_STRING);
1930                 stroff = dt_strtab_insert(yypcb->pcb_strtab, dnp->dn_string);
1931 
1932                 if (stroff == -1L)
1933                         longjmp(yypcb->pcb_jmpbuf, EDT_NOMEM);
1934                 if (stroff > DIF_STROFF_MAX)
1935                         longjmp(yypcb->pcb_jmpbuf, EDT_STR2BIG);
1936 
1937                 instr = DIF_INSTR_SETS((ulong_t)stroff, dnp->dn_reg);
1938                 dt_irlist_append(dlp, dt_cg_node_alloc(DT_LBL_NONE, instr));
1939                 break;
1940 
1941         case DT_TOK_IDENT:
1942                 /*
1943                  * If the specified identifier is a variable on which we have
1944                  * set the code generator register flag, then this variable
1945                  * has already had code generated for it and saved in di_id.
1946                  * Allocate a new register and copy the existing value to it.
1947                  */
1948                 if (dnp->dn_kind == DT_NODE_VAR &&
1949                     (dnp->dn_ident->di_flags & DT_IDFLG_CGREG)) {
1950                         dnp->dn_reg = dt_regset_alloc(drp);
1951                         instr = DIF_INSTR_MOV(dnp->dn_ident->di_id,
1952                             dnp->dn_reg);
1953                         dt_irlist_append(dlp,
1954                             dt_cg_node_alloc(DT_LBL_NONE, instr));
1955                         break;
1956                 }
1957 
1958                 /*
1959                  * Identifiers can represent function calls, variable refs, or
1960                  * symbols.  First we check for inlined variables, and handle
1961                  * them by generating code for the inline parse tree.
1962                  */
1963                 if (dnp->dn_kind == DT_NODE_VAR &&
1964                     (dnp->dn_ident->di_flags & DT_IDFLG_INLINE)) {
1965                         dt_cg_inline(dnp, dlp, drp);
1966                         break;
1967                 }
1968 
1969                 switch (dnp->dn_kind) {
1970                 case DT_NODE_FUNC:
1971                         if ((idp = dnp->dn_ident)->di_kind != DT_IDENT_FUNC) {
1972                                 dnerror(dnp, D_CG_EXPR, "%s %s( ) may not be "
1973                                     "called from a D expression (D program "
1974                                     "context required)\n",
1975                                     dt_idkind_name(idp->di_kind), idp->di_name);
1976                         }
1977 
1978                         dt_cg_arglist(dnp->dn_ident, dnp->dn_args, dlp, drp);
1979 
1980                         dnp->dn_reg = dt_regset_alloc(drp);
1981                         instr = DIF_INSTR_CALL(dnp->dn_ident->di_id,
1982                             dnp->dn_reg);
1983 
1984                         dt_irlist_append(dlp,
1985                             dt_cg_node_alloc(DT_LBL_NONE, instr));
1986 
1987                         break;
1988 
1989                 case DT_NODE_VAR:
1990                         if (dnp->dn_ident->di_kind == DT_IDENT_XLSOU ||
1991                             dnp->dn_ident->di_kind == DT_IDENT_XLPTR) {
1992                                 /*
1993                                  * This can only happen if we have translated
1994                                  * args[].  See dt_idcook_args() for details.
1995                                  */
1996                                 assert(dnp->dn_ident->di_id == DIF_VAR_ARGS);
1997                                 dt_cg_array_op(dnp, dlp, drp);
1998                                 break;
1999                         }
2000 
2001                         if (dnp->dn_ident->di_kind == DT_IDENT_ARRAY) {
2002                                 if (dnp->dn_ident->di_id > DIF_VAR_ARRAY_MAX)
2003                                         dt_cg_assoc_op(dnp, dlp, drp);
2004                                 else
2005                                         dt_cg_array_op(dnp, dlp, drp);
2006                                 break;
2007                         }
2008 
2009                         dnp->dn_reg = dt_regset_alloc(drp);
2010 
2011                         if (dnp->dn_ident->di_flags & DT_IDFLG_LOCAL)
2012                                 op = DIF_OP_LDLS;
2013                         else if (dnp->dn_ident->di_flags & DT_IDFLG_TLS)
2014                                 op = DIF_OP_LDTS;
2015                         else
2016                                 op = DIF_OP_LDGS;
2017 
2018                         dnp->dn_ident->di_flags |= DT_IDFLG_DIFR;
2019 
2020                         instr = DIF_INSTR_LDV(op,
2021                             dnp->dn_ident->di_id, dnp->dn_reg);
2022 
2023                         dt_irlist_append(dlp,
2024                             dt_cg_node_alloc(DT_LBL_NONE, instr));
2025                         break;
2026 
2027                 case DT_NODE_SYM: {
2028                         dtrace_hdl_t *dtp = yypcb->pcb_hdl;
2029                         dtrace_syminfo_t *sip = dnp->dn_ident->di_data;
2030                         GElf_Sym sym;
2031 
2032                         if (dtrace_lookup_by_name(dtp,
2033                             sip->dts_object, sip->dts_name, &sym, NULL) == -1) {
2034                                 xyerror(D_UNKNOWN, "cg failed for symbol %s`%s:"
2035                                     " %s\n", sip->dts_object, sip->dts_name,
2036                                     dtrace_errmsg(dtp, dtrace_errno(dtp)));
2037                         }
2038 
2039                         dnp->dn_reg = dt_regset_alloc(drp);
2040                         dt_cg_xsetx(dlp, dnp->dn_ident,
2041                             DT_LBL_NONE, dnp->dn_reg, sym.st_value);
2042 
2043                         if (!(dnp->dn_flags & DT_NF_REF)) {
2044                                 instr = DIF_INSTR_LOAD(dt_cg_load(dnp, ctfp,
2045                                     dnp->dn_type), dnp->dn_reg, dnp->dn_reg);
2046                                 dt_irlist_append(dlp,
2047                                     dt_cg_node_alloc(DT_LBL_NONE, instr));
2048                         }
2049                         break;
2050                 }
2051 
2052                 default:
2053                         xyerror(D_UNKNOWN, "internal error -- node type %u is "
2054                             "not valid for an identifier\n", dnp->dn_kind);
2055                 }
2056                 break;
2057 
2058         case DT_TOK_INT:
2059                 dnp->dn_reg = dt_regset_alloc(drp);
2060                 dt_cg_setx(dlp, dnp->dn_reg, dnp->dn_value);
2061                 break;
2062 
2063         default:
2064                 xyerror(D_UNKNOWN, "internal error -- token type %u is not a "
2065                     "valid D compilation token\n", dnp->dn_op);
2066         }
2067 }
2068 
2069 void
2070 dt_cg(dt_pcb_t *pcb, dt_node_t *dnp)
2071 {
2072         dif_instr_t instr;
2073         dt_xlator_t *dxp;
2074         dt_ident_t *idp;
2075 
2076         if (pcb->pcb_regs == NULL && (pcb->pcb_regs =
2077             dt_regset_create(pcb->pcb_hdl->dt_conf.dtc_difintregs)) == NULL)
2078                 longjmp(pcb->pcb_jmpbuf, EDT_NOMEM);
2079 
2080         dt_regset_reset(pcb->pcb_regs);
2081         (void) dt_regset_alloc(pcb->pcb_regs); /* allocate %r0 */
2082 
2083         if (pcb->pcb_inttab != NULL)
2084                 dt_inttab_destroy(pcb->pcb_inttab);
2085 
2086         if ((pcb->pcb_inttab = dt_inttab_create(yypcb->pcb_hdl)) == NULL)
2087                 longjmp(pcb->pcb_jmpbuf, EDT_NOMEM);
2088 
2089         if (pcb->pcb_strtab != NULL)
2090                 dt_strtab_destroy(pcb->pcb_strtab);
2091 
2092         if ((pcb->pcb_strtab = dt_strtab_create(BUFSIZ)) == NULL)
2093                 longjmp(pcb->pcb_jmpbuf, EDT_NOMEM);
2094 
2095         dt_irlist_destroy(&pcb->pcb_ir);
2096         dt_irlist_create(&pcb->pcb_ir);
2097 
2098         assert(pcb->pcb_dret == NULL);
2099         pcb->pcb_dret = dnp;
2100 
2101         if (dt_node_resolve(dnp, DT_IDENT_XLPTR) != NULL) {
2102                 dnerror(dnp, D_CG_DYN, "expression cannot evaluate to result "
2103                     "of a translated pointer\n");
2104         }
2105 
2106         /*
2107          * If we're generating code for a translator body, assign the input
2108          * parameter to the first available register (i.e. caller passes %r1).
2109          */
2110         if (dnp->dn_kind == DT_NODE_MEMBER) {
2111                 dxp = dnp->dn_membxlator;
2112                 dnp = dnp->dn_membexpr;
2113 
2114                 dxp->dx_ident->di_flags |= DT_IDFLG_CGREG;
2115                 dxp->dx_ident->di_id = dt_regset_alloc(pcb->pcb_regs);
2116         }
2117 
2118         dt_cg_node(dnp, &pcb->pcb_ir, pcb->pcb_regs);
2119 
2120         if ((idp = dt_node_resolve(dnp, DT_IDENT_XLSOU)) != NULL) {
2121                 int reg = dt_cg_xlate_expand(dnp, idp,
2122                     &pcb->pcb_ir, pcb->pcb_regs);
2123                 dt_regset_free(pcb->pcb_regs, dnp->dn_reg);
2124                 dnp->dn_reg = reg;
2125         }
2126 
2127         instr = DIF_INSTR_RET(dnp->dn_reg);
2128         dt_regset_free(pcb->pcb_regs, dnp->dn_reg);
2129         dt_irlist_append(&pcb->pcb_ir, dt_cg_node_alloc(DT_LBL_NONE, instr));
2130 
2131         if (dnp->dn_kind == DT_NODE_MEMBER) {
2132                 dt_regset_free(pcb->pcb_regs, dxp->dx_ident->di_id);
2133                 dxp->dx_ident->di_id = 0;
2134                 dxp->dx_ident->di_flags &= ~DT_IDFLG_CGREG;
2135         }
2136 
2137         dt_regset_free(pcb->pcb_regs, 0);
2138         dt_regset_assert_free(pcb->pcb_regs);
2139 }