Refactor how we track in-progress operations.

I think the functionality is unchanged.

5 files changed, 267 insertions, 174 deletions

diff --git a/debug_rom/debug_rom.S b/debug_rom/debug_rom.S
index ca58ee4..9825d48 100755
--- a/debug_rom/debug_rom.S
+++ b/debug_rom/debug_rom.S
@@ -32,6 +32,7 @@ resume:
 clear_debint:
         csrr    s1, CSR_MHARTID
         sw      s1, CLEARDEBINT(zero)
+        # TODO: race: what if the debugger sets debug int at this point?
 clear_debint_loop:
         csrr    s1, DCSR
         andi    s1, s1, (1<<DCSR_DEBUGINT_OFFSET)
diff --git a/riscv/debug_module.h b/riscv/debug_module.h
index 040ad1b..734a2a3 100644
--- a/riscv/debug_module.h
+++ b/riscv/debug_module.h
@@ -19,9 +19,11 @@ class debug_module_t : public abstract_device_t
     uint32_t ram_read32(unsigned int index);
 
     void set_interrupt(uint32_t hartid) {
+      fprintf(stderr, "set debug interrupt 0x%x\n", hartid);
       interrupt.insert(hartid);
     }
     void clear_interrupt(uint32_t hartid) {
+      fprintf(stderr, "clear debug interrupt 0x%x\n", hartid);
       interrupt.erase(hartid);
     }
     bool get_interrupt(uint32_t hartid) const {
diff --git a/riscv/gdbserver.cc b/riscv/gdbserver.cc
index a4f8c0b..629dc27 100644
--- a/riscv/gdbserver.cc
+++ b/riscv/gdbserver.cc
@@ -20,7 +20,29 @@
 #define C_EBREAK        0x9002
 #define EBREAK          0x00100073
 
-// Functions to generate RISC-V opcodes.
+//////////////////////////////////////// Utility Functions
+
+void die(const char* msg)
+{
+  fprintf(stderr, "gdbserver code died: %s\n", msg);
+  abort();
+}
+
+// gdb's register list is defined in riscv_gdb_reg_names gdb/riscv-tdep.c in
+// its source tree. We must interpret the numbers the same here.
+enum {
+  REG_XPR0 = 0,
+  REG_XPR31 = 31,
+  REG_PC = 32,
+  REG_FPR0 = 33,
+  REG_FPR31 = 64,
+  REG_CSR0 = 65,
+  REG_CSR4095 = 4160,
+  REG_END = 4161
+};
+
+//////////////////////////////////////// Functions to generate RISC-V opcodes.
+
 // TODO: Does this already exist somewhere?
 
 // Using regnames.cc as source. The RVG Calling Convention of the 2.0 RISC-V
@@ -66,7 +88,7 @@ static uint32_t sw(unsigned int src, unsigned int base, uint16_t offset)
 static uint32_t sd(unsigned int src, unsigned int base, uint16_t offset)
 {
   return (bits(offset, 11, 5) << 25) |
-    (src << 20) |
+    (bits(src, 4, 0) << 20) |
     (base << 15) |
     (bits(offset, 4, 0) << 7) |
     MATCH_SD;
@@ -138,10 +160,148 @@ void circular_buffer_t<T>::append(const T *src, unsigned int count)
   }
 }
 
