| 1 | /* |
| 2 | This file is part of KDE. |
| 3 | |
| 4 | SPDX-FileCopyrightText: 1999 Matthias Kalle Dalheimer <kalle@kde.org> |
| 5 | SPDX-FileCopyrightText: 2000 Charles Samuels <charles@kde.org> |
| 6 | SPDX-FileCopyrightText: 2005 Joseph Wenninger <kde@jowenn.at> |
| 7 | SPDX-FileCopyrightText: 2009 Frederik Gladhorn <gladhorn@kde.org> |
| 8 | |
| 9 | SPDX-License-Identifier: LGPL-2.1-only OR LGPL-3.0-only OR LicenseRef-KDE-Accepted-LGPL |
| 10 | */ |
| 11 | |
| 12 | #include "postfiledata.h" |
| 13 | |
| 14 | #include <QDateTime> |
| 15 | #include <QDebug> |
| 16 | #include <QRandomGenerator> |
| 17 | |
| 18 | namespace Attica |
| 19 | { |
| 20 | class PostFileDataPrivate |
| 21 | { |
| 22 | public: |
| 23 | QByteArray buffer; |
| 24 | QByteArray boundary; |
| 25 | QUrl url; |
| 26 | bool finished = false; |
| 27 | }; |
| 28 | |
| 29 | PostFileData::PostFileData(const QUrl &url) |
| 30 | : d(std::make_unique<PostFileDataPrivate>()) |
| 31 | { |
| 32 | d->url = url; |
| 33 | d->boundary = "----------" + randomString(length: 42 + 13).toLatin1(); |
| 34 | } |
| 35 | |
| 36 | PostFileData::~PostFileData() = default; |
| 37 | |
| 38 | QString PostFileData::randomString(int length) |
| 39 | { |
| 40 | if (length <= 0) { |
| 41 | return QString(); |
| 42 | } |
| 43 | |
| 44 | QString str; |
| 45 | str.resize(size: length); |
| 46 | int i = 0; |
| 47 | auto *generator = QRandomGenerator::global(); |
| 48 | while (length--) { |
| 49 | int r = generator->bounded(highest: 62); |
| 50 | r += 48; |
| 51 | if (r > 57) { |
| 52 | r += 7; |
| 53 | } |
| 54 | if (r > 90) { |
| 55 | r += 6; |
| 56 | } |
| 57 | str[i++] = QLatin1Char(char(r)); |
| 58 | } |
| 59 | return str; |
| 60 | } |
| 61 | |
| 62 | void PostFileData::addArgument(const QString &key, const QString &value) |
| 63 | { |
| 64 | if (d->finished) { |
| 65 | qWarning() << "PostFileData::addFile: should not add data after calling request() or data()" ; |
| 66 | } |
| 67 | QByteArray data( |
| 68 | "--" + d->boundary + "\r\n" |
| 69 | "Content-Disposition: form-data; name=\"" + key.toLatin1() + |
| 70 | "\"\r\n\r\n" + value.toUtf8() + "\r\n" ); |
| 71 | |
| 72 | d->buffer.append(a: data); |
| 73 | } |
| 74 | |
| 75 | /* |
| 76 | void PostFileData::addFile(const QString& fileName, QIODevice* file, const QString& mimeType) |
| 77 | { |
| 78 | if (d->finished) { |
| 79 | qCDebug(ATTICA) << "PostFileData::addFile: should not add data after calling request() or data()"; |
| 80 | } |
| 81 | QByteArray data = file->readAll(); |
| 82 | addFile(fileName, data, mimeType); |
| 83 | } |
| 84 | */ |
| 85 | |
| 86 | void PostFileData::addFile(const QString &fileName, const QByteArray &file, const QString &mimeType, const QString &fieldName) |
| 87 | { |
| 88 | if (d->finished) { |
| 89 | qWarning() << "PostFileData::addFile: should not add data after calling request() or data()" ; |
| 90 | } |
| 91 | |
| 92 | QByteArray data( |
| 93 | "--" + d->boundary + "\r\n" |
| 94 | "Content-Disposition: form-data; name=\"" ); |
| 95 | data.append(a: fieldName.toLatin1()); |
| 96 | data.append(a: "\"; filename=\"" + fileName.toUtf8() + "\"\r\nContent-Type: " + mimeType.toLatin1() + "\r\n\r\n" ); |
| 97 | |
| 98 | d->buffer.append(a: data); |
| 99 | d->buffer.append(a: file + QByteArray("\r\n" )); |
| 100 | } |
| 101 | |
| 102 | QNetworkRequest PostFileData::request() |
| 103 | { |
| 104 | if (!d->finished) { |
| 105 | finish(); |
| 106 | } |
| 107 | QNetworkRequest request; |
| 108 | request.setUrl(d->url); |
| 109 | request.setHeader(header: QNetworkRequest::ContentTypeHeader, value: QByteArray("multipart/form-data; boundary=" + d->boundary)); |
| 110 | request.setHeader(header: QNetworkRequest::ContentLengthHeader, value: d->buffer.length()); |
| 111 | return request; |
| 112 | } |
| 113 | |
| 114 | QByteArray PostFileData::data() |
| 115 | { |
| 116 | if (!d->finished) { |
| 117 | finish(); |
| 118 | } |
| 119 | return d->buffer; |
| 120 | } |
| 121 | |
| 122 | void PostFileData::finish() |
| 123 | { |
| 124 | Q_ASSERT(!d->finished); |
| 125 | d->finished = true; |
| 126 | d->buffer.append(a: "--" + d->boundary + "--" ); |
| 127 | } |
| 128 | |
| 129 | } |
| 130 | |