8c8f0b1b700c602ccbb7e792ad1604846f74db37
[linux-2.6-microblaze.git] / mm / shmem.c
1 /*
2  * Resizable virtual memory filesystem for Linux.
3  *
4  * Copyright (C) 2000 Linus Torvalds.
5  *               2000 Transmeta Corp.
6  *               2000-2001 Christoph Rohland
7  *               2000-2001 SAP AG
8  *               2002 Red Hat Inc.
9  * Copyright (C) 2002-2011 Hugh Dickins.
10  * Copyright (C) 2011 Google Inc.
11  * Copyright (C) 2002-2005 VERITAS Software Corporation.
12  * Copyright (C) 2004 Andi Kleen, SuSE Labs
13  *
14  * Extended attribute support for tmpfs:
15  * Copyright (c) 2004, Luke Kenneth Casson Leighton <lkcl@lkcl.net>
16  * Copyright (c) 2004 Red Hat, Inc., James Morris <jmorris@redhat.com>
17  *
18  * tiny-shmem:
19  * Copyright (c) 2004, 2008 Matt Mackall <mpm@selenic.com>
20  *
21  * This file is released under the GPL.
22  */
23
24 #include <linux/fs.h>
25 #include <linux/init.h>
26 #include <linux/vfs.h>
27 #include <linux/mount.h>
28 #include <linux/ramfs.h>
29 #include <linux/pagemap.h>
30 #include <linux/file.h>
31 #include <linux/fileattr.h>
32 #include <linux/mm.h>
33 #include <linux/random.h>
34 #include <linux/sched/signal.h>
35 #include <linux/export.h>
36 #include <linux/shmem_fs.h>
37 #include <linux/swap.h>
38 #include <linux/uio.h>
39 #include <linux/hugetlb.h>
40 #include <linux/fs_parser.h>
41 #include <linux/swapfile.h>
42 #include <linux/iversion.h>
43 #include "swap.h"
44
45 static struct vfsmount *shm_mnt;
46
47 #ifdef CONFIG_SHMEM
48 /*
49  * This virtual memory filesystem is heavily based on the ramfs. It
50  * extends ramfs by the ability to use swap and honor resource limits
51  * which makes it a completely usable filesystem.
52  */
53
54 #include <linux/xattr.h>
55 #include <linux/exportfs.h>
56 #include <linux/posix_acl.h>
57 #include <linux/posix_acl_xattr.h>
58 #include <linux/mman.h>
59 #include <linux/string.h>
60 #include <linux/slab.h>
61 #include <linux/backing-dev.h>
62 #include <linux/writeback.h>
63 #include <linux/pagevec.h>
64 #include <linux/percpu_counter.h>
65 #include <linux/falloc.h>
66 #include <linux/splice.h>
67 #include <linux/security.h>
68 #include <linux/swapops.h>
69 #include <linux/mempolicy.h>
70 #include <linux/namei.h>
71 #include <linux/ctype.h>
72 #include <linux/migrate.h>
73 #include <linux/highmem.h>
74 #include <linux/seq_file.h>
75 #include <linux/magic.h>
76 #include <linux/syscalls.h>
77 #include <linux/fcntl.h>
78 #include <uapi/linux/memfd.h>
79 #include <linux/rmap.h>
80 #include <linux/uuid.h>
81 #include <linux/quotaops.h>
82
83 #include <linux/uaccess.h>
84
85 #include "internal.h"
86
87 #define BLOCKS_PER_PAGE  (PAGE_SIZE/512)
88 #define VM_ACCT(size)    (PAGE_ALIGN(size) >> PAGE_SHIFT)
89
90 /* Pretend that each entry is of this size in directory's i_size */
91 #define BOGO_DIRENT_SIZE 20
92
93 /* Pretend that one inode + its dentry occupy this much memory */
94 #define BOGO_INODE_SIZE 1024
95
96 /* Symlink up to this size is kmalloc'ed instead of using a swappable page */
97 #define SHORT_SYMLINK_LEN 128
98
99 /*
100  * shmem_fallocate communicates with shmem_fault or shmem_writepage via
101  * inode->i_private (with i_rwsem making sure that it has only one user at
102  * a time): we would prefer not to enlarge the shmem inode just for that.
103  */
104 struct shmem_falloc {
105         wait_queue_head_t *waitq; /* faults into hole wait for punch to end */
106         pgoff_t start;          /* start of range currently being fallocated */
107         pgoff_t next;           /* the next page offset to be fallocated */
108         pgoff_t nr_falloced;    /* how many new pages have been fallocated */
109         pgoff_t nr_unswapped;   /* how often writepage refused to swap out */
110 };
111
112 struct shmem_options {
113         unsigned long long blocks;
114         unsigned long long inodes;
115         struct mempolicy *mpol;
116         kuid_t uid;
117         kgid_t gid;
118         umode_t mode;
119         bool full_inums;
120         int huge;
121         int seen;
122         bool noswap;
123         unsigned short quota_types;
124         struct shmem_quota_limits qlimits;
125 #define SHMEM_SEEN_BLOCKS 1
126 #define SHMEM_SEEN_INODES 2
127 #define SHMEM_SEEN_HUGE 4
128 #define SHMEM_SEEN_INUMS 8
129 #define SHMEM_SEEN_NOSWAP 16
130 #define SHMEM_SEEN_QUOTA 32
131 };
132
133 #ifdef CONFIG_TMPFS
134 static unsigned long shmem_default_max_blocks(void)
135 {
136         return totalram_pages() / 2;
137 }
138
139 static unsigned long shmem_default_max_inodes(void)
140 {
141         unsigned long nr_pages = totalram_pages();
142
143         return min3(nr_pages - totalhigh_pages(), nr_pages / 2,
144                         ULONG_MAX / BOGO_INODE_SIZE);
145 }
146 #endif
147
148 static int shmem_swapin_folio(struct inode *inode, pgoff_t index,
149                              struct folio **foliop, enum sgp_type sgp,
150                              gfp_t gfp, struct vm_area_struct *vma,
151                              vm_fault_t *fault_type);
152
153 static inline struct shmem_sb_info *SHMEM_SB(struct super_block *sb)
154 {
155         return sb->s_fs_info;
156 }
157
158 /*
159  * shmem_file_setup pre-accounts the whole fixed size of a VM object,
160  * for shared memory and for shared anonymous (/dev/zero) mappings
161  * (unless MAP_NORESERVE and sysctl_overcommit_memory <= 1),
162  * consistent with the pre-accounting of private mappings ...
163  */
164 static inline int shmem_acct_size(unsigned long flags, loff_t size)
165 {
166         return (flags & VM_NORESERVE) ?
167                 0 : security_vm_enough_memory_mm(current->mm, VM_ACCT(size));
168 }
169
170 static inline void shmem_unacct_size(unsigned long flags, loff_t size)
171 {
172         if (!(flags & VM_NORESERVE))
173                 vm_unacct_memory(VM_ACCT(size));
174 }
175
176 static inline int shmem_reacct_size(unsigned long flags,
177                 loff_t oldsize, loff_t newsize)
178 {
179         if (!(flags & VM_NORESERVE)) {
180                 if (VM_ACCT(newsize) > VM_ACCT(oldsize))
181                         return security_vm_enough_memory_mm(current->mm,
182                                         VM_ACCT(newsize) - VM_ACCT(oldsize));
183                 else if (VM_ACCT(newsize) < VM_ACCT(oldsize))
184                         vm_unacct_memory(VM_ACCT(oldsize) - VM_ACCT(newsize));
185         }
186         return 0;
187 }
188
189 /*
190  * ... whereas tmpfs objects are accounted incrementally as
191  * pages are allocated, in order to allow large sparse files.
192  * shmem_get_folio reports shmem_acct_block failure as -ENOSPC not -ENOMEM,
193  * so that a failure on a sparse tmpfs mapping will give SIGBUS not OOM.
194  */
195 static inline int shmem_acct_block(unsigned long flags, long pages)
196 {
197         if (!(flags & VM_NORESERVE))
198                 return 0;
199
200         return security_vm_enough_memory_mm(current->mm,
201                         pages * VM_ACCT(PAGE_SIZE));
202 }
203
204 static inline void shmem_unacct_blocks(unsigned long flags, long pages)
205 {
206         if (flags & VM_NORESERVE)
207                 vm_unacct_memory(pages * VM_ACCT(PAGE_SIZE));
208 }
209
210 static int shmem_inode_acct_block(struct inode *inode, long pages)
211 {
212         struct shmem_inode_info *info = SHMEM_I(inode);
213         struct shmem_sb_info *sbinfo = SHMEM_SB(inode->i_sb);
214         int err = -ENOSPC;
215
216         if (shmem_acct_block(info->flags, pages))
217                 return err;
218
219         might_sleep();  /* when quotas */
220         if (sbinfo->max_blocks) {
221                 if (percpu_counter_compare(&sbinfo->used_blocks,
222                                            sbinfo->max_blocks - pages) > 0)
223                         goto unacct;
224
225                 err = dquot_alloc_block_nodirty(inode, pages);
226                 if (err)
227                         goto unacct;
228
229                 percpu_counter_add(&sbinfo->used_blocks, pages);
230         } else {
231                 err = dquot_alloc_block_nodirty(inode, pages);
232                 if (err)
233                         goto unacct;
234         }
235
236         return 0;
237
238 unacct:
239         shmem_unacct_blocks(info->flags, pages);
240         return err;
241 }
242
243 static void shmem_inode_unacct_blocks(struct inode *inode, long pages)
244 {
245         struct shmem_inode_info *info = SHMEM_I(inode);
246         struct shmem_sb_info *sbinfo = SHMEM_SB(inode->i_sb);
247
248         might_sleep();  /* when quotas */
249         dquot_free_block_nodirty(inode, pages);
250
251         if (sbinfo->max_blocks)
252                 percpu_counter_sub(&sbinfo->used_blocks, pages);
253         shmem_unacct_blocks(info->flags, pages);
254 }
255
256 static const struct super_operations shmem_ops;
257 const struct address_space_operations shmem_aops;
258 static const struct file_operations shmem_file_operations;
259 static const struct inode_operations shmem_inode_operations;
260 static const struct inode_operations shmem_dir_inode_operations;
261 static const struct inode_operations shmem_special_inode_operations;
262 static const struct vm_operations_struct shmem_vm_ops;
263 static const struct vm_operations_struct shmem_anon_vm_ops;
264 static struct file_system_type shmem_fs_type;
265
266 bool vma_is_anon_shmem(struct vm_area_struct *vma)
267 {
268         return vma->vm_ops == &shmem_anon_vm_ops;
269 }
270
271 bool vma_is_shmem(struct vm_area_struct *vma)
272 {
273         return vma_is_anon_shmem(vma) || vma->vm_ops == &shmem_vm_ops;
274 }
275
276 static LIST_HEAD(shmem_swaplist);
277 static DEFINE_MUTEX(shmem_swaplist_mutex);
278
279 #ifdef CONFIG_TMPFS_QUOTA
280
281 static int shmem_enable_quotas(struct super_block *sb,
282                                unsigned short quota_types)
283 {
284         int type, err = 0;
285
286         sb_dqopt(sb)->flags |= DQUOT_QUOTA_SYS_FILE | DQUOT_NOLIST_DIRTY;
287         for (type = 0; type < SHMEM_MAXQUOTAS; type++) {
288                 if (!(quota_types & (1 << type)))
289                         continue;
290                 err = dquot_load_quota_sb(sb, type, QFMT_SHMEM,
291                                           DQUOT_USAGE_ENABLED |
292                                           DQUOT_LIMITS_ENABLED);
293                 if (err)
294                         goto out_err;
295         }
296         return 0;
297
298 out_err:
299         pr_warn("tmpfs: failed to enable quota tracking (type=%d, err=%d)\n",
300                 type, err);
301         for (type--; type >= 0; type--)
302                 dquot_quota_off(sb, type);
303         return err;
304 }
305
306 static void shmem_disable_quotas(struct super_block *sb)
307 {
308         int type;
309
310         for (type = 0; type < SHMEM_MAXQUOTAS; type++)
311                 dquot_quota_off(sb, type);
312 }
313
314 static struct dquot **shmem_get_dquots(struct inode *inode)
315 {
316         return SHMEM_I(inode)->i_dquot;
317 }
318 #endif /* CONFIG_TMPFS_QUOTA */
319
320 /*
321  * shmem_reserve_inode() performs bookkeeping to reserve a shmem inode, and
322  * produces a novel ino for the newly allocated inode.
323  *
324  * It may also be called when making a hard link to permit the space needed by
325  * each dentry. However, in that case, no new inode number is needed since that
326  * internally draws from another pool of inode numbers (currently global
327  * get_next_ino()). This case is indicated by passing NULL as inop.
328  */
329 #define SHMEM_INO_BATCH 1024
330 static int shmem_reserve_inode(struct super_block *sb, ino_t *inop)
331 {
332         struct shmem_sb_info *sbinfo = SHMEM_SB(sb);
333         ino_t ino;
334
335         if (!(sb->s_flags & SB_KERNMOUNT)) {
336                 raw_spin_lock(&sbinfo->stat_lock);
337                 if (sbinfo->max_inodes) {
338                         if (sbinfo->free_ispace < BOGO_INODE_SIZE) {
339                                 raw_spin_unlock(&sbinfo->stat_lock);
340                                 return -ENOSPC;
341                         }
342                         sbinfo->free_ispace -= BOGO_INODE_SIZE;
343                 }
344                 if (inop) {
345                         ino = sbinfo->next_ino++;
346                         if (unlikely(is_zero_ino(ino)))
347                                 ino = sbinfo->next_ino++;
348                         if (unlikely(!sbinfo->full_inums &&
349                                      ino > UINT_MAX)) {
350                                 /*
351                                  * Emulate get_next_ino uint wraparound for
352                                  * compatibility
353                                  */
354                                 if (IS_ENABLED(CONFIG_64BIT))
355                                         pr_warn("%s: inode number overflow on device %d, consider using inode64 mount option\n",
356                                                 __func__, MINOR(sb->s_dev));
357                                 sbinfo->next_ino = 1;
358                                 ino = sbinfo->next_ino++;
359                         }
360                         *inop = ino;
361                 }
362                 raw_spin_unlock(&sbinfo->stat_lock);
363         } else if (inop) {
364                 /*
365                  * __shmem_file_setup, one of our callers, is lock-free: it
366                  * doesn't hold stat_lock in shmem_reserve_inode since
367                  * max_inodes is always 0, and is called from potentially
368                  * unknown contexts. As such, use a per-cpu batched allocator
369                  * which doesn't require the per-sb stat_lock unless we are at
370                  * the batch boundary.
371                  *
372                  * We don't need to worry about inode{32,64} since SB_KERNMOUNT
373                  * shmem mounts are not exposed to userspace, so we don't need
374                  * to worry about things like glibc compatibility.
375                  */
376                 ino_t *next_ino;
377
378                 next_ino = per_cpu_ptr(sbinfo->ino_batch, get_cpu());
379                 ino = *next_ino;
380                 if (unlikely(ino % SHMEM_INO_BATCH == 0)) {
381                         raw_spin_lock(&sbinfo->stat_lock);
382                         ino = sbinfo->next_ino;
383                         sbinfo->next_ino += SHMEM_INO_BATCH;
384                         raw_spin_unlock(&sbinfo->stat_lock);
385                         if (unlikely(is_zero_ino(ino)))
386                                 ino++;
387                 }
388                 *inop = ino;
389                 *next_ino = ++ino;
390                 put_cpu();
391         }
392
393         return 0;
394 }
395
396 static void shmem_free_inode(struct super_block *sb)
397 {
398         struct shmem_sb_info *sbinfo = SHMEM_SB(sb);
399         if (sbinfo->max_inodes) {
400                 raw_spin_lock(&sbinfo->stat_lock);
401                 sbinfo->free_ispace += BOGO_INODE_SIZE;
402                 raw_spin_unlock(&sbinfo->stat_lock);
403         }
404 }
405
406 /**
407  * shmem_recalc_inode - recalculate the block usage of an inode
408  * @inode: inode to recalc
409  * @alloced: the change in number of pages allocated to inode
410  * @swapped: the change in number of pages swapped from inode
411  *
412  * We have to calculate the free blocks since the mm can drop
413  * undirtied hole pages behind our back.
414  *
415  * But normally   info->alloced == inode->i_mapping->nrpages + info->swapped
416  * So mm freed is info->alloced - (inode->i_mapping->nrpages + info->swapped)
417  */
418 static void shmem_recalc_inode(struct inode *inode, long alloced, long swapped)
419 {
420         struct shmem_inode_info *info = SHMEM_I(inode);
421         long freed;
422
423         spin_lock(&info->lock);
424         info->alloced += alloced;
425         info->swapped += swapped;
426         freed = info->alloced - info->swapped -
427                 READ_ONCE(inode->i_mapping->nrpages);
428         /*
429          * Special case: whereas normally shmem_recalc_inode() is called
430          * after i_mapping->nrpages has already been adjusted (up or down),
431          * shmem_writepage() has to raise swapped before nrpages is lowered -
432          * to stop a racing shmem_recalc_inode() from thinking that a page has
433          * been freed.  Compensate here, to avoid the need for a followup call.
434          */
435         if (swapped > 0)
436                 freed += swapped;
437         if (freed > 0)
438                 info->alloced -= freed;
439         spin_unlock(&info->lock);
440
441         /* The quota case may block */
442         if (freed > 0)
443                 shmem_inode_unacct_blocks(inode, freed);
444 }
445
446 bool shmem_charge(struct inode *inode, long pages)
447 {
448         struct address_space *mapping = inode->i_mapping;
449
450         if (shmem_inode_acct_block(inode, pages))
451                 return false;
452
453         /* nrpages adjustment first, then shmem_recalc_inode() when balanced */
454         xa_lock_irq(&mapping->i_pages);
455         mapping->nrpages += pages;
456         xa_unlock_irq(&mapping->i_pages);
457
458         shmem_recalc_inode(inode, pages, 0);
459         return true;
460 }
461
462 void shmem_uncharge(struct inode *inode, long pages)
463 {
464         /* pages argument is currently unused: keep it to help debugging */
465         /* nrpages adjustment done by __filemap_remove_folio() or caller */
466
467         shmem_recalc_inode(inode, 0, 0);
468 }
469
470 /*
471  * Replace item expected in xarray by a new item, while holding xa_lock.
472  */
473 static int shmem_replace_entry(struct address_space *mapping,
474                         pgoff_t index, void *expected, void *replacement)
475 {
476         XA_STATE(xas, &mapping->i_pages, index);
477         void *item;
478
479         VM_BUG_ON(!expected);
480         VM_BUG_ON(!replacement);
481         item = xas_load(&xas);
482         if (item != expected)
483                 return -ENOENT;
484         xas_store(&xas, replacement);
485         return 0;
486 }
487
488 /*
489  * Sometimes, before we decide whether to proceed or to fail, we must check
490  * that an entry was not already brought back from swap by a racing thread.
491  *
492  * Checking page is not enough: by the time a SwapCache page is locked, it
493  * might be reused, and again be SwapCache, using the same swap as before.
494  */
495 static bool shmem_confirm_swap(struct address_space *mapping,
496                                pgoff_t index, swp_entry_t swap)
497 {
498         return xa_load(&mapping->i_pages, index) == swp_to_radix_entry(swap);
499 }
500
501 /*
502  * Definitions for "huge tmpfs": tmpfs mounted with the huge= option
503  *
504  * SHMEM_HUGE_NEVER:
505  *      disables huge pages for the mount;
506  * SHMEM_HUGE_ALWAYS:
507  *      enables huge pages for the mount;
508  * SHMEM_HUGE_WITHIN_SIZE:
509  *      only allocate huge pages if the page will be fully within i_size,
510  *      also respect fadvise()/madvise() hints;
511  * SHMEM_HUGE_ADVISE:
512  *      only allocate huge pages if requested with fadvise()/madvise();
513  */
514
515 #define SHMEM_HUGE_NEVER        0
516 #define SHMEM_HUGE_ALWAYS       1
517 #define SHMEM_HUGE_WITHIN_SIZE  2
518 #define SHMEM_HUGE_ADVISE       3
519
520 /*
521  * Special values.
522  * Only can be set via /sys/kernel/mm/transparent_hugepage/shmem_enabled:
523  *
524  * SHMEM_HUGE_DENY:
525  *      disables huge on shm_mnt and all mounts, for emergency use;
526  * SHMEM_HUGE_FORCE:
527  *      enables huge on shm_mnt and all mounts, w/o needing option, for testing;
528  *
529  */
530 #define SHMEM_HUGE_DENY         (-1)
531 #define SHMEM_HUGE_FORCE        (-2)
532
533 #ifdef CONFIG_TRANSPARENT_HUGEPAGE
534 /* ifdef here to avoid bloating shmem.o when not necessary */
535
536 static int shmem_huge __read_mostly = SHMEM_HUGE_NEVER;
537
538 bool shmem_is_huge(struct inode *inode, pgoff_t index, bool shmem_huge_force,
539                    struct mm_struct *mm, unsigned long vm_flags)
540 {
541         loff_t i_size;
542
543         if (!S_ISREG(inode->i_mode))
544                 return false;
545         if (mm && ((vm_flags & VM_NOHUGEPAGE) || test_bit(MMF_DISABLE_THP, &mm->flags)))
546                 return false;
547         if (shmem_huge == SHMEM_HUGE_DENY)
548                 return false;
549         if (shmem_huge_force || shmem_huge == SHMEM_HUGE_FORCE)
550                 return true;
551
552         switch (SHMEM_SB(inode->i_sb)->huge) {
553         case SHMEM_HUGE_ALWAYS:
554                 return true;
555         case SHMEM_HUGE_WITHIN_SIZE:
556                 index = round_up(index + 1, HPAGE_PMD_NR);
557                 i_size = round_up(i_size_read(inode), PAGE_SIZE);
558                 if (i_size >> PAGE_SHIFT >= index)
559                         return true;
560                 fallthrough;
561         case SHMEM_HUGE_ADVISE:
562                 if (mm && (vm_flags & VM_HUGEPAGE))
563                         return true;
564                 fallthrough;
565         default:
566                 return false;
567         }
568 }
569
570 #if defined(CONFIG_SYSFS)
571 static int shmem_parse_huge(const char *str)
572 {
573         if (!strcmp(str, "never"))
574                 return SHMEM_HUGE_NEVER;
575         if (!strcmp(str, "always"))
576                 return SHMEM_HUGE_ALWAYS;
577         if (!strcmp(str, "within_size"))
578                 return SHMEM_HUGE_WITHIN_SIZE;
579         if (!strcmp(str, "advise"))
580                 return SHMEM_HUGE_ADVISE;
581         if (!strcmp(str, "deny"))
582                 return SHMEM_HUGE_DENY;
583         if (!strcmp(str, "force"))
584                 return SHMEM_HUGE_FORCE;
585         return -EINVAL;
586 }
587 #endif
588
589 #if defined(CONFIG_SYSFS) || defined(CONFIG_TMPFS)
590 static const char *shmem_format_huge(int huge)
591 {
592         switch (huge) {
593         case SHMEM_HUGE_NEVER:
594                 return "never";
595         case SHMEM_HUGE_ALWAYS:
596                 return "always";
597         case SHMEM_HUGE_WITHIN_SIZE:
598                 return "within_size";
599         case SHMEM_HUGE_ADVISE:
600                 return "advise";
601         case SHMEM_HUGE_DENY:
602                 return "deny";
603         case SHMEM_HUGE_FORCE:
604                 return "force";
605         default:
606                 VM_BUG_ON(1);
607                 return "bad_val";
608         }
609 }
610 #endif
611
612 static unsigned long shmem_unused_huge_shrink(struct shmem_sb_info *sbinfo,
613                 struct shrink_control *sc, unsigned long nr_to_split)
614 {
615         LIST_HEAD(list), *pos, *next;
616         LIST_HEAD(to_remove);
617         struct inode *inode;
618         struct shmem_inode_info *info;
619         struct folio *folio;
620         unsigned long batch = sc ? sc->nr_to_scan : 128;
621         int split = 0;
622
623         if (list_empty(&sbinfo->shrinklist))
624                 return SHRINK_STOP;
625
626         spin_lock(&sbinfo->shrinklist_lock);
627         list_for_each_safe(pos, next, &sbinfo->shrinklist) {
628                 info = list_entry(pos, struct shmem_inode_info, shrinklist);
629
630                 /* pin the inode */
631                 inode = igrab(&info->vfs_inode);
632
633                 /* inode is about to be evicted */
634                 if (!inode) {
635                         list_del_init(&info->shrinklist);
636                         goto next;
637                 }
638
639                 /* Check if there's anything to gain */
640                 if (round_up(inode->i_size, PAGE_SIZE) ==
641                                 round_up(inode->i_size, HPAGE_PMD_SIZE)) {
642                         list_move(&info->shrinklist, &to_remove);
643                         goto next;
644                 }
645
646                 list_move(&info->shrinklist, &list);
647 next:
648                 sbinfo->shrinklist_len--;
649                 if (!--batch)
650                         break;
651         }
652         spin_unlock(&sbinfo->shrinklist_lock);
653
654         list_for_each_safe(pos, next, &to_remove) {
655                 info = list_entry(pos, struct shmem_inode_info, shrinklist);
656                 inode = &info->vfs_inode;
657                 list_del_init(&info->shrinklist);
658                 iput(inode);
659         }
660
661         list_for_each_safe(pos, next, &list) {
662                 int ret;
663                 pgoff_t index;
664
665                 info = list_entry(pos, struct shmem_inode_info, shrinklist);
666                 inode = &info->vfs_inode;
667
668                 if (nr_to_split && split >= nr_to_split)
669                         goto move_back;
670
671                 index = (inode->i_size & HPAGE_PMD_MASK) >> PAGE_SHIFT;
672                 folio = filemap_get_folio(inode->i_mapping, index);
673                 if (IS_ERR(folio))
674                         goto drop;
675
676                 /* No huge page at the end of the file: nothing to split */
677                 if (!folio_test_large(folio)) {
678                         folio_put(folio);
679                         goto drop;
680                 }
681
682                 /*
683                  * Move the inode on the list back to shrinklist if we failed
684                  * to lock the page at this time.
685                  *
686                  * Waiting for the lock may lead to deadlock in the
687                  * reclaim path.
688                  */
689                 if (!folio_trylock(folio)) {
690                         folio_put(folio);
691                         goto move_back;
692                 }
693
694                 ret = split_folio(folio);
695                 folio_unlock(folio);
696                 folio_put(folio);
697
698                 /* If split failed move the inode on the list back to shrinklist */
699                 if (ret)
700                         goto move_back;
701
702                 split++;
703 drop:
704                 list_del_init(&info->shrinklist);
705                 goto put;
706 move_back:
707                 /*
708                  * Make sure the inode is either on the global list or deleted
709                  * from any local list before iput() since it could be deleted
710                  * in another thread once we put the inode (then the local list
711                  * is corrupted).
712                  */
713                 spin_lock(&sbinfo->shrinklist_lock);
714                 list_move(&info->shrinklist, &sbinfo->shrinklist);
715                 sbinfo->shrinklist_len++;
716                 spin_unlock(&sbinfo->shrinklist_lock);
717 put:
718                 iput(inode);
719         }
720
721         return split;
722 }
723
724 static long shmem_unused_huge_scan(struct super_block *sb,
725                 struct shrink_control *sc)
726 {
727         struct shmem_sb_info *sbinfo = SHMEM_SB(sb);
728
729         if (!READ_ONCE(sbinfo->shrinklist_len))
730                 return SHRINK_STOP;
731
732         return shmem_unused_huge_shrink(sbinfo, sc, 0);
733 }
734
735 static long shmem_unused_huge_count(struct super_block *sb,
736                 struct shrink_control *sc)
737 {
738         struct shmem_sb_info *sbinfo = SHMEM_SB(sb);
739         return READ_ONCE(sbinfo->shrinklist_len);
740 }
741 #else /* !CONFIG_TRANSPARENT_HUGEPAGE */
742
743 #define shmem_huge SHMEM_HUGE_DENY
744
745 bool shmem_is_huge(struct inode *inode, pgoff_t index, bool shmem_huge_force,
746                    struct mm_struct *mm, unsigned long vm_flags)
747 {
748         return false;
749 }
750
751 static unsigned long shmem_unused_huge_shrink(struct shmem_sb_info *sbinfo,
752                 struct shrink_control *sc, unsigned long nr_to_split)
753 {
754         return 0;
755 }
756 #endif /* CONFIG_TRANSPARENT_HUGEPAGE */
757
758 /*
759  * Like filemap_add_folio, but error if expected item has gone.
760  */
761 static int shmem_add_to_page_cache(struct folio *folio,
762                                    struct address_space *mapping,
763                                    pgoff_t index, void *expected, gfp_t gfp,
764                                    struct mm_struct *charge_mm)
765 {
766         XA_STATE_ORDER(xas, &mapping->i_pages, index, folio_order(folio));
767         long nr = folio_nr_pages(folio);
768         int error;
769
770         VM_BUG_ON_FOLIO(index != round_down(index, nr), folio);
771         VM_BUG_ON_FOLIO(!folio_test_locked(folio), folio);
772         VM_BUG_ON_FOLIO(!folio_test_swapbacked(folio), folio);
773         VM_BUG_ON(expected && folio_test_large(folio));
774
775         folio_ref_add(folio, nr);
776         folio->mapping = mapping;
777         folio->index = index;
778
779         if (!folio_test_swapcache(folio)) {
780                 error = mem_cgroup_charge(folio, charge_mm, gfp);
781                 if (error) {
782                         if (folio_test_pmd_mappable(folio)) {
783                                 count_vm_event(THP_FILE_FALLBACK);
784                                 count_vm_event(THP_FILE_FALLBACK_CHARGE);
785                         }
786                         goto error;
787                 }
788         }
789         folio_throttle_swaprate(folio, gfp);
790
791         do {
792                 xas_lock_irq(&xas);
793                 if (expected != xas_find_conflict(&xas)) {
794                         xas_set_err(&xas, -EEXIST);
795                         goto unlock;
796                 }
797                 if (expected && xas_find_conflict(&xas)) {
798                         xas_set_err(&xas, -EEXIST);
799                         goto unlock;
800                 }
801                 xas_store(&xas, folio);
802                 if (xas_error(&xas))
803                         goto unlock;
804                 if (folio_test_pmd_mappable(folio)) {
805                         count_vm_event(THP_FILE_ALLOC);
806                         __lruvec_stat_mod_folio(folio, NR_SHMEM_THPS, nr);
807                 }
808                 mapping->nrpages += nr;
809                 __lruvec_stat_mod_folio(folio, NR_FILE_PAGES, nr);
810                 __lruvec_stat_mod_folio(folio, NR_SHMEM, nr);
811 unlock:
812                 xas_unlock_irq(&xas);
813         } while (xas_nomem(&xas, gfp));
814
815         if (xas_error(&xas)) {
816                 error = xas_error(&xas);
817                 goto error;
818         }
819
820         return 0;
821 error:
822         folio->mapping = NULL;
823         folio_ref_sub(folio, nr);
824         return error;
825 }
826
827 /*
828  * Like delete_from_page_cache, but substitutes swap for @folio.
829  */
830 static void shmem_delete_from_page_cache(struct folio *folio, void *radswap)
831 {
832         struct address_space *mapping = folio->mapping;
833         long nr = folio_nr_pages(folio);
834         int error;
835
836         xa_lock_irq(&mapping->i_pages);
837         error = shmem_replace_entry(mapping, folio->index, folio, radswap);
838         folio->mapping = NULL;
839         mapping->nrpages -= nr;
840         __lruvec_stat_mod_folio(folio, NR_FILE_PAGES, -nr);
841         __lruvec_stat_mod_folio(folio, NR_SHMEM, -nr);
842         xa_unlock_irq(&mapping->i_pages);
843         folio_put(folio);
844         BUG_ON(error);
845 }
846
847 /*
848  * Remove swap entry from page cache, free the swap and its page cache.
849  */
850 static int shmem_free_swap(struct address_space *mapping,
851                            pgoff_t index, void *radswap)
852 {
853         void *old;
854
855         old = xa_cmpxchg_irq(&mapping->i_pages, index, radswap, NULL, 0);
856         if (old != radswap)
857                 return -ENOENT;
858         free_swap_and_cache(radix_to_swp_entry(radswap));
859         return 0;
860 }
861
862 /*
863  * Determine (in bytes) how many of the shmem object's pages mapped by the
864  * given offsets are swapped out.
865  *
866  * This is safe to call without i_rwsem or the i_pages lock thanks to RCU,
867  * as long as the inode doesn't go away and racy results are not a problem.
868  */
869 unsigned long shmem_partial_swap_usage(struct address_space *mapping,
870                                                 pgoff_t start, pgoff_t end)
871 {
872         XA_STATE(xas, &mapping->i_pages, start);
873         struct page *page;
874         unsigned long swapped = 0;
875
876         rcu_read_lock();
877         xas_for_each(&xas, page, end - 1) {
878                 if (xas_retry(&xas, page))
879                         continue;
880                 if (xa_is_value(page))
881                         swapped++;
882
883                 if (need_resched()) {
884                         xas_pause(&xas);
885                         cond_resched_rcu();
886                 }
887         }
888
889         rcu_read_unlock();
890
891         return swapped << PAGE_SHIFT;
892 }
893
894 /*
895  * Determine (in bytes) how many of the shmem object's pages mapped by the
896  * given vma is swapped out.
897  *
898  * This is safe to call without i_rwsem or the i_pages lock thanks to RCU,
899  * as long as the inode doesn't go away and racy results are not a problem.
900  */
901 unsigned long shmem_swap_usage(struct vm_area_struct *vma)
902 {
903         struct inode *inode = file_inode(vma->vm_file);
904         struct shmem_inode_info *info = SHMEM_I(inode);
905         struct address_space *mapping = inode->i_mapping;
906         unsigned long swapped;
907
908         /* Be careful as we don't hold info->lock */
909         swapped = READ_ONCE(info->swapped);
910
911         /*
912          * The easier cases are when the shmem object has nothing in swap, or
913          * the vma maps it whole. Then we can simply use the stats that we
914          * already track.
915          */
916         if (!swapped)
917                 return 0;
918
919         if (!vma->vm_pgoff && vma->vm_end - vma->vm_start >= inode->i_size)
920                 return swapped << PAGE_SHIFT;
921
922         /* Here comes the more involved part */
923         return shmem_partial_swap_usage(mapping, vma->vm_pgoff,
924                                         vma->vm_pgoff + vma_pages(vma));
925 }
926
927 /*
928  * SysV IPC SHM_UNLOCK restore Unevictable pages to their evictable lists.
929  */
930 void shmem_unlock_mapping(struct address_space *mapping)
931 {
932         struct folio_batch fbatch;
933         pgoff_t index = 0;
934
935         folio_batch_init(&fbatch);
936         /*
937          * Minor point, but we might as well stop if someone else SHM_LOCKs it.
938          */
939         while (!mapping_unevictable(mapping) &&
940                filemap_get_folios(mapping, &index, ~0UL, &fbatch)) {
941                 check_move_unevictable_folios(&fbatch);
942                 folio_batch_release(&fbatch);
943                 cond_resched();
944         }
945 }
946
947 static struct folio *shmem_get_partial_folio(struct inode *inode, pgoff_t index)
948 {
949         struct folio *folio;
950
951         /*
952          * At first avoid shmem_get_folio(,,,SGP_READ): that fails
953          * beyond i_size, and reports fallocated folios as holes.
954          */
955         folio = filemap_get_entry(inode->i_mapping, index);
956         if (!folio)
957                 return folio;
958         if (!xa_is_value(folio)) {
959                 folio_lock(folio);
960                 if (folio->mapping == inode->i_mapping)
961                         return folio;
962                 /* The folio has been swapped out */
963                 folio_unlock(folio);
964                 folio_put(folio);
965         }
966         /*
967          * But read a folio back from swap if any of it is within i_size
968          * (although in some cases this is just a waste of time).
969          */
970         folio = NULL;
971         shmem_get_folio(inode, index, &folio, SGP_READ);
972         return folio;
973 }
974
975 /*
976  * Remove range of pages and swap entries from page cache, and free them.
977  * If !unfalloc, truncate or punch hole; if unfalloc, undo failed fallocate.
978  */
979 static void shmem_undo_range(struct inode *inode, loff_t lstart, loff_t lend,
980                                                                  bool unfalloc)
981 {
982         struct address_space *mapping = inode->i_mapping;
983         struct shmem_inode_info *info = SHMEM_I(inode);
984         pgoff_t start = (lstart + PAGE_SIZE - 1) >> PAGE_SHIFT;
985         pgoff_t end = (lend + 1) >> PAGE_SHIFT;
986         struct folio_batch fbatch;
987         pgoff_t indices[PAGEVEC_SIZE];
988         struct folio *folio;
989         bool same_folio;
990         long nr_swaps_freed = 0;
991         pgoff_t index;
992         int i;
993
994         if (lend == -1)
995                 end = -1;       /* unsigned, so actually very big */
996
997         if (info->fallocend > start && info->fallocend <= end && !unfalloc)
998                 info->fallocend = start;
999
1000         folio_batch_init(&fbatch);
1001         index = start;
1002         while (index < end && find_lock_entries(mapping, &index, end - 1,
1003                         &fbatch, indices)) {
1004                 for (i = 0; i < folio_batch_count(&fbatch); i++) {
1005                         folio = fbatch.folios[i];
1006
1007                         if (xa_is_value(folio)) {
1008                                 if (unfalloc)
1009                                         continue;
1010                                 nr_swaps_freed += !shmem_free_swap(mapping,
1011                                                         indices[i], folio);
1012                                 continue;
1013                         }
1014
1015                         if (!unfalloc || !folio_test_uptodate(folio))
1016                                 truncate_inode_folio(mapping, folio);
1017                         folio_unlock(folio);
1018                 }
1019                 folio_batch_remove_exceptionals(&fbatch);
1020                 folio_batch_release(&fbatch);
1021                 cond_resched();
1022         }
1023
1024         /*
1025          * When undoing a failed fallocate, we want none of the partial folio
1026          * zeroing and splitting below, but shall want to truncate the whole
1027          * folio when !uptodate indicates that it was added by this fallocate,
1028          * even when [lstart, lend] covers only a part of the folio.
1029          */
1030         if (unfalloc)
1031                 goto whole_folios;
1032
1033         same_folio = (lstart >> PAGE_SHIFT) == (lend >> PAGE_SHIFT);
1034         folio = shmem_get_partial_folio(inode, lstart >> PAGE_SHIFT);
1035         if (folio) {
1036                 same_folio = lend < folio_pos(folio) + folio_size(folio);
1037                 folio_mark_dirty(folio);
1038                 if (!truncate_inode_partial_folio(folio, lstart, lend)) {
1039                         start = folio->index + folio_nr_pages(folio);
1040                         if (same_folio)
1041                                 end = folio->index;
1042                 }
1043                 folio_unlock(folio);
1044                 folio_put(folio);
1045                 folio = NULL;
1046         }
1047
1048         if (!same_folio)
1049                 folio = shmem_get_partial_folio(inode, lend >> PAGE_SHIFT);
1050         if (folio) {
1051                 folio_mark_dirty(folio);
1052                 if (!truncate_inode_partial_folio(folio, lstart, lend))
1053                         end = folio->index;
1054                 folio_unlock(folio);
1055                 folio_put(folio);
1056         }
1057
1058 whole_folios:
1059
1060         index = start;
1061         while (index < end) {
1062                 cond_resched();
1063
1064                 if (!find_get_entries(mapping, &index, end - 1, &fbatch,
1065                                 indices)) {
1066                         /* If all gone or hole-punch or unfalloc, we're done */
1067                         if (index == start || end != -1)
1068                                 break;
1069                         /* But if truncating, restart to make sure all gone */
1070                         index = start;
1071                         continue;
1072                 }
1073                 for (i = 0; i < folio_batch_count(&fbatch); i++) {
1074                         folio = fbatch.folios[i];
1075
1076                         if (xa_is_value(folio)) {
1077                                 if (unfalloc)
1078                                         continue;
1079                                 if (shmem_free_swap(mapping, indices[i], folio)) {
1080                                         /* Swap was replaced by page: retry */
1081                                         index = indices[i];
1082                                         break;
1083                                 }
1084                                 nr_swaps_freed++;
1085                                 continue;
1086                         }
1087
1088                         folio_lock(folio);
1089
1090                         if (!unfalloc || !folio_test_uptodate(folio)) {
1091                                 if (folio_mapping(folio) != mapping) {
1092                                         /* Page was replaced by swap: retry */
1093                                         folio_unlock(folio);
1094                                         index = indices[i];
1095                                         break;
1096                                 }
1097                                 VM_BUG_ON_FOLIO(folio_test_writeback(folio),
1098                                                 folio);
1099                                 truncate_inode_folio(mapping, folio);
1100                         }
1101                         folio_unlock(folio);
1102                 }
1103                 folio_batch_remove_exceptionals(&fbatch);
1104                 folio_batch_release(&fbatch);
1105         }
1106
1107         shmem_recalc_inode(inode, 0, -nr_swaps_freed);
1108 }
1109
1110 void shmem_truncate_range(struct inode *inode, loff_t lstart, loff_t lend)
1111 {
1112         shmem_undo_range(inode, lstart, lend, false);
1113         inode->i_ctime = inode->i_mtime = current_time(inode);
1114         inode_inc_iversion(inode);
1115 }
1116 EXPORT_SYMBOL_GPL(shmem_truncate_range);
1117
1118 static int shmem_getattr(struct mnt_idmap *idmap,
1119                          const struct path *path, struct kstat *stat,
1120                          u32 request_mask, unsigned int query_flags)
1121 {
1122         struct inode *inode = path->dentry->d_inode;
1123         struct shmem_inode_info *info = SHMEM_I(inode);
1124
1125         if (info->alloced - info->swapped != inode->i_mapping->nrpages)
1126                 shmem_recalc_inode(inode, 0, 0);
1127
1128         if (info->fsflags & FS_APPEND_FL)
1129                 stat->attributes |= STATX_ATTR_APPEND;
1130         if (info->fsflags & FS_IMMUTABLE_FL)
1131                 stat->attributes |= STATX_ATTR_IMMUTABLE;
1132         if (info->fsflags & FS_NODUMP_FL)
1133                 stat->attributes |= STATX_ATTR_NODUMP;
1134         stat->attributes_mask |= (STATX_ATTR_APPEND |
1135                         STATX_ATTR_IMMUTABLE |
1136                         STATX_ATTR_NODUMP);
1137         generic_fillattr(idmap, inode, stat);
1138
1139         if (shmem_is_huge(inode, 0, false, NULL, 0))
1140                 stat->blksize = HPAGE_PMD_SIZE;
1141
1142         if (request_mask & STATX_BTIME) {
1143                 stat->result_mask |= STATX_BTIME;
1144                 stat->btime.tv_sec = info->i_crtime.tv_sec;
1145                 stat->btime.tv_nsec = info->i_crtime.tv_nsec;
1146         }
1147
1148         return 0;
1149 }
1150
1151 static int shmem_setattr(struct mnt_idmap *idmap,
1152                          struct dentry *dentry, struct iattr *attr)
1153 {
1154         struct inode *inode = d_inode(dentry);
1155         struct shmem_inode_info *info = SHMEM_I(inode);
1156         int error;
1157         bool update_mtime = false;
1158         bool update_ctime = true;
1159
1160         error = setattr_prepare(idmap, dentry, attr);
1161         if (error)
1162                 return error;
1163
1164         if ((info->seals & F_SEAL_EXEC) && (attr->ia_valid & ATTR_MODE)) {
1165                 if ((inode->i_mode ^ attr->ia_mode) & 0111) {
1166                         return -EPERM;
1167                 }
1168         }
1169
1170         if (S_ISREG(inode->i_mode) && (attr->ia_valid & ATTR_SIZE)) {
1171                 loff_t oldsize = inode->i_size;
1172                 loff_t newsize = attr->ia_size;
1173
1174                 /* protected by i_rwsem */
1175                 if ((newsize < oldsize && (info->seals & F_SEAL_SHRINK)) ||
1176                     (newsize > oldsize && (info->seals & F_SEAL_GROW)))
1177                         return -EPERM;
1178
1179                 if (newsize != oldsize) {
1180                         error = shmem_reacct_size(SHMEM_I(inode)->flags,
1181                                         oldsize, newsize);
1182                         if (error)
1183                                 return error;
1184                         i_size_write(inode, newsize);
1185                         update_mtime = true;
1186                 } else {
1187                         update_ctime = false;
1188                 }
1189                 if (newsize <= oldsize) {
1190                         loff_t holebegin = round_up(newsize, PAGE_SIZE);
1191                         if (oldsize > holebegin)
1192                                 unmap_mapping_range(inode->i_mapping,
1193                                                         holebegin, 0, 1);
1194                         if (info->alloced)
1195                                 shmem_truncate_range(inode,
1196                                                         newsize, (loff_t)-1);
1197                         /* unmap again to remove racily COWed private pages */
1198                         if (oldsize > holebegin)
1199                                 unmap_mapping_range(inode->i_mapping,
1200                                                         holebegin, 0, 1);
1201                 }
1202         }
1203
1204         if (is_quota_modification(idmap, inode, attr)) {
1205                 error = dquot_initialize(inode);
1206                 if (error)
1207                         return error;
1208         }
1209
1210         /* Transfer quota accounting */
1211         if (i_uid_needs_update(idmap, attr, inode) ||
1212             i_gid_needs_update(idmap, attr, inode)) {
1213                 error = dquot_transfer(idmap, inode, attr);
1214
1215                 if (error)
1216                         return error;
1217         }
1218
1219         setattr_copy(idmap, inode, attr);
1220         if (attr->ia_valid & ATTR_MODE)
1221                 error = posix_acl_chmod(idmap, dentry, inode->i_mode);
1222         if (!error && update_ctime) {
1223                 inode->i_ctime = current_time(inode);
1224                 if (update_mtime)
1225                         inode->i_mtime = inode->i_ctime;
1226                 inode_inc_iversion(inode);
1227         }
1228         return error;
1229 }
1230
1231 static void shmem_evict_inode(struct inode *inode)
1232 {
1233         struct shmem_inode_info *info = SHMEM_I(inode);
1234         struct shmem_sb_info *sbinfo = SHMEM_SB(inode->i_sb);
1235
1236         if (shmem_mapping(inode->i_mapping)) {
1237                 shmem_unacct_size(info->flags, inode->i_size);
1238                 inode->i_size = 0;
1239                 mapping_set_exiting(inode->i_mapping);
1240                 shmem_truncate_range(inode, 0, (loff_t)-1);
1241                 if (!list_empty(&info->shrinklist)) {
1242                         spin_lock(&sbinfo->shrinklist_lock);
1243                         if (!list_empty(&info->shrinklist)) {
1244                                 list_del_init(&info->shrinklist);
1245                                 sbinfo->shrinklist_len--;
1246                         }
1247                         spin_unlock(&sbinfo->shrinklist_lock);
1248                 }
1249                 while (!list_empty(&info->swaplist)) {
1250                         /* Wait while shmem_unuse() is scanning this inode... */
1251                         wait_var_event(&info->stop_eviction,
1252                                        !atomic_read(&info->stop_eviction));
1253                         mutex_lock(&shmem_swaplist_mutex);
1254                         /* ...but beware of the race if we peeked too early */
1255                         if (!atomic_read(&info->stop_eviction))
1256                                 list_del_init(&info->swaplist);
1257                         mutex_unlock(&shmem_swaplist_mutex);
1258                 }
1259         }
1260
1261         simple_xattrs_free(&info->xattrs);
1262         WARN_ON(inode->i_blocks);
1263         shmem_free_inode(inode->i_sb);
1264         clear_inode(inode);
1265 #ifdef CONFIG_TMPFS_QUOTA
1266         dquot_free_inode(inode);
1267         dquot_drop(inode);
1268 #endif
1269 }
1270
1271 static int shmem_find_swap_entries(struct address_space *mapping,
1272                                    pgoff_t start, struct folio_batch *fbatch,
1273                                    pgoff_t *indices, unsigned int type)
1274 {
1275         XA_STATE(xas, &mapping->i_pages, start);
1276         struct folio *folio;
1277         swp_entry_t entry;
1278
1279         rcu_read_lock();
1280         xas_for_each(&xas, folio, ULONG_MAX) {
1281                 if (xas_retry(&xas, folio))
1282                         continue;
1283
1284                 if (!xa_is_value(folio))
1285                         continue;
1286
1287                 entry = radix_to_swp_entry(folio);
1288                 /*
1289                  * swapin error entries can be found in the mapping. But they're
1290                  * deliberately ignored here as we've done everything we can do.
1291                  */
1292                 if (swp_type(entry) != type)
1293                         continue;
1294
1295                 indices[folio_batch_count(fbatch)] = xas.xa_index;
1296                 if (!folio_batch_add(fbatch, folio))
1297                         break;
1298
1299                 if (need_resched()) {
1300                         xas_pause(&xas);
1301                         cond_resched_rcu();
1302                 }
1303         }
1304         rcu_read_unlock();
1305
1306         return xas.xa_index;
1307 }
1308
1309 /*
1310  * Move the swapped pages for an inode to page cache. Returns the count
1311  * of pages swapped in, or the error in case of failure.
1312  */
1313 static int shmem_unuse_swap_entries(struct inode *inode,
1314                 struct folio_batch *fbatch, pgoff_t *indices)
1315 {
1316         int i = 0;
1317         int ret = 0;
1318         int error = 0;
1319         struct address_space *mapping = inode->i_mapping;
1320
1321         for (i = 0; i < folio_batch_count(fbatch); i++) {
1322                 struct folio *folio = fbatch->folios[i];
1323
1324                 if (!xa_is_value(folio))
1325                         continue;
1326                 error = shmem_swapin_folio(inode, indices[i],
1327                                           &folio, SGP_CACHE,
1328                                           mapping_gfp_mask(mapping),
1329                                           NULL, NULL);
1330                 if (error == 0) {
1331                         folio_unlock(folio);
1332                         folio_put(folio);
1333                         ret++;
1334                 }
1335                 if (error == -ENOMEM)
1336                         break;
1337                 error = 0;
1338         }
1339         return error ? error : ret;
1340 }
1341
1342 /*
1343  * If swap found in inode, free it and move page from swapcache to filecache.
1344  */
1345 static int shmem_unuse_inode(struct inode *inode, unsigned int type)
1346 {
1347         struct address_space *mapping = inode->i_mapping;
1348         pgoff_t start = 0;
1349         struct folio_batch fbatch;
1350         pgoff_t indices[PAGEVEC_SIZE];
1351         int ret = 0;
1352
1353         do {
1354                 folio_batch_init(&fbatch);
1355                 shmem_find_swap_entries(mapping, start, &fbatch, indices, type);
1356                 if (folio_batch_count(&fbatch) == 0) {
1357                         ret = 0;
1358                         break;
1359                 }
1360
1361                 ret = shmem_unuse_swap_entries(inode, &fbatch, indices);
1362                 if (ret < 0)
1363                         break;
1364
1365                 start = indices[folio_batch_count(&fbatch) - 1];
1366         } while (true);
1367
1368         return ret;
1369 }
1370
1371 /*
1372  * Read all the shared memory data that resides in the swap
1373  * device 'type' back into memory, so the swap device can be
1374  * unused.
1375  */
1376 int shmem_unuse(unsigned int type)
1377 {
1378         struct shmem_inode_info *info, *next;
1379         int error = 0;
1380
1381         if (list_empty(&shmem_swaplist))
1382                 return 0;
1383
1384         mutex_lock(&shmem_swaplist_mutex);
1385         list_for_each_entry_safe(info, next, &shmem_swaplist, swaplist) {
1386                 if (!info->swapped) {
1387                         list_del_init(&info->swaplist);
1388                         continue;
1389                 }
1390                 /*
1391                  * Drop the swaplist mutex while searching the inode for swap;
1392                  * but before doing so, make sure shmem_evict_inode() will not
1393                  * remove placeholder inode from swaplist, nor let it be freed
1394                  * (igrab() would protect from unlink, but not from unmount).
1395                  */
1396                 atomic_inc(&info->stop_eviction);
1397                 mutex_unlock(&shmem_swaplist_mutex);
1398
1399                 error = shmem_unuse_inode(&info->vfs_inode, type);
1400                 cond_resched();
1401
1402                 mutex_lock(&shmem_swaplist_mutex);
1403                 next = list_next_entry(info, swaplist);
1404                 if (!info->swapped)
1405                         list_del_init(&info->swaplist);
1406                 if (atomic_dec_and_test(&info->stop_eviction))
1407                         wake_up_var(&info->stop_eviction);
1408                 if (error)
1409                         break;
1410         }
1411         mutex_unlock(&shmem_swaplist_mutex);
1412
1413         return error;
1414 }
1415
1416 /*
1417  * Move the page from the page cache to the swap cache.
1418  */
1419 static int shmem_writepage(struct page *page, struct writeback_control *wbc)
1420 {
1421         struct folio *folio = page_folio(page);
1422         struct address_space *mapping = folio->mapping;
1423         struct inode *inode = mapping->host;
1424         struct shmem_inode_info *info = SHMEM_I(inode);
1425         struct shmem_sb_info *sbinfo = SHMEM_SB(inode->i_sb);
1426         swp_entry_t swap;
1427         pgoff_t index;
1428
1429         /*
1430          * Our capabilities prevent regular writeback or sync from ever calling
1431          * shmem_writepage; but a stacking filesystem might use ->writepage of
1432          * its underlying filesystem, in which case tmpfs should write out to
1433          * swap only in response to memory pressure, and not for the writeback
1434          * threads or sync.
1435          */
1436         if (WARN_ON_ONCE(!wbc->for_reclaim))
1437                 goto redirty;
1438
1439         if (WARN_ON_ONCE((info->flags & VM_LOCKED) || sbinfo->noswap))
1440                 goto redirty;
1441
1442         if (!total_swap_pages)
1443                 goto redirty;
1444
1445         /*
1446          * If /sys/kernel/mm/transparent_hugepage/shmem_enabled is "always" or
1447          * "force", drivers/gpu/drm/i915/gem/i915_gem_shmem.c gets huge pages,
1448          * and its shmem_writeback() needs them to be split when swapping.
1449          */
1450         if (folio_test_large(folio)) {
1451                 /* Ensure the subpages are still dirty */
1452                 folio_test_set_dirty(folio);
1453                 if (split_huge_page(page) < 0)
1454                         goto redirty;
1455                 folio = page_folio(page);
1456                 folio_clear_dirty(folio);
1457         }
1458
1459         index = folio->index;
1460
1461         /*
1462          * This is somewhat ridiculous, but without plumbing a SWAP_MAP_FALLOC
1463          * value into swapfile.c, the only way we can correctly account for a
1464          * fallocated folio arriving here is now to initialize it and write it.
1465          *
1466          * That's okay for a folio already fallocated earlier, but if we have
1467          * not yet completed the fallocation, then (a) we want to keep track
1468          * of this folio in case we have to undo it, and (b) it may not be a
1469          * good idea to continue anyway, once we're pushing into swap.  So
1470          * reactivate the folio, and let shmem_fallocate() quit when too many.
1471          */
1472         if (!folio_test_uptodate(folio)) {
1473                 if (inode->i_private) {
1474                         struct shmem_falloc *shmem_falloc;
1475                         spin_lock(&inode->i_lock);
1476                         shmem_falloc = inode->i_private;
1477                         if (shmem_falloc &&
1478                             !shmem_falloc->waitq &&
1479                             index >= shmem_falloc->start &&
1480                             index < shmem_falloc->next)
1481                                 shmem_falloc->nr_unswapped++;
1482                         else
1483                                 shmem_falloc = NULL;
1484                         spin_unlock(&inode->i_lock);
1485                         if (shmem_falloc)
1486                                 goto redirty;
1487                 }
1488                 folio_zero_range(folio, 0, folio_size(folio));
1489                 flush_dcache_folio(folio);
1490                 folio_mark_uptodate(folio);
1491         }
1492
1493         swap = folio_alloc_swap(folio);
1494         if (!swap.val)
1495                 goto redirty;
1496
1497         /*
1498          * Add inode to shmem_unuse()'s list of swapped-out inodes,
1499          * if it's not already there.  Do it now before the folio is
1500          * moved to swap cache, when its pagelock no longer protects
1501          * the inode from eviction.  But don't unlock the mutex until
1502          * we've incremented swapped, because shmem_unuse_inode() will
1503          * prune a !swapped inode from the swaplist under this mutex.
1504          */
1505         mutex_lock(&shmem_swaplist_mutex);
1506         if (list_empty(&info->swaplist))
1507                 list_add(&info->swaplist, &shmem_swaplist);
1508
1509         if (add_to_swap_cache(folio, swap,
1510                         __GFP_HIGH | __GFP_NOMEMALLOC | __GFP_NOWARN,
1511                         NULL) == 0) {
1512                 shmem_recalc_inode(inode, 0, 1);
1513                 swap_shmem_alloc(swap);
1514                 shmem_delete_from_page_cache(folio, swp_to_radix_entry(swap));
1515
1516                 mutex_unlock(&shmem_swaplist_mutex);
1517                 BUG_ON(folio_mapped(folio));
1518                 swap_writepage(&folio->page, wbc);
1519                 return 0;
1520         }
1521
1522         mutex_unlock(&shmem_swaplist_mutex);
1523         put_swap_folio(folio, swap);
1524 redirty:
1525         folio_mark_dirty(folio);
1526         if (wbc->for_reclaim)
1527                 return AOP_WRITEPAGE_ACTIVATE;  /* Return with folio locked */
1528         folio_unlock(folio);
1529         return 0;
1530 }
1531
1532 #if defined(CONFIG_NUMA) && defined(CONFIG_TMPFS)
1533 static void shmem_show_mpol(struct seq_file *seq, struct mempolicy *mpol)
1534 {
1535         char buffer[64];
1536
1537         if (!mpol || mpol->mode == MPOL_DEFAULT)
1538                 return;         /* show nothing */
1539
1540         mpol_to_str(buffer, sizeof(buffer), mpol);
1541
1542         seq_printf(seq, ",mpol=%s", buffer);
1543 }
1544
1545 static struct mempolicy *shmem_get_sbmpol(struct shmem_sb_info *sbinfo)
1546 {
1547         struct mempolicy *mpol = NULL;
1548         if (sbinfo->mpol) {
1549                 raw_spin_lock(&sbinfo->stat_lock);      /* prevent replace/use races */
1550                 mpol = sbinfo->mpol;
1551                 mpol_get(mpol);
1552                 raw_spin_unlock(&sbinfo->stat_lock);
1553         }
1554         return mpol;
1555 }
1556 #else /* !CONFIG_NUMA || !CONFIG_TMPFS */
1557 static inline void shmem_show_mpol(struct seq_file *seq, struct mempolicy *mpol)
1558 {
1559 }
1560 static inline struct mempolicy *shmem_get_sbmpol(struct shmem_sb_info *sbinfo)
1561 {
1562         return NULL;
1563 }
1564 #endif /* CONFIG_NUMA && CONFIG_TMPFS */
1565 #ifndef CONFIG_NUMA
1566 #define vm_policy vm_private_data
1567 #endif
1568
1569 static void shmem_pseudo_vma_init(struct vm_area_struct *vma,
1570                 struct shmem_inode_info *info, pgoff_t index)
1571 {
1572         /* Create a pseudo vma that just contains the policy */
1573         vma_init(vma, NULL);
1574         /* Bias interleave by inode number to distribute better across nodes */
1575         vma->vm_pgoff = index + info->vfs_inode.i_ino;
1576         vma->vm_policy = mpol_shared_policy_lookup(&info->policy, index);
1577 }
1578
1579 static void shmem_pseudo_vma_destroy(struct vm_area_struct *vma)
1580 {
1581         /* Drop reference taken by mpol_shared_policy_lookup() */
1582         mpol_cond_put(vma->vm_policy);
1583 }
1584
1585 static struct folio *shmem_swapin(swp_entry_t swap, gfp_t gfp,
1586                         struct shmem_inode_info *info, pgoff_t index)
1587 {
1588         struct vm_area_struct pvma;
1589         struct page *page;
1590         struct vm_fault vmf = {
1591                 .vma = &pvma,
1592         };
1593
1594         shmem_pseudo_vma_init(&pvma, info, index);
1595         page = swap_cluster_readahead(swap, gfp, &vmf);
1596         shmem_pseudo_vma_destroy(&pvma);
1597
1598         if (!page)
1599                 return NULL;
1600         return page_folio(page);
1601 }
1602
1603 /*
1604  * Make sure huge_gfp is always more limited than limit_gfp.
1605  * Some of the flags set permissions, while others set limitations.
1606  */
1607 static gfp_t limit_gfp_mask(gfp_t huge_gfp, gfp_t limit_gfp)
1608 {
1609         gfp_t allowflags = __GFP_IO | __GFP_FS | __GFP_RECLAIM;
1610         gfp_t denyflags = __GFP_NOWARN | __GFP_NORETRY;
1611         gfp_t zoneflags = limit_gfp & GFP_ZONEMASK;
1612         gfp_t result = huge_gfp & ~(allowflags | GFP_ZONEMASK);
1613
1614         /* Allow allocations only from the originally specified zones. */
1615         result |= zoneflags;
1616
1617         /*
1618          * Minimize the result gfp by taking the union with the deny flags,
1619          * and the intersection of the allow flags.
1620          */
1621         result |= (limit_gfp & denyflags);
1622         result |= (huge_gfp & limit_gfp) & allowflags;
1623
1624         return result;
1625 }
1626
1627 static struct folio *shmem_alloc_hugefolio(gfp_t gfp,
1628                 struct shmem_inode_info *info, pgoff_t index)
1629 {
1630         struct vm_area_struct pvma;
1631         struct address_space *mapping = info->vfs_inode.i_mapping;
1632         pgoff_t hindex;
1633         struct folio *folio;
1634
1635         hindex = round_down(index, HPAGE_PMD_NR);
1636         if (xa_find(&mapping->i_pages, &hindex, hindex + HPAGE_PMD_NR - 1,
1637                                                                 XA_PRESENT))
1638                 return NULL;
1639
1640         shmem_pseudo_vma_init(&pvma, info, hindex);
1641         folio = vma_alloc_folio(gfp, HPAGE_PMD_ORDER, &pvma, 0, true);
1642         shmem_pseudo_vma_destroy(&pvma);
1643         if (!folio)
1644                 count_vm_event(THP_FILE_FALLBACK);
1645         return folio;
1646 }
1647
1648 static struct folio *shmem_alloc_folio(gfp_t gfp,
1649                         struct shmem_inode_info *info, pgoff_t index)
1650 {
1651         struct vm_area_struct pvma;
1652         struct folio *folio;
1653
1654         shmem_pseudo_vma_init(&pvma, info, index);
1655         folio = vma_alloc_folio(gfp, 0, &pvma, 0, false);
1656         shmem_pseudo_vma_destroy(&pvma);
1657
1658         return folio;
1659 }
1660
1661 static struct folio *shmem_alloc_and_acct_folio(gfp_t gfp, struct inode *inode,
1662                 pgoff_t index, bool huge)
1663 {
1664         struct shmem_inode_info *info = SHMEM_I(inode);
1665         struct folio *folio;
1666         int nr;
1667         int err;
1668
1669         if (!IS_ENABLED(CONFIG_TRANSPARENT_HUGEPAGE))
1670                 huge = false;
1671         nr = huge ? HPAGE_PMD_NR : 1;
1672
1673         err = shmem_inode_acct_block(inode, nr);
1674         if (err)
1675                 goto failed;
1676
1677         if (huge)
1678                 folio = shmem_alloc_hugefolio(gfp, info, index);
1679         else
1680                 folio = shmem_alloc_folio(gfp, info, index);
1681         if (folio) {
1682                 __folio_set_locked(folio);
1683                 __folio_set_swapbacked(folio);
1684                 return folio;
1685         }
1686
1687         err = -ENOMEM;
1688         shmem_inode_unacct_blocks(inode, nr);
1689 failed:
1690         return ERR_PTR(err);
1691 }
1692
1693 /*
1694  * When a page is moved from swapcache to shmem filecache (either by the
1695  * usual swapin of shmem_get_folio_gfp(), or by the less common swapoff of
1696  * shmem_unuse_inode()), it may have been read in earlier from swap, in
1697  * ignorance of the mapping it belongs to.  If that mapping has special
1698  * constraints (like the gma500 GEM driver, which requires RAM below 4GB),
1699  * we may need to copy to a suitable page before moving to filecache.
1700  *
1701  * In a future release, this may well be extended to respect cpuset and
1702  * NUMA mempolicy, and applied also to anonymous pages in do_swap_page();
1703  * but for now it is a simple matter of zone.
1704  */
1705 static bool shmem_should_replace_folio(struct folio *folio, gfp_t gfp)
1706 {
1707         return folio_zonenum(folio) > gfp_zone(gfp);
1708 }
1709
1710 static int shmem_replace_folio(struct folio **foliop, gfp_t gfp,
1711                                 struct shmem_inode_info *info, pgoff_t index)
1712 {
1713         struct folio *old, *new;
1714         struct address_space *swap_mapping;
1715         swp_entry_t entry;
1716         pgoff_t swap_index;
1717         int error;
1718
1719         old = *foliop;
1720         entry = folio_swap_entry(old);
1721         swap_index = swp_offset(entry);
1722         swap_mapping = swap_address_space(entry);
1723
1724         /*
1725          * We have arrived here because our zones are constrained, so don't
1726          * limit chance of success by further cpuset and node constraints.
1727          */
1728         gfp &= ~GFP_CONSTRAINT_MASK;
1729         VM_BUG_ON_FOLIO(folio_test_large(old), old);
1730         new = shmem_alloc_folio(gfp, info, index);
1731         if (!new)
1732                 return -ENOMEM;
1733
1734         folio_get(new);
1735         folio_copy(new, old);
1736         flush_dcache_folio(new);
1737
1738         __folio_set_locked(new);
1739         __folio_set_swapbacked(new);
1740         folio_mark_uptodate(new);
1741         folio_set_swap_entry(new, entry);
1742         folio_set_swapcache(new);
1743
1744         /*
1745          * Our caller will very soon move newpage out of swapcache, but it's
1746          * a nice clean interface for us to replace oldpage by newpage there.
1747          */
1748         xa_lock_irq(&swap_mapping->i_pages);
1749         error = shmem_replace_entry(swap_mapping, swap_index, old, new);
1750         if (!error) {
1751                 mem_cgroup_migrate(old, new);
1752                 __lruvec_stat_mod_folio(new, NR_FILE_PAGES, 1);
1753                 __lruvec_stat_mod_folio(new, NR_SHMEM, 1);
1754                 __lruvec_stat_mod_folio(old, NR_FILE_PAGES, -1);
1755                 __lruvec_stat_mod_folio(old, NR_SHMEM, -1);
1756         }
1757         xa_unlock_irq(&swap_mapping->i_pages);
1758
1759         if (unlikely(error)) {
1760                 /*
1761                  * Is this possible?  I think not, now that our callers check
1762                  * both PageSwapCache and page_private after getting page lock;
1763                  * but be defensive.  Reverse old to newpage for clear and free.
1764                  */
1765                 old = new;
1766         } else {
1767                 folio_add_lru(new);
1768                 *foliop = new;
1769         }
1770
1771         folio_clear_swapcache(old);
1772         old->private = NULL;
1773
1774         folio_unlock(old);
1775         folio_put_refs(old, 2);
1776         return error;
1777 }
1778
1779 static void shmem_set_folio_swapin_error(struct inode *inode, pgoff_t index,
1780                                          struct folio *folio, swp_entry_t swap)
1781 {
1782         struct address_space *mapping = inode->i_mapping;
1783         swp_entry_t swapin_error;
1784         void *old;
1785
1786         swapin_error = make_swapin_error_entry();
1787         old = xa_cmpxchg_irq(&mapping->i_pages, index,
1788                              swp_to_radix_entry(swap),
1789                              swp_to_radix_entry(swapin_error), 0);
1790         if (old != swp_to_radix_entry(swap))
1791                 return;
1792
1793         folio_wait_writeback(folio);
1794         delete_from_swap_cache(folio);
1795         /*
1796          * Don't treat swapin error folio as alloced. Otherwise inode->i_blocks
1797          * won't be 0 when inode is released and thus trigger WARN_ON(i_blocks)
1798          * in shmem_evict_inode().
1799          */
1800         shmem_recalc_inode(inode, -1, -1);
1801         swap_free(swap);
1802 }
1803
1804 /*
1805  * Swap in the folio pointed to by *foliop.
1806  * Caller has to make sure that *foliop contains a valid swapped folio.
1807  * Returns 0 and the folio in foliop if success. On failure, returns the
1808  * error code and NULL in *foliop.
1809  */
1810 static int shmem_swapin_folio(struct inode *inode, pgoff_t index,
1811                              struct folio **foliop, enum sgp_type sgp,
1812                              gfp_t gfp, struct vm_area_struct *vma,
1813                              vm_fault_t *fault_type)
1814 {
1815         struct address_space *mapping = inode->i_mapping;
1816         struct shmem_inode_info *info = SHMEM_I(inode);
1817         struct mm_struct *charge_mm = vma ? vma->vm_mm : NULL;
1818         struct swap_info_struct *si;
1819         struct folio *folio = NULL;
1820         swp_entry_t swap;
1821         int error;
1822
1823         VM_BUG_ON(!*foliop || !xa_is_value(*foliop));
1824         swap = radix_to_swp_entry(*foliop);
1825         *foliop = NULL;
1826
1827         if (is_swapin_error_entry(swap))
1828                 return -EIO;
1829
1830         si = get_swap_device(swap);
1831         if (!si) {
1832                 if (!shmem_confirm_swap(mapping, index, swap))
1833                         return -EEXIST;
1834                 else
1835                         return -EINVAL;
1836         }
1837
1838         /* Look it up and read it in.. */
1839         folio = swap_cache_get_folio(swap, NULL, 0);
1840         if (!folio) {
1841                 /* Or update major stats only when swapin succeeds?? */
1842                 if (fault_type) {
1843                         *fault_type |= VM_FAULT_MAJOR;
1844                         count_vm_event(PGMAJFAULT);
1845                         count_memcg_event_mm(charge_mm, PGMAJFAULT);
1846                 }
1847                 /* Here we actually start the io */
1848                 folio = shmem_swapin(swap, gfp, info, index);
1849                 if (!folio) {
1850                         error = -ENOMEM;
1851                         goto failed;
1852                 }
1853         }
1854
1855         /* We have to do this with folio locked to prevent races */
1856         folio_lock(folio);
1857         if (!folio_test_swapcache(folio) ||
1858             folio_swap_entry(folio).val != swap.val ||
1859             !shmem_confirm_swap(mapping, index, swap)) {
1860                 error = -EEXIST;
1861                 goto unlock;
1862         }
1863         if (!folio_test_uptodate(folio)) {
1864                 error = -EIO;
1865                 goto failed;
1866         }
1867         folio_wait_writeback(folio);
1868
1869         /*
1870          * Some architectures may have to restore extra metadata to the
1871          * folio after reading from swap.
1872          */
1873         arch_swap_restore(swap, folio);
1874
1875         if (shmem_should_replace_folio(folio, gfp)) {
1876                 error = shmem_replace_folio(&folio, gfp, info, index);
1877                 if (error)
1878                         goto failed;
1879         }
1880
1881         error = shmem_add_to_page_cache(folio, mapping, index,
1882                                         swp_to_radix_entry(swap), gfp,
1883                                         charge_mm);
1884         if (error)
1885                 goto failed;
1886
1887         shmem_recalc_inode(inode, 0, -1);
1888
1889         if (sgp == SGP_WRITE)
1890                 folio_mark_accessed(folio);
1891
1892         delete_from_swap_cache(folio);
1893         folio_mark_dirty(folio);
1894         swap_free(swap);
1895         put_swap_device(si);
1896
1897         *foliop = folio;
1898         return 0;
1899 failed:
1900         if (!shmem_confirm_swap(mapping, index, swap))
1901                 error = -EEXIST;
1902         if (error == -EIO)
1903                 shmem_set_folio_swapin_error(inode, index, folio, swap);
1904 unlock:
1905         if (folio) {
1906                 folio_unlock(folio);
1907                 folio_put(folio);
1908         }
1909         put_swap_device(si);
1910
1911         return error;
1912 }
1913
1914 /*
1915  * shmem_get_folio_gfp - find page in cache, or get from swap, or allocate
1916  *
1917  * If we allocate a new one we do not mark it dirty. That's up to the
1918  * vm. If we swap it in we mark it dirty since we also free the swap
1919  * entry since a page cannot live in both the swap and page cache.
1920  *
1921  * vma, vmf, and fault_type are only supplied by shmem_fault:
1922  * otherwise they are NULL.
1923  */
1924 static int shmem_get_folio_gfp(struct inode *inode, pgoff_t index,
1925                 struct folio **foliop, enum sgp_type sgp, gfp_t gfp,
1926                 struct vm_area_struct *vma, struct vm_fault *vmf,
1927                 vm_fault_t *fault_type)
1928 {
1929         struct address_space *mapping = inode->i_mapping;
1930         struct shmem_inode_info *info = SHMEM_I(inode);
1931         struct shmem_sb_info *sbinfo;
1932         struct mm_struct *charge_mm;
1933         struct folio *folio;
1934         pgoff_t hindex;
1935         gfp_t huge_gfp;
1936         int error;
1937         int once = 0;
1938         int alloced = 0;
1939
1940         if (index > (MAX_LFS_FILESIZE >> PAGE_SHIFT))
1941                 return -EFBIG;
1942 repeat:
1943         if (sgp <= SGP_CACHE &&
1944             ((loff_t)index << PAGE_SHIFT) >= i_size_read(inode)) {
1945                 return -EINVAL;
1946         }
1947
1948         sbinfo = SHMEM_SB(inode->i_sb);
1949         charge_mm = vma ? vma->vm_mm : NULL;
1950
1951         folio = filemap_get_entry(mapping, index);
1952         if (folio && vma && userfaultfd_minor(vma)) {
1953                 if (!xa_is_value(folio))
1954                         folio_put(folio);
1955                 *fault_type = handle_userfault(vmf, VM_UFFD_MINOR);
1956                 return 0;
1957         }
1958
1959         if (xa_is_value(folio)) {
1960                 error = shmem_swapin_folio(inode, index, &folio,
1961                                           sgp, gfp, vma, fault_type);
1962                 if (error == -EEXIST)
1963                         goto repeat;
1964
1965                 *foliop = folio;
1966                 return error;
1967         }
1968
1969         if (folio) {
1970                 folio_lock(folio);
1971
1972                 /* Has the folio been truncated or swapped out? */
1973                 if (unlikely(folio->mapping != mapping)) {
1974                         folio_unlock(folio);
1975                         folio_put(folio);
1976                         goto repeat;
1977                 }
1978                 if (sgp == SGP_WRITE)
1979                         folio_mark_accessed(folio);
1980                 if (folio_test_uptodate(folio))
1981                         goto out;
1982                 /* fallocated folio */
1983                 if (sgp != SGP_READ)
1984                         goto clear;
1985                 folio_unlock(folio);
1986                 folio_put(folio);
1987         }
1988
1989         /*
1990          * SGP_READ: succeed on hole, with NULL folio, letting caller zero.
1991          * SGP_NOALLOC: fail on hole, with NULL folio, letting caller fail.
1992          */
1993         *foliop = NULL;
1994         if (sgp == SGP_READ)
1995                 return 0;
1996         if (sgp == SGP_NOALLOC)
1997                 return -ENOENT;
1998
1999         /*
2000          * Fast cache lookup and swap lookup did not find it: allocate.
2001          */
2002
2003         if (vma && userfaultfd_missing(vma)) {
2004                 *fault_type = handle_userfault(vmf, VM_UFFD_MISSING);
2005                 return 0;
2006         }
2007
2008         if (!shmem_is_huge(inode, index, false,
2009                            vma ? vma->vm_mm : NULL, vma ? vma->vm_flags : 0))
2010                 goto alloc_nohuge;
2011
2012         huge_gfp = vma_thp_gfp_mask(vma);
2013         huge_gfp = limit_gfp_mask(huge_gfp, gfp);
2014         folio = shmem_alloc_and_acct_folio(huge_gfp, inode, index, true);
2015         if (IS_ERR(folio)) {
2016 alloc_nohuge:
2017                 folio = shmem_alloc_and_acct_folio(gfp, inode, index, false);
2018         }
2019         if (IS_ERR(folio)) {
2020                 int retry = 5;
2021
2022                 error = PTR_ERR(folio);
2023                 folio = NULL;
2024                 if (error != -ENOSPC)
2025                         goto unlock;
2026                 /*
2027                  * Try to reclaim some space by splitting a large folio
2028                  * beyond i_size on the filesystem.
2029                  */
2030                 while (retry--) {
2031                         int ret;
2032
2033                         ret = shmem_unused_huge_shrink(sbinfo, NULL, 1);
2034                         if (ret == SHRINK_STOP)
2035                                 break;
2036                         if (ret)
2037                                 goto alloc_nohuge;
2038                 }
2039                 goto unlock;
2040         }
2041
2042         hindex = round_down(index, folio_nr_pages(folio));
2043
2044         if (sgp == SGP_WRITE)
2045                 __folio_set_referenced(folio);
2046
2047         error = shmem_add_to_page_cache(folio, mapping, hindex,
2048                                         NULL, gfp & GFP_RECLAIM_MASK,
2049                                         charge_mm);
2050         if (error)
2051                 goto unacct;
2052
2053         folio_add_lru(folio);
2054         shmem_recalc_inode(inode, folio_nr_pages(folio), 0);
2055         alloced = true;
2056
2057         if (folio_test_pmd_mappable(folio) &&
2058             DIV_ROUND_UP(i_size_read(inode), PAGE_SIZE) <
2059                                         folio_next_index(folio) - 1) {
2060                 /*
2061                  * Part of the large folio is beyond i_size: subject
2062                  * to shrink under memory pressure.
2063                  */
2064                 spin_lock(&sbinfo->shrinklist_lock);
2065                 /*
2066                  * _careful to defend against unlocked access to
2067                  * ->shrink_list in shmem_unused_huge_shrink()
2068                  */
2069                 if (list_empty_careful(&info->shrinklist)) {
2070                         list_add_tail(&info->shrinklist,
2071                                       &sbinfo->shrinklist);
2072                         sbinfo->shrinklist_len++;
2073                 }
2074                 spin_unlock(&sbinfo->shrinklist_lock);
2075         }
2076
2077         /*
2078          * Let SGP_FALLOC use the SGP_WRITE optimization on a new folio.
2079          */
2080         if (sgp == SGP_FALLOC)
2081                 sgp = SGP_WRITE;
2082 clear:
2083         /*
2084          * Let SGP_WRITE caller clear ends if write does not fill folio;
2085          * but SGP_FALLOC on a folio fallocated earlier must initialize
2086          * it now, lest undo on failure cancel our earlier guarantee.
2087          */
2088         if (sgp != SGP_WRITE && !folio_test_uptodate(folio)) {
2089                 long i, n = folio_nr_pages(folio);
2090
2091                 for (i = 0; i < n; i++)
2092                         clear_highpage(folio_page(folio, i));
2093                 flush_dcache_folio(folio);
2094                 folio_mark_uptodate(folio);
2095         }
2096
2097         /* Perhaps the file has been truncated since we checked */
2098         if (sgp <= SGP_CACHE &&
2099             ((loff_t)index << PAGE_SHIFT) >= i_size_read(inode)) {
2100                 if (alloced) {
2101                         folio_clear_dirty(folio);
2102                         filemap_remove_folio(folio);
2103                         shmem_recalc_inode(inode, 0, 0);
2104                 }
2105                 error = -EINVAL;
2106                 goto unlock;
2107         }
2108 out:
2109         *foliop = folio;
2110         return 0;
2111
2112         /*
2113          * Error recovery.
2114          */
2115 unacct:
2116         shmem_inode_unacct_blocks(inode, folio_nr_pages(folio));
2117
2118         if (folio_test_large(folio)) {
2119                 folio_unlock(folio);
2120                 folio_put(folio);
2121                 goto alloc_nohuge;
2122         }
2123 unlock:
2124         if (folio) {
2125                 folio_unlock(folio);
2126                 folio_put(folio);
2127         }
2128         if (error == -ENOSPC && !once++) {
2129                 shmem_recalc_inode(inode, 0, 0);
2130                 goto repeat;
2131         }
2132         if (error == -EEXIST)
2133                 goto repeat;
2134         return error;
2135 }
2136
2137 int shmem_get_folio(struct inode *inode, pgoff_t index, struct folio **foliop,
2138                 enum sgp_type sgp)
2139 {
2140         return shmem_get_folio_gfp(inode, index, foliop, sgp,
2141                         mapping_gfp_mask(inode->i_mapping), NULL, NULL, NULL);
2142 }
2143
2144 /*
2145  * This is like autoremove_wake_function, but it removes the wait queue
2146  * entry unconditionally - even if something else had already woken the
2147  * target.
2148  */
2149 static int synchronous_wake_function(wait_queue_entry_t *wait, unsigned mode, int sync, void *key)
2150 {
2151         int ret = default_wake_function(wait, mode, sync, key);
2152         list_del_init(&wait->entry);
2153         return ret;
2154 }
2155
2156 static vm_fault_t shmem_fault(struct vm_fault *vmf)
2157 {
2158         struct vm_area_struct *vma = vmf->vma;
2159         struct inode *inode = file_inode(vma->vm_file);
2160         gfp_t gfp = mapping_gfp_mask(inode->i_mapping);
2161         struct folio *folio = NULL;
2162         int err;
2163         vm_fault_t ret = VM_FAULT_LOCKED;
2164
2165         /*
2166          * Trinity finds that probing a hole which tmpfs is punching can
2167          * prevent the hole-punch from ever completing: which in turn
2168          * locks writers out with its hold on i_rwsem.  So refrain from
2169          * faulting pages into the hole while it's being punched.  Although
2170          * shmem_undo_range() does remove the additions, it may be unable to
2171          * keep up, as each new page needs its own unmap_mapping_range() call,
2172          * and the i_mmap tree grows ever slower to scan if new vmas are added.
2173          *
2174          * It does not matter if we sometimes reach this check just before the
2175          * hole-punch begins, so that one fault then races with the punch:
2176          * we just need to make racing faults a rare case.
2177          *
2178          * The implementation below would be much simpler if we just used a
2179          * standard mutex or completion: but we cannot take i_rwsem in fault,
2180          * and bloating every shmem inode for this unlikely case would be sad.
2181          */
2182         if (unlikely(inode->i_private)) {
2183                 struct shmem_falloc *shmem_falloc;
2184
2185                 spin_lock(&inode->i_lock);
2186                 shmem_falloc = inode->i_private;
2187                 if (shmem_falloc &&
2188                     shmem_falloc->waitq &&
2189                     vmf->pgoff >= shmem_falloc->start &&
2190                     vmf->pgoff < shmem_falloc->next) {
2191                         struct file *fpin;
2192                         wait_queue_head_t *shmem_falloc_waitq;
2193                         DEFINE_WAIT_FUNC(shmem_fault_wait, synchronous_wake_function);
2194
2195                         ret = VM_FAULT_NOPAGE;
2196                         fpin = maybe_unlock_mmap_for_io(vmf, NULL);
2197                         if (fpin)
2198                                 ret = VM_FAULT_RETRY;
2199
2200                         shmem_falloc_waitq = shmem_falloc->waitq;
2201                         prepare_to_wait(shmem_falloc_waitq, &shmem_fault_wait,
2202                                         TASK_UNINTERRUPTIBLE);
2203                         spin_unlock(&inode->i_lock);
2204                         schedule();
2205
2206                         /*
2207                          * shmem_falloc_waitq points into the shmem_fallocate()
2208                          * stack of the hole-punching task: shmem_falloc_waitq
2209                          * is usually invalid by the time we reach here, but
2210                          * finish_wait() does not dereference it in that case;
2211                          * though i_lock needed lest racing with wake_up_all().
2212                          */
2213                         spin_lock(&inode->i_lock);
2214                         finish_wait(shmem_falloc_waitq, &shmem_fault_wait);
2215                         spin_unlock(&inode->i_lock);
2216
2217                         if (fpin)
2218                                 fput(fpin);
2219                         return ret;
2220                 }
2221                 spin_unlock(&inode->i_lock);
2222         }
2223
2224         err = shmem_get_folio_gfp(inode, vmf->pgoff, &folio, SGP_CACHE,
2225                                   gfp, vma, vmf, &ret);
2226         if (err)
2227                 return vmf_error(err);
2228         if (folio)
2229                 vmf->page = folio_file_page(folio, vmf->pgoff);
2230         return ret;
2231 }
2232
2233 unsigned long shmem_get_unmapped_area(struct file *file,
2234                                       unsigned long uaddr, unsigned long len,
2235                                       unsigned long pgoff, unsigned long flags)
2236 {
2237         unsigned long (*get_area)(struct file *,
2238                 unsigned long, unsigned long, unsigned long, unsigned long);
2239         unsigned long addr;
2240         unsigned long offset;
2241         unsigned long inflated_len;
2242         unsigned long inflated_addr;
2243         unsigned long inflated_offset;
2244
2245         if (len > TASK_SIZE)
2246                 return -ENOMEM;
2247
2248         get_area = current->mm->get_unmapped_area;
2249         addr = get_area(file, uaddr, len, pgoff, flags);
2250
2251         if (!IS_ENABLED(CONFIG_TRANSPARENT_HUGEPAGE))
2252                 return addr;
2253         if (IS_ERR_VALUE(addr))
2254                 return addr;
2255         if (addr & ~PAGE_MASK)
2256                 return addr;
2257         if (addr > TASK_SIZE - len)
2258                 return addr;
2259
2260         if (shmem_huge == SHMEM_HUGE_DENY)
2261                 return addr;
2262         if (len < HPAGE_PMD_SIZE)
2263                 return addr;
2264         if (flags & MAP_FIXED)
2265                 return addr;
2266         /*
2267          * Our priority is to support MAP_SHARED mapped hugely;
2268          * and support MAP_PRIVATE mapped hugely too, until it is COWed.
2269          * But if caller specified an address hint and we allocated area there
2270          * successfully, respect that as before.
2271          */
2272         if (uaddr == addr)
2273                 return addr;
2274
2275         if (shmem_huge != SHMEM_HUGE_FORCE) {
2276                 struct super_block *sb;
2277
2278                 if (file) {
2279                         VM_BUG_ON(file->f_op != &shmem_file_operations);
2280                         sb = file_inode(file)->i_sb;
2281                 } else {
2282                         /*
2283                          * Called directly from mm/mmap.c, or drivers/char/mem.c
2284                          * for "/dev/zero", to create a shared anonymous object.
2285                          */
2286                         if (IS_ERR(shm_mnt))
2287                                 return addr;
2288                         sb = shm_mnt->mnt_sb;
2289                 }
2290                 if (SHMEM_SB(sb)->huge == SHMEM_HUGE_NEVER)
2291                         return addr;
2292         }
2293
2294         offset = (pgoff << PAGE_SHIFT) & (HPAGE_PMD_SIZE-1);
2295         if (offset && offset + len < 2 * HPAGE_PMD_SIZE)
2296                 return addr;
2297         if ((addr & (HPAGE_PMD_SIZE-1)) == offset)
2298                 return addr;
2299
2300         inflated_len = len + HPAGE_PMD_SIZE - PAGE_SIZE;
2301         if (inflated_len > TASK_SIZE)
2302                 return addr;
2303         if (inflated_len < len)
2304                 return addr;
2305
2306         inflated_addr = get_area(NULL, uaddr, inflated_len, 0, flags);
2307         if (IS_ERR_VALUE(inflated_addr))
2308                 return addr;
2309         if (inflated_addr & ~PAGE_MASK)
2310                 return addr;
2311
2312         inflated_offset = inflated_addr & (HPAGE_PMD_SIZE-1);
2313         inflated_addr += offset - inflated_offset;
2314         if (inflated_offset > offset)
2315                 inflated_addr += HPAGE_PMD_SIZE;
2316
2317         if (inflated_addr > TASK_SIZE - len)
2318                 return addr;
2319         return inflated_addr;
2320 }
2321
2322 #ifdef CONFIG_NUMA
2323 static int shmem_set_policy(struct vm_area_struct *vma, struct mempolicy *mpol)
2324 {
2325         struct inode *inode = file_inode(vma->vm_file);
2326         return mpol_set_shared_policy(&SHMEM_I(inode)->policy, vma, mpol);
2327 }
2328
2329 static struct mempolicy *shmem_get_policy(struct vm_area_struct *vma,
2330                                           unsigned long addr)
2331 {
2332         struct inode *inode = file_inode(vma->vm_file);
2333         pgoff_t index;
2334
2335         index = ((addr - vma->vm_start) >> PAGE_SHIFT) + vma->vm_pgoff;
2336         return mpol_shared_policy_lookup(&SHMEM_I(inode)->policy, index);
2337 }
2338 #endif
2339
2340 int shmem_lock(struct file *file, int lock, struct ucounts *ucounts)
2341 {
2342         struct inode *inode = file_inode(file);
2343         struct shmem_inode_info *info = SHMEM_I(inode);
2344         int retval = -ENOMEM;
2345
2346         /*
2347          * What serializes the accesses to info->flags?
2348          * ipc_lock_object() when called from shmctl_do_lock(),
2349          * no serialization needed when called from shm_destroy().
2350          */
2351         if (lock && !(info->flags & VM_LOCKED)) {
2352                 if (!user_shm_lock(inode->i_size, ucounts))
2353                         goto out_nomem;
2354                 info->flags |= VM_LOCKED;
2355                 mapping_set_unevictable(file->f_mapping);
2356         }
2357         if (!lock && (info->flags & VM_LOCKED) && ucounts) {
2358                 user_shm_unlock(inode->i_size, ucounts);
2359                 info->flags &= ~VM_LOCKED;
2360                 mapping_clear_unevictable(file->f_mapping);
2361         }
2362         retval = 0;
2363
2364 out_nomem:
2365         return retval;
2366 }
2367
2368 static int shmem_mmap(struct file *file, struct vm_area_struct *vma)
2369 {
2370         struct inode *inode = file_inode(file);
2371         struct shmem_inode_info *info = SHMEM_I(inode);
2372         int ret;
2373
2374         ret = seal_check_future_write(info->seals, vma);
2375         if (ret)
2376                 return ret;
2377
2378         /* arm64 - allow memory tagging on RAM-based files */
2379         vm_flags_set(vma, VM_MTE_ALLOWED);
2380
2381         file_accessed(file);
2382         /* This is anonymous shared memory if it is unlinked at the time of mmap */
2383         if (inode->i_nlink)
2384                 vma->vm_ops = &shmem_vm_ops;
2385         else
2386                 vma->vm_ops = &shmem_anon_vm_ops;
2387         return 0;
2388 }
2389
2390 #ifdef CONFIG_TMPFS_XATTR
2391 static int shmem_initxattrs(struct inode *, const struct xattr *, void *);
2392
2393 /*
2394  * chattr's fsflags are unrelated to extended attributes,
2395  * but tmpfs has chosen to enable them under the same config option.
2396  */
2397 static void shmem_set_inode_flags(struct inode *inode, unsigned int fsflags)
2398 {
2399         unsigned int i_flags = 0;
2400
2401         if (fsflags & FS_NOATIME_FL)
2402                 i_flags |= S_NOATIME;
2403         if (fsflags & FS_APPEND_FL)
2404                 i_flags |= S_APPEND;
2405         if (fsflags & FS_IMMUTABLE_FL)
2406                 i_flags |= S_IMMUTABLE;
2407         /*
2408          * But FS_NODUMP_FL does not require any action in i_flags.
2409          */
2410         inode_set_flags(inode, i_flags, S_NOATIME | S_APPEND | S_IMMUTABLE);
2411 }
2412 #else
2413 static void shmem_set_inode_flags(struct inode *inode, unsigned int fsflags)
2414 {
2415 }
2416 #define shmem_initxattrs NULL
2417 #endif
2418
2419 static struct offset_ctx *shmem_get_offset_ctx(struct inode *inode)
2420 {
2421         return &SHMEM_I(inode)->dir_offsets;
2422 }
2423
2424 static struct inode *__shmem_get_inode(struct mnt_idmap *idmap,
2425                                              struct super_block *sb,
2426                                              struct inode *dir, umode_t mode,
2427                                              dev_t dev, unsigned long flags)
2428 {
2429         struct inode *inode;
2430         struct shmem_inode_info *info;
2431         struct shmem_sb_info *sbinfo = SHMEM_SB(sb);
2432         ino_t ino;
2433         int err;
2434
2435         err = shmem_reserve_inode(sb, &ino);
2436         if (err)
2437                 return ERR_PTR(err);
2438
2439
2440         inode = new_inode(sb);
2441
2442         if (!inode) {
2443                 shmem_free_inode(sb);
2444                 return ERR_PTR(-ENOSPC);
2445         }
2446
2447         inode->i_ino = ino;
2448         inode_init_owner(idmap, inode, dir, mode);
2449         inode->i_blocks = 0;
2450         inode->i_atime = inode->i_mtime = inode->i_ctime = current_time(inode);
2451         inode->i_generation = get_random_u32();
2452         info = SHMEM_I(inode);
2453         memset(info, 0, (char *)inode - (char *)info);
2454         spin_lock_init(&info->lock);
2455         atomic_set(&info->stop_eviction, 0);
2456         info->seals = F_SEAL_SEAL;
2457         info->flags = flags & VM_NORESERVE;
2458         info->i_crtime = inode->i_mtime;
2459         info->fsflags = (dir == NULL) ? 0 :
2460                 SHMEM_I(dir)->fsflags & SHMEM_FL_INHERITED;
2461         if (info->fsflags)
2462                 shmem_set_inode_flags(inode, info->fsflags);
2463         INIT_LIST_HEAD(&info->shrinklist);
2464         INIT_LIST_HEAD(&info->swaplist);
2465         INIT_LIST_HEAD(&info->swaplist);
2466         if (sbinfo->noswap)
2467                 mapping_set_unevictable(inode->i_mapping);
2468         simple_xattrs_init(&info->xattrs);
2469         cache_no_acl(inode);
2470         mapping_set_large_folios(inode->i_mapping);
2471
2472         switch (mode & S_IFMT) {
2473         default:
2474                 inode->i_op = &shmem_special_inode_operations;
2475                 init_special_inode(inode, mode, dev);
2476                 break;
2477         case S_IFREG:
2478                 inode->i_mapping->a_ops = &shmem_aops;
2479                 inode->i_op = &shmem_inode_operations;
2480                 inode->i_fop = &shmem_file_operations;
2481                 mpol_shared_policy_init(&info->policy,
2482                                          shmem_get_sbmpol(sbinfo));
2483                 break;
2484         case S_IFDIR:
2485                 inc_nlink(inode);
2486                 /* Some things misbehave if size == 0 on a directory */
2487                 inode->i_size = 2 * BOGO_DIRENT_SIZE;
2488                 inode->i_op = &shmem_dir_inode_operations;
2489                 inode->i_fop = &simple_offset_dir_operations;
2490                 simple_offset_init(shmem_get_offset_ctx(inode));
2491                 break;
2492         case S_IFLNK:
2493                 /*
2494                  * Must not load anything in the rbtree,
2495                  * mpol_free_shared_policy will not be called.
2496                  */
2497                 mpol_shared_policy_init(&info->policy, NULL);
2498                 break;
2499         }
2500
2501         lockdep_annotate_inode_mutex_key(inode);
2502         return inode;
2503 }
2504
2505 #ifdef CONFIG_TMPFS_QUOTA
2506 static struct inode *shmem_get_inode(struct mnt_idmap *idmap,
2507                                      struct super_block *sb, struct inode *dir,
2508                                      umode_t mode, dev_t dev, unsigned long flags)
2509 {
2510         int err;
2511         struct inode *inode;
2512
2513         inode = __shmem_get_inode(idmap, sb, dir, mode, dev, flags);
2514         if (IS_ERR(inode))
2515                 return inode;
2516
2517         err = dquot_initialize(inode);
2518         if (err)
2519                 goto errout;
2520
2521         err = dquot_alloc_inode(inode);
2522         if (err) {
2523                 dquot_drop(inode);
2524                 goto errout;
2525         }
2526         return inode;
2527
2528 errout:
2529         inode->i_flags |= S_NOQUOTA;
2530         iput(inode);
2531         return ERR_PTR(err);
2532 }
2533 #else
2534 static inline struct inode *shmem_get_inode(struct mnt_idmap *idmap,
2535                                      struct super_block *sb, struct inode *dir,
2536                                      umode_t mode, dev_t dev, unsigned long flags)
2537 {
2538         return __shmem_get_inode(idmap, sb, dir, mode, dev, flags);
2539 }
2540 #endif /* CONFIG_TMPFS_QUOTA */
2541
2542 #ifdef CONFIG_USERFAULTFD
2543 int shmem_mfill_atomic_pte(pmd_t *dst_pmd,
2544                            struct vm_area_struct *dst_vma,
2545                            unsigned long dst_addr,
2546                            unsigned long src_addr,
2547                            uffd_flags_t flags,
2548                            struct folio **foliop)
2549 {
2550         struct inode *inode = file_inode(dst_vma->vm_file);
2551         struct shmem_inode_info *info = SHMEM_I(inode);
2552         struct address_space *mapping = inode->i_mapping;
2553         gfp_t gfp = mapping_gfp_mask(mapping);
2554         pgoff_t pgoff = linear_page_index(dst_vma, dst_addr);
2555         void *page_kaddr;
2556         struct folio *folio;
2557         int ret;
2558         pgoff_t max_off;
2559
2560         if (shmem_inode_acct_block(inode, 1)) {
2561                 /*
2562                  * We may have got a page, returned -ENOENT triggering a retry,
2563                  * and now we find ourselves with -ENOMEM. Release the page, to
2564                  * avoid a BUG_ON in our caller.
2565                  */
2566                 if (unlikely(*foliop)) {
2567                         folio_put(*foliop);
2568                         *foliop = NULL;
2569                 }
2570                 return -ENOMEM;
2571         }
2572
2573         if (!*foliop) {
2574                 ret = -ENOMEM;
2575                 folio = shmem_alloc_folio(gfp, info, pgoff);
2576                 if (!folio)
2577                         goto out_unacct_blocks;
2578
2579                 if (uffd_flags_mode_is(flags, MFILL_ATOMIC_COPY)) {
2580                         page_kaddr = kmap_local_folio(folio, 0);
2581                         /*
2582                          * The read mmap_lock is held here.  Despite the
2583                          * mmap_lock being read recursive a deadlock is still
2584                          * possible if a writer has taken a lock.  For example:
2585                          *
2586                          * process A thread 1 takes read lock on own mmap_lock
2587                          * process A thread 2 calls mmap, blocks taking write lock
2588                          * process B thread 1 takes page fault, read lock on own mmap lock
2589                          * process B thread 2 calls mmap, blocks taking write lock
2590                          * process A thread 1 blocks taking read lock on process B
2591                          * process B thread 1 blocks taking read lock on process A
2592                          *
2593                          * Disable page faults to prevent potential deadlock
2594                          * and retry the copy outside the mmap_lock.
2595                          */
2596                         pagefault_disable();
2597                         ret = copy_from_user(page_kaddr,
2598                                              (const void __user *)src_addr,
2599                                              PAGE_SIZE);
2600                         pagefault_enable();
2601                         kunmap_local(page_kaddr);
2602
2603                         /* fallback to copy_from_user outside mmap_lock */
2604                         if (unlikely(ret)) {
2605                                 *foliop = folio;
2606                                 ret = -ENOENT;
2607                                 /* don't free the page */
2608                                 goto out_unacct_blocks;
2609                         }
2610
2611                         flush_dcache_folio(folio);
2612                 } else {                /* ZEROPAGE */
2613                         clear_user_highpage(&folio->page, dst_addr);
2614                 }
2615         } else {
2616                 folio = *foliop;
2617                 VM_BUG_ON_FOLIO(folio_test_large(folio), folio);
2618                 *foliop = NULL;
2619         }
2620
2621         VM_BUG_ON(folio_test_locked(folio));
2622         VM_BUG_ON(folio_test_swapbacked(folio));
2623         __folio_set_locked(folio);
2624         __folio_set_swapbacked(folio);
2625         __folio_mark_uptodate(folio);
2626
2627         ret = -EFAULT;
2628         max_off = DIV_ROUND_UP(i_size_read(inode), PAGE_SIZE);
2629         if (unlikely(pgoff >= max_off))
2630                 goto out_release;
2631
2632         ret = shmem_add_to_page_cache(folio, mapping, pgoff, NULL,
2633                                       gfp & GFP_RECLAIM_MASK, dst_vma->vm_mm);
2634         if (ret)
2635                 goto out_release;
2636
2637         ret = mfill_atomic_install_pte(dst_pmd, dst_vma, dst_addr,
2638                                        &folio->page, true, flags);
2639         if (ret)
2640                 goto out_delete_from_cache;
2641
2642         shmem_recalc_inode(inode, 1, 0);
2643         folio_unlock(folio);
2644         return 0;
2645 out_delete_from_cache:
2646         filemap_remove_folio(folio);
2647 out_release:
2648         folio_unlock(folio);
2649         folio_put(folio);
2650 out_unacct_blocks:
2651         shmem_inode_unacct_blocks(inode, 1);
2652         return ret;
2653 }
2654 #endif /* CONFIG_USERFAULTFD */
2655
2656 #ifdef CONFIG_TMPFS
2657 static const struct inode_operations shmem_symlink_inode_operations;
2658 static const struct inode_operations shmem_short_symlink_operations;
2659
2660 static int
2661 shmem_write_begin(struct file *file, struct address_space *mapping,
2662                         loff_t pos, unsigned len,
2663                         struct page **pagep, void **fsdata)
2664 {
2665         struct inode *inode = mapping->host;
2666         struct shmem_inode_info *info = SHMEM_I(inode);
2667         pgoff_t index = pos >> PAGE_SHIFT;
2668         struct folio *folio;
2669         int ret = 0;
2670
2671         /* i_rwsem is held by caller */
2672         if (unlikely(info->seals & (F_SEAL_GROW |
2673                                    F_SEAL_WRITE | F_SEAL_FUTURE_WRITE))) {
2674                 if (info->seals & (F_SEAL_WRITE | F_SEAL_FUTURE_WRITE))
2675                         return -EPERM;
2676                 if ((info->seals & F_SEAL_GROW) && pos + len > inode->i_size)
2677                         return -EPERM;
2678         }
2679
2680         ret = shmem_get_folio(inode, index, &folio, SGP_WRITE);
2681
2682         if (ret)
2683                 return ret;
2684
2685         *pagep = folio_file_page(folio, index);
2686         if (PageHWPoison(*pagep)) {
2687                 folio_unlock(folio);
2688                 folio_put(folio);
2689                 *pagep = NULL;
2690                 return -EIO;
2691         }
2692
2693         return 0;
2694 }
2695
2696 static int
2697 shmem_write_end(struct file *file, struct address_space *mapping,
2698                         loff_t pos, unsigned len, unsigned copied,
2699                         struct page *page, void *fsdata)
2700 {
2701         struct folio *folio = page_folio(page);
2702         struct inode *inode = mapping->host;
2703
2704         if (pos + copied > inode->i_size)
2705                 i_size_write(inode, pos + copied);
2706
2707         if (!folio_test_uptodate(folio)) {
2708                 if (copied < folio_size(folio)) {
2709                         size_t from = offset_in_folio(folio, pos);
2710                         folio_zero_segments(folio, 0, from,
2711                                         from + copied, folio_size(folio));
2712                 }
2713                 folio_mark_uptodate(folio);
2714         }
2715         folio_mark_dirty(folio);
2716         folio_unlock(folio);
2717         folio_put(folio);
2718
2719         return copied;
2720 }
2721
2722 static ssize_t shmem_file_read_iter(struct kiocb *iocb, struct iov_iter *to)
2723 {
2724         struct file *file = iocb->ki_filp;
2725         struct inode *inode = file_inode(file);
2726         struct address_space *mapping = inode->i_mapping;
2727         pgoff_t index;
2728         unsigned long offset;
2729         int error = 0;
2730         ssize_t retval = 0;
2731         loff_t *ppos = &iocb->ki_pos;
2732
2733         index = *ppos >> PAGE_SHIFT;
2734         offset = *ppos & ~PAGE_MASK;
2735
2736         for (;;) {
2737                 struct folio *folio = NULL;
2738                 struct page *page = NULL;
2739                 pgoff_t end_index;
2740                 unsigned long nr, ret;
2741                 loff_t i_size = i_size_read(inode);
2742
2743                 end_index = i_size >> PAGE_SHIFT;
2744                 if (index > end_index)
2745                         break;
2746                 if (index == end_index) {
2747                         nr = i_size & ~PAGE_MASK;
2748                         if (nr <= offset)
2749                                 break;
2750                 }
2751
2752                 error = shmem_get_folio(inode, index, &folio, SGP_READ);
2753                 if (error) {
2754                         if (error == -EINVAL)
2755                                 error = 0;
2756                         break;
2757                 }
2758                 if (folio) {
2759                         folio_unlock(folio);
2760
2761                         page = folio_file_page(folio, index);
2762                         if (PageHWPoison(page)) {
2763                                 folio_put(folio);
2764                                 error = -EIO;
2765                                 break;
2766                         }
2767                 }
2768
2769                 /*
2770                  * We must evaluate after, since reads (unlike writes)
2771                  * are called without i_rwsem protection against truncate
2772                  */
2773                 nr = PAGE_SIZE;
2774                 i_size = i_size_read(inode);
2775                 end_index = i_size >> PAGE_SHIFT;
2776                 if (index == end_index) {
2777                         nr = i_size & ~PAGE_MASK;
2778                         if (nr <= offset) {
2779                                 if (folio)
2780                                         folio_put(folio);
2781                                 break;
2782                         }
2783                 }
2784                 nr -= offset;
2785
2786                 if (folio) {
2787                         /*
2788                          * If users can be writing to this page using arbitrary
2789                          * virtual addresses, take care about potential aliasing
2790                          * before reading the page on the kernel side.
2791                          */
2792                         if (mapping_writably_mapped(mapping))
2793                                 flush_dcache_page(page);
2794                         /*
2795                          * Mark the page accessed if we read the beginning.
2796                          */
2797                         if (!offset)
2798                                 folio_mark_accessed(folio);
2799                         /*
2800                          * Ok, we have the page, and it's up-to-date, so
2801                          * now we can copy it to user space...
2802                          */
2803                         ret = copy_page_to_iter(page, offset, nr, to);
2804                         folio_put(folio);
2805
2806                 } else if (user_backed_iter(to)) {
2807                         /*
2808                          * Copy to user tends to be so well optimized, but
2809                          * clear_user() not so much, that it is noticeably
2810                          * faster to copy the zero page instead of clearing.
2811                          */
2812                         ret = copy_page_to_iter(ZERO_PAGE(0), offset, nr, to);
2813                 } else {
2814                         /*
2815                          * But submitting the same page twice in a row to
2816                          * splice() - or others? - can result in confusion:
2817                          * so don't attempt that optimization on pipes etc.
2818                          */
2819                         ret = iov_iter_zero(nr, to);
2820                 }
2821
2822                 retval += ret;
2823                 offset += ret;
2824                 index += offset >> PAGE_SHIFT;
2825                 offset &= ~PAGE_MASK;
2826
2827                 if (!iov_iter_count(to))
2828                         break;
2829                 if (ret < nr) {
2830                         error = -EFAULT;
2831                         break;
2832                 }
2833                 cond_resched();
2834         }
2835
2836         *ppos = ((loff_t) index << PAGE_SHIFT) + offset;
2837         file_accessed(file);
2838         return retval ? retval : error;
2839 }
2840
2841 static bool zero_pipe_buf_get(struct pipe_inode_info *pipe,
2842                               struct pipe_buffer *buf)
2843 {
2844         return true;
2845 }
2846
2847 static void zero_pipe_buf_release(struct pipe_inode_info *pipe,
2848                                   struct pipe_buffer *buf)
2849 {
2850 }
2851
2852 static bool zero_pipe_buf_try_steal(struct pipe_inode_info *pipe,
2853                                     struct pipe_buffer *buf)
2854 {
2855         return false;
2856 }
2857
2858 static const struct pipe_buf_operations zero_pipe_buf_ops = {
2859         .release        = zero_pipe_buf_release,
2860         .try_steal      = zero_pipe_buf_try_steal,
2861         .get            = zero_pipe_buf_get,
2862 };
2863
2864 static size_t splice_zeropage_into_pipe(struct pipe_inode_info *pipe,
2865                                         loff_t fpos, size_t size)
2866 {
2867         size_t offset = fpos & ~PAGE_MASK;
2868
2869         size = min_t(size_t, size, PAGE_SIZE - offset);
2870
2871         if (!pipe_full(pipe->head, pipe->tail, pipe->max_usage)) {
2872                 struct pipe_buffer *buf = pipe_head_buf(pipe);
2873
2874                 *buf = (struct pipe_buffer) {
2875                         .ops    = &zero_pipe_buf_ops,
2876                         .page   = ZERO_PAGE(0),
2877                         .offset = offset,
2878                         .len    = size,
2879                 };
2880                 pipe->head++;
2881         }
2882
2883         return size;
2884 }
2885
2886 static ssize_t shmem_file_splice_read(struct file *in, loff_t *ppos,
2887                                       struct pipe_inode_info *pipe,
2888                                       size_t len, unsigned int flags)
2889 {
2890         struct inode *inode = file_inode(in);
2891         struct address_space *mapping = inode->i_mapping;
2892         struct folio *folio = NULL;
2893         size_t total_spliced = 0, used, npages, n, part;
2894         loff_t isize;
2895         int error = 0;
2896
2897         /* Work out how much data we can actually add into the pipe */
2898         used = pipe_occupancy(pipe->head, pipe->tail);
2899         npages = max_t(ssize_t, pipe->max_usage - used, 0);
2900         len = min_t(size_t, len, npages * PAGE_SIZE);
2901
2902         do {
2903                 if (*ppos >= i_size_read(inode))
2904                         break;
2905
2906                 error = shmem_get_folio(inode, *ppos / PAGE_SIZE, &folio,
2907                                         SGP_READ);
2908                 if (error) {
2909                         if (error == -EINVAL)
2910                                 error = 0;
2911                         break;
2912                 }
2913                 if (folio) {
2914                         folio_unlock(folio);
2915
2916                         if (folio_test_hwpoison(folio) ||
2917                             (folio_test_large(folio) &&
2918                              folio_test_has_hwpoisoned(folio))) {
2919                                 error = -EIO;
2920                                 break;
2921                         }
2922                 }
2923
2924                 /*
2925                  * i_size must be checked after we know the pages are Uptodate.
2926                  *
2927                  * Checking i_size after the check allows us to calculate
2928                  * the correct value for "nr", which means the zero-filled
2929                  * part of the page is not copied back to userspace (unless
2930                  * another truncate extends the file - this is desired though).
2931                  */
2932                 isize = i_size_read(inode);
2933                 if (unlikely(*ppos >= isize))
2934                         break;
2935                 part = min_t(loff_t, isize - *ppos, len);
2936
2937                 if (folio) {
2938                         /*
2939                          * If users can be writing to this page using arbitrary
2940                          * virtual addresses, take care about potential aliasing
2941                          * before reading the page on the kernel side.
2942                          */
2943                         if (mapping_writably_mapped(mapping))
2944                                 flush_dcache_folio(folio);
2945                         folio_mark_accessed(folio);
2946                         /*
2947                          * Ok, we have the page, and it's up-to-date, so we can
2948                          * now splice it into the pipe.
2949                          */
2950                         n = splice_folio_into_pipe(pipe, folio, *ppos, part);
2951                         folio_put(folio);
2952                         folio = NULL;
2953                 } else {
2954                         n = splice_zeropage_into_pipe(pipe, *ppos, part);
2955                 }
2956
2957                 if (!n)
2958                         break;
2959                 len -= n;
2960                 total_spliced += n;
2961                 *ppos += n;
2962                 in->f_ra.prev_pos = *ppos;
2963                 if (pipe_full(pipe->head, pipe->tail, pipe->max_usage))
2964                         break;
2965
2966                 cond_resched();
2967         } while (len);
2968
2969         if (folio)
2970                 folio_put(folio);
2971
2972         file_accessed(in);
2973         return total_spliced ? total_spliced : error;
2974 }
2975
2976 static loff_t shmem_file_llseek(struct file *file, loff_t offset, int whence)
2977 {
2978         struct address_space *mapping = file->f_mapping;
2979         struct inode *inode = mapping->host;
2980
2981         if (whence != SEEK_DATA && whence != SEEK_HOLE)
2982                 return generic_file_llseek_size(file, offset, whence,
2983                                         MAX_LFS_FILESIZE, i_size_read(inode));
2984         if (offset < 0)
2985                 return -ENXIO;
2986
2987         inode_lock(inode);
2988         /* We're holding i_rwsem so we can access i_size directly */
2989         offset = mapping_seek_hole_data(mapping, offset, inode->i_size, whence);
2990         if (offset >= 0)
2991                 offset = vfs_setpos(file, offset, MAX_LFS_FILESIZE);
2992         inode_unlock(inode);
2993         return offset;
2994 }
2995
2996 static long shmem_fallocate(struct file *file, int mode, loff_t offset,
2997                                                          loff_t len)
2998 {
2999         struct inode *inode = file_inode(file);
3000         struct shmem_sb_info *sbinfo = SHMEM_SB(inode->i_sb);
3001         struct shmem_inode_info *info = SHMEM_I(inode);
3002         struct shmem_falloc shmem_falloc;
3003         pgoff_t start, index, end, undo_fallocend;
3004         int error;
3005
3006         if (mode & ~(FALLOC_FL_KEEP_SIZE | FALLOC_FL_PUNCH_HOLE))
3007                 return -EOPNOTSUPP;
3008
3009         inode_lock(inode);
3010
3011         if (mode & FALLOC_FL_PUNCH_HOLE) {
3012                 struct address_space *mapping = file->f_mapping;
3013                 loff_t unmap_start = round_up(offset, PAGE_SIZE);
3014                 loff_t unmap_end = round_down(offset + len, PAGE_SIZE) - 1;
3015                 DECLARE_WAIT_QUEUE_HEAD_ONSTACK(shmem_falloc_waitq);
3016
3017                 /* protected by i_rwsem */
3018                 if (info->seals & (F_SEAL_WRITE | F_SEAL_FUTURE_WRITE)) {
3019                         error = -EPERM;
3020                         goto out;
3021                 }
3022
3023                 shmem_falloc.waitq = &shmem_falloc_waitq;
3024                 shmem_falloc.start = (u64)unmap_start >> PAGE_SHIFT;
3025                 shmem_falloc.next = (unmap_end + 1) >> PAGE_SHIFT;
3026                 spin_lock(&inode->i_lock);
3027                 inode->i_private = &shmem_falloc;
3028                 spin_unlock(&inode->i_lock);
3029
3030                 if ((u64)unmap_end > (u64)unmap_start)
3031                         unmap_mapping_range(mapping, unmap_start,
3032                                             1 + unmap_end - unmap_start, 0);
3033                 shmem_truncate_range(inode, offset, offset + len - 1);
3034                 /* No need to unmap again: hole-punching leaves COWed pages */
3035
3036                 spin_lock(&inode->i_lock);
3037                 inode->i_private = NULL;
3038                 wake_up_all(&shmem_falloc_waitq);
3039                 WARN_ON_ONCE(!list_empty(&shmem_falloc_waitq.head));
3040                 spin_unlock(&inode->i_lock);
3041                 error = 0;
3042                 goto out;
3043         }
3044
3045         /* We need to check rlimit even when FALLOC_FL_KEEP_SIZE */
3046         error = inode_newsize_ok(inode, offset + len);
3047         if (error)
3048                 goto out;
3049
3050         if ((info->seals & F_SEAL_GROW) && offset + len > inode->i_size) {
3051                 error = -EPERM;
3052                 goto out;
3053         }
3054
3055         start = offset >> PAGE_SHIFT;
3056         end = (offset + len + PAGE_SIZE - 1) >> PAGE_SHIFT;
3057         /* Try to avoid a swapstorm if len is impossible to satisfy */
3058         if (sbinfo->max_blocks && end - start > sbinfo->max_blocks) {
3059                 error = -ENOSPC;
3060                 goto out;
3061         }
3062
3063         shmem_falloc.waitq = NULL;
3064         shmem_falloc.start = start;
3065         shmem_falloc.next  = start;
3066         shmem_falloc.nr_falloced = 0;
3067         shmem_falloc.nr_unswapped = 0;
3068         spin_lock(&inode->i_lock);
3069         inode->i_private = &shmem_falloc;
3070         spin_unlock(&inode->i_lock);
3071
3072         /*
3073          * info->fallocend is only relevant when huge pages might be
3074          * involved: to prevent split_huge_page() freeing fallocated
3075          * pages when FALLOC_FL_KEEP_SIZE committed beyond i_size.
3076          */
3077         undo_fallocend = info->fallocend;
3078         if (info->fallocend < end)
3079                 info->fallocend = end;
3080
3081         for (index = start; index < end; ) {
3082                 struct folio *folio;
3083
3084                 /*
3085                  * Good, the fallocate(2) manpage permits EINTR: we may have
3086                  * been interrupted because we are using up too much memory.
3087                  */
3088                 if (signal_pending(current))
3089                         error = -EINTR;
3090                 else if (shmem_falloc.nr_unswapped > shmem_falloc.nr_falloced)
3091                         error = -ENOMEM;
3092                 else
3093                         error = shmem_get_folio(inode, index, &folio,
3094                                                 SGP_FALLOC);
3095                 if (error) {
3096                         info->fallocend = undo_fallocend;
3097                         /* Remove the !uptodate folios we added */
3098                         if (index > start) {
3099                                 shmem_undo_range(inode,
3100                                     (loff_t)start << PAGE_SHIFT,
3101                                     ((loff_t)index << PAGE_SHIFT) - 1, true);
3102                         }
3103                         goto undone;
3104                 }
3105
3106                 /*
3107                  * Here is a more important optimization than it appears:
3108                  * a second SGP_FALLOC on the same large folio will clear it,
3109                  * making it uptodate and un-undoable if we fail later.
3110                  */
3111                 index = folio_next_index(folio);
3112                 /* Beware 32-bit wraparound */
3113                 if (!index)
3114                         index--;
3115
3116                 /*
3117                  * Inform shmem_writepage() how far we have reached.
3118                  * No need for lock or barrier: we have the page lock.
3119                  */
3120                 if (!folio_test_uptodate(folio))
3121                         shmem_falloc.nr_falloced += index - shmem_falloc.next;
3122                 shmem_falloc.next = index;
3123
3124                 /*
3125                  * If !uptodate, leave it that way so that freeable folios
3126                  * can be recognized if we need to rollback on error later.
3127                  * But mark it dirty so that memory pressure will swap rather
3128                  * than free the folios we are allocating (and SGP_CACHE folios
3129                  * might still be clean: we now need to mark those dirty too).
3130                  */
3131                 folio_mark_dirty(folio);
3132                 folio_unlock(folio);
3133                 folio_put(folio);
3134                 cond_resched();
3135         }
3136
3137         if (!(mode & FALLOC_FL_KEEP_SIZE) && offset + len > inode->i_size)
3138                 i_size_write(inode, offset + len);
3139 undone:
3140         spin_lock(&inode->i_lock);
3141         inode->i_private = NULL;
3142         spin_unlock(&inode->i_lock);
3143 out:
3144         if (!error)
3145                 file_modified(file);
3146         inode_unlock(inode);
3147         return error;
3148 }
3149
3150 static int shmem_statfs(struct dentry *dentry, struct kstatfs *buf)
3151 {
3152         struct shmem_sb_info *sbinfo = SHMEM_SB(dentry->d_sb);
3153
3154         buf->f_type = TMPFS_MAGIC;
3155         buf->f_bsize = PAGE_SIZE;
3156         buf->f_namelen = NAME_MAX;
3157         if (sbinfo->max_blocks) {
3158                 buf->f_blocks = sbinfo->max_blocks;
3159                 buf->f_bavail =
3160                 buf->f_bfree  = sbinfo->max_blocks -
3161                                 percpu_counter_sum(&sbinfo->used_blocks);
3162         }
3163         if (sbinfo->max_inodes) {
3164                 buf->f_files = sbinfo->max_inodes;
3165                 buf->f_ffree = sbinfo->free_ispace / BOGO_INODE_SIZE;
3166         }
3167         /* else leave those fields 0 like simple_statfs */
3168
3169         buf->f_fsid = uuid_to_fsid(dentry->d_sb->s_uuid.b);
3170
3171         return 0;
3172 }
3173
3174 /*
3175  * File creation. Allocate an inode, and we're done..
3176  */
3177 static int
3178 shmem_mknod(struct mnt_idmap *idmap, struct inode *dir,
3179             struct dentry *dentry, umode_t mode, dev_t dev)
3180 {
3181         struct inode *inode;
3182         int error;
3183
3184         inode = shmem_get_inode(idmap, dir->i_sb, dir, mode, dev, VM_NORESERVE);
3185
3186         if (IS_ERR(inode))
3187                 return PTR_ERR(inode);
3188
3189         error = simple_acl_create(dir, inode);
3190         if (error)
3191                 goto out_iput;
3192         error = security_inode_init_security(inode, dir,
3193                                              &dentry->d_name,
3194                                              shmem_initxattrs, NULL);
3195         if (error && error != -EOPNOTSUPP)
3196                 goto out_iput;
3197
3198         error = simple_offset_add(shmem_get_offset_ctx(dir), dentry);
3199         if (error)
3200                 goto out_iput;
3201
3202         dir->i_size += BOGO_DIRENT_SIZE;
3203         dir->i_ctime = dir->i_mtime = current_time(dir);
3204         inode_inc_iversion(dir);
3205         d_instantiate(dentry, inode);
3206         dget(dentry); /* Extra count - pin the dentry in core */
3207         return error;
3208
3209 out_iput:
3210         iput(inode);
3211         return error;
3212 }
3213
3214 static int
3215 shmem_tmpfile(struct mnt_idmap *idmap, struct inode *dir,
3216               struct file *file, umode_t mode)
3217 {
3218         struct inode *inode;
3219         int error;
3220
3221         inode = shmem_get_inode(idmap, dir->i_sb, dir, mode, 0, VM_NORESERVE);
3222
3223         if (IS_ERR(inode)) {
3224                 error = PTR_ERR(inode);
3225                 goto err_out;
3226         }
3227
3228         error = security_inode_init_security(inode, dir,
3229                                              NULL,
3230                                              shmem_initxattrs, NULL);
3231         if (error && error != -EOPNOTSUPP)
3232                 goto out_iput;
3233         error = simple_acl_create(dir, inode);
3234         if (error)
3235                 goto out_iput;
3236         d_tmpfile(file, inode);
3237
3238 err_out:
3239         return finish_open_simple(file, error);
3240 out_iput:
3241         iput(inode);
3242         return error;
3243 }
3244
3245 static int shmem_mkdir(struct mnt_idmap *idmap, struct inode *dir,
3246                        struct dentry *dentry, umode_t mode)
3247 {
3248         int error;
3249
3250         error = shmem_mknod(idmap, dir, dentry, mode | S_IFDIR, 0);
3251         if (error)
3252                 return error;
3253         inc_nlink(dir);
3254         return 0;
3255 }
3256
3257 static int shmem_create(struct mnt_idmap *idmap, struct inode *dir,
3258                         struct dentry *dentry, umode_t mode, bool excl)
3259 {
3260         return shmem_mknod(idmap, dir, dentry, mode | S_IFREG, 0);
3261 }
3262
3263 /*
3264  * Link a file..
3265  */
3266 static int shmem_link(struct dentry *old_dentry, struct inode *dir, struct dentry *dentry)
3267 {
3268         struct inode *inode = d_inode(old_dentry);
3269         int ret = 0;
3270
3271         /*
3272          * No ordinary (disk based) filesystem counts links as inodes;
3273          * but each new link needs a new dentry, pinning lowmem, and
3274          * tmpfs dentries cannot be pruned until they are unlinked.
3275          * But if an O_TMPFILE file is linked into the tmpfs, the
3276          * first link must skip that, to get the accounting right.
3277          */
3278         if (inode->i_nlink) {
3279                 ret = shmem_reserve_inode(inode->i_sb, NULL);
3280                 if (ret)
3281                         goto out;
3282         }
3283
3284         ret = simple_offset_add(shmem_get_offset_ctx(dir), dentry);
3285         if (ret) {
3286                 if (inode->i_nlink)
3287                         shmem_free_inode(inode->i_sb);
3288                 goto out;
3289         }
3290
3291         dir->i_size += BOGO_DIRENT_SIZE;
3292         inode->i_ctime = dir->i_ctime = dir->i_mtime = current_time(inode);
3293         inode_inc_iversion(dir);
3294         inc_nlink(inode);
3295         ihold(inode);   /* New dentry reference */
3296         dget(dentry);           /* Extra pinning count for the created dentry */
3297         d_instantiate(dentry, inode);
3298 out:
3299         return ret;
3300 }
3301
3302 static int shmem_unlink(struct inode *dir, struct dentry *dentry)
3303 {
3304         struct inode *inode = d_inode(dentry);
3305
3306         if (inode->i_nlink > 1 && !S_ISDIR(inode->i_mode))
3307                 shmem_free_inode(inode->i_sb);
3308
3309         simple_offset_remove(shmem_get_offset_ctx(dir), dentry);
3310
3311         dir->i_size -= BOGO_DIRENT_SIZE;
3312         inode->i_ctime = dir->i_ctime = dir->i_mtime = current_time(inode);
3313         inode_inc_iversion(dir);
3314         drop_nlink(inode);
3315         dput(dentry);   /* Undo the count from "create" - this does all the work */
3316         return 0;
3317 }
3318
3319 static int shmem_rmdir(struct inode *dir, struct dentry *dentry)
3320 {
3321         if (!simple_empty(dentry))
3322                 return -ENOTEMPTY;
3323
3324         drop_nlink(d_inode(dentry));
3325         drop_nlink(dir);
3326         return shmem_unlink(dir, dentry);
3327 }
3328
3329 static int shmem_whiteout(struct mnt_idmap *idmap,
3330                           struct inode *old_dir, struct dentry *old_dentry)
3331 {
3332         struct dentry *whiteout;
3333         int error;
3334
3335         whiteout = d_alloc(old_dentry->d_parent, &old_dentry->d_name);
3336         if (!whiteout)
3337                 return -ENOMEM;
3338
3339         error = shmem_mknod(idmap, old_dir, whiteout,
3340                             S_IFCHR | WHITEOUT_MODE, WHITEOUT_DEV);
3341         dput(whiteout);
3342         if (error)
3343                 return error;
3344
3345         /*
3346          * Cheat and hash the whiteout while the old dentry is still in
3347          * place, instead of playing games with FS_RENAME_DOES_D_MOVE.
3348          *
3349          * d_lookup() will consistently find one of them at this point,
3350          * not sure which one, but that isn't even important.
3351          */
3352         d_rehash(whiteout);
3353         return 0;
3354 }
3355
3356 /*
3357  * The VFS layer already does all the dentry stuff for rename,
3358  * we just have to decrement the usage count for the target if
3359  * it exists so that the VFS layer correctly free's it when it
3360  * gets overwritten.
3361  */
3362 static int shmem_rename2(struct mnt_idmap *idmap,
3363                          struct inode *old_dir, struct dentry *old_dentry,
3364                          struct inode *new_dir, struct dentry *new_dentry,
3365                          unsigned int flags)
3366 {
3367         struct inode *inode = d_inode(old_dentry);
3368         int they_are_dirs = S_ISDIR(inode->i_mode);
3369         int error;
3370
3371         if (flags & ~(RENAME_NOREPLACE | RENAME_EXCHANGE | RENAME_WHITEOUT))
3372                 return -EINVAL;
3373
3374         if (flags & RENAME_EXCHANGE)
3375                 return simple_offset_rename_exchange(old_dir, old_dentry,
3376                                                      new_dir, new_dentry);
3377
3378         if (!simple_empty(new_dentry))
3379                 return -ENOTEMPTY;
3380
3381         if (flags & RENAME_WHITEOUT) {
3382                 error = shmem_whiteout(idmap, old_dir, old_dentry);
3383                 if (error)
3384                         return error;
3385         }
3386
3387         simple_offset_remove(shmem_get_offset_ctx(old_dir), old_dentry);
3388         error = simple_offset_add(shmem_get_offset_ctx(new_dir), old_dentry);
3389         if (error)
3390                 return error;
3391
3392         if (d_really_is_positive(new_dentry)) {
3393                 (void) shmem_unlink(new_dir, new_dentry);
3394                 if (they_are_dirs) {
3395                         drop_nlink(d_inode(new_dentry));
3396                         drop_nlink(old_dir);
3397                 }
3398         } else if (they_are_dirs) {
3399                 drop_nlink(old_dir);
3400                 inc_nlink(new_dir);
3401         }
3402
3403         old_dir->i_size -= BOGO_DIRENT_SIZE;
3404         new_dir->i_size += BOGO_DIRENT_SIZE;
3405         old_dir->i_ctime = old_dir->i_mtime =
3406         new_dir->i_ctime = new_dir->i_mtime =
3407         inode->i_ctime = current_time(old_dir);
3408         inode_inc_iversion(old_dir);
3409         inode_inc_iversion(new_dir);
3410         return 0;
3411 }
3412
3413 static int shmem_symlink(struct mnt_idmap *idmap, struct inode *dir,
3414                          struct dentry *dentry, const char *symname)
3415 {
3416         int error;
3417         int len;
3418         struct inode *inode;
3419         struct folio *folio;
3420
3421         len = strlen(symname) + 1;
3422         if (len > PAGE_SIZE)
3423                 return -ENAMETOOLONG;
3424
3425         inode = shmem_get_inode(idmap, dir->i_sb, dir, S_IFLNK | 0777, 0,
3426                                 VM_NORESERVE);
3427
3428         if (IS_ERR(inode))
3429                 return PTR_ERR(inode);
3430
3431         error = security_inode_init_security(inode, dir, &dentry->d_name,
3432                                              shmem_initxattrs, NULL);
3433         if (error && error != -EOPNOTSUPP)
3434                 goto out_iput;
3435
3436         error = simple_offset_add(shmem_get_offset_ctx(dir), dentry);
3437         if (error)
3438                 goto out_iput;
3439
3440         inode->i_size = len-1;
3441         if (len <= SHORT_SYMLINK_LEN) {
3442                 inode->i_link = kmemdup(symname, len, GFP_KERNEL);
3443                 if (!inode->i_link) {
3444                         error = -ENOMEM;
3445                         goto out_remove_offset;
3446                 }
3447                 inode->i_op = &shmem_short_symlink_operations;
3448         } else {
3449                 inode_nohighmem(inode);
3450                 error = shmem_get_folio(inode, 0, &folio, SGP_WRITE);
3451                 if (error)
3452                         goto out_remove_offset;
3453                 inode->i_mapping->a_ops = &shmem_aops;
3454                 inode->i_op = &shmem_symlink_inode_operations;
3455                 memcpy(folio_address(folio), symname, len);
3456                 folio_mark_uptodate(folio);
3457                 folio_mark_dirty(folio);
3458                 folio_unlock(folio);
3459                 folio_put(folio);
3460         }
3461         dir->i_size += BOGO_DIRENT_SIZE;
3462         dir->i_ctime = dir->i_mtime = current_time(dir);
3463         inode_inc_iversion(dir);
3464         d_instantiate(dentry, inode);
3465         dget(dentry);
3466         return 0;
3467
3468 out_remove_offset:
3469         simple_offset_remove(shmem_get_offset_ctx(dir), dentry);
3470 out_iput:
3471         iput(inode);
3472         return error;
3473 }
3474
3475 static void shmem_put_link(void *arg)
3476 {
3477         folio_mark_accessed(arg);
3478         folio_put(arg);
3479 }
3480
3481 static const char *shmem_get_link(struct dentry *dentry,
3482                                   struct inode *inode,
3483                                   struct delayed_call *done)
3484 {
3485         struct folio *folio = NULL;
3486         int error;
3487
3488         if (!dentry) {
3489                 folio = filemap_get_folio(inode->i_mapping, 0);
3490                 if (IS_ERR(folio))
3491                         return ERR_PTR(-ECHILD);
3492                 if (PageHWPoison(folio_page(folio, 0)) ||
3493                     !folio_test_uptodate(folio)) {
3494                         folio_put(folio);
3495                         return ERR_PTR(-ECHILD);
3496                 }
3497         } else {
3498                 error = shmem_get_folio(inode, 0, &folio, SGP_READ);
3499                 if (error)
3500                         return ERR_PTR(error);
3501                 if (!folio)
3502                         return ERR_PTR(-ECHILD);
3503                 if (PageHWPoison(folio_page(folio, 0))) {
3504                         folio_unlock(folio);
3505                         folio_put(folio);
3506                         return ERR_PTR(-ECHILD);
3507                 }
3508                 folio_unlock(folio);
3509         }
3510         set_delayed_call(done, shmem_put_link, folio);
3511         return folio_address(folio);
3512 }
3513
3514 #ifdef CONFIG_TMPFS_XATTR
3515
3516 static int shmem_fileattr_get(struct dentry *dentry, struct fileattr *fa)
3517 {
3518         struct shmem_inode_info *info = SHMEM_I(d_inode(dentry));
3519
3520         fileattr_fill_flags(fa, info->fsflags & SHMEM_FL_USER_VISIBLE);
3521
3522         return 0;
3523 }
3524
3525 static int shmem_fileattr_set(struct mnt_idmap *idmap,
3526                               struct dentry *dentry, struct fileattr *fa)
3527 {
3528         struct inode *inode = d_inode(dentry);
3529         struct shmem_inode_info *info = SHMEM_I(inode);
3530
3531         if (fileattr_has_fsx(fa))
3532                 return -EOPNOTSUPP;
3533         if (fa->flags & ~SHMEM_FL_USER_MODIFIABLE)
3534                 return -EOPNOTSUPP;
3535
3536         info->fsflags = (info->fsflags & ~SHMEM_FL_USER_MODIFIABLE) |
3537                 (fa->flags & SHMEM_FL_USER_MODIFIABLE);
3538
3539         shmem_set_inode_flags(inode, info->fsflags);
3540         inode->i_ctime = current_time(inode);
3541         inode_inc_iversion(inode);
3542         return 0;
3543 }
3544
3545 /*
3546  * Superblocks without xattr inode operations may get some security.* xattr
3547  * support from the LSM "for free". As soon as we have any other xattrs
3548  * like ACLs, we also need to implement the security.* handlers at
3549  * filesystem level, though.
3550  */
3551
3552 /*
3553  * Callback for security_inode_init_security() for acquiring xattrs.
3554  */
3555 static int shmem_initxattrs(struct inode *inode,
3556                             const struct xattr *xattr_array,
3557                             void *fs_info)
3558 {
3559         struct shmem_inode_info *info = SHMEM_I(inode);
3560         const struct xattr *xattr;
3561         struct simple_xattr *new_xattr;
3562         size_t len;
3563
3564         for (xattr = xattr_array; xattr->name != NULL; xattr++) {
3565                 new_xattr = simple_xattr_alloc(xattr->value, xattr->value_len);
3566                 if (!new_xattr)
3567                         return -ENOMEM;
3568
3569                 len = strlen(xattr->name) + 1;
3570                 new_xattr->name = kmalloc(XATTR_SECURITY_PREFIX_LEN + len,
3571                                           GFP_KERNEL);
3572                 if (!new_xattr->name) {
3573                         kvfree(new_xattr);
3574                         return -ENOMEM;
3575                 }
3576
3577                 memcpy(new_xattr->name, XATTR_SECURITY_PREFIX,
3578                        XATTR_SECURITY_PREFIX_LEN);
3579                 memcpy(new_xattr->name + XATTR_SECURITY_PREFIX_LEN,
3580                        xattr->name, len);
3581
3582                 simple_xattr_add(&info->xattrs, new_xattr);
3583         }
3584
3585         return 0;
3586 }
3587
3588 static int shmem_xattr_handler_get(const struct xattr_handler *handler,
3589                                    struct dentry *unused, struct inode *inode,
3590                                    const char *name, void *buffer, size_t size)
3591 {
3592         struct shmem_inode_info *info = SHMEM_I(inode);
3593
3594         name = xattr_full_name(handler, name);
3595         return simple_xattr_get(&info->xattrs, name, buffer, size);
3596 }
3597
3598 static int shmem_xattr_handler_set(const struct xattr_handler *handler,
3599                                    struct mnt_idmap *idmap,
3600                                    struct dentry *unused, struct inode *inode,
3601                                    const char *name, const void *value,
3602                                    size_t size, int flags)
3603 {
3604         struct shmem_inode_info *info = SHMEM_I(inode);
3605         struct simple_xattr *old_xattr;
3606
3607         name = xattr_full_name(handler, name);
3608         old_xattr = simple_xattr_set(&info->xattrs, name, value, size, flags);
3609         if (!IS_ERR(old_xattr)) {
3610                 simple_xattr_free(old_xattr);
3611                 old_xattr = NULL;
3612                 inode->i_ctime = current_time(inode);
3613                 inode_inc_iversion(inode);
3614         }
3615         return PTR_ERR(old_xattr);
3616 }
3617
3618 static const struct xattr_handler shmem_security_xattr_handler = {
3619         .prefix = XATTR_SECURITY_PREFIX,
3620         .get = shmem_xattr_handler_get,
3621         .set = shmem_xattr_handler_set,
3622 };
3623
3624 static const struct xattr_handler shmem_trusted_xattr_handler = {
3625         .prefix = XATTR_TRUSTED_PREFIX,
3626         .get = shmem_xattr_handler_get,
3627         .set = shmem_xattr_handler_set,
3628 };
3629
3630 static const struct xattr_handler *shmem_xattr_handlers[] = {
3631         &shmem_security_xattr_handler,
3632         &shmem_trusted_xattr_handler,
3633         NULL
3634 };
3635
3636 static ssize_t shmem_listxattr(struct dentry *dentry, char *buffer, size_t size)
3637 {
3638         struct shmem_inode_info *info = SHMEM_I(d_inode(dentry));
3639         return simple_xattr_list(d_inode(dentry), &info->xattrs, buffer, size);
3640 }
3641 #endif /* CONFIG_TMPFS_XATTR */
3642
3643 static const struct inode_operations shmem_short_symlink_operations = {
3644         .getattr        = shmem_getattr,
3645         .setattr        = shmem_setattr,
3646         .get_link       = simple_get_link,
3647 #ifdef CONFIG_TMPFS_XATTR
3648         .listxattr      = shmem_listxattr,
3649 #endif
3650 };
3651
3652 static const struct inode_operations shmem_symlink_inode_operations = {
3653         .getattr        = shmem_getattr,
3654         .setattr        = shmem_setattr,
3655         .get_link       = shmem_get_link,
3656 #ifdef CONFIG_TMPFS_XATTR
3657         .listxattr      = shmem_listxattr,
3658 #endif
3659 };
3660
3661 static struct dentry *shmem_get_parent(struct dentry *child)
3662 {
3663         return ERR_PTR(-ESTALE);
3664 }
3665
3666 static int shmem_match(struct inode *ino, void *vfh)
3667 {
3668         __u32 *fh = vfh;
3669         __u64 inum = fh[2];
3670         inum = (inum << 32) | fh[1];
3671         return ino->i_ino == inum && fh[0] == ino->i_generation;
3672 }
3673
3674 /* Find any alias of inode, but prefer a hashed alias */
3675 static struct dentry *shmem_find_alias(struct inode *inode)
3676 {
3677         struct dentry *alias = d_find_alias(inode);
3678
3679         return alias ?: d_find_any_alias(inode);
3680 }
3681
3682
3683 static struct dentry *shmem_fh_to_dentry(struct super_block *sb,
3684                 struct fid *fid, int fh_len, int fh_type)
3685 {
3686         struct inode *inode;
3687         struct dentry *dentry = NULL;
3688         u64 inum;
3689
3690         if (fh_len < 3)
3691                 return NULL;
3692
3693         inum = fid->raw[2];
3694         inum = (inum << 32) | fid->raw[1];
3695
3696         inode = ilookup5(sb, (unsigned long)(inum + fid->raw[0]),
3697                         shmem_match, fid->raw);
3698         if (inode) {
3699                 dentry = shmem_find_alias(inode);
3700                 iput(inode);
3701         }
3702
3703         return dentry;
3704 }
3705
3706 static int shmem_encode_fh(struct inode *inode, __u32 *fh, int *len,
3707                                 struct inode *parent)
3708 {
3709         if (*len < 3) {
3710                 *len = 3;
3711                 return FILEID_INVALID;
3712         }
3713
3714         if (inode_unhashed(inode)) {
3715                 /* Unfortunately insert_inode_hash is not idempotent,
3716                  * so as we hash inodes here rather than at creation
3717                  * time, we need a lock to ensure we only try
3718                  * to do it once
3719                  */
3720                 static DEFINE_SPINLOCK(lock);
3721                 spin_lock(&lock);
3722                 if (inode_unhashed(inode))
3723                         __insert_inode_hash(inode,
3724                                             inode->i_ino + inode->i_generation);
3725                 spin_unlock(&lock);
3726         }
3727
3728         fh[0] = inode->i_generation;
3729         fh[1] = inode->i_ino;
3730         fh[2] = ((__u64)inode->i_ino) >> 32;
3731
3732         *len = 3;
3733         return 1;
3734 }
3735
3736 static const struct export_operations shmem_export_ops = {
3737         .get_parent     = shmem_get_parent,
3738         .encode_fh      = shmem_encode_fh,
3739         .fh_to_dentry   = shmem_fh_to_dentry,
3740 };
3741
3742 enum shmem_param {
3743         Opt_gid,
3744         Opt_huge,
3745         Opt_mode,
3746         Opt_mpol,
3747         Opt_nr_blocks,
3748         Opt_nr_inodes,
3749         Opt_size,
3750         Opt_uid,
3751         Opt_inode32,
3752         Opt_inode64,
3753         Opt_noswap,
3754         Opt_quota,
3755         Opt_usrquota,
3756         Opt_grpquota,
3757         Opt_usrquota_block_hardlimit,
3758         Opt_usrquota_inode_hardlimit,
3759         Opt_grpquota_block_hardlimit,
3760         Opt_grpquota_inode_hardlimit,
3761 };
3762
3763 static const struct constant_table shmem_param_enums_huge[] = {
3764         {"never",       SHMEM_HUGE_NEVER },
3765         {"always",      SHMEM_HUGE_ALWAYS },
3766         {"within_size", SHMEM_HUGE_WITHIN_SIZE },
3767         {"advise",      SHMEM_HUGE_ADVISE },
3768         {}
3769 };
3770
3771 const struct fs_parameter_spec shmem_fs_parameters[] = {
3772         fsparam_u32   ("gid",           Opt_gid),
3773         fsparam_enum  ("huge",          Opt_huge,  shmem_param_enums_huge),
3774         fsparam_u32oct("mode",          Opt_mode),
3775         fsparam_string("mpol",          Opt_mpol),
3776         fsparam_string("nr_blocks",     Opt_nr_blocks),
3777         fsparam_string("nr_inodes",     Opt_nr_inodes),
3778         fsparam_string("size",          Opt_size),
3779         fsparam_u32   ("uid",           Opt_uid),
3780         fsparam_flag  ("inode32",       Opt_inode32),
3781         fsparam_flag  ("inode64",       Opt_inode64),
3782         fsparam_flag  ("noswap",        Opt_noswap),
3783 #ifdef CONFIG_TMPFS_QUOTA
3784         fsparam_flag  ("quota",         Opt_quota),
3785         fsparam_flag  ("usrquota",      Opt_usrquota),
3786         fsparam_flag  ("grpquota",      Opt_grpquota),
3787         fsparam_string("usrquota_block_hardlimit", Opt_usrquota_block_hardlimit),
3788         fsparam_string("usrquota_inode_hardlimit", Opt_usrquota_inode_hardlimit),
3789         fsparam_string("grpquota_block_hardlimit", Opt_grpquota_block_hardlimit),
3790         fsparam_string("grpquota_inode_hardlimit", Opt_grpquota_inode_hardlimit),
3791 #endif
3792         {}
3793 };
3794
3795 static int shmem_parse_one(struct fs_context *fc, struct fs_parameter *param)
3796 {
3797         struct shmem_options *ctx = fc->fs_private;
3798         struct fs_parse_result result;
3799         unsigned long long size;
3800         char *rest;
3801         int opt;
3802         kuid_t kuid;
3803         kgid_t kgid;
3804
3805         opt = fs_parse(fc, shmem_fs_parameters, param, &result);
3806         if (opt < 0)
3807                 return opt;
3808
3809         switch (opt) {
3810         case Opt_size:
3811                 size = memparse(param->string, &rest);
3812                 if (*rest == '%') {
3813                         size <<= PAGE_SHIFT;
3814                         size *= totalram_pages();
3815                         do_div(size, 100);
3816                         rest++;
3817                 }
3818                 if (*rest)
3819                         goto bad_value;
3820                 ctx->blocks = DIV_ROUND_UP(size, PAGE_SIZE);
3821                 ctx->seen |= SHMEM_SEEN_BLOCKS;
3822                 break;
3823         case Opt_nr_blocks:
3824                 ctx->blocks = memparse(param->string, &rest);
3825                 if (*rest || ctx->blocks > LONG_MAX)
3826                         goto bad_value;
3827                 ctx->seen |= SHMEM_SEEN_BLOCKS;
3828                 break;
3829         case Opt_nr_inodes:
3830                 ctx->inodes = memparse(param->string, &rest);
3831                 if (*rest || ctx->inodes > ULONG_MAX / BOGO_INODE_SIZE)
3832                         goto bad_value;
3833                 ctx->seen |= SHMEM_SEEN_INODES;
3834                 break;
3835         case Opt_mode:
3836                 ctx->mode = result.uint_32 & 07777;
3837                 break;
3838         case Opt_uid:
3839                 kuid = make_kuid(current_user_ns(), result.uint_32);
3840                 if (!uid_valid(kuid))
3841                         goto bad_value;
3842
3843                 /*
3844                  * The requested uid must be representable in the
3845                  * filesystem's idmapping.
3846                  */
3847                 if (!kuid_has_mapping(fc->user_ns, kuid))
3848                         goto bad_value;
3849
3850                 ctx->uid = kuid;
3851                 break;
3852         case Opt_gid:
3853                 kgid = make_kgid(current_user_ns(), result.uint_32);
3854                 if (!gid_valid(kgid))
3855                         goto bad_value;
3856
3857                 /*
3858                  * The requested gid must be representable in the
3859                  * filesystem's idmapping.
3860                  */
3861                 if (!kgid_has_mapping(fc->user_ns, kgid))
3862                         goto bad_value;
3863
3864                 ctx->gid = kgid;
3865                 break;
3866         case Opt_huge:
3867                 ctx->huge = result.uint_32;
3868                 if (ctx->huge != SHMEM_HUGE_NEVER &&
3869                     !(IS_ENABLED(CONFIG_TRANSPARENT_HUGEPAGE) &&
3870                       has_transparent_hugepage()))
3871                         goto unsupported_parameter;
3872                 ctx->seen |= SHMEM_SEEN_HUGE;
3873                 break;
3874         case Opt_mpol:
3875                 if (IS_ENABLED(CONFIG_NUMA)) {
3876                         mpol_put(ctx->mpol);
3877                         ctx->mpol = NULL;
3878                         if (mpol_parse_str(param->string, &ctx->mpol))
3879                                 goto bad_value;
3880                         break;
3881                 }
3882                 goto unsupported_parameter;
3883         case Opt_inode32:
3884                 ctx->full_inums = false;
3885                 ctx->seen |= SHMEM_SEEN_INUMS;
3886                 break;
3887         case Opt_inode64:
3888                 if (sizeof(ino_t) < 8) {
3889                         return invalfc(fc,
3890                                        "Cannot use inode64 with <64bit inums in kernel\n");
3891                 }
3892                 ctx->full_inums = true;
3893                 ctx->seen |= SHMEM_SEEN_INUMS;
3894                 break;
3895         case Opt_noswap:
3896                 if ((fc->user_ns != &init_user_ns) || !capable(CAP_SYS_ADMIN)) {
3897                         return invalfc(fc,
3898                                        "Turning off swap in unprivileged tmpfs mounts unsupported");
3899                 }
3900                 ctx->noswap = true;
3901                 ctx->seen |= SHMEM_SEEN_NOSWAP;
3902                 break;
3903         case Opt_quota:
3904                 if (fc->user_ns != &init_user_ns)
3905                         return invalfc(fc, "Quotas in unprivileged tmpfs mounts are unsupported");
3906                 ctx->seen |= SHMEM_SEEN_QUOTA;
3907                 ctx->quota_types |= (QTYPE_MASK_USR | QTYPE_MASK_GRP);
3908                 break;
3909         case Opt_usrquota:
3910                 if (fc->user_ns != &init_user_ns)
3911                         return invalfc(fc, "Quotas in unprivileged tmpfs mounts are unsupported");
3912                 ctx->seen |= SHMEM_SEEN_QUOTA;
3913                 ctx->quota_types |= QTYPE_MASK_USR;
3914                 break;
3915         case Opt_grpquota:
3916                 if (fc->user_ns != &init_user_ns)
3917                         return invalfc(fc, "Quotas in unprivileged tmpfs mounts are unsupported");
3918                 ctx->seen |= SHMEM_SEEN_QUOTA;
3919                 ctx->quota_types |= QTYPE_MASK_GRP;
3920                 break;
3921         case Opt_usrquota_block_hardlimit:
3922                 size = memparse(param->string, &rest);
3923                 if (*rest || !size)
3924                         goto bad_value;
3925                 if (size > SHMEM_QUOTA_MAX_SPC_LIMIT)
3926                         return invalfc(fc,
3927                                        "User quota block hardlimit too large.");
3928                 ctx->qlimits.usrquota_bhardlimit = size;
3929                 break;
3930         case Opt_grpquota_block_hardlimit:
3931                 size = memparse(param->string, &rest);
3932                 if (*rest || !size)
3933                         goto bad_value;
3934                 if (size > SHMEM_QUOTA_MAX_SPC_LIMIT)
3935                         return invalfc(fc,
3936                                        "Group quota block hardlimit too large.");
3937                 ctx->qlimits.grpquota_bhardlimit = size;
3938                 break;
3939         case Opt_usrquota_inode_hardlimit:
3940                 size = memparse(param->string, &rest);
3941                 if (*rest || !size)
3942                         goto bad_value;
3943                 if (size > SHMEM_QUOTA_MAX_INO_LIMIT)
3944                         return invalfc(fc,
3945                                        "User quota inode hardlimit too large.");
3946                 ctx->qlimits.usrquota_ihardlimit = size;
3947                 break;
3948         case Opt_grpquota_inode_hardlimit:
3949                 size = memparse(param->string, &rest);
3950                 if (*rest || !size)
3951                         goto bad_value;
3952                 if (size > SHMEM_QUOTA_MAX_INO_LIMIT)
3953                         return invalfc(fc,
3954                                        "Group quota inode hardlimit too large.");
3955                 ctx->qlimits.grpquota_ihardlimit = size;
3956                 break;
3957         }
3958         return 0;
3959
3960 unsupported_parameter:
3961         return invalfc(fc, "Unsupported parameter '%s'", param->key);
3962 bad_value:
3963         return invalfc(fc, "Bad value for '%s'", param->key);
3964 }
3965
3966 static int shmem_parse_options(struct fs_context *fc, void *data)
3967 {
3968         char *options = data;
3969
3970         if (options) {
3971                 int err = security_sb_eat_lsm_opts(options, &fc->security);
3972                 if (err)
3973                         return err;
3974         }
3975
3976         while (options != NULL) {
3977                 char *this_char = options;
3978                 for (;;) {
3979                         /*
3980                          * NUL-terminate this option: unfortunately,
3981                          * mount options form a comma-separated list,
3982                          * but mpol's nodelist may also contain commas.
3983                          */
3984                         options = strchr(options, ',');
3985                         if (options == NULL)
3986                                 break;
3987                         options++;
3988                         if (!isdigit(*options)) {
3989                                 options[-1] = '\0';
3990                                 break;
3991                         }
3992                 }
3993                 if (*this_char) {
3994                         char *value = strchr(this_char, '=');
3995                         size_t len = 0;
3996                         int err;
3997
3998                         if (value) {
3999                                 *value++ = '\0';
4000                                 len = strlen(value);
4001                         }
4002                         err = vfs_parse_fs_string(fc, this_char, value, len);
4003                         if (err < 0)
4004                                 return err;
4005                 }
4006         }
4007         return 0;
4008 }
4009
4010 /*
4011  * Reconfigure a shmem filesystem.
4012  */
4013 static int shmem_reconfigure(struct fs_context *fc)
4014 {
4015         struct shmem_options *ctx = fc->fs_private;
4016         struct shmem_sb_info *sbinfo = SHMEM_SB(fc->root->d_sb);
4017         unsigned long used_isp;
4018         struct mempolicy *mpol = NULL;
4019         const char *err;
4020
4021         raw_spin_lock(&sbinfo->stat_lock);
4022         used_isp = sbinfo->max_inodes * BOGO_INODE_SIZE - sbinfo->free_ispace;
4023
4024         if ((ctx->seen & SHMEM_SEEN_BLOCKS) && ctx->blocks) {
4025                 if (!sbinfo->max_blocks) {
4026                         err = "Cannot retroactively limit size";
4027                         goto out;
4028                 }
4029                 if (percpu_counter_compare(&sbinfo->used_blocks,
4030                                            ctx->blocks) > 0) {
4031                         err = "Too small a size for current use";
4032                         goto out;
4033                 }
4034         }
4035         if ((ctx->seen & SHMEM_SEEN_INODES) && ctx->inodes) {
4036                 if (!sbinfo->max_inodes) {
4037                         err = "Cannot retroactively limit inodes";
4038                         goto out;
4039                 }
4040                 if (ctx->inodes * BOGO_INODE_SIZE < used_isp) {
4041                         err = "Too few inodes for current use";
4042                         goto out;
4043                 }
4044         }
4045
4046         if ((ctx->seen & SHMEM_SEEN_INUMS) && !ctx->full_inums &&
4047             sbinfo->next_ino > UINT_MAX) {
4048                 err = "Current inum too high to switch to 32-bit inums";
4049                 goto out;
4050         }
4051         if ((ctx->seen & SHMEM_SEEN_NOSWAP) && ctx->noswap && !sbinfo->noswap) {
4052                 err = "Cannot disable swap on remount";
4053                 goto out;
4054         }
4055         if (!(ctx->seen & SHMEM_SEEN_NOSWAP) && !ctx->noswap && sbinfo->noswap) {
4056                 err = "Cannot enable swap on remount if it was disabled on first mount";
4057                 goto out;
4058         }
4059
4060         if (ctx->seen & SHMEM_SEEN_QUOTA &&
4061             !sb_any_quota_loaded(fc->root->d_sb)) {
4062                 err = "Cannot enable quota on remount";
4063                 goto out;
4064         }
4065
4066 #ifdef CONFIG_TMPFS_QUOTA
4067 #define CHANGED_LIMIT(name)                                             \
4068         (ctx->qlimits.name## hardlimit &&                               \
4069         (ctx->qlimits.name## hardlimit != sbinfo->qlimits.name## hardlimit))
4070
4071         if (CHANGED_LIMIT(usrquota_b) || CHANGED_LIMIT(usrquota_i) ||
4072             CHANGED_LIMIT(grpquota_b) || CHANGED_LIMIT(grpquota_i)) {
4073                 err = "Cannot change global quota limit on remount";
4074                 goto out;
4075         }
4076 #endif /* CONFIG_TMPFS_QUOTA */
4077
4078         if (ctx->seen & SHMEM_SEEN_HUGE)
4079                 sbinfo->huge = ctx->huge;
4080         if (ctx->seen & SHMEM_SEEN_INUMS)
4081                 sbinfo->full_inums = ctx->full_inums;
4082         if (ctx->seen & SHMEM_SEEN_BLOCKS)
4083                 sbinfo->max_blocks  = ctx->blocks;
4084         if (ctx->seen & SHMEM_SEEN_INODES) {
4085                 sbinfo->max_inodes  = ctx->inodes;
4086                 sbinfo->free_ispace = ctx->inodes * BOGO_INODE_SIZE - used_isp;
4087         }
4088
4089         /*
4090          * Preserve previous mempolicy unless mpol remount option was specified.
4091          */
4092         if (ctx->mpol) {
4093                 mpol = sbinfo->mpol;
4094                 sbinfo->mpol = ctx->mpol;       /* transfers initial ref */
4095                 ctx->mpol = NULL;
4096         }
4097
4098         if (ctx->noswap)
4099                 sbinfo->noswap = true;
4100
4101         raw_spin_unlock(&sbinfo->stat_lock);
4102         mpol_put(mpol);
4103         return 0;
4104 out:
4105         raw_spin_unlock(&sbinfo->stat_lock);
4106         return invalfc(fc, "%s", err);
4107 }
4108
4109 static int shmem_show_options(struct seq_file *seq, struct dentry *root)
4110 {
4111         struct shmem_sb_info *sbinfo = SHMEM_SB(root->d_sb);
4112         struct mempolicy *mpol;
4113
4114         if (sbinfo->max_blocks != shmem_default_max_blocks())
4115                 seq_printf(seq, ",size=%luk",
4116                         sbinfo->max_blocks << (PAGE_SHIFT - 10));
4117         if (sbinfo->max_inodes != shmem_default_max_inodes())
4118                 seq_printf(seq, ",nr_inodes=%lu", sbinfo->max_inodes);
4119         if (sbinfo->mode != (0777 | S_ISVTX))
4120                 seq_printf(seq, ",mode=%03ho", sbinfo->mode);
4121         if (!uid_eq(sbinfo->uid, GLOBAL_ROOT_UID))
4122                 seq_printf(seq, ",uid=%u",
4123                                 from_kuid_munged(&init_user_ns, sbinfo->uid));
4124         if (!gid_eq(sbinfo->gid, GLOBAL_ROOT_GID))
4125                 seq_printf(seq, ",gid=%u",
4126                                 from_kgid_munged(&init_user_ns, sbinfo->gid));
4127
4128         /*
4129          * Showing inode{64,32} might be useful even if it's the system default,
4130          * since then people don't have to resort to checking both here and
4131          * /proc/config.gz to confirm 64-bit inums were successfully applied
4132          * (which may not even exist if IKCONFIG_PROC isn't enabled).
4133          *
4134          * We hide it when inode64 isn't the default and we are using 32-bit
4135          * inodes, since that probably just means the feature isn't even under
4136          * consideration.
4137          *
4138          * As such:
4139          *
4140          *                     +-----------------+-----------------+
4141          *                     | TMPFS_INODE64=y | TMPFS_INODE64=n |
4142          *  +------------------+-----------------+-----------------+
4143          *  | full_inums=true  | show            | show            |
4144          *  | full_inums=false | show            | hide            |
4145          *  +------------------+-----------------+-----------------+
4146          *
4147          */
4148         if (IS_ENABLED(CONFIG_TMPFS_INODE64) || sbinfo->full_inums)
4149                 seq_printf(seq, ",inode%d", (sbinfo->full_inums ? 64 : 32));
4150 #ifdef CONFIG_TRANSPARENT_HUGEPAGE
4151         /* Rightly or wrongly, show huge mount option unmasked by shmem_huge */
4152         if (sbinfo->huge)
4153                 seq_printf(seq, ",huge=%s", shmem_format_huge(sbinfo->huge));
4154 #endif
4155         mpol = shmem_get_sbmpol(sbinfo);
4156         shmem_show_mpol(seq, mpol);
4157         mpol_put(mpol);
4158         if (sbinfo->noswap)
4159                 seq_printf(seq, ",noswap");
4160         return 0;
4161 }
4162
4163 #endif /* CONFIG_TMPFS */
4164
4165 static void shmem_put_super(struct super_block *sb)
4166 {
4167         struct shmem_sb_info *sbinfo = SHMEM_SB(sb);
4168
4169 #ifdef CONFIG_TMPFS_QUOTA
4170         shmem_disable_quotas(sb);
4171 #endif
4172         free_percpu(sbinfo->ino_batch);
4173         percpu_counter_destroy(&sbinfo->used_blocks);
4174         mpol_put(sbinfo->mpol);
4175         kfree(sbinfo);
4176         sb->s_fs_info = NULL;
4177 }
4178
4179 static int shmem_fill_super(struct super_block *sb, struct fs_context *fc)
4180 {
4181         struct shmem_options *ctx = fc->fs_private;
4182         struct inode *inode;
4183         struct shmem_sb_info *sbinfo;
4184         int error = -ENOMEM;
4185
4186         /* Round up to L1_CACHE_BYTES to resist false sharing */
4187         sbinfo = kzalloc(max((int)sizeof(struct shmem_sb_info),
4188                                 L1_CACHE_BYTES), GFP_KERNEL);
4189         if (!sbinfo)
4190                 return error;
4191
4192         sb->s_fs_info = sbinfo;
4193
4194 #ifdef CONFIG_TMPFS
4195         /*
4196          * Per default we only allow half of the physical ram per
4197          * tmpfs instance, limiting inodes to one per page of lowmem;
4198          * but the internal instance is left unlimited.
4199          */
4200         if (!(sb->s_flags & SB_KERNMOUNT)) {
4201                 if (!(ctx->seen & SHMEM_SEEN_BLOCKS))
4202                         ctx->blocks = shmem_default_max_blocks();
4203                 if (!(ctx->seen & SHMEM_SEEN_INODES))
4204                         ctx->inodes = shmem_default_max_inodes();
4205                 if (!(ctx->seen & SHMEM_SEEN_INUMS))
4206                         ctx->full_inums = IS_ENABLED(CONFIG_TMPFS_INODE64);
4207                 sbinfo->noswap = ctx->noswap;
4208         } else {
4209                 sb->s_flags |= SB_NOUSER;
4210         }
4211         sb->s_export_op = &shmem_export_ops;
4212         sb->s_flags |= SB_NOSEC | SB_I_VERSION;
4213 #else
4214         sb->s_flags |= SB_NOUSER;
4215 #endif
4216         sbinfo->max_blocks = ctx->blocks;
4217         sbinfo->max_inodes = ctx->inodes;
4218         sbinfo->free_ispace = sbinfo->max_inodes * BOGO_INODE_SIZE;
4219         if (sb->s_flags & SB_KERNMOUNT) {
4220                 sbinfo->ino_batch = alloc_percpu(ino_t);
4221                 if (!sbinfo->ino_batch)
4222                         goto failed;
4223         }
4224         sbinfo->uid = ctx->uid;
4225         sbinfo->gid = ctx->gid;
4226         sbinfo->full_inums = ctx->full_inums;
4227         sbinfo->mode = ctx->mode;
4228         sbinfo->huge = ctx->huge;
4229         sbinfo->mpol = ctx->mpol;
4230         ctx->mpol = NULL;
4231
4232         raw_spin_lock_init(&sbinfo->stat_lock);
4233         if (percpu_counter_init(&sbinfo->used_blocks, 0, GFP_KERNEL))
4234                 goto failed;
4235         spin_lock_init(&sbinfo->shrinklist_lock);
4236         INIT_LIST_HEAD(&sbinfo->shrinklist);
4237
4238         sb->s_maxbytes = MAX_LFS_FILESIZE;
4239         sb->s_blocksize = PAGE_SIZE;
4240         sb->s_blocksize_bits = PAGE_SHIFT;
4241         sb->s_magic = TMPFS_MAGIC;
4242         sb->s_op = &shmem_ops;
4243         sb->s_time_gran = 1;
4244 #ifdef CONFIG_TMPFS_XATTR
4245         sb->s_xattr = shmem_xattr_handlers;
4246 #endif
4247 #ifdef CONFIG_TMPFS_POSIX_ACL
4248         sb->s_flags |= SB_POSIXACL;
4249 #endif
4250         uuid_gen(&sb->s_uuid);
4251
4252 #ifdef CONFIG_TMPFS_QUOTA
4253         if (ctx->seen & SHMEM_SEEN_QUOTA) {
4254                 sb->dq_op = &shmem_quota_operations;
4255                 sb->s_qcop = &dquot_quotactl_sysfile_ops;
4256                 sb->s_quota_types = QTYPE_MASK_USR | QTYPE_MASK_GRP;
4257
4258                 /* Copy the default limits from ctx into sbinfo */
4259                 memcpy(&sbinfo->qlimits, &ctx->qlimits,
4260                        sizeof(struct shmem_quota_limits));
4261
4262                 if (shmem_enable_quotas(sb, ctx->quota_types))
4263                         goto failed;
4264         }
4265 #endif /* CONFIG_TMPFS_QUOTA */
4266
4267         inode = shmem_get_inode(&nop_mnt_idmap, sb, NULL, S_IFDIR | sbinfo->mode, 0,
4268                                 VM_NORESERVE);
4269         if (IS_ERR(inode)) {
4270                 error = PTR_ERR(inode);
4271                 goto failed;
4272         }
4273         inode->i_uid = sbinfo->uid;
4274         inode->i_gid = sbinfo->gid;
4275         sb->s_root = d_make_root(inode);
4276         if (!sb->s_root)
4277                 goto failed;
4278         return 0;
4279
4280 failed:
4281         shmem_put_super(sb);
4282         return error;
4283 }
4284
4285 static int shmem_get_tree(struct fs_context *fc)
4286 {
4287         return get_tree_nodev(fc, shmem_fill_super);
4288 }
4289
4290 static void shmem_free_fc(struct fs_context *fc)
4291 {
4292         struct shmem_options *ctx = fc->fs_private;
4293
4294         if (ctx) {
4295                 mpol_put(ctx->mpol);
4296                 kfree(ctx);
4297         }
4298 }
4299
4300 static const struct fs_context_operations shmem_fs_context_ops = {
4301         .free                   = shmem_free_fc,
4302         .get_tree               = shmem_get_tree,
4303 #ifdef CONFIG_TMPFS
4304         .parse_monolithic       = shmem_parse_options,
4305         .parse_param            = shmem_parse_one,
4306         .reconfigure            = shmem_reconfigure,
4307 #endif
4308 };
4309
4310 static struct kmem_cache *shmem_inode_cachep;
4311
4312 static struct inode *shmem_alloc_inode(struct super_block *sb)
4313 {
4314         struct shmem_inode_info *info;
4315         info = alloc_inode_sb(sb, shmem_inode_cachep, GFP_KERNEL);
4316         if (!info)
4317                 return NULL;
4318         return &info->vfs_inode;
4319 }
4320
4321 static void shmem_free_in_core_inode(struct inode *inode)
4322 {
4323         if (S_ISLNK(inode->i_mode))
4324                 kfree(inode->i_link);
4325         kmem_cache_free(shmem_inode_cachep, SHMEM_I(inode));
4326 }
4327
4328 static void shmem_destroy_inode(struct inode *inode)
4329 {
4330         if (S_ISREG(inode->i_mode))
4331                 mpol_free_shared_policy(&SHMEM_I(inode)->policy);
4332         if (S_ISDIR(inode->i_mode))
4333                 simple_offset_destroy(shmem_get_offset_ctx(inode));
4334 }
4335
4336 static void shmem_init_inode(void *foo)
4337 {
4338         struct shmem_inode_info *info = foo;
4339         inode_init_once(&info->vfs_inode);
4340 }
4341
4342 static void shmem_init_inodecache(void)
4343 {
4344         shmem_inode_cachep = kmem_cache_create("shmem_inode_cache",
4345                                 sizeof(struct shmem_inode_info),
4346                                 0, SLAB_PANIC|SLAB_ACCOUNT, shmem_init_inode);
4347 }
4348
4349 static void shmem_destroy_inodecache(void)
4350 {
4351         kmem_cache_destroy(shmem_inode_cachep);
4352 }
4353
4354 /* Keep the page in page cache instead of truncating it */
4355 static int shmem_error_remove_page(struct address_space *mapping,
4356                                    struct page *page)
4357 {
4358         return 0;
4359 }
4360
4361 const struct address_space_operations shmem_aops = {
4362         .writepage      = shmem_writepage,
4363         .dirty_folio    = noop_dirty_folio,
4364 #ifdef CONFIG_TMPFS
4365         .write_begin    = shmem_write_begin,
4366         .write_end      = shmem_write_end,
4367 #endif
4368 #ifdef CONFIG_MIGRATION
4369         .migrate_folio  = migrate_folio,
4370 #endif
4371         .error_remove_page = shmem_error_remove_page,
4372 };
4373 EXPORT_SYMBOL(shmem_aops);
4374
4375 static const struct file_operations shmem_file_operations = {
4376         .mmap           = shmem_mmap,
4377         .open           = generic_file_open,
4378         .get_unmapped_area = shmem_get_unmapped_area,
4379 #ifdef CONFIG_TMPFS
4380         .llseek         = shmem_file_llseek,
4381         .read_iter      = shmem_file_read_iter,
4382         .write_iter     = generic_file_write_iter,
4383         .fsync          = noop_fsync,
4384         .splice_read    = shmem_file_splice_read,
4385         .splice_write   = iter_file_splice_write,
4386         .fallocate      = shmem_fallocate,
4387 #endif
4388 };
4389
4390 static const struct inode_operations shmem_inode_operations = {
4391         .getattr        = shmem_getattr,
4392         .setattr        = shmem_setattr,
4393 #ifdef CONFIG_TMPFS_XATTR
4394         .listxattr      = shmem_listxattr,
4395         .set_acl        = simple_set_acl,
4396         .fileattr_get   = shmem_fileattr_get,
4397         .fileattr_set   = shmem_fileattr_set,
4398 #endif
4399 };
4400
4401 static const struct inode_operations shmem_dir_inode_operations = {
4402 #ifdef CONFIG_TMPFS
4403         .getattr        = shmem_getattr,
4404         .create         = shmem_create,
4405         .lookup         = simple_lookup,
4406         .link           = shmem_link,
4407         .unlink         = shmem_unlink,
4408         .symlink        = shmem_symlink,
4409         .mkdir          = shmem_mkdir,
4410         .rmdir          = shmem_rmdir,
4411         .mknod          = shmem_mknod,
4412         .rename         = shmem_rename2,
4413         .tmpfile        = shmem_tmpfile,
4414         .get_offset_ctx = shmem_get_offset_ctx,
4415 #endif
4416 #ifdef CONFIG_TMPFS_XATTR
4417         .listxattr      = shmem_listxattr,
4418         .fileattr_get   = shmem_fileattr_get,
4419         .fileattr_set   = shmem_fileattr_set,
4420 #endif
4421 #ifdef CONFIG_TMPFS_POSIX_ACL
4422         .setattr        = shmem_setattr,
4423         .set_acl        = simple_set_acl,
4424 #endif
4425 };
4426
4427 static const struct inode_operations shmem_special_inode_operations = {
4428         .getattr        = shmem_getattr,
4429 #ifdef CONFIG_TMPFS_XATTR
4430         .listxattr      = shmem_listxattr,
4431 #endif
4432 #ifdef CONFIG_TMPFS_POSIX_ACL
4433         .setattr        = shmem_setattr,
4434         .set_acl        = simple_set_acl,
4435 #endif
4436 };
4437
4438 static const struct super_operations shmem_ops = {
4439         .alloc_inode    = shmem_alloc_inode,
4440         .free_inode     = shmem_free_in_core_inode,
4441         .destroy_inode  = shmem_destroy_inode,
4442 #ifdef CONFIG_TMPFS
4443         .statfs         = shmem_statfs,
4444         .show_options   = shmem_show_options,
4445 #endif
4446 #ifdef CONFIG_TMPFS_QUOTA
4447         .get_dquots     = shmem_get_dquots,
4448 #endif
4449         .evict_inode    = shmem_evict_inode,
4450         .drop_inode     = generic_delete_inode,
4451         .put_super      = shmem_put_super,
4452 #ifdef CONFIG_TRANSPARENT_HUGEPAGE
4453         .nr_cached_objects      = shmem_unused_huge_count,
4454         .free_cached_objects    = shmem_unused_huge_scan,
4455 #endif
4456 };
4457
4458 static const struct vm_operations_struct shmem_vm_ops = {
4459         .fault          = shmem_fault,
4460         .map_pages      = filemap_map_pages,
4461 #ifdef CONFIG_NUMA
4462         .set_policy     = shmem_set_policy,
4463         .get_policy     = shmem_get_policy,
4464 #endif
4465 };
4466
4467 static const struct vm_operations_struct shmem_anon_vm_ops = {
4468         .fault          = shmem_fault,
4469         .map_pages      = filemap_map_pages,
4470 #ifdef CONFIG_NUMA
4471         .set_policy     = shmem_set_policy,
4472         .get_policy     = shmem_get_policy,
4473 #endif
4474 };
4475
4476 int shmem_init_fs_context(struct fs_context *fc)
4477 {
4478         struct shmem_options *ctx;
4479
4480         ctx = kzalloc(sizeof(struct shmem_options), GFP_KERNEL);
4481         if (!ctx)
4482                 return -ENOMEM;
4483
4484         ctx->mode = 0777 | S_ISVTX;
4485         ctx->uid = current_fsuid();
4486         ctx->gid = current_fsgid();
4487
4488         fc->fs_private = ctx;
4489         fc->ops = &shmem_fs_context_ops;
4490         return 0;
4491 }
4492
4493 static struct file_system_type shmem_fs_type = {
4494         .owner          = THIS_MODULE,
4495         .name           = "tmpfs",
4496         .init_fs_context = shmem_init_fs_context,
4497 #ifdef CONFIG_TMPFS
4498         .parameters     = shmem_fs_parameters,
4499 #endif
4500         .kill_sb        = kill_litter_super,
4501 #ifdef CONFIG_SHMEM
4502         .fs_flags       = FS_USERNS_MOUNT | FS_ALLOW_IDMAP,
4503 #else
4504         .fs_flags       = FS_USERNS_MOUNT,
4505 #endif
4506 };
4507
4508 void __init shmem_init(void)
4509 {
4510         int error;
4511
4512         shmem_init_inodecache();
4513
4514 #ifdef CONFIG_TMPFS_QUOTA
4515         error = register_quota_format(&shmem_quota_format);
4516         if (error < 0) {
4517                 pr_err("Could not register quota format\n");
4518                 goto out3;
4519         }
4520 #endif
4521
4522         error = register_filesystem(&shmem_fs_type);
4523         if (error) {
4524                 pr_err("Could not register tmpfs\n");
4525                 goto out2;
4526         }
4527
4528         shm_mnt = kern_mount(&shmem_fs_type);
4529         if (IS_ERR(shm_mnt)) {
4530                 error = PTR_ERR(shm_mnt);
4531                 pr_err("Could not kern_mount tmpfs\n");
4532                 goto out1;
4533         }
4534
4535 #ifdef CONFIG_TRANSPARENT_HUGEPAGE
4536         if (has_transparent_hugepage() && shmem_huge > SHMEM_HUGE_DENY)
4537                 SHMEM_SB(shm_mnt->mnt_sb)->huge = shmem_huge;
4538         else
4539                 shmem_huge = SHMEM_HUGE_NEVER; /* just in case it was patched */
4540 #endif
4541         return;
4542
4543 out1:
4544         unregister_filesystem(&shmem_fs_type);
4545 out2:
4546 #ifdef CONFIG_TMPFS_QUOTA
4547         unregister_quota_format(&shmem_quota_format);
4548 out3:
4549 #endif
4550         shmem_destroy_inodecache();
4551         shm_mnt = ERR_PTR(error);
4552 }
4553
4554 #if defined(CONFIG_TRANSPARENT_HUGEPAGE) && defined(CONFIG_SYSFS)
4555 static ssize_t shmem_enabled_show(struct kobject *kobj,
4556                                   struct kobj_attribute *attr, char *buf)
4557 {
4558         static const int values[] = {
4559                 SHMEM_HUGE_ALWAYS,
4560                 SHMEM_HUGE_WITHIN_SIZE,
4561                 SHMEM_HUGE_ADVISE,
4562                 SHMEM_HUGE_NEVER,
4563                 SHMEM_HUGE_DENY,
4564                 SHMEM_HUGE_FORCE,
4565         };
4566         int len = 0;
4567         int i;
4568
4569         for (i = 0; i < ARRAY_SIZE(values); i++) {
4570                 len += sysfs_emit_at(buf, len,
4571                                      shmem_huge == values[i] ? "%s[%s]" : "%s%s",
4572                                      i ? " " : "",
4573                                      shmem_format_huge(values[i]));
4574         }
4575
4576         len += sysfs_emit_at(buf, len, "\n");
4577
4578         return len;
4579 }
4580
4581 static ssize_t shmem_enabled_store(struct kobject *kobj,
4582                 struct kobj_attribute *attr, const char *buf, size_t count)
4583 {
4584         char tmp[16];
4585         int huge;
4586
4587         if (count + 1 > sizeof(tmp))
4588                 return -EINVAL;
4589         memcpy(tmp, buf, count);
4590         tmp[count] = '\0';
4591         if (count && tmp[count - 1] == '\n')
4592                 tmp[count - 1] = '\0';
4593
4594         huge = shmem_parse_huge(tmp);
4595         if (huge == -EINVAL)
4596                 return -EINVAL;
4597         if (!has_transparent_hugepage() &&
4598                         huge != SHMEM_HUGE_NEVER && huge != SHMEM_HUGE_DENY)
4599                 return -EINVAL;
4600
4601         shmem_huge = huge;
4602         if (shmem_huge > SHMEM_HUGE_DENY)
4603                 SHMEM_SB(shm_mnt->mnt_sb)->huge = shmem_huge;
4604         return count;
4605 }
4606
4607 struct kobj_attribute shmem_enabled_attr = __ATTR_RW(shmem_enabled);
4608 #endif /* CONFIG_TRANSPARENT_HUGEPAGE && CONFIG_SYSFS */
4609
4610 #else /* !CONFIG_SHMEM */
4611
4612 /*
4613  * tiny-shmem: simple shmemfs and tmpfs using ramfs code
4614  *
4615  * This is intended for small system where the benefits of the full
4616  * shmem code (swap-backed and resource-limited) are outweighed by
4617  * their complexity. On systems without swap this code should be
4618  * effectively equivalent, but much lighter weight.
4619  */
4620
4621 static struct file_system_type shmem_fs_type = {
4622         .name           = "tmpfs",
4623         .init_fs_context = ramfs_init_fs_context,
4624         .parameters     = ramfs_fs_parameters,
4625         .kill_sb        = ramfs_kill_sb,
4626         .fs_flags       = FS_USERNS_MOUNT,
4627 };
4628
4629 void __init shmem_init(void)
4630 {
4631         BUG_ON(register_filesystem(&shmem_fs_type) != 0);
4632
4633         shm_mnt = kern_mount(&shmem_fs_type);
4634         BUG_ON(IS_ERR(shm_mnt));
4635 }
4636
4637 int shmem_unuse(unsigned int type)
4638 {
4639         return 0;
4640 }
4641
4642 int shmem_lock(struct file *file, int lock, struct ucounts *ucounts)
4643 {
4644         return 0;
4645 }
4646
4647 void shmem_unlock_mapping(struct address_space *mapping)
4648 {
4649 }
4650
4651 #ifdef CONFIG_MMU
4652 unsigned long shmem_get_unmapped_area(struct file *file,
4653                                       unsigned long addr, unsigned long len,
4654                                       unsigned long pgoff, unsigned long flags)
4655 {
4656         return current->mm->get_unmapped_area(file, addr, len, pgoff, flags);
4657 }
4658 #endif
4659
4660 void shmem_truncate_range(struct inode *inode, loff_t lstart, loff_t lend)
4661 {
4662         truncate_inode_pages_range(inode->i_mapping, lstart, lend);
4663 }
4664 EXPORT_SYMBOL_GPL(shmem_truncate_range);
4665
4666 #define shmem_vm_ops                            generic_file_vm_ops
4667 #define shmem_anon_vm_ops                       generic_file_vm_ops
4668 #define shmem_file_operations                   ramfs_file_operations
4669 #define shmem_acct_size(flags, size)            0
4670 #define shmem_unacct_size(flags, size)          do {} while (0)
4671
4672 static inline struct inode *shmem_get_inode(struct mnt_idmap *idmap, struct super_block *sb, struct inode *dir,
4673                                             umode_t mode, dev_t dev, unsigned long flags)
4674 {
4675         struct inode *inode = ramfs_get_inode(sb, dir, mode, dev);
4676         return inode ? inode : ERR_PTR(-ENOSPC);
4677 }
4678
4679 #endif /* CONFIG_SHMEM */
4680
4681 /* common code */
4682
4683 static struct file *__shmem_file_setup(struct vfsmount *mnt, const char *name, loff_t size,
4684                                        unsigned long flags, unsigned int i_flags)
4685 {
4686         struct inode *inode;
4687         struct file *res;
4688
4689         if (IS_ERR(mnt))
4690                 return ERR_CAST(mnt);
4691
4692         if (size < 0 || size > MAX_LFS_FILESIZE)
4693                 return ERR_PTR(-EINVAL);
4694
4695         if (shmem_acct_size(flags, size))
4696                 return ERR_PTR(-ENOMEM);
4697
4698         if (is_idmapped_mnt(mnt))
4699                 return ERR_PTR(-EINVAL);
4700
4701         inode = shmem_get_inode(&nop_mnt_idmap, mnt->mnt_sb, NULL,
4702                                 S_IFREG | S_IRWXUGO, 0, flags);
4703
4704         if (IS_ERR(inode)) {
4705                 shmem_unacct_size(flags, size);
4706                 return ERR_CAST(inode);
4707         }
4708         inode->i_flags |= i_flags;
4709         inode->i_size = size;
4710         clear_nlink(inode);     /* It is unlinked */
4711         res = ERR_PTR(ramfs_nommu_expand_for_mapping(inode, size));
4712         if (!IS_ERR(res))
4713                 res = alloc_file_pseudo(inode, mnt, name, O_RDWR,
4714                                 &shmem_file_operations);
4715         if (IS_ERR(res))
4716                 iput(inode);
4717         return res;
4718 }
4719
4720 /**
4721  * shmem_kernel_file_setup - get an unlinked file living in tmpfs which must be
4722  *      kernel internal.  There will be NO LSM permission checks against the
4723  *      underlying inode.  So users of this interface must do LSM checks at a
4724  *      higher layer.  The users are the big_key and shm implementations.  LSM
4725  *      checks are provided at the key or shm level rather than the inode.
4726  * @name: name for dentry (to be seen in /proc/<pid>/maps
4727  * @size: size to be set for the file
4728  * @flags: VM_NORESERVE suppresses pre-accounting of the entire object size
4729  */
4730 struct file *shmem_kernel_file_setup(const char *name, loff_t size, unsigned long flags)
4731 {
4732         return __shmem_file_setup(shm_mnt, name, size, flags, S_PRIVATE);
4733 }
4734
4735 /**
4736  * shmem_file_setup - get an unlinked file living in tmpfs
4737  * @name: name for dentry (to be seen in /proc/<pid>/maps
4738  * @size: size to be set for the file
4739  * @flags: VM_NORESERVE suppresses pre-accounting of the entire object size
4740  */
4741 struct file *shmem_file_setup(const char *name, loff_t size, unsigned long flags)
4742 {
4743         return __shmem_file_setup(shm_mnt, name, size, flags, 0);
4744 }
4745 EXPORT_SYMBOL_GPL(shmem_file_setup);
4746
4747 /**
4748  * shmem_file_setup_with_mnt - get an unlinked file living in tmpfs
4749  * @mnt: the tmpfs mount where the file will be created
4750  * @name: name for dentry (to be seen in /proc/<pid>/maps
4751  * @size: size to be set for the file
4752  * @flags: VM_NORESERVE suppresses pre-accounting of the entire object size
4753  */
4754 struct file *shmem_file_setup_with_mnt(struct vfsmount *mnt, const char *name,
4755                                        loff_t size, unsigned long flags)
4756 {
4757         return __shmem_file_setup(mnt, name, size, flags, 0);
4758 }
4759 EXPORT_SYMBOL_GPL(shmem_file_setup_with_mnt);
4760
4761 /**
4762  * shmem_zero_setup - setup a shared anonymous mapping
4763  * @vma: the vma to be mmapped is prepared by do_mmap
4764  */
4765 int shmem_zero_setup(struct vm_area_struct *vma)
4766 {
4767         struct file *file;
4768         loff_t size = vma->vm_end - vma->vm_start;
4769
4770         /*
4771          * Cloning a new file under mmap_lock leads to a lock ordering conflict
4772          * between XFS directory reading and selinux: since this file is only
4773          * accessible to the user through its mapping, use S_PRIVATE flag to
4774          * bypass file security, in the same way as shmem_kernel_file_setup().
4775          */
4776         file = shmem_kernel_file_setup("dev/zero", size, vma->vm_flags);
4777         if (IS_ERR(file))
4778                 return PTR_ERR(file);
4779
4780         if (vma->vm_file)
4781                 fput(vma->vm_file);
4782         vma->vm_file = file;
4783         vma->vm_ops = &shmem_anon_vm_ops;
4784
4785         return 0;
4786 }
4787
4788 /**
4789  * shmem_read_folio_gfp - read into page cache, using specified page allocation flags.
4790  * @mapping:    the folio's address_space
4791  * @index:      the folio index
4792  * @gfp:        the page allocator flags to use if allocating
4793  *
4794  * This behaves as a tmpfs "read_cache_page_gfp(mapping, index, gfp)",
4795  * with any new page allocations done using the specified allocation flags.
4796  * But read_cache_page_gfp() uses the ->read_folio() method: which does not
4797  * suit tmpfs, since it may have pages in swapcache, and needs to find those
4798  * for itself; although drivers/gpu/drm i915 and ttm rely upon this support.
4799  *
4800  * i915_gem_object_get_pages_gtt() mixes __GFP_NORETRY | __GFP_NOWARN in
4801  * with the mapping_gfp_mask(), to avoid OOMing the machine unnecessarily.
4802  */
4803 struct folio *shmem_read_folio_gfp(struct address_space *mapping,
4804                 pgoff_t index, gfp_t gfp)
4805 {
4806 #ifdef CONFIG_SHMEM
4807         struct inode *inode = mapping->host;
4808         struct folio *folio;
4809         int error;
4810
4811         BUG_ON(!shmem_mapping(mapping));
4812         error = shmem_get_folio_gfp(inode, index, &folio, SGP_CACHE,
4813                                   gfp, NULL, NULL, NULL);
4814         if (error)
4815                 return ERR_PTR(error);
4816
4817         folio_unlock(folio);
4818         return folio;
4819 #else
4820         /*
4821          * The tiny !SHMEM case uses ramfs without swap
4822          */
4823         return mapping_read_folio_gfp(mapping, index, gfp);
4824 #endif
4825 }
4826 EXPORT_SYMBOL_GPL(shmem_read_folio_gfp);
4827
4828 struct page *shmem_read_mapping_page_gfp(struct address_space *mapping,
4829                                          pgoff_t index, gfp_t gfp)
4830 {
4831         struct folio *folio = shmem_read_folio_gfp(mapping, index, gfp);
4832         struct page *page;
4833
4834         if (IS_ERR(folio))
4835                 return &folio->page;
4836
4837         page = folio_file_page(folio, index);
4838         if (PageHWPoison(page)) {
4839                 folio_put(folio);
4840                 return ERR_PTR(-EIO);
4841         }
4842
4843         return page;
4844 }
4845 EXPORT_SYMBOL_GPL(shmem_read_mapping_page_gfp);