projects/linux·blob master · refs

141 lines · 4.3 KiB

1// SPDX-License-Identifier: GPL-2.0-or-later
2
3#pragma once
4
5#include <stdbool.h>
6#include <stdio.h>
7#include <stdlib.h>
8
9#include "generated/bit-length.h"
10#include "maple-shared.h"
11#include "vma_internal.h"
12#include "../../../mm/vma.h"
13
14/* Simple test runner. Assumes local num_[fail, tests] counters. */
15#define TEST(name) \
16 do { \
17 (*num_tests)++; \
18 if (!test_##name()) { \
19 (*num_fail)++; \
20 fprintf(stderr, "Test " #name " FAILED\n"); \
21 } \
22 } while (0)
23
24#define __ASSERT_TRUE(_expr, _fmt, ...) \
25 do { \
26 if (!(_expr)) { \
27 fprintf(stderr, \
28 "Assert FAILED at %s:%d:%s(): %s is FALSE" \
29 _fmt ".\n", \
30 __FILE__, __LINE__, __FUNCTION__, #_expr \
31 __VA_OPT__(,) __VA_ARGS__); \
32 return false; \
33 } \
34 } while (0)
35
36#define __TO_SCALAR(x) ((unsigned long long)(uintptr_t)(x))
37
38#define ASSERT_TRUE(_expr) __ASSERT_TRUE(_expr, "")
39#define ASSERT_FALSE(_expr) ASSERT_TRUE(!(_expr))
40#define ASSERT_EQ(_val1, _val2) do { \
41 __typeof__(_val1) __val1 = (_val1); \
42 __typeof__(_val2) __val2 = (_val2); \
43 __ASSERT_TRUE(__val1 == __val2, " (0x%llx != 0x%llx)", \
44 __TO_SCALAR(__val1), __TO_SCALAR(__val2)); \
45 } while (0)
46
47#define ASSERT_NE(_val1, _val2) do { \
48 __typeof__(_val1) __val1 = (_val1); \
49 __typeof__(_val2) __val2 = (_val2); \
50 __ASSERT_TRUE(__val1 != __val2, " (0x%llx == 0x%llx)", \
51 __TO_SCALAR(__val1), __TO_SCALAR(__val2)); \
52 } while (0)
53
54#define ASSERT_FLAGS_SAME_MASK(_flags, _flags_other) \
55 ASSERT_TRUE(vma_flags_same_mask((_flags), (_flags_other)))
56
57#define ASSERT_FLAGS_NOT_SAME_MASK(_flags, _flags_other) \
58 ASSERT_FALSE(vma_flags_same_mask((_flags), (_flags_other)))
59
60#define ASSERT_FLAGS_SAME(_flags, ...) \
61 ASSERT_TRUE(vma_flags_same(_flags, __VA_ARGS__))
62
63#define ASSERT_FLAGS_NOT_SAME(_flags, ...) \
64 ASSERT_FALSE(vma_flags_same(_flags, __VA_ARGS__))
65
66#define ASSERT_FLAGS_EMPTY(_flags) \
67 ASSERT_TRUE(vma_flags_empty(_flags))
68
69#define ASSERT_FLAGS_NONEMPTY(_flags) \
70 ASSERT_FALSE(vma_flags_empty(_flags))
71
72extern bool fail_prealloc;
73
74/* Override vma_iter_prealloc() so we can choose to fail it. */
75#define vma_iter_prealloc(vmi, vma) \
76 (fail_prealloc ? -ENOMEM : mas_preallocate(&(vmi)->mas, (vma), GFP_KERNEL))
77
78#define CONFIG_DEFAULT_MMAP_MIN_ADDR 65536
79
80extern unsigned long mmap_min_addr;
81extern unsigned long dac_mmap_min_addr;
82extern unsigned long stack_guard_gap;
83
84extern const struct vm_operations_struct vma_dummy_vm_ops;
85extern struct anon_vma dummy_anon_vma;
86extern struct task_struct __current;
87
88/*
89 * Helper function which provides a wrapper around a merge existing VMA
90 * operation.
91 *
92 * Declared in main.c as uses static VMA function.
93 */
94struct vm_area_struct *merge_existing(struct vma_merge_struct *vmg);
95
96/*
97 * Helper function to allocate a VMA and link it to the tree.
98 *
99 * Declared in main.c as uses static VMA function.
100 */
101int attach_vma(struct mm_struct *mm, struct vm_area_struct *vma);
102
103/* Helper function providing a dummy vm_ops->close() method.*/
104static inline void dummy_close(struct vm_area_struct *)
105{
106}
107
108/* Helper function to simply allocate a VMA. */
109struct vm_area_struct *alloc_vma(struct mm_struct *mm,
110 unsigned long start, unsigned long end,
111 pgoff_t pgoff, vma_flags_t vma_flags);
112
113/* Helper function to detach and free a VMA. */
114void detach_free_vma(struct vm_area_struct *vma);
115
116/* Helper function to allocate a VMA and link it to the tree. */
117struct vm_area_struct *alloc_and_link_vma(struct mm_struct *mm,
118 unsigned long start, unsigned long end,
119 pgoff_t pgoff, vma_flags_t vma_flags);
120
121/*
122 * Helper function to reset the dummy anon_vma to indicate it has not been
123 * duplicated.
124 */
125void reset_dummy_anon_vma(void);
126
127/*
128 * Helper function to remove all VMAs and destroy the maple tree associated with
129 * a virtual address space. Returns a count of VMAs in the tree.
130 */
131int cleanup_mm(struct mm_struct *mm, struct vma_iterator *vmi);
132
133/* Helper function to determine if VMA has had vma_start_write() performed. */
134bool vma_write_started(struct vm_area_struct *vma);
135
136void __vma_set_dummy_anon_vma(struct vm_area_struct *vma,
137 struct anon_vma_chain *avc, struct anon_vma *anon_vma);
138
139/* Provide a simple dummy VMA/anon_vma dummy setup for testing. */
140void vma_set_dummy_anon_vma(struct vm_area_struct *vma,
141 struct anon_vma_chain *avc);