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