+////////////////////////////// Debug Operations
+
+class halt_op_t : public operation_t
+{
+  public:
+    halt_op_t(gdbserver_t& gdbserver) : operation_t(gdbserver) {};
+
+    bool start()
+    {
+      // TODO: For now we just assume the target is 64-bit.
+      gs.write_debug_ram(0, csrsi(DCSR_ADDRESS, DCSR_HALT_MASK));
+      gs.write_debug_ram(1, csrr(S0, DPC_ADDRESS));
+      gs.write_debug_ram(2, sd(S0, 0, (uint16_t) DEBUG_RAM_START));
+      gs.write_debug_ram(3, csrr(S0, DCSR_ADDRESS));
+      gs.write_debug_ram(4, sd(S0, 0, (uint16_t) DEBUG_RAM_START + 8));
+      gs.write_debug_ram(5, jal(0, (uint32_t) (DEBUG_ROM_RESUME - (DEBUG_RAM_START + 4*5))));
+      gs.set_interrupt(0);
+      return false;
+    }
+
+    bool step()
+    {
+      return true;
+    }
+};
+
+class general_registers_read_op_t : public operation_t
+{
+  // Register order that gdb expects is:
+  //   "x0",  "x1",  "x2",  "x3",  "x4",  "x5",  "x6",  "x7",
+  //   "x8",  "x9",  "x10", "x11", "x12", "x13", "x14", "x15",
+  //   "x16", "x17", "x18", "x19", "x20", "x21", "x22", "x23",
+  //   "x24", "x25", "x26", "x27", "x28", "x29", "x30", "x31",
+
+  // Each byte of register data is described by two hex digits. The bytes with
+  // the register are transmitted in target byte order. The size of each
+  // register and their position within the ‘g’ packet are determined by the
+  // gdb internal gdbarch functions DEPRECATED_REGISTER_RAW_SIZE and
+  // gdbarch_register_name.
+
+  public:
+    general_registers_read_op_t(gdbserver_t& gdbserver) :
+      operation_t(gdbserver), current_reg(0) {};
+
+    bool start()
+    {
+      gs.start_packet();
+
+      // x0 is always zero.
+      gs.send((reg_t) 0);
+
+      gs.write_debug_ram(0, sd(1, 0, (uint16_t) DEBUG_RAM_START + 16));
+      gs.write_debug_ram(1, sd(2, 0, (uint16_t) DEBUG_RAM_START + 0));
+      gs.write_debug_ram(2, jal(0, (uint32_t) (DEBUG_ROM_RESUME - (DEBUG_RAM_START + 4*2))));
+      gs.set_interrupt(0);
+      current_reg = 1;
+      return false;
+    }
+
+    bool step()
+    {
+      fprintf(stderr, "step %d\n", current_reg);
+      gs.send(((uint64_t) gs.read_debug_ram(5) << 32) | gs.read_debug_ram(4));
+      if (current_reg >= 31) {
+        gs.end_packet();
+        return true;
+      }
+
+      gs.send(((uint64_t) gs.read_debug_ram(1) << 32) | gs.read_debug_ram(0));
+
+      current_reg += 2;
+      // TODO properly read s0 and s1
+      gs.write_debug_ram(0, sd(current_reg, 0, (uint16_t) DEBUG_RAM_START + 16));
+      gs.write_debug_ram(1, sd(current_reg+1, 0, (uint16_t) DEBUG_RAM_START + 0));
+      gs.write_debug_ram(2, jal(0, (uint32_t) (DEBUG_ROM_RESUME - (DEBUG_RAM_START + 4*2))));
+      gs.set_interrupt(0);
+
+      return false;
+    }
+
+  private:
+    unsigned int current_reg;
+};
+
+class register_read_op_t : public operation_t
+{
+  public:
+    register_read_op_t(gdbserver_t& gdbserver, unsigned int reg) :
+      operation_t(gdbserver), reg(reg) {};
+
+    bool start()
+    {
+      if (reg >= REG_XPR0 && reg <= REG_XPR31) {
+        die("handle_register_read");
+        // send(p->state.XPR[reg - REG_XPR0]);
+      } else if (reg == REG_PC) {
+        gs.write_debug_ram(0, csrr(S0, DPC_ADDRESS));
+        gs.write_debug_ram(1, sd(S0, 0, (uint16_t) DEBUG_RAM_START));
+        gs.write_debug_ram(2, jal(0, (uint32_t) (DEBUG_ROM_RESUME - (DEBUG_RAM_START + 4*2))));
+        gs.set_interrupt(0);
+      } else if (reg >= REG_FPR0 && reg <= REG_FPR31) {
+        die("handle_register_read");
+        // send(p->state.FPR[reg - REG_FPR0]);
+      } else if (reg >= REG_CSR0 && reg <= REG_CSR4095) {
+        try {
+          die("handle_register_read");
+          // send(p->get_csr(reg - REG_CSR0));
+        } catch(trap_t& t) {
+          // It would be nicer to return an error here, but if you do that then gdb
+          // exits out of 'info registers all' as soon as it encounters a register
+          // that can't be read.
+          gs.start_packet();
+          gs.send((reg_t) 0);
+          gs.end_packet();
+        }
+      } else {
+        gs.send_packet("E02");
+        return true;
+      }
+
+      return false;
+    }
+
+    bool step()
+    {
+      gs.start_packet();
+      gs.send(((uint64_t) gs.read_debug_ram(1) << 32) | gs.read_debug_ram(0));
+      gs.end_packet();
+      return true;
+    }
+
+  private:
+    unsigned int reg;
+};
+
+////////////////////////////// gdbserver itself
+
 gdbserver_t::gdbserver_t(uint16_t port, sim_t *sim) :
   sim(sim),
   client_fd(0),
