Add groth16_test_env()

src/prover.cpp

@@ -3,6 +3,7 @@
 #include <string>
 #include <cstring>
 #include <stdexcept>
+#include <sstream>
 #include <alt_bn128.hpp>
 #include <nlohmann/json.hpp>
 
@@ -222,3 +223,33 @@ groth16_prover_zkey_file(const char *zkey_file_path,
                           public_buffer, public_size,
                           error_msg, error_msg_maxsize);
 }
+
+int
+groth16_test_env(char *buffer, unsigned long buffer_maxsize)
+{
+    std::ostringstream  stream;
+
+#ifdef USE_ASM
+#if defined(ARCH_X86_64)
+    stream << "ASM: x86_64" << std::endl;
+#elif defined(ARCH_ARM64)
+    stream << "ASM: arm64" << std::endl;
+#endif
+#else
+    stream << "ASM is disabled" << std::endl;
+#endif
+
+#ifdef USE_OPENMP
+    stream << "OpenMP max threads: " << omp_get_max_threads() << std::endl;
+#else
+    stream << "OpenMP is disabled" << std::endl;
+#endif
+
+    if (buffer_maxsize < stream.str().size() + 1) {
+        return PROVER_ERROR_SHORT_BUFFER;
+    }
+
+    strncpy(buffer, stream.str().c_str(), buffer_maxsize);
+
+    return PROVER_OK;
+}

src/prover.h

@@ -62,6 +62,16 @@ groth16_prover_zkey_file(const char *zkey_file_path,
                          char *public_buffer, unsigned long *public_size,
                          char *error_msg, unsigned long error_msg_maxsize);
 
+/**
+ * groth16_test_env
+ * @return error code:
+ *         PROVER_OK - in case of success
+ *         PROVER_ERROR_SHORT_BUFFER - in case of a short buffer error
+ */
+int
+groth16_test_env(char *buffer, unsigned long buffer_maxsize);
+
+
 #ifdef __cplusplus
 }
 #endif