Merge pull request #458 from fengzeroz/main

protocol buf
emqx · May 5, 2022 · a09abf2 · a09abf2
2 parents adf6128 + 4c96c11
commit a09abf2
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/include/neuron/utils/protocol_buf.h b/include/neuron/utils/protocol_buf.h
@@ -100,6 +100,16 @@ inline static uint8_t *neu_protocol_unpack_buf(neu_protocol_unpack_buf_t *buf,
     return buf->base + buf->offset - size;
 }
 
+inline static uint8_t *
+neu_protocol_unpack_buf_get(neu_protocol_unpack_buf_t *buf, uint16_t size)
+{
+    if (buf->size - buf->offset < size) {
+        return NULL;
+    }
+
+    return buf->base + buf->offset;
+}
+
 inline static uint8_t *neu_protocol_pack_buf(neu_protocol_pack_buf_t *buf,
                                              uint16_t                 size)
 {
@@ -118,6 +128,12 @@ neu_protocol_pack_buf_unused_size(neu_protocol_pack_buf_t *buf)
     return buf->offset;
 }
 
+inline static uint16_t
+neu_protocol_pack_buf_used_size(neu_protocol_pack_buf_t *buf)
+{
+    return buf->size - buf->offset;
+}
+
 inline static uint16_t
 neu_protocol_unpack_buf_unused_size(neu_protocol_unpack_buf_t *buf)
 {

diff --git a/package/create_neuron_sdk_deb.py b/package/create_neuron_sdk_deb.py
@@ -58,6 +58,8 @@
              "/usr/local/include/neuron/utils/"))
 rules.append(mkdeb.FileMap("../include/neuron/utils/zlog.h",
              "/usr/local/include/neuron/utils/"))
+rules.append(mkdeb.FileMap("../include/neuron/utils/protocol_buf.h",
+             "/usr/local/include/neuron/utils/"))
 rules.append(mkdeb.FileMap("../include/neuron/adapter.h",
              "/usr/local/include/neuron/"))
 rules.append(mkdeb.FileMap("../include/neuron/adapter_msg.h",