99 lines
2.9 KiB
C++
99 lines
2.9 KiB
C++
#include "byte_stream.hh"
|
|
#include "fsm_stream_reassembler_harness.hh"
|
|
#include "stream_reassembler.hh"
|
|
#include "util.hh"
|
|
|
|
#include <exception>
|
|
#include <iostream>
|
|
|
|
using namespace std;
|
|
|
|
int main() {
|
|
try {
|
|
auto rd = get_random_generator();
|
|
|
|
{
|
|
ReassemblerTestHarness test{65000};
|
|
|
|
test.execute(SubmitSegment{"abcd", 0});
|
|
test.execute(BytesAssembled(4));
|
|
test.execute(BytesAvailable("abcd"));
|
|
test.execute(NotAtEof{});
|
|
|
|
test.execute(SubmitSegment{"abcd", 0});
|
|
test.execute(BytesAssembled(4));
|
|
test.execute(BytesAvailable(""));
|
|
test.execute(NotAtEof{});
|
|
}
|
|
|
|
{
|
|
ReassemblerTestHarness test{65000};
|
|
|
|
test.execute(SubmitSegment{"abcd", 0});
|
|
test.execute(BytesAssembled(4));
|
|
test.execute(BytesAvailable("abcd"));
|
|
test.execute(NotAtEof{});
|
|
|
|
test.execute(SubmitSegment{"abcd", 4});
|
|
test.execute(BytesAssembled(8));
|
|
test.execute(BytesAvailable("abcd"));
|
|
test.execute(NotAtEof{});
|
|
|
|
test.execute(SubmitSegment{"abcd", 0});
|
|
test.execute(BytesAssembled(8));
|
|
test.execute(BytesAvailable(""));
|
|
test.execute(NotAtEof{});
|
|
|
|
test.execute(SubmitSegment{"abcd", 4});
|
|
test.execute(BytesAssembled(8));
|
|
test.execute(BytesAvailable(""));
|
|
test.execute(NotAtEof{});
|
|
}
|
|
|
|
{
|
|
ReassemblerTestHarness test{65000};
|
|
|
|
test.execute(SubmitSegment{"abcdefgh", 0});
|
|
test.execute(BytesAssembled(8));
|
|
test.execute(BytesAvailable("abcdefgh"));
|
|
test.execute(NotAtEof{});
|
|
string data = "abcdefgh";
|
|
|
|
for (size_t i = 0; i < 1000; ++i) {
|
|
size_t start_i = uniform_int_distribution<size_t>{0, 8}(rd);
|
|
auto start = data.begin();
|
|
std::advance(start, start_i);
|
|
|
|
size_t end_i = uniform_int_distribution<size_t>{start_i, 8}(rd);
|
|
auto end = data.begin();
|
|
std::advance(end, end_i);
|
|
|
|
test.execute(SubmitSegment{string{start, end}, start_i});
|
|
test.execute(BytesAssembled(8));
|
|
test.execute(BytesAvailable(""));
|
|
test.execute(NotAtEof{});
|
|
}
|
|
}
|
|
|
|
{
|
|
ReassemblerTestHarness test{65000};
|
|
|
|
test.execute(SubmitSegment{"abcd", 0});
|
|
test.execute(BytesAssembled(4));
|
|
test.execute(BytesAvailable("abcd"));
|
|
test.execute(NotAtEof{});
|
|
|
|
test.execute(SubmitSegment{"abcdef", 0});
|
|
test.execute(BytesAssembled(6));
|
|
test.execute(BytesAvailable("ef"));
|
|
test.execute(NotAtEof{});
|
|
}
|
|
|
|
} catch (const exception &e) {
|
|
cerr << "Exception: " << e.what() << endl;
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|