summaryrefslogtreecommitdiffstats
path: root/tests/bugs/glusterfs/getlk_owner.c
blob: cbe277318c12d90d8829a888c457ea33bdbcf667 (plain)
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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
#include <stdio.h>
#include <unistd.h>
#include <fcntl.h>
#include <string.h>

#define GETLK_OWNER_CHECK(f, cp, label)                                        \
    do {                                                                       \
        switch (f.l_type) {                                                    \
            case F_RDLCK:                                                      \
            case F_WRLCK:                                                      \
                ret = 1;                                                       \
                goto label;                                                    \
            case F_UNLCK:                                                      \
                if (!are_flocks_sane(&f, &cp)) {                               \
                    ret = 1;                                                   \
                    goto label;                                                \
                }                                                              \
                break;                                                         \
        }                                                                      \
    } while (0)

void
flock_init(struct flock *f, short int type, off_t start, off_t len)
{
    f->l_type = type;
    f->l_start = start;
    f->l_len = len;
}

int
flock_cp(struct flock *dst, struct flock *src)
{
    memcpy((void *)dst, (void *)src, sizeof(struct flock));
}

int
are_flocks_sane(struct flock *src, struct flock *cpy)
{
    return ((src->l_whence == cpy->l_whence) &&
            (src->l_start == cpy->l_start) && (src->l_len == cpy->l_len));
}

/*
 * Test description:
 * SETLK (0,3), F_WRLCK
 * SETLK (3,3), F_WRLCK
 *
 * the following GETLK requests must return flock struct unmodified
 * except for l_type to F_UNLCK
 * GETLK (3,3), F_WRLCK
 * GETLK (3,3), F_RDLCK
 *
 * */

int
main(int argc, char **argv)
{
    int fd = -1;
    int ret = 1;
    char *fname = NULL;
    struct flock f = {
        0,
    };
    struct flock cp = {
        0,
    };

    if (argc < 2)
        goto out;

    fname = argv[1];
    fd = open(fname, O_RDWR);
    if (fd == -1) {
        perror("open");
        goto out;
    }

    flock_init(&f, F_WRLCK, 0, 3);
    flock_cp(&cp, &f);
    ret = fcntl(fd, F_SETLK, &f);
    if (ret) {
        perror("fcntl");
        goto out;
    }
    if (!are_flocks_sane(&f, &cp)) {
        ret = 1;
        goto out;
    }

    flock_init(&f, F_WRLCK, 3, 3);
    flock_cp(&cp, &f);
    ret = fcntl(fd, F_SETLK, &f);
    if (ret) {
        perror("fcntl");
        goto out;
    }
    if (!are_flocks_sane(&f, &cp)) {
        ret = 1;
        goto out;
    }

    flock_init(&f, F_WRLCK, 3, 3);
    flock_cp(&cp, &f);
    ret = fcntl(fd, F_GETLK, &f);
    if (ret) {
        perror("fcntl");
        return 1;
    }
    GETLK_OWNER_CHECK(f, cp, out);

    flock_init(&f, F_RDLCK, 3, 3);
    flock_cp(&cp, &f);
    ret = fcntl(fd, F_GETLK, &f);
    if (ret) {
        perror("fcntl");
        return 1;
    }
    GETLK_OWNER_CHECK(f, cp, out);

out:
    if (fd != -1)
        close(fd);
    return ret;
}