Rebase dev 2.0

1 files changed, 13 insertions, 1 deletions

diff --git a/infra/packet_manager/packet_manager.c b/infra/packet_manager/packet_manager.c
index eba87ec..e492414 100644
--- a/infra/packet_manager/packet_manager.c
+++ b/infra/packet_manager/packet_manager.c
@@ -4,7 +4,8 @@
 #include "packet_internal.h"
 #include "packet_manager_runtime.h"
 #include "packet_manager_internal.h"
-#include "stellar/module_manager.h"
+
+#define PACKET_MANAGER_MODULE_NAME "packet_manager_module"
 
 struct packet_manager_config
 {
@@ -290,6 +291,17 @@ void packet_manager_print_stat(struct packet_manager *pkt_mgr, uint16_t thread_i
  * packet manager module
  ******************************************************************************/
 
+struct packet_manager *stellar_module_get_packet_manager(struct stellar_module_manager *mod_mgr)
+{
+    assert(mod_mgr);
+    struct stellar_module *pkt_mgr_mod = stellar_module_manager_get_module(mod_mgr, PACKET_MANAGER_MODULE_NAME);
+    if (pkt_mgr_mod == NULL)
+    {
+        return NULL;
+    }
+    return (struct packet_manager*)stellar_module_get_ctx(pkt_mgr_mod);
+}
+
 struct stellar_module *packet_manager_on_init(struct stellar_module_manager *mod_mgr)
 {
     assert(mod_mgr);