summaryrefslogtreecommitdiffstats
path: root/ipc/glue/InputStreamUtils.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/glue/InputStreamUtils.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/glue/InputStreamUtils.cpp')
-rw-r--r--ipc/glue/InputStreamUtils.cpp192
1 files changed, 192 insertions, 0 deletions
diff --git a/ipc/glue/InputStreamUtils.cpp b/ipc/glue/InputStreamUtils.cpp
new file mode 100644
index 000000000..bbc863efd
--- /dev/null
+++ b/ipc/glue/InputStreamUtils.cpp
@@ -0,0 +1,192 @@
+/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim: set ts=8 sts=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 "InputStreamUtils.h"
+
+#include "nsIIPCSerializableInputStream.h"
+
+#include "mozilla/Assertions.h"
+#include "mozilla/dom/File.h"
+#include "mozilla/dom/ipc/BlobChild.h"
+#include "mozilla/dom/ipc/BlobParent.h"
+#include "nsComponentManagerUtils.h"
+#include "nsDebug.h"
+#include "nsID.h"
+#include "nsIXULRuntime.h"
+#include "nsMIMEInputStream.h"
+#include "nsMultiplexInputStream.h"
+#include "nsNetCID.h"
+#include "nsStringStream.h"
+#include "nsXULAppAPI.h"
+
+using namespace mozilla::dom;
+
+namespace {
+
+NS_DEFINE_CID(kStringInputStreamCID, NS_STRINGINPUTSTREAM_CID);
+NS_DEFINE_CID(kFileInputStreamCID, NS_LOCALFILEINPUTSTREAM_CID);
+NS_DEFINE_CID(kPartialFileInputStreamCID, NS_PARTIALLOCALFILEINPUTSTREAM_CID);
+NS_DEFINE_CID(kBufferedInputStreamCID, NS_BUFFEREDINPUTSTREAM_CID);
+NS_DEFINE_CID(kMIMEInputStreamCID, NS_MIMEINPUTSTREAM_CID);
+NS_DEFINE_CID(kMultiplexInputStreamCID, NS_MULTIPLEXINPUTSTREAM_CID);
+
+} // namespace
+
+namespace mozilla {
+namespace ipc {
+
+void
+SerializeInputStream(nsIInputStream* aInputStream,
+ InputStreamParams& aParams,
+ nsTArray<FileDescriptor>& aFileDescriptors)
+{
+ MOZ_ASSERT(aInputStream);
+
+ nsCOMPtr<nsIIPCSerializableInputStream> serializable =
+ do_QueryInterface(aInputStream);
+ if (!serializable) {
+ MOZ_CRASH("Input stream is not serializable!");
+ }
+
+ serializable->Serialize(aParams, aFileDescriptors);
+
+ if (aParams.type() == InputStreamParams::T__None) {
+ MOZ_CRASH("Serialize failed!");
+ }
+}
+
+void
+SerializeInputStream(nsIInputStream* aInputStream,
+ OptionalInputStreamParams& aParams,
+ nsTArray<FileDescriptor>& aFileDescriptors)
+{
+ if (aInputStream) {
+ InputStreamParams params;
+ SerializeInputStream(aInputStream, params, aFileDescriptors);
+ aParams = params;
+ }
+ else {
+ aParams = mozilla::void_t();
+ }
+}
+
+already_AddRefed<nsIInputStream>
+DeserializeInputStream(const InputStreamParams& aParams,
+ const nsTArray<FileDescriptor>& aFileDescriptors)
+{
+ nsCOMPtr<nsIInputStream> stream;
+ nsCOMPtr<nsIIPCSerializableInputStream> serializable;
+
+ switch (aParams.type()) {
+ case InputStreamParams::TStringInputStreamParams:
+ serializable = do_CreateInstance(kStringInputStreamCID);
+ break;
+
+ case InputStreamParams::TFileInputStreamParams:
+ serializable = do_CreateInstance(kFileInputStreamCID);
+ break;
+
+ case InputStreamParams::TPartialFileInputStreamParams:
+ serializable = do_CreateInstance(kPartialFileInputStreamCID);
+ break;
+
+ case InputStreamParams::TTemporaryFileInputStreamParams:
+ serializable = new nsTemporaryFileInputStream();
+ break;
+
+ case InputStreamParams::TBufferedInputStreamParams:
+ serializable = do_CreateInstance(kBufferedInputStreamCID);
+ break;
+
+ case InputStreamParams::TMIMEInputStreamParams:
+ serializable = do_CreateInstance(kMIMEInputStreamCID);
+ break;
+
+ case InputStreamParams::TMultiplexInputStreamParams:
+ serializable = do_CreateInstance(kMultiplexInputStreamCID);
+ break;
+
+ // When the input stream already exists in this process, all we need to do
+ // is retrieve the original instead of sending any data over the wire.
+ case InputStreamParams::TRemoteInputStreamParams: {
+ if (NS_WARN_IF(!XRE_IsParentProcess())) {
+ return nullptr;
+ }
+
+ const nsID& id = aParams.get_RemoteInputStreamParams().id();
+
+ RefPtr<BlobImpl> blobImpl = BlobParent::GetBlobImplForID(id);
+
+ MOZ_ASSERT(blobImpl, "Invalid blob contents");
+
+ // If fetching the internal stream fails, we ignore it and return a
+ // null stream.
+ ErrorResult rv;
+ nsCOMPtr<nsIInputStream> stream;
+ blobImpl->GetInternalStream(getter_AddRefs(stream), rv);
+ if (NS_WARN_IF(rv.Failed()) || !stream) {
+ NS_WARNING("Couldn't obtain a valid stream from the blob");
+ rv.SuppressException();
+ }
+ return stream.forget();
+ }
+
+ case InputStreamParams::TSameProcessInputStreamParams: {
+ MOZ_ASSERT(aFileDescriptors.IsEmpty());
+
+ const SameProcessInputStreamParams& params =
+ aParams.get_SameProcessInputStreamParams();
+
+ stream = dont_AddRef(
+ reinterpret_cast<nsIInputStream*>(params.addRefedInputStream()));
+ MOZ_ASSERT(stream);
+
+ return stream.forget();
+ }
+
+ default:
+ MOZ_ASSERT(false, "Unknown params!");
+ return nullptr;
+ }
+
+ MOZ_ASSERT(serializable);
+
+ if (!serializable->Deserialize(aParams, aFileDescriptors)) {
+ MOZ_ASSERT(false, "Deserialize failed!");
+ return nullptr;
+ }
+
+ stream = do_QueryInterface(serializable);
+ MOZ_ASSERT(stream);
+
+ return stream.forget();
+}
+
+already_AddRefed<nsIInputStream>
+DeserializeInputStream(const OptionalInputStreamParams& aParams,
+ const nsTArray<FileDescriptor>& aFileDescriptors)
+{
+ nsCOMPtr<nsIInputStream> stream;
+
+ switch (aParams.type()) {
+ case OptionalInputStreamParams::Tvoid_t:
+ // Leave stream null.
+ break;
+
+ case OptionalInputStreamParams::TInputStreamParams:
+ stream = DeserializeInputStream(aParams.get_InputStreamParams(),
+ aFileDescriptors);
+ break;
+
+ default:
+ MOZ_ASSERT(false, "Unknown params!");
+ }
+
+ return stream.forget();
+}
+
+} // namespace ipc
+} // namespace mozilla