diff --git a/include/neuron/utils/protocol_buf.h b/include/neuron/utils/protocol_buf.h
index d8ff7895..69db0aa8 100644
--- a/include/neuron/utils/protocol_buf.h
+++ b/include/neuron/utils/protocol_buf.h
@@ -192,4 +192,16 @@ neu_protocol_unpack_buf_used_size(neu_protocol_unpack_buf_t *buf)
     return buf->offset;
 }
 
+inline static void
+neu_protocol_unpack_buf_use_all(neu_protocol_unpack_buf_t *buf)
+{
+    buf->offset = buf->size;
+}
+
+inline static uint16_t
+neu_protocol_unpack_buf_size(neu_protocol_unpack_buf_t *buf)
+{
+    return buf->size;
+}
+
 #endif