From mboxrd@z Thu Jan 1 00:00:00 1970 Received: from eggs.gnu.org ([2001:4830:134:3::10]:51267) by lists.gnu.org with esmtp (Exim 4.71) (envelope-from ) id 1XtBAL-0003n7-Ty for qemu-devel@nongnu.org; Tue, 25 Nov 2014 03:08:12 -0500 Received: from Debian-exim by eggs.gnu.org with spam-scanned (Exim 4.71) (envelope-from ) id 1XtBAF-0002ll-Ly for qemu-devel@nongnu.org; Tue, 25 Nov 2014 03:08:05 -0500 Received: from mx1.redhat.com ([209.132.183.28]:54990) by eggs.gnu.org with esmtp (Exim 4.71) (envelope-from ) id 1XtBAF-0002lc-EB for qemu-devel@nongnu.org; Tue, 25 Nov 2014 03:07:59 -0500 Received: from int-mx13.intmail.prod.int.phx2.redhat.com (int-mx13.intmail.prod.int.phx2.redhat.com [10.5.11.26]) by mx1.redhat.com (8.14.4/8.14.4) with ESMTP id sAP87wZK027941 (version=TLSv1/SSLv3 cipher=DHE-RSA-AES256-GCM-SHA384 bits=256 verify=FAIL) for ; Tue, 25 Nov 2014 03:07:59 -0500 From: Fam Zheng Date: Tue, 25 Nov 2014 16:07:59 +0800 Message-Id: <1416902879-17422-6-git-send-email-famz@redhat.com> In-Reply-To: <1416902879-17422-1-git-send-email-famz@redhat.com> References: <1416902879-17422-1-git-send-email-famz@redhat.com> Subject: [Qemu-devel] [PATCH for-2.3 5/5] tests: Add test case for qemu_poll List-Id: List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , To: qemu-devel@nongnu.org Cc: Kevin Wolf , Paolo Bonzini , Stefan Hajnoczi Signed-off-by: Fam Zheng --- tests/Makefile | 2 + tests/test-poll.c | 272 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 274 insertions(+) create mode 100644 tests/test-poll.c diff --git a/tests/Makefile b/tests/Makefile index 16f0e4c..79f399d 100644 --- a/tests/Makefile +++ b/tests/Makefile @@ -41,6 +41,7 @@ check-unit-$(CONFIG_POSIX) += tests/test-rfifolock$(EXESUF) check-unit-y += tests/test-throttle$(EXESUF) gcov-files-test-aio-$(CONFIG_WIN32) = aio-win32.c gcov-files-test-aio-$(CONFIG_POSIX) = aio-posix.c +check-unit-y += tests/test-poll$(EXESUF) check-unit-y += tests/test-thread-pool$(EXESUF) gcov-files-test-thread-pool-y = thread-pool.c gcov-files-test-hbitmap-y = util/hbitmap.c @@ -241,6 +242,7 @@ tests/check-qjson$(EXESUF): tests/check-qjson.o libqemuutil.a libqemustub.a tests/check-qom-interface$(EXESUF): tests/check-qom-interface.o $(qom-core-obj) libqemuutil.a libqemustub.a tests/test-coroutine$(EXESUF): tests/test-coroutine.o $(block-obj-y) libqemuutil.a libqemustub.a tests/test-aio$(EXESUF): tests/test-aio.o $(block-obj-y) libqemuutil.a libqemustub.a +tests/test-poll$(EXESUF): tests/test-poll.o $(block-obj-y) libqemuutil.a libqemustub.a tests/test-rfifolock$(EXESUF): tests/test-rfifolock.o libqemuutil.a libqemustub.a tests/test-throttle$(EXESUF): tests/test-throttle.o $(block-obj-y) libqemuutil.a libqemustub.a tests/test-thread-pool$(EXESUF): tests/test-thread-pool.o $(block-obj-y) libqemuutil.a libqemustub.a diff --git a/tests/test-poll.c b/tests/test-poll.c new file mode 100644 index 0000000..75074d8 --- /dev/null +++ b/tests/test-poll.c @@ -0,0 +1,272 @@ +/* + * QTest testcase for QEMU poll + * + * Copyright Red Hat, Inc. 2014 + * + * Authors: + * Fam Zheng + * + * This work is licensed under the terms of the GNU GPL, version 2 or later. + * See the COPYING file in the top-level directory. + */ + +#include +#include "qemu/poll.h" +#include "qemu/event_notifier.h" + +static EventNotifier *poll_add_one(QEMUPoll *qpoll) +{ + EventNotifier *e = g_new(EventNotifier, 1); + + event_notifier_init(e, false); + qemu_poll_add(qpoll, event_notifier_get_fd(e), + G_IO_IN, + NULL); + return e; +} + +static int poll_del_one(QEMUPoll *qpoll, EventNotifier *e) +{ + int r = qemu_poll_del(qpoll, event_notifier_get_fd(e)); + event_notifier_cleanup(e); + g_free(e); + return r; +} + +static void test_poll_single(void) +{ + QEMUPoll *qpoll; + EventNotifier *e; + int r, i; + + qpoll = qemu_poll_new(); + g_assert(qpoll); + + e = poll_add_one(qpoll); + + /* Write some data and poll */ + event_notifier_set(e); + r = qemu_poll(qpoll, 0); + g_assert_cmpint(r, ==, 1); + + /* Clear data and poll */ + r = event_notifier_test_and_clear(e); + g_assert(r); + r = qemu_poll(qpoll, 0); + g_assert_cmpint(r, ==, 0); + + /* Write a lot of data and poll */ + for (i = 0; i < 10000; i++) { + event_notifier_set(e); + } + r = qemu_poll(qpoll, 0); + g_assert_cmpint(r, ==, 1); + + /* Clear data and poll again */ + r = event_notifier_test_and_clear(e); + g_assert(r); + r = qemu_poll(qpoll, 0); + g_assert_cmpint(r, ==, 0); + + /* Clean up */ + poll_del_one(qpoll, e); + qemu_poll_free(qpoll); +} + +static void test_poll_multiple(void) +{ + QEMUPoll *qpoll; + const int N = 32; + EventNotifier *e[N]; + QEMUPollEvent events[N]; + int r, s, i; + + qpoll = qemu_poll_new(); + g_assert(qpoll); + + for (i = 0; i < N; i++) { + e[i] = poll_add_one(qpoll); + } + + /* Write some data for each and poll */ + for (i = 0; i < N; i++) { + event_notifier_set(e[i]); + } + r = qemu_poll(qpoll, 0); + g_assert_cmpint(r, ==, N); + + /* Clear data and poll */ + for (i = 0; i < N; i++) { + r = event_notifier_test_and_clear(e[i]); + g_assert(r); + } + r = qemu_poll(qpoll, 0); + g_assert_cmpint(r, ==, 0); + + /* Write some data for first half and poll */ + for (i = 0; i < N / 2; i++) { + event_notifier_set(e[i]); + } + r = qemu_poll(qpoll, 0); + g_assert_cmpint(r, ==, N / 2); + s = qemu_poll_get_events(qpoll, events, N); + g_assert_cmpint(s, ==, r); + for (i = 0; i < s; i++) { + g_assert(events[i].revents & G_IO_IN); + } + + /* Clean up */ + for (i = 0; i < N; i++) { + poll_del_one(qpoll, e[i]); + } + qemu_poll_free(qpoll); +} + +static void test_poll_add_del(void) +{ + QEMUPoll *qpoll; + EventNotifier *e1, *e2; + QEMUPollEvent events[2]; + int r; + + qpoll = qemu_poll_new(); + g_assert(qpoll); + + e1 = poll_add_one(qpoll); + e2 = poll_add_one(qpoll); + + /* Write some data for each and poll */ + event_notifier_set(e1); + event_notifier_set(e2); + r = qemu_poll(qpoll, 0); + g_assert_cmpint(r, ==, 2); + + /* Clear e1 and poll */ + r = event_notifier_test_and_clear(e1); + r = qemu_poll(qpoll, 0); + g_assert_cmpint(r, ==, 1); + r = qemu_poll_get_events(qpoll, events, 2); + g_assert_cmpint(r, ==, 1); + g_assert_cmpint(events[0].fd, ==, event_notifier_get_fd(e2)); + + /* Write to both but remove one and poll the other */ + event_notifier_set(e1); + event_notifier_set(e2); + qemu_poll_del(qpoll, event_notifier_get_fd(e2)); + r = qemu_poll(qpoll, 0); + g_assert_cmpint(r, ==, 1); + r = qemu_poll_get_events(qpoll, events, 2); + g_assert_cmpint(r, ==, 1); + g_assert_cmpint(events[0].fd, ==, event_notifier_get_fd(e1)); + + r = qemu_poll_del(qpoll, event_notifier_get_fd(e2)); + g_assert_cmpint(r, ==, -ENOENT); + + /* Add it back and poll both */ + qemu_poll_add(qpoll, event_notifier_get_fd(e2), + G_IO_IN, NULL); + r = qemu_poll(qpoll, 0); + g_assert_cmpint(r, ==, 2); + + event_notifier_test_and_clear(e1); + event_notifier_test_and_clear(e2); + + r = qemu_poll_add(qpoll, event_notifier_get_fd(e2), + G_IO_IN, NULL); + g_assert_cmpint(r, ==, -EEXIST); + + /* Clean up */ + poll_del_one(qpoll, e1); + poll_del_one(qpoll, e2); + qemu_poll_free(qpoll); +} + +static void gpollfd_fill(GPollFD *fds, int start, EventNotifier **e, int n) +{ + int i; + for (i = 0; i < n; i++) { + fds[i + start] = (GPollFD) { + .fd = event_notifier_get_fd(e[i]), + .events = G_IO_IN, + }; + } +} + +static void test_poll_set_fds(void) +{ + QEMUPoll *qpoll; + const int M = 20; + const int N = 10; + EventNotifier *em[M], *en[N]; + GPollFD fds[M + N]; + int i, r; + + qpoll = qemu_poll_new(); + g_assert(qpoll); + + for (i = 0; i < M; i++) { + en[i] = poll_add_one(qpoll); + } + + for (i = 0; i < M; i++) { + em[i] = poll_add_one(qpoll); + } + + gpollfd_fill(fds, 0, em, M); + gpollfd_fill(fds, M, en, N); + + /* Set N */ + for (i = 0; i < N; i++) { + event_notifier_set(en[i]); + } + + /* Poll M + N should get N */ + r = qemu_poll_set_fds(qpoll, fds, M + N); + g_assert_cmpint(r, ==, M + N); + r = qemu_poll(qpoll, 0); + g_assert_cmpint(r, ==, N); + + /* Poll M should get 0 */ + r = qemu_poll_set_fds(qpoll, fds, M); + g_assert_cmpint(r, ==, M); + r = qemu_poll(qpoll, 0); + g_assert_cmpint(r, ==, 0); + + /* Poll N should get N */ + r = qemu_poll_set_fds(qpoll, fds + M, N); + g_assert_cmpint(r, ==, N); + r = qemu_poll(qpoll, 0); + g_assert_cmpint(r, ==, N); + + /* Poll M + N / 2 should get N / 2 */ + r = qemu_poll_set_fds(qpoll, fds, M + N / 2); + g_assert_cmpint(r, ==, M + N / 2); + r = qemu_poll(qpoll, 0); + g_assert_cmpint(r, ==, N / 2); + + /* Clean up */ + for (i = 0; i < M; i++) { + poll_del_one(qpoll, em[i]); + } + + for (i = 0; i < N; i++) { + poll_del_one(qpoll, en[i]); + } + qemu_poll_free(qpoll); +} + +int main(int argc, char **argv) +{ + int ret; + + g_test_init(&argc, &argv, NULL); + + g_test_add_func("/poll/single", test_poll_single); + g_test_add_func("/poll/multiple", test_poll_multiple); + g_test_add_func("/poll/add-del", test_poll_add_del); + g_test_add_func("/poll/set-fds", test_poll_set_fds); + + ret = g_test_run(); + + return ret; +} -- 1.9.3