blob: 0bf8e294b9589360a5af05d39e1efc8200d904f9 [file] [log] [blame]
Elliott Hughesbfeab1b2012-09-05 17:47:37 -07001/*
2 * Copyright (C) 2012 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17#include <gtest/gtest.h>
18
19#include <errno.h>
Elliott Hughes5b9310e2013-10-02 16:59:05 -070020#include <inttypes.h>
Elliott Hughesb95cf0d2013-07-15 14:51:07 -070021#include <limits.h>
Elliott Hughes04620a32014-03-07 17:59:05 -080022#include <malloc.h>
Elliott Hughesbfeab1b2012-09-05 17:47:37 -070023#include <pthread.h>
Christopher Ferrisf04935c2013-12-20 18:43:21 -080024#include <signal.h>
Yabin Cui140f3672015-02-03 10:32:00 -080025#include <stdio.h>
Elliott Hughes70b24b12013-11-15 11:51:07 -080026#include <sys/mman.h>
Elliott Hughes4d098ca2016-04-11 12:43:05 -070027#include <sys/prctl.h>
George Burgess IV08fd0722019-01-15 19:00:11 -080028#include <sys/resource.h>
Elliott Hughes57b7a612014-08-25 17:26:50 -070029#include <sys/syscall.h>
Narayan Kamath51e6cb32014-03-03 15:38:51 +000030#include <time.h>
Elliott Hughes4d014e12012-09-07 16:47:54 -070031#include <unistd.h>
Yabin Cui33ac04a2015-09-22 11:16:15 -070032#include <unwind.h>
Elliott Hughesbfeab1b2012-09-05 17:47:37 -070033
Yabin Cui08ee8d22015-02-11 17:04:36 -080034#include <atomic>
Josh Gaoddf757e2018-10-17 15:23:03 -070035#include <future>
Yabin Cuib5845722015-03-16 22:46:42 -070036#include <vector>
Yabin Cui08ee8d22015-02-11 17:04:36 -080037
Elliott Hughes5e62b342018-10-25 11:00:00 -070038#include <android-base/macros.h>
Yabin Cui6b9c85b2018-01-23 12:56:18 -080039#include <android-base/parseint.h>
Tom Cherryb8ab6182017-04-05 16:20:29 -070040#include <android-base/scopeguard.h>
Yabin Cui6b9c85b2018-01-23 12:56:18 -080041#include <android-base/strings.h>
Tom Cherryb8ab6182017-04-05 16:20:29 -070042
Yabin Cuic9a659c2015-11-05 15:36:08 -080043#include "private/bionic_constants.h"
Yabin Cui17393b02015-03-21 15:08:25 -070044#include "BionicDeathTest.h"
Elliott Hughes71ba5892018-02-07 12:44:45 -080045#include "SignalUtils.h"
Elliott Hughes15dfd632015-09-22 16:40:14 -070046#include "utils.h"
47
Elliott Hughesbfeab1b2012-09-05 17:47:37 -070048TEST(pthread, pthread_key_create) {
49 pthread_key_t key;
Yi Kong32bc0fc2018-08-02 17:31:13 -070050 ASSERT_EQ(0, pthread_key_create(&key, nullptr));
Elliott Hughesbfeab1b2012-09-05 17:47:37 -070051 ASSERT_EQ(0, pthread_key_delete(key));
52 // Can't delete a key that's already been deleted.
53 ASSERT_EQ(EINVAL, pthread_key_delete(key));
54}
Elliott Hughes4d014e12012-09-07 16:47:54 -070055
Dan Albertc4bcc752014-09-30 11:48:24 -070056TEST(pthread, pthread_keys_max) {
Yabin Cui6c238f22014-12-11 20:50:41 -080057 // POSIX says PTHREAD_KEYS_MAX should be at least _POSIX_THREAD_KEYS_MAX.
58 ASSERT_GE(PTHREAD_KEYS_MAX, _POSIX_THREAD_KEYS_MAX);
Dan Albertc4bcc752014-09-30 11:48:24 -070059}
Elliott Hughes718a5b52014-01-28 17:02:03 -080060
Yabin Cui6c238f22014-12-11 20:50:41 -080061TEST(pthread, sysconf_SC_THREAD_KEYS_MAX_eq_PTHREAD_KEYS_MAX) {
Dan Albertc4bcc752014-09-30 11:48:24 -070062 int sysconf_max = sysconf(_SC_THREAD_KEYS_MAX);
Yabin Cui6c238f22014-12-11 20:50:41 -080063 ASSERT_EQ(sysconf_max, PTHREAD_KEYS_MAX);
Dan Albertc4bcc752014-09-30 11:48:24 -070064}
65
66TEST(pthread, pthread_key_many_distinct) {
Yabin Cui6c238f22014-12-11 20:50:41 -080067 // As gtest uses pthread keys, we can't allocate exactly PTHREAD_KEYS_MAX
68 // pthread keys, but We should be able to allocate at least this many keys.
69 int nkeys = PTHREAD_KEYS_MAX / 2;
Dan Albertc4bcc752014-09-30 11:48:24 -070070 std::vector<pthread_key_t> keys;
71
Tom Cherryb8ab6182017-04-05 16:20:29 -070072 auto scope_guard = android::base::make_scope_guard([&keys] {
Elliott Hughes0b2acdf2015-10-02 18:25:19 -070073 for (const auto& key : keys) {
Dan Albertc4bcc752014-09-30 11:48:24 -070074 EXPECT_EQ(0, pthread_key_delete(key));
75 }
76 });
77
78 for (int i = 0; i < nkeys; ++i) {
79 pthread_key_t key;
Elliott Hughes61706932015-03-31 10:56:58 -070080 // If this fails, it's likely that LIBC_PTHREAD_KEY_RESERVED_COUNT is wrong.
Yi Kong32bc0fc2018-08-02 17:31:13 -070081 ASSERT_EQ(0, pthread_key_create(&key, nullptr)) << i << " of " << nkeys;
Dan Albertc4bcc752014-09-30 11:48:24 -070082 keys.push_back(key);
83 ASSERT_EQ(0, pthread_setspecific(key, reinterpret_cast<void*>(i)));
84 }
85
86 for (int i = keys.size() - 1; i >= 0; --i) {
87 ASSERT_EQ(reinterpret_cast<void*>(i), pthread_getspecific(keys.back()));
88 pthread_key_t key = keys.back();
89 keys.pop_back();
90 ASSERT_EQ(0, pthread_key_delete(key));
91 }
92}
93
Yabin Cui6c238f22014-12-11 20:50:41 -080094TEST(pthread, pthread_key_not_exceed_PTHREAD_KEYS_MAX) {
Elliott Hughes44b53ad2013-02-11 20:18:47 +000095 std::vector<pthread_key_t> keys;
Dan Albertc4bcc752014-09-30 11:48:24 -070096 int rv = 0;
Yabin Cui6c238f22014-12-11 20:50:41 -080097
98 // Pthread keys are used by gtest, so PTHREAD_KEYS_MAX should
99 // be more than we are allowed to allocate now.
100 for (int i = 0; i < PTHREAD_KEYS_MAX; i++) {
Elliott Hughes44b53ad2013-02-11 20:18:47 +0000101 pthread_key_t key;
Yi Kong32bc0fc2018-08-02 17:31:13 -0700102 rv = pthread_key_create(&key, nullptr);
Dan Albertc4bcc752014-09-30 11:48:24 -0700103 if (rv == EAGAIN) {
104 break;
105 }
106 EXPECT_EQ(0, rv);
Elliott Hughes44b53ad2013-02-11 20:18:47 +0000107 keys.push_back(key);
108 }
109
Dan Albertc4bcc752014-09-30 11:48:24 -0700110 // Don't leak keys.
Elliott Hughes0b2acdf2015-10-02 18:25:19 -0700111 for (const auto& key : keys) {
Dan Albertc4bcc752014-09-30 11:48:24 -0700112 EXPECT_EQ(0, pthread_key_delete(key));
Elliott Hughes44b53ad2013-02-11 20:18:47 +0000113 }
Dan Albertc4bcc752014-09-30 11:48:24 -0700114 keys.clear();
115
116 // We should have eventually reached the maximum number of keys and received
117 // EAGAIN.
118 ASSERT_EQ(EAGAIN, rv);
Elliott Hughes44b53ad2013-02-11 20:18:47 +0000119}
120
Elliott Hughesebb770f2014-06-25 13:46:46 -0700121TEST(pthread, pthread_key_delete) {
122 void* expected = reinterpret_cast<void*>(1234);
123 pthread_key_t key;
Yi Kong32bc0fc2018-08-02 17:31:13 -0700124 ASSERT_EQ(0, pthread_key_create(&key, nullptr));
Elliott Hughesebb770f2014-06-25 13:46:46 -0700125 ASSERT_EQ(0, pthread_setspecific(key, expected));
126 ASSERT_EQ(expected, pthread_getspecific(key));
127 ASSERT_EQ(0, pthread_key_delete(key));
Yi Kong32bc0fc2018-08-02 17:31:13 -0700128 // After deletion, pthread_getspecific returns nullptr.
129 ASSERT_EQ(nullptr, pthread_getspecific(key));
Elliott Hughesebb770f2014-06-25 13:46:46 -0700130 // And you can't use pthread_setspecific with the deleted key.
131 ASSERT_EQ(EINVAL, pthread_setspecific(key, expected));
132}
133
Elliott Hughes40a52172014-07-30 14:48:10 -0700134TEST(pthread, pthread_key_fork) {
135 void* expected = reinterpret_cast<void*>(1234);
136 pthread_key_t key;
Yi Kong32bc0fc2018-08-02 17:31:13 -0700137 ASSERT_EQ(0, pthread_key_create(&key, nullptr));
Elliott Hughes40a52172014-07-30 14:48:10 -0700138 ASSERT_EQ(0, pthread_setspecific(key, expected));
139 ASSERT_EQ(expected, pthread_getspecific(key));
140
141 pid_t pid = fork();
142 ASSERT_NE(-1, pid) << strerror(errno);
143
144 if (pid == 0) {
145 // The surviving thread inherits all the forking thread's TLS values...
146 ASSERT_EQ(expected, pthread_getspecific(key));
147 _exit(99);
148 }
149
Elliott Hughes33697a02016-01-26 13:04:57 -0800150 AssertChildExited(pid, 99);
Elliott Hughes40a52172014-07-30 14:48:10 -0700151
152 ASSERT_EQ(expected, pthread_getspecific(key));
Dan Albert1d53ae22014-09-02 15:24:26 -0700153 ASSERT_EQ(0, pthread_key_delete(key));
Elliott Hughes40a52172014-07-30 14:48:10 -0700154}
155
156static void* DirtyKeyFn(void* key) {
157 return pthread_getspecific(*reinterpret_cast<pthread_key_t*>(key));
158}
159
160TEST(pthread, pthread_key_dirty) {
161 pthread_key_t key;
Yi Kong32bc0fc2018-08-02 17:31:13 -0700162 ASSERT_EQ(0, pthread_key_create(&key, nullptr));
Elliott Hughes40a52172014-07-30 14:48:10 -0700163
Yabin Cuia36158a2015-11-16 21:06:16 -0800164 size_t stack_size = 640 * 1024;
Yi Kong32bc0fc2018-08-02 17:31:13 -0700165 void* stack = mmap(nullptr, stack_size, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANONYMOUS, -1, 0);
Elliott Hughes40a52172014-07-30 14:48:10 -0700166 ASSERT_NE(MAP_FAILED, stack);
167 memset(stack, 0xff, stack_size);
168
169 pthread_attr_t attr;
170 ASSERT_EQ(0, pthread_attr_init(&attr));
171 ASSERT_EQ(0, pthread_attr_setstack(&attr, stack, stack_size));
172
173 pthread_t t;
174 ASSERT_EQ(0, pthread_create(&t, &attr, DirtyKeyFn, &key));
175
176 void* result;
177 ASSERT_EQ(0, pthread_join(t, &result));
178 ASSERT_EQ(nullptr, result); // Not ~0!
179
180 ASSERT_EQ(0, munmap(stack, stack_size));
Dan Albert1d53ae22014-09-02 15:24:26 -0700181 ASSERT_EQ(0, pthread_key_delete(key));
Elliott Hughes40a52172014-07-30 14:48:10 -0700182}
183
Yabin Cui5ddbb3f2015-03-05 20:35:32 -0800184TEST(pthread, static_pthread_key_used_before_creation) {
185#if defined(__BIONIC__)
186 // See http://b/19625804. The bug is about a static/global pthread key being used before creation.
187 // So here tests if the static/global default value 0 can be detected as invalid key.
188 static pthread_key_t key;
189 ASSERT_EQ(nullptr, pthread_getspecific(key));
190 ASSERT_EQ(EINVAL, pthread_setspecific(key, nullptr));
191 ASSERT_EQ(EINVAL, pthread_key_delete(key));
192#else
Elliott Hughesbcaa4542019-03-08 15:20:23 -0800193 GTEST_SKIP() << "bionic-only test";
Yabin Cui5ddbb3f2015-03-05 20:35:32 -0800194#endif
195}
196
Elliott Hughes4d014e12012-09-07 16:47:54 -0700197static void* IdFn(void* arg) {
198 return arg;
199}
200
Yabin Cui63481602014-12-01 17:41:04 -0800201class SpinFunctionHelper {
202 public:
203 SpinFunctionHelper() {
204 SpinFunctionHelper::spin_flag_ = true;
Sergey Melnikov10ce9692012-10-26 14:06:43 +0400205 }
Elliott Hughes0bd9d132017-11-02 13:11:13 -0700206
Yabin Cui63481602014-12-01 17:41:04 -0800207 ~SpinFunctionHelper() {
208 UnSpin();
209 }
Elliott Hughes0bd9d132017-11-02 13:11:13 -0700210
Yabin Cui63481602014-12-01 17:41:04 -0800211 auto GetFunction() -> void* (*)(void*) {
212 return SpinFunctionHelper::SpinFn;
213 }
214
215 void UnSpin() {
216 SpinFunctionHelper::spin_flag_ = false;
217 }
218
219 private:
220 static void* SpinFn(void*) {
221 while (spin_flag_) {}
Yi Kong32bc0fc2018-08-02 17:31:13 -0700222 return nullptr;
Yabin Cui63481602014-12-01 17:41:04 -0800223 }
Yabin Cuia36158a2015-11-16 21:06:16 -0800224 static std::atomic<bool> spin_flag_;
Yabin Cui63481602014-12-01 17:41:04 -0800225};
226
227// It doesn't matter if spin_flag_ is used in several tests,
228// because it is always set to false after each test. Each thread
229// loops on spin_flag_ can find it becomes false at some time.
Yabin Cuia36158a2015-11-16 21:06:16 -0800230std::atomic<bool> SpinFunctionHelper::spin_flag_;
Sergey Melnikov10ce9692012-10-26 14:06:43 +0400231
Elliott Hughes4d014e12012-09-07 16:47:54 -0700232static void* JoinFn(void* arg) {
Yi Kong32bc0fc2018-08-02 17:31:13 -0700233 return reinterpret_cast<void*>(pthread_join(reinterpret_cast<pthread_t>(arg), nullptr));
Elliott Hughes4d014e12012-09-07 16:47:54 -0700234}
235
Sergey Melnikov10ce9692012-10-26 14:06:43 +0400236static void AssertDetached(pthread_t t, bool is_detached) {
237 pthread_attr_t attr;
238 ASSERT_EQ(0, pthread_getattr_np(t, &attr));
239 int detach_state;
240 ASSERT_EQ(0, pthread_attr_getdetachstate(&attr, &detach_state));
241 pthread_attr_destroy(&attr);
242 ASSERT_EQ(is_detached, (detach_state == PTHREAD_CREATE_DETACHED));
243}
244
Elliott Hughes7484c212017-02-02 02:41:38 +0000245static void MakeDeadThread(pthread_t& t) {
Yi Kong32bc0fc2018-08-02 17:31:13 -0700246 ASSERT_EQ(0, pthread_create(&t, nullptr, IdFn, nullptr));
247 ASSERT_EQ(0, pthread_join(t, nullptr));
Elliott Hughes7484c212017-02-02 02:41:38 +0000248}
249
Elliott Hughes4d014e12012-09-07 16:47:54 -0700250TEST(pthread, pthread_create) {
251 void* expected_result = reinterpret_cast<void*>(123);
252 // Can we create a thread?
253 pthread_t t;
Yi Kong32bc0fc2018-08-02 17:31:13 -0700254 ASSERT_EQ(0, pthread_create(&t, nullptr, IdFn, expected_result));
Elliott Hughes4d014e12012-09-07 16:47:54 -0700255 // If we join, do we get the expected value back?
256 void* result;
257 ASSERT_EQ(0, pthread_join(t, &result));
258 ASSERT_EQ(expected_result, result);
259}
260
Elliott Hughes3e898472013-02-12 16:40:24 +0000261TEST(pthread, pthread_create_EAGAIN) {
262 pthread_attr_t attributes;
263 ASSERT_EQ(0, pthread_attr_init(&attributes));
264 ASSERT_EQ(0, pthread_attr_setstacksize(&attributes, static_cast<size_t>(-1) & ~(getpagesize() - 1)));
265
266 pthread_t t;
Yi Kong32bc0fc2018-08-02 17:31:13 -0700267 ASSERT_EQ(EAGAIN, pthread_create(&t, &attributes, IdFn, nullptr));
Elliott Hughes3e898472013-02-12 16:40:24 +0000268}
269
Elliott Hughes4d014e12012-09-07 16:47:54 -0700270TEST(pthread, pthread_no_join_after_detach) {
Elliott Hughes725b2a92016-03-23 11:20:47 -0700271 SpinFunctionHelper spin_helper;
Yabin Cui63481602014-12-01 17:41:04 -0800272
Elliott Hughes4d014e12012-09-07 16:47:54 -0700273 pthread_t t1;
Yi Kong32bc0fc2018-08-02 17:31:13 -0700274 ASSERT_EQ(0, pthread_create(&t1, nullptr, spin_helper.GetFunction(), nullptr));
Elliott Hughes4d014e12012-09-07 16:47:54 -0700275
276 // After a pthread_detach...
277 ASSERT_EQ(0, pthread_detach(t1));
Sergey Melnikov10ce9692012-10-26 14:06:43 +0400278 AssertDetached(t1, true);
Elliott Hughes4d014e12012-09-07 16:47:54 -0700279
280 // ...pthread_join should fail.
Yi Kong32bc0fc2018-08-02 17:31:13 -0700281 ASSERT_EQ(EINVAL, pthread_join(t1, nullptr));
Elliott Hughes4d014e12012-09-07 16:47:54 -0700282}
283
284TEST(pthread, pthread_no_op_detach_after_join) {
Elliott Hughes725b2a92016-03-23 11:20:47 -0700285 SpinFunctionHelper spin_helper;
Sergey Melnikov10ce9692012-10-26 14:06:43 +0400286
Elliott Hughes4d014e12012-09-07 16:47:54 -0700287 pthread_t t1;
Yi Kong32bc0fc2018-08-02 17:31:13 -0700288 ASSERT_EQ(0, pthread_create(&t1, nullptr, spin_helper.GetFunction(), nullptr));
Elliott Hughes4d014e12012-09-07 16:47:54 -0700289
290 // If thread 2 is already waiting to join thread 1...
291 pthread_t t2;
Yi Kong32bc0fc2018-08-02 17:31:13 -0700292 ASSERT_EQ(0, pthread_create(&t2, nullptr, JoinFn, reinterpret_cast<void*>(t1)));
Elliott Hughes4d014e12012-09-07 16:47:54 -0700293
Sergey Melnikov10ce9692012-10-26 14:06:43 +0400294 sleep(1); // (Give t2 a chance to call pthread_join.)
Elliott Hughes4d014e12012-09-07 16:47:54 -0700295
Yabin Cuibbb04322015-03-19 15:19:25 -0700296#if defined(__BIONIC__)
297 ASSERT_EQ(EINVAL, pthread_detach(t1));
298#else
Sergey Melnikov10ce9692012-10-26 14:06:43 +0400299 ASSERT_EQ(0, pthread_detach(t1));
Yabin Cuibbb04322015-03-19 15:19:25 -0700300#endif
Sergey Melnikov10ce9692012-10-26 14:06:43 +0400301 AssertDetached(t1, false);
302
Elliott Hughes725b2a92016-03-23 11:20:47 -0700303 spin_helper.UnSpin();
Sergey Melnikov10ce9692012-10-26 14:06:43 +0400304
305 // ...but t2's join on t1 still goes ahead (which we can tell because our join on t2 finishes).
Elliott Hughes4d014e12012-09-07 16:47:54 -0700306 void* join_result;
307 ASSERT_EQ(0, pthread_join(t2, &join_result));
Elliott Hughes5b9310e2013-10-02 16:59:05 -0700308 ASSERT_EQ(0U, reinterpret_cast<uintptr_t>(join_result));
Elliott Hughes4d014e12012-09-07 16:47:54 -0700309}
Elliott Hughes14f19592012-10-29 10:19:44 -0700310
311TEST(pthread, pthread_join_self) {
Yi Kong32bc0fc2018-08-02 17:31:13 -0700312 ASSERT_EQ(EDEADLK, pthread_join(pthread_self(), nullptr));
Elliott Hughes14f19592012-10-29 10:19:44 -0700313}
Elliott Hughes4f251be2012-11-01 16:33:29 -0700314
Elliott Hughes877ec6d2013-11-15 17:40:18 -0800315struct TestBug37410 {
316 pthread_t main_thread;
317 pthread_mutex_t mutex;
Elliott Hughes4f251be2012-11-01 16:33:29 -0700318
Elliott Hughes877ec6d2013-11-15 17:40:18 -0800319 static void main() {
320 TestBug37410 data;
321 data.main_thread = pthread_self();
Yi Kong32bc0fc2018-08-02 17:31:13 -0700322 ASSERT_EQ(0, pthread_mutex_init(&data.mutex, nullptr));
Elliott Hughes877ec6d2013-11-15 17:40:18 -0800323 ASSERT_EQ(0, pthread_mutex_lock(&data.mutex));
324
325 pthread_t t;
Yi Kong32bc0fc2018-08-02 17:31:13 -0700326 ASSERT_EQ(0, pthread_create(&t, nullptr, TestBug37410::thread_fn, reinterpret_cast<void*>(&data)));
Elliott Hughes877ec6d2013-11-15 17:40:18 -0800327
328 // Wait for the thread to be running...
329 ASSERT_EQ(0, pthread_mutex_lock(&data.mutex));
330 ASSERT_EQ(0, pthread_mutex_unlock(&data.mutex));
331
332 // ...and exit.
Yi Kong32bc0fc2018-08-02 17:31:13 -0700333 pthread_exit(nullptr);
Elliott Hughes877ec6d2013-11-15 17:40:18 -0800334 }
335
336 private:
337 static void* thread_fn(void* arg) {
338 TestBug37410* data = reinterpret_cast<TestBug37410*>(arg);
339
Evgenii Stepanov352853a2019-02-05 17:37:37 -0800340 // Unlocking data->mutex will cause the main thread to exit, invalidating *data. Save the handle.
341 pthread_t main_thread = data->main_thread;
342
Elliott Hughes877ec6d2013-11-15 17:40:18 -0800343 // Let the main thread know we're running.
344 pthread_mutex_unlock(&data->mutex);
345
346 // And wait for the main thread to exit.
Evgenii Stepanov352853a2019-02-05 17:37:37 -0800347 pthread_join(main_thread, nullptr);
Elliott Hughes877ec6d2013-11-15 17:40:18 -0800348
Yi Kong32bc0fc2018-08-02 17:31:13 -0700349 return nullptr;
Elliott Hughes877ec6d2013-11-15 17:40:18 -0800350 }
351};
Elliott Hughes4f251be2012-11-01 16:33:29 -0700352
Elliott Hughes7fd803c2013-02-14 16:33:52 -0800353// Even though this isn't really a death test, we have to say "DeathTest" here so gtest knows to
354// run this test (which exits normally) in its own process.
Yabin Cui9df70402014-11-05 18:01:01 -0800355
356class pthread_DeathTest : public BionicDeathTest {};
357
358TEST_F(pthread_DeathTest, pthread_bug_37410) {
Elliott Hughes4f251be2012-11-01 16:33:29 -0700359 // http://code.google.com/p/android/issues/detail?id=37410
Elliott Hughes877ec6d2013-11-15 17:40:18 -0800360 ASSERT_EXIT(TestBug37410::main(), ::testing::ExitedWithCode(0), "");
Elliott Hughes4f251be2012-11-01 16:33:29 -0700361}
Elliott Hughesc5d028f2013-01-10 14:42:14 -0800362
363static void* SignalHandlerFn(void* arg) {
Elliott Hughes5905d6f2018-01-30 15:09:51 -0800364 sigset64_t wait_set;
365 sigfillset64(&wait_set);
366 return reinterpret_cast<void*>(sigwait64(&wait_set, reinterpret_cast<int*>(arg)));
Elliott Hughesc5d028f2013-01-10 14:42:14 -0800367}
368
369TEST(pthread, pthread_sigmask) {
Elliott Hughes19e62322013-10-15 11:23:57 -0700370 // Check that SIGUSR1 isn't blocked.
371 sigset_t original_set;
372 sigemptyset(&original_set);
Yi Kong32bc0fc2018-08-02 17:31:13 -0700373 ASSERT_EQ(0, pthread_sigmask(SIG_BLOCK, nullptr, &original_set));
Elliott Hughes19e62322013-10-15 11:23:57 -0700374 ASSERT_FALSE(sigismember(&original_set, SIGUSR1));
375
Elliott Hughesc5d028f2013-01-10 14:42:14 -0800376 // Block SIGUSR1.
377 sigset_t set;
378 sigemptyset(&set);
379 sigaddset(&set, SIGUSR1);
Yi Kong32bc0fc2018-08-02 17:31:13 -0700380 ASSERT_EQ(0, pthread_sigmask(SIG_BLOCK, &set, nullptr));
Elliott Hughesc5d028f2013-01-10 14:42:14 -0800381
Elliott Hughes19e62322013-10-15 11:23:57 -0700382 // Check that SIGUSR1 is blocked.
383 sigset_t final_set;
384 sigemptyset(&final_set);
Yi Kong32bc0fc2018-08-02 17:31:13 -0700385 ASSERT_EQ(0, pthread_sigmask(SIG_BLOCK, nullptr, &final_set));
Elliott Hughes19e62322013-10-15 11:23:57 -0700386 ASSERT_TRUE(sigismember(&final_set, SIGUSR1));
387 // ...and that sigprocmask agrees with pthread_sigmask.
388 sigemptyset(&final_set);
Yi Kong32bc0fc2018-08-02 17:31:13 -0700389 ASSERT_EQ(0, sigprocmask(SIG_BLOCK, nullptr, &final_set));
Elliott Hughes19e62322013-10-15 11:23:57 -0700390 ASSERT_TRUE(sigismember(&final_set, SIGUSR1));
391
Elliott Hughesc5d028f2013-01-10 14:42:14 -0800392 // Spawn a thread that calls sigwait and tells us what it received.
393 pthread_t signal_thread;
394 int received_signal = -1;
Yi Kong32bc0fc2018-08-02 17:31:13 -0700395 ASSERT_EQ(0, pthread_create(&signal_thread, nullptr, SignalHandlerFn, &received_signal));
Elliott Hughesc5d028f2013-01-10 14:42:14 -0800396
397 // Send that thread SIGUSR1.
398 pthread_kill(signal_thread, SIGUSR1);
399
400 // See what it got.
401 void* join_result;
402 ASSERT_EQ(0, pthread_join(signal_thread, &join_result));
403 ASSERT_EQ(SIGUSR1, received_signal);
Elliott Hughes5b9310e2013-10-02 16:59:05 -0700404 ASSERT_EQ(0U, reinterpret_cast<uintptr_t>(join_result));
Elliott Hughes19e62322013-10-15 11:23:57 -0700405
406 // Restore the original signal mask.
Yi Kong32bc0fc2018-08-02 17:31:13 -0700407 ASSERT_EQ(0, pthread_sigmask(SIG_SETMASK, &original_set, nullptr));
Elliott Hughesc5d028f2013-01-10 14:42:14 -0800408}
Elliott Hughes5e3fc432013-02-11 16:36:48 -0800409
Elliott Hughes5905d6f2018-01-30 15:09:51 -0800410TEST(pthread, pthread_sigmask64_SIGTRMIN) {
411 // Check that SIGRTMIN isn't blocked.
412 sigset64_t original_set;
413 sigemptyset64(&original_set);
Yi Kong32bc0fc2018-08-02 17:31:13 -0700414 ASSERT_EQ(0, pthread_sigmask64(SIG_BLOCK, nullptr, &original_set));
Elliott Hughes5905d6f2018-01-30 15:09:51 -0800415 ASSERT_FALSE(sigismember64(&original_set, SIGRTMIN));
416
417 // Block SIGRTMIN.
418 sigset64_t set;
419 sigemptyset64(&set);
420 sigaddset64(&set, SIGRTMIN);
Yi Kong32bc0fc2018-08-02 17:31:13 -0700421 ASSERT_EQ(0, pthread_sigmask64(SIG_BLOCK, &set, nullptr));
Elliott Hughes5905d6f2018-01-30 15:09:51 -0800422
423 // Check that SIGRTMIN is blocked.
424 sigset64_t final_set;
425 sigemptyset64(&final_set);
Yi Kong32bc0fc2018-08-02 17:31:13 -0700426 ASSERT_EQ(0, pthread_sigmask64(SIG_BLOCK, nullptr, &final_set));
Elliott Hughes5905d6f2018-01-30 15:09:51 -0800427 ASSERT_TRUE(sigismember64(&final_set, SIGRTMIN));
428 // ...and that sigprocmask64 agrees with pthread_sigmask64.
429 sigemptyset64(&final_set);
Yi Kong32bc0fc2018-08-02 17:31:13 -0700430 ASSERT_EQ(0, sigprocmask64(SIG_BLOCK, nullptr, &final_set));
Elliott Hughes5905d6f2018-01-30 15:09:51 -0800431 ASSERT_TRUE(sigismember64(&final_set, SIGRTMIN));
432
433 // Spawn a thread that calls sigwait64 and tells us what it received.
434 pthread_t signal_thread;
435 int received_signal = -1;
Yi Kong32bc0fc2018-08-02 17:31:13 -0700436 ASSERT_EQ(0, pthread_create(&signal_thread, nullptr, SignalHandlerFn, &received_signal));
Elliott Hughes5905d6f2018-01-30 15:09:51 -0800437
438 // Send that thread SIGRTMIN.
439 pthread_kill(signal_thread, SIGRTMIN);
440
441 // See what it got.
442 void* join_result;
443 ASSERT_EQ(0, pthread_join(signal_thread, &join_result));
444 ASSERT_EQ(SIGRTMIN, received_signal);
445 ASSERT_EQ(0U, reinterpret_cast<uintptr_t>(join_result));
446
447 // Restore the original signal mask.
Yi Kong32bc0fc2018-08-02 17:31:13 -0700448 ASSERT_EQ(0, pthread_sigmask64(SIG_SETMASK, &original_set, nullptr));
Elliott Hughes5905d6f2018-01-30 15:09:51 -0800449}
450
Elliott Hughes725b2a92016-03-23 11:20:47 -0700451static void test_pthread_setname_np__pthread_getname_np(pthread_t t) {
452 ASSERT_EQ(0, pthread_setname_np(t, "short"));
453 char name[32];
454 ASSERT_EQ(0, pthread_getname_np(t, name, sizeof(name)));
455 ASSERT_STREQ("short", name);
456
Elliott Hughesd1aea302015-04-25 10:05:24 -0700457 // The limit is 15 characters --- the kernel's buffer is 16, but includes a NUL.
Elliott Hughes725b2a92016-03-23 11:20:47 -0700458 ASSERT_EQ(0, pthread_setname_np(t, "123456789012345"));
459 ASSERT_EQ(0, pthread_getname_np(t, name, sizeof(name)));
460 ASSERT_STREQ("123456789012345", name);
461
462 ASSERT_EQ(ERANGE, pthread_setname_np(t, "1234567890123456"));
463
464 // The passed-in buffer should be at least 16 bytes.
465 ASSERT_EQ(0, pthread_getname_np(t, name, 16));
466 ASSERT_EQ(ERANGE, pthread_getname_np(t, name, 15));
Elliott Hughes3e898472013-02-12 16:40:24 +0000467}
468
Elliott Hughes725b2a92016-03-23 11:20:47 -0700469TEST(pthread, pthread_setname_np__pthread_getname_np__self) {
470 test_pthread_setname_np__pthread_getname_np(pthread_self());
Elliott Hughes3e898472013-02-12 16:40:24 +0000471}
472
Elliott Hughes725b2a92016-03-23 11:20:47 -0700473TEST(pthread, pthread_setname_np__pthread_getname_np__other) {
474 SpinFunctionHelper spin_helper;
Yabin Cui63481602014-12-01 17:41:04 -0800475
Elliott Hughes725b2a92016-03-23 11:20:47 -0700476 pthread_t t;
Elliott Hughes4d098ca2016-04-11 12:43:05 -0700477 ASSERT_EQ(0, pthread_create(&t, nullptr, spin_helper.GetFunction(), nullptr));
478 test_pthread_setname_np__pthread_getname_np(t);
479 spin_helper.UnSpin();
480 ASSERT_EQ(0, pthread_join(t, nullptr));
481}
482
483// http://b/28051133: a kernel misfeature means that you can't change the
484// name of another thread if you've set PR_SET_DUMPABLE to 0.
485TEST(pthread, pthread_setname_np__pthread_getname_np__other_PR_SET_DUMPABLE) {
486 ASSERT_EQ(0, prctl(PR_SET_DUMPABLE, 0)) << strerror(errno);
487
488 SpinFunctionHelper spin_helper;
489
490 pthread_t t;
491 ASSERT_EQ(0, pthread_create(&t, nullptr, spin_helper.GetFunction(), nullptr));
Elliott Hughes725b2a92016-03-23 11:20:47 -0700492 test_pthread_setname_np__pthread_getname_np(t);
493 spin_helper.UnSpin();
494 ASSERT_EQ(0, pthread_join(t, nullptr));
Elliott Hughes3e898472013-02-12 16:40:24 +0000495}
496
Elliott Hughes11859d42017-02-13 17:59:29 -0800497TEST_F(pthread_DeathTest, pthread_setname_np__no_such_thread) {
Elliott Hughesbcb15292017-02-07 21:05:30 +0000498 pthread_t dead_thread;
499 MakeDeadThread(dead_thread);
500
Elliott Hughes5bb113c2019-02-01 16:31:10 -0800501 EXPECT_DEATH(pthread_setname_np(dead_thread, "short 3"),
502 "invalid pthread_t (.*) passed to pthread_setname_np");
Elliott Hughes6ce686c2017-02-21 13:15:20 -0800503}
504
505TEST_F(pthread_DeathTest, pthread_setname_np__null_thread) {
506 pthread_t null_thread = 0;
507 EXPECT_EQ(ENOENT, pthread_setname_np(null_thread, "short 3"));
Elliott Hughes11859d42017-02-13 17:59:29 -0800508}
509
510TEST_F(pthread_DeathTest, pthread_getname_np__no_such_thread) {
511 pthread_t dead_thread;
512 MakeDeadThread(dead_thread);
513
Elliott Hughesbcb15292017-02-07 21:05:30 +0000514 char name[64];
Elliott Hughes5bb113c2019-02-01 16:31:10 -0800515 EXPECT_DEATH(pthread_getname_np(dead_thread, name, sizeof(name)),
516 "invalid pthread_t (.*) passed to pthread_getname_np");
Elliott Hughes6ce686c2017-02-21 13:15:20 -0800517}
518
519TEST_F(pthread_DeathTest, pthread_getname_np__null_thread) {
520 pthread_t null_thread = 0;
521
522 char name[64];
523 EXPECT_EQ(ENOENT, pthread_getname_np(null_thread, name, sizeof(name)));
Elliott Hughesbcb15292017-02-07 21:05:30 +0000524}
525
Elliott Hughes9d23e042013-02-15 19:21:51 -0800526TEST(pthread, pthread_kill__0) {
527 // Signal 0 just tests that the thread exists, so it's safe to call on ourselves.
528 ASSERT_EQ(0, pthread_kill(pthread_self(), 0));
529}
530
531TEST(pthread, pthread_kill__invalid_signal) {
532 ASSERT_EQ(EINVAL, pthread_kill(pthread_self(), -1));
533}
534
Elliott Hughesfae89fc2013-02-21 11:22:23 -0800535static void pthread_kill__in_signal_handler_helper(int signal_number) {
536 static int count = 0;
537 ASSERT_EQ(SIGALRM, signal_number);
538 if (++count == 1) {
539 // Can we call pthread_kill from a signal handler?
540 ASSERT_EQ(0, pthread_kill(pthread_self(), SIGALRM));
541 }
542}
543
544TEST(pthread, pthread_kill__in_signal_handler) {
Elliott Hughes4b558f52014-03-04 15:58:02 -0800545 ScopedSignalHandler ssh(SIGALRM, pthread_kill__in_signal_handler_helper);
Elliott Hughesfae89fc2013-02-21 11:22:23 -0800546 ASSERT_EQ(0, pthread_kill(pthread_self(), SIGALRM));
547}
548
Josh Gaoddf757e2018-10-17 15:23:03 -0700549TEST(pthread, pthread_kill__exited_thread) {
550 static std::promise<pid_t> tid_promise;
551 pthread_t thread;
552 ASSERT_EQ(0, pthread_create(&thread, nullptr,
553 [](void*) -> void* {
554 tid_promise.set_value(gettid());
555 return nullptr;
556 },
557 nullptr));
558
559 pid_t tid = tid_promise.get_future().get();
560 while (TEMP_FAILURE_RETRY(syscall(__NR_tgkill, getpid(), tid, 0)) != -1) {
561 continue;
562 }
563 ASSERT_EQ(ESRCH, errno);
564
565 ASSERT_EQ(ESRCH, pthread_kill(thread, 0));
566}
567
Elliott Hughes11859d42017-02-13 17:59:29 -0800568TEST_F(pthread_DeathTest, pthread_detach__no_such_thread) {
Elliott Hughes7484c212017-02-02 02:41:38 +0000569 pthread_t dead_thread;
570 MakeDeadThread(dead_thread);
571
Elliott Hughes5bb113c2019-02-01 16:31:10 -0800572 EXPECT_DEATH(pthread_detach(dead_thread),
573 "invalid pthread_t (.*) passed to pthread_detach");
Elliott Hughes6ce686c2017-02-21 13:15:20 -0800574}
575
576TEST_F(pthread_DeathTest, pthread_detach__null_thread) {
577 pthread_t null_thread = 0;
578 EXPECT_EQ(ESRCH, pthread_detach(null_thread));
Elliott Hughes7484c212017-02-02 02:41:38 +0000579}
580
Jeff Hao9b06cc32013-08-15 14:51:16 -0700581TEST(pthread, pthread_getcpuclockid__clock_gettime) {
Elliott Hughes725b2a92016-03-23 11:20:47 -0700582 SpinFunctionHelper spin_helper;
Yabin Cui63481602014-12-01 17:41:04 -0800583
Jeff Hao9b06cc32013-08-15 14:51:16 -0700584 pthread_t t;
Yi Kong32bc0fc2018-08-02 17:31:13 -0700585 ASSERT_EQ(0, pthread_create(&t, nullptr, spin_helper.GetFunction(), nullptr));
Jeff Hao9b06cc32013-08-15 14:51:16 -0700586
587 clockid_t c;
588 ASSERT_EQ(0, pthread_getcpuclockid(t, &c));
589 timespec ts;
590 ASSERT_EQ(0, clock_gettime(c, &ts));
Elliott Hughes725b2a92016-03-23 11:20:47 -0700591 spin_helper.UnSpin();
Yabin Cuia36158a2015-11-16 21:06:16 -0800592 ASSERT_EQ(0, pthread_join(t, nullptr));
Jeff Hao9b06cc32013-08-15 14:51:16 -0700593}
594
Elliott Hughes11859d42017-02-13 17:59:29 -0800595TEST_F(pthread_DeathTest, pthread_getcpuclockid__no_such_thread) {
Elliott Hughesbcb15292017-02-07 21:05:30 +0000596 pthread_t dead_thread;
597 MakeDeadThread(dead_thread);
598
599 clockid_t c;
Elliott Hughes5bb113c2019-02-01 16:31:10 -0800600 EXPECT_DEATH(pthread_getcpuclockid(dead_thread, &c),
601 "invalid pthread_t (.*) passed to pthread_getcpuclockid");
Elliott Hughes6ce686c2017-02-21 13:15:20 -0800602}
603
604TEST_F(pthread_DeathTest, pthread_getcpuclockid__null_thread) {
605 pthread_t null_thread = 0;
606 clockid_t c;
607 EXPECT_EQ(ESRCH, pthread_getcpuclockid(null_thread, &c));
Elliott Hughesbcb15292017-02-07 21:05:30 +0000608}
609
Elliott Hughes11859d42017-02-13 17:59:29 -0800610TEST_F(pthread_DeathTest, pthread_getschedparam__no_such_thread) {
Elliott Hughesbcb15292017-02-07 21:05:30 +0000611 pthread_t dead_thread;
612 MakeDeadThread(dead_thread);
613
614 int policy;
615 sched_param param;
Elliott Hughes5bb113c2019-02-01 16:31:10 -0800616 EXPECT_DEATH(pthread_getschedparam(dead_thread, &policy, &param),
617 "invalid pthread_t (.*) passed to pthread_getschedparam");
Elliott Hughes6ce686c2017-02-21 13:15:20 -0800618}
619
620TEST_F(pthread_DeathTest, pthread_getschedparam__null_thread) {
621 pthread_t null_thread = 0;
622 int policy;
623 sched_param param;
624 EXPECT_EQ(ESRCH, pthread_getschedparam(null_thread, &policy, &param));
Elliott Hughesbcb15292017-02-07 21:05:30 +0000625}
626
Elliott Hughes11859d42017-02-13 17:59:29 -0800627TEST_F(pthread_DeathTest, pthread_setschedparam__no_such_thread) {
Elliott Hughesbcb15292017-02-07 21:05:30 +0000628 pthread_t dead_thread;
629 MakeDeadThread(dead_thread);
630
631 int policy = 0;
632 sched_param param;
Elliott Hughes5bb113c2019-02-01 16:31:10 -0800633 EXPECT_DEATH(pthread_setschedparam(dead_thread, policy, &param),
634 "invalid pthread_t (.*) passed to pthread_setschedparam");
Elliott Hughes6ce686c2017-02-21 13:15:20 -0800635}
636
637TEST_F(pthread_DeathTest, pthread_setschedparam__null_thread) {
638 pthread_t null_thread = 0;
639 int policy = 0;
640 sched_param param;
641 EXPECT_EQ(ESRCH, pthread_setschedparam(null_thread, policy, &param));
Elliott Hughesbcb15292017-02-07 21:05:30 +0000642}
643
Elliott Hughesdff08ce2017-10-16 09:58:45 -0700644TEST_F(pthread_DeathTest, pthread_setschedprio__no_such_thread) {
645 pthread_t dead_thread;
646 MakeDeadThread(dead_thread);
647
Elliott Hughes5bb113c2019-02-01 16:31:10 -0800648 EXPECT_DEATH(pthread_setschedprio(dead_thread, 123),
649 "invalid pthread_t (.*) passed to pthread_setschedprio");
Elliott Hughesdff08ce2017-10-16 09:58:45 -0700650}
651
652TEST_F(pthread_DeathTest, pthread_setschedprio__null_thread) {
653 pthread_t null_thread = 0;
654 EXPECT_EQ(ESRCH, pthread_setschedprio(null_thread, 123));
655}
656
Elliott Hughes11859d42017-02-13 17:59:29 -0800657TEST_F(pthread_DeathTest, pthread_join__no_such_thread) {
Elliott Hughes7484c212017-02-02 02:41:38 +0000658 pthread_t dead_thread;
659 MakeDeadThread(dead_thread);
660
Elliott Hughes5bb113c2019-02-01 16:31:10 -0800661 EXPECT_DEATH(pthread_join(dead_thread, nullptr),
662 "invalid pthread_t (.*) passed to pthread_join");
Elliott Hughes6ce686c2017-02-21 13:15:20 -0800663}
664
665TEST_F(pthread_DeathTest, pthread_join__null_thread) {
666 pthread_t null_thread = 0;
Yi Kong32bc0fc2018-08-02 17:31:13 -0700667 EXPECT_EQ(ESRCH, pthread_join(null_thread, nullptr));
Elliott Hughes7484c212017-02-02 02:41:38 +0000668}
669
Elliott Hughes11859d42017-02-13 17:59:29 -0800670TEST_F(pthread_DeathTest, pthread_kill__no_such_thread) {
Elliott Hughes7484c212017-02-02 02:41:38 +0000671 pthread_t dead_thread;
672 MakeDeadThread(dead_thread);
673
Elliott Hughes5bb113c2019-02-01 16:31:10 -0800674 EXPECT_DEATH(pthread_kill(dead_thread, 0),
675 "invalid pthread_t (.*) passed to pthread_kill");
Elliott Hughes6ce686c2017-02-21 13:15:20 -0800676}
677
678TEST_F(pthread_DeathTest, pthread_kill__null_thread) {
679 pthread_t null_thread = 0;
680 EXPECT_EQ(ESRCH, pthread_kill(null_thread, 0));
Elliott Hughes7484c212017-02-02 02:41:38 +0000681}
682
msg5550f020d12013-06-06 14:59:28 -0400683TEST(pthread, pthread_join__multijoin) {
Elliott Hughes725b2a92016-03-23 11:20:47 -0700684 SpinFunctionHelper spin_helper;
msg5550f020d12013-06-06 14:59:28 -0400685
686 pthread_t t1;
Yi Kong32bc0fc2018-08-02 17:31:13 -0700687 ASSERT_EQ(0, pthread_create(&t1, nullptr, spin_helper.GetFunction(), nullptr));
msg5550f020d12013-06-06 14:59:28 -0400688
689 pthread_t t2;
Yi Kong32bc0fc2018-08-02 17:31:13 -0700690 ASSERT_EQ(0, pthread_create(&t2, nullptr, JoinFn, reinterpret_cast<void*>(t1)));
msg5550f020d12013-06-06 14:59:28 -0400691
692 sleep(1); // (Give t2 a chance to call pthread_join.)
693
694 // Multiple joins to the same thread should fail.
Yi Kong32bc0fc2018-08-02 17:31:13 -0700695 ASSERT_EQ(EINVAL, pthread_join(t1, nullptr));
msg5550f020d12013-06-06 14:59:28 -0400696
Elliott Hughes725b2a92016-03-23 11:20:47 -0700697 spin_helper.UnSpin();
msg5550f020d12013-06-06 14:59:28 -0400698
699 // ...but t2's join on t1 still goes ahead (which we can tell because our join on t2 finishes).
700 void* join_result;
701 ASSERT_EQ(0, pthread_join(t2, &join_result));
Elliott Hughes5b9310e2013-10-02 16:59:05 -0700702 ASSERT_EQ(0U, reinterpret_cast<uintptr_t>(join_result));
msg5550f020d12013-06-06 14:59:28 -0400703}
Elliott Hughesb95cf0d2013-07-15 14:51:07 -0700704
Elliott Hughes70b24b12013-11-15 11:51:07 -0800705TEST(pthread, pthread_join__race) {
706 // http://b/11693195 --- pthread_join could return before the thread had actually exited.
707 // If the joiner unmapped the thread's stack, that could lead to SIGSEGV in the thread.
708 for (size_t i = 0; i < 1024; ++i) {
Yabin Cuia36158a2015-11-16 21:06:16 -0800709 size_t stack_size = 640*1024;
Yi Kong32bc0fc2018-08-02 17:31:13 -0700710 void* stack = mmap(nullptr, stack_size, PROT_READ|PROT_WRITE, MAP_ANON|MAP_PRIVATE, -1, 0);
Elliott Hughes70b24b12013-11-15 11:51:07 -0800711
712 pthread_attr_t a;
713 pthread_attr_init(&a);
714 pthread_attr_setstack(&a, stack, stack_size);
715
716 pthread_t t;
Yi Kong32bc0fc2018-08-02 17:31:13 -0700717 ASSERT_EQ(0, pthread_create(&t, &a, IdFn, nullptr));
718 ASSERT_EQ(0, pthread_join(t, nullptr));
Elliott Hughes70b24b12013-11-15 11:51:07 -0800719 ASSERT_EQ(0, munmap(stack, stack_size));
720 }
721}
722
Elliott Hughesb95cf0d2013-07-15 14:51:07 -0700723static void* GetActualGuardSizeFn(void* arg) {
724 pthread_attr_t attributes;
725 pthread_getattr_np(pthread_self(), &attributes);
726 pthread_attr_getguardsize(&attributes, reinterpret_cast<size_t*>(arg));
Yi Kong32bc0fc2018-08-02 17:31:13 -0700727 return nullptr;
Elliott Hughesb95cf0d2013-07-15 14:51:07 -0700728}
729
730static size_t GetActualGuardSize(const pthread_attr_t& attributes) {
731 size_t result;
732 pthread_t t;
733 pthread_create(&t, &attributes, GetActualGuardSizeFn, &result);
Yi Kong32bc0fc2018-08-02 17:31:13 -0700734 pthread_join(t, nullptr);
Elliott Hughesb95cf0d2013-07-15 14:51:07 -0700735 return result;
736}
737
738static void* GetActualStackSizeFn(void* arg) {
739 pthread_attr_t attributes;
740 pthread_getattr_np(pthread_self(), &attributes);
741 pthread_attr_getstacksize(&attributes, reinterpret_cast<size_t*>(arg));
Yi Kong32bc0fc2018-08-02 17:31:13 -0700742 return nullptr;
Elliott Hughesb95cf0d2013-07-15 14:51:07 -0700743}
744
745static size_t GetActualStackSize(const pthread_attr_t& attributes) {
746 size_t result;
747 pthread_t t;
748 pthread_create(&t, &attributes, GetActualStackSizeFn, &result);
Yi Kong32bc0fc2018-08-02 17:31:13 -0700749 pthread_join(t, nullptr);
Elliott Hughesb95cf0d2013-07-15 14:51:07 -0700750 return result;
751}
752
Elliott Hughesd6c678c2017-06-27 17:01:57 -0700753TEST(pthread, pthread_attr_setguardsize_tiny) {
Elliott Hughesb95cf0d2013-07-15 14:51:07 -0700754 pthread_attr_t attributes;
755 ASSERT_EQ(0, pthread_attr_init(&attributes));
756
Elliott Hughesb95cf0d2013-07-15 14:51:07 -0700757 // No such thing as too small: will be rounded up to one page by pthread_create.
758 ASSERT_EQ(0, pthread_attr_setguardsize(&attributes, 128));
759 size_t guard_size;
760 ASSERT_EQ(0, pthread_attr_getguardsize(&attributes, &guard_size));
761 ASSERT_EQ(128U, guard_size);
762 ASSERT_EQ(4096U, GetActualGuardSize(attributes));
Elliott Hughesd6c678c2017-06-27 17:01:57 -0700763}
764
765TEST(pthread, pthread_attr_setguardsize_reasonable) {
766 pthread_attr_t attributes;
767 ASSERT_EQ(0, pthread_attr_init(&attributes));
Elliott Hughesb95cf0d2013-07-15 14:51:07 -0700768
769 // Large enough and a multiple of the page size.
770 ASSERT_EQ(0, pthread_attr_setguardsize(&attributes, 32*1024));
Elliott Hughesd6c678c2017-06-27 17:01:57 -0700771 size_t guard_size;
Elliott Hughesb95cf0d2013-07-15 14:51:07 -0700772 ASSERT_EQ(0, pthread_attr_getguardsize(&attributes, &guard_size));
773 ASSERT_EQ(32*1024U, guard_size);
Elliott Hughesd6c678c2017-06-27 17:01:57 -0700774 ASSERT_EQ(32*1024U, GetActualGuardSize(attributes));
775}
Elliott Hughesb95cf0d2013-07-15 14:51:07 -0700776
Elliott Hughesd6c678c2017-06-27 17:01:57 -0700777TEST(pthread, pthread_attr_setguardsize_needs_rounding) {
778 pthread_attr_t attributes;
779 ASSERT_EQ(0, pthread_attr_init(&attributes));
780
781 // Large enough but not a multiple of the page size.
Elliott Hughesb95cf0d2013-07-15 14:51:07 -0700782 ASSERT_EQ(0, pthread_attr_setguardsize(&attributes, 32*1024 + 1));
Elliott Hughesd6c678c2017-06-27 17:01:57 -0700783 size_t guard_size;
Elliott Hughesb95cf0d2013-07-15 14:51:07 -0700784 ASSERT_EQ(0, pthread_attr_getguardsize(&attributes, &guard_size));
785 ASSERT_EQ(32*1024U + 1, guard_size);
Elliott Hughesd6c678c2017-06-27 17:01:57 -0700786 ASSERT_EQ(36*1024U, GetActualGuardSize(attributes));
787}
788
789TEST(pthread, pthread_attr_setguardsize_enormous) {
790 pthread_attr_t attributes;
791 ASSERT_EQ(0, pthread_attr_init(&attributes));
792
793 // Larger than the stack itself. (Historically we mistakenly carved
794 // the guard out of the stack itself, rather than adding it after the
795 // end.)
796 ASSERT_EQ(0, pthread_attr_setguardsize(&attributes, 32*1024*1024));
797 size_t guard_size;
798 ASSERT_EQ(0, pthread_attr_getguardsize(&attributes, &guard_size));
799 ASSERT_EQ(32*1024*1024U, guard_size);
800 ASSERT_EQ(32*1024*1024U, GetActualGuardSize(attributes));
Elliott Hughesb95cf0d2013-07-15 14:51:07 -0700801}
802
803TEST(pthread, pthread_attr_setstacksize) {
804 pthread_attr_t attributes;
805 ASSERT_EQ(0, pthread_attr_init(&attributes));
806
807 // Get the default stack size.
808 size_t default_stack_size;
809 ASSERT_EQ(0, pthread_attr_getstacksize(&attributes, &default_stack_size));
810
811 // Too small.
812 ASSERT_EQ(EINVAL, pthread_attr_setstacksize(&attributes, 128));
813 size_t stack_size;
814 ASSERT_EQ(0, pthread_attr_getstacksize(&attributes, &stack_size));
815 ASSERT_EQ(default_stack_size, stack_size);
816 ASSERT_GE(GetActualStackSize(attributes), default_stack_size);
817
Yabin Cui917d3902015-01-08 12:32:42 -0800818 // Large enough and a multiple of the page size; may be rounded up by pthread_create.
Elliott Hughesb95cf0d2013-07-15 14:51:07 -0700819 ASSERT_EQ(0, pthread_attr_setstacksize(&attributes, 32*1024));
820 ASSERT_EQ(0, pthread_attr_getstacksize(&attributes, &stack_size));
821 ASSERT_EQ(32*1024U, stack_size);
Yabin Cui917d3902015-01-08 12:32:42 -0800822 ASSERT_GE(GetActualStackSize(attributes), 32*1024U);
Elliott Hughesb95cf0d2013-07-15 14:51:07 -0700823
Yabin Cui917d3902015-01-08 12:32:42 -0800824 // Large enough but not aligned; will be rounded up by pthread_create.
Elliott Hughesb95cf0d2013-07-15 14:51:07 -0700825 ASSERT_EQ(0, pthread_attr_setstacksize(&attributes, 32*1024 + 1));
826 ASSERT_EQ(0, pthread_attr_getstacksize(&attributes, &stack_size));
827 ASSERT_EQ(32*1024U + 1, stack_size);
Christopher Ferrisf04935c2013-12-20 18:43:21 -0800828#if defined(__BIONIC__)
Yabin Cui917d3902015-01-08 12:32:42 -0800829 ASSERT_GT(GetActualStackSize(attributes), 32*1024U + 1);
Christopher Ferrisf04935c2013-12-20 18:43:21 -0800830#else // __BIONIC__
Elliott Hughesb95cf0d2013-07-15 14:51:07 -0700831 // glibc rounds down, in violation of POSIX. They document this in their BUGS section.
832 ASSERT_EQ(GetActualStackSize(attributes), 32*1024U);
Christopher Ferrisf04935c2013-12-20 18:43:21 -0800833#endif // __BIONIC__
Elliott Hughesb95cf0d2013-07-15 14:51:07 -0700834}
Elliott Hughesc3f11402013-10-30 14:40:09 -0700835
Yabin Cui76615da2015-03-17 14:22:09 -0700836TEST(pthread, pthread_rwlockattr_smoke) {
837 pthread_rwlockattr_t attr;
838 ASSERT_EQ(0, pthread_rwlockattr_init(&attr));
839
840 int pshared_value_array[] = {PTHREAD_PROCESS_PRIVATE, PTHREAD_PROCESS_SHARED};
841 for (size_t i = 0; i < sizeof(pshared_value_array) / sizeof(pshared_value_array[0]); ++i) {
842 ASSERT_EQ(0, pthread_rwlockattr_setpshared(&attr, pshared_value_array[i]));
843 int pshared;
844 ASSERT_EQ(0, pthread_rwlockattr_getpshared(&attr, &pshared));
845 ASSERT_EQ(pshared_value_array[i], pshared);
846 }
847
848 int kind_array[] = {PTHREAD_RWLOCK_PREFER_READER_NP,
849 PTHREAD_RWLOCK_PREFER_WRITER_NONRECURSIVE_NP};
850 for (size_t i = 0; i < sizeof(kind_array) / sizeof(kind_array[0]); ++i) {
851 ASSERT_EQ(0, pthread_rwlockattr_setkind_np(&attr, kind_array[i]));
852 int kind;
853 ASSERT_EQ(0, pthread_rwlockattr_getkind_np(&attr, &kind));
854 ASSERT_EQ(kind_array[i], kind);
855 }
856
857 ASSERT_EQ(0, pthread_rwlockattr_destroy(&attr));
858}
859
860TEST(pthread, pthread_rwlock_init_same_as_PTHREAD_RWLOCK_INITIALIZER) {
861 pthread_rwlock_t lock1 = PTHREAD_RWLOCK_INITIALIZER;
862 pthread_rwlock_t lock2;
Yi Kong32bc0fc2018-08-02 17:31:13 -0700863 ASSERT_EQ(0, pthread_rwlock_init(&lock2, nullptr));
Yabin Cui76615da2015-03-17 14:22:09 -0700864 ASSERT_EQ(0, memcmp(&lock1, &lock2, sizeof(lock1)));
865}
866
Elliott Hughesc3f11402013-10-30 14:40:09 -0700867TEST(pthread, pthread_rwlock_smoke) {
868 pthread_rwlock_t l;
Yi Kong32bc0fc2018-08-02 17:31:13 -0700869 ASSERT_EQ(0, pthread_rwlock_init(&l, nullptr));
Elliott Hughesc3f11402013-10-30 14:40:09 -0700870
Calin Juravle76f352e2014-05-19 13:41:10 +0100871 // Single read lock
Elliott Hughesc3f11402013-10-30 14:40:09 -0700872 ASSERT_EQ(0, pthread_rwlock_rdlock(&l));
873 ASSERT_EQ(0, pthread_rwlock_unlock(&l));
874
Calin Juravle76f352e2014-05-19 13:41:10 +0100875 // Multiple read lock
876 ASSERT_EQ(0, pthread_rwlock_rdlock(&l));
877 ASSERT_EQ(0, pthread_rwlock_rdlock(&l));
878 ASSERT_EQ(0, pthread_rwlock_unlock(&l));
879 ASSERT_EQ(0, pthread_rwlock_unlock(&l));
880
881 // Write lock
Calin Juravle92687e42014-05-22 19:21:22 +0100882 ASSERT_EQ(0, pthread_rwlock_wrlock(&l));
883 ASSERT_EQ(0, pthread_rwlock_unlock(&l));
Calin Juravle76f352e2014-05-19 13:41:10 +0100884
885 // Try writer lock
886 ASSERT_EQ(0, pthread_rwlock_trywrlock(&l));
887 ASSERT_EQ(EBUSY, pthread_rwlock_trywrlock(&l));
888 ASSERT_EQ(EBUSY, pthread_rwlock_tryrdlock(&l));
889 ASSERT_EQ(0, pthread_rwlock_unlock(&l));
890
891 // Try reader lock
892 ASSERT_EQ(0, pthread_rwlock_tryrdlock(&l));
893 ASSERT_EQ(0, pthread_rwlock_tryrdlock(&l));
894 ASSERT_EQ(EBUSY, pthread_rwlock_trywrlock(&l));
895 ASSERT_EQ(0, pthread_rwlock_unlock(&l));
896 ASSERT_EQ(0, pthread_rwlock_unlock(&l));
897
898 // Try writer lock after unlock
Elliott Hughesc3f11402013-10-30 14:40:09 -0700899 ASSERT_EQ(0, pthread_rwlock_wrlock(&l));
900 ASSERT_EQ(0, pthread_rwlock_unlock(&l));
901
Calin Juravle76f352e2014-05-19 13:41:10 +0100902 // EDEADLK in "read after write"
903 ASSERT_EQ(0, pthread_rwlock_wrlock(&l));
904 ASSERT_EQ(EDEADLK, pthread_rwlock_rdlock(&l));
905 ASSERT_EQ(0, pthread_rwlock_unlock(&l));
906
907 // EDEADLK in "write after write"
908 ASSERT_EQ(0, pthread_rwlock_wrlock(&l));
909 ASSERT_EQ(EDEADLK, pthread_rwlock_wrlock(&l));
910 ASSERT_EQ(0, pthread_rwlock_unlock(&l));
Calin Juravle76f352e2014-05-19 13:41:10 +0100911
Elliott Hughesc3f11402013-10-30 14:40:09 -0700912 ASSERT_EQ(0, pthread_rwlock_destroy(&l));
913}
914
Yabin Cui08ee8d22015-02-11 17:04:36 -0800915struct RwlockWakeupHelperArg {
916 pthread_rwlock_t lock;
917 enum Progress {
918 LOCK_INITIALIZED,
919 LOCK_WAITING,
920 LOCK_RELEASED,
Yabin Cuic9a659c2015-11-05 15:36:08 -0800921 LOCK_ACCESSED,
922 LOCK_TIMEDOUT,
Yabin Cui08ee8d22015-02-11 17:04:36 -0800923 };
924 std::atomic<Progress> progress;
Yabin Cuif7969852015-04-02 17:47:48 -0700925 std::atomic<pid_t> tid;
Yabin Cuic9a659c2015-11-05 15:36:08 -0800926 std::function<int (pthread_rwlock_t*)> trylock_function;
927 std::function<int (pthread_rwlock_t*)> lock_function;
928 std::function<int (pthread_rwlock_t*, const timespec*)> timed_lock_function;
Tom Cherryc6b5bcd2018-03-05 14:14:44 -0800929 clockid_t clock;
Yabin Cui08ee8d22015-02-11 17:04:36 -0800930};
931
Yabin Cuic9a659c2015-11-05 15:36:08 -0800932static void pthread_rwlock_wakeup_helper(RwlockWakeupHelperArg* arg) {
Yabin Cuif7969852015-04-02 17:47:48 -0700933 arg->tid = gettid();
Yabin Cui08ee8d22015-02-11 17:04:36 -0800934 ASSERT_EQ(RwlockWakeupHelperArg::LOCK_INITIALIZED, arg->progress);
935 arg->progress = RwlockWakeupHelperArg::LOCK_WAITING;
936
Yabin Cuic9a659c2015-11-05 15:36:08 -0800937 ASSERT_EQ(EBUSY, arg->trylock_function(&arg->lock));
938 ASSERT_EQ(0, arg->lock_function(&arg->lock));
Yabin Cui08ee8d22015-02-11 17:04:36 -0800939 ASSERT_EQ(RwlockWakeupHelperArg::LOCK_RELEASED, arg->progress);
940 ASSERT_EQ(0, pthread_rwlock_unlock(&arg->lock));
941
942 arg->progress = RwlockWakeupHelperArg::LOCK_ACCESSED;
943}
944
Yabin Cuic9a659c2015-11-05 15:36:08 -0800945static void test_pthread_rwlock_reader_wakeup_writer(std::function<int (pthread_rwlock_t*)> lock_function) {
Yabin Cui08ee8d22015-02-11 17:04:36 -0800946 RwlockWakeupHelperArg wakeup_arg;
Yi Kong32bc0fc2018-08-02 17:31:13 -0700947 ASSERT_EQ(0, pthread_rwlock_init(&wakeup_arg.lock, nullptr));
Yabin Cui08ee8d22015-02-11 17:04:36 -0800948 ASSERT_EQ(0, pthread_rwlock_rdlock(&wakeup_arg.lock));
949 wakeup_arg.progress = RwlockWakeupHelperArg::LOCK_INITIALIZED;
Yabin Cuif7969852015-04-02 17:47:48 -0700950 wakeup_arg.tid = 0;
Tom Cherry60ddedf2018-02-20 15:40:02 -0800951 wakeup_arg.trylock_function = &pthread_rwlock_trywrlock;
Yabin Cuic9a659c2015-11-05 15:36:08 -0800952 wakeup_arg.lock_function = lock_function;
Yabin Cui08ee8d22015-02-11 17:04:36 -0800953
954 pthread_t thread;
Yi Kong32bc0fc2018-08-02 17:31:13 -0700955 ASSERT_EQ(0, pthread_create(&thread, nullptr,
Yabin Cuic9a659c2015-11-05 15:36:08 -0800956 reinterpret_cast<void* (*)(void*)>(pthread_rwlock_wakeup_helper), &wakeup_arg));
Yabin Cuif7969852015-04-02 17:47:48 -0700957 WaitUntilThreadSleep(wakeup_arg.tid);
958 ASSERT_EQ(RwlockWakeupHelperArg::LOCK_WAITING, wakeup_arg.progress);
959
Yabin Cui08ee8d22015-02-11 17:04:36 -0800960 wakeup_arg.progress = RwlockWakeupHelperArg::LOCK_RELEASED;
961 ASSERT_EQ(0, pthread_rwlock_unlock(&wakeup_arg.lock));
962
Yi Kong32bc0fc2018-08-02 17:31:13 -0700963 ASSERT_EQ(0, pthread_join(thread, nullptr));
Yabin Cui08ee8d22015-02-11 17:04:36 -0800964 ASSERT_EQ(RwlockWakeupHelperArg::LOCK_ACCESSED, wakeup_arg.progress);
965 ASSERT_EQ(0, pthread_rwlock_destroy(&wakeup_arg.lock));
966}
967
Yabin Cuic9a659c2015-11-05 15:36:08 -0800968TEST(pthread, pthread_rwlock_reader_wakeup_writer) {
969 test_pthread_rwlock_reader_wakeup_writer(pthread_rwlock_wrlock);
Yabin Cui08ee8d22015-02-11 17:04:36 -0800970}
971
Yabin Cuic9a659c2015-11-05 15:36:08 -0800972TEST(pthread, pthread_rwlock_reader_wakeup_writer_timedwait) {
973 timespec ts;
974 ASSERT_EQ(0, clock_gettime(CLOCK_REALTIME, &ts));
975 ts.tv_sec += 1;
976 test_pthread_rwlock_reader_wakeup_writer([&](pthread_rwlock_t* lock) {
977 return pthread_rwlock_timedwrlock(lock, &ts);
978 });
979}
980
Tom Cherryc6b5bcd2018-03-05 14:14:44 -0800981TEST(pthread, pthread_rwlock_reader_wakeup_writer_timedwait_monotonic_np) {
982#if defined(__BIONIC__)
983 timespec ts;
984 ASSERT_EQ(0, clock_gettime(CLOCK_MONOTONIC, &ts));
985 ts.tv_sec += 1;
986 test_pthread_rwlock_reader_wakeup_writer(
987 [&](pthread_rwlock_t* lock) { return pthread_rwlock_timedwrlock_monotonic_np(lock, &ts); });
988#else // __BIONIC__
Elliott Hughesbcaa4542019-03-08 15:20:23 -0800989 GTEST_SKIP() << "pthread_rwlock_timedwrlock_monotonic_np not available";
Tom Cherryc6b5bcd2018-03-05 14:14:44 -0800990#endif // __BIONIC__
991}
992
Yabin Cuic9a659c2015-11-05 15:36:08 -0800993static void test_pthread_rwlock_writer_wakeup_reader(std::function<int (pthread_rwlock_t*)> lock_function) {
Yabin Cui08ee8d22015-02-11 17:04:36 -0800994 RwlockWakeupHelperArg wakeup_arg;
Yi Kong32bc0fc2018-08-02 17:31:13 -0700995 ASSERT_EQ(0, pthread_rwlock_init(&wakeup_arg.lock, nullptr));
Yabin Cui08ee8d22015-02-11 17:04:36 -0800996 ASSERT_EQ(0, pthread_rwlock_wrlock(&wakeup_arg.lock));
997 wakeup_arg.progress = RwlockWakeupHelperArg::LOCK_INITIALIZED;
Yabin Cuif7969852015-04-02 17:47:48 -0700998 wakeup_arg.tid = 0;
Tom Cherry60ddedf2018-02-20 15:40:02 -0800999 wakeup_arg.trylock_function = &pthread_rwlock_tryrdlock;
Yabin Cuic9a659c2015-11-05 15:36:08 -08001000 wakeup_arg.lock_function = lock_function;
Yabin Cui08ee8d22015-02-11 17:04:36 -08001001
1002 pthread_t thread;
Yi Kong32bc0fc2018-08-02 17:31:13 -07001003 ASSERT_EQ(0, pthread_create(&thread, nullptr,
Yabin Cuic9a659c2015-11-05 15:36:08 -08001004 reinterpret_cast<void* (*)(void*)>(pthread_rwlock_wakeup_helper), &wakeup_arg));
Yabin Cuif7969852015-04-02 17:47:48 -07001005 WaitUntilThreadSleep(wakeup_arg.tid);
1006 ASSERT_EQ(RwlockWakeupHelperArg::LOCK_WAITING, wakeup_arg.progress);
1007
Yabin Cui08ee8d22015-02-11 17:04:36 -08001008 wakeup_arg.progress = RwlockWakeupHelperArg::LOCK_RELEASED;
1009 ASSERT_EQ(0, pthread_rwlock_unlock(&wakeup_arg.lock));
1010
Yi Kong32bc0fc2018-08-02 17:31:13 -07001011 ASSERT_EQ(0, pthread_join(thread, nullptr));
Yabin Cui08ee8d22015-02-11 17:04:36 -08001012 ASSERT_EQ(RwlockWakeupHelperArg::LOCK_ACCESSED, wakeup_arg.progress);
1013 ASSERT_EQ(0, pthread_rwlock_destroy(&wakeup_arg.lock));
1014}
1015
Yabin Cuic9a659c2015-11-05 15:36:08 -08001016TEST(pthread, pthread_rwlock_writer_wakeup_reader) {
1017 test_pthread_rwlock_writer_wakeup_reader(pthread_rwlock_rdlock);
1018}
1019
1020TEST(pthread, pthread_rwlock_writer_wakeup_reader_timedwait) {
1021 timespec ts;
1022 ASSERT_EQ(0, clock_gettime(CLOCK_REALTIME, &ts));
1023 ts.tv_sec += 1;
1024 test_pthread_rwlock_writer_wakeup_reader([&](pthread_rwlock_t* lock) {
1025 return pthread_rwlock_timedrdlock(lock, &ts);
1026 });
1027}
1028
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08001029TEST(pthread, pthread_rwlock_writer_wakeup_reader_timedwait_monotonic_np) {
1030#if defined(__BIONIC__)
1031 timespec ts;
1032 ASSERT_EQ(0, clock_gettime(CLOCK_MONOTONIC, &ts));
1033 ts.tv_sec += 1;
1034 test_pthread_rwlock_writer_wakeup_reader(
1035 [&](pthread_rwlock_t* lock) { return pthread_rwlock_timedrdlock_monotonic_np(lock, &ts); });
1036#else // __BIONIC__
Elliott Hughesbcaa4542019-03-08 15:20:23 -08001037 GTEST_SKIP() << "pthread_rwlock_timedrdlock_monotonic_np not available";
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08001038#endif // __BIONIC__
1039}
1040
Yabin Cuic9a659c2015-11-05 15:36:08 -08001041static void pthread_rwlock_wakeup_timeout_helper(RwlockWakeupHelperArg* arg) {
1042 arg->tid = gettid();
1043 ASSERT_EQ(RwlockWakeupHelperArg::LOCK_INITIALIZED, arg->progress);
1044 arg->progress = RwlockWakeupHelperArg::LOCK_WAITING;
1045
1046 ASSERT_EQ(EBUSY, arg->trylock_function(&arg->lock));
1047
1048 timespec ts;
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08001049 ASSERT_EQ(0, clock_gettime(arg->clock, &ts));
Yabin Cuic9a659c2015-11-05 15:36:08 -08001050 ASSERT_EQ(ETIMEDOUT, arg->timed_lock_function(&arg->lock, &ts));
1051 ts.tv_nsec = -1;
1052 ASSERT_EQ(EINVAL, arg->timed_lock_function(&arg->lock, &ts));
1053 ts.tv_nsec = NS_PER_S;
1054 ASSERT_EQ(EINVAL, arg->timed_lock_function(&arg->lock, &ts));
1055 ts.tv_nsec = NS_PER_S - 1;
1056 ts.tv_sec = -1;
1057 ASSERT_EQ(ETIMEDOUT, arg->timed_lock_function(&arg->lock, &ts));
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08001058 ASSERT_EQ(0, clock_gettime(arg->clock, &ts));
Yabin Cuic9a659c2015-11-05 15:36:08 -08001059 ts.tv_sec += 1;
1060 ASSERT_EQ(ETIMEDOUT, arg->timed_lock_function(&arg->lock, &ts));
1061 ASSERT_EQ(RwlockWakeupHelperArg::LOCK_WAITING, arg->progress);
1062 arg->progress = RwlockWakeupHelperArg::LOCK_TIMEDOUT;
1063}
1064
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08001065static void pthread_rwlock_timedrdlock_timeout_helper(
1066 clockid_t clock, int (*lock_function)(pthread_rwlock_t* __rwlock, const timespec* __timeout)) {
Yabin Cuic9a659c2015-11-05 15:36:08 -08001067 RwlockWakeupHelperArg wakeup_arg;
1068 ASSERT_EQ(0, pthread_rwlock_init(&wakeup_arg.lock, nullptr));
1069 ASSERT_EQ(0, pthread_rwlock_wrlock(&wakeup_arg.lock));
1070 wakeup_arg.progress = RwlockWakeupHelperArg::LOCK_INITIALIZED;
1071 wakeup_arg.tid = 0;
Tom Cherry60ddedf2018-02-20 15:40:02 -08001072 wakeup_arg.trylock_function = &pthread_rwlock_tryrdlock;
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08001073 wakeup_arg.timed_lock_function = lock_function;
1074 wakeup_arg.clock = clock;
1075
1076 pthread_t thread;
1077 ASSERT_EQ(0, pthread_create(&thread, nullptr,
1078 reinterpret_cast<void* (*)(void*)>(pthread_rwlock_wakeup_timeout_helper), &wakeup_arg));
1079 WaitUntilThreadSleep(wakeup_arg.tid);
1080 ASSERT_EQ(RwlockWakeupHelperArg::LOCK_WAITING, wakeup_arg.progress);
1081
1082 ASSERT_EQ(0, pthread_join(thread, nullptr));
1083 ASSERT_EQ(RwlockWakeupHelperArg::LOCK_TIMEDOUT, wakeup_arg.progress);
1084 ASSERT_EQ(0, pthread_rwlock_unlock(&wakeup_arg.lock));
1085 ASSERT_EQ(0, pthread_rwlock_destroy(&wakeup_arg.lock));
1086}
1087
1088TEST(pthread, pthread_rwlock_timedrdlock_timeout) {
1089 pthread_rwlock_timedrdlock_timeout_helper(CLOCK_REALTIME, pthread_rwlock_timedrdlock);
1090}
1091
1092TEST(pthread, pthread_rwlock_timedrdlock_monotonic_np_timeout) {
1093#if defined(__BIONIC__)
1094 pthread_rwlock_timedrdlock_timeout_helper(CLOCK_MONOTONIC,
1095 pthread_rwlock_timedrdlock_monotonic_np);
1096#else // __BIONIC__
Elliott Hughesbcaa4542019-03-08 15:20:23 -08001097 GTEST_SKIP() << "pthread_rwlock_timedrdlock_monotonic_np not available";
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08001098#endif // __BIONIC__
1099}
1100
1101static void pthread_rwlock_timedwrlock_timeout_helper(
1102 clockid_t clock, int (*lock_function)(pthread_rwlock_t* __rwlock, const timespec* __timeout)) {
1103 RwlockWakeupHelperArg wakeup_arg;
1104 ASSERT_EQ(0, pthread_rwlock_init(&wakeup_arg.lock, nullptr));
1105 ASSERT_EQ(0, pthread_rwlock_rdlock(&wakeup_arg.lock));
1106 wakeup_arg.progress = RwlockWakeupHelperArg::LOCK_INITIALIZED;
1107 wakeup_arg.tid = 0;
1108 wakeup_arg.trylock_function = &pthread_rwlock_trywrlock;
1109 wakeup_arg.timed_lock_function = lock_function;
1110 wakeup_arg.clock = clock;
Yabin Cuic9a659c2015-11-05 15:36:08 -08001111
1112 pthread_t thread;
1113 ASSERT_EQ(0, pthread_create(&thread, nullptr,
1114 reinterpret_cast<void* (*)(void*)>(pthread_rwlock_wakeup_timeout_helper), &wakeup_arg));
1115 WaitUntilThreadSleep(wakeup_arg.tid);
1116 ASSERT_EQ(RwlockWakeupHelperArg::LOCK_WAITING, wakeup_arg.progress);
1117
1118 ASSERT_EQ(0, pthread_join(thread, nullptr));
1119 ASSERT_EQ(RwlockWakeupHelperArg::LOCK_TIMEDOUT, wakeup_arg.progress);
1120 ASSERT_EQ(0, pthread_rwlock_unlock(&wakeup_arg.lock));
1121 ASSERT_EQ(0, pthread_rwlock_destroy(&wakeup_arg.lock));
1122}
1123
1124TEST(pthread, pthread_rwlock_timedwrlock_timeout) {
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08001125 pthread_rwlock_timedwrlock_timeout_helper(CLOCK_REALTIME, pthread_rwlock_timedwrlock);
1126}
Yabin Cuic9a659c2015-11-05 15:36:08 -08001127
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08001128TEST(pthread, pthread_rwlock_timedwrlock_monotonic_np_timeout) {
1129#if defined(__BIONIC__)
1130 pthread_rwlock_timedwrlock_timeout_helper(CLOCK_MONOTONIC,
1131 pthread_rwlock_timedwrlock_monotonic_np);
1132#else // __BIONIC__
Elliott Hughesbcaa4542019-03-08 15:20:23 -08001133 GTEST_SKIP() << "pthread_rwlock_timedwrlock_monotonic_np not available";
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08001134#endif // __BIONIC__
Yabin Cuic9a659c2015-11-05 15:36:08 -08001135}
1136
Yabin Cui76615da2015-03-17 14:22:09 -07001137class RwlockKindTestHelper {
1138 private:
1139 struct ThreadArg {
1140 RwlockKindTestHelper* helper;
1141 std::atomic<pid_t>& tid;
1142
1143 ThreadArg(RwlockKindTestHelper* helper, std::atomic<pid_t>& tid)
1144 : helper(helper), tid(tid) { }
1145 };
1146
1147 public:
1148 pthread_rwlock_t lock;
1149
1150 public:
Chih-Hung Hsieh62e3a072016-05-03 12:08:05 -07001151 explicit RwlockKindTestHelper(int kind_type) {
Yabin Cui76615da2015-03-17 14:22:09 -07001152 InitRwlock(kind_type);
1153 }
1154
1155 ~RwlockKindTestHelper() {
1156 DestroyRwlock();
1157 }
1158
1159 void CreateWriterThread(pthread_t& thread, std::atomic<pid_t>& tid) {
1160 tid = 0;
1161 ThreadArg* arg = new ThreadArg(this, tid);
Yi Kong32bc0fc2018-08-02 17:31:13 -07001162 ASSERT_EQ(0, pthread_create(&thread, nullptr,
Yabin Cui76615da2015-03-17 14:22:09 -07001163 reinterpret_cast<void* (*)(void*)>(WriterThreadFn), arg));
1164 }
1165
1166 void CreateReaderThread(pthread_t& thread, std::atomic<pid_t>& tid) {
1167 tid = 0;
1168 ThreadArg* arg = new ThreadArg(this, tid);
Yi Kong32bc0fc2018-08-02 17:31:13 -07001169 ASSERT_EQ(0, pthread_create(&thread, nullptr,
Yabin Cui76615da2015-03-17 14:22:09 -07001170 reinterpret_cast<void* (*)(void*)>(ReaderThreadFn), arg));
1171 }
1172
1173 private:
1174 void InitRwlock(int kind_type) {
1175 pthread_rwlockattr_t attr;
1176 ASSERT_EQ(0, pthread_rwlockattr_init(&attr));
1177 ASSERT_EQ(0, pthread_rwlockattr_setkind_np(&attr, kind_type));
1178 ASSERT_EQ(0, pthread_rwlock_init(&lock, &attr));
1179 ASSERT_EQ(0, pthread_rwlockattr_destroy(&attr));
1180 }
1181
1182 void DestroyRwlock() {
1183 ASSERT_EQ(0, pthread_rwlock_destroy(&lock));
1184 }
1185
1186 static void WriterThreadFn(ThreadArg* arg) {
1187 arg->tid = gettid();
1188
1189 RwlockKindTestHelper* helper = arg->helper;
1190 ASSERT_EQ(0, pthread_rwlock_wrlock(&helper->lock));
1191 ASSERT_EQ(0, pthread_rwlock_unlock(&helper->lock));
1192 delete arg;
1193 }
1194
1195 static void ReaderThreadFn(ThreadArg* arg) {
1196 arg->tid = gettid();
1197
1198 RwlockKindTestHelper* helper = arg->helper;
1199 ASSERT_EQ(0, pthread_rwlock_rdlock(&helper->lock));
1200 ASSERT_EQ(0, pthread_rwlock_unlock(&helper->lock));
1201 delete arg;
1202 }
1203};
1204
1205TEST(pthread, pthread_rwlock_kind_PTHREAD_RWLOCK_PREFER_READER_NP) {
1206 RwlockKindTestHelper helper(PTHREAD_RWLOCK_PREFER_READER_NP);
1207 ASSERT_EQ(0, pthread_rwlock_rdlock(&helper.lock));
1208
1209 pthread_t writer_thread;
1210 std::atomic<pid_t> writer_tid;
1211 helper.CreateWriterThread(writer_thread, writer_tid);
1212 WaitUntilThreadSleep(writer_tid);
1213
1214 pthread_t reader_thread;
1215 std::atomic<pid_t> reader_tid;
1216 helper.CreateReaderThread(reader_thread, reader_tid);
Yi Kong32bc0fc2018-08-02 17:31:13 -07001217 ASSERT_EQ(0, pthread_join(reader_thread, nullptr));
Yabin Cui76615da2015-03-17 14:22:09 -07001218
1219 ASSERT_EQ(0, pthread_rwlock_unlock(&helper.lock));
Yi Kong32bc0fc2018-08-02 17:31:13 -07001220 ASSERT_EQ(0, pthread_join(writer_thread, nullptr));
Yabin Cui76615da2015-03-17 14:22:09 -07001221}
1222
1223TEST(pthread, pthread_rwlock_kind_PTHREAD_RWLOCK_PREFER_WRITER_NONRECURSIVE_NP) {
1224 RwlockKindTestHelper helper(PTHREAD_RWLOCK_PREFER_WRITER_NONRECURSIVE_NP);
1225 ASSERT_EQ(0, pthread_rwlock_rdlock(&helper.lock));
1226
1227 pthread_t writer_thread;
1228 std::atomic<pid_t> writer_tid;
1229 helper.CreateWriterThread(writer_thread, writer_tid);
1230 WaitUntilThreadSleep(writer_tid);
1231
1232 pthread_t reader_thread;
1233 std::atomic<pid_t> reader_tid;
1234 helper.CreateReaderThread(reader_thread, reader_tid);
1235 WaitUntilThreadSleep(reader_tid);
1236
1237 ASSERT_EQ(0, pthread_rwlock_unlock(&helper.lock));
Yi Kong32bc0fc2018-08-02 17:31:13 -07001238 ASSERT_EQ(0, pthread_join(writer_thread, nullptr));
1239 ASSERT_EQ(0, pthread_join(reader_thread, nullptr));
Yabin Cui76615da2015-03-17 14:22:09 -07001240}
1241
Elliott Hughes1728b232014-05-14 10:02:03 -07001242static int g_once_fn_call_count = 0;
Elliott Hughesc3f11402013-10-30 14:40:09 -07001243static void OnceFn() {
Elliott Hughes1728b232014-05-14 10:02:03 -07001244 ++g_once_fn_call_count;
Elliott Hughesc3f11402013-10-30 14:40:09 -07001245}
1246
1247TEST(pthread, pthread_once_smoke) {
1248 pthread_once_t once_control = PTHREAD_ONCE_INIT;
1249 ASSERT_EQ(0, pthread_once(&once_control, OnceFn));
1250 ASSERT_EQ(0, pthread_once(&once_control, OnceFn));
Elliott Hughes1728b232014-05-14 10:02:03 -07001251 ASSERT_EQ(1, g_once_fn_call_count);
Elliott Hughesc3f11402013-10-30 14:40:09 -07001252}
1253
Elliott Hughes3694ec62014-05-14 11:46:08 -07001254static std::string pthread_once_1934122_result = "";
1255
1256static void Routine2() {
1257 pthread_once_1934122_result += "2";
1258}
1259
1260static void Routine1() {
1261 pthread_once_t once_control_2 = PTHREAD_ONCE_INIT;
1262 pthread_once_1934122_result += "1";
1263 pthread_once(&once_control_2, &Routine2);
1264}
1265
1266TEST(pthread, pthread_once_1934122) {
1267 // Very old versions of Android couldn't call pthread_once from a
1268 // pthread_once init routine. http://b/1934122.
1269 pthread_once_t once_control_1 = PTHREAD_ONCE_INIT;
1270 ASSERT_EQ(0, pthread_once(&once_control_1, &Routine1));
1271 ASSERT_EQ("12", pthread_once_1934122_result);
1272}
1273
Elliott Hughes1728b232014-05-14 10:02:03 -07001274static int g_atfork_prepare_calls = 0;
Dmitriy Ivanovea295f62014-11-20 20:47:02 -08001275static void AtForkPrepare1() { g_atfork_prepare_calls = (g_atfork_prepare_calls * 10) + 1; }
1276static void AtForkPrepare2() { g_atfork_prepare_calls = (g_atfork_prepare_calls * 10) + 2; }
Elliott Hughes1728b232014-05-14 10:02:03 -07001277static int g_atfork_parent_calls = 0;
Dmitriy Ivanovea295f62014-11-20 20:47:02 -08001278static void AtForkParent1() { g_atfork_parent_calls = (g_atfork_parent_calls * 10) + 1; }
1279static void AtForkParent2() { g_atfork_parent_calls = (g_atfork_parent_calls * 10) + 2; }
Elliott Hughes1728b232014-05-14 10:02:03 -07001280static int g_atfork_child_calls = 0;
Dmitriy Ivanovea295f62014-11-20 20:47:02 -08001281static void AtForkChild1() { g_atfork_child_calls = (g_atfork_child_calls * 10) + 1; }
1282static void AtForkChild2() { g_atfork_child_calls = (g_atfork_child_calls * 10) + 2; }
Elliott Hughesc3f11402013-10-30 14:40:09 -07001283
Dmitriy Ivanov00e37812014-11-20 16:53:47 -08001284TEST(pthread, pthread_atfork_smoke) {
Dmitriy Ivanovcb0443c2015-03-16 14:15:46 -07001285 ASSERT_EQ(0, pthread_atfork(AtForkPrepare1, AtForkParent1, AtForkChild1));
1286 ASSERT_EQ(0, pthread_atfork(AtForkPrepare2, AtForkParent2, AtForkChild2));
Elliott Hughesc3f11402013-10-30 14:40:09 -07001287
Elliott Hughes33697a02016-01-26 13:04:57 -08001288 pid_t pid = fork();
Dmitriy Ivanovcb0443c2015-03-16 14:15:46 -07001289 ASSERT_NE(-1, pid) << strerror(errno);
Elliott Hughesc3f11402013-10-30 14:40:09 -07001290
Dmitriy Ivanovcb0443c2015-03-16 14:15:46 -07001291 // Child and parent calls are made in the order they were registered.
1292 if (pid == 0) {
Dmitriy Ivanovea295f62014-11-20 20:47:02 -08001293 ASSERT_EQ(12, g_atfork_child_calls);
Dmitriy Ivanovcb0443c2015-03-16 14:15:46 -07001294 _exit(0);
1295 }
Dmitriy Ivanovea295f62014-11-20 20:47:02 -08001296 ASSERT_EQ(12, g_atfork_parent_calls);
Elliott Hughesc3f11402013-10-30 14:40:09 -07001297
Dmitriy Ivanovcb0443c2015-03-16 14:15:46 -07001298 // Prepare calls are made in the reverse order.
Dmitriy Ivanovea295f62014-11-20 20:47:02 -08001299 ASSERT_EQ(21, g_atfork_prepare_calls);
Elliott Hughes33697a02016-01-26 13:04:57 -08001300 AssertChildExited(pid, 0);
Dmitriy Ivanovea295f62014-11-20 20:47:02 -08001301}
1302
Elliott Hughesc3f11402013-10-30 14:40:09 -07001303TEST(pthread, pthread_attr_getscope) {
1304 pthread_attr_t attr;
1305 ASSERT_EQ(0, pthread_attr_init(&attr));
1306
1307 int scope;
1308 ASSERT_EQ(0, pthread_attr_getscope(&attr, &scope));
1309 ASSERT_EQ(PTHREAD_SCOPE_SYSTEM, scope);
1310}
Narayan Kamath51e6cb32014-03-03 15:38:51 +00001311
1312TEST(pthread, pthread_condattr_init) {
1313 pthread_condattr_t attr;
1314 pthread_condattr_init(&attr);
1315
1316 clockid_t clock;
1317 ASSERT_EQ(0, pthread_condattr_getclock(&attr, &clock));
1318 ASSERT_EQ(CLOCK_REALTIME, clock);
1319
1320 int pshared;
1321 ASSERT_EQ(0, pthread_condattr_getpshared(&attr, &pshared));
1322 ASSERT_EQ(PTHREAD_PROCESS_PRIVATE, pshared);
1323}
1324
1325TEST(pthread, pthread_condattr_setclock) {
1326 pthread_condattr_t attr;
1327 pthread_condattr_init(&attr);
1328
1329 ASSERT_EQ(0, pthread_condattr_setclock(&attr, CLOCK_REALTIME));
1330 clockid_t clock;
1331 ASSERT_EQ(0, pthread_condattr_getclock(&attr, &clock));
1332 ASSERT_EQ(CLOCK_REALTIME, clock);
1333
1334 ASSERT_EQ(0, pthread_condattr_setclock(&attr, CLOCK_MONOTONIC));
1335 ASSERT_EQ(0, pthread_condattr_getclock(&attr, &clock));
1336 ASSERT_EQ(CLOCK_MONOTONIC, clock);
1337
1338 ASSERT_EQ(EINVAL, pthread_condattr_setclock(&attr, CLOCK_PROCESS_CPUTIME_ID));
1339}
1340
1341TEST(pthread, pthread_cond_broadcast__preserves_condattr_flags) {
Yabin Cui32651b82015-03-13 20:30:00 -07001342#if defined(__BIONIC__)
Narayan Kamath51e6cb32014-03-03 15:38:51 +00001343 pthread_condattr_t attr;
1344 pthread_condattr_init(&attr);
1345
1346 ASSERT_EQ(0, pthread_condattr_setclock(&attr, CLOCK_MONOTONIC));
1347 ASSERT_EQ(0, pthread_condattr_setpshared(&attr, PTHREAD_PROCESS_SHARED));
1348
1349 pthread_cond_t cond_var;
1350 ASSERT_EQ(0, pthread_cond_init(&cond_var, &attr));
1351
1352 ASSERT_EQ(0, pthread_cond_signal(&cond_var));
1353 ASSERT_EQ(0, pthread_cond_broadcast(&cond_var));
1354
Yabin Cui32651b82015-03-13 20:30:00 -07001355 attr = static_cast<pthread_condattr_t>(*reinterpret_cast<uint32_t*>(cond_var.__private));
Narayan Kamath51e6cb32014-03-03 15:38:51 +00001356 clockid_t clock;
1357 ASSERT_EQ(0, pthread_condattr_getclock(&attr, &clock));
1358 ASSERT_EQ(CLOCK_MONOTONIC, clock);
1359 int pshared;
1360 ASSERT_EQ(0, pthread_condattr_getpshared(&attr, &pshared));
1361 ASSERT_EQ(PTHREAD_PROCESS_SHARED, pshared);
Yabin Cui32651b82015-03-13 20:30:00 -07001362#else // !defined(__BIONIC__)
Elliott Hughesbcaa4542019-03-08 15:20:23 -08001363 GTEST_SKIP() << "bionic-only test";
Yabin Cui32651b82015-03-13 20:30:00 -07001364#endif // !defined(__BIONIC__)
1365}
1366
1367class pthread_CondWakeupTest : public ::testing::Test {
1368 protected:
1369 pthread_mutex_t mutex;
1370 pthread_cond_t cond;
1371
1372 enum Progress {
1373 INITIALIZED,
1374 WAITING,
1375 SIGNALED,
1376 FINISHED,
1377 };
1378 std::atomic<Progress> progress;
1379 pthread_t thread;
Yabin Cuic9a659c2015-11-05 15:36:08 -08001380 std::function<int (pthread_cond_t* cond, pthread_mutex_t* mutex)> wait_function;
Yabin Cui32651b82015-03-13 20:30:00 -07001381
1382 protected:
Yabin Cuic9a659c2015-11-05 15:36:08 -08001383 void SetUp() override {
1384 ASSERT_EQ(0, pthread_mutex_init(&mutex, nullptr));
1385 }
1386
1387 void InitCond(clockid_t clock=CLOCK_REALTIME) {
1388 pthread_condattr_t attr;
1389 ASSERT_EQ(0, pthread_condattr_init(&attr));
1390 ASSERT_EQ(0, pthread_condattr_setclock(&attr, clock));
1391 ASSERT_EQ(0, pthread_cond_init(&cond, &attr));
1392 ASSERT_EQ(0, pthread_condattr_destroy(&attr));
1393 }
1394
1395 void StartWaitingThread(std::function<int (pthread_cond_t* cond, pthread_mutex_t* mutex)> wait_function) {
Yabin Cui32651b82015-03-13 20:30:00 -07001396 progress = INITIALIZED;
Yabin Cuic9a659c2015-11-05 15:36:08 -08001397 this->wait_function = wait_function;
Yi Kong32bc0fc2018-08-02 17:31:13 -07001398 ASSERT_EQ(0, pthread_create(&thread, nullptr, reinterpret_cast<void* (*)(void*)>(WaitThreadFn), this));
Yabin Cuic9a659c2015-11-05 15:36:08 -08001399 while (progress != WAITING) {
Yabin Cui32651b82015-03-13 20:30:00 -07001400 usleep(5000);
1401 }
1402 usleep(5000);
1403 }
1404
Yabin Cuic9a659c2015-11-05 15:36:08 -08001405 void TearDown() override {
1406 ASSERT_EQ(0, pthread_join(thread, nullptr));
1407 ASSERT_EQ(FINISHED, progress);
1408 ASSERT_EQ(0, pthread_cond_destroy(&cond));
1409 ASSERT_EQ(0, pthread_mutex_destroy(&mutex));
1410 }
1411
Yabin Cui32651b82015-03-13 20:30:00 -07001412 private:
1413 static void WaitThreadFn(pthread_CondWakeupTest* test) {
1414 ASSERT_EQ(0, pthread_mutex_lock(&test->mutex));
1415 test->progress = WAITING;
1416 while (test->progress == WAITING) {
Yabin Cuic9a659c2015-11-05 15:36:08 -08001417 ASSERT_EQ(0, test->wait_function(&test->cond, &test->mutex));
Yabin Cui32651b82015-03-13 20:30:00 -07001418 }
1419 ASSERT_EQ(SIGNALED, test->progress);
1420 test->progress = FINISHED;
1421 ASSERT_EQ(0, pthread_mutex_unlock(&test->mutex));
1422 }
1423};
1424
Yabin Cuic9a659c2015-11-05 15:36:08 -08001425TEST_F(pthread_CondWakeupTest, signal_wait) {
1426 InitCond();
1427 StartWaitingThread([](pthread_cond_t* cond, pthread_mutex_t* mutex) {
1428 return pthread_cond_wait(cond, mutex);
1429 });
Yabin Cui32651b82015-03-13 20:30:00 -07001430 progress = SIGNALED;
Yabin Cuic9a659c2015-11-05 15:36:08 -08001431 ASSERT_EQ(0, pthread_cond_signal(&cond));
Yabin Cui32651b82015-03-13 20:30:00 -07001432}
1433
Yabin Cuic9a659c2015-11-05 15:36:08 -08001434TEST_F(pthread_CondWakeupTest, broadcast_wait) {
1435 InitCond();
1436 StartWaitingThread([](pthread_cond_t* cond, pthread_mutex_t* mutex) {
1437 return pthread_cond_wait(cond, mutex);
1438 });
Yabin Cui32651b82015-03-13 20:30:00 -07001439 progress = SIGNALED;
Yabin Cuic9a659c2015-11-05 15:36:08 -08001440 ASSERT_EQ(0, pthread_cond_broadcast(&cond));
Narayan Kamath51e6cb32014-03-03 15:38:51 +00001441}
Elliott Hughes0e714a52014-03-03 16:42:47 -08001442
Yabin Cuic9a659c2015-11-05 15:36:08 -08001443TEST_F(pthread_CondWakeupTest, signal_timedwait_CLOCK_REALTIME) {
1444 InitCond(CLOCK_REALTIME);
Elliott Hughes0e714a52014-03-03 16:42:47 -08001445 timespec ts;
1446 ASSERT_EQ(0, clock_gettime(CLOCK_REALTIME, &ts));
Yabin Cuic9a659c2015-11-05 15:36:08 -08001447 ts.tv_sec += 1;
1448 StartWaitingThread([&](pthread_cond_t* cond, pthread_mutex_t* mutex) {
1449 return pthread_cond_timedwait(cond, mutex, &ts);
1450 });
1451 progress = SIGNALED;
1452 ASSERT_EQ(0, pthread_cond_signal(&cond));
1453}
Elliott Hughes0e714a52014-03-03 16:42:47 -08001454
Yabin Cuic9a659c2015-11-05 15:36:08 -08001455TEST_F(pthread_CondWakeupTest, signal_timedwait_CLOCK_MONOTONIC) {
1456 InitCond(CLOCK_MONOTONIC);
1457 timespec ts;
1458 ASSERT_EQ(0, clock_gettime(CLOCK_MONOTONIC, &ts));
1459 ts.tv_sec += 1;
1460 StartWaitingThread([&](pthread_cond_t* cond, pthread_mutex_t* mutex) {
1461 return pthread_cond_timedwait(cond, mutex, &ts);
1462 });
1463 progress = SIGNALED;
1464 ASSERT_EQ(0, pthread_cond_signal(&cond));
1465}
Elliott Hughes0e714a52014-03-03 16:42:47 -08001466
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08001467TEST_F(pthread_CondWakeupTest, signal_timedwait_CLOCK_MONOTONIC_np) {
1468#if defined(__BIONIC__)
1469 InitCond(CLOCK_REALTIME);
1470 timespec ts;
1471 ASSERT_EQ(0, clock_gettime(CLOCK_MONOTONIC, &ts));
1472 ts.tv_sec += 1;
1473 StartWaitingThread([&](pthread_cond_t* cond, pthread_mutex_t* mutex) {
1474 return pthread_cond_timedwait_monotonic_np(cond, mutex, &ts);
1475 });
1476 progress = SIGNALED;
1477 ASSERT_EQ(0, pthread_cond_signal(&cond));
1478#else // __BIONIC__
Elliott Hughesbcaa4542019-03-08 15:20:23 -08001479 GTEST_SKIP() << "pthread_cond_timedwait_monotonic_np not available";
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08001480#endif // __BIONIC__
1481}
1482
1483static void pthread_cond_timedwait_timeout_helper(clockid_t clock,
1484 int (*wait_function)(pthread_cond_t* __cond,
1485 pthread_mutex_t* __mutex,
1486 const timespec* __timeout)) {
Yabin Cuic9a659c2015-11-05 15:36:08 -08001487 pthread_mutex_t mutex;
1488 ASSERT_EQ(0, pthread_mutex_init(&mutex, nullptr));
1489 pthread_cond_t cond;
1490 ASSERT_EQ(0, pthread_cond_init(&cond, nullptr));
1491 ASSERT_EQ(0, pthread_mutex_lock(&mutex));
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08001492
Yabin Cuic9a659c2015-11-05 15:36:08 -08001493 timespec ts;
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08001494 ASSERT_EQ(0, clock_gettime(clock, &ts));
1495 ASSERT_EQ(ETIMEDOUT, wait_function(&cond, &mutex, &ts));
Yabin Cuic9a659c2015-11-05 15:36:08 -08001496 ts.tv_nsec = -1;
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08001497 ASSERT_EQ(EINVAL, wait_function(&cond, &mutex, &ts));
Yabin Cuic9a659c2015-11-05 15:36:08 -08001498 ts.tv_nsec = NS_PER_S;
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08001499 ASSERT_EQ(EINVAL, wait_function(&cond, &mutex, &ts));
Yabin Cuic9a659c2015-11-05 15:36:08 -08001500 ts.tv_nsec = NS_PER_S - 1;
1501 ts.tv_sec = -1;
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08001502 ASSERT_EQ(ETIMEDOUT, wait_function(&cond, &mutex, &ts));
Yabin Cuic9a659c2015-11-05 15:36:08 -08001503 ASSERT_EQ(0, pthread_mutex_unlock(&mutex));
Elliott Hughes0e714a52014-03-03 16:42:47 -08001504}
Elliott Hughes57b7a612014-08-25 17:26:50 -07001505
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08001506TEST(pthread, pthread_cond_timedwait_timeout) {
1507 pthread_cond_timedwait_timeout_helper(CLOCK_REALTIME, pthread_cond_timedwait);
1508}
1509
1510TEST(pthread, pthread_cond_timedwait_monotonic_np_timeout) {
1511#if defined(__BIONIC__)
1512 pthread_cond_timedwait_timeout_helper(CLOCK_MONOTONIC, pthread_cond_timedwait_monotonic_np);
1513#else // __BIONIC__
Elliott Hughesbcaa4542019-03-08 15:20:23 -08001514 GTEST_SKIP() << "pthread_cond_timedwait_monotonic_np not available";
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08001515#endif // __BIONIC__
1516}
1517
Elliott Hughes57b7a612014-08-25 17:26:50 -07001518TEST(pthread, pthread_attr_getstack__main_thread) {
1519 // This test is only meaningful for the main thread, so make sure we're running on it!
1520 ASSERT_EQ(getpid(), syscall(__NR_gettid));
1521
1522 // Get the main thread's attributes.
1523 pthread_attr_t attributes;
1524 ASSERT_EQ(0, pthread_getattr_np(pthread_self(), &attributes));
1525
1526 // Check that we correctly report that the main thread has no guard page.
1527 size_t guard_size;
1528 ASSERT_EQ(0, pthread_attr_getguardsize(&attributes, &guard_size));
1529 ASSERT_EQ(0U, guard_size); // The main thread has no guard page.
1530
1531 // Get the stack base and the stack size (both ways).
1532 void* stack_base;
1533 size_t stack_size;
1534 ASSERT_EQ(0, pthread_attr_getstack(&attributes, &stack_base, &stack_size));
1535 size_t stack_size2;
1536 ASSERT_EQ(0, pthread_attr_getstacksize(&attributes, &stack_size2));
1537
1538 // The two methods of asking for the stack size should agree.
1539 EXPECT_EQ(stack_size, stack_size2);
1540
Yabin Cuib0c6f2db2015-05-19 15:09:23 -07001541#if defined(__BIONIC__)
dimitry6dfa5b52018-01-30 13:24:28 +01001542 // Find stack in /proc/self/maps using a pointer to the stack.
1543 //
1544 // We do not use "[stack]" label because in native-bridge environment it is not
1545 // guaranteed to point to the right stack. A native bridge implementation may
1546 // keep separate stack for the guest code.
Yi Kong32bc0fc2018-08-02 17:31:13 -07001547 void* maps_stack_hi = nullptr;
Elliott Hughes15dfd632015-09-22 16:40:14 -07001548 std::vector<map_record> maps;
1549 ASSERT_TRUE(Maps::parse_maps(&maps));
Evgenii Stepanov7cc67062019-02-05 18:43:34 -08001550 uintptr_t stack_address = reinterpret_cast<uintptr_t>(untag_address(&maps_stack_hi));
Elliott Hughes0b2acdf2015-10-02 18:25:19 -07001551 for (const auto& map : maps) {
dimitry6dfa5b52018-01-30 13:24:28 +01001552 if (map.addr_start <= stack_address && map.addr_end > stack_address){
Elliott Hughes15dfd632015-09-22 16:40:14 -07001553 maps_stack_hi = reinterpret_cast<void*>(map.addr_end);
Elliott Hughes57b7a612014-08-25 17:26:50 -07001554 break;
1555 }
1556 }
Elliott Hughes57b7a612014-08-25 17:26:50 -07001557
dimitry6dfa5b52018-01-30 13:24:28 +01001558 // The high address of the /proc/self/maps stack region should equal stack_base + stack_size.
Yabin Cuib0c6f2db2015-05-19 15:09:23 -07001559 // Remember that the stack grows down (and is mapped in on demand), so the low address of the
1560 // region isn't very interesting.
1561 EXPECT_EQ(maps_stack_hi, reinterpret_cast<uint8_t*>(stack_base) + stack_size);
1562
Elliott Hughes9e4ffa72014-08-27 15:32:01 -07001563 // The stack size should correspond to RLIMIT_STACK.
Elliott Hughes57b7a612014-08-25 17:26:50 -07001564 rlimit rl;
Elliott Hughes9e4ffa72014-08-27 15:32:01 -07001565 ASSERT_EQ(0, getrlimit(RLIMIT_STACK, &rl));
Elliott Hughes27a9aed2014-09-04 16:09:25 -07001566 uint64_t original_rlim_cur = rl.rlim_cur;
Elliott Hughes27a9aed2014-09-04 16:09:25 -07001567 if (rl.rlim_cur == RLIM_INFINITY) {
1568 rl.rlim_cur = 8 * 1024 * 1024; // Bionic reports unlimited stacks as 8MiB.
1569 }
Elliott Hughes9e4ffa72014-08-27 15:32:01 -07001570 EXPECT_EQ(rl.rlim_cur, stack_size);
1571
Tom Cherryb8ab6182017-04-05 16:20:29 -07001572 auto guard = android::base::make_scope_guard([&rl, original_rlim_cur]() {
Elliott Hughes27a9aed2014-09-04 16:09:25 -07001573 rl.rlim_cur = original_rlim_cur;
1574 ASSERT_EQ(0, setrlimit(RLIMIT_STACK, &rl));
1575 });
1576
Elliott Hughes9e4ffa72014-08-27 15:32:01 -07001577 //
1578 // What if RLIMIT_STACK is smaller than the stack's current extent?
1579 //
Elliott Hughes57b7a612014-08-25 17:26:50 -07001580 rl.rlim_cur = rl.rlim_max = 1024; // 1KiB. We know the stack must be at least a page already.
1581 rl.rlim_max = RLIM_INFINITY;
1582 ASSERT_EQ(0, setrlimit(RLIMIT_STACK, &rl));
1583
1584 ASSERT_EQ(0, pthread_getattr_np(pthread_self(), &attributes));
1585 ASSERT_EQ(0, pthread_attr_getstack(&attributes, &stack_base, &stack_size));
1586 ASSERT_EQ(0, pthread_attr_getstacksize(&attributes, &stack_size2));
1587
1588 EXPECT_EQ(stack_size, stack_size2);
1589 ASSERT_EQ(1024U, stack_size);
1590
1591 //
Elliott Hughes9e4ffa72014-08-27 15:32:01 -07001592 // What if RLIMIT_STACK isn't a whole number of pages?
Elliott Hughes57b7a612014-08-25 17:26:50 -07001593 //
1594 rl.rlim_cur = rl.rlim_max = 6666; // Not a whole number of pages.
1595 rl.rlim_max = RLIM_INFINITY;
1596 ASSERT_EQ(0, setrlimit(RLIMIT_STACK, &rl));
1597
1598 ASSERT_EQ(0, pthread_getattr_np(pthread_self(), &attributes));
1599 ASSERT_EQ(0, pthread_attr_getstack(&attributes, &stack_base, &stack_size));
1600 ASSERT_EQ(0, pthread_attr_getstacksize(&attributes, &stack_size2));
1601
1602 EXPECT_EQ(stack_size, stack_size2);
1603 ASSERT_EQ(6666U, stack_size);
Yabin Cuib0c6f2db2015-05-19 15:09:23 -07001604#endif
Elliott Hughes57b7a612014-08-25 17:26:50 -07001605}
Elliott Hughes8fb639c2014-09-12 14:43:07 -07001606
Mor-sarid, Nitzan56933322015-09-11 05:31:36 +00001607struct GetStackSignalHandlerArg {
1608 volatile bool done;
Chih-Hung Hsieh9af13d22016-06-02 14:40:09 -07001609 void* signal_stack_base;
1610 size_t signal_stack_size;
Mor-sarid, Nitzan56933322015-09-11 05:31:36 +00001611 void* main_stack_base;
1612 size_t main_stack_size;
1613};
1614
1615static GetStackSignalHandlerArg getstack_signal_handler_arg;
1616
1617static void getstack_signal_handler(int sig) {
1618 ASSERT_EQ(SIGUSR1, sig);
1619 // Use sleep() to make current thread be switched out by the kernel to provoke the error.
1620 sleep(1);
1621 pthread_attr_t attr;
1622 ASSERT_EQ(0, pthread_getattr_np(pthread_self(), &attr));
1623 void* stack_base;
1624 size_t stack_size;
1625 ASSERT_EQ(0, pthread_attr_getstack(&attr, &stack_base, &stack_size));
Chih-Hung Hsieh9af13d22016-06-02 14:40:09 -07001626
1627 // Verify if the stack used by the signal handler is the alternate stack just registered.
1628 ASSERT_LE(getstack_signal_handler_arg.signal_stack_base, &attr);
Evgenii Stepanov7cc67062019-02-05 18:43:34 -08001629 ASSERT_LT(static_cast<void*>(untag_address(&attr)),
Chih-Hung Hsieh9af13d22016-06-02 14:40:09 -07001630 static_cast<char*>(getstack_signal_handler_arg.signal_stack_base) +
Evgenii Stepanov7cc67062019-02-05 18:43:34 -08001631 getstack_signal_handler_arg.signal_stack_size);
Chih-Hung Hsieh9af13d22016-06-02 14:40:09 -07001632
1633 // Verify if the main thread's stack got in the signal handler is correct.
1634 ASSERT_EQ(getstack_signal_handler_arg.main_stack_base, stack_base);
1635 ASSERT_LE(getstack_signal_handler_arg.main_stack_size, stack_size);
1636
Mor-sarid, Nitzan56933322015-09-11 05:31:36 +00001637 getstack_signal_handler_arg.done = true;
1638}
1639
1640// The previous code obtained the main thread's stack by reading the entry in
1641// /proc/self/task/<pid>/maps that was labeled [stack]. Unfortunately, on x86/x86_64, the kernel
1642// relies on sp0 in task state segment(tss) to label the stack map with [stack]. If the kernel
1643// switches a process while the main thread is in an alternate stack, then the kernel will label
1644// the wrong map with [stack]. This test verifies that when the above situation happens, the main
1645// thread's stack is found correctly.
1646TEST(pthread, pthread_attr_getstack_in_signal_handler) {
Yabin Cui61e4d462016-03-07 17:44:58 -08001647 // This test is only meaningful for the main thread, so make sure we're running on it!
1648 ASSERT_EQ(getpid(), syscall(__NR_gettid));
1649
Mor-sarid, Nitzan56933322015-09-11 05:31:36 +00001650 const size_t sig_stack_size = 16 * 1024;
Yi Kong32bc0fc2018-08-02 17:31:13 -07001651 void* sig_stack = mmap(nullptr, sig_stack_size, PROT_READ | PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS,
Mor-sarid, Nitzan56933322015-09-11 05:31:36 +00001652 -1, 0);
1653 ASSERT_NE(MAP_FAILED, sig_stack);
1654 stack_t ss;
1655 ss.ss_sp = sig_stack;
1656 ss.ss_size = sig_stack_size;
1657 ss.ss_flags = 0;
1658 stack_t oss;
1659 ASSERT_EQ(0, sigaltstack(&ss, &oss));
1660
Yabin Cui61e4d462016-03-07 17:44:58 -08001661 pthread_attr_t attr;
1662 ASSERT_EQ(0, pthread_getattr_np(pthread_self(), &attr));
1663 void* main_stack_base;
1664 size_t main_stack_size;
1665 ASSERT_EQ(0, pthread_attr_getstack(&attr, &main_stack_base, &main_stack_size));
1666
Mor-sarid, Nitzan56933322015-09-11 05:31:36 +00001667 ScopedSignalHandler handler(SIGUSR1, getstack_signal_handler, SA_ONSTACK);
1668 getstack_signal_handler_arg.done = false;
Chih-Hung Hsieh9af13d22016-06-02 14:40:09 -07001669 getstack_signal_handler_arg.signal_stack_base = sig_stack;
1670 getstack_signal_handler_arg.signal_stack_size = sig_stack_size;
1671 getstack_signal_handler_arg.main_stack_base = main_stack_base;
1672 getstack_signal_handler_arg.main_stack_size = main_stack_size;
Mor-sarid, Nitzan56933322015-09-11 05:31:36 +00001673 kill(getpid(), SIGUSR1);
1674 ASSERT_EQ(true, getstack_signal_handler_arg.done);
1675
Mor-sarid, Nitzan56933322015-09-11 05:31:36 +00001676 ASSERT_EQ(0, sigaltstack(&oss, nullptr));
1677 ASSERT_EQ(0, munmap(sig_stack, sig_stack_size));
1678}
1679
Yabin Cui917d3902015-01-08 12:32:42 -08001680static void pthread_attr_getstack_18908062_helper(void*) {
1681 char local_variable;
1682 pthread_attr_t attributes;
1683 pthread_getattr_np(pthread_self(), &attributes);
1684 void* stack_base;
1685 size_t stack_size;
1686 pthread_attr_getstack(&attributes, &stack_base, &stack_size);
1687
1688 // Test whether &local_variable is in [stack_base, stack_base + stack_size).
1689 ASSERT_LE(reinterpret_cast<char*>(stack_base), &local_variable);
Evgenii Stepanov7cc67062019-02-05 18:43:34 -08001690 ASSERT_LT(untag_address(&local_variable), reinterpret_cast<char*>(stack_base) + stack_size);
Yabin Cui917d3902015-01-08 12:32:42 -08001691}
1692
1693// Check whether something on stack is in the range of
1694// [stack_base, stack_base + stack_size). see b/18908062.
1695TEST(pthread, pthread_attr_getstack_18908062) {
1696 pthread_t t;
Yi Kong32bc0fc2018-08-02 17:31:13 -07001697 ASSERT_EQ(0, pthread_create(&t, nullptr,
Yabin Cui917d3902015-01-08 12:32:42 -08001698 reinterpret_cast<void* (*)(void*)>(pthread_attr_getstack_18908062_helper),
Yi Kong32bc0fc2018-08-02 17:31:13 -07001699 nullptr));
1700 ASSERT_EQ(0, pthread_join(t, nullptr));
Yabin Cui917d3902015-01-08 12:32:42 -08001701}
1702
Elliott Hughes8fb639c2014-09-12 14:43:07 -07001703#if defined(__BIONIC__)
Elliott Hughesf2083612015-11-11 13:32:28 -08001704static pthread_mutex_t pthread_gettid_np_mutex = PTHREAD_MUTEX_INITIALIZER;
1705
Elliott Hughes8fb639c2014-09-12 14:43:07 -07001706static void* pthread_gettid_np_helper(void* arg) {
1707 *reinterpret_cast<pid_t*>(arg) = gettid();
Elliott Hughesf2083612015-11-11 13:32:28 -08001708
1709 // Wait for our parent to call pthread_gettid_np on us before exiting.
1710 pthread_mutex_lock(&pthread_gettid_np_mutex);
1711 pthread_mutex_unlock(&pthread_gettid_np_mutex);
Yi Kong32bc0fc2018-08-02 17:31:13 -07001712 return nullptr;
Elliott Hughes8fb639c2014-09-12 14:43:07 -07001713}
1714#endif
1715
1716TEST(pthread, pthread_gettid_np) {
1717#if defined(__BIONIC__)
1718 ASSERT_EQ(gettid(), pthread_gettid_np(pthread_self()));
1719
Elliott Hughesf2083612015-11-11 13:32:28 -08001720 // Ensure the other thread doesn't exit until after we've called
1721 // pthread_gettid_np on it.
1722 pthread_mutex_lock(&pthread_gettid_np_mutex);
1723
Elliott Hughes8fb639c2014-09-12 14:43:07 -07001724 pid_t t_gettid_result;
1725 pthread_t t;
Yi Kong32bc0fc2018-08-02 17:31:13 -07001726 pthread_create(&t, nullptr, pthread_gettid_np_helper, &t_gettid_result);
Elliott Hughes8fb639c2014-09-12 14:43:07 -07001727
1728 pid_t t_pthread_gettid_np_result = pthread_gettid_np(t);
1729
Elliott Hughesf2083612015-11-11 13:32:28 -08001730 // Release the other thread and wait for it to exit.
1731 pthread_mutex_unlock(&pthread_gettid_np_mutex);
Yi Kong32bc0fc2018-08-02 17:31:13 -07001732 ASSERT_EQ(0, pthread_join(t, nullptr));
Elliott Hughes8fb639c2014-09-12 14:43:07 -07001733
1734 ASSERT_EQ(t_gettid_result, t_pthread_gettid_np_result);
1735#else
Elliott Hughesbcaa4542019-03-08 15:20:23 -08001736 GTEST_SKIP() << "pthread_gettid_np not available";
Elliott Hughes8fb639c2014-09-12 14:43:07 -07001737#endif
1738}
Elliott Hughes34c987a2014-09-22 16:01:26 -07001739
1740static size_t cleanup_counter = 0;
1741
Derek Xue41996952014-09-25 11:05:32 +01001742static void AbortCleanupRoutine(void*) {
Elliott Hughes34c987a2014-09-22 16:01:26 -07001743 abort();
1744}
1745
Derek Xue41996952014-09-25 11:05:32 +01001746static void CountCleanupRoutine(void*) {
Elliott Hughes34c987a2014-09-22 16:01:26 -07001747 ++cleanup_counter;
1748}
1749
Derek Xue41996952014-09-25 11:05:32 +01001750static void PthreadCleanupTester() {
Yi Kong32bc0fc2018-08-02 17:31:13 -07001751 pthread_cleanup_push(CountCleanupRoutine, nullptr);
1752 pthread_cleanup_push(CountCleanupRoutine, nullptr);
1753 pthread_cleanup_push(AbortCleanupRoutine, nullptr);
Elliott Hughes34c987a2014-09-22 16:01:26 -07001754
1755 pthread_cleanup_pop(0); // Pop the abort without executing it.
1756 pthread_cleanup_pop(1); // Pop one count while executing it.
1757 ASSERT_EQ(1U, cleanup_counter);
1758 // Exit while the other count is still on the cleanup stack.
Yi Kong32bc0fc2018-08-02 17:31:13 -07001759 pthread_exit(nullptr);
Elliott Hughes34c987a2014-09-22 16:01:26 -07001760
1761 // Calls to pthread_cleanup_pop/pthread_cleanup_push must always be balanced.
1762 pthread_cleanup_pop(0);
1763}
1764
Derek Xue41996952014-09-25 11:05:32 +01001765static void* PthreadCleanupStartRoutine(void*) {
Elliott Hughes34c987a2014-09-22 16:01:26 -07001766 PthreadCleanupTester();
Yi Kong32bc0fc2018-08-02 17:31:13 -07001767 return nullptr;
Elliott Hughes34c987a2014-09-22 16:01:26 -07001768}
1769
1770TEST(pthread, pthread_cleanup_push__pthread_cleanup_pop) {
1771 pthread_t t;
Yi Kong32bc0fc2018-08-02 17:31:13 -07001772 ASSERT_EQ(0, pthread_create(&t, nullptr, PthreadCleanupStartRoutine, nullptr));
1773 ASSERT_EQ(0, pthread_join(t, nullptr));
Elliott Hughes34c987a2014-09-22 16:01:26 -07001774 ASSERT_EQ(2U, cleanup_counter);
1775}
Derek Xue41996952014-09-25 11:05:32 +01001776
1777TEST(pthread, PTHREAD_MUTEX_DEFAULT_is_PTHREAD_MUTEX_NORMAL) {
1778 ASSERT_EQ(PTHREAD_MUTEX_NORMAL, PTHREAD_MUTEX_DEFAULT);
1779}
1780
1781TEST(pthread, pthread_mutexattr_gettype) {
1782 pthread_mutexattr_t attr;
1783 ASSERT_EQ(0, pthread_mutexattr_init(&attr));
1784
1785 int attr_type;
1786
1787 ASSERT_EQ(0, pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_NORMAL));
1788 ASSERT_EQ(0, pthread_mutexattr_gettype(&attr, &attr_type));
1789 ASSERT_EQ(PTHREAD_MUTEX_NORMAL, attr_type);
1790
1791 ASSERT_EQ(0, pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_ERRORCHECK));
1792 ASSERT_EQ(0, pthread_mutexattr_gettype(&attr, &attr_type));
1793 ASSERT_EQ(PTHREAD_MUTEX_ERRORCHECK, attr_type);
1794
1795 ASSERT_EQ(0, pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE));
1796 ASSERT_EQ(0, pthread_mutexattr_gettype(&attr, &attr_type));
1797 ASSERT_EQ(PTHREAD_MUTEX_RECURSIVE, attr_type);
Yabin Cui5b8e7cd2015-03-04 17:36:59 -08001798
1799 ASSERT_EQ(0, pthread_mutexattr_destroy(&attr));
1800}
1801
Yabin Cui6b9c85b2018-01-23 12:56:18 -08001802TEST(pthread, pthread_mutexattr_protocol) {
1803 pthread_mutexattr_t attr;
1804 ASSERT_EQ(0, pthread_mutexattr_init(&attr));
1805
1806 int protocol;
1807 ASSERT_EQ(0, pthread_mutexattr_getprotocol(&attr, &protocol));
1808 ASSERT_EQ(PTHREAD_PRIO_NONE, protocol);
1809 for (size_t repeat = 0; repeat < 2; ++repeat) {
1810 for (int set_protocol : {PTHREAD_PRIO_NONE, PTHREAD_PRIO_INHERIT}) {
1811 ASSERT_EQ(0, pthread_mutexattr_setprotocol(&attr, set_protocol));
1812 ASSERT_EQ(0, pthread_mutexattr_getprotocol(&attr, &protocol));
1813 ASSERT_EQ(protocol, set_protocol);
1814 }
1815 }
1816}
1817
Yabin Cui17393b02015-03-21 15:08:25 -07001818struct PthreadMutex {
1819 pthread_mutex_t lock;
1820
Yabin Cui6b9c85b2018-01-23 12:56:18 -08001821 explicit PthreadMutex(int mutex_type, int protocol = PTHREAD_PRIO_NONE) {
1822 init(mutex_type, protocol);
Yabin Cui17393b02015-03-21 15:08:25 -07001823 }
1824
1825 ~PthreadMutex() {
1826 destroy();
1827 }
1828
1829 private:
Yabin Cui6b9c85b2018-01-23 12:56:18 -08001830 void init(int mutex_type, int protocol) {
Yabin Cui17393b02015-03-21 15:08:25 -07001831 pthread_mutexattr_t attr;
1832 ASSERT_EQ(0, pthread_mutexattr_init(&attr));
1833 ASSERT_EQ(0, pthread_mutexattr_settype(&attr, mutex_type));
Yabin Cui6b9c85b2018-01-23 12:56:18 -08001834 ASSERT_EQ(0, pthread_mutexattr_setprotocol(&attr, protocol));
Yabin Cui17393b02015-03-21 15:08:25 -07001835 ASSERT_EQ(0, pthread_mutex_init(&lock, &attr));
1836 ASSERT_EQ(0, pthread_mutexattr_destroy(&attr));
1837 }
1838
1839 void destroy() {
1840 ASSERT_EQ(0, pthread_mutex_destroy(&lock));
1841 }
1842
1843 DISALLOW_COPY_AND_ASSIGN(PthreadMutex);
1844};
Derek Xue41996952014-09-25 11:05:32 +01001845
Ryan Prichard4b6c0f52019-04-18 22:47:04 -07001846static int UnlockFromAnotherThread(pthread_mutex_t* mutex) {
1847 pthread_t thread;
1848 pthread_create(&thread, nullptr, [](void* mutex_voidp) -> void* {
1849 pthread_mutex_t* mutex = static_cast<pthread_mutex_t*>(mutex_voidp);
1850 intptr_t result = pthread_mutex_unlock(mutex);
1851 return reinterpret_cast<void*>(result);
1852 }, mutex);
1853 void* result;
1854 EXPECT_EQ(0, pthread_join(thread, &result));
1855 return reinterpret_cast<intptr_t>(result);
1856};
1857
Yabin Cui6b9c85b2018-01-23 12:56:18 -08001858static void TestPthreadMutexLockNormal(int protocol) {
1859 PthreadMutex m(PTHREAD_MUTEX_NORMAL, protocol);
Derek Xue41996952014-09-25 11:05:32 +01001860
Yabin Cui17393b02015-03-21 15:08:25 -07001861 ASSERT_EQ(0, pthread_mutex_lock(&m.lock));
Ryan Prichard4b6c0f52019-04-18 22:47:04 -07001862 if (protocol == PTHREAD_PRIO_INHERIT) {
1863 ASSERT_EQ(EPERM, UnlockFromAnotherThread(&m.lock));
1864 }
Yabin Cui17393b02015-03-21 15:08:25 -07001865 ASSERT_EQ(0, pthread_mutex_unlock(&m.lock));
Elliott Hughesd31d4c12015-12-14 17:35:10 -08001866 ASSERT_EQ(0, pthread_mutex_trylock(&m.lock));
1867 ASSERT_EQ(EBUSY, pthread_mutex_trylock(&m.lock));
1868 ASSERT_EQ(0, pthread_mutex_unlock(&m.lock));
Derek Xue41996952014-09-25 11:05:32 +01001869}
1870
Yabin Cui6b9c85b2018-01-23 12:56:18 -08001871static void TestPthreadMutexLockErrorCheck(int protocol) {
1872 PthreadMutex m(PTHREAD_MUTEX_ERRORCHECK, protocol);
Derek Xue41996952014-09-25 11:05:32 +01001873
Yabin Cui17393b02015-03-21 15:08:25 -07001874 ASSERT_EQ(0, pthread_mutex_lock(&m.lock));
Ryan Prichard4b6c0f52019-04-18 22:47:04 -07001875 ASSERT_EQ(EPERM, UnlockFromAnotherThread(&m.lock));
Yabin Cui17393b02015-03-21 15:08:25 -07001876 ASSERT_EQ(EDEADLK, pthread_mutex_lock(&m.lock));
1877 ASSERT_EQ(0, pthread_mutex_unlock(&m.lock));
1878 ASSERT_EQ(0, pthread_mutex_trylock(&m.lock));
Yabin Cui6b9c85b2018-01-23 12:56:18 -08001879 if (protocol == PTHREAD_PRIO_NONE) {
1880 ASSERT_EQ(EBUSY, pthread_mutex_trylock(&m.lock));
1881 } else {
1882 ASSERT_EQ(EDEADLK, pthread_mutex_trylock(&m.lock));
1883 }
Yabin Cui17393b02015-03-21 15:08:25 -07001884 ASSERT_EQ(0, pthread_mutex_unlock(&m.lock));
1885 ASSERT_EQ(EPERM, pthread_mutex_unlock(&m.lock));
Derek Xue41996952014-09-25 11:05:32 +01001886}
1887
Yabin Cui6b9c85b2018-01-23 12:56:18 -08001888static void TestPthreadMutexLockRecursive(int protocol) {
1889 PthreadMutex m(PTHREAD_MUTEX_RECURSIVE, protocol);
Derek Xue41996952014-09-25 11:05:32 +01001890
Yabin Cui17393b02015-03-21 15:08:25 -07001891 ASSERT_EQ(0, pthread_mutex_lock(&m.lock));
Ryan Prichard4b6c0f52019-04-18 22:47:04 -07001892 ASSERT_EQ(EPERM, UnlockFromAnotherThread(&m.lock));
Yabin Cui17393b02015-03-21 15:08:25 -07001893 ASSERT_EQ(0, pthread_mutex_lock(&m.lock));
Ryan Prichard4b6c0f52019-04-18 22:47:04 -07001894 ASSERT_EQ(EPERM, UnlockFromAnotherThread(&m.lock));
Yabin Cui17393b02015-03-21 15:08:25 -07001895 ASSERT_EQ(0, pthread_mutex_unlock(&m.lock));
1896 ASSERT_EQ(0, pthread_mutex_unlock(&m.lock));
1897 ASSERT_EQ(0, pthread_mutex_trylock(&m.lock));
Elliott Hughesd31d4c12015-12-14 17:35:10 -08001898 ASSERT_EQ(0, pthread_mutex_trylock(&m.lock));
1899 ASSERT_EQ(0, pthread_mutex_unlock(&m.lock));
Yabin Cui17393b02015-03-21 15:08:25 -07001900 ASSERT_EQ(0, pthread_mutex_unlock(&m.lock));
1901 ASSERT_EQ(EPERM, pthread_mutex_unlock(&m.lock));
1902}
1903
Yabin Cui6b9c85b2018-01-23 12:56:18 -08001904TEST(pthread, pthread_mutex_lock_NORMAL) {
1905 TestPthreadMutexLockNormal(PTHREAD_PRIO_NONE);
1906}
1907
1908TEST(pthread, pthread_mutex_lock_ERRORCHECK) {
1909 TestPthreadMutexLockErrorCheck(PTHREAD_PRIO_NONE);
1910}
1911
1912TEST(pthread, pthread_mutex_lock_RECURSIVE) {
1913 TestPthreadMutexLockRecursive(PTHREAD_PRIO_NONE);
1914}
1915
1916TEST(pthread, pthread_mutex_lock_pi) {
Yabin Cui6b9c85b2018-01-23 12:56:18 -08001917 TestPthreadMutexLockNormal(PTHREAD_PRIO_INHERIT);
1918 TestPthreadMutexLockErrorCheck(PTHREAD_PRIO_INHERIT);
1919 TestPthreadMutexLockRecursive(PTHREAD_PRIO_INHERIT);
1920}
1921
Yabin Cui5a00ba72018-01-26 17:32:31 -08001922TEST(pthread, pthread_mutex_pi_count_limit) {
1923#if defined(__BIONIC__) && !defined(__LP64__)
1924 // Bionic only supports 65536 pi mutexes in 32-bit programs.
1925 pthread_mutexattr_t attr;
1926 ASSERT_EQ(0, pthread_mutexattr_init(&attr));
1927 ASSERT_EQ(0, pthread_mutexattr_setprotocol(&attr, PTHREAD_PRIO_INHERIT));
1928 std::vector<pthread_mutex_t> mutexes(65536);
1929 // Test if we can use 65536 pi mutexes at the same time.
1930 // Run 2 times to check if freed pi mutexes can be recycled.
1931 for (int repeat = 0; repeat < 2; ++repeat) {
1932 for (auto& m : mutexes) {
1933 ASSERT_EQ(0, pthread_mutex_init(&m, &attr));
1934 }
1935 pthread_mutex_t m;
1936 ASSERT_EQ(ENOMEM, pthread_mutex_init(&m, &attr));
1937 for (auto& m : mutexes) {
1938 ASSERT_EQ(0, pthread_mutex_lock(&m));
1939 }
1940 for (auto& m : mutexes) {
1941 ASSERT_EQ(0, pthread_mutex_unlock(&m));
1942 }
1943 for (auto& m : mutexes) {
1944 ASSERT_EQ(0, pthread_mutex_destroy(&m));
1945 }
1946 }
1947 ASSERT_EQ(0, pthread_mutexattr_destroy(&attr));
1948#else
Elliott Hughesbcaa4542019-03-08 15:20:23 -08001949 GTEST_SKIP() << "pi mutex count not limited to 64Ki";
Yabin Cui5a00ba72018-01-26 17:32:31 -08001950#endif
1951}
1952
Yabin Cui17393b02015-03-21 15:08:25 -07001953TEST(pthread, pthread_mutex_init_same_as_static_initializers) {
1954 pthread_mutex_t lock_normal = PTHREAD_MUTEX_INITIALIZER;
1955 PthreadMutex m1(PTHREAD_MUTEX_NORMAL);
1956 ASSERT_EQ(0, memcmp(&lock_normal, &m1.lock, sizeof(pthread_mutex_t)));
1957 pthread_mutex_destroy(&lock_normal);
1958
1959 pthread_mutex_t lock_errorcheck = PTHREAD_ERRORCHECK_MUTEX_INITIALIZER_NP;
1960 PthreadMutex m2(PTHREAD_MUTEX_ERRORCHECK);
1961 ASSERT_EQ(0, memcmp(&lock_errorcheck, &m2.lock, sizeof(pthread_mutex_t)));
1962 pthread_mutex_destroy(&lock_errorcheck);
1963
1964 pthread_mutex_t lock_recursive = PTHREAD_RECURSIVE_MUTEX_INITIALIZER_NP;
1965 PthreadMutex m3(PTHREAD_MUTEX_RECURSIVE);
1966 ASSERT_EQ(0, memcmp(&lock_recursive, &m3.lock, sizeof(pthread_mutex_t)));
1967 ASSERT_EQ(0, pthread_mutex_destroy(&lock_recursive));
Derek Xue41996952014-09-25 11:05:32 +01001968}
Yabin Cui5a00ba72018-01-26 17:32:31 -08001969
Yabin Cui5b8e7cd2015-03-04 17:36:59 -08001970class MutexWakeupHelper {
1971 private:
Yabin Cui17393b02015-03-21 15:08:25 -07001972 PthreadMutex m;
Yabin Cui5b8e7cd2015-03-04 17:36:59 -08001973 enum Progress {
1974 LOCK_INITIALIZED,
1975 LOCK_WAITING,
1976 LOCK_RELEASED,
1977 LOCK_ACCESSED
1978 };
1979 std::atomic<Progress> progress;
Yabin Cuif7969852015-04-02 17:47:48 -07001980 std::atomic<pid_t> tid;
Yabin Cui5b8e7cd2015-03-04 17:36:59 -08001981
1982 static void thread_fn(MutexWakeupHelper* helper) {
Yabin Cuif7969852015-04-02 17:47:48 -07001983 helper->tid = gettid();
Yabin Cui5b8e7cd2015-03-04 17:36:59 -08001984 ASSERT_EQ(LOCK_INITIALIZED, helper->progress);
1985 helper->progress = LOCK_WAITING;
1986
Yabin Cui17393b02015-03-21 15:08:25 -07001987 ASSERT_EQ(0, pthread_mutex_lock(&helper->m.lock));
Yabin Cui5b8e7cd2015-03-04 17:36:59 -08001988 ASSERT_EQ(LOCK_RELEASED, helper->progress);
Yabin Cui17393b02015-03-21 15:08:25 -07001989 ASSERT_EQ(0, pthread_mutex_unlock(&helper->m.lock));
Yabin Cui5b8e7cd2015-03-04 17:36:59 -08001990
1991 helper->progress = LOCK_ACCESSED;
1992 }
1993
1994 public:
Chih-Hung Hsieh62e3a072016-05-03 12:08:05 -07001995 explicit MutexWakeupHelper(int mutex_type) : m(mutex_type) {
Yabin Cui17393b02015-03-21 15:08:25 -07001996 }
1997
1998 void test() {
1999 ASSERT_EQ(0, pthread_mutex_lock(&m.lock));
Yabin Cui5b8e7cd2015-03-04 17:36:59 -08002000 progress = LOCK_INITIALIZED;
Yabin Cuif7969852015-04-02 17:47:48 -07002001 tid = 0;
Yabin Cui5b8e7cd2015-03-04 17:36:59 -08002002
2003 pthread_t thread;
Yi Kong32bc0fc2018-08-02 17:31:13 -07002004 ASSERT_EQ(0, pthread_create(&thread, nullptr,
Yabin Cui5b8e7cd2015-03-04 17:36:59 -08002005 reinterpret_cast<void* (*)(void*)>(MutexWakeupHelper::thread_fn), this));
2006
Yabin Cuif7969852015-04-02 17:47:48 -07002007 WaitUntilThreadSleep(tid);
2008 ASSERT_EQ(LOCK_WAITING, progress);
2009
Yabin Cui5b8e7cd2015-03-04 17:36:59 -08002010 progress = LOCK_RELEASED;
Yabin Cui17393b02015-03-21 15:08:25 -07002011 ASSERT_EQ(0, pthread_mutex_unlock(&m.lock));
Yabin Cui5b8e7cd2015-03-04 17:36:59 -08002012
Yi Kong32bc0fc2018-08-02 17:31:13 -07002013 ASSERT_EQ(0, pthread_join(thread, nullptr));
Yabin Cui5b8e7cd2015-03-04 17:36:59 -08002014 ASSERT_EQ(LOCK_ACCESSED, progress);
Yabin Cui5b8e7cd2015-03-04 17:36:59 -08002015 }
2016};
2017
2018TEST(pthread, pthread_mutex_NORMAL_wakeup) {
Yabin Cui17393b02015-03-21 15:08:25 -07002019 MutexWakeupHelper helper(PTHREAD_MUTEX_NORMAL);
2020 helper.test();
Yabin Cui5b8e7cd2015-03-04 17:36:59 -08002021}
2022
2023TEST(pthread, pthread_mutex_ERRORCHECK_wakeup) {
Yabin Cui17393b02015-03-21 15:08:25 -07002024 MutexWakeupHelper helper(PTHREAD_MUTEX_ERRORCHECK);
2025 helper.test();
Yabin Cui5b8e7cd2015-03-04 17:36:59 -08002026}
2027
2028TEST(pthread, pthread_mutex_RECURSIVE_wakeup) {
Yabin Cui17393b02015-03-21 15:08:25 -07002029 MutexWakeupHelper helper(PTHREAD_MUTEX_RECURSIVE);
2030 helper.test();
Yabin Cui5b8e7cd2015-03-04 17:36:59 -08002031}
2032
Yabin Cui6b9c85b2018-01-23 12:56:18 -08002033static int GetThreadPriority(pid_t tid) {
2034 // sched_getparam() returns the static priority of a thread, which can't reflect a thread's
2035 // priority after priority inheritance. So read /proc/<pid>/stat to get the dynamic priority.
2036 std::string filename = android::base::StringPrintf("/proc/%d/stat", tid);
2037 std::string content;
2038 int result = INT_MAX;
2039 if (!android::base::ReadFileToString(filename, &content)) {
2040 return result;
2041 }
2042 std::vector<std::string> strs = android::base::Split(content, " ");
2043 if (strs.size() < 18) {
2044 return result;
2045 }
2046 if (!android::base::ParseInt(strs[17], &result)) {
2047 return INT_MAX;
2048 }
2049 return result;
2050}
2051
2052class PIMutexWakeupHelper {
2053private:
2054 PthreadMutex m;
2055 int protocol;
2056 enum Progress {
2057 LOCK_INITIALIZED,
2058 LOCK_CHILD_READY,
2059 LOCK_WAITING,
2060 LOCK_RELEASED,
2061 };
2062 std::atomic<Progress> progress;
2063 std::atomic<pid_t> main_tid;
2064 std::atomic<pid_t> child_tid;
2065 PthreadMutex start_thread_m;
2066
2067 static void thread_fn(PIMutexWakeupHelper* helper) {
2068 helper->child_tid = gettid();
2069 ASSERT_EQ(LOCK_INITIALIZED, helper->progress);
2070 ASSERT_EQ(0, setpriority(PRIO_PROCESS, gettid(), 1));
2071 ASSERT_EQ(21, GetThreadPriority(gettid()));
2072 ASSERT_EQ(0, pthread_mutex_lock(&helper->m.lock));
2073 helper->progress = LOCK_CHILD_READY;
2074 ASSERT_EQ(0, pthread_mutex_lock(&helper->start_thread_m.lock));
2075
2076 ASSERT_EQ(0, pthread_mutex_unlock(&helper->start_thread_m.lock));
2077 WaitUntilThreadSleep(helper->main_tid);
2078 ASSERT_EQ(LOCK_WAITING, helper->progress);
2079
2080 if (helper->protocol == PTHREAD_PRIO_INHERIT) {
2081 ASSERT_EQ(20, GetThreadPriority(gettid()));
2082 } else {
2083 ASSERT_EQ(21, GetThreadPriority(gettid()));
2084 }
2085 helper->progress = LOCK_RELEASED;
2086 ASSERT_EQ(0, pthread_mutex_unlock(&helper->m.lock));
2087 }
2088
2089public:
2090 explicit PIMutexWakeupHelper(int mutex_type, int protocol)
2091 : m(mutex_type, protocol), protocol(protocol), start_thread_m(PTHREAD_MUTEX_NORMAL) {
2092 }
2093
2094 void test() {
2095 ASSERT_EQ(0, pthread_mutex_lock(&start_thread_m.lock));
2096 main_tid = gettid();
2097 ASSERT_EQ(20, GetThreadPriority(main_tid));
2098 progress = LOCK_INITIALIZED;
2099 child_tid = 0;
2100
2101 pthread_t thread;
Yi Kong32bc0fc2018-08-02 17:31:13 -07002102 ASSERT_EQ(0, pthread_create(&thread, nullptr,
Yabin Cui6b9c85b2018-01-23 12:56:18 -08002103 reinterpret_cast<void* (*)(void*)>(PIMutexWakeupHelper::thread_fn), this));
2104
2105 WaitUntilThreadSleep(child_tid);
2106 ASSERT_EQ(LOCK_CHILD_READY, progress);
2107 ASSERT_EQ(0, pthread_mutex_unlock(&start_thread_m.lock));
2108 progress = LOCK_WAITING;
2109 ASSERT_EQ(0, pthread_mutex_lock(&m.lock));
2110
2111 ASSERT_EQ(LOCK_RELEASED, progress);
2112 ASSERT_EQ(0, pthread_mutex_unlock(&m.lock));
2113 ASSERT_EQ(0, pthread_join(thread, nullptr));
2114 }
2115};
2116
2117TEST(pthread, pthread_mutex_pi_wakeup) {
Yabin Cui6b9c85b2018-01-23 12:56:18 -08002118 for (int type : {PTHREAD_MUTEX_NORMAL, PTHREAD_MUTEX_RECURSIVE, PTHREAD_MUTEX_ERRORCHECK}) {
2119 for (int protocol : {PTHREAD_PRIO_INHERIT}) {
2120 PIMutexWakeupHelper helper(type, protocol);
2121 helper.test();
2122 }
2123 }
2124}
2125
Yabin Cui140f3672015-02-03 10:32:00 -08002126TEST(pthread, pthread_mutex_owner_tid_limit) {
Yabin Cuie69c2452015-02-13 16:21:25 -08002127#if defined(__BIONIC__) && !defined(__LP64__)
Yabin Cui140f3672015-02-03 10:32:00 -08002128 FILE* fp = fopen("/proc/sys/kernel/pid_max", "r");
Yi Kong32bc0fc2018-08-02 17:31:13 -07002129 ASSERT_TRUE(fp != nullptr);
Yabin Cui140f3672015-02-03 10:32:00 -08002130 long pid_max;
2131 ASSERT_EQ(1, fscanf(fp, "%ld", &pid_max));
2132 fclose(fp);
Yabin Cuie69c2452015-02-13 16:21:25 -08002133 // Bionic's pthread_mutex implementation on 32-bit devices uses 16 bits to represent owner tid.
Yabin Cui140f3672015-02-03 10:32:00 -08002134 ASSERT_LE(pid_max, 65536);
Yabin Cuie69c2452015-02-13 16:21:25 -08002135#else
Elliott Hughesbcaa4542019-03-08 15:20:23 -08002136 GTEST_SKIP() << "pthread_mutex supports 32-bit tid";
Yabin Cuie69c2452015-02-13 16:21:25 -08002137#endif
Yabin Cui140f3672015-02-03 10:32:00 -08002138}
Yabin Cuib5845722015-03-16 22:46:42 -07002139
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08002140static void pthread_mutex_timedlock_helper(clockid_t clock,
2141 int (*lock_function)(pthread_mutex_t* __mutex,
2142 const timespec* __timeout)) {
Yabin Cuic9a659c2015-11-05 15:36:08 -08002143 pthread_mutex_t m;
2144 ASSERT_EQ(0, pthread_mutex_init(&m, nullptr));
2145
2146 // If the mutex is already locked, pthread_mutex_timedlock should time out.
2147 ASSERT_EQ(0, pthread_mutex_lock(&m));
2148
2149 timespec ts;
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08002150 ASSERT_EQ(0, clock_gettime(clock, &ts));
2151 ASSERT_EQ(ETIMEDOUT, lock_function(&m, &ts));
Yabin Cuic9a659c2015-11-05 15:36:08 -08002152 ts.tv_nsec = -1;
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08002153 ASSERT_EQ(EINVAL, lock_function(&m, &ts));
Yabin Cuic9a659c2015-11-05 15:36:08 -08002154 ts.tv_nsec = NS_PER_S;
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08002155 ASSERT_EQ(EINVAL, lock_function(&m, &ts));
Yabin Cuic9a659c2015-11-05 15:36:08 -08002156 ts.tv_nsec = NS_PER_S - 1;
2157 ts.tv_sec = -1;
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08002158 ASSERT_EQ(ETIMEDOUT, lock_function(&m, &ts));
Yabin Cuic9a659c2015-11-05 15:36:08 -08002159
2160 // If the mutex is unlocked, pthread_mutex_timedlock should succeed.
2161 ASSERT_EQ(0, pthread_mutex_unlock(&m));
2162
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08002163 ASSERT_EQ(0, clock_gettime(clock, &ts));
Yabin Cuic9a659c2015-11-05 15:36:08 -08002164 ts.tv_sec += 1;
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08002165 ASSERT_EQ(0, lock_function(&m, &ts));
Yabin Cuic9a659c2015-11-05 15:36:08 -08002166
2167 ASSERT_EQ(0, pthread_mutex_unlock(&m));
2168 ASSERT_EQ(0, pthread_mutex_destroy(&m));
2169}
2170
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08002171TEST(pthread, pthread_mutex_timedlock) {
2172 pthread_mutex_timedlock_helper(CLOCK_REALTIME, pthread_mutex_timedlock);
2173}
2174
2175TEST(pthread, pthread_mutex_timedlock_monotonic_np) {
2176#if defined(__BIONIC__)
2177 pthread_mutex_timedlock_helper(CLOCK_MONOTONIC, pthread_mutex_timedlock_monotonic_np);
2178#else // __BIONIC__
Elliott Hughesbcaa4542019-03-08 15:20:23 -08002179 GTEST_SKIP() << "pthread_mutex_timedlock_monotonic_np not available";
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08002180#endif // __BIONIC__
2181}
2182
2183static void pthread_mutex_timedlock_pi_helper(clockid_t clock,
2184 int (*lock_function)(pthread_mutex_t* __mutex,
2185 const timespec* __timeout)) {
Yabin Cui6b9c85b2018-01-23 12:56:18 -08002186 PthreadMutex m(PTHREAD_MUTEX_NORMAL, PTHREAD_PRIO_INHERIT);
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08002187
Yabin Cui6b9c85b2018-01-23 12:56:18 -08002188 timespec ts;
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08002189 clock_gettime(clock, &ts);
Yabin Cui6b9c85b2018-01-23 12:56:18 -08002190 ts.tv_sec += 1;
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08002191 ASSERT_EQ(0, lock_function(&m.lock, &ts));
2192
2193 struct ThreadArgs {
2194 clockid_t clock;
2195 int (*lock_function)(pthread_mutex_t* __mutex, const timespec* __timeout);
2196 PthreadMutex& m;
2197 };
2198
2199 ThreadArgs thread_args = {
2200 .clock = clock,
2201 .lock_function = lock_function,
2202 .m = m,
2203 };
Yabin Cui6b9c85b2018-01-23 12:56:18 -08002204
2205 auto ThreadFn = [](void* arg) -> void* {
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08002206 auto args = static_cast<ThreadArgs*>(arg);
Yabin Cui6b9c85b2018-01-23 12:56:18 -08002207 timespec ts;
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08002208 clock_gettime(args->clock, &ts);
Yabin Cui6b9c85b2018-01-23 12:56:18 -08002209 ts.tv_sec += 1;
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08002210 intptr_t result = args->lock_function(&args->m.lock, &ts);
Yabin Cui6b9c85b2018-01-23 12:56:18 -08002211 return reinterpret_cast<void*>(result);
2212 };
2213
2214 pthread_t thread;
Yi Kong32bc0fc2018-08-02 17:31:13 -07002215 ASSERT_EQ(0, pthread_create(&thread, nullptr, ThreadFn, &thread_args));
Yabin Cui6b9c85b2018-01-23 12:56:18 -08002216 void* result;
2217 ASSERT_EQ(0, pthread_join(thread, &result));
2218 ASSERT_EQ(ETIMEDOUT, reinterpret_cast<intptr_t>(result));
2219 ASSERT_EQ(0, pthread_mutex_unlock(&m.lock));
2220}
2221
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08002222TEST(pthread, pthread_mutex_timedlock_pi) {
2223 pthread_mutex_timedlock_pi_helper(CLOCK_REALTIME, pthread_mutex_timedlock);
2224}
2225
2226TEST(pthread, pthread_mutex_timedlock_monotonic_np_pi) {
2227#if defined(__BIONIC__)
2228 pthread_mutex_timedlock_pi_helper(CLOCK_MONOTONIC, pthread_mutex_timedlock_monotonic_np);
2229#else // __BIONIC__
Elliott Hughesbcaa4542019-03-08 15:20:23 -08002230 GTEST_SKIP() << "pthread_mutex_timedlock_monotonic_np not available";
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08002231#endif // __BIONIC__
2232}
2233
Yabin Cui9651fdf2018-03-14 12:02:21 -07002234TEST(pthread, pthread_mutex_using_destroyed_mutex) {
2235#if defined(__BIONIC__)
2236 pthread_mutex_t m;
2237 ASSERT_EQ(0, pthread_mutex_init(&m, nullptr));
2238 ASSERT_EQ(0, pthread_mutex_destroy(&m));
2239 ASSERT_EXIT(pthread_mutex_lock(&m), ::testing::KilledBySignal(SIGABRT),
2240 "pthread_mutex_lock called on a destroyed mutex");
2241 ASSERT_EXIT(pthread_mutex_unlock(&m), ::testing::KilledBySignal(SIGABRT),
2242 "pthread_mutex_unlock called on a destroyed mutex");
2243 ASSERT_EXIT(pthread_mutex_trylock(&m), ::testing::KilledBySignal(SIGABRT),
2244 "pthread_mutex_trylock called on a destroyed mutex");
2245 timespec ts;
2246 ASSERT_EXIT(pthread_mutex_timedlock(&m, &ts), ::testing::KilledBySignal(SIGABRT),
2247 "pthread_mutex_timedlock called on a destroyed mutex");
Tom Cherryc6b5bcd2018-03-05 14:14:44 -08002248 ASSERT_EXIT(pthread_mutex_timedlock_monotonic_np(&m, &ts), ::testing::KilledBySignal(SIGABRT),
2249 "pthread_mutex_timedlock_monotonic_np called on a destroyed mutex");
Yabin Cui9651fdf2018-03-14 12:02:21 -07002250 ASSERT_EXIT(pthread_mutex_destroy(&m), ::testing::KilledBySignal(SIGABRT),
2251 "pthread_mutex_destroy called on a destroyed mutex");
2252#else
Elliott Hughesbcaa4542019-03-08 15:20:23 -08002253 GTEST_SKIP() << "bionic-only test";
Yabin Cui9651fdf2018-03-14 12:02:21 -07002254#endif
2255}
2256
Yabin Cuib5845722015-03-16 22:46:42 -07002257class StrictAlignmentAllocator {
2258 public:
2259 void* allocate(size_t size, size_t alignment) {
2260 char* p = new char[size + alignment * 2];
2261 allocated_array.push_back(p);
2262 while (!is_strict_aligned(p, alignment)) {
2263 ++p;
2264 }
2265 return p;
2266 }
2267
2268 ~StrictAlignmentAllocator() {
Elliott Hughes0b2acdf2015-10-02 18:25:19 -07002269 for (const auto& p : allocated_array) {
2270 delete[] p;
Yabin Cuib5845722015-03-16 22:46:42 -07002271 }
2272 }
2273
2274 private:
2275 bool is_strict_aligned(char* p, size_t alignment) {
2276 return (reinterpret_cast<uintptr_t>(p) % (alignment * 2)) == alignment;
2277 }
2278
2279 std::vector<char*> allocated_array;
2280};
2281
2282TEST(pthread, pthread_types_allow_four_bytes_alignment) {
2283#if defined(__BIONIC__)
2284 // For binary compatibility with old version, we need to allow 4-byte aligned data for pthread types.
2285 StrictAlignmentAllocator allocator;
2286 pthread_mutex_t* mutex = reinterpret_cast<pthread_mutex_t*>(
2287 allocator.allocate(sizeof(pthread_mutex_t), 4));
Yi Kong32bc0fc2018-08-02 17:31:13 -07002288 ASSERT_EQ(0, pthread_mutex_init(mutex, nullptr));
Yabin Cuib5845722015-03-16 22:46:42 -07002289 ASSERT_EQ(0, pthread_mutex_lock(mutex));
2290 ASSERT_EQ(0, pthread_mutex_unlock(mutex));
2291 ASSERT_EQ(0, pthread_mutex_destroy(mutex));
2292
2293 pthread_cond_t* cond = reinterpret_cast<pthread_cond_t*>(
2294 allocator.allocate(sizeof(pthread_cond_t), 4));
Yi Kong32bc0fc2018-08-02 17:31:13 -07002295 ASSERT_EQ(0, pthread_cond_init(cond, nullptr));
Yabin Cuib5845722015-03-16 22:46:42 -07002296 ASSERT_EQ(0, pthread_cond_signal(cond));
2297 ASSERT_EQ(0, pthread_cond_broadcast(cond));
2298 ASSERT_EQ(0, pthread_cond_destroy(cond));
2299
2300 pthread_rwlock_t* rwlock = reinterpret_cast<pthread_rwlock_t*>(
2301 allocator.allocate(sizeof(pthread_rwlock_t), 4));
Yi Kong32bc0fc2018-08-02 17:31:13 -07002302 ASSERT_EQ(0, pthread_rwlock_init(rwlock, nullptr));
Yabin Cuib5845722015-03-16 22:46:42 -07002303 ASSERT_EQ(0, pthread_rwlock_rdlock(rwlock));
2304 ASSERT_EQ(0, pthread_rwlock_unlock(rwlock));
2305 ASSERT_EQ(0, pthread_rwlock_wrlock(rwlock));
2306 ASSERT_EQ(0, pthread_rwlock_unlock(rwlock));
2307 ASSERT_EQ(0, pthread_rwlock_destroy(rwlock));
2308
2309#else
Elliott Hughesbcaa4542019-03-08 15:20:23 -08002310 GTEST_SKIP() << "bionic-only test";
Yabin Cuib5845722015-03-16 22:46:42 -07002311#endif
2312}
Christopher Ferris60907c72015-06-09 18:46:15 -07002313
2314TEST(pthread, pthread_mutex_lock_null_32) {
2315#if defined(__BIONIC__) && !defined(__LP64__)
Dan Albertbaa2a972015-08-13 16:58:50 -07002316 // For LP32, the pthread lock/unlock functions allow a NULL mutex and return
2317 // EINVAL in that case: http://b/19995172.
2318 //
2319 // We decorate the public defintion with _Nonnull so that people recompiling
2320 // their code with get a warning and might fix their bug, but need to pass
2321 // NULL here to test that we remain compatible.
2322 pthread_mutex_t* null_value = nullptr;
2323 ASSERT_EQ(EINVAL, pthread_mutex_lock(null_value));
Christopher Ferris60907c72015-06-09 18:46:15 -07002324#else
Elliott Hughesbcaa4542019-03-08 15:20:23 -08002325 GTEST_SKIP() << "32-bit bionic-only test";
Christopher Ferris60907c72015-06-09 18:46:15 -07002326#endif
2327}
2328
2329TEST(pthread, pthread_mutex_unlock_null_32) {
2330#if defined(__BIONIC__) && !defined(__LP64__)
Dan Albertbaa2a972015-08-13 16:58:50 -07002331 // For LP32, the pthread lock/unlock functions allow a NULL mutex and return
2332 // EINVAL in that case: http://b/19995172.
2333 //
2334 // We decorate the public defintion with _Nonnull so that people recompiling
2335 // their code with get a warning and might fix their bug, but need to pass
2336 // NULL here to test that we remain compatible.
2337 pthread_mutex_t* null_value = nullptr;
2338 ASSERT_EQ(EINVAL, pthread_mutex_unlock(null_value));
Christopher Ferris60907c72015-06-09 18:46:15 -07002339#else
Elliott Hughesbcaa4542019-03-08 15:20:23 -08002340 GTEST_SKIP() << "32-bit bionic-only test";
Christopher Ferris60907c72015-06-09 18:46:15 -07002341#endif
2342}
2343
2344TEST_F(pthread_DeathTest, pthread_mutex_lock_null_64) {
2345#if defined(__BIONIC__) && defined(__LP64__)
2346 pthread_mutex_t* null_value = nullptr;
2347 ASSERT_EXIT(pthread_mutex_lock(null_value), testing::KilledBySignal(SIGSEGV), "");
2348#else
Elliott Hughesbcaa4542019-03-08 15:20:23 -08002349 GTEST_SKIP() << "64-bit bionic-only test";
Christopher Ferris60907c72015-06-09 18:46:15 -07002350#endif
2351}
2352
2353TEST_F(pthread_DeathTest, pthread_mutex_unlock_null_64) {
2354#if defined(__BIONIC__) && defined(__LP64__)
2355 pthread_mutex_t* null_value = nullptr;
2356 ASSERT_EXIT(pthread_mutex_unlock(null_value), testing::KilledBySignal(SIGSEGV), "");
2357#else
Elliott Hughesbcaa4542019-03-08 15:20:23 -08002358 GTEST_SKIP() << "64-bit bionic-only test";
Christopher Ferris60907c72015-06-09 18:46:15 -07002359#endif
2360}
Yabin Cui33ac04a2015-09-22 11:16:15 -07002361
2362extern _Unwind_Reason_Code FrameCounter(_Unwind_Context* ctx, void* arg);
2363
2364static volatile bool signal_handler_on_altstack_done;
2365
Josh Gao61db9ac2017-03-15 19:42:05 -07002366__attribute__((__noinline__))
2367static void signal_handler_backtrace() {
2368 // Check if we have enough stack space for unwinding.
2369 int count = 0;
2370 _Unwind_Backtrace(FrameCounter, &count);
2371 ASSERT_GT(count, 0);
2372}
2373
2374__attribute__((__noinline__))
2375static void signal_handler_logging() {
2376 // Check if we have enough stack space for logging.
2377 std::string s(2048, '*');
2378 GTEST_LOG_(INFO) << s;
2379 signal_handler_on_altstack_done = true;
2380}
2381
2382__attribute__((__noinline__))
2383static void signal_handler_snprintf() {
2384 // Check if we have enough stack space for snprintf to a PATH_MAX buffer, plus some extra.
2385 char buf[PATH_MAX + 2048];
2386 ASSERT_GT(snprintf(buf, sizeof(buf), "/proc/%d/status", getpid()), 0);
2387}
2388
Yabin Cui33ac04a2015-09-22 11:16:15 -07002389static void SignalHandlerOnAltStack(int signo, siginfo_t*, void*) {
2390 ASSERT_EQ(SIGUSR1, signo);
Josh Gao61db9ac2017-03-15 19:42:05 -07002391 signal_handler_backtrace();
2392 signal_handler_logging();
2393 signal_handler_snprintf();
Yabin Cui33ac04a2015-09-22 11:16:15 -07002394}
2395
Josh Gao415daa82017-03-06 17:45:33 -08002396TEST(pthread, big_enough_signal_stack) {
Yabin Cui33ac04a2015-09-22 11:16:15 -07002397 signal_handler_on_altstack_done = false;
2398 ScopedSignalHandler handler(SIGUSR1, SignalHandlerOnAltStack, SA_SIGINFO | SA_ONSTACK);
2399 kill(getpid(), SIGUSR1);
2400 ASSERT_TRUE(signal_handler_on_altstack_done);
2401}
Yabin Cuie7c2fff2015-11-05 22:06:09 -08002402
2403TEST(pthread, pthread_barrierattr_smoke) {
2404 pthread_barrierattr_t attr;
2405 ASSERT_EQ(0, pthread_barrierattr_init(&attr));
2406 int pshared;
2407 ASSERT_EQ(0, pthread_barrierattr_getpshared(&attr, &pshared));
2408 ASSERT_EQ(PTHREAD_PROCESS_PRIVATE, pshared);
2409 ASSERT_EQ(0, pthread_barrierattr_setpshared(&attr, PTHREAD_PROCESS_SHARED));
2410 ASSERT_EQ(0, pthread_barrierattr_getpshared(&attr, &pshared));
2411 ASSERT_EQ(PTHREAD_PROCESS_SHARED, pshared);
2412 ASSERT_EQ(0, pthread_barrierattr_destroy(&attr));
2413}
2414
Yabin Cui81d27972016-03-22 13:45:55 -07002415struct BarrierTestHelperData {
2416 size_t thread_count;
2417 pthread_barrier_t barrier;
2418 std::atomic<int> finished_mask;
2419 std::atomic<int> serial_thread_count;
Yabin Cuie7c2fff2015-11-05 22:06:09 -08002420 size_t iteration_count;
Yabin Cui81d27972016-03-22 13:45:55 -07002421 std::atomic<size_t> finished_iteration_count;
2422
2423 BarrierTestHelperData(size_t thread_count, size_t iteration_count)
2424 : thread_count(thread_count), finished_mask(0), serial_thread_count(0),
2425 iteration_count(iteration_count), finished_iteration_count(0) {
2426 }
2427};
2428
2429struct BarrierTestHelperArg {
2430 int id;
2431 BarrierTestHelperData* data;
Yabin Cuie7c2fff2015-11-05 22:06:09 -08002432};
2433
2434static void BarrierTestHelper(BarrierTestHelperArg* arg) {
Yabin Cui81d27972016-03-22 13:45:55 -07002435 for (size_t i = 0; i < arg->data->iteration_count; ++i) {
2436 int result = pthread_barrier_wait(&arg->data->barrier);
2437 if (result == PTHREAD_BARRIER_SERIAL_THREAD) {
2438 arg->data->serial_thread_count++;
2439 } else {
2440 ASSERT_EQ(0, result);
2441 }
Yabin Cuid5c04c52017-05-02 12:57:39 -07002442 int mask = arg->data->finished_mask.fetch_or(1 << arg->id);
Yabin Cuiab4cddc2017-05-02 16:18:13 -07002443 mask |= 1 << arg->id;
Yabin Cuid5c04c52017-05-02 12:57:39 -07002444 if (mask == ((1 << arg->data->thread_count) - 1)) {
Yabin Cui81d27972016-03-22 13:45:55 -07002445 ASSERT_EQ(1, arg->data->serial_thread_count);
2446 arg->data->finished_iteration_count++;
2447 arg->data->finished_mask = 0;
2448 arg->data->serial_thread_count = 0;
2449 }
Yabin Cuie7c2fff2015-11-05 22:06:09 -08002450 }
2451}
2452
2453TEST(pthread, pthread_barrier_smoke) {
2454 const size_t BARRIER_ITERATION_COUNT = 10;
2455 const size_t BARRIER_THREAD_COUNT = 10;
Yabin Cui81d27972016-03-22 13:45:55 -07002456 BarrierTestHelperData data(BARRIER_THREAD_COUNT, BARRIER_ITERATION_COUNT);
2457 ASSERT_EQ(0, pthread_barrier_init(&data.barrier, nullptr, data.thread_count));
2458 std::vector<pthread_t> threads(data.thread_count);
Yabin Cuie7c2fff2015-11-05 22:06:09 -08002459 std::vector<BarrierTestHelperArg> args(threads.size());
2460 for (size_t i = 0; i < threads.size(); ++i) {
Yabin Cui81d27972016-03-22 13:45:55 -07002461 args[i].id = i;
2462 args[i].data = &data;
Yabin Cuie7c2fff2015-11-05 22:06:09 -08002463 ASSERT_EQ(0, pthread_create(&threads[i], nullptr,
2464 reinterpret_cast<void* (*)(void*)>(BarrierTestHelper), &args[i]));
2465 }
Yabin Cuie7c2fff2015-11-05 22:06:09 -08002466 for (size_t i = 0; i < threads.size(); ++i) {
2467 ASSERT_EQ(0, pthread_join(threads[i], nullptr));
2468 }
Yabin Cui81d27972016-03-22 13:45:55 -07002469 ASSERT_EQ(data.iteration_count, data.finished_iteration_count);
2470 ASSERT_EQ(0, pthread_barrier_destroy(&data.barrier));
2471}
2472
2473struct BarrierDestroyTestArg {
2474 std::atomic<int> tid;
2475 pthread_barrier_t* barrier;
2476};
2477
2478static void BarrierDestroyTestHelper(BarrierDestroyTestArg* arg) {
2479 arg->tid = gettid();
2480 ASSERT_EQ(0, pthread_barrier_wait(arg->barrier));
Yabin Cuie7c2fff2015-11-05 22:06:09 -08002481}
2482
2483TEST(pthread, pthread_barrier_destroy) {
2484 pthread_barrier_t barrier;
2485 ASSERT_EQ(0, pthread_barrier_init(&barrier, nullptr, 2));
2486 pthread_t thread;
Yabin Cui81d27972016-03-22 13:45:55 -07002487 BarrierDestroyTestArg arg;
Yabin Cuie7c2fff2015-11-05 22:06:09 -08002488 arg.tid = 0;
2489 arg.barrier = &barrier;
Yabin Cuie7c2fff2015-11-05 22:06:09 -08002490 ASSERT_EQ(0, pthread_create(&thread, nullptr,
Yabin Cui81d27972016-03-22 13:45:55 -07002491 reinterpret_cast<void* (*)(void*)>(BarrierDestroyTestHelper), &arg));
Yabin Cuie7c2fff2015-11-05 22:06:09 -08002492 WaitUntilThreadSleep(arg.tid);
2493 ASSERT_EQ(EBUSY, pthread_barrier_destroy(&barrier));
2494 ASSERT_EQ(PTHREAD_BARRIER_SERIAL_THREAD, pthread_barrier_wait(&barrier));
2495 // Verify if the barrier can be destroyed directly after pthread_barrier_wait().
2496 ASSERT_EQ(0, pthread_barrier_destroy(&barrier));
2497 ASSERT_EQ(0, pthread_join(thread, nullptr));
2498#if defined(__BIONIC__)
2499 ASSERT_EQ(EINVAL, pthread_barrier_destroy(&barrier));
2500#endif
2501}
2502
2503struct BarrierOrderingTestHelperArg {
2504 pthread_barrier_t* barrier;
2505 size_t* array;
2506 size_t array_length;
2507 size_t id;
2508};
2509
2510void BarrierOrderingTestHelper(BarrierOrderingTestHelperArg* arg) {
2511 const size_t ITERATION_COUNT = 10000;
2512 for (size_t i = 1; i <= ITERATION_COUNT; ++i) {
2513 arg->array[arg->id] = i;
Yabin Cuic9a659c2015-11-05 15:36:08 -08002514 int result = pthread_barrier_wait(arg->barrier);
2515 ASSERT_TRUE(result == 0 || result == PTHREAD_BARRIER_SERIAL_THREAD);
Yabin Cuie7c2fff2015-11-05 22:06:09 -08002516 for (size_t j = 0; j < arg->array_length; ++j) {
2517 ASSERT_EQ(i, arg->array[j]);
2518 }
Yabin Cuic9a659c2015-11-05 15:36:08 -08002519 result = pthread_barrier_wait(arg->barrier);
2520 ASSERT_TRUE(result == 0 || result == PTHREAD_BARRIER_SERIAL_THREAD);
Yabin Cuie7c2fff2015-11-05 22:06:09 -08002521 }
2522}
2523
2524TEST(pthread, pthread_barrier_check_ordering) {
2525 const size_t THREAD_COUNT = 4;
2526 pthread_barrier_t barrier;
2527 ASSERT_EQ(0, pthread_barrier_init(&barrier, nullptr, THREAD_COUNT));
2528 size_t array[THREAD_COUNT];
2529 std::vector<pthread_t> threads(THREAD_COUNT);
2530 std::vector<BarrierOrderingTestHelperArg> args(THREAD_COUNT);
2531 for (size_t i = 0; i < THREAD_COUNT; ++i) {
2532 args[i].barrier = &barrier;
2533 args[i].array = array;
2534 args[i].array_length = THREAD_COUNT;
2535 args[i].id = i;
2536 ASSERT_EQ(0, pthread_create(&threads[i], nullptr,
2537 reinterpret_cast<void* (*)(void*)>(BarrierOrderingTestHelper),
2538 &args[i]));
2539 }
2540 for (size_t i = 0; i < THREAD_COUNT; ++i) {
2541 ASSERT_EQ(0, pthread_join(threads[i], nullptr));
2542 }
2543}
Yabin Cuife3a83a2015-11-17 16:03:18 -08002544
Elliott Hughes463faad2018-07-06 14:34:49 -07002545TEST(pthread, pthread_barrier_init_zero_count) {
2546 pthread_barrier_t barrier;
2547 ASSERT_EQ(EINVAL, pthread_barrier_init(&barrier, nullptr, 0));
2548}
2549
Yabin Cuife3a83a2015-11-17 16:03:18 -08002550TEST(pthread, pthread_spinlock_smoke) {
2551 pthread_spinlock_t lock;
2552 ASSERT_EQ(0, pthread_spin_init(&lock, 0));
2553 ASSERT_EQ(0, pthread_spin_trylock(&lock));
2554 ASSERT_EQ(0, pthread_spin_unlock(&lock));
2555 ASSERT_EQ(0, pthread_spin_lock(&lock));
2556 ASSERT_EQ(EBUSY, pthread_spin_trylock(&lock));
2557 ASSERT_EQ(0, pthread_spin_unlock(&lock));
2558 ASSERT_EQ(0, pthread_spin_destroy(&lock));
2559}
Elliott Hughes53dc9dd2017-09-19 14:02:50 -07002560
Elliott Hughes8aecba72017-10-17 15:34:41 -07002561TEST(pthread, pthread_attr_getdetachstate__pthread_attr_setdetachstate) {
Elliott Hughes53dc9dd2017-09-19 14:02:50 -07002562 pthread_attr_t attr;
2563 ASSERT_EQ(0, pthread_attr_init(&attr));
2564
Elliott Hughes8aecba72017-10-17 15:34:41 -07002565 int state;
Elliott Hughes53dc9dd2017-09-19 14:02:50 -07002566 ASSERT_EQ(0, pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED));
Elliott Hughes8aecba72017-10-17 15:34:41 -07002567 ASSERT_EQ(0, pthread_attr_getdetachstate(&attr, &state));
2568 ASSERT_EQ(PTHREAD_CREATE_DETACHED, state);
2569
Elliott Hughes53dc9dd2017-09-19 14:02:50 -07002570 ASSERT_EQ(0, pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_JOINABLE));
Elliott Hughes8aecba72017-10-17 15:34:41 -07002571 ASSERT_EQ(0, pthread_attr_getdetachstate(&attr, &state));
2572 ASSERT_EQ(PTHREAD_CREATE_JOINABLE, state);
2573
Elliott Hughes53dc9dd2017-09-19 14:02:50 -07002574 ASSERT_EQ(EINVAL, pthread_attr_setdetachstate(&attr, 123));
Elliott Hughes8aecba72017-10-17 15:34:41 -07002575 ASSERT_EQ(0, pthread_attr_getdetachstate(&attr, &state));
2576 ASSERT_EQ(PTHREAD_CREATE_JOINABLE, state);
Elliott Hughes53dc9dd2017-09-19 14:02:50 -07002577}
2578
2579TEST(pthread, pthread_create__mmap_failures) {
2580 pthread_attr_t attr;
2581 ASSERT_EQ(0, pthread_attr_init(&attr));
2582 ASSERT_EQ(0, pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED));
2583
2584 const auto kPageSize = sysconf(_SC_PAGE_SIZE);
2585
Elliott Hughes57512982017-10-02 22:49:18 -07002586 // Use up all the VMAs. By default this is 64Ki (though some will already be in use).
Elliott Hughes53dc9dd2017-09-19 14:02:50 -07002587 std::vector<void*> pages;
Elliott Hughes57512982017-10-02 22:49:18 -07002588 pages.reserve(64 * 1024);
Elliott Hughes53dc9dd2017-09-19 14:02:50 -07002589 int prot = PROT_NONE;
2590 while (true) {
2591 void* page = mmap(nullptr, kPageSize, prot, MAP_ANON|MAP_PRIVATE, -1, 0);
2592 if (page == MAP_FAILED) break;
2593 pages.push_back(page);
2594 prot = (prot == PROT_NONE) ? PROT_READ : PROT_NONE;
2595 }
2596
2597 // Try creating threads, freeing up a page each time we fail.
2598 size_t EAGAIN_count = 0;
2599 size_t i = 0;
2600 for (; i < pages.size(); ++i) {
2601 pthread_t t;
2602 int status = pthread_create(&t, &attr, IdFn, nullptr);
2603 if (status != EAGAIN) break;
2604 ++EAGAIN_count;
2605 ASSERT_EQ(0, munmap(pages[i], kPageSize));
2606 }
2607
Ryan Prichard45d13492019-01-03 02:51:30 -08002608 // Creating a thread uses at least three VMAs: the combined stack and TLS, and a guard on each
2609 // side. So we should have seen at least three failures.
2610 ASSERT_GE(EAGAIN_count, 3U);
Elliott Hughes53dc9dd2017-09-19 14:02:50 -07002611
2612 for (; i < pages.size(); ++i) {
2613 ASSERT_EQ(0, munmap(pages[i], kPageSize));
2614 }
2615}
Elliott Hughesdff08ce2017-10-16 09:58:45 -07002616
2617TEST(pthread, pthread_setschedparam) {
2618 sched_param p = { .sched_priority = INT_MIN };
2619 ASSERT_EQ(EINVAL, pthread_setschedparam(pthread_self(), INT_MIN, &p));
2620}
2621
2622TEST(pthread, pthread_setschedprio) {
2623 ASSERT_EQ(EINVAL, pthread_setschedprio(pthread_self(), INT_MIN));
2624}
Elliott Hughes8aecba72017-10-17 15:34:41 -07002625
2626TEST(pthread, pthread_attr_getinheritsched__pthread_attr_setinheritsched) {
2627 pthread_attr_t attr;
2628 ASSERT_EQ(0, pthread_attr_init(&attr));
2629
2630 int state;
2631 ASSERT_EQ(0, pthread_attr_setinheritsched(&attr, PTHREAD_INHERIT_SCHED));
2632 ASSERT_EQ(0, pthread_attr_getinheritsched(&attr, &state));
2633 ASSERT_EQ(PTHREAD_INHERIT_SCHED, state);
2634
2635 ASSERT_EQ(0, pthread_attr_setinheritsched(&attr, PTHREAD_EXPLICIT_SCHED));
2636 ASSERT_EQ(0, pthread_attr_getinheritsched(&attr, &state));
2637 ASSERT_EQ(PTHREAD_EXPLICIT_SCHED, state);
2638
2639 ASSERT_EQ(EINVAL, pthread_attr_setinheritsched(&attr, 123));
2640 ASSERT_EQ(0, pthread_attr_getinheritsched(&attr, &state));
2641 ASSERT_EQ(PTHREAD_EXPLICIT_SCHED, state);
2642}
2643
2644TEST(pthread, pthread_attr_setinheritsched__PTHREAD_INHERIT_SCHED__PTHREAD_EXPLICIT_SCHED) {
2645 pthread_attr_t attr;
2646 ASSERT_EQ(0, pthread_attr_init(&attr));
2647
2648 // If we set invalid scheduling attributes but choose to inherit, everything's fine...
2649 sched_param param = { .sched_priority = sched_get_priority_max(SCHED_FIFO) + 1 };
2650 ASSERT_EQ(0, pthread_attr_setschedparam(&attr, &param));
2651 ASSERT_EQ(0, pthread_attr_setschedpolicy(&attr, SCHED_FIFO));
2652 ASSERT_EQ(0, pthread_attr_setinheritsched(&attr, PTHREAD_INHERIT_SCHED));
2653
2654 pthread_t t;
2655 ASSERT_EQ(0, pthread_create(&t, &attr, IdFn, nullptr));
2656 ASSERT_EQ(0, pthread_join(t, nullptr));
2657
Elliott Hughes7a660662017-10-30 09:26:06 -07002658#if defined(__LP64__)
2659 // If we ask to use them, though, we'll see a failure...
Elliott Hughes8aecba72017-10-17 15:34:41 -07002660 ASSERT_EQ(0, pthread_attr_setinheritsched(&attr, PTHREAD_EXPLICIT_SCHED));
2661 ASSERT_EQ(EINVAL, pthread_create(&t, &attr, IdFn, nullptr));
Elliott Hughes7a660662017-10-30 09:26:06 -07002662#else
2663 // For backwards compatibility with broken apps, we just ignore failures
2664 // to set scheduler attributes on LP32.
2665#endif
Elliott Hughes8aecba72017-10-17 15:34:41 -07002666}
2667
2668TEST(pthread, pthread_attr_setinheritsched_PTHREAD_INHERIT_SCHED_takes_effect) {
2669 sched_param param = { .sched_priority = sched_get_priority_min(SCHED_FIFO) };
2670 int rc = pthread_setschedparam(pthread_self(), SCHED_FIFO, &param);
Elliott Hughesbcaa4542019-03-08 15:20:23 -08002671 if (rc == EPERM) GTEST_SKIP() << "pthread_setschedparam failed with EPERM";
Elliott Hughes8aecba72017-10-17 15:34:41 -07002672 ASSERT_EQ(0, rc);
2673
2674 pthread_attr_t attr;
2675 ASSERT_EQ(0, pthread_attr_init(&attr));
2676 ASSERT_EQ(0, pthread_attr_setinheritsched(&attr, PTHREAD_INHERIT_SCHED));
2677
Elliott Hughes0bd9d132017-11-02 13:11:13 -07002678 SpinFunctionHelper spin_helper;
Elliott Hughes8aecba72017-10-17 15:34:41 -07002679 pthread_t t;
Elliott Hughes0bd9d132017-11-02 13:11:13 -07002680 ASSERT_EQ(0, pthread_create(&t, &attr, spin_helper.GetFunction(), nullptr));
Elliott Hughes8aecba72017-10-17 15:34:41 -07002681 int actual_policy;
2682 sched_param actual_param;
2683 ASSERT_EQ(0, pthread_getschedparam(t, &actual_policy, &actual_param));
2684 ASSERT_EQ(SCHED_FIFO, actual_policy);
Elliott Hughes0bd9d132017-11-02 13:11:13 -07002685 spin_helper.UnSpin();
Elliott Hughes8aecba72017-10-17 15:34:41 -07002686 ASSERT_EQ(0, pthread_join(t, nullptr));
2687}
2688
2689TEST(pthread, pthread_attr_setinheritsched_PTHREAD_EXPLICIT_SCHED_takes_effect) {
2690 sched_param param = { .sched_priority = sched_get_priority_min(SCHED_FIFO) };
2691 int rc = pthread_setschedparam(pthread_self(), SCHED_FIFO, &param);
Elliott Hughesbcaa4542019-03-08 15:20:23 -08002692 if (rc == EPERM) GTEST_SKIP() << "pthread_setschedparam failed with EPERM";
Elliott Hughes8aecba72017-10-17 15:34:41 -07002693 ASSERT_EQ(0, rc);
2694
2695 pthread_attr_t attr;
2696 ASSERT_EQ(0, pthread_attr_init(&attr));
2697 ASSERT_EQ(0, pthread_attr_setinheritsched(&attr, PTHREAD_EXPLICIT_SCHED));
2698 ASSERT_EQ(0, pthread_attr_setschedpolicy(&attr, SCHED_OTHER));
2699
Elliott Hughes0bd9d132017-11-02 13:11:13 -07002700 SpinFunctionHelper spin_helper;
Elliott Hughes8aecba72017-10-17 15:34:41 -07002701 pthread_t t;
Elliott Hughes0bd9d132017-11-02 13:11:13 -07002702 ASSERT_EQ(0, pthread_create(&t, &attr, spin_helper.GetFunction(), nullptr));
Elliott Hughes8aecba72017-10-17 15:34:41 -07002703 int actual_policy;
2704 sched_param actual_param;
2705 ASSERT_EQ(0, pthread_getschedparam(t, &actual_policy, &actual_param));
2706 ASSERT_EQ(SCHED_OTHER, actual_policy);
Elliott Hughes0bd9d132017-11-02 13:11:13 -07002707 spin_helper.UnSpin();
Elliott Hughes8aecba72017-10-17 15:34:41 -07002708 ASSERT_EQ(0, pthread_join(t, nullptr));
2709}
2710
2711TEST(pthread, pthread_attr_setinheritsched__takes_effect_despite_SCHED_RESET_ON_FORK) {
2712 sched_param param = { .sched_priority = sched_get_priority_min(SCHED_FIFO) };
2713 int rc = pthread_setschedparam(pthread_self(), SCHED_FIFO | SCHED_RESET_ON_FORK, &param);
Elliott Hughesbcaa4542019-03-08 15:20:23 -08002714 if (rc == EPERM) GTEST_SKIP() << "pthread_setschedparam failed with EPERM";
Elliott Hughes8aecba72017-10-17 15:34:41 -07002715 ASSERT_EQ(0, rc);
2716
2717 pthread_attr_t attr;
2718 ASSERT_EQ(0, pthread_attr_init(&attr));
2719 ASSERT_EQ(0, pthread_attr_setinheritsched(&attr, PTHREAD_INHERIT_SCHED));
2720
Elliott Hughes0bd9d132017-11-02 13:11:13 -07002721 SpinFunctionHelper spin_helper;
Elliott Hughes8aecba72017-10-17 15:34:41 -07002722 pthread_t t;
Elliott Hughes0bd9d132017-11-02 13:11:13 -07002723 ASSERT_EQ(0, pthread_create(&t, &attr, spin_helper.GetFunction(), nullptr));
Elliott Hughes8aecba72017-10-17 15:34:41 -07002724 int actual_policy;
2725 sched_param actual_param;
2726 ASSERT_EQ(0, pthread_getschedparam(t, &actual_policy, &actual_param));
2727 ASSERT_EQ(SCHED_FIFO | SCHED_RESET_ON_FORK, actual_policy);
Elliott Hughes0bd9d132017-11-02 13:11:13 -07002728 spin_helper.UnSpin();
Elliott Hughes8aecba72017-10-17 15:34:41 -07002729 ASSERT_EQ(0, pthread_join(t, nullptr));
2730}