operationHelpers.scala 25.9 KB
Newer Older
Kunshan Wang's avatar
Kunshan Wang committed
1 2
package uvm.refimpl.itpr

3 4
import java.nio.charset.Charset

Kunshan Wang's avatar
Kunshan Wang committed
5
import uvm.refimpl._
6
import uvm.refimpl.mem.MemorySupport
7 8 9 10 11 12
import uvm.refimpl.mem.Mutator
import uvm.refimpl.mem.TypeSizes._
import uvm.refimpl.mem.TypeSizes
import uvm.ssavariables._
import uvm.ssavariables.AtomicRMWOptr._
import uvm.types._
13
import uvm.IRNode
Kunshan Wang's avatar
Kunshan Wang committed
14 15 16 17
import uvm.Function
import uvm.refimpl.mem.EntityAccessors.CanBeIntegerized
import uvm.ir.irbuilder.IRBuilder
import uvm.refimpl.mem.EntityAccessors._
Kunshan Wang's avatar
Kunshan Wang committed
18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39

object OpHelper {

  val ONE = BigInt(1)

  def mask(n: Int): BigInt = {
    (ONE << n) - ONE
  }

  def truncFromBigInt(n: BigInt, len: Int): BigInt = n & mask(len)

  def zextToBigInt(n: BigInt, len: Int): BigInt = n & mask(len)

  def sextToBigInt(n: BigInt, len: Int): BigInt = {
    val bit = n.testBit(len - 1)
    if (bit) {
      n | (~mask(len - 1))
    } else {
      n & (mask(len - 1))
    }
  }

40 41 42 43 44
  // The BigInt in a BoxInt is always truncated to len bits.
  //
  // "prepare" means sign- or zero-extend the content to get the real math value.
  // "unprepare" means truncating a math value to len bits to be stored in a BoxInt.

Kunshan Wang's avatar
Kunshan Wang committed
45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101
  def prepareUnsigned(n: BigInt, len: Int): BigInt = truncFromBigInt(n, len)

  def prepareSigned(n: BigInt, len: Int): BigInt = {
    sextToBigInt(truncFromBigInt(n, len), len)
  }

  def unprepare(n: BigInt, len: Int): BigInt = truncFromBigInt(n, len)

  def trunc(n: BigInt, toLen: Int): BigInt = truncFromBigInt(n, toLen)

  def zext(n: BigInt, fromLen: Int, toLen: Int): BigInt = truncFromBigInt(n, fromLen)

  def sext(n: BigInt, fromLen: Int, toLen: Int): BigInt = {
    truncFromBigInt(sextToBigInt(n, fromLen), toLen)
  }

  def maxSInt(l: Int): BigInt = (BigInt(1) << (l - 1)) - 1
  def minSIntAbs(l: Int): BigInt = BigInt(1) << (l - 1)
  def maxUInt(l: Int): BigInt = (BigInt(1) << l) - 1

  /**
   * Convert a float to an integer. The result only has the lowest iLen binary digits.
   */
  def floatToI(n: Float, iLen: Int, signed: Boolean): BigInt = {
    val lExp = Math.getExponent(n)
    val rExp = lExp - 23
    val frac = (java.lang.Float.floatToRawIntBits(n) & 0x7fffff) | 0x800000;

    if (java.lang.Float.isNaN(n)) 0
    else if (signed) {
      if (java.lang.Float.isInfinite(n)) { if (n > 0.0F) maxSInt(iLen) else minSIntAbs(iLen) }
      else {
        if (lExp >= (iLen - 1)) if (n > 0.0F) maxSInt(iLen) else minSIntAbs(iLen)
        else if (lExp < 0) 0
        else {
          val abs = BigInt(frac) << rExp
          unprepare(if (n < 0.0F) -abs else abs, iLen)
        }
      }
    } else {
      if (n < 0.0F) 0
      else if (java.lang.Float.isInfinite(n)) maxUInt(iLen)
      else {
        if (lExp >= iLen) maxUInt(iLen)
        else if (lExp < 0) 0
        else unprepare(BigInt(frac) << rExp, iLen)
      }
    }
  }

  /**
   * Convert a float to an integer. The result only has the lowest iLen binary digits.
   */
  def doubleToI(n: Double, iLen: Int, signed: Boolean): BigInt = {
    val lExp = Math.getExponent(n)
    val rExp = lExp - 52
    val frac = (java.lang.Double.doubleToRawLongBits(n) & 0xfffffffffffffL) | 0x10000000000000L;
102

Kunshan Wang's avatar
Kunshan Wang committed
103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128
    if (java.lang.Double.isNaN(n)) 0
    else if (signed) {
      if (java.lang.Double.isInfinite(n)) { if (n > 0.0D) maxSInt(iLen) else minSIntAbs(iLen) }
      else {
        if (lExp >= (iLen - 1)) if (n > 0.0D) maxSInt(iLen) else minSIntAbs(iLen)
        else if (lExp < 0) 0
        else {
          val abs = BigInt(frac) << rExp
          unprepare(if (n < 0.0D) -abs else abs, iLen)
        }
      }
    } else {
      if (n < 0.0D) 0
      else if (java.lang.Double.isInfinite(n)) maxUInt(iLen)
      else {
        if (lExp >= iLen) maxUInt(iLen)
        else if (lExp < 0) 0
        else unprepare(BigInt(frac) << rExp, iLen)
      }
    }
  }

