pthread_introspection.c [plain text]
#include <stdatomic.h>
#include <pthread.h>
#include <pthread/introspection_private.h>
#include <dispatch/dispatch.h>
#include "darwintest_defaults.h"
static pthread_introspection_hook_t prev_pthread_introspection_hook;
static pthread_key_t key;
static atomic_int keys_set;
static atomic_int keys_cleared;
static const char value_for_key[] = "used as a TSD value";
static void
key_destructor(void *ctx)
{
T_EXPECT_EQ(ctx, (void *)value_for_key, "check value");
keys_cleared++;
}
#define THREAD_COUNT 3
static void my_pthread_introspection_hook(unsigned int event, pthread_t thread,
void *addr, size_t size)
{
static atomic_int create_count;
static atomic_int terminate_count;
static atomic_int destroy_count;
uint64_t tid;
pthread_threadid_np(NULL, &tid);
switch (event) {
case PTHREAD_INTROSPECTION_THREAD_CREATE:
atomic_fetch_add(&keys_set, 1);
pthread_introspection_setspecific_np(thread, key, value_for_key);
T_LOG("event = PTHREAD_INTROSPECTION_THREAD_CREATE, thread = %p:%lld, addr = %p, size = 0x%zx", thread, tid, addr, size);
create_count++;
break;
case PTHREAD_INTROSPECTION_THREAD_TERMINATE:
T_LOG("event = PTHREAD_INTROSPECTION_THREAD_TERMINATE, thread = %p:%lld, addr = %p, size = 0x%zx", thread, tid, addr, size);
terminate_count++;
T_ASSERT_GE(create_count, THREAD_COUNT, NULL);
T_PASS("Got termination events");
break;
case PTHREAD_INTROSPECTION_THREAD_DESTROY:
T_LOG("event = PTHREAD_INTROSPECTION_THREAD_DESTROY, thread = %p:%lld, addr = %p, size = 0x%zx", thread, tid, addr, size);
destroy_count++;
T_ASSERT_NULL(pthread_introspection_getspecific_np(thread, key), "should have cleared");
T_ASSERT_NE(keys_cleared, 0, "should have cleared a key");
T_PASS("Got destruction events");
T_END;
}
if (prev_pthread_introspection_hook != NULL){
prev_pthread_introspection_hook(event, thread, addr, size);
}
}
T_DECL(pthread_introspection, "PR-25679871",
T_META_TIMEOUT(30), T_META_ALL_VALID_ARCHS(YES))
{
prev_pthread_introspection_hook = pthread_introspection_hook_install(&my_pthread_introspection_hook);
pthread_key_create(&key, key_destructor);
for (int i = 0; i < THREAD_COUNT - 1; i++) {
T_LOG("Creating dispatch_async thread %d.", i);
dispatch_async(dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0), ^{
T_LOG("Started dispatch_async thread %d.", i);
sleep(3);
});
}
dispatch_queue_t serial_queue = dispatch_queue_create("test queue", NULL);
__block dispatch_block_t looping_block = ^{
static int count;
if (count++ < 20) {
dispatch_after(dispatch_time(DISPATCH_TIME_NOW, 50 * NSEC_PER_MSEC), serial_queue, looping_block);
} else {
T_LOG("Looping block complete");
}
};
T_LOG("Starting looping block");
dispatch_async(serial_queue, looping_block);
sleep(30);
T_FAIL("Why are we still alive?");
}