Alex Hultman
72 lines
2.2 KiB
C++
72 lines
2.2 KiB
C++
#include <iostream>
|
|
#include <cassert>
|
|
#include <sstream>
|
|
#include <iomanip>
|
|
#include <vector>
|
|
#include <climits>
|
|
|
|
#include "../src/ChunkedEncoding.h"
|
|
|
|
void runTest(unsigned int maxConsume) {
|
|
/* A list of chunks */
|
|
std::vector<std::string_view> chunks = {
|
|
"Hello there I am the first segment",
|
|
"Why hello there",
|
|
"",
|
|
"I am last?",
|
|
"And I am a little longer but it doesn't matter",
|
|
""
|
|
};
|
|
|
|
/* Encode them in chunked encoding */
|
|
std::stringstream ss;
|
|
for (std::string_view chunk : chunks) {
|
|
/* Generic chunked encoding format */
|
|
ss << std::hex << chunk.length() << "\r\n" << chunk << "\r\n";
|
|
|
|
/* Every null chunk is followed by an empty trailer */
|
|
if (chunk.length() == 0) {
|
|
ss << "\r\n";
|
|
}
|
|
}
|
|
std::string buffer = ss.str();
|
|
|
|
/* Begin with a clear state and the full data */
|
|
unsigned int state = 0;
|
|
unsigned int chunkOffset = 0;
|
|
std::string_view chunkEncoded = buffer;
|
|
while (chunkEncoded.length()) {
|
|
/* Parse a small part of the given data */
|
|
std::string_view data = chunkEncoded.substr(0, std::min<size_t>(maxConsume, chunkEncoded.length()));
|
|
chunkEncoded.remove_prefix(data.length());
|
|
|
|
/* Whatever chunk we emit, or part of chunk, it must match the expected one */
|
|
for (auto chunk : uWS::ChunkIterator(data, &state)) {
|
|
std::cout << "<" << chunk << ">" << std::endl;
|
|
|
|
/* Run check here */
|
|
if (!chunk.length() && chunks[chunkOffset].length()) {
|
|
std::cout << "We got emitted an empty chunk but expected a non-empty one" << std::endl;
|
|
std::abort();
|
|
}
|
|
|
|
if (chunks[chunkOffset].starts_with(chunk)) {
|
|
chunks[chunkOffset].remove_prefix(chunk.length());
|
|
if (!chunks[chunkOffset].length()) {
|
|
chunkOffset++;
|
|
}
|
|
} else {
|
|
std::cerr << "Chunk does not match! Should be <" << chunks[chunkOffset] << ">" << std::endl;
|
|
std::abort();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
int main() {
|
|
for (int i = 1; i < 1000; i++) {
|
|
runTest(i);
|
|
}
|
|
|
|
std::cout << "ALL BRUTEFORCE DONE" << std::endl;
|
|
} |