diff mbox

[PATCHv3] arm64: Rework valid_user_regs

Message ID 1455646517-22903-1-git-send-email-mark.rutland@arm.com
State Superseded
Headers show

Commit Message

Mark Rutland Feb. 16, 2016, 6:15 p.m. UTC
We validate pstate using PSR_MODE32_BIT, which is part of the
user-provided pstate (and cannot be trusted). Also, we conflate
validation of AArch32 and AArch64 pstate values, making the code
difficult to reason about.

Instead, validate the pstate value based on the associated task. The
task may or may not be current (e.g. when using ptrace), so this must be
passed explicitly by callers. To avoid circular header dependencies via
sched.h, is_compat_task is pulled out of asm/ptrace.h.

To make the code possible to reason about, the AArch64 and AArch32
validation is split into separate functions. Software must respect the
RES0 policy for SPSR bits, and thus the kernel mirrors the hardware
policy (RAZ/WI) for bits as-yet unallocated. When these acquire an
architected meaning writes may be permitted (potentially with additional
validation).

Signed-off-by: Mark Rutland <mark.rutland@arm.com>

Cc: Catalin Marinas <catalin.marinas@arm.com>
Cc: Dave Martin <dave.martin@arm.com>
Cc: James Morse <james.morse@arm.com>
Cc: Peter Maydell <peter.maydell@linaro.org>
Cc: Will Deacon <will.deacon@arm.com>
---
 arch/arm64/include/asm/ptrace.h | 33 +++---------------
 arch/arm64/kernel/ptrace.c      | 77 +++++++++++++++++++++++++++++++++++++++--
 arch/arm64/kernel/signal.c      |  4 +--
 arch/arm64/kernel/signal32.c    |  2 +-
 4 files changed, 82 insertions(+), 34 deletions(-)

Since v1 [1]:
* Clear the full complement of RES0 bits
* Treat IL as RES0
* Treat SS as RES0 when not tracing
* renames: s/aa64/native, s/aa32/compat/

Since v2 [2]:
* Kill PSR_{f,s,x} and refer to fields explicitly
* Add COMPAT_PSR_GE_MASK

Mark.

[1] http://lists.infradead.org/pipermail/linux-arm-kernel/2016-February/406424.html
[2] http://lists.infradead.org/pipermail/linux-arm-kernel/2016-February/407386.html

-- 
1.9.1


_______________________________________________
linux-arm-kernel mailing list
linux-arm-kernel@lists.infradead.org
http://lists.infradead.org/mailman/listinfo/linux-arm-kernel

Comments

Will Deacon Feb. 16, 2016, 6:20 p.m. UTC | #1
On Tue, Feb 16, 2016 at 06:15:17PM +0000, Mark Rutland wrote:
> We validate pstate using PSR_MODE32_BIT, which is part of the

> user-provided pstate (and cannot be trusted). Also, we conflate

> validation of AArch32 and AArch64 pstate values, making the code

> difficult to reason about.

> 

> Instead, validate the pstate value based on the associated task. The

> task may or may not be current (e.g. when using ptrace), so this must be

> passed explicitly by callers. To avoid circular header dependencies via

> sched.h, is_compat_task is pulled out of asm/ptrace.h.

> 

> To make the code possible to reason about, the AArch64 and AArch32

> validation is split into separate functions. Software must respect the

> RES0 policy for SPSR bits, and thus the kernel mirrors the hardware

> policy (RAZ/WI) for bits as-yet unallocated. When these acquire an

> architected meaning writes may be permitted (potentially with additional

> validation).

> 

> Signed-off-by: Mark Rutland <mark.rutland@arm.com>

> Cc: Catalin Marinas <catalin.marinas@arm.com>

> Cc: Dave Martin <dave.martin@arm.com>

> Cc: James Morse <james.morse@arm.com>

> Cc: Peter Maydell <peter.maydell@linaro.org>

> Cc: Will Deacon <will.deacon@arm.com>

> ---

>  arch/arm64/include/asm/ptrace.h | 33 +++---------------

