summaryrefslogtreecommitdiffstats
path: root/tests/auto/shared/httpreqrep.cpp
blob: eb2db68901ac4bb72cf2c9f45440c893c44b8a35 (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
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
/****************************************************************************
**
** Copyright (C) 2017 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 "httpreqrep.h"

HttpReqRep::HttpReqRep(QTcpSocket *socket, QObject *parent)
    : QObject(parent), m_socket(socket)
{
    m_socket->setParent(this);
    connect(m_socket, &QIODevice::readyRead, this, &HttpReqRep::handleReadyRead);
}

void HttpReqRep::sendResponse()
{
    m_socket->write("HTTP/1.1 ");
    m_socket->write(QByteArray::number(m_responseStatusCode));
    m_socket->write(" OK?\r\n");
    for (const auto & kv : m_responseHeaders) {
        m_socket->write(kv.first);
        m_socket->write(": ");
        m_socket->write(kv.second);
        m_socket->write("\r\n");
    }
    m_socket->write("\r\n");
    m_socket->write(m_responseBody);
    m_socket->close();
}

QByteArray HttpReqRep::requestHeader(const QByteArray &key) const
{
    auto it = m_requestHeaders.find(key);
    if (it != m_requestHeaders.end())
        return it->second;
    return {};
}

void HttpReqRep::handleReadyRead()
{
    const auto requestLine = m_socket->readLine();
    const auto requestLineParts = requestLine.split(' ');
    if (requestLineParts.size() != 3 || !requestLineParts[2].toUpper().startsWith("HTTP/")) {
        qWarning("HttpReqRep: invalid request line");
        Q_EMIT readFinished(false);
        return;
    }

    decltype(m_requestHeaders) headers;
    for (;;) {
        const auto headerLine = m_socket->readLine();
        if (headerLine == QByteArrayLiteral("\r\n"))
            break;
        int colonIndex = headerLine.indexOf(':');
        if (colonIndex < 0) {
            qWarning("HttpReqRep: invalid header line");
            Q_EMIT readFinished(false);
            return;
        }
        auto headerKey = headerLine.left(colonIndex).trimmed().toLower();
        auto headerValue = headerLine.mid(colonIndex + 1).trimmed().toLower();
        headers.emplace(headerKey, headerValue);
    }

    m_requestMethod = requestLineParts[0];
    m_requestPath = requestLineParts[1];
    m_requestHeaders = headers;
    Q_EMIT readFinished(true);
}