summaryrefslogtreecommitdiffstats
path: root/ipc/hal/DaemonSocketPDU.cpp
diff options
context:
space:
mode:
authorMatt A. Tobin <mattatobin@localhost.localdomain>2018-02-02 04:16:08 -0500
committerMatt A. Tobin <mattatobin@localhost.localdomain>2018-02-02 04:16:08 -0500
commit5f8de423f190bbb79a62f804151bc24824fa32d8 (patch)
tree10027f336435511475e392454359edea8e25895d /ipc/hal/DaemonSocketPDU.cpp
parent49ee0794b5d912db1f95dce6eb52d781dc210db5 (diff)
downloadUXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.gz
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.lz
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.xz
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.zip
Add m-esr52 at 52.6.0
Diffstat (limited to 'ipc/hal/DaemonSocketPDU.cpp')
-rw-r--r--ipc/hal/DaemonSocketPDU.cpp202
1 files changed, 202 insertions, 0 deletions
diff --git a/ipc/hal/DaemonSocketPDU.cpp b/ipc/hal/DaemonSocketPDU.cpp
new file mode 100644
index 000000000..de6aa0da7
--- /dev/null
+++ b/ipc/hal/DaemonSocketPDU.cpp
@@ -0,0 +1,202 @@
+/* -*- 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 "DaemonSocketPDU.h"
+#include "mozilla/ipc/DaemonSocketConsumer.h"
+#include "nsISupportsImpl.h" // for MOZ_COUNT_CTOR, MOZ_COUNT_DTOR
+
+#ifdef CHROMIUM_LOG
+#undef CHROMIUM_LOG
+#endif
+
+#if defined(MOZ_WIDGET_GONK)
+#include <android/log.h>
+#define CHROMIUM_LOG(args...) __android_log_print(ANDROID_LOG_INFO, "I/O", args);
+#else
+#include <stdio.h>
+#define IODEBUG true
+#define CHROMIUM_LOG(args...) if (IODEBUG) printf(args);
+#endif
+
+namespace mozilla {
+namespace ipc {
+
+//
+// DaemonSocketPDU
+//
+
+DaemonSocketPDU::DaemonSocketPDU(uint8_t aService, uint8_t aOpcode,
+ uint16_t aPayloadSize)
+ : mConsumer(nullptr)
+{
+ MOZ_COUNT_CTOR_INHERITED(DaemonSocketPDU, UnixSocketIOBuffer);
+
+ // Allocate memory
+ size_t availableSpace = PDU_HEADER_SIZE + aPayloadSize;
+ ResetBuffer(new uint8_t[availableSpace], 0, 0, availableSpace);
+
+ // Reserve PDU header
+ uint8_t* data = Append(PDU_HEADER_SIZE);
+ MOZ_ASSERT(data);
+
+ // Setup PDU header
+ data[PDU_OFF_SERVICE] = aService;
+ data[PDU_OFF_OPCODE] = aOpcode;
+ memcpy(data + PDU_OFF_LENGTH, &aPayloadSize, sizeof(aPayloadSize));
+}
+
+DaemonSocketPDU::DaemonSocketPDU(size_t aPayloadSize)
+ : mConsumer(nullptr)
+{
+ MOZ_COUNT_CTOR_INHERITED(DaemonSocketPDU, UnixSocketIOBuffer);
+
+ size_t availableSpace = PDU_HEADER_SIZE + aPayloadSize;
+ ResetBuffer(new uint8_t[availableSpace], 0, 0, availableSpace);
+}
+
+DaemonSocketPDU::~DaemonSocketPDU()
+{
+ MOZ_COUNT_DTOR_INHERITED(DaemonSocketPDU, UnixSocketIOBuffer);
+
+ UniquePtr<uint8_t[]> data(GetBuffer());
+ ResetBuffer(nullptr, 0, 0, 0);
+}
+
+void
+DaemonSocketPDU::GetHeader(uint8_t& aService, uint8_t& aOpcode,
+ uint16_t& aPayloadSize)
+{
+ memcpy(&aService, GetData(PDU_OFF_SERVICE), sizeof(aService));
+ memcpy(&aOpcode, GetData(PDU_OFF_OPCODE), sizeof(aOpcode));
+ memcpy(&aPayloadSize, GetData(PDU_OFF_LENGTH), sizeof(aPayloadSize));
+}
+
+ssize_t
+DaemonSocketPDU::Send(int aFd)
+{
+ struct iovec iv;
+ memset(&iv, 0, sizeof(iv));
+ iv.iov_base = GetData(GetLeadingSpace());
+ iv.iov_len = GetSize();
+
+ struct msghdr msg;
+ memset(&msg, 0, sizeof(msg));
+ msg.msg_iov = &iv;
+ msg.msg_iovlen = 1;
+ msg.msg_control = nullptr;
+ msg.msg_controllen = 0;
+
+ ssize_t res = TEMP_FAILURE_RETRY(sendmsg(aFd, &msg, 0));
+ if (res < 0) {
+ MOZ_ASSERT(errno != EBADF); /* internal error */
+ OnError("sendmsg", errno);
+ return -1;
+ }
+
+ Consume(res);
+
+ if (mConsumer) {
+ // We successfully sent a PDU, now store the
+ // result handler in the consumer.
+ mConsumer->StoreResultHandler(*this);
+ }
+
+ return res;
+}
+
+#define CMSGHDR_CONTAINS_FD(_cmsghdr) \
+ ( ((_cmsghdr)->cmsg_level == SOL_SOCKET) && \
+ ((_cmsghdr)->cmsg_type == SCM_RIGHTS) )
+
+ssize_t
+DaemonSocketPDU::Receive(int aFd)
+{
+ struct iovec iv;
+ memset(&iv, 0, sizeof(iv));
+ iv.iov_base = GetData(0);
+ iv.iov_len = GetAvailableSpace();
+
+ uint8_t cmsgbuf[CMSG_SPACE(sizeof(int)* MAX_NFDS)];
+
+ struct msghdr msg;
+ memset(&msg, 0, sizeof(msg));
+ msg.msg_iov = &iv;
+ msg.msg_iovlen = 1;
+ msg.msg_control = cmsgbuf;
+ msg.msg_controllen = sizeof(cmsgbuf);
+
+ ssize_t res = TEMP_FAILURE_RETRY(recvmsg(aFd, &msg, MSG_NOSIGNAL));
+ if (res < 0) {
+ MOZ_ASSERT(errno != EBADF); /* internal error */
+ OnError("recvmsg", errno);
+ return -1;
+ }
+ if (msg.msg_flags & (MSG_CTRUNC | MSG_OOB | MSG_ERRQUEUE)) {
+ return -1;
+ }
+
+ SetRange(0, res);
+
+ struct cmsghdr* chdr = CMSG_FIRSTHDR(&msg);
+
+ for (; chdr; chdr = CMSG_NXTHDR(&msg, chdr)) {
+ if (NS_WARN_IF(!CMSGHDR_CONTAINS_FD(chdr))) {
+ continue;
+ }
+ // Retrieve sent file descriptors.
+ size_t fdCount = (chdr->cmsg_len - CMSG_ALIGN(sizeof(struct cmsghdr))) / sizeof(int);
+ for (size_t i = 0; i < fdCount; i++) {
+ int* receivedFd = static_cast<int*>(CMSG_DATA(chdr)) + i;
+ mReceivedFds.AppendElement(*receivedFd);
+ }
+ }
+
+ return res;
+}
+
+nsTArray<int>
+DaemonSocketPDU::AcquireFds()
+{
+ // Forget all RAII object to avoid closing the fds.
+ nsTArray<int> fds;
+ for (auto& fd : mReceivedFds) {
+ fds.AppendElement(fd.forget());
+ }
+ mReceivedFds.Clear();
+ return fds;
+}
+
+nsresult
+DaemonSocketPDU::UpdateHeader()
+{
+ size_t len = GetPayloadSize();
+ if (len >= PDU_MAX_PAYLOAD_LENGTH) {
+ return NS_ERROR_ILLEGAL_VALUE;
+ }
+ uint16_t len16 = static_cast<uint16_t>(len);
+
+ memcpy(GetData(PDU_OFF_LENGTH), &len16, sizeof(len16));
+
+ return NS_OK;
+}
+
+size_t
+DaemonSocketPDU::GetPayloadSize() const
+{
+ MOZ_ASSERT(GetSize() >= PDU_HEADER_SIZE);
+
+ return GetSize() - PDU_HEADER_SIZE;
+}
+
+void
+DaemonSocketPDU::OnError(const char* aFunction, int aErrno)
+{
+ CHROMIUM_LOG("%s failed with error %d (%s)",
+ aFunction, aErrno, strerror(aErrno));
+}
+
+}
+}