-  recv_buf(64 * 1024), send_buf(64 * 1024)
+  recv_buf(64 * 1024), send_buf(64 * 1024),
+  operation(NULL)
 {
   socket_fd = socket(AF_INET, SOCK_STREAM, 0);
   if (socket_fd == -1) {
@@ -184,17 +344,14 @@ uint32_t gdbserver_t::read_debug_ram(unsigned int index)
   return sim->debug_module.ram_read32(index);
 }
 
-void gdbserver_t::halt()
+void gdbserver_t::set_operation(operation_t* operation)
 {
-  // TODO: For now we just assume the target is 64-bit.
-  write_debug_ram(0, csrsi(DCSR_ADDRESS, DCSR_HALT_MASK));
-  write_debug_ram(1, csrr(S0, DPC_ADDRESS));
-  write_debug_ram(2, sd(S0, 0, (uint16_t) DEBUG_RAM_START));
-  write_debug_ram(3, csrr(S0, DCSR_ADDRESS));
-  write_debug_ram(4, sd(S0, 0, (uint16_t) DEBUG_RAM_START + 8));
-  write_debug_ram(5, jal(0, (uint32_t) (DEBUG_ROM_RESUME - (DEBUG_RAM_START + 4*5))));
-  sim->debug_module.set_interrupt(0);
-  state = STATE_HALTING;
+  assert(this->operation == NULL || operation == NULL);
+  if (operation && operation->start()) {
+    delete operation;
+  } else {
+    this->operation = operation;
+  }
 }
 
 void gdbserver_t::accept()
@@ -215,7 +372,7 @@ void gdbserver_t::accept()
     extended_mode = false;
 
     // gdb wants the core to be halted when it attaches.
-    halt();
+    set_operation(new halt_op_t(*this));
   }
 }
 
@@ -365,59 +522,13 @@ void gdbserver_t::handle_halt_reason(const std::vector<uint8_t> &packet)
   send_packet("S00");
 }
 
-void die(const char* msg)
-{
-  fprintf(stderr, "gdbserver code died: %s\n", msg);
-  abort();
-}
-
 void gdbserver_t::handle_general_registers_read(const std::vector<uint8_t> &packet)
 {
-  // Register order that gdb expects is:
-  //   "x0",  "x1",  "x2",  "x3",  "x4",  "x5",  "x6",  "x7",
-  //   "x8",  "x9",  "x10", "x11", "x12", "x13", "x14", "x15",
-  //   "x16", "x17", "x18", "x19", "x20", "x21", "x22", "x23",
-  //   "x24", "x25", "x26", "x27", "x28", "x29", "x30", "x31",
-
-  // Each byte of register data is described by two hex digits. The bytes with
-  // the register are transmitted in target byte order. The size of each
-  // register and their position within the ‘g’ packet are determined by the
-  // gdb internal gdbarch functions DEPRECATED_REGISTER_RAW_SIZE and
-  // gdbarch_register_name.
-
-  send("$");
-  running_checksum = 0;
-  processor_t *p = sim->get_core(0);
-
-  // x0 is always zero.
-  send((reg_t) 0);
-
-  write_debug_ram(0, sd(1, 0, (uint16_t) DEBUG_RAM_START + 16));
-  write_debug_ram(1, sd(2, 0, (uint16_t) DEBUG_RAM_START + 0));
-  write_debug_ram(2, jal(0, (uint32_t) (DEBUG_ROM_RESUME - (DEBUG_RAM_START + 4*2))));
-  sim->debug_module.set_interrupt(0);
-  state = STATE_CONT_GENERAL_REGISTERS;
-  state_argument = 1;
+  set_operation(new general_registers_read_op_t(*this));
 }
 
