Line data Source code
1 : /*
2 : ___________________________________________
3 : | _ ___ _ |
4 : | | | |__ \ | | |
5 : | | |__ ) |__ _ __ _ ___ _ __ | |_ |
6 : | | '_ \ / // _` |/ _` |/ _ \ '_ \| __| | HTTP/2 AGENT FOR MOCK TESTING
7 : | | | | |/ /| (_| | (_| | __/ | | | |_ | Version 0.0.z
8 : | |_| |_|____\__,_|\__, |\___|_| |_|\__| | https://github.com/testillano/h2agent
9 : | __/ | |
10 : | |___/ |
11 : |___________________________________________|
12 :
13 : Licensed under the MIT License <http://opensource.org/licenses/MIT>.
14 : SPDX-License-Identifier: MIT
15 : Copyright (c) 2021 Eduardo Ramos
16 :
17 : Permission is hereby granted, free of charge, to any person obtaining a copy
18 : of this software and associated documentation files (the "Software"), to deal
19 : in the Software without restriction, including without limitation the rights
20 : to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
21 : copies of the Software, and to permit persons to whom the Software is
22 : furnished to do so, subject to the following conditions:
23 :
24 : The above copyright notice and this permission notice shall be included in all
25 : copies or substantial portions of the Software.
26 :
27 : THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
28 : IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
29 : FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
30 : AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
31 : LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
32 : OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
33 : SOFTWARE.
34 : */
35 :
36 : #pragma once
37 :
38 : #include <vector>
39 : #include <string>
40 : #include <memory>
41 : #include <cstdint>
42 : #include <atomic>
43 : #include <chrono>
44 :
45 : #include <boost/asio.hpp>
46 :
47 : #include <JsonSchema.hpp>
48 :
49 : #include <ert/metrics/Metrics.hpp>
50 :
51 : #include <ert/http2comm/Http2Server.hpp>
52 :
53 : namespace h2agent
54 : {
55 : namespace model
56 : {
57 : class MockServerData;
58 : class MockClientData;
59 : class Configuration;
60 : class GlobalVariable;
61 : class FileManager;
62 : class SocketManager;
63 : class AdminData;
64 : }
65 :
66 : namespace http2
67 : {
68 :
69 : class MyTrafficHttp2Server: public ert::http2comm::Http2Server
70 : {
71 : bool server_data_{};
72 : bool server_data_key_history_{};
73 : bool purge_execution_{};
74 :
75 : model::AdminData *admin_data_{};
76 : model::MockServerData *mock_server_events_data_{};
77 : model::MockClientData *mock_client_events_data_{};
78 :
79 : // metrics:
80 : ert::metrics::Metrics *metrics_{};
81 :
82 : ert::metrics::counter_t *provisioned_requests_successful_counter_{};
83 : ert::metrics::counter_t *provisioned_requests_failed_counter_{};
84 : ert::metrics::counter_t *purged_contexts_successful_counter_{};
85 : ert::metrics::counter_t *purged_contexts_failed_counter_{};
86 :
87 : std::atomic<int> max_busy_threads_{0};
88 : std::atomic<bool> receive_request_body_{true};
89 : std::atomic<bool> pre_reserve_request_body_{true};
90 :
91 : public:
92 : MyTrafficHttp2Server(const std::string &name, size_t workerThreads, size_t maxWorkerThreads, boost::asio::io_context *timersIoContext, int maxQueueDispatcherSize);
93 60 : ~MyTrafficHttp2Server() {;}
94 :
95 : /**
96 : * Enable metrics
97 : *
98 : * @param metrics Optional metrics object to compute counters
99 : * @param source Source label for prometheus metrics. If missing, class name will be taken (even being redundant with
100 : * family name prefix as will ease metrics filtering anyway). A good source convention could be the process name and
101 : * the endpoint identification:
102 : *
103 : * - h2agent[_traffic_server]: optional endpoint category, as it would be deducted from family name
104 : * - h2agentB
105 : */
106 : void enableMyMetrics(ert::metrics::Metrics *metrics, const std::string &source = "");
107 :
108 : bool checkMethodIsAllowed(
109 : const nghttp2::asio_http2::server::request& req,
110 : std::vector<std::string>& allowedMethods);
111 :
112 : bool checkMethodIsImplemented(
113 : const nghttp2::asio_http2::server::request& req);
114 :
115 : bool checkHeaders(const nghttp2::asio_http2::server::request& req);
116 :
117 : bool receiveDataLen(const nghttp2::asio_http2::server::request& req); //virtual
118 : bool preReserveRequestBody(); //virtual
119 :
120 : void receive(const std::uint64_t &receptionId,
121 : const nghttp2::asio_http2::server::request& req,
122 : const std::string &requestBody,
123 : const std::chrono::microseconds &receptionTimestampUs,
124 : unsigned int& statusCode, nghttp2::asio_http2::header_map& headers,
125 : std::string& responseBody, unsigned int &responseDelayMs);
126 :
127 49 : void setAdminData(model::AdminData *p) {
128 49 : admin_data_ = p;
129 49 : }
130 40 : model::AdminData *getAdminData() const {
131 40 : return admin_data_;
132 : }
133 :
134 49 : void setMockServerData(model::MockServerData *p) {
135 49 : mock_server_events_data_ = p;
136 49 : }
137 24 : model::MockServerData *getMockServerData() const {
138 24 : return mock_server_events_data_;
139 : }
140 :
141 49 : void setMockClientData(model::MockClientData *p) {
142 49 : mock_client_events_data_ = p;
143 49 : }
144 : model::MockClientData *getMockClientData() const {
145 : return mock_client_events_data_;
146 : }
147 :
148 : std::string dataConfigurationAsJsonString() const;
149 : std::string configurationAsJsonString() const;
150 :
151 2 : void discardData(bool discard = true) {
152 2 : server_data_ = !discard;
153 2 : }
154 :
155 2 : void discardDataKeyHistory(bool discard = true) {
156 2 : server_data_key_history_ = !discard;
157 2 : }
158 :
159 2 : void disablePurge(bool disable = true) {
160 2 : purge_execution_ = !disable;
161 2 : }
162 :
163 2 : void setReceiveRequestBody(bool receive = true) {
164 2 : receive_request_body_.store(receive);
165 2 : }
166 :
167 2 : void setPreReserveRequestBody(bool preReserve = true) {
168 2 : pre_reserve_request_body_.store(preReserve);
169 2 : }
170 : };
171 :
172 : }
173 : }
174 :
|