  def tr64IsInt(opnd: Long): Boolean = {
    (opnd & 0x7ff0000000000001L) == 0x7ff0000000000001L
  }

129
  def tr64IsFP(opnd: Long): Boolean = {
Kunshan Wang's avatar
Kunshan Wang committed
130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160
    (opnd & 0x7ff0000000000001L) != 0x7ff0000000000001L &&
      (opnd & 0x7ff0000000000003L) != 0x7ff0000000000002L
  }

  def tr64IsRef(opnd: Long): Boolean = {
    (opnd & 0x7ff0000000000003L) == 0x7ff0000000000002L
  }

  def intToTr64(opnd: Long): Long = {
    (0x7ff0000000000001L | ((opnd & 0x7ffffffffffffL) << 1) |
      ((opnd & 0x8000000000000L) << 12))
  }

  def fpToTr64(opnd: Double): Long = {
    var bits = java.lang.Double.doubleToRawLongBits(opnd)
    if (java.lang.Double.isNaN(opnd)) {
      bits = bits & 0xfff8000000000000L | 0x0000000000000008L
    }
    bits
  }

  def refToTr64(opnd: Long, tag: Long): Long = {
    (0x7ff0000000000002L | (opnd & 0x7ffffffffff8L) | ((opnd & 0x800000000000L) << 16) |
      ((tag & 0x3eL) << 46) |
      ((tag & 0x1) << 2))
  }

  def tr64ToInt(opnd: Long): Long = {
    (((opnd & 0xffffffffffffeL) >> 1) | ((opnd & 0x8000000000000000L) >> 12) & (1L << 51))
  }

161
  def tr64ToFP(opnd: Long): Double = java.lang.Double.longBitsToDouble(opnd)
Kunshan Wang's avatar
Kunshan Wang committed
162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189

  def tr64ToRef(opnd: Long): Long = {
    ((opnd & 0x7ffffffffff8L) |
      (((~(((opnd & 0x8000000000000000L) << 1) - 1)) >> 17) &
        0xffff800000000000L))
  }

  def tr64ToTag(opnd: Long): Long = {
    (((opnd & 0x000f800000000000L) >> 46) | ((opnd & 0x4) >> 2))
  }
}

object PrimOpHelpers {
  @throws(classOf[UvmDivisionByZeroException])
  def intBinOp(op: BinOptr.BinOptr, l: Int, op1v: BigInt, op2v: BigInt, ctx: => String): BigInt = {
    def pu(v: BigInt): BigInt = OpHelper.prepareUnsigned(v, l)
    def ps(v: BigInt): BigInt = OpHelper.prepareSigned(v, l)
    def up(v: BigInt): BigInt = OpHelper.unprepare(v, l)
    def shiftMask = {
      var i = 1
      while (i < l) { i <<= 1 }
      i - 1
    }
    def checkDivByZero(): Unit = {
      if (op2v == 0) throw new UvmDivisionByZeroException(ctx + "Division by zero.")
    }

    up(op match {
190 191 192
      case BinOptr.ADD  => pu(op1v) + pu(op2v)
      case BinOptr.SUB  => pu(op1v) - pu(op2v)
      case BinOptr.MUL  => pu(op1v) * pu(op2v)
Kunshan Wang's avatar
Kunshan Wang committed
193 194 195 196
      case BinOptr.UDIV => { checkDivByZero(); pu(op1v) / pu(op2v) }
      case BinOptr.SDIV => { checkDivByZero(); ps(op1v) / ps(op2v) }
      case BinOptr.UREM => { checkDivByZero(); pu(op1v) % pu(op2v) }
      case BinOptr.SREM => { checkDivByZero(); ps(op1v) % ps(op2v) }
197
      case BinOptr.SHL  => pu(op1v) << (op2v.intValue & shiftMask)
Kunshan Wang's avatar
Kunshan Wang committed
198 199
      case BinOptr.LSHR => pu(op1v) >> (op2v.intValue & shiftMask)
      case BinOptr.ASHR => ps(op1v) >> (op2v.intValue & shiftMask)
200 201 202 203
      case BinOptr.AND  => pu(op1v) & pu(op2v)
      case BinOptr.OR   => pu(op1v) | pu(op2v)
      case BinOptr.XOR  => pu(op1v) ^ pu(op2v)
      case _            => throw new UvmRuntimeException(ctx + "Binary operator %s is not suitable for integer.".format(op))
Kunshan Wang's avatar
Kunshan Wang committed
204 205 206 207 208 209 210 211 212 213
    })
  }

