1 files changed, 41 insertions, 0 deletions
diff --git a/tools/src/smp.h b/tools/src/smp.h
new file mode 100644
index 0000000..c802f3e
--- /dev/null
+++ b/tools/src/smp.h
@@ -0,0 +1,41 @@
+/* SPDX-License-Identifier: MIT */
+
+#ifndef __SMP_H__
+#define __SMP_H__
+
+#include "types.h"
+#include "utils.h"
+
+#define MAX_CPUS 20
+
+#define SECONDARY_STACK_SIZE 0x10000
+extern u8 *secondary_stacks[MAX_CPUS];
+
+void smp_secondary_entry(void);
+
+void smp_start_secondaries(void);
+
+#define smp_call0(i, f)          smp_call4(i, f, 0, 0, 0, 0)
+#define smp_call1(i, f, a)       smp_call4(i, f, a, 0, 0, 0)
+#define smp_call2(i, f, a, b)    smp_call4(i, f, a, b, 0, 0)
+#define smp_call3(i, f, a, b, c) smp_call4(i, f, a, b, c, 0)
+
+void smp_call4(int cpu, void *func, u64 arg0, u64 arg1, u64 arg2, u64 arg3);
+
+u64 smp_wait(int cpu);
+
+bool smp_is_alive(int cpu);
+uint64_t smp_get_mpidr(int cpu);
+u64 smp_get_release_addr(int cpu);
+void smp_set_wfe_mode(bool new_mode);
+void smp_send_ipi(int cpu);
+
+static inline int smp_id(void)
+{
+    if (in_el2())
+        return mrs(TPIDR_EL2);
+    else
+        return mrs(TPIDR_EL1);
+}
+
+#endif