137 lines
3.5 KiB
C++
137 lines
3.5 KiB
C++
/* -*- 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");
|
|
}
|
|
}
|
|
|
|
}
|
|
}
|