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_BOOL) {
-    return BoolToPPBool(
-        request->SetBooleanProperty(property,
-                                    PPBoolToBool(var.value.as_bool)));
+  PP_Bool result = PP_FALSE;
+  switch (var.type) {
+    case PP_VARTYPE_UNDEFINED:
+      result = BoolToPPBool(request->SetUndefinedProperty(property));
+      break;
+    case PP_VARTYPE_BOOL:
+      result = BoolToPPBool(
+          request->SetBooleanProperty(property,
+                                      PPBoolToBool(var.value.as_bool)));
+      break;
+    case PP_VARTYPE_STRING: {
+      scoped_refptr<StringVar> string(StringVar::FromPPVar(var));
+      if (string)
+        result = BoolToPPBool(request->SetStringProperty(property,
+                                                         string->value()));
+      break;
+    }
+    default:
+      break;
   }
-
-  if (var.type == PP_VARTYPE_STRING) {
-    scoped_refptr<StringVar> string(StringVar::FromPPVar(var));
-    if (string) {
-      return BoolToPPBool(request->SetStringProperty(property,
-                                                     string->value()));
-    }
-  }
-
-  return PP_FALSE;
+  return result;
 }
 
 PP_Bool AppendDataToBody(PP_Resource request_id,
                          const char* data,
                          uint32_t len) {
   scoped_refptr<PPB_URLRequestInfo_Impl> request(
       Resource::GetAs<PPB_URLRequestInfo_Impl>(request_id));
   if (!request)
     return PP_FALSE;
 
@@ skipping 54 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
-