1 | /* |
2 | * librdkafka - Apache Kafka C library |
3 | * |
4 | * Copyright (c) 2018 Magnus Edenhill |
5 | * All rights reserved. |
6 | * |
7 | * Redistribution and use in source and binary forms, with or without |
8 | * modification, are permitted provided that the following conditions are met: |
9 | * |
10 | * 1. Redistributions of source code must retain the above copyright notice, |
11 | * this list of conditions and the following disclaimer. |
12 | * 2. Redistributions in binary form must reproduce the above copyright notice, |
13 | * this list of conditions and the following disclaimer in the documentation |
14 | * and/or other materials provided with the distribution. |
15 | * |
16 | * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
17 | * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
18 | * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
19 | * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE |
20 | * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR |
21 | * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF |
22 | * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS |
23 | * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN |
24 | * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) |
25 | * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE |
26 | * POSSIBILITY OF SUCH DAMAGE. |
27 | */ |
28 | |
29 | |
30 | /** |
31 | * @brief Extra methods added to tinychtread/c11threads |
32 | */ |
33 | |
34 | |
35 | #ifndef _TINYCTHREAD_EXTRA_H_ |
36 | #define |
37 | |
38 | |
39 | #ifndef _MSC_VER |
40 | #include <pthread.h> /* needed for rwlock_t */ |
41 | #endif |
42 | |
43 | |
44 | /** |
45 | * @brief Set thread system name if platform supports it (pthreads) |
46 | * @return thrd_success or thrd_error |
47 | */ |
48 | int thrd_setname (const char *name); |
49 | |
50 | /** |
51 | * @brief Checks if passed thread is the current thread. |
52 | * @return non-zero if same thread, else 0. |
53 | */ |
54 | int thrd_is_current(thrd_t thr); |
55 | |
56 | |
57 | |
58 | |
59 | /** |
60 | * @brief Same as cnd_timedwait() but takes a relative timeout in milliseconds. |
61 | */ |
62 | int cnd_timedwait_ms(cnd_t *cnd, mtx_t *mtx, int timeout_ms); |
63 | |
64 | /** |
65 | * @brief Same as cnd_timedwait_ms() but updates the remaining time. |
66 | */ |
67 | int cnd_timedwait_msp (cnd_t *cnd, mtx_t *mtx, int *timeout_msp); |
68 | |
69 | /** |
70 | * @brief Same as cnd_timedwait() but honours |
71 | * RD_POLL_INFINITE (uses cnd_wait()), |
72 | * and RD_POLL_NOWAIT (return thrd_timedout immediately). |
73 | * |
74 | * @remark Set up \p tspec with rd_timeout_init_timespec(). |
75 | */ |
76 | int cnd_timedwait_abs (cnd_t *cnd, mtx_t *mtx, const struct timespec *tspec); |
77 | |
78 | |
79 | |
80 | |
81 | /** |
82 | * @brief Read-write locks |
83 | */ |
84 | |
85 | #if defined(_TTHREAD_WIN32_) |
86 | typedef struct rwlock_t { |
87 | SRWLOCK lock; |
88 | int rcnt; |
89 | int wcnt; |
90 | } rwlock_t; |
91 | #define rwlock_init(rwl) do { (rwl)->rcnt = (rwl)->wcnt = 0; InitializeSRWLock(&(rwl)->lock); } while (0) |
92 | #define rwlock_destroy(rwl) |
93 | #define rwlock_rdlock(rwl) do { if (0) printf("Thr %i: at %i: RDLOCK %p %s (%i, %i)\n", GetCurrentThreadId(), __LINE__, rwl, __FUNCTION__, (rwl)->rcnt, (rwl)->wcnt); assert((rwl)->rcnt >= 0 && (rwl)->wcnt >= 0); AcquireSRWLockShared(&(rwl)->lock); InterlockedIncrement(&(rwl)->rcnt); } while (0) |
94 | #define rwlock_wrlock(rwl) do { if (0) printf("Thr %i: at %i: WRLOCK %p %s (%i, %i)\n", GetCurrentThreadId(), __LINE__, rwl, __FUNCTION__, (rwl)->rcnt, (rwl)->wcnt); assert((rwl)->rcnt >= 0 && (rwl)->wcnt >= 0); AcquireSRWLockExclusive(&(rwl)->lock); InterlockedIncrement(&(rwl)->wcnt); } while (0) |
95 | #define rwlock_rdunlock(rwl) do { if (0) printf("Thr %i: at %i: RDUNLOCK %p %s (%i, %i)\n", GetCurrentThreadId(), __LINE__, rwl, __FUNCTION__, (rwl)->rcnt, (rwl)->wcnt); assert((rwl)->rcnt > 0 && (rwl)->wcnt >= 0); ReleaseSRWLockShared(&(rwl)->lock); InterlockedDecrement(&(rwl)->rcnt); } while (0) |
96 | #define rwlock_wrunlock(rwl) do { if (0) printf("Thr %i: at %i: RWUNLOCK %p %s (%i, %i)\n", GetCurrentThreadId(), __LINE__, rwl, __FUNCTION__, (rwl)->rcnt, (rwl)->wcnt); assert((rwl)->rcnt >= 0 && (rwl)->wcnt > 0); ReleaseSRWLockExclusive(&(rwl)->lock); InterlockedDecrement(&(rwl)->wcnt); } while (0) |
97 | |
98 | #define rwlock_rdlock_d(rwl) do { if (1) printf("Thr %i: at %i: RDLOCK %p %s (%i, %i)\n", GetCurrentThreadId(), __LINE__, rwl, __FUNCTION__, (rwl)->rcnt, (rwl)->wcnt); assert((rwl)->rcnt >= 0 && (rwl)->wcnt >= 0); AcquireSRWLockShared(&(rwl)->lock); InterlockedIncrement(&(rwl)->rcnt); } while (0) |
99 | #define rwlock_wrlock_d(rwl) do { if (1) printf("Thr %i: at %i: WRLOCK %p %s (%i, %i)\n", GetCurrentThreadId(), __LINE__, rwl, __FUNCTION__, (rwl)->rcnt, (rwl)->wcnt); assert((rwl)->rcnt >= 0 && (rwl)->wcnt >= 0); AcquireSRWLockExclusive(&(rwl)->lock); InterlockedIncrement(&(rwl)->wcnt); } while (0) |
100 | #define rwlock_rdunlock_d(rwl) do { if (1) printf("Thr %i: at %i: RDUNLOCK %p %s (%i, %i)\n", GetCurrentThreadId(), __LINE__, rwl, __FUNCTION__, (rwl)->rcnt, (rwl)->wcnt); assert((rwl)->rcnt > 0 && (rwl)->wcnt >= 0); ReleaseSRWLockShared(&(rwl)->lock); InterlockedDecrement(&(rwl)->rcnt); } while (0) |
101 | #define rwlock_wrunlock_d(rwl) do { if (1) printf("Thr %i: at %i: RWUNLOCK %p %s (%i, %i)\n", GetCurrentThreadId(), __LINE__, rwl, __FUNCTION__, (rwl)->rcnt, (rwl)->wcnt); assert((rwl)->rcnt >= 0 && (rwl)->wcnt > 0); ReleaseSRWLockExclusive(&(rwl)->lock); InterlockedDecrement(&(rwl)->wcnt); } while (0) |
102 | |
103 | |
104 | #else |
105 | typedef pthread_rwlock_t rwlock_t; |
106 | |
107 | int rwlock_init (rwlock_t *rwl); |
108 | int rwlock_destroy (rwlock_t *rwl); |
109 | int rwlock_rdlock (rwlock_t *rwl); |
110 | int rwlock_wrlock (rwlock_t *rwl); |
111 | int rwlock_rdunlock (rwlock_t *rwl); |
112 | int rwlock_wrunlock (rwlock_t *rwl); |
113 | |
114 | #endif |
115 | |
116 | |
117 | #endif /* _TINYCTHREAD_EXTRA_H_ */ |
118 | |