aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/core/platform.h6
-rw-r--r--src/core/strs.c4
-rw-r--r--src/platform/posix/posix_aio.h2
-rw-r--r--src/platform/posix/posix_pipedesc.c24
-rw-r--r--src/platform/posix/posix_tcp.c12
-rw-r--r--src/platform/windows/win_debug.c1
-rw-r--r--src/platform/windows/win_tcp.c73
-rw-r--r--src/transport/tcp/tcp.c32
8 files changed, 144 insertions, 10 deletions
diff --git a/src/core/platform.h b/src/core/platform.h
index 7bfb370a..10074677 100644
--- a/src/core/platform.h
+++ b/src/core/platform.h
@@ -233,6 +233,12 @@ extern void nni_plat_tcp_pipe_send(nni_plat_tcp_pipe *, nni_aio *);
// The platform may modify the iovs.
extern void nni_plat_tcp_pipe_recv(nni_plat_tcp_pipe *, nni_aio *);
+// nni_plat_tcp_pipe_peername gets the peer name.
+extern int nni_plat_tcp_pipe_peername(nni_plat_tcp_pipe *, nni_sockaddr *);
+
+// nni_plat_tcp_pipe_sockname gets the local name.
+extern int nni_plat_tcp_pipe_sockname(nni_plat_tcp_pipe *, nni_sockaddr *);
+
// nni_plat_tcp_resolv resolves a TCP name asynchronously. The family
// should be one of NNG_AF_INET, NNG_AF_INET6, or NNG_AF_UNSPEC. The
// first two constrain the name to those families, while the third will
diff --git a/src/core/strs.c b/src/core/strs.c
index 08338499..6cee605b 100644
--- a/src/core/strs.c
+++ b/src/core/strs.c
@@ -21,7 +21,11 @@ char *
nni_strdup(const char *src)
{
#ifdef NNG_HAVE_STRDUP
+#ifdef _WIN32
+ return (_strdup(src));
+#else
return (strdup(src));
+#endif
#else
char * dst;
size_t len = strlen(src);
diff --git a/src/platform/posix/posix_aio.h b/src/platform/posix/posix_aio.h
index 6c7db9e1..fe677591 100644
--- a/src/platform/posix/posix_aio.h
+++ b/src/platform/posix/posix_aio.h
@@ -28,6 +28,8 @@ extern void nni_posix_pipedesc_fini(nni_posix_pipedesc *);
extern void nni_posix_pipedesc_recv(nni_posix_pipedesc *, nni_aio *);
extern void nni_posix_pipedesc_send(nni_posix_pipedesc *, nni_aio *);
extern void nni_posix_pipedesc_close(nni_posix_pipedesc *);
+extern int nni_posix_pipedesc_peername(nni_posix_pipedesc *, nni_sockaddr *);
+extern int nni_posix_pipedesc_sockname(nni_posix_pipedesc *, nni_sockaddr *);
extern int nni_posix_epdesc_init(nni_posix_epdesc **);
extern void nni_posix_epdesc_set_local(nni_posix_epdesc *, void *, int);
diff --git a/src/platform/posix/posix_pipedesc.c b/src/platform/posix/posix_pipedesc.c
index 8a087256..a025f83c 100644
--- a/src/platform/posix/posix_pipedesc.c
+++ b/src/platform/posix/posix_pipedesc.c
@@ -288,6 +288,30 @@ nni_posix_pipedesc_send(nni_posix_pipedesc *pd, nni_aio *aio)
}
int
+nni_posix_pipedesc_peername(nni_posix_pipedesc *pd, nni_sockaddr *sa)
+{
+ struct sockaddr_storage ss;
+ socklen_t sslen = sizeof(ss);
+
+ if (getpeername(pd->node.fd, (void *) &ss, &sslen) != 0) {
+ return (nni_plat_errno(errno));
+ }
+ return (nni_posix_sockaddr2nn(sa, &ss));
+}
+
+int
+nni_posix_pipedesc_sockname(nni_posix_pipedesc *pd, nni_sockaddr *sa)
+{
+ struct sockaddr_storage ss;
+ socklen_t sslen = sizeof(ss);
+
+ if (getsockname(pd->node.fd, (void *) &ss, &sslen) != 0) {
+ return (nni_plat_errno(errno));
+ }
+ return (nni_posix_sockaddr2nn(sa, &ss));
+}
+
+int
nni_posix_pipedesc_init(nni_posix_pipedesc **pdp, int fd)
{
nni_posix_pipedesc *pd;
diff --git a/src/platform/posix/posix_tcp.c b/src/platform/posix/posix_tcp.c
index cc5466cb..e03f8056 100644
--- a/src/platform/posix/posix_tcp.c
+++ b/src/platform/posix/posix_tcp.c
@@ -105,4 +105,16 @@ nni_plat_tcp_pipe_recv(nni_plat_tcp_pipe *p, nni_aio *aio)
nni_posix_pipedesc_recv((void *) p, aio);
}
+int
+nni_plat_tcp_pipe_peername(nni_plat_tcp_pipe *p, nni_sockaddr *sa)
+{
+ return (nni_posix_pipedesc_peername((void *) p, sa));
+}
+
+int
+nni_plat_tcp_pipe_sockname(nni_plat_tcp_pipe *p, nni_sockaddr *sa)
+{
+ return (nni_posix_pipedesc_sockname((void *) p, sa));
+}
+
#endif // NNG_PLATFORM_POSIX
diff --git a/src/platform/windows/win_debug.c b/src/platform/windows/win_debug.c
index c6ee3fed..5c6c3fb5 100644
--- a/src/platform/windows/win_debug.c
+++ b/src/platform/windows/win_debug.c
@@ -109,6 +109,7 @@ static struct {
{ WSAEWOULDBLOCK, NNG_EAGAIN },
{ WSAEINPROGRESS, NNG_EAGAIN },
{ WSAENOTSOCK, NNG_ECLOSED },
+ { WSAEINVAL, NNG_EINVAL },
{ WSAEMSGSIZE, NNG_EMSGSIZE },
{ WSAENOPROTOOPT, NNG_ENOTSUP },
{ WSAEPROTONOSUPPORT, NNG_ENOTSUP },
diff --git a/src/platform/windows/win_tcp.c b/src/platform/windows/win_tcp.c
index d34ef7a6..5e27d4ee 100644
--- a/src/platform/windows/win_tcp.c
+++ b/src/platform/windows/win_tcp.c
@@ -15,9 +15,11 @@
#include <stdio.h>
struct nni_plat_tcp_pipe {
- SOCKET s;
- nni_win_event rcv_ev;
- nni_win_event snd_ev;
+ SOCKET s;
+ nni_win_event rcv_ev;
+ nni_win_event snd_ev;
+ SOCKADDR_STORAGE sockname;
+ SOCKADDR_STORAGE peername;
};
struct nni_plat_tcp_ep {
@@ -36,9 +38,11 @@ struct nni_plat_tcp_ep {
char buf[512]; // to hold acceptex results
// We have to lookup some function pointers using ioctls. Winsock,
- // gotta love it.
- LPFN_CONNECTEX connectex;
- LPFN_ACCEPTEX acceptex;
+ // gotta love it. Especially I love that asynch accept means that
+ // getsockname and getpeername don't work.
+ LPFN_CONNECTEX connectex;
+ LPFN_ACCEPTEX acceptex;
+ LPFN_GETACCEPTEXSOCKADDRS getacceptexsockaddrs;
};
static int nni_win_tcp_pipe_start(nni_win_event *, nni_aio *);
@@ -248,6 +252,24 @@ nni_plat_tcp_pipe_close(nni_plat_tcp_pipe *pipe)
}
}
+int
+nni_plat_tcp_pipe_peername(nni_plat_tcp_pipe *pipe, nni_sockaddr *sa)
+{
+ if (nni_win_sockaddr2nn(sa, &pipe->peername) < 0) {
+ return (NNG_EADDRINVAL);
+ }
+ return (0);
+}
+
+int
+nni_plat_tcp_pipe_sockname(nni_plat_tcp_pipe *pipe, nni_sockaddr *sa)
+{
+ if (nni_win_sockaddr2nn(sa, &pipe->sockname) < 0) {
+ return (NNG_EADDRINVAL);
+ }
+ return (0);
+}
+
void
nni_plat_tcp_pipe_fini(nni_plat_tcp_pipe *pipe)
{
@@ -268,6 +290,7 @@ nni_plat_tcp_ep_init(nni_plat_tcp_ep **epp, const nni_sockaddr *lsa,
DWORD nbytes;
GUID guid1 = WSAID_CONNECTEX;
GUID guid2 = WSAID_ACCEPTEX;
+ GUID guid3 = WSAID_GETACCEPTEXSOCKADDRS;
if ((ep = NNI_ALLOC_STRUCT(ep)) == NULL) {
return (NNG_ENOMEM);
@@ -303,6 +326,14 @@ nni_plat_tcp_ep_init(nni_plat_tcp_ep **epp, const nni_sockaddr *lsa,
rv = nni_win_error(GetLastError());
goto fail;
}
+ if (WSAIoctl(s, SIO_GET_EXTENSION_FUNCTION_POINTER, &guid3,
+ sizeof(guid3), &ep->getacceptexsockaddrs,
+ sizeof(ep->getacceptexsockaddrs), &nbytes, NULL,
+ NULL) == SOCKET_ERROR) {
+ rv = nni_win_error(GetLastError());
+ goto fail;
+ }
+
closesocket(s);
s = INVALID_SOCKET;
@@ -432,6 +463,10 @@ nni_win_tcp_acc_finish(nni_win_event *evt, nni_aio *aio)
nni_plat_tcp_pipe *pipe;
SOCKET s;
int rv;
+ int len1;
+ int len2;
+ SOCKADDR * sa1;
+ SOCKADDR * sa2;
s = ep->acc_s;
ep->acc_s = INVALID_SOCKET;
@@ -448,6 +483,19 @@ nni_win_tcp_acc_finish(nni_win_event *evt, nni_aio *aio)
return;
}
+ // Collect the local and peer addresses, because normal getsockname
+ // and getpeername don't work with AcceptEx.
+ len1 = sizeof(pipe->sockname);
+ len2 = sizeof(pipe->peername);
+ ep->getacceptexsockaddrs(
+ ep->buf, 0, 256, 256, &sa1, &len1, &sa2, &len2);
+ NNI_ASSERT(len1 > 0);
+ NNI_ASSERT(len1 < sizeof(SOCKADDR_STORAGE));
+ NNI_ASSERT(len2 > 0);
+ NNI_ASSERT(len2 < sizeof(SOCKADDR_STORAGE));
+ memcpy(&pipe->sockname, sa1, len1);
+ memcpy(&pipe->peername, sa2, len2);
+
nni_aio_finish_pipe(aio, pipe);
}
@@ -513,6 +561,8 @@ nni_win_tcp_con_finish(nni_win_event *evt, nni_aio *aio)
nni_plat_tcp_pipe *pipe;
SOCKET s;
int rv;
+ DWORD yes = 1;
+ int len;
s = ep->s;
ep->s = INVALID_SOCKET;
@@ -528,6 +578,17 @@ nni_win_tcp_con_finish(nni_win_event *evt, nni_aio *aio)
return;
}
+ (void) setsockopt(s, SOL_SOCKET, SO_UPDATE_CONNECT_CONTEXT,
+ (char *) &yes, sizeof(yes));
+
+ // Windows seems to be unable to get peernames for sockets on
+ // connect - perhaps because we supplied it already with connectex.
+ // Rather than debugging it, just steal the address from the endpoint.
+ memcpy(&pipe->peername, &ep->remaddr, ep->remlen);
+
+ len = sizeof(pipe->sockname);
+ (void) getsockname(s, (SOCKADDR *) &pipe->sockname, &len);
+
aio->a_pipe = pipe;
nni_aio_finish(aio, 0, 0);
}
diff --git a/src/transport/tcp/tcp.c b/src/transport/tcp/tcp.c
index 413d8fa5..37bd49c7 100644
--- a/src/transport/tcp/tcp.c
+++ b/src/transport/tcp/tcp.c
@@ -407,6 +407,32 @@ nni_tcp_pipe_peer(void *arg)
}
static int
+nni_tcp_pipe_getopt_locaddr(void *arg, void *v, size_t *szp)
+{
+ nni_tcp_pipe *p = arg;
+ int rv;
+ nng_sockaddr sa;
+
+ if ((rv = nni_plat_tcp_pipe_sockname(p->tpp, &sa)) == 0) {
+ rv = nni_getopt_sockaddr(&sa, v, szp);
+ }
+ return (rv);
+}
+
+static int
+nni_tcp_pipe_getopt_remaddr(void *arg, void *v, size_t *szp)
+{
+ nni_tcp_pipe *p = arg;
+ int rv;
+ nng_sockaddr sa;
+
+ if ((rv = nni_plat_tcp_pipe_peername(p->tpp, &sa)) == 0) {
+ rv = nni_getopt_sockaddr(&sa, v, szp);
+ }
+ return (rv);
+}
+
+static int
nni_tcp_parse_pair(char *pair, char **hostp, char **servp)
{
char *host, *serv, *end;
@@ -785,10 +811,8 @@ nni_tcp_ep_getopt_linger(void *arg, void *v, size_t *szp)
}
static nni_tran_pipe_option nni_tcp_pipe_options[] = {
-#if 0
- { NNG_OPT_LOCADDR, nni_tcp_pipe_get_locaddr },
- { NNG_OPT_REMADDR, nni_tcp_pipe_get_remaddr },
-#endif
+ { NNG_OPT_LOCADDR, nni_tcp_pipe_getopt_locaddr },
+ { NNG_OPT_REMADDR, nni_tcp_pipe_getopt_remaddr },
// terminate list
{ NULL, NULL }
};
generated by cgit v1.2.3-70-g09d2 (git 2.46.0) at 2026-01-18 17:45:03 +0000