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 <mqueue.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 MQ_MAX_PER_PROCESS_COUNT = 255;
44 static const int MQUEUE_STANDARD_NAME_LENGTH = 50;
45
ItProcessPlimitsIpc005(void)46 void ItProcessPlimitsIpc005(void)
47 {
48 INT32 ret;
49 mode_t mode;
50 char buf[g_buffSize] = { 0 };
51 CHAR mqname[MQUEUE_STANDARD_NAME_LENGTH] = "";
52 mqd_t mqueue[MQ_MAX_PER_PROCESS_COUNT];
53 struct mq_attr attr = { 0 };
54 int index = 0;
55 attr.mq_msgsize = MQUEUE_STANDARD_NAME_LENGTH;
56 attr.mq_maxmsg = 1;
57
58 std::string plimitsPath = "/proc/plimits/test";
59
60 ret = mkdir(plimitsPath.c_str(), S_IFDIR | mode);
61 ASSERT_EQ(ret, 0);
62
63 ret = ReadFile("/proc/plimits/test/ipc.mq_limit", buf);
64 ASSERT_STREQ(buf, "1024\n");
65
66 for (index = 0; index < MQ_MAX_PER_PROCESS_COUNT; ++index) {
67 ret = snprintf_s(mqname, sizeof(mqname), MQUEUE_STANDARD_NAME_LENGTH, "/mq_005_%d_%d",
68 index, LosCurTaskIDGet());
69 ASSERT_NE(ret, -1);
70 mqueue[index] = mq_open(mqname, O_CREAT | O_RDWR, S_IRUSR | S_IWUSR, &attr);
71 ASSERT_NE(mqueue[index], (mqd_t)-1);
72 }
73 for (int k = 0; k < MQ_MAX_PER_PROCESS_COUNT; ++k) {
74 ret = mq_close(mqueue[k]);
75 ASSERT_NE(ret, -1);
76 ret = snprintf_s(mqname, sizeof(mqname), MQUEUE_STANDARD_NAME_LENGTH, "/mq_005_%d_%d", k, LosCurTaskIDGet());
77 ASSERT_NE(ret, -1);
78 ret = mq_unlink(mqname);
79 ASSERT_EQ(ret, 0);
80 }
81
82 ret = rmdir(plimitsPath.c_str());
83 ASSERT_EQ(ret, 0);
84 return;
85 }
86