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 volatile int *g_shmptr = NULL;
35
Handler(int sig)36 static void Handler(int sig)
37 {
38 (*g_shmptr)++;
39 }
40
TestCase(void)41 static int TestCase(void)
42 {
43 int status;
44 (void)signal(SIGUSR2, Handler);
45 int ret;
46 pid_t pid;
47
48 int shmid = shmget(IPC_PRIVATE, sizeof(int), IPC_CREAT | 0600); // 0600 config of shmget
49 ICUNIT_ASSERT_NOT_EQUAL(shmid, -1, shmid);
50
51 g_shmptr = (int *)shmat(shmid, nullptr, 0);
52 ICUNIT_GOTO_NOT_EQUAL(g_shmptr, (int *)-1, g_shmptr, EXIT);
53
54 *g_shmptr = 0;
55
56 pid = fork();
57 ICUNIT_GOTO_WITHIN_EQUAL(pid, 0, 100000, pid, EXIT); // 100000, assert pid equal to this.
58
59 if (pid == 0) {
60 pid_t pid1 = fork();
61 ICUNIT_GOTO_WITHIN_EQUAL(pid1, 0, 100000, pid1, EXIT); // 100000, assert pid1 equal to this.
62
63 if (pid1 == 0) {
64 ret = killpg(getpgrp(), SIGUSR2);
65 ICUNIT_ASSERT_EQUAL(ret, 0, ret);
66
67 while (*g_shmptr < 3) { // 3, Number of cycles.
68 sleep(1);
69 }
70
71 exit(11); // 11, exit args
72 }
73
74 while (*g_shmptr < 3) { // 3, Number of cycles.
75 sleep(1);
76 }
77
78 ret = waitpid(pid1, &status, 0);
79 ICUNIT_GOTO_EQUAL(ret, pid1, ret, EXIT);
80
81 status = WEXITSTATUS(status);
82 ICUNIT_GOTO_EQUAL(status, 11, status, EXIT); // 11, assert status equal to this.
83
84 exit(10); // 10, exit args
85 }
86
87 while (*g_shmptr < 3) { // 3, wait function running.
88 sleep(1);
89 }
90 ICUNIT_ASSERT_EQUAL(*g_shmptr, 3, *g_shmptr); // 3, assert that function Result is equal to this.
91
92 ret = waitpid(pid, &status, 0);
93 ICUNIT_GOTO_EQUAL(ret, pid, ret, EXIT);
94
95 status = WEXITSTATUS(status);
96 ICUNIT_GOTO_EQUAL(status, 10, status, EXIT); // 10, assert that function Result is equal to this.
97
98 shmdt(static_cast<void *>(const_cast<int *>(g_shmptr)));
99 shmctl(shmid, IPC_RMID, NULL);
100
101 return 0;
102 EXIT:
103 shmdt(static_cast<void *>(const_cast<int *>(g_shmptr)));
104 shmctl(shmid, IPC_RMID, NULL);
105 return 1;
106 }
107
ItTestProcess062(void)108 void ItTestProcess062(void)
109 {
110 TEST_ADD_CASE("IT_POSIX_PROCESS_062", TestCase, TEST_POSIX, TEST_MEM, TEST_LEVEL0, TEST_FUNCTION);
111 }