1 files changed, 8 insertions, 3 deletions

diff --git a/arch/s390/kernel/ipl.c b/arch/s390/kernel/ipl.c
index 39badb9ca0b3..5c8651f36509 100644
--- a/arch/s390/kernel/ipl.c
+++ b/arch/s390/kernel/ipl.c
@@ -2074,7 +2074,8 @@ static void do_reset_calls(void)
 
 u32 dump_prefix_page;
 
-void s390_reset_system(void (*func)(void *), void *data)
+void s390_reset_system(void (*fn_pre)(void),
+                       void (*fn_post)(void *), void *data)
 {
         struct _lowcore *lc;
 
@@ -2112,7 +2113,11 @@ void s390_reset_system(void (*func)(void *), void *data)
         /* Store status at absolute zero */
         store_status();
 
+        /* Call function before reset */
+        if (fn_pre)
+                fn_pre();
         do_reset_calls();
-        if (func)
-                func(data);
+        /* Call function after reset */
+        if (fn_post)
+                fn_post(data);
 }