Blame view

mm/swapfile.c 65.4 KB
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1
2
3
4
5
6
  /*
   *  linux/mm/swapfile.c
   *
   *  Copyright (C) 1991, 1992, 1993, 1994  Linus Torvalds
   *  Swap reorganised 29.12.95, Stephen Tweedie
   */
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
7
8
9
10
11
12
13
14
15
  #include <linux/mm.h>
  #include <linux/hugetlb.h>
  #include <linux/mman.h>
  #include <linux/slab.h>
  #include <linux/kernel_stat.h>
  #include <linux/swap.h>
  #include <linux/vmalloc.h>
  #include <linux/pagemap.h>
  #include <linux/namei.h>
072441e21   Hugh Dickins   mm: move shmem pr...
16
  #include <linux/shmem_fs.h>
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
17
  #include <linux/blkdev.h>
20137a490   Hugh Dickins   swapfile: swapon ...
18
  #include <linux/random.h>
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
19
20
21
22
  #include <linux/writeback.h>
  #include <linux/proc_fs.h>
  #include <linux/seq_file.h>
  #include <linux/init.h>
5ad646880   Hugh Dickins   ksm: let shared p...
23
  #include <linux/ksm.h>
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
24
25
26
  #include <linux/rmap.h>
  #include <linux/security.h>
  #include <linux/backing-dev.h>
fc0abb145   Ingo Molnar   [PATCH] sem2mutex...
27
  #include <linux/mutex.h>
c59ede7b7   Randy.Dunlap   [PATCH] move capa...
28
  #include <linux/capability.h>
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
29
  #include <linux/syscalls.h>
8a9f3ccd2   Balbir Singh   Memory controller...
30
  #include <linux/memcontrol.h>
66d7dd518   Kay Sievers   /proc/swaps: supp...
31
  #include <linux/poll.h>
72788c385   David Rientjes   oom: replace PF_O...
32
  #include <linux/oom.h>
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
33
34
35
36
  
  #include <asm/pgtable.h>
  #include <asm/tlbflush.h>
  #include <linux/swapops.h>
27a7faa07   KAMEZAWA Hiroyuki   memcg: swap cgrou...
37
  #include <linux/page_cgroup.h>
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
38

570a335b8   Hugh Dickins   swap_info: swap c...
39
40
41
  static bool swap_count_continued(struct swap_info_struct *, pgoff_t,
  				 unsigned char);
  static void free_swap_count_continuations(struct swap_info_struct *);
d4906e1aa   Lee Schermerhorn   swap: rework map_...
42
  static sector_t map_swap_entry(swp_entry_t, struct block_device**);
570a335b8   Hugh Dickins   swap_info: swap c...
43

7c363b8c6   Adrian Bunk   mm/swapfile.c: ma...
44
45
  static DEFINE_SPINLOCK(swap_lock);
  static unsigned int nr_swapfiles;
b962716b4   Hugh Dickins   mm: optimize get_...
46
  long nr_swap_pages;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
47
  long total_swap_pages;
78ecba081   Hugh Dickins   mm: fix ever-decr...
48
  static int least_priority;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
49

1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
50
51
52
53
  static const char Bad_file[] = "Bad swap file entry ";
  static const char Unused_file[] = "Unused swap file entry ";
  static const char Bad_offset[] = "Bad swap offset entry ";
  static const char Unused_offset[] = "Unused swap offset entry ";
7c363b8c6   Adrian Bunk   mm/swapfile.c: ma...
54
  static struct swap_list_t swap_list = {-1, -1};
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
55

efa90a981   Hugh Dickins   swap_info: change...
56
  static struct swap_info_struct *swap_info[MAX_SWAPFILES];
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
57

fc0abb145   Ingo Molnar   [PATCH] sem2mutex...
58
  static DEFINE_MUTEX(swapon_mutex);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
59

66d7dd518   Kay Sievers   /proc/swaps: supp...
60
61
62
  static DECLARE_WAIT_QUEUE_HEAD(proc_poll_wait);
  /* Activity counter to indicate that a swapon or swapoff has occurred */
  static atomic_t proc_poll_event = ATOMIC_INIT(0);
8d69aaee8   Hugh Dickins   swap_info: swap_m...
63
  static inline unsigned char swap_count(unsigned char ent)
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
64
  {
570a335b8   Hugh Dickins   swap_info: swap c...
65
  	return ent & ~SWAP_HAS_CACHE;	/* may include SWAP_HAS_CONT flag */
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
66
  }
efa90a981   Hugh Dickins   swap_info: change...
67
  /* returns 1 if swap entry is freed */
c9e444103   KAMEZAWA Hiroyuki   mm: reuse unused ...
68
69
70
  static int
  __try_to_reclaim_swap(struct swap_info_struct *si, unsigned long offset)
  {
efa90a981   Hugh Dickins   swap_info: change...
71
  	swp_entry_t entry = swp_entry(si->type, offset);
c9e444103   KAMEZAWA Hiroyuki   mm: reuse unused ...
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
  	struct page *page;
  	int ret = 0;
  
  	page = find_get_page(&swapper_space, entry.val);
  	if (!page)
  		return 0;
  	/*
  	 * This function is called from scan_swap_map() and it's called
  	 * by vmscan.c at reclaiming pages. So, we hold a lock on a page, here.
  	 * We have to use trylock for avoiding deadlock. This is a special
  	 * case and you should use try_to_free_swap() with explicit lock_page()
  	 * in usual operations.
  	 */
  	if (trylock_page(page)) {
  		ret = try_to_free_swap(page);
  		unlock_page(page);
  	}
  	page_cache_release(page);
  	return ret;
  }
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
92

1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
93
  /*
6a6ba8317   Hugh Dickins   swapfile: swapon ...
94
95
96
97
98
99
   * swapon tell device that all the old swap contents can be discarded,
   * to allow the swap device to optimize its wear-levelling.
   */
  static int discard_swap(struct swap_info_struct *si)
  {
  	struct swap_extent *se;
9625a5f28   Hugh Dickins   swap_info: includ...
100
101
  	sector_t start_block;
  	sector_t nr_blocks;
6a6ba8317   Hugh Dickins   swapfile: swapon ...
102
  	int err = 0;
9625a5f28   Hugh Dickins   swap_info: includ...
103
104
105
106
107
108
  	/* Do not discard the swap header page! */
  	se = &si->first_swap_extent;
  	start_block = (se->start_block + 1) << (PAGE_SHIFT - 9);
  	nr_blocks = ((sector_t)se->nr_pages - 1) << (PAGE_SHIFT - 9);
  	if (nr_blocks) {
  		err = blkdev_issue_discard(si->bdev, start_block,
dd3932edd   Christoph Hellwig   block: remove BLK...
109
  				nr_blocks, GFP_KERNEL, 0);
9625a5f28   Hugh Dickins   swap_info: includ...
110
111
112
113
  		if (err)
  			return err;
  		cond_resched();
  	}
6a6ba8317   Hugh Dickins   swapfile: swapon ...
114

9625a5f28   Hugh Dickins   swap_info: includ...
115
116
117
  	list_for_each_entry(se, &si->first_swap_extent.list, list) {
  		start_block = se->start_block << (PAGE_SHIFT - 9);
  		nr_blocks = (sector_t)se->nr_pages << (PAGE_SHIFT - 9);
6a6ba8317   Hugh Dickins   swapfile: swapon ...
118
119
  
  		err = blkdev_issue_discard(si->bdev, start_block,
dd3932edd   Christoph Hellwig   block: remove BLK...
120
  				nr_blocks, GFP_KERNEL, 0);
6a6ba8317   Hugh Dickins   swapfile: swapon ...
121
122
123
124
125
126
127
  		if (err)
  			break;
  
  		cond_resched();
  	}
  	return err;		/* That will often be -EOPNOTSUPP */
  }
7992fde72   Hugh Dickins   swapfile: swap al...
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
  /*
   * swap allocation tell device that a cluster of swap can now be discarded,
   * to allow the swap device to optimize its wear-levelling.
   */
  static void discard_swap_cluster(struct swap_info_struct *si,
  				 pgoff_t start_page, pgoff_t nr_pages)
  {
  	struct swap_extent *se = si->curr_swap_extent;
  	int found_extent = 0;
  
  	while (nr_pages) {
  		struct list_head *lh;
  
  		if (se->start_page <= start_page &&
  		    start_page < se->start_page + se->nr_pages) {
  			pgoff_t offset = start_page - se->start_page;
  			sector_t start_block = se->start_block + offset;
858a29900   Hugh Dickins   swapfile: change ...
145
  			sector_t nr_blocks = se->nr_pages - offset;
7992fde72   Hugh Dickins   swapfile: swap al...
146
147
148
149
150
151
152
153
154
155
156
157
  
  			if (nr_blocks > nr_pages)
  				nr_blocks = nr_pages;
  			start_page += nr_blocks;
  			nr_pages -= nr_blocks;
  
  			if (!found_extent++)
  				si->curr_swap_extent = se;
  
  			start_block <<= PAGE_SHIFT - 9;
  			nr_blocks <<= PAGE_SHIFT - 9;
  			if (blkdev_issue_discard(si->bdev, start_block,
dd3932edd   Christoph Hellwig   block: remove BLK...
158
  				    nr_blocks, GFP_NOIO, 0))
7992fde72   Hugh Dickins   swapfile: swap al...
159
160
161
162
  				break;
  		}
  
  		lh = se->list.next;
7992fde72   Hugh Dickins   swapfile: swap al...
163
164
165
166
167
168
169
170
171
  		se = list_entry(lh, struct swap_extent, list);
  	}
  }
  
  static int wait_for_discard(void *word)
  {
  	schedule();
  	return 0;
  }
048c27fd7   Hugh Dickins   [PATCH] swap: sca...
172
173
  #define SWAPFILE_CLUSTER	256
  #define LATENCY_LIMIT		256
