FreeBSD patches

Author: yurivict

src/CMakeLists.txt

@@ -298,6 +298,11 @@ else()
     set(EXTRA_LIBS ${EXTRA_LIBS} ${GMP_LIBRARIES})
 endif()
 
+# system-specific libraries
+if (${CMAKE_SYSTEM_NAME} MATCHES "FreeBSD")
+    set(EXTRA_UTIL_LIBS ${EXTRA_UTIL_LIBS} "-lprocstat")
+endif()
+
 # DL
 if (EMSCRIPTEN)
     # no dlopen

src/util/debug.cpp

@@ -118,7 +118,11 @@ void invoke_debugger() {
         case 'g': {
             std::cerr << "INVOKING GDB...\n";
             std::ostringstream buffer;
+#if defined(__FreeBSD__)
+            buffer << "gdb -nw /proc/" << getpid() << "/file " << getpid();
+#else
             buffer << "gdb -nw /proc/" << getpid() << "/exe " << getpid();
+#endif
             if (system(buffer.str().c_str()) == 0) {
                 std::cerr << "continuing the execution...\n";
             } else {

src/util/memory.cpp

@@ -64,7 +64,7 @@ size_t get_current_rss() {
    --------------------------------------------------- */
 #else
 /* ----------------------------------------------------
-   Linux/OSX version for get_peak_rss and get_current_rss
+   Linux/FreeBSD/OSX version for get_peak_rss and get_current_rss
    --------------------------------------------------- */
 #include <unistd.h>
 #include <sys/resource.h>
@@ -73,6 +73,20 @@ size_t get_current_rss() {
 #include <mach/mach.h>
 #endif
 
+#if defined(__FreeBSD__)
+// for getpagesize
+#include <unistd.h>
+// for procstat_*
+#include <sys/param.h>
+#include <sys/queue.h>
+#include <sys/socket.h>
+#include <libprocstat.h>
+// for struct kinfo_proc
+#include <sys/user.h>
+// for KERN_PROC_PID
+#include <sys/sysctl.h>
+#endif
+
 namespace lean {
 size_t get_peak_rss() {
     struct rusage rusage;
@@ -91,6 +105,28 @@ size_t get_current_rss() {
     if (task_info(mach_task_self(), MACH_TASK_BASIC_INFO, reinterpret_cast<task_info_t>(&info), &infoCount) != KERN_SUCCESS)
         return static_cast<size_t>(0);
     return static_cast<size_t>(info.resident_size);
+#elif defined(__FreeBSD__)
+    // initialize
+    unsigned int count = 0;
+
+    struct procstat *pstat = procstat_open_sysctl();
+    if (!pstat)
+        return static_cast<size_t>(0);
+
+    struct kinfo_proc *kp = procstat_getprocs(pstat, KERN_PROC_PID, getpid(), &count);
+    if (!kp) {
+        procstat_close(pstat);
+        return static_cast<size_t>(0);
+    }
+
+    // compute
+    size_t rss_size = kp->ki_rssize*getpagesize();
+
+    // cleanup
+    procstat_freeprocs(pstat, kp);
+    procstat_close(pstat);
+
+    return rss_size;
 #else
     long rss  = 0;
     FILE * fp = nullptr;