MicroVM.scala 2.66 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()
35
  val clientAgents = new HashSet[ClientAgent]()
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 66 67
  /**
   * Create a new ClientAgent.
   */
Kunshan Wang's avatar
Kunshan Wang committed
68 69 70 71 72 73
  def newClientAgent(): ClientAgent = {
    val mutator = microVM.memoryManager.heap.makeMutator() // This may trigger GC
    val ca = new ClientAgent(mutator)
    clientAgents.add(ca)
    ca
  }
74 75 76 77
  /**
   * Given a name, get the ID of an identified entity.
   */
  def idOf(name: String): Int = globalBundle.allNs(name).id
Kunshan Wang's avatar
Kunshan Wang committed
78

79 80 81 82
  /**
   * Given an ID, get the name of an identified entity.
   */
  def nameOf(id: Int): String = globalBundle.allNs(id).name.get
Kunshan Wang's avatar
Kunshan Wang committed
83 84 85 86 87 88 89
  
  /**
   * Execute. This is the external pusher of the execution.
   */
  def execute(): Unit = {
    threadStackManager.execute()
  }
90
}