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
|
// SPDX-FileCopyrightText: 2021 Carl Schwan <carlschwan@kde.org>
//
// SPDX-License-Identifier: LGPL-2.1-or-later
#include "e2ee/qolminboundsession.h"
#include <iostream>
using namespace Quotient;
QOlmError lastError(OlmInboundGroupSession *session) {
return fromString(olm_inbound_group_session_last_error(session));
}
QOlmInboundGroupSession::QOlmInboundGroupSession(OlmInboundGroupSession *session)
: m_groupSession(session)
{
}
QOlmInboundGroupSession::~QOlmInboundGroupSession()
{
olm_clear_inbound_group_session(m_groupSession);
//delete[](reinterpret_cast<uint8_t *>(m_groupSession));
}
std::unique_ptr<QOlmInboundGroupSession> QOlmInboundGroupSession::create(const QByteArray &key)
{
const auto olmInboundGroupSession = olm_inbound_group_session(new uint8_t[olm_inbound_group_session_size()]);
const auto error = olm_init_inbound_group_session(olmInboundGroupSession,
reinterpret_cast<const uint8_t *>(key.constData()), key.size());
if (error == olm_error()) {
throw lastError(olmInboundGroupSession);
}
return std::make_unique<QOlmInboundGroupSession>(olmInboundGroupSession);
}
std::unique_ptr<QOlmInboundGroupSession> QOlmInboundGroupSession::import(const QByteArray &key)
{
const auto olmInboundGroupSession = olm_inbound_group_session(new uint8_t[olm_inbound_group_session_size()]);
QByteArray keyBuf = key;
const auto error = olm_import_inbound_group_session(olmInboundGroupSession,
reinterpret_cast<const uint8_t *>(keyBuf.data()), keyBuf.size());
if (error == olm_error()) {
throw lastError(olmInboundGroupSession);
}
return std::make_unique<QOlmInboundGroupSession>(olmInboundGroupSession);
}
QByteArray toKey(const PicklingMode &mode)
{
if (std::holds_alternative<Unencrypted>(mode)) {
return "";
}
return std::get<Encrypted>(mode).key;
}
QByteArray QOlmInboundGroupSession::pickle(const PicklingMode &mode) const
{
QByteArray pickledBuf(olm_pickle_inbound_group_session_length(m_groupSession), '0');
const QByteArray key = toKey(mode);
const auto error = olm_pickle_inbound_group_session(m_groupSession, key.data(), key.length(), pickledBuf.data(),
pickledBuf.length());
if (error == olm_error()) {
throw lastError(m_groupSession);
}
return pickledBuf;
}
std::variant<std::unique_ptr<QOlmInboundGroupSession>, QOlmError> QOlmInboundGroupSession::unpickle(const QByteArray &pickled, const PicklingMode &mode)
{
QByteArray pickledBuf = pickled;
const auto groupSession = olm_inbound_group_session(new uint8_t[olm_inbound_group_session_size()]);
QByteArray key = toKey(mode);
const auto error = olm_unpickle_inbound_group_session(groupSession, key.data(), key.length(),
pickledBuf.data(), pickledBuf.size());
if (error == olm_error()) {
return lastError(groupSession);
}
key.clear();
return std::make_unique<QOlmInboundGroupSession>(groupSession);
}
std::variant<std::pair<QString, uint32_t>, QOlmError> QOlmInboundGroupSession::decrypt(const QByteArray &message)
{
// This is for capturing the output of olm_group_decrypt
uint32_t messageIndex = 0;
// We need to clone the message because
// olm_decrypt_max_plaintext_length destroys the input buffer
QByteArray messageBuf(message.length(), '0');
std::copy(message.begin(), message.end(), messageBuf.begin());
QByteArray plaintextBuf(olm_group_decrypt_max_plaintext_length(m_groupSession,
reinterpret_cast<uint8_t *>(messageBuf.data()), messageBuf.length()), '0');
messageBuf = QByteArray(message.length(), '0');
std::copy(message.begin(), message.end(), messageBuf.begin());
const auto plaintextLen = olm_group_decrypt(m_groupSession, reinterpret_cast<uint8_t *>(messageBuf.data()),
messageBuf.length(), reinterpret_cast<uint8_t *>(plaintextBuf.data()), plaintextBuf.length(), &messageIndex);
// Error code or plaintext length is returned
const auto decryptError = plaintextLen;
if (decryptError == olm_error()) {
return lastError(m_groupSession);
}
QByteArray output(plaintextLen, '0');
std::memcpy(output.data(), plaintextBuf.data(), plaintextLen);
return std::make_pair<QString, qint32>(QString(output), messageIndex);
}
std::variant<QByteArray, QOlmError> QOlmInboundGroupSession::exportSession(uint32_t messageIndex)
{
const auto keyLength = olm_export_inbound_group_session_length(m_groupSession);
QByteArray keyBuf(keyLength, '0');
const auto error = olm_export_inbound_group_session(m_groupSession, reinterpret_cast<uint8_t *>(keyBuf.data()), keyLength, messageIndex);
if (error == olm_error()) {
return lastError(m_groupSession);
}
return keyBuf;
}
uint32_t QOlmInboundGroupSession::firstKnownIndex() const
{
return olm_inbound_group_session_first_known_index(m_groupSession);
}
QByteArray QOlmInboundGroupSession::sessionId() const
{
QByteArray sessionIdBuf(olm_inbound_group_session_id_length(m_groupSession), '0');
const auto error = olm_inbound_group_session_id(m_groupSession, reinterpret_cast<uint8_t *>(sessionIdBuf.data()),
sessionIdBuf.length());
if (error == olm_error()) {
throw lastError(m_groupSession);
}
return sessionIdBuf;
}
bool QOlmInboundGroupSession::isVerified() const
{
return olm_inbound_group_session_is_verified(m_groupSession) != 0;
}
|
