Blame jemalloc/include/jemalloc/internal/tsd_generic.h

Packit Service 724aca
#ifdef JEMALLOC_INTERNAL_TSD_GENERIC_H
Packit Service 724aca
#error This file should be included only once, by tsd.h.
Packit Service 724aca
#endif
Packit Service 724aca
#define JEMALLOC_INTERNAL_TSD_GENERIC_H
Packit Service 724aca
Packit Service 724aca
typedef struct tsd_init_block_s tsd_init_block_t;
Packit Service 724aca
struct tsd_init_block_s {
Packit Service 724aca
	ql_elm(tsd_init_block_t) link;
Packit Service 724aca
	pthread_t thread;
Packit Service 724aca
	void *data;
Packit Service 724aca
};
Packit Service 724aca
Packit Service 724aca
/* Defined in tsd.c, to allow the mutex headers to have tsd dependencies. */
Packit Service 724aca
typedef struct tsd_init_head_s tsd_init_head_t;
Packit Service 724aca
Packit Service 724aca
typedef struct {
Packit Service 724aca
	bool initialized;
Packit Service 724aca
	tsd_t val;
Packit Service 724aca
} tsd_wrapper_t;
Packit Service 724aca
Packit Service 724aca
void *tsd_init_check_recursion(tsd_init_head_t *head,
Packit Service 724aca
    tsd_init_block_t *block);
