Loading...
1#include <api/fd/array.h>
2#include "util/debug.h"
3#include "tests/tests.h"
4
5static void fdarray__init_revents(struct fdarray *fda, short revents)
6{
7 int fd;
8
9 fda->nr = fda->nr_alloc;
10
11 for (fd = 0; fd < fda->nr; ++fd) {
12 fda->entries[fd].fd = fda->nr - fd;
13 fda->entries[fd].revents = revents;
14 }
15}
16
17static int fdarray__fprintf_prefix(struct fdarray *fda, const char *prefix, FILE *fp)
18{
19 int printed = 0;
20
21 if (!verbose)
22 return 0;
23
24 printed += fprintf(fp, "\n%s: ", prefix);
25 return printed + fdarray__fprintf(fda, fp);
26}
27
28int test__fdarray__filter(int subtest __maybe_unused)
29{
30 int nr_fds, expected_fd[2], fd, err = TEST_FAIL;
31 struct fdarray *fda = fdarray__new(5, 5);
32
33 if (fda == NULL) {
34 pr_debug("\nfdarray__new() failed!");
35 goto out;
36 }
37
38 fdarray__init_revents(fda, POLLIN);
39 nr_fds = fdarray__filter(fda, POLLHUP, NULL);
40 if (nr_fds != fda->nr_alloc) {
41 pr_debug("\nfdarray__filter()=%d != %d shouldn't have filtered anything",
42 nr_fds, fda->nr_alloc);
43 goto out_delete;
44 }
45
46 fdarray__init_revents(fda, POLLHUP);
47 nr_fds = fdarray__filter(fda, POLLHUP, NULL);
48 if (nr_fds != 0) {
49 pr_debug("\nfdarray__filter()=%d != %d, should have filtered all fds",
50 nr_fds, fda->nr_alloc);
51 goto out_delete;
52 }
53
54 fdarray__init_revents(fda, POLLHUP);
55 fda->entries[2].revents = POLLIN;
56 expected_fd[0] = fda->entries[2].fd;
57
58 pr_debug("\nfiltering all but fda->entries[2]:");
59 fdarray__fprintf_prefix(fda, "before", stderr);
60 nr_fds = fdarray__filter(fda, POLLHUP, NULL);
61 fdarray__fprintf_prefix(fda, " after", stderr);
62 if (nr_fds != 1) {
63 pr_debug("\nfdarray__filter()=%d != 1, should have left just one event", nr_fds);
64 goto out_delete;
65 }
66
67 if (fda->entries[0].fd != expected_fd[0]) {
68 pr_debug("\nfda->entries[0].fd=%d != %d\n",
69 fda->entries[0].fd, expected_fd[0]);
70 goto out_delete;
71 }
72
73 fdarray__init_revents(fda, POLLHUP);
74 fda->entries[0].revents = POLLIN;
75 expected_fd[0] = fda->entries[0].fd;
76 fda->entries[3].revents = POLLIN;
77 expected_fd[1] = fda->entries[3].fd;
78
79 pr_debug("\nfiltering all but (fda->entries[0], fda->entries[3]):");
80 fdarray__fprintf_prefix(fda, "before", stderr);
81 nr_fds = fdarray__filter(fda, POLLHUP, NULL);
82 fdarray__fprintf_prefix(fda, " after", stderr);
83 if (nr_fds != 2) {
84 pr_debug("\nfdarray__filter()=%d != 2, should have left just two events",
85 nr_fds);
86 goto out_delete;
87 }
88
89 for (fd = 0; fd < 2; ++fd) {
90 if (fda->entries[fd].fd != expected_fd[fd]) {
91 pr_debug("\nfda->entries[%d].fd=%d != %d\n", fd,
92 fda->entries[fd].fd, expected_fd[fd]);
93 goto out_delete;
94 }
95 }
96
97 pr_debug("\n");
98
99 err = 0;
100out_delete:
101 fdarray__delete(fda);
102out:
103 return err;
104}
105
106int test__fdarray__add(int subtest __maybe_unused)
107{
108 int err = TEST_FAIL;
109 struct fdarray *fda = fdarray__new(2, 2);
110
111 if (fda == NULL) {
112 pr_debug("\nfdarray__new() failed!");
113 goto out;
114 }
115
116#define FDA_CHECK(_idx, _fd, _revents) \
117 if (fda->entries[_idx].fd != _fd) { \
118 pr_debug("\n%d: fda->entries[%d](%d) != %d!", \
119 __LINE__, _idx, fda->entries[1].fd, _fd); \
120 goto out_delete; \
121 } \
122 if (fda->entries[_idx].events != (_revents)) { \
123 pr_debug("\n%d: fda->entries[%d].revents(%d) != %d!", \
124 __LINE__, _idx, fda->entries[_idx].fd, _revents); \
125 goto out_delete; \
126 }
127
128#define FDA_ADD(_idx, _fd, _revents, _nr) \
129 if (fdarray__add(fda, _fd, _revents) < 0) { \
130 pr_debug("\n%d: fdarray__add(fda, %d, %d) failed!", \
131 __LINE__,_fd, _revents); \
132 goto out_delete; \
133 } \
134 if (fda->nr != _nr) { \
135 pr_debug("\n%d: fdarray__add(fda, %d, %d)=%d != %d", \
136 __LINE__,_fd, _revents, fda->nr, _nr); \
137 goto out_delete; \
138 } \
139 FDA_CHECK(_idx, _fd, _revents)
140
141 FDA_ADD(0, 1, POLLIN, 1);
142 FDA_ADD(1, 2, POLLERR, 2);
143
144 fdarray__fprintf_prefix(fda, "before growing array", stderr);
145
146 FDA_ADD(2, 35, POLLHUP, 3);
147
148 if (fda->entries == NULL) {
149 pr_debug("\nfdarray__add(fda, 35, POLLHUP) should have allocated fda->pollfd!");
150 goto out_delete;
151 }
152
153 fdarray__fprintf_prefix(fda, "after 3rd add", stderr);
154
155 FDA_ADD(3, 88, POLLIN | POLLOUT, 4);
156
157 fdarray__fprintf_prefix(fda, "after 4th add", stderr);
158
159 FDA_CHECK(0, 1, POLLIN);
160 FDA_CHECK(1, 2, POLLERR);
161 FDA_CHECK(2, 35, POLLHUP);
162 FDA_CHECK(3, 88, POLLIN | POLLOUT);
163
164#undef FDA_ADD
165#undef FDA_CHECK
166
167 pr_debug("\n");
168
169 err = 0;
170out_delete:
171 fdarray__delete(fda);
172out:
173 return err;
174}
1// SPDX-License-Identifier: GPL-2.0
2#include <api/fd/array.h>
3#include <poll.h>
4#include "util/debug.h"
5#include "tests/tests.h"
6
7static void fdarray__init_revents(struct fdarray *fda, short revents)
8{
9 int fd;
10
11 fda->nr = fda->nr_alloc;
12
13 for (fd = 0; fd < fda->nr; ++fd) {
14 fda->entries[fd].fd = fda->nr - fd;
15 fda->entries[fd].events = revents;
16 fda->entries[fd].revents = revents;
17 }
18}
19
20static int fdarray__fprintf_prefix(struct fdarray *fda, const char *prefix, FILE *fp)
21{
22 int printed = 0;
23
24 if (verbose <= 0)
25 return 0;
26
27 printed += fprintf(fp, "\n%s: ", prefix);
28 return printed + fdarray__fprintf(fda, fp);
29}
30
31static int test__fdarray__filter(struct test_suite *test __maybe_unused, int subtest __maybe_unused)
32{
33 int nr_fds, err = TEST_FAIL;
34 struct fdarray *fda = fdarray__new(5, 5);
35
36 if (fda == NULL) {
37 pr_debug("\nfdarray__new() failed!");
38 goto out;
39 }
40
41 fdarray__init_revents(fda, POLLIN);
42 nr_fds = fdarray__filter(fda, POLLHUP, NULL, NULL);
43 if (nr_fds != fda->nr_alloc) {
44 pr_debug("\nfdarray__filter()=%d != %d shouldn't have filtered anything",
45 nr_fds, fda->nr_alloc);
46 goto out_delete;
47 }
48
49 fdarray__init_revents(fda, POLLHUP);
50 nr_fds = fdarray__filter(fda, POLLHUP, NULL, NULL);
51 if (nr_fds != 0) {
52 pr_debug("\nfdarray__filter()=%d != %d, should have filtered all fds",
53 nr_fds, fda->nr_alloc);
54 goto out_delete;
55 }
56
57 fdarray__init_revents(fda, POLLHUP);
58 fda->entries[2].revents = POLLIN;
59
60 pr_debug("\nfiltering all but fda->entries[2]:");
61 fdarray__fprintf_prefix(fda, "before", stderr);
62 nr_fds = fdarray__filter(fda, POLLHUP, NULL, NULL);
63 fdarray__fprintf_prefix(fda, " after", stderr);
64 if (nr_fds != 1) {
65 pr_debug("\nfdarray__filter()=%d != 1, should have left just one event", nr_fds);
66 goto out_delete;
67 }
68
69 fdarray__init_revents(fda, POLLHUP);
70 fda->entries[0].revents = POLLIN;
71 fda->entries[3].revents = POLLIN;
72
73 pr_debug("\nfiltering all but (fda->entries[0], fda->entries[3]):");
74 fdarray__fprintf_prefix(fda, "before", stderr);
75 nr_fds = fdarray__filter(fda, POLLHUP, NULL, NULL);
76 fdarray__fprintf_prefix(fda, " after", stderr);
77 if (nr_fds != 2) {
78 pr_debug("\nfdarray__filter()=%d != 2, should have left just two events",
79 nr_fds);
80 goto out_delete;
81 }
82
83 pr_debug("\n");
84
85 err = 0;
86out_delete:
87 fdarray__delete(fda);
88out:
89 return err;
90}
91
92static int test__fdarray__add(struct test_suite *test __maybe_unused, int subtest __maybe_unused)
93{
94 int err = TEST_FAIL;
95 struct fdarray *fda = fdarray__new(2, 2);
96
97 if (fda == NULL) {
98 pr_debug("\nfdarray__new() failed!");
99 goto out;
100 }
101
102#define FDA_CHECK(_idx, _fd, _revents) \
103 if (fda->entries[_idx].fd != _fd) { \
104 pr_debug("\n%d: fda->entries[%d](%d) != %d!", \
105 __LINE__, _idx, fda->entries[1].fd, _fd); \
106 goto out_delete; \
107 } \
108 if (fda->entries[_idx].events != (_revents)) { \
109 pr_debug("\n%d: fda->entries[%d].revents(%d) != %d!", \
110 __LINE__, _idx, fda->entries[_idx].fd, _revents); \
111 goto out_delete; \
112 }
113
114#define FDA_ADD(_idx, _fd, _revents, _nr) \
115 if (fdarray__add(fda, _fd, _revents, fdarray_flag__default) < 0) { \
116 pr_debug("\n%d: fdarray__add(fda, %d, %d) failed!", \
117 __LINE__,_fd, _revents); \
118 goto out_delete; \
119 } \
120 if (fda->nr != _nr) { \
121 pr_debug("\n%d: fdarray__add(fda, %d, %d)=%d != %d", \
122 __LINE__,_fd, _revents, fda->nr, _nr); \
123 goto out_delete; \
124 } \
125 FDA_CHECK(_idx, _fd, _revents)
126
127 FDA_ADD(0, 1, POLLIN, 1);
128 FDA_ADD(1, 2, POLLERR, 2);
129
130 fdarray__fprintf_prefix(fda, "before growing array", stderr);
131
132 FDA_ADD(2, 35, POLLHUP, 3);
133
134 if (fda->entries == NULL) {
135 pr_debug("\nfdarray__add(fda, 35, POLLHUP) should have allocated fda->pollfd!");
136 goto out_delete;
137 }
138
139 fdarray__fprintf_prefix(fda, "after 3rd add", stderr);
140
141 FDA_ADD(3, 88, POLLIN | POLLOUT, 4);
142
143 fdarray__fprintf_prefix(fda, "after 4th add", stderr);
144
145 FDA_CHECK(0, 1, POLLIN);
146 FDA_CHECK(1, 2, POLLERR);
147 FDA_CHECK(2, 35, POLLHUP);
148 FDA_CHECK(3, 88, POLLIN | POLLOUT);
149
150#undef FDA_ADD
151#undef FDA_CHECK
152
153 pr_debug("\n");
154
155 err = 0;
156out_delete:
157 fdarray__delete(fda);
158out:
159 return err;
160}
161
162DEFINE_SUITE("Filter fds with revents mask in a fdarray", fdarray__filter);
163DEFINE_SUITE("Add fd to a fdarray, making it autogrow", fdarray__add);