1 /*
2 * Copyright (c) 2023-2023 Huawei Device Co., Ltd. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without modification,
5 * are permitted provided that the following conditions are met:
6 *
7 * 1. Redistributions of source code must retain the above copyright notice, this list of
8 * conditions and the following disclaimer.
9 *
10 * 2. Redistributions in binary form must reproduce the above copyright notice, this list
11 * of conditions and the following disclaimer in the documentation and/or other materials
12 * provided with the distribution.
13 *
14 * 3. Neither the name of the copyright holder nor the names of its contributors may be used
15 * to endorse or promote products derived from this software without specific prior written
16 * permission.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
20 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
21 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
22 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
23 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
24 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
25 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
26 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
27 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
28 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29 */
30 #include <fcntl.h>
31 #include <cstdio>
32 #include <cstdlib>
33 #include <sys/stat.h>
34 #include <sys/types.h>
35 #include <unistd.h>
36 #include <cstring>
37 #include <sys/shm.h>
38 #include <gtest/gtest.h>
39 #include "It_process_plimits.h"
40
41 static const int g_buffSize = 512;
42 static const int g_arryLen = 4;
43 static const int g_readLen = 254;
44
ItProcessPlimitsIpc010(void)45 void ItProcessPlimitsIpc010(void)
46 {
47 mode_t mode;
48 mode_t shmAcess = 0666;
49 char buf[g_buffSize] = { 0 };
50 char *array[g_arryLen] = { nullptr };
51 int mqSuccessCount;
52 int mqFailedCount;
53 int shmSuccessSize;
54 int shmFailedCount;
55 int shmid = -1;
56 int shmid_1 = -1;
57 int index = 0;
58 int shmCkeckSize = PAGE_SIZE * 9;
59 void *shared = nullptr;
60 size_t shmSize = PAGE_SIZE * 9;
61 std::string subPlimitsPath = "/proc/plimits/test";
62 std::string configFileShmSize = "/proc/plimits/test/ipc.shm_limit";
63 char shmemLimit[6] = "40960";
64
65 int ret = mkdir(subPlimitsPath.c_str(), S_IFDIR | mode);
66 ASSERT_EQ(ret, 0);
67
68 ret = WriteFile(configFileShmSize.c_str(), shmemLimit);
69 ASSERT_NE(ret, -1);
70
71 shmid = shmget(IPC_PRIVATE, shmSize, shmAcess | IPC_CREAT);
72 ASSERT_NE(shmid, -1);
73 shmid_1 = shmget(IPC_PRIVATE, 2 * PAGE_SIZE, shmAcess | IPC_CREAT); /* 2: PAGE num */
74 ASSERT_EQ(shmid_1, -1);
75
76 ret = ReadFile("/proc/plimits/test/ipc.stat", buf);
77 GetLine(buf, g_arryLen, g_readLen, array);
78 mqSuccessCount = atoi(array[index++] + strlen("mq count: "));
79 mqFailedCount = atoi(array[index++] + strlen("mq failed count: "));
80 shmSuccessSize = atoi(array[index++] + strlen("shm size: "));
81 shmFailedCount = atoi(array[index++] + strlen("shm failed count: "));
82
83 ASSERT_EQ(shmSuccessSize, shmCkeckSize);
84 ASSERT_EQ(shmFailedCount, 1);
85
86 shared = shmat(shmid, nullptr, 0);
87 ASSERT_NE(shared, (void *)-1);
88
89 ret = shmdt(shared);
90 ASSERT_NE(ret, -1);
91
92 ret = shmctl(shmid, IPC_RMID, nullptr);
93 ASSERT_NE(ret, -1);
94 shmctl(shmid_1, IPC_RMID, nullptr);
95
96 ret = rmdir(subPlimitsPath.c_str());
97 ASSERT_EQ(ret, 0);
98 return;
99 }
100