| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "content/browser/renderer_host/resource_dispatcher_host.h" | 5 #include "content/browser/renderer_host/resource_dispatcher_host.h" |
| 6 | 6 |
| 7 #include <vector> | 7 #include <vector> |
| 8 | 8 |
| 9 #include "base/bind.h" | 9 #include "base/bind.h" |
| 10 #include "base/file_path.h" | 10 #include "base/file_path.h" |
| 11 #include "base/message_loop.h" | 11 #include "base/message_loop.h" |
| 12 #include "base/process_util.h" | 12 #include "base/process_util.h" |
| 13 #include "content/browser/child_process_security_policy.h" | 13 #include "content/browser/child_process_security_policy.h" |
| 14 #include "content/browser/download/download_id.h" | 14 #include "content/browser/download/download_id.h" |
| 15 #include "content/browser/download/download_id_factory.h" |
| 15 #include "content/browser/mock_resource_context.h" | 16 #include "content/browser/mock_resource_context.h" |
| 16 #include "content/browser/renderer_host/dummy_resource_handler.h" | 17 #include "content/browser/renderer_host/dummy_resource_handler.h" |
| 17 #include "content/browser/renderer_host/global_request_id.h" | 18 #include "content/browser/renderer_host/global_request_id.h" |
| 18 #include "content/browser/renderer_host/resource_dispatcher_host.h" | 19 #include "content/browser/renderer_host/resource_dispatcher_host.h" |
| 19 #include "content/browser/renderer_host/resource_dispatcher_host_request_info.h" | 20 #include "content/browser/renderer_host/resource_dispatcher_host_request_info.h" |
| 20 #include "content/browser/renderer_host/resource_handler.h" | 21 #include "content/browser/renderer_host/resource_handler.h" |
| 21 #include "content/browser/renderer_host/resource_message_filter.h" | 22 #include "content/browser/renderer_host/resource_message_filter.h" |
| 22 #include "content/common/resource_messages.h" | 23 #include "content/common/resource_messages.h" |
| 23 #include "content/common/resource_response.h" | 24 #include "content/common/resource_response.h" |
| 24 #include "content/common/view_messages.h" | 25 #include "content/common/view_messages.h" |
| (...skipping 1129 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1154 "Content-disposition: attachment; filename=foo\n\n"); | 1155 "Content-disposition: attachment; filename=foo\n\n"); |
| 1155 std::string raw_headers(net::HttpUtil::AssembleRawHeaders(response.data(), | 1156 std::string raw_headers(net::HttpUtil::AssembleRawHeaders(response.data(), |
| 1156 response.size())); | 1157 response.size())); |
| 1157 std::string response_data("01234567890123456789\x01foobar"); | 1158 std::string response_data("01234567890123456789\x01foobar"); |
| 1158 | 1159 |
| 1159 SetResponse(raw_headers, response_data); | 1160 SetResponse(raw_headers, response_data); |
| 1160 SetResourceType(ResourceType::MAIN_FRAME); | 1161 SetResourceType(ResourceType::MAIN_FRAME); |
| 1161 HandleScheme("http"); | 1162 HandleScheme("http"); |
| 1162 | 1163 |
| 1163 MakeTestRequest(render_view_id, request_id, GURL("http://example.com/blah")); | 1164 MakeTestRequest(render_view_id, request_id, GURL("http://example.com/blah")); |
| 1164 content::MockResourceContext::GetInstance()->set_next_download_id_thunk( | 1165 scoped_refptr<DownloadIdFactory> id_factory( |
| 1165 base::Bind(&MockNextDownloadId)); | 1166 new DownloadIdFactory("valid DownloadId::Domain")); |
| 1167 content::MockResourceContext::GetInstance()->set_download_id_factory( |
| 1168 id_factory); |
| 1166 // Return some data so that the request is identified as a download | 1169 // Return some data so that the request is identified as a download |
| 1167 // and the proper resource handlers are created. | 1170 // and the proper resource handlers are created. |
| 1168 EXPECT_TRUE(net::URLRequestTestJob::ProcessOnePendingMessage()); | 1171 EXPECT_TRUE(net::URLRequestTestJob::ProcessOnePendingMessage()); |
| 1169 | 1172 |
| 1170 // And now simulate a cancellation coming from the renderer. | 1173 // And now simulate a cancellation coming from the renderer. |
| 1171 ResourceHostMsg_CancelRequest msg(filter_->child_id(), request_id); | 1174 ResourceHostMsg_CancelRequest msg(filter_->child_id(), request_id); |
| 1172 bool msg_was_ok; | 1175 bool msg_was_ok; |
| 1173 host_.OnMessageReceived(msg, filter_.get(), &msg_was_ok); | 1176 host_.OnMessageReceived(msg, filter_.get(), &msg_was_ok); |
| 1174 | 1177 |
| 1175 // Since the request had already started processing as a download, | 1178 // Since the request had already started processing as a download, |
| (...skipping 15 matching lines...) Expand all Loading... |
| 1191 "Content-disposition: attachment; filename=foo\n\n"); | 1194 "Content-disposition: attachment; filename=foo\n\n"); |
| 1192 std::string raw_headers(net::HttpUtil::AssembleRawHeaders(response.data(), | 1195 std::string raw_headers(net::HttpUtil::AssembleRawHeaders(response.data(), |
| 1193 response.size())); | 1196 response.size())); |
| 1194 std::string response_data("01234567890123456789\x01foobar"); | 1197 std::string response_data("01234567890123456789\x01foobar"); |
| 1195 | 1198 |
| 1196 SetResponse(raw_headers, response_data); | 1199 SetResponse(raw_headers, response_data); |
| 1197 SetResourceType(ResourceType::MAIN_FRAME); | 1200 SetResourceType(ResourceType::MAIN_FRAME); |
| 1198 HandleScheme("http"); | 1201 HandleScheme("http"); |
| 1199 | 1202 |
| 1200 MakeTestRequest(render_view_id, request_id, GURL("http://example.com/blah")); | 1203 MakeTestRequest(render_view_id, request_id, GURL("http://example.com/blah")); |
| 1201 content::MockResourceContext::GetInstance()->set_next_download_id_thunk( | 1204 scoped_refptr<DownloadIdFactory> id_factory( |
| 1202 base::Bind(&MockNextDownloadId)); | 1205 new DownloadIdFactory("valid DownloadId::Domain")); |
| 1206 content::MockResourceContext::GetInstance()->set_download_id_factory( |
| 1207 id_factory); |
| 1203 // Return some data so that the request is identified as a download | 1208 // Return some data so that the request is identified as a download |
| 1204 // and the proper resource handlers are created. | 1209 // and the proper resource handlers are created. |
| 1205 EXPECT_TRUE(net::URLRequestTestJob::ProcessOnePendingMessage()); | 1210 EXPECT_TRUE(net::URLRequestTestJob::ProcessOnePendingMessage()); |
| 1206 | 1211 |
| 1207 // And now simulate a cancellation coming from the renderer. | 1212 // And now simulate a cancellation coming from the renderer. |
| 1208 ResourceHostMsg_CancelRequest msg(filter_->child_id(), request_id); | 1213 ResourceHostMsg_CancelRequest msg(filter_->child_id(), request_id); |
| 1209 bool msg_was_ok; | 1214 bool msg_was_ok; |
| 1210 host_.OnMessageReceived(msg, filter_.get(), &msg_was_ok); | 1215 host_.OnMessageReceived(msg, filter_.get(), &msg_was_ok); |
| 1211 | 1216 |
| 1212 // Since the request had already started processing as a download, | 1217 // Since the request had already started processing as a download, |
| 1213 // the cancellation above should have been ignored and the request | 1218 // the cancellation above should have been ignored and the request |
| 1214 // should still be alive. | 1219 // should still be alive. |
| 1215 EXPECT_EQ(1, host_.pending_requests()); | 1220 EXPECT_EQ(1, host_.pending_requests()); |
| 1216 | 1221 |
| 1217 // Cancelling by other methods shouldn't work either. | 1222 // Cancelling by other methods shouldn't work either. |
| 1218 host_.CancelRequestsForProcess(render_view_id); | 1223 host_.CancelRequestsForProcess(render_view_id); |
| 1219 EXPECT_EQ(1, host_.pending_requests()); | 1224 EXPECT_EQ(1, host_.pending_requests()); |
| 1220 | 1225 |
| 1221 // Cancelling by context should work. | 1226 // Cancelling by context should work. |
| 1222 host_.CancelRequestsForContext(&filter_->resource_context()); | 1227 host_.CancelRequestsForContext(&filter_->resource_context()); |
| 1223 EXPECT_EQ(0, host_.pending_requests()); | 1228 EXPECT_EQ(0, host_.pending_requests()); |
| 1224 } | 1229 } |
| OLD | NEW |