Index: chrome_frame/html_utils.cc |
diff --git a/chrome_frame/html_utils.cc b/chrome_frame/html_utils.cc |
deleted file mode 100644 |
index 6a6bbea0784e1de84419829f6d3aaa674c190550..0000000000000000000000000000000000000000 |
--- a/chrome_frame/html_utils.cc |
+++ /dev/null |
@@ -1,475 +0,0 @@ |
-// Copyright (c) 2012 The Chromium Authors. All rights reserved. |
-// Use of this source code is governed by a BSD-style license that can be |
-// found in the LICENSE file. |
-// |
-#include "chrome_frame/html_utils.h" |
- |
-#include <atlbase.h> |
-#include <urlmon.h> |
- |
-#include "base/strings/string_tokenizer.h" |
-#include "base/strings/string_util.h" |
-#include "base/strings/stringprintf.h" |
-#include "chrome/common/chrome_version_info.h" |
-#include "chrome_frame/utils.h" |
-#include "net/base/net_util.h" |
-#include "webkit/common/user_agent/user_agent_util.h" |
- |
-const wchar_t kQuotes[] = L"\"'"; |
-const char kXFrameOptionsHeader[] = "X-Frame-Options"; |
-const char kXFrameOptionsValueAllowAll[] = "allowall"; |
- |
-HTMLScanner::StringRange::StringRange() { |
-} |
- |
-HTMLScanner::StringRange::StringRange(StrPos start, StrPos end) |
- : start_(start), end_(end) { |
-} |
- |
-bool HTMLScanner::StringRange::LowerCaseEqualsASCII(const char* other) const { |
- return ::LowerCaseEqualsASCII(start_, end_, other); |
-} |
- |
-bool HTMLScanner::StringRange::Equals(const wchar_t* other) const { |
- int ret = wcsncmp(&start_[0], other, end_ - start_); |
- if (ret == 0) |
- ret = (other[end_ - start_] == L'\0') ? 0 : -1; |
- return ret == 0; |
-} |
- |
-std::wstring HTMLScanner::StringRange::Copy() const { |
- return std::wstring(start_, end_); |
-} |
- |
-bool HTMLScanner::StringRange::GetTagName(std::wstring* tag_name) const { |
- if (*start_ != L'<') { |
- LOG(ERROR) << "Badly formatted tag found"; |
- return false; |
- } |
- |
- StrPos name_start = start_; |
- name_start++; |
- while (name_start < end_ && IsWhitespace(*name_start)) |
- name_start++; |
- |
- if (name_start >= end_) { |
- // We seem to have a degenerate tag (i.e. < >). Return false here. |
- return false; |
- } |
- |
- StrPos name_end = name_start + 1; |
- while (name_end < end_ && !IsWhitespace(*name_end)) |
- name_end++; |
- |
- if (name_end > end_) { |
- // This looks like an improperly formatted tab ('<foo'). Return false here. |
- return false; |
- } |
- |
- tag_name->assign(name_start, name_end); |
- return true; |
-} |
- |
- |
-bool HTMLScanner::StringRange::GetTagAttribute(const wchar_t* attribute_name, |
- StringRange* attribute_value) const { |
- if (NULL == attribute_name || NULL == attribute_value) { |
- NOTREACHED(); |
- return false; |
- } |
- |
- // Use this so we can use the convenience method LowerCaseEqualsASCII() |
- // from string_util.h. |
- std::string search_name_ascii(WideToASCII(attribute_name)); |
- |
- base::WStringTokenizer tokenizer(start_, end_, L" =/"); |
- tokenizer.set_options(base::WStringTokenizer::RETURN_DELIMS); |
- |
- // Set up the quote chars so that we get quoted attribute values as single |
- // tokens. |
- tokenizer.set_quote_chars(L"\"'"); |
- |
- const bool PARSE_STATE_NAME = true; |
- const bool PARSE_STATE_VALUE = false; |
- bool parse_state = PARSE_STATE_NAME; |
- |
- // Used to skip the first token, which is the tag name. |
- bool first_token_skipped = false; |
- |
- // This is set during a loop iteration in which an '=' sign was spotted. |
- // It is used to filter out degenerate tags such as: |
- // <meta foo==bar> |
- bool last_token_was_delim = false; |
- |
- // Set this if the attribute name has been found that we might then |
- // pick up the value in the next loop iteration. |
- bool attribute_name_found = false; |
- |
- while (tokenizer.GetNext()) { |
- // If we have a whitespace delimiter, just keep going. Cases of this should |
- // be reduced by the CollapseWhitespace call. If we have an '=' character, |
- // we update our state and reiterate. |
- if (tokenizer.token_is_delim()) { |
- if (*tokenizer.token_begin() == L'=') { |
- if (last_token_was_delim) { |
- // Looks like we have a badly formed tag, just stop parsing now. |
- return false; |
- } |
- parse_state = !parse_state; |
- last_token_was_delim = true; |
- } |
- continue; |
- } |
- |
- last_token_was_delim = false; |
- |
- // The first non-delimiter token is the tag name, which we don't want. |
- if (!first_token_skipped) { |
- first_token_skipped = true; |
- continue; |
- } |
- |
- if (PARSE_STATE_NAME == parse_state) { |
- // We have a tag name, check to see if it matches our target name: |
- if (::LowerCaseEqualsASCII(tokenizer.token_begin(), tokenizer.token_end(), |
- search_name_ascii.c_str())) { |
- attribute_name_found = true; |
- continue; |
- } |
- } else if (PARSE_STATE_VALUE == parse_state && attribute_name_found) { |
- attribute_value->start_ = tokenizer.token_begin(); |
- attribute_value->end_ = tokenizer.token_end(); |
- |
- // Unquote the attribute value if need be. |
- attribute_value->UnQuote(); |
- |
- return true; |
- } else if (PARSE_STATE_VALUE == parse_state) { |
- // If we haven't found the attribute name we want yet, ignore this token |
- // and go back to looking for our name. |
- parse_state = PARSE_STATE_NAME; |
- } |
- } |
- |
- return false; |
-} |
- |
-bool HTMLScanner::StringRange::UnQuote() { |
- if (start_ + 2 > end_) { |
- // String's too short to be quoted, bail. |
- return false; |
- } |
- |
- if ((*start_ == L'\'' && *(end_ - 1) == L'\'') || |
- (*start_ == L'"' && *(end_ - 1) == L'"')) { |
- start_ = start_ + 1; |
- end_ = end_ - 1; |
- return true; |
- } |
- |
- return false; |
-} |
- |
-HTMLScanner::HTMLScanner(const wchar_t* html_string) |
- : html_string_(CollapseWhitespace(html_string, true)), |
- quotes_(kQuotes) { |
-} |
- |
-void HTMLScanner::GetTagsByName(const wchar_t* name, StringRangeList* tag_list, |
- const wchar_t* stop_tag) { |
- DCHECK(NULL != name); |
- DCHECK(NULL != tag_list); |
- DCHECK(NULL != stop_tag); |
- |
- StringRange remaining_html(html_string_.begin(), html_string_.end()); |
- |
- std::wstring search_name(name); |
- TrimWhitespace(search_name, TRIM_ALL, &search_name); |
- |
- // Use this so we can use the convenience method LowerCaseEqualsASCII() |
- // from string_util.h. |
- std::string search_name_ascii(WideToASCII(search_name)); |
- std::string stop_tag_ascii(WideToASCII(stop_tag)); |
- |
- StringRange current_tag; |
- std::wstring current_name; |
- while (NextTag(&remaining_html, ¤t_tag)) { |
- if (current_tag.GetTagName(¤t_name)) { |
- if (LowerCaseEqualsASCII(current_name, search_name_ascii.c_str())) { |
- tag_list->push_back(current_tag); |
- } else if (LowerCaseEqualsASCII(current_name, stop_tag_ascii.c_str())) { |
- // We hit the stop tag so it's time to go home. |
- break; |
- } |
- } |
- } |
-} |
- |
-struct ScanState { |
- bool in_quote; |
- bool in_escape; |
- wchar_t quote_char; |
- ScanState() : in_quote(false), in_escape(false) {} |
-}; |
- |
-bool HTMLScanner::IsQuote(wchar_t c) { |
- return quotes_.find(c) != std::wstring::npos; |
-} |
- |
-bool HTMLScanner::IsHTMLCommentClose(const StringRange* html_string, |
- StrPos pos) { |
- if (pos < html_string->end_ && pos > html_string->start_ + 2 && |
- *pos == L'>') { |
- return *(pos-1) == L'-' && *(pos-2) == L'-'; |
- } |
- return false; |
-} |
- |
-bool HTMLScanner::IsIEConditionalCommentClose(const StringRange* html_string, |
- StrPos pos) { |
- if (pos < html_string->end_ && pos > html_string->start_ + 2 && |
- *pos == L'>') { |
- return *(pos-1) == L']'; |
- } |
- return false; |
-} |
- |
- |
-bool HTMLScanner::NextTag(StringRange* html_string, StringRange* tag) { |
- DCHECK(NULL != html_string); |
- DCHECK(NULL != tag); |
- |
- tag->start_ = html_string->start_; |
- while (tag->start_ < html_string->end_ && *tag->start_ != L'<') { |
- tag->start_++; |
- } |
- |
- // we went past the end of the string. |
- if (tag->start_ >= html_string->end_) { |
- return false; |
- } |
- |
- tag->end_ = tag->start_ + 1; |
- |
- // Get the tag name to see if we are in an HTML comment. If we are, then |
- // don't consider quotes. This should work for example: |
- // <!-- foo ' --> <meta foo='bar'> |
- std::wstring tag_name; |
- StringRange start_range(tag->start_, html_string->end_); |
- start_range.GetTagName(&tag_name); |
- if (StartsWith(tag_name, L"!--[if", true)) { |
- // This looks like the beginning of an IE conditional comment, scan until |
- // we hit the end which always looks like ']>'. For now we disregard the |
- // contents of the condition, and always assume true. |
- // TODO(robertshield): Optionally support the grammar defined by |
- // http://msdn.microsoft.com/en-us/library/ms537512(VS.85).aspx#syntax. |
- while (tag->end_ < html_string->end_ && |
- !IsIEConditionalCommentClose(html_string, tag->end_)) { |
- tag->end_++; |
- } |
- } else if (StartsWith(tag_name, L"!--", true)) { |
- // We're inside a comment tag which ends in '-->'. Keep going until we |
- // reach the end. |
- while (tag->end_ < html_string->end_ && |
- !IsHTMLCommentClose(html_string, tag->end_)) { |
- tag->end_++; |
- } |
- } else if (StartsWith(tag_name, L"![endif", true)) { |
- // We're inside the closing tag of an IE conditional comment which ends in |
- // either '-->' of ']>'. Keep going until we reach the end. |
- while (tag->end_ < html_string->end_ && |
- !IsIEConditionalCommentClose(html_string, tag->end_) && |
- !IsHTMLCommentClose(html_string, tag->end_)) { |
- tag->end_++; |
- } |
- } else { |
- // Properly handle quoted strings within non-comment tags by maintaining |
- // some state while scanning. Specifically, we have to maintain state on |
- // whether we are inside a string, what the string terminating character |
- // will be and whether we are inside an escape sequence. |
- ScanState state; |
- while (tag->end_ < html_string->end_) { |
- if (state.in_quote) { |
- if (state.in_escape) { |
- state.in_escape = false; |
- } else if (*tag->end_ == '\\') { |
- state.in_escape = true; |
- } else if (*tag->end_ == state.quote_char) { |
- state.in_quote = false; |
- } |
- } else { |
- state.in_quote = IsQuote(state.quote_char = *tag->end_); |
- } |
- |
- if (!state.in_quote && *tag->end_ == L'>') { |
- break; |
- } |
- tag->end_++; |
- } |
- } |
- |
- // We hit the end_ but found no matching tag closure. Consider this an |
- // incomplete tag and do not report it. |
- if (tag->end_ >= html_string->end_) |
- return false; |
- |
- // Modify html_string to point to just beyond the end_ of the current tag. |
- html_string->start_ = tag->end_ + 1; |
- |
- return true; |
-} |
- |
-namespace http_utils { |
- |
-const char kChromeFrameUserAgent[] = "chromeframe"; |
-static char g_cf_user_agent[100] = {0}; |
-static char g_chrome_user_agent[255] = {0}; |
- |
-const char* GetChromeFrameUserAgent() { |
- if (!g_cf_user_agent[0]) { |
- _pAtlModule->m_csStaticDataInitAndTypeInfo.Lock(); |
- if (!g_cf_user_agent[0]) { |
- uint32 high_version = 0, low_version = 0; |
- GetModuleVersion(reinterpret_cast<HMODULE>(&__ImageBase), &high_version, |
- &low_version); |
- wsprintfA(g_cf_user_agent, "%s/%i.%i.%i.%i", kChromeFrameUserAgent, |
- HIWORD(high_version), LOWORD(high_version), |
- HIWORD(low_version), LOWORD(low_version)); |
- } |
- _pAtlModule->m_csStaticDataInitAndTypeInfo.Unlock(); |
- } |
- return g_cf_user_agent; |
-} |
- |
-std::string AddChromeFrameToUserAgentValue(const std::string& value) { |
- if (value.empty()) { |
- return value; |
- } |
- |
- if (value.find(kChromeFrameUserAgent) != std::string::npos) { |
- // Our user agent has already been added. |
- return value; |
- } |
- |
- std::string ret(value); |
- size_t insert_position = ret.find(')'); |
- if (insert_position != std::string::npos) { |
- if (insert_position > 1 && isalnum(ret[insert_position - 1])) |
- ret.insert(insert_position++, ";"); |
- ret.insert(insert_position++, " "); |
- ret.insert(insert_position, GetChromeFrameUserAgent()); |
- } else { |
- ret += " "; |
- ret += GetChromeFrameUserAgent(); |
- } |
- |
- return ret; |
-} |
- |
-std::string RemoveChromeFrameFromUserAgentValue(const std::string& value) { |
- size_t cf_start = value.find(kChromeFrameUserAgent); |
- if (cf_start == std::string::npos) { |
- // The user agent is not present. |
- return value; |
- } |
- |
- size_t offset = 0; |
- // If we prepended a '; ' or a ' ' then remove that in the output. |
- if (cf_start > 1 && value[cf_start - 1] == ' ') |
- ++offset; |
- if (cf_start > 3 && |
- value[cf_start - 2] == ';' && |
- isalnum(value[cf_start - 3])) { |
- ++offset; |
- } |
- |
- std::string ret(value, 0, std::max(cf_start - offset, 0U)); |
- cf_start += strlen(kChromeFrameUserAgent); |
- while (cf_start < value.length() && |
- ((value[cf_start] >= '0' && value[cf_start] <= '9') || |
- value[cf_start] == '.' || |
- value[cf_start] == '/')) { |
- ++cf_start; |
- } |
- |
- if (cf_start < value.length()) |
- ret.append(value, cf_start, std::string::npos); |
- |
- return ret; |
-} |
- |
-std::string GetDefaultUserAgentHeaderWithCFTag() { |
- std::string ua(GetDefaultUserAgent()); |
- return "User-Agent: " + AddChromeFrameToUserAgentValue(ua); |
-} |
- |
-const char* GetChromeUserAgent() { |
- if (!g_chrome_user_agent[0]) { |
- _pAtlModule->m_csStaticDataInitAndTypeInfo.Lock(); |
- if (!g_chrome_user_agent[0]) { |
- std::string ua; |
- |
- chrome::VersionInfo version_info; |
- std::string product("Chrome/"); |
- product += version_info.is_valid() ? version_info.Version() |
- : "0.0.0.0"; |
- |
- ua = webkit_glue::BuildUserAgentFromProduct(product); |
- |
- DCHECK(ua.length() < arraysize(g_chrome_user_agent)); |
- lstrcpynA(g_chrome_user_agent, ua.c_str(), |
- arraysize(g_chrome_user_agent) - 1); |
- } |
- _pAtlModule->m_csStaticDataInitAndTypeInfo.Unlock(); |
- } |
- return g_chrome_user_agent; |
-} |
- |
-std::string GetDefaultUserAgent() { |
- std::string ret; |
- DWORD size = MAX_PATH; |
- HRESULT hr = E_OUTOFMEMORY; |
- for (int retries = 1; hr == E_OUTOFMEMORY && retries <= 10; ++retries) { |
- hr = ::ObtainUserAgentString(0, WriteInto(&ret, size + 1), &size); |
- if (hr == E_OUTOFMEMORY) { |
- size = MAX_PATH * retries; |
- } else if (SUCCEEDED(hr)) { |
- // Truncate the extra allocation. |
- DCHECK_GT(size, 0U); |
- ret.resize(size - 1); |
- } |
- } |
- |
- if (FAILED(hr)) { |
- NOTREACHED() << base::StringPrintf("ObtainUserAgentString==0x%08X", hr); |
- return std::string(); |
- } |
- |
- return ret; |
-} |
- |
-bool HasFrameBustingHeader(const std::string& http_headers) { |
- // NOTE: We cannot use net::GetSpecificHeader() here since when there are |
- // multiple instances of a header that returns the first value seen, and we |
- // need to look at all instances. |
- net::HttpUtil::HeadersIterator it(http_headers.begin(), http_headers.end(), |
- "\r\n"); |
- while (it.GetNext()) { |
- if (!lstrcmpiA(it.name().c_str(), kXFrameOptionsHeader) && |
- lstrcmpiA(it.values().c_str(), kXFrameOptionsValueAllowAll)) |
- return true; |
- } |
- return false; |
-} |
- |
-std::string GetHttpHeaderFromHeaderList(const std::string& header, |
- const std::string& headers) { |
- net::HttpUtil::HeadersIterator it(headers.begin(), headers.end(), "\r\n"); |
- while (it.GetNext()) { |
- if (!lstrcmpiA(it.name().c_str(), header.c_str())) |
- return std::string(it.values_begin(), it.values_end()); |
- } |
- return std::string(); |
-} |
- |
-} // namespace http_utils |