0001
0002 #include <api/fd/array.h>
0003 #include <poll.h>
0004 #include "util/debug.h"
0005 #include "tests/tests.h"
0006
0007 static void fdarray__init_revents(struct fdarray *fda, short revents)
0008 {
0009 int fd;
0010
0011 fda->nr = fda->nr_alloc;
0012
0013 for (fd = 0; fd < fda->nr; ++fd) {
0014 fda->entries[fd].fd = fda->nr - fd;
0015 fda->entries[fd].events = revents;
0016 fda->entries[fd].revents = revents;
0017 }
0018 }
0019
0020 static int fdarray__fprintf_prefix(struct fdarray *fda, const char *prefix, FILE *fp)
0021 {
0022 int printed = 0;
0023
0024 if (verbose <= 0)
0025 return 0;
0026
0027 printed += fprintf(fp, "\n%s: ", prefix);
0028 return printed + fdarray__fprintf(fda, fp);
0029 }
0030
0031 static int test__fdarray__filter(struct test_suite *test __maybe_unused, int subtest __maybe_unused)
0032 {
0033 int nr_fds, err = TEST_FAIL;
0034 struct fdarray *fda = fdarray__new(5, 5);
0035
0036 if (fda == NULL) {
0037 pr_debug("\nfdarray__new() failed!");
0038 goto out;
0039 }
0040
0041 fdarray__init_revents(fda, POLLIN);
0042 nr_fds = fdarray__filter(fda, POLLHUP, NULL, NULL);
0043 if (nr_fds != fda->nr_alloc) {
0044 pr_debug("\nfdarray__filter()=%d != %d shouldn't have filtered anything",
0045 nr_fds, fda->nr_alloc);
0046 goto out_delete;
0047 }
0048
0049 fdarray__init_revents(fda, POLLHUP);
0050 nr_fds = fdarray__filter(fda, POLLHUP, NULL, NULL);
0051 if (nr_fds != 0) {
0052 pr_debug("\nfdarray__filter()=%d != %d, should have filtered all fds",
0053 nr_fds, fda->nr_alloc);
0054 goto out_delete;
0055 }
0056
0057 fdarray__init_revents(fda, POLLHUP);
0058 fda->entries[2].revents = POLLIN;
0059
0060 pr_debug("\nfiltering all but fda->entries[2]:");
0061 fdarray__fprintf_prefix(fda, "before", stderr);
0062 nr_fds = fdarray__filter(fda, POLLHUP, NULL, NULL);
0063 fdarray__fprintf_prefix(fda, " after", stderr);
0064 if (nr_fds != 1) {
0065 pr_debug("\nfdarray__filter()=%d != 1, should have left just one event", nr_fds);
0066 goto out_delete;
0067 }
0068
0069 fdarray__init_revents(fda, POLLHUP);
0070 fda->entries[0].revents = POLLIN;
0071 fda->entries[3].revents = POLLIN;
0072
0073 pr_debug("\nfiltering all but (fda->entries[0], fda->entries[3]):");
0074 fdarray__fprintf_prefix(fda, "before", stderr);
0075 nr_fds = fdarray__filter(fda, POLLHUP, NULL, NULL);
0076 fdarray__fprintf_prefix(fda, " after", stderr);
0077 if (nr_fds != 2) {
0078 pr_debug("\nfdarray__filter()=%d != 2, should have left just two events",
0079 nr_fds);
0080 goto out_delete;
0081 }
0082
0083 pr_debug("\n");
0084
0085 err = 0;
0086 out_delete:
0087 fdarray__delete(fda);
0088 out:
0089 return err;
0090 }
0091
0092 static int test__fdarray__add(struct test_suite *test __maybe_unused, int subtest __maybe_unused)
0093 {
0094 int err = TEST_FAIL;
0095 struct fdarray *fda = fdarray__new(2, 2);
0096
0097 if (fda == NULL) {
0098 pr_debug("\nfdarray__new() failed!");
0099 goto out;
0100 }
0101
0102 #define FDA_CHECK(_idx, _fd, _revents) \
0103 if (fda->entries[_idx].fd != _fd) { \
0104 pr_debug("\n%d: fda->entries[%d](%d) != %d!", \
0105 __LINE__, _idx, fda->entries[1].fd, _fd); \
0106 goto out_delete; \
0107 } \
0108 if (fda->entries[_idx].events != (_revents)) { \
0109 pr_debug("\n%d: fda->entries[%d].revents(%d) != %d!", \
0110 __LINE__, _idx, fda->entries[_idx].fd, _revents); \
0111 goto out_delete; \
0112 }
0113
0114 #define FDA_ADD(_idx, _fd, _revents, _nr) \
0115 if (fdarray__add(fda, _fd, _revents, fdarray_flag__default) < 0) { \
0116 pr_debug("\n%d: fdarray__add(fda, %d, %d) failed!", \
0117 __LINE__,_fd, _revents); \
0118 goto out_delete; \
0119 } \
0120 if (fda->nr != _nr) { \
0121 pr_debug("\n%d: fdarray__add(fda, %d, %d)=%d != %d", \
0122 __LINE__,_fd, _revents, fda->nr, _nr); \
0123 goto out_delete; \
0124 } \
0125 FDA_CHECK(_idx, _fd, _revents)
0126
0127 FDA_ADD(0, 1, POLLIN, 1);
0128 FDA_ADD(1, 2, POLLERR, 2);
0129
0130 fdarray__fprintf_prefix(fda, "before growing array", stderr);
0131
0132 FDA_ADD(2, 35, POLLHUP, 3);
0133
0134 if (fda->entries == NULL) {
0135 pr_debug("\nfdarray__add(fda, 35, POLLHUP) should have allocated fda->pollfd!");
0136 goto out_delete;
0137 }
0138
0139 fdarray__fprintf_prefix(fda, "after 3rd add", stderr);
0140
0141 FDA_ADD(3, 88, POLLIN | POLLOUT, 4);
0142
0143 fdarray__fprintf_prefix(fda, "after 4th add", stderr);
0144
0145 FDA_CHECK(0, 1, POLLIN);
0146 FDA_CHECK(1, 2, POLLERR);
0147 FDA_CHECK(2, 35, POLLHUP);
0148 FDA_CHECK(3, 88, POLLIN | POLLOUT);
0149
0150 #undef FDA_ADD
0151 #undef FDA_CHECK
0152
0153 pr_debug("\n");
0154
0155 err = 0;
0156 out_delete:
0157 fdarray__delete(fda);
0158 out:
0159 return err;
0160 }
0161
0162 DEFINE_SUITE("Filter fds with revents mask in a fdarray", fdarray__filter);
0163 DEFINE_SUITE("Add fd to a fdarray, making it autogrow", fdarray__add);