24b8ff7c2   Cesar Eduardo Barros   mm: remove inline...
174
175
  static unsigned long scan_swap_map(struct swap_info_struct *si,
  				   unsigned char usage)
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
176
  {
ebebbbe90   Hugh Dickins   swapfile: rearran...
177
  	unsigned long offset;
c60aa176c   Hugh Dickins   swapfile: swap al...
178
  	unsigned long scan_base;
7992fde72   Hugh Dickins   swapfile: swap al...
179
  	unsigned long last_in_cluster = 0;
048c27fd7   Hugh Dickins   [PATCH] swap: sca...
180
  	int latency_ration = LATENCY_LIMIT;
7992fde72   Hugh Dickins   swapfile: swap al...
181
  	int found_free_cluster = 0;
7dfad4183   Hugh Dickins   [PATCH] swap: sca...
182

886bb7e9c   Hugh Dickins   swapfile: remove ...
183
  	/*
7dfad4183   Hugh Dickins   [PATCH] swap: sca...
184
185
186
187
188
189
190
  	 * We try to cluster swap pages by allocating them sequentially
  	 * in swap.  Once we've allocated SWAPFILE_CLUSTER pages this
  	 * way, however, we resort to first-free allocation, starting
  	 * a new cluster.  This prevents us from scattering swap pages
  	 * all over the entire swap partition, so that we reduce
  	 * overall disk seek times between swap pages.  -- sct
  	 * But we do now try to find an empty cluster.  -Andrea
c60aa176c   Hugh Dickins   swapfile: swap al...
191
  	 * And we let swap pages go all over an SSD partition.  Hugh
7dfad4183   Hugh Dickins   [PATCH] swap: sca...
192
  	 */
52b7efdbe   Hugh Dickins   [PATCH] swap: sca...
193
  	si->flags += SWP_SCANNING;
c60aa176c   Hugh Dickins   swapfile: swap al...
194
  	scan_base = offset = si->cluster_next;
ebebbbe90   Hugh Dickins   swapfile: rearran...
195
196
197
198
199
200
  
  	if (unlikely(!si->cluster_nr--)) {
  		if (si->pages - si->inuse_pages < SWAPFILE_CLUSTER) {
  			si->cluster_nr = SWAPFILE_CLUSTER - 1;
  			goto checks;
  		}
7992fde72   Hugh Dickins   swapfile: swap al...
201
202
203
204
205
206
207
208
209
210
211
212
213
  		if (si->flags & SWP_DISCARDABLE) {
  			/*
  			 * Start range check on racing allocations, in case
  			 * they overlap the cluster we eventually decide on
  			 * (we scan without swap_lock to allow preemption).
  			 * It's hardly conceivable that cluster_nr could be
  			 * wrapped during our scan, but don't depend on it.
  			 */
  			if (si->lowest_alloc)
  				goto checks;
  			si->lowest_alloc = si->max;
  			si->highest_alloc = 0;
  		}
5d337b919   Hugh Dickins   [PATCH] swap: swa...
214
  		spin_unlock(&swap_lock);
7dfad4183   Hugh Dickins   [PATCH] swap: sca...
215

c60aa176c   Hugh Dickins   swapfile: swap al...
216
217
218
219
220
221
222
223
224
225
  		/*
  		 * If seek is expensive, start searching for new cluster from
  		 * start of partition, to minimize the span of allocated swap.
  		 * But if seek is cheap, search from our current position, so
  		 * that swap is allocated from all over the partition: if the
  		 * Flash Translation Layer only remaps within limited zones,
  		 * we don't want to wear out the first zone too quickly.
  		 */
  		if (!(si->flags & SWP_SOLIDSTATE))
  			scan_base = offset = si->lowest_bit;
7dfad4183   Hugh Dickins   [PATCH] swap: sca...
226
227
228
229
  		last_in_cluster = offset + SWAPFILE_CLUSTER - 1;
  
  		/* Locate the first empty (unaligned) cluster */
  		for (; last_in_cluster <= si->highest_bit; offset++) {
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
230
  			if (si->swap_map[offset])
7dfad4183   Hugh Dickins   [PATCH] swap: sca...
231
232
  				last_in_cluster = offset + SWAPFILE_CLUSTER;
  			else if (offset == last_in_cluster) {
5d337b919   Hugh Dickins   [PATCH] swap: swa...
233
  				spin_lock(&swap_lock);
ebebbbe90   Hugh Dickins   swapfile: rearran...
234
235
236
  				offset -= SWAPFILE_CLUSTER - 1;
  				si->cluster_next = offset;
  				si->cluster_nr = SWAPFILE_CLUSTER - 1;
7992fde72   Hugh Dickins   swapfile: swap al...
237
  				found_free_cluster = 1;
ebebbbe90   Hugh Dickins   swapfile: rearran...
238
  				goto checks;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
239
  			}
048c27fd7   Hugh Dickins   [PATCH] swap: sca...
240
241
242
243
  			if (unlikely(--latency_ration < 0)) {
  				cond_resched();
  				latency_ration = LATENCY_LIMIT;
  			}
7dfad4183   Hugh Dickins   [PATCH] swap: sca...
244
  		}
ebebbbe90   Hugh Dickins   swapfile: rearran...
245
246
  
  		offset = si->lowest_bit;
c60aa176c   Hugh Dickins   swapfile: swap al...
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
  		last_in_cluster = offset + SWAPFILE_CLUSTER - 1;
  
  		/* Locate the first empty (unaligned) cluster */
  		for (; last_in_cluster < scan_base; offset++) {
  			if (si->swap_map[offset])
  				last_in_cluster = offset + SWAPFILE_CLUSTER;
  			else if (offset == last_in_cluster) {
  				spin_lock(&swap_lock);
  				offset -= SWAPFILE_CLUSTER - 1;
  				si->cluster_next = offset;
  				si->cluster_nr = SWAPFILE_CLUSTER - 1;
  				found_free_cluster = 1;
  				goto checks;
  			}
  			if (unlikely(--latency_ration < 0)) {
  				cond_resched();
  				latency_ration = LATENCY_LIMIT;
  			}
  		}
  
  		offset = scan_base;
5d337b919   Hugh Dickins   [PATCH] swap: swa...
268
  		spin_lock(&swap_lock);
ebebbbe90   Hugh Dickins   swapfile: rearran...
269
  		si->cluster_nr = SWAPFILE_CLUSTER - 1;
7992fde72   Hugh Dickins   swapfile: swap al...
270
  		si->lowest_alloc = 0;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
271
  	}
7dfad4183   Hugh Dickins   [PATCH] swap: sca...
272

ebebbbe90   Hugh Dickins   swapfile: rearran...
273
274
  checks:
  	if (!(si->flags & SWP_WRITEOK))
52b7efdbe   Hugh Dickins   [PATCH] swap: sca...
275
  		goto no_page;
7dfad4183   Hugh Dickins   [PATCH] swap: sca...
276
277
  	if (!si->highest_bit)
  		goto no_page;
ebebbbe90   Hugh Dickins   swapfile: rearran...
278
  	if (offset > si->highest_bit)
c60aa176c   Hugh Dickins   swapfile: swap al...
279
  		scan_base = offset = si->lowest_bit;
c9e444103   KAMEZAWA Hiroyuki   mm: reuse unused ...
280

b73d7fcec   Hugh Dickins   swap: prevent reu...
281
282
  	/* reuse swap entry of cache-only swap if not busy. */
  	if (vm_swap_full() && si->swap_map[offset] == SWAP_HAS_CACHE) {
c9e444103   KAMEZAWA Hiroyuki   mm: reuse unused ...
283
284
285
286
287
288
289
290
291
  		int swap_was_freed;
  		spin_unlock(&swap_lock);
  		swap_was_freed = __try_to_reclaim_swap(si, offset);
  		spin_lock(&swap_lock);
  		/* entry was freed successfully, try to use this again */
  		if (swap_was_freed)
  			goto checks;
  		goto scan; /* check next one */
  	}
ebebbbe90   Hugh Dickins   swapfile: rearran...
292
293
294
295
296
297
298
299
300
301
302
  	if (si->swap_map[offset])
  		goto scan;
  
  	if (offset == si->lowest_bit)
  		si->lowest_bit++;
  	if (offset == si->highest_bit)
  		si->highest_bit--;
  	si->inuse_pages++;
  	if (si->inuse_pages == si->pages) {
  		si->lowest_bit = si->max;
  		si->highest_bit = 0;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
303
  	}
253d553ba   Hugh Dickins   swap_info: SWAP_H...
304
  	si->swap_map[offset] = usage;
ebebbbe90   Hugh Dickins   swapfile: rearran...
305
306
  	si->cluster_next = offset + 1;
  	si->flags -= SWP_SCANNING;
7992fde72   Hugh Dickins   swapfile: swap al...
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
  
  	if (si->lowest_alloc) {
  		/*
  		 * Only set when SWP_DISCARDABLE, and there's a scan
  		 * for a free cluster in progress or just completed.
  		 */
  		if (found_free_cluster) {
  			/*
  			 * To optimize wear-levelling, discard the
  			 * old data of the cluster, taking care not to
  			 * discard any of its pages that have already
  			 * been allocated by racing tasks (offset has
  			 * already stepped over any at the beginning).
  			 */
  			if (offset < si->highest_alloc &&
  			    si->lowest_alloc <= last_in_cluster)
  				last_in_cluster = si->lowest_alloc - 1;
  			si->flags |= SWP_DISCARDING;
  			spin_unlock(&swap_lock);
  
  			if (offset < last_in_cluster)
  				discard_swap_cluster(si, offset,
  					last_in_cluster - offset + 1);
  
  			spin_lock(&swap_lock);
  			si->lowest_alloc = 0;
  			si->flags &= ~SWP_DISCARDING;
  
  			smp_mb();	/* wake_up_bit advises this */
  			wake_up_bit(&si->flags, ilog2(SWP_DISCARDING));
  
  		} else if (si->flags & SWP_DISCARDING) {
  			/*
  			 * Delay using pages allocated by racing tasks
  			 * until the whole discard has been issued. We
  			 * could defer that delay until swap_writepage,
  			 * but it's easier to keep this self-contained.
  			 */
  			spin_unlock(&swap_lock);
  			wait_on_bit(&si->flags, ilog2(SWP_DISCARDING),
  				wait_for_discard, TASK_UNINTERRUPTIBLE);
  			spin_lock(&swap_lock);
  		} else {
  			/*
  			 * Note pages allocated by racing tasks while
  			 * scan for a free cluster is in progress, so
  			 * that its final discard can exclude them.
  			 */
  			if (offset < si->lowest_alloc)
  				si->lowest_alloc = offset;
  			if (offset > si->highest_alloc)
  				si->highest_alloc = offset;
  		}
  	}
ebebbbe90   Hugh Dickins   swapfile: rearran...
361
  	return offset;
7dfad4183   Hugh Dickins   [PATCH] swap: sca...
362

ebebbbe90   Hugh Dickins   swapfile: rearran...
363
  scan:
5d337b919   Hugh Dickins   [PATCH] swap: swa...
364
  	spin_unlock(&swap_lock);
7dfad4183   Hugh Dickins   [PATCH] swap: sca...
365
  	while (++offset <= si->highest_bit) {
52b7efdbe   Hugh Dickins   [PATCH] swap: sca...
366
  		if (!si->swap_map[offset]) {
5d337b919   Hugh Dickins   [PATCH] swap: swa...
367
  			spin_lock(&swap_lock);
52b7efdbe   Hugh Dickins   [PATCH] swap: sca...
368
369
  			goto checks;
  		}
c9e444103   KAMEZAWA Hiroyuki   mm: reuse unused ...
370
371
372
373
  		if (vm_swap_full() && si->swap_map[offset] == SWAP_HAS_CACHE) {
  			spin_lock(&swap_lock);
  			goto checks;
  		}
048c27fd7   Hugh Dickins   [PATCH] swap: sca...
374
375
376
377
  		if (unlikely(--latency_ration < 0)) {
  			cond_resched();
  			latency_ration = LATENCY_LIMIT;
  		}
7dfad4183   Hugh Dickins   [PATCH] swap: sca...
378
  	}
c60aa176c   Hugh Dickins   swapfile: swap al...
379
380
381
382
383
384
  	offset = si->lowest_bit;
  	while (++offset < scan_base) {
  		if (!si->swap_map[offset]) {
  			spin_lock(&swap_lock);
  			goto checks;
  		}
c9e444103   KAMEZAWA Hiroyuki   mm: reuse unused ...
385
386
387
388
  		if (vm_swap_full() && si->swap_map[offset] == SWAP_HAS_CACHE) {
  			spin_lock(&swap_lock);
  			goto checks;
  		}
c60aa176c   Hugh Dickins   swapfile: swap al...
389
390
391
392
393
  		if (unlikely(--latency_ration < 0)) {
  			cond_resched();
  			latency_ration = LATENCY_LIMIT;
  		}
  	}
5d337b919   Hugh Dickins   [PATCH] swap: swa...
394
  	spin_lock(&swap_lock);
7dfad4183   Hugh Dickins   [PATCH] swap: sca...
395
396
  
  no_page:
52b7efdbe   Hugh Dickins   [PATCH] swap: sca...
397
  	si->flags -= SWP_SCANNING;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
398
399
400
401
402
  	return 0;
  }
  
  swp_entry_t get_swap_page(void)
  {
fb4f88dca   Hugh Dickins   [PATCH] swap: get...
403
404
405
406
  	struct swap_info_struct *si;
  	pgoff_t offset;
  	int type, next;
  	int wrapped = 0;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
407

5d337b919   Hugh Dickins   [PATCH] swap: swa...
408
  	spin_lock(&swap_lock);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
409
  	if (nr_swap_pages <= 0)
fb4f88dca   Hugh Dickins   [PATCH] swap: get...
410
411
412
413
  		goto noswap;
  	nr_swap_pages--;
  
  	for (type = swap_list.next; type >= 0 && wrapped < 2; type = next) {
efa90a981   Hugh Dickins   swap_info: change...
414
  		si = swap_info[type];
fb4f88dca   Hugh Dickins   [PATCH] swap: get...
415
416
  		next = si->next;
  		if (next < 0 ||
efa90a981   Hugh Dickins   swap_info: change...
417
  		    (!wrapped && si->prio != swap_info[next]->prio)) {
fb4f88dca   Hugh Dickins   [PATCH] swap: get...
418
419
  			next = swap_list.head;
  			wrapped++;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
420
  		}
fb4f88dca   Hugh Dickins   [PATCH] swap: get...
421
422
423
424
425
426
427
  
  		if (!si->highest_bit)
  			continue;
  		if (!(si->flags & SWP_WRITEOK))
  			continue;
  
  		swap_list.next = next;
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
428
  		/* This is called for allocating swap entry for cache */
253d553ba   Hugh Dickins   swap_info: SWAP_H...
429
  		offset = scan_swap_map(si, SWAP_HAS_CACHE);
5d337b919   Hugh Dickins   [PATCH] swap: swa...
430
431
  		if (offset) {
  			spin_unlock(&swap_lock);
fb4f88dca   Hugh Dickins   [PATCH] swap: get...
432
  			return swp_entry(type, offset);
5d337b919   Hugh Dickins   [PATCH] swap: swa...
433
  		}
fb4f88dca   Hugh Dickins   [PATCH] swap: get...
434
  		next = swap_list.next;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
435
  	}
fb4f88dca   Hugh Dickins   [PATCH] swap: get...
436
437
438
  
  	nr_swap_pages++;
  noswap:
5d337b919   Hugh Dickins   [PATCH] swap: swa...
439
  	spin_unlock(&swap_lock);
fb4f88dca   Hugh Dickins   [PATCH] swap: get...
440
  	return (swp_entry_t) {0};
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
441
  }
910321ea8   Hugh Dickins   swap: revert spec...
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
  /* The only caller of this function is now susupend routine */
  swp_entry_t get_swap_page_of_type(int type)
  {
  	struct swap_info_struct *si;
  	pgoff_t offset;
  
  	spin_lock(&swap_lock);
  	si = swap_info[type];
  	if (si && (si->flags & SWP_WRITEOK)) {
  		nr_swap_pages--;
  		/* This is called for allocating swap entry, not cache */
  		offset = scan_swap_map(si, 1);
  		if (offset) {
  			spin_unlock(&swap_lock);
  			return swp_entry(type, offset);
  		}
  		nr_swap_pages++;
  	}
  	spin_unlock(&swap_lock);
  	return (swp_entry_t) {0};
  }
