1 /*
2 * Copyright (C) 2024 HiHope Open Source Organization.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16 #include <cerrno>
17 #include <cstdio>
18 #include <cstdlib>
19 #include <string>
20 #include <vector>
21 #include <fcntl.h>
22 #include <unistd.h>
23 #include <gtest/gtest.h>
24 #include <sys/file.h>
25 #include <sys/stat.h>
26 #include <sys/types.h>
27 #include <sys/xattr.h>
28 #include "securec.h"
29
30 using namespace testing::ext;
31 using namespace std;
32
33 static const char *TEST_FILE = "/data/local/tmp/flock.txt";
34 mode_t MODE_0600 = S_IRUSR | S_IWUSR;
35 mode_t MODE_0644 = S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH;
36
37 class FlockApiTest : public testing::Test {
38 public:
39 static void SetUpTestCase();
40 static void TearDownTestCase();
41 void SetUp();
42 void TearDown();
43 private:
44 };
SetUp()45 void FlockApiTest::SetUp()
46 {
47 }
TearDown()48 void FlockApiTest::TearDown()
49 {
50 }
SetUpTestCase()51 void FlockApiTest::SetUpTestCase()
52 {
53 }
TearDownTestCase()54 void FlockApiTest::TearDownTestCase()
55 {
56 }
57
58 /*
59 * @tc.number : SUB_KERNEL_SYSCALL_FLOCK_0100
60 * @tc.name : FlockValidFdSuccess_0001
61 * @tc.desc : flock lock valid file with share lock and exclusive lock success.
62 * @tc.size : MediumTest
63 * @tc.type : Function
64 * @tc.level : Level 1
65 */
HWTEST_F(FlockApiTest, FlockValidFdSuccess_0001, Function | MediumTest | Level1)66 HWTEST_F(FlockApiTest, FlockValidFdSuccess_0001, Function | MediumTest | Level1)
67 {
68 int ret = -1;
69 int fd = open(TEST_FILE, O_RDWR | O_CREAT | O_TRUNC, MODE_0644);
70 EXPECT_TRUE(fd > 0);
71
72 ret = flock(fd, LOCK_SH);
73 EXPECT_EQ(ret, 0);
74 ret = flock(fd, LOCK_UN);
75 EXPECT_EQ(ret, 0);
76
77 ret = flock(fd, LOCK_EX);
78 EXPECT_EQ(ret, 0);
79 ret = flock(fd, LOCK_UN);
80 EXPECT_EQ(ret, 0);
81
82 close(fd);
83 remove(TEST_FILE);
84 }
85
86 /*
87 * @tc.number : SUB_KERNEL_SYSCALL_FLOCK_0200
88 * @tc.name : FlockInvalidFdFail_0002
89 * @tc.desc : flock lock invalid file with share lock fail, errno EBADF.
90 * @tc.size : MediumTest
91 * @tc.type : Function
92 * @tc.level : Level 2
93 */
HWTEST_F(FlockApiTest, FlockInvalidFdFail_0002, Function | MediumTest | Level2)94 HWTEST_F(FlockApiTest, FlockInvalidFdFail_0002, Function | MediumTest | Level2)
95 {
96 int ret = -1;
97 int invalidFd = -1;
98
99 errno = 0;
100 ret = flock(invalidFd, LOCK_SH);
101 EXPECT_EQ(ret, -1);
102 EXPECT_EQ(errno, EBADF);
103 }
104
105 /*
106 * @tc.number : SUB_KERNEL_SYSCALL_FLOCK_0300
107 * @tc.name : FlockLockTwiceFail_0003
108 * @tc.desc : flock try to acquire an exclusive lock on the second descriptor fail, errno EWOULDBLOCK.
109 * @tc.size : MediumTest
110 * @tc.type : Function
111 * @tc.level : Level 2
112 */
HWTEST_F(FlockApiTest, FlockLockTwiceFail_0003, Function | MediumTest | Level2)113 HWTEST_F(FlockApiTest, FlockLockTwiceFail_0003, Function | MediumTest | Level2)
114 {
115 int ret = -1;
116 int fd1 = open(TEST_FILE, O_RDWR | O_CREAT | O_TRUNC, MODE_0600);
117 EXPECT_TRUE(fd1 > 0);
118 int fd2 = open(TEST_FILE, O_RDWR | O_CREAT | O_TRUNC, MODE_0600);
119 EXPECT_TRUE(fd2 > 0);
120
121 ret = flock(fd1, LOCK_EX);
122 EXPECT_EQ(ret, 0);
123
124 errno = 0;
125 ret = flock(fd2, LOCK_EX | LOCK_NB);
126 EXPECT_NE(ret, 0);
127 EXPECT_EQ(errno, EWOULDBLOCK);
128
129 flock(fd1, LOCK_UN);
130 close(fd1);
131 close(fd2);
132 remove(TEST_FILE);
133 }
134