Green Sky
227425b90e
git-subtree-dir: external/toxcore/c-toxcore git-subtree-split: 67badf69416a74e74f6d7eb51dd96f37282b8455
135 lines
3.8 KiB
C++
135 lines
3.8 KiB
C++
#include "bin_pack.h"
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <array>
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
#include "bin_unpack.h"
|
|
|
|
namespace {
|
|
|
|
struct Bin_Pack_Deleter {
|
|
void operator()(Bin_Pack *bp) const { bin_pack_free(bp); }
|
|
};
|
|
|
|
using Bin_Pack_Ptr = std::unique_ptr<Bin_Pack, Bin_Pack_Deleter>;
|
|
|
|
struct Bin_Unpack_Deleter {
|
|
void operator()(Bin_Unpack *bu) const { bin_unpack_free(bu); }
|
|
};
|
|
|
|
using Bin_Unpack_Ptr = std::unique_ptr<Bin_Unpack, Bin_Unpack_Deleter>;
|
|
|
|
TEST(BinPack, TooSmallBufferIsNotExceeded)
|
|
{
|
|
std::array<uint8_t, 7> buf;
|
|
Bin_Pack_Ptr bp(bin_pack_new(buf.data(), buf.size()));
|
|
ASSERT_NE(bp, nullptr);
|
|
EXPECT_FALSE(bin_pack_u64_b(bp.get(), 1234567812345678LL));
|
|
}
|
|
|
|
TEST(BinPack, PackedUint64CanBeUnpacked)
|
|
{
|
|
std::array<uint8_t, 8> buf;
|
|
Bin_Pack_Ptr bp(bin_pack_new(buf.data(), buf.size()));
|
|
ASSERT_NE(bp, nullptr);
|
|
ASSERT_TRUE(bin_pack_u64_b(bp.get(), 1234567812345678LL));
|
|
|
|
Bin_Unpack_Ptr bu(bin_unpack_new(buf.data(), buf.size()));
|
|
ASSERT_NE(bu, nullptr);
|
|
uint64_t val;
|
|
ASSERT_TRUE(bin_unpack_u64_b(bu.get(), &val));
|
|
EXPECT_EQ(val, 1234567812345678LL);
|
|
}
|
|
|
|
TEST(BinPack, MsgPackedUint8CanBeUnpackedAsUint32)
|
|
{
|
|
std::array<uint8_t, 2> buf;
|
|
Bin_Pack_Ptr bp(bin_pack_new(buf.data(), buf.size()));
|
|
ASSERT_NE(bp, nullptr);
|
|
ASSERT_TRUE(bin_pack_u08(bp.get(), 123));
|
|
|
|
Bin_Unpack_Ptr bu(bin_unpack_new(buf.data(), buf.size()));
|
|
ASSERT_NE(bu, nullptr);
|
|
uint32_t val;
|
|
ASSERT_TRUE(bin_unpack_u32(bu.get(), &val));
|
|
EXPECT_EQ(val, 123);
|
|
}
|
|
|
|
TEST(BinPack, MsgPackedUint32CanBeUnpackedAsUint8IfSmallEnough)
|
|
{
|
|
std::array<uint8_t, 2> buf;
|
|
Bin_Pack_Ptr bp(bin_pack_new(buf.data(), buf.size()));
|
|
ASSERT_NE(bp, nullptr);
|
|
ASSERT_TRUE(bin_pack_u32(bp.get(), 123));
|
|
|
|
Bin_Unpack_Ptr bu(bin_unpack_new(buf.data(), buf.size()));
|
|
ASSERT_NE(bu, nullptr);
|
|
uint8_t val;
|
|
ASSERT_TRUE(bin_unpack_u08(bu.get(), &val));
|
|
EXPECT_EQ(val, 123);
|
|
}
|
|
|
|
TEST(BinPack, LargeMsgPackedUint32CannotBeUnpackedAsUint8)
|
|
{
|
|
std::array<uint8_t, 5> buf;
|
|
Bin_Pack_Ptr bp(bin_pack_new(buf.data(), buf.size()));
|
|
ASSERT_NE(bp, nullptr);
|
|
ASSERT_TRUE(bin_pack_u32(bp.get(), 1234567));
|
|
|
|
Bin_Unpack_Ptr bu(bin_unpack_new(buf.data(), buf.size()));
|
|
ASSERT_NE(bu, nullptr);
|
|
uint8_t val;
|
|
EXPECT_FALSE(bin_unpack_u08(bu.get(), &val));
|
|
}
|
|
|
|
TEST(BinPack, BinCanHoldPackedInts)
|
|
{
|
|
std::array<uint8_t, 12> buf;
|
|
Bin_Pack_Ptr bp(bin_pack_new(buf.data(), buf.size()));
|
|
ASSERT_NE(bp, nullptr);
|
|
ASSERT_TRUE(bin_pack_bin_marker(bp.get(), 8));
|
|
ASSERT_TRUE(bin_pack_u64_b(bp.get(), 1234567812345678LL));
|
|
ASSERT_TRUE(bin_pack_u16_b(bp.get(), 54321));
|
|
|
|
Bin_Unpack_Ptr bu(bin_unpack_new(buf.data(), buf.size()));
|
|
ASSERT_NE(bu, nullptr);
|
|
uint32_t size;
|
|
EXPECT_TRUE(bin_unpack_bin_size(bu.get(), &size));
|
|
EXPECT_EQ(size, 8);
|
|
uint64_t val1;
|
|
EXPECT_TRUE(bin_unpack_u64_b(bu.get(), &val1));
|
|
EXPECT_EQ(val1, 1234567812345678LL);
|
|
uint16_t val2;
|
|
EXPECT_TRUE(bin_unpack_u16_b(bu.get(), &val2));
|
|
EXPECT_EQ(val2, 54321);
|
|
}
|
|
|
|
TEST(BinPack, BinCanHoldArbitraryData)
|
|
{
|
|
std::array<uint8_t, 7> buf;
|
|
Bin_Pack_Ptr bp(bin_pack_new(buf.data(), buf.size()));
|
|
ASSERT_NE(bp, nullptr);
|
|
ASSERT_TRUE(bin_pack_bin_marker(bp.get(), 5));
|
|
ASSERT_TRUE(bin_pack_bin_b(bp.get(), reinterpret_cast<const uint8_t *>("hello"), 5));
|
|
|
|
Bin_Unpack_Ptr bu(bin_unpack_new(buf.data(), buf.size()));
|
|
ASSERT_NE(bu, nullptr);
|
|
std::array<uint8_t, 5> str;
|
|
EXPECT_TRUE(bin_unpack_bin_fixed(bu.get(), str.data(), str.size()));
|
|
EXPECT_EQ(str, (std::array<uint8_t, 5>{'h', 'e', 'l', 'l', 'o'}));
|
|
}
|
|
|
|
TEST(BinPack, OversizedArrayFailsUnpack)
|
|
{
|
|
std::array<uint8_t, 1> buf = {0x91};
|
|
|
|
Bin_Unpack_Ptr bu(bin_unpack_new(buf.data(), buf.size()));
|
|
uint32_t size;
|
|
EXPECT_FALSE(bin_unpack_array(bu.get(), &size));
|
|
}
|
|
|
|
} // namespace
|