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/webdata/token_service_table.h" | |
6 | |
7 #include <map> | |
8 #include <string> | |
9 | |
10 #include "base/logging.h" | |
11 #include "components/webdata/common/web_database.h" | |
12 #include "components/webdata/encryptor/encryptor.h" | |
13 #include "sql/statement.h" | |
14 | |
15 namespace { | |
16 | |
17 WebDatabaseTable::TypeKey GetKey() { | |
18 // We just need a unique constant. Use the address of a static that | |
19 // COMDAT folding won't touch in an optimizing linker. | |
20 static int table_key = 0; | |
21 return reinterpret_cast<void*>(&table_key); | |
22 } | |
23 | |
24 } // namespace | |
25 | |
26 TokenServiceTable* TokenServiceTable::FromWebDatabase(WebDatabase* db) { | |
27 return static_cast<TokenServiceTable*>(db->GetTable(GetKey())); | |
28 | |
29 } | |
30 | |
31 WebDatabaseTable::TypeKey TokenServiceTable::GetTypeKey() const { | |
32 return GetKey(); | |
33 } | |
34 | |
35 bool TokenServiceTable::Init(sql::Connection* db, sql::MetaTable* meta_table) { | |
36 WebDatabaseTable::Init(db, meta_table); | |
37 if (!db_->DoesTableExist("token_service")) { | |
38 if (!db_->Execute("CREATE TABLE token_service (" | |
39 "service VARCHAR PRIMARY KEY NOT NULL," | |
40 "encrypted_token BLOB)")) { | |
41 NOTREACHED(); | |
42 return false; | |
43 } | |
44 } | |
45 return true; | |
46 } | |
47 | |
48 bool TokenServiceTable::IsSyncable() { | |
49 return true; | |
50 } | |
51 | |
52 bool TokenServiceTable::MigrateToVersion(int version, | |
53 bool* update_compatible_version) { | |
54 return true; | |
55 } | |
56 | |
57 bool TokenServiceTable::RemoveAllTokens() { | |
58 sql::Statement s(db_->GetUniqueStatement( | |
59 "DELETE FROM token_service")); | |
60 | |
61 return s.Run(); | |
62 } | |
63 | |
64 bool TokenServiceTable::RemoveTokenForService(const std::string& service) { | |
65 sql::Statement s(db_->GetUniqueStatement( | |
66 "DELETE FROM token_service WHERE service = ?")); | |
67 s.BindString(0, service); | |
68 | |
69 return s.Run(); | |
70 } | |
71 | |
72 bool TokenServiceTable::SetTokenForService(const std::string& service, | |
73 const std::string& token) { | |
74 std::string encrypted_token; | |
75 bool encrypted = Encryptor::EncryptString(token, &encrypted_token); | |
76 if (!encrypted) { | |
77 return false; | |
78 } | |
79 | |
80 // Don't bother with a cached statement since this will be a relatively | |
81 // infrequent operation. | |
82 sql::Statement s(db_->GetUniqueStatement( | |
83 "INSERT OR REPLACE INTO token_service " | |
84 "(service, encrypted_token) VALUES (?, ?)")); | |
85 s.BindString(0, service); | |
86 s.BindBlob(1, encrypted_token.data(), | |
87 static_cast<int>(encrypted_token.length())); | |
88 | |
89 return s.Run(); | |
90 } | |
91 | |
92 bool TokenServiceTable::GetAllTokens( | |
93 std::map<std::string, std::string>* tokens) { | |
94 sql::Statement s(db_->GetUniqueStatement( | |
95 "SELECT service, encrypted_token FROM token_service")); | |
96 | |
97 if (!s.is_valid()) | |
98 return false; | |
99 | |
100 while (s.Step()) { | |
101 std::string encrypted_token; | |
102 std::string decrypted_token; | |
103 std::string service; | |
104 service = s.ColumnString(0); | |
105 bool entry_ok = !service.empty() && | |
106 s.ColumnBlobAsString(1, &encrypted_token); | |
107 if (entry_ok) { | |
108 Encryptor::DecryptString(encrypted_token, &decrypted_token); | |
109 (*tokens)[service] = decrypted_token; | |
110 } else { | |
111 NOTREACHED(); | |
112 return false; | |
113 } | |
114 } | |
115 return true; | |
116 } | |
OLD | NEW |