-void gdbserver_t::continue_general_registers_read()
-{
-  send(((uint64_t) read_debug_ram(5) << 32) | read_debug_ram(4));
-  if (state_argument >= 31) {
-    send_running_checksum();
-    expect_ack = true;
-    state = STATE_HALTED;
-  } else {
-    send(((uint64_t) read_debug_ram(1) << 32) | read_debug_ram(0));
-
-    state_argument += 2;
-    // TODO properly read s0 and s1
-    write_debug_ram(0, sd(state_argument, 0, (uint16_t) DEBUG_RAM_START + 16));
-    write_debug_ram(1, sd(state_argument+1, 0, (uint16_t) DEBUG_RAM_START + 0));
-    write_debug_ram(2, jal(0, (uint32_t) (DEBUG_ROM_RESUME - (DEBUG_RAM_START + 4*2))));
-    sim->debug_module.set_interrupt(0);
-    state = STATE_CONT_GENERAL_REGISTERS;
-  }
+void gdbserver_t::set_interrupt(uint32_t hartid) {
+  sim->debug_module.set_interrupt(hartid);
 }
 
 // First byte is the most-significant one.
@@ -471,19 +582,6 @@ void consume_string(std::string &str, std::vector<uint8_t>::const_iterator &iter
   }
 }
 
-// gdb's register list is defined in riscv_gdb_reg_names gdb/riscv-tdep.c in
-// its source tree. We must interpret the numbers the same here.
-enum {
-  REG_XPR0 = 0,
-  REG_XPR31 = 31,
-  REG_PC = 32,
-  REG_FPR0 = 33,
-  REG_FPR31 = 64,
-  REG_CSR0 = 65,
-  REG_CSR4095 = 4160,
-  REG_END = 4161
-};
-
 void gdbserver_t::handle_register_read(const std::vector<uint8_t> &packet)
 {
   // p n
@@ -493,46 +591,7 @@ void gdbserver_t::handle_register_read(const std::vector<uint8_t> &packet)
   if (*iter != '#')
     return send_packet("E01");
 
-  state = STATE_CONT_REGISTER_READ;
-  state_argument = n;
-
-  if (n >= REG_XPR0 && n <= REG_XPR31) {
-    die("handle_register_read");
-    // send(p->state.XPR[n - REG_XPR0]);
-  } else if (n == REG_PC) {
-    write_debug_ram(0, csrr(S0, DPC_ADDRESS));
-    write_debug_ram(1, sd(S0, 0, (uint16_t) DEBUG_RAM_START));
-    write_debug_ram(2, jal(0, (uint32_t) (DEBUG_ROM_RESUME - (DEBUG_RAM_START + 4*2))));
-    sim->debug_module.set_interrupt(0);
-  } else if (n >= REG_FPR0 && n <= REG_FPR31) {
-    die("handle_register_read");
-    // send(p->state.FPR[n - REG_FPR0]);
-  } else if (n >= REG_CSR0 && n <= REG_CSR4095) {
-    try {
-      die("handle_register_read");
-      // send(p->get_csr(n - REG_CSR0));
-    } catch(trap_t& t) {
-      // It would be nicer to return an error here, but if you do that then gdb
-      // exits out of 'info registers all' as soon as it encounters a register
-      // that can't be read.
-      send((reg_t) 0);
-    }
-  } else {
-    state = STATE_HALTED;
-    return send_packet("E02");
-  }
-}
-
-void gdbserver_t::continue_register_read()
-{
-  send("$");
-  running_checksum = 0;
-
-  send(((uint64_t) read_debug_ram(1) << 32) | read_debug_ram(0));
-
-  send_running_checksum();
-  expect_ack = true;
-  state = STATE_HALTED;
+  set_operation(new register_read_op_t(*this, n));
 }
 
 void gdbserver_t::handle_register_write(const std::vector<uint8_t> &packet)
@@ -585,8 +644,7 @@ void gdbserver_t::handle_memory_read(const std::vector<uint8_t> &packet)
   if (*iter != '#')
     return send_packet("E11");
 