  def floatBinOp(op: BinOptr.BinOptr, op1v: Float, op2v: Float, ctx: => String): Float = {
    op match {
      case BinOptr.FADD => op1v + op2v
      case BinOptr.FSUB => op1v - op2v
      case BinOptr.FMUL => op1v * op2v
      case BinOptr.FDIV => op1v / op2v
      case BinOptr.FREM => Math.IEEEremainder(op1v, op2v).toFloat
214
      case _            => throw new UvmRuntimeException(ctx + "Binary operator %s is not suitable for float.".format(op))
Kunshan Wang's avatar
Kunshan Wang committed
215 216 217 218 219 220 221 222 223 224
    }
  }

  def doubleBinOp(op: BinOptr.BinOptr, op1v: Double, op2v: Double, ctx: => String): Double = {
    op match {
      case BinOptr.FADD => op1v + op2v
      case BinOptr.FSUB => op1v - op2v
      case BinOptr.FMUL => op1v * op2v
      case BinOptr.FDIV => op1v / op2v
      case BinOptr.FREM => Math.IEEEremainder(op1v, op2v)
225
      case _            => throw new UvmRuntimeException(ctx + "Binary operator %s is not suitable for double.".format(op))
Kunshan Wang's avatar
Kunshan Wang committed
226 227 228 229 230 231 232 233
    }
  }

  def intCmp(op: CmpOptr.CmpOptr, l: Int, op1v: BigInt, op2v: BigInt, ctx: => String): Boolean = {
    def pu(v: BigInt): BigInt = OpHelper.prepareUnsigned(v, l)
    def ps(v: BigInt): BigInt = OpHelper.prepareSigned(v, l)

    op match {
234 235
      case CmpOptr.EQ  => pu(op1v) == pu(op2v)
      case CmpOptr.NE  => pu(op1v) != pu(op2v)
Kunshan Wang's avatar
Kunshan Wang committed
236 237 238 239 240 241 242 243
      case CmpOptr.UGT => pu(op1v) > pu(op2v)
      case CmpOptr.UGE => pu(op1v) >= pu(op2v)
      case CmpOptr.ULT => pu(op1v) < pu(op2v)
      case CmpOptr.ULE => pu(op1v) <= pu(op2v)
      case CmpOptr.SGT => ps(op1v) > ps(op2v)
      case CmpOptr.SGE => ps(op1v) >= ps(op2v)
      case CmpOptr.SLT => ps(op1v) < ps(op2v)
      case CmpOptr.SLE => ps(op1v) <= ps(op2v)
244
      case _           => throw new UvmRuntimeException(ctx + "Comparison operator %s not suitable for integers".format(op))
Kunshan Wang's avatar
Kunshan Wang committed
245 246 247 248 249 250 251 252
    }
  }

  def floatCmp(op: CmpOptr.CmpOptr, op1v: Float, op2v: Float, ctx: => String): Boolean = {
    import java.lang.Float.isNaN
    def ord = !isNaN(op1v) && !isNaN(op2v)
    def uno = isNaN(op1v) || isNaN(op2v)
    op match {
253
      case CmpOptr.FTRUE  => true
Kunshan Wang's avatar
Kunshan Wang committed
254
      case CmpOptr.FFALSE => false
255 256 257 258 259 260 261 262 263 264 265 266 267 268 269
      case CmpOptr.FOEQ   => ord && op1v == op2v
      case CmpOptr.FONE   => ord && op1v != op2v
      case CmpOptr.FOGT   => ord && op1v > op2v
      case CmpOptr.FOGE   => ord && op1v >= op2v
      case CmpOptr.FOLT   => ord && op1v < op2v
      case CmpOptr.FOLE   => ord && op1v <= op2v
      case CmpOptr.FORD   => ord
      case CmpOptr.FUEQ   => uno || op1v == op2v
      case CmpOptr.FUNE   => uno || op1v != op2v
      case CmpOptr.FUGT   => uno || op1v > op2v
      case CmpOptr.FUGE   => uno || op1v >= op2v
      case CmpOptr.FULT   => uno || op1v < op2v
      case CmpOptr.FULE   => uno || op1v <= op2v
      case CmpOptr.FUNO   => uno
      case _              => throw new UvmRuntimeException(ctx + "Comparison operator %s is not suitable for float.".format(op))
Kunshan Wang's avatar
Kunshan Wang committed
270 271 272 273 274 275 276 277
    }
  }

