]> git.saurik.com Git - apple/libc.git/blobdiff - gen/nanosleep.c
Libc-498.1.7.tar.gz
[apple/libc.git] / gen / nanosleep.c
index 717f30cd44a2095ab307ff06c20b0041c92e28f2..ae732f82f67697a9739357a8584e5d472e7f16c9 100644 (file)
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 1999 Apple Computer, Inc. All rights reserved.
+ * Copyright (c) 1999, 2003, 2006, 2007 Apple Inc. All rights reserved.
  *
  * @APPLE_LICENSE_HEADER_START@
  * 
 
 #include <errno.h>
 #include <sys/time.h>
-#include <mach/message.h>
 #include <mach/mach_error.h>
-#include <mach/mach_syscalls.h>
-#include <mach/clock.h>
-#include <mach/clock_types.h>
+#include <mach/mach_time.h>
 #include <stdio.h>
 
+
+#if __DARWIN_UNIX03
+#include "pthread_internals.h"
+#include <mach/clock.h>
+
+extern int __unix_conforming;
 extern mach_port_t clock_port;
+extern semaphore_t clock_sem;
+#ifdef VARIANT_CANCELABLE
+extern void _pthread_testcancel(pthread_t thread, int isconforming);
+extern int __semwait_signal(int cond_sem, int mutex_sem, int timeout, int relative, time_t tv_sec, __int32_t tv_nsec);
+#define        SEMWAIT_SIGNAL  __semwait_signal
+#else /* !VARIANT_CANCELABLE */
+extern int __semwait_signal_nocancel(int cond_sem, int mutex_sem, int timeout, int relative, time_t tv_sec, __int32_t tv_nsec);
+#define        SEMWAIT_SIGNAL  __semwait_signal_nocancel
+#endif /* VARIANT_CANCELABLE */
 
 int
 nanosleep(const struct timespec *requested_time, struct timespec *remaining_time) {
-    kern_return_t ret;
+    kern_return_t kret;
+    int ret;
     mach_timespec_t remain;
     mach_timespec_t current;
-    
-    if ((requested_time == NULL) || (requested_time->tv_sec < 0) || (requested_time->tv_nsec > NSEC_PER_SEC)) {
+   
+       if (__unix_conforming == 0)
+               __unix_conforming = 1;
+
+#ifdef VARIANT_CANCELABLE
+    _pthread_testcancel(pthread_self(), 1);
+#endif /* VARIANT_CANCELABLE */
+
+    if ((requested_time == NULL) || (requested_time->tv_sec < 0) || (requested_time->tv_nsec >= NSEC_PER_SEC)) {
         errno = EINVAL;
         return -1;
     }
 
-    ret = clock_get_time(clock_port, &current);
-    if (ret != KERN_SUCCESS) {
-        fprintf(stderr, "clock_get_time() failed: %s\n", mach_error_string(ret));
-        return -1;
+
+    if (remaining_time != NULL) {
+        kret = clock_get_time(clock_port, &current);
+        if (kret != KERN_SUCCESS) {
+            fprintf(stderr, "clock_get_time() failed: %s\n", mach_error_string(kret));
+            return -1;
+        }
     }
-    /* This depends on the layout of a mach_timespec_t and timespec_t being equivalent */
-    ret = clock_sleep_trap(clock_port, TIME_RELATIVE, requested_time->tv_sec, requested_time->tv_nsec, &remain);
-    if (ret != KERN_SUCCESS) {
-        if (ret == KERN_ABORTED) {
-            errno = EINTR;
+    ret = SEMWAIT_SIGNAL(clock_sem, MACH_PORT_NULL, 1, 1, requested_time->tv_sec, requested_time->tv_nsec);
+    if (ret < 0) {
+        if (errno == ETIMEDOUT) {
+               return 0;
+        } else if (errno == EINTR) {
             if (remaining_time != NULL) {
                 ret = clock_get_time(clock_port, &remain);
                 if (ret != KERN_SUCCESS) {
                     fprintf(stderr, "clock_get_time() failed: %s\n", mach_error_string(ret));
                     return -1;
                 }
+                /* This depends on the layout of a mach_timespec_t and timespec_t being equivalent */
                 ADD_MACH_TIMESPEC(&current, requested_time);
                 SUB_MACH_TIMESPEC(&current, &remain);
                 remaining_time->tv_sec = current.tv_sec;
@@ -66,8 +90,187 @@ nanosleep(const struct timespec *requested_time, struct timespec *remaining_time
             }
         } else {
             errno = EINVAL;
+       }
+    }
+    return -1;
+}
+
+
+#else /* !__DARWIN_UNIX03 */
+
+typedef struct {
+    uint64_t high;
+    uint64_t low;
+} uint128_t;
+
+/* 128-bit addition: acc += add */
+static inline void
+add128_128(uint128_t *acc, uint128_t *add)
+{
+    acc->high += add->high;
+    acc->low += add->low;
+    if(acc->low < add->low)
+       acc->high++; // carry
+}
+
+/* 128-bit subtraction: acc -= sub */
+static inline void
+sub128_128(uint128_t *acc, uint128_t *sub)
+{
+    acc->high -= sub->high;
+    if(acc->low < sub->low)
+       acc->high--; // borrow
+    acc->low -= sub->low;
+}
+
+#define TWO64  (((double)(1ULL << 32)) * ((double)(1ULL << 32)))
+
+static inline double
+uint128_double(uint128_t *u)
+{
+    return TWO64 * u->high + u->low; // may loses precision
+}
+
+/* 64x64 -> 128 bit multiplication */
+static inline void
+mul64x64(uint64_t x, uint64_t y, uint128_t *prod)
+{
+    uint128_t add;
+    /*
+     * Split the two 64-bit multiplicands into 32-bit parts:
+     * x => 2^32 * x1 + x2
+     * y => 2^32 * y1 + y2
+     */
+    uint32_t x1 = (uint32_t)(x >> 32);
+    uint32_t x2 = (uint32_t)x;
+    uint32_t y1 = (uint32_t)(y >> 32);
+    uint32_t y2 = (uint32_t)y;
+    /*
+     * direct multiplication:
+     * x * y => 2^64 * (x1 * y1) + 2^32 (x1 * y2 + x2 * y1) + (x2 * y2)
+     * The first and last terms are direct assignmenet into the uint128_t
+     * structure.  Then we add the middle two terms separately, to avoid
+     * 64-bit overflow.  (We could use the Karatsuba algorithm to save
+     * one multiply, but it is harder to deal with 64-bit overflows.)
+     */
+    prod->high = (uint64_t)x1 * (uint64_t)y1;
+    prod->low = (uint64_t)x2 * (uint64_t)y2;
+    add.low = (uint64_t)x1 * (uint64_t)y2;
+    add.high = (add.low >> 32);
+    add.low <<= 32;
+    add128_128(prod, &add);
+    add.low = (uint64_t)x2 * (uint64_t)y1;
+    add.high = (add.low >> 32);
+    add.low <<= 32;
+    add128_128(prod, &add);
+}
+
+/* calculate (x * y / divisor), using 128-bit internal calculations */
+static int
+muldiv128(uint64_t x, uint64_t y, uint64_t divisor, uint64_t *res)
+{
+    uint128_t temp;
+    uint128_t divisor128 = {0, divisor};
+    uint64_t result = 0;
+    double recip;
+
+    /* calculate (x * y) */
+    mul64x64(x, y, &temp);
+    /*
+     * Now divide by the divisor.  We use floating point to calculate an
+     * approximate answer and update the results.  Then we iterate and
+     * calculate a correction from the difference.
+     */
+    recip = 1.0 / ((double)divisor);
+    while(temp.high || temp.low >= divisor) {
+       uint128_t backmul;
+       uint64_t uapprox;
+       double approx = uint128_double(&temp) * recip;
+
+       if(approx > __LONG_LONG_MAX__)
+           return 0; // answer overflows 64-bits
+       uapprox = (uint64_t)approx;
+       mul64x64(uapprox, divisor, &backmul);
+       /*
+        * Because we are using unsigned integers, we need to approach the
+        * answer from the lesser side.  So if our estimate is too large
+        * we need to decrease it until it is smaller.
+        */
+       while(backmul.high > temp.high || backmul.high == temp.high && backmul.low > temp.low) {
+           sub128_128(&backmul, &divisor128);
+           uapprox--;
+       }
+       sub128_128(&temp, &backmul);
+       result += uapprox;
+    }
+    *res = result;
+    return 1;
+}
+
+int
+nanosleep(const struct timespec *requested_time, struct timespec *remaining_time) {
+    kern_return_t ret;
+    uint64_t end, units;
+    static struct mach_timebase_info info = {0, 0};
+    static int unity;
+    
+    if ((requested_time == NULL) || (requested_time->tv_sec < 0) || (requested_time->tv_nsec > NSEC_PER_SEC)) {
+        errno = EINVAL;
+        return -1;
+    }
+
+    if (info.denom == 0) {
+        ret = mach_timebase_info(&info);
+        if (ret != KERN_SUCCESS) {
+            fprintf(stderr, "mach_timebase_info() failed: %s\n", mach_error_string(ret));
+            errno = EAGAIN;
+            return -1;
+        }
+       /* If numer == denom == 1 (as in intel), no conversion needed */
+       unity = (info.numer == info.denom);
+    }
+
+    if(unity)
+       units = (uint64_t)requested_time->tv_sec * NSEC_PER_SEC;
+    else if(!muldiv128((uint64_t)info.denom * NSEC_PER_SEC,
+                      (uint64_t)requested_time->tv_sec,
+                      (uint64_t)info.numer,
+                      &units))
+    {
+       errno = EINVAL;
+       return -1;
+    }
+    end = mach_absolute_time()
+       + units
+       + (uint64_t)info.denom * requested_time->tv_nsec / info.numer;
+    ret = mach_wait_until(end);
+    if (ret != KERN_SUCCESS) {
+        if (ret == KERN_ABORTED) {
+            errno = EINTR;
+            if (remaining_time != NULL) {
+                uint64_t now = mach_absolute_time();
+                if (now >= end) {
+                   remaining_time->tv_sec = 0;
+                   remaining_time->tv_nsec = 0;
+               } else {
+                   if(unity)
+                       units = (end - now);
+                   else
+                       muldiv128((uint64_t)info.numer,
+                                 (end - now),
+                                 (uint64_t)info.denom,
+                                 &units); // this can't overflow
+                   remaining_time->tv_sec = units / NSEC_PER_SEC;
+                   remaining_time->tv_nsec = units % NSEC_PER_SEC;
+               }
+            }
+        } else {
+            errno = EINVAL;
         }
         return -1;
     }
     return 0;
 }
+
+
+#endif /* __DARWIN_UNIX03 */