1 files changed, 99 insertions, 0 deletions
diff --git a/reference/ssmalloc/include-x86_64/atomic.h b/reference/ssmalloc/include-x86_64/atomic.h
new file mode 100644
index 0000000..2431300
--- /dev/null
+++ b/reference/ssmalloc/include-x86_64/atomic.h
@@ -0,0 +1,99 @@
+#ifndef __SYNCHRO_ATOMIC_H__
+#define __SYNCHRO_ATOMIC_H__
+
+#define mb()		asm volatile ("sync" : : : "memory")
+#define LOCK_PREFIX	"lock ; "
+
+static inline unsigned long fetch_and_store(volatile unsigned int *address, unsigned int value)
+{
+	asm volatile("xchgl %k0,%1"
+		: "=r" (value)
+		: "m" (*address), "0" (value)
+		: "memory");
+
+	return value;
+}
+
+static inline int atmc_fetch_and_add(volatile unsigned int *address, int value)
+{
+	int prev = value;
+
+	asm volatile(
+		LOCK_PREFIX "xaddl %0, %1"
+		: "+r" (value), "+m" (*address)
+		: : "memory");
+
+	return prev + value;
+}
+
+static inline long long atmc_fetch_and_add64(volatile unsigned long long *address, long long value)
+{
+	int prev = value;
+
+	asm volatile(
+		LOCK_PREFIX "xaddq %0, %1"
+		: "+r" (value), "+m" (*address)
+		: : "memory");
+
+	return prev + value;
+}
+
+static inline void atmc_add32(volatile unsigned int* address, int value)
+{
+	asm volatile(
+		LOCK_PREFIX "addl %1,%0"
+		: "=m" (*address)
+		: "ir" (value), "m" (*address));
+}
+
+static inline void atmc_add64(volatile unsigned long long* address, unsigned long long value)
+{
+	asm volatile(
+		LOCK_PREFIX "addq %1,%0"
+		: "=m" (*address)
+		: "ir" (value), "m" (*address));
+}
+
+static inline unsigned int compare_and_swap32(volatile unsigned int *address, unsigned int old_value, unsigned int new_value)
+{
+	unsigned long prev = 0;
+
+	asm volatile(LOCK_PREFIX "cmpxchgl %k1,%2"
+		: "=a"(prev)
+		: "r"(new_value), "m"(*address), "0"(old_value)
+		: "memory");
+
+	return prev == old_value;
+}
+
+static inline unsigned int compare_and_swap64(volatile unsigned long long *address, unsigned long old_value, unsigned long new_value)
+{
+	unsigned long prev = 0;
+
+	asm volatile(LOCK_PREFIX "cmpxchgq %1,%2"
+		: "=a"(prev)
+		: "r"(new_value), "m"(*address), "0"(old_value)
+		: "memory");
+
+	return prev == old_value;
+}
+
+static inline unsigned long compare_and_swap64_out(volatile unsigned long long *address, unsigned long old_value, unsigned long new_value)
+{
+	unsigned long prev = 0;
+
+	asm volatile(LOCK_PREFIX "cmpxchgq %1,%2"
+		: "=a"(prev)
+		: "r"(new_value), "m"(*address), "0"(old_value)
+		: "memory");
+
+	return prev;
+}
+
+static inline unsigned long compare_and_swap_ptr(volatile void *address, void* old_ptr, void* new_ptr)
+{
+	return compare_and_swap64((volatile unsigned long long *)address, (unsigned long)old_ptr, (unsigned long)new_ptr); 
+}
+
+#endif
+