From eef63c3da8e85f9f155dbc313ec8a87bafd883fd Mon Sep 17 00:00:00 2001
From: Johannes Stoelp <johannes.stoelp@gmail.com>
Date: Fri, 27 Oct 2023 00:13:22 +0200
Subject: lt_bus: add tagged initiator socket and dmi invalidate support

---
 src/models/lt_bus.h | 77 ++++++++++++++++++++++++++++++++++++++++++-----------
 1 file changed, 61 insertions(+), 16 deletions(-)

diff --git a/src/models/lt_bus.h b/src/models/lt_bus.h
index 9ae66b0..bab1866 100644
--- a/src/models/lt_bus.h
+++ b/src/models/lt_bus.h
@@ -44,6 +44,48 @@ struct range {
   std::uint64_t end;
 };
 
+template <typename Module>
+class tlm_initiator_socket_tagged : public tlm::tlm_initiator_socket<>,
+                                    public tlm::tlm_bw_transport_if<> {
+  using cb_invalidate_direct_mem_ptr = void (Module::*)(std::size_t,
+                                                        sc_dt::uint64,
+                                                        sc_dt::uint64);
+
+ public:
+  explicit tlm_initiator_socket_tagged(const char* name,
+                                       std::size_t id,
+                                       Module* mod,
+                                       cb_invalidate_direct_mem_ptr i)
+      : tlm_initiator_socket<>{name},
+        m_mod{mod},
+        m_id{id},
+        m_invalidate_direct_mem_ptr{i} {
+    bind(*static_cast<tlm::tlm_bw_transport_if<>*>(this));
+  }
+
+ private:
+  // -- TLM_BW_TRANSPORT_IF ----------------------------------------------------
+
+  virtual void invalidate_direct_mem_ptr(sc_dt::uint64 start,
+                                         sc_dt::uint64 end) override {
+    (m_mod->*m_invalidate_direct_mem_ptr)(m_id, start, end);
+  }
+
+  virtual tlm::tlm_sync_enum nb_transport_bw(tlm::tlm_generic_payload&,
+                                             tlm::tlm_phase&,
+                                             sc_core::sc_time&) override {
+    std::fprintf(
+        stderr, "tlm_initiator_socket_tagged: nb_transport_bw not supported\n");
+    std::abort();
+  }
+
+  // -- MEMBER -----------------------------------------------------------------
+
+  std::size_t m_id{0};
+  Module* m_mod{nullptr};
+  cb_invalidate_direct_mem_ptr m_invalidate_direct_mem_ptr{nullptr};
+};
+
 template <typename Module>
 class tlm_target_socket_tagged : public tlm::tlm_target_socket<>,
                                  public tlm::tlm_fw_transport_if<> {
@@ -107,11 +149,11 @@ class tlm_target_socket_tagged : public tlm::tlm_target_socket<>,
   cb_transport_dbg m_transport_dbg{nullptr};
 };
 
-class lt_bus : public sc_core::sc_module, public tlm::tlm_bw_transport_if<> {
+class lt_bus : public sc_core::sc_module {
   using target_socket = tlm_target_socket_tagged<lt_bus>;
   using target_socket_ptr = std::unique_ptr<target_socket>;
 
-  using initiator_socket = tlm::tlm_initiator_socket<>;
+  using initiator_socket = tlm_initiator_socket_tagged<lt_bus>;
   using initiator_socket_ptr = std::unique_ptr<initiator_socket>;
 
  public:
@@ -125,7 +167,7 @@ class lt_bus : public sc_core::sc_module, public tlm::tlm_bw_transport_if<> {
     const std::string name = "init" + std::to_string(id);
     {
       // Add current module on top of module stack for tlm sockets.
-      scoped_push_hierarchy g(*this);
+      scoped_push_hierarchy h(*this);
 
       // Add new target socket to connect BUS INITIATOR.
       m_initiators.push_back(std::make_unique<target_socket>(
@@ -161,15 +203,15 @@ class lt_bus : public sc_core::sc_module, public tlm::tlm_bw_transport_if<> {
     const std::string name = "target" + std::to_string(id);
     {
       // Add current module on top of module stack for tlm sockets.
-      scoped_push_hierarchy g(*this);
+      scoped_push_hierarchy h(*this);
 
       // Add new initiator socket to connect BUS TARGET.
-      m_targets.push_back(std::make_unique<initiator_socket>(name.c_str()));
+      m_targets.push_back(std::make_unique<initiator_socket>(
+          name.c_str(), id, this, &lt_bus::invalidate_direct_mem_ptr));
     }
 
     // Bind sockets.
     auto& init = m_targets.back();
-    init->bind(*this);
     init->bind(target);
 
     // Insert new mapping, id is equal to idx into socket vector.
@@ -179,16 +221,19 @@ class lt_bus : public sc_core::sc_module, public tlm::tlm_bw_transport_if<> {
  private:
   // -- TLM_BW_TRANSPORT_IF ----------------------------------------------------
 
-  virtual tlm::tlm_sync_enum nb_transport_bw(tlm::tlm_generic_payload&,
-                                             tlm::tlm_phase&,
-                                             sc_core::sc_time&) override {
-    std::fprintf(stderr, "lt_bus: nb_transport_bw not supported\n");
-    std::abort();
-  }
-
-  virtual void invalidate_direct_mem_ptr(sc_dt::uint64 start,
-                                         sc_dt::uint64 end) override {
-    assert(false);
+  void invalidate_direct_mem_ptr(std::size_t id,
+                                 sc_dt::uint64 start,
+                                 sc_dt::uint64 end) {
+    for (const auto& map : m_mappings) {
+      if (map.idx != id) {
+        continue;
+      }
+      const sc_dt::uint64 s = map.addr.start + start;
+      const sc_dt::uint64 e = map.addr.start + end;
+      for (auto& sock : m_initiators) {
+        (*sock)->invalidate_direct_mem_ptr(s, e);
+      }
+    }
   }
 
   // -- TLM_FW_TRANSPORT_IF ----------------------------------------------------
-- 
cgit v1.2.3