From eca56660a1b22f12f1c38c2db3ad102153eb91c8 Mon Sep 17 00:00:00 2001
From: ko1N <ko1N1337@gmail.com>
Date: Fri, 16 Apr 2021 19:00:20 +0200
Subject: [PATCH] Updated set_mem_map function to latest memflow/next

---
 memflow-pcileech/src/lib.rs | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/memflow-pcileech/src/lib.rs b/memflow-pcileech/src/lib.rs
index 3f35008..161ff56 100644
--- a/memflow-pcileech/src/lib.rs
+++ b/memflow-pcileech/src/lib.rs
@@ -342,9 +342,8 @@ impl PhysicalMemory for PciLeech {
     }
 
     fn set_mem_map(&mut self, mem_map: MemoryMap<(Address, usize)>) {
-        // TODO: check if current mem_map is empty
-        // TODO: update metadata.size
-        self.mem_map = mem_map;
+        self.metadata.size = mem_map.max_address().as_usize();
+        self.mem_map.merge(mem_map)
     }
 }