1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/ipc/glue/InputStreamUtils.cpp Wed Dec 31 06:09:35 2014 +0100
1.3 @@ -0,0 +1,178 @@
1.4 +/* This Source Code Form is subject to the terms of the Mozilla Public
1.5 + * License, v. 2.0. If a copy of the MPL was not distributed with this
1.6 + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
1.7 +
1.8 +#include "InputStreamUtils.h"
1.9 +
1.10 +#include "nsIIPCSerializableInputStream.h"
1.11 +
1.12 +#include "mozilla/Assertions.h"
1.13 +#include "mozilla/dom/ipc/Blob.h"
1.14 +#include "nsComponentManagerUtils.h"
1.15 +#include "nsDebug.h"
1.16 +#include "nsID.h"
1.17 +#include "nsIDOMFile.h"
1.18 +#include "nsIXULRuntime.h"
1.19 +#include "nsMIMEInputStream.h"
1.20 +#include "nsMultiplexInputStream.h"
1.21 +#include "nsNetCID.h"
1.22 +#include "nsStringStream.h"
1.23 +#include "nsThreadUtils.h"
1.24 +#include "nsXULAppAPI.h"
1.25 +
1.26 +using mozilla::dom::BlobChild;
1.27 +using mozilla::dom::BlobParent;
1.28 +
1.29 +namespace {
1.30 +
1.31 +NS_DEFINE_CID(kStringInputStreamCID, NS_STRINGINPUTSTREAM_CID);
1.32 +NS_DEFINE_CID(kFileInputStreamCID, NS_LOCALFILEINPUTSTREAM_CID);
1.33 +NS_DEFINE_CID(kPartialFileInputStreamCID, NS_PARTIALLOCALFILEINPUTSTREAM_CID);
1.34 +NS_DEFINE_CID(kBufferedInputStreamCID, NS_BUFFEREDINPUTSTREAM_CID);
1.35 +NS_DEFINE_CID(kMIMEInputStreamCID, NS_MIMEINPUTSTREAM_CID);
1.36 +NS_DEFINE_CID(kMultiplexInputStreamCID, NS_MULTIPLEXINPUTSTREAM_CID);
1.37 +
1.38 +} // anonymous namespace
1.39 +
1.40 +namespace mozilla {
1.41 +namespace ipc {
1.42 +
1.43 +void
1.44 +SerializeInputStream(nsIInputStream* aInputStream,
1.45 + InputStreamParams& aParams,
1.46 + nsTArray<FileDescriptor>& aFileDescriptors)
1.47 +{
1.48 + MOZ_ASSERT(NS_IsMainThread());
1.49 + MOZ_ASSERT(aInputStream);
1.50 +
1.51 + nsCOMPtr<nsIIPCSerializableInputStream> serializable =
1.52 + do_QueryInterface(aInputStream);
1.53 + if (!serializable) {
1.54 + MOZ_CRASH("Input stream is not serializable!");
1.55 + }
1.56 +
1.57 + serializable->Serialize(aParams, aFileDescriptors);
1.58 +
1.59 + if (aParams.type() == InputStreamParams::T__None) {
1.60 + MOZ_CRASH("Serialize failed!");
1.61 + }
1.62 +}
1.63 +
1.64 +void
1.65 +SerializeInputStream(nsIInputStream* aInputStream,
1.66 + OptionalInputStreamParams& aParams,
1.67 + nsTArray<FileDescriptor>& aFileDescriptors)
1.68 +{
1.69 + MOZ_ASSERT(NS_IsMainThread());
1.70 +
1.71 + if (aInputStream) {
1.72 + InputStreamParams params;
1.73 + SerializeInputStream(aInputStream, params, aFileDescriptors);
1.74 + aParams = params;
1.75 + }
1.76 + else {
1.77 + aParams = mozilla::void_t();
1.78 + }
1.79 +}
1.80 +
1.81 +already_AddRefed<nsIInputStream>
1.82 +DeserializeInputStream(const InputStreamParams& aParams,
1.83 + const nsTArray<FileDescriptor>& aFileDescriptors)
1.84 +{
1.85 + MOZ_ASSERT(NS_IsMainThread());
1.86 +
1.87 + nsCOMPtr<nsIIPCSerializableInputStream> serializable;
1.88 +
1.89 + switch (aParams.type()) {
1.90 + case InputStreamParams::TStringInputStreamParams:
1.91 + serializable = do_CreateInstance(kStringInputStreamCID);
1.92 + break;
1.93 +
1.94 + case InputStreamParams::TFileInputStreamParams:
1.95 + serializable = do_CreateInstance(kFileInputStreamCID);
1.96 + break;
1.97 +
1.98 + case InputStreamParams::TPartialFileInputStreamParams:
1.99 + serializable = do_CreateInstance(kPartialFileInputStreamCID);
1.100 + break;
1.101 +
1.102 + case InputStreamParams::TBufferedInputStreamParams:
1.103 + serializable = do_CreateInstance(kBufferedInputStreamCID);
1.104 + break;
1.105 +
1.106 + case InputStreamParams::TMIMEInputStreamParams:
1.107 + serializable = do_CreateInstance(kMIMEInputStreamCID);
1.108 + break;
1.109 +
1.110 + case InputStreamParams::TMultiplexInputStreamParams:
1.111 + serializable = do_CreateInstance(kMultiplexInputStreamCID);
1.112 + break;
1.113 +
1.114 + // When the input stream already exists in this process, all we need to do
1.115 + // is retrieve the original instead of sending any data over the wire.
1.116 + case InputStreamParams::TRemoteInputStreamParams: {
1.117 + nsCOMPtr<nsIDOMBlob> domBlob;
1.118 + const RemoteInputStreamParams& params =
1.119 + aParams.get_RemoteInputStreamParams();
1.120 +
1.121 + domBlob = params.remoteBlobParent() ?
1.122 + static_cast<BlobParent*>(params.remoteBlobParent())->GetBlob() :
1.123 + static_cast<BlobChild*>(params.remoteBlobChild())->GetBlob();
1.124 +
1.125 + MOZ_ASSERT(domBlob, "Invalid blob contents");
1.126 +
1.127 + // If fetching the internal stream fails, we ignore it and return a
1.128 + // null stream.
1.129 + nsCOMPtr<nsIInputStream> stream;
1.130 + nsresult rv = domBlob->GetInternalStream(getter_AddRefs(stream));
1.131 + if (NS_FAILED(rv) || !stream) {
1.132 + NS_WARNING("Couldn't obtain a valid stream from the blob");
1.133 + }
1.134 + return stream.forget();
1.135 + }
1.136 +
1.137 + default:
1.138 + MOZ_ASSERT(false, "Unknown params!");
1.139 + return nullptr;
1.140 + }
1.141 +
1.142 + MOZ_ASSERT(serializable);
1.143 +
1.144 + if (!serializable->Deserialize(aParams, aFileDescriptors)) {
1.145 + MOZ_ASSERT(false, "Deserialize failed!");
1.146 + return nullptr;
1.147 + }
1.148 +
1.149 + nsCOMPtr<nsIInputStream> stream = do_QueryInterface(serializable);
1.150 + MOZ_ASSERT(stream);
1.151 +
1.152 + return stream.forget();
1.153 +}
1.154 +
1.155 +already_AddRefed<nsIInputStream>
1.156 +DeserializeInputStream(const OptionalInputStreamParams& aParams,
1.157 + const nsTArray<FileDescriptor>& aFileDescriptors)
1.158 +{
1.159 + MOZ_ASSERT(NS_IsMainThread());
1.160 +
1.161 + nsCOMPtr<nsIInputStream> stream;
1.162 +
1.163 + switch (aParams.type()) {
1.164 + case OptionalInputStreamParams::Tvoid_t:
1.165 + // Leave stream null.
1.166 + break;
1.167 +
1.168 + case OptionalInputStreamParams::TInputStreamParams:
1.169 + stream = DeserializeInputStream(aParams.get_InputStreamParams(),
1.170 + aFileDescriptors);
1.171 + break;
1.172 +
1.173 + default:
1.174 + MOZ_ASSERT(false, "Unknown params!");
1.175 + }
1.176 +
1.177 + return stream.forget();
1.178 +}
1.179 +
1.180 +} // namespace ipc
1.181 +} // namespace mozilla
