simdjson/tests/allparserscheckfile.cpp

69 lines
2.1 KiB
C++

#include "jsonparser/jsonparser.h"
// #define RAPIDJSON_SSE2 // bad
// #define RAPIDJSON_SSE42 // bad
#include "rapidjson/document.h"
#include "rapidjson/reader.h" // you have to check in the submodule
#include "rapidjson/stringbuffer.h"
#include "rapidjson/writer.h"
#include "sajson.h"
using namespace rapidjson;
using namespace std;
int main(int argc, char *argv[]) {
if (argc < 2) {
cerr << "Usage: " << argv[0] << " <jsonfile>\n";
cerr << "Or " << argv[0] << " -v <jsonfile>\n";
exit(1);
}
bool verbose = false;
if (argc > 2) {
if (strcmp(argv[1], "-v"))
verbose = true;
}
pair<u8 *, size_t> p = get_corpus(argv[argc - 1]);
if (verbose) {
std::cout << "Input has ";
if (p.second > 1024 * 1024)
std::cout << p.second / (1024 * 1024) << " MB ";
else if (p.second > 1024)
std::cout << p.second / 1024 << " KB ";
else
std::cout << p.second << " B ";
std::cout << std::endl;
}
ParsedJson *pj_ptr = allocate_ParsedJson(p.second);
if (pj_ptr == NULL) {
std::cerr << "can't allocate memory" << std::endl;
return EXIT_FAILURE;
}
ParsedJson &pj(*pj_ptr);
bool ours_correct = json_parse(p.first, p.second, pj);
rapidjson::Document d;
char *buffer = (char *)malloc(p.second + 1);
memcpy(buffer, p.first, p.second);
buffer[p.second] = '\0';
bool rapid_correct = (d.Parse((const char *)buffer).HasParseError() == false);
bool rapid_correct_checkencoding = (d.Parse<kParseValidateEncodingFlag>((const char *)buffer).HasParseError() == false);
bool sajson_correct = sajson::parse(sajson::dynamic_allocation(), sajson::mutable_string_view(p.second, buffer)).is_valid();
printf("our parser : %s \n", ours_correct ? "correct":"invalid");
printf("rapid : %s \n", rapid_correct ? "correct":"invalid");
printf("rapid (check encoding) : %s \n", rapid_correct_checkencoding ? "correct":"invalid");
printf("sajson : %s \n", sajson_correct ? "correct":"invalid");
free(buffer);
free(p.first);
deallocate_ParsedJson(pj_ptr);
return EXIT_SUCCESS;
}