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 84 matching lines @@
 }
 
 PP_Bool SetProperty(PP_Resource request_id,
                     PP_URLRequestProperty property,
                     PP_Var var) {
   scoped_refptr<PPB_URLRequestInfo_Impl> request(
       Resource::GetAs<PPB_URLRequestInfo_Impl>(request_id));
   if (!request)
     return PP_FALSE;
 
+  if (var.type == PP_VARTYPE_UNDEFINED)

[brettw, 2011/03/11 23:31:05]

Now that we have 3 of these, I think it would be more clear if this was a switch
statement on the var.type.

[viettrungluu, 2011/03/14 17:13:53]

Done.

+    return BoolToPPBool(request->SetUndefinedProperty(property));
+
   if (var.type == PP_VARTYPE_BOOL) {
     return BoolToPPBool(
         request->SetBooleanProperty(property,
                                     PPBoolToBool(var.value.as_bool)));
   }
 
   if (var.type == PP_VARTYPE_STRING) {
     scoped_refptr<StringVar> string(StringVar::FromPPVar(var));
     if (string) {
       return BoolToPPBool(request->SetStringProperty(property,
@@ skipping 69 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),
+      has_custom_referrer_url_(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::SetUndefinedProperty(
+    PP_URLRequestProperty property) {
+  switch (property) {
+    case PP_URLREQUESTPROPERTY_CUSTOMREFERRERURL:
+      has_custom_referrer_url_ = false;
+      custom_referrer_url_ = std::string();
+      return true;
+    default:
+      return false;
+  }
+}
+
 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:
@@ skipping 15 matching lines @@
       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:
+      has_custom_referrer_url_ = true;
+      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 (has_custom_referrer_url_) {
+    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 has_custom_referrer_url_;
+}
+
 }  // namespace ppapi
 }  // namespace webkit
-