about summary refs log tree commit diff
path: root/io/tst-fcntl-lock.c
blob: 3c32ca32253401e3788114c2f885527032d30b46 (plain) (blame)
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
/* Test for advisory record locking.
   Copyright (C) 2023-2024 Free Software Foundation, Inc.

   This program is free software; you can redistribute it and/or
   modify it under the terms of the GNU General Public License
   as published by the Free Software Foundation; either version 2
   of the License, or (at your option) any later version.

   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program; if not, see <https://www.gnu.org/licenses/>.
*/

#include <fcntl.h>
#include <errno.h>
#include <unistd.h>

/* This is essentially the POSIX lockf.  */

static int
fcntl_lockf (int fd, int cmd, off_t len)
{
  struct flock fl = {
    .l_type = F_WRLCK,
    .l_whence = SEEK_CUR,
    .l_len = len
  };

  switch (cmd)
    {
    case F_TEST:
      fl.l_type = F_RDLCK;
      if (fcntl (fd, F_GETLK, &fl) < 0)
	return -1;
      if (fl.l_type == F_UNLCK || fl.l_pid == getpid ())
	return 0;
      errno = EACCES;
      return -1;

    case F_ULOCK:
      fl.l_type = F_UNLCK;
      return fcntl (fd, F_SETLK, &fl);

    case F_LOCK:
      return fcntl (fd, F_SETLKW, &fl);

    case F_TLOCK:
      return fcntl (fd, F_SETLK, &fl);
    }

  errno = EINVAL;
  return -1;
}

static int
fcntl64_lockf (int fd, int cmd, off64_t len64)
  {
  struct flock64 fl64 = {
    .l_type = F_WRLCK,
    .l_whence = SEEK_CUR,
    .l_len = len64
  };

  switch (cmd)
    {
    case F_TEST:
      fl64.l_type = F_RDLCK;
      if (fcntl64 (fd, F_GETLK64, &fl64) < 0)
	return -1;
      if (fl64.l_type == F_UNLCK || fl64.l_pid == getpid ())
	return 0;
      errno = EACCES;
      return -1;

    case F_ULOCK:
      fl64.l_type = F_UNLCK;
      return fcntl64 (fd, F_SETLK64, &fl64);

    case F_LOCK:
      return fcntl64 (fd, F_SETLKW64, &fl64);

    case F_TLOCK:
      return fcntl64 (fd, F_SETLK64, &fl64);
    }

  errno = EINVAL;
  return -1;
}

#define TST_LOCKFD  "tst-fcntl-lock."
#define LOCKF       fcntl_lockf
#define LOCKF64     fcntl64_lockf
#include "tst-lockf.c"