kernel/tools/perf/tests/fdarray.c

   1 #include <api/fd/array.h>
   2 #include "util/debug.h"
   3 #include "tests/tests.h"
   4
   5 static 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
  17 static 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
  28 int test__fdarray__filter(void)
  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;
 100 out_delete:
 101         fdarray__delete(fda);
 102 out:
 103         return err;
 104 }
 105
 106 int test__fdarray__add(void)
 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;
 170 out_delete:
 171         fdarray__delete(fda);
 172 out:
 173         return err;
 174 }