diff --git a/README.md b/README.md new file mode 100644 index 0000000..ac8dc18 --- /dev/null +++ b/README.md @@ -0,0 +1,2 @@ +# openharmonydocs + diff --git a/pthread_atfork_test_002.cpp b/pthread_atfork_test_002.cpp new file mode 100644 index 0000000..2f7cd13 --- /dev/null +++ b/pthread_atfork_test_002.cpp @@ -0,0 +1,211 @@ +/* + * Copyright (c) 2013-2019 Huawei Technologies Co., Ltd. All rights reserved. + * Copyright (c) 2020-2021 Huawei Device Co., Ltd. All rights reserved. + * + * Redistribution and use in source and binary forms, with or without modification, + * are permitted provided that the following conditions are met: + * + * 1. Redistributions of source code must retain the above copyright notice, this list of + * conditions and the following disclaimer. + * + * 2. Redistributions in binary form must reproduce the above copyright notice, this list + * of conditions and the following disclaimer in the documentation and/or other materials + * provided with the distribution. + * + * 3. Neither the name of the copyright holder nor the names of its contributors may be used + * to endorse or promote products derived from this software without specific prior written + * permission. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS + * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, + * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR + * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR + * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, + * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, + * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; + * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, + * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR + * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF + * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ +#include "it_pthread_test.h" +#include + +// 引入系统函数nanosleep,该函数用于使当前线程休眠指定的时间。 +extern int nanosleep(const struct timespec *req, struct timespec *rem); + +// 定义全局互斥锁 g_mux,用于同步多线程操作。 +static pthread_mutex_t g_mux = PTHREAD_MUTEX_INITIALIZER; + +// 定义全局变量,跟踪fork相关的操作计数和状态。 +static volatile int g_testAtforkCount = 0; // 记录fork次数的计数器 +static int g_testAtforkPrepare = 0; // 用于标识准备状态的计数器 +static int g_testAtforkParent = 0; // 用于标识父进程状态的计数器 + +// 线程函数Doit,用于在锁保护下进行nanosleep操作。 +static void *Doit(void *arg) +{ + int err; + struct timespec ts = { 2, 0 }; // 设置休眠时间为2秒,第二个参数为纳秒(0表示不需要额外的纳秒部分)。 + + // 检查g_testAtforkCount是否为1,如果不是,跳转到EXIT标签。 + ICUNIT_GOTO_EQUAL(g_testAtforkCount, 1, g_testAtforkCount, EXIT); + + // 锁住g_mux互斥锁,保证对共享资源的访问是线程安全的。 + err = pthread_mutex_lock(&g_mux); + ICUNIT_GOTO_EQUAL(err, 0, err, EXIT); + + // 让当前线程休眠2秒。 + (void)nanosleep(&ts, NULL); + + // 解锁g_mux互斥锁。 + err = pthread_mutex_unlock(&g_mux); + ICUNIT_GOTO_EQUAL(err, 0, err, EXIT); + +EXIT: + // 函数结束,返回NULL表示没有错误。 + return NULL; +} + +// 线程函数Doit1,与Doit函数功能相同,都是进行互斥锁保护下的nanosleep操作。 +static void *Doit1(void *arg) +{ + int err; + struct timespec ts = { 2, 0 }; // 设置休眠时间为2秒。 + + // 检查g_testAtforkCount是否为1,如果不是,跳转到EXIT标签。 + ICUNIT_GOTO_EQUAL(g_testAtforkCount, 1, g_testAtforkCount, EXIT); + + // 锁住g_mux互斥锁,保证对共享资源的访问是线程安全的。 + err = pthread_mutex_lock(&g_mux); + ICUNIT_GOTO_EQUAL(err, 0, err, EXIT); + + // 让当前线程休眠2秒。 + (void)nanosleep(&ts, NULL); + + // 解锁g_mux互斥锁。 + err = pthread_mutex_unlock(&g_mux); + ICUNIT_GOTO_EQUAL(err, 0, err, EXIT); + +EXIT: + // 函数结束,返回NULL表示没有错误。 + return NULL; +} + +// Prepare函数用于准备阶段,释放锁并更新g_testAtforkPrepare计数器。 +static void Prepare(void) +{ + int err; + + // 尝试解锁g_mux互斥锁,若没有锁住,则返回错误。 + err = pthread_mutex_unlock(&g_mux); + ICUNIT_ASSERT_EQUAL_VOID(err, 0, err); // 校验解锁是否成功,若不成功则会引发错误。 + + // 更新g_testAtforkPrepare计数器,表示已准备好。 + g_testAtforkPrepare++; +} + +// Parent函数用于父进程阶段,获取锁并更新g_testAtforkParent计数器。 +static void Parent(void) +{ + int err = pthread_mutex_lock(&g_mux); // 锁住g_mux互斥锁。 + ICUNIT_ASSERT_EQUAL_VOID(err, 0, err); // 校验锁住是否成功,若不成功则会引发错误。 + + // 更新g_testAtforkParent计数器,表示父进程已进入阶段。 + g_testAtforkParent++; +} + +// PthreadAtforkTest函数的作用是进行多线程与fork操作的测试,验证pthread_atfork()的行为。 +static void *PthreadAtforkTest(void *arg) +{ + int err, ret; + int pid; + int status = 0; + struct timespec ts = { 1, 0 }; // 定义一个timespec结构,指定nanosleep时休眠1秒。 + pthread_t tid; // 定义一个线程ID变量,用于存储创建的线程的ID。 + + // 初始化全局变量,跟踪fork相关的操作计数和状态。 + g_testAtforkCount = 0; + g_testAtforkPrepare = 0; + g_testAtforkParent = 0; + + // 使用pthread_atfork注册处理函数,分别为子进程准备函数Prepare和父进程函数Parent。 + // NULL表示没有为子进程注册处理函数。 + err = pthread_atfork(Prepare, Parent, NULL); + ICUNIT_GOTO_EQUAL(err, 0, err, EXIT); // 如果pthread_atfork返回错误,跳转到EXIT标签。 + + // 将g_testAtforkCount计数器加1,表示准备开始执行fork操作。 + g_testAtforkCount++; + + // 创建一个新线程,线程的执行函数是Doit。 + err = pthread_create(&tid, NULL, Doit, NULL); + ICUNIT_GOTO_EQUAL(err, 0, err, EXIT); // 如果pthread_create失败,跳转到EXIT标签。 + + // 让当前线程休眠1秒,确保线程和主线程有足够的时间进入适当的执行阶段。 + nanosleep(&ts, NULL); + + // 执行fork系统调用,创建一个新进程。 + pid = fork(); + ICUNIT_GOTO_EQUAL(g_testAtforkPrepare, 1, g_testAtforkPrepare, EXIT); // 确保在fork后,Prepare函数已经被调用。 + + if (pid == 0) { // 子进程执行的代码。 + // 调用Doit1函数,子进程将在此执行,休眠2秒。 + Doit1(NULL); + exit(10); // 10表示子进程的退出状态。 + } + + // 检查pid值是否合法,确保fork操作成功并且pid不会超过100000。 + ICUNIT_GOTO_WITHIN_EQUAL(pid, 0, 100000, pid, EXIT_WAIT); + + // 确保父进程调用了Parent函数,更新了g_testAtforkParent计数器。 + ICUNIT_GOTO_EQUAL(g_testAtforkParent, 1, g_testAtforkParent, EXIT_WAIT); + + // 等待创建的线程结束。 + err = pthread_join(tid, NULL); + ICUNIT_GOTO_EQUAL(err, 0, err, EXIT_WAIT); // 如果pthread_join失败,跳转到EXIT_WAIT标签。 + + // 等待子进程结束,并获取其退出状态。 + err = waitpid(pid, &status, 0); + status = WEXITSTATUS(status); // 获取子进程的退出状态。 + ICUNIT_GOTO_EQUAL(err, pid, err, EXIT); // 检查waitpid返回值是否与pid一致。 + ICUNIT_GOTO_EQUAL(status, 10, status, EXIT); // 确保子进程的退出状态是10。 + +EXIT: + return NULL; // 函数结束,返回NULL。 + +EXIT_WAIT: + // 如果到达EXIT_WAIT标签,表示需要等待子进程的结束,防止僵尸进程。 + (void)waitpid(pid, 0, 0); + return NULL; +} + +static int Testcase(void) +{ + int ret; + pthread_t newPthread; + int curThreadPri, curThreadPolicy; + pthread_attr_t a = { 0 }; + struct sched_param param = { 0 }; + + ret = pthread_getschedparam(pthread_self(), &curThreadPolicy, ¶m); + ICUNIT_ASSERT_EQUAL(ret, 0, -ret); + + curThreadPri = param.sched_priority; + + ret = pthread_attr_init(&a); + pthread_attr_setinheritsched(&a, PTHREAD_EXPLICIT_SCHED); + param.sched_priority = curThreadPri + 2; // 2, adjust the priority. + pthread_attr_setschedparam(&a, ¶m); + ret = pthread_create(&newPthread, &a, PthreadAtforkTest, 0); + ICUNIT_ASSERT_EQUAL(ret, 0, ret); + + ret = pthread_join(newPthread, NULL); + ICUNIT_ASSERT_EQUAL(ret, 0, ret); + + return 0; +} + +void ItTestPthreadAtfork002(void) +{ + TEST_ADD_CASE("IT_PTHREAD_ATFORK_002", Testcase, TEST_POSIX, TEST_MEM, TEST_LEVEL0, TEST_FUNCTION); +}