/*
* (C) 2015 Jack Lloyd
*
* Botan is released under the Simplified BSD License (see license.txt)
*/

#include "tests.h"
#include <chrono>

#if defined(BOTAN_HAS_X509_CERTIFICATES)
  #include <botan/x509cert.h>
  #include <botan/x509_crl.h>
  #include <botan/base64.h>
  #include <botan/internal/filesystem.h>
#endif

namespace Botan_Tests {

namespace {

class Fuzzer_Input_Tests : public Test
   {
   public:
      std::vector<Test::Result> run() override
         {
         std::vector<Test::Result> results;
#if defined(BOTAN_HAS_X509_CERTIFICATES)
         results.push_back(test_x509_fuzz());
#endif
         return results;
         }

   private:

#if defined(BOTAN_HAS_X509_CERTIFICATES)
      Test::Result test_x509_fuzz()
         {
         const std::string TEST_DATA_DIR_FUZZ_X509 = TEST_DATA_DIR "/fuzz/x509";

         Test::Result result("X.509 fuzzing");

         std::vector<std::string> files;

         try
            {
            files = Botan::get_files_recursive(TEST_DATA_DIR_FUZZ_X509);
            }
         catch(Botan::No_Filesystem_Access)
            {
            result.note_missing("Filesystem access");
            return result;
            }

         for(auto vec_file: files)
            {
            auto start = std::chrono::steady_clock::now();

            try
               {
               // TODO: check for memory consumption?
               Botan::X509_Certificate cert(vec_file);
               }
            catch(std::exception& e)
               {
               }

            result.test_success();

            auto end = std::chrono::steady_clock::now();

            uint64_t duration = std::chrono::duration_cast<std::chrono::milliseconds>(end - start).count();

            if(duration > 100)
               {
               result.test_note("Fuzzer test " + vec_file + " took " + std::to_string(duration) + " ms");
               }
            }

         return result;
         }
#endif
   };

BOTAN_REGISTER_TEST("fuzzer", Fuzzer_Input_Tests);


}

}