added input params to generate memory files through cmd (#327)

* added input params to generate memory files through cmd

* removed comments

---------

Co-authored-by: lanaivina <31368580+lana-shanghai@users.noreply.github.com>

src/cmd/cmd.zig

@@ -145,6 +145,7 @@ var app = &cli.App{
                         &enable_trace,
                         &program_option,
                         &output_trace,
+                        &output_memory,
                     },
                     // Action to be executed for the subcommand.
                     .target = .{ .action = .{ .exec = execute } },

src/vm/cairo_run.zig

@@ -63,24 +63,25 @@ pub fn runConfig(allocator: Allocator, config: Config) !void {
     try runner.endRun();
     // TODO readReturnValues necessary for builtins
 
-    if (config.output_trace) |trace_path| {
+    if (config.output_trace != null or config.output_memory != null) {
         try runner.relocate();
+    }
 
+    if (config.output_trace) |trace_path| {
         const trace_file = try std.fs.cwd().createFile(trace_path, .{});
         defer trace_file.close();
 
         var trace_writer = trace_file.writer();
         try writeEncodedTrace(runner.relocated_trace, &trace_writer);
     }
 
-    // blocked until memory relocation is implemented
-    // if (config.output_memory) |mem_path| {
-    //     const mem_file = try std.fs.cwd().createFile(mem_path, .{});
-    //     defer mem_file.close();
+    if (config.output_memory) |mem_path| {
+        const mem_file = try std.fs.cwd().createFile(mem_path, .{});
+        defer mem_file.close();
 
-    //     var mem_writer = mem_file.writer();
-    //     try writeEncodedMemory(runner.relocated_trace, &mem_writer);
-    // }
+        var mem_writer = mem_file.writer();
+        try writeEncodedMemory(runner.relocated_memory.items, &mem_writer);
+    }
 }
 
 const expect = std.testing.expect;

src/vm/runners/cairo_runner.zig

@@ -327,7 +327,7 @@ pub const CairoRunner = struct {
 
         const relocation_table = try self.vm.segments.relocateSegments(self.allocator);
         try self.vm.relocateTrace(relocation_table);
-        // relocate_memory here
+        try self.relocateMemory(relocation_table);
         self.relocated_trace = try self.vm.getRelocatedTrace();
     }