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}