Signed-off-by: Andrew Jones <drjones@xxxxxxxxxx>
---
lib/arm/asm/thread_info.h | 1 +
lib/arm/mmu.c | 5 +++--
lib/arm/processor.c | 1 +
lib/arm/smp.c | 1 +
lib/arm64/processor.c | 1 +
5 files changed, 7 insertions(+), 2 deletions(-)
diff --git a/lib/arm/asm/thread_info.h b/lib/arm/asm/thread_info.h
index c2ecebb47d5a..80ab3954a6b0 100644
--- a/lib/arm/asm/thread_info.h
+++ b/lib/arm/asm/thread_info.h
@@ -47,6 +47,7 @@ static inline void *thread_stack_alloc(void)
struct thread_info {
int cpu;
unsigned int flags;
+ void *pgtable;
#ifdef __arm__
exception_fn exception_handlers[EXCPTN_MAX];
#else
diff --git a/lib/arm/mmu.c b/lib/arm/mmu.c
index 9e420269f1c0..b9387efe0065 100644
--- a/lib/arm/mmu.c
+++ b/lib/arm/mmu.c
@@ -54,12 +54,13 @@ void mmu_mark_disabled(int cpu)
extern void asm_mmu_enable(phys_addr_t pgtable);
void mmu_enable(pgd_t *pgtable)
{
- int cpu = current_thread_info()->cpu;
+ struct thread_info *info = current_thread_info();
asm_mmu_enable(__pa(pgtable));
flush_tlb_all();
- mmu_mark_enabled(cpu);
+ info->pgtable = pgtable;
+ mmu_mark_enabled(info->cpu);
}
extern void asm_mmu_disable(void);
diff --git a/lib/arm/processor.c b/lib/arm/processor.c
index 54fdb87ef019..49e852c06527 100644
--- a/lib/arm/processor.c
+++ b/lib/arm/processor.c
@@ -124,6 +124,7 @@ void start_usr(void (*func)(void *arg), void *arg, unsigned long sp_usr)
sp_usr &= (~7UL); /* stack ptr needs 8-byte alignment */
thread_info_init(thread_info_sp(sp_usr), TIF_USER_MODE);
+ thread_info_sp(sp_usr)->pgtable = current_thread_info()->pgtable;
asm volatile(
"mrs r0, cpsr\n"
diff --git a/lib/arm/smp.c b/lib/arm/smp.c
index 3c4e307489b2..27f6fcd07109 100644
--- a/lib/arm/smp.c
+++ b/lib/arm/smp.c
@@ -33,6 +33,7 @@ secondary_entry_fn secondary_cinit(void)
secondary_entry_fn entry;
thread_info_init(ti, 0);
+ ti->pgtable = mmu_idmap;
mmu_mark_enabled(ti->cpu);
/*
diff --git a/lib/arm64/processor.c b/lib/arm64/processor.c
index 165980a97d44..96995ccb7733 100644
--- a/lib/arm64/processor.c
+++ b/lib/arm64/processor.c
@@ -238,6 +238,7 @@ void start_usr(void (*func)(void *arg), void *arg, unsigned long sp_usr)
sp_usr &= (~15UL); /* stack ptr needs 16-byte alignment */
__thread_info_init(thread_info_sp(sp_usr), TIF_USER_MODE);
+ thread_info_sp(sp_usr)->pgtable = current_thread_info()->pgtable;
asm volatile(
"mov x0, %0\n"
--
2.13.6
