MicroVM.scala 2.48 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 Bundle()
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
  
Kunshan Wang's avatar
Kunshan Wang committed
31
  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 46 47 48
    // So the GC must know about this type because the GC looks up the globalBundle for types.
    globalBundle.allNs.add(InternalTypes.VOID)
    globalBundle.typeNs.add(InternalTypes.VOID)
    globalBundle.allNs.add(InternalTypes.BYTE)
    globalBundle.typeNs.add(InternalTypes.BYTE)
    globalBundle.allNs.add(InternalTypes.BYTE_ARRAY)
    globalBundle.typeNs.add(InternalTypes.BYTE_ARRAY)
  }
49 50 51 52 53 54 55 56 57 58 59 60 61 62

  /**
   * Add things from a bundle to the Micro VM.
   */
  def addBundle(bundle: Bundle) {
    globalBundle.merge(bundle);

    for (gc <- bundle.globalCellNs.all) {
      memoryManager.globalMemory.addGlobalCell(gc)
    }
    for (g <- bundle.globalVarNs.all) {
      constantPool.addGlobalVar(g)
    }
  }
Kunshan Wang's avatar
Kunshan Wang committed
63

64 65 66
  /**
   * Create a new ClientAgent.
   */
Kunshan Wang's avatar
Kunshan Wang committed
67 68 69 70 71 72
  def newClientAgent(): ClientAgent = {
    val mutator = microVM.memoryManager.heap.makeMutator() // This may trigger GC
    val ca = new ClientAgent(mutator)
    clientAgents.add(ca)
    ca
  }
73 74 75 76
  /**
   * 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
77

78 79 80 81
  /**
   * Given an ID, get the name of an identified entity.
   */
  def nameOf(id: Int): String = globalBundle.allNs(id).name.get
82
}