test: insomnia hack

Author: ohaiibuzzle

PlayTools/PlayLoader.m

@@ -230,11 +230,58 @@ static int pt_unlink(char const* path) {
     return unlink(ue_fix_filename(path));
 }
 
+static NSMutableDictionary *thread_sleep_counters = nil;
+static NSMutableDictionary *last_sleep_attempts = nil;
+static dispatch_once_t thread_sleep_once;
+
+static int pt_usleep(useconds_t time) {
+  dispatch_once(&thread_sleep_once, ^{
+    thread_sleep_counters = [NSMutableDictionary dictionary];
+    last_sleep_attempts = [NSMutableDictionary dictionary];
+  });
+
+  int thread_id = pthread_mach_thread_np(pthread_self());
+  NSNumber *threadKey = @(thread_id);
+
+  int thread_sleep_counter = [thread_sleep_counters[threadKey] intValue];
+  int last_sleep_attempt = [last_sleep_attempts[threadKey] intValue];
+
+  NSLog(@"[PC] Thread %i called usleep(%u) - count %i", thread_id, time,
+        thread_sleep_counter);
+
+  if (time == 100000) {
+    int timestamp = (int)[[NSDate date] timeIntervalSince1970];
+    // If it sleeps too fast, increase counter
+    if (timestamp - last_sleep_attempt < 2) {
+      thread_sleep_counter++;
+    } else {
+      thread_sleep_counter = 0;
+    }
+    last_sleep_attempt = timestamp;
+    thread_sleep_counters[threadKey] = @(thread_sleep_counter);
+    last_sleep_attempts[threadKey] = @(last_sleep_attempt);
+  }
+
+  if (thread_sleep_counter > 100) {
+    // Stop this thread from spamming usleep calls
+    NSLog(@"[PC] Thread %i exceeded usleep limit. Seem sus, stopping this "
+          @"thread FOREVER",
+          thread_id);
+    while (1) {
+      [NSThread sleepForTimeInterval:1000];
+    }
+    return 0;
+  }
+
+  return usleep(time);
+}
+
 DYLD_INTERPOSE(pt_open, open)
 DYLD_INTERPOSE(pt_stat, stat)
 DYLD_INTERPOSE(pt_access, access)
 DYLD_INTERPOSE(pt_rename, rename)
 DYLD_INTERPOSE(pt_unlink, unlink)
+DYLD_INTERPOSE(pt_usleep, usleep)
 
 @implementation PlayLoader