assemble.c

   1 /* ----------------------------------------------------------------------- *
   2  *
   3  *   Copyright 1996-2013 The NASM Authors - All Rights Reserved
   4  *   See the file AUTHORS included with the NASM distribution for
   5  *   the specific copyright holders.
   6  *
   7  *   Redistribution and use in source and binary forms, with or without
   8  *   modification, are permitted provided that the following
   9  *   conditions are met:
  10  *
  11  *   * Redistributions of source code must retain the above copyright
  12  *     notice, this list of conditions and the following disclaimer.
  13  *   * Redistributions in binary form must reproduce the above
  14  *     copyright notice, this list of conditions and the following
  15  *     disclaimer in the documentation and/or other materials provided
  16  *     with the distribution.
  17  *
  18  *     THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
  19  *     CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
  20  *     INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
  21  *     MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  22  *     DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
  23  *     CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  24  *     SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  25  *     NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  26  *     LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  27  *     HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  28  *     CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
  29  *     OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
  30  *     EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  31  *
  32  * ----------------------------------------------------------------------- */
  33
  34 /*
  35  * assemble.c   code generation for the Netwide Assembler
  36  *
  37  * the actual codes (C syntax, i.e. octal):
  38  * \0            - terminates the code. (Unless it's a literal of course.)
  39  * \1..\4        - that many literal bytes follow in the code stream
  40  * \5            - add 4 to the primary operand number (b, low octdigit)
  41  * \6            - add 4 to the secondary operand number (a, middle octdigit)
  42  * \7            - add 4 to both the primary and the secondary operand number
  43  * \10..\13      - a literal byte follows in the code stream, to be added
  44  *                 to the register value of operand 0..3
  45  * \20..\23      - a byte immediate operand, from operand 0..3
  46  * \24..\27      - a zero-extended byte immediate operand, from operand 0..3
  47  * \30..\33      - a word immediate operand, from operand 0..3
  48  * \34..\37      - select between \3[0-3] and \4[0-3] depending on 16/32 bit
  49  *                 assembly mode or the operand-size override on the operand
  50  * \40..\43      - a long immediate operand, from operand 0..3
  51  * \44..\47      - select between \3[0-3], \4[0-3] and \5[4-7]
  52  *                 depending on the address size of the instruction.
  53  * \50..\53      - a byte relative operand, from operand 0..3
  54  * \54..\57      - a qword immediate operand, from operand 0..3
  55  * \60..\63      - a word relative operand, from operand 0..3
  56  * \64..\67      - select between \6[0-3] and \7[0-3] depending on 16/32 bit
  57  *                 assembly mode or the operand-size override on the operand
  58  * \70..\73      - a long relative operand, from operand 0..3
  59  * \74..\77      - a word constant, from the _segment_ part of operand 0..3
  60  * \1ab          - a ModRM, calculated on EA in operand a, with the spare
  61  *                 field the register value of operand b.
  62  * \172\ab       - the register number from operand a in bits 7..4, with
  63  *                 the 4-bit immediate from operand b in bits 3..0.
  64  * \173\xab      - the register number from operand a in bits 7..4, with
  65  *                 the value b in bits 3..0.
  66  * \174..\177    - the register number from operand 0..3 in bits 7..4, and
  67  *                 an arbitrary value in bits 3..0 (assembled as zero.)
  68  * \2ab          - a ModRM, calculated on EA in operand a, with the spare
  69  *                 field equal to digit b.
  70  *
  71  * \240..\243    - this instruction uses EVEX rather than REX or VEX/XOP, with the
  72  *                 V field taken from operand 0..3.
  73  * \250          - this instruction uses EVEX rather than REX or VEX/XOP, with the
  74  *                 V field set to 1111b.
  75  * EVEX prefixes are followed by the sequence:
  76  * \cm\wlp\tup    where cm is:
  77  *                  cc 000 0mm
  78  *                  c = 2 for EVEX and m is the legacy escape (0f, 0f38, 0f3a)
  79  *                and wlp is:
  80  *                  00 wwl lpp
  81  *                  [l0]  ll = 0 (.128, .lz)
  82  *                  [l1]  ll = 1 (.256)
  83  *                  [l2]  ll = 2 (.512)
  84  *                  [lig] ll = 3 for EVEX.L'L don't care (always assembled as 0)
  85  *
  86  *                  [w0]  ww = 0 for W = 0
  87  *                  [w1]  ww = 1 for W = 1
  88  *                  [wig] ww = 2 for W don't care (always assembled as 0)
  89  *                  [ww]  ww = 3 for W used as REX.W
  90  *
  91  *                  [p0]  pp = 0 for no prefix
  92  *                  [60]  pp = 1 for legacy prefix 60
  93  *                  [f3]  pp = 2
  94  *                  [f2]  pp = 3
  95  *
  96  *                tup is tuple type for Disp8*N from %tuple_codes in insns.pl
  97  *                    (compressed displacement encoding)
  98  *
  99  * \254..\257    - a signed 32-bit operand to be extended to 64 bits.
 100  * \260..\263    - this instruction uses VEX/XOP rather than REX, with the
 101  *                 V field taken from operand 0..3.
 102  * \270          - this instruction uses VEX/XOP rather than REX, with the
 103  *                 V field set to 1111b.
 104  *
 105  * VEX/XOP prefixes are followed by the sequence:
 106  * \tmm\wlp        where mm is the M field; and wlp is:
 107  *                 00 wwl lpp
 108  *                 [l0]  ll = 0 for L = 0 (.128, .lz)
 109  *                 [l1]  ll = 1 for L = 1 (.256)
 110  *                 [lig] ll = 2 for L don't care (always assembled as 0)
 111  *
 112  *                 [w0]  ww = 0 for W = 0
 113  *                 [w1 ] ww = 1 for W = 1
 114  *                 [wig] ww = 2 for W don't care (always assembled as 0)
 115  *                 [ww]  ww = 3 for W used as REX.W
 116  *
 117  * t = 0 for VEX (C4/C5), t = 1 for XOP (8F).
 118  *
 119  * \271          - instruction takes XRELEASE (F3) with or without lock
 120  * \272          - instruction takes XACQUIRE/XRELEASE with or without lock
 121  * \273          - instruction takes XACQUIRE/XRELEASE with lock only
 122  * \274..\277    - a byte immediate operand, from operand 0..3, sign-extended
 123  *                 to the operand size (if o16/o32/o64 present) or the bit size
 124  * \310          - indicates fixed 16-bit address size, i.e. optional 0x67.
 125  * \311          - indicates fixed 32-bit address size, i.e. optional 0x67.
 126  * \312          - (disassembler only) invalid with non-default address size.
 127  * \313          - indicates fixed 64-bit address size, 0x67 invalid.
 128  * \314          - (disassembler only) invalid with REX.B
 129  * \315          - (disassembler only) invalid with REX.X
 130  * \316          - (disassembler only) invalid with REX.R
 131  * \317          - (disassembler only) invalid with REX.W
 132  * \320          - indicates fixed 16-bit operand size, i.e. optional 0x66.
 133  * \321          - indicates fixed 32-bit operand size, i.e. optional 0x66.
 134  * \322          - indicates that this instruction is only valid when the
 135  *                 operand size is the default (instruction to disassembler,
 136  *                 generates no code in the assembler)
 137  * \323          - indicates fixed 64-bit operand size, REX on extensions only.
 138  * \324          - indicates 64-bit operand size requiring REX prefix.
 139  * \325          - instruction which always uses spl/bpl/sil/dil
 140  * \326          - instruction not valid with 0xF3 REP prefix.  Hint for
 141                    disassembler only; for SSE instructions.
 142  * \330          - a literal byte follows in the code stream, to be added
 143  *                 to the condition code value of the instruction.
 144  * \331          - instruction not valid with REP prefix.  Hint for
 145  *                 disassembler only; for SSE instructions.
 146  * \332          - REP prefix (0xF2 byte) used as opcode extension.
 147  * \333          - REP prefix (0xF3 byte) used as opcode extension.
 148  * \334          - LOCK prefix used as REX.R (used in non-64-bit mode)
 149  * \335          - disassemble a rep (0xF3 byte) prefix as repe not rep.
 150  * \336          - force a REP(E) prefix (0xF3) even if not specified.
 151  * \337          - force a REPNE prefix (0xF2) even if not specified.
 152  *                 \336-\337 are still listed as prefixes in the disassembler.
 153  * \340          - reserve <operand 0> bytes of uninitialized storage.
 154  *                 Operand 0 had better be a segmentless constant.
 155  * \341          - this instruction needs a WAIT "prefix"
 156  * \360          - no SSE prefix (== \364\331)
 157  * \361          - 66 SSE prefix (== \366\331)
 158  * \364          - operand-size prefix (0x66) not permitted
 159  * \365          - address-size prefix (0x67) not permitted
 160  * \366          - operand-size prefix (0x66) used as opcode extension
 161  * \367          - address-size prefix (0x67) used as opcode extension
 162  * \370,\371     - match only if operand 0 meets byte jump criteria.
 163  *                 370 is used for Jcc, 371 is used for JMP.
 164  * \373          - assemble 0x03 if bits==16, 0x05 if bits==32;
 165  *                 used for conditional jump over longer jump
 166  * \374          - this instruction takes an XMM VSIB memory EA
 167  * \375          - this instruction takes an YMM VSIB memory EA
 168  * \376          - this instruction takes an ZMM VSIB memory EA
 169  */
 170
 171 #include "compiler.h"
 172
 173 #include <stdio.h>
 174 #include <string.h>
 175 #include <inttypes.h>
 176
 177 #include "nasm.h"
 178 #include "nasmlib.h"
 179 #include "assemble.h"
 180 #include "insns.h"
 181 #include "tables.h"
 182
 183 enum match_result {
 184     /*
 185      * Matching errors.  These should be sorted so that more specific
 186      * errors come later in the sequence.
 187      */
 188     MERR_INVALOP,
 189     MERR_OPSIZEMISSING,
 190     MERR_OPSIZEMISMATCH,
 191     MERR_BADCPU,
 192     MERR_BADMODE,
 193     MERR_BADHLE,
 194     /*
 195      * Matching success; the conditional ones first
 196      */
 197     MOK_JUMP,   /* Matching OK but needs jmp_match() */
 198     MOK_GOOD    /* Matching unconditionally OK */
 199 };
 200
 201 typedef struct {
 202     enum ea_type type;            /* what kind of EA is this? */
 203     int sib_present;              /* is a SIB byte necessary? */
 204     int bytes;                    /* # of bytes of offset needed */
 205     int size;                     /* lazy - this is sib+bytes+1 */
 206     uint8_t modrm, sib, rex, rip; /* the bytes themselves */
 207     int8_t disp8;                  /* compressed displacement for EVEX */
 208 } ea;
 209
 210 #define GEN_SIB(scale, index, base)                 \
 211         (((scale) << 6) | ((index) << 3) | ((base)))
 212
 213 #define GEN_MODRM(mod, reg, rm)                     \
 214         (((mod) << 6) | (((reg) & 7) << 3) | ((rm) & 7))
 215
 216 static uint32_t cpu;            /* cpu level received from nasm.c */
 217 static efunc errfunc;
 218 static struct ofmt *outfmt;
 219 static ListGen *list;
 220
 221 static int64_t calcsize(int32_t, int64_t, int, insn *,
 222                         const struct itemplate *);
 223 static void gencode(int32_t segment, int64_t offset, int bits,
 224                     insn * ins, const struct itemplate *temp,
 225                     int64_t insn_end);
 226 static enum match_result find_match(const struct itemplate **tempp,
 227                                     insn *instruction,
 228                                     int32_t segment, int64_t offset, int bits);
 229 static enum match_result matches(const struct itemplate *, insn *, int bits);
 230 static opflags_t regflag(const operand *);
 231 static int32_t regval(const operand *);
 232 static int rexflags(int, opflags_t, int);
 233 static int op_rexflags(const operand *, int);
 234 static int op_evexflags(const operand *, int, uint8_t);
 235 static void add_asp(insn *, int);
 236
 237 static enum ea_type process_ea(operand *, ea *, int, int, opflags_t, insn *);
 238
 239 static int has_prefix(insn * ins, enum prefix_pos pos, int prefix)
 240 {
 241     return ins->prefixes[pos] == prefix;
 242 }
 243
 244 static void assert_no_prefix(insn * ins, enum prefix_pos pos)
 245 {
 246     if (ins->prefixes[pos])
 247         errfunc(ERR_NONFATAL, "invalid %s prefix",
 248                 prefix_name(ins->prefixes[pos]));
 249 }
 250
 251 static const char *size_name(int size)
 252 {
 253     switch (size) {
 254     case 1:
 255         return "byte";
 256     case 2:
 257         return "word";
 258     case 4:
 259         return "dword";
 260     case 8:
 261         return "qword";
 262     case 10:
 263         return "tword";
 264     case 16:
 265         return "oword";
 266     case 32:
 267         return "yword";
 268     case 64:
 269         return "zword";
 270     default:
 271         return "???";
 272     }
 273 }
 274
 275 static void warn_overflow(int pass, int size)
 276 {
 277     errfunc(ERR_WARNING | pass | ERR_WARN_NOV,
 278             "%s data exceeds bounds", size_name(size));
 279 }
 280
 281 static void warn_overflow_const(int64_t data, int size)
 282 {
 283     if (overflow_general(data, size))
 284         warn_overflow(ERR_PASS1, size);
 285 }
 286
 287 static void warn_overflow_opd(const struct operand *o, int size)
 288 {
 289     if (o->wrt == NO_SEG && o->segment == NO_SEG) {
 290         if (overflow_general(o->offset, size))
 291             warn_overflow(ERR_PASS2, size);
 292     }
 293 }
 294
 295 /*
 296  * This routine wrappers the real output format's output routine,
 297  * in order to pass a copy of the data off to the listing file
 298  * generator at the same time.
 299  */
 300 static void out(int64_t offset, int32_t segto, const void *data,
 301                 enum out_type type, uint64_t size,
 302                 int32_t segment, int32_t wrt)
 303 {
 304     static int32_t lineno = 0;     /* static!!! */
 305     static char *lnfname = NULL;
 306     uint8_t p[8];
 307
 308     if (type == OUT_ADDRESS && segment == NO_SEG && wrt == NO_SEG) {
 309         /*
 310          * This is a non-relocated address, and we're going to
 311          * convert it into RAWDATA format.
 312          */
 313         uint8_t *q = p;
 314
 315         if (size > 8) {
 316             errfunc(ERR_PANIC, "OUT_ADDRESS with size > 8");
 317             return;
 318         }
 319
 320         WRITEADDR(q, *(int64_t *)data, size);
 321         data = p;
 322         type = OUT_RAWDATA;
 323     }
 324
 325     list->output(offset, data, type, size);
 326
 327     /*
 328      * this call to src_get determines when we call the
 329      * debug-format-specific "linenum" function
 330      * it updates lineno and lnfname to the current values
 331      * returning 0 if "same as last time", -2 if lnfname
 332      * changed, and the amount by which lineno changed,
 333      * if it did. thus, these variables must be static
 334      */
 335
 336     if (src_get(&lineno, &lnfname))
 337         outfmt->current_dfmt->linenum(lnfname, lineno, segto);
 338
 339     outfmt->output(segto, data, type, size, segment, wrt);
 340 }
 341
 342 static void out_imm8(int64_t offset, int32_t segment, struct operand *opx)
 343 {
 344     if (opx->segment != NO_SEG) {
 345         uint64_t data = opx->offset;
 346         out(offset, segment, &data, OUT_ADDRESS, 1, opx->segment, opx->wrt);
 347     } else {
 348         uint8_t byte = opx->offset;
 349         out(offset, segment, &byte, OUT_RAWDATA, 1, NO_SEG, NO_SEG);
 350     }
 351 }
 352
 353 static bool jmp_match(int32_t segment, int64_t offset, int bits,
 354                       insn * ins, const struct itemplate *temp)
 355 {
 356     int64_t isize;
 357     const uint8_t *code = temp->code;
 358     uint8_t c = code[0];
 359
 360     if (((c & ~1) != 0370) || (ins->oprs[0].type & STRICT))
 361         return false;
 362     if (!optimizing)
 363         return false;
 364     if (optimizing < 0 && c == 0371)
 365         return false;
 366
 367     isize = calcsize(segment, offset, bits, ins, temp);
 368
 369     if (ins->oprs[0].opflags & OPFLAG_UNKNOWN)
 370         /* Be optimistic in pass 1 */
 371         return true;
 372
 373     if (ins->oprs[0].segment != segment)
 374         return false;
 375
 376     isize = ins->oprs[0].offset - offset - isize; /* isize is delta */
 377     return (isize >= -128 && isize <= 127); /* is it byte size? */
 378 }
 379
 380 int64_t assemble(int32_t segment, int64_t offset, int bits, uint32_t cp,
 381                  insn * instruction, struct ofmt *output, efunc error,
 382                  ListGen * listgen)
 383 {
 384     const struct itemplate *temp;
 385     int j;
 386     enum match_result m;
 387     int64_t insn_end;
 388     int32_t itimes;
 389     int64_t start = offset;
 390     int64_t wsize;              /* size for DB etc. */
 391
 392     errfunc = error;            /* to pass to other functions */
 393     cpu = cp;
 394     outfmt = output;            /* likewise */
 395     list = listgen;             /* and again */
 396
 397     wsize = idata_bytes(instruction->opcode);
 398     if (wsize == -1)
 399         return 0;
 400
 401     if (wsize) {
 402         extop *e;
 403         int32_t t = instruction->times;
 404         if (t < 0)
 405             errfunc(ERR_PANIC,
 406                     "instruction->times < 0 (%ld) in assemble()", t);
 407
 408         while (t--) {           /* repeat TIMES times */
 409             list_for_each(e, instruction->eops) {
 410                 if (e->type == EOT_DB_NUMBER) {
 411                     if (wsize > 8) {
 412                         errfunc(ERR_NONFATAL,
 413                                 "integer supplied to a DT, DO or DY"
 414                                 " instruction");
 415                     } else {
 416                         out(offset, segment, &e->offset,
 417                             OUT_ADDRESS, wsize, e->segment, e->wrt);
 418                         offset += wsize;
 419                     }
 420                 } else if (e->type == EOT_DB_STRING ||
 421                            e->type == EOT_DB_STRING_FREE) {
 422                     int align;
 423
 424                     out(offset, segment, e->stringval,
 425                         OUT_RAWDATA, e->stringlen, NO_SEG, NO_SEG);
 426                     align = e->stringlen % wsize;
 427
 428                     if (align) {
 429                         align = wsize - align;
 430                         out(offset, segment, zero_buffer,
 431                             OUT_RAWDATA, align, NO_SEG, NO_SEG);
 432                     }
 433                     offset += e->stringlen + align;
 434                 }
 435             }
 436             if (t > 0 && t == instruction->times - 1) {
 437                 /*
 438                  * Dummy call to list->output to give the offset to the
 439                  * listing module.
 440                  */
 441                 list->output(offset, NULL, OUT_RAWDATA, 0);
 442                 list->uplevel(LIST_TIMES);
 443             }
 444         }
 445         if (instruction->times > 1)
 446             list->downlevel(LIST_TIMES);
 447         return offset - start;
 448     }
 449
 450     if (instruction->opcode == I_INCBIN) {
 451         const char *fname = instruction->eops->stringval;
 452         FILE *fp;
 453
 454         fp = fopen(fname, "rb");
 455         if (!fp) {
 456             error(ERR_NONFATAL, "`incbin': unable to open file `%s'",
 457                   fname);
 458         } else if (fseek(fp, 0L, SEEK_END) < 0) {
 459             error(ERR_NONFATAL, "`incbin': unable to seek on file `%s'",
 460                   fname);
 461             fclose(fp);
 462         } else {
 463             static char buf[4096];
 464             size_t t = instruction->times;
 465             size_t base = 0;
 466             size_t len;
 467
 468             len = ftell(fp);
 469             if (instruction->eops->next) {
 470                 base = instruction->eops->next->offset;
 471                 len -= base;
 472                 if (instruction->eops->next->next &&
 473                     len > (size_t)instruction->eops->next->next->offset)
 474                     len = (size_t)instruction->eops->next->next->offset;
 475             }
 476             /*
 477              * Dummy call to list->output to give the offset to the
 478              * listing module.
 479              */
 480             list->output(offset, NULL, OUT_RAWDATA, 0);
 481             list->uplevel(LIST_INCBIN);
 482             while (t--) {
 483                 size_t l;
 484
 485                 fseek(fp, base, SEEK_SET);
 486                 l = len;
 487                 while (l > 0) {
 488                     int32_t m;
 489                     m = fread(buf, 1, l > sizeof(buf) ? sizeof(buf) : l, fp);
 490                     if (!m) {
 491                         /*
 492                          * This shouldn't happen unless the file
 493                          * actually changes while we are reading
 494                          * it.
 495                          */
 496                         error(ERR_NONFATAL,
 497                               "`incbin': unexpected EOF while"
 498                               " reading file `%s'", fname);
 499                         t = 0;  /* Try to exit cleanly */
 500                         break;
 501                     }
 502                     out(offset, segment, buf, OUT_RAWDATA, m,
 503                         NO_SEG, NO_SEG);
 504                     l -= m;
 505                 }
 506             }
 507             list->downlevel(LIST_INCBIN);
 508             if (instruction->times > 1) {
 509                 /*
 510                  * Dummy call to list->output to give the offset to the
 511                  * listing module.
 512                  */
 513                 list->output(offset, NULL, OUT_RAWDATA, 0);
 514                 list->uplevel(LIST_TIMES);
 515                 list->downlevel(LIST_TIMES);
 516             }
 517             fclose(fp);
 518             return instruction->times * len;
 519         }
 520         return 0;               /* if we're here, there's an error */
 521     }
 522
 523     /* Check to see if we need an address-size prefix */
 524     add_asp(instruction, bits);
 525
 526     m = find_match(&temp, instruction, segment, offset, bits);
 527
 528     if (m == MOK_GOOD) {
 529         /* Matches! */
 530         int64_t insn_size = calcsize(segment, offset, bits, instruction, temp);
 531         itimes = instruction->times;
 532         if (insn_size < 0)  /* shouldn't be, on pass two */
 533             error(ERR_PANIC, "errors made it through from pass one");
 534         else
 535             while (itimes--) {
 536                 for (j = 0; j < MAXPREFIX; j++) {
 537                     uint8_t c = 0;
 538                     switch (instruction->prefixes[j]) {
 539                     case P_WAIT:
 540                         c = 0x9B;
 541                         break;
 542                     case P_LOCK:
 543                         c = 0xF0;
 544                         break;
 545                     case P_REPNE:
 546                     case P_REPNZ:
 547                     case P_XACQUIRE:
 548                         c = 0xF2;
 549                         break;
 550                     case P_REPE:
 551                     case P_REPZ:
 552                     case P_REP:
 553                     case P_XRELEASE:
 554                         c = 0xF3;
 555                         break;
 556                     case R_CS:
 557                         if (bits == 64) {
 558                             error(ERR_WARNING | ERR_PASS2,
 559                                   "cs segment base generated, but will be ignored in 64-bit mode");
 560                         }
 561                         c = 0x2E;
 562                         break;
 563                     case R_DS:
 564                         if (bits == 64) {
 565                             error(ERR_WARNING | ERR_PASS2,
 566                                   "ds segment base generated, but will be ignored in 64-bit mode");
 567                         }
 568                         c = 0x3E;
 569                         break;
 570                     case R_ES:
 571                         if (bits == 64) {
 572                             error(ERR_WARNING | ERR_PASS2,
 573                                   "es segment base generated, but will be ignored in 64-bit mode");
 574                         }
 575                         c = 0x26;
 576                         break;
 577                     case R_FS:
 578                         c = 0x64;
 579                         break;
 580                     case R_GS:
 581                         c = 0x65;
 582                         break;
 583                     case R_SS:
 584                         if (bits == 64) {
 585                             error(ERR_WARNING | ERR_PASS2,
 586                                   "ss segment base generated, but will be ignored in 64-bit mode");
 587                         }
 588                         c = 0x36;
 589                         break;
 590                     case R_SEGR6:
 591                     case R_SEGR7:
 592                         error(ERR_NONFATAL,
 593                               "segr6 and segr7 cannot be used as prefixes");
 594                         break;
 595                     case P_A16:
 596                         if (bits == 64) {
 597                             error(ERR_NONFATAL,
 598                                   "16-bit addressing is not supported "
 599                                   "in 64-bit mode");
 600                         } else if (bits != 16)
 601                             c = 0x67;
 602                         break;
 603                     case P_A32:
 604                         if (bits != 32)
 605                             c = 0x67;
 606                         break;
 607                     case P_A64:
 608                         if (bits != 64) {
 609                             error(ERR_NONFATAL,
 610                                   "64-bit addressing is only supported "
 611                                   "in 64-bit mode");
 612                         }
 613                         break;
 614                     case P_ASP:
 615                         c = 0x67;
 616                         break;
 617                     case P_O16:
 618                         if (bits != 16)
 619                             c = 0x66;
 620                         break;
 621                     case P_O32:
 622                         if (bits == 16)
 623                             c = 0x66;
 624                         break;
 625                     case P_O64:
 626                         /* REX.W */
 627                         break;
 628                     case P_OSP:
 629                         c = 0x66;
 630                         break;
 631                     case P_none:
 632                         break;
 633                     default:
 634                         error(ERR_PANIC, "invalid instruction prefix");
 635                     }
 636                     if (c != 0) {
 637                         out(offset, segment, &c, OUT_RAWDATA, 1,
 638                             NO_SEG, NO_SEG);
 639                         offset++;
 640                     }
 641                 }
 642                 insn_end = offset + insn_size;
 643                 gencode(segment, offset, bits, instruction,
 644                         temp, insn_end);
 645                 offset += insn_size;
 646                 if (itimes > 0 && itimes == instruction->times - 1) {
 647                     /*
 648                      * Dummy call to list->output to give the offset to the
 649                      * listing module.
 650                      */
 651                     list->output(offset, NULL, OUT_RAWDATA, 0);
 652                     list->uplevel(LIST_TIMES);
 653                 }
 654             }
 655         if (instruction->times > 1)
 656             list->downlevel(LIST_TIMES);
 657         return offset - start;
 658     } else {
 659         /* No match */
 660         switch (m) {
 661         case MERR_OPSIZEMISSING:
 662             error(ERR_NONFATAL, "operation size not specified");
 663             break;
 664         case MERR_OPSIZEMISMATCH:
 665             error(ERR_NONFATAL, "mismatch in operand sizes");
 666             break;
 667         case MERR_BADCPU:
 668             error(ERR_NONFATAL, "no instruction for this cpu level");
 669             break;
 670         case MERR_BADMODE:
 671             error(ERR_NONFATAL, "instruction not supported in %d-bit mode",
 672                   bits);
 673             break;
 674         default:
 675             error(ERR_NONFATAL,
 676                   "invalid combination of opcode and operands");
 677             break;
 678         }
 679     }
 680     return 0;
 681 }
 682
 683 int64_t insn_size(int32_t segment, int64_t offset, int bits, uint32_t cp,
 684                   insn * instruction, efunc error)
 685 {
 686     const struct itemplate *temp;
 687     enum match_result m;
 688
 689     errfunc = error;            /* to pass to other functions */
 690     cpu = cp;
 691
 692     if (instruction->opcode == I_none)
 693         return 0;
 694
 695     if (instruction->opcode == I_DB || instruction->opcode == I_DW ||
 696         instruction->opcode == I_DD || instruction->opcode == I_DQ ||
 697         instruction->opcode == I_DT || instruction->opcode == I_DO ||
 698         instruction->opcode == I_DY) {
 699         extop *e;
 700         int32_t isize, osize, wsize;
 701
 702         isize = 0;
 703         wsize = idata_bytes(instruction->opcode);
 704
 705         list_for_each(e, instruction->eops) {
 706             int32_t align;
 707
 708             osize = 0;
 709             if (e->type == EOT_DB_NUMBER) {
 710                 osize = 1;
 711                 warn_overflow_const(e->offset, wsize);
 712             } else if (e->type == EOT_DB_STRING ||
 713                        e->type == EOT_DB_STRING_FREE)
 714                 osize = e->stringlen;
 715
 716             align = (-osize) % wsize;
 717             if (align < 0)
 718                 align += wsize;
 719             isize += osize + align;
 720         }
 721         return isize * instruction->times;
 722     }
 723
 724     if (instruction->opcode == I_INCBIN) {
 725         const char *fname = instruction->eops->stringval;
 726         FILE *fp;
 727         int64_t val = 0;
 728         size_t len;
 729
 730         fp = fopen(fname, "rb");
 731         if (!fp)
 732             error(ERR_NONFATAL, "`incbin': unable to open file `%s'",
 733                   fname);
 734         else if (fseek(fp, 0L, SEEK_END) < 0)
 735             error(ERR_NONFATAL, "`incbin': unable to seek on file `%s'",
 736                   fname);
 737         else {
 738             len = ftell(fp);
 739             if (instruction->eops->next) {
 740                 len -= instruction->eops->next->offset;
 741                 if (instruction->eops->next->next &&
 742                     len > (size_t)instruction->eops->next->next->offset) {
 743                     len = (size_t)instruction->eops->next->next->offset;
 744                 }
 745             }
 746             val = instruction->times * len;
 747         }
 748         if (fp)
 749             fclose(fp);
 750         return val;
 751     }
 752
 753     /* Check to see if we need an address-size prefix */
 754     add_asp(instruction, bits);
 755
 756     m = find_match(&temp, instruction, segment, offset, bits);
 757     if (m == MOK_GOOD) {
 758         /* we've matched an instruction. */
 759         int64_t isize;
 760         int j;
 761
 762         isize = calcsize(segment, offset, bits, instruction, temp);
 763         if (isize < 0)
 764             return -1;
 765         for (j = 0; j < MAXPREFIX; j++) {
 766             switch (instruction->prefixes[j]) {
 767             case P_A16:
 768                 if (bits != 16)
 769                     isize++;
 770                 break;
 771             case P_A32:
 772                 if (bits != 32)
 773                     isize++;
 774                 break;
 775             case P_O16:
 776                 if (bits != 16)
 777                     isize++;
 778                 break;
 779             case P_O32:
 780                 if (bits == 16)
 781                     isize++;
 782                 break;
 783             case P_A64:
 784             case P_O64:
 785             case P_none:
 786                 break;
 787             default:
 788                 isize++;
 789                 break;
 790             }
 791         }
 792         return isize * instruction->times;
 793     } else {
 794         return -1;                  /* didn't match any instruction */
 795     }
 796 }
 797
 798 static void bad_hle_warn(const insn * ins, uint8_t hleok)
 799 {
 800     enum prefixes rep_pfx = ins->prefixes[PPS_REP];
 801     enum whatwarn { w_none, w_lock, w_inval } ww;
 802     static const enum whatwarn warn[2][4] =
 803     {
 804         { w_inval, w_inval, w_none, w_lock }, /* XACQUIRE */
 805         { w_inval, w_none,  w_none, w_lock }, /* XRELEASE */
 806     };
 807     unsigned int n;
 808
 809     n = (unsigned int)rep_pfx - P_XACQUIRE;
 810     if (n > 1)
 811         return;                 /* Not XACQUIRE/XRELEASE */
 812
 813     ww = warn[n][hleok];
 814     if (!is_class(MEMORY, ins->oprs[0].type))
 815         ww = w_inval;           /* HLE requires operand 0 to be memory */
 816
 817     switch (ww) {
 818     case w_none:
 819         break;
 820
 821     case w_lock:
 822         if (ins->prefixes[PPS_LOCK] != P_LOCK) {
 823             errfunc(ERR_WARNING | ERR_WARN_HLE | ERR_PASS2,
 824                     "%s with this instruction requires lock",
 825                     prefix_name(rep_pfx));
 826         }
 827         break;
 828
 829     case w_inval:
 830         errfunc(ERR_WARNING | ERR_WARN_HLE | ERR_PASS2,
 831                 "%s invalid with this instruction",
 832                 prefix_name(rep_pfx));
 833         break;
 834     }
 835 }
 836
 837 /* Common construct */
 838 #define case3(x) case (x): case (x)+1: case (x)+2
 839 #define case4(x) case3(x): case (x)+3
 840
 841 static int64_t calcsize(int32_t segment, int64_t offset, int bits,
 842                         insn * ins, const struct itemplate *temp)
 843 {
 844     const uint8_t *codes = temp->code;
 845     int64_t length = 0;
 846     uint8_t c;
 847     int rex_mask = ~0;
 848     int op1, op2;
 849     struct operand *opx;
 850     uint8_t opex = 0;
 851     enum ea_type eat;
 852     uint8_t hleok = 0;
 853     bool lockcheck = true;
 854
 855     ins->rex = 0;               /* Ensure REX is reset */
 856     eat = EA_SCALAR;            /* Expect a scalar EA */
 857     memset(ins->evex_p, 0, 3);  /* Ensure EVEX is reset */
 858
 859     if (ins->prefixes[PPS_OSIZE] == P_O64)
 860         ins->rex |= REX_W;
 861
 862     (void)segment;              /* Don't warn that this parameter is unused */
 863     (void)offset;               /* Don't warn that this parameter is unused */
 864
 865     while (*codes) {
 866         c = *codes++;
 867         op1 = (c & 3) + ((opex & 1) << 2);
 868         op2 = ((c >> 3) & 3) + ((opex & 2) << 1);
 869         opx = &ins->oprs[op1];
 870         opex = 0;               /* For the next iteration */
 871
 872         switch (c) {
 873         case4(01):
 874             codes += c, length += c;
 875             break;
 876
 877         case3(05):
 878             opex = c;
 879             break;
 880
 881         case4(010):
 882             ins->rex |=
 883                 op_rexflags(opx, REX_B|REX_H|REX_P|REX_W);
 884             codes++, length++;
 885             break;
 886
 887         case4(020):
 888         case4(024):
 889             length++;
 890             break;
 891
 892         case4(030):
 893             length += 2;
 894             break;
 895
 896         case4(034):
 897             if (opx->type & (BITS16 | BITS32 | BITS64))
 898                 length += (opx->type & BITS16) ? 2 : 4;
 899             else
 900                 length += (bits == 16) ? 2 : 4;
 901             break;
 902
 903         case4(040):
 904             length += 4;
 905             break;
 906
 907         case4(044):
 908             length += ins->addr_size >> 3;
 909             break;
 910
 911         case4(050):
 912             length++;
 913             break;
 914
 915         case4(054):
 916             length += 8; /* MOV reg64/imm */
 917             break;
 918
 919         case4(060):
 920             length += 2;
 921             break;
 922
 923         case4(064):
 924             if (opx->type & (BITS16 | BITS32 | BITS64))
 925                 length += (opx->type & BITS16) ? 2 : 4;
 926             else
 927                 length += (bits == 16) ? 2 : 4;
 928             break;
 929
 930         case4(070):
 931             length += 4;
 932             break;
 933
 934         case4(074):
 935             length += 2;
 936             break;
 937
 938         case 0172:
 939         case 0173:
 940             codes++;
 941             length++;
 942             break;
 943
 944         case4(0174):
 945             length++;
 946             break;
 947
 948         case4(0240):
 949             ins->rex |= REX_EV;
 950             ins->vexreg = regval(opx);
 951             ins->evex_p[2] |= op_evexflags(opx, EVEX_P2VP, 2); /* High-16 NDS */
 952             ins->vex_cm = *codes++;
 953             ins->vex_wlp = *codes++;
 954             ins->evex_tuple = (*codes++ - 0300);
 955             break;
 956
 957         case 0250:
 958             ins->rex |= REX_EV;
 959             ins->vexreg = 0;
 960             ins->vex_cm = *codes++;
 961             ins->vex_wlp = *codes++;
 962             ins->evex_tuple = (*codes++ - 0300);
 963             break;
 964
 965         case4(0254):
 966             length += 4;
 967             break;
 968
 969         case4(0260):
 970             ins->rex |= REX_V;
 971             ins->vexreg = regval(opx);
 972             ins->vex_cm = *codes++;
 973             ins->vex_wlp = *codes++;
 974             break;
 975
 976         case 0270:
 977             ins->rex |= REX_V;
 978             ins->vexreg = 0;
 979             ins->vex_cm = *codes++;
 980             ins->vex_wlp = *codes++;
 981             break;
 982
 983         case3(0271):
 984             hleok = c & 3;
 985             break;
 986
 987         case4(0274):
 988             length++;
 989             break;
 990
 991         case4(0300):
 992             break;
 993
 994         case 0310:
 995             if (bits == 64)
 996                 return -1;
 997             length += (bits != 16) && !has_prefix(ins, PPS_ASIZE, P_A16);
 998             break;
 999
1000         case 0311:
1001             length += (bits != 32) && !has_prefix(ins, PPS_ASIZE, P_A32);
1002             break;
1003
1004         case 0312:
1005             break;
1006
1007         case 0313:
1008             if (bits != 64 || has_prefix(ins, PPS_ASIZE, P_A16) ||
1009                 has_prefix(ins, PPS_ASIZE, P_A32))
1010                 return -1;
1011             break;
1012
1013         case4(0314):
1014             break;
1015
1016         case 0320:
1017         {
1018             enum prefixes pfx = ins->prefixes[PPS_OSIZE];
1019             if (pfx == P_O16)
1020                 break;
1021             if (pfx != P_none)
1022                 errfunc(ERR_WARNING | ERR_PASS2, "invalid operand size prefix");
1023             else
1024                 ins->prefixes[PPS_OSIZE] = P_O16;
1025             break;
1026         }
1027
1028         case 0321:
1029         {
1030             enum prefixes pfx = ins->prefixes[PPS_OSIZE];
1031             if (pfx == P_O32)
1032                 break;
1033             if (pfx != P_none)
1034                 errfunc(ERR_WARNING | ERR_PASS2, "invalid operand size prefix");
1035             else
1036                 ins->prefixes[PPS_OSIZE] = P_O32;
1037             break;
1038         }
1039
1040         case 0322:
1041             break;
1042
1043         case 0323:
1044             rex_mask &= ~REX_W;
1045             break;
1046
1047         case 0324:
1048             ins->rex |= REX_W;
1049             break;
1050
1051         case 0325:
1052             ins->rex |= REX_NH;
1053             break;
1054
1055         case 0326:
1056             break;
1057
1058         case 0330:
1059             codes++, length++;
1060             break;
1061
1062         case 0331:
1063             break;
1064
1065         case 0332:
1066         case 0333:
1067             length++;
1068             break;
1069
1070         case 0334:
1071             ins->rex |= REX_L;
1072             break;
1073
1074         case 0335:
1075             break;
1076
1077         case 0336:
1078             if (!ins->prefixes[PPS_REP])
1079                 ins->prefixes[PPS_REP] = P_REP;
1080             break;
1081
1082         case 0337:
1083             if (!ins->prefixes[PPS_REP])
1084                 ins->prefixes[PPS_REP] = P_REPNE;
1085             break;
1086
1087         case 0340:
1088             if (ins->oprs[0].segment != NO_SEG)
1089                 errfunc(ERR_NONFATAL, "attempt to reserve non-constant"
1090                         " quantity of BSS space");
1091             else
1092                 length += ins->oprs[0].offset;
1093             break;
1094
1095         case 0341:
1096             if (!ins->prefixes[PPS_WAIT])
1097                 ins->prefixes[PPS_WAIT] = P_WAIT;
1098             break;
1099
1100         case 0360:
1101             break;
1102
1103         case 0361:
1104             length++;
1105             break;
1106
1107         case 0364:
1108         case 0365:
1109             break;
1110
1111         case 0366:
1112         case 0367:
1113             length++;
1114             break;
1115
1116         case3(0370):
1117             break;
1118
1119         case 0373:
1120             length++;
1121             break;
1122
1123         case 0374:
1124             eat = EA_XMMVSIB;
1125             break;
1126
1127         case 0375:
1128             eat = EA_YMMVSIB;
1129             break;
1130
1131         case 0376:
1132             eat = EA_ZMMVSIB;
1133             break;
1134
1135         case4(0100):
1136         case4(0110):
1137         case4(0120):
1138         case4(0130):
1139         case4(0200):
1140         case4(0204):
1141         case4(0210):
1142         case4(0214):
1143         case4(0220):
1144         case4(0224):
1145         case4(0230):
1146         case4(0234):
1147             {
1148                 ea ea_data;
1149                 int rfield;
1150                 opflags_t rflags;
1151                 struct operand *opy = &ins->oprs[op2];
1152                 struct operand *oplast;
1153
1154                 ea_data.rex = 0;           /* Ensure ea.REX is initially 0 */
1155
1156                 if (c <= 0177) {
1157                     /* pick rfield from operand b (opx) */
1158                     rflags = regflag(opx);
1159                     rfield = nasm_regvals[opx->basereg];
1160                     /* find the last SIMD operand where ER decorator resides */
1161                     oplast = &ins->oprs[op1 > op2 ? op1 : op2];
1162                 } else {
1163                     rflags = 0;
1164                     rfield = c & 7;
1165                     oplast = opy;
1166                 }
1167
1168                 if (oplast->decoflags & ER) {
1169                     /* set EVEX.RC (rounding control) and b */
1170                     ins->evex_p[2] |= (((ins->evex_rm - BRC_RN) << 5) & EVEX_P2LL) |
1171                                       EVEX_P2B;
1172                 } else {
1173                     /* set EVEX.L'L (vector length) */
1174                     ins->evex_p[2] |= ((ins->vex_wlp << (5 - 2)) & EVEX_P2LL);
1175                     if ((oplast->decoflags & SAE) ||
1176                         (opy->decoflags & BRDCAST_MASK)) {
1177                         /* set EVEX.b */
1178                         ins->evex_p[2] |= EVEX_P2B;
1179                     }
1180                 }
1181
1182                 if (process_ea(opy, &ea_data, bits,
1183                                rfield, rflags, ins) != eat) {
1184                     errfunc(ERR_NONFATAL, "invalid effective address");
1185                     return -1;
1186                 } else {
1187                     ins->rex |= ea_data.rex;
1188                     length += ea_data.size;
1189                 }
1190             }
1191             break;
1192
1193         default:
1194             errfunc(ERR_PANIC, "internal instruction table corrupt"
1195                     ": instruction code \\%o (0x%02X) given", c, c);
1196             break;
1197         }
1198     }
1199
1200     ins->rex &= rex_mask;
1201
1202     if (ins->rex & REX_NH) {
1203         if (ins->rex & REX_H) {
1204             errfunc(ERR_NONFATAL, "instruction cannot use high registers");
1205             return -1;
1206         }
1207         ins->rex &= ~REX_P;        /* Don't force REX prefix due to high reg */
1208     }
1209
1210     if (ins->rex & (REX_V | REX_EV)) {
1211         int bad32 = REX_R|REX_W|REX_X|REX_B;
1212
1213         if (ins->rex & REX_H) {
1214             errfunc(ERR_NONFATAL, "cannot use high register in AVX instruction");
1215             return -1;
1216         }
1217         switch (ins->vex_wlp & 060) {
1218         case 000:
1219         case 040:
1220             ins->rex &= ~REX_W;
1221             break;
1222         case 020:
1223             ins->rex |= REX_W;
1224             bad32 &= ~REX_W;
1225             break;
1226         case 060:
1227             /* Follow REX_W */
1228             break;
1229         }
1230
1231         if (bits != 64 && ((ins->rex & bad32) || ins->vexreg > 7)) {
1232             errfunc(ERR_NONFATAL, "invalid operands in non-64-bit mode");
1233             return -1;
1234         }
1235         if (ins->rex & REX_EV)
1236             length += 4;
1237         else if (ins->vex_cm != 1 || (ins->rex & (REX_W|REX_X|REX_B)))
1238             length += 3;
1239         else
1240             length += 2;
1241     } else if (ins->rex & REX_REAL) {
1242         if (ins->rex & REX_H) {
1243             errfunc(ERR_NONFATAL, "cannot use high register in rex instruction");
1244             return -1;
1245         } else if (bits == 64) {
1246             length++;
1247         } else if ((ins->rex & REX_L) &&
1248                    !(ins->rex & (REX_P|REX_W|REX_X|REX_B)) &&
1249                    cpu >= IF_X86_64) {
1250             /* LOCK-as-REX.R */
1251             assert_no_prefix(ins, PPS_LOCK);
1252             lockcheck = false;  /* Already errored, no need for warning */
1253             length++;
1254         } else {
1255             errfunc(ERR_NONFATAL, "invalid operands in non-64-bit mode");
1256             return -1;
1257         }
1258     }
1259
1260     if (has_prefix(ins, PPS_LOCK, P_LOCK) && lockcheck &&
1261         (!(temp->flags & IF_LOCK) || !is_class(MEMORY, ins->oprs[0].type))) {
1262         errfunc(ERR_WARNING | ERR_WARN_LOCK | ERR_PASS2 ,
1263                 "instruction is not lockable");
1264     }
1265
1266     bad_hle_warn(ins, hleok);
1267
1268     return length;
1269 }
1270
1271 static inline unsigned int emit_rex(insn *ins, int32_t segment, int64_t offset, int bits)
1272 {
1273     if (bits == 64) {
1274         if ((ins->rex & REX_REAL) && !(ins->rex & (REX_V | REX_EV))) {
1275             ins->rex = (ins->rex & REX_REAL) | REX_P;
1276             out(offset, segment, &ins->rex, OUT_RAWDATA, 1, NO_SEG, NO_SEG);
1277             ins->rex = 0;
1278             return 1;
1279         }
1280     }
1281
1282     return 0;
1283 }
1284
1285 static void gencode(int32_t segment, int64_t offset, int bits,
1286                     insn * ins, const struct itemplate *temp,
1287                     int64_t insn_end)
1288 {
1289     uint8_t c;
1290     uint8_t bytes[4];
1291     int64_t size;
1292     int64_t data;
1293     int op1, op2;
1294     struct operand *opx;
1295     const uint8_t *codes = temp->code;
1296     uint8_t opex = 0;
1297     enum ea_type eat = EA_SCALAR;
1298
1299     while (*codes) {
1300         c = *codes++;
1301         op1 = (c & 3) + ((opex & 1) << 2);
1302         op2 = ((c >> 3) & 3) + ((opex & 2) << 1);
1303         opx = &ins->oprs[op1];
1304         opex = 0;                /* For the next iteration */
1305
1306         switch (c) {
1307         case 01:
1308         case 02:
1309         case 03:
1310         case 04:
1311             offset += emit_rex(ins, segment, offset, bits);
1312             out(offset, segment, codes, OUT_RAWDATA, c, NO_SEG, NO_SEG);
1313             codes += c;
1314             offset += c;
1315             break;
1316
1317         case 05:
1318         case 06:
1319         case 07:
1320             opex = c;
1321             break;
1322
1323         case4(010):
1324             offset += emit_rex(ins, segment, offset, bits);
1325             bytes[0] = *codes++ + (regval(opx) & 7);
1326             out(offset, segment, bytes, OUT_RAWDATA, 1, NO_SEG, NO_SEG);
1327             offset += 1;
1328             break;
1329
1330         case4(020):
1331             if (opx->offset < -256 || opx->offset > 255) {
1332                 errfunc(ERR_WARNING | ERR_PASS2 | ERR_WARN_NOV,
1333                         "byte value exceeds bounds");
1334             }
1335             out_imm8(offset, segment, opx);
1336             offset += 1;
1337             break;
1338
1339         case4(024):
1340             if (opx->offset < 0 || opx->offset > 255)
1341                 errfunc(ERR_WARNING | ERR_PASS2 | ERR_WARN_NOV,
1342                         "unsigned byte value exceeds bounds");
1343             out_imm8(offset, segment, opx);
1344             offset += 1;
1345             break;
1346
1347         case4(030):
1348             warn_overflow_opd(opx, 2);
1349             data = opx->offset;
1350             out(offset, segment, &data, OUT_ADDRESS, 2,
1351                 opx->segment, opx->wrt);
1352             offset += 2;
1353             break;
1354
1355         case4(034):
1356             if (opx->type & (BITS16 | BITS32))
1357                 size = (opx->type & BITS16) ? 2 : 4;
1358             else
1359                 size = (bits == 16) ? 2 : 4;
1360             warn_overflow_opd(opx, size);
1361             data = opx->offset;
1362             out(offset, segment, &data, OUT_ADDRESS, size,
1363                 opx->segment, opx->wrt);
1364             offset += size;
1365             break;
1366
1367         case4(040):
1368             warn_overflow_opd(opx, 4);
1369             data = opx->offset;
1370             out(offset, segment, &data, OUT_ADDRESS, 4,
1371                 opx->segment, opx->wrt);
1372             offset += 4;
1373             break;
1374
1375         case4(044):
1376             data = opx->offset;
1377             size = ins->addr_size >> 3;
1378             warn_overflow_opd(opx, size);
1379             out(offset, segment, &data, OUT_ADDRESS, size,
1380                 opx->segment, opx->wrt);
1381             offset += size;
1382             break;
1383
1384         case4(050):
1385             if (opx->segment != segment) {
1386                 data = opx->offset;
1387                 out(offset, segment, &data,
1388                     OUT_REL1ADR, insn_end - offset,
1389                     opx->segment, opx->wrt);
1390             } else {
1391                 data = opx->offset - insn_end;
1392                 if (data > 127 || data < -128)
1393                     errfunc(ERR_NONFATAL, "short jump is out of range");
1394                 out(offset, segment, &data,
1395                     OUT_ADDRESS, 1, NO_SEG, NO_SEG);
1396             }
1397             offset += 1;
1398             break;
1399
1400         case4(054):
1401             data = (int64_t)opx->offset;
1402             out(offset, segment, &data, OUT_ADDRESS, 8,
1403                 opx->segment, opx->wrt);
1404             offset += 8;
1405             break;
1406
1407         case4(060):
1408             if (opx->segment != segment) {
1409                 data = opx->offset;
1410                 out(offset, segment, &data,
1411                     OUT_REL2ADR, insn_end - offset,
1412                     opx->segment, opx->wrt);
1413             } else {
1414                 data = opx->offset - insn_end;
1415                 out(offset, segment, &data,
1416                     OUT_ADDRESS, 2, NO_SEG, NO_SEG);
1417             }
1418             offset += 2;
1419             break;
1420
1421         case4(064):
1422             if (opx->type & (BITS16 | BITS32 | BITS64))
1423                 size = (opx->type & BITS16) ? 2 : 4;
1424             else
1425                 size = (bits == 16) ? 2 : 4;
1426             if (opx->segment != segment) {
1427                 data = opx->offset;
1428                 out(offset, segment, &data,
1429                     size == 2 ? OUT_REL2ADR : OUT_REL4ADR,
1430                     insn_end - offset, opx->segment, opx->wrt);
1431             } else {
1432                 data = opx->offset - insn_end;
1433                 out(offset, segment, &data,
1434                     OUT_ADDRESS, size, NO_SEG, NO_SEG);
1435             }
1436             offset += size;
1437             break;
1438
1439         case4(070):
1440             if (opx->segment != segment) {
1441                 data = opx->offset;
1442                 out(offset, segment, &data,
1443                     OUT_REL4ADR, insn_end - offset,
1444                     opx->segment, opx->wrt);
1445             } else {
1446                 data = opx->offset - insn_end;
1447                 out(offset, segment, &data,
1448                     OUT_ADDRESS, 4, NO_SEG, NO_SEG);
1449             }
1450             offset += 4;
1451             break;
1452
1453         case4(074):
1454             if (opx->segment == NO_SEG)
1455                 errfunc(ERR_NONFATAL, "value referenced by FAR is not"
1456                         " relocatable");
1457             data = 0;
1458             out(offset, segment, &data, OUT_ADDRESS, 2,
1459                 outfmt->segbase(1 + opx->segment),
1460                 opx->wrt);
1461             offset += 2;
1462             break;
1463
1464         case 0172:
1465             c = *codes++;
1466             opx = &ins->oprs[c >> 3];
1467             bytes[0] = nasm_regvals[opx->basereg] << 4;
1468             opx = &ins->oprs[c & 7];
1469             if (opx->segment != NO_SEG || opx->wrt != NO_SEG) {
1470                 errfunc(ERR_NONFATAL,
1471                         "non-absolute expression not permitted as argument %d",
1472                         c & 7);
1473             } else {
1474                 if (opx->offset & ~15) {
1475                     errfunc(ERR_WARNING | ERR_PASS2 | ERR_WARN_NOV,
1476                             "four-bit argument exceeds bounds");
1477                 }
1478                 bytes[0] |= opx->offset & 15;
1479             }
1480             out(offset, segment, bytes, OUT_RAWDATA, 1, NO_SEG, NO_SEG);
1481             offset++;
1482             break;
1483
1484         case 0173:
1485             c = *codes++;
1486             opx = &ins->oprs[c >> 4];
1487             bytes[0] = nasm_regvals[opx->basereg] << 4;
1488             bytes[0] |= c & 15;
1489             out(offset, segment, bytes, OUT_RAWDATA, 1, NO_SEG, NO_SEG);
1490             offset++;
1491             break;
1492
1493         case4(0174):
1494             bytes[0] = nasm_regvals[opx->basereg] << 4;
1495             out(offset, segment, bytes, OUT_RAWDATA, 1, NO_SEG, NO_SEG);
1496             offset++;
1497             break;
1498
1499         case4(0254):
1500             data = opx->offset;
1501             if (opx->wrt == NO_SEG && opx->segment == NO_SEG &&
1502                 (int32_t)data != (int64_t)data) {
1503                 errfunc(ERR_WARNING | ERR_PASS2 | ERR_WARN_NOV,
1504                         "signed dword immediate exceeds bounds");
1505             }
1506             out(offset, segment, &data, OUT_ADDRESS, 4,
1507                 opx->segment, opx->wrt);
1508             offset += 4;
1509             break;
1510
1511         case4(0240):
1512         case 0250:
1513             codes += 3;
1514             ins->evex_p[2] |= op_evexflags(&ins->oprs[0],
1515                                            EVEX_P2Z | EVEX_P2AAA, 2);
1516             ins->evex_p[2] ^= EVEX_P2VP;        /* 1's complement */
1517             bytes[0] = 0x62;
1518             /* EVEX.X can be set by either REX or EVEX for different reasons */
1519             bytes[1] = (~(((ins->rex & 7) << 5) |
1520                           (ins->evex_p[0] & (EVEX_P0X | EVEX_P0RP))) & 0xf0) |
1521                         (ins->vex_cm & 3);
1522             bytes[2] = ((ins->rex & REX_W) << (7 - 3)) |
1523                        ((~ins->vexreg & 15) << 3) |
1524                        (1 << 2) | (ins->vex_wlp & 3);
1525             bytes[3] = ins->evex_p[2];
1526             out(offset, segment, &bytes, OUT_RAWDATA, 4, NO_SEG, NO_SEG);
1527             offset += 4;
1528             break;
1529
1530         case4(0260):
1531         case 0270:
1532             codes += 2;
1533             if (ins->vex_cm != 1 || (ins->rex & (REX_W|REX_X|REX_B))) {
1534                 bytes[0] = (ins->vex_cm >> 6) ? 0x8f : 0xc4;
1535                 bytes[1] = (ins->vex_cm & 31) | ((~ins->rex & 7) << 5);
1536                 bytes[2] = ((ins->rex & REX_W) << (7-3)) |
1537                     ((~ins->vexreg & 15)<< 3) | (ins->vex_wlp & 07);
1538                 out(offset, segment, &bytes, OUT_RAWDATA, 3, NO_SEG, NO_SEG);
1539                 offset += 3;
1540             } else {
1541                 bytes[0] = 0xc5;
1542                 bytes[1] = ((~ins->rex & REX_R) << (7-2)) |
1543                     ((~ins->vexreg & 15) << 3) | (ins->vex_wlp & 07);
1544                 out(offset, segment, &bytes, OUT_RAWDATA, 2, NO_SEG, NO_SEG);
1545                 offset += 2;
1546             }
1547             break;
1548
1549         case 0271:
1550         case 0272:
1551         case 0273:
1552             break;
1553
1554         case4(0274):
1555         {
1556             uint64_t uv, um;
1557             int s;
1558
1559             if (ins->rex & REX_W)
1560                 s = 64;
1561             else if (ins->prefixes[PPS_OSIZE] == P_O16)
1562                 s = 16;
1563             else if (ins->prefixes[PPS_OSIZE] == P_O32)
1564                 s = 32;
1565             else
1566                 s = bits;
1567
1568             um = (uint64_t)2 << (s-1);
1569             uv = opx->offset;
1570
1571             if (uv > 127 && uv < (uint64_t)-128 &&
1572                 (uv < um-128 || uv > um-1)) {
1573                 /* If this wasn't explicitly byte-sized, warn as though we
1574                  * had fallen through to the imm16/32/64 case.
1575                  */
1576                 errfunc(ERR_WARNING | ERR_PASS2 | ERR_WARN_NOV,
1577                         "%s value exceeds bounds",
1578                         (opx->type & BITS8) ? "signed byte" :
1579                         s == 16 ? "word" :
1580                         s == 32 ? "dword" :
1581                         "signed dword");
1582             }
1583             if (opx->segment != NO_SEG) {
1584                 data = uv;
1585                 out(offset, segment, &data, OUT_ADDRESS, 1,
1586                     opx->segment, opx->wrt);
1587             } else {
1588                 bytes[0] = uv;
1589                 out(offset, segment, bytes, OUT_RAWDATA, 1, NO_SEG,
1590                     NO_SEG);
1591             }
1592             offset += 1;
1593             break;
1594         }
1595
1596         case4(0300):
1597             break;
1598
1599         case 0310:
1600             if (bits == 32 && !has_prefix(ins, PPS_ASIZE, P_A16)) {
1601                 *bytes = 0x67;
1602                 out(offset, segment, bytes, OUT_RAWDATA, 1, NO_SEG, NO_SEG);
1603                 offset += 1;
1604             } else
1605                 offset += 0;
1606             break;
1607
1608         case 0311:
1609             if (bits != 32 && !has_prefix(ins, PPS_ASIZE, P_A32)) {
1610                 *bytes = 0x67;
1611                 out(offset, segment, bytes, OUT_RAWDATA, 1, NO_SEG, NO_SEG);
1612                 offset += 1;
1613             } else
1614                 offset += 0;
1615             break;
1616
1617         case 0312:
1618             break;
1619
1620         case 0313:
1621             ins->rex = 0;
1622             break;
1623
1624         case4(0314):
1625             break;
1626
1627         case 0320:
1628         case 0321:
1629             break;
1630
1631         case 0322:
1632         case 0323:
1633             break;
1634
1635         case 0324:
1636             ins->rex |= REX_W;
1637             break;
1638
1639         case 0325:
1640             break;
1641
1642         case 0326:
1643             break;
1644
1645         case 0330:
1646             *bytes = *codes++ ^ get_cond_opcode(ins->condition);
1647             out(offset, segment, bytes, OUT_RAWDATA, 1, NO_SEG, NO_SEG);
1648             offset += 1;
1649             break;
1650
1651         case 0331:
1652             break;
1653
1654         case 0332:
1655         case 0333:
1656             *bytes = c - 0332 + 0xF2;
1657             out(offset, segment, bytes, OUT_RAWDATA, 1, NO_SEG, NO_SEG);
1658             offset += 1;
1659             break;
1660
1661         case 0334:
1662             if (ins->rex & REX_R) {
1663                 *bytes = 0xF0;
1664                 out(offset, segment, bytes, OUT_RAWDATA, 1, NO_SEG, NO_SEG);
1665                 offset += 1;
1666             }
1667             ins->rex &= ~(REX_L|REX_R);
1668             break;
1669
1670         case 0335:
1671             break;
1672
1673         case 0336:
1674         case 0337:
1675             break;
1676
1677         case 0340:
1678             if (ins->oprs[0].segment != NO_SEG)
1679                 errfunc(ERR_PANIC, "non-constant BSS size in pass two");
1680             else {
1681                 int64_t size = ins->oprs[0].offset;
1682                 if (size > 0)
1683                     out(offset, segment, NULL,
1684                         OUT_RESERVE, size, NO_SEG, NO_SEG);
1685                 offset += size;
1686             }
1687             break;
1688
1689         case 0341:
1690             break;
1691
1692         case 0360:
1693             break;
1694
1695         case 0361:
1696             bytes[0] = 0x66;
1697             out(offset, segment, bytes, OUT_RAWDATA, 1, NO_SEG, NO_SEG);
1698             offset += 1;
1699             break;
1700
1701         case 0364:
1702         case 0365:
1703             break;
1704
1705         case 0366:
1706         case 0367:
1707             *bytes = c - 0366 + 0x66;
1708             out(offset, segment, bytes, OUT_RAWDATA, 1, NO_SEG, NO_SEG);
1709             offset += 1;
1710             break;
1711
1712         case 0370:
1713         case 0371:
1714             break;
1715
1716         case 0373:
1717             *bytes = bits == 16 ? 3 : 5;
1718             out(offset, segment, bytes, OUT_RAWDATA, 1, NO_SEG, NO_SEG);
1719             offset += 1;
1720             break;
1721
1722         case 0374:
1723             eat = EA_XMMVSIB;
1724             break;
1725
1726         case 0375:
1727             eat = EA_YMMVSIB;
1728             break;
1729
1730         case 0376:
1731             eat = EA_ZMMVSIB;
1732             break;
1733
1734         case4(0100):
1735         case4(0110):
1736         case4(0120):
1737         case4(0130):
1738         case4(0200):
1739         case4(0204):
1740         case4(0210):
1741         case4(0214):
1742         case4(0220):
1743         case4(0224):
1744         case4(0230):
1745         case4(0234):
1746             {
1747                 ea ea_data;
1748                 int rfield;
1749                 opflags_t rflags;
1750                 uint8_t *p;
1751                 int32_t s;
1752                 struct operand *opy = &ins->oprs[op2];
1753
1754                 if (c <= 0177) {
1755                     /* pick rfield from operand b (opx) */
1756                     rflags = regflag(opx);
1757                     rfield = nasm_regvals[opx->basereg];
1758                 } else {
1759                     /* rfield is constant */
1760                     rflags = 0;
1761                     rfield = c & 7;
1762                 }
1763
1764                 if (process_ea(opy, &ea_data, bits,
1765                                rfield, rflags, ins) != eat)
1766                     errfunc(ERR_NONFATAL, "invalid effective address");
1767
1768                 p = bytes;
1769                 *p++ = ea_data.modrm;
1770                 if (ea_data.sib_present)
1771                     *p++ = ea_data.sib;
1772
1773                 s = p - bytes;
1774                 out(offset, segment, bytes, OUT_RAWDATA, s, NO_SEG, NO_SEG);
1775
1776                 /*
1777                  * Make sure the address gets the right offset in case
1778                  * the line breaks in the .lst file (BR 1197827)
1779                  */
1780                 offset += s;
1781                 s = 0;
1782
1783                 switch (ea_data.bytes) {
1784                 case 0:
1785                     break;
1786                 case 1:
1787                 case 2:
1788                 case 4:
1789                 case 8:
1790                     /* use compressed displacement, if available */
1791                     data = ea_data.disp8 ? ea_data.disp8 : opy->offset;
1792                     s += ea_data.bytes;
1793                     if (ea_data.rip) {
1794                         if (opy->segment == segment) {
1795                             data -= insn_end;
1796                             if (overflow_signed(data, ea_data.bytes))
1797                                 warn_overflow(ERR_PASS2, ea_data.bytes);
1798                             out(offset, segment, &data, OUT_ADDRESS,
1799                                 ea_data.bytes, NO_SEG, NO_SEG);
1800                         } else {
1801                             /* overflow check in output/linker? */
1802                             out(offset, segment, &data,        OUT_REL4ADR,
1803                                 insn_end - offset, opy->segment, opy->wrt);
1804                         }
1805                     } else {
1806                         if (overflow_general(data, ins->addr_size >> 3) ||
1807                             signed_bits(data, ins->addr_size) !=
1808                             signed_bits(data, ea_data.bytes * 8))
1809                             warn_overflow(ERR_PASS2, ea_data.bytes);
1810
1811                         out(offset, segment, &data, OUT_ADDRESS,
1812                             ea_data.bytes, opy->segment, opy->wrt);
1813                     }
1814                     break;
1815                 default:
1816                     /* Impossible! */
1817                     errfunc(ERR_PANIC,
1818                             "Invalid amount of bytes (%d) for offset?!",
1819                             ea_data.bytes);
1820                     break;
1821                 }
1822                 offset += s;
1823             }
1824             break;
1825
1826         default:
1827             errfunc(ERR_PANIC, "internal instruction table corrupt"
1828                     ": instruction code \\%o (0x%02X) given", c, c);
1829             break;
1830         }
1831     }
1832 }
1833
1834 static opflags_t regflag(const operand * o)
1835 {
1836     if (!is_register(o->basereg))
1837         errfunc(ERR_PANIC, "invalid operand passed to regflag()");
1838     return nasm_reg_flags[o->basereg];
1839 }
1840
1841 static int32_t regval(const operand * o)
1842 {
1843     if (!is_register(o->basereg))
1844         errfunc(ERR_PANIC, "invalid operand passed to regval()");
1845     return nasm_regvals[o->basereg];
1846 }
1847
1848 static int op_rexflags(const operand * o, int mask)
1849 {
1850     opflags_t flags;
1851     int val;
1852
1853     if (!is_register(o->basereg))
1854         errfunc(ERR_PANIC, "invalid operand passed to op_rexflags()");
1855
1856     flags = nasm_reg_flags[o->basereg];
1857     val = nasm_regvals[o->basereg];
1858
1859     return rexflags(val, flags, mask);
1860 }
1861
1862 static int rexflags(int val, opflags_t flags, int mask)
1863 {
1864     int rex = 0;
1865
1866     if (val >= 8)
1867         rex |= REX_B|REX_X|REX_R;
1868     if (flags & BITS64)
1869         rex |= REX_W;
1870     if (!(REG_HIGH & ~flags))                   /* AH, CH, DH, BH */
1871         rex |= REX_H;
1872     else if (!(REG8 & ~flags) && val >= 4)      /* SPL, BPL, SIL, DIL */
1873         rex |= REX_P;
1874
1875     return rex & mask;
1876 }
1877
1878 static int evexflags(int val, decoflags_t deco,
1879                      int mask, uint8_t byte)
1880 {
1881     int evex = 0;
1882
1883     switch(byte) {
1884     case 0:
1885         if (val >= 16)
1886             evex |= (EVEX_P0RP | EVEX_P0X);
1887         break;
1888     case 2:
1889         if (val >= 16)
1890             evex |= EVEX_P2VP;
1891         if (deco & Z)
1892             evex |= EVEX_P2Z;
1893         if (deco & OPMASK_MASK)
1894             evex |= deco & EVEX_P2AAA;
1895         break;
1896     }
1897     return evex & mask;
1898 }
1899
1900 static int op_evexflags(const operand * o, int mask, uint8_t byte)
1901 {
1902     int val;
1903
1904     if (!is_register(o->basereg))
1905         errfunc(ERR_PANIC, "invalid operand passed to op_evexflags()");
1906
1907     val = nasm_regvals[o->basereg];
1908
1909     return evexflags(val, o->decoflags, mask, byte);
1910 }
1911
1912 static enum match_result find_match(const struct itemplate **tempp,
1913                                     insn *instruction,
1914                                     int32_t segment, int64_t offset, int bits)
1915 {
1916     const struct itemplate *temp;
1917     enum match_result m, merr;
1918     opflags_t xsizeflags[MAX_OPERANDS];
1919     bool opsizemissing = false;
1920     int8_t broadcast = -1;
1921     int i;
1922
1923     /* find the position of broadcasting operand */
1924     for (i = 0; i < instruction->operands; i++)
1925         if (instruction->oprs[i].decoflags & BRDCAST_MASK) {
1926             broadcast = i;
1927             break;
1928         }
1929
1930     /* broadcasting uses a different data element size */
1931     for (i = 0; i < instruction->operands; i++)
1932         if (i == broadcast)
1933             xsizeflags[i] = instruction->oprs[i].decoflags & BRSIZE_MASK;
1934         else
1935             xsizeflags[i] = instruction->oprs[i].type & SIZE_MASK;
1936
1937     merr = MERR_INVALOP;
1938
1939     for (temp = nasm_instructions[instruction->opcode];
1940          temp->opcode != I_none; temp++) {
1941         m = matches(temp, instruction, bits);
1942         if (m == MOK_JUMP) {
1943             if (jmp_match(segment, offset, bits, instruction, temp))
1944                 m = MOK_GOOD;
1945             else
1946                 m = MERR_INVALOP;
1947         } else if (m == MERR_OPSIZEMISSING &&
1948                    (temp->flags & IF_SMASK) != IF_SX) {
1949             /*
1950              * Missing operand size and a candidate for fuzzy matching...
1951              */
1952             for (i = 0; i < temp->operands; i++)
1953                 if (i == broadcast)
1954                     xsizeflags[i] |= temp->deco[i] & BRSIZE_MASK;
1955                 else
1956                     xsizeflags[i] |= temp->opd[i] & SIZE_MASK;
1957             opsizemissing = true;
1958         }
1959         if (m > merr)
1960             merr = m;
1961         if (merr == MOK_GOOD)
1962             goto done;
1963     }
1964
1965     /* No match, but see if we can get a fuzzy operand size match... */
1966     if (!opsizemissing)
1967         goto done;
1968
1969     for (i = 0; i < instruction->operands; i++) {
1970         /*
1971          * We ignore extrinsic operand sizes on registers, so we should
1972          * never try to fuzzy-match on them.  This also resolves the case
1973          * when we have e.g. "xmmrm128" in two different positions.
1974          */
1975         if (is_class(REGISTER, instruction->oprs[i].type))
1976             continue;
1977
1978         /* This tests if xsizeflags[i] has more than one bit set */
1979         if ((xsizeflags[i] & (xsizeflags[i]-1)))
1980             goto done;                /* No luck */
1981
1982         if (i == broadcast)
1983             instruction->oprs[i].decoflags |= xsizeflags[i];
1984         else
1985             instruction->oprs[i].type |= xsizeflags[i]; /* Set the size */
1986     }
1987
1988     /* Try matching again... */
1989     for (temp = nasm_instructions[instruction->opcode];
1990          temp->opcode != I_none; temp++) {
1991         m = matches(temp, instruction, bits);
1992         if (m == MOK_JUMP) {
1993             if (jmp_match(segment, offset, bits, instruction, temp))
1994                 m = MOK_GOOD;
1995             else
1996                 m = MERR_INVALOP;
1997         }
1998         if (m > merr)
1999             merr = m;
2000         if (merr == MOK_GOOD)
2001             goto done;
2002     }
2003
2004 done:
2005     *tempp = temp;
2006     return merr;
2007 }
2008
2009 static enum match_result matches(const struct itemplate *itemp,
2010                                  insn *instruction, int bits)
2011 {
2012     opflags_t size[MAX_OPERANDS], asize;
2013     bool opsizemissing = false;
2014     int i, oprs;
2015
2016     /*
2017      * Check the opcode
2018      */
2019     if (itemp->opcode != instruction->opcode)
2020         return MERR_INVALOP;
2021
2022     /*
2023      * Count the operands
2024      */
2025     if (itemp->operands != instruction->operands)
2026         return MERR_INVALOP;
2027
2028     /*
2029      * Is it legal?
2030      */
2031     if (!(optimizing > 0) && (itemp->flags & IF_OPT))
2032         return MERR_INVALOP;
2033
2034     /*
2035      * Check that no spurious colons or TOs are present
2036      */
2037     for (i = 0; i < itemp->operands; i++)
2038         if (instruction->oprs[i].type & ~itemp->opd[i] & (COLON | TO))
2039             return MERR_INVALOP;
2040
2041     /*
2042      * Process size flags
2043      */
2044     switch (itemp->flags & IF_SMASK) {
2045     case IF_SB:
2046         asize = BITS8;
2047         break;
2048     case IF_SW:
2049         asize = BITS16;
2050         break;
2051     case IF_SD:
2052         asize = BITS32;
2053         break;
2054     case IF_SQ:
2055         asize = BITS64;
2056         break;
2057     case IF_SO:
2058         asize = BITS128;
2059         break;
2060     case IF_SY:
2061         asize = BITS256;
2062         break;
2063     case IF_SZ:
2064         asize = BITS512;
2065         break;
2066     case IF_SIZE:
2067         switch (bits) {
2068         case 16:
2069             asize = BITS16;
2070             break;
2071         case 32:
2072             asize = BITS32;
2073             break;
2074         case 64:
2075             asize = BITS64;
2076             break;
2077         default:
2078             asize = 0;
2079             break;
2080         }
2081         break;
2082     default:
2083         asize = 0;
2084         break;
2085     }
2086
2087     if (itemp->flags & IF_ARMASK) {
2088         /* S- flags only apply to a specific operand */
2089         i = ((itemp->flags & IF_ARMASK) >> IF_ARSHFT) - 1;
2090         memset(size, 0, sizeof size);
2091         size[i] = asize;
2092     } else {
2093         /* S- flags apply to all operands */
2094         for (i = 0; i < MAX_OPERANDS; i++)
2095             size[i] = asize;
2096     }
2097
2098     /*
2099      * Check that the operand flags all match up,
2100      * it's a bit tricky so lets be verbose:
2101      *
2102      * 1) Find out the size of operand. If instruction
2103      *    doesn't have one specified -- we're trying to
2104      *    guess it either from template (IF_S* flag) or
2105      *    from code bits.
2106      *
2107      * 2) If template operand do not match the instruction OR
2108      *    template has an operand size specified AND this size differ
2109      *    from which instruction has (perhaps we got it from code bits)
2110      *    we are:
2111      *      a)  Check that only size of instruction and operand is differ
2112      *          other characteristics do match
2113      *      b)  Perhaps it's a register specified in instruction so
2114      *          for such a case we just mark that operand as "size
2115      *          missing" and this will turn on fuzzy operand size
2116      *          logic facility (handled by a caller)
2117      */
2118     for (i = 0; i < itemp->operands; i++) {
2119         opflags_t type = instruction->oprs[i].type;
2120         decoflags_t deco = instruction->oprs[i].decoflags;
2121         if (!(type & SIZE_MASK))
2122             type |= size[i];
2123
2124         if ((itemp->opd[i] & ~type & ~SIZE_MASK) ||
2125             (itemp->deco[i] & deco) != deco) {
2126             return MERR_INVALOP;
2127         } else if ((itemp->opd[i] & SIZE_MASK) &&
2128                    (itemp->opd[i] & SIZE_MASK) != (type & SIZE_MASK)) {
2129             if (type & SIZE_MASK) {
2130                 /*
2131                  * when broadcasting, the element size depends on
2132                  * the instruction type. decorator flag should match.
2133                  */
2134 #define MATCH_BRSZ(bits) (((type & SIZE_MASK) == BITS##bits) &&             \
2135                           ((itemp->deco[i] & BRSIZE_MASK) == BR_BITS##bits))
2136                 if (!((deco & BRDCAST_MASK) &&
2137                       (MATCH_BRSZ(32) || MATCH_BRSZ(64)))) {
2138                     return MERR_INVALOP;
2139                 }
2140             } else if (!is_class(REGISTER, type)) {
2141                 /*
2142                  * Note: we don't honor extrinsic operand sizes for registers,
2143                  * so "missing operand size" for a register should be
2144                  * considered a wildcard match rather than an error.
2145                  */
2146                 opsizemissing = true;
2147             }
2148         }
2149     }
2150
2151     if (opsizemissing)
2152         return MERR_OPSIZEMISSING;
2153
2154     /*
2155      * Check operand sizes
2156      */
2157     if (itemp->flags & (IF_SM | IF_SM2)) {
2158         oprs = (itemp->flags & IF_SM2 ? 2 : itemp->operands);
2159         for (i = 0; i < oprs; i++) {
2160             asize = itemp->opd[i] & SIZE_MASK;
2161             if (asize) {
2162                 for (i = 0; i < oprs; i++)
2163                     size[i] = asize;
2164                 break;
2165             }
2166         }
2167     } else {
2168         oprs = itemp->operands;
2169     }
2170
2171     for (i = 0; i < itemp->operands; i++) {
2172         if (!(itemp->opd[i] & SIZE_MASK) &&
2173             (instruction->oprs[i].type & SIZE_MASK & ~size[i]))
2174             return MERR_OPSIZEMISMATCH;
2175     }
2176
2177     /*
2178      * Check template is okay at the set cpu level
2179      */
2180     if (((itemp->flags & IF_PLEVEL) > cpu))
2181         return MERR_BADCPU;
2182
2183     /*
2184      * Verify the appropriate long mode flag.
2185      */
2186     if ((itemp->flags & (bits == 64 ? IF_NOLONG : IF_LONG)))
2187         return MERR_BADMODE;
2188
2189     /*
2190      * If we have a HLE prefix, look for the NOHLE flag
2191      */
2192     if ((itemp->flags & IF_NOHLE) &&
2193         (has_prefix(instruction, PPS_REP, P_XACQUIRE) ||
2194          has_prefix(instruction, PPS_REP, P_XRELEASE)))
2195         return MERR_BADHLE;
2196
2197     /*
2198      * Check if special handling needed for Jumps
2199      */
2200     if ((itemp->code[0] & ~1) == 0370)
2201         return MOK_JUMP;
2202
2203     return MOK_GOOD;
2204 }
2205
2206 /*
2207  * Check if offset is a multiple of N with corresponding tuple type
2208  * if Disp8*N is available, compressed displacement is stored in compdisp
2209  */
2210 static bool is_disp8n(operand *input, insn *ins, int8_t *compdisp)
2211 {
2212     const uint8_t fv_n[2][2][VLMAX] = {{{16, 32, 64}, {4, 4, 4}},
2213                                        {{16, 32, 64}, {8, 8, 8}}};
2214     const uint8_t hv_n[2][VLMAX]    =  {{8, 16, 32}, {4, 4, 4}};
2215     const uint8_t dup_n[VLMAX]      =   {8, 32, 64};
2216
2217     bool evex_b           = input->decoflags & BRDCAST_MASK;
2218     enum ttypes   tuple   = ins->evex_tuple;
2219     /* vex_wlp composed as [wwllpp] */
2220     enum vectlens vectlen = (ins->vex_wlp & 0x0c) >> 2;
2221     /* wig(=2) is treated as w0(=0) */
2222     bool evex_w           = (ins->vex_wlp & 0x10) >> 4;
2223     int32_t off           = input->offset;
2224     uint8_t n = 0;
2225     int32_t disp8;
2226
2227     switch(tuple) {
2228     case FV:
2229         n = fv_n[evex_w][evex_b][vectlen];
2230         break;
2231     case HV:
2232         n = hv_n[evex_b][vectlen];
2233         break;
2234
2235     case FVM:
2236         /* 16, 32, 64 for VL 128, 256, 512 respectively*/
2237         n = 1 << (vectlen + 4);
2238         break;
2239     case T1S8:  /* N = 1 */
2240     case T1S16: /* N = 2 */
2241         n = tuple - T1S8 + 1;
2242         break;
2243     case T1S:
2244         /* N = 4 for 32bit, 8 for 64bit */
2245         n = evex_w ? 8 : 4;
2246         break;
2247     case T1F32:
2248     case T1F64:
2249         /* N = 4 for 32bit, 8 for 64bit */
2250         n = (tuple == T1F32 ? 4 : 8);
2251         break;
2252     case T2:
2253     case T4:
2254     case T8:
2255         if (vectlen + 7 <= (evex_w + 5) + (tuple - T2 + 1))
2256             n = 0;
2257         else
2258             n = 1 << (tuple - T2 + evex_w + 4);
2259         break;
2260     case HVM:
2261     case QVM:
2262     case OVM:
2263         n = 1 << (OVM - tuple + vectlen + 1);
2264         break;
2265     case M128:
2266         n = 16;
2267         break;
2268     case DUP:
2269         n = dup_n[vectlen];
2270         break;
2271
2272     default:
2273         break;
2274     }
2275
2276     if (n && !(off & (n - 1))) {
2277         disp8 = off / n;
2278         /* if it fits in Disp8 */
2279         if (disp8 >= -128 && disp8 <= 127) {
2280             *compdisp = disp8;
2281             return true;
2282         }
2283     }
2284
2285     *compdisp = 0;
2286     return false;
2287 }
2288
2289 /*
2290  * Check if ModR/M.mod should/can be 01.
2291  * - EAF_BYTEOFFS is set
2292  * - offset can fit in a byte when EVEX is not used
2293  * - offset can be compressed when EVEX is used
2294  */
2295 #define IS_MOD_01()     (input->eaflags & EAF_BYTEOFFS ||       \
2296                          (o >= -128 && o <= 127 &&              \
2297                           seg == NO_SEG && !forw_ref &&         \
2298                           !(input->eaflags & EAF_WORDOFFS) &&   \
2299                           !(ins->rex & REX_EV)) ||              \
2300                          (ins->rex & REX_EV &&                  \
2301                           is_disp8n(input, ins, &output->disp8)))
2302
2303 static enum ea_type process_ea(operand *input, ea *output, int bits,
2304                                int rfield, opflags_t rflags, insn *ins)
2305 {
2306     bool forw_ref = !!(input->opflags & OPFLAG_UNKNOWN);
2307     int addrbits = ins->addr_size;
2308
2309     output->type    = EA_SCALAR;
2310     output->rip     = false;
2311
2312     /* REX flags for the rfield operand */
2313     output->rex     |= rexflags(rfield, rflags, REX_R | REX_P | REX_W | REX_H);
2314     /* EVEX.R' flag for the REG operand */
2315     ins->evex_p[0]  |= evexflags(rfield, 0, EVEX_P0RP, 0);
2316
2317     if (is_class(REGISTER, input->type)) {
2318         /*
2319          * It's a direct register.
2320          */
2321         if (!is_register(input->basereg))
2322             goto err;
2323
2324         if (!is_reg_class(REG_EA, input->basereg))
2325             goto err;
2326
2327         /* broadcasting is not available with a direct register operand. */
2328         if (input->decoflags & BRDCAST_MASK) {
2329             nasm_error(ERR_NONFATAL, "Broadcasting not allowed from a register");
2330             goto err;
2331         }
2332
2333         output->rex         |= op_rexflags(input, REX_B | REX_P | REX_W | REX_H);
2334         ins->evex_p[0]      |= op_evexflags(input, EVEX_P0X, 0);
2335         output->sib_present = false;    /* no SIB necessary */
2336         output->bytes       = 0;        /* no offset necessary either */
2337         output->modrm       = GEN_MODRM(3, rfield, nasm_regvals[input->basereg]);
2338     } else {
2339         /*
2340          * It's a memory reference.
2341          */
2342
2343         /* Embedded rounding or SAE is not available with a mem ref operand. */
2344         if (input->decoflags & (ER | SAE)) {
2345             nasm_error(ERR_NONFATAL,
2346                        "Embedded rounding is available only with reg-reg op.");
2347             return -1;
2348         }
2349
2350         if (input->basereg == -1 &&
2351             (input->indexreg == -1 || input->scale == 0)) {
2352             /*
2353              * It's a pure offset.
2354              */
2355             if (bits == 64 && ((input->type & IP_REL) == IP_REL) &&
2356                 input->segment == NO_SEG) {
2357                 nasm_error(ERR_WARNING | ERR_PASS1, "absolute address can not be RIP-relative");
2358                 input->type &= ~IP_REL;
2359                 input->type |= MEMORY;
2360             }
2361
2362             if (input->eaflags & EAF_BYTEOFFS ||
2363                 (input->eaflags & EAF_WORDOFFS &&
2364                  input->disp_size != (addrbits != 16 ? 32 : 16))) {
2365                 nasm_error(ERR_WARNING | ERR_PASS1, "displacement size ignored on absolute address");
2366             }
2367
2368             if (bits == 64 && (~input->type & IP_REL)) {
2369                 output->sib_present = true;
2370                 output->sib         = GEN_SIB(0, 4, 5);
2371                 output->bytes       = 4;
2372                 output->modrm       = GEN_MODRM(0, rfield, 4);
2373                 output->rip         = false;
2374             } else {
2375                 output->sib_present = false;
2376                 output->bytes       = (addrbits != 16 ? 4 : 2);
2377                 output->modrm       = GEN_MODRM(0, rfield, (addrbits != 16 ? 5 : 6));
2378                 output->rip         = bits == 64;
2379             }
2380         } else {
2381             /*
2382              * It's an indirection.
2383              */
2384             int i = input->indexreg, b = input->basereg, s = input->scale;
2385             int32_t seg = input->segment;
2386             int hb = input->hintbase, ht = input->hinttype;
2387             int t, it, bt;              /* register numbers */
2388             opflags_t x, ix, bx;        /* register flags */
2389
2390             if (s == 0)
2391                 i = -1;         /* make this easy, at least */
2392
2393             if (is_register(i)) {
2394                 it = nasm_regvals[i];
2395                 ix = nasm_reg_flags[i];
2396             } else {
2397                 it = -1;
2398                 ix = 0;
2399             }
2400
2401             if (is_register(b)) {
2402                 bt = nasm_regvals[b];
2403                 bx = nasm_reg_flags[b];
2404             } else {
2405                 bt = -1;
2406                 bx = 0;
2407             }
2408
2409             /* if either one are a vector register... */
2410             if ((ix|bx) & (XMMREG|YMMREG|ZMMREG) & ~REG_EA) {
2411                 opflags_t sok = BITS32 | BITS64;
2412                 int32_t o = input->offset;
2413                 int mod, scale, index, base;
2414
2415                 /*
2416                  * For a vector SIB, one has to be a vector and the other,
2417                  * if present, a GPR.  The vector must be the index operand.
2418                  */
2419                 if (it == -1 || (bx & (XMMREG|YMMREG|ZMMREG) & ~REG_EA)) {
2420                     if (s == 0)
2421                         s = 1;
2422                     else if (s != 1)
2423                         goto err;
2424
2425                     t = bt, bt = it, it = t;
2426                     x = bx, bx = ix, ix = x;
2427                 }
2428
2429                 if (bt != -1) {
2430                     if (REG_GPR & ~bx)
2431                         goto err;
2432                     if (!(REG64 & ~bx) || !(REG32 & ~bx))
2433                         sok &= bx;
2434                     else
2435                         goto err;
2436                 }
2437
2438                 /*
2439                  * While we're here, ensure the user didn't specify
2440                  * WORD or QWORD
2441                  */
2442                 if (input->disp_size == 16 || input->disp_size == 64)
2443                     goto err;
2444
2445                 if (addrbits == 16 ||
2446                     (addrbits == 32 && !(sok & BITS32)) ||
2447                     (addrbits == 64 && !(sok & BITS64)))
2448                     goto err;
2449
2450                 output->type = ((ix & ZMMREG & ~REG_EA) ? EA_ZMMVSIB
2451                                 : ((ix & YMMREG & ~REG_EA)
2452                                 ? EA_YMMVSIB : EA_XMMVSIB));
2453
2454                 output->rex    |= rexflags(it, ix, REX_X);
2455                 output->rex    |= rexflags(bt, bx, REX_B);
2456                 ins->evex_p[2] |= evexflags(it, 0, EVEX_P2VP, 2);
2457
2458                 index = it & 7; /* it is known to be != -1 */
2459
2460                 switch (s) {
2461                 case 1:
2462                     scale = 0;
2463                     break;
2464                 case 2:
2465                     scale = 1;
2466                     break;
2467                 case 4:
2468                     scale = 2;
2469                     break;
2470                 case 8:
2471                     scale = 3;
2472                     break;
2473                 default:   /* then what the smeg is it? */
2474                     goto err;    /* panic */
2475                 }
2476
2477                 if (bt == -1) {
2478                     base = 5;
2479                     mod = 0;
2480                 } else {
2481                     base = (bt & 7);
2482                     if (base != REG_NUM_EBP && o == 0 &&
2483                         seg == NO_SEG && !forw_ref &&
2484                         !(input->eaflags & (EAF_BYTEOFFS | EAF_WORDOFFS)))
2485                         mod = 0;
2486                     else if (IS_MOD_01())
2487                         mod = 1;
2488                     else
2489                         mod = 2;
2490                 }
2491
2492                 output->sib_present = true;
2493                 output->bytes       = (bt == -1 || mod == 2 ? 4 : mod);
2494                 output->modrm       = GEN_MODRM(mod, rfield, 4);
2495                 output->sib         = GEN_SIB(scale, index, base);
2496             } else if ((ix|bx) & (BITS32|BITS64)) {
2497                 /*
2498                  * it must be a 32/64-bit memory reference. Firstly we have
2499                  * to check that all registers involved are type E/Rxx.
2500                  */
2501                 opflags_t sok = BITS32 | BITS64;
2502                 int32_t o = input->offset;
2503
2504                 if (it != -1) {
2505                     if (!(REG64 & ~ix) || !(REG32 & ~ix))
2506                         sok &= ix;
2507                     else
2508                         goto err;
2509                 }
2510
2511                 if (bt != -1) {
2512                     if (REG_GPR & ~bx)
2513                         goto err; /* Invalid register */
2514                     if (~sok & bx & SIZE_MASK)
2515                         goto err; /* Invalid size */
2516                     sok &= bx;
2517                 }
2518
2519                 /*
2520                  * While we're here, ensure the user didn't specify
2521                  * WORD or QWORD
2522                  */
2523                 if (input->disp_size == 16 || input->disp_size == 64)
2524                     goto err;
2525
2526                 if (addrbits == 16 ||
2527                     (addrbits == 32 && !(sok & BITS32)) ||
2528                     (addrbits == 64 && !(sok & BITS64)))
2529                     goto err;
2530
2531                 /* now reorganize base/index */
2532                 if (s == 1 && bt != it && bt != -1 && it != -1 &&
2533                     ((hb == b && ht == EAH_NOTBASE) ||
2534                      (hb == i && ht == EAH_MAKEBASE))) {
2535                     /* swap if hints say so */
2536                     t = bt, bt = it, it = t;
2537                     x = bx, bx = ix, ix = x;
2538                 }
2539                 if (bt == it)     /* convert EAX+2*EAX to 3*EAX */
2540                     bt = -1, bx = 0, s++;
2541                 if (bt == -1 && s == 1 && !(hb == it && ht == EAH_NOTBASE)) {
2542                     /* make single reg base, unless hint */
2543                     bt = it, bx = ix, it = -1, ix = 0;
2544                 }
2545                 if (((s == 2 && it != REG_NUM_ESP && !(input->eaflags & EAF_TIMESTWO)) ||
2546                       s == 3 || s == 5 || s == 9) && bt == -1)
2547                     bt = it, bx = ix, s--; /* convert 3*EAX to EAX+2*EAX */
2548                 if (it == -1 && (bt & 7) != REG_NUM_ESP &&
2549                     (input->eaflags & EAF_TIMESTWO))
2550                     it = bt, ix = bx, bt = -1, bx = 0, s = 1;
2551                 /* convert [NOSPLIT EAX] to sib format with 0x0 displacement */
2552                 if (s == 1 && it == REG_NUM_ESP) {
2553                     /* swap ESP into base if scale is 1 */
2554                     t = it, it = bt, bt = t;
2555                     x = ix, ix = bx, bx = x;
2556                 }
2557                 if (it == REG_NUM_ESP ||
2558                     (s != 1 && s != 2 && s != 4 && s != 8 && it != -1))
2559                     goto err;        /* wrong, for various reasons */
2560
2561                 output->rex |= rexflags(it, ix, REX_X);
2562                 output->rex |= rexflags(bt, bx, REX_B);
2563
2564                 if (it == -1 && (bt & 7) != REG_NUM_ESP) {
2565                     /* no SIB needed */
2566                     int mod, rm;
2567
2568                     if (bt == -1) {
2569                         rm = 5;
2570                         mod = 0;
2571                     } else {
2572                         rm = (bt & 7);
2573                         if (rm != REG_NUM_EBP && o == 0 &&
2574                             seg == NO_SEG && !forw_ref &&
2575                             !(input->eaflags & (EAF_BYTEOFFS | EAF_WORDOFFS)))
2576                             mod = 0;
2577                         else if (IS_MOD_01())
2578                             mod = 1;
2579                         else
2580                             mod = 2;
2581                     }
2582
2583                     output->sib_present = false;
2584                     output->bytes       = (bt == -1 || mod == 2 ? 4 : mod);
2585                     output->modrm       = GEN_MODRM(mod, rfield, rm);
2586                 } else {
2587                     /* we need a SIB */
2588                     int mod, scale, index, base;
2589
2590                     if (it == -1)
2591                         index = 4, s = 1;
2592                     else
2593                         index = (it & 7);
2594
2595                     switch (s) {
2596                     case 1:
2597                         scale = 0;
2598                         break;
2599                     case 2:
2600                         scale = 1;
2601                         break;
2602                     case 4:
2603                         scale = 2;
2604                         break;
2605                     case 8:
2606                         scale = 3;
2607                         break;
2608                     default:   /* then what the smeg is it? */
2609                         goto err;    /* panic */
2610                     }
2611
2612                     if (bt == -1) {
2613                         base = 5;
2614                         mod = 0;
2615                     } else {
2616                         base = (bt & 7);
2617                         if (base != REG_NUM_EBP && o == 0 &&
2618                             seg == NO_SEG && !forw_ref &&
2619                             !(input->eaflags & (EAF_BYTEOFFS | EAF_WORDOFFS)))
2620                             mod = 0;
2621                         else if (IS_MOD_01())
2622                             mod = 1;
2623                         else
2624                             mod = 2;
2625                     }
2626
2627                     output->sib_present = true;
2628                     output->bytes       = (bt == -1 || mod == 2 ? 4 : mod);
2629                     output->modrm       = GEN_MODRM(mod, rfield, 4);
2630                     output->sib         = GEN_SIB(scale, index, base);
2631                 }
2632             } else {            /* it's 16-bit */
2633                 int mod, rm;
2634                 int16_t o = input->offset;
2635
2636                 /* check for 64-bit long mode */
2637                 if (addrbits == 64)
2638                     goto err;
2639
2640                 /* check all registers are BX, BP, SI or DI */
2641                 if ((b != -1 && b != R_BP && b != R_BX && b != R_SI && b != R_DI) ||
2642                     (i != -1 && i != R_BP && i != R_BX && i != R_SI && i != R_DI))
2643                     goto err;
2644
2645                 /* ensure the user didn't specify DWORD/QWORD */
2646                 if (input->disp_size == 32 || input->disp_size == 64)
2647                     goto err;
2648
2649                 if (s != 1 && i != -1)
2650                     goto err;        /* no can do, in 16-bit EA */
2651                 if (b == -1 && i != -1) {
2652                     int tmp = b;
2653                     b = i;
2654                     i = tmp;
2655                 }               /* swap */
2656                 if ((b == R_SI || b == R_DI) && i != -1) {
2657                     int tmp = b;
2658                     b = i;
2659                     i = tmp;
2660                 }
2661                 /* have BX/BP as base, SI/DI index */
2662                 if (b == i)
2663                     goto err;        /* shouldn't ever happen, in theory */
2664                 if (i != -1 && b != -1 &&
2665                     (i == R_BP || i == R_BX || b == R_SI || b == R_DI))
2666                     goto err;        /* invalid combinations */
2667                 if (b == -1)            /* pure offset: handled above */
2668                     goto err;        /* so if it gets to here, panic! */
2669
2670                 rm = -1;
2671                 if (i != -1)
2672                     switch (i * 256 + b) {
2673                     case R_SI * 256 + R_BX:
2674                         rm = 0;
2675                         break;
2676                     case R_DI * 256 + R_BX:
2677                         rm = 1;
2678                         break;
2679                     case R_SI * 256 + R_BP:
2680                         rm = 2;
2681                         break;
2682                     case R_DI * 256 + R_BP:
2683                         rm = 3;
2684                         break;
2685                 } else
2686                     switch (b) {
2687                     case R_SI:
2688                         rm = 4;
2689                         break;
2690                     case R_DI:
2691                         rm = 5;
2692                         break;
2693                     case R_BP:
2694                         rm = 6;
2695                         break;
2696                     case R_BX:
2697                         rm = 7;
2698                         break;
2699                     }
2700                 if (rm == -1)           /* can't happen, in theory */
2701                     goto err;        /* so panic if it does */
2702
2703                 if (o == 0 && seg == NO_SEG && !forw_ref && rm != 6 &&
2704                     !(input->eaflags & (EAF_BYTEOFFS | EAF_WORDOFFS)))
2705                     mod = 0;
2706                 else if (IS_MOD_01())
2707                     mod = 1;
2708                 else
2709                     mod = 2;
2710
2711                 output->sib_present = false;    /* no SIB - it's 16-bit */
2712                 output->bytes       = mod;      /* bytes of offset needed */
2713                 output->modrm       = GEN_MODRM(mod, rfield, rm);
2714             }
2715         }
2716     }
2717
2718     output->size = 1 + output->sib_present + output->bytes;
2719     return output->type;
2720
2721 err:
2722     return output->type = EA_INVALID;
2723 }
2724
2725 static void add_asp(insn *ins, int addrbits)
2726 {
2727     int j, valid;
2728     int defdisp;
2729
2730     valid = (addrbits == 64) ? 64|32 : 32|16;
2731
2732     switch (ins->prefixes[PPS_ASIZE]) {
2733     case P_A16:
2734         valid &= 16;
2735         break;
2736     case P_A32:
2737         valid &= 32;
2738         break;
2739     case P_A64:
2740         valid &= 64;
2741         break;
2742     case P_ASP:
2743         valid &= (addrbits == 32) ? 16 : 32;
2744         break;
2745     default:
2746         break;
2747     }
2748
2749     for (j = 0; j < ins->operands; j++) {
2750         if (is_class(MEMORY, ins->oprs[j].type)) {
2751             opflags_t i, b;
2752
2753             /* Verify as Register */
2754             if (!is_register(ins->oprs[j].indexreg))
2755                 i = 0;
2756             else
2757                 i = nasm_reg_flags[ins->oprs[j].indexreg];
2758
2759             /* Verify as Register */
2760             if (!is_register(ins->oprs[j].basereg))
2761                 b = 0;
2762             else
2763                 b = nasm_reg_flags[ins->oprs[j].basereg];
2764
2765             if (ins->oprs[j].scale == 0)
2766                 i = 0;
2767
2768             if (!i && !b) {
2769                 int ds = ins->oprs[j].disp_size;
2770                 if ((addrbits != 64 && ds > 8) ||
2771                     (addrbits == 64 && ds == 16))
2772                     valid &= ds;
2773             } else {
2774                 if (!(REG16 & ~b))
2775                     valid &= 16;
2776                 if (!(REG32 & ~b))
2777                     valid &= 32;
2778                 if (!(REG64 & ~b))
2779                     valid &= 64;
2780
2781                 if (!(REG16 & ~i))
2782                     valid &= 16;
2783                 if (!(REG32 & ~i))
2784                     valid &= 32;
2785                 if (!(REG64 & ~i))
2786                     valid &= 64;
2787             }
2788         }
2789     }
2790
2791     if (valid & addrbits) {
2792         ins->addr_size = addrbits;
2793     } else if (valid & ((addrbits == 32) ? 16 : 32)) {
2794         /* Add an address size prefix */
2795         ins->prefixes[PPS_ASIZE] = (addrbits == 32) ? P_A16 : P_A32;;
2796         ins->addr_size = (addrbits == 32) ? 16 : 32;
2797     } else {
2798         /* Impossible... */
2799         errfunc(ERR_NONFATAL, "impossible combination of address sizes");
2800         ins->addr_size = addrbits; /* Error recovery */
2801     }
2802
2803     defdisp = ins->addr_size == 16 ? 16 : 32;
2804
2805     for (j = 0; j < ins->operands; j++) {
2806         if (!(MEM_OFFS & ~ins->oprs[j].type) &&
2807             (ins->oprs[j].disp_size ? ins->oprs[j].disp_size : defdisp) != ins->addr_size) {
2808             /*
2809              * mem_offs sizes must match the address size; if not,
2810              * strip the MEM_OFFS bit and match only EA instructions
2811              */
2812             ins->oprs[j].type &= ~(MEM_OFFS & ~MEMORY);
2813         }
2814     }
2815 }