>  arch/arm64/kernel/ptrace.c      | 77 +++++++++++++++++++++++++++++++++++++++--

>  arch/arm64/kernel/signal.c      |  4 +--

>  arch/arm64/kernel/signal32.c    |  2 +-

>  4 files changed, 82 insertions(+), 34 deletions(-)


[...]

> diff --git a/arch/arm64/kernel/ptrace.c b/arch/arm64/kernel/ptrace.c

> index ff7f132..5026ffb 100644

> --- a/arch/arm64/kernel/ptrace.c

> +++ b/arch/arm64/kernel/ptrace.c

> @@ -500,7 +500,7 @@ static int gpr_set(struct task_struct *target, const struct user_regset *regset,

>  	if (ret)

>  		return ret;

>  

> -	if (!valid_user_regs(&newregs))

> +	if (!valid_user_regs(&newregs, target))

>  		return -EINVAL;

>  

>  	task_pt_regs(target)->user_regs = newregs;

> @@ -770,7 +770,7 @@ static int compat_gpr_set(struct task_struct *target,

>  

>  	}

>  

> -	if (valid_user_regs(&newregs.user_regs))

> +	if (valid_user_regs(&newregs.user_regs, target))

>  		*task_pt_regs(target) = newregs;

>  	else

>  		ret = -EINVAL;

> @@ -1272,3 +1272,76 @@ asmlinkage void syscall_trace_exit(struct pt_regs *regs)

>  	if (test_thread_flag(TIF_SYSCALL_TRACE))

>  		tracehook_report_syscall(regs, PTRACE_SYSCALL_EXIT);

>  }

> +

> +/*

> + * Bits which are always architecturally RES0 per ARM DDI 0487A.h

> + * Userspace cannot use these until they have an architectural meaning.

> + * We also reserve IL for the kernel; SS is handled dynamically.

> + */

> +#define SPSR_EL1_AARCH64_RES0_BITS \

> +	(GENMASK_ULL(63,32) | GENMASK_ULL(27, 22) | GENMASK_ULL(20, 10) | \

> +	 GENMASK_ULL(5, 5))

> +#define SPSR_EL1_AARCH32_RES0_BITS \

> +	(GENMASK_ULL(63,32) | GENMASK_ULL(24, 22) | GENMASK_ULL(20,20))

> +

> +static int valid_compat_regs(struct user_pt_regs *regs)

> +{

> +	regs->pstate &= ~SPSR_EL1_AARCH32_RES0_BITS;

> +

> +	if (!system_supports_mixed_endian_el0()) {

> +		if (IS_ENABLED(CONFIG_CPU_BIG_ENDIAN))

> +			regs->pstate |= COMPAT_PSR_E_BIT;

> +		else

> +			regs->pstate &= ~COMPAT_PSR_E_BIT;

> +	}

> +

> +	if (user_mode(regs) && (regs->pstate & PSR_MODE32_BIT) &&

> +	    (regs->pstate & COMPAT_PSR_A_BIT) == 0 &&

> +	    (regs->pstate & COMPAT_PSR_I_BIT) == 0 &&

> +	    (regs->pstate & COMPAT_PSR_F_BIT) == 0) {

> +		return 1;

> +	}

> +

> +	/* Force PSR to a valid 32-bit EL0t */

> +	regs->pstate &= COMPAT_PSR_N_BIT | COMPAT_PSR_Z_BIT |

> +			COMPAT_PSR_C_BIT | COMPAT_PSR_V_BIT |

> +			COMPAT_PSR_Q_BIT | COMPAT_PSR_IT_MASK |

> +			COMPAT_PSR_GE_MASK | COMPAT_PSR_E_BIT |

> +			COMPAT_PSR_T_BIT;

> +	regs->pstate |= PSR_MODE32_BIT;


Might be worth an explicit comment to say that we're mirroring arch/arm/
behaviour here.

> +

> +	return 0;

> +}

> +

> +static int valid_native_regs(struct user_pt_regs *regs)

