summaryrefslogtreecommitdiffstats
path: root/dom/archivereader/ArchiveReader.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'dom/archivereader/ArchiveReader.cpp')
-rw-r--r--dom/archivereader/ArchiveReader.cpp217
1 files changed, 217 insertions, 0 deletions
diff --git a/dom/archivereader/ArchiveReader.cpp b/dom/archivereader/ArchiveReader.cpp
new file mode 100644
index 000000000..f6985d989
--- /dev/null
+++ b/dom/archivereader/ArchiveReader.cpp
@@ -0,0 +1,217 @@
+/* -*- 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 "ArchiveReader.h"
+#include "ArchiveRequest.h"
+#include "ArchiveEvent.h"
+#include "ArchiveZipEvent.h"
+
+#include "nsIURI.h"
+#include "nsNetCID.h"
+
+#include "mozilla/dom/ArchiveReaderBinding.h"
+#include "mozilla/dom/BindingDeclarations.h"
+#include "mozilla/dom/File.h"
+#include "mozilla/dom/EncodingUtils.h"
+#include "mozilla/Preferences.h"
+
+using namespace mozilla;
+using namespace mozilla::dom;
+USING_ARCHIVEREADER_NAMESPACE
+
+/* static */ already_AddRefed<ArchiveReader>
+ArchiveReader::Constructor(const GlobalObject& aGlobal,
+ Blob& aBlob,
+ const ArchiveReaderOptions& aOptions,
+ ErrorResult& aError)
+{
+ nsCOMPtr<nsPIDOMWindowInner> window = do_QueryInterface(aGlobal.GetAsSupports());
+ if (!window) {
+ aError.Throw(NS_ERROR_UNEXPECTED);
+ return nullptr;
+ }
+
+ nsAutoCString encoding;
+ if (!EncodingUtils::FindEncodingForLabelNoReplacement(aOptions.mEncoding,
+ encoding)) {
+ aError.ThrowRangeError<MSG_ENCODING_NOT_SUPPORTED>(aOptions.mEncoding);
+ return nullptr;
+ }
+
+ RefPtr<ArchiveReader> reader =
+ new ArchiveReader(aBlob, window, encoding);
+ return reader.forget();
+}
+
+ArchiveReader::ArchiveReader(Blob& aBlob, nsPIDOMWindowInner* aWindow,
+ const nsACString& aEncoding)
+ : mBlobImpl(aBlob.Impl())
+ , mWindow(aWindow)
+ , mStatus(NOT_STARTED)
+ , mEncoding(aEncoding)
+{
+ MOZ_ASSERT(aWindow);
+}
+
+ArchiveReader::~ArchiveReader()
+{
+}
+
+/* virtual */ JSObject*
+ArchiveReader::WrapObject(JSContext* aCx, JS::Handle<JSObject*> aGivenProto)
+{
+ return ArchiveReaderBinding::Wrap(aCx, this, aGivenProto);
+}
+
+nsresult
+ArchiveReader::RegisterRequest(ArchiveRequest* aRequest)
+{
+ switch (mStatus) {
+ // Append to the list and let's start to work:
+ case NOT_STARTED:
+ mRequests.AppendElement(aRequest);
+ return OpenArchive();
+
+ // Just append to the list:
+ case WORKING:
+ mRequests.AppendElement(aRequest);
+ return NS_OK;
+
+ // Return data!
+ case READY:
+ RequestReady(aRequest);
+ return NS_OK;
+ }
+
+ NS_ASSERTION(false, "unexpected mStatus value");
+ return NS_OK;
+}
+
+// This returns the input stream
+nsresult
+ArchiveReader::GetInputStream(nsIInputStream** aInputStream)
+{
+ // Getting the input stream
+ ErrorResult rv;
+ mBlobImpl->GetInternalStream(aInputStream, rv);
+ if (NS_WARN_IF(rv.Failed())) {
+ return rv.StealNSResult();
+ }
+
+ return NS_OK;
+}
+
+nsresult
+ArchiveReader::GetSize(uint64_t* aSize)
+{
+ ErrorResult rv;
+ *aSize = mBlobImpl->GetSize(rv);
+ return rv.StealNSResult();
+}
+
+// Here we open the archive:
+nsresult
+ArchiveReader::OpenArchive()
+{
+ mStatus = WORKING;
+ nsresult rv;
+
+ // Target:
+ nsCOMPtr<nsIEventTarget> target = do_GetService(NS_STREAMTRANSPORTSERVICE_CONTRACTID);
+ NS_ASSERTION(target, "Must have stream transport service");
+
+ // Here a Event to make everything async:
+ RefPtr<ArchiveReaderEvent> event;
+
+ /* FIXME: If we want to support more than 1 format we should check the content type here: */
+ event = new ArchiveReaderZipEvent(this, mEncoding);
+ rv = target->Dispatch(event, NS_DISPATCH_NORMAL);
+ NS_ENSURE_SUCCESS(rv, rv);
+
+ // In order to be sure that this object exists when the event finishes its task,
+ // we increase the refcount here:
+ AddRef();
+
+ return NS_OK;
+}
+
+// Data received from the dispatched event:
+void
+ArchiveReader::Ready(nsTArray<RefPtr<File>>& aFileList,
+ nsresult aStatus)
+{
+ mStatus = READY;
+
+ // Let's store the values:
+ mData.fileList = aFileList;
+ mData.status = aStatus;
+
+ // Propagate the results:
+ for (uint32_t index = 0; index < mRequests.Length(); ++index) {
+ RefPtr<ArchiveRequest> request = mRequests[index];
+ RequestReady(request);
+ }
+
+ mRequests.Clear();
+
+ // The async operation is concluded, we can decrease the reference:
+ Release();
+}
+
+void
+ArchiveReader::RequestReady(ArchiveRequest* aRequest)
+{
+ // The request will do the rest:
+ aRequest->ReaderReady(mData.fileList, mData.status);
+}
+
+already_AddRefed<ArchiveRequest>
+ArchiveReader::GetFilenames()
+{
+ RefPtr<ArchiveRequest> request = GenerateArchiveRequest();
+ request->OpGetFilenames();
+
+ return request.forget();
+}
+
+already_AddRefed<ArchiveRequest>
+ArchiveReader::GetFile(const nsAString& filename)
+{
+ RefPtr<ArchiveRequest> request = GenerateArchiveRequest();
+ request->OpGetFile(filename);
+
+ return request.forget();
+}
+
+already_AddRefed<ArchiveRequest>
+ArchiveReader::GetFiles()
+{
+ RefPtr<ArchiveRequest> request = GenerateArchiveRequest();
+ request->OpGetFiles();
+
+ return request.forget();
+}
+
+already_AddRefed<ArchiveRequest>
+ArchiveReader::GenerateArchiveRequest()
+{
+ NS_ASSERTION(NS_IsMainThread(), "Wrong thread!");
+ return ArchiveRequest::Create(mWindow, this);
+}
+
+NS_IMPL_CYCLE_COLLECTION_WRAPPERCACHE(ArchiveReader,
+ mBlobImpl,
+ mWindow,
+ mData.fileList,
+ mRequests)
+
+NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(ArchiveReader)
+ NS_WRAPPERCACHE_INTERFACE_MAP_ENTRY
+ NS_INTERFACE_MAP_ENTRY(nsISupports)
+NS_INTERFACE_MAP_END
+
+NS_IMPL_CYCLE_COLLECTING_ADDREF(ArchiveReader)
+NS_IMPL_CYCLE_COLLECTING_RELEASE(ArchiveReader)