tests/zfs-tests/cmd/mmap_libaio/mmap_libaio.c

   1 /*
   2  * CDDL HEADER START
   3  *
   4  * The contents of this file are subject to the terms of the
   5  * Common Development and Distribution License (the "License").
   6  * You may not use this file except in compliance with the License.
   7  *
   8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
   9  * or http://www.opensolaris.org/os/licensing.
  10  * See the License for the specific language governing permissions
  11  * and limitations under the License.
  12  *
  13  * When distributing Covered Code, include this CDDL HEADER in each
  14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
  15  * If applicable, add the following below this CDDL HEADER, with the
  16  * fields enclosed by brackets "[]" replaced with your own identifying
  17  * information: Portions Copyright [yyyy] [name of copyright owner]
  18  *
  19  * CDDL HEADER END
  20  */
  21
  22 /*
  23  * Copyright 2018 Canonical.  All rights reserved.
  24  */
  25
  26 #include <stdio.h>
  27 #include <stdlib.h>
  28 #include <unistd.h>
  29 #include <fcntl.h>
  30 #include <sys/mman.h>
  31 #include <sys/types.h>
  32 #include <sys/stat.h>
  33 #include <libaio.h>
  34 #include <err.h>
  35
  36 io_context_t io_ctx;
  37
  38 static void
  39 do_sync_io(struct iocb *iocb)
  40 {
  41         struct io_event event;
  42         struct iocb *iocbs[] = { iocb };
  43         struct timespec ts = { 30, 0 };
  44
  45         if (io_submit(io_ctx, 1, iocbs) != 1)
  46                 err(1, "io_submit failed");
  47
  48         if (io_getevents(io_ctx, 0, 1, &event, &ts) != 1)
  49                 err(1, "io_getevents failed");
  50 }
  51
  52 int
  53 main(int argc, char **argv)
  54 {
  55         char *buf;
  56         int page_size = getpagesize();
  57         int buf_size = strtol(argv[2], NULL, 0);
  58         int rwfd;
  59         struct iocb iocb;
  60
  61         if (io_queue_init(1024, &io_ctx))
  62                 err(1, "io_queue_init failed");
  63
  64         rwfd = open(argv[1], O_RDWR | O_CREAT, S_IRUSR | S_IWUSR);
  65         if (rwfd < 0)
  66                 err(1, "open failed");
  67
  68         if (ftruncate(rwfd, buf_size) < 0)
  69                 err(1, "ftruncate failed");
  70
  71         buf = mmap(0, page_size, PROT_READ | PROT_WRITE, MAP_SHARED, rwfd, 0);
  72         if (buf == MAP_FAILED)
  73                 err(1, "mmap failed");
  74
  75         (void) io_prep_pwrite(&iocb, rwfd, buf, buf_size, 0);
  76         do_sync_io(&iocb);
  77
  78         (void) io_prep_pread(&iocb, rwfd, buf, buf_size, 0);
  79         do_sync_io(&iocb);
  80
  81         if (close(rwfd))
  82                 err(1, "close failed");
  83
  84         if (io_queue_release(io_ctx) != 0)
  85                 err(1, "io_queue_release failed");
  86
  87         return (0);
  88 }