f1a8d7cd3d
Summary: The network layer now has a `Session` that handles all things that should be done before the `Execute` method is called on sessions. Also, all sessions now communicate using streams instead of holding the input buffer and writing to the `Socket`. This design will allow implementation of a SSL middleware. Reviewers: buda, dgleich Reviewed By: buda Subscribers: pullbot Differential Revision: https://phabricator.memgraph.io/D1314
132 lines
3.5 KiB
C++
132 lines
3.5 KiB
C++
#include "bolt_common.hpp"
|
|
#include "communication/bolt/v1/decoder/buffer.hpp"
|
|
#include "communication/bolt/v1/decoder/chunked_decoder_buffer.hpp"
|
|
|
|
constexpr const int SIZE = 131072;
|
|
uint8_t data[SIZE];
|
|
|
|
using BufferT = communication::bolt::Buffer<>;
|
|
using StreamBufferT = io::network::StreamBuffer;
|
|
using DecoderBufferT = communication::bolt::ChunkedDecoderBuffer<BufferT>;
|
|
using ChunkStateT = communication::bolt::ChunkState;
|
|
|
|
TEST(BoltBuffer, CorrectChunk) {
|
|
uint8_t tmp[2000];
|
|
BufferT buffer;
|
|
DecoderBufferT decoder_buffer(buffer);
|
|
StreamBufferT sb = buffer.Allocate();
|
|
|
|
sb.data[0] = 0x03;
|
|
sb.data[1] = 0xe8;
|
|
memcpy(sb.data + 2, data, 1000);
|
|
sb.data[1002] = 0;
|
|
sb.data[1003] = 0;
|
|
buffer.Written(1004);
|
|
|
|
ASSERT_EQ(decoder_buffer.GetChunk(), ChunkStateT::Whole);
|
|
ASSERT_EQ(decoder_buffer.GetChunk(), ChunkStateT::Done);
|
|
|
|
ASSERT_EQ(decoder_buffer.Read(tmp, 1000), true);
|
|
for (int i = 0; i < 1000; ++i) EXPECT_EQ(data[i], tmp[i]);
|
|
|
|
ASSERT_EQ(buffer.size(), 0);
|
|
}
|
|
|
|
TEST(BoltBuffer, CorrectChunkTrailingData) {
|
|
uint8_t tmp[2000];
|
|
BufferT buffer;
|
|
DecoderBufferT decoder_buffer(buffer);
|
|
StreamBufferT sb = buffer.Allocate();
|
|
|
|
sb.data[0] = 0x03;
|
|
sb.data[1] = 0xe8;
|
|
memcpy(sb.data + 2, data, 2002);
|
|
sb.data[1002] = 0;
|
|
sb.data[1003] = 0;
|
|
buffer.Written(2004);
|
|
|
|
ASSERT_EQ(decoder_buffer.GetChunk(), ChunkStateT::Whole);
|
|
ASSERT_EQ(decoder_buffer.GetChunk(), ChunkStateT::Done);
|
|
|
|
ASSERT_EQ(decoder_buffer.Read(tmp, 1000), true);
|
|
for (int i = 0; i < 1000; ++i) EXPECT_EQ(data[i], tmp[i]);
|
|
|
|
uint8_t *leftover = buffer.data();
|
|
ASSERT_EQ(buffer.size(), 1000);
|
|
for (int i = 0; i < 1000; ++i) EXPECT_EQ(data[i + 1002], leftover[i]);
|
|
}
|
|
|
|
TEST(BoltBuffer, GraduallyPopulatedChunk) {
|
|
uint8_t tmp[2000];
|
|
BufferT buffer;
|
|
DecoderBufferT decoder_buffer(buffer);
|
|
StreamBufferT sb = buffer.Allocate();
|
|
|
|
sb.data[0] = 0x03;
|
|
sb.data[1] = 0xe8;
|
|
buffer.Written(2);
|
|
|
|
for (int i = 0; i < 5; ++i) {
|
|
ASSERT_EQ(decoder_buffer.GetChunk(), ChunkStateT::Partial);
|
|
sb = buffer.Allocate();
|
|
memcpy(sb.data, data + 200 * i, 200);
|
|
buffer.Written(200);
|
|
}
|
|
|
|
sb = buffer.Allocate();
|
|
sb.data[0] = 0;
|
|
sb.data[1] = 0;
|
|
buffer.Written(2);
|
|
ASSERT_EQ(decoder_buffer.GetChunk(), ChunkStateT::Whole);
|
|
ASSERT_EQ(decoder_buffer.GetChunk(), ChunkStateT::Done);
|
|
|
|
ASSERT_EQ(decoder_buffer.Read(tmp, 1000), true);
|
|
for (int i = 0; i < 1000; ++i) EXPECT_EQ(data[i], tmp[i]);
|
|
|
|
ASSERT_EQ(buffer.size(), 0);
|
|
}
|
|
|
|
TEST(BoltBuffer, GraduallyPopulatedChunkTrailingData) {
|
|
uint8_t tmp[2000];
|
|
BufferT buffer;
|
|
DecoderBufferT decoder_buffer(buffer);
|
|
StreamBufferT sb = buffer.Allocate();
|
|
|
|
sb.data[0] = 0x03;
|
|
sb.data[1] = 0xe8;
|
|
buffer.Written(2);
|
|
|
|
for (int i = 0; i < 5; ++i) {
|
|
ASSERT_EQ(decoder_buffer.GetChunk(), ChunkStateT::Partial);
|
|
sb = buffer.Allocate();
|
|
memcpy(sb.data, data + 200 * i, 200);
|
|
buffer.Written(200);
|
|
}
|
|
|
|
sb = buffer.Allocate();
|
|
sb.data[0] = 0;
|
|
sb.data[1] = 0;
|
|
buffer.Written(2);
|
|
|
|
sb = buffer.Allocate();
|
|
memcpy(sb.data, data, 1000);
|
|
buffer.Written(1000);
|
|
|
|
ASSERT_EQ(decoder_buffer.GetChunk(), ChunkStateT::Whole);
|
|
ASSERT_EQ(decoder_buffer.GetChunk(), ChunkStateT::Done);
|
|
|
|
ASSERT_EQ(decoder_buffer.Read(tmp, 1000), true);
|
|
for (int i = 0; i < 1000; ++i) EXPECT_EQ(data[i], tmp[i]);
|
|
|
|
uint8_t *leftover = buffer.data();
|
|
ASSERT_EQ(buffer.size(), 1000);
|
|
for (int i = 0; i < 1000; ++i) EXPECT_EQ(data[i], leftover[i]);
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
InitializeData(data, SIZE);
|
|
google::InitGoogleLogging(argv[0]);
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|