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
|
// Copyright (C) 2017 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only
#ifndef HTTPREQREP_H
#define HTTPREQREP_H
#include <QTcpSocket>
#include <map>
#include <utility>
// Represents an HTTP request-response exchange.
class HttpReqRep : public QObject
{
Q_OBJECT
public:
explicit HttpReqRep(QTcpSocket *socket, QObject *parent = nullptr);
Q_INVOKABLE void sendResponse(int statusCode = 200);
void sendResponse(const QByteArray &response);
void close();
bool isClosed() const { return m_state == State::DISCONNECTED; }
// Request parameters (only valid after requestReceived())
QByteArray requestMethod() const { return m_requestMethod; }
QByteArray requestPath() const { return m_requestPath; }
QByteArray requestHeader(const QByteArray &key) const;
bool hasRequestHeader(const QByteArray &key) const;
// Response parameters (can be set until sendResponse()/close()).
int responseStatus() const { return m_responseStatusCode; }
void setResponseStatus(int statusCode)
{
m_responseStatusCode = statusCode;
}
void setResponseHeader(const QByteArray &key, QByteArray value)
{
m_responseHeaders[key.toLower()] = std::move(value);
}
QByteArray responseBody() const { return m_responseBody; }
Q_INVOKABLE void setResponseBody(QByteArray content)
{
m_responseHeaders["content-length"] = QByteArray::number(content.size());
m_responseBody = std::move(content);
}
Q_SIGNALS:
// Emitted when the request has been correctly parsed.
void requestReceived();
// Emitted on first call to sendResponse().
void responseSent();
// Emitted when something goes wrong.
void error(const QString &error);
// Emitted during or some time after sendResponse() or close().
void closed();
private Q_SLOTS:
void handleReadyRead();
void handleDisconnected();
private:
enum class State {
// Waiting for first line of request.
RECEIVING_REQUEST, // Next: RECEIVING_HEADERS or DISCONNECTING.
// Waiting for header lines.
RECEIVING_HEADERS, // Next: REQUEST_RECEIVED or DISCONNECTING.
// Request parsing succeeded, waiting for sendResponse() or close().
REQUEST_RECEIVED, // Next: DISCONNECTING.
// Waiting for network.
DISCONNECTING, // Next: DISCONNECTED.
// Connection is dead.
DISCONNECTED, // Next: -
};
QTcpSocket *m_socket = nullptr;
State m_state = State::RECEIVING_REQUEST;
QByteArray m_requestMethod;
QByteArray m_requestPath;
std::map<QByteArray, QByteArray> m_requestHeaders;
int m_responseStatusCode = -1;
std::map<QByteArray, QByteArray> m_responseHeaders;
QByteArray m_responseBody;
};
#endif // !HTTPREQREP_H
|
