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