blob: a7bbd6934f0e9eb40932bae257efaf73b745c206 (
plain)
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
|
// Copyright (C) 2018 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only
#include "proxyserver.h"
#include <QDataStream>
#include <QTcpSocket>
#include <QDebug>
ProxyServer::ProxyServer(QObject *parent) : QObject(parent),
m_port(5555)
{
connect(&m_server, &QTcpServer::newConnection, this, &ProxyServer::handleNewConnection);
}
void ProxyServer::setPort(int port)
{
m_port = port;
}
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());
//simply wait for whole request
if (!m_data.endsWith("\r\n\r\n"))
return;
// add fake proxy authetication
socket->write("HTTP/1.1 407 Proxy Auth Required\nProxy-Authenticate: "
"Basic realm=\"Proxy requires authentication\"\r\n\r\n");
m_data.clear();
socket->disconnectFromHost();
emit requestReceived();
}
|