src/platform/Memory.cpp

   1 #include "Memory.h"
   2 #include "BlockEvent.h"
   3
   4 namespace Aesalon {
   5 namespace Platform {
   6
   7 Misc::SmartPointer<MemoryBlock> Memory::get_block(MemoryAddress address) const {
   8     for(block_set_t::const_iterator i = block_set.begin(); i != block_set.end(); i ++) {
   9         if((*i)->get_address() == address) return *i;
  10     }
  11     return NULL;
  12 }
  13
  14 void Memory::remove_block(Misc::SmartPointer<MemoryBlock> block) {
  15     for(block_set_t::iterator i = block_set.begin(); i != block_set.end(); i ++) {
  16         if((*i) != block) continue;
  17         block_set.erase(i);
  18         break;
  19     }
  20 }
  21
  22 void Memory::handle_event(Misc::SmartPointer<Event> event) {
  23     if(event->get_type() == Event::BLOCK_EVENT) {
  24         Misc::SmartPointer<BlockEvent> be = event.to<BlockEvent>();
  25         switch(be->get_block_type()) {
  26             case BlockEvent::ALLOC_EVENT:
  27                 add_block(new MemoryBlock(be->get_address(), be->get_size()));
  28                 break;
  29             case BlockEvent::FREE_EVENT:
  30                 remove_block(get_block(be->get_address()));
  31                 break;
  32             case BlockEvent::REALLOC_EVENT:
  33                 Misc::SmartPointer<MemoryBlock> block = get_block(be->get_address());
  34                 block->set_address(be->get_new_address());
  35                 block->set_size(be->get_size());
  36                 break;
  37         }
  38     }
  39 }
  40
  41 } // namespace Platform
  42 } // namespace Aesalon