blob: 6827e45d4a7c3d52d25cd275411bcfceeccf9de7 [file] [log] [blame]
/*
* Copyright (c) 2016 Fujitsu Ltd.
* Author: Guangwen Feng <fenggw-fnst@cn.fujitsu.com>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See
* the GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program.
*/
/*
* Description:
* Basic test for epoll_wait(2).
* Check that epoll_wait(2) works for EPOLLOUT and EPOLLIN events
* on a epoll instance and that struct epoll_event is set correctly.
*/
#include <sys/epoll.h>
#include <poll.h>
#include <string.h>
#include <errno.h>
#include "tst_test.h"
static int write_size, epfd, fds[2];
static int get_writesize(void)
{
int nfd, write_size = 0;
char buf[4096];
struct pollfd pfd[] = {
{.fd = fds[1], .events = POLLOUT},
};
memset(buf, 'a', sizeof(buf));
do {
write_size += SAFE_WRITE(0, fds[1], buf, sizeof(buf));
nfd = poll(pfd, 1, 1);
if (nfd == -1)
tst_brk(TBROK | TERRNO, "poll() failed");
} while (nfd > 0);
char read_buf[write_size];
SAFE_READ(1, fds[0], read_buf, sizeof(read_buf));
tst_res(TINFO, "Pipe buffer size is %i bytes", write_size);
return write_size;
}
static void setup(void)
{
static struct epoll_event epevs[2] = {
{.events = EPOLLIN},
{.events = EPOLLOUT},
};
SAFE_PIPE(fds);
epevs[0].data.fd = fds[0];
epevs[1].data.fd = fds[1];
write_size = get_writesize();
epfd = epoll_create(3);
if (epfd == -1)
tst_brk(TBROK | TERRNO, "epoll_create() failed");
if (epoll_ctl(epfd, EPOLL_CTL_ADD, fds[0], &epevs[0]) ||
epoll_ctl(epfd, EPOLL_CTL_ADD, fds[1], &epevs[1])) {
tst_brk(TBROK | TERRNO, "epoll_ctl() failed");
}
}
static int has_event(struct epoll_event *epevs, int epevs_len,
int fd, uint32_t events)
{
int i;
for (i = 0; i < epevs_len; i++) {
if ((epevs[i].data.fd == fd) && (epevs[i].events == events))
return 1;
}
return 0;
}
static void dump_epevs(struct epoll_event *epevs, int epevs_len)
{
int i;
for (i = 0; i < epevs_len; i++) {
tst_res(TINFO, "epevs[%d]: epoll.data.fd %d, epoll.events %x",
i, epevs[i].data.fd, epevs[i].events);
}
}
static void verify_epollout(void)
{
struct epoll_event ret_evs = {.events = 0, .data.fd = 0};
TEST(epoll_wait(epfd, &ret_evs, 1, -1));
if (TEST_RETURN == -1) {
tst_res(TFAIL | TTERRNO, "epoll_wait() epollout failed");
return;
}
if (TEST_RETURN != 1) {
tst_res(TFAIL, "epoll_wait() returned %li, expected 1",
TEST_RETURN);
return;
}
if (ret_evs.data.fd != fds[1]) {
tst_res(TFAIL, "epoll.data.fd %i, expected %i",
ret_evs.data.fd, fds[1]);
return;
}
if (ret_evs.events != EPOLLOUT) {
tst_res(TFAIL, "epoll.events %x, expected EPOLLOUT %x",
ret_evs.events, EPOLLOUT);
return;
}
tst_res(TPASS, "epoll_wait() epollout");
}
static void verify_epollin(void)
{
char write_buf[write_size];
char read_buf[sizeof(write_buf)];
struct epoll_event ret_evs = {.events = 0, .data.fd = 0};
memset(write_buf, 'a', sizeof(write_buf));
SAFE_WRITE(1, fds[1], write_buf, sizeof(write_buf));
TEST(epoll_wait(epfd, &ret_evs, 1, -1));
if (TEST_RETURN == -1) {
tst_res(TFAIL | TTERRNO, "epoll_wait() epollin failed");
goto end;
}
if (TEST_RETURN != 1) {
tst_res(TFAIL, "epoll_wait() returned %li, expected 1",
TEST_RETURN);
goto end;
}
if (ret_evs.data.fd != fds[0]) {
tst_res(TFAIL, "epoll.data.fd %i, expected %i",
ret_evs.data.fd, fds[0]);
goto end;
}
if (ret_evs.events != EPOLLIN) {
tst_res(TFAIL, "epoll.events %x, expected EPOLLIN %x",
ret_evs.events, EPOLLIN);
goto end;
}
tst_res(TPASS, "epoll_wait() epollin");
end:
SAFE_READ(1, fds[0], read_buf, sizeof(write_buf));
}
static void verify_epollio(void)
{
char write_buf[] = "Testing";
char read_buf[sizeof(write_buf)];
uint32_t events = EPOLLIN | EPOLLOUT;
struct epoll_event ret_evs[2];
SAFE_WRITE(1, fds[1], write_buf, sizeof(write_buf));
while (events) {
int events_matched = 0;
bzero(ret_evs, sizeof(ret_evs));
TEST(epoll_wait(epfd, ret_evs, 2, -1));
if (TEST_RETURN <= 0) {
tst_res(TFAIL | TTERRNO, "epoll_wait() returned %li",
TEST_RETURN);
goto end;
}
if ((events & EPOLLIN) &&
has_event(ret_evs, 2, fds[0], EPOLLIN)) {
events_matched++;
events &= ~EPOLLIN;
}
if ((events & EPOLLOUT) &&
has_event(ret_evs, 2, fds[1], EPOLLOUT)) {
events_matched++;
events &= ~EPOLLOUT;
}
if (TEST_RETURN != events_matched) {
tst_res(TFAIL,
"epoll_wait() returned unexpected events");
dump_epevs(ret_evs, 2);
goto end;
}
}
tst_res(TPASS, "epoll_wait() epollio");
end:
SAFE_READ(1, fds[0], read_buf, sizeof(write_buf));
}
static void cleanup(void)
{
if (epfd > 0)
SAFE_CLOSE(epfd);
if (fds[0]) {
SAFE_CLOSE(fds[0]);
SAFE_CLOSE(fds[1]);
}
}
static void do_test(unsigned int n)
{
switch (n) {
case 0:
verify_epollout();
break;
case 1:
verify_epollin();
break;
case 2:
verify_epollio();
break;
}
}
static struct tst_test test = {
.setup = setup,
.cleanup = cleanup,
.test = do_test,
.tcnt = 3,
};