Add address environment support to ALL implementatins of up_reprioritize_rtr()

1 files changed, 13 insertions, 2 deletions

diff --git a/nuttx/arch/z80/src/common/up_reprioritizertr.c b/nuttx/arch/z80/src/common/up_reprioritizertr.c
index c470c0863..ff7cb64ab 100644
--- a/nuttx/arch/z80/src/common/up_reprioritizertr.c
+++ b/nuttx/arch/z80/src/common/up_reprioritizertr.c
@@ -1,7 +1,7 @@
 /****************************************************************************
  * arch/z80/src/common/up_reprioritizertr.c
  *
- *   Copyright (C) 2007-2009, 2013 Gregory Nutt. All rights reserved.
+ *   Copyright (C) 2007-2009, 2013-2014 Gregory Nutt. All rights reserved.
  *   Author: Gregory Nutt <gnutt@nuttx.org>
  *
  * Redistribution and use in source and binary forms, with or without
@@ -49,6 +49,7 @@
 #include "chip/chip.h"
 #include "chip/switch.h"
 #include "sched/sched.h"
+#include "group/group.h"
 #include "up_internal.h"
 
 /****************************************************************************
@@ -163,7 +164,8 @@ void up_reprioritize_rtr(FAR struct tcb_s *tcb, uint8_t priority)
               slldbg("New Active Task TCB=%p\n", rtcb);
 
               /* Then setup so that the context will be performed on exit
-               * from the interrupt.
+               * from the interrupt.  Any necessary address environment
+               * changes will be made when the interrupt returns.
                */
 
                SET_IRQCONTEXT(rtcb);
@@ -183,6 +185,15 @@ void up_reprioritize_rtr(FAR struct tcb_s *tcb, uint8_t priority)
               rtcb = (FAR struct tcb_s*)g_readytorun.head;
               slldbg("New Active Task TCB=%p\n", rtcb);
 
+#ifdef CONFIG_ARCH_ADDRENV
+              /* Make sure that the address environment for the previously
+               * running task is closed down gracefully (data caches dump,
+               * MMU flushed) and set up the address environment for the new
+               * thread at the head of the ready-to-run list.
+               */
+
+              (void)group_addrenv(rtcb);
+#endif
               /* Then switch contexts */
 
               RESTORE_USERCONTEXT(rtcb);