blob: 8b5700f5bf6e56c02b2d4b6ab0c9e255151c9067 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
/* -*- Mode: C++; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* 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/. */
#ifndef mozilla_net_MemoryDownloader_h__
#define mozilla_net_MemoryDownloader_h__
#include "mozilla/UniquePtr.h"
#include "nsCOMPtr.h"
#include "nsIStreamListener.h"
#include "nsTArray.h"
/**
* mozilla::net::MemoryDownloader
*
* This class is similar to nsIDownloader, but stores the downloaded
* stream in memory instead of a file. Ownership of the temporary
* memory is transferred to the observer when download is complete;
* there is no need to retain a reference to the downloader.
*/
namespace mozilla {
namespace net {
class MemoryDownloader final : public nsIStreamListener {
public:
NS_DECL_ISUPPORTS
NS_DECL_NSIREQUESTOBSERVER
NS_DECL_NSISTREAMLISTENER
using Data = mozilla::UniquePtr<FallibleTArray<uint8_t>>;
class IObserver : public nsISupports {
public:
// Note: aData may be null if (and only if) aStatus indicates failure.
virtual void OnDownloadComplete(MemoryDownloader* aDownloader,
nsIRequest* aRequest, nsresult aStatus,
Data aData) = 0;
};
explicit MemoryDownloader(IObserver* aObserver);
private:
virtual ~MemoryDownloader() = default;
static nsresult ConsumeData(nsIInputStream* in, void* closure,
const char* fromRawSegment, uint32_t toOffset,
uint32_t count, uint32_t* writeCount);
RefPtr<IObserver> mObserver;
Data mData;
nsresult mStatus;
};
} // namespace net
} // namespace mozilla
#endif // mozilla_net_MemoryDownloader_h__
|