cachepc-linux

Fork of AMDESE/linux with modifications for CachePC side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-linux
Log | Files | Refs | README | LICENSE | sfeed.txt

process.c (6013B)


      1// SPDX-License-Identifier: GPL-2.0
      2/*
      3 * This file handles the architecture dependent parts of process handling.
      4 *
      5 *    Copyright IBM Corp. 1999, 2009
      6 *    Author(s): Martin Schwidefsky <schwidefsky@de.ibm.com>,
      7 *		 Hartmut Penner <hp@de.ibm.com>,
      8 *		 Denis Joseph Barrow,
      9 */
     10
     11#include <linux/elf-randomize.h>
     12#include <linux/compiler.h>
     13#include <linux/cpu.h>
     14#include <linux/sched.h>
     15#include <linux/sched/debug.h>
     16#include <linux/sched/task.h>
     17#include <linux/sched/task_stack.h>
     18#include <linux/kernel.h>
     19#include <linux/mm.h>
     20#include <linux/elfcore.h>
     21#include <linux/smp.h>
     22#include <linux/slab.h>
     23#include <linux/interrupt.h>
     24#include <linux/tick.h>
     25#include <linux/personality.h>
     26#include <linux/syscalls.h>
     27#include <linux/compat.h>
     28#include <linux/kprobes.h>
     29#include <linux/random.h>
     30#include <linux/export.h>
     31#include <linux/init_task.h>
     32#include <linux/entry-common.h>
     33#include <asm/cpu_mf.h>
     34#include <asm/io.h>
     35#include <asm/processor.h>
     36#include <asm/vtimer.h>
     37#include <asm/exec.h>
     38#include <asm/irq.h>
     39#include <asm/nmi.h>
     40#include <asm/smp.h>
     41#include <asm/stacktrace.h>
     42#include <asm/switch_to.h>
     43#include <asm/runtime_instr.h>
     44#include <asm/unwind.h>
     45#include "entry.h"
     46
     47void ret_from_fork(void) asm("ret_from_fork");
     48
     49void __ret_from_fork(struct task_struct *prev, struct pt_regs *regs)
     50{
     51	void (*func)(void *arg);
     52
     53	schedule_tail(prev);
     54
     55	if (!user_mode(regs)) {
     56		/* Kernel thread */
     57		func = (void *)regs->gprs[9];
     58		func((void *)regs->gprs[10]);
     59	}
     60	clear_pt_regs_flag(regs, PIF_SYSCALL);
     61	syscall_exit_to_user_mode(regs);
     62}
     63
     64void flush_thread(void)
     65{
     66}
     67
     68void arch_setup_new_exec(void)
     69{
     70	if (S390_lowcore.current_pid != current->pid) {
     71		S390_lowcore.current_pid = current->pid;
     72		if (test_facility(40))
     73			lpp(&S390_lowcore.lpp);
     74	}
     75}
     76
     77void arch_release_task_struct(struct task_struct *tsk)
     78{
     79	runtime_instr_release(tsk);
     80	guarded_storage_release(tsk);
     81}
     82
     83int arch_dup_task_struct(struct task_struct *dst, struct task_struct *src)
     84{
     85	/*
     86	 * Save the floating-point or vector register state of the current
     87	 * task and set the CIF_FPU flag to lazy restore the FPU register
     88	 * state when returning to user space.
     89	 */
     90	save_fpu_regs();
     91
     92	memcpy(dst, src, arch_task_struct_size);
     93	dst->thread.fpu.regs = dst->thread.fpu.fprs;
     94	return 0;
     95}
     96
     97int copy_thread(struct task_struct *p, const struct kernel_clone_args *args)
     98{
     99	unsigned long clone_flags = args->flags;
    100	unsigned long new_stackp = args->stack;
    101	unsigned long tls = args->tls;
    102	struct fake_frame
    103	{
    104		struct stack_frame sf;
    105		struct pt_regs childregs;
    106	} *frame;
    107
    108	frame = container_of(task_pt_regs(p), struct fake_frame, childregs);
    109	p->thread.ksp = (unsigned long) frame;
    110	/* Save access registers to new thread structure. */
    111	save_access_regs(&p->thread.acrs[0]);
    112	/* start new process with ar4 pointing to the correct address space */
    113	/* Don't copy debug registers */
    114	memset(&p->thread.per_user, 0, sizeof(p->thread.per_user));
    115	memset(&p->thread.per_event, 0, sizeof(p->thread.per_event));
    116	clear_tsk_thread_flag(p, TIF_SINGLE_STEP);
    117	p->thread.per_flags = 0;
    118	/* Initialize per thread user and system timer values */
    119	p->thread.user_timer = 0;
    120	p->thread.guest_timer = 0;
    121	p->thread.system_timer = 0;
    122	p->thread.hardirq_timer = 0;
    123	p->thread.softirq_timer = 0;
    124	p->thread.last_break = 1;
    125
    126	frame->sf.back_chain = 0;
    127	frame->sf.gprs[5] = (unsigned long)frame + sizeof(struct stack_frame);
    128	frame->sf.gprs[6] = (unsigned long)p;
    129	/* new return point is ret_from_fork */
    130	frame->sf.gprs[8] = (unsigned long)ret_from_fork;
    131	/* fake return stack for resume(), don't go back to schedule */
    132	frame->sf.gprs[9] = (unsigned long)frame;
    133
    134	/* Store access registers to kernel stack of new process. */
    135	if (unlikely(args->fn)) {
    136		/* kernel thread */
    137		memset(&frame->childregs, 0, sizeof(struct pt_regs));
    138		frame->childregs.psw.mask = PSW_KERNEL_BITS | PSW_MASK_DAT |
    139				PSW_MASK_IO | PSW_MASK_EXT | PSW_MASK_MCHECK;
    140		frame->childregs.psw.addr =
    141				(unsigned long)__ret_from_fork;
    142		frame->childregs.gprs[9] = (unsigned long)args->fn;
    143		frame->childregs.gprs[10] = (unsigned long)args->fn_arg;
    144		frame->childregs.orig_gpr2 = -1;
    145		frame->childregs.last_break = 1;
    146		return 0;
    147	}
    148	frame->childregs = *current_pt_regs();
    149	frame->childregs.gprs[2] = 0;	/* child returns 0 on fork. */
    150	frame->childregs.flags = 0;
    151	if (new_stackp)
    152		frame->childregs.gprs[15] = new_stackp;
    153
    154	/* Don't copy runtime instrumentation info */
    155	p->thread.ri_cb = NULL;
    156	frame->childregs.psw.mask &= ~PSW_MASK_RI;
    157	/* Don't copy guarded storage control block */
    158	p->thread.gs_cb = NULL;
    159	p->thread.gs_bc_cb = NULL;
    160
    161	/* Set a new TLS ?  */
    162	if (clone_flags & CLONE_SETTLS) {
    163		if (is_compat_task()) {
    164			p->thread.acrs[0] = (unsigned int)tls;
    165		} else {
    166			p->thread.acrs[0] = (unsigned int)(tls >> 32);
    167			p->thread.acrs[1] = (unsigned int)tls;
    168		}
    169	}
    170	/*
    171	 * s390 stores the svc return address in arch_data when calling
    172	 * sigreturn()/restart_syscall() via vdso. 1 means no valid address
    173	 * stored.
    174	 */
    175	p->restart_block.arch_data = 1;
    176	return 0;
    177}
    178
    179void execve_tail(void)
    180{
    181	current->thread.fpu.fpc = 0;
    182	asm volatile("sfpc %0" : : "d" (0));
    183}
    184
    185unsigned long __get_wchan(struct task_struct *p)
    186{
    187	struct unwind_state state;
    188	unsigned long ip = 0;
    189
    190	if (!task_stack_page(p))
    191		return 0;
    192
    193	if (!try_get_task_stack(p))
    194		return 0;
    195
    196	unwind_for_each_frame(&state, p, NULL, 0) {
    197		if (state.stack_info.type != STACK_TYPE_TASK) {
    198			ip = 0;
    199			break;
    200		}
    201
    202		ip = unwind_get_return_address(&state);
    203		if (!ip)
    204			break;
    205
    206		if (!in_sched_functions(ip))
    207			break;
    208	}
    209
    210	put_task_stack(p);
    211	return ip;
    212}
    213
    214unsigned long arch_align_stack(unsigned long sp)
    215{
    216	if (!(current->personality & ADDR_NO_RANDOMIZE) && randomize_va_space)
    217		sp -= get_random_int() & ~PAGE_MASK;
    218	return sp & ~0xf;
    219}
    220
    221static inline unsigned long brk_rnd(void)
    222{
    223	return (get_random_int() & BRK_RND_MASK) << PAGE_SHIFT;
    224}
    225
    226unsigned long arch_randomize_brk(struct mm_struct *mm)
    227{
    228	unsigned long ret;
    229
    230	ret = PAGE_ALIGN(mm->brk + brk_rnd());
    231	return (ret > mm->brk) ? ret : mm->brk;
    232}