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_pthread_test.h"
32 #define TEST_THREAD_COUNT 2000
33
34 static int g_testCnt = 0;
35
PthreadTest116(VOID * arg)36 static void *PthreadTest116(VOID *arg)
37 {
38 return NULL;
39 }
40
PthreadTest115(VOID * arg)41 static void *PthreadTest115(VOID *arg)
42 {
43 pthread_t th1;
44
45 int ret = pthread_create(&th1, NULL, PthreadTest116, 0);
46 ICUNIT_ASSERT_EQUAL_NULL(ret, 0, (void *)(intptr_t)ret);
47
48 ret = pthread_join(th1, NULL);
49 ICUNIT_ASSERT_EQUAL_NULL(ret, 0, (void *)(intptr_t)ret);
50
51 g_testCnt++;
52
53 return NULL;
54 }
55
GroupProcess(void)56 static int GroupProcess(void)
57 {
58 int ret;
59 pthread_t th1;
60 pthread_attr_t attr;
61 pthread_attr_init(&attr);
62 int policy = 0;
63 unsigned int *getStack = NULL;
64 size_t getSize;
65 struct sched_param param = { 0 };
66 int stackSize = 0x3000;
67
68 g_testCnt = 0;
69
70 void *stack = malloc(stackSize);
71 ICUNIT_ASSERT_NOT_EQUAL((int)stack, 0, stack);
72
73 ret = pthread_attr_setstack(&attr, stack, stackSize);
74 ICUNIT_ASSERT_EQUAL(ret, 0, ret);
75
76 ret = pthread_attr_getstack(&attr, (void **)&getStack, &getSize);
77 ICUNIT_ASSERT_EQUAL(ret, 0, ret);
78 ICUNIT_ASSERT_EQUAL(getStack, stack, getStack);
79 ICUNIT_ASSERT_EQUAL(getSize, stackSize, getSize);
80
81 ret = pthread_create(&th1, &attr, PthreadTest115, 0);
82 ICUNIT_ASSERT_EQUAL(ret, 0, ret);
83
84 ret = pthread_join(th1, NULL);
85 ICUNIT_ASSERT_EQUAL(ret, 0, ret);
86
87 ICUNIT_ASSERT_EQUAL(g_testCnt, 1, g_testCnt);
88
89 free(stack);
90
91 stackSize = 3000;
92
93 pthread_attr_init(&attr);
94 stack = malloc(stackSize);
95 ICUNIT_ASSERT_NOT_EQUAL((int)stack, 0, stack);
96
97 ret = pthread_attr_setstack(&attr, stack, stackSize);
98 ICUNIT_ASSERT_EQUAL(ret, 0, ret);
99
100 ret = pthread_attr_getstack(&attr, (void **)&getStack, &getSize);
101 ICUNIT_ASSERT_EQUAL(ret, 0, ret);
102 ICUNIT_ASSERT_EQUAL(getStack, stack, getStack);
103 ICUNIT_ASSERT_EQUAL(getSize, stackSize, getSize);
104
105 ret = pthread_create(&th1, &attr, PthreadTest115, 0);
106 ICUNIT_ASSERT_EQUAL(ret, 0, ret);
107
108 ret = pthread_join(th1, NULL);
109 ICUNIT_ASSERT_EQUAL(ret, 0, ret);
110
111 ICUNIT_ASSERT_EQUAL(g_testCnt, 2, g_testCnt); // 2, here, assert the result.
112
113 free(stack);
114
115 exit(255); // 255, set a special exit code.
116 return 0;
117 }
118
TestCase(void)119 static int TestCase(void)
120 {
121 int ret;
122 int status = 0;
123 pid_t pid;
124 pid = fork();
125 ICUNIT_GOTO_WITHIN_EQUAL(pid, 0, 100000, pid, EXIT); // 100000, The pid will never exceed 100000.
126
127 if (pid == 0) {
128 prctl(PR_SET_NAME, "mainFork", 0UL, 0UL, 0UL);
129 GroupProcess();
130 printf("[Failed] - [Errline : %d RetCode : 0x%x\n", g_iCunitErrLineNo, g_iCunitErrCode);
131 exit(g_iCunitErrLineNo);
132 }
133
134 ret = waitpid(pid, &status, 0);
135 ICUNIT_ASSERT_EQUAL(ret, pid, ret);
136 status = WEXITSTATUS(status);
137 ICUNIT_GOTO_EQUAL(status, 255, status, EXIT); // 255, assert the special exit code.
138
139 return 0;
140 EXIT:
141 return 1;
142 }
143
ItTestPthread018(void)144 void ItTestPthread018(void)
145 {
146 TEST_ADD_CASE("IT_POSIX_PTHREAD_018", TestCase, TEST_POSIX, TEST_MEM, TEST_LEVEL0, TEST_FUNCTION);
147 }
148