135 lines
4.5 KiB
C++
135 lines
4.5 KiB
C++
/*
|
|
This file is part of the PhantomJS project from Ofi Labs.
|
|
|
|
Copyright (C) 2011 Ariya Hidayat <ariya.hidayat@gmail.com>
|
|
Copyright (C) 2011 Ivan De Marino <ivan.de.marino@gmail.com>
|
|
|
|
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 <organization> 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 <COPYRIGHT HOLDER> 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.
|
|
*/
|
|
|
|
#ifndef NETWORKACCESSMANAGER_H
|
|
#define NETWORKACCESSMANAGER_H
|
|
|
|
#include <QNetworkAccessManager>
|
|
#include <QNetworkReply>
|
|
#include <QSslConfiguration>
|
|
#include <QTimer>
|
|
#include <QStringList>
|
|
|
|
class Config;
|
|
class QAuthenticator;
|
|
class QNetworkDiskCache;
|
|
class QSslConfiguration;
|
|
|
|
class TimeoutTimer : public QTimer
|
|
{
|
|
Q_OBJECT
|
|
|
|
public:
|
|
TimeoutTimer(QObject* parent = 0);
|
|
QNetworkReply* reply;
|
|
QVariantMap data;
|
|
};
|
|
|
|
class JsNetworkRequest : public QObject
|
|
{
|
|
Q_OBJECT
|
|
|
|
public:
|
|
JsNetworkRequest(QNetworkRequest* request, QObject* parent = 0);
|
|
Q_INVOKABLE void abort();
|
|
Q_INVOKABLE void changeUrl(const QString& url);
|
|
Q_INVOKABLE bool setHeader(const QString& name, const QVariant& value);
|
|
|
|
private:
|
|
QNetworkRequest* m_networkRequest;
|
|
};
|
|
|
|
class NoFileAccessReply : public QNetworkReply
|
|
{
|
|
Q_OBJECT
|
|
|
|
public:
|
|
NoFileAccessReply(QObject* parent, const QNetworkRequest& req, const QNetworkAccessManager::Operation op);
|
|
~NoFileAccessReply();
|
|
void abort() {}
|
|
protected:
|
|
qint64 readData(char*, qint64) { return -1; }
|
|
};
|
|
|
|
class NetworkAccessManager : public QNetworkAccessManager
|
|
{
|
|
Q_OBJECT
|
|
public:
|
|
NetworkAccessManager(QObject* parent, const Config* config);
|
|
void setUserName(const QString& userName);
|
|
void setPassword(const QString& password);
|
|
void setMaxAuthAttempts(int maxAttempts);
|
|
void setResourceTimeout(int resourceTimeout);
|
|
void setCustomHeaders(const QVariantMap& headers);
|
|
QVariantMap customHeaders() const;
|
|
QStringList captureContent() const;
|
|
void setCaptureContent(const QStringList& patterns);
|
|
|
|
void setCookieJar(QNetworkCookieJar* cookieJar);
|
|
|
|
protected:
|
|
bool m_ignoreSslErrors;
|
|
bool m_localUrlAccessEnabled;
|
|
int m_authAttempts;
|
|
int m_maxAuthAttempts;
|
|
int m_resourceTimeout;
|
|
QString m_userName;
|
|
QString m_password;
|
|
QNetworkReply* createRequest(Operation op, const QNetworkRequest& req, QIODevice* outgoingData = 0);
|
|
void handleFinished(QNetworkReply* reply, const QVariant& status, const QVariant& statusText);
|
|
|
|
signals:
|
|
void resourceRequested(const QVariant& data, QObject*);
|
|
void resourceReceived(const QVariant& data);
|
|
void resourceError(const QVariant& data);
|
|
void resourceTimeout(const QVariant& data);
|
|
|
|
private slots:
|
|
void handleStarted();
|
|
void handleFinished(QNetworkReply* reply);
|
|
void provideAuthentication(QNetworkReply* reply, QAuthenticator* authenticator);
|
|
void handleSslErrors(const QList<QSslError>& errors);
|
|
void handleNetworkError();
|
|
void handleTimeout();
|
|
|
|
private:
|
|
void prepareSslConfiguration(const Config* config);
|
|
QVariantList getHeadersFromReply(const QNetworkReply* reply);
|
|
|
|
QHash<QNetworkReply*, int> m_ids;
|
|
QSet<QNetworkReply*> m_started;
|
|
int m_idCounter;
|
|
QNetworkDiskCache* m_networkDiskCache;
|
|
QVariantMap m_customHeaders;
|
|
QSslConfiguration m_sslConfiguration;
|
|
};
|
|
|
|
#endif // NETWORKACCESSMANAGER_H
|