MicroVM.scala 3.04 KB
Newer Older
1 2 3 4 5 6 7 8 9
package uvm.refimpl

import uvm._
import uvm.refimpl.itpr._
import uvm.refimpl.mem._
import uvm.refimpl.mem.TypeSizes.Word
import scala.collection.mutable.HashSet
import uvm.ir.textinput.UIRTextReader
import uvm.ir.textinput.IDFactory
Kunshan Wang's avatar
Kunshan Wang committed
10
import uvm.refimpl.nat.NativeCallHelper
11 12 13 14 15 16 17 18

object MicroVM {
  val DEFAULT_HEAP_SIZE: Word = 4L * 1024L * 1024L; // 4MiB
  val DEFAULT_GLOBAL_SIZE: Word = 1L * 1024L * 1024L; // 1MiB
  val DEFAULT_STACK_SIZE: Word = 63L * 1024L; // 60KiB per stack
}

class MicroVM(heapSize: Word = MicroVM.DEFAULT_HEAP_SIZE,
Kunshan Wang's avatar
Kunshan Wang committed
19 20 21
              globalSize: Word = MicroVM.DEFAULT_GLOBAL_SIZE,
              stackSize: Word = MicroVM.DEFAULT_STACK_SIZE) {

22 23
  // implicitly injected resources
  private implicit val microVM = this
24

25
  val globalBundle = new GlobalBundle()
26 27
  val constantPool = new ConstantPool()
  val memoryManager = new MemoryManager(heapSize, globalSize, stackSize)
Kunshan Wang's avatar
Kunshan Wang committed
28 29

  private implicit val memorySupport = memoryManager.memorySupport
Kunshan Wang's avatar
Kunshan Wang committed
30

31
  implicit val nativeCallHelper = new NativeCallHelper()
Kunshan Wang's avatar
Kunshan Wang committed
32

33 34
  val threadStackManager = new ThreadStackManager()
  val trapManager = new TrapManager()
Kunshan Wang's avatar
Kunshan Wang committed
35
  val contexts = new HashSet[MuCtx]()
Kunshan Wang's avatar
Kunshan Wang committed
36

37
  val irReader = new UIRTextReader(new IDFactory())
Kunshan Wang's avatar
Kunshan Wang committed
38

39
  {
Kunshan Wang's avatar
Kunshan Wang committed
40
    // The micro VM allocates stacks on the heap in the large object space. It is represented as a bug chunk of byte array.
41 42 43 44 45
    // So the GC must know about this type because the GC looks up the globalBundle for types.
    globalBundle.typeNs.add(InternalTypes.VOID)
    globalBundle.typeNs.add(InternalTypes.BYTE)
    globalBundle.typeNs.add(InternalTypes.BYTE_ARRAY)
  }
46 47 48 49

  /**
   * Add things from a bundle to the Micro VM.
   */
50
  def addBundle(bundle: TrantientBundle) {
51 52 53 54 55
    globalBundle.merge(bundle);

    for (gc <- bundle.globalCellNs.all) {
      memoryManager.globalMemory.addGlobalCell(gc)
    }
56
    for (ef <- bundle.expFuncNs.all) {
57
      nativeCallHelper.exposeFuncStatic(ef)
58 59
    }
    // Must allocate the memory and expose the functions before making constants.
60 61 62 63
    for (g <- bundle.globalVarNs.all) {
      constantPool.addGlobalVar(g)
    }
  }
Kunshan Wang's avatar
Kunshan Wang committed
64

65
  /**
Kunshan Wang's avatar
Kunshan Wang committed
66
   * Create a new MuCtx. Part of the API.
67
   */
Kunshan Wang's avatar
Kunshan Wang committed
68
  def newContext(): MuCtx = {
Kunshan Wang's avatar
Kunshan Wang committed
69
    val mutator = microVM.memoryManager.heap.makeMutator() // This may trigger GC
Kunshan Wang's avatar
Kunshan Wang committed
70 71
    val ca = new MuCtx(mutator)
    contexts.add(ca)
Kunshan Wang's avatar
Kunshan Wang committed
72 73
    ca
  }
74 75 76
  /**
   * Given a name, get the ID of an identified entity.
   */
Kunshan Wang's avatar
Kunshan Wang committed
77 78 79 80 81
  def idOf(name: String): Int = try {
    globalBundle.allNs(name).id
  } catch {
    case e: NoSuchElementException => throw new UvmRefImplException("No Mu entity has name %s.".format(name), e)
  }
Kunshan Wang's avatar
Kunshan Wang committed
82

83 84 85
  /**
   * Given an ID, get the name of an identified entity.
   */
Kunshan Wang's avatar
Kunshan Wang committed
86 87 88 89 90
  def nameOf(id: Int): String = try {
    globalBundle.allNs(id).name.get
  } catch {
    case e: NoSuchElementException => throw new UvmRefImplException("No Mu entity has ID %d.".format(id), e)
  }
Kunshan Wang's avatar
Kunshan Wang committed
91
  
Kunshan Wang's avatar
Kunshan Wang committed
92 93 94 95 96 97 98
  /**
   * Set the trap handler.
   */
  def setTrapHandler(trapHandler: TrapHandler): Unit = {
    trapManager.trapHandler = trapHandler
  }

Kunshan Wang's avatar
Kunshan Wang committed
99 100 101 102 103 104
  /**
   * Execute. This is the external pusher of the execution.
   */
  def execute(): Unit = {
    threadStackManager.execute()
  }
105
}