tmpfs,xattr: enable limited user extended attributes
[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, size_t freed_ispace)
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 + freed_ispace;
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         size_t freed = 0;
1236
1237         if (shmem_mapping(inode->i_mapping)) {
1238                 shmem_unacct_size(info->flags, inode->i_size);
1239                 inode->i_size = 0;
1240                 mapping_set_exiting(inode->i_mapping);
1241                 shmem_truncate_range(inode, 0, (loff_t)-1);
1242                 if (!list_empty(&info->shrinklist)) {
1243                         spin_lock(&sbinfo->shrinklist_lock);
1244                         if (!list_empty(&info->shrinklist)) {
1245                                 list_del_init(&info->shrinklist);
1246                                 sbinfo->shrinklist_len--;
1247                         }
1248                         spin_unlock(&sbinfo->shrinklist_lock);
1249                 }
1250                 while (!list_empty(&info->swaplist)) {
1251                         /* Wait while shmem_unuse() is scanning this inode... */
1252                         wait_var_event(&info->stop_eviction,
1253                                        !atomic_read(&info->stop_eviction));
1254                         mutex_lock(&shmem_swaplist_mutex);
1255                         /* ...but beware of the race if we peeked too early */
1256                         if (!atomic_read(&info->stop_eviction))
1257                                 list_del_init(&info->swaplist);
1258                         mutex_unlock(&shmem_swaplist_mutex);
1259                 }
1260         }
1261
1262         simple_xattrs_free(&info->xattrs, sbinfo->max_inodes ? &freed : NULL);
1263         shmem_free_inode(inode->i_sb, freed);
1264         WARN_ON(inode->i_blocks);
1265         clear_inode(inode);
1266 #ifdef CONFIG_TMPFS_QUOTA
1267         dquot_free_inode(inode);
1268         dquot_drop(inode);
1269 #endif
1270 }
1271
1272 static int shmem_find_swap_entries(struct address_space *mapping,
1273                                    pgoff_t start, struct folio_batch *fbatch,
1274                                    pgoff_t *indices, unsigned int type)
1275 {
1276         XA_STATE(xas, &mapping->i_pages, start);
1277         struct folio *folio;
1278         swp_entry_t entry;
1279
1280         rcu_read_lock();
1281         xas_for_each(&xas, folio, ULONG_MAX) {
1282                 if (xas_retry(&xas, folio))
1283                         continue;
1284
1285                 if (!xa_is_value(folio))
1286                         continue;
1287
1288                 entry = radix_to_swp_entry(folio);
1289                 /*
1290                  * swapin error entries can be found in the mapping. But they're
1291                  * deliberately ignored here as we've done everything we can do.
1292                  */
1293                 if (swp_type(entry) != type)
1294                         continue;
1295
1296                 indices[folio_batch_count(fbatch)] = xas.xa_index;
1297                 if (!folio_batch_add(fbatch, folio))
1298                         break;
1299
1300                 if (need_resched()) {
1301                         xas_pause(&xas);
1302                         cond_resched_rcu();
1303                 }
1304         }
1305         rcu_read_unlock();
1306
1307         return xas.xa_index;
1308 }
1309
1310 /*
1311  * Move the swapped pages for an inode to page cache. Returns the count
1312  * of pages swapped in, or the error in case of failure.
1313  */
1314 static int shmem_unuse_swap_entries(struct inode *inode,
1315                 struct folio_batch *fbatch, pgoff_t *indices)
1316 {
1317         int i = 0;
1318         int ret = 0;
1319         int error = 0;
1320         struct address_space *mapping = inode->i_mapping;
1321
1322         for (i = 0; i < folio_batch_count(fbatch); i++) {
1323                 struct folio *folio = fbatch->folios[i];
1324
1325                 if (!xa_is_value(folio))
1326                         continue;
1327                 error = shmem_swapin_folio(inode, indices[i],
1328                                           &folio, SGP_CACHE,
1329                                           mapping_gfp_mask(mapping),
1330                                           NULL, NULL);
1331                 if (error == 0) {
1332                         folio_unlock(folio);
1333                         folio_put(folio);
1334                         ret++;
1335                 }
1336                 if (error == -ENOMEM)
1337                         break;
1338                 error = 0;
1339         }
1340         return error ? error : ret;
1341 }
1342
1343 /*
1344  * If swap found in inode, free it and move page from swapcache to filecache.
1345  */
1346 static int shmem_unuse_inode(struct inode *inode, unsigned int type)
1347 {
1348         struct address_space *mapping = inode->i_mapping;
1349         pgoff_t start = 0;
1350         struct folio_batch fbatch;
1351         pgoff_t indices[PAGEVEC_SIZE];
1352         int ret = 0;
1353
1354         do {
1355                 folio_batch_init(&fbatch);
1356                 shmem_find_swap_entries(mapping, start, &fbatch, indices, type);
1357                 if (folio_batch_count(&fbatch) == 0) {
1358                         ret = 0;
1359                         break;
1360                 }
1361
1362                 ret = shmem_unuse_swap_entries(inode, &fbatch, indices);
1363                 if (ret < 0)
1364                         break;
1365
1366                 start = indices[folio_batch_count(&fbatch) - 1];
1367         } while (true);
1368
1369         return ret;
1370 }
1371
1372 /*
1373  * Read all the shared memory data that resides in the swap
1374  * device 'type' back into memory, so the swap device can be
1375  * unused.
1376  */
1377 int shmem_unuse(unsigned int type)
1378 {
1379         struct shmem_inode_info *info, *next;
1380         int error = 0;
1381
1382         if (list_empty(&shmem_swaplist))
1383                 return 0;
1384
1385         mutex_lock(&shmem_swaplist_mutex);
1386         list_for_each_entry_safe(info, next, &shmem_swaplist, swaplist) {
1387                 if (!info->swapped) {
1388                         list_del_init(&info->swaplist);
1389                         continue;
1390                 }
1391                 /*
1392                  * Drop the swaplist mutex while searching the inode for swap;
1393                  * but before doing so, make sure shmem_evict_inode() will not
1394                  * remove placeholder inode from swaplist, nor let it be freed
1395                  * (igrab() would protect from unlink, but not from unmount).
1396                  */
1397                 atomic_inc(&info->stop_eviction);
1398                 mutex_unlock(&shmem_swaplist_mutex);
1399
1400                 error = shmem_unuse_inode(&info->vfs_inode, type);
1401                 cond_resched();
1402
1403                 mutex_lock(&shmem_swaplist_mutex);
1404                 next = list_next_entry(info, swaplist);
1405                 if (!info->swapped)
1406                         list_del_init(&info->swaplist);
1407                 if (atomic_dec_and_test(&info->stop_eviction))
1408                         wake_up_var(&info->stop_eviction);
1409                 if (error)
1410                         break;
1411         }
1412         mutex_unlock(&shmem_swaplist_mutex);
1413
1414         return error;
1415 }
1416
1417 /*
1418  * Move the page from the page cache to the swap cache.
1419  */
1420 static int shmem_writepage(struct page *page, struct writeback_control *wbc)
1421 {
1422         struct folio *folio = page_folio(page);
1423         struct address_space *mapping = folio->mapping;
1424         struct inode *inode = mapping->host;
1425         struct shmem_inode_info *info = SHMEM_I(inode);
1426         struct shmem_sb_info *sbinfo = SHMEM_SB(inode->i_sb);
1427         swp_entry_t swap;
1428         pgoff_t index;
1429
1430         /*
1431          * Our capabilities prevent regular writeback or sync from ever calling
1432          * shmem_writepage; but a stacking filesystem might use ->writepage of
1433          * its underlying filesystem, in which case tmpfs should write out to
1434          * swap only in response to memory pressure, and not for the writeback
1435          * threads or sync.
1436          */
1437         if (WARN_ON_ONCE(!wbc->for_reclaim))
1438                 goto redirty;
1439
1440         if (WARN_ON_ONCE((info->flags & VM_LOCKED) || sbinfo->noswap))
1441                 goto redirty;
1442
1443         if (!total_swap_pages)
1444                 goto redirty;
1445
1446         /*
1447          * If /sys/kernel/mm/transparent_hugepage/shmem_enabled is "always" or
1448          * "force", drivers/gpu/drm/i915/gem/i915_gem_shmem.c gets huge pages,
1449          * and its shmem_writeback() needs them to be split when swapping.
1450          */
1451         if (folio_test_large(folio)) {
1452                 /* Ensure the subpages are still dirty */
1453                 folio_test_set_dirty(folio);
1454                 if (split_huge_page(page) < 0)
1455                         goto redirty;
1456                 folio = page_folio(page);
1457                 folio_clear_dirty(folio);
1458         }
1459
1460         index = folio->index;
1461
1462         /*
1463          * This is somewhat ridiculous, but without plumbing a SWAP_MAP_FALLOC
1464          * value into swapfile.c, the only way we can correctly account for a
1465          * fallocated folio arriving here is now to initialize it and write it.
1466          *
1467          * That's okay for a folio already fallocated earlier, but if we have
1468          * not yet completed the fallocation, then (a) we want to keep track
1469          * of this folio in case we have to undo it, and (b) it may not be a
1470          * good idea to continue anyway, once we're pushing into swap.  So
1471          * reactivate the folio, and let shmem_fallocate() quit when too many.
1472          */
1473         if (!folio_test_uptodate(folio)) {
1474                 if (inode->i_private) {
1475                         struct shmem_falloc *shmem_falloc;
1476                         spin_lock(&inode->i_lock);
1477                         shmem_falloc = inode->i_private;
1478                         if (shmem_falloc &&
1479                             !shmem_falloc->waitq &&
1480                             index >= shmem_falloc->start &&
1481                             index < shmem_falloc->next)
1482                                 shmem_falloc->nr_unswapped++;
1483                         else
1484                                 shmem_falloc = NULL;
1485                         spin_unlock(&inode->i_lock);
1486                         if (shmem_falloc)
1487                                 goto redirty;
1488                 }
1489                 folio_zero_range(folio, 0, folio_size(folio));
1490                 flush_dcache_folio(folio);
1491                 folio_mark_uptodate(folio);
1492         }
1493
1494         swap = folio_alloc_swap(folio);
1495         if (!swap.val)
1496                 goto redirty;
1497
1498         /*
1499          * Add inode to shmem_unuse()'s list of swapped-out inodes,
1500          * if it's not already there.  Do it now before the folio is
1501          * moved to swap cache, when its pagelock no longer protects
1502          * the inode from eviction.  But don't unlock the mutex until
1503          * we've incremented swapped, because shmem_unuse_inode() will
1504          * prune a !swapped inode from the swaplist under this mutex.
1505          */
1506         mutex_lock(&shmem_swaplist_mutex);
1507         if (list_empty(&info->swaplist))
1508                 list_add(&info->swaplist, &shmem_swaplist);
1509
1510         if (add_to_swap_cache(folio, swap,
1511                         __GFP_HIGH | __GFP_NOMEMALLOC | __GFP_NOWARN,
1512                         NULL) == 0) {
1513                 shmem_recalc_inode(inode, 0, 1);
1514                 swap_shmem_alloc(swap);
1515                 shmem_delete_from_page_cache(folio, swp_to_radix_entry(swap));
1516
1517                 mutex_unlock(&shmem_swaplist_mutex);
1518                 BUG_ON(folio_mapped(folio));
1519                 swap_writepage(&folio->page, wbc);
1520                 return 0;
1521         }
1522
1523         mutex_unlock(&shmem_swaplist_mutex);
1524         put_swap_folio(folio, swap);
1525 redirty:
1526         folio_mark_dirty(folio);
1527         if (wbc->for_reclaim)
1528                 return AOP_WRITEPAGE_ACTIVATE;  /* Return with folio locked */
1529         folio_unlock(folio);
1530         return 0;
1531 }
1532
1533 #if defined(CONFIG_NUMA) && defined(CONFIG_TMPFS)
1534 static void shmem_show_mpol(struct seq_file *seq, struct mempolicy *mpol)
1535 {
1536         char buffer[64];
1537
1538         if (!mpol || mpol->mode == MPOL_DEFAULT)
1539                 return;         /* show nothing */
1540
1541         mpol_to_str(buffer, sizeof(buffer), mpol);
1542
1543         seq_printf(seq, ",mpol=%s", buffer);
1544 }
1545
1546 static struct mempolicy *shmem_get_sbmpol(struct shmem_sb_info *sbinfo)
1547 {
1548         struct mempolicy *mpol = NULL;
1549         if (sbinfo->mpol) {
1550                 raw_spin_lock(&sbinfo->stat_lock);      /* prevent replace/use races */
1551                 mpol = sbinfo->mpol;
1552                 mpol_get(mpol);
1553                 raw_spin_unlock(&sbinfo->stat_lock);
1554         }
1555         return mpol;
1556 }
1557 #else /* !CONFIG_NUMA || !CONFIG_TMPFS */
1558 static inline void shmem_show_mpol(struct seq_file *seq, struct mempolicy *mpol)
1559 {
1560 }
1561 static inline struct mempolicy *shmem_get_sbmpol(struct shmem_sb_info *sbinfo)
1562 {
1563         return NULL;
1564 }
1565 #endif /* CONFIG_NUMA && CONFIG_TMPFS */
1566 #ifndef CONFIG_NUMA
1567 #define vm_policy vm_private_data
1568 #endif
1569
1570 static void shmem_pseudo_vma_init(struct vm_area_struct *vma,
1571                 struct shmem_inode_info *info, pgoff_t index)
1572 {
1573         /* Create a pseudo vma that just contains the policy */
1574         vma_init(vma, NULL);
1575         /* Bias interleave by inode number to distribute better across nodes */
1576         vma->vm_pgoff = index + info->vfs_inode.i_ino;
1577         vma->vm_policy = mpol_shared_policy_lookup(&info->policy, index);
1578 }
1579
1580 static void shmem_pseudo_vma_destroy(struct vm_area_struct *vma)
1581 {
1582         /* Drop reference taken by mpol_shared_policy_lookup() */
1583         mpol_cond_put(vma->vm_policy);
1584 }
1585
1586 static struct folio *shmem_swapin(swp_entry_t swap, gfp_t gfp,
1587                         struct shmem_inode_info *info, pgoff_t index)
1588 {
1589         struct vm_area_struct pvma;
1590         struct page *page;
1591         struct vm_fault vmf = {
1592                 .vma = &pvma,
1593         };
1594
1595         shmem_pseudo_vma_init(&pvma, info, index);
1596         page = swap_cluster_readahead(swap, gfp, &vmf);
1597         shmem_pseudo_vma_destroy(&pvma);
1598
1599         if (!page)
1600                 return NULL;
1601         return page_folio(page);
1602 }
1603
1604 /*
1605  * Make sure huge_gfp is always more limited than limit_gfp.
1606  * Some of the flags set permissions, while others set limitations.
1607  */
1608 static gfp_t limit_gfp_mask(gfp_t huge_gfp, gfp_t limit_gfp)
1609 {
1610         gfp_t allowflags = __GFP_IO | __GFP_FS | __GFP_RECLAIM;
1611         gfp_t denyflags = __GFP_NOWARN | __GFP_NORETRY;
1612         gfp_t zoneflags = limit_gfp & GFP_ZONEMASK;
1613         gfp_t result = huge_gfp & ~(allowflags | GFP_ZONEMASK);
1614
1615         /* Allow allocations only from the originally specified zones. */
1616         result |= zoneflags;
1617
1618         /*
1619          * Minimize the result gfp by taking the union with the deny flags,
1620          * and the intersection of the allow flags.
1621          */
1622         result |= (limit_gfp & denyflags);
1623         result |= (huge_gfp & limit_gfp) & allowflags;
1624
1625         return result;
1626 }
1627
1628 static struct folio *shmem_alloc_hugefolio(gfp_t gfp,
1629                 struct shmem_inode_info *info, pgoff_t index)
1630 {
1631         struct vm_area_struct pvma;
1632         struct address_space *mapping = info->vfs_inode.i_mapping;
1633         pgoff_t hindex;
1634         struct folio *folio;
1635
1636         hindex = round_down(index, HPAGE_PMD_NR);
1637         if (xa_find(&mapping->i_pages, &hindex, hindex + HPAGE_PMD_NR - 1,
1638                                                                 XA_PRESENT))
1639                 return NULL;
1640
1641         shmem_pseudo_vma_init(&pvma, info, hindex);
1642         folio = vma_alloc_folio(gfp, HPAGE_PMD_ORDER, &pvma, 0, true);
1643         shmem_pseudo_vma_destroy(&pvma);
1644         if (!folio)
1645                 count_vm_event(THP_FILE_FALLBACK);
1646         return folio;
1647 }
1648
1649 static struct folio *shmem_alloc_folio(gfp_t gfp,
1650                         struct shmem_inode_info *info, pgoff_t index)
1651 {
1652         struct vm_area_struct pvma;
1653         struct folio *folio;
1654
1655         shmem_pseudo_vma_init(&pvma, info, index);
1656         folio = vma_alloc_folio(gfp, 0, &pvma, 0, false);
1657         shmem_pseudo_vma_destroy(&pvma);
1658
1659         return folio;
1660 }
1661
1662 static struct folio *shmem_alloc_and_acct_folio(gfp_t gfp, struct inode *inode,
1663                 pgoff_t index, bool huge)
1664 {
1665         struct shmem_inode_info *info = SHMEM_I(inode);
1666         struct folio *folio;
1667         int nr;
1668         int err;
1669
1670         if (!IS_ENABLED(CONFIG_TRANSPARENT_HUGEPAGE))
1671                 huge = false;
1672         nr = huge ? HPAGE_PMD_NR : 1;
1673
1674         err = shmem_inode_acct_block(inode, nr);
1675         if (err)
1676                 goto failed;
1677
1678         if (huge)
1679                 folio = shmem_alloc_hugefolio(gfp, info, index);
1680         else
1681                 folio = shmem_alloc_folio(gfp, info, index);
1682         if (folio) {
1683                 __folio_set_locked(folio);
1684                 __folio_set_swapbacked(folio);
1685                 return folio;
1686         }
1687
1688         err = -ENOMEM;
1689         shmem_inode_unacct_blocks(inode, nr);
1690 failed:
1691         return ERR_PTR(err);
1692 }
1693
1694 /*
1695  * When a page is moved from swapcache to shmem filecache (either by the
1696  * usual swapin of shmem_get_folio_gfp(), or by the less common swapoff of
1697  * shmem_unuse_inode()), it may have been read in earlier from swap, in
1698  * ignorance of the mapping it belongs to.  If that mapping has special
1699  * constraints (like the gma500 GEM driver, which requires RAM below 4GB),
1700  * we may need to copy to a suitable page before moving to filecache.
1701  *
1702  * In a future release, this may well be extended to respect cpuset and
1703  * NUMA mempolicy, and applied also to anonymous pages in do_swap_page();
1704  * but for now it is a simple matter of zone.
1705  */
1706 static bool shmem_should_replace_folio(struct folio *folio, gfp_t gfp)
1707 {
1708         return folio_zonenum(folio) > gfp_zone(gfp);
1709 }
1710
1711 static int shmem_replace_folio(struct folio **foliop, gfp_t gfp,
1712                                 struct shmem_inode_info *info, pgoff_t index)
1713 {
1714         struct folio *old, *new;
1715         struct address_space *swap_mapping;
1716         swp_entry_t entry;
1717         pgoff_t swap_index;
1718         int error;
1719
1720         old = *foliop;
1721         entry = folio_swap_entry(old);
1722         swap_index = swp_offset(entry);
1723         swap_mapping = swap_address_space(entry);
1724
1725         /*
1726          * We have arrived here because our zones are constrained, so don't
1727          * limit chance of success by further cpuset and node constraints.
1728          */
1729         gfp &= ~GFP_CONSTRAINT_MASK;
1730         VM_BUG_ON_FOLIO(folio_test_large(old), old);
1731         new = shmem_alloc_folio(gfp, info, index);
1732         if (!new)
1733                 return -ENOMEM;
1734
1735         folio_get(new);
1736         folio_copy(new, old);
1737         flush_dcache_folio(new);
1738
1739         __folio_set_locked(new);
1740         __folio_set_swapbacked(new);
1741         folio_mark_uptodate(new);
1742         folio_set_swap_entry(new, entry);
1743         folio_set_swapcache(new);
1744
1745         /*
1746          * Our caller will very soon move newpage out of swapcache, but it's
1747          * a nice clean interface for us to replace oldpage by newpage there.
1748          */
1749         xa_lock_irq(&swap_mapping->i_pages);
1750         error = shmem_replace_entry(swap_mapping, swap_index, old, new);
1751         if (!error) {
1752                 mem_cgroup_migrate(old, new);
1753                 __lruvec_stat_mod_folio(new, NR_FILE_PAGES, 1);
1754                 __lruvec_stat_mod_folio(new, NR_SHMEM, 1);
1755                 __lruvec_stat_mod_folio(old, NR_FILE_PAGES, -1);
1756                 __lruvec_stat_mod_folio(old, NR_SHMEM, -1);
1757         }
1758         xa_unlock_irq(&swap_mapping->i_pages);
1759
1760         if (unlikely(error)) {
1761                 /*
1762                  * Is this possible?  I think not, now that our callers check
1763                  * both PageSwapCache and page_private after getting page lock;
1764                  * but be defensive.  Reverse old to newpage for clear and free.
1765                  */
1766                 old = new;
1767         } else {
1768                 folio_add_lru(new);
1769                 *foliop = new;
1770         }
1771
1772         folio_clear_swapcache(old);
1773         old->private = NULL;
1774
1775         folio_unlock(old);
1776         folio_put_refs(old, 2);
1777         return error;
1778 }
1779
1780 static void shmem_set_folio_swapin_error(struct inode *inode, pgoff_t index,
1781                                          struct folio *folio, swp_entry_t swap)
1782 {
1783         struct address_space *mapping = inode->i_mapping;
1784         swp_entry_t swapin_error;
1785         void *old;
1786
1787         swapin_error = make_swapin_error_entry();
1788         old = xa_cmpxchg_irq(&mapping->i_pages, index,
1789                              swp_to_radix_entry(swap),
1790                              swp_to_radix_entry(swapin_error), 0);
1791         if (old != swp_to_radix_entry(swap))
1792                 return;
1793
1794         folio_wait_writeback(folio);
1795         delete_from_swap_cache(folio);
1796         /*
1797          * Don't treat swapin error folio as alloced. Otherwise inode->i_blocks
1798          * won't be 0 when inode is released and thus trigger WARN_ON(i_blocks)
1799          * in shmem_evict_inode().
1800          */
1801         shmem_recalc_inode(inode, -1, -1);
1802         swap_free(swap);
1803 }
1804
1805 /*
1806  * Swap in the folio pointed to by *foliop.
1807  * Caller has to make sure that *foliop contains a valid swapped folio.
1808  * Returns 0 and the folio in foliop if success. On failure, returns the
1809  * error code and NULL in *foliop.
1810  */
1811 static int shmem_swapin_folio(struct inode *inode, pgoff_t index,
1812                              struct folio **foliop, enum sgp_type sgp,
1813                              gfp_t gfp, struct vm_area_struct *vma,
1814                              vm_fault_t *fault_type)
1815 {
1816         struct address_space *mapping = inode->i_mapping;
1817         struct shmem_inode_info *info = SHMEM_I(inode);
1818         struct mm_struct *charge_mm = vma ? vma->vm_mm : NULL;
1819         struct swap_info_struct *si;
1820         struct folio *folio = NULL;
1821         swp_entry_t swap;
1822         int error;
1823
1824         VM_BUG_ON(!*foliop || !xa_is_value(*foliop));
1825         swap = radix_to_swp_entry(*foliop);
1826         *foliop = NULL;
1827
1828         if (is_swapin_error_entry(swap))
1829                 return -EIO;
1830
1831         si = get_swap_device(swap);
1832         if (!si) {
1833                 if (!shmem_confirm_swap(mapping, index, swap))
1834                         return -EEXIST;
1835                 else
1836                         return -EINVAL;
1837         }
1838
1839         /* Look it up and read it in.. */
1840         folio = swap_cache_get_folio(swap, NULL, 0);
1841         if (!folio) {
1842                 /* Or update major stats only when swapin succeeds?? */
1843                 if (fault_type) {
1844                         *fault_type |= VM_FAULT_MAJOR;
1845                         count_vm_event(PGMAJFAULT);
1846                         count_memcg_event_mm(charge_mm, PGMAJFAULT);
1847                 }
1848                 /* Here we actually start the io */
1849                 folio = shmem_swapin(swap, gfp, info, index);
1850                 if (!folio) {
1851                         error = -ENOMEM;
1852                         goto failed;
1853                 }
1854         }
1855
1856         /* We have to do this with folio locked to prevent races */
1857         folio_lock(folio);
1858         if (!folio_test_swapcache(folio) ||
1859             folio_swap_entry(folio).val != swap.val ||
1860             !shmem_confirm_swap(mapping, index, swap)) {
1861                 error = -EEXIST;
1862                 goto unlock;
1863         }
1864         if (!folio_test_uptodate(folio)) {
1865                 error = -EIO;
1866                 goto failed;
1867         }
1868         folio_wait_writeback(folio);
1869
1870         /*
1871          * Some architectures may have to restore extra metadata to the
1872          * folio after reading from swap.
1873          */
1874         arch_swap_restore(swap, folio);
1875
1876         if (shmem_should_replace_folio(folio, gfp)) {
1877                 error = shmem_replace_folio(&folio, gfp, info, index);
1878                 if (error)
1879                         goto failed;
1880         }
1881
1882         error = shmem_add_to_page_cache(folio, mapping, index,
1883                                         swp_to_radix_entry(swap), gfp,
1884                                         charge_mm);
1885         if (error)
1886                 goto failed;
1887
1888         shmem_recalc_inode(inode, 0, -1);
1889
1890         if (sgp == SGP_WRITE)
1891                 folio_mark_accessed(folio);
1892
1893         delete_from_swap_cache(folio);
1894         folio_mark_dirty(folio);
1895         swap_free(swap);
1896         put_swap_device(si);
1897
1898         *foliop = folio;
1899         return 0;
1900 failed:
1901         if (!shmem_confirm_swap(mapping, index, swap))
1902                 error = -EEXIST;
1903         if (error == -EIO)
1904                 shmem_set_folio_swapin_error(inode, index, folio, swap);
1905 unlock:
1906         if (folio) {
1907                 folio_unlock(folio);
1908                 folio_put(folio);
1909         }
1910         put_swap_device(si);
1911
1912         return error;
1913 }
1914
1915 /*
1916  * shmem_get_folio_gfp - find page in cache, or get from swap, or allocate
1917  *
1918  * If we allocate a new one we do not mark it dirty. That's up to the
1919  * vm. If we swap it in we mark it dirty since we also free the swap
1920  * entry since a page cannot live in both the swap and page cache.
1921  *
1922  * vma, vmf, and fault_type are only supplied by shmem_fault:
1923  * otherwise they are NULL.
1924  */
1925 static int shmem_get_folio_gfp(struct inode *inode, pgoff_t index,
1926                 struct folio **foliop, enum sgp_type sgp, gfp_t gfp,
1927                 struct vm_area_struct *vma, struct vm_fault *vmf,
1928                 vm_fault_t *fault_type)
1929 {
1930         struct address_space *mapping = inode->i_mapping;
1931         struct shmem_inode_info *info = SHMEM_I(inode);
1932         struct shmem_sb_info *sbinfo;
1933         struct mm_struct *charge_mm;
1934         struct folio *folio;
1935         pgoff_t hindex;
1936         gfp_t huge_gfp;
1937         int error;
1938         int once = 0;
1939         int alloced = 0;
1940
1941         if (index > (MAX_LFS_FILESIZE >> PAGE_SHIFT))
1942                 return -EFBIG;
1943 repeat:
1944         if (sgp <= SGP_CACHE &&
1945             ((loff_t)index << PAGE_SHIFT) >= i_size_read(inode)) {
1946                 return -EINVAL;
1947         }
1948
1949         sbinfo = SHMEM_SB(inode->i_sb);
1950         charge_mm = vma ? vma->vm_mm : NULL;
1951
1952         folio = filemap_get_entry(mapping, index);
1953         if (folio && vma && userfaultfd_minor(vma)) {
1954                 if (!xa_is_value(folio))
1955                         folio_put(folio);
1956                 *fault_type = handle_userfault(vmf, VM_UFFD_MINOR);
1957                 return 0;
1958         }
1959
1960         if (xa_is_value(folio)) {
1961                 error = shmem_swapin_folio(inode, index, &folio,
1962                                           sgp, gfp, vma, fault_type);
1963                 if (error == -EEXIST)
1964                         goto repeat;
1965
1966                 *foliop = folio;
1967                 return error;
1968         }
1969
1970         if (folio) {
1971                 folio_lock(folio);
1972
1973                 /* Has the folio been truncated or swapped out? */
1974                 if (unlikely(folio->mapping != mapping)) {
1975                         folio_unlock(folio);
1976                         folio_put(folio);
1977                         goto repeat;
1978                 }
1979                 if (sgp == SGP_WRITE)
1980                         folio_mark_accessed(folio);
1981                 if (folio_test_uptodate(folio))
1982                         goto out;
1983                 /* fallocated folio */
1984                 if (sgp != SGP_READ)
1985                         goto clear;
1986                 folio_unlock(folio);
1987                 folio_put(folio);
1988         }
1989
1990         /*
1991          * SGP_READ: succeed on hole, with NULL folio, letting caller zero.
1992          * SGP_NOALLOC: fail on hole, with NULL folio, letting caller fail.
1993          */
1994         *foliop = NULL;
1995         if (sgp == SGP_READ)
1996                 return 0;
1997         if (sgp == SGP_NOALLOC)
1998                 return -ENOENT;
1999
2000         /*
2001          * Fast cache lookup and swap lookup did not find it: allocate.
2002          */
2003
2004         if (vma && userfaultfd_missing(vma)) {
2005                 *fault_type = handle_userfault(vmf, VM_UFFD_MISSING);
2006                 return 0;
2007         }
2008
2009         if (!shmem_is_huge(inode, index, false,
2010                            vma ? vma->vm_mm : NULL, vma ? vma->vm_flags : 0))
2011                 goto alloc_nohuge;
2012
2013         huge_gfp = vma_thp_gfp_mask(vma);
2014         huge_gfp = limit_gfp_mask(huge_gfp, gfp);
2015         folio = shmem_alloc_and_acct_folio(huge_gfp, inode, index, true);
2016         if (IS_ERR(folio)) {
2017 alloc_nohuge:
2018                 folio = shmem_alloc_and_acct_folio(gfp, inode, index, false);
2019         }
2020         if (IS_ERR(folio)) {
2021                 int retry = 5;
2022
2023                 error = PTR_ERR(folio);
2024                 folio = NULL;
2025                 if (error != -ENOSPC)
2026                         goto unlock;
2027                 /*
2028                  * Try to reclaim some space by splitting a large folio
2029                  * beyond i_size on the filesystem.
2030                  */
2031                 while (retry--) {
2032                         int ret;
2033
2034                         ret = shmem_unused_huge_shrink(sbinfo, NULL, 1);
2035                         if (ret == SHRINK_STOP)
2036                                 break;
2037                         if (ret)
2038                                 goto alloc_nohuge;
2039                 }
2040                 goto unlock;
2041         }
2042
2043         hindex = round_down(index, folio_nr_pages(folio));
2044
2045         if (sgp == SGP_WRITE)
2046                 __folio_set_referenced(folio);
2047
2048         error = shmem_add_to_page_cache(folio, mapping, hindex,
2049                                         NULL, gfp & GFP_RECLAIM_MASK,
2050                                         charge_mm);
2051         if (error)
2052                 goto unacct;
2053
2054         folio_add_lru(folio);
2055         shmem_recalc_inode(inode, folio_nr_pages(folio), 0);
2056         alloced = true;
2057
2058         if (folio_test_pmd_mappable(folio) &&
2059             DIV_ROUND_UP(i_size_read(inode), PAGE_SIZE) <
2060                                         folio_next_index(folio) - 1) {
2061                 /*
2062                  * Part of the large folio is beyond i_size: subject
2063                  * to shrink under memory pressure.
2064                  */
2065                 spin_lock(&sbinfo->shrinklist_lock);
2066                 /*
2067                  * _careful to defend against unlocked access to
2068                  * ->shrink_list in shmem_unused_huge_shrink()
2069                  */
2070                 if (list_empty_careful(&info->shrinklist)) {
2071                         list_add_tail(&info->shrinklist,
2072                                       &sbinfo->shrinklist);
2073                         sbinfo->shrinklist_len++;
2074                 }
2075                 spin_unlock(&sbinfo->shrinklist_lock);
2076         }
2077
2078         /*
2079          * Let SGP_FALLOC use the SGP_WRITE optimization on a new folio.
2080          */
2081         if (sgp == SGP_FALLOC)
2082                 sgp = SGP_WRITE;
2083 clear:
2084         /*
2085          * Let SGP_WRITE caller clear ends if write does not fill folio;
2086          * but SGP_FALLOC on a folio fallocated earlier must initialize
2087          * it now, lest undo on failure cancel our earlier guarantee.
2088          */
2089         if (sgp != SGP_WRITE && !folio_test_uptodate(folio)) {
2090                 long i, n = folio_nr_pages(folio);
2091
2092                 for (i = 0; i < n; i++)
2093                         clear_highpage(folio_page(folio, i));
2094                 flush_dcache_folio(folio);
2095                 folio_mark_uptodate(folio);
2096         }
2097
2098         /* Perhaps the file has been truncated since we checked */
2099         if (sgp <= SGP_CACHE &&
2100             ((loff_t)index << PAGE_SHIFT) >= i_size_read(inode)) {
2101                 if (alloced) {
2102                         folio_clear_dirty(folio);
2103                         filemap_remove_folio(folio);
2104                         shmem_recalc_inode(inode, 0, 0);
2105                 }
2106                 error = -EINVAL;
2107                 goto unlock;
2108         }
2109 out:
2110         *foliop = folio;
2111         return 0;
2112
2113         /*
2114          * Error recovery.
2115          */
2116 unacct:
2117         shmem_inode_unacct_blocks(inode, folio_nr_pages(folio));
2118
2119         if (folio_test_large(folio)) {
2120                 folio_unlock(folio);
2121                 folio_put(folio);
2122                 goto alloc_nohuge;
2123         }
2124 unlock:
2125         if (folio) {
2126                 folio_unlock(folio);
2127                 folio_put(folio);
2128         }
2129         if (error == -ENOSPC && !once++) {
2130                 shmem_recalc_inode(inode, 0, 0);
2131                 goto repeat;
2132         }
2133         if (error == -EEXIST)
2134                 goto repeat;
2135         return error;
2136 }
2137
2138 int shmem_get_folio(struct inode *inode, pgoff_t index, struct folio **foliop,
2139                 enum sgp_type sgp)
2140 {
2141         return shmem_get_folio_gfp(inode, index, foliop, sgp,
2142                         mapping_gfp_mask(inode->i_mapping), NULL, NULL, NULL);
2143 }
2144
2145 /*
2146  * This is like autoremove_wake_function, but it removes the wait queue
2147  * entry unconditionally - even if something else had already woken the
2148  * target.
2149  */
2150 static int synchronous_wake_function(wait_queue_entry_t *wait, unsigned mode, int sync, void *key)
2151 {
2152         int ret = default_wake_function(wait, mode, sync, key);
2153         list_del_init(&wait->entry);
2154         return ret;
2155 }
2156
2157 static vm_fault_t shmem_fault(struct vm_fault *vmf)
2158 {
2159         struct vm_area_struct *vma = vmf->vma;
2160         struct inode *inode = file_inode(vma->vm_file);
2161         gfp_t gfp = mapping_gfp_mask(inode->i_mapping);
2162         struct folio *folio = NULL;
2163         int err;
2164         vm_fault_t ret = VM_FAULT_LOCKED;
2165
2166         /*
2167          * Trinity finds that probing a hole which tmpfs is punching can
2168          * prevent the hole-punch from ever completing: which in turn
2169          * locks writers out with its hold on i_rwsem.  So refrain from
2170          * faulting pages into the hole while it's being punched.  Although
2171          * shmem_undo_range() does remove the additions, it may be unable to
2172          * keep up, as each new page needs its own unmap_mapping_range() call,
2173          * and the i_mmap tree grows ever slower to scan if new vmas are added.
2174          *
2175          * It does not matter if we sometimes reach this check just before the
2176          * hole-punch begins, so that one fault then races with the punch:
2177          * we just need to make racing faults a rare case.
2178          *
2179          * The implementation below would be much simpler if we just used a
2180          * standard mutex or completion: but we cannot take i_rwsem in fault,
2181          * and bloating every shmem inode for this unlikely case would be sad.
2182          */
2183         if (unlikely(inode->i_private)) {
2184                 struct shmem_falloc *shmem_falloc;
2185
2186                 spin_lock(&inode->i_lock);
2187                 shmem_falloc = inode->i_private;
2188                 if (shmem_falloc &&
2189                     shmem_falloc->waitq &&
2190                     vmf->pgoff >= shmem_falloc->start &&
2191                     vmf->pgoff < shmem_falloc->next) {
2192                         struct file *fpin;
2193                         wait_queue_head_t *shmem_falloc_waitq;
2194                         DEFINE_WAIT_FUNC(shmem_fault_wait, synchronous_wake_function);
2195
2196                         ret = VM_FAULT_NOPAGE;
2197                         fpin = maybe_unlock_mmap_for_io(vmf, NULL);
2198                         if (fpin)
2199                                 ret = VM_FAULT_RETRY;
2200
2201                         shmem_falloc_waitq = shmem_falloc->waitq;
2202                         prepare_to_wait(shmem_falloc_waitq, &shmem_fault_wait,
2203                                         TASK_UNINTERRUPTIBLE);
2204                         spin_unlock(&inode->i_lock);
2205                         schedule();
2206
2207                         /*
2208                          * shmem_falloc_waitq points into the shmem_fallocate()
2209                          * stack of the hole-punching task: shmem_falloc_waitq
2210                          * is usually invalid by the time we reach here, but
2211                          * finish_wait() does not dereference it in that case;
2212                          * though i_lock needed lest racing with wake_up_all().
2213                          */
2214                         spin_lock(&inode->i_lock);
2215                         finish_wait(shmem_falloc_waitq, &shmem_fault_wait);
2216                         spin_unlock(&inode->i_lock);
2217
2218                         if (fpin)
2219                                 fput(fpin);
2220                         return ret;
2221                 }
2222                 spin_unlock(&inode->i_lock);
2223         }
2224
2225         err = shmem_get_folio_gfp(inode, vmf->pgoff, &folio, SGP_CACHE,
2226                                   gfp, vma, vmf, &ret);
2227         if (err)
2228                 return vmf_error(err);
2229         if (folio)
2230                 vmf->page = folio_file_page(folio, vmf->pgoff);
2231         return ret;
2232 }
2233
2234 unsigned long shmem_get_unmapped_area(struct file *file,
2235                                       unsigned long uaddr, unsigned long len,
2236                                       unsigned long pgoff, unsigned long flags)
2237 {
2238         unsigned long (*get_area)(struct file *,
2239                 unsigned long, unsigned long, unsigned long, unsigned long);
2240         unsigned long addr;
2241         unsigned long offset;
2242         unsigned long inflated_len;
2243         unsigned long inflated_addr;
2244         unsigned long inflated_offset;
2245
2246         if (len > TASK_SIZE)
2247                 return -ENOMEM;
2248
2249         get_area = current->mm->get_unmapped_area;
2250         addr = get_area(file, uaddr, len, pgoff, flags);
2251
2252         if (!IS_ENABLED(CONFIG_TRANSPARENT_HUGEPAGE))
2253                 return addr;
2254         if (IS_ERR_VALUE(addr))
2255                 return addr;
2256         if (addr & ~PAGE_MASK)
2257                 return addr;
2258         if (addr > TASK_SIZE - len)
2259                 return addr;
2260
2261         if (shmem_huge == SHMEM_HUGE_DENY)
2262                 return addr;
2263         if (len < HPAGE_PMD_SIZE)
2264                 return addr;
2265         if (flags & MAP_FIXED)
2266                 return addr;
2267         /*
2268          * Our priority is to support MAP_SHARED mapped hugely;
2269          * and support MAP_PRIVATE mapped hugely too, until it is COWed.
2270          * But if caller specified an address hint and we allocated area there
2271          * successfully, respect that as before.
2272          */
2273         if (uaddr == addr)
2274                 return addr;
2275
2276         if (shmem_huge != SHMEM_HUGE_FORCE) {
2277                 struct super_block *sb;
2278
2279                 if (file) {
2280                         VM_BUG_ON(file->f_op != &shmem_file_operations);
2281                         sb = file_inode(file)->i_sb;
2282                 } else {
2283                         /*
2284                          * Called directly from mm/mmap.c, or drivers/char/mem.c
2285                          * for "/dev/zero", to create a shared anonymous object.
2286                          */
2287                         if (IS_ERR(shm_mnt))
2288                                 return addr;
2289                         sb = shm_mnt->mnt_sb;
2290                 }
2291                 if (SHMEM_SB(sb)->huge == SHMEM_HUGE_NEVER)
2292                         return addr;
2293         }
2294
2295         offset = (pgoff << PAGE_SHIFT) & (HPAGE_PMD_SIZE-1);
2296         if (offset && offset + len < 2 * HPAGE_PMD_SIZE)
2297                 return addr;
2298         if ((addr & (HPAGE_PMD_SIZE-1)) == offset)
2299                 return addr;
2300
2301         inflated_len = len + HPAGE_PMD_SIZE - PAGE_SIZE;
2302         if (inflated_len > TASK_SIZE)
2303                 return addr;
2304         if (inflated_len < len)
2305                 return addr;
2306
2307         inflated_addr = get_area(NULL, uaddr, inflated_len, 0, flags);
2308         if (IS_ERR_VALUE(inflated_addr))
2309                 return addr;
2310         if (inflated_addr & ~PAGE_MASK)
2311                 return addr;
2312
2313         inflated_offset = inflated_addr & (HPAGE_PMD_SIZE-1);
2314         inflated_addr += offset - inflated_offset;
2315         if (inflated_offset > offset)
2316                 inflated_addr += HPAGE_PMD_SIZE;
2317
2318         if (inflated_addr > TASK_SIZE - len)
2319                 return addr;
2320         return inflated_addr;
2321 }
2322
2323 #ifdef CONFIG_NUMA
2324 static int shmem_set_policy(struct vm_area_struct *vma, struct mempolicy *mpol)
2325 {
2326         struct inode *inode = file_inode(vma->vm_file);
2327         return mpol_set_shared_policy(&SHMEM_I(inode)->policy, vma, mpol);
2328 }
2329
2330 static struct mempolicy *shmem_get_policy(struct vm_area_struct *vma,
2331                                           unsigned long addr)
2332 {
2333         struct inode *inode = file_inode(vma->vm_file);
2334         pgoff_t index;
2335
2336         index = ((addr - vma->vm_start) >> PAGE_SHIFT) + vma->vm_pgoff;
2337         return mpol_shared_policy_lookup(&SHMEM_I(inode)->policy, index);
2338 }
2339 #endif
2340
2341 int shmem_lock(struct file *file, int lock, struct ucounts *ucounts)
2342 {
2343         struct inode *inode = file_inode(file);
2344         struct shmem_inode_info *info = SHMEM_I(inode);
2345         int retval = -ENOMEM;
2346
2347         /*
2348          * What serializes the accesses to info->flags?
2349          * ipc_lock_object() when called from shmctl_do_lock(),
2350          * no serialization needed when called from shm_destroy().
2351          */
2352         if (lock && !(info->flags & VM_LOCKED)) {
2353                 if (!user_shm_lock(inode->i_size, ucounts))
2354                         goto out_nomem;
2355                 info->flags |= VM_LOCKED;
2356                 mapping_set_unevictable(file->f_mapping);
2357         }
2358         if (!lock && (info->flags & VM_LOCKED) && ucounts) {
2359                 user_shm_unlock(inode->i_size, ucounts);
2360                 info->flags &= ~VM_LOCKED;
2361                 mapping_clear_unevictable(file->f_mapping);
2362         }
2363         retval = 0;
2364
2365 out_nomem:
2366         return retval;
2367 }
2368
2369 static int shmem_mmap(struct file *file, struct vm_area_struct *vma)
2370 {
2371         struct inode *inode = file_inode(file);
2372         struct shmem_inode_info *info = SHMEM_I(inode);
2373         int ret;
2374
2375         ret = seal_check_future_write(info->seals, vma);
2376         if (ret)
2377                 return ret;
2378
2379         /* arm64 - allow memory tagging on RAM-based files */
2380         vm_flags_set(vma, VM_MTE_ALLOWED);
2381
2382         file_accessed(file);
2383         /* This is anonymous shared memory if it is unlinked at the time of mmap */
2384         if (inode->i_nlink)
2385                 vma->vm_ops = &shmem_vm_ops;
2386         else
2387                 vma->vm_ops = &shmem_anon_vm_ops;
2388         return 0;
2389 }
2390
2391 #ifdef CONFIG_TMPFS_XATTR
2392 static int shmem_initxattrs(struct inode *, const struct xattr *, void *);
2393
2394 /*
2395  * chattr's fsflags are unrelated to extended attributes,
2396  * but tmpfs has chosen to enable them under the same config option.
2397  */
2398 static void shmem_set_inode_flags(struct inode *inode, unsigned int fsflags)
2399 {
2400         unsigned int i_flags = 0;
2401
2402         if (fsflags & FS_NOATIME_FL)
2403                 i_flags |= S_NOATIME;
2404         if (fsflags & FS_APPEND_FL)
2405                 i_flags |= S_APPEND;
2406         if (fsflags & FS_IMMUTABLE_FL)
2407                 i_flags |= S_IMMUTABLE;
2408         /*
2409          * But FS_NODUMP_FL does not require any action in i_flags.
2410          */
2411         inode_set_flags(inode, i_flags, S_NOATIME | S_APPEND | S_IMMUTABLE);
2412 }
2413 #else
2414 static void shmem_set_inode_flags(struct inode *inode, unsigned int fsflags)
2415 {
2416 }
2417 #define shmem_initxattrs NULL
2418 #endif
2419
2420 static struct offset_ctx *shmem_get_offset_ctx(struct inode *inode)
2421 {
2422         return &SHMEM_I(inode)->dir_offsets;
2423 }
2424
2425 static struct inode *__shmem_get_inode(struct mnt_idmap *idmap,
2426                                              struct super_block *sb,
2427                                              struct inode *dir, umode_t mode,
2428                                              dev_t dev, unsigned long flags)
2429 {
2430         struct inode *inode;
2431         struct shmem_inode_info *info;
2432         struct shmem_sb_info *sbinfo = SHMEM_SB(sb);
2433         ino_t ino;
2434         int err;
2435
2436         err = shmem_reserve_inode(sb, &ino);
2437         if (err)
2438                 return ERR_PTR(err);
2439
2440
2441         inode = new_inode(sb);
2442
2443         if (!inode) {
2444                 shmem_free_inode(sb, 0);
2445                 return ERR_PTR(-ENOSPC);
2446         }
2447
2448         inode->i_ino = ino;
2449         inode_init_owner(idmap, inode, dir, mode);
2450         inode->i_blocks = 0;
2451         inode->i_atime = inode->i_mtime = inode->i_ctime = current_time(inode);
2452         inode->i_generation = get_random_u32();
2453         info = SHMEM_I(inode);
2454         memset(info, 0, (char *)inode - (char *)info);
2455         spin_lock_init(&info->lock);
2456         atomic_set(&info->stop_eviction, 0);
2457         info->seals = F_SEAL_SEAL;
2458         info->flags = flags & VM_NORESERVE;
2459         info->i_crtime = inode->i_mtime;
2460         info->fsflags = (dir == NULL) ? 0 :
2461                 SHMEM_I(dir)->fsflags & SHMEM_FL_INHERITED;
2462         if (info->fsflags)
2463                 shmem_set_inode_flags(inode, info->fsflags);
2464         INIT_LIST_HEAD(&info->shrinklist);
2465         INIT_LIST_HEAD(&info->swaplist);
2466         INIT_LIST_HEAD(&info->swaplist);
2467         if (sbinfo->noswap)
2468                 mapping_set_unevictable(inode->i_mapping);
2469         simple_xattrs_init(&info->xattrs);
2470         cache_no_acl(inode);
2471         mapping_set_large_folios(inode->i_mapping);
2472
2473         switch (mode & S_IFMT) {
2474         default:
2475                 inode->i_op = &shmem_special_inode_operations;
2476                 init_special_inode(inode, mode, dev);
2477                 break;
2478         case S_IFREG:
2479                 inode->i_mapping->a_ops = &shmem_aops;
2480                 inode->i_op = &shmem_inode_operations;
2481                 inode->i_fop = &shmem_file_operations;
2482                 mpol_shared_policy_init(&info->policy,
2483                                          shmem_get_sbmpol(sbinfo));
2484                 break;
2485         case S_IFDIR:
2486                 inc_nlink(inode);
2487                 /* Some things misbehave if size == 0 on a directory */
2488                 inode->i_size = 2 * BOGO_DIRENT_SIZE;
2489                 inode->i_op = &shmem_dir_inode_operations;
2490                 inode->i_fop = &simple_offset_dir_operations;
2491                 simple_offset_init(shmem_get_offset_ctx(inode));
2492                 break;
2493         case S_IFLNK:
2494                 /*
2495                  * Must not load anything in the rbtree,
2496                  * mpol_free_shared_policy will not be called.
2497                  */
2498                 mpol_shared_policy_init(&info->policy, NULL);
2499                 break;
2500         }
2501
2502         lockdep_annotate_inode_mutex_key(inode);
2503         return inode;
2504 }
2505
2506 #ifdef CONFIG_TMPFS_QUOTA
2507 static struct inode *shmem_get_inode(struct mnt_idmap *idmap,
2508                                      struct super_block *sb, struct inode *dir,
2509                                      umode_t mode, dev_t dev, unsigned long flags)
2510 {
2511         int err;
2512         struct inode *inode;
2513
2514         inode = __shmem_get_inode(idmap, sb, dir, mode, dev, flags);
2515         if (IS_ERR(inode))
2516                 return inode;
2517
2518         err = dquot_initialize(inode);
2519         if (err)
2520                 goto errout;
2521
2522         err = dquot_alloc_inode(inode);
2523         if (err) {
2524                 dquot_drop(inode);
2525                 goto errout;
2526         }
2527         return inode;
2528
2529 errout:
2530         inode->i_flags |= S_NOQUOTA;
2531         iput(inode);
2532         return ERR_PTR(err);
2533 }
2534 #else
2535 static inline struct inode *shmem_get_inode(struct mnt_idmap *idmap,
2536                                      struct super_block *sb, struct inode *dir,
2537                                      umode_t mode, dev_t dev, unsigned long flags)
2538 {
2539         return __shmem_get_inode(idmap, sb, dir, mode, dev, flags);
2540 }
2541 #endif /* CONFIG_TMPFS_QUOTA */
2542
2543 #ifdef CONFIG_USERFAULTFD
2544 int shmem_mfill_atomic_pte(pmd_t *dst_pmd,
2545                            struct vm_area_struct *dst_vma,
2546                            unsigned long dst_addr,
2547                            unsigned long src_addr,
2548                            uffd_flags_t flags,
2549                            struct folio **foliop)
2550 {
2551         struct inode *inode = file_inode(dst_vma->vm_file);
2552         struct shmem_inode_info *info = SHMEM_I(inode);
2553         struct address_space *mapping = inode->i_mapping;
2554         gfp_t gfp = mapping_gfp_mask(mapping);
2555         pgoff_t pgoff = linear_page_index(dst_vma, dst_addr);
2556         void *page_kaddr;
2557         struct folio *folio;
2558         int ret;
2559         pgoff_t max_off;
2560
2561         if (shmem_inode_acct_block(inode, 1)) {
2562                 /*
2563                  * We may have got a page, returned -ENOENT triggering a retry,
2564                  * and now we find ourselves with -ENOMEM. Release the page, to
2565                  * avoid a BUG_ON in our caller.
2566                  */
2567                 if (unlikely(*foliop)) {
2568                         folio_put(*foliop);
2569                         *foliop = NULL;
2570                 }
2571                 return -ENOMEM;
2572         }
2573
2574         if (!*foliop) {
2575                 ret = -ENOMEM;
2576                 folio = shmem_alloc_folio(gfp, info, pgoff);
2577                 if (!folio)
2578                         goto out_unacct_blocks;
2579
2580                 if (uffd_flags_mode_is(flags, MFILL_ATOMIC_COPY)) {
2581                         page_kaddr = kmap_local_folio(folio, 0);
2582                         /*
2583                          * The read mmap_lock is held here.  Despite the
2584                          * mmap_lock being read recursive a deadlock is still
2585                          * possible if a writer has taken a lock.  For example:
2586                          *
2587                          * process A thread 1 takes read lock on own mmap_lock
2588                          * process A thread 2 calls mmap, blocks taking write lock
2589                          * process B thread 1 takes page fault, read lock on own mmap lock
2590                          * process B thread 2 calls mmap, blocks taking write lock
2591                          * process A thread 1 blocks taking read lock on process B
2592                          * process B thread 1 blocks taking read lock on process A
2593                          *
2594                          * Disable page faults to prevent potential deadlock
2595                          * and retry the copy outside the mmap_lock.
2596                          */
2597                         pagefault_disable();
2598                         ret = copy_from_user(page_kaddr,
2599                                              (const void __user *)src_addr,
2600                                              PAGE_SIZE);
2601                         pagefault_enable();
2602                         kunmap_local(page_kaddr);
2603
2604                         /* fallback to copy_from_user outside mmap_lock */
2605                         if (unlikely(ret)) {
2606                                 *foliop = folio;
2607                                 ret = -ENOENT;
2608                                 /* don't free the page */
2609                                 goto out_unacct_blocks;
2610                         }
2611
2612                         flush_dcache_folio(folio);
2613                 } else {                /* ZEROPAGE */
2614                         clear_user_highpage(&folio->page, dst_addr);
2615                 }
2616         } else {
2617                 folio = *foliop;
2618                 VM_BUG_ON_FOLIO(folio_test_large(folio), folio);
2619                 *foliop = NULL;
2620         }
2621
2622         VM_BUG_ON(folio_test_locked(folio));
2623         VM_BUG_ON(folio_test_swapbacked(folio));
2624         __folio_set_locked(folio);
2625         __folio_set_swapbacked(folio);
2626         __folio_mark_uptodate(folio);
2627
2628         ret = -EFAULT;
2629         max_off = DIV_ROUND_UP(i_size_read(inode), PAGE_SIZE);
2630         if (unlikely(pgoff >= max_off))
2631                 goto out_release;
2632
2633         ret = shmem_add_to_page_cache(folio, mapping, pgoff, NULL,
2634                                       gfp & GFP_RECLAIM_MASK, dst_vma->vm_mm);
2635         if (ret)
2636                 goto out_release;
2637
2638         ret = mfill_atomic_install_pte(dst_pmd, dst_vma, dst_addr,
2639                                        &folio->page, true, flags);
2640         if (ret)
2641                 goto out_delete_from_cache;
2642
2643         shmem_recalc_inode(inode, 1, 0);
2644         folio_unlock(folio);
2645         return 0;
2646 out_delete_from_cache:
2647         filemap_remove_folio(folio);
2648 out_release:
2649         folio_unlock(folio);
2650         folio_put(folio);
2651 out_unacct_blocks:
2652         shmem_inode_unacct_blocks(inode, 1);
2653         return ret;
2654 }
2655 #endif /* CONFIG_USERFAULTFD */
2656
2657 #ifdef CONFIG_TMPFS
2658 static const struct inode_operations shmem_symlink_inode_operations;
2659 static const struct inode_operations shmem_short_symlink_operations;
2660
2661 static int
2662 shmem_write_begin(struct file *file, struct address_space *mapping,
2663                         loff_t pos, unsigned len,
2664                         struct page **pagep, void **fsdata)
2665 {
2666         struct inode *inode = mapping->host;
2667         struct shmem_inode_info *info = SHMEM_I(inode);
2668         pgoff_t index = pos >> PAGE_SHIFT;
2669         struct folio *folio;
2670         int ret = 0;
2671
2672         /* i_rwsem is held by caller */
2673         if (unlikely(info->seals & (F_SEAL_GROW |
2674                                    F_SEAL_WRITE | F_SEAL_FUTURE_WRITE))) {
2675                 if (info->seals & (F_SEAL_WRITE | F_SEAL_FUTURE_WRITE))
2676                         return -EPERM;
2677                 if ((info->seals & F_SEAL_GROW) && pos + len > inode->i_size)
2678                         return -EPERM;
2679         }
2680
2681         ret = shmem_get_folio(inode, index, &folio, SGP_WRITE);
2682
2683         if (ret)
2684                 return ret;
2685
2686         *pagep = folio_file_page(folio, index);
2687         if (PageHWPoison(*pagep)) {
2688                 folio_unlock(folio);
2689                 folio_put(folio);
2690                 *pagep = NULL;
2691                 return -EIO;
2692         }
2693
2694         return 0;
2695 }
2696
2697 static int
2698 shmem_write_end(struct file *file, struct address_space *mapping,
2699                         loff_t pos, unsigned len, unsigned copied,
2700                         struct page *page, void *fsdata)
2701 {
2702         struct folio *folio = page_folio(page);
2703         struct inode *inode = mapping->host;
2704
2705         if (pos + copied > inode->i_size)
2706                 i_size_write(inode, pos + copied);
2707
2708         if (!folio_test_uptodate(folio)) {
2709                 if (copied < folio_size(folio)) {
2710                         size_t from = offset_in_folio(folio, pos);
2711                         folio_zero_segments(folio, 0, from,
2712                                         from + copied, folio_size(folio));
2713                 }
2714                 folio_mark_uptodate(folio);
2715         }
2716         folio_mark_dirty(folio);
2717         folio_unlock(folio);
2718         folio_put(folio);
2719
2720         return copied;
2721 }
2722
2723 static ssize_t shmem_file_read_iter(struct kiocb *iocb, struct iov_iter *to)
2724 {
2725         struct file *file = iocb->ki_filp;
2726         struct inode *inode = file_inode(file);
2727         struct address_space *mapping = inode->i_mapping;
2728         pgoff_t index;
2729         unsigned long offset;
2730         int error = 0;
2731         ssize_t retval = 0;
2732         loff_t *ppos = &iocb->ki_pos;
2733
2734         index = *ppos >> PAGE_SHIFT;
2735         offset = *ppos & ~PAGE_MASK;
2736
2737         for (;;) {
2738                 struct folio *folio = NULL;
2739                 struct page *page = NULL;
2740                 pgoff_t end_index;
2741                 unsigned long nr, ret;
2742                 loff_t i_size = i_size_read(inode);
2743
2744                 end_index = i_size >> PAGE_SHIFT;
2745                 if (index > end_index)
2746                         break;
2747                 if (index == end_index) {
2748                         nr = i_size & ~PAGE_MASK;
2749                         if (nr <= offset)
2750                                 break;
2751                 }
2752
2753                 error = shmem_get_folio(inode, index, &folio, SGP_READ);
2754                 if (error) {
2755                         if (error == -EINVAL)
2756                                 error = 0;
2757                         break;
2758                 }
2759                 if (folio) {
2760                         folio_unlock(folio);
2761
2762                         page = folio_file_page(folio, index);
2763                         if (PageHWPoison(page)) {
2764                                 folio_put(folio);
2765                                 error = -EIO;
2766                                 break;
2767                         }
2768                 }
2769
2770                 /*
2771                  * We must evaluate after, since reads (unlike writes)
2772                  * are called without i_rwsem protection against truncate
2773                  */
2774                 nr = PAGE_SIZE;
2775                 i_size = i_size_read(inode);
2776                 end_index = i_size >> PAGE_SHIFT;
2777                 if (index == end_index) {
2778                         nr = i_size & ~PAGE_MASK;
2779                         if (nr <= offset) {
2780                                 if (folio)
2781                                         folio_put(folio);
2782                                 break;
2783                         }
2784                 }
2785                 nr -= offset;
2786
2787                 if (folio) {
2788                         /*
2789                          * If users can be writing to this page using arbitrary
2790                          * virtual addresses, take care about potential aliasing
2791                          * before reading the page on the kernel side.
2792                          */
2793                         if (mapping_writably_mapped(mapping))
2794                                 flush_dcache_page(page);
2795                         /*
2796                          * Mark the page accessed if we read the beginning.
2797                          */
2798                         if (!offset)
2799                                 folio_mark_accessed(folio);
2800                         /*
2801                          * Ok, we have the page, and it's up-to-date, so
2802                          * now we can copy it to user space...
2803                          */
2804                         ret = copy_page_to_iter(page, offset, nr, to);
2805                         folio_put(folio);
2806
2807                 } else if (user_backed_iter(to)) {
2808                         /*
2809                          * Copy to user tends to be so well optimized, but
2810                          * clear_user() not so much, that it is noticeably
2811                          * faster to copy the zero page instead of clearing.
2812                          */
2813                         ret = copy_page_to_iter(ZERO_PAGE(0), offset, nr, to);
2814                 } else {
2815                         /*
2816                          * But submitting the same page twice in a row to
2817                          * splice() - or others? - can result in confusion:
2818                          * so don't attempt that optimization on pipes etc.
2819                          */
2820                         ret = iov_iter_zero(nr, to);
2821                 }
2822
2823                 retval += ret;
2824                 offset += ret;
2825                 index += offset >> PAGE_SHIFT;
2826                 offset &= ~PAGE_MASK;
2827
2828                 if (!iov_iter_count(to))
2829                         break;
2830                 if (ret < nr) {
2831                         error = -EFAULT;
2832                         break;
2833                 }
2834                 cond_resched();
2835         }
2836
2837         *ppos = ((loff_t) index << PAGE_SHIFT) + offset;
2838         file_accessed(file);
2839         return retval ? retval : error;
2840 }
2841
2842 static bool zero_pipe_buf_get(struct pipe_inode_info *pipe,
2843                               struct pipe_buffer *buf)
2844 {
2845         return true;
2846 }
2847
2848 static void zero_pipe_buf_release(struct pipe_inode_info *pipe,
2849                                   struct pipe_buffer *buf)
2850 {
2851 }
2852
2853 static bool zero_pipe_buf_try_steal(struct pipe_inode_info *pipe,
2854                                     struct pipe_buffer *buf)
2855 {
2856         return false;
2857 }
2858
2859 static const struct pipe_buf_operations zero_pipe_buf_ops = {
2860         .release        = zero_pipe_buf_release,
2861         .try_steal      = zero_pipe_buf_try_steal,
2862         .get            = zero_pipe_buf_get,
2863 };
2864
2865 static size_t splice_zeropage_into_pipe(struct pipe_inode_info *pipe,
2866                                         loff_t fpos, size_t size)
2867 {
2868         size_t offset = fpos & ~PAGE_MASK;
2869
2870         size = min_t(size_t, size, PAGE_SIZE - offset);
2871
2872         if (!pipe_full(pipe->head, pipe->tail, pipe->max_usage)) {
2873                 struct pipe_buffer *buf = pipe_head_buf(pipe);
2874
2875                 *buf = (struct pipe_buffer) {
2876                         .ops    = &zero_pipe_buf_ops,
2877                         .page   = ZERO_PAGE(0),
2878                         .offset = offset,
2879                         .len    = size,
2880                 };
2881                 pipe->head++;
2882         }
2883
2884         return size;
2885 }
2886
2887 static ssize_t shmem_file_splice_read(struct file *in, loff_t *ppos,
2888                                       struct pipe_inode_info *pipe,
2889                                       size_t len, unsigned int flags)
2890 {
2891         struct inode *inode = file_inode(in);
2892         struct address_space *mapping = inode->i_mapping;
2893         struct folio *folio = NULL;
2894         size_t total_spliced = 0, used, npages, n, part;
2895         loff_t isize;
2896         int error = 0;
2897
2898         /* Work out how much data we can actually add into the pipe */
2899         used = pipe_occupancy(pipe->head, pipe->tail);
2900         npages = max_t(ssize_t, pipe->max_usage - used, 0);
2901         len = min_t(size_t, len, npages * PAGE_SIZE);
2902
2903         do {
2904                 if (*ppos >= i_size_read(inode))
2905                         break;
2906
2907                 error = shmem_get_folio(inode, *ppos / PAGE_SIZE, &folio,
2908                                         SGP_READ);
2909                 if (error) {
2910                         if (error == -EINVAL)
2911                                 error = 0;
2912                         break;
2913                 }
2914                 if (folio) {
2915                         folio_unlock(folio);
2916
2917                         if (folio_test_hwpoison(folio) ||
2918                             (folio_test_large(folio) &&
2919                              folio_test_has_hwpoisoned(folio))) {
2920                                 error = -EIO;
2921                                 break;
2922                         }
2923                 }
2924
2925                 /*
2926                  * i_size must be checked after we know the pages are Uptodate.
2927                  *
2928                  * Checking i_size after the check allows us to calculate
2929                  * the correct value for "nr", which means the zero-filled
2930                  * part of the page is not copied back to userspace (unless
2931                  * another truncate extends the file - this is desired though).
2932                  */
2933                 isize = i_size_read(inode);
2934                 if (unlikely(*ppos >= isize))
2935                         break;
2936                 part = min_t(loff_t, isize - *ppos, len);
2937
2938                 if (folio) {
2939                         /*
2940                          * If users can be writing to this page using arbitrary
2941                          * virtual addresses, take care about potential aliasing
2942                          * before reading the page on the kernel side.
2943                          */
2944                         if (mapping_writably_mapped(mapping))
2945                                 flush_dcache_folio(folio);
2946                         folio_mark_accessed(folio);
2947                         /*
2948                          * Ok, we have the page, and it's up-to-date, so we can
2949                          * now splice it into the pipe.
2950                          */
2951                         n = splice_folio_into_pipe(pipe, folio, *ppos, part);
2952                         folio_put(folio);
2953                         folio = NULL;
2954                 } else {
2955                         n = splice_zeropage_into_pipe(pipe, *ppos, part);
2956                 }
2957
2958                 if (!n)
2959                         break;
2960                 len -= n;
2961                 total_spliced += n;
2962                 *ppos += n;
2963                 in->f_ra.prev_pos = *ppos;
2964                 if (pipe_full(pipe->head, pipe->tail, pipe->max_usage))
2965                         break;
2966
2967                 cond_resched();
2968         } while (len);
2969
2970         if (folio)
2971                 folio_put(folio);
2972
2973         file_accessed(in);
2974         return total_spliced ? total_spliced : error;
2975 }
2976
2977 static loff_t shmem_file_llseek(struct file *file, loff_t offset, int whence)
2978 {
2979         struct address_space *mapping = file->f_mapping;
2980         struct inode *inode = mapping->host;
2981
2982         if (whence != SEEK_DATA && whence != SEEK_HOLE)
2983                 return generic_file_llseek_size(file, offset, whence,
2984                                         MAX_LFS_FILESIZE, i_size_read(inode));
2985         if (offset < 0)
2986                 return -ENXIO;
2987
2988         inode_lock(inode);
2989         /* We're holding i_rwsem so we can access i_size directly */
2990         offset = mapping_seek_hole_data(mapping, offset, inode->i_size, whence);
2991         if (offset >= 0)
2992                 offset = vfs_setpos(file, offset, MAX_LFS_FILESIZE);
2993         inode_unlock(inode);
2994         return offset;
2995 }
2996
2997 static long shmem_fallocate(struct file *file, int mode, loff_t offset,
2998                                                          loff_t len)
2999 {
3000         struct inode *inode = file_inode(file);
3001         struct shmem_sb_info *sbinfo = SHMEM_SB(inode->i_sb);
3002         struct shmem_inode_info *info = SHMEM_I(inode);
3003         struct shmem_falloc shmem_falloc;
3004         pgoff_t start, index, end, undo_fallocend;
3005         int error;
3006
3007         if (mode & ~(FALLOC_FL_KEEP_SIZE | FALLOC_FL_PUNCH_HOLE))
3008                 return -EOPNOTSUPP;
3009
3010         inode_lock(inode);
3011
3012         if (mode & FALLOC_FL_PUNCH_HOLE) {
3013                 struct address_space *mapping = file->f_mapping;
3014                 loff_t unmap_start = round_up(offset, PAGE_SIZE);
3015                 loff_t unmap_end = round_down(offset + len, PAGE_SIZE) - 1;
3016                 DECLARE_WAIT_QUEUE_HEAD_ONSTACK(shmem_falloc_waitq);
3017
3018                 /* protected by i_rwsem */
3019                 if (info->seals & (F_SEAL_WRITE | F_SEAL_FUTURE_WRITE)) {
3020                         error = -EPERM;
3021                         goto out;
3022                 }
3023
3024                 shmem_falloc.waitq = &shmem_falloc_waitq;
3025                 shmem_falloc.start = (u64)unmap_start >> PAGE_SHIFT;
3026                 shmem_falloc.next = (unmap_end + 1) >> PAGE_SHIFT;
3027                 spin_lock(&inode->i_lock);
3028                 inode->i_private = &shmem_falloc;
3029                 spin_unlock(&inode->i_lock);
3030
3031                 if ((u64)unmap_end > (u64)unmap_start)
3032                         unmap_mapping_range(mapping, unmap_start,
3033                                             1 + unmap_end - unmap_start, 0);
3034                 shmem_truncate_range(inode, offset, offset + len - 1);
3035                 /* No need to unmap again: hole-punching leaves COWed pages */
3036
3037                 spin_lock(&inode->i_lock);
3038                 inode->i_private = NULL;
3039                 wake_up_all(&shmem_falloc_waitq);
3040                 WARN_ON_ONCE(!list_empty(&shmem_falloc_waitq.head));
3041                 spin_unlock(&inode->i_lock);
3042                 error = 0;
3043                 goto out;
3044         }
3045
3046         /* We need to check rlimit even when FALLOC_FL_KEEP_SIZE */
3047         error = inode_newsize_ok(inode, offset + len);
3048         if (error)
3049                 goto out;
3050
3051         if ((info->seals & F_SEAL_GROW) && offset + len > inode->i_size) {
3052                 error = -EPERM;
3053                 goto out;
3054         }
3055
3056         start = offset >> PAGE_SHIFT;
3057         end = (offset + len + PAGE_SIZE - 1) >> PAGE_SHIFT;
3058         /* Try to avoid a swapstorm if len is impossible to satisfy */
3059         if (sbinfo->max_blocks && end - start > sbinfo->max_blocks) {
3060                 error = -ENOSPC;
3061                 goto out;
3062         }
3063
3064         shmem_falloc.waitq = NULL;
3065         shmem_falloc.start = start;
3066         shmem_falloc.next  = start;
3067         shmem_falloc.nr_falloced = 0;
3068         shmem_falloc.nr_unswapped = 0;
3069         spin_lock(&inode->i_lock);
3070         inode->i_private = &shmem_falloc;
3071         spin_unlock(&inode->i_lock);
3072
3073         /*
3074          * info->fallocend is only relevant when huge pages might be
3075          * involved: to prevent split_huge_page() freeing fallocated
3076          * pages when FALLOC_FL_KEEP_SIZE committed beyond i_size.
3077          */
3078         undo_fallocend = info->fallocend;
3079         if (info->fallocend < end)
3080                 info->fallocend = end;
3081
3082         for (index = start; index < end; ) {
3083                 struct folio *folio;
3084
3085                 /*
3086                  * Good, the fallocate(2) manpage permits EINTR: we may have
3087                  * been interrupted because we are using up too much memory.
3088                  */
3089                 if (signal_pending(current))
3090                         error = -EINTR;
3091                 else if (shmem_falloc.nr_unswapped > shmem_falloc.nr_falloced)
3092                         error = -ENOMEM;
3093                 else
3094                         error = shmem_get_folio(inode, index, &folio,
3095                                                 SGP_FALLOC);
3096                 if (error) {
3097                         info->fallocend = undo_fallocend;
3098                         /* Remove the !uptodate folios we added */
3099                         if (index > start) {
3100                                 shmem_undo_range(inode,
3101                                     (loff_t)start << PAGE_SHIFT,
3102                                     ((loff_t)index << PAGE_SHIFT) - 1, true);
3103                         }
3104                         goto undone;
3105                 }
3106
3107                 /*
3108                  * Here is a more important optimization than it appears:
3109                  * a second SGP_FALLOC on the same large folio will clear it,
3110                  * making it uptodate and un-undoable if we fail later.
3111                  */
3112                 index = folio_next_index(folio);
3113                 /* Beware 32-bit wraparound */
3114                 if (!index)
3115                         index--;
3116
3117                 /*
3118                  * Inform shmem_writepage() how far we have reached.
3119                  * No need for lock or barrier: we have the page lock.
3120                  */
3121                 if (!folio_test_uptodate(folio))
3122                         shmem_falloc.nr_falloced += index - shmem_falloc.next;
3123                 shmem_falloc.next = index;
3124
3125                 /*
3126                  * If !uptodate, leave it that way so that freeable folios
3127                  * can be recognized if we need to rollback on error later.
3128                  * But mark it dirty so that memory pressure will swap rather
3129                  * than free the folios we are allocating (and SGP_CACHE folios
3130                  * might still be clean: we now need to mark those dirty too).
3131                  */
3132                 folio_mark_dirty(folio);
3133                 folio_unlock(folio);
3134                 folio_put(folio);
3135                 cond_resched();
3136         }
3137
3138         if (!(mode & FALLOC_FL_KEEP_SIZE) && offset + len > inode->i_size)
3139                 i_size_write(inode, offset + len);
3140 undone:
3141         spin_lock(&inode->i_lock);
3142         inode->i_private = NULL;
3143         spin_unlock(&inode->i_lock);
3144 out:
3145         if (!error)
3146                 file_modified(file);
3147         inode_unlock(inode);
3148         return error;
3149 }
3150
3151 static int shmem_statfs(struct dentry *dentry, struct kstatfs *buf)
3152 {
3153         struct shmem_sb_info *sbinfo = SHMEM_SB(dentry->d_sb);
3154
3155         buf->f_type = TMPFS_MAGIC;
3156         buf->f_bsize = PAGE_SIZE;
3157         buf->f_namelen = NAME_MAX;
3158         if (sbinfo->max_blocks) {
3159                 buf->f_blocks = sbinfo->max_blocks;
3160                 buf->f_bavail =
3161                 buf->f_bfree  = sbinfo->max_blocks -
3162                                 percpu_counter_sum(&sbinfo->used_blocks);
3163         }
3164         if (sbinfo->max_inodes) {
3165                 buf->f_files = sbinfo->max_inodes;
3166                 buf->f_ffree = sbinfo->free_ispace / BOGO_INODE_SIZE;
3167         }
3168         /* else leave those fields 0 like simple_statfs */
3169
3170         buf->f_fsid = uuid_to_fsid(dentry->d_sb->s_uuid.b);
3171
3172         return 0;
3173 }
3174
3175 /*
3176  * File creation. Allocate an inode, and we're done..
3177  */
3178 static int
3179 shmem_mknod(struct mnt_idmap *idmap, struct inode *dir,
3180             struct dentry *dentry, umode_t mode, dev_t dev)
3181 {
3182         struct inode *inode;
3183         int error;
3184
3185         inode = shmem_get_inode(idmap, dir->i_sb, dir, mode, dev, VM_NORESERVE);
3186
3187         if (IS_ERR(inode))
3188                 return PTR_ERR(inode);
3189
3190         error = simple_acl_create(dir, inode);
3191         if (error)
3192                 goto out_iput;
3193         error = security_inode_init_security(inode, dir,
3194                                              &dentry->d_name,
3195                                              shmem_initxattrs, NULL);
3196         if (error && error != -EOPNOTSUPP)
3197                 goto out_iput;
3198
3199         error = simple_offset_add(shmem_get_offset_ctx(dir), dentry);
3200         if (error)
3201                 goto out_iput;
3202
3203         dir->i_size += BOGO_DIRENT_SIZE;
3204         dir->i_ctime = dir->i_mtime = current_time(dir);
3205         inode_inc_iversion(dir);
3206         d_instantiate(dentry, inode);
3207         dget(dentry); /* Extra count - pin the dentry in core */
3208         return error;
3209
3210 out_iput:
3211         iput(inode);
3212         return error;
3213 }
3214
3215 static int
3216 shmem_tmpfile(struct mnt_idmap *idmap, struct inode *dir,
3217               struct file *file, umode_t mode)
3218 {
3219         struct inode *inode;
3220         int error;
3221
3222         inode = shmem_get_inode(idmap, dir->i_sb, dir, mode, 0, VM_NORESERVE);
3223
3224         if (IS_ERR(inode)) {
3225                 error = PTR_ERR(inode);
3226                 goto err_out;
3227         }
3228
3229         error = security_inode_init_security(inode, dir,
3230                                              NULL,
3231                                              shmem_initxattrs, NULL);
3232         if (error && error != -EOPNOTSUPP)
3233                 goto out_iput;
3234         error = simple_acl_create(dir, inode);
3235         if (error)
3236                 goto out_iput;
3237         d_tmpfile(file, inode);
3238
3239 err_out:
3240         return finish_open_simple(file, error);
3241 out_iput:
3242         iput(inode);
3243         return error;
3244 }
3245
3246 static int shmem_mkdir(struct mnt_idmap *idmap, struct inode *dir,
3247                        struct dentry *dentry, umode_t mode)
3248 {
3249         int error;
3250
3251         error = shmem_mknod(idmap, dir, dentry, mode | S_IFDIR, 0);
3252         if (error)
3253                 return error;
3254         inc_nlink(dir);
3255         return 0;
3256 }
3257
3258 static int shmem_create(struct mnt_idmap *idmap, struct inode *dir,
3259                         struct dentry *dentry, umode_t mode, bool excl)
3260 {
3261         return shmem_mknod(idmap, dir, dentry, mode | S_IFREG, 0);
3262 }
3263
3264 /*
3265  * Link a file..
3266  */
3267 static int shmem_link(struct dentry *old_dentry, struct inode *dir, struct dentry *dentry)
3268 {
3269         struct inode *inode = d_inode(old_dentry);
3270         int ret = 0;
3271
3272         /*
3273          * No ordinary (disk based) filesystem counts links as inodes;
3274          * but each new link needs a new dentry, pinning lowmem, and
3275          * tmpfs dentries cannot be pruned until they are unlinked.
3276          * But if an O_TMPFILE file is linked into the tmpfs, the
3277          * first link must skip that, to get the accounting right.
3278          */
3279         if (inode->i_nlink) {
3280                 ret = shmem_reserve_inode(inode->i_sb, NULL);
3281                 if (ret)
3282                         goto out;
3283         }
3284
3285         ret = simple_offset_add(shmem_get_offset_ctx(dir), dentry);
3286         if (ret) {
3287                 if (inode->i_nlink)
3288                         shmem_free_inode(inode->i_sb, 0);
3289                 goto out;
3290         }
3291
3292         dir->i_size += BOGO_DIRENT_SIZE;
3293         inode->i_ctime = dir->i_ctime = dir->i_mtime = current_time(inode);
3294         inode_inc_iversion(dir);
3295         inc_nlink(inode);
3296         ihold(inode);   /* New dentry reference */
3297         dget(dentry);           /* Extra pinning count for the created dentry */
3298         d_instantiate(dentry, inode);
3299 out:
3300         return ret;
3301 }
3302
3303 static int shmem_unlink(struct inode *dir, struct dentry *dentry)
3304 {
3305         struct inode *inode = d_inode(dentry);
3306
3307         if (inode->i_nlink > 1 && !S_ISDIR(inode->i_mode))
3308                 shmem_free_inode(inode->i_sb, 0);
3309
3310         simple_offset_remove(shmem_get_offset_ctx(dir), dentry);
3311
3312         dir->i_size -= BOGO_DIRENT_SIZE;
3313         inode->i_ctime = dir->i_ctime = dir->i_mtime = current_time(inode);
3314         inode_inc_iversion(dir);
3315         drop_nlink(inode);
3316         dput(dentry);   /* Undo the count from "create" - this does all the work */
3317         return 0;
3318 }
3319
3320 static int shmem_rmdir(struct inode *dir, struct dentry *dentry)
3321 {
3322         if (!simple_empty(dentry))
3323                 return -ENOTEMPTY;
3324
3325         drop_nlink(d_inode(dentry));
3326         drop_nlink(dir);
3327         return shmem_unlink(dir, dentry);
3328 }
3329
3330 static int shmem_whiteout(struct mnt_idmap *idmap,
3331                           struct inode *old_dir, struct dentry *old_dentry)
3332 {
3333         struct dentry *whiteout;
3334         int error;
3335
3336         whiteout = d_alloc(old_dentry->d_parent, &old_dentry->d_name);
3337         if (!whiteout)
3338                 return -ENOMEM;
3339
3340         error = shmem_mknod(idmap, old_dir, whiteout,
3341                             S_IFCHR | WHITEOUT_MODE, WHITEOUT_DEV);
3342         dput(whiteout);
3343         if (error)
3344                 return error;
3345
3346         /*
3347          * Cheat and hash the whiteout while the old dentry is still in
3348          * place, instead of playing games with FS_RENAME_DOES_D_MOVE.
3349          *
3350          * d_lookup() will consistently find one of them at this point,
3351          * not sure which one, but that isn't even important.
3352          */
3353         d_rehash(whiteout);
3354         return 0;
3355 }
3356
3357 /*
3358  * The VFS layer already does all the dentry stuff for rename,
3359  * we just have to decrement the usage count for the target if
3360  * it exists so that the VFS layer correctly free's it when it
3361  * gets overwritten.
3362  */
3363 static int shmem_rename2(struct mnt_idmap *idmap,
3364                          struct inode *old_dir, struct dentry *old_dentry,
3365                          struct inode *new_dir, struct dentry *new_dentry,
3366                          unsigned int flags)
3367 {
3368         struct inode *inode = d_inode(old_dentry);
3369         int they_are_dirs = S_ISDIR(inode->i_mode);
3370         int error;
3371
3372         if (flags & ~(RENAME_NOREPLACE | RENAME_EXCHANGE | RENAME_WHITEOUT))
3373                 return -EINVAL;
3374
3375         if (flags & RENAME_EXCHANGE)
3376                 return simple_offset_rename_exchange(old_dir, old_dentry,
3377                                                      new_dir, new_dentry);
3378
3379         if (!simple_empty(new_dentry))
3380                 return -ENOTEMPTY;
3381
3382         if (flags & RENAME_WHITEOUT) {
3383                 error = shmem_whiteout(idmap, old_dir, old_dentry);
3384                 if (error)
3385                         return error;
3386         }
3387
3388         simple_offset_remove(shmem_get_offset_ctx(old_dir), old_dentry);
3389         error = simple_offset_add(shmem_get_offset_ctx(new_dir), old_dentry);
3390         if (error)
3391                 return error;
3392
3393         if (d_really_is_positive(new_dentry)) {
3394                 (void) shmem_unlink(new_dir, new_dentry);
3395                 if (they_are_dirs) {
3396                         drop_nlink(d_inode(new_dentry));
3397                         drop_nlink(old_dir);
3398                 }
3399         } else if (they_are_dirs) {
3400                 drop_nlink(old_dir);
3401                 inc_nlink(new_dir);
3402         }
3403
3404         old_dir->i_size -= BOGO_DIRENT_SIZE;
3405         new_dir->i_size += BOGO_DIRENT_SIZE;
3406         old_dir->i_ctime = old_dir->i_mtime =
3407         new_dir->i_ctime = new_dir->i_mtime =
3408         inode->i_ctime = current_time(old_dir);
3409         inode_inc_iversion(old_dir);
3410         inode_inc_iversion(new_dir);
3411         return 0;
3412 }
3413
3414 static int shmem_symlink(struct mnt_idmap *idmap, struct inode *dir,
3415                          struct dentry *dentry, const char *symname)
3416 {
3417         int error;
3418         int len;
3419         struct inode *inode;
3420         struct folio *folio;
3421
3422         len = strlen(symname) + 1;
3423         if (len > PAGE_SIZE)
3424                 return -ENAMETOOLONG;
3425
3426         inode = shmem_get_inode(idmap, dir->i_sb, dir, S_IFLNK | 0777, 0,
3427                                 VM_NORESERVE);
3428
3429         if (IS_ERR(inode))
3430                 return PTR_ERR(inode);
3431
3432         error = security_inode_init_security(inode, dir, &dentry->d_name,
3433                                              shmem_initxattrs, NULL);
3434         if (error && error != -EOPNOTSUPP)
3435                 goto out_iput;
3436
3437         error = simple_offset_add(shmem_get_offset_ctx(dir), dentry);
3438         if (error)
3439                 goto out_iput;
3440
3441         inode->i_size = len-1;
3442         if (len <= SHORT_SYMLINK_LEN) {
3443                 inode->i_link = kmemdup(symname, len, GFP_KERNEL);
3444                 if (!inode->i_link) {
3445                         error = -ENOMEM;
3446                         goto out_remove_offset;
3447                 }
3448                 inode->i_op = &shmem_short_symlink_operations;
3449         } else {
3450                 inode_nohighmem(inode);
3451                 error = shmem_get_folio(inode, 0, &folio, SGP_WRITE);
3452                 if (error)
3453                         goto out_remove_offset;
3454                 inode->i_mapping->a_ops = &shmem_aops;
3455                 inode->i_op = &shmem_symlink_inode_operations;
3456                 memcpy(folio_address(folio), symname, len);
3457                 folio_mark_uptodate(folio);
3458                 folio_mark_dirty(folio);
3459                 folio_unlock(folio);
3460                 folio_put(folio);
3461         }
3462         dir->i_size += BOGO_DIRENT_SIZE;
3463         dir->i_ctime = dir->i_mtime = current_time(dir);
3464         inode_inc_iversion(dir);
3465         d_instantiate(dentry, inode);
3466         dget(dentry);
3467         return 0;
3468
3469 out_remove_offset:
3470         simple_offset_remove(shmem_get_offset_ctx(dir), dentry);
3471 out_iput:
3472         iput(inode);
3473         return error;
3474 }
3475
3476 static void shmem_put_link(void *arg)
3477 {
3478         folio_mark_accessed(arg);
3479         folio_put(arg);
3480 }
3481
3482 static const char *shmem_get_link(struct dentry *dentry,
3483                                   struct inode *inode,
3484                                   struct delayed_call *done)
3485 {
3486         struct folio *folio = NULL;
3487         int error;
3488
3489         if (!dentry) {
3490                 folio = filemap_get_folio(inode->i_mapping, 0);
3491                 if (IS_ERR(folio))
3492                         return ERR_PTR(-ECHILD);
3493                 if (PageHWPoison(folio_page(folio, 0)) ||
3494                     !folio_test_uptodate(folio)) {
3495                         folio_put(folio);
3496                         return ERR_PTR(-ECHILD);
3497                 }
3498         } else {
3499                 error = shmem_get_folio(inode, 0, &folio, SGP_READ);
3500                 if (error)
3501                         return ERR_PTR(error);
3502                 if (!folio)
3503                         return ERR_PTR(-ECHILD);
3504                 if (PageHWPoison(folio_page(folio, 0))) {
3505                         folio_unlock(folio);
3506                         folio_put(folio);
3507                         return ERR_PTR(-ECHILD);
3508                 }
3509                 folio_unlock(folio);
3510         }
3511         set_delayed_call(done, shmem_put_link, folio);
3512         return folio_address(folio);
3513 }
3514
3515 #ifdef CONFIG_TMPFS_XATTR
3516
3517 static int shmem_fileattr_get(struct dentry *dentry, struct fileattr *fa)
3518 {
3519         struct shmem_inode_info *info = SHMEM_I(d_inode(dentry));
3520
3521         fileattr_fill_flags(fa, info->fsflags & SHMEM_FL_USER_VISIBLE);
3522
3523         return 0;
3524 }
3525
3526 static int shmem_fileattr_set(struct mnt_idmap *idmap,
3527                               struct dentry *dentry, struct fileattr *fa)
3528 {
3529         struct inode *inode = d_inode(dentry);
3530         struct shmem_inode_info *info = SHMEM_I(inode);
3531
3532         if (fileattr_has_fsx(fa))
3533                 return -EOPNOTSUPP;
3534         if (fa->flags & ~SHMEM_FL_USER_MODIFIABLE)
3535                 return -EOPNOTSUPP;
3536
3537         info->fsflags = (info->fsflags & ~SHMEM_FL_USER_MODIFIABLE) |
3538                 (fa->flags & SHMEM_FL_USER_MODIFIABLE);
3539
3540         shmem_set_inode_flags(inode, info->fsflags);
3541         inode->i_ctime = current_time(inode);
3542         inode_inc_iversion(inode);
3543         return 0;
3544 }
3545
3546 /*
3547  * Superblocks without xattr inode operations may get some security.* xattr
3548  * support from the LSM "for free". As soon as we have any other xattrs
3549  * like ACLs, we also need to implement the security.* handlers at
3550  * filesystem level, though.
3551  */
3552
3553 /*
3554  * Callback for security_inode_init_security() for acquiring xattrs.
3555  */
3556 static int shmem_initxattrs(struct inode *inode,
3557                             const struct xattr *xattr_array,
3558                             void *fs_info)
3559 {
3560         struct shmem_inode_info *info = SHMEM_I(inode);
3561         struct shmem_sb_info *sbinfo = SHMEM_SB(inode->i_sb);
3562         const struct xattr *xattr;
3563         struct simple_xattr *new_xattr;
3564         size_t ispace = 0;
3565         size_t len;
3566
3567         if (sbinfo->max_inodes) {
3568                 for (xattr = xattr_array; xattr->name != NULL; xattr++) {
3569                         ispace += simple_xattr_space(xattr->name,
3570                                 xattr->value_len + XATTR_SECURITY_PREFIX_LEN);
3571                 }
3572                 if (ispace) {
3573                         raw_spin_lock(&sbinfo->stat_lock);
3574                         if (sbinfo->free_ispace < ispace)
3575                                 ispace = 0;
3576                         else
3577                                 sbinfo->free_ispace -= ispace;
3578                         raw_spin_unlock(&sbinfo->stat_lock);
3579                         if (!ispace)
3580                                 return -ENOSPC;
3581                 }
3582         }
3583
3584         for (xattr = xattr_array; xattr->name != NULL; xattr++) {
3585                 new_xattr = simple_xattr_alloc(xattr->value, xattr->value_len);
3586                 if (!new_xattr)
3587                         break;
3588
3589                 len = strlen(xattr->name) + 1;
3590                 new_xattr->name = kmalloc(XATTR_SECURITY_PREFIX_LEN + len,
3591                                           GFP_KERNEL);
3592                 if (!new_xattr->name) {
3593                         kvfree(new_xattr);
3594                         break;
3595                 }
3596
3597                 memcpy(new_xattr->name, XATTR_SECURITY_PREFIX,
3598                        XATTR_SECURITY_PREFIX_LEN);
3599                 memcpy(new_xattr->name + XATTR_SECURITY_PREFIX_LEN,
3600                        xattr->name, len);
3601
3602                 simple_xattr_add(&info->xattrs, new_xattr);
3603         }
3604
3605         if (xattr->name != NULL) {
3606                 if (ispace) {
3607                         raw_spin_lock(&sbinfo->stat_lock);
3608                         sbinfo->free_ispace += ispace;
3609                         raw_spin_unlock(&sbinfo->stat_lock);
3610                 }
3611                 simple_xattrs_free(&info->xattrs, NULL);
3612                 return -ENOMEM;
3613         }
3614
3615         return 0;
3616 }
3617
3618 static int shmem_xattr_handler_get(const struct xattr_handler *handler,
3619                                    struct dentry *unused, struct inode *inode,
3620                                    const char *name, void *buffer, size_t size)
3621 {
3622         struct shmem_inode_info *info = SHMEM_I(inode);
3623
3624         name = xattr_full_name(handler, name);
3625         return simple_xattr_get(&info->xattrs, name, buffer, size);
3626 }
3627
3628 static int shmem_xattr_handler_set(const struct xattr_handler *handler,
3629                                    struct mnt_idmap *idmap,
3630                                    struct dentry *unused, struct inode *inode,
3631                                    const char *name, const void *value,
3632                                    size_t size, int flags)
3633 {
3634         struct shmem_inode_info *info = SHMEM_I(inode);
3635         struct shmem_sb_info *sbinfo = SHMEM_SB(inode->i_sb);
3636         struct simple_xattr *old_xattr;
3637         size_t ispace = 0;
3638
3639         name = xattr_full_name(handler, name);
3640         if (value && sbinfo->max_inodes) {
3641                 ispace = simple_xattr_space(name, size);
3642                 raw_spin_lock(&sbinfo->stat_lock);
3643                 if (sbinfo->free_ispace < ispace)
3644                         ispace = 0;
3645                 else
3646                         sbinfo->free_ispace -= ispace;
3647                 raw_spin_unlock(&sbinfo->stat_lock);
3648                 if (!ispace)
3649                         return -ENOSPC;
3650         }
3651
3652         old_xattr = simple_xattr_set(&info->xattrs, name, value, size, flags);
3653         if (!IS_ERR(old_xattr)) {
3654                 ispace = 0;
3655                 if (old_xattr && sbinfo->max_inodes)
3656                         ispace = simple_xattr_space(old_xattr->name,
3657                                                     old_xattr->size);
3658                 simple_xattr_free(old_xattr);
3659                 old_xattr = NULL;
3660                 inode->i_ctime = current_time(inode);
3661                 inode_inc_iversion(inode);
3662         }
3663         if (ispace) {
3664                 raw_spin_lock(&sbinfo->stat_lock);
3665                 sbinfo->free_ispace += ispace;
3666                 raw_spin_unlock(&sbinfo->stat_lock);
3667         }
3668         return PTR_ERR(old_xattr);
3669 }
3670
3671 static const struct xattr_handler shmem_security_xattr_handler = {
3672         .prefix = XATTR_SECURITY_PREFIX,
3673         .get = shmem_xattr_handler_get,
3674         .set = shmem_xattr_handler_set,
3675 };
3676
3677 static const struct xattr_handler shmem_trusted_xattr_handler = {
3678         .prefix = XATTR_TRUSTED_PREFIX,
3679         .get = shmem_xattr_handler_get,
3680         .set = shmem_xattr_handler_set,
3681 };
3682
3683 static const struct xattr_handler shmem_user_xattr_handler = {
3684         .prefix = XATTR_USER_PREFIX,
3685         .get = shmem_xattr_handler_get,
3686         .set = shmem_xattr_handler_set,
3687 };
3688
3689 static const struct xattr_handler *shmem_xattr_handlers[] = {
3690         &shmem_security_xattr_handler,
3691         &shmem_trusted_xattr_handler,
3692         &shmem_user_xattr_handler,
3693         NULL
3694 };
3695
3696 static ssize_t shmem_listxattr(struct dentry *dentry, char *buffer, size_t size)
3697 {
3698         struct shmem_inode_info *info = SHMEM_I(d_inode(dentry));
3699         return simple_xattr_list(d_inode(dentry), &info->xattrs, buffer, size);
3700 }
3701 #endif /* CONFIG_TMPFS_XATTR */
3702
3703 static const struct inode_operations shmem_short_symlink_operations = {
3704         .getattr        = shmem_getattr,
3705         .setattr        = shmem_setattr,
3706         .get_link       = simple_get_link,
3707 #ifdef CONFIG_TMPFS_XATTR
3708         .listxattr      = shmem_listxattr,
3709 #endif
3710 };
3711
3712 static const struct inode_operations shmem_symlink_inode_operations = {
3713         .getattr        = shmem_getattr,
3714         .setattr        = shmem_setattr,
3715         .get_link       = shmem_get_link,
3716 #ifdef CONFIG_TMPFS_XATTR
3717         .listxattr      = shmem_listxattr,
3718 #endif
3719 };
3720
3721 static struct dentry *shmem_get_parent(struct dentry *child)
3722 {
3723         return ERR_PTR(-ESTALE);
3724 }
3725
3726 static int shmem_match(struct inode *ino, void *vfh)
3727 {
3728         __u32 *fh = vfh;
3729         __u64 inum = fh[2];
3730         inum = (inum << 32) | fh[1];
3731         return ino->i_ino == inum && fh[0] == ino->i_generation;
3732 }
3733
3734 /* Find any alias of inode, but prefer a hashed alias */
3735 static struct dentry *shmem_find_alias(struct inode *inode)
3736 {
3737         struct dentry *alias = d_find_alias(inode);
3738
3739         return alias ?: d_find_any_alias(inode);
3740 }
3741
3742
3743 static struct dentry *shmem_fh_to_dentry(struct super_block *sb,
3744                 struct fid *fid, int fh_len, int fh_type)
3745 {
3746         struct inode *inode;
3747         struct dentry *dentry = NULL;
3748         u64 inum;
3749
3750         if (fh_len < 3)
3751                 return NULL;
3752
3753         inum = fid->raw[2];
3754         inum = (inum << 32) | fid->raw[1];
3755
3756         inode = ilookup5(sb, (unsigned long)(inum + fid->raw[0]),
3757                         shmem_match, fid->raw);
3758         if (inode) {
3759                 dentry = shmem_find_alias(inode);
3760                 iput(inode);
3761         }
3762
3763         return dentry;
3764 }
3765
3766 static int shmem_encode_fh(struct inode *inode, __u32 *fh, int *len,
3767                                 struct inode *parent)
3768 {
3769         if (*len < 3) {
3770                 *len = 3;
3771                 return FILEID_INVALID;
3772         }
3773
3774         if (inode_unhashed(inode)) {
3775                 /* Unfortunately insert_inode_hash is not idempotent,
3776                  * so as we hash inodes here rather than at creation
3777                  * time, we need a lock to ensure we only try
3778                  * to do it once
3779                  */
3780                 static DEFINE_SPINLOCK(lock);
3781                 spin_lock(&lock);
3782                 if (inode_unhashed(inode))
3783                         __insert_inode_hash(inode,
3784                                             inode->i_ino + inode->i_generation);
3785                 spin_unlock(&lock);
3786         }
3787
3788         fh[0] = inode->i_generation;
3789         fh[1] = inode->i_ino;
3790         fh[2] = ((__u64)inode->i_ino) >> 32;
3791
3792         *len = 3;
3793         return 1;
3794 }
3795
3796 static const struct export_operations shmem_export_ops = {
3797         .get_parent     = shmem_get_parent,
3798         .encode_fh      = shmem_encode_fh,
3799         .fh_to_dentry   = shmem_fh_to_dentry,
3800 };
3801
3802 enum shmem_param {
3803         Opt_gid,
3804         Opt_huge,
3805         Opt_mode,
3806         Opt_mpol,
3807         Opt_nr_blocks,
3808         Opt_nr_inodes,
3809         Opt_size,
3810         Opt_uid,
3811         Opt_inode32,
3812         Opt_inode64,
3813         Opt_noswap,
3814         Opt_quota,
3815         Opt_usrquota,
3816         Opt_grpquota,
3817         Opt_usrquota_block_hardlimit,
3818         Opt_usrquota_inode_hardlimit,
3819         Opt_grpquota_block_hardlimit,
3820         Opt_grpquota_inode_hardlimit,
3821 };
3822
3823 static const struct constant_table shmem_param_enums_huge[] = {
3824         {"never",       SHMEM_HUGE_NEVER },
3825         {"always",      SHMEM_HUGE_ALWAYS },
3826         {"within_size", SHMEM_HUGE_WITHIN_SIZE },
3827         {"advise",      SHMEM_HUGE_ADVISE },
3828         {}
3829 };
3830
3831 const struct fs_parameter_spec shmem_fs_parameters[] = {
3832         fsparam_u32   ("gid",           Opt_gid),
3833         fsparam_enum  ("huge",          Opt_huge,  shmem_param_enums_huge),
3834         fsparam_u32oct("mode",          Opt_mode),
3835         fsparam_string("mpol",          Opt_mpol),
3836         fsparam_string("nr_blocks",     Opt_nr_blocks),
3837         fsparam_string("nr_inodes",     Opt_nr_inodes),
3838         fsparam_string("size",          Opt_size),
3839         fsparam_u32   ("uid",           Opt_uid),
3840         fsparam_flag  ("inode32",       Opt_inode32),
3841         fsparam_flag  ("inode64",       Opt_inode64),
3842         fsparam_flag  ("noswap",        Opt_noswap),
3843 #ifdef CONFIG_TMPFS_QUOTA
3844         fsparam_flag  ("quota",         Opt_quota),
3845         fsparam_flag  ("usrquota",      Opt_usrquota),
3846         fsparam_flag  ("grpquota",      Opt_grpquota),
3847         fsparam_string("usrquota_block_hardlimit", Opt_usrquota_block_hardlimit),
3848         fsparam_string("usrquota_inode_hardlimit", Opt_usrquota_inode_hardlimit),
3849         fsparam_string("grpquota_block_hardlimit", Opt_grpquota_block_hardlimit),
3850         fsparam_string("grpquota_inode_hardlimit", Opt_grpquota_inode_hardlimit),
3851 #endif
3852         {}
3853 };
3854
3855 static int shmem_parse_one(struct fs_context *fc, struct fs_parameter *param)
3856 {
3857         struct shmem_options *ctx = fc->fs_private;
3858         struct fs_parse_result result;
3859         unsigned long long size;
3860         char *rest;
3861         int opt;
3862         kuid_t kuid;
3863         kgid_t kgid;
3864
3865         opt = fs_parse(fc, shmem_fs_parameters, param, &result);
3866         if (opt < 0)
3867                 return opt;
3868
3869         switch (opt) {
3870         case Opt_size:
3871                 size = memparse(param->string, &rest);
3872                 if (*rest == '%') {
3873                         size <<= PAGE_SHIFT;
3874                         size *= totalram_pages();
3875                         do_div(size, 100);
3876                         rest++;
3877                 }
3878                 if (*rest)
3879                         goto bad_value;
3880                 ctx->blocks = DIV_ROUND_UP(size, PAGE_SIZE);
3881                 ctx->seen |= SHMEM_SEEN_BLOCKS;
3882                 break;
3883         case Opt_nr_blocks:
3884                 ctx->blocks = memparse(param->string, &rest);
3885                 if (*rest || ctx->blocks > LONG_MAX)
3886                         goto bad_value;
3887                 ctx->seen |= SHMEM_SEEN_BLOCKS;
3888                 break;
3889         case Opt_nr_inodes:
3890                 ctx->inodes = memparse(param->string, &rest);
3891                 if (*rest || ctx->inodes > ULONG_MAX / BOGO_INODE_SIZE)
3892                         goto bad_value;
3893                 ctx->seen |= SHMEM_SEEN_INODES;
3894                 break;
3895         case Opt_mode:
3896                 ctx->mode = result.uint_32 & 07777;
3897                 break;
3898         case Opt_uid:
3899                 kuid = make_kuid(current_user_ns(), result.uint_32);
3900                 if (!uid_valid(kuid))
3901                         goto bad_value;
3902
3903                 /*
3904                  * The requested uid must be representable in the
3905                  * filesystem's idmapping.
3906                  */
3907                 if (!kuid_has_mapping(fc->user_ns, kuid))
3908                         goto bad_value;
3909
3910                 ctx->uid = kuid;
3911                 break;
3912         case Opt_gid:
3913                 kgid = make_kgid(current_user_ns(), result.uint_32);
3914                 if (!gid_valid(kgid))
3915                         goto bad_value;
3916
3917                 /*
3918                  * The requested gid must be representable in the
3919                  * filesystem's idmapping.
3920                  */
3921                 if (!kgid_has_mapping(fc->user_ns, kgid))
3922                         goto bad_value;
3923
3924                 ctx->gid = kgid;
3925                 break;
3926         case Opt_huge:
3927                 ctx->huge = result.uint_32;
3928                 if (ctx->huge != SHMEM_HUGE_NEVER &&
3929                     !(IS_ENABLED(CONFIG_TRANSPARENT_HUGEPAGE) &&
3930                       has_transparent_hugepage()))
3931                         goto unsupported_parameter;
3932                 ctx->seen |= SHMEM_SEEN_HUGE;
3933                 break;
3934         case Opt_mpol:
3935                 if (IS_ENABLED(CONFIG_NUMA)) {
3936                         mpol_put(ctx->mpol);
3937                         ctx->mpol = NULL;
3938                         if (mpol_parse_str(param->string, &ctx->mpol))
3939                                 goto bad_value;
3940                         break;
3941                 }
3942                 goto unsupported_parameter;
3943         case Opt_inode32:
3944                 ctx->full_inums = false;
3945                 ctx->seen |= SHMEM_SEEN_INUMS;
3946                 break;
3947         case Opt_inode64:
3948                 if (sizeof(ino_t) < 8) {
3949                         return invalfc(fc,
3950                                        "Cannot use inode64 with <64bit inums in kernel\n");
3951                 }
3952                 ctx->full_inums = true;
3953                 ctx->seen |= SHMEM_SEEN_INUMS;
3954                 break;
3955         case Opt_noswap:
3956                 if ((fc->user_ns != &init_user_ns) || !capable(CAP_SYS_ADMIN)) {
3957                         return invalfc(fc,
3958                                        "Turning off swap in unprivileged tmpfs mounts unsupported");
3959                 }
3960                 ctx->noswap = true;
3961                 ctx->seen |= SHMEM_SEEN_NOSWAP;
3962                 break;
3963         case Opt_quota:
3964                 if (fc->user_ns != &init_user_ns)
3965                         return invalfc(fc, "Quotas in unprivileged tmpfs mounts are unsupported");
3966                 ctx->seen |= SHMEM_SEEN_QUOTA;
3967                 ctx->quota_types |= (QTYPE_MASK_USR | QTYPE_MASK_GRP);
3968                 break;
3969         case Opt_usrquota:
3970                 if (fc->user_ns != &init_user_ns)
3971                         return invalfc(fc, "Quotas in unprivileged tmpfs mounts are unsupported");
3972                 ctx->seen |= SHMEM_SEEN_QUOTA;
3973                 ctx->quota_types |= QTYPE_MASK_USR;
3974                 break;
3975         case Opt_grpquota:
3976                 if (fc->user_ns != &init_user_ns)
3977                         return invalfc(fc, "Quotas in unprivileged tmpfs mounts are unsupported");
3978                 ctx->seen |= SHMEM_SEEN_QUOTA;
3979                 ctx->quota_types |= QTYPE_MASK_GRP;
3980                 break;
3981         case Opt_usrquota_block_hardlimit:
3982                 size = memparse(param->string, &rest);
3983                 if (*rest || !size)
3984                         goto bad_value;
3985                 if (size > SHMEM_QUOTA_MAX_SPC_LIMIT)
3986                         return invalfc(fc,
3987                                        "User quota block hardlimit too large.");
3988                 ctx->qlimits.usrquota_bhardlimit = size;
3989                 break;
3990         case Opt_grpquota_block_hardlimit:
3991                 size = memparse(param->string, &rest);
3992                 if (*rest || !size)
3993                         goto bad_value;
3994                 if (size > SHMEM_QUOTA_MAX_SPC_LIMIT)
3995                         return invalfc(fc,
3996                                        "Group quota block hardlimit too large.");
3997                 ctx->qlimits.grpquota_bhardlimit = size;
3998                 break;
3999         case Opt_usrquota_inode_hardlimit:
4000                 size = memparse(param->string, &rest);
4001                 if (*rest || !size)
4002                         goto bad_value;
4003                 if (size > SHMEM_QUOTA_MAX_INO_LIMIT)
4004                         return invalfc(fc,
4005                                        "User quota inode hardlimit too large.");
4006                 ctx->qlimits.usrquota_ihardlimit = size;
4007                 break;
4008         case Opt_grpquota_inode_hardlimit:
4009                 size = memparse(param->string, &rest);
4010                 if (*rest || !size)
4011                         goto bad_value;
4012                 if (size > SHMEM_QUOTA_MAX_INO_LIMIT)
4013                         return invalfc(fc,
4014                                        "Group quota inode hardlimit too large.");
4015                 ctx->qlimits.grpquota_ihardlimit = size;
4016                 break;
4017         }
4018         return 0;
4019
4020 unsupported_parameter:
4021         return invalfc(fc, "Unsupported parameter '%s'", param->key);
4022 bad_value:
4023         return invalfc(fc, "Bad value for '%s'", param->key);
4024 }
4025
4026 static int shmem_parse_options(struct fs_context *fc, void *data)
4027 {
4028         char *options = data;
4029
4030         if (options) {
4031                 int err = security_sb_eat_lsm_opts(options, &fc->security);
4032                 if (err)
4033                         return err;
4034         }
4035
4036         while (options != NULL) {
4037                 char *this_char = options;
4038                 for (;;) {
4039                         /*
4040                          * NUL-terminate this option: unfortunately,
4041                          * mount options form a comma-separated list,
4042                          * but mpol's nodelist may also contain commas.
4043                          */
4044                         options = strchr(options, ',');
4045                         if (options == NULL)
4046                                 break;
4047                         options++;
4048                         if (!isdigit(*options)) {
4049                                 options[-1] = '\0';
4050                                 break;
4051                         }
4052                 }
4053                 if (*this_char) {
4054                         char *value = strchr(this_char, '=');
4055                         size_t len = 0;
4056                         int err;
4057
4058                         if (value) {
4059                                 *value++ = '\0';
4060                                 len = strlen(value);
4061                         }
4062                         err = vfs_parse_fs_string(fc, this_char, value, len);
4063                         if (err < 0)
4064                                 return err;
4065                 }
4066         }
4067         return 0;
4068 }
4069
4070 /*
4071  * Reconfigure a shmem filesystem.
4072  */
4073 static int shmem_reconfigure(struct fs_context *fc)
4074 {
4075         struct shmem_options *ctx = fc->fs_private;
4076         struct shmem_sb_info *sbinfo = SHMEM_SB(fc->root->d_sb);
4077         unsigned long used_isp;
4078         struct mempolicy *mpol = NULL;
4079         const char *err;
4080
4081         raw_spin_lock(&sbinfo->stat_lock);
4082         used_isp = sbinfo->max_inodes * BOGO_INODE_SIZE - sbinfo->free_ispace;
4083
4084         if ((ctx->seen & SHMEM_SEEN_BLOCKS) && ctx->blocks) {
4085                 if (!sbinfo->max_blocks) {
4086                         err = "Cannot retroactively limit size";
4087                         goto out;
4088                 }
4089                 if (percpu_counter_compare(&sbinfo->used_blocks,
4090                                            ctx->blocks) > 0) {
4091                         err = "Too small a size for current use";
4092                         goto out;
4093                 }
4094         }
4095         if ((ctx->seen & SHMEM_SEEN_INODES) && ctx->inodes) {
4096                 if (!sbinfo->max_inodes) {
4097                         err = "Cannot retroactively limit inodes";
4098                         goto out;
4099                 }
4100                 if (ctx->inodes * BOGO_INODE_SIZE < used_isp) {
4101                         err = "Too few inodes for current use";
4102                         goto out;
4103                 }
4104         }
4105
4106         if ((ctx->seen & SHMEM_SEEN_INUMS) && !ctx->full_inums &&
4107             sbinfo->next_ino > UINT_MAX) {
4108                 err = "Current inum too high to switch to 32-bit inums";
4109                 goto out;
4110         }
4111         if ((ctx->seen & SHMEM_SEEN_NOSWAP) && ctx->noswap && !sbinfo->noswap) {
4112                 err = "Cannot disable swap on remount";
4113                 goto out;
4114         }
4115         if (!(ctx->seen & SHMEM_SEEN_NOSWAP) && !ctx->noswap && sbinfo->noswap) {
4116                 err = "Cannot enable swap on remount if it was disabled on first mount";
4117                 goto out;
4118         }
4119
4120         if (ctx->seen & SHMEM_SEEN_QUOTA &&
4121             !sb_any_quota_loaded(fc->root->d_sb)) {
4122                 err = "Cannot enable quota on remount";
4123                 goto out;
4124         }
4125
4126 #ifdef CONFIG_TMPFS_QUOTA
4127 #define CHANGED_LIMIT(name)                                             \
4128         (ctx->qlimits.name## hardlimit &&                               \
4129         (ctx->qlimits.name## hardlimit != sbinfo->qlimits.name## hardlimit))
4130
4131         if (CHANGED_LIMIT(usrquota_b) || CHANGED_LIMIT(usrquota_i) ||
4132             CHANGED_LIMIT(grpquota_b) || CHANGED_LIMIT(grpquota_i)) {
4133                 err = "Cannot change global quota limit on remount";
4134                 goto out;
4135         }
4136 #endif /* CONFIG_TMPFS_QUOTA */
4137
4138         if (ctx->seen & SHMEM_SEEN_HUGE)
4139                 sbinfo->huge = ctx->huge;
4140         if (ctx->seen & SHMEM_SEEN_INUMS)
4141                 sbinfo->full_inums = ctx->full_inums;
4142         if (ctx->seen & SHMEM_SEEN_BLOCKS)
4143                 sbinfo->max_blocks  = ctx->blocks;
4144         if (ctx->seen & SHMEM_SEEN_INODES) {
4145                 sbinfo->max_inodes  = ctx->inodes;
4146                 sbinfo->free_ispace = ctx->inodes * BOGO_INODE_SIZE - used_isp;
4147         }
4148
4149         /*
4150          * Preserve previous mempolicy unless mpol remount option was specified.
4151          */
4152         if (ctx->mpol) {
4153                 mpol = sbinfo->mpol;
4154                 sbinfo->mpol = ctx->mpol;       /* transfers initial ref */
4155                 ctx->mpol = NULL;
4156         }
4157
4158         if (ctx->noswap)
4159                 sbinfo->noswap = true;
4160
4161         raw_spin_unlock(&sbinfo->stat_lock);
4162         mpol_put(mpol);
4163         return 0;
4164 out:
4165         raw_spin_unlock(&sbinfo->stat_lock);
4166         return invalfc(fc, "%s", err);
4167 }
4168
4169 static int shmem_show_options(struct seq_file *seq, struct dentry *root)
4170 {
4171         struct shmem_sb_info *sbinfo = SHMEM_SB(root->d_sb);
4172         struct mempolicy *mpol;
4173
4174         if (sbinfo->max_blocks != shmem_default_max_blocks())
4175                 seq_printf(seq, ",size=%luk",
4176                         sbinfo->max_blocks << (PAGE_SHIFT - 10));
4177         if (sbinfo->max_inodes != shmem_default_max_inodes())
4178                 seq_printf(seq, ",nr_inodes=%lu", sbinfo->max_inodes);
4179         if (sbinfo->mode != (0777 | S_ISVTX))
4180                 seq_printf(seq, ",mode=%03ho", sbinfo->mode);
4181         if (!uid_eq(sbinfo->uid, GLOBAL_ROOT_UID))
4182                 seq_printf(seq, ",uid=%u",
4183                                 from_kuid_munged(&init_user_ns, sbinfo->uid));
4184         if (!gid_eq(sbinfo->gid, GLOBAL_ROOT_GID))
4185                 seq_printf(seq, ",gid=%u",
4186                                 from_kgid_munged(&init_user_ns, sbinfo->gid));
4187
4188         /*
4189          * Showing inode{64,32} might be useful even if it's the system default,
4190          * since then people don't have to resort to checking both here and
4191          * /proc/config.gz to confirm 64-bit inums were successfully applied
4192          * (which may not even exist if IKCONFIG_PROC isn't enabled).
4193          *
4194          * We hide it when inode64 isn't the default and we are using 32-bit
4195          * inodes, since that probably just means the feature isn't even under
4196          * consideration.
4197          *
4198          * As such:
4199          *
4200          *                     +-----------------+-----------------+
4201          *                     | TMPFS_INODE64=y | TMPFS_INODE64=n |
4202          *  +------------------+-----------------+-----------------+
4203          *  | full_inums=true  | show            | show            |
4204          *  | full_inums=false | show            | hide            |
4205          *  +------------------+-----------------+-----------------+
4206          *
4207          */
4208         if (IS_ENABLED(CONFIG_TMPFS_INODE64) || sbinfo->full_inums)
4209                 seq_printf(seq, ",inode%d", (sbinfo->full_inums ? 64 : 32));
4210 #ifdef CONFIG_TRANSPARENT_HUGEPAGE
4211         /* Rightly or wrongly, show huge mount option unmasked by shmem_huge */
4212         if (sbinfo->huge)
4213                 seq_printf(seq, ",huge=%s", shmem_format_huge(sbinfo->huge));
4214 #endif
4215         mpol = shmem_get_sbmpol(sbinfo);
4216         shmem_show_mpol(seq, mpol);
4217         mpol_put(mpol);
4218         if (sbinfo->noswap)
4219                 seq_printf(seq, ",noswap");
4220         return 0;
4221 }
4222
4223 #endif /* CONFIG_TMPFS */
4224
4225 static void shmem_put_super(struct super_block *sb)
4226 {
4227         struct shmem_sb_info *sbinfo = SHMEM_SB(sb);
4228
4229 #ifdef CONFIG_TMPFS_QUOTA
4230         shmem_disable_quotas(sb);
4231 #endif
4232         free_percpu(sbinfo->ino_batch);
4233         percpu_counter_destroy(&sbinfo->used_blocks);
4234         mpol_put(sbinfo->mpol);
4235         kfree(sbinfo);
4236         sb->s_fs_info = NULL;
4237 }
4238
4239 static int shmem_fill_super(struct super_block *sb, struct fs_context *fc)
4240 {
4241         struct shmem_options *ctx = fc->fs_private;
4242         struct inode *inode;
4243         struct shmem_sb_info *sbinfo;
4244         int error = -ENOMEM;
4245
4246         /* Round up to L1_CACHE_BYTES to resist false sharing */
4247         sbinfo = kzalloc(max((int)sizeof(struct shmem_sb_info),
4248                                 L1_CACHE_BYTES), GFP_KERNEL);
4249         if (!sbinfo)
4250                 return error;
4251
4252         sb->s_fs_info = sbinfo;
4253
4254 #ifdef CONFIG_TMPFS
4255         /*
4256          * Per default we only allow half of the physical ram per
4257          * tmpfs instance, limiting inodes to one per page of lowmem;
4258          * but the internal instance is left unlimited.
4259          */
4260         if (!(sb->s_flags & SB_KERNMOUNT)) {
4261                 if (!(ctx->seen & SHMEM_SEEN_BLOCKS))
4262                         ctx->blocks = shmem_default_max_blocks();
4263                 if (!(ctx->seen & SHMEM_SEEN_INODES))
4264                         ctx->inodes = shmem_default_max_inodes();
4265                 if (!(ctx->seen & SHMEM_SEEN_INUMS))
4266                         ctx->full_inums = IS_ENABLED(CONFIG_TMPFS_INODE64);
4267                 sbinfo->noswap = ctx->noswap;
4268         } else {
4269                 sb->s_flags |= SB_NOUSER;
4270         }
4271         sb->s_export_op = &shmem_export_ops;
4272         sb->s_flags |= SB_NOSEC | SB_I_VERSION;
4273 #else
4274         sb->s_flags |= SB_NOUSER;
4275 #endif
4276         sbinfo->max_blocks = ctx->blocks;
4277         sbinfo->max_inodes = ctx->inodes;
4278         sbinfo->free_ispace = sbinfo->max_inodes * BOGO_INODE_SIZE;
4279         if (sb->s_flags & SB_KERNMOUNT) {
4280                 sbinfo->ino_batch = alloc_percpu(ino_t);
4281                 if (!sbinfo->ino_batch)
4282                         goto failed;
4283         }
4284         sbinfo->uid = ctx->uid;
4285         sbinfo->gid = ctx->gid;
4286         sbinfo->full_inums = ctx->full_inums;
4287         sbinfo->mode = ctx->mode;
4288         sbinfo->huge = ctx->huge;
4289         sbinfo->mpol = ctx->mpol;
4290         ctx->mpol = NULL;
4291
4292         raw_spin_lock_init(&sbinfo->stat_lock);
4293         if (percpu_counter_init(&sbinfo->used_blocks, 0, GFP_KERNEL))
4294                 goto failed;
4295         spin_lock_init(&sbinfo->shrinklist_lock);
4296         INIT_LIST_HEAD(&sbinfo->shrinklist);
4297
4298         sb->s_maxbytes = MAX_LFS_FILESIZE;
4299         sb->s_blocksize = PAGE_SIZE;
4300         sb->s_blocksize_bits = PAGE_SHIFT;
4301         sb->s_magic = TMPFS_MAGIC;
4302         sb->s_op = &shmem_ops;
4303         sb->s_time_gran = 1;
4304 #ifdef CONFIG_TMPFS_XATTR
4305         sb->s_xattr = shmem_xattr_handlers;
4306 #endif
4307 #ifdef CONFIG_TMPFS_POSIX_ACL
4308         sb->s_flags |= SB_POSIXACL;
4309 #endif
4310         uuid_gen(&sb->s_uuid);
4311
4312 #ifdef CONFIG_TMPFS_QUOTA
4313         if (ctx->seen & SHMEM_SEEN_QUOTA) {
4314                 sb->dq_op = &shmem_quota_operations;
4315                 sb->s_qcop = &dquot_quotactl_sysfile_ops;
4316                 sb->s_quota_types = QTYPE_MASK_USR | QTYPE_MASK_GRP;
4317
4318                 /* Copy the default limits from ctx into sbinfo */
4319                 memcpy(&sbinfo->qlimits, &ctx->qlimits,
4320                        sizeof(struct shmem_quota_limits));
4321
4322                 if (shmem_enable_quotas(sb, ctx->quota_types))
4323                         goto failed;
4324         }
4325 #endif /* CONFIG_TMPFS_QUOTA */
4326
4327         inode = shmem_get_inode(&nop_mnt_idmap, sb, NULL, S_IFDIR | sbinfo->mode, 0,
4328                                 VM_NORESERVE);
4329         if (IS_ERR(inode)) {
4330                 error = PTR_ERR(inode);
4331                 goto failed;
4332         }
4333         inode->i_uid = sbinfo->uid;
4334         inode->i_gid = sbinfo->gid;
4335         sb->s_root = d_make_root(inode);
4336         if (!sb->s_root)
4337                 goto failed;
4338         return 0;
4339
4340 failed:
4341         shmem_put_super(sb);
4342         return error;
4343 }
4344
4345 static int shmem_get_tree(struct fs_context *fc)
4346 {
4347         return get_tree_nodev(fc, shmem_fill_super);
4348 }
4349
4350 static void shmem_free_fc(struct fs_context *fc)
4351 {
4352         struct shmem_options *ctx = fc->fs_private;
4353
4354         if (ctx) {
4355                 mpol_put(ctx->mpol);
4356                 kfree(ctx);
4357         }
4358 }
4359
4360 static const struct fs_context_operations shmem_fs_context_ops = {
4361         .free                   = shmem_free_fc,
4362         .get_tree               = shmem_get_tree,
4363 #ifdef CONFIG_TMPFS
4364         .parse_monolithic       = shmem_parse_options,
4365         .parse_param            = shmem_parse_one,
4366         .reconfigure            = shmem_reconfigure,
4367 #endif
4368 };
4369
4370 static struct kmem_cache *shmem_inode_cachep;
4371
4372 static struct inode *shmem_alloc_inode(struct super_block *sb)
4373 {
4374         struct shmem_inode_info *info;
4375         info = alloc_inode_sb(sb, shmem_inode_cachep, GFP_KERNEL);
4376         if (!info)
4377                 return NULL;
4378         return &info->vfs_inode;
4379 }
4380
4381 static void shmem_free_in_core_inode(struct inode *inode)
4382 {
4383         if (S_ISLNK(inode->i_mode))
4384                 kfree(inode->i_link);
4385         kmem_cache_free(shmem_inode_cachep, SHMEM_I(inode));
4386 }
4387
4388 static void shmem_destroy_inode(struct inode *inode)
4389 {
4390         if (S_ISREG(inode->i_mode))
4391                 mpol_free_shared_policy(&SHMEM_I(inode)->policy);
4392         if (S_ISDIR(inode->i_mode))
4393                 simple_offset_destroy(shmem_get_offset_ctx(inode));
4394 }
4395
4396 static void shmem_init_inode(void *foo)
4397 {
4398         struct shmem_inode_info *info = foo;
4399         inode_init_once(&info->vfs_inode);
4400 }
4401
4402 static void shmem_init_inodecache(void)
4403 {
4404         shmem_inode_cachep = kmem_cache_create("shmem_inode_cache",
4405                                 sizeof(struct shmem_inode_info),
4406                                 0, SLAB_PANIC|SLAB_ACCOUNT, shmem_init_inode);
4407 }
4408
4409 static void shmem_destroy_inodecache(void)
4410 {
4411         kmem_cache_destroy(shmem_inode_cachep);
4412 }
4413
4414 /* Keep the page in page cache instead of truncating it */
4415 static int shmem_error_remove_page(struct address_space *mapping,
4416                                    struct page *page)
4417 {
4418         return 0;
4419 }
4420
4421 const struct address_space_operations shmem_aops = {
4422         .writepage      = shmem_writepage,
4423         .dirty_folio    = noop_dirty_folio,
4424 #ifdef CONFIG_TMPFS
4425         .write_begin    = shmem_write_begin,
4426         .write_end      = shmem_write_end,
4427 #endif
4428 #ifdef CONFIG_MIGRATION
4429         .migrate_folio  = migrate_folio,
4430 #endif
4431         .error_remove_page = shmem_error_remove_page,
4432 };
4433 EXPORT_SYMBOL(shmem_aops);
4434
4435 static const struct file_operations shmem_file_operations = {
4436         .mmap           = shmem_mmap,
4437         .open           = generic_file_open,
4438         .get_unmapped_area = shmem_get_unmapped_area,
4439 #ifdef CONFIG_TMPFS
4440         .llseek         = shmem_file_llseek,
4441         .read_iter      = shmem_file_read_iter,
4442         .write_iter     = generic_file_write_iter,
4443         .fsync          = noop_fsync,
4444         .splice_read    = shmem_file_splice_read,
4445         .splice_write   = iter_file_splice_write,
4446         .fallocate      = shmem_fallocate,
4447 #endif
4448 };
4449
4450 static const struct inode_operations shmem_inode_operations = {
4451         .getattr        = shmem_getattr,
4452         .setattr        = shmem_setattr,
4453 #ifdef CONFIG_TMPFS_XATTR
4454         .listxattr      = shmem_listxattr,
4455         .set_acl        = simple_set_acl,
4456         .fileattr_get   = shmem_fileattr_get,
4457         .fileattr_set   = shmem_fileattr_set,
4458 #endif
4459 };
4460
4461 static const struct inode_operations shmem_dir_inode_operations = {
4462 #ifdef CONFIG_TMPFS
4463         .getattr        = shmem_getattr,
4464         .create         = shmem_create,
4465         .lookup         = simple_lookup,
4466         .link           = shmem_link,
4467         .unlink         = shmem_unlink,
4468         .symlink        = shmem_symlink,
4469         .mkdir          = shmem_mkdir,
4470         .rmdir          = shmem_rmdir,
4471         .mknod          = shmem_mknod,
4472         .rename         = shmem_rename2,
4473         .tmpfile        = shmem_tmpfile,
4474         .get_offset_ctx = shmem_get_offset_ctx,
4475 #endif
4476 #ifdef CONFIG_TMPFS_XATTR
4477         .listxattr      = shmem_listxattr,
4478         .fileattr_get   = shmem_fileattr_get,
4479         .fileattr_set   = shmem_fileattr_set,
4480 #endif
4481 #ifdef CONFIG_TMPFS_POSIX_ACL
4482         .setattr        = shmem_setattr,
4483         .set_acl        = simple_set_acl,
4484 #endif
4485 };
4486
4487 static const struct inode_operations shmem_special_inode_operations = {
4488         .getattr        = shmem_getattr,
4489 #ifdef CONFIG_TMPFS_XATTR
4490         .listxattr      = shmem_listxattr,
4491 #endif
4492 #ifdef CONFIG_TMPFS_POSIX_ACL
4493         .setattr        = shmem_setattr,
4494         .set_acl        = simple_set_acl,
4495 #endif
4496 };
4497
4498 static const struct super_operations shmem_ops = {
4499         .alloc_inode    = shmem_alloc_inode,
4500         .free_inode     = shmem_free_in_core_inode,
4501         .destroy_inode  = shmem_destroy_inode,
4502 #ifdef CONFIG_TMPFS
4503         .statfs         = shmem_statfs,
4504         .show_options   = shmem_show_options,
4505 #endif
4506 #ifdef CONFIG_TMPFS_QUOTA
4507         .get_dquots     = shmem_get_dquots,
4508 #endif
4509         .evict_inode    = shmem_evict_inode,
4510         .drop_inode     = generic_delete_inode,
4511         .put_super      = shmem_put_super,
4512 #ifdef CONFIG_TRANSPARENT_HUGEPAGE
4513         .nr_cached_objects      = shmem_unused_huge_count,
4514         .free_cached_objects    = shmem_unused_huge_scan,
4515 #endif
4516 };
4517
4518 static const struct vm_operations_struct shmem_vm_ops = {
4519         .fault          = shmem_fault,
4520         .map_pages      = filemap_map_pages,
4521 #ifdef CONFIG_NUMA
4522         .set_policy     = shmem_set_policy,
4523         .get_policy     = shmem_get_policy,
4524 #endif
4525 };
4526
4527 static const struct vm_operations_struct shmem_anon_vm_ops = {
4528         .fault          = shmem_fault,
4529         .map_pages      = filemap_map_pages,
4530 #ifdef CONFIG_NUMA
4531         .set_policy     = shmem_set_policy,
4532         .get_policy     = shmem_get_policy,
4533 #endif
4534 };
4535
4536 int shmem_init_fs_context(struct fs_context *fc)
4537 {
4538         struct shmem_options *ctx;
4539
4540         ctx = kzalloc(sizeof(struct shmem_options), GFP_KERNEL);
4541         if (!ctx)
4542                 return -ENOMEM;
4543
4544         ctx->mode = 0777 | S_ISVTX;
4545         ctx->uid = current_fsuid();
4546         ctx->gid = current_fsgid();
4547
4548         fc->fs_private = ctx;
4549         fc->ops = &shmem_fs_context_ops;
4550         return 0;
4551 }
4552
4553 static struct file_system_type shmem_fs_type = {
4554         .owner          = THIS_MODULE,
4555         .name           = "tmpfs",
4556         .init_fs_context = shmem_init_fs_context,
4557 #ifdef CONFIG_TMPFS
4558         .parameters     = shmem_fs_parameters,
4559 #endif
4560         .kill_sb        = kill_litter_super,
4561 #ifdef CONFIG_SHMEM
4562         .fs_flags       = FS_USERNS_MOUNT | FS_ALLOW_IDMAP,
4563 #else
4564         .fs_flags       = FS_USERNS_MOUNT,
4565 #endif
4566 };
4567
4568 void __init shmem_init(void)
4569 {
4570         int error;
4571
4572         shmem_init_inodecache();
4573
4574 #ifdef CONFIG_TMPFS_QUOTA
4575         error = register_quota_format(&shmem_quota_format);
4576         if (error < 0) {
4577                 pr_err("Could not register quota format\n");
4578                 goto out3;
4579         }
4580 #endif
4581
4582         error = register_filesystem(&shmem_fs_type);
4583         if (error) {
4584                 pr_err("Could not register tmpfs\n");
4585                 goto out2;
4586         }
4587
4588         shm_mnt = kern_mount(&shmem_fs_type);
4589         if (IS_ERR(shm_mnt)) {
4590                 error = PTR_ERR(shm_mnt);
4591                 pr_err("Could not kern_mount tmpfs\n");
4592                 goto out1;
4593         }
4594
4595 #ifdef CONFIG_TRANSPARENT_HUGEPAGE
4596         if (has_transparent_hugepage() && shmem_huge > SHMEM_HUGE_DENY)
4597                 SHMEM_SB(shm_mnt->mnt_sb)->huge = shmem_huge;
4598         else
4599                 shmem_huge = SHMEM_HUGE_NEVER; /* just in case it was patched */
4600 #endif
4601         return;
4602
4603 out1:
4604         unregister_filesystem(&shmem_fs_type);
4605 out2:
4606 #ifdef CONFIG_TMPFS_QUOTA
4607         unregister_quota_format(&shmem_quota_format);
4608 out3:
4609 #endif
4610         shmem_destroy_inodecache();
4611         shm_mnt = ERR_PTR(error);
4612 }
4613
4614 #if defined(CONFIG_TRANSPARENT_HUGEPAGE) && defined(CONFIG_SYSFS)
4615 static ssize_t shmem_enabled_show(struct kobject *kobj,
4616                                   struct kobj_attribute *attr, char *buf)
4617 {
4618         static const int values[] = {
4619                 SHMEM_HUGE_ALWAYS,
4620                 SHMEM_HUGE_WITHIN_SIZE,
4621                 SHMEM_HUGE_ADVISE,
4622                 SHMEM_HUGE_NEVER,
4623                 SHMEM_HUGE_DENY,
4624                 SHMEM_HUGE_FORCE,
4625         };
4626         int len = 0;
4627         int i;
4628
4629         for (i = 0; i < ARRAY_SIZE(values); i++) {
4630                 len += sysfs_emit_at(buf, len,
4631                                      shmem_huge == values[i] ? "%s[%s]" : "%s%s",
4632                                      i ? " " : "",
4633                                      shmem_format_huge(values[i]));
4634         }
4635
4636         len += sysfs_emit_at(buf, len, "\n");
4637
4638         return len;
4639 }
4640
4641 static ssize_t shmem_enabled_store(struct kobject *kobj,
4642                 struct kobj_attribute *attr, const char *buf, size_t count)
4643 {
4644         char tmp[16];
4645         int huge;
4646
4647         if (count + 1 > sizeof(tmp))
4648                 return -EINVAL;
4649         memcpy(tmp, buf, count);
4650         tmp[count] = '\0';
4651         if (count && tmp[count - 1] == '\n')
4652                 tmp[count - 1] = '\0';
4653
4654         huge = shmem_parse_huge(tmp);
4655         if (huge == -EINVAL)
4656                 return -EINVAL;
4657         if (!has_transparent_hugepage() &&
4658                         huge != SHMEM_HUGE_NEVER && huge != SHMEM_HUGE_DENY)
4659                 return -EINVAL;
4660
4661         shmem_huge = huge;
4662         if (shmem_huge > SHMEM_HUGE_DENY)
4663                 SHMEM_SB(shm_mnt->mnt_sb)->huge = shmem_huge;
4664         return count;
4665 }
4666
4667 struct kobj_attribute shmem_enabled_attr = __ATTR_RW(shmem_enabled);
4668 #endif /* CONFIG_TRANSPARENT_HUGEPAGE && CONFIG_SYSFS */
4669
4670 #else /* !CONFIG_SHMEM */
4671
4672 /*
4673  * tiny-shmem: simple shmemfs and tmpfs using ramfs code
4674  *
4675  * This is intended for small system where the benefits of the full
4676  * shmem code (swap-backed and resource-limited) are outweighed by
4677  * their complexity. On systems without swap this code should be
4678  * effectively equivalent, but much lighter weight.
4679  */
4680
4681 static struct file_system_type shmem_fs_type = {
4682         .name           = "tmpfs",
4683         .init_fs_context = ramfs_init_fs_context,
4684         .parameters     = ramfs_fs_parameters,
4685         .kill_sb        = ramfs_kill_sb,
4686         .fs_flags       = FS_USERNS_MOUNT,
4687 };
4688
4689 void __init shmem_init(void)
4690 {
4691         BUG_ON(register_filesystem(&shmem_fs_type) != 0);
4692
4693         shm_mnt = kern_mount(&shmem_fs_type);
4694         BUG_ON(IS_ERR(shm_mnt));
4695 }
4696
4697 int shmem_unuse(unsigned int type)
4698 {
4699         return 0;
4700 }
4701
4702 int shmem_lock(struct file *file, int lock, struct ucounts *ucounts)
4703 {
4704         return 0;
4705 }
4706
4707 void shmem_unlock_mapping(struct address_space *mapping)
4708 {
4709 }
4710
4711 #ifdef CONFIG_MMU
4712 unsigned long shmem_get_unmapped_area(struct file *file,
4713                                       unsigned long addr, unsigned long len,
4714                                       unsigned long pgoff, unsigned long flags)
4715 {
4716         return current->mm->get_unmapped_area(file, addr, len, pgoff, flags);
4717 }
4718 #endif
4719
4720 void shmem_truncate_range(struct inode *inode, loff_t lstart, loff_t lend)
4721 {
4722         truncate_inode_pages_range(inode->i_mapping, lstart, lend);
4723 }
4724 EXPORT_SYMBOL_GPL(shmem_truncate_range);
4725
4726 #define shmem_vm_ops                            generic_file_vm_ops
4727 #define shmem_anon_vm_ops                       generic_file_vm_ops
4728 #define shmem_file_operations                   ramfs_file_operations
4729 #define shmem_acct_size(flags, size)            0
4730 #define shmem_unacct_size(flags, size)          do {} while (0)
4731
4732 static inline struct inode *shmem_get_inode(struct mnt_idmap *idmap, struct super_block *sb, struct inode *dir,
4733                                             umode_t mode, dev_t dev, unsigned long flags)
4734 {
4735         struct inode *inode = ramfs_get_inode(sb, dir, mode, dev);
4736         return inode ? inode : ERR_PTR(-ENOSPC);
4737 }
4738
4739 #endif /* CONFIG_SHMEM */
4740
4741 /* common code */
4742
4743 static struct file *__shmem_file_setup(struct vfsmount *mnt, const char *name, loff_t size,
4744                                        unsigned long flags, unsigned int i_flags)
4745 {
4746         struct inode *inode;
4747         struct file *res;
4748
4749         if (IS_ERR(mnt))
4750                 return ERR_CAST(mnt);
4751
4752         if (size < 0 || size > MAX_LFS_FILESIZE)
4753                 return ERR_PTR(-EINVAL);
4754
4755         if (shmem_acct_size(flags, size))
4756                 return ERR_PTR(-ENOMEM);
4757
4758         if (is_idmapped_mnt(mnt))
4759                 return ERR_PTR(-EINVAL);
4760
4761         inode = shmem_get_inode(&nop_mnt_idmap, mnt->mnt_sb, NULL,
4762                                 S_IFREG | S_IRWXUGO, 0, flags);
4763
4764         if (IS_ERR(inode)) {
4765                 shmem_unacct_size(flags, size);
4766                 return ERR_CAST(inode);
4767         }
4768         inode->i_flags |= i_flags;
4769         inode->i_size = size;
4770         clear_nlink(inode);     /* It is unlinked */
4771         res = ERR_PTR(ramfs_nommu_expand_for_mapping(inode, size));
4772         if (!IS_ERR(res))
4773                 res = alloc_file_pseudo(inode, mnt, name, O_RDWR,
4774                                 &shmem_file_operations);
4775         if (IS_ERR(res))
4776                 iput(inode);
4777         return res;
4778 }
4779
4780 /**
4781  * shmem_kernel_file_setup - get an unlinked file living in tmpfs which must be
4782  *      kernel internal.  There will be NO LSM permission checks against the
4783  *      underlying inode.  So users of this interface must do LSM checks at a
4784  *      higher layer.  The users are the big_key and shm implementations.  LSM
4785  *      checks are provided at the key or shm level rather than the inode.
4786  * @name: name for dentry (to be seen in /proc/<pid>/maps
4787  * @size: size to be set for the file
4788  * @flags: VM_NORESERVE suppresses pre-accounting of the entire object size
4789  */
4790 struct file *shmem_kernel_file_setup(const char *name, loff_t size, unsigned long flags)
4791 {
4792         return __shmem_file_setup(shm_mnt, name, size, flags, S_PRIVATE);
4793 }
4794
4795 /**
4796  * shmem_file_setup - get an unlinked file living in tmpfs
4797  * @name: name for dentry (to be seen in /proc/<pid>/maps
4798  * @size: size to be set for the file
4799  * @flags: VM_NORESERVE suppresses pre-accounting of the entire object size
4800  */
4801 struct file *shmem_file_setup(const char *name, loff_t size, unsigned long flags)
4802 {
4803         return __shmem_file_setup(shm_mnt, name, size, flags, 0);
4804 }
4805 EXPORT_SYMBOL_GPL(shmem_file_setup);
4806
4807 /**
4808  * shmem_file_setup_with_mnt - get an unlinked file living in tmpfs
4809  * @mnt: the tmpfs mount where the file will be created
4810  * @name: name for dentry (to be seen in /proc/<pid>/maps
4811  * @size: size to be set for the file
4812  * @flags: VM_NORESERVE suppresses pre-accounting of the entire object size
4813  */
4814 struct file *shmem_file_setup_with_mnt(struct vfsmount *mnt, const char *name,
4815                                        loff_t size, unsigned long flags)
4816 {
4817         return __shmem_file_setup(mnt, name, size, flags, 0);
4818 }
4819 EXPORT_SYMBOL_GPL(shmem_file_setup_with_mnt);
4820
4821 /**
4822  * shmem_zero_setup - setup a shared anonymous mapping
4823  * @vma: the vma to be mmapped is prepared by do_mmap
4824  */
4825 int shmem_zero_setup(struct vm_area_struct *vma)
4826 {
4827         struct file *file;
4828         loff_t size = vma->vm_end - vma->vm_start;
4829
4830         /*
4831          * Cloning a new file under mmap_lock leads to a lock ordering conflict
4832          * between XFS directory reading and selinux: since this file is only
4833          * accessible to the user through its mapping, use S_PRIVATE flag to
4834          * bypass file security, in the same way as shmem_kernel_file_setup().
4835          */
4836         file = shmem_kernel_file_setup("dev/zero", size, vma->vm_flags);
4837         if (IS_ERR(file))
4838                 return PTR_ERR(file);
4839
4840         if (vma->vm_file)
4841                 fput(vma->vm_file);
4842         vma->vm_file = file;
4843         vma->vm_ops = &shmem_anon_vm_ops;
4844
4845         return 0;
4846 }
4847
4848 /**
4849  * shmem_read_folio_gfp - read into page cache, using specified page allocation flags.
4850  * @mapping:    the folio's address_space
4851  * @index:      the folio index
4852  * @gfp:        the page allocator flags to use if allocating
4853  *
4854  * This behaves as a tmpfs "read_cache_page_gfp(mapping, index, gfp)",
4855  * with any new page allocations done using the specified allocation flags.
4856  * But read_cache_page_gfp() uses the ->read_folio() method: which does not
4857  * suit tmpfs, since it may have pages in swapcache, and needs to find those
4858  * for itself; although drivers/gpu/drm i915 and ttm rely upon this support.
4859  *
4860  * i915_gem_object_get_pages_gtt() mixes __GFP_NORETRY | __GFP_NOWARN in
4861  * with the mapping_gfp_mask(), to avoid OOMing the machine unnecessarily.
4862  */
4863 struct folio *shmem_read_folio_gfp(struct address_space *mapping,
4864                 pgoff_t index, gfp_t gfp)
4865 {
4866 #ifdef CONFIG_SHMEM
4867         struct inode *inode = mapping->host;
4868         struct folio *folio;
4869         int error;
4870
4871         BUG_ON(!shmem_mapping(mapping));
4872         error = shmem_get_folio_gfp(inode, index, &folio, SGP_CACHE,
4873                                   gfp, NULL, NULL, NULL);
4874         if (error)
4875                 return ERR_PTR(error);
4876
4877         folio_unlock(folio);
4878         return folio;
4879 #else
4880         /*
4881          * The tiny !SHMEM case uses ramfs without swap
4882          */
4883         return mapping_read_folio_gfp(mapping, index, gfp);
4884 #endif
4885 }
4886 EXPORT_SYMBOL_GPL(shmem_read_folio_gfp);
4887
4888 struct page *shmem_read_mapping_page_gfp(struct address_space *mapping,
4889                                          pgoff_t index, gfp_t gfp)
4890 {
4891         struct folio *folio = shmem_read_folio_gfp(mapping, index, gfp);
4892         struct page *page;
4893
4894         if (IS_ERR(folio))
4895                 return &folio->page;
4896
4897         page = folio_file_page(folio, index);
4898         if (PageHWPoison(page)) {
4899                 folio_put(folio);
4900                 return ERR_PTR(-EIO);
4901         }
4902
4903         return page;
4904 }
4905 EXPORT_SYMBOL_GPL(shmem_read_mapping_page_gfp);