/* SPDX-License-Identifier: GPL-2.0-or-later */ /* * Copyright (C) 2019, Google Inc. * * Threads test */ #include #include #include #include #include #include #include #include #include #include "test.h" using namespace std; using namespace libcamera; class DelayThread : public Thread { public: DelayThread(chrono::steady_clock::duration duration) : duration_(duration) { } protected: void run() { this_thread::sleep_for(duration_); } private: chrono::steady_clock::duration duration_; }; class CancelThread : public Thread { public: CancelThread(bool &cancelled) : cancelled_(cancelled) { } protected: void run() { cancelled_ = true; /* * Cancel the thread and call a guaranteed cancellation point * (nanosleep). */ pthread_cancel(pthread_self()); struct timespec req{ 0, 100*000*000 }; nanosleep(&req, nullptr); cancelled_ = false; } private: bool &cancelled_; }; class CpuSetTester : public Object { public: CpuSetTester(unsigned int cpuset) : cpuset_(cpuset) {} bool testCpuSet() { int ret = sched_getcpu(); if (static_cast(cpuset_) != ret) { cout << "Invalid cpuset: " << ret << ", expecting: " << cpuset_ << endl; return false; } return true; } private: const unsigned int cpuset_; }; class ThreadTest : public Test { protected: int init() { return 0; } int run() { /* Test Thread() retrieval for the main thread. */ Thread *mainThread = Thread::current(); if (!mainThread) { cout << "Thread::current() failed to main thread" << endl; return TestFail; } if (!mainThread->isRunning()) { cout << "Main thread is not running" << endl; return TestFail; } /* Test starting the main thread, the test shall not crash. */ mainThread->start(); /* Test the running state of a custom thread. */ std::unique_ptr thread = std::make_unique(); thread->start(); if (!thread->isRunning()) { cout << "Thread is not running after being started" << endl; return TestFail; } thread->exit(0); thread->wait(); if (thread->isRunning()) { cout << "Thread is still running after finishing" << endl; return TestFail; } /* Test waiting for completion with a timeout. */ thread = std::make_unique(chrono::milliseconds(500)); thread->start(); thread->exit(0); bool timeout = !thread->wait(chrono::milliseconds(100)); if (!timeout) { cout << "Waiting for thread didn't time out" << endl; return TestFail; } timeout = !thread->wait(chrono::milliseconds(1000)); if (timeout) { cout << "Waiting for thread timed out" << endl; return TestFail; } /* Test waiting on a thread that isn't running. */ thread = std::make_unique(); timeout = !thread->wait(); if (timeout) { cout << "Waiting for non-started thread timed out" << endl; return TestFail; } thread->start(); thread->exit(0); thread->wait(); timeout = !thread->wait(); if (timeout) { cout << "Waiting for already stopped thread timed out" << endl; return TestFail; } /* Test thread cleanup upon abnormal termination. */ bool cancelled = false; bool finished = false; thread = std::make_unique(cancelled); thread->finished.connect(this, [&finished]() { finished = true; }); thread->start(); thread->exit(0); thread->wait(chrono::milliseconds(1000)); if (!cancelled || !finished) { cout << "Cleanup failed upon abnormal termination" << endl; return TestFail; } const unsigned int numCpus = std::thread::hardware_concurrency(); for (unsigned int i = 0; i < numCpus; ++i) { thread = std::make_unique(); const std::array cpus{ i }; thread->setThreadAffinity(cpus); thread->start(); CpuSetTester tester(i); tester.moveToThread(thread.get()); if (!tester.invokeMethod(&CpuSetTester::testCpuSet, ConnectionTypeBlocking)) return TestFail; thread->exit(0); thread->wait(); } return TestPass; } void cleanup() { } }; TEST_REGISTER(ThreadTest)