1c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// Copyright (c) 2010 The Chromium Authors. All rights reserved. 2c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// Use of this source code is governed by a BSD-style license that can be 3c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// found in the LICENSE file. 4c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 5c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "net/http/http_request_headers.h" 6c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 7c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "base/logging.h" 83345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick#include "base/string_split.h" 9c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "base/string_util.h" 10c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "net/http/http_util.h" 11c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 12c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochnamespace net { 13c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 14c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochconst char HttpRequestHeaders::kGetMethod[] = "GET"; 15c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochconst char HttpRequestHeaders::kAcceptCharset[] = "Accept-Charset"; 16c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochconst char HttpRequestHeaders::kAcceptEncoding[] = "Accept-Encoding"; 17c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochconst char HttpRequestHeaders::kAcceptLanguage[] = "Accept-Language"; 18c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochconst char HttpRequestHeaders::kCacheControl[] = "Cache-Control"; 19c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochconst char HttpRequestHeaders::kConnection[] = "Connection"; 20c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochconst char HttpRequestHeaders::kContentLength[] = "Content-Length"; 21c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochconst char HttpRequestHeaders::kContentType[] = "Content-Type"; 22c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochconst char HttpRequestHeaders::kCookie[] = "Cookie"; 23c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochconst char HttpRequestHeaders::kHost[] = "Host"; 24c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochconst char HttpRequestHeaders::kIfModifiedSince[] = "If-Modified-Since"; 25c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochconst char HttpRequestHeaders::kIfNoneMatch[] = "If-None-Match"; 26c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochconst char HttpRequestHeaders::kIfRange[] = "If-Range"; 27c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochconst char HttpRequestHeaders::kOrigin[] = "Origin"; 28c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochconst char HttpRequestHeaders::kPragma[] = "Pragma"; 29c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochconst char HttpRequestHeaders::kProxyConnection[] = "Proxy-Connection"; 30c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochconst char HttpRequestHeaders::kRange[] = "Range"; 31c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochconst char HttpRequestHeaders::kReferer[] = "Referer"; 32c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochconst char HttpRequestHeaders::kUserAgent[] = "User-Agent"; 3372a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsenconst char HttpRequestHeaders::kTransferEncoding[] = "Transfer-Encoding"; 34c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 353345a6884c488ff3a535c2c9acdd33d74b37e311Iain MerrickHttpRequestHeaders::HeaderKeyValuePair::HeaderKeyValuePair() { 363345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick} 373345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick 383345a6884c488ff3a535c2c9acdd33d74b37e311Iain MerrickHttpRequestHeaders::HeaderKeyValuePair::HeaderKeyValuePair( 393345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick const base::StringPiece& key, const base::StringPiece& value) 403345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick : key(key.data(), key.size()), value(value.data(), value.size()) { 413345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick} 423345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick 433345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick 44c407dc5cd9bdc5668497f21b26b09d988ab439deBen MurdochHttpRequestHeaders::Iterator::Iterator(const HttpRequestHeaders& headers) 45c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch : started_(false), 46c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch curr_(headers.headers_.begin()), 47c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch end_(headers.headers_.end()) {} 48c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 49c407dc5cd9bdc5668497f21b26b09d988ab439deBen MurdochHttpRequestHeaders::Iterator::~Iterator() {} 50c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 51c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochbool HttpRequestHeaders::Iterator::GetNext() { 52c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (!started_) { 53c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch started_ = true; 54c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return curr_ != end_; 55c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 56c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 57c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (curr_ == end_) 58c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return false; 59c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 60c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch ++curr_; 61c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return curr_ != end_; 62c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 63c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 64c407dc5cd9bdc5668497f21b26b09d988ab439deBen MurdochHttpRequestHeaders::HttpRequestHeaders() {} 65c407dc5cd9bdc5668497f21b26b09d988ab439deBen MurdochHttpRequestHeaders::~HttpRequestHeaders() {} 66c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 67c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochbool HttpRequestHeaders::GetHeader(const base::StringPiece& key, 68c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch std::string* out) const { 69c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch HeaderVector::const_iterator it = FindHeader(key); 70c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (it == headers_.end()) 71c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return false; 72c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch out->assign(it->value); 73c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return true; 74c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 75c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 76c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochvoid HttpRequestHeaders::Clear() { 77c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch headers_.clear(); 78c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 79c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 80c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochvoid HttpRequestHeaders::SetHeader(const base::StringPiece& key, 81c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch const base::StringPiece& value) { 82c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch HeaderVector::iterator it = FindHeader(key); 83c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (it != headers_.end()) 84c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch it->value = value.as_string(); 85c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch else 86c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch headers_.push_back(HeaderKeyValuePair(key.as_string(), value.as_string())); 87c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 88c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 8972a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsenvoid HttpRequestHeaders::SetHeaderIfMissing(const base::StringPiece& key, 9072a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen const base::StringPiece& value) { 9172a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen HeaderVector::iterator it = FindHeader(key); 9272a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen if (it == headers_.end()) 9372a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen headers_.push_back(HeaderKeyValuePair(key.as_string(), value.as_string())); 9472a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen} 9572a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen 96c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochvoid HttpRequestHeaders::RemoveHeader(const base::StringPiece& key) { 97c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch HeaderVector::iterator it = FindHeader(key); 98c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (it != headers_.end()) 99c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch headers_.erase(it); 100c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 101c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 102c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochvoid HttpRequestHeaders::AddHeaderFromString( 103c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch const base::StringPiece& header_line) { 104c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch DCHECK_EQ(std::string::npos, header_line.find("\r\n")) 105c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch << "\"" << header_line << "\" contains CRLF."; 106c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 107c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch const std::string::size_type key_end_index = header_line.find(":"); 108c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (key_end_index == std::string::npos) { 109c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch LOG(DFATAL) << "\"" << header_line << "\" is missing colon delimiter."; 110c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return; 111c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 112c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 113c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (key_end_index == 0) { 114c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch LOG(DFATAL) << "\"" << header_line << "\" is missing header key."; 115c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return; 116c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 117c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 118c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch const base::StringPiece header_key(header_line.data(), key_end_index); 119c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 120c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch const std::string::size_type value_index = key_end_index + 1; 121c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 122c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (value_index < header_line.size()) { 123c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch std::string header_value(header_line.data() + value_index, 124c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch header_line.size() - value_index); 125c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch std::string::const_iterator header_value_begin = 126c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch header_value.begin(); 127c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch std::string::const_iterator header_value_end = 128c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch header_value.end(); 129c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch HttpUtil::TrimLWS(&header_value_begin, &header_value_end); 130c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 131c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (header_value_begin == header_value_end) { 132c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // Value was all LWS. 133c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch SetHeader(header_key, ""); 134c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } else { 135c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch SetHeader(header_key, 136c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch base::StringPiece(&*header_value_begin, 137c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch header_value_end - header_value_begin)); 138c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 139c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } else if (value_index == header_line.size()) { 140c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch SetHeader(header_key, ""); 141c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } else { 142c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch NOTREACHED(); 143c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 144c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 145c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 146c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochvoid HttpRequestHeaders::AddHeadersFromString( 147c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch const base::StringPiece& headers) { 148c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // TODO(willchan): Consider adding more StringPiece support in string_util.h 149c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // to eliminate copies. 150c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch std::vector<std::string> header_line_vector; 1513345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick base::SplitStringUsingSubstr(headers.as_string(), "\r\n", 1523345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick &header_line_vector); 153c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch for (std::vector<std::string>::const_iterator it = header_line_vector.begin(); 154c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch it != header_line_vector.end(); ++it) { 155c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (!it->empty()) 156c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch AddHeaderFromString(*it); 157c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 158c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 159c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 160c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochvoid HttpRequestHeaders::MergeFrom(const HttpRequestHeaders& other) { 161c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch for (HeaderVector::const_iterator it = other.headers_.begin(); 162c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch it != other.headers_.end(); ++it ) { 163c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch SetHeader(it->key, it->value); 164c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 165c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 166c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 167c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochstd::string HttpRequestHeaders::ToString() const { 168c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch std::string output; 169c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch for (HeaderVector::const_iterator it = headers_.begin(); 170c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch it != headers_.end(); ++it) { 1714a5e2dc747d50c653511c68ccb2cfbfb740bd5a7Ben Murdoch if (!it->value.empty()) { 1724a5e2dc747d50c653511c68ccb2cfbfb740bd5a7Ben Murdoch base::StringAppendF(&output, "%s: %s\r\n", 1734a5e2dc747d50c653511c68ccb2cfbfb740bd5a7Ben Murdoch it->key.c_str(), it->value.c_str()); 1744a5e2dc747d50c653511c68ccb2cfbfb740bd5a7Ben Murdoch } else { 1754a5e2dc747d50c653511c68ccb2cfbfb740bd5a7Ben Murdoch base::StringAppendF(&output, "%s:\r\n", it->key.c_str()); 1764a5e2dc747d50c653511c68ccb2cfbfb740bd5a7Ben Murdoch } 177c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 178c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch output.append("\r\n"); 179c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return output; 180c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 181c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 182c407dc5cd9bdc5668497f21b26b09d988ab439deBen MurdochHttpRequestHeaders::HeaderVector::iterator 183c407dc5cd9bdc5668497f21b26b09d988ab439deBen MurdochHttpRequestHeaders::FindHeader(const base::StringPiece& key) { 184c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch for (HeaderVector::iterator it = headers_.begin(); 185c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch it != headers_.end(); ++it) { 186c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (key.length() == it->key.length() && 187c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch !base::strncasecmp(key.data(), it->key.data(), key.length())) 188c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return it; 189c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 190c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 191c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return headers_.end(); 192c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 193c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 194c407dc5cd9bdc5668497f21b26b09d988ab439deBen MurdochHttpRequestHeaders::HeaderVector::const_iterator 195c407dc5cd9bdc5668497f21b26b09d988ab439deBen MurdochHttpRequestHeaders::FindHeader(const base::StringPiece& key) const { 196c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch for (HeaderVector::const_iterator it = headers_.begin(); 197c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch it != headers_.end(); ++it) { 198c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (key.length() == it->key.length() && 199c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch !base::strncasecmp(key.data(), it->key.data(), key.length())) 200c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return it; 201c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 202c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 203c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return headers_.end(); 204c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 205c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 206c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} // namespace net 207