[libc-commits] [libc] [libc] created tan function fuzzer (PR #101570)

[via libc-commits]

https://github.com/RoseZhang03 created https://github.com/llvm/llvm-project/pull/101570

Also edited file header formatting on sin_fuz and cos_fuzz


>From 6fb615a1a8fb49f682c4a2e35a2dfb630b051cee Mon Sep 17 00:00:00 2001
From: Rose Zhang <rosezhang at google.com>
Date: Thu, 1 Aug 2024 22:11:55 +0000
Subject: [PATCH] [libc] created tan function fuzzer

Also edited file header formatting on sin_fuz and cos_fuzz
---
 libc/fuzzing/math/CMakeLists.txt |  9 +++++++
 libc/fuzzing/math/cos_fuzz.cpp   |  2 +-
 libc/fuzzing/math/sin_fuzz.cpp   |  2 +-
 libc/fuzzing/math/tan_fuzz.cpp   | 40 ++++++++++++++++++++++++++++++++
 4 files changed, 51 insertions(+), 2 deletions(-)
 create mode 100644 libc/fuzzing/math/tan_fuzz.cpp

diff --git a/libc/fuzzing/math/CMakeLists.txt b/libc/fuzzing/math/CMakeLists.txt
index 7faf8fd53c849..31336fa3e49f3 100644
--- a/libc/fuzzing/math/CMakeLists.txt
+++ b/libc/fuzzing/math/CMakeLists.txt
@@ -79,3 +79,12 @@ add_libc_fuzzer(
   DEPENDS
     libc.src.math.cos
 )
+
+add_libc_fuzzer(
+  tan_fuzz
+  NEED_MPFR
+  SRCS
+    tan_fuzz.cpp
+  DEPENDS
+    libc.src.math.tan
+)
diff --git a/libc/fuzzing/math/cos_fuzz.cpp b/libc/fuzzing/math/cos_fuzz.cpp
index e26687d7f6ea8..3e4ad659fd3a5 100644
--- a/libc/fuzzing/math/cos_fuzz.cpp
+++ b/libc/fuzzing/math/cos_fuzz.cpp
@@ -1,4 +1,4 @@
-//===-- cos_fuzz.cpp ----------------------------------------------------===//
+//===-- 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.
diff --git a/libc/fuzzing/math/sin_fuzz.cpp b/libc/fuzzing/math/sin_fuzz.cpp
index 0293afb062839..009b9f2080298 100644
--- a/libc/fuzzing/math/sin_fuzz.cpp
+++ b/libc/fuzzing/math/sin_fuzz.cpp
@@ -1,4 +1,4 @@
-//===-- sin_fuzz.cpp ----------------------------------------------------===//
+//===-- sin_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.
diff --git a/libc/fuzzing/math/tan_fuzz.cpp b/libc/fuzzing/math/tan_fuzz.cpp
new file mode 100644
index 0000000000000..700ebb80974c8
--- /dev/null
+++ b/libc/fuzzing/math/tan_fuzz.cpp
@@ -0,0 +1,40 @@
+//===-- tan_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 tan implementation.
+///
+//===----------------------------------------------------------------------===//
+
+#include "src/math/tan.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_tan(input, input, MPFR_RNDN);
+  mpfr_subnormalize(input, output, MPFR_RNDN);
+  double to_compare = mpfr_get_d(input, MPFR_RNDN);
+
+  double result = LIBC_NAMESPACE::tan(x);
+
+  if (result != to_compare)
+    __builtin_trap();
+
+  mpfr_clear(input);
+  return 0;
+}