summaryrefslogtreecommitdiff
path: root/src/fd_signaler.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/fd_signaler.cpp')
-rw-r--r--src/fd_signaler.cpp278
1 files changed, 278 insertions, 0 deletions
diff --git a/src/fd_signaler.cpp b/src/fd_signaler.cpp
new file mode 100644
index 0000000..005dd86
--- /dev/null
+++ b/src/fd_signaler.cpp
@@ -0,0 +1,278 @@
+/*
+ Copyright (c) 2007-2009 FastMQ Inc.
+
+ This file is part of 0MQ.
+
+ 0MQ is free software; you can redistribute it and/or modify it under
+ the terms of the Lesser GNU General Public License as published by
+ the Free Software Foundation; either version 3 of the License, or
+ (at your option) any later version.
+
+ 0MQ is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ Lesser GNU General Public License for more details.
+
+ You should have received a copy of the Lesser GNU General Public License
+ along with this program. If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include "fd_signaler.hpp"
+#include "platform.hpp"
+#include "err.hpp"
+#include "fd.hpp"
+
+#if defined ZS_HAVE_OPENVMS
+#include <netinet/tcp.h>
+#elif defined ZS_HAVE_WINDOWS
+#include "windows.hpp"
+#else
+#include <unistd.h>
+#include <fcntl.h>
+#endif
+
+#if defined ZS_HAVE_EVENTFD
+
+#include <sys/eventfd.h>
+
+zs::fd_signaler_t::fd_signaler_t ()
+{
+ // Create eventfd object.
+ fd = eventfd (0, 0);
+ errno_assert (fd != -1);
+
+ // Set to non-blocking mode.
+ int flags = fcntl (fd, F_GETFL, 0);
+ if (flags == -1)
+ flags = 0;
+ int rc = fcntl (fd, F_SETFL, flags | O_NONBLOCK);
+ errno_assert (rc != -1);
+}
+
+zs::fd_signaler_t::~fd_signaler_t ()
+{
+ int rc = close (fd);
+ errno_assert (rc != -1);
+}
+
+void zs::fd_signaler_t::signal (int signal_)
+{
+ zs_assert (signal_ >= 0 && signal_ < 64);
+ signals_t inc = 1;
+ inc <<= signal_;
+ ssize_t sz = write (fd, &inc, sizeof (signals_t));
+ errno_assert (sz == sizeof (signals_t));
+}
+
+zs::fd_signaler_t::signals_t zs::fd_signaler_t::check ()
+{
+ signals_t val;
+ ssize_t sz = read (fd, &val, sizeof (signals_t));
+ if (sz == -1 && (errno == EAGAIN || errno == EWOULDBLOCK ||
+ errno == EINTR))
+ return 0;
+ errno_assert (sz != -1);
+ return val;
+}
+
+zs::fd_t zs::fd_signaler_t::get_fd ()
+{
+ return fd;
+}
+
+#elif defined ZS_HAVE_WINDOWS
+
+zs::fd_signaler_t::fd_signaler_t ()
+{
+ struct sockaddr_in addr;
+ SOCKET listener;
+ int addrlen = sizeof (addr);
+
+ w = INVALID_SOCKET;
+ r = INVALID_SOCKET;
+
+ fd_t rcs = (listener = socket (AF_INET, SOCK_STREAM, 0));
+ wsa_assert (rcs != INVALID_SOCKET);
+
+ memset (&addr, 0, sizeof (addr));
+ addr.sin_family = AF_INET;
+ resolve_ip_hostname (&addr, "127.0.0.1:0");
+
+ int rc = bind (listener, (const struct sockaddr*) &addr, sizeof (addr));
+ wsa_assert (rc != SOCKET_ERROR);
+
+ rc = getsockname (listener, (struct sockaddr*) &addr, &addrlen);
+ wsa_assert (rc != SOCKET_ERROR);
+
+ // Listen for incomming connections.
+ rc = listen (listener, 1);
+ wsa_assert (rc != SOCKET_ERROR);
+
+ // Create the socket.
+ w = WSASocket (AF_INET, SOCK_STREAM, 0, NULL, 0, 0);
+ wsa_assert (w != INVALID_SOCKET);
+
+ // Connect to the remote peer.
+ rc = connect (w, (sockaddr *) &addr, sizeof (addr));
+ wsa_assert (rc != SOCKET_ERROR);
+
+ // Accept connection from w
+ r = accept (listener, NULL, NULL);
+ wsa_assert (r != INVALID_SOCKET);
+
+ rc = closesocket (listener);
+ wsa_assert (rc != SOCKET_ERROR);
+}
+
+zs::fd_signaler_t::~fd_signaler_t ()
+{
+ int rc = closesocket (w);
+ wsa_assert (rc != SOCKET_ERROR);
+
+ rc = closesocket (r);
+ wsa_assert (rc != SOCKET_ERROR);
+}
+
+void zs::fd_signaler_t::signal (int signal_)
+{
+ zs_assert (signal_ >= 0 && signal_ < 64);
+ char c = (char) signal_;
+ int rc = send (w, &c, 1, 0);
+ win_assert (rc != SOCKET_ERROR);
+}
+
+zs::fd_signaler_t::signals_t zs::fd_signaler_t::check ()
+{
+ char buffer [32];
+ int nbytes = recv (r, buffer, 32, 0);
+ win_assert (nbytes != -1);
+
+ signals_t signals = 0;
+ for (int pos = 0; pos != nbytes; pos++) {
+ zs_assert (buffer [pos] < 64);
+ signals |= (1 << (buffer [pos]));
+ }
+ return signals;
+}
+
+zs::fd_t zs::fd_signaler_t::get_fd ()
+{
+ return r;
+}
+
+#else
+
+#include <sys/types.h>
+#include <sys/socket.h>
+
+zs::fd_signaler_t::fd_signaler_t ()
+{
+ int sv [2];
+ int rc = socketpair (AF_UNIX, SOCK_STREAM, 0, sv);
+ errno_assert (rc == 0);
+ w = sv [0];
+ r = sv [1];
+
+ // Set to non-blocking mode.
+ int flags = fcntl (r, F_GETFL, 0);
+ if (flags == -1)
+ flags = 0;
+ rc = fcntl (r, F_SETFL, flags | O_NONBLOCK);
+ errno_assert (rc != -1);
+}
+
+zs::fd_signaler_t::~fd_signaler_t ()
+{
+ close (w);
+ close (r);
+}
+
+void zs::fd_signaler_t::signal (int signal_)
+{
+ zs_assert (signal_ >= 0 && signal_ < 64);
+ unsigned char c = (unsigned char) signal_;
+ ssize_t nbytes = send (w, &c, 1, 0);
+ errno_assert (nbytes == 1);
+}
+
+zs::fd_signaler_t::signals_t zs::fd_signaler_t::check ()
+{
+ unsigned char buffer [32];
+ ssize_t nbytes = recv (r, buffer, 32, 0);
+ errno_assert (nbytes != -1);
+ signals_t signals = 0;
+ for (int pos = 0; pos != nbytes; pos ++) {
+ zs_assert (buffer [pos] < 64);
+ signals |= (1 << (buffer [pos]));
+ }
+ return signals;
+}
+
+zs::fd_t zs::fd_signaler_t::get_fd ()
+{
+ return r;
+}
+
+#endif
+
+#if defined ZS_HAVE_OPENVMS
+
+int zs::fd_signaler_t::socketpair (int domain_, int type_, int protocol_,
+ int sv_ [2])
+{
+ int listener;
+ sockaddr_in lcladdr;
+ socklen_t lcladdr_len;
+ int rc;
+ int on = 1;
+
+ zs_assert (type_ == SOCK_STREAM);
+
+ // Fill in the localhost address (127.0.0.1).
+ memset (&lcladdr, 0, sizeof (lcladdr));
+ lcladdr.sin_family = AF_INET;
+ lcladdr.sin_addr.s_addr = 0x0100007f;
+ lcladdr.sin_port = INADDR_ANY;
+
+ listener = socket (AF_INET, SOCK_STREAM, 0);
+ errno_assert (listener != -1);
+
+ rc = setsockopt (listener, IPPROTO_TCP, TCP_NODELAY, &on, sizeof (on));
+ errno_assert (rc != -1);
+
+ rc = setsockopt (listener, IPPROTO_TCP, TCP_NODELACK, &on, sizeof (on));
+ errno_assert (rc != -1);
+
+ rc = bind(listener, (struct sockaddr*) &lcladdr, sizeof (lcladdr));
+ errno_assert (rc != -1);
+
+ lcladdr_len = sizeof (lcladdr);
+
+ rc = getsockname (listener, (struct sockaddr*) &lcladdr, &lcladdr_len);
+ errno_assert (rc != -1);
+
+ rc = listen (listener, 1);
+ errno_assert (rc != -1);
+
+ sv_ [0] = socket (AF_INET, SOCK_STREAM, 0);
+ errno_assert (rc != -1);
+
+ rc = setsockopt (sv_ [0], IPPROTO_TCP, TCP_NODELAY, &on, sizeof (on));
+ errno_assert (rc != -1);
+
+ rc = setsockopt (sv_ [0], IPPROTO_TCP, TCP_NODELACK, &on, sizeof (on));
+ errno_assert (rc != -1);
+
+ rc = connect (sv_ [0], (struct sockaddr*) &lcladdr, sizeof (lcladdr));
+ errno_assert (rc != -1);
+
+ sv_ [1] = accept (listener, NULL, NULL);
+ errno_assert (sv_ [1] != -1);
+
+ close (listener);
+
+ return 0;
+}
+
+#endif
+
generated by cgit v1.2.3 (git 2.25.1) at 2024-06-26 17:52:00 +0000