blob: 26326e2a49c12798c3d9a8a0a29a56251f946ca6 (
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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*-
*
* 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 nsWyciwygChannel_h___
#define nsWyciwygChannel_h___
#include "nsString.h"
#include "nsCOMPtr.h"
#include "nsILoadInfo.h"
#include "nsIWyciwygChannel.h"
#include "nsIStreamListener.h"
#include "nsICacheEntryOpenCallback.h"
#include "PrivateBrowsingChannel.h"
#include "mozilla/BasePrincipal.h"
class nsICacheEntry;
class nsICacheStorage;
class nsIInputStream;
class nsIInputStreamPump;
class nsILoadGroup;
class nsIOutputStream;
class nsIProgressEventSink;
class nsIURI;
extern mozilla::LazyLogModule gWyciwygLog;
//-----------------------------------------------------------------------------
class nsWyciwygChannel final: public nsIWyciwygChannel,
public nsIStreamListener,
public nsICacheEntryOpenCallback,
public mozilla::net::PrivateBrowsingChannel<nsWyciwygChannel>
{
public:
NS_DECL_THREADSAFE_ISUPPORTS
NS_DECL_NSIREQUEST
NS_DECL_NSICHANNEL
NS_DECL_NSIWYCIWYGCHANNEL
NS_DECL_NSIREQUESTOBSERVER
NS_DECL_NSISTREAMLISTENER
NS_DECL_NSICACHEENTRYOPENCALLBACK
// nsWyciwygChannel methods:
nsWyciwygChannel();
nsresult Init(nsIURI *uri);
protected:
virtual ~nsWyciwygChannel();
nsresult ReadFromCache();
nsresult EnsureWriteCacheEntry();
nsresult GetCacheStorage(nsICacheStorage **_retval);
nsresult OpenCacheEntryForReading(nsIURI *aURI);
nsresult OpenCacheEntryForWriting(nsIURI *aURI);
void WriteCharsetAndSourceToCache(int32_t aSource,
const nsCString& aCharset);
void NotifyListener();
friend class mozilla::net::PrivateBrowsingChannel<nsWyciwygChannel>;
enum EMode {
NONE,
WRITING,
READING
};
EMode mMode;
nsresult mStatus;
bool mIsPending;
bool mNeedToWriteCharset;
int32_t mCharsetSource;
nsCString mCharset;
int64_t mContentLength;
uint32_t mLoadFlags;
mozilla::NeckoOriginAttributes mOriginAttributes;
nsCOMPtr<nsIURI> mURI;
nsCOMPtr<nsIURI> mOriginalURI;
nsCOMPtr<nsISupports> mOwner;
nsCOMPtr<nsILoadInfo> mLoadInfo;
nsCOMPtr<nsIInterfaceRequestor> mCallbacks;
nsCOMPtr<nsIProgressEventSink> mProgressSink;
nsCOMPtr<nsILoadGroup> mLoadGroup;
nsCOMPtr<nsIStreamListener> mListener;
nsCOMPtr<nsISupports> mListenerContext;
// reuse as much of this channel implementation as we can
nsCOMPtr<nsIInputStreamPump> mPump;
// Cache related stuff
nsCOMPtr<nsICacheEntry> mCacheEntry;
nsCOMPtr<nsIOutputStream> mCacheOutputStream;
nsCOMPtr<nsIInputStream> mCacheInputStream;
bool mNeedToSetSecurityInfo;
nsCOMPtr<nsISupports> mSecurityInfo;
};
/**
* Casting nsWyciwygChannel to nsISupports is ambiguous.
* This method handles that.
*/
inline nsISupports*
ToSupports(nsWyciwygChannel* p)
{
return NS_ISUPPORTS_CAST(nsIStreamListener*, p);
}
#endif /* nsWyciwygChannel_h___ */
|
