libsgxstep: fix remap memory leak.

jovanbulck · Apr 25, 2019 · e982624 · e982624
1 parent 0ce3066
commit e982624
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/libsgxstep/pt.c b/libsgxstep/pt.c
@@ -90,6 +90,11 @@ void *remap(uint64_t phys)
     return (void*) virt;
 }
 
+void free_map(void *p)
+{
+    ASSERT( !munmap((void*)(((uintptr_t) p) & ~PFN_MASK), 0x1000) );
+}
+
 void *remap_page_table_level( void *address, pt_level_t level )
 {
 	address_mapping_t *map = get_mappings( address );
@@ -241,6 +246,7 @@ void print_pte_adrs( void *adrs)
     uint64_t *pte = NULL;
     ASSERT( pte = remap_page_table_level( adrs, PTE) );
     print_pte(pte);
+    free_map(pte);
 }
 
 void print_pte( uint64_t *pte )

diff --git a/libsgxstep/pt.h b/libsgxstep/pt.h
@@ -161,6 +161,7 @@ void __attribute__((destructor))  tear_down_sgx_step( void );
 typedef enum {PGD, PUD, PMD, PTE, PAGE} pt_level_t;
 
 void *remap(uint64_t phys);
+void free_map(void *p);
 void *remap_page_table_level( void *address, pt_level_t level );
 void flush_tlb(void *adrs);