tests/mmfs_readwrite.c

   1 /*
   2  * Copyright © 2008 Intel Corporation
   3  *
   4  * Permission is hereby granted, free of charge, to any person obtaining a
   5  * copy of this software and associated documentation files (the "Software"),
   6  * to deal in the Software without restriction, including without limitation
   7  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
   8  * and/or sell copies of the Software, and to permit persons to whom the
   9  * Software is furnished to do so, subject to the following conditions:
  10  *
  11  * The above copyright notice and this permission notice (including the next
  12  * paragraph) shall be included in all copies or substantial portions of the
  13  * Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  18  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  20  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  21  * IN THE SOFTWARE.
  22  *
  23  * Authors:
  24  *    Eric Anholt <eric@anholt.net>
  25  *
  26  */
  27
  28 #include <stdlib.h>
  29 #include <stdio.h>
  30 #include <string.h>
  31 #include <assert.h>
  32 #include <fcntl.h>
  33 #include <inttypes.h>
  34 #include <errno.h>
  35 #include <sys/stat.h>
  36 #include "mmfs.h"
  37
  38 #define MMFS_BUFFER_SIZE 16384
  39
  40 int do_read(int fd, int handle, void *buf, int offset, int size)
  41 {
  42         struct mmfs_pread_args read;
  43
  44         /* Ensure that we don't have any convenient data in buf in case
  45          * we fail.
  46          */
  47         memset(buf, 0xd0, size);
  48
  49         memset(&read, 0, sizeof(read));
  50         read.handle = handle;
  51         read.data = buf;
  52         read.size = size;
  53         read.offset = offset;
  54
  55         return ioctl(fd, MMFS_IOCTL_PREAD, &read);
  56 }
  57
  58 int do_write(int fd, int handle, void *buf, int offset, int size)
  59 {
  60         struct mmfs_pwrite_args write;
  61
  62         memset(&write, 0, sizeof(write));
  63         write.handle = handle;
  64         write.data = buf;
  65         write.size = size;
  66         write.offset = offset;
  67
  68         return ioctl(fd, MMFS_IOCTL_PWRITE, &write);
  69 }
  70
  71 int main(int argc, char **argv)
  72 {
  73         int fd;
  74         struct mmfs_alloc_args alloc;
  75         uint8_t expected[MMFS_BUFFER_SIZE];
  76         uint8_t buf[MMFS_BUFFER_SIZE];
  77         int ret;
  78         int handle;
  79
  80         fd = open_mmfs_device();
  81
  82         memset(&alloc, 0, sizeof(alloc));
  83         alloc.size = MMFS_BUFFER_SIZE;
  84         ret = ioctl(fd, MMFS_IOCTL_ALLOC, &alloc);
  85         assert(ret == 0);
  86         handle = alloc.handle;
  87
  88         printf("Testing contents of newly allocated object.\n");
  89         ret = do_read(fd, handle, buf, 0, MMFS_BUFFER_SIZE);
  90         assert(ret == 0);
  91         memset(&expected, 0, sizeof(expected));
  92         assert(memcmp(expected, buf, sizeof(expected)) == 0);
  93
  94         printf("Testing read beyond end of buffer.\n");
  95         ret = do_read(fd, handle, buf, MMFS_BUFFER_SIZE / 2, MMFS_BUFFER_SIZE);
  96         assert(ret == -1 && errno == EINVAL);
  97
  98         printf("Testing full write of buffer\n");
  99         memset(buf, 0, sizeof(buf));
 100         memset(buf + 1024, 0x01, 1024);
 101         memset(expected + 1024, 0x01, 1024);
 102         ret = do_write(fd, handle, buf, 0, MMFS_BUFFER_SIZE);
 103         assert(ret == 0);
 104         ret = do_read(fd, handle, buf, 0, MMFS_BUFFER_SIZE);
 105         assert(ret == 0);
 106         assert(memcmp(buf, expected, sizeof(buf)) == 0);
 107
 108         printf("Testing partial write of buffer\n");
 109         memset(buf + 4096, 0x02, 1024);
 110         memset(expected + 4096, 0x02, 1024);
 111         ret = do_write(fd, handle, buf + 4096, 4096, 1024);
 112         assert(ret == 0);
 113         ret = do_read(fd, handle, buf, 0, MMFS_BUFFER_SIZE);
 114         assert(ret == 0);
 115         assert(memcmp(buf, expected, sizeof(buf)) == 0);
 116
 117         printf("Testing partial read of buffer\n");
 118         ret = do_read(fd, handle, buf, 512, 1024);
 119         assert(ret == 0);
 120         assert(memcmp(buf, expected + 512, 1024) == 0);
 121
 122         close(fd);
 123
 124         return 0;
 125 }