[libc-commits] [libc] b45d362 - [libc] added cos function fuzzing test (#101556)

[via libc-commits]

Author: RoseZhang03
Date: 2024-08-01T21:59:55Z
New Revision: b45d36299e11708fe6de0104bc287328f7340e2d

URL: https://github.com/llvm/llvm-project/commit/b45d36299e11708fe6de0104bc287328f7340e2d
DIFF: https://github.com/llvm/llvm-project/commit/b45d36299e11708fe6de0104bc287328f7340e2d.diff

LOG: [libc] added cos function fuzzing test (#101556)

Added: 
    libc/fuzzing/math/cos_fuzz.cpp

Modified: 
    libc/fuzzing/math/CMakeLists.txt
    libc/fuzzing/math/sin_fuzz.cpp

Removed: 
    


################################################################################
diff  --git a/libc/fuzzing/math/CMakeLists.txt b/libc/fuzzing/math/CMakeLists.txt
index ad163cc53eae3..7faf8fd53c849 100644
--- a/libc/fuzzing/math/CMakeLists.txt
+++ b/libc/fuzzing/math/CMakeLists.txt
@@ -70,3 +70,12 @@ add_libc_fuzzer(
   DEPENDS
     libc.src.math.sin
 )
+
+add_libc_fuzzer(
+  cos_fuzz
+  NEED_MPFR
+  SRCS
+    cos_fuzz.cpp
+  DEPENDS
+    libc.src.math.cos
+)

diff  --git a/libc/fuzzing/math/cos_fuzz.cpp b/libc/fuzzing/math/cos_fuzz.cpp
new file mode 100644
index 0000000000000..e26687d7f6ea8
--- /dev/null
+++ b/libc/fuzzing/math/cos_fuzz.cpp
@@ -0,0 +1,40 @@
+//===-- cos_fuzz.cpp ----------------------------------------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+///
+/// Fuzzing test for llvm-libc cos implementation.
+///
+//===----------------------------------------------------------------------===//
+
+#include "src/math/cos.h"
+#include <math.h>
+#include <mpfr.h>
+
+extern "C" int LLVMFuzzerTestOneInput(const double x) {
+  // remove NaN and inf as preconditions
+  if (isnan(x))
+    return 0;
+  if (isinf(x))
+    return 0;
+  // signed zeros already tested in unit tests
+  if (signbit(x) && x == 0.0)
+    return 0;
+  mpfr_t input;
+  mpfr_init2(input, 53);
+  mpfr_set_d(input, x, MPFR_RNDN);
+  int output = mpfr_cos(input, input, MPFR_RNDN);
+  mpfr_subnormalize(input, output, MPFR_RNDN);
+  double to_compare = mpfr_get_d(input, MPFR_RNDN);
+
+  double result = LIBC_NAMESPACE::cos(x);
+
+  if (result != to_compare)
+    __builtin_trap();
+
+  mpfr_clear(input);
+  return 0;
+}

diff  --git a/libc/fuzzing/math/sin_fuzz.cpp b/libc/fuzzing/math/sin_fuzz.cpp
index 1e607d461978d..0293afb062839 100644
--- a/libc/fuzzing/math/sin_fuzz.cpp
+++ b/libc/fuzzing/math/sin_fuzz.cpp
@@ -11,17 +11,17 @@
 //===----------------------------------------------------------------------===//
 
 #include "src/math/sin.h"
-#include <cmath>
+#include <math.h>
 #include <mpfr.h>
 
 extern "C" int LLVMFuzzerTestOneInput(const double x) {
   // remove NaN and inf as preconditions
-  if (std::isnan(x))
+  if (isnan(x))
     return 0;
-  if (std::isinf(x))
+  if (isinf(x))
     return 0;
   // signed zeros already tested in unit tests
-  if (std::signbit(x) && x == 0.0)
+  if (signbit(x) && x == 0.0)
     return 0;
   mpfr_t input;
   mpfr_init2(input, 53);