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 sockets (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 : #include <string>
37 : #include <memory>
38 :
39 : #include <ert/tracing/Logger.hpp>
40 :
41 : #include <SafeSocket.hpp>
42 : #include <SocketManager.hpp>
43 : #include <functions.hpp>
44 :
45 :
46 : namespace h2agent
47 : {
48 : namespace model
49 : {
50 :
51 :
52 5 : SafeSocket::SafeSocket (SocketManager *socketManager, const std::string& path, boost::asio::io_context *timersIoContext):
53 5 : path_(path),
54 5 : io_context_(timersIoContext),
55 5 : socket_manager_(socketManager)
56 : {
57 5 : open();
58 5 : }
59 :
60 1 : void SafeSocket::delayedWrite(unsigned int writeDelayUs, const std::string &data) {
61 : // metrics
62 1 : socket_manager_->incrementObservedDelayedWriteOperationCounter();
63 :
64 : //if (!io_context_) return; // protection
65 1 : auto timer = std::make_shared<boost::asio::steady_timer>(*io_context_, std::chrono::microseconds(writeDelayUs));
66 1 : timer->expires_after(std::chrono::microseconds(writeDelayUs));
67 1 : timer->async_wait([this, timer, data] (const boost::system::error_code& e) {
68 1 : if( e ) return; // probably, we were cancelled (boost::asio::error::operation_aborted)
69 1 : write(data, 0, true /* from delayed */);
70 : });
71 1 : }
72 :
73 5 : bool SafeSocket::open() {
74 :
75 5 : socket_ = socket(AF_UNIX, SOCK_DGRAM, 0);
76 5 : if (socket_ >= 0) {
77 : // metrics
78 5 : socket_manager_->incrementObservedOpenOperationCounter();
79 :
80 5 : memset(&server_addr_, 0, sizeof(struct sockaddr_un));
81 5 : server_addr_.sun_family = AF_UNIX;
82 5 : strcpy(server_addr_.sun_path, path_.c_str());
83 : }
84 : else {
85 0 : LOGWARNING(ert::tracing::Logger::warning(ert::tracing::Logger::asString("Failed to open '%s'", path_.c_str()), ERT_FILE_LOCATION));
86 : // metrics
87 0 : socket_manager_->incrementObservedErrorOpenOperationCounter();
88 0 : return false;
89 : }
90 :
91 5 : return true;
92 : }
93 :
94 10 : nlohmann::json SafeSocket::getJson() const {
95 10 : nlohmann::json result;
96 :
97 10 : result["path"] = path_;
98 10 : result["socket"] = socket_;
99 :
100 10 : return result;
101 0 : }
102 :
103 6 : void SafeSocket::write (const std::string& data, unsigned int writeDelayUs, bool fromDelayed) {
104 :
105 : // trace data & delay
106 6 : LOGDEBUG(
107 : ert::tracing::Logger::debug(ert::tracing::Logger::asString("Data for write operation is: %s", data.c_str()), ERT_FILE_LOCATION);
108 : if (writeDelayUs != 0) ert::tracing::Logger::debug(ert::tracing::Logger::asString("Delay for write operation is: %lu", writeDelayUs), ERT_FILE_LOCATION);
109 : );
110 :
111 : // Write socket:
112 : //std::lock_guard<std::mutex> lock(mutex_);
113 :
114 : // metrics
115 6 : if (writeDelayUs == 0) {
116 5 : socket_manager_->incrementObservedWriteOperationCounter();
117 5 : if (!fromDelayed) socket_manager_->incrementObservedInstantWriteOperationCounter();
118 : }
119 :
120 : // Close socket:
121 6 : if (io_context_ && writeDelayUs != 0) {
122 1 : delayedWrite(writeDelayUs, data);
123 : }
124 : else {
125 5 : sendto(socket_, data.c_str(), data.length(), 0, (struct sockaddr*)&server_addr_, sizeof(struct sockaddr_un));
126 5 : LOGDEBUG(ert::tracing::Logger::debug(ert::tracing::Logger::asString("Data written into '%s'", path_.c_str()), ERT_FILE_LOCATION));
127 : }
128 6 : }
129 :
130 : }
131 : }
132 :
|
