vanja/yabridge
1
0
Fork 0
mirror of https://github.com/robbert-vdh/yabridge.git synced 2026-05-09 20:29:10 +02:00
Code Issues Packages Projects Releases Wiki Activity

Use boost::asio::transfer_exactly

Browse Source 
Instead of doing assertions.
This commit is contained in:
Robbert van der Helm
2020-12-10 15:03:26 +01:00
parent e07467697a
commit 5f7c105b6d
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/common/communication.h
+4 -3
View File
@@ -106,7 +106,8 @@ template <typename T, typename Socket>
inline T read_object(Socket& socket, std::vector<uint8_t>& buffer) { inline T read_object(Socket& socket, std::vector<uint8_t>& buffer) {
// See the note above on the use of `uint64_t` instead of `size_t` // See the note above on the use of `uint64_t` instead of `size_t`
std::array<uint64_t, 1> message_length; std::array<uint64_t, 1> message_length;
boost::asio::read(socket, boost::asio::buffer(message_length)); boost::asio::read(socket, boost::asio::buffer(message_length),
boost::asio::transfer_exactly(sizeof(message_length)));
// Make sure the buffer is large enough // Make sure the buffer is large enough
const size_t size = message_length[0]; const size_t size = message_length[0];
@@ -116,8 +117,8 @@ inline T read_object(Socket& socket, std::vector<uint8_t>& buffer) {
// merging for us, since local domain sockets have packet limits somewhere // merging for us, since local domain sockets have packet limits somewhere
// in the hundreds of kilobytes // in the hundreds of kilobytes
const auto actual_size = const auto actual_size =
boost::asio::read(socket, boost::asio::buffer(buffer)); boost::asio::read(socket, boost::asio::buffer(buffer),
assert(size == actual_size); boost::asio::transfer_exactly(size));
T object; T object;
auto [_, success] = auto [_, success] =