tests/mm_mmap.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 "drm.h"
  37
  38 #define OBJECT_SIZE 16384
  39
  40 int do_read(int fd, int handle, void *buf, int offset, int size)
  41 {
  42         struct drm_mm_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, DRM_IOCTL_MM_PREAD, &read);
  56 }
  57
  58 int do_write(int fd, int handle, void *buf, int offset, int size)
  59 {
  60         struct drm_mm_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, DRM_IOCTL_MM_PWRITE, &write);
  69 }
  70
  71 int main(int argc, char **argv)
  72 {
  73         int fd;
  74         struct drm_mm_alloc_args alloc;
  75         struct drm_mm_mmap_args mmap;
  76         struct drm_mm_unreference_args unref;
  77         uint8_t expected[OBJECT_SIZE];
  78         uint8_t buf[OBJECT_SIZE];
  79         int ret;
  80         int handle;
  81
  82         fd = drm_open_any();
  83
  84         memset(&mmap, 0, sizeof(mmap));
  85         mmap.handle = 0x10101010;
  86         mmap.offset = 0;
  87         mmap.size = 4096;
  88         printf("Testing mmaping of bad object.\n");
  89         ret = ioctl(fd, DRM_IOCTL_MM_MMAP, &mmap);
  90         assert(ret == -1 && errno == EINVAL);
  91
  92         memset(&alloc, 0, sizeof(alloc));
  93         alloc.size = OBJECT_SIZE;
  94         ret = ioctl(fd, DRM_IOCTL_MM_ALLOC, &alloc);
  95         assert(ret == 0);
  96         handle = alloc.handle;
  97
  98         printf("Testing mmaping of newly allocated object.\n");
  99         mmap.handle = handle;
 100         mmap.offset = 0;
 101         mmap.size = OBJECT_SIZE;
 102         ret = ioctl(fd, DRM_IOCTL_MM_MMAP, &mmap);
 103         assert(ret == 0);
 104
 105         printf("Testing contents of newly allocated object.\n");
 106         memset(expected, 0, sizeof(expected));
 107         assert(memcmp(mmap.addr, expected, sizeof(expected)) == 0);
 108
 109         printf("Testing coherency of writes and mmap reads.\n");
 110         memset(buf, 0, sizeof(buf));
 111         memset(buf + 1024, 0x01, 1024);
 112         memset(expected + 1024, 0x01, 1024);
 113         ret = do_write(fd, handle, buf, 0, OBJECT_SIZE);
 114         assert(ret == 0);
 115         assert(memcmp(buf, mmap.addr, sizeof(buf)) == 0);
 116
 117         printf("Testing that mapping stays after unreference\n");
 118         unref.handle = handle;
 119         ret = ioctl(fd, DRM_IOCTL_MM_UNREFERENCE, &unref);
 120         assert(ret == 0);
 121         assert(memcmp(buf, mmap.addr, sizeof(buf)) == 0);
 122
 123         printf("Testing unmapping\n");
 124         munmap(mmap.addr, OBJECT_SIZE);
 125
 126         close(fd);
 127
 128         return 0;
 129 }