From 87c4aa1308ad369a10388ae74ea8cd08d58f8498 Mon Sep 17 00:00:00 2001
From: Owen Anderson <resistor@mac.com>
Date: Sat, 24 Aug 2024 02:19:51 +0000
Subject: [PATCH] arm64: Remember to switch address spaces on context switch.

Change-Id: Iaace0dc005b843830122bede0320d080e50ec348
Reviewed-on: https://review.haiku-os.org/c/haiku/+/8097
Reviewed-by: waddlesplash <waddlesplash@gmail.com>
---
 src/system/kernel/arch/arm64/arch_thread.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/system/kernel/arch/arm64/arch_thread.cpp b/src/system/kernel/arch/arm64/arch_thread.cpp
index 6975d0215b..a7f152e7ae 100644
--- a/src/system/kernel/arch/arm64/arch_thread.cpp
+++ b/src/system/kernel/arch/arm64/arch_thread.cpp
@@ -15,6 +15,7 @@
 #include <vm/vm_types.h>
 #include <vm/VMAddressSpace.h>
 #include <arch_vm.h>
+#include <arch/vm.h>
 #include <arch/vm_translation_map.h>
 
 #include <string.h>
@@ -96,6 +97,7 @@ extern "C" void _arch_context_swap(arch_thread *from, arch_thread *to);
 void
 arch_thread_context_switch(Thread *from, Thread *to)
 {
+	arch_vm_aspace_swap(from->team->address_space, to->team->address_space);
 	arm64_set_tls_context(to);
 	_arch_context_swap(&from->arch_info, &to->arch_info);
 }