fix: add workaround to build on ubuntu 20.04 (#5)

Signed-off-by: wep21 <daisuke.nishimatsu1021@gmail.com>
tier4 · Aug 11, 2023 · 99c2628 · 99c2628
1 parent ec260d7
commit 99c2628
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -26,10 +26,25 @@ endif()
 
 add_executable(app src/app.cpp)
 
+include(CheckSymbolExists)
+check_symbol_exists(mallinfo2 malloc.h HAVE_MALLINFO2)
+
 add_library(preloaded_heaptrack SHARED src/preloaded.cpp)
+if(HAVE_MALLINFO2)
+  target_compile_definitions(preloaded_heaptrack
+    PRIVATE
+      HAVE_MALLINFO2
+  )
+endif()
 
 add_library(preloaded_tlsf SHARED src/preloaded_tlsf.cpp)
 target_link_libraries(preloaded_tlsf PRIVATE tlsf::tlsf)
+if(HAVE_MALLINFO2)
+  target_compile_definitions(preloaded_tlsf
+    PRIVATE
+      HAVE_MALLINFO2
+  )
+endif()
 
 install(TARGETS preloaded_heaptrack preloaded_tlsf DESTINATION lib)
 install(TARGETS app DESTINATION bin)

diff --git a/src/preloaded.cpp b/src/preloaded.cpp
@@ -364,13 +364,15 @@ struct mallinfo mallinfo()
   return orig();
 }
 
+#ifdef HAVE_MALLINFO2
 using mallinfo2_type = struct mallinfo2 (*)( void);
 struct mallinfo2 mallinfo2()
 {
   static mallinfo2_type orig = reinterpret_cast<mallinfo2_type>(dlsym(RTLD_NEXT, "mallinfo2"));
   printf("hoge: mallinfo2 called\n");
   return orig();
 }
+#endif
 
 using mallopt_type = int (*)(int, int);
 int mallopt(int param, int value)

diff --git a/src/preloaded_tlsf.cpp b/src/preloaded_tlsf.cpp
@@ -370,13 +370,15 @@ struct mallinfo mallinfo()
   return orig();
 }
 
+#ifdef HAVE_MALLINFO2
 using mallinfo2_type = struct mallinfo2 (*)( void);
 struct mallinfo2 mallinfo2()
 {
   static mallinfo2_type orig = reinterpret_cast<mallinfo2_type>(dlsym(RTLD_NEXT, "mallinfo2"));
   printf("hoge: mallinfo2 called\n");
   return orig();
 }
+#endif
 
 using mallopt_type = int (*)(int, int);
 int mallopt(int param, int value)