Support WebSocket in WebRequest API.

extensions/browser/api/web_request/web_request_api_helpers.cc

 // 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 "extensions/browser/api/web_request/web_request_api_helpers.h"
 
 #include <stddef.h>
 #include <stdint.h>
 
 #include <algorithm>
@@ skipping 446 matching lines @@
 }
 
 // Helper function for MergeRedirectUrlOfResponses() that allows ignoring
 // all redirects but those to data:// urls and about:blank. This is important
 // to treat these URLs as "cancel urls", i.e. URLs that extensions redirect
 // to if they want to express that they want to cancel a request. This reduces
 // the number of conflicts that we need to flag, as canceling is considered
 // a higher precedence operation that redirects.
 // Returns whether a redirect occurred.
 static bool MergeRedirectUrlOfResponsesHelper(
+    const GURL& url,
     const EventResponseDeltas& deltas,
     GURL* new_url,
     extensions::WarningSet* conflicting_extensions,
     const net::NetLogWithSource* net_log,
     bool consider_only_cancel_scheme_urls) {
+  // Redirecting WebSocket handshake request is prohibited.
+  if (url.SchemeIsWSOrWSS())
+    return false;
+
   bool redirected = false;
 
   // Extension that determines the |new_url|.
   std::string winning_extension_id;
   EventResponseDeltas::const_iterator delta;
   for (delta = deltas.begin(); delta != deltas.end(); ++delta) {
     if ((*delta)->new_url.is_empty())
       continue;
     if (consider_only_cancel_scheme_urls &&
         !(*delta)->new_url.SchemeIs(url::kDataScheme) &&
@@ skipping 16 matching lines @@
               (*delta)->new_url,
               *new_url));
       net_log->AddEvent(
           net::NetLogEventType::CHROME_EXTENSION_IGNORED_DUE_TO_CONFLICT,
           CreateNetLogExtensionIdCallback(delta->get()));
     }
   }
   return redirected;
 }
 
-void MergeRedirectUrlOfResponses(const EventResponseDeltas& deltas,
+void MergeRedirectUrlOfResponses(const GURL& url,
+                                 const EventResponseDeltas& deltas,
                                  GURL* new_url,
                                  extensions::WarningSet* conflicting_extensions,
                                  const net::NetLogWithSource* net_log) {
   // First handle only redirects to data:// URLs and about:blank. These are a
   // special case as they represent a way of cancelling a request.
   if (MergeRedirectUrlOfResponsesHelper(
-          deltas, new_url, conflicting_extensions, net_log, true)) {
+          url, deltas, new_url, conflicting_extensions, net_log, true)) {
     // If any extension cancelled a request by redirecting to a data:// URL or
     // about:blank, we don't consider the other redirects.
     return;
   }
 
   // Handle all other redirects.
-  MergeRedirectUrlOfResponsesHelper(
-      deltas, new_url, conflicting_extensions, net_log, false);
+  MergeRedirectUrlOfResponsesHelper(url, deltas, new_url,
+                                    conflicting_extensions, net_log, false);
 }
 
 void MergeOnBeforeRequestResponses(
+    const GURL& url,
     const EventResponseDeltas& deltas,
     GURL* new_url,
     extensions::WarningSet* conflicting_extensions,
     const net::NetLogWithSource* net_log) {
-  MergeRedirectUrlOfResponses(deltas, new_url, conflicting_extensions, net_log);
+  MergeRedirectUrlOfResponses(url, deltas, new_url, conflicting_extensions,
+                              net_log);
 }
 
 static bool DoesRequestCookieMatchFilter(
     const ParsedRequestCookie& cookie,
     RequestCookie* filter) {
   if (!filter) return true;
   if (filter->name.get() && cookie.first != *filter->name) return false;
   if (filter->value.get() && cookie.second != *filter->value) return false;
   return true;
 }
@@ skipping 535 matching lines @@
   for (const auto& delta : deltas) {
     for (const auto& deleted_hdr : delta->deleted_response_headers) {
       if (base::ToLowerASCII(deleted_hdr.first) == lower_key)
         return delta->extension_id;
     }
   }
   return std::string();
 }
 
 void MergeOnHeadersReceivedResponses(
+    const GURL& url,
     const EventResponseDeltas& deltas,
     const net::HttpResponseHeaders* original_response_headers,
     scoped_refptr<net::HttpResponseHeaders>* override_response_headers,
     GURL* allowed_unsafe_redirect_url,
     extensions::WarningSet* conflicting_extensions,
     const net::NetLogWithSource* net_log) {
   EventResponseDeltas::const_iterator delta;
 
   // Here we collect which headers we have removed or added so far due to
   // extensions of higher precedence. Header keys are always stored as
@@ skipping 66 matching lines @@
       net_log->AddEvent(
           net::NetLogEventType::CHROME_EXTENSION_IGNORED_DUE_TO_CONFLICT,
           CreateNetLogExtensionIdCallback(delta->get()));
     }
   }
 
   MergeCookiesInOnHeadersReceivedResponses(deltas, original_response_headers,
       override_response_headers, conflicting_extensions, net_log);
 
   GURL new_url;
-  MergeRedirectUrlOfResponses(
-      deltas, &new_url, conflicting_extensions, net_log);
+  MergeRedirectUrlOfResponses(url, deltas, &new_url, conflicting_extensions,
+                              net_log);
   if (new_url.is_valid()) {
     // Only create a copy if we really want to modify the response headers.
     if (override_response_headers->get() == NULL) {
       *override_response_headers = new net::HttpResponseHeaders(
           original_response_headers->raw_headers());
     }
     (*override_response_headers)->ReplaceStatusLine("HTTP/1.1 302 Found");
     (*override_response_headers)->RemoveHeader("location");
     (*override_response_headers)->AddHeader("Location: " + new_url.spec());
     // Explicitly mark the URL as safe for redirection, to prevent the request
@@ skipping 95 matching lines @@
   for (size_t i = 0; i < kResourceTypeStringsLength; ++i) {
     if (type_str == kResourceTypeStrings[i]) {
       found = true;
       types->push_back(extensions::ResourceTypeExt(kResourceTypeValues[i]));
     }
   }
   return found;
 }
 
 }  // namespace extension_web_request_api_helpers