#include <linux/sched.h>
 #include <linux/sched/mm.h>
 #include <linux/mm_inline.h>
+#include <linux/mmu_context.h>
 #include <linux/string.h>
 #include <linux/uio.h>
 #include <linux/ksm.h>
                               &guard_remove_walk_ops, NULL);
 }
 
+#ifdef CONFIG_64BIT
+/* Does the madvise operation result in discarding of mapped data? */
+static bool is_discard(int behavior)
+{
+       switch (behavior) {
+       case MADV_FREE:
+       case MADV_DONTNEED:
+       case MADV_DONTNEED_LOCKED:
+       case MADV_REMOVE:
+       case MADV_DONTFORK:
+       case MADV_WIPEONFORK:
+       case MADV_GUARD_INSTALL:
+               return true;
+       }
+
+       return false;
+}
+
+/*
+ * We are restricted from madvise()'ing mseal()'d VMAs only in very particular
+ * circumstances - discarding of data from read-only anonymous SEALED mappings.
+ *
+ * This is because users cannot trivally discard data from these VMAs, and may
+ * only do so via an appropriate madvise() call.
+ */
+static bool can_madvise_modify(struct madvise_behavior *madv_behavior)
+{
+       struct vm_area_struct *vma = madv_behavior->vma;
+
+       /* If the VMA isn't sealed we're good. */
+       if (can_modify_vma(vma))
+               return true;
+
+       /* For a sealed VMA, we only care about discard operations. */
+       if (!is_discard(madv_behavior->behavior))
+               return true;
+
+       /*
+        * We explicitly permit all file-backed mappings, whether MAP_SHARED or
+        * MAP_PRIVATE.
+        *
+        * The latter causes some complications. Because now, one can mmap()
+        * read/write a MAP_PRIVATE mapping, write to it, then mprotect()
+        * read-only, mseal() and a discard will be permitted.
+        *
+        * However, in order to avoid issues with potential use of madvise(...,
+        * MADV_DONTNEED) of mseal()'d .text mappings we, for the time being,
+        * permit this.
+        */
+       if (!vma_is_anonymous(vma))
+               return true;
+
+       /* If the user could write to the mapping anyway, then this is fine. */
+       if ((vma->vm_flags & VM_WRITE) &&
+           arch_vma_access_permitted(vma, /* write= */ true,
+                       /* execute= */ false, /* foreign= */ false))
+               return true;
+
+       /* Otherwise, we are not permitted to perform this operation. */
+       return false;
+}
+#else
+static bool can_madvise_modify(struct madvise_behavior *madv_behavior)
+{
+       return true;
+}
+#endif
+
 /*
  * Apply an madvise behavior to a region of a vma.  madvise_update_vma
  * will handle splitting a vm area into separate areas, each area with its own
        struct madvise_behavior_range *range = &madv_behavior->range;
        int error;
 
-       if (unlikely(!can_modify_vma_madv(madv_behavior->vma, behavior)))
+       if (unlikely(!can_madvise_modify(madv_behavior)))
                return -EPERM;
 
        switch (behavior) {
 
 #include <linux/mman.h>
 #include <linux/mm.h>
 #include <linux/mm_inline.h>
-#include <linux/mmu_context.h>
 #include <linux/syscalls.h>
 #include <linux/sched.h>
 #include "internal.h"
        vm_flags_set(vma, VM_SEALED);
 }
 
-static bool is_madv_discard(int behavior)
-{
-       switch (behavior) {
-       case MADV_FREE:
-       case MADV_DONTNEED:
-       case MADV_DONTNEED_LOCKED:
-       case MADV_REMOVE:
-       case MADV_DONTFORK:
-       case MADV_WIPEONFORK:
-       case MADV_GUARD_INSTALL:
-               return true;
-       }
-
-       return false;
-}
-
-static bool is_ro_anon(struct vm_area_struct *vma)
-{
-       /* check anonymous mapping. */
-       if (vma->vm_file || vma->vm_flags & VM_SHARED)
-               return false;
-
-       /*
-        * check for non-writable:
-        * PROT=RO or PKRU is not writeable.
-        */
-       if (!(vma->vm_flags & VM_WRITE) ||
-               !arch_vma_access_permitted(vma, true, false, false))
-               return true;
-
-       return false;
-}
-
-/*
- * Check if a vma is allowed to be modified by madvise.
- */
-bool can_modify_vma_madv(struct vm_area_struct *vma, int behavior)
-{
-       if (!is_madv_discard(behavior))
-               return true;
-
-       if (unlikely(!can_modify_vma(vma) && is_ro_anon(vma)))
-               return false;
-
-       /* Allow by default. */
-       return true;
-}
-
 static int mseal_fixup(struct vma_iterator *vmi, struct vm_area_struct *vma,
                struct vm_area_struct **prev, unsigned long start,
                unsigned long end, vm_flags_t newflags)