1 files changed, 115 insertions, 0 deletions
diff --git a/fesvr/dtm.h b/fesvr/dtm.h
new file mode 100644
index 0000000..fbf161e
--- /dev/null
+++ b/fesvr/dtm.h
@@ -0,0 +1,115 @@
+#ifndef _ROCKET_DTM_H
+#define _ROCKET_DTM_H
+
+#include "htif.h"
+#include "context.h"
+#include <stdint.h>
+#include <queue>
+#include <semaphore.h>
+#include <vector>
+#include <string>
+#include <stdlib.h>
+
+// abstract debug transport module
+class dtm_t : public htif_t
+{
+ public:
+  dtm_t(int argc, char**argv);
+  ~dtm_t();
+
+  struct req {
+    uint32_t addr;
+    uint32_t op;
+    uint32_t data;
+  };
+
+  struct resp {
+    uint32_t resp;
+    uint32_t data;
+  };
+
+  void tick(
+    bool  req_ready,
+    bool  resp_valid,
+    resp  resp_bits
+  );
+  // Akin to tick, but the target thread returns a response on every invocation
+  void return_resp(
+    resp  resp_bits
+  );
+
+  
+  bool req_valid() { return req_wait; }
+  req req_bits() { return req_buf; }
+  bool resp_ready() { return true; }
+
+  uint32_t read(uint32_t addr);
+  uint32_t write(uint32_t addr, uint32_t data);
+  void nop();
+
+  uint64_t read_csr(unsigned which);
+  uint64_t write_csr(unsigned which, uint64_t data);
+  uint64_t clear_csr(unsigned which, uint64_t data);
+  uint64_t set_csr(unsigned which, uint64_t data);
+  void fence_i();
+
+  void producer_thread();
+
+ protected:
+  virtual void read_chunk(addr_t taddr, size_t len, void* dst) override;
+  virtual void write_chunk(addr_t taddr, size_t len, const void* src) override;
+  virtual void clear_chunk(addr_t taddr, size_t len) override;
+  virtual size_t chunk_align() override;
+  virtual size_t chunk_max_size() override;
+  virtual void reset() override;
+  virtual void idle() override;
+
+ private:
+  context_t host;
+  context_t* target;
+  pthread_t producer;
+  sem_t req_produce;
+  sem_t req_consume;
+  sem_t resp_produce;
+  sem_t resp_consume;
+  req req_buf;
+  resp resp_buf;
+  bool running;
+
+  uint32_t run_abstract_command(uint32_t command, const uint32_t program[], size_t program_n,
+                                uint32_t data[], size_t data_n);
+
+  void die(uint32_t cmderr);
+  void halt(int);
+  int enumerate_harts();
+  void select_hart(int);
+  void resume(int);
+  uint64_t save_reg(unsigned regno);
+  void restore_reg(unsigned regno, uint64_t val);
+  
+  uint64_t modify_csr(unsigned which, uint64_t data, uint32_t type);
+
+  bool req_wait;
+  bool resp_wait;
+  uint32_t data_base;
+  
+  uint32_t xlen;
+
+  static const int max_idle_cycles = 10000;
+
+  size_t ram_words;
+  size_t data_words;
+  int num_harts;
+  int current_hart;
+  
+  uint32_t get_xlen();
+  uint32_t do_command(dtm_t::req r);
+
+  void parse_args(const std::vector<std::string>& args);
+  void register_devices();
+  void start_host_thread();
+
+  friend class memif_t;
+};
+
+#endif