1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
/****************************************************************************
**
** Copyright (C) 2019 The Qt Company Ltd.
** Contact: https://www.qt.io/licensing/
**
** This file is part of the QtWebEngine module 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 "proxy_server.h"
#include <QDataStream>
#include <QTcpSocket>
#include <QDebug>
ProxyServer::ProxyServer(QObject *parent) : QObject(parent)
{
connect(&m_server, &QTcpServer::newConnection, this, &ProxyServer::handleNewConnection);
}
void ProxyServer::setPort(int port)
{
m_port = port;
}
void ProxyServer::setCredentials(const QByteArray &user, const QByteArray password)
{
m_auth.append(user);
m_auth.append(':');
m_auth.append(password);
m_auth = m_auth.toBase64();
m_authenticate = true;
}
void ProxyServer::setCookie(const QByteArray &cookie)
{
m_cookie.append(QByteArrayLiteral("Cookie: "));
m_cookie.append(cookie);
}
bool ProxyServer::isListening()
{
return m_server.isListening();
}
void ProxyServer::run()
{
if (!m_server.listen(QHostAddress::LocalHost, m_port))
qFatal("Could not start the test server");
}
void ProxyServer::handleNewConnection()
{
// do one connection at the time
Q_ASSERT(m_data.isEmpty());
QTcpSocket *socket = m_server.nextPendingConnection();
Q_ASSERT(socket);
connect(socket, &QAbstractSocket::disconnected, socket, &QObject::deleteLater);
connect(socket, &QAbstractSocket::readyRead, this, &ProxyServer::handleReadReady);
}
void ProxyServer::handleReadReady()
{
QTcpSocket *socket = qobject_cast<QTcpSocket*>(sender());
Q_ASSERT(socket);
m_data.append(socket->readAll());
if (!m_data.endsWith("\r\n\r\n"))
return;
if (m_authenticate && !m_data.contains(QByteArrayLiteral("Proxy-Authorization: Basic"))) {
socket->write("HTTP/1.1 407 Proxy Authentication Required\nProxy-Authenticate: "
"Basic realm=\"Proxy requires authentication\"\r\n"
"content-length: 0\r\n"
"\r\n");
return;
}
if (m_authenticate && m_data.contains(m_auth)) {
emit authenticationSuccess();
}
if (m_data.contains(m_cookie)) {
emit cookieMatch();
}
m_data.clear();
emit requestReceived();
}
#include "moc_proxy_server.cpp"
|