  def doubleCmp(op: CmpOptr.CmpOptr, op1v: Double, op2v: Double, ctx: => String): Boolean = {
    import java.lang.Double.isNaN
    def ord = !isNaN(op1v) && !isNaN(op2v)
    def uno = isNaN(op1v) || isNaN(op2v)
    op match {
278
      case CmpOptr.FTRUE  => true
Kunshan Wang's avatar
Kunshan Wang committed
279
      case CmpOptr.FFALSE => false
280 281 282 283 284 285 286 287 288 289 290 291 292 293 294
      case CmpOptr.FOEQ   => ord && op1v == op2v
      case CmpOptr.FONE   => ord && op1v != op2v
      case CmpOptr.FOGT   => ord && op1v > op2v
      case CmpOptr.FOGE   => ord && op1v >= op2v
      case CmpOptr.FOLT   => ord && op1v < op2v
      case CmpOptr.FOLE   => ord && op1v <= op2v
      case CmpOptr.FORD   => ord
      case CmpOptr.FUEQ   => uno || op1v == op2v
      case CmpOptr.FUNE   => uno || op1v != op2v
      case CmpOptr.FUGT   => uno || op1v > op2v
      case CmpOptr.FUGE   => uno || op1v >= op2v
      case CmpOptr.FULT   => uno || op1v < op2v
      case CmpOptr.FULE   => uno || op1v <= op2v
      case CmpOptr.FUNO   => uno
      case _              => throw new UvmRuntimeException(ctx + "Comparison operator %s is not suitable for double.".format(op))
Kunshan Wang's avatar
Kunshan Wang committed
295 296
    }
  }
297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331

  def refCmp(op: CmpOptr.CmpOptr, op1v: Word, op2v: Word, ctx: => String): Boolean = {
    op match {
      case CmpOptr.EQ => op1v == op2v
      case CmpOptr.NE => op1v != op2v
      case _          => throw new UvmRuntimeException(ctx + "Comparison operator %s is not suitable for ref.".format(op))
    }
  }

  def irefCmp(op: CmpOptr.CmpOptr, op1b: Word, op1o: Word, op2b: Word, op2o: Word, ctx: => String): Boolean = {
    val a1 = op1b + op1o
    val a2 = op2b + op2o

    def warnDiffObj() = if (op1b != op2b) throw new UvmRuntimeException(
      ctx + "Attempt to compare order of irefs in two different objects. lhs: 0x%x+0x%x rhs: 0x%x+0x%x".format(
        op1b, op1o, op2b, op2o))

    op match {
      case CmpOptr.EQ  => a1 == a2
      case CmpOptr.NE  => a1 != a2
      case CmpOptr.ULT => { warnDiffObj(); a1 < a2 }
      case CmpOptr.ULE => { warnDiffObj(); a1 <= a2 }
      case CmpOptr.UGT => { warnDiffObj(); a1 > a2 }
      case CmpOptr.UGE => { warnDiffObj(); a1 >= a2 }
      case _           => throw new UvmRuntimeException(ctx + "Comparison operator %s is not suitable for iref.".format(op))
    }
  }

  def objCmp[T <: AnyRef](op: CmpOptr.CmpOptr, obj1: T, obj2: T, kind: String, ctx: => String): Boolean = {
    op match {
      case CmpOptr.EQ => obj1 eq obj2
      case CmpOptr.NE => obj1 ne obj2
      case _          => throw new UvmRuntimeException(ctx + "Comparison operator %s is not suitable for %s.".format(op, kind))
    }
  }
Kunshan Wang's avatar
Kunshan Wang committed
332
}
Kunshan Wang's avatar
Kunshan Wang committed
333

334
object MemoryOperations {
335 336 337 338 339 340 341 342 343

  def addressOf(ptr: Boolean, vb: ValueBox): Word = {
    if (ptr) {
      vb.asInstanceOf[BoxPointer].addr
    } else {
      val lb = vb.asInstanceOf[BoxIRef]
      lb.objRef + lb.offset
    }
  }
344

345 346 347 348 349 350 351
  def noAccessViaPointer(ptr: Boolean, ty: Type) {
    if (ptr) {
      throw new UvmIllegalMemoryAccessException("Cannot access type %s via pointer".format(ty.repr))
    }
  }

