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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
#ifndef CERT_TRANS_LOG_SQLITE_DB_H_
#define CERT_TRANS_LOG_SQLITE_DB_H_
#include <mutex>
#include <string>
#include "base/macros.h"
#include "log/database.h"
#include "log/logged_entry.h"
struct sqlite3;
namespace cert_trans {
class SQLiteDB : public Database {
public:
explicit SQLiteDB(const std::string& dbfile);
~SQLiteDB();
typedef Database::WriteResult WriteResult;
typedef Database::LookupResult LookupResult;
WriteResult CreateSequencedEntry_(const LoggedEntry& logged) override;
LookupResult LookupByHash(const std::string& hash,
LoggedEntry* result) const override;
LookupResult LookupByIndex(int64_t sequence_number,
LoggedEntry* result) const override;
std::unique_ptr<Database::Iterator> ScanEntries(
int64_t start_index) const override;
WriteResult WriteTreeHead_(const ct::SignedTreeHead& sth) override;
LookupResult LatestTreeHead(ct::SignedTreeHead* result) const override;
int64_t TreeSize() const override;
void AddNotifySTHCallback(
const Database::NotifySTHCallback* callback) override;
void RemoveNotifySTHCallback(
const Database::NotifySTHCallback* callback) override;
void InitializeNode(const std::string& node_id) override;
LookupResult NodeId(std::string* node_id) override;
// Force an STH notification. This is needed only for ct-dns-server,
// which shares a SQLite database with ct-server, but needs to
// refresh itself occasionally.
void ForceNotifySTH();
private:
class Iterator;
LookupResult LookupByIndex(const std::unique_lock<std::mutex>& lock,
int64_t sequence_number,
LoggedEntry* result) const;
// This finds the next entry with a sequence number equal or greater
// to the one specified.
LookupResult LookupNextIndex(const std::unique_lock<std::mutex>& lock,
int64_t sequence_number,
LoggedEntry* result) const;
LookupResult LatestTreeHeadNoLock(const std::unique_lock<std::mutex>& lock,
ct::SignedTreeHead* result) const;
LookupResult NodeId(const std::unique_lock<std::mutex>& lock,
std::string* node_id);
void BeginTransaction(const std::unique_lock<std::mutex>& lock);
void EndTransaction(const std::unique_lock<std::mutex>& lock);
void MaybeStartNewTransaction(const std::unique_lock<std::mutex>& lock);
mutable std::mutex lock_;
sqlite3* const db_;
// This is marked mutable, as it is a lazily updated cache updated
// from some of the getters.
mutable int64_t tree_size_;
DatabaseNotifierHelper callbacks_;
int64_t transaction_size_;
bool in_transaction_;
DISALLOW_COPY_AND_ASSIGN(SQLiteDB);
};
} // namespace cert_trans
#endif // CERT_TRANS_LOG_SQLITE_DB_H_
|