-  send("$");
-  running_checksum = 0;
+  start_packet();
   char buffer[3];
   processor_t *p = sim->get_core(0);
   mmu_t* mmu = sim->debug_mmu;
@@ -595,7 +653,7 @@ void gdbserver_t::handle_memory_read(const std::vector<uint8_t> &packet)
     sprintf(buffer, "%02x", mmu->load_uint8(address + i));
     send(buffer);
   }
-  send_running_checksum();
+  end_packet();
 }
 
 void gdbserver_t::handle_memory_binary_write(const std::vector<uint8_t> &packet)
@@ -749,8 +807,7 @@ void gdbserver_t::handle_query(const std::vector<uint8_t> &packet)
   if (iter != packet.end())
     iter++;
   if (name == "Supported") {
-    send("$");
-    running_checksum = 0;
+    start_packet();
     while (iter != packet.end()) {
       std::string feature;
       consume_string(feature, iter, packet.end(), ';');
@@ -760,7 +817,7 @@ void gdbserver_t::handle_query(const std::vector<uint8_t> &packet)
         send("swbreak+;");
       }
     }
-    return send_running_checksum();
+    return end_packet();
   }
 
   fprintf(stderr, "Unsupported query %s\n", name.c_str());
@@ -833,25 +890,23 @@ void gdbserver_t::handle()
 
     bool interrupt = sim->debug_module.get_interrupt(0);
 
-    if (state == STATE_HALTING && !interrupt) {
-      // gdb requested a halt and now it's done.
-      send_packet("T05");
-      fprintf(stderr, "DPC: 0x%x\n", read_debug_ram(0));
-      fprintf(stderr, "DCSR: 0x%x\n", read_debug_ram(2));
-      state = STATE_HALTED;
-    }
-
     if (!interrupt) {
+      if (operation && operation->step()) {
+        delete operation;
+        set_operation(NULL);
+      }
+
+      /*
       switch (state) {
-        case STATE_CONT_GENERAL_REGISTERS:
-          continue_general_registers_read();
-          break;
-        case STATE_CONT_REGISTER_READ:
-          continue_register_read();
-          break;
-        default:
+        case STATE_HALTING:
+          // gdb requested a halt and now it's done.
+          send_packet("T05");
+          fprintf(stderr, "DPC: 0x%x\n", read_debug_ram(0));
+          fprintf(stderr, "DCSR: 0x%x\n", read_debug_ram(2));
+          state = STATE_HALTED;
           break;
       }
+      */
     }
 
     /* TODO
@@ -878,20 +933,16 @@ void gdbserver_t::handle()
     }
       */
 
-    if (state == STATE_HALTED) {
-      this->read();
-      //p->debug = false;
-    } else {
-      //p->debug = true;
-    }
-
+    this->read();
     this->write();
 
   } else {
     this->accept();
   }
 
-  this->process_requests();
+  if (!operation) {
+    this->process_requests();
+  }
 }
 
 void gdbserver_t::send(const char* msg)
@@ -924,16 +975,26 @@ void gdbserver_t::send(uint32_t value)
 
 void gdbserver_t::send_packet(const char* data)
 {
-  send("$");
-  running_checksum = 0;
+  start_packet();
   send(data);
-  send_running_checksum();
+  end_packet();
   expect_ack = true;
 }
 
-void gdbserver_t::send_running_checksum()
+void gdbserver_t::start_packet()
 {
+  send("$");
+  running_checksum = 0;
+}
+
+void gdbserver_t::end_packet(const char* data)
+{
+  if (data) {
+    send(data);
+  }
+
   char checksum_string[4];
   sprintf(checksum_string, "#%02x", running_checksum);
   send(checksum_string);
+  expect_ack = true;
 }
diff --git a/riscv/gdbserver.h b/riscv/gdbserver.h
index 156001d..35e12b8 100644
--- a/riscv/gdbserver.h
+++ b/riscv/gdbserver.h
@@ -54,6 +54,32 @@ public:
   void remove(mmu_t* mmu);
 };
 