> +{

> +	regs->pstate &= ~SPSR_EL1_AARCH64_RES0_BITS;

> +

> +	if (user_mode(regs) && !(regs->pstate & PSR_MODE32_BIT) &&

> +	    (regs->pstate & PSR_D_BIT) == 0 &&

> +	    (regs->pstate & PSR_A_BIT) == 0 &&

> +	    (regs->pstate & PSR_I_BIT) == 0 &&

> +	    (regs->pstate & PSR_F_BIT) == 0) {

> +		return 1;

> +	}

> +

> +	/* Force PSR to a valid 64-bit EL0t */

> +	regs->pstate &= PSR_N_BIT | PSR_Z_BIT | PSR_C_BIT | PSR_V_BIT;


Can we not just zap the pstate to PSR_MODE_EL0t and be done with it?

Will

_______________________________________________
linux-arm-kernel mailing list
linux-arm-kernel@lists.infradead.org
http://lists.infradead.org/mailman/listinfo/linux-arm-kernel
Will Deacon Feb. 29, 2016, 7:08 p.m. UTC | #2
Mark,

On Tue, Feb 16, 2016 at 06:20:05PM +0000, Will Deacon wrote:
> On Tue, Feb 16, 2016 at 06:15:17PM +0000, Mark Rutland wrote:

> > We validate pstate using PSR_MODE32_BIT, which is part of the

> > user-provided pstate (and cannot be trusted). Also, we conflate

> > validation of AArch32 and AArch64 pstate values, making the code

> > difficult to reason about.

> > 

> > Instead, validate the pstate value based on the associated task. The

> > task may or may not be current (e.g. when using ptrace), so this must be

> > passed explicitly by callers. To avoid circular header dependencies via

> > sched.h, is_compat_task is pulled out of asm/ptrace.h.

> > 

> > To make the code possible to reason about, the AArch64 and AArch32

> > validation is split into separate functions. Software must respect the

> > RES0 policy for SPSR bits, and thus the kernel mirrors the hardware

> > policy (RAZ/WI) for bits as-yet unallocated. When these acquire an

> > architected meaning writes may be permitted (potentially with additional

> > validation).

> > 

> > Signed-off-by: Mark Rutland <mark.rutland@arm.com>

> > Cc: Catalin Marinas <catalin.marinas@arm.com>

> > Cc: Dave Martin <dave.martin@arm.com>

> > Cc: James Morse <james.morse@arm.com>

> > Cc: Peter Maydell <peter.maydell@linaro.org>

> > Cc: Will Deacon <will.deacon@arm.com>


[...]

> > +static int valid_compat_regs(struct user_pt_regs *regs)

> > +{

> > +	regs->pstate &= ~SPSR_EL1_AARCH32_RES0_BITS;

> > +

> > +	if (!system_supports_mixed_endian_el0()) {

> > +		if (IS_ENABLED(CONFIG_CPU_BIG_ENDIAN))

> > +			regs->pstate |= COMPAT_PSR_E_BIT;

> > +		else

> > +			regs->pstate &= ~COMPAT_PSR_E_BIT;

> > +	}

> > +

> > +	if (user_mode(regs) && (regs->pstate & PSR_MODE32_BIT) &&

> > +	    (regs->pstate & COMPAT_PSR_A_BIT) == 0 &&

> > +	    (regs->pstate & COMPAT_PSR_I_BIT) == 0 &&

> > +	    (regs->pstate & COMPAT_PSR_F_BIT) == 0) {

> > +		return 1;

> > +	}

> > +

> > +	/* Force PSR to a valid 32-bit EL0t */

> > +	regs->pstate &= COMPAT_PSR_N_BIT | COMPAT_PSR_Z_BIT |

> > +			COMPAT_PSR_C_BIT | COMPAT_PSR_V_BIT |

> > +			COMPAT_PSR_Q_BIT | COMPAT_PSR_IT_MASK |

> > +			COMPAT_PSR_GE_MASK | COMPAT_PSR_E_BIT |

> > +			COMPAT_PSR_T_BIT;

> > +	regs->pstate |= PSR_MODE32_BIT;

> 

> Might be worth an explicit comment to say that we're mirroring arch/arm/

> behaviour here.

> 

> > +

> > +	return 0;

> > +}

> > +

> > +static int valid_native_regs(struct user_pt_regs *regs)

> > +{

> > +	regs->pstate &= ~SPSR_EL1_AARCH64_RES0_BITS;

> > +

> > +	if (user_mode(regs) && !(regs->pstate & PSR_MODE32_BIT) &&

> > +	    (regs->pstate & PSR_D_BIT) == 0 &&

> > +	    (regs->pstate & PSR_A_BIT) == 0 &&

> > +	    (regs->pstate & PSR_I_BIT) == 0 &&

> > +	    (regs->pstate & PSR_F_BIT) == 0) {

> > +		return 1;

> > +	}

> > +

> > +	/* Force PSR to a valid 64-bit EL0t */

> > +	regs->pstate &= PSR_N_BIT | PSR_Z_BIT | PSR_C_BIT | PSR_V_BIT;

> 

> Can we not just zap the pstate to PSR_MODE_EL0t and be done with it?


Are you planning to respin this patch?

Will

_______________________________________________
linux-arm-kernel mailing list
linux-arm-kernel@lists.infradead.org
http://lists.infradead.org/mailman/listinfo/linux-arm-kernel
Mark Rutland March 1, 2016, 12:47 p.m. UTC | #3
On Tue, Feb 16, 2016 at 06:20:05PM +0000, Will Deacon wrote:
> On Tue, Feb 16, 2016 at 06:15:17PM +0000, Mark Rutland wrote:

> > We validate pstate using PSR_MODE32_BIT, which is part of the

> > user-provided pstate (and cannot be trusted). Also, we conflate

> > validation of AArch32 and AArch64 pstate values, making the code

> > difficult to reason about.

> > 

> > Instead, validate the pstate value based on the associated task. The

> > task may or may not be current (e.g. when using ptrace), so this must be

> > passed explicitly by callers. To avoid circular header dependencies via

> > sched.h, is_compat_task is pulled out of asm/ptrace.h.

> > 

> > To make the code possible to reason about, the AArch64 and AArch32

> > validation is split into separate functions. Software must respect the

> > RES0 policy for SPSR bits, and thus the kernel mirrors the hardware

> > policy (RAZ/WI) for bits as-yet unallocated. When these acquire an

> > architected meaning writes may be permitted (potentially with additional

> > validation).

> > 

> > Signed-off-by: Mark Rutland <mark.rutland@arm.com>

> > Cc: Catalin Marinas <catalin.marinas@arm.com>

> > Cc: Dave Martin <dave.martin@arm.com>

> > Cc: James Morse <james.morse@arm.com>

> > Cc: Peter Maydell <peter.maydell@linaro.org>

> > Cc: Will Deacon <will.deacon@arm.com>

> > ---

> >  arch/arm64/include/asm/ptrace.h | 33 +++---------------

> >  arch/arm64/kernel/ptrace.c      | 77 +++++++++++++++++++++++++++++++++++++++--

> >  arch/arm64/kernel/signal.c      |  4 +--

> >  arch/arm64/kernel/signal32.c    |  2 +-

> >  4 files changed, 82 insertions(+), 34 deletions(-)

> 

> [...]


> > +	/* Force PSR to a valid 32-bit EL0t */

> > +	regs->pstate &= COMPAT_PSR_N_BIT | COMPAT_PSR_Z_BIT |

> > +			COMPAT_PSR_C_BIT | COMPAT_PSR_V_BIT |

> > +			COMPAT_PSR_Q_BIT | COMPAT_PSR_IT_MASK |

> > +			COMPAT_PSR_GE_MASK | COMPAT_PSR_E_BIT |

> > +			COMPAT_PSR_T_BIT;

> > +	regs->pstate |= PSR_MODE32_BIT;

> 

> Might be worth an explicit comment to say that we're mirroring arch/arm/

> behaviour here.


Done.

> > +static int valid_native_regs(struct user_pt_regs *regs)

> > +{

> > +	regs->pstate &= ~SPSR_EL1_AARCH64_RES0_BITS;

> > +

> > +	if (user_mode(regs) && !(regs->pstate & PSR_MODE32_BIT) &&

> > +	    (regs->pstate & PSR_D_BIT) == 0 &&

> > +	    (regs->pstate & PSR_A_BIT) == 0 &&

> > +	    (regs->pstate & PSR_I_BIT) == 0 &&

> > +	    (regs->pstate & PSR_F_BIT) == 0) {

> > +		return 1;

> > +	}

> > +

> > +	/* Force PSR to a valid 64-bit EL0t */

> > +	regs->pstate &= PSR_N_BIT | PSR_Z_BIT | PSR_C_BIT | PSR_V_BIT;

> 

> Can we not just zap the pstate to PSR_MODE_EL0t and be done with it?


I'm worried that some userspace might be relying on these being
preserved. If that turns out to be the case, then adding them back could
break userspace that in the meantime ended up relying on these being
zeroed.

If we're certain that no-one is relying on these, I can zero them.
Otherwise, while it would look neater I'm not sure that we gain much
relative to the potential pain we might be causing ourselves.

Thanks,
Mark.

_______________________________________________
linux-arm-kernel mailing list
linux-arm-kernel@lists.infradead.org
http://lists.infradead.org/mailman/listinfo/linux-arm-kernel
Peter Maydell March 1, 2016, 1:08 p.m. UTC | #4
On 1 March 2016 at 12:47, Mark Rutland <mark.rutland@arm.com> wrote:
> On Tue, Feb 16, 2016 at 06:20:05PM +0000, Will Deacon wrote:

>> > +static int valid_native_regs(struct user_pt_regs *regs)

>> > +{

>> > +   regs->pstate &= ~SPSR_EL1_AARCH64_RES0_BITS;

>> > +

>> > +   if (user_mode(regs) && !(regs->pstate & PSR_MODE32_BIT) &&

>> > +       (regs->pstate & PSR_D_BIT) == 0 &&

>> > +       (regs->pstate & PSR_A_BIT) == 0 &&

>> > +       (regs->pstate & PSR_I_BIT) == 0 &&

>> > +       (regs->pstate & PSR_F_BIT) == 0) {

>> > +           return 1;

>> > +   }

>> > +

>> > +   /* Force PSR to a valid 64-bit EL0t */

>> > +   regs->pstate &= PSR_N_BIT | PSR_Z_BIT | PSR_C_BIT | PSR_V_BIT;

>>

>> Can we not just zap the pstate to PSR_MODE_EL0t and be done with it?

>

> I'm worried that some userspace might be relying on these being

> preserved.


This function is called as part of signal-return, right?
You clearly can't just zap the flag registers in that code
path, because you'd then be corrupting the flags of the
bit of userspace code that was interrupted by the signal.
(Or am I missing something?)

thanks
-- PMM

_______________________________________________
linux-arm-kernel mailing list
linux-arm-kernel@lists.infradead.org
http://lists.infradead.org/mailman/listinfo/linux-arm-kernel
Will Deacon March 1, 2016, 1:40 p.m. UTC | #5
On Tue, Mar 01, 2016 at 01:08:33PM +0000, Peter Maydell wrote:
> On 1 March 2016 at 12:47, Mark Rutland <mark.rutland@arm.com> wrote:

> > On Tue, Feb 16, 2016 at 06:20:05PM +0000, Will Deacon wrote:

> >> > +static int valid_native_regs(struct user_pt_regs *regs)

> >> > +{

> >> > +   regs->pstate &= ~SPSR_EL1_AARCH64_RES0_BITS;

> >> > +

> >> > +   if (user_mode(regs) && !(regs->pstate & PSR_MODE32_BIT) &&

> >> > +       (regs->pstate & PSR_D_BIT) == 0 &&

> >> > +       (regs->pstate & PSR_A_BIT) == 0 &&

> >> > +       (regs->pstate & PSR_I_BIT) == 0 &&

> >> > +       (regs->pstate & PSR_F_BIT) == 0) {

> >> > +           return 1;

> >> > +   }

> >> > +

> >> > +   /* Force PSR to a valid 64-bit EL0t */

> >> > +   regs->pstate &= PSR_N_BIT | PSR_Z_BIT | PSR_C_BIT | PSR_V_BIT;

> >>

> >> Can we not just zap the pstate to PSR_MODE_EL0t and be done with it?

> >

> > I'm worried that some userspace might be relying on these being

> > preserved.

> 

> This function is called as part of signal-return, right?

> You clearly can't just zap the flag registers in that code

> path, because you'd then be corrupting the flags of the

> bit of userspace code that was interrupted by the signal.

> (Or am I missing something?)


Well, it would only occur if the signal handler had tried to set pstate
to an invalid value. That said, it is a change in behaviour, so we can
leave it as Mark has suggested.

Will

_______________________________________________
linux-arm-kernel mailing list
linux-arm-kernel@lists.infradead.org
http://lists.infradead.org/mailman/listinfo/linux-arm-kernel
Mark Rutland March 1, 2016, 2:01 p.m. UTC | #6
On Tue, Mar 01, 2016 at 01:40:44PM +0000, Will Deacon wrote:
> On Tue, Mar 01, 2016 at 01:08:33PM +0000, Peter Maydell wrote:

> > On 1 March 2016 at 12:47, Mark Rutland <mark.rutland@arm.com> wrote:

> > > On Tue, Feb 16, 2016 at 06:20:05PM +0000, Will Deacon wrote:

> > >> > +static int valid_native_regs(struct user_pt_regs *regs)

> > >> > +{

> > >> > +   regs->pstate &= ~SPSR_EL1_AARCH64_RES0_BITS;

> > >> > +

> > >> > +   if (user_mode(regs) && !(regs->pstate & PSR_MODE32_BIT) &&

> > >> > +       (regs->pstate & PSR_D_BIT) == 0 &&

> > >> > +       (regs->pstate & PSR_A_BIT) == 0 &&

> > >> > +       (regs->pstate & PSR_I_BIT) == 0 &&

> > >> > +       (regs->pstate & PSR_F_BIT) == 0) {

> > >> > +           return 1;

> > >> > +   }

> > >> > +

> > >> > +   /* Force PSR to a valid 64-bit EL0t */

> > >> > +   regs->pstate &= PSR_N_BIT | PSR_Z_BIT | PSR_C_BIT | PSR_V_BIT;

> > >>

> > >> Can we not just zap the pstate to PSR_MODE_EL0t and be done with it?

> > >

> > > I'm worried that some userspace might be relying on these being

> > > preserved.

> > 

> > This function is called as part of signal-return, right?

> > You clearly can't just zap the flag registers in that code

> > path, because you'd then be corrupting the flags of the

> > bit of userspace code that was interrupted by the signal.

> > (Or am I missing something?)

> 

> Well, it would only occur if the signal handler had tried to set pstate

> to an invalid value. That said, it is a change in behaviour, so we can

> leave it as Mark has suggested.


Ok. Assuming you're happy with that I'll post a v4 shortly with the
AArch32 comment, and the forced AArch64 state left as in v3.

Thanks,
Mark.

_______________________________________________
linux-arm-kernel mailing list
linux-arm-kernel@lists.infradead.org
http://lists.infradead.org/mailman/listinfo/linux-arm-kernel
diff mbox

Patch

diff --git a/arch/arm64/include/asm/ptrace.h b/arch/arm64/include/asm/ptrace.h
index e9e5467..a307eb6 100644
--- a/arch/arm64/include/asm/ptrace.h
+++ b/arch/arm64/include/asm/ptrace.h
@@ -58,6 +58,7 @@ 
 #define COMPAT_PSR_Z_BIT	0x40000000
 #define COMPAT_PSR_N_BIT	0x80000000
 #define COMPAT_PSR_IT_MASK	0x0600fc00	/* If-Then execution state mask */
+#define COMPAT_PSR_GE_MASK	0x000f0000
 
 #ifdef CONFIG_CPU_BIG_ENDIAN
 #define COMPAT_PSR_ENDSTATE	COMPAT_PSR_E_BIT
@@ -151,35 +152,9 @@  static inline unsigned long regs_return_value(struct pt_regs *regs)
 	return regs->regs[0];
 }
 
-/*
- * Are the current registers suitable for user mode? (used to maintain
- * security in signal handlers)
- */
-static inline int valid_user_regs(struct user_pt_regs *regs)
-{
-	if (user_mode(regs) && (regs->pstate & PSR_I_BIT) == 0) {
-		regs->pstate &= ~(PSR_F_BIT | PSR_A_BIT);
-
-		/* The T bit is reserved for AArch64 */
-		if (!(regs->pstate & PSR_MODE32_BIT))
-			regs->pstate &= ~COMPAT_PSR_T_BIT;
-
-		return 1;
-	}
-
-	/*
-	 * Force PSR to something logical...
-	 */
-	regs->pstate &= PSR_f | PSR_s | (PSR_x & ~PSR_A_BIT) | \
-			COMPAT_PSR_T_BIT | PSR_MODE32_BIT;
-
-	if (!(regs->pstate & PSR_MODE32_BIT)) {
-		regs->pstate &= ~COMPAT_PSR_T_BIT;
-		regs->pstate |= PSR_MODE_EL0t;
-	}
-
-	return 0;
-}
+/* We must avoid circular header include via sched.h */
+struct task_struct;
+int valid_user_regs(struct user_pt_regs *regs, struct task_struct *task);
 
 #define instruction_pointer(regs)	((unsigned long)(regs)->pc)
 
diff --git a/arch/arm64/kernel/ptrace.c b/arch/arm64/kernel/ptrace.c
index ff7f132..5026ffb 100644
--- a/arch/arm64/kernel/ptrace.c
+++ b/arch/arm64/kernel/ptrace.c
@@ -500,7 +500,7 @@  static int gpr_set(struct task_struct *target, const struct user_regset *regset,
 	if (ret)
 		return ret;
 
-	if (!valid_user_regs(&newregs))
+	if (!valid_user_regs(&newregs, target))
 		return -EINVAL;
 
 	task_pt_regs(target)->user_regs = newregs;
@@ -770,7 +770,7 @@  static int compat_gpr_set(struct task_struct *target,
 
 	}
 
-	if (valid_user_regs(&newregs.user_regs))
+	if (valid_user_regs(&newregs.user_regs, target))
 		*task_pt_regs(target) = newregs;
 	else
 		ret = -EINVAL;
@@ -1272,3 +1272,76 @@  asmlinkage void syscall_trace_exit(struct pt_regs *regs)
 	if (test_thread_flag(TIF_SYSCALL_TRACE))
 		tracehook_report_syscall(regs, PTRACE_SYSCALL_EXIT);
 }
+
+/*
+ * Bits which are always architecturally RES0 per ARM DDI 0487A.h
+ * Userspace cannot use these until they have an architectural meaning.
+ * We also reserve IL for the kernel; SS is handled dynamically.
+ */
+#define SPSR_EL1_AARCH64_RES0_BITS \
+	(GENMASK_ULL(63,32) | GENMASK_ULL(27, 22) | GENMASK_ULL(20, 10) | \
+	 GENMASK_ULL(5, 5))
+#define SPSR_EL1_AARCH32_RES0_BITS \
+	(GENMASK_ULL(63,32) | GENMASK_ULL(24, 22) | GENMASK_ULL(20,20))
+
+static int valid_compat_regs(struct user_pt_regs *regs)
+{
+	regs->pstate &= ~SPSR_EL1_AARCH32_RES0_BITS;
+
+	if (!system_supports_mixed_endian_el0()) {
+		if (IS_ENABLED(CONFIG_CPU_BIG_ENDIAN))
+			regs->pstate |= COMPAT_PSR_E_BIT;
+		else
+			regs->pstate &= ~COMPAT_PSR_E_BIT;
+	}
+
+	if (user_mode(regs) && (regs->pstate & PSR_MODE32_BIT) &&
+	    (regs->pstate & COMPAT_PSR_A_BIT) == 0 &&
+	    (regs->pstate & COMPAT_PSR_I_BIT) == 0 &&
+	    (regs->pstate & COMPAT_PSR_F_BIT) == 0) {
+		return 1;
+	}
+
+	/* Force PSR to a valid 32-bit EL0t */
+	regs->pstate &= COMPAT_PSR_N_BIT | COMPAT_PSR_Z_BIT |
+			COMPAT_PSR_C_BIT | COMPAT_PSR_V_BIT |
+			COMPAT_PSR_Q_BIT | COMPAT_PSR_IT_MASK |
+			COMPAT_PSR_GE_MASK | COMPAT_PSR_E_BIT |
+			COMPAT_PSR_T_BIT;
+	regs->pstate |= PSR_MODE32_BIT;
+
+	return 0;
+}
+
+static int valid_native_regs(struct user_pt_regs *regs)
+{
+	regs->pstate &= ~SPSR_EL1_AARCH64_RES0_BITS;
+
+	if (user_mode(regs) && !(regs->pstate & PSR_MODE32_BIT) &&
+	    (regs->pstate & PSR_D_BIT) == 0 &&
+	    (regs->pstate & PSR_A_BIT) == 0 &&
+	    (regs->pstate & PSR_I_BIT) == 0 &&
+	    (regs->pstate & PSR_F_BIT) == 0) {
+		return 1;
+	}
+
+	/* Force PSR to a valid 64-bit EL0t */
+	regs->pstate &= PSR_N_BIT | PSR_Z_BIT | PSR_C_BIT | PSR_V_BIT;
+
+	return 0;
+}
+
+/*
+ * Are the current registers suitable for user mode? (used to maintain
+ * security in signal handlers)
+ */
+int valid_user_regs(struct user_pt_regs *regs, struct task_struct *task)
+{
+	if (!test_tsk_thread_flag(task, TIF_SINGLESTEP))
+		regs->pstate &= ~DBG_SPSR_SS;
+
+	if (is_compat_thread(task_thread_info(task)))
+		return valid_compat_regs(regs);
+	else
+		return valid_native_regs(regs);
+}
diff --git a/arch/arm64/kernel/signal.c b/arch/arm64/kernel/signal.c
index e18c48c..a8eafdb 100644
--- a/arch/arm64/kernel/signal.c
+++ b/arch/arm64/kernel/signal.c
@@ -115,7 +115,7 @@  static int restore_sigframe(struct pt_regs *regs,
 	 */
 	regs->syscallno = ~0UL;
 
-	err |= !valid_user_regs(&regs->user_regs);
+	err |= !valid_user_regs(&regs->user_regs, current);
 
 	if (err == 0) {
 		struct fpsimd_context *fpsimd_ctx =
@@ -307,7 +307,7 @@  static void handle_signal(struct ksignal *ksig, struct pt_regs *regs)
 	/*
 	 * Check that the resulting registers are actually sane.
 	 */
-	ret |= !valid_user_regs(&regs->user_regs);
+	ret |= !valid_user_regs(&regs->user_regs, current);
 
 	/*
 	 * Fast forward the stepping logic so we step into the signal
diff --git a/arch/arm64/kernel/signal32.c b/arch/arm64/kernel/signal32.c
index 71ef6dc..1073356 100644
--- a/arch/arm64/kernel/signal32.c
+++ b/arch/arm64/kernel/signal32.c
@@ -356,7 +356,7 @@  static int compat_restore_sigframe(struct pt_regs *regs,
 	 */
 	regs->syscallno = ~0UL;
 
-	err |= !valid_user_regs(&regs->user_regs);
+	err |= !valid_user_regs(&regs->user_regs, current);
 
 	aux = (struct compat_aux_sigframe __user *) sf->uc.uc_regspace;
 	if (err == 0)