libc/fuzzing/stdlib/StringParserOutputDiff.h

   1 //===-- Template to diff single-input-single-output functions ---*- C++ -*-===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #ifndef LLVM_LIBC_FUZZING_STDLIB_STRING_PARSER_OUTPUT_DIFF_H
  10 #define LLVM_LIBC_FUZZING_STDLIB_STRING_PARSER_OUTPUT_DIFF_H
  11
  12 #include "fuzzing/math/Compare.h"
  13
  14 #include <stddef.h>
  15 #include <stdint.h>
  16
  17 template <typename T> using StringInputSingleOutputFunc = T (*)(const char *);
  18
  19 template <typename T>
  20 void StringParserOutputDiff(StringInputSingleOutputFunc<T> func1,
  21                             StringInputSingleOutputFunc<T> func2,
  22                             const uint8_t *data, size_t size) {
  23   if (size < sizeof(T))
  24     return;
  25
  26   const char *x = reinterpret_cast<const char *>(data);
  27
  28   T result1 = func1(x);
  29   T result2 = func2(x);
  30
  31   if (!ValuesEqual(result1, result2))
  32     __builtin_trap();
  33 }
  34
  35 template <typename T>
  36 using StringToNumberFunc = T (*)(const char *, char **, int);
  37
  38 template <typename T>
  39 void StringToNumberOutputDiff(StringToNumberFunc<T> func1,
  40                               StringToNumberFunc<T> func2, const uint8_t *data,
  41                               size_t size) {
  42   if (size < sizeof(T))
  43     return;
  44
  45   const char *x = reinterpret_cast<const char *>(data + 1);
  46   int base = data[0] % 36;
  47   base = base + ((base == 0) ? 0 : 1);
  48
  49   char *outPtr1 = nullptr;
  50   char *outPtr2 = nullptr;
  51
  52   T result1 = func1(x, &outPtr1, base);
  53   T result2 = func2(x, &outPtr2, base);
  54
  55   if (!(ValuesEqual(result1, result2) && (*outPtr1 == *outPtr2)))
  56     __builtin_trap();
  57 }
  58
  59 #endif // LLVM_LIBC_FUZZING_STDLIB_STRING_PARSER_OUTPUT_DIFF_H