  def load(ptr: Boolean, ty: Type, loc: Word, br: ValueBox)(implicit microVM: MicroVM, memorySupport: MemorySupport): Unit = {
352 353 354
    def loadScalar(ty: Type, loc: Word, br: ValueBox): Unit = ty match {
      case TypeInt(l) =>
        val bi: BigInt = l match {
355 356 357 358
          case 8  => memorySupport.loadByte(loc, !ptr)
          case 16 => memorySupport.loadShort(loc, !ptr)
          case 32 => memorySupport.loadInt(loc, !ptr)
          case 64 => memorySupport.loadLong(loc, !ptr)
Kunshan Wang's avatar
Kunshan Wang committed
359 360 361 362 363
          case 128 => {
            val lowWord = memorySupport.loadLong(loc, !ptr)
            val highWord = memorySupport.loadLong(loc + 8, !ptr)
            (BigInt(highWord) << 64) + lowWord
          }
364
          case _ => throw new UvmUnimplementedOperationException("Loading int of length %d is not supported".format(l))
365 366 367
        }
        br.asInstanceOf[BoxInt].value = OpHelper.unprepare(bi, l)
      case _: TypeFloat =>
368
        val fv = memorySupport.loadFloat(loc, !ptr)
369 370
        br.asInstanceOf[BoxFloat].value = fv
      case _: TypeDouble =>
371
        val dv = memorySupport.loadDouble(loc, !ptr)
372
        br.asInstanceOf[BoxDouble].value = dv
373 374 375
      case _: AbstractPointerType =>
        val addr = memorySupport.loadLong(loc, !ptr)
        br.asInstanceOf[BoxPointer].addr = addr
376
      case _: TypeRef =>
377
        noAccessViaPointer(ptr, ty)
378
        val addr = memorySupport.loadLong(loc)
379 380
        br.asInstanceOf[BoxRef].objRef = addr
      case _: TypeIRef =>
381
        noAccessViaPointer(ptr, ty)
382 383
        val base = memorySupport.loadLong(loc)
        val offset = memorySupport.loadLong(loc + WORD_SIZE_BYTES)
384
        br.asInstanceOf[BoxIRef].oo = (base, offset)
385 386 387 388
      case _: TypeTagRef64 =>
        noAccessViaPointer(ptr, ty)
        val raw = memorySupport.loadLong(loc)
        br.asInstanceOf[BoxTagRef64].raw = raw
Kunshan Wang's avatar
Kunshan Wang committed
389 390 391 392 393
      case _: AbstractOpaqueRefType => {
        def loadEntity[T: CanBeIntegerized](): Unit = {
          val obj = memorySupport.loadEntity[T](loc)
          br.asInstanceOf[BoxOpaque[T]].obj = obj
        }
394
        noAccessViaPointer(ptr, ty)
Kunshan Wang's avatar
Kunshan Wang committed
395 396 397 398 399 400 401 402
        ty match {
          case _: TypeFuncRef        => loadEntity[Function]()
          case _: TypeThreadRef      => loadEntity[InterpreterThread]()
          case _: TypeStackRef       => loadEntity[InterpreterStack]()
          case _: TypeFrameCursorRef => loadEntity[FrameCursor]()
          case _: TypeIRBuilderRef   => loadEntity[IRBuilder]()
        }
      }
403
      case _ => throw new UvmUnimplementedOperationException("Loading of type %s is not supporing".format(ty.getClass.getName))
404 405 406 407
    }

    ty match {
      case TypeVector(ety, len) =>
Kunshan Wang's avatar
Kunshan Wang committed
408
        val brs = br.asInstanceOf[BoxSeq].values
409 410 411 412 413 414 415 416
        val elemSkip = alignUp(sizeOf(ety), alignOf(ety))
        for ((brElem, i) <- brs.zipWithIndex) {
          loadScalar(ety, loc + elemSkip * i, brElem)
        }
      case sty => loadScalar(sty, loc, br)
    }
  }

417
  def store(ptr: Boolean, ty: Type, loc: Word, nvb: ValueBox)(implicit microVM: MicroVM, memorySupport: MemorySupport): Unit = {
Kunshan Wang's avatar
Kunshan Wang committed
418
    def storeScalar(ty: Type, loc: Word, nvb: ValueBox): Unit = ty match {
419 420 421
      case TypeInt(l) =>
        val bi = nvb.asInstanceOf[BoxInt].value
        l match {
422 423 424 425
          case 8  => memorySupport.storeByte(loc, bi.byteValue, !ptr)
          case 16 => memorySupport.storeShort(loc, bi.shortValue, !ptr)
          case 32 => memorySupport.storeInt(loc, bi.intValue, !ptr)
          case 64 => memorySupport.storeLong(loc, bi.longValue, !ptr)
Kunshan Wang's avatar
Kunshan Wang committed
426 427 428 429
          case 128 => {
            memorySupport.storeLong(loc, (bi & 0xffffffffffffffffL).longValue, !ptr)
            memorySupport.storeLong(loc + 8, (bi >> 64).longValue, !ptr)
          }
430
          case _ => throw new UvmUnimplementedOperationException("Storing int of length %d is not supported".format(l))
431 432 433
        }
      case _: TypeFloat =>
        val fv = nvb.asInstanceOf[BoxFloat].value
434
        memorySupport.storeFloat(loc, fv, !ptr)
435 436
      case _: TypeDouble =>
        val dv = nvb.asInstanceOf[BoxDouble].value
437
        memorySupport.storeDouble(loc, dv, !ptr)
438 439 440
      case _: AbstractPointerType =>
        val addr = nvb.asInstanceOf[BoxPointer].addr
        memorySupport.storeLong(loc, addr, !ptr)
441
      case _: TypeRef =>
442
        noAccessViaPointer(ptr, ty)
443
        val addr = nvb.asInstanceOf[BoxRef].objRef
444
        memorySupport.storeLong(loc, addr)
445
      case _: TypeIRef =>
446
        noAccessViaPointer(ptr, ty)
447
        val BoxIRef(base, offset) = nvb.asInstanceOf[BoxIRef]
448 449
        memorySupport.storeLong(loc, base)
        memorySupport.storeLong(loc + WORD_SIZE_BYTES, offset)
450 451 452 453
      case _: TypeTagRef64 =>
        noAccessViaPointer(ptr, ty)
        val raw = nvb.asInstanceOf[BoxTagRef64].raw
        memorySupport.storeLong(loc, raw)
Kunshan Wang's avatar
Kunshan Wang committed
454 455 456 457 458
      case _: AbstractOpaqueRefType => {
        def storeEntity[T: CanBeIntegerized](): Unit = {
          val obj = nvb.asInstanceOf[BoxOpaque[T]].obj
          memorySupport.storeEntity[T](loc, obj)
        }
459
        noAccessViaPointer(ptr, ty)
Kunshan Wang's avatar
Kunshan Wang committed
460 461 462 463 464 465 466 467
        ty match {
          case _: TypeFuncRef        => storeEntity[Function]()
          case _: TypeThreadRef      => storeEntity[InterpreterThread]()
          case _: TypeStackRef       => storeEntity[InterpreterStack]()
          case _: TypeFrameCursorRef => storeEntity[FrameCursor]()
          case _: TypeIRBuilderRef   => storeEntity[IRBuilder]()
        }
      }
468
      case _ => throw new UvmUnimplementedOperationException("Storing of type %s is not supporing".format(ty.getClass.getName))
469 470 471 472
    }

    ty match {
      case TypeVector(ety, len) =>
Kunshan Wang's avatar
Kunshan Wang committed
473
        val nvbs = nvb.asInstanceOf[BoxSeq].values
474
        val elemSkip = alignUp(sizeOf(ety), alignOf(ety))
Kunshan Wang's avatar
Kunshan Wang committed
475 476
        for ((nvbElem, i) <- nvbs.zipWithIndex) {
          storeScalar(ety, loc + elemSkip * i, nvbElem)
477
        }
Kunshan Wang's avatar
Kunshan Wang committed
478
      case sty => storeScalar(sty, loc, nvb)
479 480 481 482 483 484
    }
  }

