|  | @@ -31,260 +31,8 @@
 | 
	
		
			
				|  |  |   *
 | 
	
		
			
				|  |  |   */
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -#include "src/core/iomgr/pollset_posix.h"
 | 
	
		
			
				|  |  | +#include <grpc/support/port_platform.h>
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -#include <errno.h>
 | 
	
		
			
				|  |  | -#include <poll.h>
 | 
	
		
			
				|  |  | -#include <stdlib.h>
 | 
	
		
			
				|  |  | -#include <string.h>
 | 
	
		
			
				|  |  | -#include <unistd.h>
 | 
	
		
			
				|  |  | +#ifdef GPR_WIN32
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -#include "src/core/iomgr/alarm_internal.h"
 | 
	
		
			
				|  |  | -#include "src/core/iomgr/fd_posix.h"
 | 
	
		
			
				|  |  | -#include "src/core/iomgr/iomgr_internal.h"
 | 
	
		
			
				|  |  | -#include "src/core/iomgr/socket_utils_posix.h"
 | 
	
		
			
				|  |  | -#include <grpc/support/alloc.h>
 | 
	
		
			
				|  |  | -#include <grpc/support/log.h>
 | 
	
		
			
				|  |  | -#include <grpc/support/thd.h>
 | 
	
		
			
				|  |  | -#include <grpc/support/useful.h>
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -static grpc_pollset g_backup_pollset;
 | 
	
		
			
				|  |  | -static int g_shutdown_backup_poller;
 | 
	
		
			
				|  |  | -static gpr_event g_backup_poller_done;
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -static void backup_poller(void *p) {
 | 
	
		
			
				|  |  | -  gpr_timespec delta = gpr_time_from_millis(100);
 | 
	
		
			
				|  |  | -  gpr_timespec last_poll = gpr_now();
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -  gpr_mu_lock(&g_backup_pollset.mu);
 | 
	
		
			
				|  |  | -  while (g_shutdown_backup_poller == 0) {
 | 
	
		
			
				|  |  | -    gpr_timespec next_poll = gpr_time_add(last_poll, delta);
 | 
	
		
			
				|  |  | -    grpc_pollset_work(&g_backup_pollset, next_poll);
 | 
	
		
			
				|  |  | -    gpr_mu_unlock(&g_backup_pollset.mu);
 | 
	
		
			
				|  |  | -    gpr_sleep_until(next_poll);
 | 
	
		
			
				|  |  | -    gpr_mu_lock(&g_backup_pollset.mu);
 | 
	
		
			
				|  |  | -    last_poll = next_poll;
 | 
	
		
			
				|  |  | -  }
 | 
	
		
			
				|  |  | -  gpr_mu_unlock(&g_backup_pollset.mu);
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -  gpr_event_set(&g_backup_poller_done, (void *)1);
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -void grpc_pollset_kick(grpc_pollset *p) {
 | 
	
		
			
				|  |  | -  if (!p->counter) return;
 | 
	
		
			
				|  |  | -  grpc_pollset_kick_kick(&p->kick_state);
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -void grpc_pollset_force_kick(grpc_pollset *p) { grpc_pollset_kick(p); }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -/* global state management */
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -grpc_pollset *grpc_backup_pollset(void) { return &g_backup_pollset; }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -void grpc_pollset_global_init(void) {
 | 
	
		
			
				|  |  | -  gpr_thd_id id;
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -  /* Initialize kick fd state */
 | 
	
		
			
				|  |  | -  grpc_pollset_kick_global_init();
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -  /* initialize the backup pollset */
 | 
	
		
			
				|  |  | -  grpc_pollset_init(&g_backup_pollset);
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -  /* start the backup poller thread */
 | 
	
		
			
				|  |  | -  g_shutdown_backup_poller = 0;
 | 
	
		
			
				|  |  | -  gpr_event_init(&g_backup_poller_done);
 | 
	
		
			
				|  |  | -  gpr_thd_new(&id, backup_poller, NULL, NULL);
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -void grpc_pollset_global_shutdown(void) {
 | 
	
		
			
				|  |  | -  /* terminate the backup poller thread */
 | 
	
		
			
				|  |  | -  gpr_mu_lock(&g_backup_pollset.mu);
 | 
	
		
			
				|  |  | -  g_shutdown_backup_poller = 1;
 | 
	
		
			
				|  |  | -  gpr_mu_unlock(&g_backup_pollset.mu);
 | 
	
		
			
				|  |  | -  gpr_event_wait(&g_backup_poller_done, gpr_inf_future);
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -  /* destroy the backup pollset */
 | 
	
		
			
				|  |  | -  grpc_pollset_destroy(&g_backup_pollset);
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -  /* destroy the kick pipes */
 | 
	
		
			
				|  |  | -  grpc_pollset_kick_global_destroy();
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -/* main interface */
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -static void become_empty_pollset(grpc_pollset *pollset);
 | 
	
		
			
				|  |  | -static void become_unary_pollset(grpc_pollset *pollset, grpc_fd *fd);
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -void grpc_pollset_init(grpc_pollset *pollset) {
 | 
	
		
			
				|  |  | -  gpr_mu_init(&pollset->mu);
 | 
	
		
			
				|  |  | -  gpr_cv_init(&pollset->cv);
 | 
	
		
			
				|  |  | -  grpc_pollset_kick_init(&pollset->kick_state);
 | 
	
		
			
				|  |  | -  become_empty_pollset(pollset);
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -void grpc_pollset_add_fd(grpc_pollset *pollset, grpc_fd *fd) {
 | 
	
		
			
				|  |  | -  gpr_mu_lock(&pollset->mu);
 | 
	
		
			
				|  |  | -  pollset->vtable->add_fd(pollset, fd);
 | 
	
		
			
				|  |  | -  gpr_cv_broadcast(&pollset->cv);
 | 
	
		
			
				|  |  | -  gpr_mu_unlock(&pollset->mu);
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -void grpc_pollset_del_fd(grpc_pollset *pollset, grpc_fd *fd) {
 | 
	
		
			
				|  |  | -  gpr_mu_lock(&pollset->mu);
 | 
	
		
			
				|  |  | -  pollset->vtable->del_fd(pollset, fd);
 | 
	
		
			
				|  |  | -  gpr_cv_broadcast(&pollset->cv);
 | 
	
		
			
				|  |  | -  gpr_mu_unlock(&pollset->mu);
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -int grpc_pollset_work(grpc_pollset *pollset, gpr_timespec deadline) {
 | 
	
		
			
				|  |  | -  /* pollset->mu already held */
 | 
	
		
			
				|  |  | -  gpr_timespec now;
 | 
	
		
			
				|  |  | -  now = gpr_now();
 | 
	
		
			
				|  |  | -  if (gpr_time_cmp(now, deadline) > 0) {
 | 
	
		
			
				|  |  | -    return 0;
 | 
	
		
			
				|  |  | -  }
 | 
	
		
			
				|  |  | -  if (grpc_maybe_call_delayed_callbacks(&pollset->mu, 1)) {
 | 
	
		
			
				|  |  | -    return 1;
 | 
	
		
			
				|  |  | -  }
 | 
	
		
			
				|  |  | -  if (grpc_alarm_check(&pollset->mu, now, &deadline)) {
 | 
	
		
			
				|  |  | -    return 1;
 | 
	
		
			
				|  |  | -  }
 | 
	
		
			
				|  |  | -  return pollset->vtable->maybe_work(pollset, deadline, now, 1);
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -void grpc_pollset_destroy(grpc_pollset *pollset) {
 | 
	
		
			
				|  |  | -  pollset->vtable->destroy(pollset);
 | 
	
		
			
				|  |  | -  grpc_pollset_kick_destroy(&pollset->kick_state);
 | 
	
		
			
				|  |  | -  gpr_mu_destroy(&pollset->mu);
 | 
	
		
			
				|  |  | -  gpr_cv_destroy(&pollset->cv);
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -/*
 | 
	
		
			
				|  |  | - * empty_pollset - a vtable that provides polling for NO file descriptors
 | 
	
		
			
				|  |  | - */
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -static void empty_pollset_add_fd(grpc_pollset *pollset, grpc_fd *fd) {
 | 
	
		
			
				|  |  | -  become_unary_pollset(pollset, fd);
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -static void empty_pollset_del_fd(grpc_pollset *pollset, grpc_fd *fd) {}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -static int empty_pollset_maybe_work(grpc_pollset *pollset,
 | 
	
		
			
				|  |  | -                                    gpr_timespec deadline, gpr_timespec now,
 | 
	
		
			
				|  |  | -                                    int allow_synchronous_callback) {
 | 
	
		
			
				|  |  | -  return 0;
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -static void empty_pollset_destroy(grpc_pollset *pollset) {}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -static const grpc_pollset_vtable empty_pollset = {
 | 
	
		
			
				|  |  | -    empty_pollset_add_fd, empty_pollset_del_fd, empty_pollset_maybe_work,
 | 
	
		
			
				|  |  | -    empty_pollset_destroy};
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -static void become_empty_pollset(grpc_pollset *pollset) {
 | 
	
		
			
				|  |  | -  pollset->vtable = &empty_pollset;
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -/*
 | 
	
		
			
				|  |  | - * unary_poll_pollset - a vtable that provides polling for one file descriptor
 | 
	
		
			
				|  |  | - *                      via poll()
 | 
	
		
			
				|  |  | - */
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -static void unary_poll_pollset_add_fd(grpc_pollset *pollset, grpc_fd *fd) {
 | 
	
		
			
				|  |  | -  grpc_fd *fds[2];
 | 
	
		
			
				|  |  | -  if (fd == pollset->data.ptr) return;
 | 
	
		
			
				|  |  | -  fds[0] = pollset->data.ptr;
 | 
	
		
			
				|  |  | -  fds[1] = fd;
 | 
	
		
			
				|  |  | -  grpc_platform_become_multipoller(pollset, fds, GPR_ARRAY_SIZE(fds));
 | 
	
		
			
				|  |  | -  grpc_fd_unref(fds[0]);
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -static void unary_poll_pollset_del_fd(grpc_pollset *pollset, grpc_fd *fd) {
 | 
	
		
			
				|  |  | -  if (fd == pollset->data.ptr) {
 | 
	
		
			
				|  |  | -    grpc_fd_unref(pollset->data.ptr);
 | 
	
		
			
				|  |  | -    become_empty_pollset(pollset);
 | 
	
		
			
				|  |  | -  }
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -static int unary_poll_pollset_maybe_work(grpc_pollset *pollset,
 | 
	
		
			
				|  |  | -                                         gpr_timespec deadline,
 | 
	
		
			
				|  |  | -                                         gpr_timespec now,
 | 
	
		
			
				|  |  | -                                         int allow_synchronous_callback) {
 | 
	
		
			
				|  |  | -  struct pollfd pfd[2];
 | 
	
		
			
				|  |  | -  grpc_fd *fd;
 | 
	
		
			
				|  |  | -  int timeout;
 | 
	
		
			
				|  |  | -  int r;
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -  if (pollset->counter) {
 | 
	
		
			
				|  |  | -    return 0;
 | 
	
		
			
				|  |  | -  }
 | 
	
		
			
				|  |  | -  fd = pollset->data.ptr;
 | 
	
		
			
				|  |  | -  if (grpc_fd_is_orphaned(fd)) {
 | 
	
		
			
				|  |  | -    grpc_fd_unref(fd);
 | 
	
		
			
				|  |  | -    become_empty_pollset(pollset);
 | 
	
		
			
				|  |  | -    return 0;
 | 
	
		
			
				|  |  | -  }
 | 
	
		
			
				|  |  | -  if (gpr_time_cmp(deadline, gpr_inf_future) == 0) {
 | 
	
		
			
				|  |  | -    timeout = -1;
 | 
	
		
			
				|  |  | -  } else {
 | 
	
		
			
				|  |  | -    timeout = gpr_time_to_millis(gpr_time_sub(deadline, now));
 | 
	
		
			
				|  |  | -    if (timeout <= 0) {
 | 
	
		
			
				|  |  | -      return 1;
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -  }
 | 
	
		
			
				|  |  | -  pfd[0].fd = grpc_pollset_kick_pre_poll(&pollset->kick_state);
 | 
	
		
			
				|  |  | -  if (pfd[0].fd < 0) {
 | 
	
		
			
				|  |  | -    /* Already kicked */
 | 
	
		
			
				|  |  | -    return 1;
 | 
	
		
			
				|  |  | -  }
 | 
	
		
			
				|  |  | -  pfd[0].events = POLLIN;
 | 
	
		
			
				|  |  | -  pfd[0].revents = 0;
 | 
	
		
			
				|  |  | -  pfd[1].fd = fd->fd;
 | 
	
		
			
				|  |  | -  pfd[1].events = grpc_fd_begin_poll(fd, pollset, POLLIN, POLLOUT);
 | 
	
		
			
				|  |  | -  pfd[1].revents = 0;
 | 
	
		
			
				|  |  | -  pollset->counter = 1;
 | 
	
		
			
				|  |  | -  gpr_mu_unlock(&pollset->mu);
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -  r = poll(pfd, GPR_ARRAY_SIZE(pfd), timeout);
 | 
	
		
			
				|  |  | -  if (r < 0) {
 | 
	
		
			
				|  |  | -    if (errno != EINTR) {
 | 
	
		
			
				|  |  | -      gpr_log(GPR_ERROR, "poll() failed: %s", strerror(errno));
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -  } else if (r == 0) {
 | 
	
		
			
				|  |  | -    /* do nothing */
 | 
	
		
			
				|  |  | -  } else {
 | 
	
		
			
				|  |  | -    if (pfd[0].revents & POLLIN) {
 | 
	
		
			
				|  |  | -      grpc_pollset_kick_consume(&pollset->kick_state);
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -    if (pfd[1].revents & POLLIN) {
 | 
	
		
			
				|  |  | -      grpc_fd_become_readable(fd, allow_synchronous_callback);
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -    if (pfd[1].revents & POLLOUT) {
 | 
	
		
			
				|  |  | -      grpc_fd_become_writable(fd, allow_synchronous_callback);
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -  }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -  grpc_pollset_kick_post_poll(&pollset->kick_state);
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -  gpr_mu_lock(&pollset->mu);
 | 
	
		
			
				|  |  | -  grpc_fd_end_poll(fd, pollset);
 | 
	
		
			
				|  |  | -  pollset->counter = 0;
 | 
	
		
			
				|  |  | -  gpr_cv_broadcast(&pollset->cv);
 | 
	
		
			
				|  |  | -  return 1;
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -static void unary_poll_pollset_destroy(grpc_pollset *pollset) {
 | 
	
		
			
				|  |  | -  GPR_ASSERT(pollset->counter == 0);
 | 
	
		
			
				|  |  | -  grpc_fd_unref(pollset->data.ptr);
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -static const grpc_pollset_vtable unary_poll_pollset = {
 | 
	
		
			
				|  |  | -    unary_poll_pollset_add_fd, unary_poll_pollset_del_fd,
 | 
	
		
			
				|  |  | -    unary_poll_pollset_maybe_work, unary_poll_pollset_destroy};
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -static void become_unary_pollset(grpc_pollset *pollset, grpc_fd *fd) {
 | 
	
		
			
				|  |  | -  pollset->vtable = &unary_poll_pollset;
 | 
	
		
			
				|  |  | -  pollset->counter = 0;
 | 
	
		
			
				|  |  | -  pollset->data.ptr = fd;
 | 
	
		
			
				|  |  | -  grpc_fd_ref(fd);
 | 
	
		
			
				|  |  | -}
 | 
	
		
			
				|  |  | +#endif /* GPR_WIN32 */
 |