diff --git a/sysdeps/aarch64/Makefile b/sysdeps/aarch64/Makefile
index 9d8844d9c8..141d7d9cc2 100644
--- a/sysdeps/aarch64/Makefile
+++ b/sysdeps/aarch64/Makefile
@@ -68,6 +68,9 @@ sysdep_routines += \
__mtag_tag_zero_region \
__mtag_tag_region \
__arm_za_disable
+
+tests += \
+ tst-sme-jmp
endif
ifeq ($(subdir),malloc)
diff --git a/sysdeps/aarch64/tst-sme-jmp.c b/sysdeps/aarch64/tst-sme-jmp.c
new file mode 100644
index 0000000000..35769340e7
--- /dev/null
+++ b/sysdeps/aarch64/tst-sme-jmp.c
@@ -0,0 +1,280 @@
+/* Test for SME longjmp.
+ Copyright (C) 2023 Free Software Foundation, Inc.
+ This file is part of the GNU C Library.
+
+ The GNU C Library is free software; you can redistribute it and/or
+ modify it under the terms of the GNU Lesser General Public
+ License as published by the Free Software Foundation; either
+ version 2.1 of the License, or (at your option) any later version.
+
+ The GNU C Library is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ Lesser General Public License for more details.
+
+ You should have received a copy of the GNU Lesser General Public
+ License along with the GNU C Library; if not, see
+ . */
+
+#include
+#include
+#include
+#include
+#include
+#include
+#include
+#include
+#include
+#include
+
+struct blk {
+ void *za_save_buffer;
+ uint16_t num_za_save_slices;
+ char __reserved[6];
+};
+
+static unsigned long svl;
+static uint8_t *za_orig;
+static uint8_t *za_dump;
+static uint8_t *za_save;
+
+static unsigned long
+get_svl (void)
+{
+ register unsigned long x0 asm ("x0");
+ asm volatile (
+ ".inst 0x04bf5820 /* rdsvl x0, 1 */\n"
+ : "=r" (x0));
+ return x0;
+}
+
+/* PSTATE.ZA = 1, set ZA state to active. */
+static void
+start_za (void)
+{
+ asm volatile (
+ ".inst 0xd503457f /* smstart za */");
+}
+
+/* Read SVCR to get SM (bit0) and ZA (bit1) state. */
+static unsigned long
+get_svcr (void)
+{
+ register unsigned long x0 asm ("x0");
+ asm volatile (
+ ".inst 0xd53b4240 /* mrs x0, svcr */\n"
+ : "=r" (x0));
+ return x0;
+}
+
+/* Load data into ZA byte by byte from p. */
+static void __attribute__ ((noinline))
+load_za (const void *p)
+{
+ register unsigned long x15 asm ("x15") = 0;
+ register unsigned long x16 asm ("x16") = (unsigned long)p;
+ register unsigned long x17 asm ("x17") = svl;
+
+ asm volatile (
+ ".inst 0xd503437f /* smstart sm */\n"
+ ".L_ldr_loop:\n"
+ ".inst 0xe1006200 /* ldr za[w15, 0], [x16] */\n"
+ "add w15, w15, 1\n"
+ ".inst 0x04305030 /* addvl x16, x16, 1 */\n"
+ "cmp w15, w17\n"
+ "bne .L_ldr_loop\n"
+ ".inst 0xd503427f /* smstop sm */\n"
+ : "+r"(x15), "+r"(x16), "+r"(x17));
+}
+
+/* Set tpidr2 to BLK. */
+static void
+set_tpidr2 (struct blk *blk)
+{
+ register unsigned long x0 asm ("x0") = (unsigned long)blk;
+ asm volatile (
+ ".inst 0xd51bd0a0 /* msr tpidr2_el0, x0 */\n"
+ :: "r"(x0) : "memory");
+}
+
+/* Returns tpidr2. */
+static void *
+get_tpidr2 (void)
+{
+ register unsigned long x0 asm ("x0");
+ asm volatile (
+ ".inst 0xd53bd0a0 /* mrs x0, tpidr2_el0 */\n"
+ : "=r"(x0) :: "memory");
+ return (void *) x0;
+}
+
+static void
+print_data(const char *msg, void *p)
+{
+ unsigned char *a = p;
+ printf ("%s:\n", msg);
+ for (int i = 0; i < svl; i++)
+ {
+ printf ("%d: ", i);
+ for (int j = 0; j < svl; j++)
+ printf("%02x,", a[i*svl+j]);
+ printf("\n");
+ }
+ printf(".\n");
+ fflush (stdout);
+}
+
+__attribute__ ((noinline))
+static void
+do_longjmp (jmp_buf env)
+{
+ longjmp (env, 1);
+}
+
+__attribute__ ((noinline))
+static void
+do_setcontext (const ucontext_t *p)
+{
+ setcontext (p);
+}
+
+static void
+longjmp_test (void)
+{
+ unsigned long svcr;
+ jmp_buf env;
+ void *p;
+ int r;
+ struct blk blk = {za_save, svl, {0}};
+
+ printf ("longjmp test:\n");
+ p = get_tpidr2 ();
+ printf ("initial tp2 = %p\n", p);
+ if (p != NULL)
+ FAIL_EXIT1 ("tpidr2 is not initialized to 0");
+ svcr = get_svcr ();
+ if (svcr != 0)
+ FAIL_EXIT1 ("svcr != 0: %lu", svcr);
+ set_tpidr2 (&blk);
+ start_za ();
+ load_za (za_orig);
+
+ print_data ("za save space", za_save);
+ p = get_tpidr2 ();
+ printf ("before setjmp: tp2 = %p\n", p);
+ if (p != &blk)
+ FAIL_EXIT1 ("tpidr2 is not set to BLK %p", (void *)&blk);
+ if (setjmp (env) == 0)
+ {
+ p = get_tpidr2 ();
+ printf ("before longjmp: tp2 = %p\n", p);
+ if (p != &blk)
+ FAIL_EXIT1 ("tpidr2 is clobbered");
+ do_longjmp (env);
+ FAIL_EXIT1 ("longjmp returned");
+ }
+ p = get_tpidr2 ();
+ printf ("after longjmp: tp2 = %p\n", p);
+ if (p != NULL)
+ FAIL_EXIT1 ("tpidr2 is not set to 0");
+ svcr = get_svcr ();
+ if (svcr != 0)
+ FAIL_EXIT1 ("svcr != 0: %lu", svcr);
+ print_data ("za save space", za_save);
+ r = memcmp (za_orig, za_save, svl*svl);
+ if (r != 0)
+ FAIL_EXIT1 ("saving za failed");
+}
+
+static void
+setcontext_test (void)
+{
+ volatile int setcontext_done = 0;
+ unsigned long svcr;
+ ucontext_t ctx;
+ void *p;
+ int r;
+ struct blk blk = {za_save, svl, {0}};
+
+ printf ("setcontext test:\n");
+ p = get_tpidr2 ();
+ printf ("initial tp2 = %p\n", p);
+ if (p != NULL)
+ FAIL_EXIT1 ("tpidr2 is not initialized to 0");
+ svcr = get_svcr ();
+ if (svcr != 0)
+ FAIL_EXIT1 ("svcr != 0: %lu", svcr);
+ set_tpidr2 (&blk);
+ start_za ();
+ load_za (za_orig);
+
+ print_data ("za save space", za_save);
+ p = get_tpidr2 ();
+ printf ("before getcontext: tp2 = %p\n", p);
+ if (p != &blk)
+ FAIL_EXIT1 ("tpidr2 is not set to BLK %p", (void *)&blk);
+ r = getcontext (&ctx);
+ if (r != 0)
+ FAIL_EXIT1 ("getcontext failed");
+ if (setcontext_done == 0)
+ {
+ p = get_tpidr2 ();
+ printf ("before setcontext: tp2 = %p\n", p);
+ if (p != &blk)
+ FAIL_EXIT1 ("tpidr2 is clobbered");
+ setcontext_done = 1;
+ do_setcontext (&ctx);
+ FAIL_EXIT1 ("setcontext returned");
+ }
+ p = get_tpidr2 ();
+ printf ("after setcontext: tp2 = %p\n", p);
+ if (p != NULL)
+ FAIL_EXIT1 ("tpidr2 is not set to 0");
+ svcr = get_svcr ();
+ if (svcr != 0)
+ FAIL_EXIT1 ("svcr != 0: %lu", svcr);
+ print_data ("za save space", za_save);
+ r = memcmp (za_orig, za_save, svl*svl);
+ if (r != 0)
+ FAIL_EXIT1 ("saving za failed");
+}
+
+static int
+do_test (void)
+{
+ unsigned long hwcap2;
+
+ hwcap2 = getauxval (AT_HWCAP2);
+ if ((hwcap2 & HWCAP2_SME) == 0)
+ return EXIT_UNSUPPORTED;
+
+ svl = get_svl ();
+ printf ("svl: %lu\n", svl);
+ if (svl < 16 || svl % 16 != 0 || svl >= (1 << 16))
+ FAIL_EXIT1 ("invalid svl");
+
+ za_orig = xmalloc (svl*svl);
+ za_save = xmalloc (svl*svl);
+ za_dump = xmalloc (svl*svl);
+ memset (za_orig, 1, svl*svl);
+ memset (za_save, 2, svl*svl);
+ memset (za_dump, 3, svl*svl);
+ for (int i = 0; i < svl; i++)
+ for (int j = 0; j < svl; j++)
+ za_orig[i*svl+j] = i*svl+j;
+ print_data ("original data", za_orig);
+
+ longjmp_test ();
+
+ memset (za_save, 2, svl*svl);
+ memset (za_dump, 3, svl*svl);
+
+ setcontext_test ();
+
+ free (za_orig);
+ free (za_save);
+ free (za_dump);
+ return 0;
+}
+
+#include