--- /dev/null
+#include "util/evlist.h"
+#include "util/debug.h"
+#include "tests/tests.h"
+
+static void perf_evlist__init_pollfd(struct perf_evlist *evlist,
+ int nr_fds_alloc, short revents)
+{
+ int fd;
+
+ evlist->nr_fds = nr_fds_alloc;
+
+ for (fd = 0; fd < nr_fds_alloc; ++fd) {
+ evlist->pollfd[fd].fd = nr_fds_alloc - fd;
+ evlist->pollfd[fd].revents = revents;
+ }
+}
+
+static int perf_evlist__fprintf_pollfd(struct perf_evlist *evlist,
+ const char *prefix, FILE *fp)
+{
+ int printed = 0, fd;
+
+ if (!verbose)
+ return 0;
+
+ printed += fprintf(fp, "\n%s: %3d [ ", prefix, evlist->nr_fds);
+ for (fd = 0; fd < evlist->nr_fds; ++fd)
+ printed += fprintf(fp, "%s%d", fd ? ", " : "", evlist->pollfd[fd].fd);
+ printed += fprintf(fp, " ]");
+ return printed;
+}
+
+int test__perf_evlist__filter_pollfd(void)
+{
+ const int nr_fds_alloc = 5;
+ int nr_fds, expected_fd[2], fd;
+ struct pollfd pollfd[nr_fds_alloc];
+ struct perf_evlist evlist_alloc = {
+ .pollfd = pollfd,
+ }, *evlist = &evlist_alloc;
+
+ perf_evlist__init_pollfd(evlist, nr_fds_alloc, POLLIN);
+ nr_fds = perf_evlist__filter_pollfd(evlist, POLLHUP);
+ if (nr_fds != nr_fds_alloc) {
+ pr_debug("\nperf_evlist__filter_pollfd()=%d != %d shouldn't have filtered anything",
+ nr_fds, nr_fds_alloc);
+ return TEST_FAIL;
+ }
+
+ perf_evlist__init_pollfd(evlist, nr_fds_alloc, POLLHUP);
+ nr_fds = perf_evlist__filter_pollfd(evlist, POLLHUP);
+ if (nr_fds != 0) {
+ pr_debug("\nperf_evlist__filter_pollfd()=%d != %d, should have filtered all fds",
+ nr_fds, nr_fds_alloc);
+ return TEST_FAIL;
+ }
+
+ perf_evlist__init_pollfd(evlist, nr_fds_alloc, POLLHUP);
+ pollfd[2].revents = POLLIN;
+ expected_fd[0] = pollfd[2].fd;
+
+ pr_debug("\nfiltering all but pollfd[2]:");
+ perf_evlist__fprintf_pollfd(evlist, "before", stderr);
+ nr_fds = perf_evlist__filter_pollfd(evlist, POLLHUP);
+ perf_evlist__fprintf_pollfd(evlist, " after", stderr);
+ if (nr_fds != 1) {
+ pr_debug("\nperf_evlist__filter_pollfd()=%d != 1, should have left just one event",
+ nr_fds);
+ return TEST_FAIL;
+ }
+
+ if (pollfd[0].fd != expected_fd[0]) {
+ pr_debug("\npollfd[0].fd=%d != %d\n", pollfd[0].fd, expected_fd[0]);
+ return TEST_FAIL;
+ }
+
+ perf_evlist__init_pollfd(evlist, nr_fds_alloc, POLLHUP);
+ pollfd[0].revents = POLLIN;
+ expected_fd[0] = pollfd[0].fd;
+ pollfd[3].revents = POLLIN;
+ expected_fd[1] = pollfd[3].fd;
+
+ pr_debug("\nfiltering all but (pollfd[0], pollfd[3]):");
+ perf_evlist__fprintf_pollfd(evlist, "before", stderr);
+ nr_fds = perf_evlist__filter_pollfd(evlist, POLLHUP);
+ perf_evlist__fprintf_pollfd(evlist, " after", stderr);
+ if (nr_fds != 2) {
+ pr_debug("\nperf_evlist__filter_pollfd()=%d != 2, should have left just two events",
+ nr_fds);
+ return TEST_FAIL;
+ }
+
+ for (fd = 0; fd < 2; ++fd) {
+ if (pollfd[fd].fd != expected_fd[fd]) {
+ pr_debug("\npollfd[%d].fd=%d != %d\n", fd, pollfd[fd].fd, expected_fd[fd]);
+ return TEST_FAIL;
+ }
+ }
+
+ pr_debug("\n");
+
+ return 0;
+}