Added move constructor for MIPSImage

spim/CPU/image.cpp

@@ -1,19 +1,46 @@
+#include "image.h"
 #include "inst.h"
 #include "spim.h"
 #include "spim-utils.h"
-#include "image.h"
 #include "sym-tbl.h"
 
 #include <iostream>
 
-MIPSImage::MIPSImage(int ctx) : ctx(ctx), std_out(ctx, std::cout), std_err(ctx, std::cerr) {}
+MIPSImage::MIPSImage(int ctx) :
+    ctx(ctx),
+    std_out(ctx, std::cout),
+    std_err(ctx, std::cerr)
+{
+    label_hash_table = (label **) zmalloc(*this, LABEL_HASH_TABLE_SIZE * sizeof(label *));
+}
 
 MIPSImage::~MIPSImage() {
     initialize_symbol_table(*this);
     for (auto it = labels_to_free.begin(); it != labels_to_free.end(); it ++) {
         free((*it)->name);
         free(*it);
     }
+    if (label_hash_table)
+        free(label_hash_table);
+}
+
+MIPSImage::MIPSImage(MIPSImage &&other) :
+    ctx(other.ctx),
+    mem_img(std::move(other.mem_img)),
+    reg_img(std::move(other.reg_img)),
+    bkpt_map(std::move(other.bkpt_map)),
+    local_labels(other.local_labels),
+    label_hash_table(other.label_hash_table),
+    labels_to_free(std::move(other.labels_to_free)),
+    std_out(std::move(other.std_out)),
+    std_err(std::move(other.std_err))
+{
+    other.mem_img = {};
+    other.reg_img = {};
+    other.bkpt_map.clear();
+    other.local_labels = NULL;
+    other.label_hash_table = NULL;
+    other.labels_to_free.clear();
 }
 
 int MIPSImage::get_ctx() const {

spim/CPU/image.h

@@ -30,14 +30,16 @@ class MIPSImage {
     std::unordered_map<mem_addr, breakpoint> bkpt_map;
     // std::unordered_map<mem_addr, label> labels;
     label *local_labels = NULL; // No allocs occur here
-    label *label_hash_table[LABEL_HASH_TABLE_SIZE] = {0};
+    label **label_hash_table = NULL; // Points to an array of size LABEL_HASH_TABLE_SIZE
     std::vector<label *> labels_to_free;
 
     MIPSImagePrintStream std_out;
     MIPSImagePrintStream std_err;
   public:
     MIPSImage(int ctx);
     ~MIPSImage();
+    MIPSImage(const MIPSImage&) = delete;
+    MIPSImage(MIPSImage&&);
 
     int get_ctx() const;
 

spim/CPU/image_print_stream.cpp

@@ -21,6 +21,19 @@ MIPSImagePrintStream::~MIPSImagePrintStream() {
     sync(); // Flush the buffer on exit
 }
 
+MIPSImagePrintStream::MIPSImagePrintStream(MIPSImagePrintStream &&other) :
+    ctx(other.ctx),
+    sink(other.sink),
+    buf(std::move(other.buf))
+{
+    char *base = &buf.front();
+    setp(base, base + buf.size() - 1);
+
+    int offset = other.pptr() - other.pbase();
+    pbump(offset);
+    other.pbump(-offset);
+}
+
 std::streamsize MIPSImagePrintStream::xsputn(const char *s, std::streamsize n) {
     std::streamsize char_pos = 0;
     while (char_pos < n) {
@@ -51,6 +64,9 @@ MIPSImagePrintStream::int_type MIPSImagePrintStream::overflow(MIPSImagePrintStre
 int MIPSImagePrintStream::sync() {
     std::ptrdiff_t n = pptr() - pbase();
     pbump(-n);
+    if (!n) {
+        return 0;
+    }
 #ifdef WASM
     char *s = new char[n + 1];
     strncpy(s, pbase(), n);

spim/CPU/image_print_stream.h

@@ -22,6 +22,7 @@ class MIPSImagePrintStream : public std::streambuf {
     public:
         explicit MIPSImagePrintStream(unsigned int ctx, std::ostream &sink, std::size_t buffer_size = 256); 
         MIPSImagePrintStream(const MIPSImagePrintStream&) = delete;
+        MIPSImagePrintStream(MIPSImagePrintStream&&);
         MIPSImagePrintStream& operator=(const MIPSImagePrintStream&) = delete;
         ~MIPSImagePrintStream();
 

spim/CPU/mem.cpp

@@ -227,6 +227,8 @@ void mem_dump_profile(MIPSImage &img) {
 void
 free_instructions (instruction **inst, int n)
 {
+  if (!inst)
+    return;
   for ( ; n > 0; n --, inst ++)
     if (*inst)
       free_inst (*inst);

spim/CPU/sym-tbl.cpp

@@ -70,6 +70,8 @@ static void resolve_a_label_sub (MIPSImage &img, label *sym, instruction *inst,
 void
 initialize_symbol_table (MIPSImage &img)
 {
+  if (!img.get_label_hash_table())
+    return;
   int i;
 
   for (i = 0; i < LABEL_HASH_TABLE_SIZE; i ++)

spim/worker.cpp

@@ -4,6 +4,7 @@
 #include <optional>
 #include <thread>
 #include <condition_variable>
+#include <utility>
 
 #include "CPU/scanner.h"
 #include "CPU/spim-utils.h"
@@ -71,16 +72,14 @@ void reset(unsigned int max_contexts, std::set<unsigned int> &active_ctxs) {
         if (i >= max_contexts) {
             continue;
         }
-        ctxs.emplace(i, i);
-        MIPSImage &new_image = ctxs.at(i);
+        MIPSImage new_image(i);
         initialize_world(new_image, DEFAULT_EXCEPTION_HANDLER, false);
         initialize_run_stack(new_image, 0, nullptr);
         char file_name[64];
         sprintf(file_name, "./input_%d.s", i);
         if (read_assembly_file(new_image, file_name)) { // check if the file exists
             new_image.reg_image().PC = starting_address(new_image);
-        } else {
-            ctxs.erase(i);
+            ctxs.emplace(i, std::move(new_image));
         }
         yylex_destroy();
     }