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
|
// SPDX-FileCopyrightText: 2022 Jonah Brüchert <jbb@kaidan.im>
// SPDX-FileCopyrightText: 2022 Linus Jahn <lnj@kaidan.im>
//
// SPDX-License-Identifier: LGPL-2.1-or-later
#include "QXmppEncryptedFileSharingProvider.h"
#include "QXmppFileEncryption.h"
#include "QXmppFileMetadata.h"
#include "QXmppFileSharingManager.h"
#include "QXmppFutureUtils_p.h"
#include "QXmppUtils.h"
#include "QcaInitializer_p.h"
#include <QMimeDatabase>
using namespace QXmpp;
using namespace QXmpp::Private;
///
/// \class QXmppEncryptedFileSharingProvider
///
/// Encrypts or decrypts files on the fly when uploading or downloading.
///
/// \since QXmpp 1.5
///
class QXmppEncryptedFileSharingProviderPrivate
{
public:
QcaInitializer init;
QXmppFileSharingManager *manager;
std::shared_ptr<QXmppFileSharingProvider> uploadBaseProvider;
};
///
/// \brief Create a new QXmppEncryptedFileSharingProvider
///
/// \param manager QXmppFileSharingManager to be used to find other providers for downloading
/// encrypted files.
/// \param uploadBaseProvider Provider to be used for uploading the encrypted files.
///
QXmppEncryptedFileSharingProvider::QXmppEncryptedFileSharingProvider(
QXmppFileSharingManager *manager,
std::shared_ptr<QXmppFileSharingProvider> uploadBaseProvider)
: d(std::make_unique<QXmppEncryptedFileSharingProviderPrivate>())
{
d->manager = manager;
d->uploadBaseProvider = std::move(uploadBaseProvider);
}
QXmppEncryptedFileSharingProvider::~QXmppEncryptedFileSharingProvider() = default;
auto QXmppEncryptedFileSharingProvider::downloadFile(const std::any &source,
std::unique_ptr<QIODevice> target,
std::function<void(quint64, quint64)> reportProgress,
std::function<void(DownloadResult)> reportFinished)
-> std::shared_ptr<Download>
{
QXmppEncryptedFileSource encryptedSource;
try {
encryptedSource = std::any_cast<QXmppEncryptedFileSource>(source);
} catch (const std::bad_any_cast &) {
qFatal("QXmppEncryptedFileSharingProvider::downloadFile can only handle QXmppEncryptedFileSource sources");
}
auto output = std::make_unique<Encryption::DecryptionDevice>(std::move(target), encryptedSource.cipher(), encryptedSource.iv(), encryptedSource.key());
// find provider for source of encrypted file
std::any httpSource = encryptedSource.httpSources().front();
if (auto provider = d->manager->providerForSource(httpSource)) {
return provider->downloadFile(httpSource, std::move(output), std::move(reportProgress), std::move(reportFinished));
}
reportFinished(QXmppError { QStringLiteral("No basic file sharing provider available for encrypted file."), {} });
return {};
}
auto QXmppEncryptedFileSharingProvider::uploadFile(std::unique_ptr<QIODevice> data,
const QXmppFileMetadata &,
std::function<void(quint64, quint64)> reportProgress,
std::function<void(UploadResult)> reportFinished)
-> std::shared_ptr<Upload>
{
auto cipher = Aes256CbcPkcs7;
auto key = Encryption::generateKey(cipher);
auto iv = Encryption::generateInitializationVector(cipher);
auto encDevice = std::make_unique<Encryption::EncryptionDevice>(std::move(data), cipher, key, iv);
auto encryptedSize = encDevice->size();
QXmppFileMetadata metadata;
metadata.setFilename(QXmppUtils::generateStanzaHash(10));
metadata.setMediaType(QMimeDatabase().mimeTypeForName("application/octet-stream"));
metadata.setSize(encryptedSize);
// find provider for source of encrypted file
Q_ASSERT(d->uploadBaseProvider);
return d->uploadBaseProvider->uploadFile(
std::move(encDevice),
metadata,
std::move(reportProgress),
[=, reportFinished = std::move(reportFinished)](UploadResult result) {
auto encryptedResult = visitForward<UploadResult>(std::move(result), [&](std::any httpSourceAny) {
QXmppEncryptedFileSource encryptedSource;
encryptedSource.setKey(key);
encryptedSource.setIv(iv);
encryptedSource.setHttpSources({ std::any_cast<QXmppHttpFileSource>(std::move(httpSourceAny)) });
return encryptedSource;
});
reportFinished(std::move(encryptedResult));
});
}
|
