blob: e534d6f06ab624d7c97c8b69f8b3240b7f1cce17 [file] [log] [blame] [edit]
// Copyright 2016 The Cobalt Authors. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#include "cobalt/browser/storage_upgrade_handler.h"
#include <vector>
#include "base/logging.h"
#include "base/memory/ref_counted.h"
#include "cobalt/dom/local_storage_database.h"
#include "cobalt/dom/storage_area.h"
#include "cobalt/network/persistent_cookie_store.h"
#include "cobalt/storage/storage_manager.h"
#include "cobalt/storage/upgrade/upgrade_reader.h"
#include "net/cookies/canonical_cookie.h"
namespace cobalt {
namespace browser {
StorageUpgradeHandler::StorageUpgradeHandler(
const GURL& url, net::NetLog* net_log,
scoped_refptr<base::SingleThreadTaskRunner> cookie_callback_thread)
: default_local_storage_origin_(loader::Origin(url)),
net_log_(net_log),
cookie_callback_thread_(cookie_callback_thread) {}
void StorageUpgradeHandler::OnUpgrade(storage::StorageManager* storage,
const char* data, int size) {
storage::upgrade::UpgradeReader upgrade_reader;
upgrade_reader.Parse(data, size);
int num_cookies = upgrade_reader.GetNumCookies();
int num_local_storage_entries = upgrade_reader.GetNumLocalStorageEntries();
DLOG(INFO) << "Upgrading legacy save data: " << num_cookies << " cookies, "
<< num_local_storage_entries << " local storage entries.";
if (num_cookies > 0) {
scoped_refptr<network::PersistentCookieStore> cookie_store(
new network::PersistentCookieStore(
storage, cookie_callback_thread_
? cookie_callback_thread_
: base::MessageLoop::current()->task_runner()));
// We don't care about the local copies of the cookies returned to us.
auto null_loaded_callback =
net::CookieMonster::PersistentCookieStore::LoadedCallback();
// Load the current cookies to ensure the database table is initialized.
cookie_store->Load(null_loaded_callback,
net::NetLogWithSource::Make(
net_log_, net::NetLogSourceType::COOKIE_STORE));
for (int i = 0; i < num_cookies; i++) {
const net::CanonicalCookie* cookie = upgrade_reader.GetCookie(i);
DCHECK(cookie);
cookie_store->AddCookie(*cookie);
}
}
if (num_local_storage_entries > 0) {
dom::LocalStorageDatabase local_storage_database(storage);
for (int i = 0; i < num_local_storage_entries; i++) {
const storage::upgrade::UpgradeReader::LocalStorageEntry*
local_storage_entry = upgrade_reader.GetLocalStorageEntry(i);
DCHECK(local_storage_entry);
local_storage_database.Write(default_local_storage_origin_,
local_storage_entry->key,
local_storage_entry->value);
}
}
}
} // namespace browser
} // namespace cobalt