668 lines · 15.4 KB · C
RawDownload
#include <linux/errno.h>#include <linux/kernel.h>#include <linux/mm.h>#include <linux/smp.h>#include <linux/prctl.h>#include <linux/slab.h>#include <linux/sched.h>#include <linux/module.h>#include <linux/pm.h>#include <linux/clockchips.h>#include <linux/random.h>#include <linux/user-return-notifier.h>#include <linux/dmi.h>#include <linux/utsname.h>#include <trace/events/power.h>#include <linux/hw_breakpoint.h>#include <asm/cpu.h>#include <asm/system.h>#include <asm/apic.h>#include <asm/syscalls.h>#include <asm/idle.h>#include <asm/uaccess.h>#include <asm/i387.h>#include <asm/debugreg.h> struct kmem_cache *task_xstate_cachep;EXPORT_SYMBOL_GPL(task_xstate_cachep); int arch_dup_task_struct(struct task_struct *dst, struct task_struct *src){	int ret; 	*dst = *src;	if (fpu_allocated(&src->thread.fpu)) {		memset(&dst->thread.fpu, 0, sizeof(dst->thread.fpu));		ret = fpu_alloc(&dst->thread.fpu);		if (ret)			return ret;		fpu_copy(&dst->thread.fpu, &src->thread.fpu);	}	return 0;} void free_thread_xstate(struct task_struct *tsk){	fpu_free(&tsk->thread.fpu);} void free_thread_info(struct thread_info *ti){	free_thread_xstate(ti->task);	free_pages((unsigned long)ti, THREAD_ORDER);} void arch_task_cache_init(void){        task_xstate_cachep =        	kmem_cache_create("task_xstate", xstate_size,				  __alignof__(union thread_xstate),				  SLAB_PANIC | SLAB_NOTRACK, NULL);} /* * Free current thread data structures etc.. */void exit_thread(void){	struct task_struct *me = current;	struct thread_struct *t = &me->thread;	unsigned long *bp = t->io_bitmap_ptr; 	if (bp) {		struct tss_struct *tss = &per_cpu(init_tss, get_cpu()); 		t->io_bitmap_ptr = NULL;		clear_thread_flag(TIF_IO_BITMAP);		/*		 * Careful, clear this in the TSS too:		 */		memset(tss->io_bitmap, 0xff, t->io_bitmap_max);		t->io_bitmap_max = 0;		put_cpu();		kfree(bp);	}} void show_regs(struct pt_regs *regs){	show_registers(regs);	show_trace(NULL, regs, (unsigned long *)kernel_stack_pointer(regs), 0);} void show_regs_common(void){	const char *vendor, *product, *board; 	vendor = dmi_get_system_info(DMI_SYS_VENDOR);	if (!vendor)		vendor = "";	product = dmi_get_system_info(DMI_PRODUCT_NAME);	if (!product)		product = ""; 	/* Board Name is optional */	board = dmi_get_system_info(DMI_BOARD_NAME); 	printk(KERN_CONT "\n");	printk(KERN_DEFAULT "Pid: %d, comm: %.20s %s %s %.*s",		current->pid, current->comm, print_tainted(),		init_utsname()->release,		(int)strcspn(init_utsname()->version, " "),		init_utsname()->version);	printk(KERN_CONT " %s %s", vendor, product);	if (board)		printk(KERN_CONT "/%s", board);	printk(KERN_CONT "\n");} void flush_thread(void){	struct task_struct *tsk = current; 	flush_ptrace_hw_breakpoint(tsk);	memset(tsk->thread.tls_array, 0, sizeof(tsk->thread.tls_array));	/*	 * Forget coprocessor state..	 */	tsk->fpu_counter = 0;	clear_fpu(tsk);	clear_used_math();} static void hard_disable_TSC(void){	write_cr4(read_cr4() | X86_CR4_TSD);} void disable_TSC(void){	preempt_disable();	if (!test_and_set_thread_flag(TIF_NOTSC))		/*		 * Must flip the CPU state synchronously with		 * TIF_NOTSC in the current running context.		 */		hard_disable_TSC();	preempt_enable();} static void hard_enable_TSC(void){	write_cr4(read_cr4() & ~X86_CR4_TSD);} static void enable_TSC(void){	preempt_disable();	if (test_and_clear_thread_flag(TIF_NOTSC))		/*		 * Must flip the CPU state synchronously with		 * TIF_NOTSC in the current running context.		 */		hard_enable_TSC();	preempt_enable();} int get_tsc_mode(unsigned long adr){	unsigned int val; 	if (test_thread_flag(TIF_NOTSC))		val = PR_TSC_SIGSEGV;	else		val = PR_TSC_ENABLE; 	return put_user(val, (unsigned int __user *)adr);} int set_tsc_mode(unsigned int val){	if (val == PR_TSC_SIGSEGV)		disable_TSC();	else if (val == PR_TSC_ENABLE)		enable_TSC();	else		return -EINVAL; 	return 0;} void __switch_to_xtra(struct task_struct *prev_p, struct task_struct *next_p,		      struct tss_struct *tss){	struct thread_struct *prev, *next; 	prev = &prev_p->thread;	next = &next_p->thread; 	if (test_tsk_thread_flag(prev_p, TIF_BLOCKSTEP) ^	    test_tsk_thread_flag(next_p, TIF_BLOCKSTEP)) {		unsigned long debugctl = get_debugctlmsr(); 		debugctl &= ~DEBUGCTLMSR_BTF;		if (test_tsk_thread_flag(next_p, TIF_BLOCKSTEP))			debugctl |= DEBUGCTLMSR_BTF; 		update_debugctlmsr(debugctl);	} 	if (test_tsk_thread_flag(prev_p, TIF_NOTSC) ^	    test_tsk_thread_flag(next_p, TIF_NOTSC)) {		/* prev and next are different */		if (test_tsk_thread_flag(next_p, TIF_NOTSC))			hard_disable_TSC();		else			hard_enable_TSC();	} 	if (test_tsk_thread_flag(next_p, TIF_IO_BITMAP)) {		/*		 * Copy the relevant range of the IO bitmap.		 * Normally this is 128 bytes or less:		 */		memcpy(tss->io_bitmap, next->io_bitmap_ptr,		       max(prev->io_bitmap_max, next->io_bitmap_max));	} else if (test_tsk_thread_flag(prev_p, TIF_IO_BITMAP)) {		/*		 * Clear any possible leftover bits:		 */		memset(tss->io_bitmap, 0xff, prev->io_bitmap_max);	}	propagate_user_return_notify(prev_p, next_p);} int sys_fork(struct pt_regs *regs){	return do_fork(SIGCHLD, regs->sp, regs, 0, NULL, NULL);} /* * This is trivial, and on the face of it looks like it * could equally well be done in user mode. * * Not so, for quite unobvious reasons - register pressure. * In user mode vfork() cannot have a stack frame, and if * done by calling the "clone()" system call directly, you * do not have enough call-clobbered registers to hold all * the information you need. */int sys_vfork(struct pt_regs *regs){	return do_fork(CLONE_VFORK | CLONE_VM | SIGCHLD, regs->sp, regs, 0,		       NULL, NULL);} longsys_clone(unsigned long clone_flags, unsigned long newsp,	  void __user *parent_tid, void __user *child_tid, struct pt_regs *regs){	if (!newsp)		newsp = regs->sp;	return do_fork(clone_flags, newsp, regs, 0, parent_tid, child_tid);} /* * This gets run with %si containing the * function to call, and %di containing * the "args". */extern void kernel_thread_helper(void); /* * Create a kernel thread */int kernel_thread(int (*fn)(void *), void *arg, unsigned long flags){	struct pt_regs regs; 	memset(&regs, 0, sizeof(regs)); 	regs.si = (unsigned long) fn;	regs.di = (unsigned long) arg; #ifdef CONFIG_X86_32	regs.ds = __USER_DS;	regs.es = __USER_DS;	regs.fs = __KERNEL_PERCPU;	regs.gs = __KERNEL_STACK_CANARY;#else	regs.ss = __KERNEL_DS;#endif 	regs.orig_ax = -1;	regs.ip = (unsigned long) kernel_thread_helper;	regs.cs = __KERNEL_CS | get_kernel_rpl();	regs.flags = X86_EFLAGS_IF | 0x2; 	/* Ok, create the new process.. */	return do_fork(flags | CLONE_VM | CLONE_UNTRACED, 0, &regs, 0, NULL, NULL);}EXPORT_SYMBOL(kernel_thread); /* * sys_execve() executes a new program. */long sys_execve(const char __user *name,		const char __user *const __user *argv,		const char __user *const __user *envp, struct pt_regs *regs){	long error;	char *filename; 	filename = getname(name);	error = PTR_ERR(filename);	if (IS_ERR(filename))		return error;	error = do_execve(filename, argv, envp, regs); #ifdef CONFIG_X86_32	if (error == 0) {		/* Make sure we don't return using sysenter.. */                set_thread_flag(TIF_IRET);        }#endif 	putname(filename);	return error;} /* * Idle related variables and functions */unsigned long boot_option_idle_override = IDLE_NO_OVERRIDE;EXPORT_SYMBOL(boot_option_idle_override); /* * Powermanagement idle function, if any.. */void (*pm_idle)(void);#ifdef CONFIG_APM_MODULEEXPORT_SYMBOL(pm_idle);#endif #ifdef CONFIG_X86_32/* * This halt magic was a workaround for ancient floppy DMA * wreckage. It should be safe to remove. */static int hlt_counter;void disable_hlt(void){	hlt_counter++;}EXPORT_SYMBOL(disable_hlt); void enable_hlt(void){	hlt_counter--;}EXPORT_SYMBOL(enable_hlt); static inline int hlt_use_halt(void){	return (!hlt_counter && boot_cpu_data.hlt_works_ok);}#elsestatic inline int hlt_use_halt(void){	return 1;}#endif /* * We use this if we don't have any better * idle routine.. */void default_idle(void){	if (hlt_use_halt()) {		trace_power_start(POWER_CSTATE, 1, smp_processor_id());		trace_cpu_idle(1, smp_processor_id());		current_thread_info()->status &= ~TS_POLLING;		/*		 * TS_POLLING-cleared state must be visible before we		 * test NEED_RESCHED:		 */		smp_mb(); 		if (!need_resched())			safe_halt();	/* enables interrupts racelessly */		else			local_irq_enable();		current_thread_info()->status |= TS_POLLING;		trace_power_end(smp_processor_id());		trace_cpu_idle(PWR_EVENT_EXIT, smp_processor_id());	} else {		local_irq_enable();		/* loop is done by the caller */		cpu_relax();	}}#ifdef CONFIG_APM_MODULEEXPORT_SYMBOL(default_idle);#endif bool set_pm_idle_to_default(void){	bool ret = !!pm_idle; 	pm_idle = default_idle; 	return ret;}void stop_this_cpu(void *dummy){	local_irq_disable();	/*	 * Remove this CPU:	 */	set_cpu_online(smp_processor_id(), false);	disable_local_APIC(); 	for (;;) {		if (hlt_works(smp_processor_id()))			halt();	}} static void do_nothing(void *unused){} /* * cpu_idle_wait - Used to ensure that all the CPUs discard old value of * pm_idle and update to new pm_idle value. Required while changing pm_idle * handler on SMP systems. * * Caller must have changed pm_idle to the new value before the call. Old * pm_idle value will not be used by any CPU after the return of this function. */void cpu_idle_wait(void){	smp_mb();	/* kick all the CPUs so that they exit out of pm_idle */	smp_call_function(do_nothing, NULL, 1);}EXPORT_SYMBOL_GPL(cpu_idle_wait); /* Default MONITOR/MWAIT with no hints, used for default C1 state */static void mwait_idle(void){	if (!need_resched()) {		trace_power_start(POWER_CSTATE, 1, smp_processor_id());		trace_cpu_idle(1, smp_processor_id());		if (this_cpu_has(X86_FEATURE_CLFLUSH_MONITOR))			clflush((void *)&current_thread_info()->flags); 		__monitor((void *)&current_thread_info()->flags, 0, 0);		smp_mb();		if (!need_resched())			__sti_mwait(0, 0);		else			local_irq_enable();		trace_power_end(smp_processor_id());		trace_cpu_idle(PWR_EVENT_EXIT, smp_processor_id());	} else		local_irq_enable();} /* * On SMP it's slightly faster (but much more power-consuming!) * to poll the ->work.need_resched flag instead of waiting for the * cross-CPU IPI to arrive. Use this option with caution. */static void poll_idle(void){	trace_power_start(POWER_CSTATE, 0, smp_processor_id());	trace_cpu_idle(0, smp_processor_id());	local_irq_enable();	while (!need_resched())		cpu_relax();	trace_power_end(smp_processor_id());	trace_cpu_idle(PWR_EVENT_EXIT, smp_processor_id());} /* * mwait selection logic: * * It depends on the CPU. For AMD CPUs that support MWAIT this is * wrong. Family 0x10 and 0x11 CPUs will enter C1 on HLT. Powersavings * then depend on a clock divisor and current Pstate of the core. If * all cores of a processor are in halt state (C1) the processor can * enter the C1E (C1 enhanced) state. If mwait is used this will never * happen. * * idle=mwait overrides this decision and forces the usage of mwait. */ #define MWAIT_INFO			0x05#define MWAIT_ECX_EXTENDED_INFO		0x01

Lines 1–500 of 668.