diff options
| author | Garrett D'Amore <garrett@damore.org> | 2017-01-21 22:05:55 -0800 |
|---|---|---|
| committer | Garrett D'Amore <garrett@damore.org> | 2017-01-21 22:05:55 -0800 |
| commit | 769f9a2b66aca629eb4dd240a072849a48aa300f (patch) | |
| tree | 5fc475839caa1e7f9bf1362f8721ce63f6c58aa5 /src/platform/windows/win_pipe.c | |
| parent | 99a78ade3a6034784e40d5dfa70cc72aa09021ca (diff) | |
| download | nng-769f9a2b66aca629eb4dd240a072849a48aa300f.tar.gz nng-769f9a2b66aca629eb4dd240a072849a48aa300f.tar.bz2 nng-769f9a2b66aca629eb4dd240a072849a48aa300f.zip | |
Whoops, forgot to add the pipe implementations to git!!
Diffstat (limited to 'src/platform/windows/win_pipe.c')
| -rw-r--r-- | src/platform/windows/win_pipe.c | 149 |
1 files changed, 149 insertions, 0 deletions
diff --git a/src/platform/windows/win_pipe.c b/src/platform/windows/win_pipe.c new file mode 100644 index 00000000..f254d9bb --- /dev/null +++ b/src/platform/windows/win_pipe.c @@ -0,0 +1,149 @@ +// +// Copyright 2017 Garrett D'Amore <garrett@damore.org> +// +// This software is supplied under the terms of the MIT License, a +// copy of which should be located in the distribution where this +// file was obtained (LICENSE.txt). A copy of the license may also be +// found online at https://opensource.org/licenses/MIT. +// + +#include "core/nng_impl.h" + +// Windows named pipes won't work for us; we *MUST* use sockets. This is +// a real sadness, but what can you do. We use an anonymous socket bound +// to localhost and a connected peer. + +#ifdef PLATFORM_WINDOWS + +int +nni_plat_pipe_open(int *wfdp, int *rfdp) +{ + SOCKET afd = INVALID_SOCKET; + SOCKET rfd = INVALID_SOCKET; + SOCKET wfd = INVALID_SOCKET; + + struct sockaddr_in addr; + socklen_t alen; + int one; + ULONG yes; + int rv; + + ZeroMemory(&addr, sizeof (addr)); + + // Restrict our bind to the loopback address. We bind to an + // ephemeral port. + addr.sin_family = AF_INET; + addr.sin_port = 0; + addr.sin_addr.s_addr = INADDR_LOOPBACK; + + afd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); + if (afd == INVALID_SOCKET) { + goto fail; + } + + // Make sure we have exclusive address use... + one = 1; + if (setsockopt(afd, SOL_SOCKET, SO_EXCLUSIVEADDRUSE, + (char *) (&one), sizeof (one)) != 0) { + goto fail; + } + alen = sizeof (addr); + if (bind(afd, (struct sockaddr *) &addr, alen) != 0) { + goto fail; + } + // What port did we bind to? + if (getsockname(afd, (struct sockaddr *) &addr, &alen) != 0) { + goto fail; + } + + // Minimum backlog -- we only expect one connection ever. + if (listen(afd, 1) != 0) { + goto fail; + } + + rfd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); + if (afd == INVALID_SOCKET) { + goto fail; + } + + if (connect(rfd, (struct sockaddr *) &addr, alen) != 0) { + goto fail; + } + + // Now we have to do the accept dance. We don't care about the + // peer adddress, since know it. + wfd = accept(afd, NULL, 0); + if (wfd == INVALID_SOCKET) { + goto fail; + } + + // Now that we are connected, mark everything non-blocking. + yes = 1; + if (ioctlsocket(rfd, FIONBIO, &yes) != 0) { + goto fail; + } + yes = 1; + if (ioctlsocket(wfd, FIONBIO, &yes) != 0) { + goto fail; + } + + // Close the listener now that we have the connection. + closesocket((SOCKET) afd); + *rfdp = (int) rfd; + *wfdp = (int) wfd; + return (0); + +fail: + rv = nni_winsock_error(WSAGetLastError()); + if (afd != INVALID_SOCKET) { + closesocket(afd); + } + if (rfd != INVALID_SOCKET) { + closesocket(rfd); + } + if (wfd != INVALID_SOCKET) { + closesocket(wfd); + } + + return (0); +} + + +void +nni_plat_pipe_raise(int wfd) +{ + char c = 1; + + send((SOCKET) wfd, &c, 1, 0); +} + + +void +nni_plat_pipe_clear(int rfd) +{ + char buf[32]; + + for (;;) { + // Completely drain the pipe, but don't wait. This coalesces + // events somewhat. + if (recv((SOCKET) rfd, buf, sizeof (buf), 0) <= 0) { + return; + } + } +} + + +void +nni_plat_pipe_close(int wfd, int rfd) +{ + closesocket((SOCKET) wfd); + closesocket((SOCKET) rfd); +} + + +#else + +// Suppress empty symbols warnings in ranlib. +int nni_win_pipe_not_used = 0; + +#endif // PLATFORM_WINDOWS |
