components/signin/core/webdata/token_service_table.cc - Issue 129253003: Revert of Create signin component and componentize TokenWebData.

Side by Side Diff: components/signin/core/webdata/token_service_table.cc

Issue 129253003: Revert of Create signin component and componentize TokenWebData. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src

Patch Set: Created 6 years, 11 months ago

Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.

Jump to:

View unified diff | Download patch | Annotate | Revision Log

OLD	NEW
	(Empty)
1 // Copyright (c) 2011 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 "components/signin/core/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