1 | // Copyright (C) 2024 The Qt Company Ltd. |
2 | // SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only |
3 | |
4 | #ifndef QHttpServerHttp2ProtocolHandler_H |
5 | #define QHttpServerHttp2ProtocolHandler_H |
6 | |
7 | #include <QtHttpServer/qthttpserverglobal.h> |
8 | #include <QtHttpServer/qhttpserverrequest.h> |
9 | #include <QtHttpServer/private/qhttpserverstream_p.h> |
10 | #include <QtNetwork/private/hpack_p.h> |
11 | #include <QtCore/qbytearray.h> |
12 | #include <QtCore/qqueue.h> |
13 | |
14 | // |
15 | // W A R N I N G |
16 | // ------------- |
17 | // |
18 | // This file is not part of the Qt API. It exists for the convenience |
19 | // of QHttpServer. This header file may change from version to |
20 | // version without notice, or even be removed. |
21 | // |
22 | // We mean it. |
23 | |
24 | QT_REQUIRE_CONFIG(http); |
25 | QT_REQUIRE_CONFIG(ssl); |
26 | |
27 | QT_BEGIN_NAMESPACE |
28 | |
29 | class QTcpSocket; |
30 | class QAbstractHttpServer; |
31 | class QHttp2Connection; |
32 | class QHttp2Stream; |
33 | |
34 | struct QHttpServerHttp2Queue |
35 | { |
36 | QQueue<QByteArray> data; |
37 | HPack::HttpHeader trailers; |
38 | bool allEnqueued = false; |
39 | }; |
40 | |
41 | class QHttpServerHttp2ProtocolHandler : public QHttpServerStream |
42 | { |
43 | Q_OBJECT |
44 | |
45 | friend class QAbstractHttpServerPrivate; |
46 | |
47 | private: |
48 | QHttpServerHttp2ProtocolHandler(QAbstractHttpServer *server, QIODevice *socket); |
49 | |
50 | void responderDestroyed() final; |
51 | void startHandlingRequest() final; |
52 | void socketDisconnected() final; |
53 | |
54 | void write(const QByteArray &body, const QHttpHeaders &, |
55 | QHttpServerResponder::StatusCode status, quint32 streamId) final; |
56 | void write(QHttpServerResponder::StatusCode status, quint32 streamId) final; |
57 | void write(QIODevice *data, const QHttpHeaders &, |
58 | QHttpServerResponder::StatusCode status, quint32 streamId) final; |
59 | void writeBeginChunked(const QHttpHeaders &, |
60 | QHttpServerResponder::StatusCode status, |
61 | quint32 streamId) final; |
62 | void writeChunk(const QByteArray &body, quint32 streamId) final; |
63 | void writeEndChunked(const QByteArray &data, |
64 | const QHttpHeaders &trailers, |
65 | quint32 streamId) final; |
66 | |
67 | void writeHeadersAndStatus(const QHttpHeaders &, |
68 | QHttpServerResponder::StatusCode status, |
69 | bool endStream, |
70 | quint32 streamId); |
71 | |
72 | private slots: |
73 | void onStreamCreated(QHttp2Stream *stream); |
74 | void onStreamClosed(quint32 streamId); |
75 | void onStreamHalfClosed(quint32 streamId); |
76 | void sendToStream(quint32 streamId); |
77 | |
78 | private: |
79 | QHttp2Stream * getStream(quint32 streamId) const; |
80 | void enqueueChunk(const QByteArray &body, bool allEnqueued, const QHttpHeaders &trailers, |
81 | quint32 streamId); |
82 | |
83 | QAbstractHttpServer *m_server; |
84 | QIODevice *m_socket; |
85 | QTcpSocket *m_tcpSocket; |
86 | QHttpServerRequest m_request; |
87 | QHttp2Connection *m_connection; |
88 | QHash<quint32, QList<QMetaObject::Connection>> m_streamConnections; |
89 | QHash<quint32, QHttpServerHttp2Queue> m_streamQueue; |
90 | qint32 m_responderCounter = 0; |
91 | }; |
92 | |
93 | QT_END_NAMESPACE |
94 | |
95 | #endif // QHttpServerHttp2ProtocolHandler_H |
96 | |