2 Copyright (C) 2006, 2007, 2008, 2009, 2010, 2011, 2012, 2013, 2014, 2015,
4 Ben Kibbey <bjk@luxsci.net>
6 This file is part of pwmd.
8 Pwmd is free software: you can redistribute it and/or modify
9 it under the terms of the GNU General Public License as published by
10 the Free Software Foundation, either version 2 of the License, or
11 (at your option) any later version.
13 Pwmd is distributed in the hope that it will be useful,
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 GNU General Public License for more details.
18 You should have received a copy of the GNU General Public License
19 along with Pwmd. If not, see <http://www.gnu.org/licenses/>.
28 #ifdef HAVE_SYS_TIME_H
36 #ifdef HAVE_SYS_SELECT_H
37 #include <sys/select.h>
39 #include <sys/types.h>
43 #ifdef HAVE_SYS_FILE_H
56 #ifndef HAVE_PTHREAD_CANCEL
57 pthread_key_t signal_thread_key
;
58 #define TEST_CANCEL() do { \
59 int *cancel = (int *) pthread_getspecific (signal_thread_key); \
60 if (cancel && *cancel) \
61 pthread_exit (NULL); \
67 #define INIT_TIMESPEC(t, ts) do { \
68 long s = (t*100000000)/1000000000; \
69 long l = (t*100000000)%1000000000; \
70 clock_gettime(CLOCK_REALTIME, &ts); \
72 if (ts.tv_nsec + l >= 1000000000) { \
81 #define MUTEX_LOCK_DEBUG(m) do { \
82 log_write("%s(%i): %s(): LOCK %p", __FILE__, __LINE__, \
86 #define MUTEX_UNLOCK_DEBUG(m) do { \
87 log_write("%s(%i): %s(): UNLOCK %p", __FILE__, __LINE__, \
91 #define MUTEX_LOCK_DEBUG(m)
92 #define MUTEX_UNLOCK_DEBUG(m)
95 #define MUTEX_LOCK(m) do { \
96 MUTEX_LOCK_DEBUG(m); \
97 while (pthread_mutex_trylock (m) == EBUSY) { \
98 struct timeval tv = { 0, 3000 }; \
100 select (0, NULL, NULL, NULL, &tv); \
104 #define MUTEX_UNLOCK(m) do { \
105 MUTEX_UNLOCK_DEBUG(m); \
106 pthread_mutex_unlock(m); \
109 #define TIMED_LOCK_DURATION 100000
111 #define TRY_FLOCK(ctx, fd, type, rc) do { \
112 long ka = (long)config_get_integer ("global", "keepalive_interval");\
113 long to = config_get_long ("global", "lock_timeout"); \
114 for (long elapsed = 0; ; elapsed++) { \
115 struct timeval tv = { 0, TIMED_LOCK_DURATION }; \
116 int n = flock (fd, type|LOCK_NB); \
117 if (n == -1 && errno == EWOULDBLOCK) { \
118 if (to == -1 || elapsed >= to) { \
119 rc = GPG_ERR_LOCKED; \
122 select (0, NULL, NULL, NULL, &tv); \
123 if (ctx && ka && elapsed && !((elapsed+1*10) % (ka*10))) { \
124 rc = send_status (ctx, STATUS_KEEPALIVE, NULL); \
131 rc = gpg_error_from_errno (errno); \
137 #define TRY_FLOCK(ctx, fd, type, rc) do { \
142 #define MUTEX_TIMED_LOCK(ctx, m, timeout, rc) do { \
143 long ka = (long)config_get_integer ("global", "keepalive_interval");\
144 for (long elapsed = 0; ; elapsed++) { \
145 struct timeval tv = { 0, TIMED_LOCK_DURATION }; \
146 int n = pthread_mutex_trylock (m); \
148 if (timeout == -1 || elapsed >= timeout) { \
149 rc = GPG_ERR_LOCKED; \
152 select (0, NULL, NULL, NULL, &tv); \
153 if (ctx && ka && elapsed && !((elapsed+1*10) % (ka*10))) { \
154 rc = send_status (ctx, STATUS_KEEPALIVE, NULL); \
161 rc = gpg_error_from_errno (n); \
165 MUTEX_LOCK_DEBUG(m); \
171 #define MUTEX_TRYLOCK(ctx, m, rc, t) do { \
172 int err = pthread_mutex_trylock (m); \
174 if (err == EBUSY) { \
177 rc = send_status(ctx, STATUS_LOCKED, NULL); \
179 MUTEX_TIMED_LOCK(ctx, m, t, rc); \
181 rc = GPG_ERR_LOCKED; \
184 rc = GPG_ERR_LOCKED; \
187 rc = gpg_error_from_errno (err); \
189 MUTEX_LOCK_DEBUG(m); \
192 #define MUTEX_TIMED_RWLOCK(ctx, m, timeout, rc, w) do { \
193 long ka = (long)config_get_integer ("global", "keepalive_interval");\
194 for (long elapsed = 0; ; elapsed++) { \
195 struct timeval tv = { 0, TIMED_LOCK_DURATION }; \
198 err = pthread_rwlock_trywrlock(m); \
200 err = pthread_rwlock_tryrdlock(m); \
201 if (err == EBUSY) { \
202 if (timeout == -1 || elapsed >= timeout) { \
203 rc = GPG_ERR_LOCKED; \
206 select (0, NULL, NULL, NULL, &tv); \
207 if (ctx && ka && elapsed && !((elapsed+1*10) % (ka*10))) { \
208 rc = send_status (ctx, STATUS_KEEPALIVE, NULL); \
215 rc = gpg_error_from_errno (err); \
219 MUTEX_LOCK_DEBUG(m); \