From 75aee14e2951cfbd4d8d22375a9f4301890da1ee Mon Sep 17 00:00:00 2001
From: Garrett D'Amore <garrett@damore.org>
Date: Wed, 14 Dec 2016 08:22:23 -0800
Subject: Protocols are 16-bit.

---
 src/core/socket.c | 8 +++++++-
 src/nng.h         | 8 +++++++-
 2 files changed, 14 insertions(+), 2 deletions(-)

(limited to 'src')

diff --git a/src/core/socket.c b/src/core/socket.c
index ee436bb2..9c289a94 100644
--- a/src/core/socket.c
+++ b/src/core/socket.c
@@ -59,7 +59,7 @@ nni_socket_recvq(nng_socket_t s)
 }
 
 int
-nng_socket_create(nng_socket_t *sockp, int proto)
+nng_socket_create(nng_socket_t *sockp, uint16_t proto)
 {
 	return (NNG_EAGAIN);	/* XXX: IMPLEMENT ME */
 }
@@ -125,3 +125,9 @@ nng_socket_sendmsg(nng_socket_t sock, nng_msg_t msg, int tmout)
 	}
 	return (rv);
 }
+
+uint16_t
+nng_socket_protocol(nng_socket_t sock)
+{
+        return (sock->s_ops.proto_self);
+}
diff --git a/src/nng.h b/src/nng.h
index 0adadb39..556ffba3 100644
--- a/src/nng.h
+++ b/src/nng.h
@@ -67,7 +67,7 @@ typedef struct nng_stat *nng_stat_t;
  * error code on failure, or zero on success.  The socket starts in cooked
  * mode.
  */
-NNG_DECL int nng_socket_create(nng_socket_t *, int proto);
+NNG_DECL int nng_socket_create(nng_socket_t *, uint16_t proto);
 
 /*
  * nng_socket_close closes the socket, terminating all activity and
@@ -79,6 +79,12 @@ NNG_DECL int nng_socket_create(nng_socket_t *, int proto);
  */
 NNG_DECL int nng_socket_close(nng_socket_t);
 
+/*
+ * nng_socket_protocol returns the protocol number the socket was created
+ * with.
+ */
+uint16_t nng_socket_protocol(nng_socket_t);
+
 /*
  * nng_socket_setopt sets an option for a specific socket.
  */
-- 
cgit v1.2.3-70-g09d2