strace/tests/aio.c
Dmitry V. Levin a5417308e4 tests/tests.h: add ARRAY_SIZE and LENGTH_OF macros
* tests/tests.h (ARRAY_SIZE, LENGTH_OF): New macros.
* tests/aio.c (ARRAY_SIZE): Remove.
* tests/mmsg.c (LENGTH_OF): Remove.
2016-01-20 15:24:45 +00:00

211 lines
6.6 KiB
C

/*
* Copyright (c) 2015-2016 Dmitry V. Levin <ldv@altlinux.org>
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. The name of the author may not be used to endorse or promote products
* derived from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
#include "tests.h"
#include <assert.h>
#include <errno.h>
#include <fcntl.h>
#include <inttypes.h>
#include <stdio.h>
#include <time.h>
#include <unistd.h>
#include <sys/syscall.h>
#if defined __NR_io_setup \
&& defined __NR_io_submit \
&& defined __NR_io_getevents \
&& defined __NR_io_cancel \
&& defined __NR_io_destroy
# include <linux/aio_abi.h>
int
main(void)
{
const unsigned int sizeof_data0 = 4096;
const unsigned int sizeof_data1 = 8192;
void *data0 = tail_alloc(sizeof_data0);
void *data1 = tail_alloc(sizeof_data1);
const struct iocb proto_cb[] = {
{
.aio_data = 0xfeedface11111111,
.aio_reqprio = 11,
.aio_buf = (unsigned long) data0,
.aio_offset = 0xdeface1facefeed,
.aio_nbytes = sizeof_data0
},
{
.aio_data = 0xfeedface22222222,
.aio_reqprio = 22,
.aio_buf = (unsigned long) data1,
.aio_offset = 0xdeface2cafef00d,
.aio_nbytes = sizeof_data1
}
};
const struct iocb *cb = tail_memdup(proto_cb, sizeof(proto_cb));
const struct iovec proto_iov0[] = {
{
.iov_base = data0,
.iov_len = sizeof_data0 / 4
},
{
.iov_base = data0 + sizeof_data0 / 4,
.iov_len = sizeof_data0 / 4 * 3
},
};
const struct iovec *iov0 = tail_memdup(proto_iov0, sizeof(proto_iov0));
const struct iovec proto_iov1[] = {
{
.iov_base = data1,
.iov_len = sizeof_data1 / 4
},
{
.iov_base = data1 + sizeof_data1 / 4,
.iov_len = sizeof_data1 / 4 * 3
},
};
const struct iovec *iov1 = tail_memdup(proto_iov1, sizeof(proto_iov1));
const struct iocb proto_cbv[] = {
{
.aio_data = 0xfeed11111111face,
.aio_lio_opcode = 7,
.aio_reqprio = 111,
.aio_buf = (unsigned long) iov0,
.aio_offset = 0xdeface1facefeed,
.aio_nbytes = ARRAY_SIZE(proto_iov0)
},
{
.aio_data = 0xfeed22222222face,
.aio_lio_opcode = 7,
.aio_reqprio = 222,
.aio_buf = (unsigned long) iov1,
.aio_offset = 0xdeface2cafef00d,
.aio_nbytes = ARRAY_SIZE(proto_iov1)
}
};
const struct iocb *cbv = tail_memdup(proto_cbv, sizeof(proto_cbv));
const struct iocb proto_cbc = {
.aio_data = 0xdeadbeefbadc0ded,
.aio_reqprio = 99,
.aio_fildes = -42
};
const struct iocb *cbc = tail_memdup(&proto_cbc, sizeof(proto_cbc));
const long proto_cbs[] = {
(long) &cb[0], (long) &cb[1]
};
const long *cbs = tail_memdup(proto_cbs, sizeof(proto_cbs));
const long proto_cbvs[] = {
(long) &cbv[0], (long) &cbv[1],
};
const long *cbvs = tail_memdup(proto_cbvs, sizeof(proto_cbvs));
unsigned long *ctx = tail_alloc(sizeof(unsigned long));
*ctx = 0;
const unsigned int nr = ARRAY_SIZE(proto_cb);
const unsigned long lnr = (unsigned long) (0xdeadbeef00000000ULL | nr);
const struct io_event *ev = tail_alloc(nr * sizeof(struct io_event));
const struct timespec proto_ts = { .tv_nsec = 123456789 };
const struct timespec *ts = tail_memdup(&proto_ts, sizeof(proto_ts));
(void) close(0);
if (open("/dev/zero", O_RDONLY))
perror_msg_and_skip("open: %s", "/dev/zero");
if (syscall(__NR_io_setup, lnr, ctx))
perror_msg_and_skip("io_setup");
printf("io_setup(%u, [%lu]) = 0\n", nr, *ctx);
if (syscall(__NR_io_submit, *ctx, nr, cbs) != (long) nr)
perror_msg_and_skip("io_submit");
printf("io_submit(%lu, %u, ["
"{data=%#llx, pread, reqprio=11, fildes=0, "
"buf=%p, nbytes=%u, offset=%lld}, "
"{data=%#llx, pread, reqprio=22, fildes=0, "
"buf=%p, nbytes=%u, offset=%lld}"
"]) = %u\n",
*ctx, nr,
(unsigned long long) cb[0].aio_data, data0,
sizeof_data0, (long long) cb[0].aio_offset,
(unsigned long long) cb[1].aio_data, data1,
sizeof_data1, (long long) cb[1].aio_offset,
nr);
assert(syscall(__NR_io_getevents, *ctx, nr, nr + 1, ev, ts) == (long) nr);
printf("io_getevents(%lu, %u, %u, ["
"{data=%#llx, obj=%p, res=%u, res2=0}, "
"{data=%#llx, obj=%p, res=%u, res2=0}"
"], {0, 123456789}) = %u\n",
*ctx, nr, nr + 1,
(unsigned long long) cb[0].aio_data, &cb[0], sizeof_data0,
(unsigned long long) cb[1].aio_data, &cb[1], sizeof_data1,
nr);
assert(syscall(__NR_io_cancel, *ctx, cbc, ev) == -1 && EINVAL == errno);
printf("io_cancel(%lu, {data=%#llx, pread, reqprio=99, fildes=-42}, %p) "
"= -1 EINVAL (%m)\n",
*ctx, (unsigned long long) cbc->aio_data, ev);
if (syscall(__NR_io_submit, *ctx, nr, cbvs) != (long) nr)
perror_msg_and_skip("io_submit");
printf("io_submit(%lu, %u, ["
"{data=%#llx, preadv, reqprio=%hd, fildes=0, "
"iovec=[{%p, %u}, {%p, %u}], offset=%lld}, "
"{data=%#llx, preadv, reqprio=%hd, fildes=0, "
"iovec=[{%p, %u}, {%p, %u}], offset=%lld}"
"]) = %u\n",
*ctx, nr,
(unsigned long long) cbv[0].aio_data, cbv[0].aio_reqprio,
iov0[0].iov_base, (unsigned int) iov0[0].iov_len,
iov0[1].iov_base, (unsigned int) iov0[1].iov_len,
(long long) cbv[0].aio_offset,
(unsigned long long) cbv[1].aio_data, cbv[1].aio_reqprio,
iov1[0].iov_base, (unsigned int) iov1[0].iov_len,
iov1[1].iov_base, (unsigned int) iov1[1].iov_len,
(long long) cbv[1].aio_offset,
nr);
assert(syscall(__NR_io_destroy, *ctx) == 0);
printf("io_destroy(%lu) = 0\n", *ctx);
puts("+++ exited with 0 +++");
return 0;
}
#else
SKIP_MAIN_UNDEFINED("__NR_io_*")
#endif