318 lines
9.3 KiB
C++
318 lines
9.3 KiB
C++
#include "ns3/buffer.h"
|
|
#include "ns3/random-variable.h"
|
|
#include "ns3/test.h"
|
|
|
|
namespace ns3 {
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Unit tests
|
|
//-----------------------------------------------------------------------------
|
|
class BufferTest : public TestCase {
|
|
private:
|
|
void EnsureWrittenBytes (Buffer b, uint32_t n, uint8_t array[], const char *file, int line);
|
|
public:
|
|
virtual void DoRun (void);
|
|
BufferTest ();
|
|
};
|
|
|
|
|
|
BufferTest::BufferTest ()
|
|
: TestCase ("Buffer") {
|
|
}
|
|
|
|
void
|
|
BufferTest::EnsureWrittenBytes (Buffer b, uint32_t n, uint8_t array[], const char *file, int line)
|
|
{
|
|
bool success = true;
|
|
uint8_t *expected = array;
|
|
uint8_t const*got;
|
|
got = b.PeekData ();
|
|
for (uint32_t j = 0; j < n; j++)
|
|
{
|
|
if (got[j] != expected[j])
|
|
{
|
|
success = false;
|
|
}
|
|
}
|
|
if (!success)
|
|
{
|
|
std::ostringstream failure;
|
|
failure << "Buffer -- ";
|
|
failure << "expected: n=";
|
|
failure << n << ", ";
|
|
failure.setf (std::ios::hex, std::ios::basefield);
|
|
for (uint32_t j = 0; j < n; j++)
|
|
{
|
|
failure << (uint16_t)expected[j] << " ";
|
|
}
|
|
failure.setf (std::ios::dec, std::ios::basefield);
|
|
failure << "got: ";
|
|
failure.setf (std::ios::hex, std::ios::basefield);
|
|
for (uint32_t j = 0; j < n; j++)
|
|
{
|
|
failure << (uint16_t)got[j] << " ";
|
|
}
|
|
failure << std::endl;
|
|
NS_TEST_ASSERT_MSG_EQ_INTERNAL (true, false, failure.str (), file, line);
|
|
}
|
|
}
|
|
|
|
/* Note: works only when variadic macros are
|
|
* available which is the case for gcc.
|
|
* XXX
|
|
*/
|
|
#define ENSURE_WRITTEN_BYTES(buffer, n, ...) \
|
|
{ \
|
|
uint8_t bytes[] = { __VA_ARGS__}; \
|
|
EnsureWrittenBytes (buffer, n, bytes, __FILE__, __LINE__); \
|
|
}
|
|
|
|
void
|
|
BufferTest::DoRun (void)
|
|
{
|
|
Buffer buffer;
|
|
Buffer::Iterator i;
|
|
buffer.AddAtStart (6);
|
|
i = buffer.Begin ();
|
|
i.WriteU8 (0x66);
|
|
ENSURE_WRITTEN_BYTES (buffer, 1, 0x66);
|
|
i = buffer.Begin ();
|
|
i.WriteU8 (0x67);
|
|
ENSURE_WRITTEN_BYTES (buffer, 1, 0x67);
|
|
i.WriteHtonU16 (0x6568);
|
|
i = buffer.Begin ();
|
|
ENSURE_WRITTEN_BYTES (buffer, 3, 0x67, 0x65, 0x68);
|
|
i.WriteHtonU16 (0x6369);
|
|
ENSURE_WRITTEN_BYTES (buffer, 3, 0x63, 0x69, 0x68);
|
|
i.WriteHtonU32 (0xdeadbeaf);
|
|
ENSURE_WRITTEN_BYTES (buffer, 6, 0x63, 0x69, 0xde, 0xad, 0xbe, 0xaf);
|
|
buffer.AddAtStart (2);
|
|
i = buffer.Begin ();
|
|
i.WriteU16 (0);
|
|
ENSURE_WRITTEN_BYTES (buffer, 8, 0, 0, 0x63, 0x69, 0xde, 0xad, 0xbe, 0xaf);
|
|
buffer.AddAtEnd (2);
|
|
i = buffer.Begin ();
|
|
i.Next (8);
|
|
i.WriteU16 (0);
|
|
ENSURE_WRITTEN_BYTES (buffer, 10, 0, 0, 0x63, 0x69, 0xde, 0xad, 0xbe, 0xaf, 0, 0);
|
|
buffer.RemoveAtStart (3);
|
|
i = buffer.Begin ();
|
|
ENSURE_WRITTEN_BYTES (buffer, 7, 0x69, 0xde, 0xad, 0xbe, 0xaf, 0, 0);
|
|
buffer.RemoveAtEnd (4);
|
|
i = buffer.Begin ();
|
|
ENSURE_WRITTEN_BYTES (buffer, 3, 0x69, 0xde, 0xad);
|
|
buffer.AddAtStart (1);
|
|
i = buffer.Begin ();
|
|
i.WriteU8 (0xff);
|
|
ENSURE_WRITTEN_BYTES (buffer, 4, 0xff, 0x69, 0xde, 0xad);
|
|
buffer.AddAtEnd (1);
|
|
i = buffer.Begin ();
|
|
i.Next (4);
|
|
i.WriteU8 (0xff);
|
|
i.Prev (2);
|
|
uint16_t saved = i.ReadU16 ();
|
|
i.Prev (2);
|
|
i.WriteHtonU16 (0xff00);
|
|
i.Prev (2);
|
|
NS_TEST_ASSERT_MSG_EQ (i.ReadNtohU16 (), 0xff00, "Could not read expected value");
|
|
i.Prev (2);
|
|
i.WriteU16 (saved);
|
|
ENSURE_WRITTEN_BYTES (buffer, 5, 0xff, 0x69, 0xde, 0xad, 0xff);
|
|
Buffer o = buffer;
|
|
ENSURE_WRITTEN_BYTES (o, 5, 0xff, 0x69, 0xde, 0xad, 0xff);
|
|
o.AddAtStart (1);
|
|
i = o.Begin ();
|
|
i.WriteU8 (0xfe);
|
|
ENSURE_WRITTEN_BYTES (o, 6, 0xfe, 0xff, 0x69, 0xde, 0xad, 0xff);
|
|
buffer.AddAtStart (2);
|
|
i = buffer.Begin ();
|
|
i.WriteU8 (0xfd);
|
|
i.WriteU8 (0xfd);
|
|
ENSURE_WRITTEN_BYTES (o, 6, 0xfe, 0xff, 0x69, 0xde, 0xad, 0xff);
|
|
ENSURE_WRITTEN_BYTES (buffer, 7, 0xfd, 0xfd, 0xff, 0x69, 0xde, 0xad, 0xff);
|
|
|
|
// test 64-bit read/write
|
|
Buffer buff64;
|
|
buff64.AddAtStart (8);
|
|
i = buff64.Begin ();
|
|
i.WriteU64 (0x0123456789ABCDEFllu);
|
|
ENSURE_WRITTEN_BYTES (buff64, 8, 0xef, 0xcd, 0xab, 0x89, 0x67, 0x45, 0x23, 0x01);
|
|
i = buff64.Begin ();
|
|
NS_TEST_ASSERT_MSG_EQ (i.ReadLsbtohU64 (), 0x0123456789abcdefllu, "Could not read expected value");
|
|
i = buff64.Begin ();
|
|
i.WriteHtolsbU64 (0x0123456789ABCDEFllu);
|
|
ENSURE_WRITTEN_BYTES (buff64, 8, 0xef, 0xcd, 0xab, 0x89, 0x67, 0x45, 0x23, 0x01);
|
|
i = buff64.Begin ();
|
|
NS_TEST_ASSERT_MSG_EQ (i.ReadLsbtohU64 (), 0x0123456789abcdefllu, "Could not read expected value");
|
|
i = buff64.Begin ();
|
|
i.WriteHtonU64 (0x0123456789ABCDEFllu);
|
|
ENSURE_WRITTEN_BYTES (buff64, 8, 0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef);
|
|
i = buff64.Begin ();
|
|
NS_TEST_ASSERT_MSG_EQ (i.ReadNtohU64 (), 0x0123456789abcdefllu, "could not read expected value");
|
|
|
|
// test self-assignment
|
|
{
|
|
Buffer a = o;
|
|
a = a;
|
|
}
|
|
|
|
// test Remove start.
|
|
buffer = Buffer (5);
|
|
ENSURE_WRITTEN_BYTES (buffer, 5, 0, 0, 0, 0, 0);
|
|
buffer.RemoveAtStart (1);
|
|
ENSURE_WRITTEN_BYTES (buffer, 4, 0, 0, 0, 0);
|
|
buffer.AddAtStart (1);
|
|
buffer.Begin ().WriteU8 (0xff);
|
|
ENSURE_WRITTEN_BYTES (buffer, 5, 0xff, 0, 0, 0, 0);
|
|
buffer.RemoveAtStart (3);
|
|
ENSURE_WRITTEN_BYTES (buffer, 2, 0, 0);
|
|
buffer.AddAtStart (4);
|
|
buffer.Begin ().WriteHtonU32 (0xdeadbeaf);
|
|
ENSURE_WRITTEN_BYTES (buffer, 6, 0xde, 0xad, 0xbe, 0xaf, 0, 0);
|
|
buffer.RemoveAtStart (2);
|
|
ENSURE_WRITTEN_BYTES (buffer, 4, 0xbe, 0xaf, 0, 0);
|
|
buffer.AddAtEnd (4);
|
|
i = buffer.Begin ();
|
|
i.Next (4);
|
|
i.WriteHtonU32 (0xdeadbeaf);
|
|
ENSURE_WRITTEN_BYTES (buffer, 8, 0xbe, 0xaf, 0, 0, 0xde, 0xad, 0xbe, 0xaf);
|
|
buffer.RemoveAtStart (5);
|
|
ENSURE_WRITTEN_BYTES (buffer, 3, 0xad, 0xbe, 0xaf);
|
|
// test Remove end
|
|
buffer = Buffer (5);
|
|
ENSURE_WRITTEN_BYTES (buffer, 5, 0, 0, 0, 0, 0);
|
|
buffer.RemoveAtEnd (1);
|
|
ENSURE_WRITTEN_BYTES (buffer, 4, 0, 0, 0, 0);
|
|
buffer.AddAtEnd (2);
|
|
i = buffer.Begin ();
|
|
i.Next (4);
|
|
i.WriteU8 (0xab);
|
|
i.WriteU8 (0xac);
|
|
ENSURE_WRITTEN_BYTES (buffer, 6, 0, 0, 0, 0, 0xab, 0xac);
|
|
buffer.RemoveAtEnd (1);
|
|
ENSURE_WRITTEN_BYTES (buffer, 5, 0, 0, 0, 0, 0xab);
|
|
buffer.RemoveAtEnd (3);
|
|
ENSURE_WRITTEN_BYTES (buffer, 2, 0, 0);
|
|
buffer.AddAtEnd (6);
|
|
i = buffer.Begin ();
|
|
i.Next (2);
|
|
i.WriteU8 (0xac);
|
|
i.WriteU8 (0xad);
|
|
i.WriteU8 (0xae);
|
|
i.WriteU8 (0xaf);
|
|
i.WriteU8 (0xba);
|
|
i.WriteU8 (0xbb);
|
|
ENSURE_WRITTEN_BYTES (buffer, 8, 0, 0, 0xac, 0xad, 0xae, 0xaf, 0xba, 0xbb);
|
|
buffer.AddAtStart (3);
|
|
i = buffer.Begin ();
|
|
i.WriteU8 (0x30);
|
|
i.WriteU8 (0x31);
|
|
i.WriteU8 (0x32);
|
|
ENSURE_WRITTEN_BYTES (buffer, 11, 0x30, 0x31, 0x32, 0, 0, 0xac, 0xad, 0xae, 0xaf, 0xba, 0xbb);
|
|
buffer.RemoveAtEnd (9);
|
|
ENSURE_WRITTEN_BYTES (buffer, 2, 0x30, 0x31);
|
|
buffer = Buffer (3);
|
|
buffer.AddAtEnd (2);
|
|
i = buffer.Begin ();
|
|
i.Next (3);
|
|
i.WriteHtonU16 (0xabcd);
|
|
buffer.AddAtStart (1);
|
|
buffer.Begin ().WriteU8 (0x21);
|
|
ENSURE_WRITTEN_BYTES (buffer, 6, 0x21, 0, 0, 0, 0xab, 0xcd);
|
|
buffer.RemoveAtEnd (8);
|
|
NS_TEST_ASSERT_MSG_EQ (buffer.GetSize (), 0, "Buffer size not zero");
|
|
|
|
buffer = Buffer (6);
|
|
buffer.AddAtStart (9);
|
|
buffer.AddAtEnd (3);
|
|
i = buffer.End ();
|
|
i.Prev (1);
|
|
i.WriteU8 (1, 1);
|
|
|
|
buffer = Buffer (6);
|
|
buffer.AddAtStart (3);
|
|
buffer.RemoveAtEnd (8);
|
|
buffer.AddAtEnd (4);
|
|
i = buffer.End ();
|
|
i.Prev (4);
|
|
i.WriteU8 (1, 4);
|
|
|
|
buffer = Buffer (1);
|
|
buffer.AddAtEnd (100);
|
|
i = buffer.End ();
|
|
i.Prev (100);
|
|
i.WriteU8 (1, 100);
|
|
|
|
// Bug #54
|
|
{
|
|
const uint32_t actualSize = 72602;
|
|
const uint32_t chunkSize = 67624;
|
|
UniformVariable bytesRng (0, 256);
|
|
|
|
Buffer inputBuffer;
|
|
Buffer outputBuffer;
|
|
|
|
inputBuffer.AddAtEnd (actualSize);
|
|
{
|
|
Buffer::Iterator iter = inputBuffer.Begin ();
|
|
for (uint32_t i = 0; i < actualSize; i++)
|
|
iter.WriteU8 (static_cast<uint8_t> (bytesRng.GetValue ()));
|
|
}
|
|
|
|
outputBuffer.AddAtEnd (chunkSize);
|
|
Buffer::Iterator iter = outputBuffer.End ();
|
|
iter.Prev (chunkSize);
|
|
iter.Write (inputBuffer.PeekData (), chunkSize);
|
|
|
|
NS_TEST_EXPECT_MSG_EQ (memcmp (inputBuffer.PeekData (), outputBuffer.PeekData (), chunkSize), 0, "memcp works");
|
|
}
|
|
|
|
buffer = Buffer (5);
|
|
buffer.AddAtEnd (2);
|
|
i = buffer.End ();
|
|
i.Prev (2);
|
|
i.WriteU8 (0);
|
|
i.WriteU8 (0x66);
|
|
ENSURE_WRITTEN_BYTES (buffer, 7, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x66);
|
|
Buffer frag0 = buffer.CreateFragment (0, 2);
|
|
ENSURE_WRITTEN_BYTES (frag0, 2, 0x00, 0x00);
|
|
Buffer frag1 = buffer.CreateFragment (2, 5);
|
|
ENSURE_WRITTEN_BYTES (frag1, 5, 0x00, 0x00, 0x00, 0x00, 0x66);
|
|
frag0.AddAtEnd (frag1);
|
|
ENSURE_WRITTEN_BYTES (buffer, 7, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x66);
|
|
ENSURE_WRITTEN_BYTES (frag0, 7, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x66);
|
|
|
|
buffer = Buffer (5);
|
|
buffer.AddAtStart (2);
|
|
i = buffer.Begin ();
|
|
i.WriteU8 (0x1);
|
|
i.WriteU8 (0x2);
|
|
buffer.AddAtEnd (2);
|
|
i = buffer.End ();
|
|
i.Prev (2);
|
|
i.WriteU8 (0x3);
|
|
i.WriteU8 (0x4);
|
|
ENSURE_WRITTEN_BYTES (buffer, 9, 0x1, 0x2, 0x00, 0x00, 0x00, 0x00, 0x00, 0x3, 0x4);
|
|
Buffer other;
|
|
other.AddAtStart (9);
|
|
i = other.Begin ();
|
|
i.Write (buffer.Begin (), buffer.End ());
|
|
ENSURE_WRITTEN_BYTES (other, 9, 0x1, 0x2, 0x00, 0x00, 0x00, 0x00, 0x00, 0x3, 0x4);
|
|
}
|
|
//-----------------------------------------------------------------------------
|
|
class BufferTestSuite : public TestSuite
|
|
{
|
|
public:
|
|
BufferTestSuite ();
|
|
};
|
|
|
|
BufferTestSuite::BufferTestSuite ()
|
|
: TestSuite ("buffer", UNIT)
|
|
{
|
|
AddTestCase (new BufferTest);
|
|
}
|
|
|
|
static BufferTestSuite g_bufferTestSuite;
|
|
|
|
} // namespace ns3
|