Loading...
1#include <stdio.h>
2#include <assert.h>
3
4#include <linux/scatterlist.h>
5
6#define MAX_PAGES (64)
7
8struct test {
9 int alloc_ret;
10 unsigned num_pages;
11 unsigned *pfn;
12 unsigned *pfn_app;
13 unsigned size;
14 unsigned int max_seg;
15 unsigned int expected_segments;
16};
17
18static void set_pages(struct page **pages, const unsigned *array, unsigned num)
19{
20 unsigned int i;
21
22 assert(num < MAX_PAGES);
23 for (i = 0; i < num; i++)
24 pages[i] = (struct page *)(unsigned long)
25 ((1 + array[i]) * PAGE_SIZE);
26}
27
28#define pfn(...) (unsigned []){ __VA_ARGS__ }
29
30static void fail(struct test *test, struct sg_table *st, const char *cond)
31{
32 unsigned int i;
33
34 fprintf(stderr, "Failed on '%s'!\n\n", cond);
35
36 printf("size = %u, max segment = %u, expected nents = %u\nst->nents = %u, st->orig_nents= %u\n",
37 test->size, test->max_seg, test->expected_segments, st->nents,
38 st->orig_nents);
39
40 printf("%u input PFNs:", test->num_pages);
41 for (i = 0; i < test->num_pages; i++)
42 printf(" %x", test->pfn[i]);
43 printf("\n");
44
45 exit(1);
46}
47
48#define VALIDATE(cond, st, test) \
49 if (!(cond)) \
50 fail((test), (st), #cond);
51
52int main(void)
53{
54 const unsigned int sgmax = UINT_MAX;
55 struct test *test, tests[] = {
56 { -EINVAL, 1, pfn(0), NULL, PAGE_SIZE, 0, 1 },
57 { 0, 1, pfn(0), NULL, PAGE_SIZE, PAGE_SIZE + 1, 1 },
58 { 0, 1, pfn(0), NULL, PAGE_SIZE, sgmax, 1 },
59 { 0, 1, pfn(0), NULL, 1, sgmax, 1 },
60 { 0, 2, pfn(0, 1), NULL, 2 * PAGE_SIZE, sgmax, 1 },
61 { 0, 2, pfn(1, 0), NULL, 2 * PAGE_SIZE, sgmax, 2 },
62 { 0, 3, pfn(0, 1, 2), NULL, 3 * PAGE_SIZE, sgmax, 1 },
63 { 0, 3, pfn(0, 1, 2), NULL, 3 * PAGE_SIZE, sgmax, 1 },
64 { 0, 3, pfn(0, 1, 2), pfn(3, 4, 5), 3 * PAGE_SIZE, sgmax, 1 },
65 { 0, 3, pfn(0, 1, 2), pfn(4, 5, 6), 3 * PAGE_SIZE, sgmax, 2 },
66 { 0, 3, pfn(0, 2, 1), NULL, 3 * PAGE_SIZE, sgmax, 3 },
67 { 0, 3, pfn(0, 1, 3), NULL, 3 * PAGE_SIZE, sgmax, 2 },
68 { 0, 3, pfn(1, 2, 4), NULL, 3 * PAGE_SIZE, sgmax, 2 },
69 { 0, 3, pfn(1, 3, 4), NULL, 3 * PAGE_SIZE, sgmax, 2 },
70 { 0, 4, pfn(0, 1, 3, 4), NULL, 4 * PAGE_SIZE, sgmax, 2 },
71 { 0, 5, pfn(0, 1, 3, 4, 5), NULL, 5 * PAGE_SIZE, sgmax, 2 },
72 { 0, 5, pfn(0, 1, 3, 4, 6), NULL, 5 * PAGE_SIZE, sgmax, 3 },
73 { 0, 5, pfn(0, 1, 2, 3, 4), NULL, 5 * PAGE_SIZE, sgmax, 1 },
74 { 0, 5, pfn(0, 1, 2, 3, 4), NULL, 5 * PAGE_SIZE, 2 * PAGE_SIZE,
75 3 },
76 { 0, 6, pfn(0, 1, 2, 3, 4, 5), NULL, 6 * PAGE_SIZE,
77 2 * PAGE_SIZE, 3 },
78 { 0, 6, pfn(0, 2, 3, 4, 5, 6), NULL, 6 * PAGE_SIZE,
79 2 * PAGE_SIZE, 4 },
80 { 0, 6, pfn(0, 1, 3, 4, 5, 6), pfn(7, 8, 9, 10, 11, 12),
81 6 * PAGE_SIZE, 12 * PAGE_SIZE, 2 },
82 { 0, 0, NULL, NULL, 0, 0, 0 },
83 };
84 unsigned int i;
85
86 for (i = 0, test = tests; test->expected_segments; test++, i++) {
87 int left_pages = test->pfn_app ? test->num_pages : 0;
88 struct sg_append_table append = {};
89 struct page *pages[MAX_PAGES];
90 int ret;
91
92 set_pages(pages, test->pfn, test->num_pages);
93
94 if (test->pfn_app)
95 ret = sg_alloc_append_table_from_pages(
96 &append, pages, test->num_pages, 0, test->size,
97 test->max_seg, left_pages, GFP_KERNEL);
98 else
99 ret = sg_alloc_table_from_pages_segment(
100 &append.sgt, pages, test->num_pages, 0,
101 test->size, test->max_seg, GFP_KERNEL);
102
103 assert(ret == test->alloc_ret);
104
105 if (test->alloc_ret)
106 continue;
107
108 if (test->pfn_app) {
109 set_pages(pages, test->pfn_app, test->num_pages);
110 ret = sg_alloc_append_table_from_pages(
111 &append, pages, test->num_pages, 0, test->size,
112 test->max_seg, 0, GFP_KERNEL);
113
114 assert(ret == test->alloc_ret);
115 }
116
117 VALIDATE(append.sgt.nents == test->expected_segments,
118 &append.sgt, test);
119 if (!test->pfn_app)
120 VALIDATE(append.sgt.orig_nents ==
121 test->expected_segments,
122 &append.sgt, test);
123
124 if (test->pfn_app)
125 sg_free_append_table(&append);
126 else
127 sg_free_table(&append.sgt);
128 }
129
130 assert(i == (sizeof(tests) / sizeof(tests[0])) - 1);
131
132 return 0;
133}
1#include <stdio.h>
2#include <assert.h>
3
4#include <linux/scatterlist.h>
5
6#define MAX_PAGES (64)
7
8static void set_pages(struct page **pages, const unsigned *array, unsigned num)
9{
10 unsigned int i;
11
12 assert(num < MAX_PAGES);
13 for (i = 0; i < num; i++)
14 pages[i] = (struct page *)(unsigned long)
15 ((1 + array[i]) * PAGE_SIZE);
16}
17
18#define pfn(...) (unsigned []){ __VA_ARGS__ }
19
20int main(void)
21{
22 const unsigned int sgmax = SCATTERLIST_MAX_SEGMENT;
23 struct test {
24 int alloc_ret;
25 unsigned num_pages;
26 unsigned *pfn;
27 unsigned size;
28 unsigned int max_seg;
29 unsigned int expected_segments;
30 } *test, tests[] = {
31 { -EINVAL, 1, pfn(0), PAGE_SIZE, PAGE_SIZE + 1, 1 },
32 { -EINVAL, 1, pfn(0), PAGE_SIZE, 0, 1 },
33 { -EINVAL, 1, pfn(0), PAGE_SIZE, sgmax + 1, 1 },
34 { 0, 1, pfn(0), PAGE_SIZE, sgmax, 1 },
35 { 0, 1, pfn(0), 1, sgmax, 1 },
36 { 0, 2, pfn(0, 1), 2 * PAGE_SIZE, sgmax, 1 },
37 { 0, 2, pfn(1, 0), 2 * PAGE_SIZE, sgmax, 2 },
38 { 0, 3, pfn(0, 1, 2), 3 * PAGE_SIZE, sgmax, 1 },
39 { 0, 3, pfn(0, 2, 1), 3 * PAGE_SIZE, sgmax, 3 },
40 { 0, 3, pfn(0, 1, 3), 3 * PAGE_SIZE, sgmax, 2 },
41 { 0, 3, pfn(1, 2, 4), 3 * PAGE_SIZE, sgmax, 2 },
42 { 0, 3, pfn(1, 3, 4), 3 * PAGE_SIZE, sgmax, 2 },
43 { 0, 4, pfn(0, 1, 3, 4), 4 * PAGE_SIZE, sgmax, 2 },
44 { 0, 5, pfn(0, 1, 3, 4, 5), 5 * PAGE_SIZE, sgmax, 2 },
45 { 0, 5, pfn(0, 1, 3, 4, 6), 5 * PAGE_SIZE, sgmax, 3 },
46 { 0, 5, pfn(0, 1, 2, 3, 4), 5 * PAGE_SIZE, sgmax, 1 },
47 { 0, 5, pfn(0, 1, 2, 3, 4), 5 * PAGE_SIZE, 2 * PAGE_SIZE, 3 },
48 { 0, 6, pfn(0, 1, 2, 3, 4, 5), 6 * PAGE_SIZE, 2 * PAGE_SIZE, 3 },
49 { 0, 6, pfn(0, 2, 3, 4, 5, 6), 6 * PAGE_SIZE, 2 * PAGE_SIZE, 4 },
50 { 0, 6, pfn(0, 1, 3, 4, 5, 6), 6 * PAGE_SIZE, 2 * PAGE_SIZE, 3 },
51 { 0, 0, NULL, 0, 0, 0 },
52 };
53 unsigned int i;
54
55 for (i = 0, test = tests; test->expected_segments; test++, i++) {
56 struct page *pages[MAX_PAGES];
57 struct sg_table st;
58 int ret;
59
60 set_pages(pages, test->pfn, test->num_pages);
61
62 ret = __sg_alloc_table_from_pages(&st, pages, test->num_pages,
63 0, test->size, test->max_seg,
64 GFP_KERNEL);
65 assert(ret == test->alloc_ret);
66
67 if (test->alloc_ret)
68 continue;
69
70 assert(st.nents == test->expected_segments);
71 assert(st.orig_nents == test->expected_segments);
72
73 sg_free_table(&st);
74 }
75
76 assert(i == (sizeof(tests) / sizeof(tests[0])) - 1);
77
78 return 0;
79}