Compilation of 'threading.c' fixed for macos

2 files changed, 63 insertions, 0 deletions

diff --git a/src/threading.c b/src/threading.c
index c0e6a55..7b75986 100644
--- a/src/threading.c
+++ b/src/threading.c
@@ -58,6 +58,10 @@ THE SOFTWARE.
   volatile bool_t sudo;
 #endif
 
+#ifdef PLATFORM_OSX
+# include "threading_osx.h"
+#endif
+
 /* Linux with older glibc (such as Debian) don't have pthread_setname_np, but have prctl
 */
 #if defined PLATFORM_LINUX
diff --git a/src/threading_osx.h b/src/threading_osx.h
new file mode 100644
index 0000000..93da8c3
--- /dev/null
+++ b/src/threading_osx.h
@@ -0,0 +1,59 @@
+/*
+ * THREADING_OSX.H
+ * http://yyshen.github.io/2015/01/18/binding_threads_to_cores_osx.html
+ */
+#ifndef __threading_osx_h__
+#define __threading_osx_h__ 1
+
+#include <mach/mach_types.h>
+#include <mach/thread_act.h>
+#include <sys/sysctl.h>
+
+#define SYSCTL_CORE_COUNT "machdep.cpu.core_count"
+
+typedef struct cpu_set {
+        uint32_t count;
+} cpu_set_t;
+
+static inline void CPU_ZERO(cpu_set_t *cs) { cs->count = 0; }
+static inline void CPU_SET(int num, cpu_set_t *cs) { cs->count |= (1 << num); }
+static inline int CPU_ISSET(int num, cpu_set_t *cs) { return (cs->count & (1 << num)); }
+
+int sched_getaffinity(pid_t pid, size_t cpu_size, cpu_set_t *cpu_set)
+{
+        int32_t core_count = 0;
+        size_t  len = sizeof(core_count);
+        int ret = sysctlbyname(SYSCTL_CORE_COUNT, &core_count, &len, 0, 0);
+        if (ret)
+        {
+//              printf("error while get core count %d\n", ret);
+                return -1;
+        }
+        cpu_set->count = 0;
+        for (int i = 0; i < core_count; i++)
+        {
+                cpu_set->count |= (1 << i);
+        }
+
+        return 0;
+}
+
+int pthread_setaffinity_np(pthread_t thread, size_t cpu_size, cpu_set_t *cpu_set)
+{
+        thread_port_t mach_thread;
+        int core = 0;
+
+        for (core = 0; core < 8 * cpu_size; core++)
+        {
+                if (CPU_ISSET(core, cpu_set))
+                        break;
+        }
+//      printf("binding to core %d\n", core);
+        thread_affinity_policy_data_t policy = { core };
+        mach_thread = pthread_mach_thread_np(thread);
+        thread_policy_set(mach_thread, THREAD_AFFINITY_POLICY, (thread_policy_t)&policy, 1);
+
+        return 0;
+}
+
+#endif