fixes #691 Desire 16 and 64 bit message manipulators

We use macros to generate message bodies for each of the various
variants, reducing source code size (but not compiled size).  The
documentation is updated to indicate each of these variants.

4 files changed, 183 insertions, 114 deletions

diff --git a/src/core/message.c b/src/core/message.c
index f17240a4..f3151230 100644
--- a/src/core/message.c
+++ b/src/core/message.c
@@ -286,6 +286,42 @@ nni_chunk_insert(nni_chunk *ch, const void *data, size_t len)
 }
 
 static int
+nni_chunk_insert_u16(nni_chunk *ch, uint16_t val)
+{
+	unsigned char buf[sizeof(uint16_t)];
+	NNI_PUT16(buf, val);
+	return (nni_chunk_insert(ch, buf, sizeof(buf)));
+}
+
+static int
+nni_chunk_append_u16(nni_chunk *ch, uint16_t val)
+{
+	unsigned char buf[sizeof(uint16_t)];
+	NNI_PUT16(buf, val);
+	return (nni_chunk_append(ch, buf, sizeof(buf)));
+}
+
+static uint16_t
+nni_chunk_trim_u16(nni_chunk *ch)
+{
+	uint16_t v;
+	NNI_ASSERT(ch->ch_len >= sizeof(v));
+	NNI_GET16(ch->ch_ptr, v);
+	nni_chunk_trim(ch, sizeof(v));
+	return (v);
+}
+
+static uint16_t
+nni_chunk_chop_u16(nni_chunk *ch)
+{
+	uint16_t v;
+	NNI_ASSERT(ch->ch_len >= sizeof(v));
+	NNI_GET16(ch->ch_ptr + ch->ch_len - sizeof(v), v);
+	nni_chunk_chop(ch, sizeof(v));
+	return (v);
+}
+
+static int
 nni_chunk_insert_u32(nni_chunk *ch, uint32_t val)
 {
 	unsigned char buf[sizeof(uint32_t)];
@@ -321,6 +357,42 @@ nni_chunk_chop_u32(nni_chunk *ch)
 	return (v);
 }
 
+static int
+nni_chunk_insert_u64(nni_chunk *ch, uint64_t val)
+{
+	unsigned char buf[sizeof(uint64_t)];
+	NNI_PUT64(buf, val);
+	return (nni_chunk_insert(ch, buf, sizeof(buf)));
+}
+
+static int
+nni_chunk_append_u64(nni_chunk *ch, uint64_t val)
+{
+	unsigned char buf[sizeof(uint64_t)];
+	NNI_PUT64(buf, val);
+	return (nni_chunk_append(ch, buf, sizeof(buf)));
+}
+
+static uint64_t
+nni_chunk_trim_u64(nni_chunk *ch)
+{
+	uint64_t v;
+	NNI_ASSERT(ch->ch_len >= sizeof(v));
+	NNI_GET64(ch->ch_ptr, v);
+	nni_chunk_trim(ch, sizeof(v));
+	return (v);
+}
+
+static uint64_t
+nni_chunk_chop_u64(nni_chunk *ch)
+{
+	uint64_t v;
+	NNI_ASSERT(ch->ch_len >= sizeof(v));
+	NNI_GET64(ch->ch_ptr + ch->ch_len - sizeof(v), v);
+	nni_chunk_chop(ch, sizeof(v));
+	return (v);
+}
+
 int
 nni_msg_alloc(nni_msg **mp, size_t sz)
 {
@@ -558,52 +630,40 @@ nni_msg_header_chop(nni_msg *m, size_t len)
 	return (nni_chunk_chop(&m->m_header, len));
 }
 
