qt5base-lts/tests/manual/qnetworkaccessmanager/qget/qget.cpp
Shane Kearns 32adb82741 Enable specifying raw headers for the request
Use "--headers=file" where the file contains the raw headers to send.
This is useful for replaying requests from log files.

Change-Id: I3bbe582d96fc9797f692a0d5772e8164f8265ce0
Reviewed-by: Martin Petersson <Martin.Petersson@nokia.com>
2012-05-05 03:50:41 +02:00

211 lines
8.6 KiB
C++

/****************************************************************************
**
** Copyright (C) 2012 Nokia Corporation and/or its subsidiary(-ies).
** Contact: http://www.qt-project.org/
**
** This file is part of the test suite of the Qt Toolkit.
**
** $QT_BEGIN_LICENSE:LGPL$
** GNU Lesser General Public License Usage
** This file may be used under the terms of the GNU Lesser General Public
** License version 2.1 as published by the Free Software Foundation and
** appearing in the file LICENSE.LGPL included in the packaging of this
** file. Please review the following information to ensure the GNU Lesser
** General Public License version 2.1 requirements will be met:
** http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
**
** In addition, as a special exception, Nokia gives you certain additional
** rights. These rights are described in the Nokia Qt LGPL Exception
** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
**
** GNU General Public License Usage
** Alternatively, this file may be used under the terms of the GNU General
** Public License version 3.0 as published by the Free Software Foundation
** and appearing in the file LICENSE.GPL included in the packaging of this
** file. Please review the following information to ensure the GNU General
** Public License version 3.0 requirements will be met:
** http://www.gnu.org/copyleft/gpl.html.
**
** Other Usage
** Alternatively, this file may be used in accordance with the terms and
** conditions contained in a signed written agreement between you and Nokia.
**
**
**
**
**
**
** $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("-"))
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();
}