Packit Service 724aca
void tsd_init_finish(tsd_init_head_t *head, tsd_init_block_t *block);
Packit Service 724aca
Packit Service 724aca
extern pthread_key_t tsd_tsd;
Packit Service 724aca
extern tsd_init_head_t tsd_init_head;
Packit Service 724aca
extern tsd_wrapper_t tsd_boot_wrapper;
Packit Service 724aca
extern bool tsd_booted;
Packit Service 724aca
Packit Service 724aca
/* Initialization/cleanup. */
Packit Service 724aca
JEMALLOC_ALWAYS_INLINE void
Packit Service 724aca
tsd_cleanup_wrapper(void *arg) {
Packit Service 724aca
	tsd_wrapper_t *wrapper = (tsd_wrapper_t *)arg;
Packit Service 724aca
Packit Service 724aca
	if (wrapper->initialized) {
Packit Service 724aca
		wrapper->initialized = false;
Packit Service 724aca
		tsd_cleanup(&wrapper->val);
Packit Service 724aca
		if (wrapper->initialized) {
Packit Service 724aca
			/* Trigger another cleanup round. */
Packit Service 724aca
			if (pthread_setspecific(tsd_tsd, (void *)wrapper) != 0)
Packit Service 724aca
			{
Packit Service 724aca
				malloc_write("<jemalloc>: Error setting TSD\n");
Packit Service 724aca
				if (opt_abort) {
Packit Service 724aca
					abort();
Packit Service 724aca
				}
Packit Service 724aca
			}
Packit Service 724aca
			return;
Packit Service 724aca
		}
Packit Service 724aca
	}
Packit Service 724aca
	malloc_tsd_dalloc(wrapper);
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
JEMALLOC_ALWAYS_INLINE void
Packit Service 724aca
tsd_wrapper_set(tsd_wrapper_t *wrapper) {
Packit Service 724aca
	if (pthread_setspecific(tsd_tsd, (void *)wrapper) != 0) {
Packit Service 724aca
		malloc_write("<jemalloc>: Error setting TSD\n");
Packit Service 724aca
		abort();
Packit Service 724aca
	}
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
JEMALLOC_ALWAYS_INLINE tsd_wrapper_t *
Packit Service 724aca
tsd_wrapper_get(bool init) {
Packit Service 724aca
	tsd_wrapper_t *wrapper = (tsd_wrapper_t *)pthread_getspecific(tsd_tsd);
Packit Service 724aca
Packit Service 724aca
	if (init && unlikely(wrapper == NULL)) {
Packit Service 724aca
		tsd_init_block_t block;
Packit Service 724aca
		wrapper = (tsd_wrapper_t *)
Packit Service 724aca
		    tsd_init_check_recursion(&tsd_init_head, &block);
Packit Service 724aca
		if (wrapper) {
Packit Service 724aca
			return wrapper;
Packit Service 724aca
		}
Packit Service 724aca
		wrapper = (tsd_wrapper_t *)
Packit Service 724aca
		    malloc_tsd_malloc(sizeof(tsd_wrapper_t));
Packit Service 724aca
		block.data = (void *)wrapper;
Packit Service 724aca
		if (wrapper == NULL) {
Packit Service 724aca
			malloc_write("<jemalloc>: Error allocating TSD\n");
Packit Service 724aca
			abort();
Packit Service 724aca
		} else {
Packit Service 724aca
			wrapper->initialized = false;
Packit Service 724aca
      JEMALLOC_DIAGNOSTIC_PUSH
Packit Service 724aca
      JEMALLOC_DIAGNOSTIC_IGNORE_MISSING_STRUCT_FIELD_INITIALIZERS
Packit Service 724aca
			tsd_t initializer = TSD_INITIALIZER;
Packit Service 724aca
      JEMALLOC_DIAGNOSTIC_POP
Packit Service 724aca
			wrapper->val = initializer;
Packit Service 724aca
		}
Packit Service 724aca
		tsd_wrapper_set(wrapper);
Packit Service 724aca
		tsd_init_finish(&tsd_init_head, &block);
Packit Service 724aca
	}
Packit Service 724aca
	return wrapper;
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
JEMALLOC_ALWAYS_INLINE bool
Packit Service 724aca
tsd_boot0(void) {
Packit Service 724aca
	if (pthread_key_create(&tsd_tsd, tsd_cleanup_wrapper) != 0) {
Packit Service 724aca
		return true;
Packit Service 724aca
	}
Packit Service 724aca
	tsd_wrapper_set(&tsd_boot_wrapper);
Packit Service 724aca
	tsd_booted = true;
Packit Service 724aca
	return false;
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
JEMALLOC_ALWAYS_INLINE void
Packit Service 724aca
tsd_boot1(void) {
Packit Service 724aca
	tsd_wrapper_t *wrapper;
Packit Service 724aca
	wrapper = (tsd_wrapper_t *)malloc_tsd_malloc(sizeof(tsd_wrapper_t));
Packit Service 724aca
	if (wrapper == NULL) {
Packit Service 724aca
		malloc_write("<jemalloc>: Error allocating TSD\n");
Packit Service 724aca
		abort();
Packit Service 724aca
	}
Packit Service 724aca
	tsd_boot_wrapper.initialized = false;
Packit Service 724aca
	tsd_cleanup(&tsd_boot_wrapper.val);
Packit Service 724aca
	wrapper->initialized = false;
Packit Service 724aca
  JEMALLOC_DIAGNOSTIC_PUSH
Packit Service 724aca
  JEMALLOC_DIAGNOSTIC_IGNORE_MISSING_STRUCT_FIELD_INITIALIZERS
Packit Service 724aca
	tsd_t initializer = TSD_INITIALIZER;
Packit Service 724aca
  JEMALLOC_DIAGNOSTIC_POP
Packit Service 724aca
	wrapper->val = initializer;
Packit Service 724aca
	tsd_wrapper_set(wrapper);
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
JEMALLOC_ALWAYS_INLINE bool
Packit Service 724aca
tsd_boot(void) {
Packit Service 724aca
	if (tsd_boot0()) {
Packit Service 724aca
		return true;
Packit Service 724aca
	}
Packit Service 724aca
	tsd_boot1();
Packit Service 724aca
	return false;
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
JEMALLOC_ALWAYS_INLINE bool
Packit Service 724aca
tsd_booted_get(void) {
Packit Service 724aca
	return tsd_booted;
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
JEMALLOC_ALWAYS_INLINE bool
Packit Service 724aca
tsd_get_allocates(void) {
Packit Service 724aca
	return true;
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
/* Get/set. */
Packit Service 724aca
JEMALLOC_ALWAYS_INLINE tsd_t *
Packit Service 724aca
tsd_get(bool init) {
Packit Service 724aca
	tsd_wrapper_t *wrapper;
Packit Service 724aca
Packit Service 724aca
	assert(tsd_booted);
Packit Service 724aca
	wrapper = tsd_wrapper_get(init);
Packit Service 724aca
	if (tsd_get_allocates() && !init && wrapper == NULL) {
Packit Service 724aca
		return NULL;
Packit Service 724aca
	}
Packit Service 724aca
	return &wrapper->val;
Packit Service 724aca
}
Packit Service 724aca
Packit Service 724aca
JEMALLOC_ALWAYS_INLINE void
Packit Service 724aca
tsd_set(tsd_t *val) {
Packit Service 724aca
	tsd_wrapper_t *wrapper;
Packit Service 724aca
Packit Service 724aca
	assert(tsd_booted);
Packit Service 724aca
	wrapper = tsd_wrapper_get(true);
Packit Service 724aca
	if (likely(&wrapper->val != val)) {
Packit Service 724aca
		wrapper->val = *(val);
Packit Service 724aca
	}
Packit Service 724aca
	wrapper->initialized = true;
Packit Service 724aca
}