  /**
   * Compare exchange. The result (the old value) is written into br. Return true if successful, false otherwise.
   */
485
  def cmpXchg(ptr: Boolean, ty: Type, loc: Word, eb: ValueBox, db: ValueBox, br: ValueBox)(implicit microVM: MicroVM, memorySupport: MemorySupport): Boolean = {
486 487 488 489 490 491
    ty match {
      case TypeInt(l) =>
        val ebi = eb.asInstanceOf[BoxInt].value
        val dbi = db.asInstanceOf[BoxInt].value
        val (succ, rbi) = l match {
          case 32 => {
492
            val (succ2, rv) = memorySupport.cmpXchgInt(loc, ebi.intValue, dbi.intValue, !ptr)
493 494 495
            (succ2, BigInt(rv))
          }
          case 64 => {
496
            val (succ2, rv) = memorySupport.cmpXchgLong(loc, ebi.longValue, dbi.longValue, !ptr)
497 498
            (succ2, BigInt(rv))
          }
499
          case _ => throw new UvmUnimplementedOperationException("CmpXchg on int of length %d is not supported".format(l))
500 501 502
        }
        br.asInstanceOf[BoxInt].value = OpHelper.unprepare(rbi, l)
        succ
503 504 505 506 507 508
      case _: AbstractPointerType =>
        val el = eb.asInstanceOf[BoxPointer].addr
        val dl = db.asInstanceOf[BoxPointer].addr
        val (succ, rl) = memorySupport.cmpXchgLong(loc, el, dl, !ptr)
        br.asInstanceOf[BoxPointer].addr = rl
        succ
509
      case _: TypeRef =>
510
        noAccessViaPointer(ptr, ty)
511 512
        val el = eb.asInstanceOf[BoxRef].objRef
        val dl = db.asInstanceOf[BoxRef].objRef
513
        val (succ, rl) = memorySupport.cmpXchgLong(loc, el, dl)
514 515 516
        br.asInstanceOf[BoxRef].objRef = rl
        succ
      case _: TypeIRef =>
517
        noAccessViaPointer(ptr, ty)
518 519
        val BoxIRef(el, eh) = eb.asInstanceOf[BoxIRef]
        val BoxIRef(dl, dh) = db.asInstanceOf[BoxIRef]
520
        val (succ, (rl, rh)) = memorySupport.cmpXchgI128(loc, (el, eh), (dl, dh))
521 522
        br.asInstanceOf[BoxIRef].oo = (rl, rh)
        succ
Kunshan Wang's avatar
Kunshan Wang committed
523 524 525 526 527 528 529 530
      case _: AbstractOpaqueRefType => {
        def cmpXchgEntity[T: CanBeIntegerized](): Boolean = {
          val eObj = eb.asInstanceOf[BoxOpaque[T]].obj
          val dObj = db.asInstanceOf[BoxOpaque[T]].obj
          val (succ, rObj) = memorySupport.cmpXchgEntity(loc, eObj, dObj)
          br.asInstanceOf[BoxOpaque[T]].obj = rObj
          succ
        }
531
        noAccessViaPointer(ptr, ty)
Kunshan Wang's avatar
Kunshan Wang committed
532 533 534 535 536 537 538 539
        ty match {
          case _: TypeFuncRef        => cmpXchgEntity[Function]()
          case _: TypeThreadRef      => cmpXchgEntity[InterpreterThread]()
          case _: TypeStackRef       => cmpXchgEntity[InterpreterStack]()
          case _: TypeFrameCursorRef => cmpXchgEntity[FrameCursor]()
          case _: TypeIRBuilderRef   => cmpXchgEntity[IRBuilder]()
        }
      }
540
      case _ => throw new UvmUnimplementedOperationException("CmpXchg of type %s is not supporing".format(ty.getClass.getName))
541 542 543
    }
  }

544
  def atomicRMW(ptr: Boolean, ty: Type, op: AtomicRMWOptr, loc: Word, ob: ValueBox, br: ValueBox)(implicit microVM: MicroVM, memorySupport: MemorySupport): Unit = {
545 546 547 548
    ty match {
      case TypeInt(l) =>
        val obi = ob.asInstanceOf[BoxInt].value
        val rbi: BigInt = l match {
549 550
          case 32 => memorySupport.atomicRMWInt(op, loc, obi.intValue, !ptr)
          case 64 => memorySupport.atomicRMWLong(op, loc, obi.longValue, !ptr)
551
          case _  => throw new UvmUnimplementedOperationException("AtomicRMW on int of length %d is not supported".format(l))
552 553 554 555
        }
        br.asInstanceOf[BoxInt].value = OpHelper.unprepare(rbi, l)
      case _ =>
        if (op != XCHG) {
556
          throw new UvmUnimplementedOperationException("AtomicRMW operation other than XCHG only supports int. %s found.".format(ty.getClass.getName))
557 558
        } else {
          ty match {
559 560 561 562
            case _: AbstractPointerType =>
              val ol = ob.asInstanceOf[BoxPointer].addr
              val rl = memorySupport.atomicRMWLong(op, loc, ol, !ptr)
              br.asInstanceOf[BoxPointer].addr = rl
563
            case _: TypeRef =>
564
              noAccessViaPointer(ptr, ty)
565
              val ol = ob.asInstanceOf[BoxRef].objRef
566
              val rl = memorySupport.atomicRMWLong(op, loc, ol)
567 568
              br.asInstanceOf[BoxRef].objRef = rl
            case _: TypeIRef =>
569
              noAccessViaPointer(ptr, ty)
570
              val BoxIRef(ol, oh) = ob.asInstanceOf[BoxIRef]
571
              val (rl, rh) = memorySupport.xchgI128(loc, (ol, oh))
572
              br.asInstanceOf[BoxIRef].oo = (rl, rh)
Kunshan Wang's avatar
Kunshan Wang committed
573 574 575 576 577 578
            case _: AbstractOpaqueRefType => {
              def xchgEntity[T: CanBeIntegerized](): Unit = {
                val obj = ob.asInstanceOf[BoxOpaque[T]].obj
                val oldObj = memorySupport.xchgEntity(loc, obj)
                br.asInstanceOf[BoxOpaque[T]].obj = oldObj
              }
579
              noAccessViaPointer(ptr, ty)
Kunshan Wang's avatar
Kunshan Wang committed
580 581 582 583 584 585 586 587
              ty match {
                case _: TypeFuncRef        => xchgEntity[Function]()
                case _: TypeThreadRef      => xchgEntity[InterpreterThread]()
                case _: TypeStackRef       => xchgEntity[InterpreterStack]()
                case _: TypeFrameCursorRef => xchgEntity[FrameCursor]()
                case _: TypeIRBuilderRef   => xchgEntity[IRBuilder]()
              }
            }
588
            case _ =>
589
              throw new UvmUnimplementedOperationException("AtomicRMW XCHG of type %s is not supporing".format(ty.getClass.getName))
590 591 592
          }
        }
    }
Kunshan Wang's avatar
Kunshan Wang committed
593
  }
Kunshan Wang's avatar
Kunshan Wang committed
594 595 596 597

