From 39e61682e71872530afde44e4e304a0f49ab932a Mon Sep 17 00:00:00 2001
From: Evan Balster <evan@interactopia.com>
Date: Tue, 6 Jul 2021 15:39:34 -0700
Subject: Add & document msg_capacity, msg_reserve (#1458)

* Add & document msg_capacity, msg_reserve

* reserve/capacity code style

* Documentation references to reserve/capacity
---
 src/core/message.c | 12 ++++++++++++
 src/core/message.h |  2 ++
 src/nng.c          | 12 ++++++++++++
 3 files changed, 26 insertions(+)

(limited to 'src')

diff --git a/src/core/message.c b/src/core/message.c
index 824fc079..52065bdd 100644
--- a/src/core/message.c
+++ b/src/core/message.c
@@ -456,6 +456,18 @@ nni_msg_realloc(nni_msg *m, size_t sz)
 	return (0);
 }
 
+int
+nni_msg_reserve(nni_msg *m, size_t capacity)
+{
+	return (nni_chunk_grow(&m->m_body, capacity, 0));
+}
+
+size_t
+nni_msg_capacity(nni_msg *m)
+{
+	return ((size_t) ((m->m_body.ch_buf + m->m_body.ch_cap) - m->m_body.ch_ptr));
+}
+
 void *
 nni_msg_header(nni_msg *m)
 {
diff --git a/src/core/message.h b/src/core/message.h
index 6400fd55..7e35ba75 100644
--- a/src/core/message.h
+++ b/src/core/message.h
@@ -17,6 +17,8 @@
 extern int      nni_msg_alloc(nni_msg **, size_t);
 extern void     nni_msg_free(nni_msg *);
 extern int      nni_msg_realloc(nni_msg *, size_t);
+extern int      nni_msg_reserve(nni_msg *, size_t);
+extern size_t   nni_msg_capacity(nni_msg *);
 extern int      nni_msg_dup(nni_msg **, const nni_msg *);
 extern void *   nni_msg_header(nni_msg *);
 extern size_t   nni_msg_header_len(const nni_msg *);
diff --git a/src/nng.c b/src/nng.c
index 3a3c3c5c..1ccc1386 100644
--- a/src/nng.c
+++ b/src/nng.c
@@ -1325,6 +1325,18 @@ nng_msg_free(nng_msg *msg)
 	nni_msg_free(msg);
 }
 
+int
+nng_msg_reserve(nng_msg *msg, size_t capacity)
+{
+	return (nni_msg_reserve(msg, capacity));
+}
+
+size_t
+nng_msg_capacity(nng_msg *msg)
+{
+	return (nni_msg_capacity(msg));
+}
+
 void *
 nng_msg_body(nng_msg *msg)
 {
-- 
cgit v1.2.3-70-g09d2