base/base64_encode_fuzzer.cc

   1 // Copyright 2017 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include <string>
   6
   7 #include "base/base64.h"
   8 #include "base/check_op.h"
   9 #include "base/strings/string_piece.h"
  10
  11 // Encode some random data, and then decode it.
  12 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
  13   base::span<const uint8_t> data_span(data, size);
  14   base::StringPiece data_piece(reinterpret_cast<const char*>(data), size);
  15
  16   const std::string encode_output = base::Base64Encode(data_span);
  17   std::string decode_output;
  18   CHECK(base::Base64Decode(encode_output, &decode_output));
  19   CHECK_EQ(data_piece, decode_output);
  20
  21   // Also run the StringPiece variant and check that it gives the same results.
  22   std::string string_piece_encode_output;
  23   base::Base64Encode(data_piece, &string_piece_encode_output);
  24   CHECK_EQ(encode_output, string_piece_encode_output);
  25
  26   return 0;
  27 }