1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
// Copyright (c) 2009 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE entry.
//
// Implement the storage of service tokens in memory.
#include "chrome/browser/sync/util/user_settings.h"
#include "chrome/browser/password_manager/encryptor.h"
#include "chrome/browser/sync/util/query_helpers.h"
namespace browser_sync {
void UserSettings::SetAuthTokenForService(
const std::string& email,
const std::string& service_name,
const std::string& long_lived_service_token) {
std::string encrypted_service_token;
if (!Encryptor::EncryptString(long_lived_service_token,
&encrypted_service_token)) {
LOG(ERROR) << "Encrytion failed: " << long_lived_service_token;
return;
}
ScopedDBHandle dbhandle(this);
ExecOrDie(dbhandle.get(), "INSERT INTO cookies "
"(email, service_name, service_token) "
"values (?, ?, ?)", email, service_name, encrypted_service_token);
}
void UserSettings::ClearAllServiceTokens() {
ScopedDBHandle dbhandle(this);
ExecOrDie(dbhandle.get(), "DELETE FROM cookies");
}
bool UserSettings::GetLastUser(std::string* username) {
ScopedDBHandle dbhandle(this);
ScopedStatement query(PrepareQuery(dbhandle.get(),
"SELECT email FROM cookies"));
if (SQLITE_ROW == sqlite3_step(query.get())) {
GetColumn(query.get(), 0, username);
return true;
} else {
return false;
}
}
bool UserSettings::GetLastUserAndServiceToken(const std::string& service_name,
std::string* username,
std::string* service_token) {
ScopedDBHandle dbhandle(this);
ScopedStatement query(PrepareQuery(
dbhandle.get(),
"SELECT email, service_token FROM cookies WHERE service_name = ?",
service_name));
if (SQLITE_ROW == sqlite3_step(query.get())) {
std::string encrypted_service_token;
GetColumn(query.get(), 1, &encrypted_service_token);
if (!Encryptor::DecryptString(encrypted_service_token, service_token)) {
LOG(ERROR) << "Decryption failed: " << encrypted_service_token;
return false;
}
GetColumn(query.get(), 0, username);
return true;
}
return false;
}
} // namespace browser_sync
|
