1 /*
2 * Copyright (c) 2013-2019 Huawei Technologies Co., Ltd. All rights reserved.
3 * Copyright (c) 2020-2021 Huawei Device Co., Ltd. All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without modification,
6 * are permitted provided that the following conditions are met:
7 *
8 * 1. Redistributions of source code must retain the above copyright notice, this list of
9 * conditions and the following disclaimer.
10 *
11 * 2. Redistributions in binary form must reproduce the above copyright notice, this list
12 * of conditions and the following disclaimer in the documentation and/or other materials
13 * provided with the distribution.
14 *
15 * 3. Neither the name of the copyright holder nor the names of its contributors may be used
16 * to endorse or promote products derived from this software without specific prior written
17 * permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
21 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
23 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
24 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
25 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
26 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
27 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
28 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
29 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30 */
31 #include "it_test_process.h"
32 #include "sys/shm.h"
33
34 static int *g_shmptr = NULL;
35 static int g_usetTestCount = 0;
PthreadTest115(VOID * arg)36 static VOID *PthreadTest115(VOID *arg)
37 {
38 g_usetTestCount++;
39 return NULL;
40 }
41
TestThread(void * arg)42 static int TestThread(void *arg)
43 {
44 int data = *((int *)arg);
45 int ret = 0;
46 pthread_t gTh;
47 ret = pthread_create(&gTh, NULL, PthreadTest115, NULL);
48 ICUNIT_ASSERT_EQUAL(ret, 0, ret);
49
50 ret = pthread_join(gTh, NULL);
51 ICUNIT_ASSERT_EQUAL(ret, 0, ret);
52 ICUNIT_ASSERT_EQUAL(g_usetTestCount, 1, g_usetTestCount);
53
54 *g_shmptr = 100; // 100, set shared num.
55
56 pid_t pid = fork();
57
58 ICUNIT_GOTO_WITHIN_EQUAL(pid, 0, 100000, pid, EXIT); // 100000, assert pid equal to this.
59 if (pid == 0) {
60 sleep(1);
61 exit(0);
62 }
63
64 *g_shmptr = 200; // 200, set shared num.
65
66 ret = waitpid(pid, NULL, 0);
67 printf("waitpid ret : %d errno : %d pid : %d getpid : %d\n", ret, errno, pid, getpid());
68
69 ICUNIT_ASSERT_EQUAL(ret, pid, ret);
70
71 *g_shmptr = data;
72 EXIT:
73 return 0;
74 }
75
76 // This testcase us used for undefination of LOSCFG_USER_TEST_SMP
Testcase(void)77 static int Testcase(void)
78 {
79 int arg = 0x2000;
80 int status;
81 int ret;
82 char *stackTop;
83 char *stack;
84 pid_t pid;
85
86 g_usetTestCount = 0;
87 int shmid = shmget(IPC_PRIVATE, sizeof(int), IPC_CREAT | 0600); // // 0600, set shmget config.
88 ICUNIT_ASSERT_NOT_EQUAL(shmid, -1, shmid);
89
90 g_shmptr = (int *)shmat(shmid, nullptr, 0);
91 ICUNIT_ASSERT_NOT_EQUAL(g_shmptr, reinterpret_cast<int *>(-1), g_shmptr);
92
93 *g_shmptr = 0;
94
95 stack = static_cast<char *>(malloc(arg));
96 ICUNIT_GOTO_NOT_EQUAL(stack, NULL, stack, EXIT1);
97
98 stackTop = reinterpret_cast<char *>(reinterpret_cast<unsigned long>(stack) + arg);
99 pid = clone(TestThread, (void *)stackTop, CLONE_VFORK, &arg);
100
101 ICUNIT_GOTO_EQUAL(*g_shmptr, arg, *g_shmptr, EXIT2);
102
103 ret = waitpid(pid, &status, 0);
104 ICUNIT_GOTO_EQUAL(ret, pid, ret, EXIT2);
105
106 EXIT2:
107 free(stack);
108
109 EXIT1:
110 shmdt(g_shmptr);
111 shmctl(shmid, IPC_RMID, NULL);
112
113 return 0;
114 }
115
ItTestProcess049(void)116 void ItTestProcess049(void)
117 {
118 TEST_ADD_CASE("IT_POSIX_PROCESS_049", Testcase, TEST_POSIX, TEST_MEM, TEST_LEVEL0, TEST_FUNCTION);
119 }
120