espressif/esp-idf
1
0
Fork 1
mirror of https://github.com/espressif/esp-idf.git synced 2025-07-30 18:57:19 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix(tcp_transport): Fix test for socks transport

Browse Source 
- Removed out of scope stack acess
This commit is contained in:
glmfe
2025-03-26 08:23:10 -03:00
parent 796bac0b74
commit 5c76c789f8
1 changed files with 37 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

55
components/tcp_transport/host_test/main/test_socks_transport.cpp
View File

@ -1,5 +1,5 @@
/* /*
* SPDX-FileCopyrightText: 2022-2023 Espressif Systems (Shanghai) CO LTD * SPDX-FileCopyrightText: 2022-2025 Espressif Systems (Shanghai) CO LTD
* *
* SPDX-License-Identifier: Apache-2.0 * SPDX-License-Identifier: Apache-2.0
*/ */
@ -37,17 +37,27 @@ using namespace std::literals;
namespace { namespace {
/* struct CallbackContext {
* Makes possible to pass a capturing lambda as a callback esp_transport_handle_t test_parent;
*/ std::array<char, 9> expected_request;
decltype(auto) capture_lambda(auto callable) int timeout;
{ };
// make a static copy of the lambda to extend it's lifetime and avoid the capture. static CallbackContext context;
[[maybe_unused]]static auto call = callable;
return []<typename... Args>(Args... args) { struct CallbackContextGuard {
return call(args...); CallbackContext &context;
};
} // Constructor to initialize the context
CallbackContextGuard(CallbackContext &context): context(context) {}
// Destructor to clean up the context
~CallbackContextGuard() {
// Reset the context to avoid residual state
context.test_parent = nullptr;
context.expected_request.fill(0);
context.timeout = 0;
}
};
auto make_response(socks_transport_error_t response) auto make_response(socks_transport_error_t response)
{ {
@ -140,14 +150,23 @@ TEST_CASE("Requests to Proxy", "[Requests]")
auto *p_addr_info = &addr_info; auto *p_addr_info = &addr_info;
lwip_getaddrinfo_ReturnThruPtr_res(&p_addr_info); lwip_getaddrinfo_ReturnThruPtr_res(&p_addr_info);
auto expected_request = std::array<char,9>{0x04, 0x01, 0x00, 0x50, 0x5a, 0x5a, 0x5a, 0x5a, 0x00 }; auto expected_request = std::array<char,9>{0x04, 0x01, 0x00, 0x50, 0x5a, 0x5a, 0x5a, 0x5a, 0x00 };
mock_write_Stub(capture_lambda([&test_parent, expected_request, &timeout](esp_transport_handle_t transport, const char *request_sent, int len, int timeout_ms, [[maybe_unused]]int num_call) {
// Create and initialize the callback context
context = {
.test_parent = test_parent.get(),
.expected_request = expected_request,
.timeout = timeout,
};
CallbackContextGuard guard(context);
mock_write_Stub([](esp_transport_handle_t transport, const char *request_sent, int len, int timeout_ms, [[maybe_unused]]int num_call) {
using namespace Catch::Matchers; using namespace Catch::Matchers;
REQUIRE(transport == test_parent.get()); REQUIRE(transport == context.test_parent);
REQUIRE(len == expected_request.size()); REQUIRE(len == context.expected_request.size());
REQUIRE(timeout_ms == timeout); REQUIRE(timeout_ms == context.timeout);
REQUIRE(std::equal(request_sent,request_sent+len, std::begin(expected_request), std::end(expected_request))); REQUIRE(std::equal(request_sent, request_sent + len, std::begin(context.expected_request), std::end(context.expected_request)));
return len; return len;
})); });
SECTION("Successful connection request") { SECTION("Successful connection request") {