File: tst_bench_qhttpserver_transfer.cpp

package info (click to toggle)
qt6-httpserver 6.8.2-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 1,036 kB
  • sloc: cpp: 7,856; makefile: 20
file content (160 lines) | stat: -rw-r--r-- 5,283 bytes parent folder | download
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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
// Copyright (C) 2024 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only

#include <QtTest/qtest.h>

#include <QtHttpServer/qhttpserver.h>

#include <QtCore/qobject.h>
#include <QtCore/qbytearray.h>
#include <QtCore/qbuffer.h>

#include <QtNetwork/qnetworkaccessmanager.h>
#if QT_CONFIG(ssl)
#include <QtNetwork/qsslcertificate.h>
#include <QtNetwork/qsslconfiguration.h>
#include <QtNetwork/qsslkey.h>
#include <QtNetwork/qsslserver.h>
#include <QtNetwork/qsslsocket.h>
#endif

using namespace Qt::StringLiterals;
using namespace std::chrono_literals;

class tst_bench_QHttpServer_transfer : public QObject
{
    Q_OBJECT
public:
    tst_bench_QHttpServer_transfer();
private slots:
    void transferPayload_data();
    void transferPayload();
private:
    QHttpServer server;
    bool testHttp2 = false;
    bool testTls = false;
    quint16 ports[2] = {0, 0};
};

tst_bench_QHttpServer_transfer::tst_bench_QHttpServer_transfer()
{
    auto *tcpServer = new QTcpServer(&server);
    if (!tcpServer->listen())
        qFatal("Failed to listen on TCP");
    ports[0] = tcpServer->serverPort();
    server.bind(tcpServer);
    qDebug() << "Test HTTP on port" << ports[0];
#if QT_CONFIG(ssl)
    if (QSslSocket::supportsSsl()) {
        testHttp2 = QSslSocket::isFeatureSupported(QSsl::SupportedFeature::ServerSideAlpn);
        testTls = true;
        auto *sslServer = new QSslServer(&server);
        sslServer->setHandshakeTimeout(std::numeric_limits<int>::max());
        QSslConfiguration sslConfig = sslServer->sslConfiguration();
        sslConfig.setPeerVerifyMode(QSslSocket::VerifyNone);
        if (testHttp2)
            sslConfig.setAllowedNextProtocols({ QSslConfiguration::ALPNProtocolHTTP2 });
        const auto certs = QSslCertificate::fromPath(u":/cert/localhost.cert"_s);
        if (certs.isEmpty())
            qFatal("Failed to load certificate");
        sslConfig.setLocalCertificate(certs.first());
        QFile key(u":/cert/localhost.key"_s);
        if (!key.open(QIODevice::ReadOnly))
            qFatal("Failed to open key file");
        sslConfig.setPrivateKey(QSslKey(&key, QSsl::KeyAlgorithm::Rsa));
        sslServer->setSslConfiguration(sslConfig);
        if (!sslServer->listen(QHostAddress::LocalHost))
            qFatal("Failed to listen on TLS");
        ports[1] = sslServer->serverPort();
        server.bind(sslServer);
        qDebug() << "Testing TLS with" << QSslSocket::sslLibraryBuildVersionString();
        qDebug() << "Testing HTTPS on port" << ports[1];
    }
#endif

    QByteArray bigdata = "1"_ba.repeated(1024 * 1024 * 100);
    using HTTP = QHttpServerRequest::Method;
    server.route("/bytearray", HTTP::Get,
                 [=](const QHttpServerRequest &req, QHttpServerResponder &resp) {
                     resp.write(bigdata, "application/octet-stream");
                 });

    server.route("/qbuffer", HTTP::Get,
                 [=](const QHttpServerRequest &req, QHttpServerResponder &resp) {
                     auto *buf = new QBuffer;
                     buf->setData(bigdata);
                     buf->open(QIODevice::ReadOnly);
                     resp.write(buf, "application/octet-stream");
                 });
}

enum TransferProtocol {
    HTTP,
    HTTPS,
    HTTP2,
};

void tst_bench_QHttpServer_transfer::transferPayload_data()
{
    QTest::addColumn<QString>("path");
    QTest::addColumn<TransferProtocol>("transferProtocol");

    QTest::addRow("bytearray") << "/bytearray" << HTTP;
    QTest::addRow("qbuffer") << "/qbuffer" << HTTP;
    if (testHttp2) {
        QTest::addRow("bytearray-http2") << "/bytearray" << HTTP2;
        QTest::addRow("qbuffer-http2") << "/qbuffer" << HTTP2;
    }
    if (testTls) {
        QTest::addRow("bytearray-tls") << "/bytearray" << HTTPS;
        QTest::addRow("qbuffer-tls") << "/qbuffer" << HTTPS;
    }
}

void tst_bench_QHttpServer_transfer::transferPayload()
{
    QFETCH(QString, path);
    QFETCH(TransferProtocol, transferProtocol);
    const bool tls = transferProtocol != HTTP;

    QNetworkAccessManager qnam;
    QNetworkRequest req;
    QUrl url;
    url.setHost("127.0.0.1");
    url.setScheme(tls ? "https" : "http");
    url.setPort(ports[tls ? 1 : 0]);
    url.setPath(path);
    req.setUrl(url);

    req.setAttribute(QNetworkRequest::Http2AllowedAttribute, transferProtocol == HTTP2);

#if QT_CONFIG(ssl)
    if (tls) {
        QSslConfiguration sslConfig = req.sslConfiguration();
        sslConfig.setPeerVerifyMode(QSslSocket::VerifyNone);
        req.setSslConfiguration(sslConfig);
    }
#endif

    QBENCHMARK {
        QNetworkReply *reply = qnam.get(req);
        QObject::connect(reply, &QNetworkReply::sslErrors, reply,
                         qOverload<>(&QNetworkReply::ignoreSslErrors));
        qint64 total = 0;
        const auto replyReady = [&reply]() -> bool {
            return reply->isFinished() || reply->bytesAvailable();
        };
        while (!replyReady()) {
            bool res = QTest::qWaitFor(replyReady, 5s);
            QVERIFY(res);
            total += reply->readAll().size();
        }
        QCOMPARE(total, 1024 * 1024 * 100);
        delete reply;
    }

}

QTEST_MAIN(tst_bench_QHttpServer_transfer)

#include "tst_bench_qhttpserver_transfer.moc"