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
33 static int g_waitPid;
34 static int g_backPid;
35 static int g_backPid1;
36
ThreadFunc(void * arg)37 static void *ThreadFunc(void *arg)
38 {
39 int status = 0;
40
41 int ret = waitpid(g_waitPid, &status, 0);
42 if (ret == g_backPid) {
43 status = WEXITSTATUS(status);
44 ICUNIT_GOTO_EQUAL(status, 3, status, EXIT); // 3, assert status equal to this.
45 } else if (ret == g_backPid1) {
46 status = WEXITSTATUS(status);
47 ICUNIT_GOTO_EQUAL(status, 4, status, EXIT); // 4, assert status equal to this.
48 } else {
49 ICUNIT_GOTO_EQUAL(ret, 0, ret, EXIT);
50 }
51
52 ICUNIT_GOTO_EQUAL(status, 3, status, EXIT); // 3, assert status equal to this.
53
54 EXIT:
55 return NULL;
56 }
57
ProcessTest(void)58 static int ProcessTest(void)
59 {
60 int ret;
61 int status = 100;
62 pthread_t newPthread;
63 pid_t pid, pid1, pid2;
64 int count = 0;
65
66 g_waitPid = 0;
67 pid = fork();
68 ICUNIT_GOTO_WITHIN_EQUAL(pid, 0, 100000, pid, EXIT); // 100000, assert pid equal to this.
69 g_waitPid = -1;
70 g_backPid = pid;
71
72 if (pid == 0) {
73 while (count < 20) { // 20, Number of cycles.
74 count++;
75 usleep(1000 * 10 * 2); // 1000, 10, 2, Used to calculate the delay time.
76 }
77 exit(3); // 3, exit args
78 }
79
80 pid1 = fork();
81 ICUNIT_GOTO_WITHIN_EQUAL(pid1, 0, 100000, pid, EXIT); // 100000, assert pid1 equal to this.
82 g_backPid1 = pid1;
83 if (pid1 == 0) {
84 while (count < 15) { // 15, Number of cycles.
85 count++;
86 usleep(1000 * 10 * 2); // 1000, 10, 2, Used to calculate the delay time.
87 }
88 exit(4); // 4, exit args
89 }
90
91 ret = pthread_create(&newPthread, NULL, ThreadFunc, NULL);
92 ICUNIT_ASSERT_EQUAL(ret, 0, ret);
93
94 usleep(1000 * 10 * 2); // 1000, 10, 2, Used to calculate the delay time.
95
96 ret = waitpid(-1, &status, 0);
97 if (ret == pid) {
98 status = WEXITSTATUS(status);
99 ICUNIT_GOTO_EQUAL(status, 3, status, EXIT); // 3, assert status equal to this.
100 } else if (ret == pid1) {
101 status = WEXITSTATUS(status);
102 ICUNIT_GOTO_EQUAL(status, 4, status, EXIT); // 4, assert status equal to this.
103 } else {
104 ICUNIT_GOTO_EQUAL(ret, 0, ret, EXIT);
105 }
106
107 ret = pthread_join(newPthread, NULL);
108 ICUNIT_ASSERT_EQUAL(ret, 0, ret);
109
110 exit(255); // 255, exit args
111 EXIT:
112 return 0;
113 }
114
TestCase(void)115 static int TestCase(void)
116 {
117 int ret;
118 int status = 0;
119 pid_t pid = fork();
120 ICUNIT_GOTO_WITHIN_EQUAL(pid, 0, 100000, pid, EXIT); // 100000, assert pid equal to this.
121 if (pid == 0) {
122 ProcessTest();
123 exit(g_iCunitErrLineNo);
124 }
125
126 ret = waitpid(pid, &status, 0);
127 ICUNIT_GOTO_EQUAL(ret, pid, ret, EXIT);
128 status = WEXITSTATUS(status);
129 ICUNIT_GOTO_EQUAL(status, 255, status, EXIT); // 255, assert status equal to this.
130
131 return 0;
132 EXIT:
133 return 1;
134 }
135
ItTestProcess020(void)136 void ItTestProcess020(void)
137 {
138 TEST_ADD_CASE("IT_POSIX_PROCESS_020", TestCase, TEST_POSIX, TEST_MEM, TEST_LEVEL0, TEST_FUNCTION);
139 }
140