1// Copyright (c) 2010 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 "chrome/browser/net/url_request_mock_http_job.h" 6 7#include "base/file_util.h" 8#include "base/message_loop.h" 9#include "base/string_util.h" 10#include "base/threading/thread_restrictions.h" 11#include "base/utf_string_conversions.h" 12#include "chrome/common/url_constants.h" 13#include "net/base/net_util.h" 14#include "net/http/http_response_headers.h" 15#include "net/url_request/url_request_filter.h" 16 17static const char kMockHostname[] = "mock.http"; 18static const FilePath::CharType kMockHeaderFileSuffix[] = 19 FILE_PATH_LITERAL(".mock-http-headers"); 20 21FilePath URLRequestMockHTTPJob::base_path_; 22 23// static 24net::URLRequestJob* URLRequestMockHTTPJob::Factory(net::URLRequest* request, 25 const std::string& scheme) { 26 return new URLRequestMockHTTPJob(request, 27 GetOnDiskPath(base_path_, request, scheme)); 28} 29 30// static 31void URLRequestMockHTTPJob::AddUrlHandler(const FilePath& base_path) { 32 base_path_ = base_path; 33 34 // Add kMockHostname to net::URLRequestFilter. 35 net::URLRequestFilter* filter = net::URLRequestFilter::GetInstance(); 36 filter->AddHostnameHandler("http", kMockHostname, 37 URLRequestMockHTTPJob::Factory); 38} 39 40/* static */ 41GURL URLRequestMockHTTPJob::GetMockUrl(const FilePath& path) { 42 std::string url = "http://"; 43 url.append(kMockHostname); 44 url.append("/"); 45 std::string path_str = path.MaybeAsASCII(); 46 DCHECK(!path_str.empty()); // We only expect ASCII paths in tests. 47 url.append(path_str); 48 return GURL(url); 49} 50 51/* static */ 52GURL URLRequestMockHTTPJob::GetMockViewSourceUrl(const FilePath& path) { 53 std::string url = chrome::kViewSourceScheme; 54 url.append(":"); 55 url.append(GetMockUrl(path).spec()); 56 return GURL(url); 57} 58 59/* static */ 60FilePath URLRequestMockHTTPJob::GetOnDiskPath(const FilePath& base_path, 61 net::URLRequest* request, 62 const std::string& scheme) { 63 // Conceptually we just want to "return base_path + request->url().path()". 64 // But path in the request URL is in URL space (i.e. %-encoded spaces). 65 // So first we convert base FilePath to a URL, then append the URL 66 // path to that, and convert the final URL back to a FilePath. 67 GURL file_url(net::FilePathToFileURL(base_path)); 68 std::string url = file_url.spec() + request->url().path(); 69 FilePath file_path; 70 net::FileURLToFilePath(GURL(url), &file_path); 71 return file_path; 72} 73 74URLRequestMockHTTPJob::URLRequestMockHTTPJob(net::URLRequest* request, 75 const FilePath& file_path) 76 : net::URLRequestFileJob(request, file_path) { } 77 78// Public virtual version. 79void URLRequestMockHTTPJob::GetResponseInfo(net::HttpResponseInfo* info) { 80 // Forward to private const version. 81 GetResponseInfoConst(info); 82} 83 84bool URLRequestMockHTTPJob::IsRedirectResponse(GURL* location, 85 int* http_status_code) { 86 // Override the net::URLRequestFileJob implementation to invoke the default 87 // one based on HttpResponseInfo. 88 return net::URLRequestJob::IsRedirectResponse(location, http_status_code); 89} 90 91// Private const version. 92void URLRequestMockHTTPJob::GetResponseInfoConst( 93 net::HttpResponseInfo* info) const { 94 // We have to load our headers from disk, but we only use this class 95 // from tests, so allow these IO operations to happen on any thread. 96 base::ThreadRestrictions::ScopedAllowIO allow_io; 97 98 FilePath header_file = FilePath(file_path_.value() + kMockHeaderFileSuffix); 99 std::string raw_headers; 100 if (!file_util::ReadFileToString(header_file, &raw_headers)) 101 return; 102 103 // ParseRawHeaders expects \0 to end each header line. 104 ReplaceSubstringsAfterOffset(&raw_headers, 0, "\n", std::string("\0", 1)); 105 info->headers = new net::HttpResponseHeaders(raw_headers); 106} 107 108bool URLRequestMockHTTPJob::GetMimeType(std::string* mime_type) const { 109 net::HttpResponseInfo info; 110 GetResponseInfoConst(&info); 111 return info.headers && info.headers->GetMimeType(mime_type); 112} 113 114bool URLRequestMockHTTPJob::GetCharset(std::string* charset) { 115 net::HttpResponseInfo info; 116 GetResponseInfo(&info); 117 return info.headers && info.headers->GetCharset(charset); 118} 119