  /**
   * Check if a memory location still holds a particular value. Used by futex.
   */
598
  def cmpInt(len: Int, loc: Word, expected: BigInt)(implicit memorySupport: MemorySupport): Boolean = len match {
Kunshan Wang's avatar
Kunshan Wang committed
599
    case 64 => {
600
      val expNum = OpHelper.prepareSigned(expected, len).longValue
601
      val actualNum = memorySupport.loadLong(loc)
Kunshan Wang's avatar
Kunshan Wang committed
602 603 604
      expNum == actualNum
    }
    case 32 => {
605
      val expNum = OpHelper.prepareSigned(expected, len).intValue
606
      val actualNum = memorySupport.loadInt(loc)
Kunshan Wang's avatar
Kunshan Wang committed
607 608
      expNum == actualNum
    }
609
    case _ => throw new UvmUnimplementedOperationException("Futex of %d bit int is not supported".format(len))
Kunshan Wang's avatar
Kunshan Wang committed
610
  }
611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641

  val US_ASCII = Charset.forName("US-ASCII")

  /**
   * Read an ASCII string from the memory.
   *
   * @param loc A ref to a @uvm.meta.bytes object.
   */
  def bytesToStr(loc: Word)(implicit memorySupport: MemorySupport): String = {
    // It is a hybrid<@i64 @i8> object. The length is determined by the fixed part. 
    val len = memorySupport.loadLong(loc)
    val bytes = new Array[Byte](len.toInt)
    val begin = loc + TypeSizes.WORD_SIZE_BYTES
    memorySupport.loadBytes(begin, bytes, 0, len, true)

    val result = new String(bytes, US_ASCII)
    result
  }

