| /**************************************************************************** |
| ** |
| ** Copyright (C) 2016 The Qt Company Ltd. |
| ** Contact: https://www.qt.io/licensing/ |
| ** |
| ** This file is part of the test suite of the Qt Toolkit. |
| ** |
| ** $QT_BEGIN_LICENSE:GPL-EXCEPT$ |
| ** 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. |
| ** |
| ** GNU General Public License Usage |
| ** Alternatively, this file may be used under the terms of the GNU |
| ** General Public License version 3 as published by the Free Software |
| ** Foundation with exceptions as appearing in the file LICENSE.GPL3-EXCEPT |
| ** included in the packaging of this file. Please review the following |
| ** information to ensure the GNU General Public License requirements will |
| ** be met: https://www.gnu.org/licenses/gpl-3.0.html. |
| ** |
| ** $QT_END_LICENSE$ |
| ** |
| ****************************************************************************/ |
| |
| #include "qget.h" |
| |
| #include <QNetworkProxy> |
| #include <QDebug> |
| #include <QCoreApplication> |
| #include <QList> |
| #include <QStringList> |
| #include <QNetworkConfiguration> |
| #include <QNetworkConfigurationManager> |
| #include <QNetworkSession> |
| |
| void printShortUsage() |
| { |
| qDebug() << QCoreApplication::applicationName() << " [options] [list of urls]" << endl |
| << "Get one or more urls using QNetworkAccessManager" << endl |
| << "--help to display detailed usage" << endl; |
| } |
| |
| void printUsage() |
| { |
| qDebug() << QCoreApplication::applicationName() << " [options] [list of urls]" << endl |
| << "Get one or more urls using QNetworkAccessManager" << endl |
| << "Options:" |
| << "--help This message" << endl |
| << "--user=<username> Set username to use for authentication" << endl |
| << "--password=<password> Set password to use for authentication" << endl |
| << "--proxy-user=<username> Set username to use for proxy authentication" << endl |
| << "--proxy-password=<password> Set password to use for proxy authentication" << endl |
| << "--proxy=on Use system proxy (default)" << endl |
| << "--proxy=off Don't use system proxy" << endl |
| << "--proxy=<host:port>[,type] Use specified proxy" << endl |
| << " ,http HTTP proxy (default)" << endl |
| << " ,socks SOCKS5 proxy" << endl |
| << " ,ftp FTP proxy" << endl |
| << " ,httpcaching HTTP caching proxy (no CONNECT method)" << endl |
| << "--headers=filename Set request headers from file contents" << endl |
| << "--post=filename upload the file to the next url using HTTP POST" << endl |
| << "--put=filename upload the file to the next url using HTTP PUT" << endl |
| << "--content-type=<MIME> set content-type header for upload" << endl |
| << "--serial don't run requests in parallel" << endl; |
| } |
| |
| int main(int argc, char *argv[]) |
| { |
| QCoreApplication app(argc, argv); |
| if (argc < 2) { |
| printShortUsage(); |
| return EXIT_FAILURE; |
| } |
| |
| //use system proxy (by default) |
| QNetworkProxyFactory::setUseSystemConfiguration(true); |
| |
| DownloadManager dl; |
| QString uploadFileName; |
| QString contentType; |
| QString httpUser; |
| QString httpPassword; |
| QString headersFile; |
| TransferItem::Method method = TransferItem::Get; |
| //arguments match wget where possible |
| foreach (QString str, app.arguments().mid(1)) { |
| if (str == "--help") |
| printUsage(); |
| else if (str.startsWith("--user=")) |
| httpUser = str.mid(7); |
| else if (str.startsWith("--password=")) |
| httpPassword = str.mid(11); |
| else if (str.startsWith("--proxy-user=")) |
| dl.setProxyUser(str.mid(13)); |
| else if (str.startsWith("--proxy-password=")) |
| dl.setProxyPassword(str.mid(17)); |
| else if (str == "--proxy=off") |
| QNetworkProxyFactory::setUseSystemConfiguration(false); |
| else if (str == "--proxy=on") |
| QNetworkProxyFactory::setUseSystemConfiguration(true); |
| else if (str.startsWith("--proxy=")) { |
| //parse "--proxy=host:port[,type]" |
| QNetworkProxy proxy; |
| str = str.mid(8); |
| int sep = str.indexOf(':'); |
| proxy.setHostName(str.left(sep)); |
| str = str.mid(sep + 1); |
| sep = str.indexOf(','); |
| QString port; |
| if (sep < 0) { |
| port = str; |
| proxy.setType(QNetworkProxy::HttpProxy); |
| } else { |
| port = str.left(sep); |
| str = str.mid(sep + 1); |
| if (str == "socks") |
| proxy.setType(QNetworkProxy::Socks5Proxy); |
| else if (str == "ftp") |
| proxy.setType(QNetworkProxy::FtpCachingProxy); |
| else if (str == "httpcaching") |
| proxy.setType(QNetworkProxy::HttpCachingProxy); |
| else if (str == "http") |
| proxy.setType(QNetworkProxy::HttpProxy); |
| else { |
| qDebug() << "unknown proxy type"; |
| return EXIT_FAILURE; |
| } |
| } |
| bool ok; |
| quint16 p = port.toUShort(&ok); |
| if (!ok) { |
| qDebug() << "couldn't parse proxy"; |
| return EXIT_FAILURE; |
| } |
| proxy.setPort(p); |
| qDebug() << "proxy:" << proxy.hostName() << proxy.port() << proxy.type(); |
| dl.setProxy(proxy); |
| } |
| else if (str.startsWith("--put=")) { |
| method = TransferItem::Put; |
| uploadFileName = str.mid(6); |
| } |
| else if (str.startsWith("--post=")) { |
| method = TransferItem::Post; |
| uploadFileName = str.mid(7); |
| } |
| else if (str.startsWith("--content-type=")) |
| contentType=str.mid(15); |
| else if (str.startsWith("--headers=")) |
| headersFile=str.mid(10); |
| else if (str == "--serial") |
| dl.setQueueMode(DownloadManager::Serial); |
| else if (str.startsWith(QLatin1Char('-'))) |
| qDebug() << "unsupported option" << str; |
| else { |
| QUrl url(QUrl::fromUserInput(str)); |
| QNetworkRequest request(url); |
| //set headers |
| if (!headersFile.isEmpty()) { |
| QFile f(headersFile); |
| if (!f.open(QFile::ReadOnly | QFile::Text)) { |
| qDebug() << "can't open headers file: " << headersFile; |
| } else { |
| while (!f.atEnd()) { |
| QByteArray line = f.readLine().trimmed(); |
| if (line.isEmpty()) break; |
| int colon = line.indexOf(':'); |
| qDebug() << line; |
| if (colon > 0 && colon < line.length() - 2) { |
| request.setRawHeader(line.left(colon), line.mid(colon+2)); |
| } |
| } |
| f.close(); |
| } |
| } |
| if (!contentType.isEmpty()) |
| request.setHeader(QNetworkRequest::ContentTypeHeader, contentType); |
| |
| switch (method) { |
| case TransferItem::Put: |
| case TransferItem::Post: |
| dl.upload(request, httpUser, httpPassword, uploadFileName, method); |
| break; |
| case TransferItem::Get: |
| dl.get(request, httpUser, httpPassword); |
| break; |
| } |
| method = TransferItem::Get; //default for urls without a request type before it |
| } |
| } |
| QMetaObject::invokeMethod(&dl, "checkForAllDone", Qt::QueuedConnection); |
| return app.exec(); |
| } |
