| 1 | /**************************************************************************** |
| 2 | ** |
| 3 | ** Copyright (C) 2016 The Qt Company Ltd. |
| 4 | ** Contact: https://www.qt.io/licensing/ |
| 5 | ** |
| 6 | ** This file is part of the examples of the Qt Toolkit. |
| 7 | ** |
| 8 | ** $QT_BEGIN_LICENSE:BSD$ |
| 9 | ** Commercial License Usage |
| 10 | ** Licensees holding valid commercial Qt licenses may use this file in |
| 11 | ** accordance with the commercial license agreement provided with the |
| 12 | ** Software or, alternatively, in accordance with the terms contained in |
| 13 | ** a written agreement between you and The Qt Company. For licensing terms |
| 14 | ** and conditions see https://www.qt.io/terms-conditions. For further |
| 15 | ** information use the contact form at https://www.qt.io/contact-us. |
| 16 | ** |
| 17 | ** BSD License Usage |
| 18 | ** Alternatively, you may use this file under the terms of the BSD license |
| 19 | ** as follows: |
| 20 | ** |
| 21 | ** "Redistribution and use in source and binary forms, with or without |
| 22 | ** modification, are permitted provided that the following conditions are |
| 23 | ** met: |
| 24 | ** * Redistributions of source code must retain the above copyright |
| 25 | ** notice, this list of conditions and the following disclaimer. |
| 26 | ** * Redistributions in binary form must reproduce the above copyright |
| 27 | ** notice, this list of conditions and the following disclaimer in |
| 28 | ** the documentation and/or other materials provided with the |
| 29 | ** distribution. |
| 30 | ** * Neither the name of The Qt Company Ltd nor the names of its |
| 31 | ** contributors may be used to endorse or promote products derived |
| 32 | ** from this software without specific prior written permission. |
| 33 | ** |
| 34 | ** |
| 35 | ** THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| 36 | ** "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| 37 | ** LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| 38 | ** A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| 39 | ** OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| 40 | ** SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| 41 | ** LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| 42 | ** DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| 43 | ** THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| 44 | ** (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| 45 | ** OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE." |
| 46 | ** |
| 47 | ** $QT_END_LICENSE$ |
| 48 | ** |
| 49 | ****************************************************************************/ |
| 50 | |
| 51 | #include "httpwindow.h" |
| 52 | |
| 53 | #include "ui_authenticationdialog.h" |
| 54 | |
| 55 | #include <QtWidgets> |
| 56 | #include <QtNetwork> |
| 57 | #include <QUrl> |
| 58 | |
| 59 | #if QT_CONFIG(ssl) |
| 60 | const char defaultUrl[] = "https://www.qt.io/" ; |
| 61 | #else |
| 62 | const char defaultUrl[] = "http://www.qt.io/" ; |
| 63 | #endif |
| 64 | const char defaultFileName[] = "index.html" ; |
| 65 | |
| 66 | ProgressDialog::ProgressDialog(const QUrl &url, QWidget *parent) |
| 67 | : QProgressDialog(parent) |
| 68 | { |
| 69 | setWindowTitle(tr("Download Progress" )); |
| 70 | setWindowFlags(windowFlags() & ~Qt::WindowContextHelpButtonHint); |
| 71 | setLabelText(tr("Downloading %1." ).arg(url.toDisplayString())); |
| 72 | setMinimum(0); |
| 73 | setValue(0); |
| 74 | setMinimumDuration(0); |
| 75 | setMinimumSize(QSize(400, 75)); |
| 76 | } |
| 77 | |
| 78 | ProgressDialog::~ProgressDialog() |
| 79 | { |
| 80 | } |
| 81 | |
| 82 | void ProgressDialog::networkReplyProgress(qint64 bytesRead, qint64 totalBytes) |
| 83 | { |
| 84 | setMaximum(totalBytes); |
| 85 | setValue(bytesRead); |
| 86 | } |
| 87 | |
| 88 | HttpWindow::HttpWindow(QWidget *parent) |
| 89 | : QDialog(parent) |
| 90 | , statusLabel(new QLabel(tr("Please enter the URL of a file you want to download.\n\n" ), this)) |
| 91 | , urlLineEdit(new QLineEdit(defaultUrl)) |
| 92 | , downloadButton(new QPushButton(tr("Download" ))) |
| 93 | , launchCheckBox(new QCheckBox("Launch file" )) |
| 94 | , defaultFileLineEdit(new QLineEdit(defaultFileName)) |
| 95 | , downloadDirectoryLineEdit(new QLineEdit) |
| 96 | , reply(nullptr) |
| 97 | , file(nullptr) |
| 98 | , httpRequestAborted(false) |
| 99 | { |
| 100 | setWindowFlags(windowFlags() & ~Qt::WindowContextHelpButtonHint); |
| 101 | setWindowTitle(tr("HTTP" )); |
| 102 | |
| 103 | connect(&qnam, &QNetworkAccessManager::authenticationRequired, |
| 104 | this, &HttpWindow::slotAuthenticationRequired); |
| 105 | #ifndef QT_NO_SSL |
| 106 | connect(&qnam, &QNetworkAccessManager::sslErrors, |
| 107 | this, &HttpWindow::sslErrors); |
| 108 | #endif |
| 109 | |
| 110 | QFormLayout *formLayout = new QFormLayout; |
| 111 | urlLineEdit->setClearButtonEnabled(true); |
| 112 | connect(urlLineEdit, &QLineEdit::textChanged, |
| 113 | this, &HttpWindow::enableDownloadButton); |
| 114 | formLayout->addRow(tr("&URL:" ), urlLineEdit); |
| 115 | QString downloadDirectory = QStandardPaths::writableLocation(QStandardPaths::TempLocation); |
| 116 | if (downloadDirectory.isEmpty() || !QFileInfo(downloadDirectory).isDir()) |
| 117 | downloadDirectory = QDir::currentPath(); |
| 118 | downloadDirectoryLineEdit->setText(QDir::toNativeSeparators(downloadDirectory)); |
| 119 | formLayout->addRow(tr("&Download directory:" ), downloadDirectoryLineEdit); |
| 120 | formLayout->addRow(tr("Default &file:" ), defaultFileLineEdit); |
| 121 | launchCheckBox->setChecked(true); |
| 122 | formLayout->addRow(launchCheckBox); |
| 123 | |
| 124 | QVBoxLayout *mainLayout = new QVBoxLayout(this); |
| 125 | mainLayout->addLayout(formLayout); |
| 126 | |
| 127 | mainLayout->addItem(new QSpacerItem(0, 0, QSizePolicy::Ignored, QSizePolicy::MinimumExpanding)); |
| 128 | |
| 129 | statusLabel->setWordWrap(true); |
| 130 | mainLayout->addWidget(statusLabel); |
| 131 | |
| 132 | downloadButton->setDefault(true); |
| 133 | connect(downloadButton, &QAbstractButton::clicked, this, &HttpWindow::downloadFile); |
| 134 | QPushButton *quitButton = new QPushButton(tr("Quit" )); |
| 135 | quitButton->setAutoDefault(false); |
| 136 | connect(quitButton, &QAbstractButton::clicked, this, &QWidget::close); |
| 137 | QDialogButtonBox *buttonBox = new QDialogButtonBox; |
| 138 | buttonBox->addButton(downloadButton, QDialogButtonBox::ActionRole); |
| 139 | buttonBox->addButton(quitButton, QDialogButtonBox::RejectRole); |
| 140 | mainLayout->addWidget(buttonBox); |
| 141 | |
| 142 | urlLineEdit->setFocus(); |
| 143 | } |
| 144 | |
| 145 | HttpWindow::~HttpWindow() |
| 146 | { |
| 147 | } |
| 148 | |
| 149 | void HttpWindow::startRequest(const QUrl &requestedUrl) |
| 150 | { |
| 151 | url = requestedUrl; |
| 152 | httpRequestAborted = false; |
| 153 | |
| 154 | reply = qnam.get(QNetworkRequest(url)); |
| 155 | connect(reply, &QNetworkReply::finished, this, &HttpWindow::httpFinished); |
| 156 | connect(reply, &QIODevice::readyRead, this, &HttpWindow::httpReadyRead); |
| 157 | |
| 158 | ProgressDialog *progressDialog = new ProgressDialog(url, this); |
| 159 | progressDialog->setAttribute(Qt::WA_DeleteOnClose); |
| 160 | connect(progressDialog, &QProgressDialog::canceled, this, &HttpWindow::cancelDownload); |
| 161 | connect(reply, &QNetworkReply::downloadProgress, progressDialog, &ProgressDialog::networkReplyProgress); |
| 162 | connect(reply, &QNetworkReply::finished, progressDialog, &ProgressDialog::hide); |
| 163 | progressDialog->show(); |
| 164 | |
| 165 | statusLabel->setText(tr("Downloading %1..." ).arg(url.toString())); |
| 166 | } |
| 167 | |
| 168 | void HttpWindow::downloadFile() |
| 169 | { |
| 170 | const QString urlSpec = urlLineEdit->text().trimmed(); |
| 171 | if (urlSpec.isEmpty()) |
| 172 | return; |
| 173 | |
| 174 | const QUrl newUrl = QUrl::fromUserInput(urlSpec); |
| 175 | if (!newUrl.isValid()) { |
| 176 | QMessageBox::information(this, tr("Error" ), |
| 177 | tr("Invalid URL: %1: %2" ).arg(urlSpec, newUrl.errorString())); |
| 178 | return; |
| 179 | } |
| 180 | |
| 181 | QString fileName = newUrl.fileName(); |
| 182 | if (fileName.isEmpty()) |
| 183 | fileName = defaultFileLineEdit->text().trimmed(); |
| 184 | if (fileName.isEmpty()) |
| 185 | fileName = defaultFileName; |
| 186 | QString downloadDirectory = QDir::cleanPath(downloadDirectoryLineEdit->text().trimmed()); |
| 187 | bool useDirectory = !downloadDirectory.isEmpty() && QFileInfo(downloadDirectory).isDir(); |
| 188 | if (useDirectory) |
| 189 | fileName.prepend(downloadDirectory + '/'); |
| 190 | if (QFile::exists(fileName)) { |
| 191 | if (QMessageBox::question(this, tr("Overwrite Existing File" ), |
| 192 | tr("There already exists a file called %1%2." |
| 193 | " Overwrite?" ) |
| 194 | .arg(fileName, |
| 195 | useDirectory |
| 196 | ? QString() |
| 197 | : QStringLiteral(" in the current directory" )), |
| 198 | QMessageBox::Yes | QMessageBox::No, |
| 199 | QMessageBox::No) |
| 200 | == QMessageBox::No) { |
| 201 | return; |
| 202 | } |
| 203 | QFile::remove(fileName); |
| 204 | } |
| 205 | |
| 206 | file = openFileForWrite(fileName); |
| 207 | if (!file) |
| 208 | return; |
| 209 | |
| 210 | downloadButton->setEnabled(false); |
| 211 | |
| 212 | // schedule the request |
| 213 | startRequest(newUrl); |
| 214 | } |
| 215 | |
| 216 | std::unique_ptr<QFile> HttpWindow::openFileForWrite(const QString &fileName) |
| 217 | { |
| 218 | std::unique_ptr<QFile> file(new QFile(fileName)); |
| 219 | if (!file->open(QIODevice::WriteOnly)) { |
| 220 | QMessageBox::information(this, tr("Error" ), |
| 221 | tr("Unable to save the file %1: %2." ) |
| 222 | .arg(QDir::toNativeSeparators(fileName), |
| 223 | file->errorString())); |
| 224 | return nullptr; |
| 225 | } |
| 226 | return file; |
| 227 | } |
| 228 | |
| 229 | void HttpWindow::cancelDownload() |
| 230 | { |
| 231 | statusLabel->setText(tr("Download canceled." )); |
| 232 | httpRequestAborted = true; |
| 233 | reply->abort(); |
| 234 | downloadButton->setEnabled(true); |
| 235 | } |
| 236 | |
| 237 | void HttpWindow::httpFinished() |
| 238 | { |
| 239 | QFileInfo fi; |
| 240 | if (file) { |
| 241 | fi.setFile(file->fileName()); |
| 242 | file->close(); |
| 243 | file.reset(); |
| 244 | } |
| 245 | |
| 246 | if (httpRequestAborted) { |
| 247 | reply->deleteLater(); |
| 248 | reply = nullptr; |
| 249 | return; |
| 250 | } |
| 251 | |
| 252 | if (reply->error()) { |
| 253 | QFile::remove(fi.absoluteFilePath()); |
| 254 | statusLabel->setText(tr("Download failed:\n%1." ).arg(reply->errorString())); |
| 255 | downloadButton->setEnabled(true); |
| 256 | reply->deleteLater(); |
| 257 | reply = nullptr; |
| 258 | return; |
| 259 | } |
| 260 | |
| 261 | const QVariant redirectionTarget = reply->attribute(QNetworkRequest::RedirectionTargetAttribute); |
| 262 | |
| 263 | reply->deleteLater(); |
| 264 | reply = nullptr; |
| 265 | |
| 266 | if (!redirectionTarget.isNull()) { |
| 267 | const QUrl redirectedUrl = url.resolved(redirectionTarget.toUrl()); |
| 268 | if (QMessageBox::question(this, tr("Redirect" ), |
| 269 | tr("Redirect to %1 ?" ).arg(redirectedUrl.toString()), |
| 270 | QMessageBox::Yes | QMessageBox::No) == QMessageBox::No) { |
| 271 | QFile::remove(fi.absoluteFilePath()); |
| 272 | downloadButton->setEnabled(true); |
| 273 | statusLabel->setText(tr("Download failed:\nRedirect rejected." )); |
| 274 | return; |
| 275 | } |
| 276 | file = openFileForWrite(fi.absoluteFilePath()); |
| 277 | if (!file) { |
| 278 | downloadButton->setEnabled(true); |
| 279 | return; |
| 280 | } |
| 281 | startRequest(redirectedUrl); |
| 282 | return; |
| 283 | } |
| 284 | |
| 285 | statusLabel->setText(tr("Downloaded %1 bytes to %2\nin\n%3" ) |
| 286 | .arg(fi.size()).arg(fi.fileName(), QDir::toNativeSeparators(fi.absolutePath()))); |
| 287 | if (launchCheckBox->isChecked()) |
| 288 | QDesktopServices::openUrl(QUrl::fromLocalFile(fi.absoluteFilePath())); |
| 289 | downloadButton->setEnabled(true); |
| 290 | } |
| 291 | |
| 292 | void HttpWindow::httpReadyRead() |
| 293 | { |
| 294 | // this slot gets called every time the QNetworkReply has new data. |
| 295 | // We read all of its new data and write it into the file. |
| 296 | // That way we use less RAM than when reading it at the finished() |
| 297 | // signal of the QNetworkReply |
| 298 | if (file) |
| 299 | file->write(reply->readAll()); |
| 300 | } |
| 301 | |
| 302 | void HttpWindow::enableDownloadButton() |
| 303 | { |
| 304 | downloadButton->setEnabled(!urlLineEdit->text().isEmpty()); |
| 305 | } |
| 306 | |
| 307 | void HttpWindow::slotAuthenticationRequired(QNetworkReply *, QAuthenticator *authenticator) |
| 308 | { |
| 309 | QDialog authenticationDialog; |
| 310 | Ui::Dialog ui; |
| 311 | ui.setupUi(&authenticationDialog); |
| 312 | authenticationDialog.adjustSize(); |
| 313 | ui.siteDescription->setText(tr("%1 at %2" ).arg(authenticator->realm(), url.host())); |
| 314 | |
| 315 | // Did the URL have information? Fill the UI |
| 316 | // This is only relevant if the URL-supplied credentials were wrong |
| 317 | ui.userEdit->setText(url.userName()); |
| 318 | ui.passwordEdit->setText(url.password()); |
| 319 | |
| 320 | if (authenticationDialog.exec() == QDialog::Accepted) { |
| 321 | authenticator->setUser(ui.userEdit->text()); |
| 322 | authenticator->setPassword(ui.passwordEdit->text()); |
| 323 | } |
| 324 | } |
| 325 | |
| 326 | #ifndef QT_NO_SSL |
| 327 | void HttpWindow::sslErrors(QNetworkReply *, const QList<QSslError> &errors) |
| 328 | { |
| 329 | QString errorString; |
| 330 | for (const QSslError &error : errors) { |
| 331 | if (!errorString.isEmpty()) |
| 332 | errorString += '\n'; |
| 333 | errorString += error.errorString(); |
| 334 | } |
| 335 | |
| 336 | if (QMessageBox::warning(this, tr("SSL Errors" ), |
| 337 | tr("One or more SSL errors has occurred:\n%1" ).arg(errorString), |
| 338 | QMessageBox::Ignore | QMessageBox::Abort) == QMessageBox::Ignore) { |
| 339 | reply->ignoreSslErrors(); |
| 340 | } |
| 341 | } |
| 342 | #endif |
| 343 | |