freezer.c 4.44 KB
Newer Older
1 2 3 4 5 6 7 8
/*
 * kernel/freezer.c - Function to freeze a process
 *
 * Originally from kernel/power/process.c
 */

#include <linux/interrupt.h>
#include <linux/suspend.h>
9
#include <linux/export.h>
10 11
#include <linux/syscalls.h>
#include <linux/freezer.h>
12
#include <linux/kthread.h>
13

14 15 16 17 18 19 20 21
/* total number of freezing conditions in effect */
atomic_t system_freezing_cnt = ATOMIC_INIT(0);
EXPORT_SYMBOL(system_freezing_cnt);

/* indicate whether PM freezing is in effect, protected by pm_mutex */
bool pm_freezing;
bool pm_nosig_freezing;

22 23 24 25 26 27
/*
 * Temporary export for the deadlock workaround in ata_scsi_hotplug().
 * Remove once the hack becomes unnecessary.
 */
EXPORT_SYMBOL_GPL(pm_freezing);

28 29
/* protects freezing and frozen transitions */
static DEFINE_SPINLOCK(freezer_lock);
30

31 32 33 34 35 36 37 38 39 40 41
/**
 * freezing_slow_path - slow path for testing whether a task needs to be frozen
 * @p: task to be tested
 *
 * This function is called by freezing() if system_freezing_cnt isn't zero
 * and tests whether @p needs to enter and stay in frozen state.  Can be
 * called under any context.  The freezers are responsible for ensuring the
 * target tasks see the updated state.
 */
bool freezing_slow_path(struct task_struct *p)
{
42
	if (p->flags & (PF_NOFREEZE | PF_SUSPEND_TASK))
43 44
		return false;

45
	if (test_tsk_thread_flag(p, TIF_MEMDIE))
46 47
		return false;

48 49 50
	if (pm_nosig_freezing || cgroup_freezing(p))
		return true;

51
	if (pm_freezing && !(p->flags & PF_KTHREAD))
52 53 54 55 56 57
		return true;

	return false;
}
EXPORT_SYMBOL(freezing_slow_path);

58
/* Refrigerator is place where frozen processes are stored :-). */
59
bool __refrigerator(bool check_kthr_stop)
60 61 62
{
	/* Hmm, should we be allowed to suspend when there are realtime
	   processes around? */
63
	bool was_frozen = false;
64
	long save = current->state;
65 66 67 68 69

	pr_debug("%s entered refrigerator\n", current->comm);

	for (;;) {
		set_current_state(TASK_UNINTERRUPTIBLE);
70 71 72

		spin_lock_irq(&freezer_lock);
		current->flags |= PF_FROZEN;
73
		if (!freezing(current) ||
74
		    (check_kthr_stop && kthread_should_stop()))
75 76 77 78
			current->flags &= ~PF_FROZEN;
		spin_unlock_irq(&freezer_lock);

		if (!(current->flags & PF_FROZEN))
79
			break;
80
		was_frozen = true;
81 82
		schedule();
	}
83

84
	pr_debug("%s left refrigerator\n", current->comm);
85 86 87 88 89 90 91

	/*
	 * Restore saved task state before returning.  The mb'd version
	 * needs to be used; otherwise, it might silently break
	 * synchronization which depends on ordered task state change.
	 */
	set_current_state(save);
92 93

	return was_frozen;
94
}
95
EXPORT_SYMBOL(__refrigerator);
96 97 98 99 100

static void fake_signal_wake_up(struct task_struct *p)
{
	unsigned long flags;

101 102 103 104
	if (lock_task_sighand(p, &flags)) {
		signal_wake_up(p, 0);
		unlock_task_sighand(p, &flags);
	}
105 106 107
}

/**
108 109
 * freeze_task - send a freeze request to given task
 * @p: task to send the request to
110
 *
111 112 113
 * If @p is freezing, the freeze request is sent either by sending a fake
 * signal (if it's not a kernel thread) or waking it up (if it's a kernel
 * thread).
114 115 116
 *
 * RETURNS:
 * %false, if @p is not freezing or already frozen; %true, otherwise
117
 */
118
bool freeze_task(struct task_struct *p)
119
{
120 121
	unsigned long flags;

122 123 124 125 126 127 128 129 130 131 132 133
	/*
	 * This check can race with freezer_do_not_count, but worst case that
	 * will result in an extra wakeup being sent to the task.  It does not
	 * race with freezer_count(), the barriers in freezer_count() and
	 * freezer_should_skip() ensure that either freezer_count() sees
	 * freezing == true in try_to_freeze() and freezes, or
	 * freezer_should_skip() sees !PF_FREEZE_SKIP and freezes the task
	 * normally.
	 */
	if (freezer_should_skip(p))
		return false;

134
	spin_lock_irqsave(&freezer_lock, flags);
135 136 137 138
	if (!freezing(p) || frozen(p)) {
		spin_unlock_irqrestore(&freezer_lock, flags);
		return false;
	}
139

140
	if (!(p->flags & PF_KTHREAD))
141
		fake_signal_wake_up(p);
142
	else
143
		wake_up_state(p, TASK_INTERRUPTIBLE);
144

145
	spin_unlock_irqrestore(&freezer_lock, flags);
146
	return true;
147 148
}

149
void __thaw_task(struct task_struct *p)
150
{
151
	unsigned long flags;
152

153
	spin_lock_irqsave(&freezer_lock, flags);
154
	if (frozen(p))
155
		wake_up_process(p);
156
	spin_unlock_irqrestore(&freezer_lock, flags);
157
}
158 159

/**
160
 * set_freezable - make %current freezable
161 162 163
 *
 * Mark %current freezable and enter refrigerator if necessary.
 */
164
bool set_freezable(void)
165 166 167 168 169 170 171 172 173 174 175 176 177 178
{
	might_sleep();

	/*
	 * Modify flags while holding freezer_lock.  This ensures the
	 * freezer notices that we aren't frozen yet or the freezing
	 * condition is visible to try_to_freeze() below.
	 */
	spin_lock_irq(&freezer_lock);
	current->flags &= ~PF_NOFREEZE;
	spin_unlock_irq(&freezer_lock);

	return try_to_freeze();
}
179
EXPORT_SYMBOL(set_freezable);