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
|
// SPDX-FileCopyrightText: Tobias Fella <fella@posteo.de>
// SPDX-License-Identifier: LGPL-2.1-or-later
#include "mxcreply.h"
#include <QtCore/QBuffer>
#include "accountregistry.h"
#include "room.h"
#ifdef Quotient_E2EE_ENABLED
#include "events/filesourceinfo.h"
#endif
using namespace Quotient;
class MxcReply::Private
{
public:
explicit Private(QNetworkReply* r = nullptr)
: m_reply(r)
{}
QNetworkReply* m_reply;
Omittable<EncryptedFileMetadata> m_encryptedFile;
QIODevice* m_device = nullptr;
};
MxcReply::MxcReply(QNetworkReply* reply)
: d(makeImpl<Private>(reply))
{
d->m_device = d->m_reply;
reply->setParent(this);
connect(d->m_reply, &QNetworkReply::finished, this, [this]() {
setError(d->m_reply->error(), d->m_reply->errorString());
setOpenMode(ReadOnly);
Q_EMIT finished();
});
}
MxcReply::MxcReply(QNetworkReply* reply, Room* room, const QString &eventId)
: d(makeImpl<Private>(reply))
{
reply->setParent(this);
connect(d->m_reply, &QNetworkReply::finished, this, [this]() {
setError(d->m_reply->error(), d->m_reply->errorString());
#ifdef Quotient_E2EE_ENABLED
if(!d->m_encryptedFile.has_value()) {
d->m_device = d->m_reply;
} else {
auto buffer = new QBuffer(this);
buffer->setData(
decryptFile(d->m_reply->readAll(), *d->m_encryptedFile));
buffer->open(ReadOnly);
d->m_device = buffer;
}
#else
d->m_device = d->m_reply;
#endif
setOpenMode(ReadOnly);
emit finished();
});
#ifdef Quotient_E2EE_ENABLED
auto eventIt = room->findInTimeline(eventId);
if(eventIt != room->historyEdge()) {
if (auto event = eventIt->viewAs<RoomMessageEvent>()) {
if (auto* efm = std::get_if<EncryptedFileMetadata>(
&event->content()->fileInfo()->source))
d->m_encryptedFile = *efm;
}
}
#endif
}
MxcReply::MxcReply()
: d(ZeroImpl<Private>())
{
static const auto BadRequestPhrase = tr("Bad Request");
QMetaObject::invokeMethod(this, [this]() {
setAttribute(QNetworkRequest::HttpStatusCodeAttribute, 400);
setAttribute(QNetworkRequest::HttpReasonPhraseAttribute,
BadRequestPhrase);
setError(QNetworkReply::ProtocolInvalidOperationError,
BadRequestPhrase);
setFinished(true);
emit errorOccurred(QNetworkReply::ProtocolInvalidOperationError);
emit finished();
}, Qt::QueuedConnection);
}
qint64 MxcReply::readData(char *data, qint64 maxSize)
{
return d->m_device->read(data, maxSize);
}
void MxcReply::abort()
{
d->m_reply->abort();
}
|