-int
-nni_msg_append_u32(nni_msg *m, uint32_t val)
-{
-	return (nni_chunk_append_u32(&m->m_body, val));
-}
-
-int
-nni_msg_insert_u32(nni_msg *m, uint32_t val)
-{
-	return (nni_chunk_insert_u32(&m->m_body, val));
-}
-
-int
-nni_msg_header_append_u32(nni_msg *m, uint32_t val)
-{
-	return (nni_chunk_append_u32(&m->m_header, val));
-}
-
-int
-nni_msg_header_insert_u32(nni_msg *m, uint32_t val)
-{
-	return (nni_chunk_insert_u32(&m->m_header, val));
-}
-
-uint32_t
-nni_msg_chop_u32(nni_msg *m)
-{
-	return (nni_chunk_chop_u32(&m->m_body));
-}
-
-uint32_t
-nni_msg_trim_u32(nni_msg *m)
-{
-	return (nni_chunk_trim_u32(&m->m_body));
-}
-
-uint32_t
-nni_msg_header_chop_u32(nni_msg *m)
-{
-	return (nni_chunk_chop_u32(&m->m_header));
-}
-uint32_t
-nni_msg_header_trim_u32(nni_msg *m)
-{
-	return (nni_chunk_trim_u32(&m->m_header));
-}
+#define DEF_MSG_ADD_N(z, x)                                      \
+	int nni_msg_##z##_u##x(nni_msg *m, uint##x##_t v)        \
+	{                                                        \
+		return (nni_chunk_##z##_u##x(&m->m_body, v));    \
+	}                                                        \
+	int nni_msg_header_##z##_u##x(nni_msg *m, uint##x##_t v) \
+	{                                                        \
+		return (nni_chunk_##z##_u##x(&m->m_header, v));  \
+	}
+
+#define DEF_MSG_REM_N(z, x)                                  \
+	uint##x##_t nni_msg_##z##_u##x(nni_msg *m)           \
+	{                                                    \
+		return (nni_chunk_##z##_u##x(&m->m_body));   \
+	}                                                    \
+	uint##x##_t nni_msg_header_##z##_u##x(nni_msg *m)    \
+	{                                                    \
+		return (nni_chunk_##z##_u##x(&m->m_header)); \
+	}
+
+#define DEF_MSG_ADD(op) \
+	DEF_MSG_ADD_N(op, 16) DEF_MSG_ADD_N(op, 32) DEF_MSG_ADD_N(op, 64)
+#define DEF_MSG_REM(op) \
+	DEF_MSG_REM_N(op, 16) DEF_MSG_REM_N(op, 32) DEF_MSG_REM_N(op, 64)
+
+DEF_MSG_ADD(append)
+DEF_MSG_ADD(insert)
+DEF_MSG_REM(chop)
+DEF_MSG_REM(trim)
+
+#undef DEF_MSG_ADD_N
+#undef DEF_MSG_REM_N
+#undef DEF_MSG_ADD
+#undef DEF_MSG_REM
 
 void
 nni_msg_clear(nni_msg *m)
diff --git a/src/core/message.h b/src/core/message.h
index 00d31436..5ec1d9a1 100644
--- a/src/core/message.h
+++ b/src/core/message.h
@@ -35,14 +35,30 @@ extern int      nni_msg_header_chop(nni_msg *, size_t);
 extern int      nni_msg_setopt(nni_msg *, int, const void *, size_t);
 extern int      nni_msg_getopt(nni_msg *, int, void *, size_t *);
 extern void     nni_msg_dump(const char *, const nni_msg *);
+extern int      nni_msg_append_u16(nni_msg *, uint16_t);
 extern int      nni_msg_append_u32(nni_msg *, uint32_t);
+extern int      nni_msg_append_u64(nni_msg *, uint64_t);
+extern int      nni_msg_insert_u16(nni_msg *, uint16_t);
 extern int      nni_msg_insert_u32(nni_msg *, uint32_t);
+extern int      nni_msg_insert_u64(nni_msg *, uint64_t);
+extern int      nni_msg_header_append_u16(nni_msg *, uint16_t);
 extern int      nni_msg_header_append_u32(nni_msg *, uint32_t);
+extern int      nni_msg_header_append_u64(nni_msg *, uint64_t);
+extern int      nni_msg_header_insert_u16(nni_msg *, uint16_t);
 extern int      nni_msg_header_insert_u32(nni_msg *, uint32_t);
+extern int      nni_msg_header_insert_u64(nni_msg *, uint64_t);
+extern uint16_t nni_msg_trim_u16(nni_msg *);
 extern uint32_t nni_msg_trim_u32(nni_msg *);
+extern uint64_t nni_msg_trim_u64(nni_msg *);
+extern uint16_t nni_msg_chop_u16(nni_msg *);
 extern uint32_t nni_msg_chop_u32(nni_msg *);
+extern uint64_t nni_msg_chop_u64(nni_msg *);
+extern uint16_t nni_msg_header_trim_u16(nni_msg *);
 extern uint32_t nni_msg_header_trim_u32(nni_msg *);
+extern uint64_t nni_msg_header_trim_u64(nni_msg *);
+extern uint16_t nni_msg_header_chop_u16(nni_msg *);
 extern uint32_t nni_msg_header_chop_u32(nni_msg *);
+extern uint64_t nni_msg_header_chop_u64(nni_msg *);
 extern void     nni_msg_set_pipe(nni_msg *, uint32_t);
 extern uint32_t nni_msg_get_pipe(const nni_msg *);
 
diff --git a/src/nng.c b/src/nng.c
index 37ee3e5f..64968cae 100644
--- a/src/nng.c
+++ b/src/nng.c
@@ -1329,69 +1329,47 @@ nng_msg_header_append(nng_msg *msg, const void *data, size_t sz)
 	return (nni_msg_header_append(msg, data, sz));
 }
 
-int
-nng_msg_header_append_u32(nng_msg *msg, uint32_t val)
-{
-	return (nni_msg_header_append_u32(msg, val));
-}
-
-int
-nng_msg_header_insert_u32(nng_msg *msg, uint32_t val)
-{
-	return (nni_msg_header_insert_u32(msg, val));
-}
-
-int
-nng_msg_header_chop_u32(nng_msg *msg, uint32_t *valp)
-{
-	if (nni_msg_header_len(msg) < sizeof(uint32_t)) {
-		return (NNG_EINVAL);
-	}
-	*valp = nni_msg_header_chop_u32(msg);
-	return (0);
-}
-
-int
-nng_msg_header_trim_u32(nng_msg *msg, uint32_t *valp)
-{
-	if (nni_msg_header_len(msg) < sizeof(uint32_t)) {
-		return (NNG_EINVAL);
-	}
-	*valp = nni_msg_header_trim_u32(msg);
-	return (0);
-}
-
-int
-nng_msg_append_u32(nng_msg *msg, uint32_t val)
-{
-	return (nni_msg_append_u32(msg, val));
-}
-
-int
-nng_msg_insert_u32(nng_msg *msg, uint32_t val)
-{
-	return (nni_msg_insert_u32(msg, val));
-}
-
-int
-nng_msg_chop_u32(nng_msg *msg, uint32_t *valp)
-{
-	if (nni_msg_len(msg) < sizeof(uint32_t)) {
-		return (NNG_EINVAL);
-	}
-	*valp = nni_msg_chop_u32(msg);
-	return (0);
-}
-
-int
-nng_msg_trim_u32(nng_msg *msg, uint32_t *valp)
-{
-	if (nni_msg_len(msg) < sizeof(uint32_t)) {
-		return (NNG_EINVAL);
-	}
-	*valp = nni_msg_trim_u32(msg);
-	return (0);
-}
+#define DEF_MSG_ADD_N(op, n)                                      \
+	int nng_msg_header_##op##_u##n(nng_msg *m, uint##n##_t v) \
+	{                                                         \
+		return (nni_msg_header_##op##_u##n(m, v));        \
+	}                                                         \
+	int nng_msg_##op##_u##n(nng_msg *m, uint##n##_t v)        \
+	{                                                         \
+		return (nni_msg_##op##_u##n(m, v));               \
+	}
+#define DEF_MSG_REM_N(op, n)                                        \
+	int nng_msg_header_##op##_u##n(nng_msg *m, uint##n##_t *vp) \
+	{                                                           \
+		if (nni_msg_header_len(m) < sizeof(*vp)) {          \
+			return (NNG_EINVAL);                        \
+		}                                                   \
+		*vp = nni_msg_header_##op##_u##n(m);                \
+		return (0);                                         \
+	}                                                           \
+	int nng_msg_##op##_u##n(nng_msg *m, uint##n##_t *vp)        \
+	{                                                           \
+		if (nni_msg_len(m) < sizeof(*vp)) {                 \
+			return (NNG_EINVAL);                        \
+		}                                                   \
+		*vp = nni_msg_##op##_u##n(m);                       \
+		return (0);                                         \
+	}
+
+#define DEF_MSG_ADD(op) \
+	DEF_MSG_ADD_N(op, 16) DEF_MSG_ADD_N(op, 32) DEF_MSG_ADD_N(op, 64)
+#define DEF_MSG_REM(op) \
+	DEF_MSG_REM_N(op, 16) DEF_MSG_REM_N(op, 32) DEF_MSG_REM_N(op, 64)
+
+DEF_MSG_ADD(append)
+DEF_MSG_ADD(insert)
+DEF_MSG_REM(chop)
+DEF_MSG_REM(trim)
+
+#undef DEF_MSG_ADD_N
+#undef DEF_MSG_REM_N
+#undef DEF_MSG_ADD
+#undef DEF_MSG_REM
 
 int
 nng_msg_header_insert(nng_msg *msg, const void *data, size_t sz)
diff --git a/src/nng.h b/src/nng.h
index 6bc38c90..863ce363 100644
--- a/src/nng.h
+++ b/src/nng.h
@@ -610,19 +610,34 @@ NNG_DECL int    nng_msg_header_append(nng_msg *, const void *, size_t);
 NNG_DECL int    nng_msg_header_insert(nng_msg *, const void *, size_t);
 NNG_DECL int    nng_msg_header_trim(nng_msg *, size_t);
 NNG_DECL int    nng_msg_header_chop(nng_msg *, size_t);
+NNG_DECL int    nng_msg_header_append_u16(nng_msg *, uint16_t);
 NNG_DECL int    nng_msg_header_append_u32(nng_msg *, uint32_t);
+NNG_DECL int    nng_msg_header_append_u64(nng_msg *, uint64_t);
+NNG_DECL int    nng_msg_header_insert_u16(nng_msg *, uint16_t);
 NNG_DECL int    nng_msg_header_insert_u32(nng_msg *, uint32_t);
+NNG_DECL int    nng_msg_header_insert_u64(nng_msg *, uint64_t);
+NNG_DECL int    nng_msg_header_chop_u16(nng_msg *, uint16_t *);
 NNG_DECL int    nng_msg_header_chop_u32(nng_msg *, uint32_t *);
+NNG_DECL int    nng_msg_header_chop_u64(nng_msg *, uint64_t *);
+NNG_DECL int    nng_msg_header_trim_u16(nng_msg *, uint16_t *);
 NNG_DECL int    nng_msg_header_trim_u32(nng_msg *, uint32_t *);
+NNG_DECL int    nng_msg_header_trim_u64(nng_msg *, uint64_t *);
+NNG_DECL int    nng_msg_append_u16(nng_msg *, uint16_t);
 NNG_DECL int    nng_msg_append_u32(nng_msg *, uint32_t);
+NNG_DECL int    nng_msg_append_u64(nng_msg *, uint64_t);
+NNG_DECL int    nng_msg_insert_u16(nng_msg *, uint16_t);
 NNG_DECL int    nng_msg_insert_u32(nng_msg *, uint32_t);
+NNG_DECL int    nng_msg_insert_u64(nng_msg *, uint64_t);
+NNG_DECL int    nng_msg_chop_u16(nng_msg *, uint16_t *);
 NNG_DECL int    nng_msg_chop_u32(nng_msg *, uint32_t *);
+NNG_DECL int    nng_msg_chop_u64(nng_msg *, uint64_t *);
+NNG_DECL int    nng_msg_trim_u16(nng_msg *, uint16_t *);
 NNG_DECL int    nng_msg_trim_u32(nng_msg *, uint32_t *);
-
-NNG_DECL int  nng_msg_dup(nng_msg **, const nng_msg *);
-NNG_DECL void nng_msg_clear(nng_msg *);
-NNG_DECL void nng_msg_header_clear(nng_msg *);
-NNG_DECL void nng_msg_set_pipe(nng_msg *, nng_pipe);
+NNG_DECL int    nng_msg_trim_u64(nng_msg *, uint64_t *);
+NNG_DECL int    nng_msg_dup(nng_msg **, const nng_msg *);
+NNG_DECL void   nng_msg_clear(nng_msg *);
+NNG_DECL void   nng_msg_header_clear(nng_msg *);
+NNG_DECL void   nng_msg_set_pipe(nng_msg *, nng_pipe);
 NNG_DECL nng_pipe nng_msg_get_pipe(const nng_msg *);
 NNG_DECL int      nng_msg_getopt(nng_msg *, int, void *, size_t *);