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
|
// SPDX-FileCopyrightText: 2021 Tobias Fella <fella@posteo.de>
// SPDX-License-Identifier: LGPL-2.1-or-later
#pragma once
#include <QtCore/QObject>
#include <QtSql/QSqlQuery>
#include <QtCore/QVector>
#include <QtCore/QHash>
#include "e2ee/e2ee.h"
namespace Quotient {
class QUOTIENT_API Database : public QObject
{
Q_OBJECT
public:
Database(const QString& matrixId, const QString& deviceId, QObject* parent);
int version();
void transaction();
void commit();
QSqlQuery execute(const QString &queryString);
QSqlQuery execute(QSqlQuery &query);
QSqlDatabase database();
QSqlQuery prepareQuery(const QString& quaryString);
QByteArray accountPickle();
void setAccountPickle(const QByteArray &pickle);
void clear();
void saveOlmSession(const QString& senderKey, const QString& sessionId,
const QByteArray& pickle, const QDateTime& timestamp);
UnorderedMap<QString, std::vector<QOlmSessionPtr>> loadOlmSessions(
const PicklingMode& picklingMode);
UnorderedMap<QString, QOlmInboundGroupSessionPtr> loadMegolmSessions(
const QString& roomId, const PicklingMode& picklingMode);
void saveMegolmSession(const QString& roomId, const QString& sessionId,
const QByteArray& pickle, const QString& senderId,
const QString& olmSessionId);
void addGroupSessionIndexRecord(const QString& roomId,
const QString& sessionId, uint32_t index,
const QString& eventId, qint64 ts);
std::pair<QString, qint64> groupSessionIndexRecord(const QString& roomId,
const QString& sessionId,
qint64 index);
void clearRoomData(const QString& roomId);
void setOlmSessionLastReceived(const QString& sessionId,
const QDateTime& timestamp);
QOlmOutboundGroupSessionPtr loadCurrentOutboundMegolmSession(
const QString& roomId, const PicklingMode& picklingMode);
void saveCurrentOutboundMegolmSession(
const QString& roomId, const PicklingMode& picklingMode,
const QOlmOutboundGroupSession& session);
void updateOlmSession(const QString& senderKey, const QString& sessionId,
const QByteArray& pickle);
// Returns a map UserId -> [DeviceId] that have not received key yet
QMultiHash<QString, QString> devicesWithoutKey(
const QString& roomId, QMultiHash<QString, QString> devices,
const QString& sessionId);
// 'devices' contains tuples {userId, deviceId, curveKey}
void setDevicesReceivedKey(
const QString& roomId,
const QVector<std::tuple<QString, QString, QString>>& devices,
const QString& sessionId, int index);
bool isSessionVerified(const QString& edKey);
void setSessionVerified(const QString& edKeyId);
private:
void migrateTo1();
void migrateTo2();
void migrateTo3();
void migrateTo4();
void migrateTo5();
QString m_matrixId;
};
} // namespace Quotient
|
