Merge pull request #6572

438b846 ByteSlice: Fix persisting ptr to std::moved SSO buffer (Doy-lee)
This commit is contained in:
luigi1111 2020-05-21 03:06:15 -05:00
commit 4f6dbcd00e
No known key found for this signature in database
GPG Key ID: F4ACA0183641E010
2 changed files with 27 additions and 1 deletions

View File

@ -133,10 +133,13 @@ namespace epee
template<typename T> template<typename T>
byte_slice::byte_slice(const adapt_buffer, T&& buffer) byte_slice::byte_slice(const adapt_buffer, T&& buffer)
: storage_(nullptr), portion_(to_byte_span(to_span(buffer))) : storage_(nullptr), portion_(nullptr)
{ {
if (!buffer.empty()) if (!buffer.empty())
{
storage_ = allocate_slice<adapted_byte_slice<T>>(0, std::move(buffer)); storage_ = allocate_slice<adapted_byte_slice<T>>(0, std::move(buffer));
portion_ = to_byte_span(to_span(static_cast<adapted_byte_slice<T> *>(storage_.get())->buffer));
}
} }
byte_slice::byte_slice(std::initializer_list<span<const std::uint8_t>> sources) byte_slice::byte_slice(std::initializer_list<span<const std::uint8_t>> sources)

View File

@ -387,6 +387,29 @@ TEST(ByteSlice, Construction)
EXPECT_FALSE(std::is_copy_assignable<epee::byte_slice>()); EXPECT_FALSE(std::is_copy_assignable<epee::byte_slice>());
} }
TEST(ByteSlice, DataReturnedMatches)
{
for (int i = 64; i > 0; i--)
{
std::string sso_string(i, 'a');
std::string original = sso_string;
epee::byte_slice slice{std::move(sso_string)};
EXPECT_EQ(slice.size(), original.size());
EXPECT_EQ(memcmp(slice.data(), original.data(), original.size()), 0);
}
for (int i = 64; i > 0; i--)
{
std::vector<uint8_t> sso_vector(i, 'a');
std::vector<uint8_t> original = sso_vector;
epee::byte_slice slice{std::move(sso_vector)};
EXPECT_EQ(slice.size(), original.size());
EXPECT_EQ(memcmp(slice.data(), original.data(), original.size()), 0);
}
}
TEST(ByteSlice, NoExcept) TEST(ByteSlice, NoExcept)
{ {
EXPECT_TRUE(std::is_nothrow_default_constructible<epee::byte_slice>()); EXPECT_TRUE(std::is_nothrow_default_constructible<epee::byte_slice>());