multi read works
This commit is contained in:
BuildTools
parent
b40e452293
commit
9c9306b825
222
tests/sys/kqueue/libkqueue/read_m.c
Normal file
222
tests/sys/kqueue/libkqueue/read_m.c
Normal file
@ -0,0 +1,222 @@
|
|||||||
|
/*
|
||||||
|
* Copyright (c) 2009 Mark Heily <mark@heily.com>
|
||||||
|
*
|
||||||
|
* Permission to use, copy, modify, and distribute this software for any
|
||||||
|
* purpose with or without fee is hereby granted, provided that the above
|
||||||
|
* copyright notice and this permission notice appear in all copies.
|
||||||
|
*
|
||||||
|
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
||||||
|
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
||||||
|
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
||||||
|
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
||||||
|
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
||||||
|
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
||||||
|
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
||||||
|
*
|
||||||
|
* $FreeBSD$
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "common.h"
|
||||||
|
|
||||||
|
#include <sys/ioctl.h>
|
||||||
|
#include <semaphore.h>
|
||||||
|
#include <pthread.h>
|
||||||
|
|
||||||
|
#define THREAD_CNT (100)
|
||||||
|
#define PACKET_CNT (5000)
|
||||||
|
#define FKQMULTI _IO('f', 89)
|
||||||
|
#define TEST_DEBUG
|
||||||
|
|
||||||
|
struct thread_info {
|
||||||
|
pthread_t thrd;
|
||||||
|
int evcnt;
|
||||||
|
int tid;
|
||||||
|
};
|
||||||
|
|
||||||
|
int g_kqfd;
|
||||||
|
int g_sockfd[2];
|
||||||
|
int g_end;
|
||||||
|
struct thread_info g_thrd_info[THREAD_CNT];
|
||||||
|
/* Test threads signals this upon receiving events */
|
||||||
|
sem_t g_sem_driver;
|
||||||
|
|
||||||
|
static int
|
||||||
|
check_sched(void)
|
||||||
|
{
|
||||||
|
int max = 0, min = 999999;
|
||||||
|
|
||||||
|
for(int i = 0; i < THREAD_CNT; i++) {
|
||||||
|
int cur = g_thrd_info[i].evcnt;
|
||||||
|
if (cur > max) {
|
||||||
|
max = cur;
|
||||||
|
}
|
||||||
|
if (cur < min) {
|
||||||
|
min = cur;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#ifdef TEST_DEBUG
|
||||||
|
printf("READ_M: check_sched: max difference is %d\n", max - min);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
return (max - min) <= 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
socket_pop()
|
||||||
|
{
|
||||||
|
char buf[1];
|
||||||
|
|
||||||
|
/* Drain the read buffer, then make sure there are no more events. */
|
||||||
|
#ifdef TEST_DEBUG
|
||||||
|
printf("READ_M: popping the read buffer\n");
|
||||||
|
#endif
|
||||||
|
if (read(g_sockfd[0], &buf[0], 1) < 1)
|
||||||
|
err(1, "read(2)");
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
socket_push()
|
||||||
|
{
|
||||||
|
#ifdef TEST_DEBUG
|
||||||
|
printf("READ_M: pushing to socket\n");
|
||||||
|
#endif
|
||||||
|
if (write(g_sockfd[1], ".", 1) < 1) {
|
||||||
|
#ifdef TEST_DEBUG
|
||||||
|
printf("READ_M: write failed with %d\n", errno);
|
||||||
|
#endif
|
||||||
|
err(1, "write(2)");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/* for multi threaded read */
|
||||||
|
static void*
|
||||||
|
test_socket_read_thrd(void* args)
|
||||||
|
{
|
||||||
|
struct thread_info *info = (struct thread_info *) args;
|
||||||
|
|
||||||
|
struct kevent kev;
|
||||||
|
EV_SET(&kev, g_sockfd[0], EVFILT_READ, EV_ADD | EV_CLEAR, 0, 0, &g_sockfd[0]);
|
||||||
|
kev.data = 1;
|
||||||
|
|
||||||
|
while (1) {
|
||||||
|
#ifdef TEST_DEBUG
|
||||||
|
printf("READ_M: thread %d waiting for events\n", info->tid);
|
||||||
|
#endif
|
||||||
|
kevent_cmp(&kev, kevent_get(g_kqfd));
|
||||||
|
#ifdef TEST_DEBUG
|
||||||
|
printf("READ_M: thread %d woke up\n", info->tid);
|
||||||
|
#endif
|
||||||
|
socket_pop();
|
||||||
|
info->evcnt++;
|
||||||
|
|
||||||
|
/* signal the driver */
|
||||||
|
sem_post(&g_sem_driver);
|
||||||
|
|
||||||
|
if (g_end) {
|
||||||
|
#ifdef TEST_DEBUG
|
||||||
|
printf("READ_M: thread %d exiting...\n", info->tid);
|
||||||
|
#endif
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
pthread_exit(0);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
test_socket_read(void)
|
||||||
|
{
|
||||||
|
int error = 0;
|
||||||
|
struct kevent kev;
|
||||||
|
EV_SET(&kev, g_sockfd[0], EVFILT_READ, EV_ADD | EV_CLEAR, 0, 0, &g_sockfd[0]);
|
||||||
|
|
||||||
|
const char *test_id = "[Multi]kevent(EVFILT_READ)";
|
||||||
|
sem_init(&g_sem_driver, 0, 0);
|
||||||
|
g_end = 0;
|
||||||
|
|
||||||
|
test_begin(test_id);
|
||||||
|
|
||||||
|
error = kevent(g_kqfd, &kev, 1, NULL, 0, NULL);
|
||||||
|
|
||||||
|
if (error == -1) {
|
||||||
|
#ifdef TEST_DEBUG
|
||||||
|
printf("READ_M: kevent add failed with %d\n", errno);
|
||||||
|
#endif
|
||||||
|
err(1, "kevent_add");
|
||||||
|
}
|
||||||
|
|
||||||
|
#ifdef TEST_DEBUG
|
||||||
|
printf("READ_M: creating %d threads...\n", THREAD_CNT);
|
||||||
|
#endif
|
||||||
|
for (int i = 0; i < THREAD_CNT; i++) {
|
||||||
|
g_thrd_info[i].tid = i;
|
||||||
|
g_thrd_info[i].evcnt = 0;
|
||||||
|
pthread_create(&g_thrd_info[i].thrd, NULL, test_socket_read_thrd, &g_thrd_info[i]);
|
||||||
|
}
|
||||||
|
|
||||||
|
#ifdef TEST_DEBUG
|
||||||
|
printf("READ_M: waiting for threads to wait on KQ...\n");
|
||||||
|
#endif
|
||||||
|
|
||||||
|
sleep(3);
|
||||||
|
|
||||||
|
for(int i = 0; i < PACKET_CNT; i++) {
|
||||||
|
#ifdef TEST_DEBUG
|
||||||
|
printf("READ_M: processing packet %d\n", i);
|
||||||
|
#endif
|
||||||
|
socket_push();
|
||||||
|
/* wait for thread events */
|
||||||
|
sem_wait(&g_sem_driver);
|
||||||
|
|
||||||
|
if (!check_sched()) {
|
||||||
|
#ifdef TEST_DEBUG
|
||||||
|
printf("READ_M: check_sched failed...\n");
|
||||||
|
#endif
|
||||||
|
g_end = 1;
|
||||||
|
error = 2;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
if ((i + THREAD_CNT) == PACKET_CNT - 1) {
|
||||||
|
g_end = 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/* shutdown the systems */
|
||||||
|
|
||||||
|
#ifdef TEST_DEBUG
|
||||||
|
printf("READ_M: finished testing, system shutting down...\n");
|
||||||
|
#endif
|
||||||
|
for (int i = 0; i < THREAD_CNT; i++) {
|
||||||
|
pthread_join(g_thrd_info[i].thrd, NULL);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!error)
|
||||||
|
success();
|
||||||
|
else
|
||||||
|
err(error, "kevent");
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
void
|
||||||
|
test_evfilt_read_m()
|
||||||
|
{
|
||||||
|
/* Create a connected pair of full-duplex sockets for testing socket events */
|
||||||
|
if (socketpair(AF_UNIX, SOCK_STREAM, 0, &g_sockfd[0]) < 0)
|
||||||
|
abort();
|
||||||
|
|
||||||
|
g_kqfd = kqueue();
|
||||||
|
int error = ioctl(g_kqfd, FKQMULTI);
|
||||||
|
if (error == -1) {
|
||||||
|
#ifdef TEST_DEBUG
|
||||||
|
printf("READ_M: ioctl failed with %d\n", errno);
|
||||||
|
#endif
|
||||||
|
err(1, "ioctl");
|
||||||
|
}
|
||||||
|
|
||||||
|
test_socket_read();
|
||||||
|
|
||||||
|
close(g_kqfd);
|
||||||
|
close(g_sockfd[0]);
|
||||||
|
close(g_sockfd[1]);
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user