+class gdbserver_t;
+
+class operation_t
+{
+  public:
+    operation_t(gdbserver_t& gdbserver) : gs(gdbserver) {}
+    virtual ~operation_t() {}
+
+    // Called when the operation is first set as the current one.
+    // Return true if this operation is complete. In that case the object will
+    // be deleted.
+    // Return false if more steps are required the next time the debug
+    // interrupt is clear.
+    virtual bool start() { return true; }
+
+    // Perform the next step of this operation (which is probably to write to
+    // Debug RAM and assert the debug interrupt).
+    // Return true if this operation is complete. In that case the object will
+    // be deleted.
+    // Return false if more steps are required the next time the debug
+    // interrupt is clear.
+    virtual bool step() = 0;
+
+    gdbserver_t& gs;
+};
+
 class gdbserver_t
 {
 public:
@@ -84,7 +110,27 @@ public:
 
   bool connected() const { return client_fd > 0; }
 
-  void halt();
+  // TODO: Move this into its own packet sending class?
+  // Add the given message to send_buf.
+  void send(const char* msg);
+  // Hex-encode a 64-bit value, and send it to gcc in target byte order (little
+  // endian).
+  void send(uint64_t value);
+  // Hex-encode a 32-bit value, and send it to gcc in target byte order (little
+  // endian).
+  void send(uint32_t value);
+  void send_packet(const char* data);
+  uint8_t running_checksum;
+  // Send "$" and clear running checksum.
+  void start_packet();
+  // Send "#" and checksum.
+  void end_packet(const char* data=NULL);
+
+  // Write value to the index'th word in Debug RAM.
+  void write_debug_ram(unsigned int index, uint32_t value);
+  uint32_t read_debug_ram(unsigned int index);
+
+  void set_interrupt(uint32_t hartid);
 
 private:
   sim_t *sim;
@@ -98,15 +144,6 @@ private:
   // Used to track whether we think the target is running. If we think it is
   // but it isn't, we need to tell gdb about it.
   bool running;
-  enum {
-    STATE_UNKNOWN,
-    STATE_RUNNING,
-    STATE_HALTING,
-    STATE_HALTED,
-    STATE_CONT_GENERAL_REGISTERS,
-    STATE_CONT_REGISTER_READ,
-  } state;
-  uint32_t state_argument;
 
   std::map <reg_t, software_breakpoint_t> breakpoints;
 
@@ -117,21 +154,9 @@ private:
   void accept();
   // Process all complete requests in recv_buf.
   void process_requests();
-  // Add the given message to send_buf.
-  void send(const char* msg);
-  // Hex-encode a 64-bit value, and send it to gcc in target byte order (little
-  // endian).
-  void send(uint64_t value);
-  // Hex-encode a 32-bit value, and send it to gcc in target byte order (little
-  // endian).
-  void send(uint32_t value);
-  void send_packet(const char* data);
-  uint8_t running_checksum;
-  void send_running_checksum();
 
-  // Write value to the index'th word in Debug RAM.
-  void write_debug_ram(unsigned int index, uint32_t value);
-  uint32_t read_debug_ram(unsigned int index);
+  operation_t* operation;
+  void set_operation(operation_t* operation);
 };
 
 #endif
diff --git a/riscv/processor.cc b/riscv/processor.cc
index 9b97120..ad84b6e 100644
--- a/riscv/processor.cc
+++ b/riscv/processor.cc
@@ -470,7 +470,8 @@ reg_t processor_t::get_csr(int which)
     case CSR_MEDELEG: return state.medeleg;
     case CSR_MIDELEG: return state.mideleg;
     case DCSR_ADDRESS:
-      return
+      {
+        uint32_t value =
           (1 << DCSR_XDEBUGVER_OFFSET) |
           (0 << DCSR_HWBPCOUNT_OFFSET) |
           (0 << DCSR_NDRESET_OFFSET) |
@@ -486,6 +487,9 @@ reg_t processor_t::get_csr(int which)
           (state.dcsr.ebreaku << DCSR_EBREAKU_OFFSET) |
           (state.dcsr.halt << DCSR_HALT_OFFSET) |
           (state.dcsr.cause << DCSR_CAUSE_OFFSET);
+        fprintf(stderr, "DCSR: 0x%x\n", value);
+        return value;
+      }
     case DPC_ADDRESS:
       return state.dpc;
     case DSCRATCH_ADDRESS: