add zstd library (#854)

test/CMakeLists.txt

@@ -17,6 +17,7 @@ add_executable(goalc-test
         ${CMAKE_CURRENT_LIST_DIR}/test_common_util.cpp
         ${CMAKE_CURRENT_LIST_DIR}/test_pretty_print.cpp
         ${CMAKE_CURRENT_LIST_DIR}/test_math.cpp
+        ${CMAKE_CURRENT_LIST_DIR}/test_zstd.cpp
         ${CMAKE_CURRENT_LIST_DIR}/test_zydis.cpp
         ${CMAKE_CURRENT_LIST_DIR}/goalc/test_goal_kernel.cpp
         ${CMAKE_CURRENT_LIST_DIR}/decompiler/FormRegressionTest.cpp
@@ -36,7 +37,7 @@ add_executable(goalc-test
         ${GOALC_TEST_FRAMEWORK_SOURCES}
         ${GOALC_TEST_CASES})
 
-target_link_libraries(goalc-test common runtime compiler gtest decomp Zydis)
+target_link_libraries(goalc-test common runtime compiler gtest decomp Zydis libzstd_static)
 
 if(WIN32)
   target_link_libraries(goalc-test mman)

test/test_zstd.cpp

@@ -0,0 +1,58 @@
+#include <cstdio>
+#include <string>
+#include <vector>
+
+#include "common/common_types.h"
+
+#include "gtest/gtest.h"
+#include "test/all_jak1_symbols.h"
+#include "third-party/zstd/lib/zstd.h"
+#include "common/util/compress.h"
+
+TEST(ZSTD, Basic) {
+  std::string all;
+  for (auto& x : all_syms) {
+    all.append(x);
+    all.append("\n");
+  }
+
+  size_t in_size = all.size();
+  printf("size: %d\n", (int)in_size);
+
+  size_t max_out_size = ZSTD_compressBound(in_size);
+
+  std::vector<u8> out(max_out_size);
+  auto result =
+      ZSTD_compress(out.data(), out.size(), all.data(), all.length(), ZSTD_CLEVEL_DEFAULT);
+  ASSERT_FALSE(ZSTD_isError(result));
+  printf("compressed: %d\n", (int)result);
+
+  std::string uncompressed;
+  uncompressed.resize(in_size);
+
+  auto result2 = ZSTD_decompress(uncompressed.data(), uncompressed.size(), out.data(), result);
+  if (ZSTD_isError(result2)) {
+    printf("err: %s\n", ZSTD_getErrorName(result2));
+  }
+  ASSERT_FALSE(ZSTD_isError(result2));
+  EXPECT_EQ(result2, in_size);
+  EXPECT_EQ(all, uncompressed);
+}
+
+TEST(ZSTD, CommonLibrary) {
+  std::string all;
+  for (auto& x : all_syms) {
+    all.append(x);
+    all.append("\n");
+  }
+
+  auto compressed = compression::compress_zstd(all.data(), all.size());
+  auto decompressed = compression::decompress_zstd(compressed.data(), compressed.size());
+
+  ASSERT_EQ(decompressed.size(), all.size());
+  for (size_t i = 0; i < all.length(); i++) {
+    EXPECT_EQ(decompressed[i], all[i]);
+  }
+
+  EXPECT_TRUE(compressed.size() < 0.5 * all.size());
+}