1 files changed, 35 insertions, 1 deletions
diff --git a/hicn-light/src/hicn/core/msgbuf.c b/hicn-light/src/hicn/core/msgbuf.c
index 299b20f9b..c58f7a7dc 100644
--- a/hicn-light/src/hicn/core/msgbuf.c
+++ b/hicn-light/src/hicn/core/msgbuf.c
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2021 Cisco and/or its affiliates.
+ * Copyright (c) 2021-2022 Cisco and/or its affiliates.
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at:
@@ -19,3 +19,37 @@
  */
 
 #include "msgbuf.h"
+#include "../strategies/probe_generator.h"
+
+int msgbuf_initialize(msgbuf_t *msgbuf) {
+  /*
+   * We define the format and the storage area of the packet buffer we
+   * manipulate
+   */
+  hicn_packet_buffer_t *pkbuf = msgbuf_get_pkbuf(msgbuf);
+  hicn_packet_set_buffer(pkbuf, msgbuf->packet, MTU, 0);
+  hicn_packet_init_header(pkbuf, 0);
+  return 0;
+}
+
+int msgbuf_initialize_from_packet(msgbuf_t *msgbuf) {
+  hicn_packet_set_buffer(msgbuf_get_pkbuf(msgbuf), msgbuf->packet, MTU,
+                         msgbuf_get_len(msgbuf));
+  return 0;
+}
+
+bool msgbuf_is_command(const msgbuf_t *msgbuf) {
+  return (*msgbuf->packet == REQUEST_LIGHT);
+}
+
+bool msgbuf_is_probe(const msgbuf_t *msgbuf) {
+  hicn_name_t name;
+  hicn_name_suffix_t suffix;
+
+  assert(msgbuf_get_type(msgbuf) == HICN_PACKET_TYPE_DATA);
+
+  const hicn_packet_buffer_t *pkbuf = msgbuf_get_pkbuf(msgbuf);
+  hicn_data_get_name(pkbuf, &name);
+  suffix = hicn_name_get_suffix(&name);
+  return (suffix >= MIN_PROBE_SUFFIX && suffix <= MAX_PROBE_SUFFIX);
+}