add REDIRECT

hc

2024-10-09 05e59e5fb0064c97a1c10921ecd549f2d4a58565

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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
// SPDX-License-Identifier: GPL-2.0
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <errno.h>
#include <linux/fcntl.h>
#include <malloc.h>
 
#include <sys/ioctl.h>
#include <sys/syscall.h>
#include <linux/memfd.h>
#include <linux/udmabuf.h>
 
#define TEST_PREFIX    "drivers/dma-buf/udmabuf"
#define NUM_PAGES       4
 
static int memfd_create(const char *name, unsigned int flags)
{
    return syscall(__NR_memfd_create, name, flags);
}
 
int main(int argc, char *argv[])
{
    struct udmabuf_create create;
    int devfd, memfd, buf, ret;
    off_t size;
    void *mem;
 
    devfd = open("/dev/udmabuf", O_RDWR);
    if (devfd < 0) {
        printf("%s: [skip,no-udmabuf]\n", TEST_PREFIX);
        exit(77);
    }
 
    memfd = memfd_create("udmabuf-test", MFD_ALLOW_SEALING);
    if (memfd < 0) {
        printf("%s: [skip,no-memfd]\n", TEST_PREFIX);
        exit(77);
    }
 
    ret = fcntl(memfd, F_ADD_SEALS, F_SEAL_SHRINK);
    if (ret < 0) {
        printf("%s: [skip,fcntl-add-seals]\n", TEST_PREFIX);
        exit(77);
    }
 
 
    size = getpagesize() * NUM_PAGES;
    ret = ftruncate(memfd, size);
    if (ret == -1) {
        printf("%s: [FAIL,memfd-truncate]\n", TEST_PREFIX);
        exit(1);
    }
 
    memset(&create, 0, sizeof(create));
 
    /* should fail (offset not page aligned) */
    create.memfd  = memfd;
    create.offset = getpagesize()/2;
    create.size   = getpagesize();
    buf = ioctl(devfd, UDMABUF_CREATE, &create);
    if (buf >= 0) {
        printf("%s: [FAIL,test-1]\n", TEST_PREFIX);
        exit(1);
    }
 
    /* should fail (size not multiple of page) */
    create.memfd  = memfd;
    create.offset = 0;
    create.size   = getpagesize()/2;
    buf = ioctl(devfd, UDMABUF_CREATE, &create);
    if (buf >= 0) {
        printf("%s: [FAIL,test-2]\n", TEST_PREFIX);
        exit(1);
    }
 
    /* should fail (not memfd) */
    create.memfd  = 0; /* stdin */
    create.offset = 0;
    create.size   = size;
    buf = ioctl(devfd, UDMABUF_CREATE, &create);
    if (buf >= 0) {
        printf("%s: [FAIL,test-3]\n", TEST_PREFIX);
        exit(1);
    }
 
    /* should work */
    create.memfd  = memfd;
    create.offset = 0;
    create.size   = size;
    buf = ioctl(devfd, UDMABUF_CREATE, &create);
    if (buf < 0) {
        printf("%s: [FAIL,test-4]\n", TEST_PREFIX);
        exit(1);
    }
 
    fprintf(stderr, "%s: ok\n", TEST_PREFIX);
    close(buf);
    close(memfd);
    close(devfd);
    return 0;
}