tools/testing/selftests/drivers/dma-buf/udmabuf.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 #define _GNU_SOURCE
   3 #define __EXPORTED_HEADERS__
   4
   5 #include <stdio.h>
   6 #include <stdlib.h>
   7 #include <unistd.h>
   8 #include <string.h>
   9 #include <errno.h>
  10 #include <fcntl.h>
  11 #include <malloc.h>
  12
  13 #include <sys/ioctl.h>
  14 #include <sys/syscall.h>
  15 #include <linux/memfd.h>
  16 #include <linux/udmabuf.h>
  17
  18 #define TEST_PREFIX     "drivers/dma-buf/udmabuf"
  19 #define NUM_PAGES       4
  20
  21 static int memfd_create(const char *name, unsigned int flags)
  22 {
  23         return syscall(__NR_memfd_create, name, flags);
  24 }
  25
  26 int main(int argc, char *argv[])
  27 {
  28         struct udmabuf_create create;
  29         int devfd, memfd, buf, ret;
  30         off_t size;
  31         void *mem;
  32
  33         devfd = open("/dev/udmabuf", O_RDWR);
  34         if (devfd < 0) {
  35                 printf("%s: [skip,no-udmabuf: Unable to access DMA buffer device file]\n",
  36                        TEST_PREFIX);
  37                 exit(77);
  38         }
  39
  40         memfd = memfd_create("udmabuf-test", MFD_ALLOW_SEALING);
  41         if (memfd < 0) {
  42                 printf("%s: [skip,no-memfd]\n", TEST_PREFIX);
  43                 exit(77);
  44         }
  45
  46         ret = fcntl(memfd, F_ADD_SEALS, F_SEAL_SHRINK);
  47         if (ret < 0) {
  48                 printf("%s: [skip,fcntl-add-seals]\n", TEST_PREFIX);
  49                 exit(77);
  50         }
  51
  52
  53         size = getpagesize() * NUM_PAGES;
  54         ret = ftruncate(memfd, size);
  55         if (ret == -1) {
  56                 printf("%s: [FAIL,memfd-truncate]\n", TEST_PREFIX);
  57                 exit(1);
  58         }
  59
  60         memset(&create, 0, sizeof(create));
  61
  62         /* should fail (offset not page aligned) */
  63         create.memfd  = memfd;
  64         create.offset = getpagesize()/2;
  65         create.size   = getpagesize();
  66         buf = ioctl(devfd, UDMABUF_CREATE, &create);
  67         if (buf >= 0) {
  68                 printf("%s: [FAIL,test-1]\n", TEST_PREFIX);
  69                 exit(1);
  70         }
  71
  72         /* should fail (size not multiple of page) */
  73         create.memfd  = memfd;
  74         create.offset = 0;
  75         create.size   = getpagesize()/2;
  76         buf = ioctl(devfd, UDMABUF_CREATE, &create);
  77         if (buf >= 0) {
  78                 printf("%s: [FAIL,test-2]\n", TEST_PREFIX);
  79                 exit(1);
  80         }
  81
  82         /* should fail (not memfd) */
  83         create.memfd  = 0; /* stdin */
  84         create.offset = 0;
  85         create.size   = size;
  86         buf = ioctl(devfd, UDMABUF_CREATE, &create);
  87         if (buf >= 0) {
  88                 printf("%s: [FAIL,test-3]\n", TEST_PREFIX);
  89                 exit(1);
  90         }
  91
  92         /* should work */
  93         create.memfd  = memfd;
  94         create.offset = 0;
  95         create.size   = size;
  96         buf = ioctl(devfd, UDMABUF_CREATE, &create);
  97         if (buf < 0) {
  98                 printf("%s: [FAIL,test-4]\n", TEST_PREFIX);
  99                 exit(1);
 100         }
 101
 102         fprintf(stderr, "%s: ok\n", TEST_PREFIX);
 103         close(buf);
 104         close(memfd);
 105         close(devfd);
 106         return 0;
 107 }