]> git.proxmox.com Git - mirror_ubuntu-zesty-kernel.git/blame - mm/mmap.c
UBUNTU: Ubuntu-4.10.0-37.41
[mirror_ubuntu-zesty-kernel.git] / mm / mmap.c
CommitLineData
1da177e4
LT
1/*
2 * mm/mmap.c
3 *
4 * Written by obz.
5 *
046c6884 6 * Address space accounting code <alan@lxorguk.ukuu.org.uk>
1da177e4
LT
7 */
8
b1de0d13
MH
9#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
10
e8420a8e 11#include <linux/kernel.h>
1da177e4 12#include <linux/slab.h>
4af3c9cc 13#include <linux/backing-dev.h>
1da177e4 14#include <linux/mm.h>
615d6e87 15#include <linux/vmacache.h>
1da177e4
LT
16#include <linux/shm.h>
17#include <linux/mman.h>
18#include <linux/pagemap.h>
19#include <linux/swap.h>
20#include <linux/syscalls.h>
c59ede7b 21#include <linux/capability.h>
1da177e4
LT
22#include <linux/init.h>
23#include <linux/file.h>
24#include <linux/fs.h>
25#include <linux/personality.h>
26#include <linux/security.h>
27#include <linux/hugetlb.h>
c01d5b30 28#include <linux/shmem_fs.h>
1da177e4 29#include <linux/profile.h>
b95f1b31 30#include <linux/export.h>
1da177e4
LT
31#include <linux/mount.h>
32#include <linux/mempolicy.h>
33#include <linux/rmap.h>
cddb8a5c 34#include <linux/mmu_notifier.h>
82f71ae4 35#include <linux/mmdebug.h>
cdd6c482 36#include <linux/perf_event.h>
120a795d 37#include <linux/audit.h>
b15d00b6 38#include <linux/khugepaged.h>
2b144498 39#include <linux/uprobes.h>
d3737187 40#include <linux/rbtree_augmented.h>
1640879a
AS
41#include <linux/notifier.h>
42#include <linux/memory.h>
b1de0d13 43#include <linux/printk.h>
19a809af 44#include <linux/userfaultfd_k.h>
d977d56c 45#include <linux/moduleparam.h>
62b5f7d0 46#include <linux/pkeys.h>
1da177e4 47
7c0f6ba6 48#include <linux/uaccess.h>
1da177e4
LT
49#include <asm/cacheflush.h>
50#include <asm/tlb.h>
d6dd61c8 51#include <asm/mmu_context.h>
1da177e4 52
42b77728
JB
53#include "internal.h"
54
3a459756
KK
55#ifndef arch_mmap_check
56#define arch_mmap_check(addr, len, flags) (0)
57#endif
58
d07e2259
DC
59#ifdef CONFIG_HAVE_ARCH_MMAP_RND_BITS
60const int mmap_rnd_bits_min = CONFIG_ARCH_MMAP_RND_BITS_MIN;
61const int mmap_rnd_bits_max = CONFIG_ARCH_MMAP_RND_BITS_MAX;
62int mmap_rnd_bits __read_mostly = CONFIG_ARCH_MMAP_RND_BITS;
63#endif
64#ifdef CONFIG_HAVE_ARCH_MMAP_RND_COMPAT_BITS
65const int mmap_rnd_compat_bits_min = CONFIG_ARCH_MMAP_RND_COMPAT_BITS_MIN;
66const int mmap_rnd_compat_bits_max = CONFIG_ARCH_MMAP_RND_COMPAT_BITS_MAX;
67int mmap_rnd_compat_bits __read_mostly = CONFIG_ARCH_MMAP_RND_COMPAT_BITS;
68#endif
69
f4fcd558 70static bool ignore_rlimit_data;
d977d56c 71core_param(ignore_rlimit_data, ignore_rlimit_data, bool, 0644);
d07e2259 72
e0da382c
HD
73static void unmap_region(struct mm_struct *mm,
74 struct vm_area_struct *vma, struct vm_area_struct *prev,
75 unsigned long start, unsigned long end);
76
1da177e4
LT
77/* description of effects of mapping type and prot in current implementation.
78 * this is due to the limited x86 page protection hardware. The expected
79 * behavior is in parens:
80 *
81 * map_type prot
82 * PROT_NONE PROT_READ PROT_WRITE PROT_EXEC
83 * MAP_SHARED r: (no) no r: (yes) yes r: (no) yes r: (no) yes
84 * w: (no) no w: (no) no w: (yes) yes w: (no) no
85 * x: (no) no x: (no) yes x: (no) yes x: (yes) yes
cc71aba3 86 *
1da177e4
LT
87 * MAP_PRIVATE r: (no) no r: (yes) yes r: (no) yes r: (no) yes
88 * w: (no) no w: (no) no w: (copy) copy w: (no) no
89 * x: (no) no x: (no) yes x: (no) yes x: (yes) yes
90 *
cab15ce6
CM
91 * On arm64, PROT_EXEC has the following behaviour for both MAP_SHARED and
92 * MAP_PRIVATE:
93 * r: (no) no
94 * w: (no) no
95 * x: (yes) yes
1da177e4
LT
96 */
97pgprot_t protection_map[16] = {
98 __P000, __P001, __P010, __P011, __P100, __P101, __P110, __P111,
99 __S000, __S001, __S010, __S011, __S100, __S101, __S110, __S111
100};
101
804af2cf
HD
102pgprot_t vm_get_page_prot(unsigned long vm_flags)
103{
b845f313
DK
104 return __pgprot(pgprot_val(protection_map[vm_flags &
105 (VM_READ|VM_WRITE|VM_EXEC|VM_SHARED)]) |
106 pgprot_val(arch_vm_get_page_prot(vm_flags)));
804af2cf
HD
107}
108EXPORT_SYMBOL(vm_get_page_prot);
109
64e45507
PF
110static pgprot_t vm_pgprot_modify(pgprot_t oldprot, unsigned long vm_flags)
111{
112 return pgprot_modify(oldprot, vm_get_page_prot(vm_flags));
113}
114
115/* Update vma->vm_page_prot to reflect vma->vm_flags. */
116void vma_set_page_prot(struct vm_area_struct *vma)
117{
118 unsigned long vm_flags = vma->vm_flags;
6d2329f8 119 pgprot_t vm_page_prot;
64e45507 120
6d2329f8
AA
121 vm_page_prot = vm_pgprot_modify(vma->vm_page_prot, vm_flags);
122 if (vma_wants_writenotify(vma, vm_page_prot)) {
64e45507 123 vm_flags &= ~VM_SHARED;
6d2329f8 124 vm_page_prot = vm_pgprot_modify(vm_page_prot, vm_flags);
64e45507 125 }
6d2329f8
AA
126 /* remove_protection_ptes reads vma->vm_page_prot without mmap_sem */
127 WRITE_ONCE(vma->vm_page_prot, vm_page_prot);
64e45507
PF
128}
129
1da177e4 130/*
c8c06efa 131 * Requires inode->i_mapping->i_mmap_rwsem
1da177e4
LT
132 */
133static void __remove_shared_vm_struct(struct vm_area_struct *vma,
134 struct file *file, struct address_space *mapping)
135{
136 if (vma->vm_flags & VM_DENYWRITE)
496ad9aa 137 atomic_inc(&file_inode(file)->i_writecount);
1da177e4 138 if (vma->vm_flags & VM_SHARED)
4bb5f5d9 139 mapping_unmap_writable(mapping);
1da177e4
LT
140
141 flush_dcache_mmap_lock(mapping);
27ba0644 142 vma_interval_tree_remove(vma, &mapping->i_mmap);
1da177e4
LT
143 flush_dcache_mmap_unlock(mapping);
144}
145
146/*
6b2dbba8 147 * Unlink a file-based vm structure from its interval tree, to hide
a8fb5618 148 * vma from rmap and vmtruncate before freeing its page tables.
1da177e4 149 */
a8fb5618 150void unlink_file_vma(struct vm_area_struct *vma)
1da177e4
LT
151{
152 struct file *file = vma->vm_file;
153
1da177e4
LT
154 if (file) {
155 struct address_space *mapping = file->f_mapping;
83cde9e8 156 i_mmap_lock_write(mapping);
1da177e4 157 __remove_shared_vm_struct(vma, file, mapping);
83cde9e8 158 i_mmap_unlock_write(mapping);
1da177e4 159 }
a8fb5618
HD
160}
161
162/*
163 * Close a vm structure and free it, returning the next.
164 */
165static struct vm_area_struct *remove_vma(struct vm_area_struct *vma)
166{
167 struct vm_area_struct *next = vma->vm_next;
168
a8fb5618 169 might_sleep();
1da177e4
LT
170 if (vma->vm_ops && vma->vm_ops->close)
171 vma->vm_ops->close(vma);
e9714acf 172 if (vma->vm_file)
e14748e8 173 vma_fput(vma);
f0be3d32 174 mpol_put(vma_policy(vma));
1da177e4 175 kmem_cache_free(vm_area_cachep, vma);
a8fb5618 176 return next;
1da177e4
LT
177}
178
5d22fc25 179static int do_brk(unsigned long addr, unsigned long len);
e4eb1ff6 180
6a6160a7 181SYSCALL_DEFINE1(brk, unsigned long, brk)
1da177e4 182{
8764b338 183 unsigned long retval;
1da177e4
LT
184 unsigned long newbrk, oldbrk;
185 struct mm_struct *mm = current->mm;
ddeaddb9 186 struct vm_area_struct *next;
a5b4592c 187 unsigned long min_brk;
128557ff 188 bool populate;
1da177e4 189
dc0ef0df
MH
190 if (down_write_killable(&mm->mmap_sem))
191 return -EINTR;
1da177e4 192
a5b4592c 193#ifdef CONFIG_COMPAT_BRK
5520e894
JK
194 /*
195 * CONFIG_COMPAT_BRK can still be overridden by setting
196 * randomize_va_space to 2, which will still cause mm->start_brk
197 * to be arbitrarily shifted
198 */
4471a675 199 if (current->brk_randomized)
5520e894
JK
200 min_brk = mm->start_brk;
201 else
202 min_brk = mm->end_data;
a5b4592c
JK
203#else
204 min_brk = mm->start_brk;
205#endif
206 if (brk < min_brk)
1da177e4 207 goto out;
1e624196
RG
208
209 /*
210 * Check against rlimit here. If this check is done later after the test
211 * of oldbrk with newbrk then it can escape the test and let the data
212 * segment grow beyond its set limit the in case where the limit is
213 * not page aligned -Ram Gupta
214 */
8764b338
CG
215 if (check_data_rlimit(rlimit(RLIMIT_DATA), brk, mm->start_brk,
216 mm->end_data, mm->start_data))
1e624196
RG
217 goto out;
218
1da177e4
LT
219 newbrk = PAGE_ALIGN(brk);
220 oldbrk = PAGE_ALIGN(mm->brk);
221 if (oldbrk == newbrk)
222 goto set_brk;
223
224 /* Always allow shrinking brk. */
225 if (brk <= mm->brk) {
226 if (!do_munmap(mm, newbrk, oldbrk-newbrk))
227 goto set_brk;
228 goto out;
229 }
230
1da177e4 231 /* Check against existing mmap mappings. */
ddeaddb9
HD
232 next = find_vma(mm, oldbrk);
233 if (next && newbrk + PAGE_SIZE > vm_start_gap(next))
1da177e4
LT
234 goto out;
235
236 /* Ok, looks good - let it rip. */
5d22fc25 237 if (do_brk(oldbrk, newbrk-oldbrk) < 0)
1da177e4 238 goto out;
128557ff 239
1da177e4
LT
240set_brk:
241 mm->brk = brk;
128557ff
ML
242 populate = newbrk > oldbrk && (mm->def_flags & VM_LOCKED) != 0;
243 up_write(&mm->mmap_sem);
244 if (populate)
245 mm_populate(oldbrk, newbrk - oldbrk);
246 return brk;
247
1da177e4
LT
248out:
249 retval = mm->brk;
250 up_write(&mm->mmap_sem);
251 return retval;
252}
253
d3737187
ML
254static long vma_compute_subtree_gap(struct vm_area_struct *vma)
255{
ddeaddb9
HD
256 unsigned long max, prev_end, subtree_gap;
257
258 /*
259 * Note: in the rare case of a VM_GROWSDOWN above a VM_GROWSUP, we
260 * allow two stack_guard_gaps between them here, and when choosing
261 * an unmapped area; whereas when expanding we only require one.
262 * That's a little inconsistent, but keeps the code here simpler.
263 */
264 max = vm_start_gap(vma);
265 if (vma->vm_prev) {
266 prev_end = vm_end_gap(vma->vm_prev);
267 if (max > prev_end)
268 max -= prev_end;
269 else
270 max = 0;
271 }
d3737187
ML
272 if (vma->vm_rb.rb_left) {
273 subtree_gap = rb_entry(vma->vm_rb.rb_left,
274 struct vm_area_struct, vm_rb)->rb_subtree_gap;
275 if (subtree_gap > max)
276 max = subtree_gap;
277 }
278 if (vma->vm_rb.rb_right) {
279 subtree_gap = rb_entry(vma->vm_rb.rb_right,
280 struct vm_area_struct, vm_rb)->rb_subtree_gap;
281 if (subtree_gap > max)
282 max = subtree_gap;
283 }
284 return max;
285}
286
ed8ea815 287#ifdef CONFIG_DEBUG_VM_RB
acf128d0 288static int browse_rb(struct mm_struct *mm)
1da177e4 289{
acf128d0 290 struct rb_root *root = &mm->mm_rb;
5a0768f6 291 int i = 0, j, bug = 0;
1da177e4
LT
292 struct rb_node *nd, *pn = NULL;
293 unsigned long prev = 0, pend = 0;
294
295 for (nd = rb_first(root); nd; nd = rb_next(nd)) {
296 struct vm_area_struct *vma;
297 vma = rb_entry(nd, struct vm_area_struct, vm_rb);
5a0768f6 298 if (vma->vm_start < prev) {
ff26f70f
AM
299 pr_emerg("vm_start %lx < prev %lx\n",
300 vma->vm_start, prev);
5a0768f6
ML
301 bug = 1;
302 }
303 if (vma->vm_start < pend) {
ff26f70f
AM
304 pr_emerg("vm_start %lx < pend %lx\n",
305 vma->vm_start, pend);
5a0768f6
ML
306 bug = 1;
307 }
308 if (vma->vm_start > vma->vm_end) {
ff26f70f
AM
309 pr_emerg("vm_start %lx > vm_end %lx\n",
310 vma->vm_start, vma->vm_end);
5a0768f6
ML
311 bug = 1;
312 }
acf128d0 313 spin_lock(&mm->page_table_lock);
5a0768f6 314 if (vma->rb_subtree_gap != vma_compute_subtree_gap(vma)) {
8542bdfc 315 pr_emerg("free gap %lx, correct %lx\n",
5a0768f6
ML
316 vma->rb_subtree_gap,
317 vma_compute_subtree_gap(vma));
318 bug = 1;
319 }
acf128d0 320 spin_unlock(&mm->page_table_lock);
1da177e4
LT
321 i++;
322 pn = nd;
d1af65d1
DM
323 prev = vma->vm_start;
324 pend = vma->vm_end;
1da177e4
LT
325 }
326 j = 0;
5a0768f6 327 for (nd = pn; nd; nd = rb_prev(nd))
1da177e4 328 j++;
5a0768f6 329 if (i != j) {
8542bdfc 330 pr_emerg("backwards %d, forwards %d\n", j, i);
5a0768f6 331 bug = 1;
1da177e4 332 }
5a0768f6 333 return bug ? -1 : i;
1da177e4
LT
334}
335
d3737187
ML
336static void validate_mm_rb(struct rb_root *root, struct vm_area_struct *ignore)
337{
338 struct rb_node *nd;
339
340 for (nd = rb_first(root); nd; nd = rb_next(nd)) {
341 struct vm_area_struct *vma;
342 vma = rb_entry(nd, struct vm_area_struct, vm_rb);
96dad67f
SL
343 VM_BUG_ON_VMA(vma != ignore &&
344 vma->rb_subtree_gap != vma_compute_subtree_gap(vma),
345 vma);
1da177e4 346 }
1da177e4
LT
347}
348
eafd4dc4 349static void validate_mm(struct mm_struct *mm)
1da177e4
LT
350{
351 int bug = 0;
352 int i = 0;
5a0768f6 353 unsigned long highest_address = 0;
ed8ea815 354 struct vm_area_struct *vma = mm->mmap;
ff26f70f 355
ed8ea815 356 while (vma) {
12352d3c 357 struct anon_vma *anon_vma = vma->anon_vma;
ed8ea815 358 struct anon_vma_chain *avc;
ff26f70f 359
12352d3c
KK
360 if (anon_vma) {
361 anon_vma_lock_read(anon_vma);
362 list_for_each_entry(avc, &vma->anon_vma_chain, same_vma)
363 anon_vma_interval_tree_verify(avc);
364 anon_vma_unlock_read(anon_vma);
365 }
366
ddeaddb9 367 highest_address = vm_end_gap(vma);
ed8ea815 368 vma = vma->vm_next;
1da177e4
LT
369 i++;
370 }
5a0768f6 371 if (i != mm->map_count) {
8542bdfc 372 pr_emerg("map_count %d vm_next %d\n", mm->map_count, i);
5a0768f6
ML
373 bug = 1;
374 }
375 if (highest_address != mm->highest_vm_end) {
8542bdfc 376 pr_emerg("mm->highest_vm_end %lx, found %lx\n",
ff26f70f 377 mm->highest_vm_end, highest_address);
5a0768f6
ML
378 bug = 1;
379 }
acf128d0 380 i = browse_rb(mm);
5a0768f6 381 if (i != mm->map_count) {
ff26f70f
AM
382 if (i != -1)
383 pr_emerg("map_count %d rb %d\n", mm->map_count, i);
5a0768f6
ML
384 bug = 1;
385 }
96dad67f 386 VM_BUG_ON_MM(bug, mm);
1da177e4
LT
387}
388#else
d3737187 389#define validate_mm_rb(root, ignore) do { } while (0)
1da177e4
LT
390#define validate_mm(mm) do { } while (0)
391#endif
392
d3737187
ML
393RB_DECLARE_CALLBACKS(static, vma_gap_callbacks, struct vm_area_struct, vm_rb,
394 unsigned long, rb_subtree_gap, vma_compute_subtree_gap)
395
396/*
397 * Update augmented rbtree rb_subtree_gap values after vma->vm_start or
398 * vma->vm_prev->vm_end values changed, without modifying the vma's position
399 * in the rbtree.
400 */
401static void vma_gap_update(struct vm_area_struct *vma)
402{
403 /*
404 * As it turns out, RB_DECLARE_CALLBACKS() already created a callback
405 * function that does exacltly what we want.
406 */
407 vma_gap_callbacks_propagate(&vma->vm_rb, NULL);
408}
409
410static inline void vma_rb_insert(struct vm_area_struct *vma,
411 struct rb_root *root)
412{
413 /* All rb_subtree_gap values must be consistent prior to insertion */
414 validate_mm_rb(root, NULL);
415
416 rb_insert_augmented(&vma->vm_rb, root, &vma_gap_callbacks);
417}
418
8f26e0b1 419static void __vma_rb_erase(struct vm_area_struct *vma, struct rb_root *root)
d3737187 420{
d3737187
ML
421 /*
422 * Note rb_erase_augmented is a fairly large inline function,
423 * so make sure we instantiate it only once with our desired
424 * augmented rbtree callbacks.
425 */
426 rb_erase_augmented(&vma->vm_rb, root, &vma_gap_callbacks);
427}
428
8f26e0b1
AA
429static __always_inline void vma_rb_erase_ignore(struct vm_area_struct *vma,
430 struct rb_root *root,
431 struct vm_area_struct *ignore)
432{
433 /*
434 * All rb_subtree_gap values must be consistent prior to erase,
435 * with the possible exception of the "next" vma being erased if
436 * next->vm_start was reduced.
437 */
438 validate_mm_rb(root, ignore);
439
440 __vma_rb_erase(vma, root);
441}
442
443static __always_inline void vma_rb_erase(struct vm_area_struct *vma,
444 struct rb_root *root)
445{
446 /*
447 * All rb_subtree_gap values must be consistent prior to erase,
448 * with the possible exception of the vma being erased.
449 */
450 validate_mm_rb(root, vma);
451
452 __vma_rb_erase(vma, root);
453}
454
bf181b9f
ML
455/*
456 * vma has some anon_vma assigned, and is already inserted on that
457 * anon_vma's interval trees.
458 *
459 * Before updating the vma's vm_start / vm_end / vm_pgoff fields, the
460 * vma must be removed from the anon_vma's interval trees using
461 * anon_vma_interval_tree_pre_update_vma().
462 *
463 * After the update, the vma will be reinserted using
464 * anon_vma_interval_tree_post_update_vma().
465 *
466 * The entire update must be protected by exclusive mmap_sem and by
467 * the root anon_vma's mutex.
468 */
469static inline void
470anon_vma_interval_tree_pre_update_vma(struct vm_area_struct *vma)
471{
472 struct anon_vma_chain *avc;
473
474 list_for_each_entry(avc, &vma->anon_vma_chain, same_vma)
475 anon_vma_interval_tree_remove(avc, &avc->anon_vma->rb_root);
476}
477
478static inline void
479anon_vma_interval_tree_post_update_vma(struct vm_area_struct *vma)
480{
481 struct anon_vma_chain *avc;
482
483 list_for_each_entry(avc, &vma->anon_vma_chain, same_vma)
484 anon_vma_interval_tree_insert(avc, &avc->anon_vma->rb_root);
485}
486
6597d783
HD
487static int find_vma_links(struct mm_struct *mm, unsigned long addr,
488 unsigned long end, struct vm_area_struct **pprev,
489 struct rb_node ***rb_link, struct rb_node **rb_parent)
1da177e4 490{
6597d783 491 struct rb_node **__rb_link, *__rb_parent, *rb_prev;
1da177e4
LT
492
493 __rb_link = &mm->mm_rb.rb_node;
494 rb_prev = __rb_parent = NULL;
1da177e4
LT
495
496 while (*__rb_link) {
497 struct vm_area_struct *vma_tmp;
498
499 __rb_parent = *__rb_link;
500 vma_tmp = rb_entry(__rb_parent, struct vm_area_struct, vm_rb);
501
502 if (vma_tmp->vm_end > addr) {
6597d783
HD
503 /* Fail if an existing vma overlaps the area */
504 if (vma_tmp->vm_start < end)
505 return -ENOMEM;
1da177e4
LT
506 __rb_link = &__rb_parent->rb_left;
507 } else {
508 rb_prev = __rb_parent;
509 __rb_link = &__rb_parent->rb_right;
510 }
511 }
512
513 *pprev = NULL;
514 if (rb_prev)
515 *pprev = rb_entry(rb_prev, struct vm_area_struct, vm_rb);
516 *rb_link = __rb_link;
517 *rb_parent = __rb_parent;
6597d783 518 return 0;
1da177e4
LT
519}
520
e8420a8e
CH
521static unsigned long count_vma_pages_range(struct mm_struct *mm,
522 unsigned long addr, unsigned long end)
523{
524 unsigned long nr_pages = 0;
525 struct vm_area_struct *vma;
526
527 /* Find first overlaping mapping */
528 vma = find_vma_intersection(mm, addr, end);
529 if (!vma)
530 return 0;
531
532 nr_pages = (min(end, vma->vm_end) -
533 max(addr, vma->vm_start)) >> PAGE_SHIFT;
534
535 /* Iterate over the rest of the overlaps */
536 for (vma = vma->vm_next; vma; vma = vma->vm_next) {
537 unsigned long overlap_len;
538
539 if (vma->vm_start > end)
540 break;
541
542 overlap_len = min(end, vma->vm_end) - vma->vm_start;
543 nr_pages += overlap_len >> PAGE_SHIFT;
544 }
545
546 return nr_pages;
547}
548
1da177e4
LT
549void __vma_link_rb(struct mm_struct *mm, struct vm_area_struct *vma,
550 struct rb_node **rb_link, struct rb_node *rb_parent)
551{
d3737187
ML
552 /* Update tracking information for the gap following the new vma. */
553 if (vma->vm_next)
554 vma_gap_update(vma->vm_next);
555 else
ddeaddb9 556 mm->highest_vm_end = vm_end_gap(vma);
d3737187
ML
557
558 /*
559 * vma->vm_prev wasn't known when we followed the rbtree to find the
560 * correct insertion point for that vma. As a result, we could not
561 * update the vma vm_rb parents rb_subtree_gap values on the way down.
562 * So, we first insert the vma with a zero rb_subtree_gap value
563 * (to be consistent with what we did on the way down), and then
564 * immediately update the gap to the correct value. Finally we
565 * rebalance the rbtree after all augmented values have been set.
566 */
1da177e4 567 rb_link_node(&vma->vm_rb, rb_parent, rb_link);
d3737187
ML
568 vma->rb_subtree_gap = 0;
569 vma_gap_update(vma);
570 vma_rb_insert(vma, &mm->mm_rb);
1da177e4
LT
571}
572
cb8f488c 573static void __vma_link_file(struct vm_area_struct *vma)
1da177e4 574{
48aae425 575 struct file *file;
1da177e4
LT
576
577 file = vma->vm_file;
578 if (file) {
579 struct address_space *mapping = file->f_mapping;
580
581 if (vma->vm_flags & VM_DENYWRITE)
496ad9aa 582 atomic_dec(&file_inode(file)->i_writecount);
1da177e4 583 if (vma->vm_flags & VM_SHARED)
4bb5f5d9 584 atomic_inc(&mapping->i_mmap_writable);
1da177e4
LT
585
586 flush_dcache_mmap_lock(mapping);
27ba0644 587 vma_interval_tree_insert(vma, &mapping->i_mmap);
1da177e4
LT
588 flush_dcache_mmap_unlock(mapping);
589 }
590}
591
592static void
593__vma_link(struct mm_struct *mm, struct vm_area_struct *vma,
594 struct vm_area_struct *prev, struct rb_node **rb_link,
595 struct rb_node *rb_parent)
596{
597 __vma_link_list(mm, vma, prev, rb_parent);
598 __vma_link_rb(mm, vma, rb_link, rb_parent);
1da177e4
LT
599}
600
601static void vma_link(struct mm_struct *mm, struct vm_area_struct *vma,
602 struct vm_area_struct *prev, struct rb_node **rb_link,
603 struct rb_node *rb_parent)
604{
605 struct address_space *mapping = NULL;
606
64ac4940 607 if (vma->vm_file) {
1da177e4 608 mapping = vma->vm_file->f_mapping;
83cde9e8 609 i_mmap_lock_write(mapping);
64ac4940 610 }
1da177e4
LT
611
612 __vma_link(mm, vma, prev, rb_link, rb_parent);
613 __vma_link_file(vma);
614
1da177e4 615 if (mapping)
83cde9e8 616 i_mmap_unlock_write(mapping);
1da177e4
LT
617
618 mm->map_count++;
619 validate_mm(mm);
620}
621
622/*
88f6b4c3 623 * Helper for vma_adjust() in the split_vma insert case: insert a vma into the
6b2dbba8 624 * mm's list and rbtree. It has already been inserted into the interval tree.
1da177e4 625 */
48aae425 626static void __insert_vm_struct(struct mm_struct *mm, struct vm_area_struct *vma)
1da177e4 627{
6597d783 628 struct vm_area_struct *prev;
48aae425 629 struct rb_node **rb_link, *rb_parent;
1da177e4 630
6597d783
HD
631 if (find_vma_links(mm, vma->vm_start, vma->vm_end,
632 &prev, &rb_link, &rb_parent))
633 BUG();
1da177e4
LT
634 __vma_link(mm, vma, prev, rb_link, rb_parent);
635 mm->map_count++;
636}
637
e86f15ee
AA
638static __always_inline void __vma_unlink_common(struct mm_struct *mm,
639 struct vm_area_struct *vma,
640 struct vm_area_struct *prev,
8f26e0b1
AA
641 bool has_prev,
642 struct vm_area_struct *ignore)
1da177e4 643{
d3737187 644 struct vm_area_struct *next;
297c5eee 645
8f26e0b1 646 vma_rb_erase_ignore(vma, &mm->mm_rb, ignore);
e86f15ee
AA
647 next = vma->vm_next;
648 if (has_prev)
649 prev->vm_next = next;
650 else {
651 prev = vma->vm_prev;
652 if (prev)
653 prev->vm_next = next;
654 else
655 mm->mmap = next;
656 }
297c5eee
LT
657 if (next)
658 next->vm_prev = prev;
615d6e87
DB
659
660 /* Kill the cache */
661 vmacache_invalidate(mm);
1da177e4
LT
662}
663
e86f15ee
AA
664static inline void __vma_unlink_prev(struct mm_struct *mm,
665 struct vm_area_struct *vma,
666 struct vm_area_struct *prev)
667{
8f26e0b1 668 __vma_unlink_common(mm, vma, prev, true, vma);
e86f15ee
AA
669}
670
1da177e4
LT
671/*
672 * We cannot adjust vm_start, vm_end, vm_pgoff fields of a vma that
673 * is already present in an i_mmap tree without adjusting the tree.
674 * The following helper function should be used when such adjustments
675 * are necessary. The "insert" vma (if any) is to be inserted
676 * before we drop the necessary locks.
677 */
e86f15ee
AA
678int __vma_adjust(struct vm_area_struct *vma, unsigned long start,
679 unsigned long end, pgoff_t pgoff, struct vm_area_struct *insert,
680 struct vm_area_struct *expand)
1da177e4
LT
681{
682 struct mm_struct *mm = vma->vm_mm;
e86f15ee 683 struct vm_area_struct *next = vma->vm_next, *orig_vma = vma;
1da177e4 684 struct address_space *mapping = NULL;
6b2dbba8 685 struct rb_root *root = NULL;
012f1800 686 struct anon_vma *anon_vma = NULL;
1da177e4 687 struct file *file = vma->vm_file;
d3737187 688 bool start_changed = false, end_changed = false;
1da177e4
LT
689 long adjust_next = 0;
690 int remove_next = 0;
691
692 if (next && !insert) {
734537c9 693 struct vm_area_struct *exporter = NULL, *importer = NULL;
287d97ac 694
1da177e4
LT
695 if (end >= next->vm_end) {
696 /*
697 * vma expands, overlapping all the next, and
698 * perhaps the one after too (mprotect case 6).
86d12e47 699 * The only other cases that gets here are
e86f15ee 700 * case 1, case 7 and case 8.
1da177e4 701 */
e86f15ee
AA
702 if (next == expand) {
703 /*
704 * The only case where we don't expand "vma"
705 * and we expand "next" instead is case 8.
706 */
707 VM_WARN_ON(end != next->vm_end);
708 /*
709 * remove_next == 3 means we're
710 * removing "vma" and that to do so we
711 * swapped "vma" and "next".
712 */
713 remove_next = 3;
714 VM_WARN_ON(file != next->vm_file);
715 swap(vma, next);
716 } else {
717 VM_WARN_ON(expand != vma);
718 /*
719 * case 1, 6, 7, remove_next == 2 is case 6,
720 * remove_next == 1 is case 1 or 7.
721 */
722 remove_next = 1 + (end > next->vm_end);
723 VM_WARN_ON(remove_next == 2 &&
724 end != next->vm_next->vm_end);
725 VM_WARN_ON(remove_next == 1 &&
726 end != next->vm_end);
727 /* trim end to next, for case 6 first pass */
728 end = next->vm_end;
729 }
730
287d97ac 731 exporter = next;
1da177e4 732 importer = vma;
734537c9
KS
733
734 /*
735 * If next doesn't have anon_vma, import from vma after
736 * next, if the vma overlaps with it.
737 */
97a42cd4 738 if (remove_next == 2 && !next->anon_vma)
734537c9
KS
739 exporter = next->vm_next;
740
1da177e4
LT
741 } else if (end > next->vm_start) {
742 /*
743 * vma expands, overlapping part of the next:
744 * mprotect case 5 shifting the boundary up.
745 */
746 adjust_next = (end - next->vm_start) >> PAGE_SHIFT;
287d97ac 747 exporter = next;
1da177e4 748 importer = vma;
e86f15ee 749 VM_WARN_ON(expand != importer);
1da177e4
LT
750 } else if (end < vma->vm_end) {
751 /*
752 * vma shrinks, and !insert tells it's not
753 * split_vma inserting another: so it must be
754 * mprotect case 4 shifting the boundary down.
755 */
cc71aba3 756 adjust_next = -((vma->vm_end - end) >> PAGE_SHIFT);
287d97ac 757 exporter = vma;
1da177e4 758 importer = next;
e86f15ee 759 VM_WARN_ON(expand != importer);
1da177e4 760 }
1da177e4 761
5beb4930
RR
762 /*
763 * Easily overlooked: when mprotect shifts the boundary,
764 * make sure the expanding vma has anon_vma set if the
765 * shrinking vma had, to cover any anon pages imported.
766 */
287d97ac 767 if (exporter && exporter->anon_vma && !importer->anon_vma) {
c4ea95d7
DF
768 int error;
769
b800c91a 770 importer->anon_vma = exporter->anon_vma;
c4ea95d7 771 error = anon_vma_clone(importer, exporter);
3fe89b3e 772 if (error)
c4ea95d7 773 return error;
5beb4930
RR
774 }
775 }
734537c9 776again:
e86f15ee 777 vma_adjust_trans_huge(orig_vma, start, end, adjust_next);
37f9f559 778
1da177e4
LT
779 if (file) {
780 mapping = file->f_mapping;
27ba0644
KS
781 root = &mapping->i_mmap;
782 uprobe_munmap(vma, vma->vm_start, vma->vm_end);
682968e0 783
27ba0644
KS
784 if (adjust_next)
785 uprobe_munmap(next, next->vm_start, next->vm_end);
682968e0 786
83cde9e8 787 i_mmap_lock_write(mapping);
1da177e4 788 if (insert) {
1da177e4 789 /*
6b2dbba8 790 * Put into interval tree now, so instantiated pages
1da177e4
LT
791 * are visible to arm/parisc __flush_dcache_page
792 * throughout; but we cannot insert into address
793 * space until vma start or end is updated.
794 */
795 __vma_link_file(insert);
796 }
797 }
798
bf181b9f
ML
799 anon_vma = vma->anon_vma;
800 if (!anon_vma && adjust_next)
801 anon_vma = next->anon_vma;
802 if (anon_vma) {
e86f15ee
AA
803 VM_WARN_ON(adjust_next && next->anon_vma &&
804 anon_vma != next->anon_vma);
4fc3f1d6 805 anon_vma_lock_write(anon_vma);
bf181b9f
ML
806 anon_vma_interval_tree_pre_update_vma(vma);
807 if (adjust_next)
808 anon_vma_interval_tree_pre_update_vma(next);
809 }
012f1800 810
1da177e4
LT
811 if (root) {
812 flush_dcache_mmap_lock(mapping);
6b2dbba8 813 vma_interval_tree_remove(vma, root);
1da177e4 814 if (adjust_next)
6b2dbba8 815 vma_interval_tree_remove(next, root);
1da177e4
LT
816 }
817
d3737187
ML
818 if (start != vma->vm_start) {
819 vma->vm_start = start;
820 start_changed = true;
821 }
822 if (end != vma->vm_end) {
823 vma->vm_end = end;
824 end_changed = true;
825 }
1da177e4
LT
826 vma->vm_pgoff = pgoff;
827 if (adjust_next) {
828 next->vm_start += adjust_next << PAGE_SHIFT;
829 next->vm_pgoff += adjust_next;
830 }
831
832 if (root) {
833 if (adjust_next)
6b2dbba8
ML
834 vma_interval_tree_insert(next, root);
835 vma_interval_tree_insert(vma, root);
1da177e4
LT
836 flush_dcache_mmap_unlock(mapping);
837 }
838
839 if (remove_next) {
840 /*
841 * vma_merge has merged next into vma, and needs
842 * us to remove next before dropping the locks.
843 */
e86f15ee
AA
844 if (remove_next != 3)
845 __vma_unlink_prev(mm, next, vma);
846 else
8f26e0b1
AA
847 /*
848 * vma is not before next if they've been
849 * swapped.
850 *
851 * pre-swap() next->vm_start was reduced so
852 * tell validate_mm_rb to ignore pre-swap()
853 * "next" (which is stored in post-swap()
854 * "vma").
855 */
856 __vma_unlink_common(mm, next, NULL, false, vma);
1da177e4
LT
857 if (file)
858 __remove_shared_vm_struct(next, file, mapping);
1da177e4
LT
859 } else if (insert) {
860 /*
861 * split_vma has split insert from vma, and needs
862 * us to insert it before dropping the locks
863 * (it may either follow vma or precede it).
864 */
865 __insert_vm_struct(mm, insert);
d3737187
ML
866 } else {
867 if (start_changed)
868 vma_gap_update(vma);
869 if (end_changed) {
870 if (!next)
ddeaddb9 871 mm->highest_vm_end = vm_end_gap(vma);
d3737187
ML
872 else if (!adjust_next)
873 vma_gap_update(next);
874 }
1da177e4
LT
875 }
876
bf181b9f
ML
877 if (anon_vma) {
878 anon_vma_interval_tree_post_update_vma(vma);
879 if (adjust_next)
880 anon_vma_interval_tree_post_update_vma(next);
08b52706 881 anon_vma_unlock_write(anon_vma);
bf181b9f 882 }
1da177e4 883 if (mapping)
83cde9e8 884 i_mmap_unlock_write(mapping);
1da177e4 885
2b144498 886 if (root) {
7b2d81d4 887 uprobe_mmap(vma);
2b144498
SD
888
889 if (adjust_next)
7b2d81d4 890 uprobe_mmap(next);
2b144498
SD
891 }
892
1da177e4 893 if (remove_next) {
925d1c40 894 if (file) {
cbc91f71 895 uprobe_munmap(next, next->vm_start, next->vm_end);
e14748e8 896 vma_fput(vma);
925d1c40 897 }
5beb4930
RR
898 if (next->anon_vma)
899 anon_vma_merge(vma, next);
1da177e4 900 mm->map_count--;
3964acd0 901 mpol_put(vma_policy(next));
1da177e4
LT
902 kmem_cache_free(vm_area_cachep, next);
903 /*
904 * In mprotect's case 6 (see comments on vma_merge),
905 * we must remove another next too. It would clutter
906 * up the code too much to do both in one go.
907 */
e86f15ee
AA
908 if (remove_next != 3) {
909 /*
910 * If "next" was removed and vma->vm_end was
911 * expanded (up) over it, in turn
912 * "next->vm_prev->vm_end" changed and the
913 * "vma->vm_next" gap must be updated.
914 */
915 next = vma->vm_next;
916 } else {
917 /*
918 * For the scope of the comment "next" and
919 * "vma" considered pre-swap(): if "vma" was
920 * removed, next->vm_start was expanded (down)
921 * over it and the "next" gap must be updated.
922 * Because of the swap() the post-swap() "vma"
923 * actually points to pre-swap() "next"
924 * (post-swap() "next" as opposed is now a
925 * dangling pointer).
926 */
927 next = vma;
928 }
734537c9
KS
929 if (remove_next == 2) {
930 remove_next = 1;
931 end = next->vm_end;
1da177e4 932 goto again;
734537c9 933 }
d3737187
ML
934 else if (next)
935 vma_gap_update(next);
fb8c41e9
AA
936 else {
937 /*
938 * If remove_next == 2 we obviously can't
939 * reach this path.
940 *
941 * If remove_next == 3 we can't reach this
942 * path because pre-swap() next is always not
943 * NULL. pre-swap() "next" is not being
944 * removed and its next->vm_end is not altered
945 * (and furthermore "end" already matches
946 * next->vm_end in remove_next == 3).
947 *
948 * We reach this only in the remove_next == 1
949 * case if the "next" vma that was removed was
950 * the highest vma of the mm. However in such
951 * case next->vm_end == "end" and the extended
952 * "vma" has vma->vm_end == next->vm_end so
953 * mm->highest_vm_end doesn't need any update
954 * in remove_next == 1 case.
955 */
ddeaddb9 956 VM_WARN_ON(mm->highest_vm_end != vm_end_gap(vma));
fb8c41e9 957 }
1da177e4 958 }
2b144498 959 if (insert && file)
7b2d81d4 960 uprobe_mmap(insert);
1da177e4
LT
961
962 validate_mm(mm);
5beb4930
RR
963
964 return 0;
1da177e4
LT
965}
966
967/*
968 * If the vma has a ->close operation then the driver probably needs to release
969 * per-vma resources, so we don't attempt to merge those.
970 */
1da177e4 971static inline int is_mergeable_vma(struct vm_area_struct *vma,
19a809af
AA
972 struct file *file, unsigned long vm_flags,
973 struct vm_userfaultfd_ctx vm_userfaultfd_ctx)
1da177e4 974{
34228d47
CG
975 /*
976 * VM_SOFTDIRTY should not prevent from VMA merging, if we
977 * match the flags but dirty bit -- the caller should mark
978 * merged VMA as dirty. If dirty bit won't be excluded from
979 * comparison, we increase pressue on the memory system forcing
980 * the kernel to generate new VMAs when old one could be
981 * extended instead.
982 */
983 if ((vma->vm_flags ^ vm_flags) & ~VM_SOFTDIRTY)
1da177e4
LT
984 return 0;
985 if (vma->vm_file != file)
986 return 0;
987 if (vma->vm_ops && vma->vm_ops->close)
988 return 0;
19a809af
AA
989 if (!is_mergeable_vm_userfaultfd_ctx(vma, vm_userfaultfd_ctx))
990 return 0;
1da177e4
LT
991 return 1;
992}
993
994static inline int is_mergeable_anon_vma(struct anon_vma *anon_vma1,
965f55de
SL
995 struct anon_vma *anon_vma2,
996 struct vm_area_struct *vma)
1da177e4 997{
965f55de
SL
998 /*
999 * The list_is_singular() test is to avoid merging VMA cloned from
1000 * parents. This can improve scalability caused by anon_vma lock.
1001 */
1002 if ((!anon_vma1 || !anon_vma2) && (!vma ||
1003 list_is_singular(&vma->anon_vma_chain)))
1004 return 1;
1005 return anon_vma1 == anon_vma2;
1da177e4
LT
1006}
1007
1008/*
1009 * Return true if we can merge this (vm_flags,anon_vma,file,vm_pgoff)
1010 * in front of (at a lower virtual address and file offset than) the vma.
1011 *
1012 * We cannot merge two vmas if they have differently assigned (non-NULL)
1013 * anon_vmas, nor if same anon_vma is assigned but offsets incompatible.
1014 *
1015 * We don't check here for the merged mmap wrapping around the end of pagecache
1016 * indices (16TB on ia32) because do_mmap_pgoff() does not permit mmap's which
1017 * wrap, nor mmaps which cover the final page at index -1UL.
1018 */
1019static int
1020can_vma_merge_before(struct vm_area_struct *vma, unsigned long vm_flags,
19a809af
AA
1021 struct anon_vma *anon_vma, struct file *file,
1022 pgoff_t vm_pgoff,
1023 struct vm_userfaultfd_ctx vm_userfaultfd_ctx)
1da177e4 1024{
19a809af 1025 if (is_mergeable_vma(vma, file, vm_flags, vm_userfaultfd_ctx) &&
965f55de 1026 is_mergeable_anon_vma(anon_vma, vma->anon_vma, vma)) {
1da177e4
LT
1027 if (vma->vm_pgoff == vm_pgoff)
1028 return 1;
1029 }
1030 return 0;
1031}
1032
1033/*
1034 * Return true if we can merge this (vm_flags,anon_vma,file,vm_pgoff)
1035 * beyond (at a higher virtual address and file offset than) the vma.
1036 *
1037 * We cannot merge two vmas if they have differently assigned (non-NULL)
1038 * anon_vmas, nor if same anon_vma is assigned but offsets incompatible.
1039 */
1040static int
1041can_vma_merge_after(struct vm_area_struct *vma, unsigned long vm_flags,
19a809af
AA
1042 struct anon_vma *anon_vma, struct file *file,
1043 pgoff_t vm_pgoff,
1044 struct vm_userfaultfd_ctx vm_userfaultfd_ctx)
1da177e4 1045{
19a809af 1046 if (is_mergeable_vma(vma, file, vm_flags, vm_userfaultfd_ctx) &&
965f55de 1047 is_mergeable_anon_vma(anon_vma, vma->anon_vma, vma)) {
1da177e4 1048 pgoff_t vm_pglen;
d6e93217 1049 vm_pglen = vma_pages(vma);
1da177e4
LT
1050 if (vma->vm_pgoff + vm_pglen == vm_pgoff)
1051 return 1;
1052 }
1053 return 0;
1054}
1055
1056/*
1057 * Given a mapping request (addr,end,vm_flags,file,pgoff), figure out
1058 * whether that can be merged with its predecessor or its successor.
1059 * Or both (it neatly fills a hole).
1060 *
1061 * In most cases - when called for mmap, brk or mremap - [addr,end) is
1062 * certain not to be mapped by the time vma_merge is called; but when
1063 * called for mprotect, it is certain to be already mapped (either at
1064 * an offset within prev, or at the start of next), and the flags of
1065 * this area are about to be changed to vm_flags - and the no-change
1066 * case has already been eliminated.
1067 *
1068 * The following mprotect cases have to be considered, where AAAA is
1069 * the area passed down from mprotect_fixup, never extending beyond one
1070 * vma, PPPPPP is the prev vma specified, and NNNNNN the next vma after:
1071 *
1072 * AAAA AAAA AAAA AAAA
1073 * PPPPPPNNNNNN PPPPPPNNNNNN PPPPPPNNNNNN PPPPNNNNXXXX
1074 * cannot merge might become might become might become
1075 * PPNNNNNNNNNN PPPPPPPPPPNN PPPPPPPPPPPP 6 or
1076 * mmap, brk or case 4 below case 5 below PPPPPPPPXXXX 7 or
e86f15ee 1077 * mremap move: PPPPXXXXXXXX 8
1da177e4
LT
1078 * AAAA
1079 * PPPP NNNN PPPPPPPPPPPP PPPPPPPPNNNN PPPPNNNNNNNN
1080 * might become case 1 below case 2 below case 3 below
1081 *
e86f15ee
AA
1082 * It is important for case 8 that the the vma NNNN overlapping the
1083 * region AAAA is never going to extended over XXXX. Instead XXXX must
1084 * be extended in region AAAA and NNNN must be removed. This way in
1085 * all cases where vma_merge succeeds, the moment vma_adjust drops the
1086 * rmap_locks, the properties of the merged vma will be already
1087 * correct for the whole merged range. Some of those properties like
1088 * vm_page_prot/vm_flags may be accessed by rmap_walks and they must
1089 * be correct for the whole merged range immediately after the
1090 * rmap_locks are released. Otherwise if XXXX would be removed and
1091 * NNNN would be extended over the XXXX range, remove_migration_ptes
1092 * or other rmap walkers (if working on addresses beyond the "end"
1093 * parameter) may establish ptes with the wrong permissions of NNNN
1094 * instead of the right permissions of XXXX.
1da177e4
LT
1095 */
1096struct vm_area_struct *vma_merge(struct mm_struct *mm,
1097 struct vm_area_struct *prev, unsigned long addr,
1098 unsigned long end, unsigned long vm_flags,
cc71aba3 1099 struct anon_vma *anon_vma, struct file *file,
19a809af
AA
1100 pgoff_t pgoff, struct mempolicy *policy,
1101 struct vm_userfaultfd_ctx vm_userfaultfd_ctx)
1da177e4
LT
1102{
1103 pgoff_t pglen = (end - addr) >> PAGE_SHIFT;
1104 struct vm_area_struct *area, *next;
5beb4930 1105 int err;
1da177e4
LT
1106
1107 /*
1108 * We later require that vma->vm_flags == vm_flags,
1109 * so this tests vma->vm_flags & VM_SPECIAL, too.
1110 */
1111 if (vm_flags & VM_SPECIAL)
1112 return NULL;
1113
1114 if (prev)
1115 next = prev->vm_next;
1116 else
1117 next = mm->mmap;
1118 area = next;
e86f15ee 1119 if (area && area->vm_end == end) /* cases 6, 7, 8 */
1da177e4
LT
1120 next = next->vm_next;
1121
e86f15ee
AA
1122 /* verify some invariant that must be enforced by the caller */
1123 VM_WARN_ON(prev && addr <= prev->vm_start);
1124 VM_WARN_ON(area && end > area->vm_end);
1125 VM_WARN_ON(addr >= end);
1126
1da177e4
LT
1127 /*
1128 * Can it merge with the predecessor?
1129 */
1130 if (prev && prev->vm_end == addr &&
cc71aba3 1131 mpol_equal(vma_policy(prev), policy) &&
1da177e4 1132 can_vma_merge_after(prev, vm_flags,
19a809af
AA
1133 anon_vma, file, pgoff,
1134 vm_userfaultfd_ctx)) {
1da177e4
LT
1135 /*
1136 * OK, it can. Can we now merge in the successor as well?
1137 */
1138 if (next && end == next->vm_start &&
1139 mpol_equal(policy, vma_policy(next)) &&
1140 can_vma_merge_before(next, vm_flags,
19a809af
AA
1141 anon_vma, file,
1142 pgoff+pglen,
1143 vm_userfaultfd_ctx) &&
1da177e4 1144 is_mergeable_anon_vma(prev->anon_vma,
965f55de 1145 next->anon_vma, NULL)) {
1da177e4 1146 /* cases 1, 6 */
e86f15ee
AA
1147 err = __vma_adjust(prev, prev->vm_start,
1148 next->vm_end, prev->vm_pgoff, NULL,
1149 prev);
1da177e4 1150 } else /* cases 2, 5, 7 */
e86f15ee
AA
1151 err = __vma_adjust(prev, prev->vm_start,
1152 end, prev->vm_pgoff, NULL, prev);
5beb4930
RR
1153 if (err)
1154 return NULL;
6d50e60c 1155 khugepaged_enter_vma_merge(prev, vm_flags);
1da177e4
LT
1156 return prev;
1157 }
1158
1159 /*
1160 * Can this new request be merged in front of next?
1161 */
1162 if (next && end == next->vm_start &&
cc71aba3 1163 mpol_equal(policy, vma_policy(next)) &&
1da177e4 1164 can_vma_merge_before(next, vm_flags,
19a809af
AA
1165 anon_vma, file, pgoff+pglen,
1166 vm_userfaultfd_ctx)) {
1da177e4 1167 if (prev && addr < prev->vm_end) /* case 4 */
e86f15ee
AA
1168 err = __vma_adjust(prev, prev->vm_start,
1169 addr, prev->vm_pgoff, NULL, next);
1170 else { /* cases 3, 8 */
1171 err = __vma_adjust(area, addr, next->vm_end,
1172 next->vm_pgoff - pglen, NULL, next);
1173 /*
1174 * In case 3 area is already equal to next and
1175 * this is a noop, but in case 8 "area" has
1176 * been removed and next was expanded over it.
1177 */
1178 area = next;
1179 }
5beb4930
RR
1180 if (err)
1181 return NULL;
6d50e60c 1182 khugepaged_enter_vma_merge(area, vm_flags);
1da177e4
LT
1183 return area;
1184 }
1185
1186 return NULL;
1187}
1188
d0e9fe17
LT
1189/*
1190 * Rough compatbility check to quickly see if it's even worth looking
1191 * at sharing an anon_vma.
1192 *
1193 * They need to have the same vm_file, and the flags can only differ
1194 * in things that mprotect may change.
1195 *
1196 * NOTE! The fact that we share an anon_vma doesn't _have_ to mean that
1197 * we can merge the two vma's. For example, we refuse to merge a vma if
1198 * there is a vm_ops->close() function, because that indicates that the
1199 * driver is doing some kind of reference counting. But that doesn't
1200 * really matter for the anon_vma sharing case.
1201 */
1202static int anon_vma_compatible(struct vm_area_struct *a, struct vm_area_struct *b)
1203{
1204 return a->vm_end == b->vm_start &&
1205 mpol_equal(vma_policy(a), vma_policy(b)) &&
1206 a->vm_file == b->vm_file &&
34228d47 1207 !((a->vm_flags ^ b->vm_flags) & ~(VM_READ|VM_WRITE|VM_EXEC|VM_SOFTDIRTY)) &&
d0e9fe17
LT
1208 b->vm_pgoff == a->vm_pgoff + ((b->vm_start - a->vm_start) >> PAGE_SHIFT);
1209}
1210
1211/*
1212 * Do some basic sanity checking to see if we can re-use the anon_vma
1213 * from 'old'. The 'a'/'b' vma's are in VM order - one of them will be
1214 * the same as 'old', the other will be the new one that is trying
1215 * to share the anon_vma.
1216 *
1217 * NOTE! This runs with mm_sem held for reading, so it is possible that
1218 * the anon_vma of 'old' is concurrently in the process of being set up
1219 * by another page fault trying to merge _that_. But that's ok: if it
1220 * is being set up, that automatically means that it will be a singleton
1221 * acceptable for merging, so we can do all of this optimistically. But
4db0c3c2 1222 * we do that READ_ONCE() to make sure that we never re-load the pointer.
d0e9fe17
LT
1223 *
1224 * IOW: that the "list_is_singular()" test on the anon_vma_chain only
1225 * matters for the 'stable anon_vma' case (ie the thing we want to avoid
1226 * is to return an anon_vma that is "complex" due to having gone through
1227 * a fork).
1228 *
1229 * We also make sure that the two vma's are compatible (adjacent,
1230 * and with the same memory policies). That's all stable, even with just
1231 * a read lock on the mm_sem.
1232 */
1233static struct anon_vma *reusable_anon_vma(struct vm_area_struct *old, struct vm_area_struct *a, struct vm_area_struct *b)
1234{
1235 if (anon_vma_compatible(a, b)) {
4db0c3c2 1236 struct anon_vma *anon_vma = READ_ONCE(old->anon_vma);
d0e9fe17
LT
1237
1238 if (anon_vma && list_is_singular(&old->anon_vma_chain))
1239 return anon_vma;
1240 }
1241 return NULL;
1242}
1243
1da177e4
LT
1244/*
1245 * find_mergeable_anon_vma is used by anon_vma_prepare, to check
1246 * neighbouring vmas for a suitable anon_vma, before it goes off
1247 * to allocate a new anon_vma. It checks because a repetitive
1248 * sequence of mprotects and faults may otherwise lead to distinct
1249 * anon_vmas being allocated, preventing vma merge in subsequent
1250 * mprotect.
1251 */
1252struct anon_vma *find_mergeable_anon_vma(struct vm_area_struct *vma)
1253{
d0e9fe17 1254 struct anon_vma *anon_vma;
1da177e4 1255 struct vm_area_struct *near;
1da177e4
LT
1256
1257 near = vma->vm_next;
1258 if (!near)
1259 goto try_prev;
1260
d0e9fe17
LT
1261 anon_vma = reusable_anon_vma(near, vma, near);
1262 if (anon_vma)
1263 return anon_vma;
1da177e4 1264try_prev:
9be34c9d 1265 near = vma->vm_prev;
1da177e4
LT
1266 if (!near)
1267 goto none;
1268
d0e9fe17
LT
1269 anon_vma = reusable_anon_vma(near, near, vma);
1270 if (anon_vma)
1271 return anon_vma;
1da177e4
LT
1272none:
1273 /*
1274 * There's no absolute need to look only at touching neighbours:
1275 * we could search further afield for "compatible" anon_vmas.
1276 * But it would probably just be a waste of time searching,
1277 * or lead to too many vmas hanging off the same anon_vma.
1278 * We're trying to allow mprotect remerging later on,
1279 * not trying to minimize memory used for anon_vmas.
1280 */
1281 return NULL;
1282}
1283
40401530
AV
1284/*
1285 * If a hint addr is less than mmap_min_addr change hint to be as
1286 * low as possible but still greater than mmap_min_addr
1287 */
1288static inline unsigned long round_hint_to_min(unsigned long hint)
1289{
1290 hint &= PAGE_MASK;
1291 if (((void *)hint != NULL) &&
1292 (hint < mmap_min_addr))
1293 return PAGE_ALIGN(mmap_min_addr);
1294 return hint;
1295}
1296
363ee17f
DB
1297static inline int mlock_future_check(struct mm_struct *mm,
1298 unsigned long flags,
1299 unsigned long len)
1300{
1301 unsigned long locked, lock_limit;
1302
1303 /* mlock MCL_FUTURE? */
1304 if (flags & VM_LOCKED) {
1305 locked = len >> PAGE_SHIFT;
1306 locked += mm->locked_vm;
1307 lock_limit = rlimit(RLIMIT_MEMLOCK);
1308 lock_limit >>= PAGE_SHIFT;
1309 if (locked > lock_limit && !capable(CAP_IPC_LOCK))
1310 return -EAGAIN;
1311 }
1312 return 0;
1313}
1314
1da177e4 1315/*
27f5de79 1316 * The caller must hold down_write(&current->mm->mmap_sem).
1da177e4 1317 */
1fcfd8db 1318unsigned long do_mmap(struct file *file, unsigned long addr,
1da177e4 1319 unsigned long len, unsigned long prot,
1fcfd8db
ON
1320 unsigned long flags, vm_flags_t vm_flags,
1321 unsigned long pgoff, unsigned long *populate)
1da177e4 1322{
cc71aba3 1323 struct mm_struct *mm = current->mm;
62b5f7d0 1324 int pkey = 0;
1da177e4 1325
41badc15 1326 *populate = 0;
bebeb3d6 1327
e37609bb
PK
1328 if (!len)
1329 return -EINVAL;
1330
1da177e4
LT
1331 /*
1332 * Does the application expect PROT_READ to imply PROT_EXEC?
1333 *
1334 * (the exception is when the underlying filesystem is noexec
1335 * mounted, in which case we dont add PROT_EXEC.)
1336 */
1337 if ((prot & PROT_READ) && (current->personality & READ_IMPLIES_EXEC))
90f8572b 1338 if (!(file && path_noexec(&file->f_path)))
1da177e4
LT
1339 prot |= PROT_EXEC;
1340
7cd94146
EP
1341 if (!(flags & MAP_FIXED))
1342 addr = round_hint_to_min(addr);
1343
1da177e4
LT
1344 /* Careful about overflows.. */
1345 len = PAGE_ALIGN(len);
9206de95 1346 if (!len)
1da177e4
LT
1347 return -ENOMEM;
1348
1349 /* offset overflow? */
1350 if ((pgoff + (len >> PAGE_SHIFT)) < pgoff)
cc71aba3 1351 return -EOVERFLOW;
1da177e4
LT
1352
1353 /* Too many mappings? */
1354 if (mm->map_count > sysctl_max_map_count)
1355 return -ENOMEM;
1356
1357 /* Obtain the address to map to. we verify (or select) it and ensure
1358 * that it represents a valid section of the address space.
1359 */
1360 addr = get_unmapped_area(file, addr, len, pgoff, flags);
de1741a1 1361 if (offset_in_page(addr))
1da177e4
LT
1362 return addr;
1363
62b5f7d0
DH
1364 if (prot == PROT_EXEC) {
1365 pkey = execute_only_pkey(mm);
1366 if (pkey < 0)
1367 pkey = 0;
1368 }
1369
1da177e4
LT
1370 /* Do simple checking here so the lower-level routines won't have
1371 * to. we assume access permissions have been handled by the open
1372 * of the memory object, so we don't do any here.
1373 */
62b5f7d0 1374 vm_flags |= calc_vm_prot_bits(prot, pkey) | calc_vm_flag_bits(flags) |
1da177e4
LT
1375 mm->def_flags | VM_MAYREAD | VM_MAYWRITE | VM_MAYEXEC;
1376
cdf7b341 1377 if (flags & MAP_LOCKED)
1da177e4
LT
1378 if (!can_do_mlock())
1379 return -EPERM;
ba470de4 1380
363ee17f
DB
1381 if (mlock_future_check(mm, vm_flags, len))
1382 return -EAGAIN;
1da177e4 1383
1da177e4 1384 if (file) {
077bf22b
ON
1385 struct inode *inode = file_inode(file);
1386
1da177e4
LT
1387 switch (flags & MAP_TYPE) {
1388 case MAP_SHARED:
1389 if ((prot&PROT_WRITE) && !(file->f_mode&FMODE_WRITE))
1390 return -EACCES;
1391
1392 /*
1393 * Make sure we don't allow writing to an append-only
1394 * file..
1395 */
1396 if (IS_APPEND(inode) && (file->f_mode & FMODE_WRITE))
1397 return -EACCES;
1398
1399 /*
1400 * Make sure there are no mandatory locks on the file.
1401 */
d7a06983 1402 if (locks_verify_locked(file))
1da177e4
LT
1403 return -EAGAIN;
1404
1405 vm_flags |= VM_SHARED | VM_MAYSHARE;
1406 if (!(file->f_mode & FMODE_WRITE))
1407 vm_flags &= ~(VM_MAYWRITE | VM_SHARED);
1408
1409 /* fall through */
1410 case MAP_PRIVATE:
1411 if (!(file->f_mode & FMODE_READ))
1412 return -EACCES;
90f8572b 1413 if (path_noexec(&file->f_path)) {
80c5606c
LT
1414 if (vm_flags & VM_EXEC)
1415 return -EPERM;
1416 vm_flags &= ~VM_MAYEXEC;
1417 }
80c5606c 1418
72c2d531 1419 if (!file->f_op->mmap)
80c5606c 1420 return -ENODEV;
b2c56e4f
ON
1421 if (vm_flags & (VM_GROWSDOWN|VM_GROWSUP))
1422 return -EINVAL;
1da177e4
LT
1423 break;
1424
1425 default:
1426 return -EINVAL;
1427 }
1428 } else {
1429 switch (flags & MAP_TYPE) {
1430 case MAP_SHARED:
b2c56e4f
ON
1431 if (vm_flags & (VM_GROWSDOWN|VM_GROWSUP))
1432 return -EINVAL;
ce363942
TH
1433 /*
1434 * Ignore pgoff.
1435 */
1436 pgoff = 0;
1da177e4
LT
1437 vm_flags |= VM_SHARED | VM_MAYSHARE;
1438 break;
1439 case MAP_PRIVATE:
1440 /*
1441 * Set pgoff according to addr for anon_vma.
1442 */
1443 pgoff = addr >> PAGE_SHIFT;
1444 break;
1445 default:
1446 return -EINVAL;
1447 }
1448 }
1449
c22c0d63
ML
1450 /*
1451 * Set 'VM_NORESERVE' if we should not account for the
1452 * memory use of this mapping.
1453 */
1454 if (flags & MAP_NORESERVE) {
1455 /* We honor MAP_NORESERVE if allowed to overcommit */
1456 if (sysctl_overcommit_memory != OVERCOMMIT_NEVER)
1457 vm_flags |= VM_NORESERVE;
1458
1459 /* hugetlb applies strict overcommit unless MAP_NORESERVE */
1460 if (file && is_file_hugepages(file))
1461 vm_flags |= VM_NORESERVE;
1462 }
1463
1464 addr = mmap_region(file, addr, len, vm_flags, pgoff);
09a9f1d2
ML
1465 if (!IS_ERR_VALUE(addr) &&
1466 ((vm_flags & VM_LOCKED) ||
1467 (flags & (MAP_POPULATE | MAP_NONBLOCK)) == MAP_POPULATE))
41badc15 1468 *populate = len;
bebeb3d6 1469 return addr;
0165ab44 1470}
6be5ceb0 1471
66f0dc48
HD
1472SYSCALL_DEFINE6(mmap_pgoff, unsigned long, addr, unsigned long, len,
1473 unsigned long, prot, unsigned long, flags,
1474 unsigned long, fd, unsigned long, pgoff)
1475{
1476 struct file *file = NULL;
1e3ee14b 1477 unsigned long retval;
66f0dc48
HD
1478
1479 if (!(flags & MAP_ANONYMOUS)) {
120a795d 1480 audit_mmap_fd(fd, flags);
66f0dc48
HD
1481 file = fget(fd);
1482 if (!file)
1e3ee14b 1483 return -EBADF;
af73e4d9
NH
1484 if (is_file_hugepages(file))
1485 len = ALIGN(len, huge_page_size(hstate_file(file)));
493af578
JE
1486 retval = -EINVAL;
1487 if (unlikely(flags & MAP_HUGETLB && !is_file_hugepages(file)))
1488 goto out_fput;
66f0dc48
HD
1489 } else if (flags & MAP_HUGETLB) {
1490 struct user_struct *user = NULL;
c103a4dc 1491 struct hstate *hs;
af73e4d9 1492
c103a4dc 1493 hs = hstate_sizelog((flags >> MAP_HUGE_SHIFT) & SHM_HUGE_MASK);
091d0d55
LZ
1494 if (!hs)
1495 return -EINVAL;
1496
1497 len = ALIGN(len, huge_page_size(hs));
66f0dc48
HD
1498 /*
1499 * VM_NORESERVE is used because the reservations will be
1500 * taken when vm_ops->mmap() is called
1501 * A dummy user value is used because we are not locking
1502 * memory so no accounting is necessary
1503 */
af73e4d9 1504 file = hugetlb_file_setup(HUGETLB_ANON_FILE, len,
42d7395f
AK
1505 VM_NORESERVE,
1506 &user, HUGETLB_ANONHUGE_INODE,
1507 (flags >> MAP_HUGE_SHIFT) & MAP_HUGE_MASK);
66f0dc48
HD
1508 if (IS_ERR(file))
1509 return PTR_ERR(file);
1510 }
1511
1512 flags &= ~(MAP_EXECUTABLE | MAP_DENYWRITE);
1513
9fbeb5ab 1514 retval = vm_mmap_pgoff(file, addr, len, prot, flags, pgoff);
493af578 1515out_fput:
66f0dc48
HD
1516 if (file)
1517 fput(file);
66f0dc48
HD
1518 return retval;
1519}
1520
a4679373
CH
1521#ifdef __ARCH_WANT_SYS_OLD_MMAP
1522struct mmap_arg_struct {
1523 unsigned long addr;
1524 unsigned long len;
1525 unsigned long prot;
1526 unsigned long flags;
1527 unsigned long fd;
1528 unsigned long offset;
1529};
1530
1531SYSCALL_DEFINE1(old_mmap, struct mmap_arg_struct __user *, arg)
1532{
1533 struct mmap_arg_struct a;
1534
1535 if (copy_from_user(&a, arg, sizeof(a)))
1536 return -EFAULT;
de1741a1 1537 if (offset_in_page(a.offset))
a4679373
CH
1538 return -EINVAL;
1539
1540 return sys_mmap_pgoff(a.addr, a.len, a.prot, a.flags, a.fd,
1541 a.offset >> PAGE_SHIFT);
1542}
1543#endif /* __ARCH_WANT_SYS_OLD_MMAP */
1544
4e950f6f
AD
1545/*
1546 * Some shared mappigns will want the pages marked read-only
1547 * to track write events. If so, we'll downgrade vm_page_prot
1548 * to the private version (using protection_map[] without the
1549 * VM_SHARED bit).
1550 */
6d2329f8 1551int vma_wants_writenotify(struct vm_area_struct *vma, pgprot_t vm_page_prot)
4e950f6f 1552{
ca16d140 1553 vm_flags_t vm_flags = vma->vm_flags;
8a04446a 1554 const struct vm_operations_struct *vm_ops = vma->vm_ops;
4e950f6f
AD
1555
1556 /* If it was private or non-writable, the write bit is already clear */
1557 if ((vm_flags & (VM_WRITE|VM_SHARED)) != ((VM_WRITE|VM_SHARED)))
1558 return 0;
1559
1560 /* The backer wishes to know when pages are first written to? */
8a04446a 1561 if (vm_ops && (vm_ops->page_mkwrite || vm_ops->pfn_mkwrite))
4e950f6f
AD
1562 return 1;
1563
64e45507
PF
1564 /* The open routine did something to the protections that pgprot_modify
1565 * won't preserve? */
6d2329f8
AA
1566 if (pgprot_val(vm_page_prot) !=
1567 pgprot_val(vm_pgprot_modify(vm_page_prot, vm_flags)))
4e950f6f
AD
1568 return 0;
1569
64e45507
PF
1570 /* Do we need to track softdirty? */
1571 if (IS_ENABLED(CONFIG_MEM_SOFT_DIRTY) && !(vm_flags & VM_SOFTDIRTY))
1572 return 1;
1573
4e950f6f 1574 /* Specialty mapping? */
4b6e1e37 1575 if (vm_flags & VM_PFNMAP)
4e950f6f
AD
1576 return 0;
1577
1578 /* Can the mapping track the dirty pages? */
1579 return vma->vm_file && vma->vm_file->f_mapping &&
1580 mapping_cap_account_dirty(vma->vm_file->f_mapping);
1581}
1582
fc8744ad
LT
1583/*
1584 * We account for memory if it's a private writeable mapping,
5a6fe125 1585 * not hugepages and VM_NORESERVE wasn't set.
fc8744ad 1586 */
ca16d140 1587static inline int accountable_mapping(struct file *file, vm_flags_t vm_flags)
fc8744ad 1588{
5a6fe125
MG
1589 /*
1590 * hugetlb has its own accounting separate from the core VM
1591 * VM_HUGETLB may not be set yet so we cannot check for that flag.
1592 */
1593 if (file && is_file_hugepages(file))
1594 return 0;
1595
fc8744ad
LT
1596 return (vm_flags & (VM_NORESERVE | VM_SHARED | VM_WRITE)) == VM_WRITE;
1597}
1598
0165ab44 1599unsigned long mmap_region(struct file *file, unsigned long addr,
c22c0d63 1600 unsigned long len, vm_flags_t vm_flags, unsigned long pgoff)
0165ab44
MS
1601{
1602 struct mm_struct *mm = current->mm;
1603 struct vm_area_struct *vma, *prev;
0165ab44
MS
1604 int error;
1605 struct rb_node **rb_link, *rb_parent;
1606 unsigned long charged = 0;
0165ab44 1607
e8420a8e 1608 /* Check against address space limit. */
84638335 1609 if (!may_expand_vm(mm, vm_flags, len >> PAGE_SHIFT)) {
e8420a8e
CH
1610 unsigned long nr_pages;
1611
1612 /*
1613 * MAP_FIXED may remove pages of mappings that intersects with
1614 * requested mapping. Account for the pages it would unmap.
1615 */
e8420a8e
CH
1616 nr_pages = count_vma_pages_range(mm, addr, addr + len);
1617
84638335
KK
1618 if (!may_expand_vm(mm, vm_flags,
1619 (len >> PAGE_SHIFT) - nr_pages))
e8420a8e
CH
1620 return -ENOMEM;
1621 }
1622
1da177e4 1623 /* Clear old maps */
9fcd1457
RV
1624 while (find_vma_links(mm, addr, addr + len, &prev, &rb_link,
1625 &rb_parent)) {
1da177e4
LT
1626 if (do_munmap(mm, addr, len))
1627 return -ENOMEM;
1da177e4
LT
1628 }
1629
fc8744ad
LT
1630 /*
1631 * Private writable mapping: check memory availability
1632 */
5a6fe125 1633 if (accountable_mapping(file, vm_flags)) {
fc8744ad 1634 charged = len >> PAGE_SHIFT;
191c5424 1635 if (security_vm_enough_memory_mm(mm, charged))
fc8744ad
LT
1636 return -ENOMEM;
1637 vm_flags |= VM_ACCOUNT;
1da177e4
LT
1638 }
1639
1640 /*
de33c8db 1641 * Can we just expand an old mapping?
1da177e4 1642 */
19a809af
AA
1643 vma = vma_merge(mm, prev, addr, addr + len, vm_flags,
1644 NULL, file, pgoff, NULL, NULL_VM_UFFD_CTX);
de33c8db
LT
1645 if (vma)
1646 goto out;
1da177e4
LT
1647
1648 /*
1649 * Determine the object being mapped and call the appropriate
1650 * specific mapper. the address has already been validated, but
1651 * not unmapped, but the maps are removed from the list.
1652 */
c5e3b83e 1653 vma = kmem_cache_zalloc(vm_area_cachep, GFP_KERNEL);
1da177e4
LT
1654 if (!vma) {
1655 error = -ENOMEM;
1656 goto unacct_error;
1657 }
1da177e4
LT
1658
1659 vma->vm_mm = mm;
1660 vma->vm_start = addr;
1661 vma->vm_end = addr + len;
1662 vma->vm_flags = vm_flags;
3ed75eb8 1663 vma->vm_page_prot = vm_get_page_prot(vm_flags);
1da177e4 1664 vma->vm_pgoff = pgoff;
5beb4930 1665 INIT_LIST_HEAD(&vma->anon_vma_chain);
1da177e4
LT
1666
1667 if (file) {
1da177e4
LT
1668 if (vm_flags & VM_DENYWRITE) {
1669 error = deny_write_access(file);
1670 if (error)
1671 goto free_vma;
1da177e4 1672 }
4bb5f5d9
DH
1673 if (vm_flags & VM_SHARED) {
1674 error = mapping_map_writable(file->f_mapping);
1675 if (error)
1676 goto allow_write_and_free_vma;
1677 }
1678
1679 /* ->mmap() can change vma->vm_file, but must guarantee that
1680 * vma_link() below can deny write-access if VM_DENYWRITE is set
1681 * and map writably if VM_SHARED is set. This usually means the
1682 * new file must not have been exposed to user-space, yet.
1683 */
cb0942b8 1684 vma->vm_file = get_file(file);
1da177e4
LT
1685 error = file->f_op->mmap(file, vma);
1686 if (error)
1687 goto unmap_and_free_vma;
f8dbf0a7
HS
1688
1689 /* Can addr have changed??
1690 *
1691 * Answer: Yes, several device drivers can do it in their
1692 * f_op->mmap method. -DaveM
2897b4d2
JK
1693 * Bug: If addr is changed, prev, rb_link, rb_parent should
1694 * be updated for vma_link()
f8dbf0a7 1695 */
2897b4d2
JK
1696 WARN_ON_ONCE(addr != vma->vm_start);
1697
f8dbf0a7 1698 addr = vma->vm_start;
f8dbf0a7 1699 vm_flags = vma->vm_flags;
1da177e4
LT
1700 } else if (vm_flags & VM_SHARED) {
1701 error = shmem_zero_setup(vma);
1702 if (error)
1703 goto free_vma;
1704 }
1705
de33c8db 1706 vma_link(mm, vma, prev, rb_link, rb_parent);
4d3d5b41 1707 /* Once vma denies write, undo our temporary denial count */
4bb5f5d9
DH
1708 if (file) {
1709 if (vm_flags & VM_SHARED)
1710 mapping_unmap_writable(file->f_mapping);
1711 if (vm_flags & VM_DENYWRITE)
1712 allow_write_access(file);
1713 }
e8686772 1714 file = vma->vm_file;
4d3d5b41 1715out:
cdd6c482 1716 perf_event_mmap(vma);
0a4a9391 1717
84638335 1718 vm_stat_account(mm, vm_flags, len >> PAGE_SHIFT);
1da177e4 1719 if (vm_flags & VM_LOCKED) {
bebeb3d6
ML
1720 if (!((vm_flags & VM_SPECIAL) || is_vm_hugetlb_page(vma) ||
1721 vma == get_gate_vma(current->mm)))
06f9d8c2 1722 mm->locked_vm += (len >> PAGE_SHIFT);
bebeb3d6 1723 else
de60f5f1 1724 vma->vm_flags &= VM_LOCKED_CLEAR_MASK;
bebeb3d6 1725 }
2b144498 1726
c7a3a88c
ON
1727 if (file)
1728 uprobe_mmap(vma);
2b144498 1729
d9104d1c
CG
1730 /*
1731 * New (or expanded) vma always get soft dirty status.
1732 * Otherwise user-space soft-dirty page tracker won't
1733 * be able to distinguish situation when vma area unmapped,
1734 * then new mapped in-place (which must be aimed as
1735 * a completely new data area).
1736 */
1737 vma->vm_flags |= VM_SOFTDIRTY;
1738
64e45507
PF
1739 vma_set_page_prot(vma);
1740
1da177e4
LT
1741 return addr;
1742
1743unmap_and_free_vma:
e14748e8 1744 vma_fput(vma);
1da177e4 1745 vma->vm_file = NULL;
1da177e4
LT
1746
1747 /* Undo any partial mapping done by a device driver. */
e0da382c
HD
1748 unmap_region(mm, vma, prev, vma->vm_start, vma->vm_end);
1749 charged = 0;
4bb5f5d9
DH
1750 if (vm_flags & VM_SHARED)
1751 mapping_unmap_writable(file->f_mapping);
1752allow_write_and_free_vma:
1753 if (vm_flags & VM_DENYWRITE)
1754 allow_write_access(file);
1da177e4
LT
1755free_vma:
1756 kmem_cache_free(vm_area_cachep, vma);
1757unacct_error:
1758 if (charged)
1759 vm_unacct_memory(charged);
1760 return error;
1761}
1762
db4fbfb9
ML
1763unsigned long unmapped_area(struct vm_unmapped_area_info *info)
1764{
1765 /*
1766 * We implement the search by looking for an rbtree node that
1767 * immediately follows a suitable gap. That is,
1768 * - gap_start = vma->vm_prev->vm_end <= info->high_limit - length;
1769 * - gap_end = vma->vm_start >= info->low_limit + length;
1770 * - gap_end - gap_start >= length
1771 */
1772
1773 struct mm_struct *mm = current->mm;
1774 struct vm_area_struct *vma;
1775 unsigned long length, low_limit, high_limit, gap_start, gap_end;
1776
1777 /* Adjust search length to account for worst case alignment overhead */
1778 length = info->length + info->align_mask;
1779 if (length < info->length)
1780 return -ENOMEM;
1781
1782 /* Adjust search limits by the desired length */
1783 if (info->high_limit < length)
1784 return -ENOMEM;
1785 high_limit = info->high_limit - length;
1786
1787 if (info->low_limit > high_limit)
1788 return -ENOMEM;
1789 low_limit = info->low_limit + length;
1790
1791 /* Check if rbtree root looks promising */
1792 if (RB_EMPTY_ROOT(&mm->mm_rb))
1793 goto check_highest;
1794 vma = rb_entry(mm->mm_rb.rb_node, struct vm_area_struct, vm_rb);
1795 if (vma->rb_subtree_gap < length)
1796 goto check_highest;
1797
1798 while (true) {
1799 /* Visit left subtree if it looks promising */
ddeaddb9 1800 gap_end = vm_start_gap(vma);
db4fbfb9
ML
1801 if (gap_end >= low_limit && vma->vm_rb.rb_left) {
1802 struct vm_area_struct *left =
1803 rb_entry(vma->vm_rb.rb_left,
1804 struct vm_area_struct, vm_rb);
1805 if (left->rb_subtree_gap >= length) {
1806 vma = left;
1807 continue;
1808 }
1809 }
1810
ddeaddb9 1811 gap_start = vma->vm_prev ? vm_end_gap(vma->vm_prev) : 0;
db4fbfb9
ML
1812check_current:
1813 /* Check if current node has a suitable gap */
1814 if (gap_start > high_limit)
1815 return -ENOMEM;
25bca944
HD
1816 if (gap_end >= low_limit &&
1817 gap_end > gap_start && gap_end - gap_start >= length)
db4fbfb9
ML
1818 goto found;
1819
1820 /* Visit right subtree if it looks promising */
1821 if (vma->vm_rb.rb_right) {
1822 struct vm_area_struct *right =
1823 rb_entry(vma->vm_rb.rb_right,
1824 struct vm_area_struct, vm_rb);
1825 if (right->rb_subtree_gap >= length) {
1826 vma = right;
1827 continue;
1828 }
1829 }
1830
1831 /* Go back up the rbtree to find next candidate node */
1832 while (true) {
1833 struct rb_node *prev = &vma->vm_rb;
1834 if (!rb_parent(prev))
1835 goto check_highest;
1836 vma = rb_entry(rb_parent(prev),
1837 struct vm_area_struct, vm_rb);
1838 if (prev == vma->vm_rb.rb_left) {
ddeaddb9
HD
1839 gap_start = vm_end_gap(vma->vm_prev);
1840 gap_end = vm_start_gap(vma);
db4fbfb9
ML
1841 goto check_current;
1842 }
1843 }
1844 }
1845
1846check_highest:
1847 /* Check highest gap, which does not precede any rbtree node */
1848 gap_start = mm->highest_vm_end;
1849 gap_end = ULONG_MAX; /* Only for VM_BUG_ON below */
1850 if (gap_start > high_limit)
1851 return -ENOMEM;
1852
1853found:
1854 /* We found a suitable gap. Clip it with the original low_limit. */
1855 if (gap_start < info->low_limit)
1856 gap_start = info->low_limit;
1857
1858 /* Adjust gap address to the desired alignment */
1859 gap_start += (info->align_offset - gap_start) & info->align_mask;
1860
1861 VM_BUG_ON(gap_start + info->length > info->high_limit);
1862 VM_BUG_ON(gap_start + info->length > gap_end);
1863 return gap_start;
1864}
1865
1866unsigned long unmapped_area_topdown(struct vm_unmapped_area_info *info)
1867{
1868 struct mm_struct *mm = current->mm;
1869 struct vm_area_struct *vma;
1870 unsigned long length, low_limit, high_limit, gap_start, gap_end;
1871
1872 /* Adjust search length to account for worst case alignment overhead */
1873 length = info->length + info->align_mask;
1874 if (length < info->length)
1875 return -ENOMEM;
1876
1877 /*
1878 * Adjust search limits by the desired length.
1879 * See implementation comment at top of unmapped_area().
1880 */
1881 gap_end = info->high_limit;
1882 if (gap_end < length)
1883 return -ENOMEM;
1884 high_limit = gap_end - length;
1885
1886 if (info->low_limit > high_limit)
1887 return -ENOMEM;
1888 low_limit = info->low_limit + length;
1889
1890 /* Check highest gap, which does not precede any rbtree node */
1891 gap_start = mm->highest_vm_end;
1892 if (gap_start <= high_limit)
1893 goto found_highest;
1894
1895 /* Check if rbtree root looks promising */
1896 if (RB_EMPTY_ROOT(&mm->mm_rb))
1897 return -ENOMEM;
1898 vma = rb_entry(mm->mm_rb.rb_node, struct vm_area_struct, vm_rb);
1899 if (vma->rb_subtree_gap < length)
1900 return -ENOMEM;
1901
1902 while (true) {
1903 /* Visit right subtree if it looks promising */
ddeaddb9 1904 gap_start = vma->vm_prev ? vm_end_gap(vma->vm_prev) : 0;
db4fbfb9
ML
1905 if (gap_start <= high_limit && vma->vm_rb.rb_right) {
1906 struct vm_area_struct *right =
1907 rb_entry(vma->vm_rb.rb_right,
1908 struct vm_area_struct, vm_rb);
1909 if (right->rb_subtree_gap >= length) {
1910 vma = right;
1911 continue;
1912 }
1913 }
1914
1915check_current:
1916 /* Check if current node has a suitable gap */
ddeaddb9 1917 gap_end = vm_start_gap(vma);
db4fbfb9
ML
1918 if (gap_end < low_limit)
1919 return -ENOMEM;
25bca944
HD
1920 if (gap_start <= high_limit &&
1921 gap_end > gap_start && gap_end - gap_start >= length)
db4fbfb9
ML
1922 goto found;
1923
1924 /* Visit left subtree if it looks promising */
1925 if (vma->vm_rb.rb_left) {
1926 struct vm_area_struct *left =
1927 rb_entry(vma->vm_rb.rb_left,
1928 struct vm_area_struct, vm_rb);
1929 if (left->rb_subtree_gap >= length) {
1930 vma = left;
1931 continue;
1932 }
1933 }
1934
1935 /* Go back up the rbtree to find next candidate node */
1936 while (true) {
1937 struct rb_node *prev = &vma->vm_rb;
1938 if (!rb_parent(prev))
1939 return -ENOMEM;
1940 vma = rb_entry(rb_parent(prev),
1941 struct vm_area_struct, vm_rb);
1942 if (prev == vma->vm_rb.rb_right) {
1943 gap_start = vma->vm_prev ?
ddeaddb9 1944 vm_end_gap(vma->vm_prev) : 0;
db4fbfb9
ML
1945 goto check_current;
1946 }
1947 }
1948 }
1949
1950found:
1951 /* We found a suitable gap. Clip it with the original high_limit. */
1952 if (gap_end > info->high_limit)
1953 gap_end = info->high_limit;
1954
1955found_highest:
1956 /* Compute highest gap address at the desired alignment */
1957 gap_end -= info->length;
1958 gap_end -= (gap_end - info->align_offset) & info->align_mask;
1959
1960 VM_BUG_ON(gap_end < info->low_limit);
1961 VM_BUG_ON(gap_end < gap_start);
1962 return gap_end;
1963}
1964
1da177e4
LT
1965/* Get an address range which is currently unmapped.
1966 * For shmat() with addr=0.
1967 *
1968 * Ugly calling convention alert:
1969 * Return value with the low bits set means error value,
1970 * ie
1971 * if (ret & ~PAGE_MASK)
1972 * error = ret;
1973 *
1974 * This function "knows" that -ENOMEM has the bits set.
1975 */
1976#ifndef HAVE_ARCH_UNMAPPED_AREA
1977unsigned long
1978arch_get_unmapped_area(struct file *filp, unsigned long addr,
1979 unsigned long len, unsigned long pgoff, unsigned long flags)
1980{
1981 struct mm_struct *mm = current->mm;
ddeaddb9 1982 struct vm_area_struct *vma, *prev;
db4fbfb9 1983 struct vm_unmapped_area_info info;
1da177e4 1984
2afc745f 1985 if (len > TASK_SIZE - mmap_min_addr)
1da177e4
LT
1986 return -ENOMEM;
1987
06abdfb4
BH
1988 if (flags & MAP_FIXED)
1989 return addr;
1990
1da177e4
LT
1991 if (addr) {
1992 addr = PAGE_ALIGN(addr);
ddeaddb9 1993 vma = find_vma_prev(mm, addr, &prev);
2afc745f 1994 if (TASK_SIZE - len >= addr && addr >= mmap_min_addr &&
ddeaddb9
HD
1995 (!vma || addr + len <= vm_start_gap(vma)) &&
1996 (!prev || addr >= vm_end_gap(prev)))
1da177e4
LT
1997 return addr;
1998 }
1da177e4 1999
db4fbfb9
ML
2000 info.flags = 0;
2001 info.length = len;
4e99b021 2002 info.low_limit = mm->mmap_base;
db4fbfb9
ML
2003 info.high_limit = TASK_SIZE;
2004 info.align_mask = 0;
2005 return vm_unmapped_area(&info);
1da177e4 2006}
cc71aba3 2007#endif
1da177e4 2008
1da177e4
LT
2009/*
2010 * This mmap-allocator allocates new areas top-down from below the
2011 * stack's low limit (the base):
2012 */
2013#ifndef HAVE_ARCH_UNMAPPED_AREA_TOPDOWN
2014unsigned long
2015arch_get_unmapped_area_topdown(struct file *filp, const unsigned long addr0,
2016 const unsigned long len, const unsigned long pgoff,
2017 const unsigned long flags)
2018{
ddeaddb9 2019 struct vm_area_struct *vma, *prev;
1da177e4 2020 struct mm_struct *mm = current->mm;
db4fbfb9
ML
2021 unsigned long addr = addr0;
2022 struct vm_unmapped_area_info info;
1da177e4
LT
2023
2024 /* requested length too big for entire address space */
2afc745f 2025 if (len > TASK_SIZE - mmap_min_addr)
1da177e4
LT
2026 return -ENOMEM;
2027
06abdfb4
BH
2028 if (flags & MAP_FIXED)
2029 return addr;
2030
1da177e4
LT
2031 /* requesting a specific address */
2032 if (addr) {
2033 addr = PAGE_ALIGN(addr);
ddeaddb9 2034 vma = find_vma_prev(mm, addr, &prev);
2afc745f 2035 if (TASK_SIZE - len >= addr && addr >= mmap_min_addr &&
ddeaddb9
HD
2036 (!vma || addr + len <= vm_start_gap(vma)) &&
2037 (!prev || addr >= vm_end_gap(prev)))
1da177e4
LT
2038 return addr;
2039 }
2040
db4fbfb9
ML
2041 info.flags = VM_UNMAPPED_AREA_TOPDOWN;
2042 info.length = len;
2afc745f 2043 info.low_limit = max(PAGE_SIZE, mmap_min_addr);
db4fbfb9
ML
2044 info.high_limit = mm->mmap_base;
2045 info.align_mask = 0;
2046 addr = vm_unmapped_area(&info);
b716ad95 2047
1da177e4
LT
2048 /*
2049 * A failed mmap() very likely causes application failure,
2050 * so fall back to the bottom-up function here. This scenario
2051 * can happen with large stack limits and large mmap()
2052 * allocations.
2053 */
de1741a1 2054 if (offset_in_page(addr)) {
db4fbfb9
ML
2055 VM_BUG_ON(addr != -ENOMEM);
2056 info.flags = 0;
2057 info.low_limit = TASK_UNMAPPED_BASE;
2058 info.high_limit = TASK_SIZE;
2059 addr = vm_unmapped_area(&info);
2060 }
1da177e4
LT
2061
2062 return addr;
2063}
2064#endif
2065
1da177e4
LT
2066unsigned long
2067get_unmapped_area(struct file *file, unsigned long addr, unsigned long len,
2068 unsigned long pgoff, unsigned long flags)
2069{
06abdfb4
BH
2070 unsigned long (*get_area)(struct file *, unsigned long,
2071 unsigned long, unsigned long, unsigned long);
2072
9206de95
AV
2073 unsigned long error = arch_mmap_check(addr, len, flags);
2074 if (error)
2075 return error;
2076
2077 /* Careful about overflows.. */
2078 if (len > TASK_SIZE)
2079 return -ENOMEM;
2080
06abdfb4 2081 get_area = current->mm->get_unmapped_area;
c01d5b30
HD
2082 if (file) {
2083 if (file->f_op->get_unmapped_area)
2084 get_area = file->f_op->get_unmapped_area;
2085 } else if (flags & MAP_SHARED) {
2086 /*
2087 * mmap_region() will call shmem_zero_setup() to create a file,
2088 * so use shmem's get_unmapped_area in case it can be huge.
2089 * do_mmap_pgoff() will clear pgoff, so match alignment.
2090 */
2091 pgoff = 0;
2092 get_area = shmem_get_unmapped_area;
2093 }
2094
06abdfb4
BH
2095 addr = get_area(file, addr, len, pgoff, flags);
2096 if (IS_ERR_VALUE(addr))
2097 return addr;
1da177e4 2098
07ab67c8
LT
2099 if (addr > TASK_SIZE - len)
2100 return -ENOMEM;
de1741a1 2101 if (offset_in_page(addr))
07ab67c8 2102 return -EINVAL;
06abdfb4 2103
9ac4ed4b
AV
2104 error = security_mmap_addr(addr);
2105 return error ? error : addr;
1da177e4
LT
2106}
2107
2108EXPORT_SYMBOL(get_unmapped_area);
2109
2110/* Look up the first VMA which satisfies addr < vm_end, NULL if none. */
48aae425 2111struct vm_area_struct *find_vma(struct mm_struct *mm, unsigned long addr)
1da177e4 2112{
615d6e87
DB
2113 struct rb_node *rb_node;
2114 struct vm_area_struct *vma;
1da177e4 2115
841e31e5 2116 /* Check the cache first. */
615d6e87
DB
2117 vma = vmacache_find(mm, addr);
2118 if (likely(vma))
2119 return vma;
841e31e5 2120
615d6e87 2121 rb_node = mm->mm_rb.rb_node;
841e31e5 2122
615d6e87
DB
2123 while (rb_node) {
2124 struct vm_area_struct *tmp;
2125
2126 tmp = rb_entry(rb_node, struct vm_area_struct, vm_rb);
2127
2128 if (tmp->vm_end > addr) {
2129 vma = tmp;
2130 if (tmp->vm_start <= addr)
2131 break;
2132 rb_node = rb_node->rb_left;
2133 } else
2134 rb_node = rb_node->rb_right;
1da177e4 2135 }
615d6e87
DB
2136
2137 if (vma)
2138 vmacache_update(addr, vma);
1da177e4
LT
2139 return vma;
2140}
2141
2142EXPORT_SYMBOL(find_vma);
2143
6bd4837d
KM
2144/*
2145 * Same as find_vma, but also return a pointer to the previous VMA in *pprev.
6bd4837d 2146 */
1da177e4
LT
2147struct vm_area_struct *
2148find_vma_prev(struct mm_struct *mm, unsigned long addr,
2149 struct vm_area_struct **pprev)
2150{
6bd4837d 2151 struct vm_area_struct *vma;
1da177e4 2152
6bd4837d 2153 vma = find_vma(mm, addr);
83cd904d
MP
2154 if (vma) {
2155 *pprev = vma->vm_prev;
2156 } else {
2157 struct rb_node *rb_node = mm->mm_rb.rb_node;
2158 *pprev = NULL;
2159 while (rb_node) {
2160 *pprev = rb_entry(rb_node, struct vm_area_struct, vm_rb);
2161 rb_node = rb_node->rb_right;
2162 }
2163 }
6bd4837d 2164 return vma;
1da177e4
LT
2165}
2166
2167/*
2168 * Verify that the stack growth is acceptable and
2169 * update accounting. This is shared with both the
2170 * grow-up and grow-down cases.
2171 */
ddeaddb9
HD
2172static int acct_stack_growth(struct vm_area_struct *vma,
2173 unsigned long size, unsigned long grow)
1da177e4
LT
2174{
2175 struct mm_struct *mm = vma->vm_mm;
2176 struct rlimit *rlim = current->signal->rlim;
ddeaddb9 2177 unsigned long new_start;
1da177e4
LT
2178
2179 /* address space limit tests */
84638335 2180 if (!may_expand_vm(mm, vma->vm_flags, grow))
1da177e4
LT
2181 return -ENOMEM;
2182
2183 /* Stack limit test */
ddeaddb9 2184 if (size > READ_ONCE(rlim[RLIMIT_STACK].rlim_cur))
1da177e4
LT
2185 return -ENOMEM;
2186
2187 /* mlock limit tests */
2188 if (vma->vm_flags & VM_LOCKED) {
2189 unsigned long locked;
2190 unsigned long limit;
2191 locked = mm->locked_vm + grow;
4db0c3c2 2192 limit = READ_ONCE(rlim[RLIMIT_MEMLOCK].rlim_cur);
59e99e5b 2193 limit >>= PAGE_SHIFT;
1da177e4
LT
2194 if (locked > limit && !capable(CAP_IPC_LOCK))
2195 return -ENOMEM;
2196 }
2197
0d59a01b
AL
2198 /* Check to ensure the stack will not grow into a hugetlb-only region */
2199 new_start = (vma->vm_flags & VM_GROWSUP) ? vma->vm_start :
2200 vma->vm_end - size;
2201 if (is_hugepage_only_range(vma->vm_mm, new_start, size))
2202 return -EFAULT;
2203
1da177e4
LT
2204 /*
2205 * Overcommit.. This must be the final test, as it will
2206 * update security statistics.
2207 */
05fa199d 2208 if (security_vm_enough_memory_mm(mm, grow))
1da177e4
LT
2209 return -ENOMEM;
2210
1da177e4
LT
2211 return 0;
2212}
2213
46dea3d0 2214#if defined(CONFIG_STACK_GROWSUP) || defined(CONFIG_IA64)
1da177e4 2215/*
46dea3d0
HD
2216 * PA-RISC uses this for its stack; IA64 for its Register Backing Store.
2217 * vma is the last one with address > vma->vm_end. Have to extend vma.
1da177e4 2218 */
72ae531c 2219int expand_upwards(struct vm_area_struct *vma, unsigned long address)
1da177e4 2220{
09357814 2221 struct mm_struct *mm = vma->vm_mm;
ddeaddb9
HD
2222 struct vm_area_struct *next;
2223 unsigned long gap_addr;
12352d3c 2224 int error = 0;
1da177e4
LT
2225
2226 if (!(vma->vm_flags & VM_GROWSUP))
2227 return -EFAULT;
2228
bddfdc66 2229 /* Guard against exceeding limits of the address space. */
ddeaddb9 2230 address &= PAGE_MASK;
bddfdc66 2231 if (address >= TASK_SIZE)
72ae531c 2232 return -ENOMEM;
bddfdc66 2233 address += PAGE_SIZE;
72ae531c 2234
ddeaddb9
HD
2235 /* Enforce stack_guard_gap */
2236 gap_addr = address + stack_guard_gap;
bddfdc66
HD
2237
2238 /* Guard against overflow */
2239 if (gap_addr < address || gap_addr > TASK_SIZE)
2240 gap_addr = TASK_SIZE;
2241
ddeaddb9 2242 next = vma->vm_next;
3e4f09ee
MH
2243 if (next && next->vm_start < gap_addr &&
2244 (next->vm_flags & (VM_WRITE|VM_READ|VM_EXEC))) {
ddeaddb9
HD
2245 if (!(next->vm_flags & VM_GROWSUP))
2246 return -ENOMEM;
2247 /* Check that both stack segments have the same anon_vma? */
2248 }
2249
12352d3c 2250 /* We must make sure the anon_vma is allocated. */
1da177e4
LT
2251 if (unlikely(anon_vma_prepare(vma)))
2252 return -ENOMEM;
1da177e4
LT
2253
2254 /*
2255 * vma->vm_start/vm_end cannot change under us because the caller
2256 * is required to hold the mmap_sem in read mode. We need the
2257 * anon_vma lock to serialize against concurrent expand_stacks.
2258 */
12352d3c 2259 anon_vma_lock_write(vma->anon_vma);
1da177e4
LT
2260
2261 /* Somebody else might have raced and expanded it already */
2262 if (address > vma->vm_end) {
2263 unsigned long size, grow;
2264
2265 size = address - vma->vm_start;
2266 grow = (address - vma->vm_end) >> PAGE_SHIFT;
2267
42c36f63
HD
2268 error = -ENOMEM;
2269 if (vma->vm_pgoff + (size >> PAGE_SHIFT) >= vma->vm_pgoff) {
72ae531c 2270 error = acct_stack_growth(vma, size, grow);
42c36f63 2271 if (!error) {
4128997b
ML
2272 /*
2273 * vma_gap_update() doesn't support concurrent
2274 * updates, but we only hold a shared mmap_sem
2275 * lock here, so we need to protect against
2276 * concurrent vma expansions.
12352d3c 2277 * anon_vma_lock_write() doesn't help here, as
4128997b
ML
2278 * we don't guarantee that all growable vmas
2279 * in a mm share the same root anon vma.
2280 * So, we reuse mm->page_table_lock to guard
2281 * against concurrent vma expansions.
2282 */
09357814 2283 spin_lock(&mm->page_table_lock);
87e8827b 2284 if (vma->vm_flags & VM_LOCKED)
09357814 2285 mm->locked_vm += grow;
84638335 2286 vm_stat_account(mm, vma->vm_flags, grow);
bf181b9f 2287 anon_vma_interval_tree_pre_update_vma(vma);
42c36f63 2288 vma->vm_end = address;
bf181b9f 2289 anon_vma_interval_tree_post_update_vma(vma);
d3737187
ML
2290 if (vma->vm_next)
2291 vma_gap_update(vma->vm_next);
2292 else
ddeaddb9 2293 mm->highest_vm_end = vm_end_gap(vma);
09357814 2294 spin_unlock(&mm->page_table_lock);
4128997b 2295
42c36f63
HD
2296 perf_event_mmap(vma);
2297 }
3af9e859 2298 }
1da177e4 2299 }
12352d3c 2300 anon_vma_unlock_write(vma->anon_vma);
6d50e60c 2301 khugepaged_enter_vma_merge(vma, vma->vm_flags);
09357814 2302 validate_mm(mm);
1da177e4
LT
2303 return error;
2304}
46dea3d0
HD
2305#endif /* CONFIG_STACK_GROWSUP || CONFIG_IA64 */
2306
1da177e4
LT
2307/*
2308 * vma is the first one with address < vma->vm_start. Have to extend vma.
2309 */
d05f3169 2310int expand_downwards(struct vm_area_struct *vma,
72ae531c 2311 unsigned long address)
1da177e4 2312{
09357814 2313 struct mm_struct *mm = vma->vm_mm;
ddeaddb9 2314 struct vm_area_struct *prev;
1da177e4
LT
2315 int error;
2316
8869477a 2317 address &= PAGE_MASK;
e5467859 2318 error = security_mmap_addr(address);
8869477a
EP
2319 if (error)
2320 return error;
2321
ddeaddb9 2322 /* Enforce stack_guard_gap */
ddeaddb9 2323 prev = vma->vm_prev;
aea792ba
ON
2324 /* Check that both stack segments have the same anon_vma? */
2325 if (prev && !(prev->vm_flags & VM_GROWSDOWN) &&
3e4f09ee 2326 (prev->vm_flags & (VM_WRITE|VM_READ|VM_EXEC))) {
aea792ba 2327 if (address - prev->vm_end < stack_guard_gap)
ddeaddb9 2328 return -ENOMEM;
ddeaddb9
HD
2329 }
2330
12352d3c
KK
2331 /* We must make sure the anon_vma is allocated. */
2332 if (unlikely(anon_vma_prepare(vma)))
2333 return -ENOMEM;
1da177e4
LT
2334
2335 /*
2336 * vma->vm_start/vm_end cannot change under us because the caller
2337 * is required to hold the mmap_sem in read mode. We need the
2338 * anon_vma lock to serialize against concurrent expand_stacks.
2339 */
12352d3c 2340 anon_vma_lock_write(vma->anon_vma);
1da177e4
LT
2341
2342 /* Somebody else might have raced and expanded it already */
2343 if (address < vma->vm_start) {
2344 unsigned long size, grow;
2345
2346 size = vma->vm_end - address;
2347 grow = (vma->vm_start - address) >> PAGE_SHIFT;
2348
a626ca6a
LT
2349 error = -ENOMEM;
2350 if (grow <= vma->vm_pgoff) {
72ae531c 2351 error = acct_stack_growth(vma, size, grow);
a626ca6a 2352 if (!error) {
4128997b
ML
2353 /*
2354 * vma_gap_update() doesn't support concurrent
2355 * updates, but we only hold a shared mmap_sem
2356 * lock here, so we need to protect against
2357 * concurrent vma expansions.
12352d3c 2358 * anon_vma_lock_write() doesn't help here, as
4128997b
ML
2359 * we don't guarantee that all growable vmas
2360 * in a mm share the same root anon vma.
2361 * So, we reuse mm->page_table_lock to guard
2362 * against concurrent vma expansions.
2363 */
09357814 2364 spin_lock(&mm->page_table_lock);
87e8827b 2365 if (vma->vm_flags & VM_LOCKED)
09357814 2366 mm->locked_vm += grow;
84638335 2367 vm_stat_account(mm, vma->vm_flags, grow);
bf181b9f 2368 anon_vma_interval_tree_pre_update_vma(vma);
a626ca6a
LT
2369 vma->vm_start = address;
2370 vma->vm_pgoff -= grow;
bf181b9f 2371 anon_vma_interval_tree_post_update_vma(vma);
d3737187 2372 vma_gap_update(vma);
09357814 2373 spin_unlock(&mm->page_table_lock);
4128997b 2374
a626ca6a
LT
2375 perf_event_mmap(vma);
2376 }
1da177e4
LT
2377 }
2378 }
12352d3c 2379 anon_vma_unlock_write(vma->anon_vma);
6d50e60c 2380 khugepaged_enter_vma_merge(vma, vma->vm_flags);
09357814 2381 validate_mm(mm);
1da177e4
LT
2382 return error;
2383}
2384
ddeaddb9
HD
2385/* enforced gap between the expanding stack and other mappings. */
2386unsigned long stack_guard_gap = 256UL<<PAGE_SHIFT;
2387
2388static int __init cmdline_parse_stack_guard_gap(char *p)
2389{
2390 unsigned long val;
2391 char *endptr;
2392
2393 val = simple_strtoul(p, &endptr, 10);
2394 if (!*endptr)
2395 stack_guard_gap = val << PAGE_SHIFT;
2396
2397 return 0;
2398}
2399__setup("stack_guard_gap=", cmdline_parse_stack_guard_gap);
2400
b6a2fea3
OW
2401#ifdef CONFIG_STACK_GROWSUP
2402int expand_stack(struct vm_area_struct *vma, unsigned long address)
2403{
72ae531c 2404 return expand_upwards(vma, address);
b6a2fea3
OW
2405}
2406
2407struct vm_area_struct *
2408find_extend_vma(struct mm_struct *mm, unsigned long addr)
2409{
2410 struct vm_area_struct *vma, *prev;
2411
2412 addr &= PAGE_MASK;
2413 vma = find_vma_prev(mm, addr, &prev);
2414 if (vma && (vma->vm_start <= addr))
2415 return vma;
1c127185 2416 if (!prev || expand_stack(prev, addr))
b6a2fea3 2417 return NULL;
cea10a19 2418 if (prev->vm_flags & VM_LOCKED)
fc05f566 2419 populate_vma_page_range(prev, addr, prev->vm_end, NULL);
b6a2fea3
OW
2420 return prev;
2421}
2422#else
2423int expand_stack(struct vm_area_struct *vma, unsigned long address)
2424{
72ae531c 2425 return expand_downwards(vma, address);
b6a2fea3
OW
2426}
2427
1da177e4 2428struct vm_area_struct *
cc71aba3 2429find_extend_vma(struct mm_struct *mm, unsigned long addr)
1da177e4 2430{
cc71aba3 2431 struct vm_area_struct *vma;
1da177e4
LT
2432 unsigned long start;
2433
2434 addr &= PAGE_MASK;
cc71aba3 2435 vma = find_vma(mm, addr);
1da177e4
LT
2436 if (!vma)
2437 return NULL;
2438 if (vma->vm_start <= addr)
2439 return vma;
2440 if (!(vma->vm_flags & VM_GROWSDOWN))
2441 return NULL;
2442 start = vma->vm_start;
2443 if (expand_stack(vma, addr))
2444 return NULL;
cea10a19 2445 if (vma->vm_flags & VM_LOCKED)
fc05f566 2446 populate_vma_page_range(vma, addr, start, NULL);
1da177e4
LT
2447 return vma;
2448}
2449#endif
2450
e1d6d01a
JB
2451EXPORT_SYMBOL_GPL(find_extend_vma);
2452
1da177e4 2453/*
2c0b3814 2454 * Ok - we have the memory areas we should free on the vma list,
1da177e4 2455 * so release them, and do the vma updates.
2c0b3814
HD
2456 *
2457 * Called with the mm semaphore held.
1da177e4 2458 */
2c0b3814 2459static void remove_vma_list(struct mm_struct *mm, struct vm_area_struct *vma)
1da177e4 2460{
4f74d2c8
LT
2461 unsigned long nr_accounted = 0;
2462
365e9c87
HD
2463 /* Update high watermark before we lower total_vm */
2464 update_hiwater_vm(mm);
1da177e4 2465 do {
2c0b3814
HD
2466 long nrpages = vma_pages(vma);
2467
4f74d2c8
LT
2468 if (vma->vm_flags & VM_ACCOUNT)
2469 nr_accounted += nrpages;
84638335 2470 vm_stat_account(mm, vma->vm_flags, -nrpages);
a8fb5618 2471 vma = remove_vma(vma);
146425a3 2472 } while (vma);
4f74d2c8 2473 vm_unacct_memory(nr_accounted);
1da177e4
LT
2474 validate_mm(mm);
2475}
2476
2477/*
2478 * Get rid of page table information in the indicated region.
2479 *
f10df686 2480 * Called with the mm semaphore held.
1da177e4
LT
2481 */
2482static void unmap_region(struct mm_struct *mm,
e0da382c
HD
2483 struct vm_area_struct *vma, struct vm_area_struct *prev,
2484 unsigned long start, unsigned long end)
1da177e4 2485{
cc71aba3 2486 struct vm_area_struct *next = prev ? prev->vm_next : mm->mmap;
d16dfc55 2487 struct mmu_gather tlb;
1da177e4
LT
2488
2489 lru_add_drain();
2b047252 2490 tlb_gather_mmu(&tlb, mm, start, end);
365e9c87 2491 update_hiwater_rss(mm);
4f74d2c8 2492 unmap_vmas(&tlb, vma, start, end);
d16dfc55 2493 free_pgtables(&tlb, vma, prev ? prev->vm_end : FIRST_USER_ADDRESS,
6ee8630e 2494 next ? next->vm_start : USER_PGTABLES_CEILING);
d16dfc55 2495 tlb_finish_mmu(&tlb, start, end);
1da177e4
LT
2496}
2497
2498/*
2499 * Create a list of vma's touched by the unmap, removing them from the mm's
2500 * vma list as we go..
2501 */
2502static void
2503detach_vmas_to_be_unmapped(struct mm_struct *mm, struct vm_area_struct *vma,
2504 struct vm_area_struct *prev, unsigned long end)
2505{
2506 struct vm_area_struct **insertion_point;
2507 struct vm_area_struct *tail_vma = NULL;
2508
2509 insertion_point = (prev ? &prev->vm_next : &mm->mmap);
297c5eee 2510 vma->vm_prev = NULL;
1da177e4 2511 do {
d3737187 2512 vma_rb_erase(vma, &mm->mm_rb);
1da177e4
LT
2513 mm->map_count--;
2514 tail_vma = vma;
2515 vma = vma->vm_next;
2516 } while (vma && vma->vm_start < end);
2517 *insertion_point = vma;
d3737187 2518 if (vma) {
297c5eee 2519 vma->vm_prev = prev;
d3737187
ML
2520 vma_gap_update(vma);
2521 } else
ddeaddb9 2522 mm->highest_vm_end = prev ? vm_end_gap(prev) : 0;
1da177e4 2523 tail_vma->vm_next = NULL;
615d6e87
DB
2524
2525 /* Kill the cache */
2526 vmacache_invalidate(mm);
1da177e4
LT
2527}
2528
2529/*
659ace58
KM
2530 * __split_vma() bypasses sysctl_max_map_count checking. We use this on the
2531 * munmap path where it doesn't make sense to fail.
1da177e4 2532 */
cc71aba3 2533static int __split_vma(struct mm_struct *mm, struct vm_area_struct *vma,
1da177e4
LT
2534 unsigned long addr, int new_below)
2535{
1da177e4 2536 struct vm_area_struct *new;
e3975891 2537 int err;
1da177e4 2538
a5516438
AK
2539 if (is_vm_hugetlb_page(vma) && (addr &
2540 ~(huge_page_mask(hstate_vma(vma)))))
1da177e4
LT
2541 return -EINVAL;
2542
e94b1766 2543 new = kmem_cache_alloc(vm_area_cachep, GFP_KERNEL);
1da177e4 2544 if (!new)
e3975891 2545 return -ENOMEM;
1da177e4
LT
2546
2547 /* most fields are the same, copy all, and then fixup */
2548 *new = *vma;
2549
5beb4930
RR
2550 INIT_LIST_HEAD(&new->anon_vma_chain);
2551
1da177e4
LT
2552 if (new_below)
2553 new->vm_end = addr;
2554 else {
2555 new->vm_start = addr;
2556 new->vm_pgoff += ((addr - vma->vm_start) >> PAGE_SHIFT);
2557 }
2558
ef0855d3
ON
2559 err = vma_dup_policy(vma, new);
2560 if (err)
5beb4930 2561 goto out_free_vma;
1da177e4 2562
c4ea95d7
DF
2563 err = anon_vma_clone(new, vma);
2564 if (err)
5beb4930
RR
2565 goto out_free_mpol;
2566
e9714acf 2567 if (new->vm_file)
e14748e8 2568 vma_get_file(new);
1da177e4
LT
2569
2570 if (new->vm_ops && new->vm_ops->open)
2571 new->vm_ops->open(new);
2572
2573 if (new_below)
5beb4930 2574 err = vma_adjust(vma, addr, vma->vm_end, vma->vm_pgoff +
1da177e4
LT
2575 ((addr - new->vm_start) >> PAGE_SHIFT), new);
2576 else
5beb4930 2577 err = vma_adjust(vma, vma->vm_start, addr, vma->vm_pgoff, new);
1da177e4 2578
5beb4930
RR
2579 /* Success. */
2580 if (!err)
2581 return 0;
2582
2583 /* Clean everything up if vma_adjust failed. */
58927533
RR
2584 if (new->vm_ops && new->vm_ops->close)
2585 new->vm_ops->close(new);
e9714acf 2586 if (new->vm_file)
e14748e8 2587 vma_fput(new);
2aeadc30 2588 unlink_anon_vmas(new);
5beb4930 2589 out_free_mpol:
ef0855d3 2590 mpol_put(vma_policy(new));
5beb4930
RR
2591 out_free_vma:
2592 kmem_cache_free(vm_area_cachep, new);
5beb4930 2593 return err;
1da177e4
LT
2594}
2595
659ace58
KM
2596/*
2597 * Split a vma into two pieces at address 'addr', a new vma is allocated
2598 * either for the first part or the tail.
2599 */
2600int split_vma(struct mm_struct *mm, struct vm_area_struct *vma,
2601 unsigned long addr, int new_below)
2602{
2603 if (mm->map_count >= sysctl_max_map_count)
2604 return -ENOMEM;
2605
2606 return __split_vma(mm, vma, addr, new_below);
2607}
2608
1da177e4
LT
2609/* Munmap is split into 2 main parts -- this part which finds
2610 * what needs doing, and the areas themselves, which do the
2611 * work. This now handles partial unmappings.
2612 * Jeremy Fitzhardinge <jeremy@goop.org>
2613 */
2614int do_munmap(struct mm_struct *mm, unsigned long start, size_t len)
2615{
2616 unsigned long end;
146425a3 2617 struct vm_area_struct *vma, *prev, *last;
1da177e4 2618
de1741a1 2619 if ((offset_in_page(start)) || start > TASK_SIZE || len > TASK_SIZE-start)
1da177e4
LT
2620 return -EINVAL;
2621
cc71aba3 2622 len = PAGE_ALIGN(len);
2623 if (len == 0)
1da177e4
LT
2624 return -EINVAL;
2625
2626 /* Find the first overlapping VMA */
9be34c9d 2627 vma = find_vma(mm, start);
146425a3 2628 if (!vma)
1da177e4 2629 return 0;
9be34c9d 2630 prev = vma->vm_prev;
146425a3 2631 /* we have start < vma->vm_end */
1da177e4
LT
2632
2633 /* if it doesn't overlap, we have nothing.. */
2634 end = start + len;
146425a3 2635 if (vma->vm_start >= end)
1da177e4
LT
2636 return 0;
2637
2638 /*
2639 * If we need to split any vma, do it now to save pain later.
2640 *
2641 * Note: mremap's move_vma VM_ACCOUNT handling assumes a partially
2642 * unmapped vm_area_struct will remain in use: so lower split_vma
2643 * places tmp vma above, and higher split_vma places tmp vma below.
2644 */
146425a3 2645 if (start > vma->vm_start) {
659ace58
KM
2646 int error;
2647
2648 /*
2649 * Make sure that map_count on return from munmap() will
2650 * not exceed its limit; but let map_count go just above
2651 * its limit temporarily, to help free resources as expected.
2652 */
2653 if (end < vma->vm_end && mm->map_count >= sysctl_max_map_count)
2654 return -ENOMEM;
2655
2656 error = __split_vma(mm, vma, start, 0);
1da177e4
LT
2657 if (error)
2658 return error;
146425a3 2659 prev = vma;
1da177e4
LT
2660 }
2661
2662 /* Does it split the last one? */
2663 last = find_vma(mm, end);
2664 if (last && end > last->vm_start) {
659ace58 2665 int error = __split_vma(mm, last, end, 1);
1da177e4
LT
2666 if (error)
2667 return error;
2668 }
cc71aba3 2669 vma = prev ? prev->vm_next : mm->mmap;
1da177e4 2670
ba470de4
RR
2671 /*
2672 * unlock any mlock()ed ranges before detaching vmas
2673 */
2674 if (mm->locked_vm) {
2675 struct vm_area_struct *tmp = vma;
2676 while (tmp && tmp->vm_start < end) {
2677 if (tmp->vm_flags & VM_LOCKED) {
2678 mm->locked_vm -= vma_pages(tmp);
2679 munlock_vma_pages_all(tmp);
2680 }
2681 tmp = tmp->vm_next;
2682 }
2683 }
2684
1da177e4
LT
2685 /*
2686 * Remove the vma's, and unmap the actual pages
2687 */
146425a3
HD
2688 detach_vmas_to_be_unmapped(mm, vma, prev, end);
2689 unmap_region(mm, vma, prev, start, end);
1da177e4 2690
1de4fa14
DH
2691 arch_unmap(mm, vma, start, end);
2692
1da177e4 2693 /* Fix up all other VM information */
2c0b3814 2694 remove_vma_list(mm, vma);
1da177e4
LT
2695
2696 return 0;
2697}
1da177e4 2698
bfce281c 2699int vm_munmap(unsigned long start, size_t len)
1da177e4
LT
2700{
2701 int ret;
bfce281c 2702 struct mm_struct *mm = current->mm;
1da177e4 2703
ae798783
MH
2704 if (down_write_killable(&mm->mmap_sem))
2705 return -EINTR;
2706
a46ef99d 2707 ret = do_munmap(mm, start, len);
1da177e4
LT
2708 up_write(&mm->mmap_sem);
2709 return ret;
2710}
a46ef99d
LT
2711EXPORT_SYMBOL(vm_munmap);
2712
2713SYSCALL_DEFINE2(munmap, unsigned long, addr, size_t, len)
2714{
dc0ef0df
MH
2715 int ret;
2716 struct mm_struct *mm = current->mm;
2717
a46ef99d 2718 profile_munmap(addr);
dc0ef0df
MH
2719 if (down_write_killable(&mm->mmap_sem))
2720 return -EINTR;
2721 ret = do_munmap(mm, addr, len);
2722 up_write(&mm->mmap_sem);
2723 return ret;
a46ef99d 2724}
1da177e4 2725
c8d78c18
KS
2726
2727/*
2728 * Emulation of deprecated remap_file_pages() syscall.
2729 */
2730SYSCALL_DEFINE5(remap_file_pages, unsigned long, start, unsigned long, size,
2731 unsigned long, prot, unsigned long, pgoff, unsigned long, flags)
2732{
2733
2734 struct mm_struct *mm = current->mm;
2735 struct vm_area_struct *vma;
2736 unsigned long populate = 0;
2737 unsigned long ret = -EINVAL;
e14748e8 2738 struct file *file, *prfile;
c8d78c18 2739
756a025f
JP
2740 pr_warn_once("%s (%d) uses deprecated remap_file_pages() syscall. See Documentation/vm/remap_file_pages.txt.\n",
2741 current->comm, current->pid);
c8d78c18
KS
2742
2743 if (prot)
2744 return ret;
2745 start = start & PAGE_MASK;
2746 size = size & PAGE_MASK;
2747
2748 if (start + size <= start)
2749 return ret;
2750
2751 /* Does pgoff wrap? */
2752 if (pgoff + (size >> PAGE_SHIFT) < pgoff)
2753 return ret;
2754
dc0ef0df
MH
2755 if (down_write_killable(&mm->mmap_sem))
2756 return -EINTR;
2757
c8d78c18
KS
2758 vma = find_vma(mm, start);
2759
2760 if (!vma || !(vma->vm_flags & VM_SHARED))
2761 goto out;
2762
48f7df32 2763 if (start < vma->vm_start)
c8d78c18
KS
2764 goto out;
2765
48f7df32
KS
2766 if (start + size > vma->vm_end) {
2767 struct vm_area_struct *next;
2768
2769 for (next = vma->vm_next; next; next = next->vm_next) {
2770 /* hole between vmas ? */
2771 if (next->vm_start != next->vm_prev->vm_end)
2772 goto out;
2773
2774 if (next->vm_file != vma->vm_file)
2775 goto out;
2776
2777 if (next->vm_flags != vma->vm_flags)
2778 goto out;
2779
2780 if (start + size <= next->vm_end)
2781 break;
2782 }
2783
2784 if (!next)
2785 goto out;
c8d78c18
KS
2786 }
2787
2788 prot |= vma->vm_flags & VM_READ ? PROT_READ : 0;
2789 prot |= vma->vm_flags & VM_WRITE ? PROT_WRITE : 0;
2790 prot |= vma->vm_flags & VM_EXEC ? PROT_EXEC : 0;
2791
2792 flags &= MAP_NONBLOCK;
2793 flags |= MAP_SHARED | MAP_FIXED | MAP_POPULATE;
2794 if (vma->vm_flags & VM_LOCKED) {
48f7df32 2795 struct vm_area_struct *tmp;
c8d78c18 2796 flags |= MAP_LOCKED;
48f7df32 2797
c8d78c18 2798 /* drop PG_Mlocked flag for over-mapped range */
48f7df32
KS
2799 for (tmp = vma; tmp->vm_start >= start + size;
2800 tmp = tmp->vm_next) {
9a73f61b
KS
2801 /*
2802 * Split pmd and munlock page on the border
2803 * of the range.
2804 */
2805 vma_adjust_trans_huge(tmp, start, start + size, 0);
2806
48f7df32
KS
2807 munlock_vma_pages_range(tmp,
2808 max(tmp->vm_start, start),
2809 min(tmp->vm_end, start + size));
2810 }
c8d78c18
KS
2811 }
2812
e14748e8
SF
2813 vma_get_file(vma);
2814 file = vma->vm_file;
2815 prfile = vma->vm_prfile;
c8d78c18
KS
2816 ret = do_mmap_pgoff(vma->vm_file, start, size,
2817 prot, flags, pgoff, &populate);
e14748e8
SF
2818 if (!IS_ERR_VALUE(ret) && file && prfile) {
2819 struct vm_area_struct *new_vma;
2820
2821 new_vma = find_vma(mm, ret);
2822 if (!new_vma->vm_prfile)
2823 new_vma->vm_prfile = prfile;
2824 if (new_vma != vma)
2825 get_file(prfile);
2826 }
2827 /*
2828 * two fput()s instead of vma_fput(vma),
2829 * coz vma may not be available anymore.
2830 */
c8d78c18 2831 fput(file);
e14748e8
SF
2832 if (prfile)
2833 fput(prfile);
c8d78c18
KS
2834out:
2835 up_write(&mm->mmap_sem);
2836 if (populate)
2837 mm_populate(ret, populate);
2838 if (!IS_ERR_VALUE(ret))
2839 ret = 0;
2840 return ret;
2841}
2842
1da177e4
LT
2843static inline void verify_mm_writelocked(struct mm_struct *mm)
2844{
a241ec65 2845#ifdef CONFIG_DEBUG_VM
1da177e4
LT
2846 if (unlikely(down_read_trylock(&mm->mmap_sem))) {
2847 WARN_ON(1);
2848 up_read(&mm->mmap_sem);
2849 }
2850#endif
2851}
2852
2853/*
2854 * this is really a simplified "do_mmap". it only handles
2855 * anonymous maps. eventually we may be able to do some
2856 * brk-specific accounting here.
2857 */
ba093a6d 2858static int do_brk(unsigned long addr, unsigned long request)
1da177e4 2859{
cc71aba3 2860 struct mm_struct *mm = current->mm;
2861 struct vm_area_struct *vma, *prev;
ba093a6d 2862 unsigned long flags, len;
cc71aba3 2863 struct rb_node **rb_link, *rb_parent;
1da177e4 2864 pgoff_t pgoff = addr >> PAGE_SHIFT;
3a459756 2865 int error;
1da177e4 2866
ba093a6d
KC
2867 len = PAGE_ALIGN(request);
2868 if (len < request)
2869 return -ENOMEM;
1da177e4 2870 if (!len)
5d22fc25 2871 return 0;
1da177e4 2872
3a459756
KK
2873 flags = VM_DATA_DEFAULT_FLAGS | VM_ACCOUNT | mm->def_flags;
2874
2c6a1016 2875 error = get_unmapped_area(NULL, addr, len, 0, MAP_FIXED);
de1741a1 2876 if (offset_in_page(error))
3a459756
KK
2877 return error;
2878
363ee17f
DB
2879 error = mlock_future_check(mm, mm->def_flags, len);
2880 if (error)
2881 return error;
1da177e4
LT
2882
2883 /*
2884 * mm->mmap_sem is required to protect against another thread
2885 * changing the mappings in case we sleep.
2886 */
2887 verify_mm_writelocked(mm);
2888
2889 /*
2890 * Clear old maps. this also does some error checking for us
2891 */
9fcd1457
RV
2892 while (find_vma_links(mm, addr, addr + len, &prev, &rb_link,
2893 &rb_parent)) {
1da177e4
LT
2894 if (do_munmap(mm, addr, len))
2895 return -ENOMEM;
1da177e4
LT
2896 }
2897
2898 /* Check against address space limits *after* clearing old maps... */
84638335 2899 if (!may_expand_vm(mm, flags, len >> PAGE_SHIFT))
1da177e4
LT
2900 return -ENOMEM;
2901
2902 if (mm->map_count > sysctl_max_map_count)
2903 return -ENOMEM;
2904
191c5424 2905 if (security_vm_enough_memory_mm(mm, len >> PAGE_SHIFT))
1da177e4
LT
2906 return -ENOMEM;
2907
1da177e4 2908 /* Can we just expand an old private anonymous mapping? */
ba470de4 2909 vma = vma_merge(mm, prev, addr, addr + len, flags,
19a809af 2910 NULL, NULL, pgoff, NULL, NULL_VM_UFFD_CTX);
ba470de4 2911 if (vma)
1da177e4
LT
2912 goto out;
2913
2914 /*
2915 * create a vma struct for an anonymous mapping
2916 */
c5e3b83e 2917 vma = kmem_cache_zalloc(vm_area_cachep, GFP_KERNEL);
1da177e4
LT
2918 if (!vma) {
2919 vm_unacct_memory(len >> PAGE_SHIFT);
2920 return -ENOMEM;
2921 }
1da177e4 2922
5beb4930 2923 INIT_LIST_HEAD(&vma->anon_vma_chain);
1da177e4
LT
2924 vma->vm_mm = mm;
2925 vma->vm_start = addr;
2926 vma->vm_end = addr + len;
2927 vma->vm_pgoff = pgoff;
2928 vma->vm_flags = flags;
3ed75eb8 2929 vma->vm_page_prot = vm_get_page_prot(flags);
1da177e4
LT
2930 vma_link(mm, vma, prev, rb_link, rb_parent);
2931out:
3af9e859 2932 perf_event_mmap(vma);
1da177e4 2933 mm->total_vm += len >> PAGE_SHIFT;
84638335 2934 mm->data_vm += len >> PAGE_SHIFT;
128557ff
ML
2935 if (flags & VM_LOCKED)
2936 mm->locked_vm += (len >> PAGE_SHIFT);
d9104d1c 2937 vma->vm_flags |= VM_SOFTDIRTY;
5d22fc25 2938 return 0;
1da177e4
LT
2939}
2940
5d22fc25 2941int vm_brk(unsigned long addr, unsigned long len)
e4eb1ff6
LT
2942{
2943 struct mm_struct *mm = current->mm;
5d22fc25 2944 int ret;
128557ff 2945 bool populate;
e4eb1ff6 2946
2d6c9282
MH
2947 if (down_write_killable(&mm->mmap_sem))
2948 return -EINTR;
2949
e4eb1ff6 2950 ret = do_brk(addr, len);
128557ff 2951 populate = ((mm->def_flags & VM_LOCKED) != 0);
e4eb1ff6 2952 up_write(&mm->mmap_sem);
5d22fc25 2953 if (populate && !ret)
128557ff 2954 mm_populate(addr, len);
e4eb1ff6
LT
2955 return ret;
2956}
2957EXPORT_SYMBOL(vm_brk);
1da177e4
LT
2958
2959/* Release all mmaps. */
2960void exit_mmap(struct mm_struct *mm)
2961{
d16dfc55 2962 struct mmu_gather tlb;
ba470de4 2963 struct vm_area_struct *vma;
1da177e4
LT
2964 unsigned long nr_accounted = 0;
2965
d6dd61c8 2966 /* mm's last user has gone, and its about to be pulled down */
cddb8a5c 2967 mmu_notifier_release(mm);
d6dd61c8 2968
ba470de4
RR
2969 if (mm->locked_vm) {
2970 vma = mm->mmap;
2971 while (vma) {
2972 if (vma->vm_flags & VM_LOCKED)
2973 munlock_vma_pages_all(vma);
2974 vma = vma->vm_next;
2975 }
2976 }
9480c53e
JF
2977
2978 arch_exit_mmap(mm);
2979
ba470de4 2980 vma = mm->mmap;
9480c53e
JF
2981 if (!vma) /* Can happen if dup_mmap() received an OOM */
2982 return;
2983
1da177e4 2984 lru_add_drain();
1da177e4 2985 flush_cache_mm(mm);
2b047252 2986 tlb_gather_mmu(&tlb, mm, 0, -1);
901608d9 2987 /* update_hiwater_rss(mm) here? but nobody should be looking */
e0da382c 2988 /* Use -1 here to ensure all VMAs in the mm are unmapped */
4f74d2c8 2989 unmap_vmas(&tlb, vma, 0, -1);
9ba69294 2990
6ee8630e 2991 free_pgtables(&tlb, vma, FIRST_USER_ADDRESS, USER_PGTABLES_CEILING);
853f5e26 2992 tlb_finish_mmu(&tlb, 0, -1);
1da177e4 2993
1da177e4 2994 /*
8f4f8c16
HD
2995 * Walk the list again, actually closing and freeing it,
2996 * with preemption enabled, without holding any MM locks.
1da177e4 2997 */
4f74d2c8
LT
2998 while (vma) {
2999 if (vma->vm_flags & VM_ACCOUNT)
3000 nr_accounted += vma_pages(vma);
a8fb5618 3001 vma = remove_vma(vma);
4f74d2c8
LT
3002 }
3003 vm_unacct_memory(nr_accounted);
1da177e4
LT
3004}
3005
3006/* Insert vm structure into process list sorted by address
3007 * and into the inode's i_mmap tree. If vm_file is non-NULL
c8c06efa 3008 * then i_mmap_rwsem is taken here.
1da177e4 3009 */
6597d783 3010int insert_vm_struct(struct mm_struct *mm, struct vm_area_struct *vma)
1da177e4 3011{
6597d783
HD
3012 struct vm_area_struct *prev;
3013 struct rb_node **rb_link, *rb_parent;
1da177e4 3014
c9d13f5f
CG
3015 if (find_vma_links(mm, vma->vm_start, vma->vm_end,
3016 &prev, &rb_link, &rb_parent))
3017 return -ENOMEM;
3018 if ((vma->vm_flags & VM_ACCOUNT) &&
3019 security_vm_enough_memory_mm(mm, vma_pages(vma)))
3020 return -ENOMEM;
3021
1da177e4
LT
3022 /*
3023 * The vm_pgoff of a purely anonymous vma should be irrelevant
3024 * until its first write fault, when page's anon_vma and index
3025 * are set. But now set the vm_pgoff it will almost certainly
3026 * end up with (unless mremap moves it elsewhere before that
3027 * first wfault), so /proc/pid/maps tells a consistent story.
3028 *
3029 * By setting it to reflect the virtual start address of the
3030 * vma, merges and splits can happen in a seamless way, just
3031 * using the existing file pgoff checks and manipulations.
3032 * Similarly in do_mmap_pgoff and in do_brk.
3033 */
8a9cc3b5 3034 if (vma_is_anonymous(vma)) {
1da177e4
LT
3035 BUG_ON(vma->anon_vma);
3036 vma->vm_pgoff = vma->vm_start >> PAGE_SHIFT;
3037 }
2b144498 3038
1da177e4
LT
3039 vma_link(mm, vma, prev, rb_link, rb_parent);
3040 return 0;
3041}
3042
3043/*
3044 * Copy the vma structure to a new location in the same mm,
3045 * prior to moving page table entries, to effect an mremap move.
3046 */
3047struct vm_area_struct *copy_vma(struct vm_area_struct **vmap,
38a76013
ML
3048 unsigned long addr, unsigned long len, pgoff_t pgoff,
3049 bool *need_rmap_locks)
1da177e4
LT
3050{
3051 struct vm_area_struct *vma = *vmap;
3052 unsigned long vma_start = vma->vm_start;
3053 struct mm_struct *mm = vma->vm_mm;
3054 struct vm_area_struct *new_vma, *prev;
3055 struct rb_node **rb_link, *rb_parent;
948f017b 3056 bool faulted_in_anon_vma = true;
1da177e4
LT
3057
3058 /*
3059 * If anonymous vma has not yet been faulted, update new pgoff
3060 * to match new location, to increase its chance of merging.
3061 */
ce75799b 3062 if (unlikely(vma_is_anonymous(vma) && !vma->anon_vma)) {
1da177e4 3063 pgoff = addr >> PAGE_SHIFT;
948f017b
AA
3064 faulted_in_anon_vma = false;
3065 }
1da177e4 3066
6597d783
HD
3067 if (find_vma_links(mm, addr, addr + len, &prev, &rb_link, &rb_parent))
3068 return NULL; /* should never get here */
1da177e4 3069 new_vma = vma_merge(mm, prev, addr, addr + len, vma->vm_flags,
19a809af
AA
3070 vma->anon_vma, vma->vm_file, pgoff, vma_policy(vma),
3071 vma->vm_userfaultfd_ctx);
1da177e4
LT
3072 if (new_vma) {
3073 /*
3074 * Source vma may have been merged into new_vma
3075 */
948f017b
AA
3076 if (unlikely(vma_start >= new_vma->vm_start &&
3077 vma_start < new_vma->vm_end)) {
3078 /*
3079 * The only way we can get a vma_merge with
3080 * self during an mremap is if the vma hasn't
3081 * been faulted in yet and we were allowed to
3082 * reset the dst vma->vm_pgoff to the
3083 * destination address of the mremap to allow
3084 * the merge to happen. mremap must change the
3085 * vm_pgoff linearity between src and dst vmas
3086 * (in turn preventing a vma_merge) to be
3087 * safe. It is only safe to keep the vm_pgoff
3088 * linear if there are no pages mapped yet.
3089 */
81d1b09c 3090 VM_BUG_ON_VMA(faulted_in_anon_vma, new_vma);
38a76013 3091 *vmap = vma = new_vma;
108d6642 3092 }
38a76013 3093 *need_rmap_locks = (new_vma->vm_pgoff <= vma->vm_pgoff);
1da177e4 3094 } else {
e94b1766 3095 new_vma = kmem_cache_alloc(vm_area_cachep, GFP_KERNEL);
e3975891
CG
3096 if (!new_vma)
3097 goto out;
3098 *new_vma = *vma;
3099 new_vma->vm_start = addr;
3100 new_vma->vm_end = addr + len;
3101 new_vma->vm_pgoff = pgoff;
3102 if (vma_dup_policy(vma, new_vma))
3103 goto out_free_vma;
3104 INIT_LIST_HEAD(&new_vma->anon_vma_chain);
3105 if (anon_vma_clone(new_vma, vma))
3106 goto out_free_mempol;
3107 if (new_vma->vm_file)
e14748e8 3108 vma_get_file(new_vma);
e3975891
CG
3109 if (new_vma->vm_ops && new_vma->vm_ops->open)
3110 new_vma->vm_ops->open(new_vma);
3111 vma_link(mm, new_vma, prev, rb_link, rb_parent);
3112 *need_rmap_locks = false;
1da177e4
LT
3113 }
3114 return new_vma;
5beb4930 3115
e3975891 3116out_free_mempol:
ef0855d3 3117 mpol_put(vma_policy(new_vma));
e3975891 3118out_free_vma:
5beb4930 3119 kmem_cache_free(vm_area_cachep, new_vma);
e3975891 3120out:
5beb4930 3121 return NULL;
1da177e4 3122}
119f657c 3123
3124/*
3125 * Return true if the calling process may expand its vm space by the passed
3126 * number of pages
3127 */
84638335 3128bool may_expand_vm(struct mm_struct *mm, vm_flags_t flags, unsigned long npages)
119f657c 3129{
84638335
KK
3130 if (mm->total_vm + npages > rlimit(RLIMIT_AS) >> PAGE_SHIFT)
3131 return false;
119f657c 3132
d977d56c
KK
3133 if (is_data_mapping(flags) &&
3134 mm->data_vm + npages > rlimit(RLIMIT_DATA) >> PAGE_SHIFT) {
f4fcd558
KK
3135 /* Workaround for Valgrind */
3136 if (rlimit(RLIMIT_DATA) == 0 &&
3137 mm->data_vm + npages <= rlimit_max(RLIMIT_DATA) >> PAGE_SHIFT)
3138 return true;
3139 if (!ignore_rlimit_data) {
3140 pr_warn_once("%s (%d): VmData %lu exceed data ulimit %lu. Update limits or use boot option ignore_rlimit_data.\n",
d977d56c
KK
3141 current->comm, current->pid,
3142 (mm->data_vm + npages) << PAGE_SHIFT,
3143 rlimit(RLIMIT_DATA));
d977d56c 3144 return false;
f4fcd558 3145 }
d977d56c 3146 }
119f657c 3147
84638335
KK
3148 return true;
3149}
3150
3151void vm_stat_account(struct mm_struct *mm, vm_flags_t flags, long npages)
3152{
3153 mm->total_vm += npages;
3154
d977d56c 3155 if (is_exec_mapping(flags))
84638335 3156 mm->exec_vm += npages;
d977d56c 3157 else if (is_stack_mapping(flags))
84638335 3158 mm->stack_vm += npages;
d977d56c 3159 else if (is_data_mapping(flags))
84638335 3160 mm->data_vm += npages;
119f657c 3161}
fa5dc22f 3162
a62c34bd
AL
3163static int special_mapping_fault(struct vm_area_struct *vma,
3164 struct vm_fault *vmf);
3165
3166/*
3167 * Having a close hook prevents vma merging regardless of flags.
3168 */
3169static void special_mapping_close(struct vm_area_struct *vma)
3170{
3171}
3172
3173static const char *special_mapping_name(struct vm_area_struct *vma)
3174{
3175 return ((struct vm_special_mapping *)vma->vm_private_data)->name;
3176}
3177
b059a453
DS
3178static int special_mapping_mremap(struct vm_area_struct *new_vma)
3179{
3180 struct vm_special_mapping *sm = new_vma->vm_private_data;
3181
3182 if (sm->mremap)
3183 return sm->mremap(sm, new_vma);
3184 return 0;
3185}
3186
a62c34bd
AL
3187static const struct vm_operations_struct special_mapping_vmops = {
3188 .close = special_mapping_close,
3189 .fault = special_mapping_fault,
b059a453 3190 .mremap = special_mapping_mremap,
a62c34bd
AL
3191 .name = special_mapping_name,
3192};
3193
3194static const struct vm_operations_struct legacy_special_mapping_vmops = {
3195 .close = special_mapping_close,
3196 .fault = special_mapping_fault,
3197};
fa5dc22f 3198
b1d0e4f5
NP
3199static int special_mapping_fault(struct vm_area_struct *vma,
3200 struct vm_fault *vmf)
fa5dc22f 3201{
b1d0e4f5 3202 pgoff_t pgoff;
fa5dc22f
RM
3203 struct page **pages;
3204
f872f540 3205 if (vma->vm_ops == &legacy_special_mapping_vmops) {
a62c34bd 3206 pages = vma->vm_private_data;
f872f540
AL
3207 } else {
3208 struct vm_special_mapping *sm = vma->vm_private_data;
3209
3210 if (sm->fault)
3211 return sm->fault(sm, vma, vmf);
3212
3213 pages = sm->pages;
3214 }
a62c34bd 3215
8a9cc3b5 3216 for (pgoff = vmf->pgoff; pgoff && *pages; ++pages)
b1d0e4f5 3217 pgoff--;
fa5dc22f
RM
3218
3219 if (*pages) {
3220 struct page *page = *pages;
3221 get_page(page);
b1d0e4f5
NP
3222 vmf->page = page;
3223 return 0;
fa5dc22f
RM
3224 }
3225
b1d0e4f5 3226 return VM_FAULT_SIGBUS;
fa5dc22f
RM
3227}
3228
a62c34bd
AL
3229static struct vm_area_struct *__install_special_mapping(
3230 struct mm_struct *mm,
3231 unsigned long addr, unsigned long len,
27f28b97
CG
3232 unsigned long vm_flags, void *priv,
3233 const struct vm_operations_struct *ops)
fa5dc22f 3234{
462e635e 3235 int ret;
fa5dc22f
RM
3236 struct vm_area_struct *vma;
3237
3238 vma = kmem_cache_zalloc(vm_area_cachep, GFP_KERNEL);
3239 if (unlikely(vma == NULL))
3935ed6a 3240 return ERR_PTR(-ENOMEM);
fa5dc22f 3241
5beb4930 3242 INIT_LIST_HEAD(&vma->anon_vma_chain);
fa5dc22f
RM
3243 vma->vm_mm = mm;
3244 vma->vm_start = addr;
3245 vma->vm_end = addr + len;
3246
d9104d1c 3247 vma->vm_flags = vm_flags | mm->def_flags | VM_DONTEXPAND | VM_SOFTDIRTY;
3ed75eb8 3248 vma->vm_page_prot = vm_get_page_prot(vma->vm_flags);
fa5dc22f 3249
a62c34bd
AL
3250 vma->vm_ops = ops;
3251 vma->vm_private_data = priv;
fa5dc22f 3252
462e635e
TO
3253 ret = insert_vm_struct(mm, vma);
3254 if (ret)
3255 goto out;
fa5dc22f 3256
84638335 3257 vm_stat_account(mm, vma->vm_flags, len >> PAGE_SHIFT);
fa5dc22f 3258
cdd6c482 3259 perf_event_mmap(vma);
089dd79d 3260
3935ed6a 3261 return vma;
462e635e
TO
3262
3263out:
3264 kmem_cache_free(vm_area_cachep, vma);
3935ed6a
SS
3265 return ERR_PTR(ret);
3266}
3267
2eefd878
DS
3268bool vma_is_special_mapping(const struct vm_area_struct *vma,
3269 const struct vm_special_mapping *sm)
3270{
3271 return vma->vm_private_data == sm &&
3272 (vma->vm_ops == &special_mapping_vmops ||
3273 vma->vm_ops == &legacy_special_mapping_vmops);
3274}
3275
a62c34bd
AL
3276/*
3277 * Called with mm->mmap_sem held for writing.
3278 * Insert a new vma covering the given region, with the given flags.
3279 * Its pages are supplied by the given array of struct page *.
3280 * The array can be shorter than len >> PAGE_SHIFT if it's null-terminated.
3281 * The region past the last page supplied will always produce SIGBUS.
3282 * The array pointer and the pages it points to are assumed to stay alive
3283 * for as long as this mapping might exist.
3284 */
3285struct vm_area_struct *_install_special_mapping(
3286 struct mm_struct *mm,
3287 unsigned long addr, unsigned long len,
3288 unsigned long vm_flags, const struct vm_special_mapping *spec)
3289{
27f28b97
CG
3290 return __install_special_mapping(mm, addr, len, vm_flags, (void *)spec,
3291 &special_mapping_vmops);
a62c34bd
AL
3292}
3293
3935ed6a
SS
3294int install_special_mapping(struct mm_struct *mm,
3295 unsigned long addr, unsigned long len,
3296 unsigned long vm_flags, struct page **pages)
3297{
a62c34bd 3298 struct vm_area_struct *vma = __install_special_mapping(
27f28b97
CG
3299 mm, addr, len, vm_flags, (void *)pages,
3300 &legacy_special_mapping_vmops);
3935ed6a 3301
14bd5b45 3302 return PTR_ERR_OR_ZERO(vma);
fa5dc22f 3303}
7906d00c
AA
3304
3305static DEFINE_MUTEX(mm_all_locks_mutex);
3306
454ed842 3307static void vm_lock_anon_vma(struct mm_struct *mm, struct anon_vma *anon_vma)
7906d00c 3308{
bf181b9f 3309 if (!test_bit(0, (unsigned long *) &anon_vma->root->rb_root.rb_node)) {
7906d00c
AA
3310 /*
3311 * The LSB of head.next can't change from under us
3312 * because we hold the mm_all_locks_mutex.
3313 */
572043c9 3314 down_write_nest_lock(&anon_vma->root->rwsem, &mm->mmap_sem);
7906d00c
AA
3315 /*
3316 * We can safely modify head.next after taking the
5a505085 3317 * anon_vma->root->rwsem. If some other vma in this mm shares
7906d00c
AA
3318 * the same anon_vma we won't take it again.
3319 *
3320 * No need of atomic instructions here, head.next
3321 * can't change from under us thanks to the
5a505085 3322 * anon_vma->root->rwsem.
7906d00c
AA
3323 */
3324 if (__test_and_set_bit(0, (unsigned long *)
bf181b9f 3325 &anon_vma->root->rb_root.rb_node))
7906d00c
AA
3326 BUG();
3327 }
3328}
3329
454ed842 3330static void vm_lock_mapping(struct mm_struct *mm, struct address_space *mapping)
7906d00c
AA
3331{
3332 if (!test_bit(AS_MM_ALL_LOCKS, &mapping->flags)) {
3333 /*
3334 * AS_MM_ALL_LOCKS can't change from under us because
3335 * we hold the mm_all_locks_mutex.
3336 *
3337 * Operations on ->flags have to be atomic because
3338 * even if AS_MM_ALL_LOCKS is stable thanks to the
3339 * mm_all_locks_mutex, there may be other cpus
3340 * changing other bitflags in parallel to us.
3341 */
3342 if (test_and_set_bit(AS_MM_ALL_LOCKS, &mapping->flags))
3343 BUG();
c8c06efa 3344 down_write_nest_lock(&mapping->i_mmap_rwsem, &mm->mmap_sem);
7906d00c
AA
3345 }
3346}
3347
3348/*
3349 * This operation locks against the VM for all pte/vma/mm related
3350 * operations that could ever happen on a certain mm. This includes
3351 * vmtruncate, try_to_unmap, and all page faults.
3352 *
3353 * The caller must take the mmap_sem in write mode before calling
3354 * mm_take_all_locks(). The caller isn't allowed to release the
3355 * mmap_sem until mm_drop_all_locks() returns.
3356 *
3357 * mmap_sem in write mode is required in order to block all operations
3358 * that could modify pagetables and free pages without need of
27ba0644 3359 * altering the vma layout. It's also needed in write mode to avoid new
7906d00c
AA
3360 * anon_vmas to be associated with existing vmas.
3361 *
3362 * A single task can't take more than one mm_take_all_locks() in a row
3363 * or it would deadlock.
3364 *
bf181b9f 3365 * The LSB in anon_vma->rb_root.rb_node and the AS_MM_ALL_LOCKS bitflag in
7906d00c
AA
3366 * mapping->flags avoid to take the same lock twice, if more than one
3367 * vma in this mm is backed by the same anon_vma or address_space.
3368 *
88f306b6
KS
3369 * We take locks in following order, accordingly to comment at beginning
3370 * of mm/rmap.c:
3371 * - all hugetlbfs_i_mmap_rwsem_key locks (aka mapping->i_mmap_rwsem for
3372 * hugetlb mapping);
3373 * - all i_mmap_rwsem locks;
3374 * - all anon_vma->rwseml
3375 *
3376 * We can take all locks within these types randomly because the VM code
3377 * doesn't nest them and we protected from parallel mm_take_all_locks() by
3378 * mm_all_locks_mutex.
7906d00c
AA
3379 *
3380 * mm_take_all_locks() and mm_drop_all_locks are expensive operations
3381 * that may have to take thousand of locks.
3382 *
3383 * mm_take_all_locks() can fail if it's interrupted by signals.
3384 */
3385int mm_take_all_locks(struct mm_struct *mm)
3386{
3387 struct vm_area_struct *vma;
5beb4930 3388 struct anon_vma_chain *avc;
7906d00c
AA
3389
3390 BUG_ON(down_read_trylock(&mm->mmap_sem));
3391
3392 mutex_lock(&mm_all_locks_mutex);
3393
3394 for (vma = mm->mmap; vma; vma = vma->vm_next) {
3395 if (signal_pending(current))
3396 goto out_unlock;
88f306b6
KS
3397 if (vma->vm_file && vma->vm_file->f_mapping &&
3398 is_vm_hugetlb_page(vma))
3399 vm_lock_mapping(mm, vma->vm_file->f_mapping);
3400 }
3401
3402 for (vma = mm->mmap; vma; vma = vma->vm_next) {
3403 if (signal_pending(current))
3404 goto out_unlock;
3405 if (vma->vm_file && vma->vm_file->f_mapping &&
3406 !is_vm_hugetlb_page(vma))
454ed842 3407 vm_lock_mapping(mm, vma->vm_file->f_mapping);
7906d00c 3408 }
7cd5a02f
PZ
3409
3410 for (vma = mm->mmap; vma; vma = vma->vm_next) {
3411 if (signal_pending(current))
3412 goto out_unlock;
3413 if (vma->anon_vma)
5beb4930
RR
3414 list_for_each_entry(avc, &vma->anon_vma_chain, same_vma)
3415 vm_lock_anon_vma(mm, avc->anon_vma);
7906d00c 3416 }
7cd5a02f 3417
584cff54 3418 return 0;
7906d00c
AA
3419
3420out_unlock:
584cff54
KC
3421 mm_drop_all_locks(mm);
3422 return -EINTR;
7906d00c
AA
3423}
3424
3425static void vm_unlock_anon_vma(struct anon_vma *anon_vma)
3426{
bf181b9f 3427 if (test_bit(0, (unsigned long *) &anon_vma->root->rb_root.rb_node)) {
7906d00c
AA
3428 /*
3429 * The LSB of head.next can't change to 0 from under
3430 * us because we hold the mm_all_locks_mutex.
3431 *
3432 * We must however clear the bitflag before unlocking
bf181b9f 3433 * the vma so the users using the anon_vma->rb_root will
7906d00c
AA
3434 * never see our bitflag.
3435 *
3436 * No need of atomic instructions here, head.next
3437 * can't change from under us until we release the
5a505085 3438 * anon_vma->root->rwsem.
7906d00c
AA
3439 */
3440 if (!__test_and_clear_bit(0, (unsigned long *)
bf181b9f 3441 &anon_vma->root->rb_root.rb_node))
7906d00c 3442 BUG();
08b52706 3443 anon_vma_unlock_write(anon_vma);
7906d00c
AA
3444 }
3445}
3446
3447static void vm_unlock_mapping(struct address_space *mapping)
3448{
3449 if (test_bit(AS_MM_ALL_LOCKS, &mapping->flags)) {
3450 /*
3451 * AS_MM_ALL_LOCKS can't change to 0 from under us
3452 * because we hold the mm_all_locks_mutex.
3453 */
83cde9e8 3454 i_mmap_unlock_write(mapping);
7906d00c
AA
3455 if (!test_and_clear_bit(AS_MM_ALL_LOCKS,
3456 &mapping->flags))
3457 BUG();
3458 }
3459}
3460
3461/*
3462 * The mmap_sem cannot be released by the caller until
3463 * mm_drop_all_locks() returns.
3464 */
3465void mm_drop_all_locks(struct mm_struct *mm)
3466{
3467 struct vm_area_struct *vma;
5beb4930 3468 struct anon_vma_chain *avc;
7906d00c
AA
3469
3470 BUG_ON(down_read_trylock(&mm->mmap_sem));
3471 BUG_ON(!mutex_is_locked(&mm_all_locks_mutex));
3472
3473 for (vma = mm->mmap; vma; vma = vma->vm_next) {
3474 if (vma->anon_vma)
5beb4930
RR
3475 list_for_each_entry(avc, &vma->anon_vma_chain, same_vma)
3476 vm_unlock_anon_vma(avc->anon_vma);
7906d00c
AA
3477 if (vma->vm_file && vma->vm_file->f_mapping)
3478 vm_unlock_mapping(vma->vm_file->f_mapping);
3479 }
3480
3481 mutex_unlock(&mm_all_locks_mutex);
3482}
8feae131
DH
3483
3484/*
3485 * initialise the VMA slab
3486 */
3487void __init mmap_init(void)
3488{
00a62ce9
KM
3489 int ret;
3490
908c7f19 3491 ret = percpu_counter_init(&vm_committed_as, 0, GFP_KERNEL);
00a62ce9 3492 VM_BUG_ON(ret);
8feae131 3493}
c9b1d098
AS
3494
3495/*
3496 * Initialise sysctl_user_reserve_kbytes.
3497 *
3498 * This is intended to prevent a user from starting a single memory hogging
3499 * process, such that they cannot recover (kill the hog) in OVERCOMMIT_NEVER
3500 * mode.
3501 *
3502 * The default value is min(3% of free memory, 128MB)
3503 * 128MB is enough to recover with sshd/login, bash, and top/kill.
3504 */
1640879a 3505static int init_user_reserve(void)
c9b1d098
AS
3506{
3507 unsigned long free_kbytes;
3508
3509 free_kbytes = global_page_state(NR_FREE_PAGES) << (PAGE_SHIFT - 10);
3510
3511 sysctl_user_reserve_kbytes = min(free_kbytes / 32, 1UL << 17);
3512 return 0;
3513}
a64fb3cd 3514subsys_initcall(init_user_reserve);
4eeab4f5
AS
3515
3516/*
3517 * Initialise sysctl_admin_reserve_kbytes.
3518 *
3519 * The purpose of sysctl_admin_reserve_kbytes is to allow the sys admin
3520 * to log in and kill a memory hogging process.
3521 *
3522 * Systems with more than 256MB will reserve 8MB, enough to recover
3523 * with sshd, bash, and top in OVERCOMMIT_GUESS. Smaller systems will
3524 * only reserve 3% of free pages by default.
3525 */
1640879a 3526static int init_admin_reserve(void)
4eeab4f5
AS
3527{
3528 unsigned long free_kbytes;
3529
3530 free_kbytes = global_page_state(NR_FREE_PAGES) << (PAGE_SHIFT - 10);
3531
3532 sysctl_admin_reserve_kbytes = min(free_kbytes / 32, 1UL << 13);
3533 return 0;
3534}
a64fb3cd 3535subsys_initcall(init_admin_reserve);
1640879a
AS
3536
3537/*
3538 * Reinititalise user and admin reserves if memory is added or removed.
3539 *
3540 * The default user reserve max is 128MB, and the default max for the
3541 * admin reserve is 8MB. These are usually, but not always, enough to
3542 * enable recovery from a memory hogging process using login/sshd, a shell,
3543 * and tools like top. It may make sense to increase or even disable the
3544 * reserve depending on the existence of swap or variations in the recovery
3545 * tools. So, the admin may have changed them.
3546 *
3547 * If memory is added and the reserves have been eliminated or increased above
3548 * the default max, then we'll trust the admin.
3549 *
3550 * If memory is removed and there isn't enough free memory, then we
3551 * need to reset the reserves.
3552 *
3553 * Otherwise keep the reserve set by the admin.
3554 */
3555static int reserve_mem_notifier(struct notifier_block *nb,
3556 unsigned long action, void *data)
3557{
3558 unsigned long tmp, free_kbytes;
3559
3560 switch (action) {
3561 case MEM_ONLINE:
3562 /* Default max is 128MB. Leave alone if modified by operator. */
3563 tmp = sysctl_user_reserve_kbytes;
3564 if (0 < tmp && tmp < (1UL << 17))
3565 init_user_reserve();
3566
3567 /* Default max is 8MB. Leave alone if modified by operator. */
3568 tmp = sysctl_admin_reserve_kbytes;
3569 if (0 < tmp && tmp < (1UL << 13))
3570 init_admin_reserve();
3571
3572 break;
3573 case MEM_OFFLINE:
3574 free_kbytes = global_page_state(NR_FREE_PAGES) << (PAGE_SHIFT - 10);
3575
3576 if (sysctl_user_reserve_kbytes > free_kbytes) {
3577 init_user_reserve();
3578 pr_info("vm.user_reserve_kbytes reset to %lu\n",
3579 sysctl_user_reserve_kbytes);
3580 }
3581
3582 if (sysctl_admin_reserve_kbytes > free_kbytes) {
3583 init_admin_reserve();
3584 pr_info("vm.admin_reserve_kbytes reset to %lu\n",
3585 sysctl_admin_reserve_kbytes);
3586 }
3587 break;
3588 default:
3589 break;
3590 }
3591 return NOTIFY_OK;
3592}
3593
3594static struct notifier_block reserve_mem_nb = {
3595 .notifier_call = reserve_mem_notifier,
3596};
3597
3598static int __meminit init_reserve_notifier(void)
3599{
3600 if (register_hotmemory_notifier(&reserve_mem_nb))
b1de0d13 3601 pr_err("Failed registering memory add/remove notifier for admin reserve\n");
1640879a
AS
3602
3603 return 0;
3604}
a64fb3cd 3605subsys_initcall(init_reserve_notifier);