X-Git-Url: http://nsz.repo.hu/git/?a=blobdiff_plain;f=src%2Fthread%2F__timedwait.c;h=c9ec70cf6718f1296ccd336166a62a8ed2d9e79d;hb=b72cd07f176b876aa51864d93aa8101477b1d732;hp=931675eb74ed10e1781405dccf3e240864112810;hpb=ec381af90208a5675183202cf05078eac208edfb;p=musl

diff --git a/src/thread/__timedwait.c b/src/thread/__timedwait.c
index 931675eb..c9ec70cf 100644
--- a/src/thread/__timedwait.c
+++ b/src/thread/__timedwait.c
@@ -4,45 +4,39 @@
 #include "futex.h"
 #include "syscall.h"
 
-static int do_wait(volatile int *addr, int val, clockid_t clk, const struct timespec *at, int cp, int priv)
+int __pthread_setcancelstate(int, int *);
+int __clock_gettime(clockid_t, struct timespec *);
+
+int __timedwait(volatile int *addr, int val,
+	clockid_t clk, const struct timespec *at,
+	void (*cleanup)(void *), void *arg, int priv)
 {
-	int r, flag = 0;
+	int r, cs;
 	struct timespec to, *top=0;
 
-	if (!at) goto notimeout;
-	if (at->tv_nsec >= 1000000000UL)
-		return EINVAL;
-	if (clk == CLOCK_REALTIME || clk == CLOCK_MONOTONIC) {
-		if (clk == CLOCK_REALTIME) flag = FUTEX_CLOCK_REALTIME;
-		if (cp) r = -__syscall_cp(SYS_futex, addr, FUTEX_WAIT_BITSET|flag, val, at, 0, -1);
-		else r = -__syscall(SYS_futex, addr, FUTEX_WAIT_BITSET|flag, val, at, 0, -1);
-		if (r != ENOSYS) goto done;
-	}
-	if (clock_gettime(clk, &to)) return EINVAL;
-	to.tv_sec = at->tv_sec - to.tv_sec;
-	if ((to.tv_nsec = at->tv_nsec - to.tv_nsec) < 0) {
-		to.tv_sec--;
-		to.tv_nsec += 1000000000;
-	}
-	if (to.tv_sec < 0) return ETIMEDOUT;
-	top = &to;
-notimeout:
-	if (cp) r = -__syscall_cp(SYS_futex, addr, FUTEX_WAIT, val, top);
-	else r = -__syscall(SYS_futex, addr, FUTEX_WAIT, val, top);
-done:
-	if (r == EINTR || r == EINVAL || r == ETIMEDOUT) return r;
-	return 0;
-}
+	if (priv) priv = 128;
 
-int __timedwait(volatile int *addr, int val, clockid_t clk, const struct timespec *at, void (*cleanup)(void *), void *arg, int priv)
-{
-	int r;
-	if (cleanup) {
-		pthread_cleanup_push(cleanup, arg);
-		r = do_wait(addr, val, clk, at, 1, priv);
-		pthread_cleanup_pop(0);
-	} else {
-		r = do_wait(addr, val, clk, at, 0, priv);
+	if (at) {
+		if (at->tv_nsec >= 1000000000UL) return EINVAL;
+		if (__clock_gettime(clk, &to)) return EINVAL;
+		to.tv_sec = at->tv_sec - to.tv_sec;
+		if ((to.tv_nsec = at->tv_nsec - to.tv_nsec) < 0) {
+			to.tv_sec--;
+			to.tv_nsec += 1000000000;
+		}
+		if (to.tv_sec < 0) return ETIMEDOUT;
+		top = &to;
 	}
+
+	if (!cleanup) __pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, &cs);
+	pthread_cleanup_push(cleanup, arg);
+
+	r = -__syscall_cp(SYS_futex, addr, FUTEX_WAIT|priv, val, top);
+	if (r == ENOSYS) r = -__syscall_cp(SYS_futex, addr, FUTEX_WAIT, val, top);
+	if (r != EINTR && r != ETIMEDOUT) r = 0;
+
+	pthread_cleanup_pop(0);
+	if (!cleanup) __pthread_setcancelstate(cs, 0);
+
 	return r;
 }