| /**************************************************************************** |
| ** |
| ** Copyright (C) 2017 The Qt Company Ltd. |
| ** Contact: https://www.qt.io/licensing/ |
| ** |
| ** This file is part of the examples of the Qt Toolkit. |
| ** |
| ** $QT_BEGIN_LICENSE:BSD$ |
| ** Commercial License Usage |
| ** Licensees holding valid commercial Qt licenses may use this file in |
| ** accordance with the commercial license agreement provided with the |
| ** Software or, alternatively, in accordance with the terms contained in |
| ** a written agreement between you and The Qt Company. For licensing terms |
| ** and conditions see https://www.qt.io/terms-conditions. For further |
| ** information use the contact form at https://www.qt.io/contact-us. |
| ** |
| ** BSD License Usage |
| ** Alternatively, you may use this file under the terms of the BSD license |
| ** as follows: |
| ** |
| ** "Redistribution and use in source and binary forms, with or without |
| ** modification, are permitted provided that the following conditions are |
| ** met: |
| ** * Redistributions of source code must retain the above copyright |
| ** notice, this list of conditions and the following disclaimer. |
| ** * Redistributions in binary form must reproduce the above copyright |
| ** notice, this list of conditions and the following disclaimer in |
| ** the documentation and/or other materials provided with the |
| ** distribution. |
| ** * Neither the name of The Qt Company Ltd nor the names of its |
| ** contributors may be used to endorse or promote products derived |
| ** from this software without specific prior written permission. |
| ** |
| ** |
| ** THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| ** "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| ** LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| ** A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| ** OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| ** SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| ** LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| ** DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| ** THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| ** (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| ** OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE." |
| ** |
| ** $QT_END_LICENSE$ |
| ** |
| ****************************************************************************/ |
| |
| #include <QtCore> |
| #include <QtNetwork> |
| |
| #include <cstdio> |
| |
| QT_BEGIN_NAMESPACE |
| class QSslError; |
| QT_END_NAMESPACE |
| |
| using namespace std; |
| |
| class DownloadManager: public QObject |
| { |
| Q_OBJECT |
| QNetworkAccessManager manager; |
| QVector<QNetworkReply *> currentDownloads; |
| |
| public: |
| DownloadManager(); |
| void doDownload(const QUrl &url); |
| static QString saveFileName(const QUrl &url); |
| bool saveToDisk(const QString &filename, QIODevice *data); |
| static bool isHttpRedirect(QNetworkReply *reply); |
| |
| public slots: |
| void execute(); |
| void downloadFinished(QNetworkReply *reply); |
| void sslErrors(const QList<QSslError> &errors); |
| }; |
| |
| DownloadManager::DownloadManager() |
| { |
| connect(&manager, &QNetworkAccessManager::finished, |
| this, &DownloadManager::downloadFinished); |
| } |
| |
| void DownloadManager::doDownload(const QUrl &url) |
| { |
| QNetworkRequest request(url); |
| QNetworkReply *reply = manager.get(request); |
| |
| #if QT_CONFIG(ssl) |
| connect(reply, &QNetworkReply::sslErrors, |
| this, &DownloadManager::sslErrors); |
| #endif |
| |
| currentDownloads.append(reply); |
| } |
| |
| QString DownloadManager::saveFileName(const QUrl &url) |
| { |
| QString path = url.path(); |
| QString basename = QFileInfo(path).fileName(); |
| |
| if (basename.isEmpty()) |
| basename = "download"; |
| |
| if (QFile::exists(basename)) { |
| // already exists, don't overwrite |
| int i = 0; |
| basename += '.'; |
| while (QFile::exists(basename + QString::number(i))) |
| ++i; |
| |
| basename += QString::number(i); |
| } |
| |
| return basename; |
| } |
| |
| bool DownloadManager::saveToDisk(const QString &filename, QIODevice *data) |
| { |
| QFile file(filename); |
| if (!file.open(QIODevice::WriteOnly)) { |
| fprintf(stderr, "Could not open %s for writing: %s\n", |
| qPrintable(filename), |
| qPrintable(file.errorString())); |
| return false; |
| } |
| |
| file.write(data->readAll()); |
| file.close(); |
| |
| return true; |
| } |
| |
| bool DownloadManager::isHttpRedirect(QNetworkReply *reply) |
| { |
| int statusCode = reply->attribute(QNetworkRequest::HttpStatusCodeAttribute).toInt(); |
| return statusCode == 301 || statusCode == 302 || statusCode == 303 |
| || statusCode == 305 || statusCode == 307 || statusCode == 308; |
| } |
| |
| void DownloadManager::execute() |
| { |
| QStringList args = QCoreApplication::instance()->arguments(); |
| args.takeFirst(); // skip the first argument, which is the program's name |
| if (args.isEmpty()) { |
| printf("Qt Download example - downloads all URLs in parallel\n" |
| "Usage: download url1 [url2... urlN]\n" |
| "\n" |
| "Downloads the URLs passed in the command-line to the local directory\n" |
| "If the target file already exists, a .0, .1, .2, etc. is appended to\n" |
| "differentiate.\n"); |
| QCoreApplication::instance()->quit(); |
| return; |
| } |
| |
| for (const QString &arg : qAsConst(args)) { |
| QUrl url = QUrl::fromEncoded(arg.toLocal8Bit()); |
| doDownload(url); |
| } |
| } |
| |
| void DownloadManager::sslErrors(const QList<QSslError> &sslErrors) |
| { |
| #if QT_CONFIG(ssl) |
| for (const QSslError &error : sslErrors) |
| fprintf(stderr, "SSL error: %s\n", qPrintable(error.errorString())); |
| #else |
| Q_UNUSED(sslErrors); |
| #endif |
| } |
| |
| void DownloadManager::downloadFinished(QNetworkReply *reply) |
| { |
| QUrl url = reply->url(); |
| if (reply->error()) { |
| fprintf(stderr, "Download of %s failed: %s\n", |
| url.toEncoded().constData(), |
| qPrintable(reply->errorString())); |
| } else { |
| if (isHttpRedirect(reply)) { |
| fputs("Request was redirected.\n", stderr); |
| } else { |
| QString filename = saveFileName(url); |
| if (saveToDisk(filename, reply)) { |
| printf("Download of %s succeeded (saved to %s)\n", |
| url.toEncoded().constData(), qPrintable(filename)); |
| } |
| } |
| } |
| |
| currentDownloads.removeAll(reply); |
| reply->deleteLater(); |
| |
| if (currentDownloads.isEmpty()) { |
| // all downloads finished |
| QCoreApplication::instance()->quit(); |
| } |
| } |
| |
| int main(int argc, char **argv) |
| { |
| QCoreApplication app(argc, argv); |
| |
| DownloadManager manager; |
| QTimer::singleShot(0, &manager, SLOT(execute())); |
| |
| app.exec(); |
| } |
| |
| #include "main.moc" |