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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
|
#pragma once
#define OWN_CACHED_CONTACT
#include <m_db_int.h>
struct CQuery
{
~CQuery();
sqlite3_stmt *pQuery = nullptr;
};
struct DBCachedContact : public DBCachedContactBase
{
int32_t m_count;
MEVENT m_unread;
uint32_t m_unreadTimestamp;
DBCachedContact()
: m_count(-1), m_unread(0) { }
__forceinline bool HasCount() const {
return m_count > -1;
}
void AddEvent(MEVENT hDbEvent, uint32_t timestamp, bool unread);
void EditEvent(MEVENT hDbEvent, uint32_t timestamp, bool unread);
void DeleteEvent(MEVENT hDbEvent);
void MarkRead(MEVENT hDbEvent);
};
struct CDbxSQLiteEventCursor : public DB::EventCursor
{
CDbxSQLiteEventCursor(MCONTACT _1, sqlite3* m_db, MEVENT hDbEvent, bool reverse = false);
~CDbxSQLiteEventCursor() override;
MEVENT FetchNext() override;
private:
sqlite3 *m_db;
sqlite3_stmt *cursor;
};
class CDbxSQLite : public MDatabaseCommon, public MIDatabaseChecker, public MZeroedObject
{
ptrW m_wszFileName;
sqlite3 *m_db = nullptr;
struct {
sqlite3_stmt *cur;
MCONTACT hContact;
MEVENT hEvent;
void clear() {
if (cur)
sqlite3_reset(cur);
memset(this, 0, sizeof(*this));
}
}
fwd, back;
DBCachedContact m_system;
struct Impl {
CDbxSQLite &pro;
CTimer m_timer;
void OnTimer(CTimer *pTimer)
{
pTimer->Stop();
pro.DBFlush(true);
}
Impl(CDbxSQLite &_p) :
pro(_p),
m_timer(Miranda_GetSystemWindow(), UINT_PTR(this))
{
m_timer.OnEvent = Callback(this, &Impl::OnTimer);
}
} m_impl;
bool m_safetyMode, m_bReadOnly, m_bShared, m_bTranStarted;
// contacts
void InitContacts();
CQuery qCntCount, qCntAdd, qCntDel, qCntDelSettings, qCntDelEvents, qCntDelEventSrt;
// encryption
void InitEncryption();
CQuery qCryptGetMode, qCryptSetMode, qCryptGetProvider, qCryptSetProvider, qCryptGetKey, qCryptSetKey, qCryptEnc1, qCryptEnc2;
// events
LIST<char> m_modules;
void InitEvents();
void UninitEvents();
CQuery qEvCount, qEvAdd, qEvDel, qEvEdit, qEvBlobSize, qEvGet, qEvGetFlags, qEvSetFlags, qEvGetContact;
CQuery qEvFindFirst, qEvFindNext, qEvFindLast, qEvFindPrev, qEvFindUnread, qEvGetById, qEvAddSrt, qEvDelSrt, qEvMetaSplit, qEvMetaMerge;
int DeleteEventMain(MEVENT);
int DeleteEventSrt(MEVENT);
// settings
void InitSettings();
CQuery qSettModules, qSettWrite, qSettDel, qSettEnum, qSettChanges;
int DeleteContactSettingWorker(MCONTACT contactID, LPCSTR szModule, LPCSTR szSetting);
void DBFlush(bool bForce = false);
sqlite3_stmt* InitQuery(const char *szQuery, CQuery &stmt);
public:
CDbxSQLite(const wchar_t *pwszFileName, bool bReadOnly, bool bShared);
~CDbxSQLite();
int Create();
int Check();
int Load();
STDMETHODIMP_(BOOL) IsRelational(void) override;
STDMETHODIMP_(void) SetCacheSafetyMode(BOOL) override;
STDMETHODIMP_(int) GetContactCount(void) override;
STDMETHODIMP_(int) DeleteContact(MCONTACT contactID) override;
STDMETHODIMP_(MCONTACT) AddContact(void) override;
STDMETHODIMP_(BOOL) IsDbContact(MCONTACT contactID) override;
STDMETHODIMP_(int) GetContactSize(void) override;
STDMETHODIMP_(int) GetEventCount(MCONTACT contactID) override;
STDMETHODIMP_(MEVENT) AddEvent(MCONTACT contactID, const DBEVENTINFO *dbe) override;
STDMETHODIMP_(BOOL) DeleteEvent(MEVENT hDbEvent) override;
STDMETHODIMP_(BOOL) EditEvent(MCONTACT contactID, MEVENT hDbEvent, const DBEVENTINFO *dbe) override;
STDMETHODIMP_(int) GetBlobSize(MEVENT hDbEvent) override;
STDMETHODIMP_(BOOL) GetEvent(MEVENT hDbEvent, DBEVENTINFO *dbe) override;
STDMETHODIMP_(BOOL) MarkEventRead(MCONTACT contactID, MEVENT hDbEvent) override;
STDMETHODIMP_(MCONTACT) GetEventContact(MEVENT hDbEvent) override;
STDMETHODIMP_(MEVENT) FindFirstEvent(MCONTACT contactID) override;
STDMETHODIMP_(MEVENT) FindFirstUnreadEvent(MCONTACT contactID) override;
STDMETHODIMP_(MEVENT) FindLastEvent(MCONTACT contactID) override;
STDMETHODIMP_(MEVENT) FindNextEvent(MCONTACT contactID, MEVENT hDbEvent) override;
STDMETHODIMP_(MEVENT) FindPrevEvent(MCONTACT contactID, MEVENT hDbEvent) override;
STDMETHODIMP_(MEVENT) GetEventById(LPCSTR szModule, LPCSTR szId) override;
STDMETHODIMP_(BOOL) EnumModuleNames(DBMODULEENUMPROC pFunc, void *pParam) override;
STDMETHODIMP_(BOOL) ReadCryptoKey(MBinBuffer&) override;
STDMETHODIMP_(BOOL) StoreCryptoKey() override;
STDMETHODIMP_(CRYPTO_PROVIDER*) ReadProvider() override;
STDMETHODIMP_(BOOL) StoreProvider(CRYPTO_PROVIDER*) override;
STDMETHODIMP_(BOOL) EnableEncryption(BOOL) override;
STDMETHODIMP_(BOOL) ReadEncryption() override;
STDMETHODIMP_(BOOL) WriteContactSettingWorker(MCONTACT contactID, DBCONTACTWRITESETTING &dbcws) override;
STDMETHODIMP_(BOOL) DeleteContactSetting(MCONTACT contactID, LPCSTR szModule, LPCSTR szSetting) override;
STDMETHODIMP_(BOOL) EnumContactSettings(MCONTACT hContact, DBSETTINGENUMPROC pfnEnumProc, const char *szModule, void *param) override;
STDMETHODIMP_(BOOL) MetaMergeHistory(DBCachedContact *ccMeta, DBCachedContact *ccSub) override;
STDMETHODIMP_(BOOL) MetaSplitHistory(DBCachedContact *ccMeta, DBCachedContact *ccSub) override;
STDMETHODIMP_(BOOL) Compact() override;
STDMETHODIMP_(BOOL) Backup(LPCWSTR) override;
STDMETHODIMP_(BOOL) Flush() override;
STDMETHODIMP_(DATABASELINK*) GetDriver() override;
STDMETHODIMP_(DB::EventCursor*) EventCursor(MCONTACT hContact, MEVENT hDbEvent) override;
STDMETHODIMP_(DB::EventCursor*) EventCursorRev(MCONTACT hContact, MEVENT hDbEvent) override;
////////////////////////////////////////////////////////////////////////////////////////
// database checker interface implementation
protected:
STDMETHODIMP_(MIDatabaseChecker *) GetChecker() override
{ return this;
}
STDMETHODIMP_(BOOL) Start(DBCHeckCallback *callback) override
{
cb = callback;
return ERROR_SUCCESS;
}
STDMETHODIMP_(BOOL) CheckDb(int phase) override;
STDMETHODIMP_(VOID) Destroy() override
{}
DBCHeckCallback *cb;
int CheckPhase1(void);
int CheckPhase2(void);
int CheckPhase3(void);
int CheckPhase4(void);
};
|
