1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
2 /* vim: set ts=2 et sw=2 tw=80: */
3 /* This Source Code Form is subject to the terms of the Mozilla Public
4 * License, v. 2.0. If a copy of the MPL was not distributed with this
5 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
6
7 #include "ExternalHelperAppChild.h"
8 #include "mozilla/dom/BrowserChild.h"
9 #include "nsIInputStream.h"
10 #include "nsIRequest.h"
11 #include "nsIResumableChannel.h"
12 #include "nsIPropertyBag2.h"
13 #include "nsNetUtil.h"
14
15 namespace mozilla {
16 namespace dom {
17
NS_IMPL_ISUPPORTS(ExternalHelperAppChild,nsIStreamListener,nsIRequestObserver)18 NS_IMPL_ISUPPORTS(ExternalHelperAppChild, nsIStreamListener, nsIRequestObserver)
19
20 ExternalHelperAppChild::ExternalHelperAppChild() : mStatus(NS_OK) {}
21
~ExternalHelperAppChild()22 ExternalHelperAppChild::~ExternalHelperAppChild() {}
23
24 //-----------------------------------------------------------------------------
25 // nsIStreamListener
26 //-----------------------------------------------------------------------------
27 NS_IMETHODIMP
OnDataAvailable(nsIRequest * request,nsIInputStream * input,uint64_t offset,uint32_t count)28 ExternalHelperAppChild::OnDataAvailable(nsIRequest* request,
29 nsIInputStream* input, uint64_t offset,
30 uint32_t count) {
31 if (NS_FAILED(mStatus)) return mStatus;
32
33 static uint32_t const kCopyChunkSize = 128 * 1024;
34 uint32_t toRead = std::min<uint32_t>(count, kCopyChunkSize);
35
36 nsCString data;
37
38 while (count) {
39 nsresult rv = NS_ReadInputStreamToString(input, data, toRead);
40 if (NS_WARN_IF(NS_FAILED(rv))) {
41 return rv;
42 }
43
44 if (NS_WARN_IF(!SendOnDataAvailable(data, offset, toRead))) {
45 return NS_ERROR_UNEXPECTED;
46 }
47
48 count -= toRead;
49 offset += toRead;
50 toRead = std::min<uint32_t>(count, kCopyChunkSize);
51 }
52
53 return NS_OK;
54 }
55
56 //////////////////////////////////////////////////////////////////////////////
57 // nsIRequestObserver
58 //////////////////////////////////////////////////////////////////////////////
59
60 NS_IMETHODIMP
OnStartRequest(nsIRequest * request)61 ExternalHelperAppChild::OnStartRequest(nsIRequest* request) {
62 nsresult rv = mHandler->OnStartRequest(request);
63 NS_ENSURE_SUCCESS(rv, NS_ERROR_UNEXPECTED);
64
65 nsCString entityID;
66 nsCOMPtr<nsIResumableChannel> resumable(do_QueryInterface(request));
67 if (resumable) {
68 resumable->GetEntityID(entityID);
69 }
70 SendOnStartRequest(entityID);
71 return NS_OK;
72 }
73
74 NS_IMETHODIMP
OnStopRequest(nsIRequest * request,nsresult status)75 ExternalHelperAppChild::OnStopRequest(nsIRequest* request, nsresult status) {
76 // mHandler can be null if we diverted the request to the parent
77 if (mHandler) {
78 nsresult rv = mHandler->OnStopRequest(request, status);
79 SendOnStopRequest(status);
80 NS_ENSURE_SUCCESS(rv, NS_ERROR_UNEXPECTED);
81 }
82
83 return NS_OK;
84 }
85
RecvCancel(const nsresult & aStatus)86 mozilla::ipc::IPCResult ExternalHelperAppChild::RecvCancel(
87 const nsresult& aStatus) {
88 mStatus = aStatus;
89 return IPC_OK();
90 }
91
92 } // namespace dom
93 } // namespace mozilla
94