| 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/browser_thread.h" | 13 #include "content/browser/browser_thread.h" |
| 14 #include "content/browser/child_process_security_policy.h" | 14 #include "content/browser/child_process_security_policy.h" |
| 15 #include "content/browser/download/download_id.h" | 15 #include "content/browser/download/download_id.h" |
| 16 #include "content/browser/download/download_id_factory.h" |
| 16 #include "content/browser/mock_resource_context.h" | 17 #include "content/browser/mock_resource_context.h" |
| 17 #include "content/browser/renderer_host/dummy_resource_handler.h" | 18 #include "content/browser/renderer_host/dummy_resource_handler.h" |
| 18 #include "content/browser/renderer_host/global_request_id.h" | 19 #include "content/browser/renderer_host/global_request_id.h" |
| 19 #include "content/browser/renderer_host/resource_dispatcher_host.h" | 20 #include "content/browser/renderer_host/resource_dispatcher_host.h" |
| 20 #include "content/browser/renderer_host/resource_dispatcher_host_request_info.h" | 21 #include "content/browser/renderer_host/resource_dispatcher_host_request_info.h" |
| 21 #include "content/browser/renderer_host/resource_handler.h" | 22 #include "content/browser/renderer_host/resource_handler.h" |
| 22 #include "content/browser/renderer_host/resource_message_filter.h" | 23 #include "content/browser/renderer_host/resource_message_filter.h" |
| 23 #include "content/common/resource_messages.h" | 24 #include "content/common/resource_messages.h" |
| 24 #include "content/common/resource_response.h" | 25 #include "content/common/resource_response.h" |
| 25 #include "content/common/view_messages.h" | 26 #include "content/common/view_messages.h" |
| (...skipping 1102 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1128 | 1129 |
| 1129 void* iter = NULL; | 1130 void* iter = NULL; |
| 1130 EXPECT_TRUE(IPC::ReadParam(&msgs[0][0], &iter, &request_id)); | 1131 EXPECT_TRUE(IPC::ReadParam(&msgs[0][0], &iter, &request_id)); |
| 1131 EXPECT_TRUE(IPC::ReadParam(&msgs[0][0], &iter, &status)); | 1132 EXPECT_TRUE(IPC::ReadParam(&msgs[0][0], &iter, &status)); |
| 1132 | 1133 |
| 1133 EXPECT_EQ(1, request_id); | 1134 EXPECT_EQ(1, request_id); |
| 1134 EXPECT_EQ(net::URLRequestStatus::CANCELED, status.status()); | 1135 EXPECT_EQ(net::URLRequestStatus::CANCELED, status.status()); |
| 1135 EXPECT_EQ(net::ERR_FILE_NOT_FOUND, status.error()); | 1136 EXPECT_EQ(net::ERR_FILE_NOT_FOUND, status.error()); |
| 1136 } | 1137 } |
| 1137 | 1138 |
| 1138 namespace { | |
| 1139 DownloadId MockNextDownloadId() { | |
| 1140 return DownloadId(reinterpret_cast<DownloadManager*>(0xFFFFFFFF), 0); | |
| 1141 } | |
| 1142 } | |
| 1143 | |
| 1144 // Test for http://crbug.com/76202 . We don't want to destroy a | 1139 // Test for http://crbug.com/76202 . We don't want to destroy a |
| 1145 // download request prematurely when processing a cancellation from | 1140 // download request prematurely when processing a cancellation from |
| 1146 // the renderer. | 1141 // the renderer. |
| 1147 TEST_F(ResourceDispatcherHostTest, IgnoreCancelForDownloads) { | 1142 TEST_F(ResourceDispatcherHostTest, IgnoreCancelForDownloads) { |
| 1148 EXPECT_EQ(0, host_.pending_requests()); | 1143 EXPECT_EQ(0, host_.pending_requests()); |
| 1149 | 1144 |
| 1150 int render_view_id = 0; | 1145 int render_view_id = 0; |
| 1151 int request_id = 1; | 1146 int request_id = 1; |
| 1152 | 1147 |
| 1153 std::string response("HTTP\n" | 1148 std::string response("HTTP\n" |
| 1154 "Content-disposition: attachment; filename=foo\n\n"); | 1149 "Content-disposition: attachment; filename=foo\n\n"); |
| 1155 std::string raw_headers(net::HttpUtil::AssembleRawHeaders(response.data(), | 1150 std::string raw_headers(net::HttpUtil::AssembleRawHeaders(response.data(), |
| 1156 response.size())); | 1151 response.size())); |
| 1157 std::string response_data("01234567890123456789\x01foobar"); | 1152 std::string response_data("01234567890123456789\x01foobar"); |
| 1158 | 1153 |
| 1159 SetResponse(raw_headers, response_data); | 1154 SetResponse(raw_headers, response_data); |
| 1160 SetResourceType(ResourceType::MAIN_FRAME); | 1155 SetResourceType(ResourceType::MAIN_FRAME); |
| 1161 HandleScheme("http"); | 1156 HandleScheme("http"); |
| 1162 | 1157 |
| 1163 MakeTestRequest(render_view_id, request_id, GURL("http://example.com/blah")); | 1158 MakeTestRequest(render_view_id, request_id, GURL("http://example.com/blah")); |
| 1164 content::MockResourceContext::GetInstance()->set_next_download_id_thunk( | 1159 scoped_refptr<DownloadIdFactory> id_factory( |
| 1165 base::Bind(&MockNextDownloadId)); | 1160 new DownloadIdFactory("valid DownloadId::Domain")); |
| 1161 content::MockResourceContext::GetInstance()->set_download_id_factory( |
| 1162 id_factory); |
| 1166 // Return some data so that the request is identified as a download | 1163 // Return some data so that the request is identified as a download |
| 1167 // and the proper resource handlers are created. | 1164 // and the proper resource handlers are created. |
| 1168 EXPECT_TRUE(net::URLRequestTestJob::ProcessOnePendingMessage()); | 1165 EXPECT_TRUE(net::URLRequestTestJob::ProcessOnePendingMessage()); |
| 1169 | 1166 |
| 1170 // And now simulate a cancellation coming from the renderer. | 1167 // And now simulate a cancellation coming from the renderer. |
| 1171 ResourceHostMsg_CancelRequest msg(filter_->child_id(), request_id); | 1168 ResourceHostMsg_CancelRequest msg(filter_->child_id(), request_id); |
| 1172 bool msg_was_ok; | 1169 bool msg_was_ok; |
| 1173 host_.OnMessageReceived(msg, filter_.get(), &msg_was_ok); | 1170 host_.OnMessageReceived(msg, filter_.get(), &msg_was_ok); |
| 1174 | 1171 |
| 1175 // Since the request had already started processing as a download, | 1172 // 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"); | 1188 "Content-disposition: attachment; filename=foo\n\n"); |
| 1192 std::string raw_headers(net::HttpUtil::AssembleRawHeaders(response.data(), | 1189 std::string raw_headers(net::HttpUtil::AssembleRawHeaders(response.data(), |
| 1193 response.size())); | 1190 response.size())); |
| 1194 std::string response_data("01234567890123456789\x01foobar"); | 1191 std::string response_data("01234567890123456789\x01foobar"); |
| 1195 | 1192 |
| 1196 SetResponse(raw_headers, response_data); | 1193 SetResponse(raw_headers, response_data); |
| 1197 SetResourceType(ResourceType::MAIN_FRAME); | 1194 SetResourceType(ResourceType::MAIN_FRAME); |
| 1198 HandleScheme("http"); | 1195 HandleScheme("http"); |
| 1199 | 1196 |
| 1200 MakeTestRequest(render_view_id, request_id, GURL("http://example.com/blah")); | 1197 MakeTestRequest(render_view_id, request_id, GURL("http://example.com/blah")); |
| 1201 content::MockResourceContext::GetInstance()->set_next_download_id_thunk( | 1198 scoped_refptr<DownloadIdFactory> id_factory( |
| 1202 base::Bind(&MockNextDownloadId)); | 1199 new DownloadIdFactory("valid DownloadId::Domain")); |
| 1200 content::MockResourceContext::GetInstance()->set_download_id_factory( |
| 1201 id_factory); |
| 1203 // Return some data so that the request is identified as a download | 1202 // Return some data so that the request is identified as a download |
| 1204 // and the proper resource handlers are created. | 1203 // and the proper resource handlers are created. |
| 1205 EXPECT_TRUE(net::URLRequestTestJob::ProcessOnePendingMessage()); | 1204 EXPECT_TRUE(net::URLRequestTestJob::ProcessOnePendingMessage()); |
| 1206 | 1205 |
| 1207 // And now simulate a cancellation coming from the renderer. | 1206 // And now simulate a cancellation coming from the renderer. |
| 1208 ResourceHostMsg_CancelRequest msg(filter_->child_id(), request_id); | 1207 ResourceHostMsg_CancelRequest msg(filter_->child_id(), request_id); |
| 1209 bool msg_was_ok; | 1208 bool msg_was_ok; |
| 1210 host_.OnMessageReceived(msg, filter_.get(), &msg_was_ok); | 1209 host_.OnMessageReceived(msg, filter_.get(), &msg_was_ok); |
| 1211 | 1210 |
| 1212 // Since the request had already started processing as a download, | 1211 // Since the request had already started processing as a download, |
| 1213 // the cancellation above should have been ignored and the request | 1212 // the cancellation above should have been ignored and the request |
| 1214 // should still be alive. | 1213 // should still be alive. |
| 1215 EXPECT_EQ(1, host_.pending_requests()); | 1214 EXPECT_EQ(1, host_.pending_requests()); |
| 1216 | 1215 |
| 1217 // Cancelling by other methods shouldn't work either. | 1216 // Cancelling by other methods shouldn't work either. |
| 1218 host_.CancelRequestsForProcess(render_view_id); | 1217 host_.CancelRequestsForProcess(render_view_id); |
| 1219 EXPECT_EQ(1, host_.pending_requests()); | 1218 EXPECT_EQ(1, host_.pending_requests()); |
| 1220 | 1219 |
| 1221 // Cancelling by context should work. | 1220 // Cancelling by context should work. |
| 1222 host_.CancelRequestsForContext(&filter_->resource_context()); | 1221 host_.CancelRequestsForContext(&filter_->resource_context()); |
| 1223 EXPECT_EQ(0, host_.pending_requests()); | 1222 EXPECT_EQ(0, host_.pending_requests()); |
| 1224 } | 1223 } |
| OLD | NEW |