|
|
|
@ -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 */ |