15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/test/net/url_request_slow_download_job.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/bind.h" 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/compiler_specific.h" 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h" 109ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch#include "base/message_loop/message_loop.h" 11868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/string_util.h" 12868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/stringprintf.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/browser/browser_thread.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/io_buffer.h" 152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "net/base/net_errors.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/http/http_response_headers.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/url_request.h" 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/url_request_filter.h" 197dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch#include "url/gurl.h" 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace content { 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char URLRequestSlowDownloadJob::kUnknownSizeUrl[] = 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "http://url.handled.by.slow.download/download-unknown-size"; 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char URLRequestSlowDownloadJob::kKnownSizeUrl[] = 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "http://url.handled.by.slow.download/download-known-size"; 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char URLRequestSlowDownloadJob::kFinishDownloadUrl[] = 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "http://url.handled.by.slow.download/download-finish"; 292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const char URLRequestSlowDownloadJob::kErrorDownloadUrl[] = 302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) "http://url.handled.by.slow.download/download-error"; 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int URLRequestSlowDownloadJob::kFirstDownloadSize = 1024 * 35; 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int URLRequestSlowDownloadJob::kSecondDownloadSize = 1024 * 10; 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// static 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)base::LazyInstance<URLRequestSlowDownloadJob::SlowJobsSet>::Leaky 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) URLRequestSlowDownloadJob::pending_requests_ = LAZY_INSTANCE_INITIALIZER; 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLRequestSlowDownloadJob::Start() { 40c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) base::MessageLoop::current()->PostTask( 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FROM_HERE, 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Bind(&URLRequestSlowDownloadJob::StartAsync, 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) weak_factory_.GetWeakPtr())); 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// static 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLRequestSlowDownloadJob::AddUrlHandler() { 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) net::URLRequestFilter* filter = net::URLRequestFilter::GetInstance(); 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) filter->AddUrlHandler(GURL(kUnknownSizeUrl), 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &URLRequestSlowDownloadJob::Factory); 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) filter->AddUrlHandler(GURL(kKnownSizeUrl), 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &URLRequestSlowDownloadJob::Factory); 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) filter->AddUrlHandler(GURL(kFinishDownloadUrl), 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &URLRequestSlowDownloadJob::Factory); 552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) filter->AddUrlHandler(GURL(kErrorDownloadUrl), 562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) &URLRequestSlowDownloadJob::Factory); 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// static 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)net::URLRequestJob* URLRequestSlowDownloadJob::Factory( 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) net::URLRequest* request, 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) net::NetworkDelegate* network_delegate, 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& scheme) { 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) URLRequestSlowDownloadJob* job = new URLRequestSlowDownloadJob( 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) request, network_delegate); 672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (request->url().spec() != kFinishDownloadUrl && 682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) request->url().spec() != kErrorDownloadUrl) 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pending_requests_.Get().insert(job); 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return job; 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// static 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)size_t URLRequestSlowDownloadJob::NumberOutstandingRequests() { 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return pending_requests_.Get().size(); 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// static 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLRequestSlowDownloadJob::FinishPendingRequests() { 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) typedef std::set<URLRequestSlowDownloadJob*> JobList; 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (JobList::iterator it = pending_requests_.Get().begin(); it != 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pending_requests_.Get().end(); ++it) { 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) (*it)->set_should_finish_download(); 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void URLRequestSlowDownloadJob::ErrorPendingRequests() { 902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) typedef std::set<URLRequestSlowDownloadJob*> JobList; 912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for (JobList::iterator it = pending_requests_.Get().begin(); it != 932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pending_requests_.Get().end(); ++it) { 942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) (*it)->set_should_error_download(); 952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)URLRequestSlowDownloadJob::URLRequestSlowDownloadJob( 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) net::URLRequest* request, net::NetworkDelegate* network_delegate) 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : net::URLRequestJob(request, network_delegate), 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bytes_already_sent_(0), 1022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) should_error_download_(false), 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) should_finish_download_(false), 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) buffer_size_(0), 105c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) weak_factory_(this) { 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLRequestSlowDownloadJob::StartAsync() { 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (LowerCaseEqualsASCII(kFinishDownloadUrl, request_->url().spec().c_str())) 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) URLRequestSlowDownloadJob::FinishPendingRequests(); 1112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (LowerCaseEqualsASCII(kErrorDownloadUrl, request_->url().spec().c_str())) 1122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) URLRequestSlowDownloadJob::ErrorPendingRequests(); 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NotifyHeadersComplete(); 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// ReadRawData and CheckDoneStatus together implement a state 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// machine. ReadRawData may be called arbitrarily by the network stack. 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// It responds by: 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// * If there are bytes remaining in the first chunk, they are 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// returned. 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// [No bytes remaining in first chunk. ] 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// * If should_finish_download_ is not set, it returns IO_PENDING, 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// and starts calling CheckDoneStatus on a regular timer. 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// [should_finish_download_ set.] 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// * If there are bytes remaining in the second chunk, they are filled. 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// * Otherwise, return *bytes_read = 0 to indicate end of request. 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// CheckDoneStatus is called on a regular basis, in the specific 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// case where we have transmitted all of the first chunk and none of the 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// second. If should_finish_download_ becomes set, it will "complete" 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// the ReadRawData call that spawned off the CheckDoneStatus() repeated call. 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// FillBufferHelper is a helper function that does the actual work of figuring 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// out where in the state machine we are and how we should fill the buffer. 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// It returns an enum indicating the state of the read. 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)URLRequestSlowDownloadJob::ReadStatus 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)URLRequestSlowDownloadJob::FillBufferHelper( 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) net::IOBuffer* buf, int buf_size, int* bytes_written) { 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (bytes_already_sent_ < kFirstDownloadSize) { 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int bytes_to_write = std::min(kFirstDownloadSize - bytes_already_sent_, 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) buf_size); 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (int i = 0; i < bytes_to_write; ++i) { 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) buf->data()[i] = '*'; 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *bytes_written = bytes_to_write; 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bytes_already_sent_ += bytes_to_write; 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return BUFFER_FILLED; 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!should_finish_download_) 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return REQUEST_BLOCKED; 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (bytes_already_sent_ < kFirstDownloadSize + kSecondDownloadSize) { 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int bytes_to_write = 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::min(kFirstDownloadSize + kSecondDownloadSize - bytes_already_sent_, 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) buf_size); 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (int i = 0; i < bytes_to_write; ++i) { 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) buf->data()[i] = '*'; 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *bytes_written = bytes_to_write; 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bytes_already_sent_ += bytes_to_write; 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return BUFFER_FILLED; 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return REQUEST_COMPLETE; 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool URLRequestSlowDownloadJob::ReadRawData(net::IOBuffer* buf, int buf_size, 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int* bytes_read) { 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (LowerCaseEqualsASCII(kFinishDownloadUrl, 1712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) request_->url().spec().c_str()) || 1722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) LowerCaseEqualsASCII(kErrorDownloadUrl, 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) request_->url().spec().c_str())) { 1742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) VLOG(10) << __FUNCTION__ << " called w/ kFinish/ErrorDownloadUrl."; 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *bytes_read = 0; 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) VLOG(10) << __FUNCTION__ << " called at position " 1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) << bytes_already_sent_ << " in the stream."; 1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReadStatus status = FillBufferHelper(buf, buf_size, bytes_read); 1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) switch (status) { 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) case BUFFER_FILLED: 1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) case REQUEST_BLOCKED: 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) buffer_ = buf; 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) buffer_size_ = buf_size; 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetStatus(net::URLRequestStatus(net::URLRequestStatus::IO_PENDING, 0)); 189c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) base::MessageLoop::current()->PostDelayedTask( 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FROM_HERE, 1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Bind(&URLRequestSlowDownloadJob::CheckDoneStatus, 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) weak_factory_.GetWeakPtr()), 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::TimeDelta::FromMilliseconds(100)); 1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) case REQUEST_COMPLETE: 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *bytes_read = 0; 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NOTREACHED(); 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLRequestSlowDownloadJob::CheckDoneStatus() { 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (should_finish_download_) { 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) VLOG(10) << __FUNCTION__ << " called w/ should_finish_download_ set."; 206868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) DCHECK(NULL != buffer_.get()); 2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int bytes_written = 0; 208868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) ReadStatus status = 209868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) FillBufferHelper(buffer_.get(), buffer_size_, &bytes_written); 2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_EQ(BUFFER_FILLED, status); 2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) buffer_ = NULL; // Release the reference. 2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetStatus(net::URLRequestStatus()); 2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NotifyReadComplete(bytes_written); 2142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } else if (should_error_download_) { 2152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) VLOG(10) << __FUNCTION__ << " called w/ should_finish_ownload_ set."; 2162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NotifyDone(net::URLRequestStatus( 2172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) net::URLRequestStatus::FAILED, net::ERR_CONNECTION_RESET)); 2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 219c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) base::MessageLoop::current()->PostDelayedTask( 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FROM_HERE, 2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Bind(&URLRequestSlowDownloadJob::CheckDoneStatus, 2225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) weak_factory_.GetWeakPtr()), 2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::TimeDelta::FromMilliseconds(100)); 2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Public virtual version. 2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLRequestSlowDownloadJob::GetResponseInfo(net::HttpResponseInfo* info) { 2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Forward to private const version. 2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) GetResponseInfoConst(info); 2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)URLRequestSlowDownloadJob::~URLRequestSlowDownloadJob() { 2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pending_requests_.Get().erase(this); 2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Private const version. 2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void URLRequestSlowDownloadJob::GetResponseInfoConst( 2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) net::HttpResponseInfo* info) const { 2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Send back mock headers. 2425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string raw_headers; 2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (LowerCaseEqualsASCII(kFinishDownloadUrl, 2442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) request_->url().spec().c_str()) || 2452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) LowerCaseEqualsASCII(kErrorDownloadUrl, 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) request_->url().spec().c_str())) { 2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) raw_headers.append( 2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "HTTP/1.1 200 OK\n" 2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Content-type: text/plain\n"); 2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) raw_headers.append( 2525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "HTTP/1.1 200 OK\n" 2535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Content-type: application/octet-stream\n" 2545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Cache-Control: max-age=0\n"); 2555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (LowerCaseEqualsASCII(kKnownSizeUrl, request_->url().spec().c_str())) { 2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) raw_headers.append(base::StringPrintf( 2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Content-Length: %d\n", 2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kFirstDownloadSize + kSecondDownloadSize)); 2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // ParseRawHeaders expects \0 to end each header line. 2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReplaceSubstringsAfterOffset(&raw_headers, 0, "\n", std::string("\0", 1)); 2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) info->headers = new net::HttpResponseHeaders(raw_headers); 2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool URLRequestSlowDownloadJob::GetMimeType(std::string* mime_type) const { 2695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) net::HttpResponseInfo info; 2705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) GetResponseInfoConst(&info); 271868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) return info.headers.get() && info.headers->GetMimeType(mime_type); 2725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace content 275