OLD | NEW |
| (Empty) |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "webkit/plugins/npapi/test/plugin_geturl_test.h" | |
6 | |
7 #include <stdio.h> | |
8 | |
9 #include "base/basictypes.h" | |
10 #include "base/file_util.h" | |
11 #include "base/strings/string_number_conversions.h" | |
12 #include "base/strings/string_util.h" | |
13 #include "base/strings/utf_string_conversions.h" | |
14 | |
15 // url for "self". The %22%22 is to make a statement for javascript to | |
16 // evaluate and return. | |
17 #define SELF_URL "javascript:window.location+\"\"" | |
18 | |
19 // The identifier for the self url stream. | |
20 #define SELF_URL_STREAM_ID 1 | |
21 | |
22 // The identifier for the fetched url stream. | |
23 #define FETCHED_URL_STREAM_ID 2 | |
24 | |
25 // url for testing GetURL with a bogus URL. | |
26 #define BOGUS_URL "bogoproto:///x:/asdf.xysdhffieasdf.asdhj/" | |
27 | |
28 // url for testing redirect notifications sent to plugins. | |
29 #define REDIRECT_SRC_URL \ | |
30 "http://mock.http/npapi/plugin_read_page_redirect_src.html" | |
31 | |
32 // The notification id for the redirect notification url. | |
33 #define REDIRECT_SRC_URL_NOTIFICATION_ID 4 | |
34 | |
35 // The identifier for the bogus url stream. | |
36 #define BOGUS_URL_STREAM_ID 3 | |
37 | |
38 // The maximum chunk size of stream data. | |
39 #define STREAM_CHUNK 197 | |
40 | |
41 namespace NPAPIClient { | |
42 | |
43 PluginGetURLTest::PluginGetURLTest(NPP id, NPNetscapeFuncs *host_functions) | |
44 : PluginTest(id, host_functions), | |
45 tests_started_(false), | |
46 tests_in_progress_(0), | |
47 test_file_(NULL), | |
48 expect_404_response_(false), | |
49 npn_evaluate_context_(false), | |
50 handle_url_redirects_(false), | |
51 received_url_redirect_notification_(false) { | |
52 } | |
53 | |
54 PluginGetURLTest::~PluginGetURLTest() {} | |
55 | |
56 NPError PluginGetURLTest::New(uint16 mode, int16 argc, const char* argn[], | |
57 const char* argv[], NPSavedData* saved) { | |
58 const char* page_not_found_url = GetArgValue("page_not_found_url", argc, | |
59 argn, argv); | |
60 if (page_not_found_url) { | |
61 page_not_found_url_ = page_not_found_url; | |
62 expect_404_response_ = true; | |
63 } | |
64 | |
65 const char* fail_write_url = GetArgValue("fail_write_url", argc, | |
66 argn, argv); | |
67 if (fail_write_url) { | |
68 fail_write_url_ = fail_write_url; | |
69 } | |
70 | |
71 const char* referrer_target_url = GetArgValue("ref_target", argc, | |
72 argn, argv); | |
73 if (referrer_target_url) { | |
74 referrer_target_url_ = referrer_target_url; | |
75 } | |
76 | |
77 if (!base::strcasecmp(GetArgValue("name", argc, argn, argv), | |
78 "geturlredirectnotify")) { | |
79 handle_url_redirects_ = true; | |
80 } | |
81 return PluginTest::New(mode, argc, argn, argv, saved); | |
82 } | |
83 | |
84 NPError PluginGetURLTest::SetWindow(NPWindow* pNPWindow) { | |
85 #if !defined(OS_MACOSX) | |
86 if (pNPWindow->window == NULL) | |
87 return NPERR_NO_ERROR; | |
88 #endif | |
89 | |
90 if (!tests_started_) { | |
91 tests_started_ = true; | |
92 | |
93 tests_in_progress_++; | |
94 | |
95 if (expect_404_response_) { | |
96 HostFunctions()->geturl(id(), page_not_found_url_.c_str(), NULL); | |
97 return NPERR_NO_ERROR; | |
98 } else if (!fail_write_url_.empty()) { | |
99 HostFunctions()->geturl(id(), fail_write_url_.c_str(), NULL); | |
100 return NPERR_NO_ERROR; | |
101 } else if (!referrer_target_url_.empty()) { | |
102 HostFunctions()->pushpopupsenabledstate(id(), true); | |
103 HostFunctions()->geturl(id(), referrer_target_url_.c_str(), "_blank"); | |
104 HostFunctions()->poppopupsenabledstate(id()); | |
105 return NPERR_NO_ERROR; | |
106 } else if (handle_url_redirects_) { | |
107 HostFunctions()->geturlnotify( | |
108 id(), REDIRECT_SRC_URL, NULL, | |
109 reinterpret_cast<void*>(REDIRECT_SRC_URL_NOTIFICATION_ID)); | |
110 return NPERR_NO_ERROR; | |
111 } | |
112 | |
113 std::string url = SELF_URL; | |
114 HostFunctions()->geturlnotify(id(), url.c_str(), NULL, | |
115 reinterpret_cast<void*>(SELF_URL_STREAM_ID)); | |
116 | |
117 tests_in_progress_++; | |
118 std::string bogus_url = BOGUS_URL; | |
119 HostFunctions()->geturlnotify(id(), bogus_url.c_str(), NULL, | |
120 reinterpret_cast<void*>(BOGUS_URL_STREAM_ID)); | |
121 } | |
122 return NPERR_NO_ERROR; | |
123 } | |
124 | |
125 NPError PluginGetURLTest::NewStream(NPMIMEType type, NPStream* stream, | |
126 NPBool seekable, uint16* stype) { | |
127 if (stream == NULL) { | |
128 SetError("NewStream got null stream"); | |
129 return NPERR_INVALID_PARAM; | |
130 } | |
131 | |
132 if (test_completed()) { | |
133 return PluginTest::NewStream(type, stream, seekable, stype); | |
134 } | |
135 | |
136 if (!referrer_target_url_.empty()) { | |
137 return NPERR_NO_ERROR; | |
138 } | |
139 | |
140 COMPILE_ASSERT(sizeof(unsigned long) <= sizeof(stream->notifyData), | |
141 cast_validity_check); | |
142 | |
143 if (expect_404_response_) { | |
144 NPObject *window_obj = NULL; | |
145 HostFunctions()->getvalue(id(), NPNVWindowNPObject, &window_obj); | |
146 if (!window_obj) { | |
147 SetError("Failed to get NPObject for plugin instance2"); | |
148 SignalTestCompleted(); | |
149 return NPERR_NO_ERROR; | |
150 } | |
151 | |
152 std::string script = "javascript:document.title=\"OK\""; | |
153 NPString script_string; | |
154 script_string.UTF8Characters = script.c_str(); | |
155 script_string.UTF8Length = static_cast<unsigned int>(script.length()); | |
156 NPVariant result_var; | |
157 | |
158 npn_evaluate_context_ = true; | |
159 HostFunctions()->evaluate(id(), window_obj, &script_string, &result_var); | |
160 npn_evaluate_context_ = false; | |
161 return NPERR_NO_ERROR; | |
162 } | |
163 | |
164 if (!fail_write_url_.empty()) { | |
165 return NPERR_NO_ERROR; | |
166 } | |
167 | |
168 | |
169 unsigned long stream_id = reinterpret_cast<unsigned long>( | |
170 stream->notifyData); | |
171 | |
172 switch (stream_id) { | |
173 case SELF_URL_STREAM_ID: | |
174 break; | |
175 case FETCHED_URL_STREAM_ID: | |
176 { | |
177 std::string filename = self_url_; | |
178 if (filename.find("file:///", 0) != 0) { | |
179 SetError("Test expects a file-url."); | |
180 break; | |
181 } | |
182 | |
183 // TODO(evanm): use the net:: functions to convert file:// URLs to | |
184 // on-disk file paths. But it probably doesn't actually matter in | |
185 // this test. | |
186 | |
187 #if defined(OS_WIN) | |
188 filename = filename.substr(8); // remove "file:///" | |
189 // Assume an ASCII path on Windows. | |
190 base::FilePath path = base::FilePath(ASCIIToWide(filename)); | |
191 #else | |
192 filename = filename.substr(7); // remove "file://" | |
193 base::FilePath path = base::FilePath(filename); | |
194 #endif | |
195 | |
196 test_file_ = file_util::OpenFile(path, "r"); | |
197 if (!test_file_) { | |
198 SetError("Could not open source file"); | |
199 } | |
200 } | |
201 break; | |
202 case BOGUS_URL_STREAM_ID: | |
203 SetError("Unexpected NewStream for BOGUS_URL"); | |
204 break; | |
205 case REDIRECT_SRC_URL_NOTIFICATION_ID: | |
206 SetError("Should not redirect to URL when plugin denied it."); | |
207 break; | |
208 default: | |
209 SetError("Unexpected NewStream callback"); | |
210 break; | |
211 } | |
212 return NPERR_NO_ERROR; | |
213 } | |
214 | |
215 int32 PluginGetURLTest::WriteReady(NPStream *stream) { | |
216 if (test_completed()) { | |
217 return PluginTest::WriteReady(stream); | |
218 } | |
219 | |
220 if (!referrer_target_url_.empty()) { | |
221 return STREAM_CHUNK; | |
222 } | |
223 | |
224 COMPILE_ASSERT(sizeof(unsigned long) <= sizeof(stream->notifyData), | |
225 cast_validity_check); | |
226 unsigned long stream_id = reinterpret_cast<unsigned long>( | |
227 stream->notifyData); | |
228 if (stream_id == BOGUS_URL_STREAM_ID) | |
229 SetError("Received WriteReady for BOGUS_URL"); | |
230 | |
231 return STREAM_CHUNK; | |
232 } | |
233 | |
234 int32 PluginGetURLTest::Write(NPStream *stream, int32 offset, int32 len, | |
235 void *buffer) { | |
236 if (test_completed()) { | |
237 return PluginTest::Write(stream, offset, len, buffer); | |
238 } | |
239 | |
240 if (!fail_write_url_.empty()) { | |
241 SignalTestCompleted(); | |
242 return -1; | |
243 } | |
244 | |
245 if (!referrer_target_url_.empty()) { | |
246 return len; | |
247 } | |
248 | |
249 if (stream == NULL) { | |
250 SetError("Write got null stream"); | |
251 return -1; | |
252 } | |
253 if (len < 0 || len > STREAM_CHUNK) { | |
254 SetError("Write got bogus stream chunk size"); | |
255 return -1; | |
256 } | |
257 | |
258 COMPILE_ASSERT(sizeof(unsigned long) <= sizeof(stream->notifyData), | |
259 cast_validity_check); | |
260 unsigned long stream_id = reinterpret_cast<unsigned long>( | |
261 stream->notifyData); | |
262 switch (stream_id) { | |
263 case SELF_URL_STREAM_ID: | |
264 self_url_.append(static_cast<char*>(buffer), len); | |
265 break; | |
266 case FETCHED_URL_STREAM_ID: | |
267 { | |
268 char read_buffer[STREAM_CHUNK]; | |
269 int32 bytes = | |
270 static_cast<int32>(fread(read_buffer, 1, len, test_file_)); | |
271 // Technically, fread could return fewer than len | |
272 // bytes. But this is not likely. | |
273 if (bytes != len) | |
274 SetError("Did not read correct bytelength from source file"); | |
275 if (memcmp(read_buffer, buffer, len)) | |
276 SetError("Content mismatch between data and source!"); | |
277 } | |
278 break; | |
279 case BOGUS_URL_STREAM_ID: | |
280 SetError("Unexpected write callback for BOGUS_URL"); | |
281 break; | |
282 default: | |
283 SetError("Unexpected write callback"); | |
284 break; | |
285 } | |
286 // Pretend that we took all the data. | |
287 return len; | |
288 } | |
289 | |
290 | |
291 NPError PluginGetURLTest::DestroyStream(NPStream *stream, NPError reason) { | |
292 if (test_completed()) { | |
293 return PluginTest::DestroyStream(stream, reason); | |
294 } | |
295 | |
296 if (stream == NULL) { | |
297 SetError("NewStream got null stream"); | |
298 return NPERR_INVALID_PARAM; | |
299 } | |
300 | |
301 COMPILE_ASSERT(sizeof(unsigned long) <= sizeof(stream->notifyData), | |
302 cast_validity_check); | |
303 | |
304 if (expect_404_response_) { | |
305 if (npn_evaluate_context_) { | |
306 SetError("Received destroyStream in the context of NPN_Evaluate."); | |
307 } | |
308 | |
309 SignalTestCompleted(); | |
310 return NPERR_NO_ERROR; | |
311 } | |
312 | |
313 if (!referrer_target_url_.empty()) { | |
314 return NPERR_NO_ERROR; | |
315 } | |
316 | |
317 unsigned long stream_id = | |
318 reinterpret_cast<unsigned long>(stream->notifyData); | |
319 switch (stream_id) { | |
320 case SELF_URL_STREAM_ID: | |
321 // don't care | |
322 break; | |
323 case FETCHED_URL_STREAM_ID: | |
324 { | |
325 char read_buffer[STREAM_CHUNK]; | |
326 size_t bytes = fread(read_buffer, 1, sizeof(read_buffer), test_file_); | |
327 if (bytes != 0) | |
328 SetError("Data and source mismatch on length"); | |
329 file_util::CloseFile(test_file_); | |
330 } | |
331 break; | |
332 default: | |
333 SetError("Unexpected NewStream callback"); | |
334 break; | |
335 } | |
336 return NPERR_NO_ERROR; | |
337 } | |
338 | |
339 void PluginGetURLTest::StreamAsFile(NPStream* stream, const char* fname) { | |
340 if (stream == NULL) { | |
341 SetError("NewStream got null stream"); | |
342 return; | |
343 } | |
344 | |
345 COMPILE_ASSERT(sizeof(unsigned long) <= sizeof(stream->notifyData), | |
346 cast_validity_check); | |
347 unsigned long stream_id = | |
348 reinterpret_cast<unsigned long>(stream->notifyData); | |
349 switch (stream_id) { | |
350 case SELF_URL_STREAM_ID: | |
351 // don't care | |
352 break; | |
353 default: | |
354 SetError("Unexpected NewStream callback"); | |
355 break; | |
356 } | |
357 } | |
358 | |
359 void PluginGetURLTest::URLNotify(const char* url, NPReason reason, void* data) { | |
360 if (!tests_in_progress_) { | |
361 SetError("URLNotify received after tests completed"); | |
362 return; | |
363 } | |
364 | |
365 if (!url) { | |
366 SetError("URLNotify received NULL url"); | |
367 return; | |
368 } | |
369 | |
370 COMPILE_ASSERT(sizeof(unsigned long) <= sizeof(data), cast_validity_check); | |
371 unsigned long stream_id = reinterpret_cast<unsigned long>(data); | |
372 switch (stream_id) { | |
373 case SELF_URL_STREAM_ID: | |
374 if (strcmp(url, SELF_URL) != 0) | |
375 SetError("URLNotify reported incorrect url for SELF_URL"); | |
376 | |
377 // We have our stream url. Go fetch it. | |
378 HostFunctions()->geturlnotify(id(), self_url_.c_str(), NULL, | |
379 reinterpret_cast<void*>(FETCHED_URL_STREAM_ID)); | |
380 break; | |
381 case FETCHED_URL_STREAM_ID: | |
382 if (!url || strcmp(url, self_url_.c_str()) != 0) | |
383 SetError("URLNotify reported incorrect url for FETCHED_URL"); | |
384 tests_in_progress_--; | |
385 break; | |
386 case BOGUS_URL_STREAM_ID: | |
387 if (reason != NPRES_NETWORK_ERR) { | |
388 std::string err = "BOGUS_URL received unexpected URLNotify status: "; | |
389 err.append(base::IntToString(reason)); | |
390 SetError(err); | |
391 } | |
392 tests_in_progress_--; | |
393 break; | |
394 case REDIRECT_SRC_URL_NOTIFICATION_ID: { | |
395 if (!received_url_redirect_notification_) { | |
396 SetError("Failed to receive URLRedirect notification"); | |
397 } | |
398 tests_in_progress_--; | |
399 break; | |
400 } | |
401 default: | |
402 SetError("Unexpected NewStream callback"); | |
403 break; | |
404 } | |
405 | |
406 if (tests_in_progress_ == 0) | |
407 SignalTestCompleted(); | |
408 } | |
409 | |
410 void PluginGetURLTest::URLRedirectNotify(const char* url, | |
411 int32_t status, | |
412 void* notify_data) { | |
413 if (!base::strcasecmp(url, "http://mock.http/npapi/plugin_read_page.html")) { | |
414 received_url_redirect_notification_ = true; | |
415 // Disallow redirect notification. | |
416 HostFunctions()->urlredirectresponse(id(), notify_data, false); | |
417 } | |
418 } | |
419 | |
420 } // namespace NPAPIClient | |
OLD | NEW |