From 788e1e92ef092478f95d6c9f53c4417d8aa1da1e Mon Sep 17 00:00:00 2001
From: znley <shanjiantao@loongson.cn>
Date: Mon, 12 Jun 2023 15:26:35 +0800
Subject: [PATCH] zdtm: add loongarch64 support

Signed-off-by: znley <shanjiantao@loongson.cn>
---
 .../lib/arch/loongarch64/include/asm/atomic.h | 49 +++++++++++++++++++
 test/zdtm/lib/test.c                          |  2 +-
 2 files changed, 50 insertions(+), 1 deletion(-)
 create mode 100644 test/zdtm/lib/arch/loongarch64/include/asm/atomic.h

diff --git a/test/zdtm/lib/arch/loongarch64/include/asm/atomic.h b/test/zdtm/lib/arch/loongarch64/include/asm/atomic.h
new file mode 100644
index 000000000..1803aaeb4
--- /dev/null
+++ b/test/zdtm/lib/arch/loongarch64/include/asm/atomic.h
@@ -0,0 +1,49 @@
+#ifndef __CR_ATOMIC_H__
+#define __CR_ATOMIC_H__
+
+typedef uint32_t atomic_t;
+
+#define atomic_get(v)	 (*(volatile int *)v)
+#define atomic_set(v, i) (*(v) = (i))
+
+static inline int __atomic_add(int i, atomic_t *v)
+{
+	int result;
+	asm volatile("amadd_db.w %1, %2, %0" : "+ZB"(*v), "=&r"(result) : "r"(i) : "memory");
+	return result + i;
+}
+
+static inline void atomic_add(int i, atomic_t *v)
+{
+	__atomic_add(i, v);
+}
+
+static inline int atomic_add_return(int i, atomic_t *v)
+{
+	return __atomic_add(i, v);
+}
+
+#define atomic_sub(i, v)	atomic_add(-(int)i, v)
+#define atomic_sub_return(i, v) atomic_add_return(-(int)i, v)
+#define atomic_inc(v)		atomic_add_return(1, v)
+#define atomic_dec(v)		atomic_sub_return(1, v)
+#define atomic_dec_return(v)	atomic_sub_return(1, v)
+
+static inline int atomic_cmpxchg(atomic_t *ptr, int old, int new)
+{
+	int ret;
+	asm volatile("1:                     \n"
+		     " ll.w  %0, %1          \n"
+		     " bne   %0, %2, 2f      \n"
+		     " or    $t0, %3, $zero  \n"
+		     " sc.w  $t0, %1         \n"
+		     " beqz  $t0, 1b         \n"
+		     "2:                     \n"
+		     " dbar  0               \n"
+		     : "=&r"(ret), "+ZB"(*ptr)
+		     : "r"(old), "r"(new)
+		     : "t0", "memory");
+	return ret;
+}
+
+#endif /* __CR_ATOMIC_H__ */
diff --git a/test/zdtm/lib/test.c b/test/zdtm/lib/test.c
index 6291ea4a7..a5ba38b2d 100644
--- a/test/zdtm/lib/test.c
+++ b/test/zdtm/lib/test.c
@@ -406,7 +406,7 @@ pid_t sys_clone_unified(unsigned long flags, void *child_stack, void *parent_tid
 {
 #ifdef __x86_64__
 	return (pid_t)syscall(__NR_clone, flags, child_stack, parent_tid, child_tid, newtls);
-#elif (__i386__ || __arm__ || __aarch64__ || __powerpc64__ || __mips__)
+#elif (__i386__ || __arm__ || __aarch64__ || __powerpc64__ || __mips__ || __loongarch64)
 	return (pid_t)syscall(__NR_clone, flags, child_stack, parent_tid, newtls, child_tid);
 #elif __s390x__
 	return (pid_t)syscall(__NR_clone, child_stack, flags, parent_tid, child_tid, newtls);