arm64/sve: Better handle failure to allocate SVE register storage
[linux-2.6-microblaze.git] / arch / arm64 / kernel / signal.c
1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * Based on arch/arm/kernel/signal.c
4  *
5  * Copyright (C) 1995-2009 Russell King
6  * Copyright (C) 2012 ARM Ltd.
7  */
8
9 #include <linux/cache.h>
10 #include <linux/compat.h>
11 #include <linux/errno.h>
12 #include <linux/kernel.h>
13 #include <linux/signal.h>
14 #include <linux/personality.h>
15 #include <linux/freezer.h>
16 #include <linux/stddef.h>
17 #include <linux/uaccess.h>
18 #include <linux/sizes.h>
19 #include <linux/string.h>
20 #include <linux/tracehook.h>
21 #include <linux/ratelimit.h>
22 #include <linux/syscalls.h>
23
24 #include <asm/daifflags.h>
25 #include <asm/debug-monitors.h>
26 #include <asm/elf.h>
27 #include <asm/cacheflush.h>
28 #include <asm/ucontext.h>
29 #include <asm/unistd.h>
30 #include <asm/fpsimd.h>
31 #include <asm/ptrace.h>
32 #include <asm/signal32.h>
33 #include <asm/traps.h>
34 #include <asm/vdso.h>
35
36 /*
37  * Do a signal return; undo the signal stack. These are aligned to 128-bit.
38  */
39 struct rt_sigframe {
40         struct siginfo info;
41         struct ucontext uc;
42 };
43
44 struct frame_record {
45         u64 fp;
46         u64 lr;
47 };
48
49 struct rt_sigframe_user_layout {
50         struct rt_sigframe __user *sigframe;
51         struct frame_record __user *next_frame;
52
53         unsigned long size;     /* size of allocated sigframe data */
54         unsigned long limit;    /* largest allowed size */
55
56         unsigned long fpsimd_offset;
57         unsigned long esr_offset;
58         unsigned long sve_offset;
59         unsigned long extra_offset;
60         unsigned long end_offset;
61 };
62
63 #define BASE_SIGFRAME_SIZE round_up(sizeof(struct rt_sigframe), 16)
64 #define TERMINATOR_SIZE round_up(sizeof(struct _aarch64_ctx), 16)
65 #define EXTRA_CONTEXT_SIZE round_up(sizeof(struct extra_context), 16)
66
67 static void init_user_layout(struct rt_sigframe_user_layout *user)
68 {
69         const size_t reserved_size =
70                 sizeof(user->sigframe->uc.uc_mcontext.__reserved);
71
72         memset(user, 0, sizeof(*user));
73         user->size = offsetof(struct rt_sigframe, uc.uc_mcontext.__reserved);
74
75         user->limit = user->size + reserved_size;
76
77         user->limit -= TERMINATOR_SIZE;
78         user->limit -= EXTRA_CONTEXT_SIZE;
79         /* Reserve space for extension and terminator ^ */
80 }
81
82 static size_t sigframe_size(struct rt_sigframe_user_layout const *user)
83 {
84         return round_up(max(user->size, sizeof(struct rt_sigframe)), 16);
85 }
86
87 /*
88  * Sanity limit on the approximate maximum size of signal frame we'll
89  * try to generate.  Stack alignment padding and the frame record are
90  * not taken into account.  This limit is not a guarantee and is
91  * NOT ABI.
92  */
93 #define SIGFRAME_MAXSZ SZ_64K
94
95 static int __sigframe_alloc(struct rt_sigframe_user_layout *user,
96                             unsigned long *offset, size_t size, bool extend)
97 {
98         size_t padded_size = round_up(size, 16);
99
100         if (padded_size > user->limit - user->size &&
101             !user->extra_offset &&
102             extend) {
103                 int ret;
104
105                 user->limit += EXTRA_CONTEXT_SIZE;
106                 ret = __sigframe_alloc(user, &user->extra_offset,
107                                        sizeof(struct extra_context), false);
108                 if (ret) {
109                         user->limit -= EXTRA_CONTEXT_SIZE;
110                         return ret;
111                 }
112
113                 /* Reserve space for the __reserved[] terminator */
114                 user->size += TERMINATOR_SIZE;
115
116                 /*
117                  * Allow expansion up to SIGFRAME_MAXSZ, ensuring space for
118                  * the terminator:
119                  */
120                 user->limit = SIGFRAME_MAXSZ - TERMINATOR_SIZE;
121         }
122
123         /* Still not enough space?  Bad luck! */
124         if (padded_size > user->limit - user->size)
125                 return -ENOMEM;
126
127         *offset = user->size;
128         user->size += padded_size;
129
130         return 0;
131 }
132
133 /*
134  * Allocate space for an optional record of <size> bytes in the user
135  * signal frame.  The offset from the signal frame base address to the
136  * allocated block is assigned to *offset.
137  */
138 static int sigframe_alloc(struct rt_sigframe_user_layout *user,
139                           unsigned long *offset, size_t size)
140 {
141         return __sigframe_alloc(user, offset, size, true);
142 }
143
144 /* Allocate the null terminator record and prevent further allocations */
145 static int sigframe_alloc_end(struct rt_sigframe_user_layout *user)
146 {
147         int ret;
148
149         /* Un-reserve the space reserved for the terminator: */
150         user->limit += TERMINATOR_SIZE;
151
152         ret = sigframe_alloc(user, &user->end_offset,
153                              sizeof(struct _aarch64_ctx));
154         if (ret)
155                 return ret;
156
157         /* Prevent further allocation: */
158         user->limit = user->size;
159         return 0;
160 }
161
162 static void __user *apply_user_offset(
163         struct rt_sigframe_user_layout const *user, unsigned long offset)
164 {
165         char __user *base = (char __user *)user->sigframe;
166
167         return base + offset;
168 }
169
170 static int preserve_fpsimd_context(struct fpsimd_context __user *ctx)
171 {
172         struct user_fpsimd_state const *fpsimd =
173                 &current->thread.uw.fpsimd_state;
174         int err;
175
176         /* copy the FP and status/control registers */
177         err = __copy_to_user(ctx->vregs, fpsimd->vregs, sizeof(fpsimd->vregs));
178         __put_user_error(fpsimd->fpsr, &ctx->fpsr, err);
179         __put_user_error(fpsimd->fpcr, &ctx->fpcr, err);
180
181         /* copy the magic/size information */
182         __put_user_error(FPSIMD_MAGIC, &ctx->head.magic, err);
183         __put_user_error(sizeof(struct fpsimd_context), &ctx->head.size, err);
184
185         return err ? -EFAULT : 0;
186 }
187
188 static int restore_fpsimd_context(struct fpsimd_context __user *ctx)
189 {
190         struct user_fpsimd_state fpsimd;
191         __u32 magic, size;
192         int err = 0;
193
194         /* check the magic/size information */
195         __get_user_error(magic, &ctx->head.magic, err);
196         __get_user_error(size, &ctx->head.size, err);
197         if (err)
198                 return -EFAULT;
199         if (magic != FPSIMD_MAGIC || size != sizeof(struct fpsimd_context))
200                 return -EINVAL;
201
202         /* copy the FP and status/control registers */
203         err = __copy_from_user(fpsimd.vregs, ctx->vregs,
204                                sizeof(fpsimd.vregs));
205         __get_user_error(fpsimd.fpsr, &ctx->fpsr, err);
206         __get_user_error(fpsimd.fpcr, &ctx->fpcr, err);
207
208         clear_thread_flag(TIF_SVE);
209
210         /* load the hardware registers from the fpsimd_state structure */
211         if (!err)
212                 fpsimd_update_current_state(&fpsimd);
213
214         return err ? -EFAULT : 0;
215 }
216
217
218 struct user_ctxs {
219         struct fpsimd_context __user *fpsimd;
220         struct sve_context __user *sve;
221 };
222
223 #ifdef CONFIG_ARM64_SVE
224
225 static int preserve_sve_context(struct sve_context __user *ctx)
226 {
227         int err = 0;
228         u16 reserved[ARRAY_SIZE(ctx->__reserved)];
229         unsigned int vl = current->thread.sve_vl;
230         unsigned int vq = 0;
231
232         if (test_thread_flag(TIF_SVE))
233                 vq = sve_vq_from_vl(vl);
234
235         memset(reserved, 0, sizeof(reserved));
236
237         __put_user_error(SVE_MAGIC, &ctx->head.magic, err);
238         __put_user_error(round_up(SVE_SIG_CONTEXT_SIZE(vq), 16),
239                          &ctx->head.size, err);
240         __put_user_error(vl, &ctx->vl, err);
241         BUILD_BUG_ON(sizeof(ctx->__reserved) != sizeof(reserved));
242         err |= __copy_to_user(&ctx->__reserved, reserved, sizeof(reserved));
243
244         if (vq) {
245                 /*
246                  * This assumes that the SVE state has already been saved to
247                  * the task struct by calling the function
248                  * fpsimd_signal_preserve_current_state().
249                  */
250                 err |= __copy_to_user((char __user *)ctx + SVE_SIG_REGS_OFFSET,
251                                       current->thread.sve_state,
252                                       SVE_SIG_REGS_SIZE(vq));
253         }
254
255         return err ? -EFAULT : 0;
256 }
257
258 static int restore_sve_fpsimd_context(struct user_ctxs *user)
259 {
260         int err;
261         unsigned int vq;
262         struct user_fpsimd_state fpsimd;
263         struct sve_context sve;
264
265         if (__copy_from_user(&sve, user->sve, sizeof(sve)))
266                 return -EFAULT;
267
268         if (sve.vl != current->thread.sve_vl)
269                 return -EINVAL;
270
271         if (sve.head.size <= sizeof(*user->sve)) {
272                 clear_thread_flag(TIF_SVE);
273                 goto fpsimd_only;
274         }
275
276         vq = sve_vq_from_vl(sve.vl);
277
278         if (sve.head.size < SVE_SIG_CONTEXT_SIZE(vq))
279                 return -EINVAL;
280
281         /*
282          * Careful: we are about __copy_from_user() directly into
283          * thread.sve_state with preemption enabled, so protection is
284          * needed to prevent a racing context switch from writing stale
285          * registers back over the new data.
286          */
287
288         fpsimd_flush_task_state(current);
289         /* From now, fpsimd_thread_switch() won't touch thread.sve_state */
290
291         sve_alloc(current);
292         if (!current->thread.sve_state) {
293                 clear_thread_flag(TIF_SVE);
294                 return -ENOMEM;
295         }
296
297         err = __copy_from_user(current->thread.sve_state,
298                                (char __user const *)user->sve +
299                                         SVE_SIG_REGS_OFFSET,
300                                SVE_SIG_REGS_SIZE(vq));
301         if (err)
302                 return -EFAULT;
303
304         set_thread_flag(TIF_SVE);
305
306 fpsimd_only:
307         /* copy the FP and status/control registers */
308         /* restore_sigframe() already checked that user->fpsimd != NULL. */
309         err = __copy_from_user(fpsimd.vregs, user->fpsimd->vregs,
310                                sizeof(fpsimd.vregs));
311         __get_user_error(fpsimd.fpsr, &user->fpsimd->fpsr, err);
312         __get_user_error(fpsimd.fpcr, &user->fpsimd->fpcr, err);
313
314         /* load the hardware registers from the fpsimd_state structure */
315         if (!err)
316                 fpsimd_update_current_state(&fpsimd);
317
318         return err ? -EFAULT : 0;
319 }
320
321 #else /* ! CONFIG_ARM64_SVE */
322
323 /* Turn any non-optimised out attempts to use these into a link error: */
324 extern int preserve_sve_context(void __user *ctx);
325 extern int restore_sve_fpsimd_context(struct user_ctxs *user);
326
327 #endif /* ! CONFIG_ARM64_SVE */
328
329
330 static int parse_user_sigframe(struct user_ctxs *user,
331                                struct rt_sigframe __user *sf)
332 {
333         struct sigcontext __user *const sc = &sf->uc.uc_mcontext;
334         struct _aarch64_ctx __user *head;
335         char __user *base = (char __user *)&sc->__reserved;
336         size_t offset = 0;
337         size_t limit = sizeof(sc->__reserved);
338         bool have_extra_context = false;
339         char const __user *const sfp = (char const __user *)sf;
340
341         user->fpsimd = NULL;
342         user->sve = NULL;
343
344         if (!IS_ALIGNED((unsigned long)base, 16))
345                 goto invalid;
346
347         while (1) {
348                 int err = 0;
349                 u32 magic, size;
350                 char const __user *userp;
351                 struct extra_context const __user *extra;
352                 u64 extra_datap;
353                 u32 extra_size;
354                 struct _aarch64_ctx const __user *end;
355                 u32 end_magic, end_size;
356
357                 if (limit - offset < sizeof(*head))
358                         goto invalid;
359
360                 if (!IS_ALIGNED(offset, 16))
361                         goto invalid;
362
363                 head = (struct _aarch64_ctx __user *)(base + offset);
364                 __get_user_error(magic, &head->magic, err);
365                 __get_user_error(size, &head->size, err);
366                 if (err)
367                         return err;
368
369                 if (limit - offset < size)
370                         goto invalid;
371
372                 switch (magic) {
373                 case 0:
374                         if (size)
375                                 goto invalid;
376
377                         goto done;
378
379                 case FPSIMD_MAGIC:
380                         if (!system_supports_fpsimd())
381                                 goto invalid;
382                         if (user->fpsimd)
383                                 goto invalid;
384
385                         if (size < sizeof(*user->fpsimd))
386                                 goto invalid;
387
388                         user->fpsimd = (struct fpsimd_context __user *)head;
389                         break;
390
391                 case ESR_MAGIC:
392                         /* ignore */
393                         break;
394
395                 case SVE_MAGIC:
396                         if (!system_supports_sve())
397                                 goto invalid;
398
399                         if (user->sve)
400                                 goto invalid;
401
402                         if (size < sizeof(*user->sve))
403                                 goto invalid;
404
405                         user->sve = (struct sve_context __user *)head;
406                         break;
407
408                 case EXTRA_MAGIC:
409                         if (have_extra_context)
410                                 goto invalid;
411
412                         if (size < sizeof(*extra))
413                                 goto invalid;
414
415                         userp = (char const __user *)head;
416
417                         extra = (struct extra_context const __user *)userp;
418                         userp += size;
419
420                         __get_user_error(extra_datap, &extra->datap, err);
421                         __get_user_error(extra_size, &extra->size, err);
422                         if (err)
423                                 return err;
424
425                         /* Check for the dummy terminator in __reserved[]: */
426
427                         if (limit - offset - size < TERMINATOR_SIZE)
428                                 goto invalid;
429
430                         end = (struct _aarch64_ctx const __user *)userp;
431                         userp += TERMINATOR_SIZE;
432
433                         __get_user_error(end_magic, &end->magic, err);
434                         __get_user_error(end_size, &end->size, err);
435                         if (err)
436                                 return err;
437
438                         if (end_magic || end_size)
439                                 goto invalid;
440
441                         /* Prevent looping/repeated parsing of extra_context */
442                         have_extra_context = true;
443
444                         base = (__force void __user *)extra_datap;
445                         if (!IS_ALIGNED((unsigned long)base, 16))
446                                 goto invalid;
447
448                         if (!IS_ALIGNED(extra_size, 16))
449                                 goto invalid;
450
451                         if (base != userp)
452                                 goto invalid;
453
454                         /* Reject "unreasonably large" frames: */
455                         if (extra_size > sfp + SIGFRAME_MAXSZ - userp)
456                                 goto invalid;
457
458                         /*
459                          * Ignore trailing terminator in __reserved[]
460                          * and start parsing extra data:
461                          */
462                         offset = 0;
463                         limit = extra_size;
464
465                         if (!access_ok(base, limit))
466                                 goto invalid;
467
468                         continue;
469
470                 default:
471                         goto invalid;
472                 }
473
474                 if (size < sizeof(*head))
475                         goto invalid;
476
477                 if (limit - offset < size)
478                         goto invalid;
479
480                 offset += size;
481         }
482
483 done:
484         return 0;
485
486 invalid:
487         return -EINVAL;
488 }
489
490 static int restore_sigframe(struct pt_regs *regs,
491                             struct rt_sigframe __user *sf)
492 {
493         sigset_t set;
494         int i, err;
495         struct user_ctxs user;
496
497         err = __copy_from_user(&set, &sf->uc.uc_sigmask, sizeof(set));
498         if (err == 0)
499                 set_current_blocked(&set);
500
501         for (i = 0; i < 31; i++)
502                 __get_user_error(regs->regs[i], &sf->uc.uc_mcontext.regs[i],
503                                  err);
504         __get_user_error(regs->sp, &sf->uc.uc_mcontext.sp, err);
505         __get_user_error(regs->pc, &sf->uc.uc_mcontext.pc, err);
506         __get_user_error(regs->pstate, &sf->uc.uc_mcontext.pstate, err);
507
508         /*
509          * Avoid sys_rt_sigreturn() restarting.
510          */
511         forget_syscall(regs);
512
513         err |= !valid_user_regs(&regs->user_regs, current);
514         if (err == 0)
515                 err = parse_user_sigframe(&user, sf);
516
517         if (err == 0 && system_supports_fpsimd()) {
518                 if (!user.fpsimd)
519                         return -EINVAL;
520
521                 if (user.sve) {
522                         if (!system_supports_sve())
523                                 return -EINVAL;
524
525                         err = restore_sve_fpsimd_context(&user);
526                 } else {
527                         err = restore_fpsimd_context(user.fpsimd);
528                 }
529         }
530
531         return err;
532 }
533
534 SYSCALL_DEFINE0(rt_sigreturn)
535 {
536         struct pt_regs *regs = current_pt_regs();
537         struct rt_sigframe __user *frame;
538
539         /* Always make any pending restarted system calls return -EINTR */
540         current->restart_block.fn = do_no_restart_syscall;
541
542         /*
543          * Since we stacked the signal on a 128-bit boundary, then 'sp' should
544          * be word aligned here.
545          */
546         if (regs->sp & 15)
547                 goto badframe;
548
549         frame = (struct rt_sigframe __user *)regs->sp;
550
551         if (!access_ok(frame, sizeof (*frame)))
552                 goto badframe;
553
554         if (restore_sigframe(regs, frame))
555                 goto badframe;
556
557         if (restore_altstack(&frame->uc.uc_stack))
558                 goto badframe;
559
560         return regs->regs[0];
561
562 badframe:
563         arm64_notify_segfault(regs->sp);
564         return 0;
565 }
566
567 /*
568  * Determine the layout of optional records in the signal frame
569  *
570  * add_all: if true, lays out the biggest possible signal frame for
571  *      this task; otherwise, generates a layout for the current state
572  *      of the task.
573  */
574 static int setup_sigframe_layout(struct rt_sigframe_user_layout *user,
575                                  bool add_all)
576 {
577         int err;
578
579         err = sigframe_alloc(user, &user->fpsimd_offset,
580                              sizeof(struct fpsimd_context));
581         if (err)
582                 return err;
583
584         /* fault information, if valid */
585         if (add_all || current->thread.fault_code) {
586                 err = sigframe_alloc(user, &user->esr_offset,
587                                      sizeof(struct esr_context));
588                 if (err)
589                         return err;
590         }
591
592         if (system_supports_sve()) {
593                 unsigned int vq = 0;
594
595                 if (add_all || test_thread_flag(TIF_SVE)) {
596                         int vl = sve_max_vl;
597
598                         if (!add_all)
599                                 vl = current->thread.sve_vl;
600
601                         vq = sve_vq_from_vl(vl);
602                 }
603
604                 err = sigframe_alloc(user, &user->sve_offset,
605                                      SVE_SIG_CONTEXT_SIZE(vq));
606                 if (err)
607                         return err;
608         }
609
610         return sigframe_alloc_end(user);
611 }
612
613 static int setup_sigframe(struct rt_sigframe_user_layout *user,
614                           struct pt_regs *regs, sigset_t *set)
615 {
616         int i, err = 0;
617         struct rt_sigframe __user *sf = user->sigframe;
618
619         /* set up the stack frame for unwinding */
620         __put_user_error(regs->regs[29], &user->next_frame->fp, err);
621         __put_user_error(regs->regs[30], &user->next_frame->lr, err);
622
623         for (i = 0; i < 31; i++)
624                 __put_user_error(regs->regs[i], &sf->uc.uc_mcontext.regs[i],
625                                  err);
626         __put_user_error(regs->sp, &sf->uc.uc_mcontext.sp, err);
627         __put_user_error(regs->pc, &sf->uc.uc_mcontext.pc, err);
628         __put_user_error(regs->pstate, &sf->uc.uc_mcontext.pstate, err);
629
630         __put_user_error(current->thread.fault_address, &sf->uc.uc_mcontext.fault_address, err);
631
632         err |= __copy_to_user(&sf->uc.uc_sigmask, set, sizeof(*set));
633
634         if (err == 0 && system_supports_fpsimd()) {
635                 struct fpsimd_context __user *fpsimd_ctx =
636                         apply_user_offset(user, user->fpsimd_offset);
637                 err |= preserve_fpsimd_context(fpsimd_ctx);
638         }
639
640         /* fault information, if valid */
641         if (err == 0 && user->esr_offset) {
642                 struct esr_context __user *esr_ctx =
643                         apply_user_offset(user, user->esr_offset);
644
645                 __put_user_error(ESR_MAGIC, &esr_ctx->head.magic, err);
646                 __put_user_error(sizeof(*esr_ctx), &esr_ctx->head.size, err);
647                 __put_user_error(current->thread.fault_code, &esr_ctx->esr, err);
648         }
649
650         /* Scalable Vector Extension state, if present */
651         if (system_supports_sve() && err == 0 && user->sve_offset) {
652                 struct sve_context __user *sve_ctx =
653                         apply_user_offset(user, user->sve_offset);
654                 err |= preserve_sve_context(sve_ctx);
655         }
656
657         if (err == 0 && user->extra_offset) {
658                 char __user *sfp = (char __user *)user->sigframe;
659                 char __user *userp =
660                         apply_user_offset(user, user->extra_offset);
661
662                 struct extra_context __user *extra;
663                 struct _aarch64_ctx __user *end;
664                 u64 extra_datap;
665                 u32 extra_size;
666
667                 extra = (struct extra_context __user *)userp;
668                 userp += EXTRA_CONTEXT_SIZE;
669
670                 end = (struct _aarch64_ctx __user *)userp;
671                 userp += TERMINATOR_SIZE;
672
673                 /*
674                  * extra_datap is just written to the signal frame.
675                  * The value gets cast back to a void __user *
676                  * during sigreturn.
677                  */
678                 extra_datap = (__force u64)userp;
679                 extra_size = sfp + round_up(user->size, 16) - userp;
680
681                 __put_user_error(EXTRA_MAGIC, &extra->head.magic, err);
682                 __put_user_error(EXTRA_CONTEXT_SIZE, &extra->head.size, err);
683                 __put_user_error(extra_datap, &extra->datap, err);
684                 __put_user_error(extra_size, &extra->size, err);
685
686                 /* Add the terminator */
687                 __put_user_error(0, &end->magic, err);
688                 __put_user_error(0, &end->size, err);
689         }
690
691         /* set the "end" magic */
692         if (err == 0) {
693                 struct _aarch64_ctx __user *end =
694                         apply_user_offset(user, user->end_offset);
695
696                 __put_user_error(0, &end->magic, err);
697                 __put_user_error(0, &end->size, err);
698         }
699
700         return err;
701 }
702
703 static int get_sigframe(struct rt_sigframe_user_layout *user,
704                          struct ksignal *ksig, struct pt_regs *regs)
705 {
706         unsigned long sp, sp_top;
707         int err;
708
709         init_user_layout(user);
710         err = setup_sigframe_layout(user, false);
711         if (err)
712                 return err;
713
714         sp = sp_top = sigsp(regs->sp, ksig);
715
716         sp = round_down(sp - sizeof(struct frame_record), 16);
717         user->next_frame = (struct frame_record __user *)sp;
718
719         sp = round_down(sp, 16) - sigframe_size(user);
720         user->sigframe = (struct rt_sigframe __user *)sp;
721
722         /*
723          * Check that we can actually write to the signal frame.
724          */
725         if (!access_ok(user->sigframe, sp_top - sp))
726                 return -EFAULT;
727
728         return 0;
729 }
730
731 static void setup_return(struct pt_regs *regs, struct k_sigaction *ka,
732                          struct rt_sigframe_user_layout *user, int usig)
733 {
734         __sigrestore_t sigtramp;
735
736         regs->regs[0] = usig;
737         regs->sp = (unsigned long)user->sigframe;
738         regs->regs[29] = (unsigned long)&user->next_frame->fp;
739         regs->pc = (unsigned long)ka->sa.sa_handler;
740
741         /*
742          * Signal delivery is a (wacky) indirect function call in
743          * userspace, so simulate the same setting of BTYPE as a BLR
744          * <register containing the signal handler entry point>.
745          * Signal delivery to a location in a PROT_BTI guarded page
746          * that is not a function entry point will now trigger a
747          * SIGILL in userspace.
748          *
749          * If the signal handler entry point is not in a PROT_BTI
750          * guarded page, this is harmless.
751          */
752         if (system_supports_bti()) {
753                 regs->pstate &= ~PSR_BTYPE_MASK;
754                 regs->pstate |= PSR_BTYPE_C;
755         }
756
757         /* TCO (Tag Check Override) always cleared for signal handlers */
758         regs->pstate &= ~PSR_TCO_BIT;
759
760         if (ka->sa.sa_flags & SA_RESTORER)
761                 sigtramp = ka->sa.sa_restorer;
762         else
763                 sigtramp = VDSO_SYMBOL(current->mm->context.vdso, sigtramp);
764
765         regs->regs[30] = (unsigned long)sigtramp;
766 }
767
768 static int setup_rt_frame(int usig, struct ksignal *ksig, sigset_t *set,
769                           struct pt_regs *regs)
770 {
771         struct rt_sigframe_user_layout user;
772         struct rt_sigframe __user *frame;
773         int err = 0;
774
775         fpsimd_signal_preserve_current_state();
776
777         if (get_sigframe(&user, ksig, regs))
778                 return 1;
779
780         frame = user.sigframe;
781
782         __put_user_error(0, &frame->uc.uc_flags, err);
783         __put_user_error(NULL, &frame->uc.uc_link, err);
784
785         err |= __save_altstack(&frame->uc.uc_stack, regs->sp);
786         err |= setup_sigframe(&user, regs, set);
787         if (err == 0) {
788                 setup_return(regs, &ksig->ka, &user, usig);
789                 if (ksig->ka.sa.sa_flags & SA_SIGINFO) {
790                         err |= copy_siginfo_to_user(&frame->info, &ksig->info);
791                         regs->regs[1] = (unsigned long)&frame->info;
792                         regs->regs[2] = (unsigned long)&frame->uc;
793                 }
794         }
795
796         return err;
797 }
798
799 static void setup_restart_syscall(struct pt_regs *regs)
800 {
801         if (is_compat_task())
802                 compat_setup_restart_syscall(regs);
803         else
804                 regs->regs[8] = __NR_restart_syscall;
805 }
806
807 /*
808  * OK, we're invoking a handler
809  */
810 static void handle_signal(struct ksignal *ksig, struct pt_regs *regs)
811 {
812         sigset_t *oldset = sigmask_to_save();
813         int usig = ksig->sig;
814         int ret;
815
816         rseq_signal_deliver(ksig, regs);
817
818         /*
819          * Set up the stack frame
820          */
821         if (is_compat_task()) {
822                 if (ksig->ka.sa.sa_flags & SA_SIGINFO)
823                         ret = compat_setup_rt_frame(usig, ksig, oldset, regs);
824                 else
825                         ret = compat_setup_frame(usig, ksig, oldset, regs);
826         } else {
827                 ret = setup_rt_frame(usig, ksig, oldset, regs);
828         }
829
830         /*
831          * Check that the resulting registers are actually sane.
832          */
833         ret |= !valid_user_regs(&regs->user_regs, current);
834
835         /* Step into the signal handler if we are stepping */
836         signal_setup_done(ret, ksig, test_thread_flag(TIF_SINGLESTEP));
837 }
838
839 /*
840  * Note that 'init' is a special process: it doesn't get signals it doesn't
841  * want to handle. Thus you cannot kill init even with a SIGKILL even by
842  * mistake.
843  *
844  * Note that we go through the signals twice: once to check the signals that
845  * the kernel can handle, and then we build all the user-level signal handling
846  * stack-frames in one go after that.
847  */
848 static void do_signal(struct pt_regs *regs)
849 {
850         unsigned long continue_addr = 0, restart_addr = 0;
851         int retval = 0;
852         struct ksignal ksig;
853         bool syscall = in_syscall(regs);
854
855         /*
856          * If we were from a system call, check for system call restarting...
857          */
858         if (syscall) {
859                 continue_addr = regs->pc;
860                 restart_addr = continue_addr - (compat_thumb_mode(regs) ? 2 : 4);
861                 retval = regs->regs[0];
862
863                 /*
864                  * Avoid additional syscall restarting via ret_to_user.
865                  */
866                 forget_syscall(regs);
867
868                 /*
869                  * Prepare for system call restart. We do this here so that a
870                  * debugger will see the already changed PC.
871                  */
872                 switch (retval) {
873                 case -ERESTARTNOHAND:
874                 case -ERESTARTSYS:
875                 case -ERESTARTNOINTR:
876                 case -ERESTART_RESTARTBLOCK:
877                         regs->regs[0] = regs->orig_x0;
878                         regs->pc = restart_addr;
879                         break;
880                 }
881         }
882
883         /*
884          * Get the signal to deliver. When running under ptrace, at this point
885          * the debugger may change all of our registers.
886          */
887         if (get_signal(&ksig)) {
888                 /*
889                  * Depending on the signal settings, we may need to revert the
890                  * decision to restart the system call, but skip this if a
891                  * debugger has chosen to restart at a different PC.
892                  */
893                 if (regs->pc == restart_addr &&
894                     (retval == -ERESTARTNOHAND ||
895                      retval == -ERESTART_RESTARTBLOCK ||
896                      (retval == -ERESTARTSYS &&
897                       !(ksig.ka.sa.sa_flags & SA_RESTART)))) {
898                         regs->regs[0] = -EINTR;
899                         regs->pc = continue_addr;
900                 }
901
902                 handle_signal(&ksig, regs);
903                 return;
904         }
905
906         /*
907          * Handle restarting a different system call. As above, if a debugger
908          * has chosen to restart at a different PC, ignore the restart.
909          */
910         if (syscall && regs->pc == restart_addr) {
911                 if (retval == -ERESTART_RESTARTBLOCK)
912                         setup_restart_syscall(regs);
913                 user_rewind_single_step(current);
914         }
915
916         restore_saved_sigmask();
917 }
918
919 static bool cpu_affinity_invalid(struct pt_regs *regs)
920 {
921         if (!compat_user_mode(regs))
922                 return false;
923
924         /*
925          * We're preemptible, but a reschedule will cause us to check the
926          * affinity again.
927          */
928         return !cpumask_test_cpu(raw_smp_processor_id(),
929                                  system_32bit_el0_cpumask());
930 }
931
932 asmlinkage void do_notify_resume(struct pt_regs *regs,
933                                  unsigned long thread_flags)
934 {
935         do {
936                 if (thread_flags & _TIF_NEED_RESCHED) {
937                         /* Unmask Debug and SError for the next task */
938                         local_daif_restore(DAIF_PROCCTX_NOIRQ);
939
940                         schedule();
941                 } else {
942                         local_daif_restore(DAIF_PROCCTX);
943
944                         if (thread_flags & _TIF_UPROBE)
945                                 uprobe_notify_resume(regs);
946
947                         if (thread_flags & _TIF_MTE_ASYNC_FAULT) {
948                                 clear_thread_flag(TIF_MTE_ASYNC_FAULT);
949                                 send_sig_fault(SIGSEGV, SEGV_MTEAERR,
950                                                (void __user *)NULL, current);
951                         }
952
953                         if (thread_flags & (_TIF_SIGPENDING | _TIF_NOTIFY_SIGNAL))
954                                 do_signal(regs);
955
956                         if (thread_flags & _TIF_NOTIFY_RESUME) {
957                                 tracehook_notify_resume(regs);
958                                 rseq_handle_notify_resume(NULL, regs);
959
960                                 /*
961                                  * If we reschedule after checking the affinity
962                                  * then we must ensure that TIF_NOTIFY_RESUME
963                                  * is set so that we check the affinity again.
964                                  * Since tracehook_notify_resume() clears the
965                                  * flag, ensure that the compiler doesn't move
966                                  * it after the affinity check.
967                                  */
968                                 barrier();
969
970                                 if (cpu_affinity_invalid(regs))
971                                         force_sig(SIGKILL);
972                         }
973
974                         if (thread_flags & _TIF_FOREIGN_FPSTATE)
975                                 fpsimd_restore_current_state();
976                 }
977
978                 local_daif_mask();
979                 thread_flags = READ_ONCE(current_thread_info()->flags);
980         } while (thread_flags & _TIF_WORK_MASK);
981 }
982
983 unsigned long __ro_after_init signal_minsigstksz;
984
985 /*
986  * Determine the stack space required for guaranteed signal devliery.
987  * This function is used to populate AT_MINSIGSTKSZ at process startup.
988  * cpufeatures setup is assumed to be complete.
989  */
990 void __init minsigstksz_setup(void)
991 {
992         struct rt_sigframe_user_layout user;
993
994         init_user_layout(&user);
995
996         /*
997          * If this fails, SIGFRAME_MAXSZ needs to be enlarged.  It won't
998          * be big enough, but it's our best guess:
999          */
1000         if (WARN_ON(setup_sigframe_layout(&user, true)))
1001                 return;
1002
1003         signal_minsigstksz = sigframe_size(&user) +
1004                 round_up(sizeof(struct frame_record), 16) +
1005                 16; /* max alignment padding */
1006 }