1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63
|
// SPDX-License-Identifier: (MIT OR Apache-2.0)
#include "util.h"
enum {
TEST_FILE_SIZE = 2 * 1024 * 1024,
TEST_FILE_OFFSET = 0x10000,
};
static char filename[] = "write-XXXXXX";
/*
* Write a well-known pattern at an offset in a file.
*/
int main(int argc, char **argv)
{
struct test_opts opts;
struct blkio *b;
struct blkioq *q;
struct blkio_mem_region mem_region;
struct blkio_completion completion;
void *buf; /* I/O buffer */
void *pattern; /* reference data at TEST_FILE_OFFSET */
size_t buf_size;
int fd;
parse_generic_opts(&opts, argc, argv);
buf_size = sysconf(_SC_PAGESIZE);
/* Initialize pattern buffer */
pattern = malloc(buf_size);
assert(pattern);
memset(pattern, 'A', buf_size);
create_and_connect(&b, &opts, filename, TEST_FILE_SIZE);
ok(blkio_get_int(b, "fd", &fd));
assert(fd >= 0);
/* Set up I/O buffer */
ok(blkio_alloc_mem_region(b, &mem_region, buf_size));
buf = mem_region.addr;
memcpy(buf, pattern, buf_size);
ok(blkio_set_int(b, "num-queues", 1));
ok(blkio_start(b));
ok(blkio_map_mem_region(b, &mem_region));
q = blkio_get_queue(b, 0);
assert(q);
blkioq_write(q, TEST_FILE_OFFSET, buf, buf_size, NULL, 0);
assert(blkioq_do_io(q, &completion, 1, 1, NULL) == 1);
assert(completion.ret == 0);
memset(buf, 0, buf_size);
assert(pread_full(fd, buf, buf_size, TEST_FILE_OFFSET) == buf_size);
assert(memcmp(buf, pattern, buf_size) == 0);
blkio_destroy(&b);
return 0;
}
|