fdarray.c 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160
  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. static void fdarray__init_revents(struct fdarray *fda, short revents)
  7. {
  8. int fd;
  9. fda->nr = fda->nr_alloc;
  10. for (fd = 0; fd < fda->nr; ++fd) {
  11. fda->entries[fd].fd = fda->nr - fd;
  12. fda->entries[fd].events = revents;
  13. fda->entries[fd].revents = revents;
  14. }
  15. }
  16. static int fdarray__fprintf_prefix(struct fdarray *fda, const char *prefix, FILE *fp)
  17. {
  18. int printed = 0;
  19. if (verbose <= 0)
  20. return 0;
  21. printed += fprintf(fp, "\n%s: ", prefix);
  22. return printed + fdarray__fprintf(fda, fp);
  23. }
  24. int test__fdarray__filter(struct test *test __maybe_unused, int subtest __maybe_unused)
  25. {
  26. int nr_fds, err = TEST_FAIL;
  27. struct fdarray *fda = fdarray__new(5, 5);
  28. if (fda == NULL) {
  29. pr_debug("\nfdarray__new() failed!");
  30. goto out;
  31. }
  32. fdarray__init_revents(fda, POLLIN);
  33. nr_fds = fdarray__filter(fda, POLLHUP, NULL, NULL);
  34. if (nr_fds != fda->nr_alloc) {
  35. pr_debug("\nfdarray__filter()=%d != %d shouldn't have filtered anything",
  36. nr_fds, fda->nr_alloc);
  37. goto out_delete;
  38. }
  39. fdarray__init_revents(fda, POLLHUP);
  40. nr_fds = fdarray__filter(fda, POLLHUP, NULL, NULL);
  41. if (nr_fds != 0) {
  42. pr_debug("\nfdarray__filter()=%d != %d, should have filtered all fds",
  43. nr_fds, fda->nr_alloc);
  44. goto out_delete;
  45. }
  46. fdarray__init_revents(fda, POLLHUP);
  47. fda->entries[2].revents = POLLIN;
  48. pr_debug("\nfiltering all but fda->entries[2]:");
  49. fdarray__fprintf_prefix(fda, "before", stderr);
  50. nr_fds = fdarray__filter(fda, POLLHUP, NULL, NULL);
  51. fdarray__fprintf_prefix(fda, " after", stderr);
  52. if (nr_fds != 1) {
  53. pr_debug("\nfdarray__filter()=%d != 1, should have left just one event", nr_fds);
  54. goto out_delete;
  55. }
  56. fdarray__init_revents(fda, POLLHUP);
  57. fda->entries[0].revents = POLLIN;
  58. fda->entries[3].revents = POLLIN;
  59. pr_debug("\nfiltering all but (fda->entries[0], fda->entries[3]):");
  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 != 2) {
  64. pr_debug("\nfdarray__filter()=%d != 2, should have left just two events",
  65. nr_fds);
  66. goto out_delete;
  67. }
  68. pr_debug("\n");
  69. err = 0;
  70. out_delete:
  71. fdarray__delete(fda);
  72. out:
  73. return err;
  74. }
  75. int test__fdarray__add(struct test *test __maybe_unused, int subtest __maybe_unused)
  76. {
  77. int err = TEST_FAIL;
  78. struct fdarray *fda = fdarray__new(2, 2);
  79. if (fda == NULL) {
  80. pr_debug("\nfdarray__new() failed!");
  81. goto out;
  82. }
  83. #define FDA_CHECK(_idx, _fd, _revents) \
  84. if (fda->entries[_idx].fd != _fd) { \
  85. pr_debug("\n%d: fda->entries[%d](%d) != %d!", \
  86. __LINE__, _idx, fda->entries[1].fd, _fd); \
  87. goto out_delete; \
  88. } \
  89. if (fda->entries[_idx].events != (_revents)) { \
  90. pr_debug("\n%d: fda->entries[%d].revents(%d) != %d!", \
  91. __LINE__, _idx, fda->entries[_idx].fd, _revents); \
  92. goto out_delete; \
  93. }
  94. #define FDA_ADD(_idx, _fd, _revents, _nr) \
  95. if (fdarray__add(fda, _fd, _revents, fdarray_flag__default) < 0) { \
  96. pr_debug("\n%d: fdarray__add(fda, %d, %d) failed!", \
  97. __LINE__,_fd, _revents); \
  98. goto out_delete; \
  99. } \
  100. if (fda->nr != _nr) { \
  101. pr_debug("\n%d: fdarray__add(fda, %d, %d)=%d != %d", \
  102. __LINE__,_fd, _revents, fda->nr, _nr); \
  103. goto out_delete; \
  104. } \
  105. FDA_CHECK(_idx, _fd, _revents)
  106. FDA_ADD(0, 1, POLLIN, 1);
  107. FDA_ADD(1, 2, POLLERR, 2);
  108. fdarray__fprintf_prefix(fda, "before growing array", stderr);
  109. FDA_ADD(2, 35, POLLHUP, 3);
  110. if (fda->entries == NULL) {
  111. pr_debug("\nfdarray__add(fda, 35, POLLHUP) should have allocated fda->pollfd!");
  112. goto out_delete;
  113. }
  114. fdarray__fprintf_prefix(fda, "after 3rd add", stderr);
  115. FDA_ADD(3, 88, POLLIN | POLLOUT, 4);
  116. fdarray__fprintf_prefix(fda, "after 4th add", stderr);
  117. FDA_CHECK(0, 1, POLLIN);
  118. FDA_CHECK(1, 2, POLLERR);
  119. FDA_CHECK(2, 35, POLLHUP);
  120. FDA_CHECK(3, 88, POLLIN | POLLOUT);
  121. #undef FDA_ADD
  122. #undef FDA_CHECK
  123. pr_debug("\n");
  124. err = 0;
  125. out_delete:
  126. fdarray__delete(fda);
  127. out:
  128. return err;
  129. }