android_webview/native/state_serializer.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "android_webview/native/state_serializer.h"
   6
   7 #include <string>
   8
   9 #include "base/memory/scoped_vector.h"
  10 #include "base/pickle.h"
  11 #include "base/time/time.h"
  12 #include "content/public/browser/child_process_security_policy.h"
  13 #include "content/public/browser/navigation_controller.h"
  14 #include "content/public/browser/navigation_entry.h"
  15 #include "content/public/browser/render_process_host.h"
  16 #include "content/public/browser/web_contents.h"
  17 #include "content/public/common/page_state.h"
  18
  19 // Reasons for not re-using TabNavigation under chrome/ as of 20121116:
  20 // * Android WebView has different requirements for fields to store since
  21 //   we are the only ones using values like BaseURLForDataURL.
  22 // * TabNavigation does unnecessary copying of data, which in Android
  23 //   WebView case, is undesired since save/restore is called in Android
  24 //   very frequently.
  25 // * TabNavigation is tightly integrated with the rest of chrome session
  26 //   restore and sync code, and has other purpose in addition to serializing
  27 //   NavigationEntry.
  28
  29 using std::string;
  30
  31 namespace android_webview {
  32
  33 namespace {
  34
  35 // Sanity check value that we are restoring from a valid pickle.
  36 // This can potentially used as an actual serialization version number in the
  37 // future if we ever decide to support restoring from older versions.
  38 const uint32 AW_STATE_VERSION = 20130814;
  39
  40 }  // namespace
  41
  42 bool WriteToPickle(const content::WebContents& web_contents,
  43                    Pickle* pickle) {
  44   DCHECK(pickle);
  45
  46   if (!internal::WriteHeaderToPickle(pickle))
  47     return false;
  48
  49   const content::NavigationController& controller =
  50       web_contents.GetController();
  51   const int entry_count = controller.GetEntryCount();
  52   const int selected_entry = controller.GetCurrentEntryIndex();
  53   DCHECK_GE(entry_count, 0);
  54   DCHECK_GE(selected_entry, -1);  // -1 is valid
  55   DCHECK_LT(selected_entry, entry_count);
  56
  57   if (!pickle->WriteInt(entry_count))
  58     return false;
  59
  60   if (!pickle->WriteInt(selected_entry))
  61     return false;
  62
  63   for (int i = 0; i < entry_count; ++i) {
  64     if (!internal::WriteNavigationEntryToPickle(*controller.GetEntryAtIndex(i),
  65                                                 pickle))
  66       return false;
  67   }
  68
  69   // Please update AW_STATE_VERSION if serialization format is changed.
  70
  71   return true;
  72 }
  73
  74 bool RestoreFromPickle(PickleIterator* iterator,
  75                        content::WebContents* web_contents) {
  76   DCHECK(iterator);
  77   DCHECK(web_contents);
  78
  79   if (!internal::RestoreHeaderFromPickle(iterator))
  80     return false;
  81
  82   int entry_count = -1;
  83   int selected_entry = -2;  // -1 is a valid value
  84
  85   if (!iterator->ReadInt(&entry_count))
  86     return false;
  87
  88   if (!iterator->ReadInt(&selected_entry))
  89     return false;
  90
  91   if (entry_count < 0)
  92     return false;
  93   if (selected_entry < -1)
  94     return false;
  95   if (selected_entry >= entry_count)
  96     return false;
  97
  98   ScopedVector<content::NavigationEntry> restored_entries;
  99   for (int i = 0; i < entry_count; ++i) {
 100     restored_entries.push_back(content::NavigationEntry::Create());
 101     if (!internal::RestoreNavigationEntryFromPickle(iterator,
 102                                                     restored_entries[i]))
 103       return false;
 104
 105     restored_entries[i]->SetPageID(i);
 106   }
 107
 108   // |web_contents| takes ownership of these entries after this call.
 109   content::NavigationController& controller = web_contents->GetController();
 110   controller.Restore(
 111       selected_entry,
 112       content::NavigationController::RESTORE_LAST_SESSION_EXITED_CLEANLY,
 113       &restored_entries.get());
 114   DCHECK_EQ(0u, restored_entries.size());
 115
 116   if (controller.GetActiveEntry()) {
 117     // Set up the file access rights for the selected navigation entry.
 118     // TODO(joth): This is duplicated from chrome/.../session_restore.cc and
 119     // should be shared e.g. in  NavigationController. http://crbug.com/68222
 120     const int id = web_contents->GetRenderProcessHost()->GetID();
 121     const content::PageState& page_state =
 122         controller.GetActiveEntry()->GetPageState();
 123     const std::vector<base::FilePath>& file_paths =
 124         page_state.GetReferencedFiles();
 125     for (std::vector<base::FilePath>::const_iterator file = file_paths.begin();
 126          file != file_paths.end(); ++file) {
 127       content::ChildProcessSecurityPolicy::GetInstance()->GrantReadFile(id,
 128                                                                         *file);
 129     }
 130   }
 131
 132   controller.LoadIfNecessary();
 133
 134   return true;
 135 }
 136
 137 namespace internal {
 138
 139 bool WriteHeaderToPickle(Pickle* pickle) {
 140   return pickle->WriteUInt32(AW_STATE_VERSION);
 141 }
 142
 143 bool RestoreHeaderFromPickle(PickleIterator* iterator) {
 144   uint32 state_version = -1;
 145   if (!iterator->ReadUInt32(&state_version))
 146     return false;
 147
 148   if (AW_STATE_VERSION != state_version)
 149     return false;
 150
 151   return true;
 152 }
 153
 154 bool WriteNavigationEntryToPickle(const content::NavigationEntry& entry,
 155                                   Pickle* pickle) {
 156   if (!pickle->WriteString(entry.GetURL().spec()))
 157     return false;
 158
 159   if (!pickle->WriteString(entry.GetVirtualURL().spec()))
 160     return false;
 161
 162   const content::Referrer& referrer = entry.GetReferrer();
 163   if (!pickle->WriteString(referrer.url.spec()))
 164     return false;
 165   if (!pickle->WriteInt(static_cast<int>(referrer.policy)))
 166     return false;
 167
 168   if (!pickle->WriteString16(entry.GetTitle()))
 169     return false;
 170
 171   if (!pickle->WriteString(entry.GetPageState().ToEncodedData()))
 172     return false;
 173
 174   if (!pickle->WriteBool(static_cast<int>(entry.GetHasPostData())))
 175     return false;
 176
 177   if (!pickle->WriteString(entry.GetOriginalRequestURL().spec()))
 178     return false;
 179
 180   if (!pickle->WriteString(entry.GetBaseURLForDataURL().spec()))
 181     return false;
 182
 183   if (!pickle->WriteBool(static_cast<int>(entry.GetIsOverridingUserAgent())))
 184     return false;
 185
 186   if (!pickle->WriteInt64(entry.GetTimestamp().ToInternalValue()))
 187     return false;
 188
 189   if (!pickle->WriteInt(entry.GetHttpStatusCode()))
 190     return false;
 191
 192   // Please update AW_STATE_VERSION if serialization format is changed.
 193
 194   return true;
 195 }
 196
 197 bool RestoreNavigationEntryFromPickle(PickleIterator* iterator,
 198                                       content::NavigationEntry* entry) {
 199   {
 200     string url;
 201     if (!iterator->ReadString(&url))
 202       return false;
 203     entry->SetURL(GURL(url));
 204   }
 205
 206   {
 207     string virtual_url;
 208     if (!iterator->ReadString(&virtual_url))
 209       return false;
 210     entry->SetVirtualURL(GURL(virtual_url));
 211   }
 212
 213   {
 214     content::Referrer referrer;
 215     string referrer_url;
 216     int policy;
 217
 218     if (!iterator->ReadString(&referrer_url))
 219       return false;
 220     if (!iterator->ReadInt(&policy))
 221       return false;
 222
 223     referrer.url = GURL(referrer_url);
 224     referrer.policy = static_cast<blink::WebReferrerPolicy>(policy);
 225     entry->SetReferrer(referrer);
 226   }
 227
 228   {
 229     base::string16 title;
 230     if (!iterator->ReadString16(&title))
 231       return false;
 232     entry->SetTitle(title);
 233   }
 234
 235   {
 236     string content_state;
 237     if (!iterator->ReadString(&content_state))
 238       return false;
 239     entry->SetPageState(
 240         content::PageState::CreateFromEncodedData(content_state));
 241   }
 242
 243   {
 244     bool has_post_data;
 245     if (!iterator->ReadBool(&has_post_data))
 246       return false;
 247     entry->SetHasPostData(has_post_data);
 248   }
 249
 250   {
 251     string original_request_url;
 252     if (!iterator->ReadString(&original_request_url))
 253       return false;
 254     entry->SetOriginalRequestURL(GURL(original_request_url));
 255   }
 256
 257   {
 258     string base_url_for_data_url;
 259     if (!iterator->ReadString(&base_url_for_data_url))
 260       return false;
 261     entry->SetBaseURLForDataURL(GURL(base_url_for_data_url));
 262   }
 263
 264   {
 265     bool is_overriding_user_agent;
 266     if (!iterator->ReadBool(&is_overriding_user_agent))
 267       return false;
 268     entry->SetIsOverridingUserAgent(is_overriding_user_agent);
 269   }
 270
 271   {
 272     int64 timestamp;
 273     if (!iterator->ReadInt64(&timestamp))
 274       return false;
 275     entry->SetTimestamp(base::Time::FromInternalValue(timestamp));
 276   }
 277
 278   {
 279     int http_status_code;
 280     if (!iterator->ReadInt(&http_status_code))
 281       return false;
 282     entry->SetHttpStatusCode(http_status_code);
 283   }
 284
 285   return true;
 286 }
 287
 288 }  // namespace internal
 289
 290 }  // namespace android_webview