summaryrefslogtreecommitdiffstats
path: root/ipc/unixfd/UnixSocketWatcher.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'ipc/unixfd/UnixSocketWatcher.cpp')
-rw-r--r--ipc/unixfd/UnixSocketWatcher.cpp136
1 files changed, 136 insertions, 0 deletions
diff --git a/ipc/unixfd/UnixSocketWatcher.cpp b/ipc/unixfd/UnixSocketWatcher.cpp
new file mode 100644
index 000000000..2ce32e93d
--- /dev/null
+++ b/ipc/unixfd/UnixSocketWatcher.cpp
@@ -0,0 +1,136 @@
+/* -*- Mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; tab-width: 40 -*- */
+/* vim: set ts=2 et sw=2 tw=80: */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#include <fcntl.h>
+#include "UnixSocketWatcher.h"
+
+namespace mozilla {
+namespace ipc {
+
+UnixSocketWatcher::~UnixSocketWatcher()
+{
+}
+
+void UnixSocketWatcher::Close()
+{
+ MOZ_ASSERT(MessageLoopForIO::current() == GetIOLoop());
+
+ mConnectionStatus = SOCKET_IS_DISCONNECTED;
+ UnixFdWatcher::Close();
+}
+
+nsresult
+UnixSocketWatcher::Connect(const struct sockaddr* aAddr, socklen_t aAddrLen)
+{
+ MOZ_ASSERT(MessageLoopForIO::current() == GetIOLoop());
+ MOZ_ASSERT(IsOpen());
+ MOZ_ASSERT(aAddr || !aAddrLen);
+
+ if (TEMP_FAILURE_RETRY(connect(GetFd(), aAddr, aAddrLen) < 0)) {
+ if (errno == EINPROGRESS) {
+ mConnectionStatus = SOCKET_IS_CONNECTING;
+ // Set up a write watch to receive the connect signal
+ AddWatchers(WRITE_WATCHER, false);
+ return NS_OK;
+ }
+ OnError("connect", errno);
+ return NS_ERROR_FAILURE;
+ }
+
+ mConnectionStatus = SOCKET_IS_CONNECTED;
+ OnConnected();
+
+ return NS_OK;
+}
+
+nsresult
+UnixSocketWatcher::Listen(const struct sockaddr* aAddr, socklen_t aAddrLen)
+{
+ MOZ_ASSERT(MessageLoopForIO::current() == GetIOLoop());
+ MOZ_ASSERT(IsOpen());
+ MOZ_ASSERT(aAddr || !aAddrLen);
+
+ if (mConnectionStatus == SOCKET_IS_DISCONNECTED) {
+ // We init the socket descriptor when we listen for the first time.
+ if (bind(GetFd(), aAddr, aAddrLen) < 0) {
+ OnError("bind", errno);
+ return NS_ERROR_FAILURE;
+ }
+ if (listen(GetFd(), 1) < 0) {
+ OnError("listen", errno);
+ return NS_ERROR_FAILURE;
+ }
+ }
+ mConnectionStatus = SOCKET_IS_LISTENING;
+ OnListening();
+
+ return NS_OK;
+}
+
+UnixSocketWatcher::UnixSocketWatcher(MessageLoop* aIOLoop)
+: UnixFdWatcher(aIOLoop)
+, mConnectionStatus(SOCKET_IS_DISCONNECTED)
+{
+}
+
+UnixSocketWatcher::UnixSocketWatcher(MessageLoop* aIOLoop, int aFd,
+ ConnectionStatus aConnectionStatus)
+: UnixFdWatcher(aIOLoop, aFd)
+, mConnectionStatus(aConnectionStatus)
+{
+}
+
+void
+UnixSocketWatcher::SetSocket(int aFd, ConnectionStatus aConnectionStatus)
+{
+ MOZ_ASSERT(MessageLoopForIO::current() == GetIOLoop());
+
+ SetFd(aFd);
+ mConnectionStatus = aConnectionStatus;
+}
+
+void
+UnixSocketWatcher::OnFileCanReadWithoutBlocking(int aFd)
+{
+ MOZ_ASSERT(MessageLoopForIO::current() == GetIOLoop());
+ MOZ_ASSERT(aFd == GetFd());
+
+ if (mConnectionStatus == SOCKET_IS_CONNECTED) {
+ OnSocketCanReceiveWithoutBlocking();
+ } else if (mConnectionStatus == SOCKET_IS_LISTENING) {
+ OnSocketCanAcceptWithoutBlocking();
+ } else {
+ NS_NOTREACHED("invalid connection state for reading");
+ }
+}
+
+void
+UnixSocketWatcher::OnFileCanWriteWithoutBlocking(int aFd)
+{
+ MOZ_ASSERT(MessageLoopForIO::current() == GetIOLoop());
+ MOZ_ASSERT(aFd == GetFd());
+
+ if (mConnectionStatus == SOCKET_IS_CONNECTED) {
+ OnSocketCanSendWithoutBlocking();
+ } else if (mConnectionStatus == SOCKET_IS_CONNECTING) {
+ RemoveWatchers(WRITE_WATCHER);
+ int error = 0;
+ socklen_t len = sizeof(error);
+ if (getsockopt(GetFd(), SOL_SOCKET, SO_ERROR, &error, &len) < 0) {
+ OnError("getsockopt", errno);
+ } else if (error) {
+ OnError("connect", error);
+ } else {
+ mConnectionStatus = SOCKET_IS_CONNECTED;
+ OnConnected();
+ }
+ } else {
+ NS_NOTREACHED("invalid connection state for writing");
+ }
+}
+
+}
+}