  /**
   * Create a Mu @uvm.meta.bytes object to hold an ASCII string.
   *
   * @return The address of the allocated object.
   */
  def strToBytes(str: String)(implicit memorySupport: MemorySupport, mutator: Mutator): Word = {
    val bytes = str.getBytes(US_ASCII)
    val len = bytes.length
    val loc = mutator.newHybrid(InternalTypes.BYTES, len)
    memorySupport.storeLong(loc, bytes.length.toLong)
    val begin = loc + TypeSizes.WORD_SIZE_BYTES
    memorySupport.storeBytes(begin, bytes, 0, len, true)
642

643 644
    loc
  }
Kunshan Wang's avatar
Kunshan Wang committed
645

646 647 648
  /**
   * Load irnoderef value from the memory. Use fake value from microVM.irNodeRegistry
   */
Kunshan Wang's avatar
Kunshan Wang committed
649 650
  def loadIRBuilder(loc: Word)(implicit microVM: MicroVM, memorySupport: MemorySupport): Option[IRBuilder] = {
    memorySupport.loadEntity[IRBuilder](loc)
651 652 653 654 655
  }

  /**
   * Store irnoderef value into the memory. Use fake value from microVM.irNodeRegistry
   */
Kunshan Wang's avatar
Kunshan Wang committed
656 657
  def storeIRBuilder(loc: Word, maybeIRBuilder: Option[IRBuilder])(implicit microVM: MicroVM, memorySupport: MemorySupport): Unit = {
    memorySupport.storeEntity(loc, maybeIRBuilder)
658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676
  }

  def loadInt32Array(base: Word, len: Word)(implicit memorySupport: MemorySupport): IndexedSeq[Int] = {
    if (base == 0L) {
      IndexedSeq[Int]()
    } else {
      for (i <- 0L until len) yield {
        val addr = base + i * 4L
        val v = memorySupport.loadInt(addr)
        v
      }
    }
  }

  def loadInt64Array(base: Word, len: Word)(implicit memorySupport: MemorySupport): IndexedSeq[Long] = {
    if (base == 0L) {
      IndexedSeq[Long]()
    } else {
      for (i <- 0L until len) yield {
Kunshan Wang's avatar
Kunshan Wang committed
677
        val addr = base + i * 8L
678 679 680 681 682
        val v = memorySupport.loadLong(addr)
        v
      }
    }
  }
Kunshan Wang's avatar
Kunshan Wang committed
683 684
}