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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
|
// Copyright (c) 2009 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROME_FRAME_PLUGIN_URL_REQUEST_H_
#define CHROME_FRAME_PLUGIN_URL_REQUEST_H_
#include <string>
#include <vector>
#include "base/ref_counted.h"
#include "base/scoped_comptr_win.h"
#include "base/time.h"
#include "chrome_frame/chrome_frame_delegate.h"
#include "chrome_frame/urlmon_upload_data_stream.h"
#include "ipc/ipc_message.h"
#include "net/base/upload_data.h"
#include "net/url_request/url_request_status.h"
#include "webkit/glue/resource_type.h"
class PluginUrlRequest;
class PluginUrlRequestDelegate;
class PluginUrlRequestManager;
class DECLSPEC_NOVTABLE PluginUrlRequestDelegate { // NOLINT
public:
virtual void OnResponseStarted(int request_id, const char* mime_type,
const char* headers, int size, base::Time last_modified,
const std::string& redirect_url, int redirect_status) = 0;
virtual void OnReadComplete(int request_id, const std::string& data) = 0;
virtual void OnResponseEnd(int request_id,
const URLRequestStatus& status) = 0;
virtual void AddPrivacyDataForUrl(const std::string& url,
const std::string& policy_ref,
int32 flags) {}
virtual void OnCookiesRetrieved(bool success, const GURL& url,
const std::string& cookie_string,
int cookie_id) = 0;
protected:
PluginUrlRequestDelegate() {}
~PluginUrlRequestDelegate() {}
};
class DECLSPEC_NOVTABLE PluginUrlRequestManager { // NOLINT
public:
PluginUrlRequestManager() : delegate_(NULL), enable_frame_busting_(true) {}
virtual ~PluginUrlRequestManager() {}
void set_frame_busting(bool enable) {
enable_frame_busting_ = enable;
}
virtual void set_delegate(PluginUrlRequestDelegate* delegate) {
delegate_ = delegate;
}
enum ThreadSafeFlags {
NOT_THREADSAFE = 0x00,
START_REQUEST_THREADSAFE = 0x01,
STOP_REQUEST_THREADSAFE = 0x02,
READ_REQUEST_THREADSAFE = 0x04,
DOWNLOAD_REQUEST_THREADSAFE = 0x08,
COOKIE_REQUEST_THREADSAFE = 0x10
};
virtual ThreadSafeFlags GetThreadSafeFlags() = 0;
// These are called directly from Automation Client when network related
// automation messages are received from Chrome.
// Strip 'tab' handle and forward to the virtual methods implemented by
// derived classes.
void StartUrlRequest(int tab, int request_id,
const IPC::AutomationURLRequest& request_info) {
StartRequest(request_id, request_info);
}
void ReadUrlRequest(int tab, int request_id, int bytes_to_read) {
ReadRequest(request_id, bytes_to_read);
}
void EndUrlRequest(int tab, int request_id, const URLRequestStatus& s) {
EndRequest(request_id);
}
void DownloadUrlRequestInHost(int tab, int request_id) {
DownloadRequestInHost(request_id);
}
void StopAllRequests() {
StopAll();
}
void GetCookiesFromHost(int tab_handle, const GURL& url, int cookie_id) {
GetCookiesForUrl(url, cookie_id);
}
void SetCookiesInHost(int tab_handle, const GURL& url,
const std::string& cookie) {
SetCookiesForUrl(url, cookie);
}
protected:
PluginUrlRequestDelegate* delegate_;
bool enable_frame_busting_;
private:
virtual void StartRequest(int request_id,
const IPC::AutomationURLRequest& request_info) = 0;
virtual void ReadRequest(int request_id, int bytes_to_read) = 0;
virtual void EndRequest(int request_id) = 0;
virtual void DownloadRequestInHost(int request_id) = 0;
virtual void StopAll() = 0;
virtual void GetCookiesForUrl(const GURL& url, int cookie_id) = 0;
virtual void SetCookiesForUrl(const GURL& url, const std::string& cookie) = 0;
};
// Used as base class. Holds Url request properties (url, method, referrer..)
class PluginUrlRequest {
public:
PluginUrlRequest();
~PluginUrlRequest();
bool Initialize(PluginUrlRequestDelegate* delegate,
int remote_request_id, const std::string& url, const std::string& method,
const std::string& referrer, const std::string& extra_headers,
net::UploadData* upload_data, ResourceType::Type resource_type,
bool enable_frame_busting_);
// Accessors.
int id() const {
return remote_request_id_;
}
const std::string& url() const {
return url_;
}
const std::string& method() const {
return method_;
}
const std::string& referrer() const {
return referrer_;
}
const std::string& extra_headers() const {
return extra_headers_;
}
uint64 post_data_len() const {
return post_data_len_;
}
protected:
HRESULT get_upload_data(IStream** ret) {
DCHECK(ret);
if (!upload_data_.get())
return S_FALSE;
*ret = upload_data_.get();
(*ret)->AddRef();
return S_OK;
}
void set_url(const std::string& url) {
url_ = url;
}
void ClearPostData() {
upload_data_.Release();
post_data_len_ = 0;
}
void SendData();
bool enable_frame_busting_;
PluginUrlRequestDelegate* delegate_;
int remote_request_id_;
uint64 post_data_len_;
std::string url_;
std::string method_;
std::string referrer_;
std::string extra_headers_;
ResourceType::Type resource_type_;
ScopedComPtr<IStream> upload_data_;
};
#endif // CHROME_FRAME_PLUGIN_URL_REQUEST_H_
|
