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 <arpa/inet.h>
24#include <gtest/gtest.h>
25#include <netinet/in.h>
26#include <sys/stat.h>
27#include <sys/socket.h>
28#include <sys/types.h>
29
30using namespace testing::ext;
31
32class HatsTruncateTest : public testing::Test {
33public:
34static void SetUpTestCase();
35static void TearDownTestCase();
36void SetUp();
37void TearDown();
38private:
39};
40void HatsTruncateTest::SetUp()
41{
42}
43void HatsTruncateTest::TearDown()
44{
45}
46void HatsTruncateTest::SetUpTestCase()
47{
48}
49void HatsTruncateTest::TearDownTestCase()
50{
51}
52
53const char *TRUNCATE_TEST_FILE = "/data/local/tmp/tryTruncate.txt";
54const char *TEST_DATA = "1234567890";
55const int TEST_DATA_LEN = strlen(TEST_DATA);
56
57
58/*
59 * @tc.number : SUB_KERNEL_SYSCALL_TRUNCATE_0100
60 * @tc.name   : TruncateFileSuccess_0001
61 * @tc.desc   : Truncate file success.
62 * @tc.size   : MediumTest
63 * @tc.type   : Function
64 * @tc.level  : Level 1
65 */
66HWTEST_F(HatsTruncateTest, TruncateFileSuccess_0001, Function | MediumTest | Level1)
67{
68    int fd;
69    int ret;
70    char *buf = nullptr;
71    buf = new char[TEST_DATA_LEN / 2 + 1];
72
73    fd = open(TRUNCATE_TEST_FILE, O_WRONLY | O_CREAT | O_TRUNC, 0644);
74    EXPECT_TRUE(fd >= 3);
75    ret = write(fd, TEST_DATA, TEST_DATA_LEN);
76    EXPECT_TRUE(ret == TEST_DATA_LEN);
77    close(fd);
78
79    ret = truncate(TRUNCATE_TEST_FILE, TEST_DATA_LEN / 2);
80    EXPECT_TRUE(ret == 0);
81
82    fd = open(TRUNCATE_TEST_FILE, O_RDONLY, 0644);
83    EXPECT_TRUE(fd >= 3);
84    ret = read(fd, buf, TEST_DATA_LEN);
85    EXPECT_TRUE(ret == TEST_DATA_LEN / 2);
86    buf[TEST_DATA_LEN / 2] = '\0';
87    ret = strcmp(buf, "12345");
88    EXPECT_TRUE(ret == 0);
89    close(fd);
90
91    delete[] buf;
92}
93
94/*
95 * @tc.number : SUB_KERNEL_SYSCALL_TRUNCATE_0200
96 * @tc.name   : TruncateNotExistFileFail_0002
97 * @tc.desc   : Truncate file that does not exist fail.
98 * @tc.size   : MediumTest
99 * @tc.type   : Function
100 * @tc.level  : Level 2
101 */
102HWTEST_F(HatsTruncateTest, TruncateNotExistFileFail_0002, Function | MediumTest | Level2)
103{
104    int ret;
105
106    ret = remove(TRUNCATE_TEST_FILE);
107    EXPECT_TRUE(ret == 0);
108
109    errno = 0;
110    ret = truncate(TRUNCATE_TEST_FILE, TEST_DATA_LEN / 2);
111    EXPECT_TRUE(ret == -1);
112    EXPECT_EQ(errno, ENOENT);
113}
114
115/*
116 * @tc.number : SUB_KERNEL_SYSCALL_TRUNCATE_0300
117 * @tc.name   : TruncateExtendFileSuccess_0003
118 * @tc.desc   : Truncate extend file success.
119 * @tc.size   : MediumTest
120 * @tc.type   : Function
121 * @tc.level  : Level 1
122 */
123HWTEST_F(HatsTruncateTest, TruncateExtendFileSuccess_0003, Function | MediumTest | Level1)
124{
125    int fd;
126    int ret;
127    struct stat fileStat;
128    ssize_t fileSize1;
129    ssize_t fileSize2;
130
131    fd = open(TRUNCATE_TEST_FILE, O_WRONLY | O_CREAT | O_TRUNC, 0644);
132    EXPECT_TRUE(fd >= 3);
133    ret = write(fd, TEST_DATA, TEST_DATA_LEN);
134    EXPECT_TRUE(ret == TEST_DATA_LEN);
135    close(fd);
136
137    ret = stat(TRUNCATE_TEST_FILE, &fileStat);
138    EXPECT_TRUE(ret == 0);
139    fileSize1 = fileStat.st_size;
140
141    ret = truncate(TRUNCATE_TEST_FILE, fileSize1 * 2);
142    EXPECT_TRUE(ret == 0);
143
144    ret = stat(TRUNCATE_TEST_FILE, &fileStat);
145    EXPECT_TRUE(ret == 0);
146    fileSize2 = fileStat.st_size;
147
148    ret = (fileSize1 * 2 == fileSize2);
149    EXPECT_TRUE(ret);
150}
151
152/*
153 * @tc.number : SUB_KERNEL_SYSCALL_TRUNCATE_0400
154 * @tc.name   : TruncateInvaliLenFail_0004
155 * @tc.desc   : Truncate using invalid length fail.
156 * @tc.size   : MediumTest
157 * @tc.type   : Function
158 * @tc.level  : Level 2
159 */
160HWTEST_F(HatsTruncateTest, TruncateInvaliLenFail_0004, Function | MediumTest | Level2)
161{
162    int ret;
163    int truncateLen;
164
165    truncateLen = -1;
166    errno = 0;
167    ret = truncate(TRUNCATE_TEST_FILE, truncateLen);
168    EXPECT_TRUE(ret == -1);
169    EXPECT_EQ(errno, EINVAL);
170}