third_party/sqlite/fuzz/sql_strftime_fuzzer.cc

// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <cstdlib>
#include <iostream>
#include <string>
#include <vector>

#include "base/check.h"
#include "testing/libfuzzer/proto/lpm_interface.h"
#include "third_party/sqlite/fuzz/sql_query_grammar.pb.h"
#include "third_party/sqlite/fuzz/sql_query_grammar_fuzzable.pb.h"
#include "third_party/sqlite/fuzz/sql_query_proto_to_string.h"
#include "third_party/sqlite/fuzz/sql_run_queries.h"

using namespace sql_query_grammar;

DEFINE_PROTO_FUZZER(
    const fuzzable::sql_query_grammar::StrftimeFn& fuzzable_sql_strftime) {
  std::string serialized;
  CHECK(fuzzable_sql_strftime.SerializeToString(&serialized));
  sql_query_grammar::StrftimeFn sql_strftime;
  // Recursion limits can cause parsing to fail.
  if (!sql_strftime.ParseFromString(serialized)) {
    return;
  }
  std::string strftime_str = sql_fuzzer::StrftimeFnToString(sql_strftime);
  // Convert printf command into runnable SQL query.
  strftime_str = "SELECT " + strftime_str + ";";

  if (getenv("LPM_DUMP_NATIVE_INPUT")) {
    std::cout << "_________________________" << std::endl;
    std::cout << strftime_str << std::endl;
    std::cout << "------------------------" << std::endl;
  }

  std::vector<std::string> queries;
  queries.push_back(strftime_str);
  sql_fuzzer::RunSqlQueries(queries, ::getenv("LPM_SQLITE_TRACE"));
}