PrescienceLab
diff --git a/‎src/e9patch/e9api.cpp‎
Lines changed: 17 additions & 2 deletions b/‎src/e9patch/e9api.cpp‎
Lines changed: 17 additions & 2 deletions
diff --git a/‎src/e9patch/e9elf.cpp‎
Lines changed: 0 additions & 2 deletions b/‎src/e9patch/e9elf.cpp‎
Lines changed: 0 additions & 2 deletions
@@ -424,8 +424,23 @@ static void parseEmit(Binary *B, const Message &msg)
     // Create and optimize the mappings:
     MappingSet mappings;
     buildMappings(B->allocator, option_mem_mapping_size, mappings);
-    optimizeMappings(mappings);
-    putchar('\n');
+    switch (option_mem_granularity)
+    {
+        case 64:
+            optimizeMappings<Key64>(B->allocator, option_mem_mapping_size,
+                mappings);
+            break;
+        case 128:
+            optimizeMappings<Key128>(B->allocator, option_mem_mapping_size,
+                mappings);
+            break;
+        case 4096:
+            optimizeMappings<Key4096>(B->allocator, option_mem_mapping_size,
+                mappings);
+            break;
+        default:
+            error("unimplemented granularity (%zu)", option_mem_granularity);
+    }
 
     // Create the patched binary:
     B->patched.size = emitElf(B, mappings, option_mem_mapping_size);
 
@@ -649,11 +649,9 @@ size_t emitElf(const Binary *B, const MappingSet &mappings,
     {
         uint8_t *base = data + size;
         mapping->offset = (off_t)size;
-        printf("[\33[33m%.16lX\33[0m]", mapping->key);
         flattenMapping(base, mapping, /*int3=*/0xcc);
         size += mapping->size;
     }
-    putchar('\n');
 
     // Step (4): Modify the entry address.
     intptr_t old_entry = 0;
Original file line number	Diff line number	Diff line change
`@@ -649,11 +649,9 @@ size_t emitElf(const Binary *B, const MappingSet &mappings,`
`649`	`649`	`{`
`650`	`650`	`uint8_t *base = data + size;`
`651`	`651`	`mapping->offset = (off_t)size;`
`652`		`- printf("[\33[33m%.16lX\33[0m]", mapping->key);`
`653`	`652`	`flattenMapping(base, mapping, /int3=/0xcc);`
`654`	`653`	`size += mapping->size;`
`655`	`654`	`}`
`656`		`- putchar('\n');`
`657`	`655`
`658`	`656`	`// Step (4): Modify the entry address.`
`659`	`657`	`intptr_t old_entry = 0;`