OLD | NEW |
| (Empty) |
1 // Copyright 2014 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 "chrome/browser/net/quota_policy_channel_id_store.h" | |
6 | |
7 #include <list> | |
8 | |
9 #include "base/basictypes.h" | |
10 #include "base/bind.h" | |
11 #include "base/file_util.h" | |
12 #include "base/files/file_path.h" | |
13 #include "base/logging.h" | |
14 #include "base/metrics/histogram.h" | |
15 #include "base/strings/string_util.h" | |
16 #include "base/threading/thread.h" | |
17 #include "base/threading/thread_restrictions.h" | |
18 #include "net/cookies/cookie_util.h" | |
19 #include "net/extras/sqlite/sqlite_channel_id_store.h" | |
20 #include "url/gurl.h" | |
21 #include "webkit/browser/quota/special_storage_policy.h" | |
22 | |
23 QuotaPolicyChannelIDStore::QuotaPolicyChannelIDStore( | |
24 const base::FilePath& path, | |
25 const scoped_refptr<base::SequencedTaskRunner>& background_task_runner, | |
26 quota::SpecialStoragePolicy* special_storage_policy) | |
27 : special_storage_policy_(special_storage_policy), | |
28 persistent_store_( | |
29 new net::SQLiteChannelIDStore(path, background_task_runner)) { | |
30 DCHECK(background_task_runner); | |
31 } | |
32 | |
33 QuotaPolicyChannelIDStore::~QuotaPolicyChannelIDStore() { | |
34 if (!special_storage_policy_.get() || | |
35 !special_storage_policy_->HasSessionOnlyOrigins()) { | |
36 return; | |
37 } | |
38 std::list<std::string> session_only_server_identifiers; | |
39 for (std::set<std::string>::iterator it = server_identifiers_.begin(); | |
40 it != server_identifiers_.end(); | |
41 ++it) { | |
42 GURL url(net::cookie_util::CookieOriginToURL(*it, true)); | |
43 if (special_storage_policy_->IsStorageSessionOnly(url)) | |
44 session_only_server_identifiers.push_back(*it); | |
45 } | |
46 persistent_store_->DeleteAllInList(session_only_server_identifiers); | |
47 } | |
48 | |
49 void QuotaPolicyChannelIDStore::Load(const LoadedCallback& loaded_callback) { | |
50 persistent_store_->Load( | |
51 base::Bind(&QuotaPolicyChannelIDStore::OnLoad, this, loaded_callback)); | |
52 } | |
53 | |
54 void QuotaPolicyChannelIDStore::AddChannelID( | |
55 const net::DefaultChannelIDStore::ChannelID& channel_id) { | |
56 server_identifiers_.insert(channel_id.server_identifier()); | |
57 persistent_store_->AddChannelID(channel_id); | |
58 } | |
59 | |
60 void QuotaPolicyChannelIDStore::DeleteChannelID( | |
61 const net::DefaultChannelIDStore::ChannelID& channel_id) { | |
62 server_identifiers_.erase(channel_id.server_identifier()); | |
63 persistent_store_->DeleteChannelID(channel_id); | |
64 } | |
65 | |
66 void QuotaPolicyChannelIDStore::SetForceKeepSessionState() { | |
67 special_storage_policy_ = NULL; | |
68 } | |
69 | |
70 void QuotaPolicyChannelIDStore::OnLoad( | |
71 const LoadedCallback& loaded_callback, | |
72 scoped_ptr<ChannelIDVector> channel_ids) { | |
73 for (ChannelIDVector::const_iterator channel_id = channel_ids->begin(); | |
74 channel_id != channel_ids->end(); | |
75 ++channel_id) { | |
76 server_identifiers_.insert((*channel_id)->server_identifier()); | |
77 } | |
78 loaded_callback.Run(channel_ids.Pass()); | |
79 } | |
OLD | NEW |