codeberg-wownero/wownero
1
0
Fork 0
mirror of https://codeberg.org/wownero/wownero.git synced 2026-10-17 05:41:49 +02:00
Code Issues Packages Projects Releases Wiki Activity

Cleanup some of the fragmented levin handling

Browse Source 
(cherry picked from commit 017c9022d7bbc03484191177228657b14fb09f53)
This commit is contained in:
Lee *!* Clagett
2026-05-05 18:20:52 -04:00
committed by jwinterm
parent 77cd1986ac
commit e16f0a2c28
2 changed files with 33 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
contrib/epee/include/net/levin_protocol_handler_async.h
+12 -4
View File
@@ -481,18 +481,26 @@ public:
return false; return false;
} }
temp = std::move(m_fragment_buffer); temp.swap(m_fragment_buffer);
m_fragment_buffer.clear();
std::memcpy(std::addressof(m_current_head), std::addressof(temp[0]), sizeof(bucket_head2)); std::memcpy(std::addressof(m_current_head), std::addressof(temp[0]), sizeof(bucket_head2));
const std::uint64_t inner_size = SWAP64LE(m_current_head.m_cb);
buff_to_invoke = {reinterpret_cast<const uint8_t*>(temp.data()) + sizeof(bucket_head2), temp.size() - sizeof(bucket_head2)};
if (buff_to_invoke.size() < inner_size)
{
MERROR(m_connection_context << "Invalid fragmented buffer size: " << buff_to_invoke.size() << " vs " << inner_size);
return false;
}
buff_to_invoke = {buff_to_invoke.data(), std::size_t(inner_size)};
const size_t max_bytes = m_connection_context.get_max_bytes(m_current_head.m_command); const size_t max_bytes = m_connection_context.get_max_bytes(m_current_head.m_command);
if(m_current_head.m_cb > std::min<size_t>(max_packet_size, max_bytes)) if(buff_to_invoke.size() > std::min<size_t>(max_packet_size, max_bytes))
{ {
MERROR(m_connection_context << "Maximum packet size exceed!, m_max_packet_size = " << std::min<size_t>(max_packet_size, max_bytes) MERROR(m_connection_context << "Maximum packet size exceed!, m_max_packet_size = " << std::min<size_t>(max_packet_size, max_bytes)
<< ", packet header received " << m_current_head.m_cb << ", command " << m_current_head.m_command << ", packet header received " << m_current_head.m_cb << ", command " << m_current_head.m_command
<< ", connection will be closed."); << ", connection will be closed.");
return false; return false;
} }
buff_to_invoke = {reinterpret_cast<const uint8_t*>(temp.data()) + sizeof(bucket_head2), temp.size() - sizeof(bucket_head2)};
} }
bool is_response = (m_oponent_protocol_ver == LEVIN_PROTOCOL_VER_1 && m_current_head.m_flags&LEVIN_PACKET_RESPONSE); bool is_response = (m_oponent_protocol_ver == LEVIN_PROTOCOL_VER_1 && m_current_head.m_flags&LEVIN_PACKET_RESPONSE);
tests/unit_tests/epee_levin_protocol_handler_async.cpp
+21 -1
View File
@@ -506,7 +506,6 @@ TEST_F(positive_test_connection_to_levin_protocol_handler_calls, handler_process
} }
std::string compare_buffer(1024 * 4, 'c'); std::string compare_buffer(1024 * 4, 'c');
compare_buffer.resize(((1024 - sizeof(epee::levin::bucket_head2)) * 5) - sizeof(epee::levin::bucket_head2)); // add padding zeroes
ASSERT_EQ(4u, m_commands_handler.notify_counter()); ASSERT_EQ(4u, m_commands_handler.notify_counter());
ASSERT_EQ(0u, m_commands_handler.invoke_counter()); ASSERT_EQ(0u, m_commands_handler.invoke_counter());
@@ -652,3 +651,24 @@ TEST_F(test_levin_protocol_handler__hanle_recv_with_invalid_data, handles_short_
ASSERT_FALSE(m_conn->m_protocol_handler.handle_recv(m_buf.data(), m_buf.size())); ASSERT_FALSE(m_conn->m_protocol_handler.handle_recv(m_buf.data(), m_buf.size()));
} }
TEST_F(test_levin_protocol_handler__hanle_recv_with_invalid_data, handles_bad_cb)
{
m_req_head.m_cb = sizeof(epee::levin::bucket_head2);
m_req_head.m_flags = LEVIN_PACKET_BEGIN;
m_req_head.m_command = 0;
epee::levin::bucket_head2 inner{};
inner.m_cb = 2;
m_in_data.resize(sizeof(epee::levin::bucket_head2));
prepare_buf();
ASSERT_TRUE(m_conn->m_protocol_handler.handle_recv(m_buf.data(), m_buf.size()));
m_req_head.m_cb = 1;
m_req_head.m_flags = LEVIN_PACKET_END;
m_in_data.resize(1);
prepare_buf();
ASSERT_FALSE(m_conn->m_protocol_handler.handle_recv(m_buf.data(), m_buf.size()));
}