projects/pycryptodome/pcd_aes_fuzzer.cc

// Copyright 2020 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#include "AES.c"
#include "common.h"
#include <fuzzer/FuzzedDataProvider.h>

extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {

  if (!size)
    return 0;

  enum KeySize { AES128 = 16, AES192 = 24, AES256 = 32, kMaxValue = AES256 };

  FuzzedDataProvider stream(data, size);
  const KeySize keySize = stream.ConsumeEnum<KeySize>();
  if (stream.remaining_bytes() < keySize)
    return 0;

  std::vector<uint8_t> keyBuf = stream.ConsumeBytes<uint8_t>(keySize);
  const uint8_t *key = keyBuf.data();

  BlockBase *state;
  if (AES_start_operation(key, keySize, reinterpret_cast<AES_State **>(&state)))
    return 0;

  uint8_t outEnc[size];
  uint8_t outDec[size];

  AES_encrypt(reinterpret_cast<BlockBase *>(state), data, outEnc, size);
  AES_decrypt(reinterpret_cast<BlockBase *>(state), data, outDec, size);

  AES_stop_operation(reinterpret_cast<BlockBase *>(state));

  return 0;
}