Move referrer policy to a different field when serializing.

components/sessions/ios/ios_serialized_navigation_driver.cc

 // Copyright 2014 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 "components/sessions/ios/ios_serialized_navigation_driver.h"
 
 #include "base/memory/singleton.h"
 #include "components/sessions/serialized_navigation_entry.h"
 #include "ios/web/public/referrer.h"
 
@@ skipping 14 matching lines @@
 IOSSerializedNavigationDriver::IOSSerializedNavigationDriver() {
 }
 
 IOSSerializedNavigationDriver::~IOSSerializedNavigationDriver() {
 }
 
 int IOSSerializedNavigationDriver::GetDefaultReferrerPolicy() const {
   return web::ReferrerPolicyDefault;
 }
 
+bool IOSSerializedNavigationDriver::MapReferrerPolicyToOldValues(
+    int referrer_policy,
+    int* mapped_referrer_policy) const {
+  // "always" and "default" are the same value in all versions.
+  if (referrer_policy == web::ReferrerPolicyAlways ||
+      referrer_policy == web::ReferrerPolicyDefault) {
+    *mapped_referrer_policy = referrer_policy;
+    return true;
+  }
+
+  // "origin" exists in the old encoding.
+  if (referrer_policy == web::ReferrerPolicyOrigin) {
+    *mapped_referrer_policy = 3;
+    return true;
+  }
+
+  // Everything else is mapped to never.
+  *mapped_referrer_policy = 2;
+  return false;
+}
+
+bool IOSSerializedNavigationDriver::MapReferrerPolicyToNewValues(
+    int referrer_policy,
+    int* mapped_referrer_policy) const {
+  // "always" and "default" are the same value in all versions.
+  if (referrer_policy == 0 || referrer_policy == 1) {
+    *mapped_referrer_policy = referrer_policy;
+    return true;
+  }
+
+  // Since we don't know what encoding was used, we map the rest to "never".
+  *mapped_referrer_policy = web::ReferrerPolicyNever;
+  return false;
+}
+
 std::string
 IOSSerializedNavigationDriver::GetSanitizedPageStateForPickle(
     const SerializedNavigationEntry* navigation) const {
   return std::string();
 }
 
 void IOSSerializedNavigationDriver::Sanitize(
     SerializedNavigationEntry* navigation) const {
   web::Referrer referrer(
       navigation->referrer_url_,
       static_cast<web::ReferrerPolicy>(navigation->referrer_policy_));
 
   if (!navigation->virtual_url_.SchemeIsHTTPOrHTTPS() ||
       !referrer.url.SchemeIsHTTPOrHTTPS()) {
     referrer.url = GURL();
   } else {
+    if (referrer.policy < 0 || referrer.policy > web::ReferrerPolicyLast) {
+      NOTREACHED();
+      referrer.policy = web::ReferrerPolicyNever;
+    }
+    bool is_downgrade = referrer.url.SchemeIsSecure() &&
+                        !navigation->virtual_url_.SchemeIsSecure();
     switch (referrer.policy) {
       case web::ReferrerPolicyDefault:
-        if (referrer.url.SchemeIsSecure() &&
-            !navigation->virtual_url_.SchemeIsSecure()) {
+        if (is_downgrade)
           referrer.url = GURL();
-        }
         break;
+      case web::ReferrerPolicyNoReferrerWhenDowngrade:
+        if (is_downgrade)
+          referrer.url = GURL();
       case web::ReferrerPolicyAlways:
         break;
       case web::ReferrerPolicyNever:
         referrer.url = GURL();
         break;
       case web::ReferrerPolicyOrigin:
         referrer.url = referrer.url.GetOrigin();
         break;
-      default:
-        NOTREACHED();
+      case web::ReferrerPolicyOriginWhenCrossOrigin:
+        if (navigation->virtual_url_.GetOrigin() != referrer.url.GetOrigin())
+          referrer.url = referrer.url.GetOrigin();
         break;
     }
   }
 
   // Reset the referrer if it has changed.
   if (navigation->referrer_url_ != referrer.url) {
     navigation->referrer_url_ = GURL();
     navigation->referrer_policy_ = GetDefaultReferrerPolicy();
   }
 }
 
 }  // namespace sessions