Add tests for timeout on select with PROXY_TO_PTHREAD

Signed-off-by: Kohei Tokunaga <[email protected]>

Author: ktock

test/core/test_pthread_select_timeout.c

@@ -0,0 +1,129 @@
+#include <sys/select.h>
+#include <time.h>
+#include <assert.h>
+#include <stdio.h>
+#include <unistd.h>
+#include <pthread.h>
+#include <string.h>
+#include <sys/time.h>
+
+// Check if timeout works without fds
+void test_timeout_without_fds()
+{
+  struct timeval tv;
+  time_t before_select;
+
+  tv.tv_sec = 1;
+  tv.tv_usec = 0;
+  before_select = time(NULL);
+  assert(select(0, NULL, NULL, NULL, &tv) == 0);
+  assert(time(NULL) - before_select >= 1);
+}
+
+// Check if timeout works with fds without events
+void test_timeout_with_fds_without_events()
+{
+  struct timeval tv, begin, end;
+  fd_set readfds;
+  int pipe_a[2];
+
+  assert(pipe(pipe_a) == 0);
+
+  tv.tv_sec = 1;
+  tv.tv_usec = 0;
+  FD_ZERO(&readfds);
+  FD_SET(pipe_a[0], &readfds);
+  gettimeofday(&begin, NULL);
+  assert(select(pipe_a[0] + 1, &readfds, NULL, NULL, &tv) == 0);
+  gettimeofday(&end, NULL);
+  assert((end.tv_sec - begin.tv_sec) * 1000000 + end.tv_usec - begin.tv_usec > 1000000);
+
+  close(pipe_a[0]); close(pipe_a[1]);
+}
+
+int pipe_shared[2];
+
+void *wakeup_after_2s(void * arg)
+{
+  const char *t = "test\n";
+
+  sleep(2);
+  write(pipe_shared[1], t, strlen(t));
+
+  return NULL;
+}
+
+// Check if select can unblock on an event
+void test_unblock_select()
+{
+  struct timeval begin, end;
+  fd_set readfds;
+  int maxfd;
+  pthread_t tid;
+  int pipe_a[2];
+
+  assert(pipe(pipe_a) == 0);
+  assert(pipe(pipe_shared) == 0);
+
+  FD_ZERO(&readfds);
+  FD_SET(pipe_a[0], &readfds);
+  FD_SET(pipe_shared[0], &readfds);
+  maxfd = (pipe_a[0] > pipe_shared[0] ? pipe_a[0] : pipe_shared[0]);
+  assert(pthread_create(&tid, NULL, wakeup_after_2s, NULL) == 0);
+  gettimeofday(&begin, NULL);
+  assert(select(maxfd + 1, &readfds, NULL, NULL, NULL) == 1);
+  gettimeofday(&end, NULL);
+  assert(FD_ISSET(pipe_shared[0], &readfds));
+  assert((end.tv_sec - begin.tv_sec) * 1000000 + end.tv_usec - begin.tv_usec > 1000000);
+
+  pthread_join(tid, NULL);
+
+  close(pipe_a[0]); close(pipe_a[1]);
+  close(pipe_shared[0]); close(pipe_shared[1]);
+}
+
+// Check if select works with ready fds
+void test_ready_fds()
+{
+  struct timeval tv;
+  fd_set readfds;
+  int maxfd;
+  const char *t = "test\n";
+  int pipe_c[2];
+  int pipe_d[2];
+
+  assert(pipe(pipe_c) == 0);
+  assert(pipe(pipe_d) == 0);
+
+  write(pipe_c[1], t, strlen(t));
+  write(pipe_d[1], t, strlen(t));
+  maxfd = (pipe_c[0] > pipe_d[0] ? pipe_c[0] : pipe_d[0]);
+
+  tv.tv_sec = 0;
+  tv.tv_usec = 0;
+  FD_ZERO(&readfds);
+  FD_SET(pipe_c[0], &readfds);
+  FD_SET(pipe_d[0], &readfds);
+  assert(select(maxfd + 1, &readfds, NULL, NULL, &tv) == 2);
+  assert(FD_ISSET(pipe_c[0], &readfds));
+  assert(FD_ISSET(pipe_d[0], &readfds));
+
+  FD_ZERO(&readfds);
+  FD_SET(pipe_c[0], &readfds);
+  FD_SET(pipe_d[0], &readfds);
+  assert(select(maxfd + 1, &readfds, NULL, NULL, NULL) == 2);
+  assert(FD_ISSET(pipe_c[0], &readfds));
+  assert(FD_ISSET(pipe_d[0], &readfds));
+
+  close(pipe_c[0]); close(pipe_c[1]);
+  close(pipe_d[0]); close(pipe_d[1]);
+}
+
+int main()
+{
+  test_timeout_without_fds();
+  test_timeout_with_fds_without_events();
+  test_unblock_select();
+  test_ready_fds();
+  return 0;
+}

test/test_core.py

@@ -9680,6 +9680,9 @@ def test_externref_emjs(self, dynlink):
   def test_syscall_intercept(self):
     self.do_core_test('test_syscall_intercept.c')
 
+  def test_pthread_select_timeout(self):
+    self.do_runf('core/test_pthread_select_timeout.c', cflags=['-pthread', '-sPROXY_TO_PTHREAD=1', '-sEXIT_RUNTIME=1'])
+
   @also_without_bigint
   def test_jslib_i64_params(self):
     # Tests the defineI64Param and receiveI64ParamAsI53 helpers that are