Workaround to support 32-bit Linux.

peter-lawrey · peter-lawrey · commit 5e8bdb4e52be · 2013-12-17T14:12:12.000Z
diff --git a/affinity/src/main/java/net/openhft/affinity/impl/PosixJNAAffinity.java b/affinity/src/main/java/net/openhft/affinity/impl/PosixJNAAffinity.java
@@ -17,6 +17,7 @@
 package net.openhft.affinity.impl;
 
 import com.sun.jna.*;
+import com.sun.jna.ptr.IntByReference;
 import com.sun.jna.ptr.LongByReference;
 import net.openhft.affinity.IAffinity;
 
@@ -39,10 +40,23 @@ public enum PosixJNAAffinity implements IAffinity {
     public static final boolean LOADED;
     private static final Logger LOGGER = Logger.getLogger(PosixJNAAffinity.class.getName());
     private static final String LIBRARY_NAME = Platform.isWindows() ? "msvcrt" : "c";
+    private static final int PROCS = Runtime.getRuntime().availableProcessors();
 
     @Override
     public long getAffinity() {
         final CLibrary lib = CLibrary.INSTANCE;
+        if (PROCS <= 32) {
+            // TODO where are systems with 64+ cores...
+            final IntByReference cpuset = new IntByReference(0);
+            try {
+                final int ret = lib.sched_getaffinity(0, Integer.SIZE / 8, cpuset);
+                if (ret < 0)
+                    throw new IllegalStateException("sched_getaffinity((" + Integer.SIZE / 8 + ") , &(" + cpuset + ") ) return " + ret);
+                return cpuset.getValue();
+            } catch (LastErrorException e) {
+                throw new IllegalStateException("sched_getaffinity((" + Integer.SIZE / 8 + ") , &(" + cpuset + ") ) errorNo=" + e.getErrorCode(), e);
+            }
+        }
         // TODO where are systems with 64+ cores...
         final LongByReference cpuset = new LongByReference(0L);
         try {
@@ -58,6 +72,16 @@ public long getAffinity() {
     @Override
     public void setAffinity(final long affinity) {
         final CLibrary lib = CLibrary.INSTANCE;
+        if (PROCS <= 32)
+            try {
+                final int ret = lib.sched_setaffinity(0, Integer.SIZE / 8, new IntByReference((int) affinity));
+                if (ret < 0) {
+                    throw new IllegalStateException("sched_setaffinity((" + Integer.SIZE / 8 + ") , &(" + affinity + ") ) return " + ret);
+                }
+            } catch (LastErrorException e) {
+                throw new IllegalStateException("sched_getaffinity((" + Integer.SIZE / 8 + ") , &(" + affinity + ") ) errorNo=" + e.getErrorCode(), e);
+            }
+
         try {
             //fixme: where are systems with more then 64 cores...
             final int ret = lib.sched_setaffinity(0, Long.SIZE / 8, new LongByReference(affinity));
