diff --git a/include/neuron/utils/protocol_buf.h b/include/neuron/utils/protocol_buf.h index d9f8628a..8aa14e45 100644 --- 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 index 3c2fa15c..25c5f0d1 100755 --- a/package/create_neuron_sdk_deb.py +++ b/package/create_neuron_sdk_deb.py @@ -58,6 +58,8 @@ rules.append(mkdeb.FileMap("../include/neuron/utils/utlist.h", "/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",