Pepper: Add a property to URLRequestInfo to skip header validation.

webkit/plugins/ppapi/ppb_url_request_info_impl.cc

-// Copyright (c) 2010 The Chromium Authors. All rights reserved.
+// Copyright (c) 2011 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 "webkit/plugins/ppapi/ppb_url_request_info_impl.h"
 
 #include "base/logging.h"
 #include "base/string_util.h"
 #include "googleurl/src/gurl.h"
 #include "net/http/http_util.h"
 #include "ppapi/c/pp_var.h"
@@ skipping 173 matching lines @@
   int64_t start_offset;
   int64_t number_of_bytes;
   PP_Time expected_last_modified_time;
 };
 
 PPB_URLRequestInfo_Impl::PPB_URLRequestInfo_Impl(PluginInstance* instance)
     : Resource(instance),
       stream_to_file_(false),
       follow_redirects_(true),
       record_download_progress_(false),
-      record_upload_progress_(false) {
+      record_upload_progress_(false),
+      use_custom_referrer_(false) {
 }
 
 PPB_URLRequestInfo_Impl::~PPB_URLRequestInfo_Impl() {
 }
 
 // static
 const PPB_URLRequestInfo* PPB_URLRequestInfo_Impl::GetInterface() {
   return &ppb_urlrequestinfo;
 }
 
 PPB_URLRequestInfo_Impl* PPB_URLRequestInfo_Impl::AsPPB_URLRequestInfo_Impl() {
   return this;
 }
 
 bool PPB_URLRequestInfo_Impl::SetBooleanProperty(PP_URLRequestProperty property,
                                                  bool value) {
   switch (property) {
     case PP_URLREQUESTPROPERTY_STREAMTOFILE:
       stream_to_file_ = value;
       return true;
     case PP_URLREQUESTPROPERTY_FOLLOWREDIRECTS:
       follow_redirects_ = value;
       return true;
     case PP_URLREQUESTPROPERTY_RECORDDOWNLOADPROGRESS:
       record_download_progress_ = value;
       return true;
     case PP_URLREQUESTPROPERTY_RECORDUPLOADPROGRESS:
       record_upload_progress_ = value;
       return true;
+    case PP_URLREQUESTPROPERTY_USECUSTOMREFERRER:
+      use_custom_referrer_ = value;
+      return true;
     default:
       return false;
   }
 }
 
 bool PPB_URLRequestInfo_Impl::SetStringProperty(PP_URLRequestProperty property,
                                                 const std::string& value) {
   // TODO(darin): Validate input.  Perhaps at a different layer?
   switch (property) {
     case PP_URLREQUESTPROPERTY_URL:
       url_ = value;  // NOTE: This may be a relative URL.
       return true;
     case PP_URLREQUESTPROPERTY_METHOD:
       method_ = value;
       return true;
     case PP_URLREQUESTPROPERTY_HEADERS:
       if (!AreValidHeaders(value))
         return false;
       headers_ = value;
       return true;
+    case PP_URLREQUESTPROPERTY_CUSTOMREFERRERURL:
+      custom_referrer_url_ = value;
+      return true;
     default:
       return false;
   }
 }
 
 bool PPB_URLRequestInfo_Impl::AppendDataToBody(const std::string& data) {
   if (!data.empty())
     body_.push_back(BodyItem(data));
   return true;
 }
@@ skipping 49 matching lines @@
             body_[i].number_of_bytes,
             body_[i].expected_last_modified_time);
       } else {
         DCHECK(!body_[i].data.empty());
         http_body.appendData(WebData(body_[i].data));
       }
     }
     web_request.setHTTPBody(http_body);
   }
 
-  frame->setReferrerForRequest(web_request, WebURL());  // Use default.
+  if (use_custom_referrer()) {
+    if (!custom_referrer_url_.empty())
+      frame->setReferrerForRequest(web_request, GURL(custom_referrer_url_));
+  } else {
+    frame->setReferrerForRequest(web_request, WebURL());  // Use default.
+  }
+
   return web_request;
 }
 
+bool PPB_URLRequestInfo_Impl::RequiresUniversalAccess() const {
+  return use_custom_referrer();
+}
+
 }  // namespace ppapi
 }  // namespace webkit
-