From e0a7471cf8ddf819ef1dd7780e6834f48ea1a434 Mon Sep 17 00:00:00 2001
From: Robin Randhawa <robin.randhawa@arm.com>
Date: Fri, 15 Jan 2021 15:50:00 +0000
Subject: [PATCH] aarch64: Add a set_tcb method to setup tpidr_el0

---
 src/context/arch/aarch64.rs | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/context/arch/aarch64.rs b/src/context/arch/aarch64.rs
index b57bfa4..5f5e2a3 100644
--- a/src/context/arch/aarch64.rs
+++ b/src/context/arch/aarch64.rs
@@ -104,6 +104,10 @@ impl Context {
         self.lr = address;
     }
 
+    pub fn set_tcb(&mut self, pid: usize) {
+        self.tpidr_el0 = (crate::USER_TCB_OFFSET + pid * crate::PAGE_SIZE);
+    }
+
     pub fn set_fp(&mut self, address: usize) {
         self.fp = address;
     }