qt5base-lts/tests/manual/qnetworkaccessmanager/qget/transferitem.cpp
Shane Kearns a32d6834ad qget test - fix null pointer dereference
If the server sends a redirect with no body, the file is null

Change-Id: I49fd1d8a4cdd404497ebef4c7f3b478960776896
Reviewed-by: Richard J. Moore <rich@kde.org>
2012-06-27 00:46:35 +02:00

163 lines
5.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 <QDebug>
TransferItem::TransferItem(const QNetworkRequest &r, const QString &u, const QString &p, QNetworkAccessManager &n, Method m)
: method(m), request(r), reply(0), nam(n), inputFile(0), outputFile(0), user(u), password(p)
{
}
void TransferItem::progress(qint64 sent, qint64 total)
{
if (total > 0)
qDebug() << (sent*100/total) << "%";
else
qDebug() << sent << "B";
}
void TransferItem::start()
{
switch (method) {
case Get:
reply = nam.get(request);
connect(reply, SIGNAL(readyRead()), this, SLOT(readyRead()));
connect(reply, SIGNAL(downloadProgress(qint64,qint64)), this, SLOT(progress(qint64,qint64)));
break;
case Put:
reply = nam.put(request, inputFile);
connect(reply, SIGNAL(uploadProgress(qint64,qint64)), this, SLOT(progress(qint64,qint64)));
break;
case Post:
reply = nam.post(request, inputFile);
connect(reply, SIGNAL(uploadProgress(qint64,qint64)), this, SLOT(progress(qint64,qint64)));
break;
}
connect(reply, SIGNAL(finished()), this, SLOT(finished()));
}
DownloadItem::DownloadItem(const QNetworkRequest &r, const QString &user, const QString &password, QNetworkAccessManager &manager)
: TransferItem(r, user, password, manager, Get)
{
}
DownloadItem::~DownloadItem()
{
}
void DownloadItem::readyRead()
{
if (!outputFile)
outputFile = new QFile(this);
if (!outputFile->isOpen()) {
qDebug() << reply->header(QNetworkRequest::ContentTypeHeader) << reply->header(QNetworkRequest::ContentLengthHeader);
QString path = reply->url().path();
path = path.mid(path.lastIndexOf('/') + 1);
if (path.isEmpty())
path = QLatin1String("index.html");
outputFile->setFileName(path);
for (int i=1;i<1000;i++) {
if (!outputFile->exists() && outputFile->open(QIODevice::WriteOnly | QIODevice::Truncate))
break;
outputFile->setFileName(QString(QLatin1String("%1.%2")).arg(path).arg(i));
}
if (!outputFile->isOpen()) {
qDebug() << "couldn't open output file";
reply->abort();
return;
}
qDebug() << reply->url() << " -> " << outputFile->fileName();
}
outputFile->write(reply->readAll());
}
void DownloadItem::finished()
{
if (reply->attribute(QNetworkRequest::RedirectionTargetAttribute).isValid()) {
QUrl url = reply->attribute(QNetworkRequest::RedirectionTargetAttribute).toUrl();
url = reply->url().resolved(url);
qDebug() << reply->url() << "redirected to " << url;
if (redirects.contains(url)) {
qDebug() << "redirect loop detected";
} else if (redirects.count() > 10) {
qDebug() << "too many redirects";
} else {
//follow redirect
if (outputFile && outputFile->isOpen()) {
if (!outputFile->seek(0) || !outputFile->resize(0)) {
outputFile->close();
outputFile->remove();
}
}
reply->deleteLater();
reply = nam.get(QNetworkRequest(url));
reply->setParent(this);
connect(reply, SIGNAL(readyRead()), this, SLOT(readyRead()));
connect(reply, SIGNAL(finished()), this, SLOT(finished()));
redirects.append(url);
return;
}
}
if (outputFile && outputFile->isOpen()) {
outputFile->write(reply->readAll());
outputFile->close();
}
emit downloadFinished(this);
}
UploadItem::UploadItem(const QNetworkRequest &r, const QString &user, const QString &password, QNetworkAccessManager &n, QFile *f, TransferItem::Method method)
: TransferItem(r,user, password, n,method)
{
inputFile = f;
f->setParent(this);
qDebug() << f->fileName() << f->isOpen() << f->size();
}
UploadItem::~UploadItem()
{
}
void UploadItem::finished()
{
emit downloadFinished(this);
}