73c34b6ac   Hugh Dickins   swap_info: miscel...
463
  static struct swap_info_struct *swap_info_get(swp_entry_t entry)
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
464
  {
73c34b6ac   Hugh Dickins   swap_info: miscel...
465
  	struct swap_info_struct *p;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
466
467
468
469
470
471
472
  	unsigned long offset, type;
  
  	if (!entry.val)
  		goto out;
  	type = swp_type(entry);
  	if (type >= nr_swapfiles)
  		goto bad_nofile;
efa90a981   Hugh Dickins   swap_info: change...
473
  	p = swap_info[type];
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
474
475
476
477
478
479
480
  	if (!(p->flags & SWP_USED))
  		goto bad_device;
  	offset = swp_offset(entry);
  	if (offset >= p->max)
  		goto bad_offset;
  	if (!p->swap_map[offset])
  		goto bad_free;
5d337b919   Hugh Dickins   [PATCH] swap: swa...
481
  	spin_lock(&swap_lock);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
  	return p;
  
  bad_free:
  	printk(KERN_ERR "swap_free: %s%08lx
  ", Unused_offset, entry.val);
  	goto out;
  bad_offset:
  	printk(KERN_ERR "swap_free: %s%08lx
  ", Bad_offset, entry.val);
  	goto out;
  bad_device:
  	printk(KERN_ERR "swap_free: %s%08lx
  ", Unused_file, entry.val);
  	goto out;
  bad_nofile:
  	printk(KERN_ERR "swap_free: %s%08lx
  ", Bad_file, entry.val);
  out:
  	return NULL;
886bb7e9c   Hugh Dickins   swapfile: remove ...
501
  }
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
502

8d69aaee8   Hugh Dickins   swap_info: swap_m...
503
504
  static unsigned char swap_entry_free(struct swap_info_struct *p,
  				     swp_entry_t entry, unsigned char usage)
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
505
  {
253d553ba   Hugh Dickins   swap_info: SWAP_H...
506
  	unsigned long offset = swp_offset(entry);
8d69aaee8   Hugh Dickins   swap_info: swap_m...
507
508
  	unsigned char count;
  	unsigned char has_cache;
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
509

253d553ba   Hugh Dickins   swap_info: SWAP_H...
510
511
512
  	count = p->swap_map[offset];
  	has_cache = count & SWAP_HAS_CACHE;
  	count &= ~SWAP_HAS_CACHE;
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
513

253d553ba   Hugh Dickins   swap_info: SWAP_H...
514
  	if (usage == SWAP_HAS_CACHE) {
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
515
  		VM_BUG_ON(!has_cache);
253d553ba   Hugh Dickins   swap_info: SWAP_H...
516
  		has_cache = 0;
aaa468653   Hugh Dickins   swap_info: note S...
517
518
519
520
521
522
  	} else if (count == SWAP_MAP_SHMEM) {
  		/*
  		 * Or we could insist on shmem.c using a special
  		 * swap_shmem_free() and free_shmem_swap_and_cache()...
  		 */
  		count = 0;
570a335b8   Hugh Dickins   swap_info: swap c...
523
524
525
526
527
528
529
530
531
  	} else if ((count & ~COUNT_CONTINUED) <= SWAP_MAP_MAX) {
  		if (count == COUNT_CONTINUED) {
  			if (swap_count_continued(p, offset, count))
  				count = SWAP_MAP_MAX | COUNT_CONTINUED;
  			else
  				count = SWAP_MAP_MAX;
  		} else
  			count--;
  	}
253d553ba   Hugh Dickins   swap_info: SWAP_H...
532
533
534
535
536
537
  
  	if (!count)
  		mem_cgroup_uncharge_swap(entry);
  
  	usage = count | has_cache;
  	p->swap_map[offset] = usage;
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
538

355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
539
  	/* free if no reference */
253d553ba   Hugh Dickins   swap_info: SWAP_H...
540
  	if (!usage) {
b3a27d052   Nitin Gupta   swap: Add swap sl...
541
  		struct gendisk *disk = p->bdev->bd_disk;
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
542
543
544
545
  		if (offset < p->lowest_bit)
  			p->lowest_bit = offset;
  		if (offset > p->highest_bit)
  			p->highest_bit = offset;
efa90a981   Hugh Dickins   swap_info: change...
546
547
548
  		if (swap_list.next >= 0 &&
  		    p->prio > swap_info[swap_list.next]->prio)
  			swap_list.next = p->type;
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
549
550
  		nr_swap_pages++;
  		p->inuse_pages--;
b3a27d052   Nitin Gupta   swap: Add swap sl...
551
552
553
  		if ((p->flags & SWP_BLKDEV) &&
  				disk->fops->swap_slot_free_notify)
  			disk->fops->swap_slot_free_notify(p->bdev, offset);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
554
  	}
253d553ba   Hugh Dickins   swap_info: SWAP_H...
555
556
  
  	return usage;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
557
558
559
560
561
562
563
564
  }
  
  /*
   * Caller has made sure that the swapdevice corresponding to entry
   * is still around or has not been recycled.
   */
  void swap_free(swp_entry_t entry)
  {
73c34b6ac   Hugh Dickins   swap_info: miscel...
565
  	struct swap_info_struct *p;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
566
567
568
  
  	p = swap_info_get(entry);
  	if (p) {
253d553ba   Hugh Dickins   swap_info: SWAP_H...
569
  		swap_entry_free(p, entry, 1);
5d337b919   Hugh Dickins   [PATCH] swap: swa...
570
  		spin_unlock(&swap_lock);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
571
572
573
574
  	}
  }
  
  /*
cb4b86ba4   KAMEZAWA Hiroyuki   mm: add swap cach...
575
576
577
578
   * Called after dropping swapcache to decrease refcnt to swap entries.
   */
  void swapcache_free(swp_entry_t entry, struct page *page)
  {
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
579
  	struct swap_info_struct *p;
8d69aaee8   Hugh Dickins   swap_info: swap_m...
580
  	unsigned char count;
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
581

355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
582
583
  	p = swap_info_get(entry);
  	if (p) {
253d553ba   Hugh Dickins   swap_info: SWAP_H...
584
585
586
  		count = swap_entry_free(p, entry, SWAP_HAS_CACHE);
  		if (page)
  			mem_cgroup_uncharge_swapcache(page, entry, count != 0);
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
587
588
  		spin_unlock(&swap_lock);
  	}
cb4b86ba4   KAMEZAWA Hiroyuki   mm: add swap cach...
589
590
591
  }
  
  /*
c475a8ab6   Hugh Dickins   [PATCH] can_share...
592
   * How many references to page are currently swapped out?
570a335b8   Hugh Dickins   swap_info: swap c...
593
594
   * This does not give an exact answer when swap count is continued,
   * but does include the high COUNT_CONTINUED flag to allow for that.
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
595
   */
c475a8ab6   Hugh Dickins   [PATCH] can_share...
596
  static inline int page_swapcount(struct page *page)
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
597
  {
c475a8ab6   Hugh Dickins   [PATCH] can_share...
598
599
  	int count = 0;
  	struct swap_info_struct *p;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
600
  	swp_entry_t entry;
4c21e2f24   Hugh Dickins   [PATCH] mm: split...
601
  	entry.val = page_private(page);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
602
603
  	p = swap_info_get(entry);
  	if (p) {
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
604
  		count = swap_count(p->swap_map[swp_offset(entry)]);
5d337b919   Hugh Dickins   [PATCH] swap: swa...
605
  		spin_unlock(&swap_lock);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
606
  	}
c475a8ab6   Hugh Dickins   [PATCH] can_share...
607
  	return count;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
608
609
610
  }
  
  /*
7b1fe5979   Hugh Dickins   mm: reuse_swap_pa...
611
612
613
614
   * We can write to an anon page without COW if there are no other references
   * to it.  And as a side-effect, free up its swap: because the old content
   * on disk will never be read, and seeking back there to write new content
   * later would only waste time away from clustering.
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
615
   */
7b1fe5979   Hugh Dickins   mm: reuse_swap_pa...
616
  int reuse_swap_page(struct page *page)
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
617
  {
c475a8ab6   Hugh Dickins   [PATCH] can_share...
618
  	int count;
51726b122   Hugh Dickins   mm: replace some ...
619
  	VM_BUG_ON(!PageLocked(page));
5ad646880   Hugh Dickins   ksm: let shared p...
620
621
  	if (unlikely(PageKsm(page)))
  		return 0;
c475a8ab6   Hugh Dickins   [PATCH] can_share...
622
  	count = page_mapcount(page);
7b1fe5979   Hugh Dickins   mm: reuse_swap_pa...
623
  	if (count <= 1 && PageSwapCache(page)) {
c475a8ab6   Hugh Dickins   [PATCH] can_share...
624
  		count += page_swapcount(page);
7b1fe5979   Hugh Dickins   mm: reuse_swap_pa...
625
626
627
628
629
  		if (count == 1 && !PageWriteback(page)) {
  			delete_from_swap_cache(page);
  			SetPageDirty(page);
  		}
  	}
5ad646880   Hugh Dickins   ksm: let shared p...
630
  	return count <= 1;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
631
632
633
  }
  
  /*
a2c43eed8   Hugh Dickins   mm: try_to_free_s...
634
635
   * If swap is getting full, or if there are no more mappings of this page,
   * then try_to_free_swap is called to free its swap space.
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
636
   */
a2c43eed8   Hugh Dickins   mm: try_to_free_s...
637
  int try_to_free_swap(struct page *page)
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
638
  {
51726b122   Hugh Dickins   mm: replace some ...
639
  	VM_BUG_ON(!PageLocked(page));
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
640
641
642
643
644
  
  	if (!PageSwapCache(page))
  		return 0;
  	if (PageWriteback(page))
  		return 0;
a2c43eed8   Hugh Dickins   mm: try_to_free_s...
645
  	if (page_swapcount(page))
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
646
  		return 0;
b73d7fcec   Hugh Dickins   swap: prevent reu...
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
  	/*
  	 * Once hibernation has begun to create its image of memory,
  	 * there's a danger that one of the calls to try_to_free_swap()
  	 * - most probably a call from __try_to_reclaim_swap() while
  	 * hibernation is allocating its own swap pages for the image,
  	 * but conceivably even a call from memory reclaim - will free
  	 * the swap from a page which has already been recorded in the
  	 * image as a clean swapcache page, and then reuse its swap for
  	 * another page of the image.  On waking from hibernation, the
  	 * original page might be freed under memory pressure, then
  	 * later read back in from swap, now with the wrong data.
  	 *
  	 * Hibernation clears bits from gfp_allowed_mask to prevent
  	 * memory reclaim from writing to disk, so check that here.
  	 */
  	if (!(gfp_allowed_mask & __GFP_IO))
  		return 0;
a2c43eed8   Hugh Dickins   mm: try_to_free_s...
664
665
666
  	delete_from_swap_cache(page);
  	SetPageDirty(page);
  	return 1;
68a22394c   Rik van Riel   vmscan: free swap...
667
668
669
  }
  
  /*
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
670
671
672
   * Free the swap entry like above, but also try to
   * free the page cache entry if it is the last user.
   */
2509ef26d   Hugh Dickins   badpage: zap prin...
673
  int free_swap_and_cache(swp_entry_t entry)
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
674
  {
2509ef26d   Hugh Dickins   badpage: zap prin...
675
  	struct swap_info_struct *p;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
676
  	struct page *page = NULL;
a7420aa54   Andi Kleen   HWPOISON: Add sup...
677
  	if (non_swap_entry(entry))
2509ef26d   Hugh Dickins   badpage: zap prin...
678
  		return 1;
0697212a4   Christoph Lameter   [PATCH] Swapless ...
679

1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
680
681
  	p = swap_info_get(entry);
  	if (p) {
253d553ba   Hugh Dickins   swap_info: SWAP_H...
682
  		if (swap_entry_free(p, entry, 1) == SWAP_HAS_CACHE) {
93fac7041   Nick Piggin   [PATCH] mm: sched...
683
  			page = find_get_page(&swapper_space, entry.val);
8413ac9d8   Nick Piggin   mm: page lock use...
684
  			if (page && !trylock_page(page)) {
93fac7041   Nick Piggin   [PATCH] mm: sched...
685
686
687
688
  				page_cache_release(page);
  				page = NULL;
  			}
  		}
5d337b919   Hugh Dickins   [PATCH] swap: swa...
689
  		spin_unlock(&swap_lock);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
690
691
  	}
  	if (page) {
a2c43eed8   Hugh Dickins   mm: try_to_free_s...
692
693
694
695
  		/*
  		 * Not mapped elsewhere, or swap space full? Free it!
  		 * Also recheck PageSwapCache now page is locked (above).
  		 */
93fac7041   Nick Piggin   [PATCH] mm: sched...
696
  		if (PageSwapCache(page) && !PageWriteback(page) &&
a2c43eed8   Hugh Dickins   mm: try_to_free_s...
697
  				(!page_mapped(page) || vm_swap_full())) {
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
698
699
700
701
702
703
  			delete_from_swap_cache(page);
  			SetPageDirty(page);
  		}
  		unlock_page(page);
  		page_cache_release(page);
  	}
2509ef26d   Hugh Dickins   badpage: zap prin...
704
  	return p != NULL;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
705
  }
024914477   Daisuke Nishimura   memcg: move charg...
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
  #ifdef CONFIG_CGROUP_MEM_RES_CTLR
  /**
   * mem_cgroup_count_swap_user - count the user of a swap entry
   * @ent: the swap entry to be checked
   * @pagep: the pointer for the swap cache page of the entry to be stored
   *
   * Returns the number of the user of the swap entry. The number is valid only
   * for swaps of anonymous pages.
   * If the entry is found on swap cache, the page is stored to pagep with
   * refcount of it being incremented.
   */
  int mem_cgroup_count_swap_user(swp_entry_t ent, struct page **pagep)
  {
  	struct page *page;
  	struct swap_info_struct *p;
  	int count = 0;
  
  	page = find_get_page(&swapper_space, ent.val);
  	if (page)
  		count += page_mapcount(page);
  	p = swap_info_get(ent);
  	if (p) {
  		count += swap_count(p->swap_map[swp_offset(ent)]);
  		spin_unlock(&swap_lock);
  	}
  
  	*pagep = page;
  	return count;
  }
  #endif
b0cb1a19d   Rafael J. Wysocki   Replace CONFIG_SO...
736
  #ifdef CONFIG_HIBERNATION
f577eb30a   Rafael J. Wysocki   [PATCH] swsusp: l...
737
  /*
915bae9eb   Rafael J. Wysocki   [PATCH] swsusp: u...
738
   * Find the swap type that corresponds to given device (if any).
f577eb30a   Rafael J. Wysocki   [PATCH] swsusp: l...
739
   *
915bae9eb   Rafael J. Wysocki   [PATCH] swsusp: u...
740
741
742
743
   * @offset - number of the PAGE_SIZE-sized block of the device, starting
   * from 0, in which the swap header is expected to be located.
   *
   * This is needed for the suspend to disk (aka swsusp).
f577eb30a   Rafael J. Wysocki   [PATCH] swsusp: l...
744
   */
7bf236874   Rafael J. Wysocki   [PATCH] swsusp: D...
745
  int swap_type_of(dev_t device, sector_t offset, struct block_device **bdev_p)
f577eb30a   Rafael J. Wysocki   [PATCH] swsusp: l...
746
  {
915bae9eb   Rafael J. Wysocki   [PATCH] swsusp: u...
747
  	struct block_device *bdev = NULL;
efa90a981   Hugh Dickins   swap_info: change...
748
  	int type;
f577eb30a   Rafael J. Wysocki   [PATCH] swsusp: l...
749

915bae9eb   Rafael J. Wysocki   [PATCH] swsusp: u...
750
751
  	if (device)
  		bdev = bdget(device);
f577eb30a   Rafael J. Wysocki   [PATCH] swsusp: l...
752
  	spin_lock(&swap_lock);
efa90a981   Hugh Dickins   swap_info: change...
753
754
  	for (type = 0; type < nr_swapfiles; type++) {
  		struct swap_info_struct *sis = swap_info[type];
f577eb30a   Rafael J. Wysocki   [PATCH] swsusp: l...
755

915bae9eb   Rafael J. Wysocki   [PATCH] swsusp: u...
756
  		if (!(sis->flags & SWP_WRITEOK))
f577eb30a   Rafael J. Wysocki   [PATCH] swsusp: l...
757
  			continue;
b6b5bce35   Rafael J. Wysocki   [PATCH] swsusp: F...
758

915bae9eb   Rafael J. Wysocki   [PATCH] swsusp: u...
759
  		if (!bdev) {
7bf236874   Rafael J. Wysocki   [PATCH] swsusp: D...
760
  			if (bdev_p)
dddac6a7b   Alan Jenkins   PM / Hibernate: R...
761
  				*bdev_p = bdgrab(sis->bdev);
7bf236874   Rafael J. Wysocki   [PATCH] swsusp: D...
762

6e1819d61   Rafael J. Wysocki   [PATCH] swsusp: u...
763
  			spin_unlock(&swap_lock);
efa90a981   Hugh Dickins   swap_info: change...
764
  			return type;
6e1819d61   Rafael J. Wysocki   [PATCH] swsusp: u...
765
  		}
915bae9eb   Rafael J. Wysocki   [PATCH] swsusp: u...
766
  		if (bdev == sis->bdev) {
9625a5f28   Hugh Dickins   swap_info: includ...
767
  			struct swap_extent *se = &sis->first_swap_extent;
915bae9eb   Rafael J. Wysocki   [PATCH] swsusp: u...
768

915bae9eb   Rafael J. Wysocki   [PATCH] swsusp: u...
769
  			if (se->start_block == offset) {
7bf236874   Rafael J. Wysocki   [PATCH] swsusp: D...
770
  				if (bdev_p)
dddac6a7b   Alan Jenkins   PM / Hibernate: R...
771
  					*bdev_p = bdgrab(sis->bdev);
7bf236874   Rafael J. Wysocki   [PATCH] swsusp: D...
772

915bae9eb   Rafael J. Wysocki   [PATCH] swsusp: u...
773
774
  				spin_unlock(&swap_lock);
  				bdput(bdev);
efa90a981   Hugh Dickins   swap_info: change...
775
  				return type;
915bae9eb   Rafael J. Wysocki   [PATCH] swsusp: u...
776
  			}
f577eb30a   Rafael J. Wysocki   [PATCH] swsusp: l...
777
778
779
  		}
  	}
  	spin_unlock(&swap_lock);
915bae9eb   Rafael J. Wysocki   [PATCH] swsusp: u...
780
781
  	if (bdev)
  		bdput(bdev);
f577eb30a   Rafael J. Wysocki   [PATCH] swsusp: l...
782
783
784
785
  	return -ENODEV;
  }
  
  /*
73c34b6ac   Hugh Dickins   swap_info: miscel...
786
787
788
789
790
791
792
793
794
795
796
   * Get the (PAGE_SIZE) block corresponding to given offset on the swapdev
   * corresponding to given index in swap_info (swap type).
   */
  sector_t swapdev_block(int type, pgoff_t offset)
  {
  	struct block_device *bdev;
  
  	if ((unsigned int)type >= nr_swapfiles)
  		return 0;
  	if (!(swap_info[type]->flags & SWP_WRITEOK))
  		return 0;
d4906e1aa   Lee Schermerhorn   swap: rework map_...
797
  	return map_swap_entry(swp_entry(type, offset), &bdev);
73c34b6ac   Hugh Dickins   swap_info: miscel...
798
799
800
  }
  
  /*
f577eb30a   Rafael J. Wysocki   [PATCH] swsusp: l...
801
802
803
804
805
806
807
808
   * Return either the total number of swap pages of given type, or the number
   * of free pages of that type (depending on @free)
   *
   * This is needed for software suspend
   */
  unsigned int count_swap_pages(int type, int free)
  {
  	unsigned int n = 0;
efa90a981   Hugh Dickins   swap_info: change...
809
810
811
812
813
814
  	spin_lock(&swap_lock);
  	if ((unsigned int)type < nr_swapfiles) {
  		struct swap_info_struct *sis = swap_info[type];
  
  		if (sis->flags & SWP_WRITEOK) {
  			n = sis->pages;
f577eb30a   Rafael J. Wysocki   [PATCH] swsusp: l...
815
  			if (free)
efa90a981   Hugh Dickins   swap_info: change...
816
  				n -= sis->inuse_pages;
f577eb30a   Rafael J. Wysocki   [PATCH] swsusp: l...
817
  		}
f577eb30a   Rafael J. Wysocki   [PATCH] swsusp: l...
818
  	}
efa90a981   Hugh Dickins   swap_info: change...
819
  	spin_unlock(&swap_lock);
f577eb30a   Rafael J. Wysocki   [PATCH] swsusp: l...
820
821
  	return n;
  }
73c34b6ac   Hugh Dickins   swap_info: miscel...
822
  #endif /* CONFIG_HIBERNATION */
f577eb30a   Rafael J. Wysocki   [PATCH] swsusp: l...
823

1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
824
  /*
72866f6f2   Hugh Dickins   [PATCH] mm: anon ...
825
826
827
   * No need to decide whether this PTE shares the swap entry with others,
   * just let do_wp_page work it out if a write is requested later - to
   * force COW, vm_page_prot omits write permission from any private vma.
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
828
   */
044d66c1d   Hugh Dickins   memcgroup: reinst...
829
  static int unuse_pte(struct vm_area_struct *vma, pmd_t *pmd,
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
830
831
  		unsigned long addr, swp_entry_t entry, struct page *page)
  {
56039efa1   KAMEZAWA Hiroyuki   memcg: fix ugly i...
832
  	struct mem_cgroup *ptr;
044d66c1d   Hugh Dickins   memcgroup: reinst...
833
834
835
  	spinlock_t *ptl;
  	pte_t *pte;
  	int ret = 1;
85d9fc89f   KAMEZAWA Hiroyuki   memcg: fix refcnt...
836
  	if (mem_cgroup_try_charge_swapin(vma->vm_mm, page, GFP_KERNEL, &ptr)) {
044d66c1d   Hugh Dickins   memcgroup: reinst...
837
  		ret = -ENOMEM;
85d9fc89f   KAMEZAWA Hiroyuki   memcg: fix refcnt...
838
839
  		goto out_nolock;
  	}
044d66c1d   Hugh Dickins   memcgroup: reinst...
840
841
842
843
  
  	pte = pte_offset_map_lock(vma->vm_mm, pmd, addr, &ptl);
  	if (unlikely(!pte_same(*pte, swp_entry_to_pte(entry)))) {
  		if (ret > 0)
7a81b88cb   KAMEZAWA Hiroyuki   memcg: introduce ...
844
  			mem_cgroup_cancel_charge_swapin(ptr);
044d66c1d   Hugh Dickins   memcgroup: reinst...
845
846
847
  		ret = 0;
  		goto out;
  	}
8a9f3ccd2   Balbir Singh   Memory controller...
848

b084d4353   KAMEZAWA Hiroyuki   mm: count swap usage
849
  	dec_mm_counter(vma->vm_mm, MM_SWAPENTS);
d559db086   KAMEZAWA Hiroyuki   mm: clean up mm_c...
850
  	inc_mm_counter(vma->vm_mm, MM_ANONPAGES);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
851
852
853
854
  	get_page(page);
  	set_pte_at(vma->vm_mm, addr, pte,
  		   pte_mkold(mk_pte(page, vma->vm_page_prot)));
  	page_add_anon_rmap(page, vma, addr);
7a81b88cb   KAMEZAWA Hiroyuki   memcg: introduce ...
855
  	mem_cgroup_commit_charge_swapin(page, ptr);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
856
857
858
859
860
861
  	swap_free(entry);
  	/*
  	 * Move the page to the active list so it is not
  	 * immediately swapped out again after swapon.
  	 */
  	activate_page(page);
044d66c1d   Hugh Dickins   memcgroup: reinst...
862
863
  out:
  	pte_unmap_unlock(pte, ptl);
85d9fc89f   KAMEZAWA Hiroyuki   memcg: fix refcnt...
864
  out_nolock:
044d66c1d   Hugh Dickins   memcgroup: reinst...
865
  	return ret;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
866
867
868
869
870
871
  }
  
  static int unuse_pte_range(struct vm_area_struct *vma, pmd_t *pmd,
  				unsigned long addr, unsigned long end,
  				swp_entry_t entry, struct page *page)
  {
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
872
  	pte_t swp_pte = swp_entry_to_pte(entry);
705e87c0c   Hugh Dickins   [PATCH] mm: pte_o...
873
  	pte_t *pte;
8a9f3ccd2   Balbir Singh   Memory controller...
874
  	int ret = 0;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
875

044d66c1d   Hugh Dickins   memcgroup: reinst...
876
877
878
879
880
881
882
883
884
885
  	/*
  	 * We don't actually need pte lock while scanning for swp_pte: since
  	 * we hold page lock and mmap_sem, swp_pte cannot be inserted into the
  	 * page table while we're scanning; though it could get zapped, and on
  	 * some architectures (e.g. x86_32 with PAE) we might catch a glimpse
  	 * of unmatched parts which look like swp_pte, so unuse_pte must
  	 * recheck under pte lock.  Scanning without pte lock lets it be
  	 * preemptible whenever CONFIG_PREEMPT but not CONFIG_HIGHPTE.
  	 */
  	pte = pte_offset_map(pmd, addr);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
886
887
888
889
890
891
  	do {
  		/*
  		 * swapoff spends a _lot_ of time in this loop!
  		 * Test inline before going to call unuse_pte.
  		 */
  		if (unlikely(pte_same(*pte, swp_pte))) {
044d66c1d   Hugh Dickins   memcgroup: reinst...
892
893
894
895
896
  			pte_unmap(pte);
  			ret = unuse_pte(vma, pmd, addr, entry, page);
  			if (ret)
  				goto out;
  			pte = pte_offset_map(pmd, addr);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
897
898
  		}
  	} while (pte++, addr += PAGE_SIZE, addr != end);
044d66c1d   Hugh Dickins   memcgroup: reinst...
899
900
  	pte_unmap(pte - 1);
  out:
8a9f3ccd2   Balbir Singh   Memory controller...
901
  	return ret;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
902
903
904
905
906
907
908
909
  }
  
  static inline int unuse_pmd_range(struct vm_area_struct *vma, pud_t *pud,
  				unsigned long addr, unsigned long end,
  				swp_entry_t entry, struct page *page)
  {
  	pmd_t *pmd;
  	unsigned long next;
8a9f3ccd2   Balbir Singh   Memory controller...
910
  	int ret;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
911
912
913
914
  
  	pmd = pmd_offset(pud, addr);
  	do {
  		next = pmd_addr_end(addr, end);
3f04f62f9   Andrea Arcangeli   thp: split_huge_p...
915
916
  		if (unlikely(pmd_trans_huge(*pmd)))
  			continue;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
917
918
  		if (pmd_none_or_clear_bad(pmd))
  			continue;
8a9f3ccd2   Balbir Singh   Memory controller...
919
920
921
  		ret = unuse_pte_range(vma, pmd, addr, next, entry, page);
  		if (ret)
  			return ret;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
922
923
924
925
926
927
928
929
930
931
  	} while (pmd++, addr = next, addr != end);
  	return 0;
  }
  
  static inline int unuse_pud_range(struct vm_area_struct *vma, pgd_t *pgd,
  				unsigned long addr, unsigned long end,
  				swp_entry_t entry, struct page *page)
  {
  	pud_t *pud;
  	unsigned long next;
8a9f3ccd2   Balbir Singh   Memory controller...
932
  	int ret;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
933
934
935
936
937
938
  
  	pud = pud_offset(pgd, addr);
  	do {
  		next = pud_addr_end(addr, end);
  		if (pud_none_or_clear_bad(pud))
  			continue;
8a9f3ccd2   Balbir Singh   Memory controller...
939
940
941
  		ret = unuse_pmd_range(vma, pud, addr, next, entry, page);
  		if (ret)
  			return ret;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
942
943
944
945
946
947
948
949
950
  	} while (pud++, addr = next, addr != end);
  	return 0;
  }
  
  static int unuse_vma(struct vm_area_struct *vma,
  				swp_entry_t entry, struct page *page)
  {
  	pgd_t *pgd;
  	unsigned long addr, end, next;
8a9f3ccd2   Balbir Singh   Memory controller...
951
  	int ret;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
952

3ca7b3c5b   Hugh Dickins   mm: define PAGE_M...
953
  	if (page_anon_vma(page)) {
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
  		addr = page_address_in_vma(page, vma);
  		if (addr == -EFAULT)
  			return 0;
  		else
  			end = addr + PAGE_SIZE;
  	} else {
  		addr = vma->vm_start;
  		end = vma->vm_end;
  	}
  
  	pgd = pgd_offset(vma->vm_mm, addr);
  	do {
  		next = pgd_addr_end(addr, end);
  		if (pgd_none_or_clear_bad(pgd))
  			continue;
8a9f3ccd2   Balbir Singh   Memory controller...
969
970
971
  		ret = unuse_pud_range(vma, pgd, addr, next, entry, page);
  		if (ret)
  			return ret;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
972
973
974
975
976
977
978
979
  	} while (pgd++, addr = next, addr != end);
  	return 0;
  }
  
  static int unuse_mm(struct mm_struct *mm,
  				swp_entry_t entry, struct page *page)
  {
  	struct vm_area_struct *vma;
8a9f3ccd2   Balbir Singh   Memory controller...
980
  	int ret = 0;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
981
982
983
  
  	if (!down_read_trylock(&mm->mmap_sem)) {
  		/*
7d03431cf   Fernando Luis Vazquez Cao   swapfile/vmscan: ...
984
985
  		 * Activate page so shrink_inactive_list is unlikely to unmap
  		 * its ptes while lock is dropped, so swapoff can make progress.
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
986
  		 */
c475a8ab6   Hugh Dickins   [PATCH] can_share...
987
  		activate_page(page);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
988
989
990
991
  		unlock_page(page);
  		down_read(&mm->mmap_sem);
  		lock_page(page);
  	}
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
992
  	for (vma = mm->mmap; vma; vma = vma->vm_next) {
8a9f3ccd2   Balbir Singh   Memory controller...
993
  		if (vma->anon_vma && (ret = unuse_vma(vma, entry, page)))
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
994
995
  			break;
  	}
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
996
  	up_read(&mm->mmap_sem);
8a9f3ccd2   Balbir Singh   Memory controller...
997
  	return (ret < 0)? ret: 0;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
998
999
1000
1001
1002
1003
  }
  
  /*
   * Scan swap_map from current position to next entry still in use.
   * Recycle to start on reaching the end, returning 0 when empty.
   */
6eb396dc4   Hugh Dickins   [PATCH] swap: swa...
1004
1005
  static unsigned int find_next_to_unuse(struct swap_info_struct *si,
  					unsigned int prev)
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1006
  {
6eb396dc4   Hugh Dickins   [PATCH] swap: swa...
1007
1008
  	unsigned int max = si->max;
  	unsigned int i = prev;
8d69aaee8   Hugh Dickins   swap_info: swap_m...
1009
  	unsigned char count;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1010
1011
  
  	/*
5d337b919   Hugh Dickins   [PATCH] swap: swa...
1012
  	 * No need for swap_lock here: we're just looking
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1013
1014
  	 * for whether an entry is in use, not modifying it; false
  	 * hits are okay, and sys_swapoff() has already prevented new
5d337b919   Hugh Dickins   [PATCH] swap: swa...
1015
  	 * allocations from this area (while holding swap_lock).
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1016
1017
1018
1019
1020
1021
1022
1023
1024
1025
1026
1027
1028
1029
1030
1031
  	 */
  	for (;;) {
  		if (++i >= max) {
  			if (!prev) {
  				i = 0;
  				break;
  			}
  			/*
  			 * No entries in use at top of swap_map,
  			 * loop back to start and recheck there.
  			 */
  			max = prev + 1;
  			prev = 0;
  			i = 1;
  		}
  		count = si->swap_map[i];
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
1032
  		if (count && swap_count(count) != SWAP_MAP_BAD)
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1033
1034
1035
1036
1037
1038
1039
1040
1041
1042
1043
1044
  			break;
  	}
  	return i;
  }
  
  /*
   * We completely avoid races by reading each swap page in advance,
   * and then search for the process using it.  All the necessary
   * page table adjustments can then be made atomically.
   */
  static int try_to_unuse(unsigned int type)
  {
efa90a981   Hugh Dickins   swap_info: change...
1045
  	struct swap_info_struct *si = swap_info[type];
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1046
  	struct mm_struct *start_mm;
8d69aaee8   Hugh Dickins   swap_info: swap_m...
1047
1048
  	unsigned char *swap_map;
  	unsigned char swcount;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1049
1050
  	struct page *page;
  	swp_entry_t entry;
6eb396dc4   Hugh Dickins   [PATCH] swap: swa...
1051
  	unsigned int i = 0;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1052
  	int retval = 0;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1053
1054
1055
1056
1057
1058
1059
1060
1061
1062
1063
1064
1065
  
  	/*
  	 * When searching mms for an entry, a good strategy is to
  	 * start at the first mm we freed the previous entry from
  	 * (though actually we don't notice whether we or coincidence
  	 * freed the entry).  Initialize this start_mm with a hold.
  	 *
  	 * A simpler strategy would be to start at the last mm we
  	 * freed the previous entry from; but that would take less
  	 * advantage of mmlist ordering, which clusters forked mms
  	 * together, child after parent.  If we race with dup_mmap(), we
  	 * prefer to resolve parent before child, lest we miss entries
  	 * duplicated after we scanned child: using last mm would invert
570a335b8   Hugh Dickins   swap_info: swap c...
1066
  	 * that.
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1067
1068
1069
1070
1071
1072
1073
1074
1075
1076
1077
1078
1079
1080
  	 */
  	start_mm = &init_mm;
  	atomic_inc(&init_mm.mm_users);
  
  	/*
  	 * Keep on scanning until all entries have gone.  Usually,
  	 * one pass through swap_map is enough, but not necessarily:
  	 * there are races when an instance of an entry might be missed.
  	 */
  	while ((i = find_next_to_unuse(si, i)) != 0) {
  		if (signal_pending(current)) {
  			retval = -EINTR;
  			break;
  		}
886bb7e9c   Hugh Dickins   swapfile: remove ...
1081
  		/*
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1082
1083
  		 * Get a page for the entry, using the existing swap
  		 * cache page if there is one.  Otherwise, get a clean
886bb7e9c   Hugh Dickins   swapfile: remove ...
1084
  		 * page and read the swap into it.
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1085
1086
1087
  		 */
  		swap_map = &si->swap_map[i];
  		entry = swp_entry(type, i);
02098feaa   Hugh Dickins   swapin needs gfp_...
1088
1089
  		page = read_swap_cache_async(entry,
  					GFP_HIGHUSER_MOVABLE, NULL, 0);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1090
1091
1092
1093
1094
1095
1096
1097
1098
1099
1100
1101
1102
1103
1104
1105
1106
1107
1108
1109
1110
1111
1112
1113
1114
1115
1116
1117
1118
1119
1120
1121
1122
1123
1124
1125
1126
  		if (!page) {
  			/*
  			 * Either swap_duplicate() failed because entry
  			 * has been freed independently, and will not be
  			 * reused since sys_swapoff() already disabled
  			 * allocation from here, or alloc_page() failed.
  			 */
  			if (!*swap_map)
  				continue;
  			retval = -ENOMEM;
  			break;
  		}
  
  		/*
  		 * Don't hold on to start_mm if it looks like exiting.
  		 */
  		if (atomic_read(&start_mm->mm_users) == 1) {
  			mmput(start_mm);
  			start_mm = &init_mm;
  			atomic_inc(&init_mm.mm_users);
  		}
  
  		/*
  		 * Wait for and lock page.  When do_swap_page races with
  		 * try_to_unuse, do_swap_page can handle the fault much
  		 * faster than try_to_unuse can locate the entry.  This
  		 * apparently redundant "wait_on_page_locked" lets try_to_unuse
  		 * defer to do_swap_page in such a case - in some tests,
  		 * do_swap_page and try_to_unuse repeatedly compete.
  		 */
  		wait_on_page_locked(page);
  		wait_on_page_writeback(page);
  		lock_page(page);
  		wait_on_page_writeback(page);
  
  		/*
  		 * Remove all references to entry.
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1127
  		 */
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1128
  		swcount = *swap_map;
aaa468653   Hugh Dickins   swap_info: note S...
1129
1130
1131
1132
1133
1134
  		if (swap_count(swcount) == SWAP_MAP_SHMEM) {
  			retval = shmem_unuse(entry, page);
  			/* page has already been unlocked and released */
  			if (retval < 0)
  				break;
  			continue;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1135
  		}
aaa468653   Hugh Dickins   swap_info: note S...
1136
1137
  		if (swap_count(swcount) && start_mm != &init_mm)
  			retval = unuse_mm(start_mm, entry, page);
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
1138
  		if (swap_count(*swap_map)) {
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1139
1140
1141
1142
1143
1144
1145
1146
1147
  			int set_start_mm = (*swap_map >= swcount);
  			struct list_head *p = &start_mm->mmlist;
  			struct mm_struct *new_start_mm = start_mm;
  			struct mm_struct *prev_mm = start_mm;
  			struct mm_struct *mm;
  
  			atomic_inc(&new_start_mm->mm_users);
  			atomic_inc(&prev_mm->mm_users);
  			spin_lock(&mmlist_lock);
aaa468653   Hugh Dickins   swap_info: note S...
1148
  			while (swap_count(*swap_map) && !retval &&
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1149
1150
  					(p = p->next) != &start_mm->mmlist) {
  				mm = list_entry(p, struct mm_struct, mmlist);
70af7c5c6   Hugh Dickins   [PATCH] swapoff: ...
1151
  				if (!atomic_inc_not_zero(&mm->mm_users))
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1152
  					continue;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1153
1154
1155
1156
1157
1158
1159
  				spin_unlock(&mmlist_lock);
  				mmput(prev_mm);
  				prev_mm = mm;
  
  				cond_resched();
  
  				swcount = *swap_map;
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
1160
  				if (!swap_count(swcount)) /* any usage ? */
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1161
  					;
aaa468653   Hugh Dickins   swap_info: note S...
1162
  				else if (mm == &init_mm)
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1163
  					set_start_mm = 1;
aaa468653   Hugh Dickins   swap_info: note S...
1164
  				else
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1165
  					retval = unuse_mm(mm, entry, page);
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
1166

32c5fc10e   Bo Liu   mm: remove incorr...
1167
  				if (set_start_mm && *swap_map < swcount) {
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1168
1169
1170
1171
1172
1173
1174
1175
1176
1177
1178
1179
1180
1181
1182
1183
1184
1185
1186
  					mmput(new_start_mm);
  					atomic_inc(&mm->mm_users);
  					new_start_mm = mm;
  					set_start_mm = 0;
  				}
  				spin_lock(&mmlist_lock);
  			}
  			spin_unlock(&mmlist_lock);
  			mmput(prev_mm);
  			mmput(start_mm);
  			start_mm = new_start_mm;
  		}
  		if (retval) {
  			unlock_page(page);
  			page_cache_release(page);
  			break;
  		}
  
  		/*
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1187
1188
1189
1190
1191
1192
1193
1194
1195
1196
1197
  		 * If a reference remains (rare), we would like to leave
  		 * the page in the swap cache; but try_to_unmap could
  		 * then re-duplicate the entry once we drop page lock,
  		 * so we might loop indefinitely; also, that page could
  		 * not be swapped out to other storage meanwhile.  So:
  		 * delete from cache even if there's another reference,
  		 * after ensuring that the data has been saved to disk -
  		 * since if the reference remains (rarer), it will be
  		 * read from disk into another page.  Splitting into two
  		 * pages would be incorrect if swap supported "shared
  		 * private" pages, but they are handled by tmpfs files.
5ad646880   Hugh Dickins   ksm: let shared p...
1198
1199
1200
1201
1202
1203
  		 *
  		 * Given how unuse_vma() targets one particular offset
  		 * in an anon_vma, once the anon_vma has been determined,
  		 * this splitting happens to be just what is needed to
  		 * handle where KSM pages have been swapped out: re-reading
  		 * is unnecessarily slow, but we can fix that later on.
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1204
  		 */
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
1205
1206
  		if (swap_count(*swap_map) &&
  		     PageDirty(page) && PageSwapCache(page)) {
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1207
1208
1209
1210
1211
1212
1213
1214
  			struct writeback_control wbc = {
  				.sync_mode = WB_SYNC_NONE,
  			};
  
  			swap_writepage(page, &wbc);
  			lock_page(page);
  			wait_on_page_writeback(page);
  		}
68bdc8d64   Hugh Dickins   mm: try_to_unuse ...
1215
1216
1217
1218
1219
1220
1221
1222
1223
1224
  
  		/*
  		 * It is conceivable that a racing task removed this page from
  		 * swap cache just before we acquired the page lock at the top,
  		 * or while we dropped it in unuse_mm().  The page might even
  		 * be back in swap cache on another swap area: that we must not
  		 * delete, since it may not have been written out to swap yet.
  		 */
  		if (PageSwapCache(page) &&
  		    likely(page_private(page) == entry.val))
2e0e26c76   Hugh Dickins   tmpfs: open a win...
1225
  			delete_from_swap_cache(page);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1226
1227
1228
1229
  
  		/*
  		 * So we could skip searching mms once swap count went
  		 * to 1, we did not mark any present ptes as dirty: must
2706a1b89   Anderson Briglia   vmscan: fix comme...
1230
  		 * mark page dirty so shrink_page_list will preserve it.
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1231
1232
1233
1234
1235
1236
1237
1238
1239
1240
1241
1242
1243
  		 */
  		SetPageDirty(page);
  		unlock_page(page);
  		page_cache_release(page);
  
  		/*
  		 * Make sure that we aren't completely killing
  		 * interactive performance.
  		 */
  		cond_resched();
  	}
  
  	mmput(start_mm);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1244
1245
1246
1247
  	return retval;
  }
  
  /*
5d337b919   Hugh Dickins   [PATCH] swap: swa...
1248
1249
1250
   * After a successful try_to_unuse, if no swap is now in use, we know
   * we can empty the mmlist.  swap_lock must be held on entry and exit.
   * Note that mmlist_lock nests inside swap_lock, and an mm must be
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1251
1252
1253
1254
1255
   * added to the mmlist just after page_duplicate - before would be racy.
   */
  static void drain_mmlist(void)
  {
  	struct list_head *p, *next;
efa90a981   Hugh Dickins   swap_info: change...
1256
  	unsigned int type;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1257

efa90a981   Hugh Dickins   swap_info: change...
1258
1259
  	for (type = 0; type < nr_swapfiles; type++)
  		if (swap_info[type]->inuse_pages)
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1260
1261
1262
1263
1264
1265
1266
1267
1268
  			return;
  	spin_lock(&mmlist_lock);
  	list_for_each_safe(p, next, &init_mm.mmlist)
  		list_del_init(p);
  	spin_unlock(&mmlist_lock);
  }
  
  /*
   * Use this swapdev's extent info to locate the (PAGE_SIZE) block which
d4906e1aa   Lee Schermerhorn   swap: rework map_...
1269
1270
1271
   * corresponds to page offset for the specified swap entry.
   * Note that the type of this function is sector_t, but it returns page offset
   * into the bdev, not sector offset.
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1272
   */
d4906e1aa   Lee Schermerhorn   swap: rework map_...
1273
  static sector_t map_swap_entry(swp_entry_t entry, struct block_device **bdev)
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1274
  {
f29ad6a99   Hugh Dickins   swap_info: privat...
1275
1276
1277
1278
  	struct swap_info_struct *sis;
  	struct swap_extent *start_se;
  	struct swap_extent *se;
  	pgoff_t offset;
efa90a981   Hugh Dickins   swap_info: change...
1279
  	sis = swap_info[swp_type(entry)];
f29ad6a99   Hugh Dickins   swap_info: privat...
1280
1281
1282
1283
1284
  	*bdev = sis->bdev;
  
  	offset = swp_offset(entry);
  	start_se = sis->curr_swap_extent;
  	se = start_se;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1285
1286
1287
1288
1289
1290
1291
1292
  
  	for ( ; ; ) {
  		struct list_head *lh;
  
  		if (se->start_page <= offset &&
  				offset < (se->start_page + se->nr_pages)) {
  			return se->start_block + (offset - se->start_page);
  		}
11d31886d   Hugh Dickins   [PATCH] swap: swa...
1293
  		lh = se->list.next;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1294
1295
1296
1297
1298
1299
1300
  		se = list_entry(lh, struct swap_extent, list);
  		sis->curr_swap_extent = se;
  		BUG_ON(se == start_se);		/* It *must* be present */
  	}
  }
  
  /*
d4906e1aa   Lee Schermerhorn   swap: rework map_...
1301
1302
1303
1304
1305
1306
1307
1308
1309
1310
   * Returns the page offset into bdev for the specified page's swap entry.
   */
  sector_t map_swap_page(struct page *page, struct block_device **bdev)
  {
  	swp_entry_t entry;
  	entry.val = page_private(page);
  	return map_swap_entry(entry, bdev);
  }
  
  /*
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1311
1312
1313
1314
   * Free all of a swapdev's extent information
   */
  static void destroy_swap_extents(struct swap_info_struct *sis)
  {
9625a5f28   Hugh Dickins   swap_info: includ...
1315
  	while (!list_empty(&sis->first_swap_extent.list)) {
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1316
  		struct swap_extent *se;
9625a5f28   Hugh Dickins   swap_info: includ...
1317
  		se = list_entry(sis->first_swap_extent.list.next,
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1318
1319
1320
1321
  				struct swap_extent, list);
  		list_del(&se->list);
  		kfree(se);
  	}
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1322
1323
1324
1325
  }
  
  /*
   * Add a block range (and the corresponding page range) into this swapdev's
11d31886d   Hugh Dickins   [PATCH] swap: swa...
1326
   * extent list.  The extent list is kept sorted in page order.
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1327
   *
11d31886d   Hugh Dickins   [PATCH] swap: swa...
1328
   * This function rather assumes that it is called in ascending page order.
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1329
1330
1331
1332
1333
1334
1335
1336
   */
  static int
  add_swap_extent(struct swap_info_struct *sis, unsigned long start_page,
  		unsigned long nr_pages, sector_t start_block)
  {
  	struct swap_extent *se;
  	struct swap_extent *new_se;
  	struct list_head *lh;
9625a5f28   Hugh Dickins   swap_info: includ...
1337
1338
1339
1340
1341
1342
1343
1344
1345
  	if (start_page == 0) {
  		se = &sis->first_swap_extent;
  		sis->curr_swap_extent = se;
  		se->start_page = 0;
  		se->nr_pages = nr_pages;
  		se->start_block = start_block;
  		return 1;
  	} else {
  		lh = sis->first_swap_extent.list.prev;	/* Highest extent */
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1346
  		se = list_entry(lh, struct swap_extent, list);
11d31886d   Hugh Dickins   [PATCH] swap: swa...
1347
1348
  		BUG_ON(se->start_page + se->nr_pages != start_page);
  		if (se->start_block + se->nr_pages == start_block) {
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1349
1350
1351
1352
  			/* Merge it */
  			se->nr_pages += nr_pages;
  			return 0;
  		}
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1353
1354
1355
1356
1357
1358
1359
1360
1361
1362
1363
  	}
  
  	/*
  	 * No merge.  Insert a new extent, preserving ordering.
  	 */
  	new_se = kmalloc(sizeof(*se), GFP_KERNEL);
  	if (new_se == NULL)
  		return -ENOMEM;
  	new_se->start_page = start_page;
  	new_se->nr_pages = nr_pages;
  	new_se->start_block = start_block;
9625a5f28   Hugh Dickins   swap_info: includ...
1364
  	list_add_tail(&new_se->list, &sis->first_swap_extent.list);
53092a740   Hugh Dickins   [PATCH] swap: sho...
1365
  	return 1;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1366
1367
1368
1369
1370
1371
1372
1373
1374
1375
1376
1377
1378
1379
1380
1381
1382
1383
1384
1385
1386
1387
  }
  
  /*
   * A `swap extent' is a simple thing which maps a contiguous range of pages
   * onto a contiguous range of disk blocks.  An ordered list of swap extents
   * is built at swapon time and is then used at swap_writepage/swap_readpage
   * time for locating where on disk a page belongs.
   *
   * If the swapfile is an S_ISBLK block device, a single extent is installed.
   * This is done so that the main operating code can treat S_ISBLK and S_ISREG
   * swap files identically.
   *
   * Whether the swapdev is an S_ISREG file or an S_ISBLK blockdev, the swap
   * extent list operates in PAGE_SIZE disk blocks.  Both S_ISREG and S_ISBLK
   * swapfiles are handled *identically* after swapon time.
   *
   * For S_ISREG swapfiles, setup_swap_extents() will walk all the file's blocks
   * and will parse them into an ordered extent list, in PAGE_SIZE chunks.  If
   * some stray blocks are found which do not fall within the PAGE_SIZE alignment
   * requirements, they are simply tossed out - we will never use those blocks
   * for swapping.
   *
b0d9bcd4b   Hugh Dickins   [PATCH] swap: upd...
1388
   * For S_ISREG swapfiles we set S_SWAPFILE across the life of the swapon.  This
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1389
1390
1391
1392
1393
1394
1395
1396
1397
1398
   * prevents root from shooting her foot off by ftruncating an in-use swapfile,
   * which will scribble on the fs.
   *
   * The amount of disk space which a single swap extent represents varies.
   * Typically it is in the 1-4 megabyte range.  So we can have hundreds of
   * extents in the list.  To avoid much list walking, we cache the previous
   * search location in `curr_swap_extent', and start new searches from there.
   * This is extremely effective.  The average number of iterations in
   * map_swap_page() has been measured at about 0.3 per page.  - akpm.
   */
53092a740   Hugh Dickins   [PATCH] swap: sho...
1399
  static int setup_swap_extents(struct swap_info_struct *sis, sector_t *span)
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1400
1401
1402
1403
1404
1405
1406
  {
  	struct inode *inode;
  	unsigned blocks_per_page;
  	unsigned long page_no;
  	unsigned blkbits;
  	sector_t probe_block;
  	sector_t last_block;
53092a740   Hugh Dickins   [PATCH] swap: sho...
1407
1408
1409
  	sector_t lowest_block = -1;
  	sector_t highest_block = 0;
  	int nr_extents = 0;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1410
1411
1412
1413
1414
  	int ret;
  
  	inode = sis->swap_file->f_mapping->host;
  	if (S_ISBLK(inode->i_mode)) {
  		ret = add_swap_extent(sis, 0, sis->max, 0);
53092a740   Hugh Dickins   [PATCH] swap: sho...
1415
  		*span = sis->pages;
9625a5f28   Hugh Dickins   swap_info: includ...
1416
  		goto out;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1417
1418
1419
1420
1421
1422
1423
1424
1425
1426
1427
1428
1429
1430
1431
1432
1433
1434
1435
1436
1437
1438
1439
1440
1441
1442
1443
1444
1445
1446
1447
1448
1449
1450
1451
1452
1453
1454
1455
1456
1457
1458
  	}
  
  	blkbits = inode->i_blkbits;
  	blocks_per_page = PAGE_SIZE >> blkbits;
  
  	/*
  	 * Map all the blocks into the extent list.  This code doesn't try
  	 * to be very smart.
  	 */
  	probe_block = 0;
  	page_no = 0;
  	last_block = i_size_read(inode) >> blkbits;
  	while ((probe_block + blocks_per_page) <= last_block &&
  			page_no < sis->max) {
  		unsigned block_in_page;
  		sector_t first_block;
  
  		first_block = bmap(inode, probe_block);
  		if (first_block == 0)
  			goto bad_bmap;
  
  		/*
  		 * It must be PAGE_SIZE aligned on-disk
  		 */
  		if (first_block & (blocks_per_page - 1)) {
  			probe_block++;
  			goto reprobe;
  		}
  
  		for (block_in_page = 1; block_in_page < blocks_per_page;
  					block_in_page++) {
  			sector_t block;
  
  			block = bmap(inode, probe_block + block_in_page);
  			if (block == 0)
  				goto bad_bmap;
  			if (block != first_block + block_in_page) {
  				/* Discontiguity */
  				probe_block++;
  				goto reprobe;
  			}
  		}
53092a740   Hugh Dickins   [PATCH] swap: sho...
1459
1460
1461
1462
1463
1464
1465
  		first_block >>= (PAGE_SHIFT - blkbits);
  		if (page_no) {	/* exclude the header page */
  			if (first_block < lowest_block)
  				lowest_block = first_block;
  			if (first_block > highest_block)
  				highest_block = first_block;
  		}
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1466
1467
1468
  		/*
  		 * We found a PAGE_SIZE-length, PAGE_SIZE-aligned run of blocks
  		 */
53092a740   Hugh Dickins   [PATCH] swap: sho...
1469
1470
  		ret = add_swap_extent(sis, page_no, 1, first_block);
  		if (ret < 0)
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1471
  			goto out;
53092a740   Hugh Dickins   [PATCH] swap: sho...
1472
  		nr_extents += ret;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1473
1474
1475
1476
1477
  		page_no++;
  		probe_block += blocks_per_page;
  reprobe:
  		continue;
  	}
53092a740   Hugh Dickins   [PATCH] swap: sho...
1478
1479
  	ret = nr_extents;
  	*span = 1 + highest_block - lowest_block;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1480
  	if (page_no == 0)
e2244ec2e   Hugh Dickins   [PATCH] swap: cor...
1481
  		page_no = 1;	/* force Empty message */
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1482
  	sis->max = page_no;
e2244ec2e   Hugh Dickins   [PATCH] swap: cor...
1483
  	sis->pages = page_no - 1;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1484
  	sis->highest_bit = page_no - 1;
9625a5f28   Hugh Dickins   swap_info: includ...
1485
1486
  out:
  	return ret;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1487
1488
1489
1490
  bad_bmap:
  	printk(KERN_ERR "swapon: swapfile has holes
  ");
  	ret = -EINVAL;
9625a5f28   Hugh Dickins   swap_info: includ...
1491
  	goto out;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1492
  }
40531542e   Cesar Eduardo Barros   sys_swapon: separ...
1493
1494
1495
1496
1497
1498
1499
1500
1501
1502
1503
1504
1505
1506
1507
1508
1509
1510
1511
1512
1513
1514
1515
1516
1517
1518
1519
1520
1521
  static void enable_swap_info(struct swap_info_struct *p, int prio,
  				unsigned char *swap_map)
  {
  	int i, prev;
  
  	spin_lock(&swap_lock);
  	if (prio >= 0)
  		p->prio = prio;
  	else
  		p->prio = --least_priority;
  	p->swap_map = swap_map;
  	p->flags |= SWP_WRITEOK;
  	nr_swap_pages += p->pages;
  	total_swap_pages += p->pages;
  
  	/* insert swap space into swap_list: */
  	prev = -1;
  	for (i = swap_list.head; i >= 0; i = swap_info[i]->next) {
  		if (p->prio >= swap_info[i]->prio)
  			break;
  		prev = i;
  	}
  	p->next = i;
  	if (prev < 0)
  		swap_list.head = swap_list.next = p->type;
  	else
  		swap_info[prev]->next = p->type;
  	spin_unlock(&swap_lock);
  }
c4ea37c26   Heiko Carstens   [CVE-2009-0029] S...
1522
  SYSCALL_DEFINE1(swapoff, const char __user *, specialfile)
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1523
  {
73c34b6ac   Hugh Dickins   swap_info: miscel...
1524
  	struct swap_info_struct *p = NULL;
8d69aaee8   Hugh Dickins   swap_info: swap_m...
1525
  	unsigned char *swap_map;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1526
1527
1528
  	struct file *swap_file, *victim;
  	struct address_space *mapping;
  	struct inode *inode;
73c34b6ac   Hugh Dickins   swap_info: miscel...
1529
  	char *pathname;
72788c385   David Rientjes   oom: replace PF_O...
1530
  	int oom_score_adj;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1531
1532
  	int i, type, prev;
  	int err;
886bb7e9c   Hugh Dickins   swapfile: remove ...
1533

1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1534
1535
1536
1537
1538
1539
1540
1541
1542
1543
1544
1545
1546
1547
1548
1549
  	if (!capable(CAP_SYS_ADMIN))
  		return -EPERM;
  
  	pathname = getname(specialfile);
  	err = PTR_ERR(pathname);
  	if (IS_ERR(pathname))
  		goto out;
  
  	victim = filp_open(pathname, O_RDWR|O_LARGEFILE, 0);
  	putname(pathname);
  	err = PTR_ERR(victim);
  	if (IS_ERR(victim))
  		goto out;
  
  	mapping = victim->f_mapping;
  	prev = -1;
5d337b919   Hugh Dickins   [PATCH] swap: swa...
1550
  	spin_lock(&swap_lock);
efa90a981   Hugh Dickins   swap_info: change...
1551
1552
  	for (type = swap_list.head; type >= 0; type = swap_info[type]->next) {
  		p = swap_info[type];
22c6f8fdb   Hugh Dickins   swapfile: remove ...
1553
  		if (p->flags & SWP_WRITEOK) {
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1554
1555
1556
1557
1558
1559
1560
  			if (p->swap_file->f_mapping == mapping)
  				break;
  		}
  		prev = type;
  	}
  	if (type < 0) {
  		err = -EINVAL;
5d337b919   Hugh Dickins   [PATCH] swap: swa...
1561
  		spin_unlock(&swap_lock);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1562
1563
1564
1565
1566
1567
  		goto out_dput;
  	}
  	if (!security_vm_enough_memory(p->pages))
  		vm_unacct_memory(p->pages);
  	else {
  		err = -ENOMEM;
5d337b919   Hugh Dickins   [PATCH] swap: swa...
1568
  		spin_unlock(&swap_lock);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1569
1570
  		goto out_dput;
  	}
efa90a981   Hugh Dickins   swap_info: change...
1571
  	if (prev < 0)
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1572
  		swap_list.head = p->next;
efa90a981   Hugh Dickins   swap_info: change...
1573
1574
  	else
  		swap_info[prev]->next = p->next;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1575
1576
1577
1578
  	if (type == swap_list.next) {
  		/* just pick something that's safe... */
  		swap_list.next = swap_list.head;
  	}
78ecba081   Hugh Dickins   mm: fix ever-decr...
1579
  	if (p->prio < 0) {
efa90a981   Hugh Dickins   swap_info: change...
1580
1581
  		for (i = p->next; i >= 0; i = swap_info[i]->next)
  			swap_info[i]->prio = p->prio--;
78ecba081   Hugh Dickins   mm: fix ever-decr...
1582
1583
  		least_priority++;
  	}
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1584
1585
1586
  	nr_swap_pages -= p->pages;
  	total_swap_pages -= p->pages;
  	p->flags &= ~SWP_WRITEOK;
5d337b919   Hugh Dickins   [PATCH] swap: swa...
1587
  	spin_unlock(&swap_lock);
fb4f88dca   Hugh Dickins   [PATCH] swap: get...
1588

72788c385   David Rientjes   oom: replace PF_O...
1589
  	oom_score_adj = test_set_oom_score_adj(OOM_SCORE_ADJ_MAX);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1590
  	err = try_to_unuse(type);
43362a497   David Rientjes   oom: fix race whi...
1591
  	compare_swap_oom_score_adj(OOM_SCORE_ADJ_MAX, oom_score_adj);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1592

1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1593
  	if (err) {
40531542e   Cesar Eduardo Barros   sys_swapon: separ...
1594
1595
1596
1597
1598
1599
  		/*
  		 * reading p->prio and p->swap_map outside the lock is
  		 * safe here because only sys_swapon and sys_swapoff
  		 * change them, and there can be no other sys_swapon or
  		 * sys_swapoff for this swap_info_struct at this point.
  		 */
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1600
  		/* re-insert swap space back into swap_list */
40531542e   Cesar Eduardo Barros   sys_swapon: separ...
1601
  		enable_swap_info(p, p->prio, p->swap_map);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1602
1603
  		goto out_dput;
  	}
52b7efdbe   Hugh Dickins   [PATCH] swap: sca...
1604

5d337b919   Hugh Dickins   [PATCH] swap: swa...
1605
  	destroy_swap_extents(p);
570a335b8   Hugh Dickins   swap_info: swap c...
1606
1607
  	if (p->flags & SWP_CONTINUED)
  		free_swap_count_continuations(p);
fc0abb145   Ingo Molnar   [PATCH] sem2mutex...
1608
  	mutex_lock(&swapon_mutex);
5d337b919   Hugh Dickins   [PATCH] swap: swa...
1609
1610
  	spin_lock(&swap_lock);
  	drain_mmlist();
52b7efdbe   Hugh Dickins   [PATCH] swap: sca...
1611
  	/* wait for anyone still in scan_swap_map */
52b7efdbe   Hugh Dickins   [PATCH] swap: sca...
1612
1613
  	p->highest_bit = 0;		/* cuts scans short */
  	while (p->flags >= SWP_SCANNING) {
5d337b919   Hugh Dickins   [PATCH] swap: swa...
1614
  		spin_unlock(&swap_lock);
13e4b57f6   Nishanth Aravamudan   [PATCH] mm: fix-u...
1615
  		schedule_timeout_uninterruptible(1);
5d337b919   Hugh Dickins   [PATCH] swap: swa...
1616
  		spin_lock(&swap_lock);
52b7efdbe   Hugh Dickins   [PATCH] swap: sca...
1617
  	}
52b7efdbe   Hugh Dickins   [PATCH] swap: sca...
1618

1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1619
1620
1621
1622
1623
1624
  	swap_file = p->swap_file;
  	p->swap_file = NULL;
  	p->max = 0;
  	swap_map = p->swap_map;
  	p->swap_map = NULL;
  	p->flags = 0;
5d337b919   Hugh Dickins   [PATCH] swap: swa...
1625
  	spin_unlock(&swap_lock);
fc0abb145   Ingo Molnar   [PATCH] sem2mutex...
1626
  	mutex_unlock(&swapon_mutex);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1627
  	vfree(swap_map);
27a7faa07   KAMEZAWA Hiroyuki   memcg: swap cgrou...
1628
1629
  	/* Destroy swap account informatin */
  	swap_cgroup_swapoff(type);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1630
1631
1632
1633
  	inode = mapping->host;
  	if (S_ISBLK(inode->i_mode)) {
  		struct block_device *bdev = I_BDEV(inode);
  		set_blocksize(bdev, p->old_block_size);
e525fd89d   Tejun Heo   block: make blkde...
1634
  		blkdev_put(bdev, FMODE_READ | FMODE_WRITE | FMODE_EXCL);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1635
  	} else {
1b1dcc1b5   Jes Sorensen   [PATCH] mutex sub...
1636
  		mutex_lock(&inode->i_mutex);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1637
  		inode->i_flags &= ~S_SWAPFILE;
1b1dcc1b5   Jes Sorensen   [PATCH] mutex sub...
1638
  		mutex_unlock(&inode->i_mutex);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1639
1640
1641
  	}
  	filp_close(swap_file, NULL);
  	err = 0;
66d7dd518   Kay Sievers   /proc/swaps: supp...
1642
1643
  	atomic_inc(&proc_poll_event);
  	wake_up_interruptible(&proc_poll_wait);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1644
1645
1646
1647
1648
1649
1650
1651
  
  out_dput:
  	filp_close(victim, NULL);
  out:
  	return err;
  }
  
  #ifdef CONFIG_PROC_FS
66d7dd518   Kay Sievers   /proc/swaps: supp...
1652
1653
  static unsigned swaps_poll(struct file *file, poll_table *wait)
  {
f15146380   Kay Sievers   fs: seq_file - ad...
1654
  	struct seq_file *seq = file->private_data;
66d7dd518   Kay Sievers   /proc/swaps: supp...
1655
1656
  
  	poll_wait(file, &proc_poll_wait, wait);
f15146380   Kay Sievers   fs: seq_file - ad...
1657
1658
  	if (seq->poll_event != atomic_read(&proc_poll_event)) {
  		seq->poll_event = atomic_read(&proc_poll_event);
66d7dd518   Kay Sievers   /proc/swaps: supp...
1659
1660
1661
1662
1663
  		return POLLIN | POLLRDNORM | POLLERR | POLLPRI;
  	}
  
  	return POLLIN | POLLRDNORM;
  }
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1664
1665
1666
  /* iterator */
  static void *swap_start(struct seq_file *swap, loff_t *pos)
  {
efa90a981   Hugh Dickins   swap_info: change...
1667
1668
  	struct swap_info_struct *si;
  	int type;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1669
  	loff_t l = *pos;
fc0abb145   Ingo Molnar   [PATCH] sem2mutex...
1670
  	mutex_lock(&swapon_mutex);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1671

881e4aabe   Suleiman Souhlal   [PATCH] Always pr...
1672
1673
  	if (!l)
  		return SEQ_START_TOKEN;
efa90a981   Hugh Dickins   swap_info: change...
1674
1675
1676
1677
  	for (type = 0; type < nr_swapfiles; type++) {
  		smp_rmb();	/* read nr_swapfiles before swap_info[type] */
  		si = swap_info[type];
  		if (!(si->flags & SWP_USED) || !si->swap_map)
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1678
  			continue;
881e4aabe   Suleiman Souhlal   [PATCH] Always pr...
1679
  		if (!--l)
efa90a981   Hugh Dickins   swap_info: change...
1680
  			return si;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1681
1682
1683
1684
1685
1686
1687
  	}
  
  	return NULL;
  }
  
  static void *swap_next(struct seq_file *swap, void *v, loff_t *pos)
  {
efa90a981   Hugh Dickins   swap_info: change...
1688
1689
  	struct swap_info_struct *si = v;
  	int type;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1690

881e4aabe   Suleiman Souhlal   [PATCH] Always pr...
1691
  	if (v == SEQ_START_TOKEN)
efa90a981   Hugh Dickins   swap_info: change...
1692
1693
1694
  		type = 0;
  	else
  		type = si->type + 1;
881e4aabe   Suleiman Souhlal   [PATCH] Always pr...
1695

efa90a981   Hugh Dickins   swap_info: change...
1696
1697
1698
1699
  	for (; type < nr_swapfiles; type++) {
  		smp_rmb();	/* read nr_swapfiles before swap_info[type] */
  		si = swap_info[type];
  		if (!(si->flags & SWP_USED) || !si->swap_map)
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1700
1701
  			continue;
  		++*pos;
efa90a981   Hugh Dickins   swap_info: change...
1702
  		return si;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1703
1704
1705
1706
1707
1708
1709
  	}
  
  	return NULL;
  }
  
  static void swap_stop(struct seq_file *swap, void *v)
  {
fc0abb145   Ingo Molnar   [PATCH] sem2mutex...
1710
  	mutex_unlock(&swapon_mutex);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1711
1712
1713
1714
  }
  
  static int swap_show(struct seq_file *swap, void *v)
  {
efa90a981   Hugh Dickins   swap_info: change...
1715
  	struct swap_info_struct *si = v;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1716
1717
  	struct file *file;
  	int len;
efa90a981   Hugh Dickins   swap_info: change...
1718
  	if (si == SEQ_START_TOKEN) {
881e4aabe   Suleiman Souhlal   [PATCH] Always pr...
1719
1720
1721
1722
  		seq_puts(swap,"Filename\t\t\t\tType\t\tSize\tUsed\tPriority
  ");
  		return 0;
  	}
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1723

efa90a981   Hugh Dickins   swap_info: change...
1724
  	file = si->swap_file;
c32c2f63a   Jan Blunck   d_path: Make seq_...
1725
1726
  	len = seq_path(swap, &file->f_path, " \t
  \\");
6eb396dc4   Hugh Dickins   [PATCH] swap: swa...
1727
1728
  	seq_printf(swap, "%*s%s\t%u\t%u\t%d
  ",
886bb7e9c   Hugh Dickins   swapfile: remove ...
1729
1730
  			len < 40 ? 40 - len : 1, " ",
  			S_ISBLK(file->f_path.dentry->d_inode->i_mode) ?
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1731
  				"partition" : "file\t",
efa90a981   Hugh Dickins   swap_info: change...
1732
1733
1734
  			si->pages << (PAGE_SHIFT - 10),
  			si->inuse_pages << (PAGE_SHIFT - 10),
  			si->prio);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1735
1736
  	return 0;
  }
15ad7cdcf   Helge Deller   [PATCH] struct se...
1737
  static const struct seq_operations swaps_op = {
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1738
1739
1740
1741
1742
1743
1744
1745
  	.start =	swap_start,
  	.next =		swap_next,
  	.stop =		swap_stop,
  	.show =		swap_show
  };
  
  static int swaps_open(struct inode *inode, struct file *file)
  {
f15146380   Kay Sievers   fs: seq_file - ad...
1746
  	struct seq_file *seq;
66d7dd518   Kay Sievers   /proc/swaps: supp...
1747
  	int ret;
66d7dd518   Kay Sievers   /proc/swaps: supp...
1748
  	ret = seq_open(file, &swaps_op);
f15146380   Kay Sievers   fs: seq_file - ad...
1749
  	if (ret)
66d7dd518   Kay Sievers   /proc/swaps: supp...
1750
  		return ret;
66d7dd518   Kay Sievers   /proc/swaps: supp...
1751

f15146380   Kay Sievers   fs: seq_file - ad...
1752
1753
1754
  	seq = file->private_data;
  	seq->poll_event = atomic_read(&proc_poll_event);
  	return 0;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1755
  }
15ad7cdcf   Helge Deller   [PATCH] struct se...
1756
  static const struct file_operations proc_swaps_operations = {
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1757
1758
1759
1760
  	.open		= swaps_open,
  	.read		= seq_read,
  	.llseek		= seq_lseek,
  	.release	= seq_release,
66d7dd518   Kay Sievers   /proc/swaps: supp...
1761
  	.poll		= swaps_poll,
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1762
1763
1764
1765
  };
  
  static int __init procswaps_init(void)
  {
3d71f86f4   Denis V. Lunev   mm: use non-racy ...
1766
  	proc_create("swaps", 0, NULL, &proc_swaps_operations);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1767
1768
1769
1770
  	return 0;
  }
  __initcall(procswaps_init);
  #endif /* CONFIG_PROC_FS */
1796316a8   Jan Beulich   x86: consolidate ...
1771
1772
1773
1774
1775
1776
1777
1778
  #ifdef MAX_SWAPFILES_CHECK
  static int __init max_swapfiles_check(void)
  {
  	MAX_SWAPFILES_CHECK();
  	return 0;
  }
  late_initcall(max_swapfiles_check);
  #endif
53cbb2435   Cesar Eduardo Barros   sys_swapon: separ...
1779
  static struct swap_info_struct *alloc_swap_info(void)
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1780
  {
73c34b6ac   Hugh Dickins   swap_info: miscel...
1781
  	struct swap_info_struct *p;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1782
  	unsigned int type;
efa90a981   Hugh Dickins   swap_info: change...
1783
1784
1785
  
  	p = kzalloc(sizeof(*p), GFP_KERNEL);
  	if (!p)
53cbb2435   Cesar Eduardo Barros   sys_swapon: separ...
1786
  		return ERR_PTR(-ENOMEM);
efa90a981   Hugh Dickins   swap_info: change...
1787

5d337b919   Hugh Dickins   [PATCH] swap: swa...
1788
  	spin_lock(&swap_lock);
efa90a981   Hugh Dickins   swap_info: change...
1789
1790
  	for (type = 0; type < nr_swapfiles; type++) {
  		if (!(swap_info[type]->flags & SWP_USED))
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1791
  			break;
efa90a981   Hugh Dickins   swap_info: change...
1792
  	}
0697212a4   Christoph Lameter   [PATCH] Swapless ...
1793
  	if (type >= MAX_SWAPFILES) {
5d337b919   Hugh Dickins   [PATCH] swap: swa...
1794
  		spin_unlock(&swap_lock);
efa90a981   Hugh Dickins   swap_info: change...
1795
  		kfree(p);
730c0581c   Cesar Eduardo Barros   sys_swapon: simpl...
1796
  		return ERR_PTR(-EPERM);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1797
  	}
efa90a981   Hugh Dickins   swap_info: change...
1798
1799
1800
1801
1802
1803
1804
1805
1806
1807
1808
1809
1810
1811
1812
1813
1814
1815
  	if (type >= nr_swapfiles) {
  		p->type = type;
  		swap_info[type] = p;
  		/*
  		 * Write swap_info[type] before nr_swapfiles, in case a
  		 * racing procfs swap_start() or swap_next() is reading them.
  		 * (We never shrink nr_swapfiles, we never free this entry.)
  		 */
  		smp_wmb();
  		nr_swapfiles++;
  	} else {
  		kfree(p);
  		p = swap_info[type];
  		/*
  		 * Do not memset this entry: a racing procfs swap_next()
  		 * would be relying on p->type to remain valid.
  		 */
  	}
9625a5f28   Hugh Dickins   swap_info: includ...
1816
  	INIT_LIST_HEAD(&p->first_swap_extent.list);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1817
  	p->flags = SWP_USED;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1818
  	p->next = -1;
5d337b919   Hugh Dickins   [PATCH] swap: swa...
1819
  	spin_unlock(&swap_lock);
efa90a981   Hugh Dickins   swap_info: change...
1820

53cbb2435   Cesar Eduardo Barros   sys_swapon: separ...
1821
  	return p;
53cbb2435   Cesar Eduardo Barros   sys_swapon: separ...
1822
  }
4d0e1e107   Cesar Eduardo Barros   sys_swapon: separ...
1823
1824
1825
1826
1827
1828
1829
1830
1831
1832
1833
  static int claim_swapfile(struct swap_info_struct *p, struct inode *inode)
  {
  	int error;
  
  	if (S_ISBLK(inode->i_mode)) {
  		p->bdev = bdgrab(I_BDEV(inode));
  		error = blkdev_get(p->bdev,
  				   FMODE_READ | FMODE_WRITE | FMODE_EXCL,
  				   sys_swapon);
  		if (error < 0) {
  			p->bdev = NULL;
87ade72a7   Cesar Eduardo Barros   sys_swapon: simpl...
1834
  			return -EINVAL;
4d0e1e107   Cesar Eduardo Barros   sys_swapon: separ...
1835
1836
1837
1838
  		}
  		p->old_block_size = block_size(p->bdev);
  		error = set_blocksize(p->bdev, PAGE_SIZE);
  		if (error < 0)
87ade72a7   Cesar Eduardo Barros   sys_swapon: simpl...
1839
  			return error;
4d0e1e107   Cesar Eduardo Barros   sys_swapon: separ...
1840
1841
1842
1843
  		p->flags |= SWP_BLKDEV;
  	} else if (S_ISREG(inode->i_mode)) {
  		p->bdev = inode->i_sb->s_bdev;
  		mutex_lock(&inode->i_mutex);
87ade72a7   Cesar Eduardo Barros   sys_swapon: simpl...
1844
1845
1846
1847
  		if (IS_SWAPFILE(inode))
  			return -EBUSY;
  	} else
  		return -EINVAL;
4d0e1e107   Cesar Eduardo Barros   sys_swapon: separ...
1848
1849
  
  	return 0;
4d0e1e107   Cesar Eduardo Barros   sys_swapon: separ...
1850
  }
ca8bd38bf   Cesar Eduardo Barros   sys_swapon: separ...
1851
1852
1853
1854
1855
1856
1857
1858
1859
1860
1861
  static unsigned long read_swap_header(struct swap_info_struct *p,
  					union swap_header *swap_header,
  					struct inode *inode)
  {
  	int i;
  	unsigned long maxpages;
  	unsigned long swapfilepages;
  
  	if (memcmp("SWAPSPACE2", swap_header->magic.magic, 10)) {
  		printk(KERN_ERR "Unable to find swap-space signature
  ");
387190253   Cesar Eduardo Barros   sys_swapon: simpl...
1862
  		return 0;
ca8bd38bf   Cesar Eduardo Barros   sys_swapon: separ...
1863
1864
1865
1866
1867
1868
1869
1870
1871
1872
1873
1874
1875
1876
1877
1878
  	}
  
  	/* swap partition endianess hack... */
  	if (swab32(swap_header->info.version) == 1) {
  		swab32s(&swap_header->info.version);
  		swab32s(&swap_header->info.last_page);
  		swab32s(&swap_header->info.nr_badpages);
  		for (i = 0; i < swap_header->info.nr_badpages; i++)
  			swab32s(&swap_header->info.badpages[i]);
  	}
  	/* Check the swap header's sub-version */
  	if (swap_header->info.version != 1) {
  		printk(KERN_WARNING
  		       "Unable to handle swap header version %d
  ",
  		       swap_header->info.version);
387190253   Cesar Eduardo Barros   sys_swapon: simpl...
1879
  		return 0;
ca8bd38bf   Cesar Eduardo Barros   sys_swapon: separ...
1880
1881
1882
1883
1884
1885
1886
1887
  	}
  
  	p->lowest_bit  = 1;
  	p->cluster_next = 1;
  	p->cluster_nr = 0;
  
  	/*
  	 * Find out how many pages are allowed for a single swap
a2c16d6cb   Hugh Dickins   mm: let swap use ...
1888
1889
1890
1891
1892
1893
  	 * device. There are three limiting factors: 1) the number
  	 * of bits for the swap offset in the swp_entry_t type, and
  	 * 2) the number of bits in the swap pte as defined by the
  	 * the different architectures, and 3) the number of free bits
  	 * in an exceptional radix_tree entry. In order to find the
  	 * largest possible bit mask, a swap entry with swap type 0
ca8bd38bf   Cesar Eduardo Barros   sys_swapon: separ...
1894
  	 * and swap offset ~0UL is created, encoded to a swap pte,
a2c16d6cb   Hugh Dickins   mm: let swap use ...
1895
  	 * decoded to a swp_entry_t again, and finally the swap
ca8bd38bf   Cesar Eduardo Barros   sys_swapon: separ...
1896
1897
1898
  	 * offset is extracted. This will mask all the bits from
  	 * the initial ~0UL mask that can't be encoded in either
  	 * the swp_entry_t or the architecture definition of a
a2c16d6cb   Hugh Dickins   mm: let swap use ...
1899
  	 * swap pte.  Then the same is done for a radix_tree entry.
ca8bd38bf   Cesar Eduardo Barros   sys_swapon: separ...
1900
1901
  	 */
  	maxpages = swp_offset(pte_to_swp_entry(
a2c16d6cb   Hugh Dickins   mm: let swap use ...
1902
1903
1904
  			swp_entry_to_pte(swp_entry(0, ~0UL))));
  	maxpages = swp_offset(radix_to_swp_entry(
  			swp_to_radix_entry(swp_entry(0, maxpages)))) + 1;
ca8bd38bf   Cesar Eduardo Barros   sys_swapon: separ...
1905
1906
1907
1908
1909
1910
1911
1912
1913
  	if (maxpages > swap_header->info.last_page) {
  		maxpages = swap_header->info.last_page + 1;
  		/* p->max is an unsigned int: don't overflow it */
  		if ((unsigned int)maxpages == 0)
  			maxpages = UINT_MAX;
  	}
  	p->highest_bit = maxpages - 1;
  
  	if (!maxpages)
387190253   Cesar Eduardo Barros   sys_swapon: simpl...
1914
  		return 0;
ca8bd38bf   Cesar Eduardo Barros   sys_swapon: separ...
1915
1916
1917
1918
1919
  	swapfilepages = i_size_read(inode) >> PAGE_SHIFT;
  	if (swapfilepages && maxpages > swapfilepages) {
  		printk(KERN_WARNING
  		       "Swap area shorter than signature indicates
  ");
387190253   Cesar Eduardo Barros   sys_swapon: simpl...
1920
  		return 0;
ca8bd38bf   Cesar Eduardo Barros   sys_swapon: separ...
1921
1922
  	}
  	if (swap_header->info.nr_badpages && S_ISREG(inode->i_mode))
387190253   Cesar Eduardo Barros   sys_swapon: simpl...
1923
  		return 0;
ca8bd38bf   Cesar Eduardo Barros   sys_swapon: separ...
1924
  	if (swap_header->info.nr_badpages > MAX_SWAP_BADPAGES)
387190253   Cesar Eduardo Barros   sys_swapon: simpl...
1925
  		return 0;
ca8bd38bf   Cesar Eduardo Barros   sys_swapon: separ...
1926
1927
  
  	return maxpages;
ca8bd38bf   Cesar Eduardo Barros   sys_swapon: separ...
1928
  }
915d4d7bc   Cesar Eduardo Barros   sys_swapon: separ...
1929
1930
1931
1932
1933
1934
1935
  static int setup_swap_map_and_extents(struct swap_info_struct *p,
  					union swap_header *swap_header,
  					unsigned char *swap_map,
  					unsigned long maxpages,
  					sector_t *span)
  {
  	int i;
915d4d7bc   Cesar Eduardo Barros   sys_swapon: separ...
1936
1937
1938
1939
1940
1941
1942
  	unsigned int nr_good_pages;
  	int nr_extents;
  
  	nr_good_pages = maxpages - 1;	/* omit header page */
  
  	for (i = 0; i < swap_header->info.nr_badpages; i++) {
  		unsigned int page_nr = swap_header->info.badpages[i];
bdb8e3f68   Cesar Eduardo Barros   sys_swapon: simpl...
1943
1944
  		if (page_nr == 0 || page_nr > swap_header->info.last_page)
  			return -EINVAL;
915d4d7bc   Cesar Eduardo Barros   sys_swapon: separ...
1945
1946
1947
1948
1949
1950
1951
1952
1953
1954
1955
  		if (page_nr < maxpages) {
  			swap_map[page_nr] = SWAP_MAP_BAD;
  			nr_good_pages--;
  		}
  	}
  
  	if (nr_good_pages) {
  		swap_map[0] = SWAP_MAP_BAD;
  		p->max = maxpages;
  		p->pages = nr_good_pages;
  		nr_extents = setup_swap_extents(p, span);
bdb8e3f68   Cesar Eduardo Barros   sys_swapon: simpl...
1956
1957
  		if (nr_extents < 0)
  			return nr_extents;
915d4d7bc   Cesar Eduardo Barros   sys_swapon: separ...
1958
1959
1960
1961
1962
  		nr_good_pages = p->pages;
  	}
  	if (!nr_good_pages) {
  		printk(KERN_WARNING "Empty swap-file
  ");
bdb8e3f68   Cesar Eduardo Barros   sys_swapon: simpl...
1963
  		return -EINVAL;
915d4d7bc   Cesar Eduardo Barros   sys_swapon: separ...
1964
1965
1966
  	}
  
  	return nr_extents;
915d4d7bc   Cesar Eduardo Barros   sys_swapon: separ...
1967
  }
53cbb2435   Cesar Eduardo Barros   sys_swapon: separ...
1968
1969
1970
  SYSCALL_DEFINE2(swapon, const char __user *, specialfile, int, swap_flags)
  {
  	struct swap_info_struct *p;
28b36bd74   Cesar Eduardo Barros   sys_swapon: remov...
1971
  	char *name;
53cbb2435   Cesar Eduardo Barros   sys_swapon: separ...
1972
1973
  	struct file *swap_file = NULL;
  	struct address_space *mapping;
40531542e   Cesar Eduardo Barros   sys_swapon: separ...
1974
1975
  	int i;
  	int prio;
53cbb2435   Cesar Eduardo Barros   sys_swapon: separ...
1976
1977
  	int error;
  	union swap_header *swap_header;
915d4d7bc   Cesar Eduardo Barros   sys_swapon: separ...
1978
  	int nr_extents;
53cbb2435   Cesar Eduardo Barros   sys_swapon: separ...
1979
1980
  	sector_t span;
  	unsigned long maxpages;
53cbb2435   Cesar Eduardo Barros   sys_swapon: separ...
1981
1982
1983
  	unsigned char *swap_map = NULL;
  	struct page *page = NULL;
  	struct inode *inode = NULL;
53cbb2435   Cesar Eduardo Barros   sys_swapon: separ...
1984
1985
1986
1987
1988
  
  	if (!capable(CAP_SYS_ADMIN))
  		return -EPERM;
  
  	p = alloc_swap_info();
2542e5134   Cesar Eduardo Barros   sys_swapon: simpl...
1989
1990
  	if (IS_ERR(p))
  		return PTR_ERR(p);
53cbb2435   Cesar Eduardo Barros   sys_swapon: separ...
1991

1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1992
  	name = getname(specialfile);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1993
  	if (IS_ERR(name)) {
7de7fb6b3   Cesar Eduardo Barros   sys_swapon: move ...
1994
  		error = PTR_ERR(name);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1995
  		name = NULL;
bd69010b0   Cesar Eduardo Barros   sys_swapon: use a...
1996
  		goto bad_swap;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1997
1998
  	}
  	swap_file = filp_open(name, O_RDWR|O_LARGEFILE, 0);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
1999
  	if (IS_ERR(swap_file)) {
7de7fb6b3   Cesar Eduardo Barros   sys_swapon: move ...
2000
  		error = PTR_ERR(swap_file);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2001
  		swap_file = NULL;
bd69010b0   Cesar Eduardo Barros   sys_swapon: use a...
2002
  		goto bad_swap;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2003
2004
2005
2006
  	}
  
  	p->swap_file = swap_file;
  	mapping = swap_file->f_mapping;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2007

1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2008
  	for (i = 0; i < nr_swapfiles; i++) {
efa90a981   Hugh Dickins   swap_info: change...
2009
  		struct swap_info_struct *q = swap_info[i];
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2010

e8e6c2ec4   Cesar Eduardo Barros   sys_swapon: do no...
2011
  		if (q == p || !q->swap_file)
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2012
  			continue;
7de7fb6b3   Cesar Eduardo Barros   sys_swapon: move ...
2013
2014
  		if (mapping == q->swap_file->f_mapping) {
  			error = -EBUSY;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2015
  			goto bad_swap;
7de7fb6b3   Cesar Eduardo Barros   sys_swapon: move ...
2016
  		}
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2017
  	}
2130781e2   Cesar Eduardo Barros   sys_swapon: fix i...
2018
2019
  	inode = mapping->host;
  	/* If S_ISREG(inode->i_mode) will do mutex_lock(&inode->i_mutex); */
4d0e1e107   Cesar Eduardo Barros   sys_swapon: separ...
2020
2021
  	error = claim_swapfile(p, inode);
  	if (unlikely(error))
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2022
  		goto bad_swap;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2023

1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2024
2025
2026
2027
2028
2029
2030
  	/*
  	 * Read the swap header.
  	 */
  	if (!mapping->a_ops->readpage) {
  		error = -EINVAL;
  		goto bad_swap;
  	}
090d2b185   Pekka Enberg   [PATCH] read_mapp...
2031
  	page = read_mapping_page(mapping, 0, swap_file);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2032
2033
2034
2035
  	if (IS_ERR(page)) {
  		error = PTR_ERR(page);
  		goto bad_swap;
  	}
81e339712   Hugh Dickins   swapfile: remove ...
2036
  	swap_header = kmap(page);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2037

ca8bd38bf   Cesar Eduardo Barros   sys_swapon: separ...
2038
2039
  	maxpages = read_swap_header(p, swap_header, inode);
  	if (unlikely(!maxpages)) {
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2040
2041
2042
  		error = -EINVAL;
  		goto bad_swap;
  	}
886bb7e9c   Hugh Dickins   swapfile: remove ...
2043

81e339712   Hugh Dickins   swapfile: remove ...
2044
  	/* OK, set up the swap map and apply the bad block list */
803d0c835   Cesar Eduardo Barros   sys_swapon: use v...
2045
  	swap_map = vzalloc(maxpages);
81e339712   Hugh Dickins   swapfile: remove ...
2046
2047
2048
2049
  	if (!swap_map) {
  		error = -ENOMEM;
  		goto bad_swap;
  	}
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2050

1421ef3cd   Cesar Eduardo Barros   sys_swapon: call ...
2051
2052
2053
  	error = swap_cgroup_swapon(p->type, maxpages);
  	if (error)
  		goto bad_swap;
915d4d7bc   Cesar Eduardo Barros   sys_swapon: separ...
2054
2055
2056
2057
  	nr_extents = setup_swap_map_and_extents(p, swap_header, swap_map,
  		maxpages, &span);
  	if (unlikely(nr_extents < 0)) {
  		error = nr_extents;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2058
2059
  		goto bad_swap;
  	}
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2060

3bd0f0c76   Suresh Jayaraman   swapfile: avoid N...
2061
2062
2063
2064
2065
  	if (p->bdev) {
  		if (blk_queue_nonrot(bdev_get_queue(p->bdev))) {
  			p->flags |= SWP_SOLIDSTATE;
  			p->cluster_next = 1 + (random32() % p->highest_bit);
  		}
339944663   Hugh Dickins   swap: discard whi...
2066
  		if (discard_swap(p) == 0 && (swap_flags & SWAP_FLAG_DISCARD))
3bd0f0c76   Suresh Jayaraman   swapfile: avoid N...
2067
  			p->flags |= SWP_DISCARDABLE;
20137a490   Hugh Dickins   swapfile: swapon ...
2068
  	}
6a6ba8317   Hugh Dickins   swapfile: swapon ...
2069

fc0abb145   Ingo Molnar   [PATCH] sem2mutex...
2070
  	mutex_lock(&swapon_mutex);
40531542e   Cesar Eduardo Barros   sys_swapon: separ...
2071
  	prio = -1;
78ecba081   Hugh Dickins   mm: fix ever-decr...
2072
  	if (swap_flags & SWAP_FLAG_PREFER)
40531542e   Cesar Eduardo Barros   sys_swapon: separ...
2073
  		prio =
78ecba081   Hugh Dickins   mm: fix ever-decr...
2074
  		  (swap_flags & SWAP_FLAG_PRIO_MASK) >> SWAP_FLAG_PRIO_SHIFT;
40531542e   Cesar Eduardo Barros   sys_swapon: separ...
2075
  	enable_swap_info(p, prio, swap_map);
c69dbfb84   Cesar Eduardo Barros   sys_swapon: move ...
2076
2077
2078
2079
2080
2081
2082
2083
  
  	printk(KERN_INFO "Adding %uk swap on %s.  "
  			"Priority:%d extents:%d across:%lluk %s%s
  ",
  		p->pages<<(PAGE_SHIFT-10), name, p->prio,
  		nr_extents, (unsigned long long)span<<(PAGE_SHIFT-10),
  		(p->flags & SWP_SOLIDSTATE) ? "SS" : "",
  		(p->flags & SWP_DISCARDABLE) ? "D" : "");
fc0abb145   Ingo Molnar   [PATCH] sem2mutex...
2084
  	mutex_unlock(&swapon_mutex);
66d7dd518   Kay Sievers   /proc/swaps: supp...
2085
2086
  	atomic_inc(&proc_poll_event);
  	wake_up_interruptible(&proc_poll_wait);
9b01c350a   Cesar Eduardo Barros   sys_swapon: do on...
2087
2088
  	if (S_ISREG(inode->i_mode))
  		inode->i_flags |= S_SWAPFILE;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2089
2090
2091
  	error = 0;
  	goto out;
  bad_swap:
bd69010b0   Cesar Eduardo Barros   sys_swapon: use a...
2092
  	if (inode && S_ISBLK(inode->i_mode) && p->bdev) {
f2090d2df   Cesar Eduardo Barros   sys_swapon: remov...
2093
2094
  		set_blocksize(p->bdev, p->old_block_size);
  		blkdev_put(p->bdev, FMODE_READ | FMODE_WRITE | FMODE_EXCL);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2095
  	}
4cd3bb10f   Hugh Dickins   [PATCH] swap: mov...
2096
  	destroy_swap_extents(p);
e8e6c2ec4   Cesar Eduardo Barros   sys_swapon: do no...
2097
  	swap_cgroup_swapoff(p->type);
5d337b919   Hugh Dickins   [PATCH] swap: swa...
2098
  	spin_lock(&swap_lock);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2099
  	p->swap_file = NULL;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2100
  	p->flags = 0;
5d337b919   Hugh Dickins   [PATCH] swap: swa...
2101
  	spin_unlock(&swap_lock);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2102
  	vfree(swap_map);
52c50567d   Mel Gorman   mm: swap: unlock ...
2103
  	if (swap_file) {
2130781e2   Cesar Eduardo Barros   sys_swapon: fix i...
2104
  		if (inode && S_ISREG(inode->i_mode)) {
52c50567d   Mel Gorman   mm: swap: unlock ...
2105
  			mutex_unlock(&inode->i_mutex);
2130781e2   Cesar Eduardo Barros   sys_swapon: fix i...
2106
2107
  			inode = NULL;
  		}
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2108
  		filp_close(swap_file, NULL);
52c50567d   Mel Gorman   mm: swap: unlock ...
2109
  	}
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2110
2111
2112
2113
2114
2115
2116
  out:
  	if (page && !IS_ERR(page)) {
  		kunmap(page);
  		page_cache_release(page);
  	}
  	if (name)
  		putname(name);
9b01c350a   Cesar Eduardo Barros   sys_swapon: do on...
2117
  	if (inode && S_ISREG(inode->i_mode))
1b1dcc1b5   Jes Sorensen   [PATCH] mutex sub...
2118
  		mutex_unlock(&inode->i_mutex);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2119
2120
2121
2122
2123
  	return error;
  }
  
  void si_swapinfo(struct sysinfo *val)
  {
efa90a981   Hugh Dickins   swap_info: change...
2124
  	unsigned int type;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2125
  	unsigned long nr_to_be_unused = 0;
5d337b919   Hugh Dickins   [PATCH] swap: swa...
2126
  	spin_lock(&swap_lock);
efa90a981   Hugh Dickins   swap_info: change...
2127
2128
2129
2130
2131
  	for (type = 0; type < nr_swapfiles; type++) {
  		struct swap_info_struct *si = swap_info[type];
  
  		if ((si->flags & SWP_USED) && !(si->flags & SWP_WRITEOK))
  			nr_to_be_unused += si->inuse_pages;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2132
2133
2134
  	}
  	val->freeswap = nr_swap_pages + nr_to_be_unused;
  	val->totalswap = total_swap_pages + nr_to_be_unused;
5d337b919   Hugh Dickins   [PATCH] swap: swa...
2135
  	spin_unlock(&swap_lock);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2136
2137
2138
2139
2140
  }
  
  /*
   * Verify that a swap entry is valid and increment its swap map count.
   *
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
2141
2142
2143
2144
2145
2146
   * Returns error code in following case.
   * - success -> 0
   * - swp_entry is invalid -> EINVAL
   * - swp_entry is migration entry -> EINVAL
   * - swap-cache reference is requested but there is already one. -> EEXIST
   * - swap-cache reference is requested but the entry is not used. -> ENOENT
570a335b8   Hugh Dickins   swap_info: swap c...
2147
   * - swap-mapped reference requested but needs continued swap count. -> ENOMEM
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2148
   */
8d69aaee8   Hugh Dickins   swap_info: swap_m...
2149
  static int __swap_duplicate(swp_entry_t entry, unsigned char usage)
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2150
  {
73c34b6ac   Hugh Dickins   swap_info: miscel...
2151
  	struct swap_info_struct *p;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2152
  	unsigned long offset, type;
8d69aaee8   Hugh Dickins   swap_info: swap_m...
2153
2154
  	unsigned char count;
  	unsigned char has_cache;
253d553ba   Hugh Dickins   swap_info: SWAP_H...
2155
  	int err = -EINVAL;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2156

a7420aa54   Andi Kleen   HWPOISON: Add sup...
2157
  	if (non_swap_entry(entry))
253d553ba   Hugh Dickins   swap_info: SWAP_H...
2158
  		goto out;
0697212a4   Christoph Lameter   [PATCH] Swapless ...
2159

1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2160
2161
2162
  	type = swp_type(entry);
  	if (type >= nr_swapfiles)
  		goto bad_file;
efa90a981   Hugh Dickins   swap_info: change...
2163
  	p = swap_info[type];
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2164
  	offset = swp_offset(entry);
5d337b919   Hugh Dickins   [PATCH] swap: swa...
2165
  	spin_lock(&swap_lock);
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
2166
2167
  	if (unlikely(offset >= p->max))
  		goto unlock_out;
253d553ba   Hugh Dickins   swap_info: SWAP_H...
2168
2169
2170
2171
  	count = p->swap_map[offset];
  	has_cache = count & SWAP_HAS_CACHE;
  	count &= ~SWAP_HAS_CACHE;
  	err = 0;
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
2172

253d553ba   Hugh Dickins   swap_info: SWAP_H...
2173
  	if (usage == SWAP_HAS_CACHE) {
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
2174
2175
  
  		/* set SWAP_HAS_CACHE if there is no cache and entry is used */
253d553ba   Hugh Dickins   swap_info: SWAP_H...
2176
2177
2178
2179
2180
2181
  		if (!has_cache && count)
  			has_cache = SWAP_HAS_CACHE;
  		else if (has_cache)		/* someone else added cache */
  			err = -EEXIST;
  		else				/* no users remaining */
  			err = -ENOENT;
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
2182
2183
  
  	} else if (count || has_cache) {
253d553ba   Hugh Dickins   swap_info: SWAP_H...
2184

570a335b8   Hugh Dickins   swap_info: swap c...
2185
2186
2187
  		if ((count & ~COUNT_CONTINUED) < SWAP_MAP_MAX)
  			count += usage;
  		else if ((count & ~COUNT_CONTINUED) > SWAP_MAP_MAX)
253d553ba   Hugh Dickins   swap_info: SWAP_H...
2188
  			err = -EINVAL;
570a335b8   Hugh Dickins   swap_info: swap c...
2189
2190
2191
2192
  		else if (swap_count_continued(p, offset, count))
  			count = COUNT_CONTINUED;
  		else
  			err = -ENOMEM;
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
2193
  	} else
253d553ba   Hugh Dickins   swap_info: SWAP_H...
2194
2195
2196
  		err = -ENOENT;			/* unused swap entry */
  
  	p->swap_map[offset] = count | has_cache;
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
2197
  unlock_out:
5d337b919   Hugh Dickins   [PATCH] swap: swa...
2198
  	spin_unlock(&swap_lock);
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2199
  out:
253d553ba   Hugh Dickins   swap_info: SWAP_H...
2200
  	return err;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2201
2202
2203
2204
2205
2206
  
  bad_file:
  	printk(KERN_ERR "swap_dup: %s%08lx
  ", Bad_file, entry.val);
  	goto out;
  }
253d553ba   Hugh Dickins   swap_info: SWAP_H...
2207

355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
2208
  /*
aaa468653   Hugh Dickins   swap_info: note S...
2209
2210
2211
2212
2213
2214
2215
2216
2217
   * Help swapoff by noting that swap entry belongs to shmem/tmpfs
   * (in which case its reference count is never incremented).
   */
  void swap_shmem_alloc(swp_entry_t entry)
  {
  	__swap_duplicate(entry, SWAP_MAP_SHMEM);
  }
  
  /*
08259d58e   Hugh Dickins   mm: add comment o...
2218
2219
2220
2221
2222
   * Increase reference count of swap entry by 1.
   * Returns 0 for success, or -ENOMEM if a swap_count_continuation is required
   * but could not be atomically allocated.  Returns 0, just as if it succeeded,
   * if __swap_duplicate() fails for another reason (-EINVAL or -ENOENT), which
   * might occur if a page table entry has got corrupted.
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
2223
   */
570a335b8   Hugh Dickins   swap_info: swap c...
2224
  int swap_duplicate(swp_entry_t entry)
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
2225
  {
570a335b8   Hugh Dickins   swap_info: swap c...
2226
2227
2228
2229
2230
  	int err = 0;
  
  	while (!err && __swap_duplicate(entry, 1) == -ENOMEM)
  		err = add_swap_count_continuation(entry, GFP_ATOMIC);
  	return err;
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
2231
  }
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2232

cb4b86ba4   KAMEZAWA Hiroyuki   mm: add swap cach...
2233
  /*
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
2234
2235
   * @entry: swap entry for which we allocate swap cache.
   *
73c34b6ac   Hugh Dickins   swap_info: miscel...
2236
   * Called when allocating swap cache for existing swap entry,
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
2237
2238
2239
   * This can return error codes. Returns 0 at success.
   * -EBUSY means there is a swap cache.
   * Note: return code is different from swap_duplicate().
cb4b86ba4   KAMEZAWA Hiroyuki   mm: add swap cach...
2240
2241
2242
   */
  int swapcache_prepare(swp_entry_t entry)
  {
253d553ba   Hugh Dickins   swap_info: SWAP_H...
2243
  	return __swap_duplicate(entry, SWAP_HAS_CACHE);
cb4b86ba4   KAMEZAWA Hiroyuki   mm: add swap cach...
2244
  }
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2245
  /*
5d337b919   Hugh Dickins   [PATCH] swap: swa...
2246
   * swap_lock prevents swap_map being freed. Don't grab an extra
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2247
2248
2249
2250
   * reference on the swaphandle, it doesn't matter if it becomes unused.
   */
  int valid_swaphandles(swp_entry_t entry, unsigned long *offset)
  {
8952898b0   Hugh Dickins   swapin: fix valid...
2251
  	struct swap_info_struct *si;
3f9e7949f   Hugh Dickins   [PATCH] valid_swa...
2252
  	int our_page_cluster = page_cluster;
8952898b0   Hugh Dickins   swapin: fix valid...
2253
2254
2255
  	pgoff_t target, toff;
  	pgoff_t base, end;
  	int nr_pages = 0;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2256

3f9e7949f   Hugh Dickins   [PATCH] valid_swa...
2257
  	if (!our_page_cluster)	/* no readahead */
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2258
  		return 0;
8952898b0   Hugh Dickins   swapin: fix valid...
2259

efa90a981   Hugh Dickins   swap_info: change...
2260
  	si = swap_info[swp_type(entry)];
8952898b0   Hugh Dickins   swapin: fix valid...
2261
2262
2263
2264
2265
  	target = swp_offset(entry);
  	base = (target >> our_page_cluster) << our_page_cluster;
  	end = base + (1 << our_page_cluster);
  	if (!base)		/* first page is swap header */
  		base++;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2266

5d337b919   Hugh Dickins   [PATCH] swap: swa...
2267
  	spin_lock(&swap_lock);
8952898b0   Hugh Dickins   swapin: fix valid...
2268
2269
2270
2271
2272
2273
2274
2275
  	if (end > si->max)	/* don't go beyond end of map */
  		end = si->max;
  
  	/* Count contiguous allocated slots above our target */
  	for (toff = target; ++toff < end; nr_pages++) {
  		/* Don't read in free or bad pages */
  		if (!si->swap_map[toff])
  			break;
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
2276
  		if (swap_count(si->swap_map[toff]) == SWAP_MAP_BAD)
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2277
  			break;
8952898b0   Hugh Dickins   swapin: fix valid...
2278
2279
2280
  	}
  	/* Count contiguous allocated slots below our target */
  	for (toff = target; --toff >= base; nr_pages++) {
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2281
  		/* Don't read in free or bad pages */
8952898b0   Hugh Dickins   swapin: fix valid...
2282
  		if (!si->swap_map[toff])
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2283
  			break;
355cfa73d   KAMEZAWA Hiroyuki   mm: modify swap_m...
2284
  		if (swap_count(si->swap_map[toff]) == SWAP_MAP_BAD)
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2285
  			break;
8952898b0   Hugh Dickins   swapin: fix valid...
2286
  	}
5d337b919   Hugh Dickins   [PATCH] swap: swa...
2287
  	spin_unlock(&swap_lock);
8952898b0   Hugh Dickins   swapin: fix valid...
2288
2289
2290
2291
2292
2293
2294
  
  	/*
  	 * Indicate starting offset, and return number of pages to get:
  	 * if only 1, say 0, since there's then no readahead to be done.
  	 */
  	*offset = ++toff;
  	return nr_pages? ++nr_pages: 0;
1da177e4c   Linus Torvalds   Linux-2.6.12-rc2
2295
  }
570a335b8   Hugh Dickins   swap_info: swap c...
2296
2297
2298
2299
2300
2301
2302
2303
2304
2305
2306
2307
2308
2309
2310
2311
2312
2313
2314
2315
2316
2317
2318
2319
2320
2321
2322
2323
2324
2325
2326
2327
2328
2329
2330
2331
2332
2333
2334
2335
2336
2337
2338
2339
2340
2341
2342
2343
2344
2345
2346
2347
2348
2349
2350
2351
2352
2353
2354
2355
2356
2357
2358
2359
2360
2361
2362
2363
2364
2365
2366
2367
2368
2369
2370
2371
2372
2373
2374
2375
2376
2377
2378
2379
2380
2381
2382
2383
2384
2385
2386
2387
2388
2389
2390
2391
2392
2393
2394
2395
2396
2397
2398
2399
2400
2401
2402
2403
2404
2405
2406
2407
2408
2409
2410
2411
2412
2413
2414
2415
2416
2417
2418
2419
2420
2421
2422
2423
2424
2425
2426
2427
2428
2429
2430
2431
2432
2433
2434
2435
2436
2437
2438
2439
2440
2441
2442
2443
2444
2445
2446
2447
2448
2449
2450
2451
2452
2453
2454
2455
2456
2457
2458
2459
2460
2461
2462
2463
2464
2465
2466
2467
2468
2469
2470
2471
2472
2473
2474
2475
2476
2477
2478
2479
2480
2481
2482
2483
2484
2485
2486
2487
2488
2489
2490
2491
2492
2493
2494
2495
2496
2497
2498
2499
2500
2501
2502
2503
2504
2505
2506
2507
2508
2509
2510
2511
  
  /*
   * add_swap_count_continuation - called when a swap count is duplicated
   * beyond SWAP_MAP_MAX, it allocates a new page and links that to the entry's
   * page of the original vmalloc'ed swap_map, to hold the continuation count
   * (for that entry and for its neighbouring PAGE_SIZE swap entries).  Called
   * again when count is duplicated beyond SWAP_MAP_MAX * SWAP_CONT_MAX, etc.
   *
   * These continuation pages are seldom referenced: the common paths all work
   * on the original swap_map, only referring to a continuation page when the
   * low "digit" of a count is incremented or decremented through SWAP_MAP_MAX.
   *
   * add_swap_count_continuation(, GFP_ATOMIC) can be called while holding
   * page table locks; if it fails, add_swap_count_continuation(, GFP_KERNEL)
   * can be called after dropping locks.
   */
  int add_swap_count_continuation(swp_entry_t entry, gfp_t gfp_mask)
  {
  	struct swap_info_struct *si;
  	struct page *head;
  	struct page *page;
  	struct page *list_page;
  	pgoff_t offset;
  	unsigned char count;
  
  	/*
  	 * When debugging, it's easier to use __GFP_ZERO here; but it's better
  	 * for latency not to zero a page while GFP_ATOMIC and holding locks.
  	 */
  	page = alloc_page(gfp_mask | __GFP_HIGHMEM);
  
  	si = swap_info_get(entry);
  	if (!si) {
  		/*
  		 * An acceptable race has occurred since the failing
  		 * __swap_duplicate(): the swap entry has been freed,
  		 * perhaps even the whole swap_map cleared for swapoff.
  		 */
  		goto outer;
  	}
  
  	offset = swp_offset(entry);
  	count = si->swap_map[offset] & ~SWAP_HAS_CACHE;
  
  	if ((count & ~COUNT_CONTINUED) != SWAP_MAP_MAX) {
  		/*
  		 * The higher the swap count, the more likely it is that tasks
  		 * will race to add swap count continuation: we need to avoid
  		 * over-provisioning.
  		 */
  		goto out;
  	}
  
  	if (!page) {
  		spin_unlock(&swap_lock);
  		return -ENOMEM;
  	}
  
  	/*
  	 * We are fortunate that although vmalloc_to_page uses pte_offset_map,
  	 * no architecture is using highmem pages for kernel pagetables: so it
  	 * will not corrupt the GFP_ATOMIC caller's atomic pagetable kmaps.
  	 */
  	head = vmalloc_to_page(si->swap_map + offset);
  	offset &= ~PAGE_MASK;
  
  	/*
  	 * Page allocation does not initialize the page's lru field,
  	 * but it does always reset its private field.
  	 */
  	if (!page_private(head)) {
  		BUG_ON(count & COUNT_CONTINUED);
  		INIT_LIST_HEAD(&head->lru);
  		set_page_private(head, SWP_CONTINUED);
  		si->flags |= SWP_CONTINUED;
  	}
  
  	list_for_each_entry(list_page, &head->lru, lru) {
  		unsigned char *map;
  
  		/*
  		 * If the previous map said no continuation, but we've found
  		 * a continuation page, free our allocation and use this one.
  		 */
  		if (!(count & COUNT_CONTINUED))
  			goto out;
  
  		map = kmap_atomic(list_page, KM_USER0) + offset;
  		count = *map;
  		kunmap_atomic(map, KM_USER0);
  
  		/*
  		 * If this continuation count now has some space in it,
  		 * free our allocation and use this one.
  		 */
  		if ((count & ~COUNT_CONTINUED) != SWAP_CONT_MAX)
  			goto out;
  	}
  
  	list_add_tail(&page->lru, &head->lru);
  	page = NULL;			/* now it's attached, don't free it */
  out:
  	spin_unlock(&swap_lock);
  outer:
  	if (page)
  		__free_page(page);
  	return 0;
  }
  
  /*
   * swap_count_continued - when the original swap_map count is incremented
   * from SWAP_MAP_MAX, check if there is already a continuation page to carry
   * into, carry if so, or else fail until a new continuation page is allocated;
   * when the original swap_map count is decremented from 0 with continuation,
   * borrow from the continuation and report whether it still holds more.
   * Called while __swap_duplicate() or swap_entry_free() holds swap_lock.
   */
  static bool swap_count_continued(struct swap_info_struct *si,
  				 pgoff_t offset, unsigned char count)
  {
  	struct page *head;
  	struct page *page;
  	unsigned char *map;
  
  	head = vmalloc_to_page(si->swap_map + offset);
  	if (page_private(head) != SWP_CONTINUED) {
  		BUG_ON(count & COUNT_CONTINUED);
  		return false;		/* need to add count continuation */
  	}
  
  	offset &= ~PAGE_MASK;
  	page = list_entry(head->lru.next, struct page, lru);
  	map = kmap_atomic(page, KM_USER0) + offset;
  
  	if (count == SWAP_MAP_MAX)	/* initial increment from swap_map */
  		goto init_map;		/* jump over SWAP_CONT_MAX checks */
  
  	if (count == (SWAP_MAP_MAX | COUNT_CONTINUED)) { /* incrementing */
  		/*
  		 * Think of how you add 1 to 999
  		 */
  		while (*map == (SWAP_CONT_MAX | COUNT_CONTINUED)) {
  			kunmap_atomic(map, KM_USER0);
  			page = list_entry(page->lru.next, struct page, lru);
  			BUG_ON(page == head);
  			map = kmap_atomic(page, KM_USER0) + offset;
  		}
  		if (*map == SWAP_CONT_MAX) {
  			kunmap_atomic(map, KM_USER0);
  			page = list_entry(page->lru.next, struct page, lru);
  			if (page == head)
  				return false;	/* add count continuation */
  			map = kmap_atomic(page, KM_USER0) + offset;
  init_map:		*map = 0;		/* we didn't zero the page */
  		}
  		*map += 1;
  		kunmap_atomic(map, KM_USER0);
  		page = list_entry(page->lru.prev, struct page, lru);
  		while (page != head) {
  			map = kmap_atomic(page, KM_USER0) + offset;
  			*map = COUNT_CONTINUED;
  			kunmap_atomic(map, KM_USER0);
  			page = list_entry(page->lru.prev, struct page, lru);
  		}
  		return true;			/* incremented */
  
  	} else {				/* decrementing */
  		/*
  		 * Think of how you subtract 1 from 1000
  		 */
  		BUG_ON(count != COUNT_CONTINUED);
  		while (*map == COUNT_CONTINUED) {
  			kunmap_atomic(map, KM_USER0);
  			page = list_entry(page->lru.next, struct page, lru);
  			BUG_ON(page == head);
  			map = kmap_atomic(page, KM_USER0) + offset;
  		}
  		BUG_ON(*map == 0);
  		*map -= 1;
  		if (*map == 0)
  			count = 0;
  		kunmap_atomic(map, KM_USER0);
  		page = list_entry(page->lru.prev, struct page, lru);
  		while (page != head) {
  			map = kmap_atomic(page, KM_USER0) + offset;
  			*map = SWAP_CONT_MAX | count;
  			count = COUNT_CONTINUED;
  			kunmap_atomic(map, KM_USER0);
  			page = list_entry(page->lru.prev, struct page, lru);
  		}
  		return count == COUNT_CONTINUED;
  	}
  }
  
  /*
   * free_swap_count_continuations - swapoff free all the continuation pages
   * appended to the swap_map, after swap_map is quiesced, before vfree'ing it.
   */
  static void free_swap_count_continuations(struct swap_info_struct *si)
  {
  	pgoff_t offset;
  
  	for (offset = 0; offset < si->max; offset += PAGE_SIZE) {
  		struct page *head;
  		head = vmalloc_to_page(si->swap_map + offset);
  		if (page_private(head)) {
  			struct list_head *this, *next;
  			list_for_each_safe(this, next, &head->lru) {
  				struct page *page;
  				page = list_entry(this, struct page, lru);
  				list_del(this);
  				__free_page(page);
  			}
  		}
  	}
  }