2015-02-27 19:19:33 +03:00
|
|
|
/*
|
|
|
|
* QEMU I/O channel test helpers
|
|
|
|
*
|
|
|
|
* Copyright (c) 2015 Red Hat, Inc.
|
|
|
|
*
|
|
|
|
* This library is free software; you can redistribute it and/or
|
|
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
|
|
* License as published by the Free Software Foundation; either
|
|
|
|
* version 2 of the License, or (at your option) any later version.
|
|
|
|
*
|
|
|
|
* This library 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
|
|
|
|
* Lesser General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
|
|
* License along with this library; if not, see <http://www.gnu.org/licenses/>.
|
|
|
|
*
|
|
|
|
*/
|
|
|
|
|
2016-02-08 21:08:51 +03:00
|
|
|
#include "qemu/osdep.h"
|
2015-02-27 19:19:33 +03:00
|
|
|
#include "io-channel-helpers.h"
|
|
|
|
|
|
|
|
struct QIOChannelTest {
|
|
|
|
QIOChannel *src;
|
|
|
|
QIOChannel *dst;
|
|
|
|
bool blocking;
|
|
|
|
size_t len;
|
|
|
|
size_t niov;
|
|
|
|
char *input;
|
|
|
|
struct iovec *inputv;
|
|
|
|
char *output;
|
|
|
|
struct iovec *outputv;
|
|
|
|
Error *writeerr;
|
|
|
|
Error *readerr;
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
static void test_skip_iovec(struct iovec **iov,
|
|
|
|
size_t *niov,
|
|
|
|
size_t skip,
|
|
|
|
struct iovec *old)
|
|
|
|
{
|
|
|
|
size_t offset = 0;
|
|
|
|
size_t i;
|
|
|
|
|
|
|
|
for (i = 0; i < *niov; i++) {
|
|
|
|
if (skip < (*iov)[i].iov_len) {
|
|
|
|
old->iov_len = (*iov)[i].iov_len;
|
|
|
|
old->iov_base = (*iov)[i].iov_base;
|
|
|
|
|
|
|
|
(*iov)[i].iov_len -= skip;
|
|
|
|
(*iov)[i].iov_base += skip;
|
|
|
|
break;
|
|
|
|
} else {
|
|
|
|
skip -= (*iov)[i].iov_len;
|
|
|
|
|
|
|
|
if (i == 0 && old->iov_base) {
|
|
|
|
(*iov)[i].iov_len = old->iov_len;
|
|
|
|
(*iov)[i].iov_base = old->iov_base;
|
|
|
|
old->iov_len = 0;
|
|
|
|
old->iov_base = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
offset++;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
*iov = *iov + offset;
|
|
|
|
*niov -= offset;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/* This thread sends all data using iovecs */
|
|
|
|
static gpointer test_io_thread_writer(gpointer opaque)
|
|
|
|
{
|
|
|
|
QIOChannelTest *data = opaque;
|
|
|
|
struct iovec *iov = data->inputv;
|
|
|
|
size_t niov = data->niov;
|
|
|
|
struct iovec old = { 0 };
|
|
|
|
|
|
|
|
qio_channel_set_blocking(data->src, data->blocking, NULL);
|
|
|
|
|
|
|
|
while (niov) {
|
|
|
|
ssize_t ret;
|
|
|
|
ret = qio_channel_writev(data->src,
|
|
|
|
iov,
|
|
|
|
niov,
|
|
|
|
&data->writeerr);
|
|
|
|
if (ret == QIO_CHANNEL_ERR_BLOCK) {
|
|
|
|
if (data->blocking) {
|
|
|
|
error_setg(&data->writeerr,
|
|
|
|
"Unexpected I/O blocking");
|
|
|
|
break;
|
|
|
|
} else {
|
|
|
|
qio_channel_wait(data->src,
|
|
|
|
G_IO_OUT);
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
} else if (ret < 0) {
|
|
|
|
break;
|
|
|
|
} else if (ret == 0) {
|
|
|
|
error_setg(&data->writeerr,
|
|
|
|
"Unexpected zero length write");
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
test_skip_iovec(&iov, &niov, ret, &old);
|
|
|
|
}
|
|
|
|
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/* This thread receives all data using iovecs */
|
|
|
|
static gpointer test_io_thread_reader(gpointer opaque)
|
|
|
|
{
|
|
|
|
QIOChannelTest *data = opaque;
|
|
|
|
struct iovec *iov = data->outputv;
|
|
|
|
size_t niov = data->niov;
|
|
|
|
struct iovec old = { 0 };
|
|
|
|
|
|
|
|
qio_channel_set_blocking(data->dst, data->blocking, NULL);
|
|
|
|
|
|
|
|
while (niov) {
|
|
|
|
ssize_t ret;
|
|
|
|
|
|
|
|
ret = qio_channel_readv(data->dst,
|
|
|
|
iov,
|
|
|
|
niov,
|
|
|
|
&data->readerr);
|
|
|
|
|
|
|
|
if (ret == QIO_CHANNEL_ERR_BLOCK) {
|
|
|
|
if (data->blocking) {
|
|
|
|
error_setg(&data->writeerr,
|
|
|
|
"Unexpected I/O blocking");
|
|
|
|
break;
|
|
|
|
} else {
|
|
|
|
qio_channel_wait(data->dst,
|
|
|
|
G_IO_IN);
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
} else if (ret < 0) {
|
|
|
|
break;
|
|
|
|
} else if (ret == 0) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
test_skip_iovec(&iov, &niov, ret, &old);
|
|
|
|
}
|
|
|
|
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
QIOChannelTest *qio_channel_test_new(void)
|
|
|
|
{
|
|
|
|
QIOChannelTest *data = g_new0(QIOChannelTest, 1);
|
|
|
|
size_t i;
|
|
|
|
size_t offset;
|
|
|
|
|
|
|
|
|
|
|
|
/* We'll send 1 MB of data */
|
|
|
|
#define CHUNK_COUNT 250
|
|
|
|
#define CHUNK_LEN 4194
|
|
|
|
|
|
|
|
data->len = CHUNK_COUNT * CHUNK_LEN;
|
|
|
|
data->input = g_new0(char, data->len);
|
|
|
|
data->output = g_new0(gchar, data->len);
|
|
|
|
|
|
|
|
/* Fill input with a pattern */
|
|
|
|
for (i = 0; i < data->len; i += CHUNK_LEN) {
|
|
|
|
memset(data->input + i, (i / CHUNK_LEN), CHUNK_LEN);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* We'll split the data across a bunch of IO vecs */
|
|
|
|
data->niov = CHUNK_COUNT;
|
|
|
|
data->inputv = g_new0(struct iovec, data->niov);
|
|
|
|
data->outputv = g_new0(struct iovec, data->niov);
|
|
|
|
|
|
|
|
for (i = 0, offset = 0; i < data->niov; i++, offset += CHUNK_LEN) {
|
|
|
|
data->inputv[i].iov_base = data->input + offset;
|
|
|
|
data->outputv[i].iov_base = data->output + offset;
|
|
|
|
data->inputv[i].iov_len = CHUNK_LEN;
|
|
|
|
data->outputv[i].iov_len = CHUNK_LEN;
|
|
|
|
}
|
|
|
|
|
|
|
|
return data;
|
|
|
|
}
|
|
|
|
|
|
|
|
void qio_channel_test_run_threads(QIOChannelTest *test,
|
|
|
|
bool blocking,
|
|
|
|
QIOChannel *src,
|
|
|
|
QIOChannel *dst)
|
|
|
|
{
|
|
|
|
GThread *reader, *writer;
|
|
|
|
|
|
|
|
test->src = src;
|
|
|
|
test->dst = dst;
|
|
|
|
test->blocking = blocking;
|
|
|
|
|
|
|
|
reader = g_thread_new("reader",
|
|
|
|
test_io_thread_reader,
|
|
|
|
test);
|
|
|
|
writer = g_thread_new("writer",
|
|
|
|
test_io_thread_writer,
|
|
|
|
test);
|
|
|
|
|
|
|
|
g_thread_join(reader);
|
|
|
|
g_thread_join(writer);
|
|
|
|
|
|
|
|
test->dst = test->src = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void qio_channel_test_run_writer(QIOChannelTest *test,
|
|
|
|
QIOChannel *src)
|
|
|
|
{
|
|
|
|
test->src = src;
|
|
|
|
test_io_thread_writer(test);
|
|
|
|
test->src = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void qio_channel_test_run_reader(QIOChannelTest *test,
|
|
|
|
QIOChannel *dst)
|
|
|
|
{
|
|
|
|
test->dst = dst;
|
|
|
|
test_io_thread_reader(test);
|
|
|
|
test->dst = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void qio_channel_test_validate(QIOChannelTest *test)
|
|
|
|
{
|
|
|
|
g_assert_cmpint(memcmp(test->input,
|
|
|
|
test->output,
|
|
|
|
test->len), ==, 0);
|
|
|
|
g_assert(test->readerr == NULL);
|
|
|
|
g_assert(test->writeerr == NULL);
|
|
|
|
|
|
|
|
g_free(test->inputv);
|
|
|
|
g_free(test->outputv);
|
|
|
|
g_free(test->input);
|
|
|
|
g_free(test->output);
|
|
|
|
g_free(test);
|
|
|
|
}
|