Ondřej Surý
d07c4a98da
Prefer the pthread_barrier implementation on platforms where it is available over uv_barrier implementation. This also solves the problem with thread sanitizer builds on macOS that doesn't have pthread barrier.
85 lines
2.3 KiB
C
85 lines
2.3 KiB
C
/*
|
|
* Copyright (C) Internet Systems Consortium, Inc. ("ISC")
|
|
*
|
|
* SPDX-License-Identifier: MPL-2.0
|
|
*
|
|
* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, you can obtain one at https://mozilla.org/MPL/2.0/.
|
|
*
|
|
* See the COPYRIGHT file distributed with this work for additional
|
|
* information regarding copyright ownership.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <isc/util.h>
|
|
|
|
#if HAVE_PTHREAD_BARRIER_INIT
|
|
|
|
#include <pthread.h>
|
|
|
|
#if ISC_TRACK_PTHREADS_OBJECTS
|
|
typedef pthread_barrier_t *isc_barrier_t;
|
|
#else
|
|
typedef pthread_barrier_t isc_barrier_t;
|
|
#endif
|
|
|
|
#define isc__barrier_init(bp, count) \
|
|
{ \
|
|
int _ret = pthread_barrier_init(bp, NULL, count); \
|
|
PTHREADS_RUNTIME_CHECK(pthread_barrier_init, _ret); \
|
|
}
|
|
|
|
#define isc__barrier_wait(bp) pthread_barrier_wait(bp)
|
|
|
|
#define isc__barrier_destroy(bp) \
|
|
{ \
|
|
int _ret = pthread_barrier_destroy(bp); \
|
|
PTHREADS_RUNTIME_CHECK(pthread_barrier_destroy, _ret); \
|
|
}
|
|
|
|
#else
|
|
|
|
#include <uv.h>
|
|
|
|
#if ISC_TRACK_PTHREADS_OBJECTS
|
|
typedef uv_barrier_t *isc_barrier_t;
|
|
#else
|
|
typedef uv_barrier_t isc_barrier_t;
|
|
#endif
|
|
|
|
#define isc__barrier_init(bp, count) \
|
|
{ \
|
|
int _ret = uv_barrier_init(bp, count); \
|
|
UV_RUNTIME_CHECK(uv_barrier_init, _ret); \
|
|
}
|
|
|
|
#define isc__barrier_wait(bp) uv_barrier_wait(bp)
|
|
|
|
#define isc__barrier_destroy(bp) uv_barrier_destroy(bp)
|
|
|
|
#endif
|
|
|
|
#if ISC_TRACK_PTHREADS_OBJECTS
|
|
|
|
#define isc_barrier_init(bp, count) \
|
|
{ \
|
|
*bp = malloc(sizeof(**bp)); \
|
|
isc__barrier_init(*bp, count); \
|
|
}
|
|
#define isc_barrier_wait(bp) isc__barrier_wait(*bp)
|
|
#define isc_barrier_destroy(bp) \
|
|
{ \
|
|
isc__barrier_destroy(*bp); \
|
|
free(*bp); \
|
|
}
|
|
|
|
#else /* ISC_TRACK_PTHREADS_OBJECTS */
|
|
|
|
#define isc_barrier_init(bp, count) isc__barrier_init(bp, count)
|
|
#define isc_barrier_wait(bp) isc__barrier_wait(bp)
|
|
#define isc_barrier_destroy(bp) isc__barrier_destroy(bp)
|
|
|
|
#endif /* ISC_TRACK_PTHREADS_OBJECTS */
|