blob: 0c692231f2bb439f8544d7cb2f2c9d985b5eccb2 [file] [log] [blame]
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001/*
2 *
Jan Tattermusch7897ae92017-06-07 22:57:36 +02003 * Copyright 2016 gRPC authors.
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07004 *
Jan Tattermusch7897ae92017-06-07 22:57:36 +02005 * Licensed under the Apache License, Version 2.0 (the "License");
6 * you may not use this file except in compliance with the License.
7 * You may obtain a copy of the License at
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07008 *
Jan Tattermusch7897ae92017-06-07 22:57:36 +02009 * http://www.apache.org/licenses/LICENSE-2.0
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -070010 *
Jan Tattermusch7897ae92017-06-07 22:57:36 +020011 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -070016 *
17 */
18
Alexander Polcyndb3e8982018-02-21 16:59:24 -080019#include <grpc/support/port_platform.h>
20
murgatroid9954070892016-08-08 17:01:18 -070021#include "src/core/lib/iomgr/port.h"
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -070022
Yash Tibrewal4e0fe522017-10-08 18:07:15 -070023#include <grpc/grpc_posix.h>
yang-gceb24752017-11-07 12:06:37 -080024#include <grpc/support/log.h>
Yash Tibrewal4e0fe522017-10-08 18:07:15 -070025
Sree Kuchibhotla76a07952016-06-22 15:09:06 -070026/* This polling engine is only relevant on linux kernels supporting epoll() */
Mehrdad Afsharifb669002018-01-17 15:37:56 -080027#ifdef GRPC_LINUX_EPOLL_CREATE1
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -070028
Craig Tiller4509c472017-04-27 19:05:13 +000029#include "src/core/lib/iomgr/ev_epollsig_linux.h"
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -070030
31#include <assert.h>
32#include <errno.h>
Craig Tiller20397792017-07-18 11:35:27 -070033#include <limits.h>
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -070034#include <poll.h>
Sree Kuchibhotla4998e302016-08-16 07:26:31 -070035#include <pthread.h>
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -070036#include <signal.h>
37#include <string.h>
38#include <sys/epoll.h>
39#include <sys/socket.h>
40#include <unistd.h>
41
42#include <grpc/support/alloc.h>
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -070043#include <grpc/support/string_util.h>
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -070044
Craig Tillerb4bb1cd2017-07-20 14:18:17 -070045#include "src/core/lib/debug/stats.h"
Vijay Paib6cf1232018-01-25 21:02:26 -080046#include "src/core/lib/gpr/tls.h"
Vijay Paid4d0a302018-01-25 13:24:03 -080047#include "src/core/lib/gpr/useful.h"
Mark D. Roth4f2b0fd2018-01-19 12:12:23 -080048#include "src/core/lib/gprpp/manual_constructor.h"
Craig Tiller6b7c1fb2017-07-19 15:45:03 -070049#include "src/core/lib/iomgr/block_annotate.h"
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -070050#include "src/core/lib/iomgr/ev_posix.h"
51#include "src/core/lib/iomgr/iomgr_internal.h"
Craig Tiller376887d2017-04-06 08:27:03 -070052#include "src/core/lib/iomgr/lockfree_event.h"
Craig Tiller185f6c92017-03-17 08:33:19 -070053#include "src/core/lib/iomgr/timer.h"
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -070054#include "src/core/lib/iomgr/wakeup_fd_posix.h"
55#include "src/core/lib/profiling/timers.h"
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -070056
Craig Tillerbaa14a92017-11-03 09:09:36 -070057#define GRPC_POLLSET_KICK_BROADCAST ((grpc_pollset_worker*)1)
Craig Tillere24b24d2017-04-06 16:05:45 -070058
Noah Eisenc384d812017-11-12 20:14:27 -080059#define GRPC_POLLING_TRACE(...) \
ncteisen9ffb1492017-11-10 14:00:49 -080060 if (grpc_polling_trace.enabled()) { \
Noah Eisenc384d812017-11-12 20:14:27 -080061 gpr_log(GPR_INFO, __VA_ARGS__); \
Sree Kuchibhotla1e776682016-06-28 14:09:26 -070062 }
63
Sree Kuchibhotlac7be7c62016-06-09 17:08:50 -070064static int grpc_wakeup_signal = -1;
65static bool is_grpc_wakeup_signal_initialized = false;
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -070066
Sree Kuchibhotlac7be7c62016-06-09 17:08:50 -070067/* Implements the function defined in grpc_posix.h. This function might be
68 * called before even calling grpc_init() to set either a different signal to
69 * use. If signum == -1, then the use of signals is disabled */
70void grpc_use_signal(int signum) {
71 grpc_wakeup_signal = signum;
72 is_grpc_wakeup_signal_initialized = true;
73
74 if (grpc_wakeup_signal < 0) {
75 gpr_log(GPR_INFO,
76 "Use of signals is disabled. Epoll engine will not be used");
77 } else {
78 gpr_log(GPR_INFO, "epoll engine will be using signal: %d",
79 grpc_wakeup_signal);
80 }
81}
82
83struct polling_island;
Sree Kuchibhotla5855c472016-06-08 12:56:56 -070084
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -080085typedef enum {
86 POLL_OBJ_FD,
87 POLL_OBJ_POLLSET,
88 POLL_OBJ_POLLSET_SET
89} poll_obj_type;
90
91typedef struct poll_obj {
ncteisene9cd8a82017-06-29 06:03:52 -040092#ifndef NDEBUG
Sree Kuchibhotlaa0749a62016-11-18 20:22:09 -080093 poll_obj_type obj_type;
94#endif
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -080095 gpr_mu mu;
Craig Tillerbaa14a92017-11-03 09:09:36 -070096 struct polling_island* pi;
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -080097} poll_obj;
98
Craig Tillerbaa14a92017-11-03 09:09:36 -070099const char* poll_obj_string(poll_obj_type po_type) {
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -0800100 switch (po_type) {
101 case POLL_OBJ_FD:
102 return "fd";
103 case POLL_OBJ_POLLSET:
104 return "pollset";
105 case POLL_OBJ_POLLSET_SET:
106 return "pollset_set";
107 }
108
109 GPR_UNREACHABLE_CODE(return "UNKNOWN");
110}
111
Craig Tillerbaa14a92017-11-03 09:09:36 -0700112 /*******************************************************************************
113 * Fd Declarations
114 */
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -0800115
Craig Tillerbaa14a92017-11-03 09:09:36 -0700116#define FD_FROM_PO(po) ((grpc_fd*)(po))
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -0800117
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700118struct grpc_fd {
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -0800119 poll_obj po;
120
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700121 int fd;
122 /* refst format:
Sree Kuchibhotla5098f912016-05-31 10:58:17 -0700123 bit 0 : 1=Active / 0=Orphaned
124 bits 1-n : refcount
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700125 Ref/Unref by two to avoid altering the orphaned bit */
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700126 gpr_atm refst;
127
Sree Kuchibhotla99983382017-02-12 17:03:27 -0800128 /* The fd is either closed or we relinquished control of it. In either
129 cases, this indicates that the 'fd' on this structure is no longer
130 valid */
Sree Kuchibhotla79a62332016-06-04 14:01:03 -0700131 bool orphaned;
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700132
Craig Tillerfbf61bb2017-11-08 11:50:14 -0800133 grpc_core::ManualConstructor<grpc_core::LockfreeEvent> read_closure;
134 grpc_core::ManualConstructor<grpc_core::LockfreeEvent> write_closure;
Yash Tibrewaladc733f2018-04-02 18:32:06 -0700135 grpc_core::ManualConstructor<grpc_core::LockfreeEvent> error_closure;
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700136
Craig Tillerbaa14a92017-11-03 09:09:36 -0700137 struct grpc_fd* freelist_next;
138 grpc_closure* on_done_closure;
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700139
Sree Kuchibhotla82d73412017-02-09 18:27:45 -0800140 /* The pollset that last noticed that the fd is readable. The actual type
141 * stored in this is (grpc_pollset *) */
142 gpr_atm read_notifier_pollset;
Sree Kuchibhotla5855c472016-06-08 12:56:56 -0700143
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700144 grpc_iomgr_object iomgr_object;
Yash Tibrewaladc733f2018-04-02 18:32:06 -0700145
146 /* Do we need to track EPOLLERR events separately? */
147 bool track_err;
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700148};
149
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700150/* Reference counting for fds */
ncteisend39010e2017-06-08 17:08:07 -0700151#ifndef NDEBUG
Craig Tillerbaa14a92017-11-03 09:09:36 -0700152static void fd_ref(grpc_fd* fd, const char* reason, const char* file, int line);
153static void fd_unref(grpc_fd* fd, const char* reason, const char* file,
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700154 int line);
155#define GRPC_FD_REF(fd, reason) fd_ref(fd, reason, __FILE__, __LINE__)
156#define GRPC_FD_UNREF(fd, reason) fd_unref(fd, reason, __FILE__, __LINE__)
157#else
Craig Tillerbaa14a92017-11-03 09:09:36 -0700158static void fd_ref(grpc_fd* fd);
159static void fd_unref(grpc_fd* fd);
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700160#define GRPC_FD_REF(fd, reason) fd_ref(fd)
161#define GRPC_FD_UNREF(fd, reason) fd_unref(fd)
162#endif
163
164static void fd_global_init(void);
165static void fd_global_shutdown(void);
166
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700167/*******************************************************************************
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700168 * Polling island Declarations
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700169 */
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700170
ncteisena1354852017-06-08 16:25:53 -0700171#ifndef NDEBUG
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700172
Sree Kuchibhotla6a295452016-06-23 15:53:10 -0700173#define PI_ADD_REF(p, r) pi_add_ref_dbg((p), (r), __FILE__, __LINE__)
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800174#define PI_UNREF(p, r) pi_unref_dbg((p), (r), __FILE__, __LINE__)
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700175
ncteisend39010e2017-06-08 17:08:07 -0700176#else
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700177
Sree Kuchibhotla6a295452016-06-23 15:53:10 -0700178#define PI_ADD_REF(p, r) pi_add_ref((p))
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800179#define PI_UNREF(p, r) pi_unref((p))
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700180
ncteisena1354852017-06-08 16:25:53 -0700181#endif
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700182
Craig Tiller460502e2016-10-13 10:02:08 -0700183/* This is also used as grpc_workqueue (by directly casing it) */
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700184typedef struct polling_island {
185 gpr_mu mu;
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700186 /* Ref count. Use PI_ADD_REF() and PI_UNREF() macros to increment/decrement
187 the refcount.
188 Once the ref count becomes zero, this structure is destroyed which means
189 we should ensure that there is never a scenario where a PI_ADD_REF() is
190 racing with a PI_UNREF() that just made the ref_count zero. */
Craig Tiller15007612016-07-06 09:36:16 -0700191 gpr_atm ref_count;
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700192
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700193 /* Pointer to the polling_island this merged into.
194 * merged_to value is only set once in polling_island's lifetime (and that too
195 * only if the island is merged with another island). Because of this, we can
196 * use gpr_atm type here so that we can do atomic access on this and reduce
197 * lock contention on 'mu' mutex.
198 *
199 * Note that if this field is not NULL (i.e not 0), all the remaining fields
200 * (except mu and ref_count) are invalid and must be ignored. */
201 gpr_atm merged_to;
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700202
Craig Tiller460502e2016-10-13 10:02:08 -0700203 /* Number of threads currently polling on this island */
Craig Tillerd8a3c042016-09-09 12:42:37 -0700204 gpr_atm poller_count;
Craig Tillerb39307d2016-06-30 15:39:13 -0700205
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700206 /* The fd of the underlying epoll set */
207 int epoll_fd;
208
209 /* The file descriptors in the epoll set */
210 size_t fd_cnt;
211 size_t fd_capacity;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700212 grpc_fd** fds;
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700213} polling_island;
214
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700215/*******************************************************************************
216 * Pollset Declarations
217 */
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700218struct grpc_pollset_worker {
Sree Kuchibhotla34217242016-06-29 00:19:07 -0700219 /* Thread id of this worker */
220 pthread_t pt_id;
221
222 /* Used to prevent a worker from getting kicked multiple times */
223 gpr_atm is_kicked;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700224 struct grpc_pollset_worker* next;
225 struct grpc_pollset_worker* prev;
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700226};
227
228struct grpc_pollset {
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -0800229 poll_obj po;
230
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700231 grpc_pollset_worker root_worker;
232 bool kicked_without_pollers;
233
234 bool shutting_down; /* Is the pollset shutting down ? */
235 bool finish_shutdown_called; /* Is the 'finish_shutdown_locked()' called ? */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700236 grpc_closure* shutdown_done; /* Called after after shutdown is complete */
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700237};
238
239/*******************************************************************************
240 * Pollset-set Declarations
241 */
242struct grpc_pollset_set {
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -0800243 poll_obj po;
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700244};
245
246/*******************************************************************************
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -0700247 * Common helpers
248 */
249
Craig Tillerbaa14a92017-11-03 09:09:36 -0700250static bool append_error(grpc_error** composite, grpc_error* error,
251 const char* desc) {
Craig Tillerf975f742016-07-01 14:56:27 -0700252 if (error == GRPC_ERROR_NONE) return true;
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -0700253 if (*composite == GRPC_ERROR_NONE) {
Noah Eisen3005ce82017-03-14 13:38:41 -0700254 *composite = GRPC_ERROR_CREATE_FROM_COPIED_STRING(desc);
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -0700255 }
256 *composite = grpc_error_add_child(*composite, error);
Craig Tillerf975f742016-07-01 14:56:27 -0700257 return false;
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -0700258}
259
260/*******************************************************************************
Sree Kuchibhotla24b10622016-06-08 15:20:17 -0700261 * Polling island Definitions
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700262 */
263
Sree Kuchibhotla24b10622016-06-08 15:20:17 -0700264/* The wakeup fd that is used to wake up all threads in a Polling island. This
265 is useful in the polling island merge operation where we need to wakeup all
266 the threads currently polling the smaller polling island (so that they can
267 start polling the new/merged polling island)
268
269 NOTE: This fd is initialized to be readable and MUST NOT be consumed i.e the
270 threads that woke up MUST NOT call grpc_wakeup_fd_consume_wakeup() */
271static grpc_wakeup_fd polling_island_wakeup_fd;
272
Craig Tiller2e620132016-10-10 15:27:44 -0700273/* The polling island being polled right now.
274 See comments in workqueue_maybe_wakeup for why this is tracked. */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700275static __thread polling_island* g_current_thread_polling_island;
Craig Tillerd8a3c042016-09-09 12:42:37 -0700276
Craig Tillerb39307d2016-06-30 15:39:13 -0700277/* Forward declaration */
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800278static void polling_island_delete(polling_island* pi);
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700279
Sree Kuchibhotlaad2c4772016-06-13 19:06:54 -0700280#ifdef GRPC_TSAN
Sree Kuchibhotla41622a82016-06-13 16:43:14 -0700281/* Currently TSAN may incorrectly flag data races between epoll_ctl and
282 epoll_wait for any grpc_fd structs that are added to the epoll set via
283 epoll_ctl and are returned (within a very short window) via epoll_wait().
284
285 To work-around this race, we establish a happens-before relation between
286 the code just-before epoll_ctl() and the code after epoll_wait() by using
287 this atomic */
288gpr_atm g_epoll_sync;
Sree Kuchibhotlaad2c4772016-06-13 19:06:54 -0700289#endif /* defined(GRPC_TSAN) */
Sree Kuchibhotla41622a82016-06-13 16:43:14 -0700290
Craig Tillerbaa14a92017-11-03 09:09:36 -0700291static void pi_add_ref(polling_island* pi);
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800292static void pi_unref(polling_island* pi);
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700293
ncteisena1354852017-06-08 16:25:53 -0700294#ifndef NDEBUG
Craig Tillerbaa14a92017-11-03 09:09:36 -0700295static void pi_add_ref_dbg(polling_island* pi, const char* reason,
296 const char* file, int line) {
ncteisen9ffb1492017-11-10 14:00:49 -0800297 if (grpc_polling_trace.enabled()) {
ncteisen3ac64f82017-06-19 17:35:44 -0700298 gpr_atm old_cnt = gpr_atm_acq_load(&pi->ref_count);
Craig Tillerbaa14a92017-11-03 09:09:36 -0700299 gpr_log(GPR_DEBUG,
300 "Add ref pi: %p, old:%" PRIdPTR " -> new:%" PRIdPTR
301 " (%s) - (%s, %d)",
ncteisen3ac64f82017-06-19 17:35:44 -0700302 pi, old_cnt, old_cnt + 1, reason, file, line);
ncteisena1354852017-06-08 16:25:53 -0700303 }
304 pi_add_ref(pi);
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700305}
306
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800307static void pi_unref_dbg(polling_island* pi, const char* reason,
308 const char* file, int line) {
ncteisen9ffb1492017-11-10 14:00:49 -0800309 if (grpc_polling_trace.enabled()) {
ncteisen3ac64f82017-06-19 17:35:44 -0700310 gpr_atm old_cnt = gpr_atm_acq_load(&pi->ref_count);
Craig Tillerbaa14a92017-11-03 09:09:36 -0700311 gpr_log(GPR_DEBUG,
312 "Unref pi: %p, old:%" PRIdPTR " -> new:%" PRIdPTR
313 " (%s) - (%s, %d)",
ncteisen3ac64f82017-06-19 17:35:44 -0700314 pi, old_cnt, (old_cnt - 1), reason, file, line);
ncteisena1354852017-06-08 16:25:53 -0700315 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800316 pi_unref(pi);
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700317}
318#endif
319
Craig Tillerbaa14a92017-11-03 09:09:36 -0700320static void pi_add_ref(polling_island* pi) {
Craig Tiller15007612016-07-06 09:36:16 -0700321 gpr_atm_no_barrier_fetch_add(&pi->ref_count, 1);
322}
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700323
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800324static void pi_unref(polling_island* pi) {
Craig Tillerd8a3c042016-09-09 12:42:37 -0700325 /* If ref count went to zero, delete the polling island.
Sree Kuchibhotla6a295452016-06-23 15:53:10 -0700326 Note that this deletion not be done under a lock. Once the ref count goes
327 to zero, we are guaranteed that no one else holds a reference to the
328 polling island (and that there is no racing pi_add_ref() call either).
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700329
330 Also, if we are deleting the polling island and the merged_to field is
331 non-empty, we should remove a ref to the merged_to polling island
332 */
Craig Tillerd8a3c042016-09-09 12:42:37 -0700333 if (1 == gpr_atm_full_fetch_add(&pi->ref_count, -1)) {
Craig Tillerbaa14a92017-11-03 09:09:36 -0700334 polling_island* next = (polling_island*)gpr_atm_acq_load(&pi->merged_to);
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800335 polling_island_delete(pi);
Craig Tiller4782d922017-11-10 09:53:21 -0800336 if (next != nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800337 PI_UNREF(next, "pi_delete"); /* Recursive call */
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700338 }
339 }
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700340}
341
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700342/* The caller is expected to hold pi->mu lock before calling this function */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700343static void polling_island_add_fds_locked(polling_island* pi, grpc_fd** fds,
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -0700344 size_t fd_count, bool add_fd_refs,
Craig Tillerbaa14a92017-11-03 09:09:36 -0700345 grpc_error** error) {
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700346 int err;
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700347 size_t i;
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700348 struct epoll_event ev;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700349 char* err_msg;
350 const char* err_desc = "polling_island_add_fds";
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700351
Sree Kuchibhotlaad2c4772016-06-13 19:06:54 -0700352#ifdef GRPC_TSAN
Sree Kuchibhotla41622a82016-06-13 16:43:14 -0700353 /* See the definition of g_epoll_sync for more context */
Sree Kuchibhotla0224dcc2016-06-22 18:04:00 -0700354 gpr_atm_rel_store(&g_epoll_sync, (gpr_atm)0);
Sree Kuchibhotlaad2c4772016-06-13 19:06:54 -0700355#endif /* defined(GRPC_TSAN) */
Sree Kuchibhotla41622a82016-06-13 16:43:14 -0700356
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700357 for (i = 0; i < fd_count; i++) {
Noah Eisenbe82e642018-02-09 09:16:55 -0800358 ev.events = static_cast<uint32_t>(EPOLLIN | EPOLLOUT | EPOLLET);
Yash Tibrewaladc733f2018-04-02 18:32:06 -0700359 /* Use the least significant bit of ev.data.ptr to store track_err to avoid
360 * synchronization issues when accessing it after receiving an event */
361 ev.data.ptr = reinterpret_cast<void*>(reinterpret_cast<intptr_t>(fds[i]) |
362 (fds[i]->track_err ? 1 : 0));
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700363 err = epoll_ctl(pi->epoll_fd, EPOLL_CTL_ADD, fds[i]->fd, &ev);
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700364
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700365 if (err < 0) {
366 if (errno != EEXIST) {
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -0700367 gpr_asprintf(
368 &err_msg,
369 "epoll_ctl (epoll_fd: %d) add fd: %d failed with error: %d (%s)",
370 pi->epoll_fd, fds[i]->fd, errno, strerror(errno));
371 append_error(error, GRPC_OS_ERROR(errno, err_msg), err_desc);
372 gpr_free(err_msg);
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700373 }
374
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700375 continue;
376 }
377
378 if (pi->fd_cnt == pi->fd_capacity) {
379 pi->fd_capacity = GPR_MAX(pi->fd_capacity + 8, pi->fd_cnt * 3 / 2);
Noah Eisen4d20a662018-02-09 09:34:04 -0800380 pi->fds = static_cast<grpc_fd**>(
381 gpr_realloc(pi->fds, sizeof(grpc_fd*) * pi->fd_capacity));
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700382 }
383
384 pi->fds[pi->fd_cnt++] = fds[i];
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700385 if (add_fd_refs) {
386 GRPC_FD_REF(fds[i], "polling_island");
387 }
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700388 }
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700389}
390
Sree Kuchibhotla24b10622016-06-08 15:20:17 -0700391/* The caller is expected to hold pi->mu before calling this */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700392static void polling_island_add_wakeup_fd_locked(polling_island* pi,
393 grpc_wakeup_fd* wakeup_fd,
394 grpc_error** error) {
Sree Kuchibhotla24b10622016-06-08 15:20:17 -0700395 struct epoll_event ev;
396 int err;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700397 char* err_msg;
398 const char* err_desc = "polling_island_add_wakeup_fd";
Sree Kuchibhotla24b10622016-06-08 15:20:17 -0700399
Noah Eisenbe82e642018-02-09 09:16:55 -0800400 ev.events = static_cast<uint32_t>(EPOLLIN | EPOLLET);
Sree Kuchibhotla24b10622016-06-08 15:20:17 -0700401 ev.data.ptr = wakeup_fd;
402 err = epoll_ctl(pi->epoll_fd, EPOLL_CTL_ADD,
403 GRPC_WAKEUP_FD_GET_READ_FD(wakeup_fd), &ev);
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -0700404 if (err < 0 && errno != EEXIST) {
405 gpr_asprintf(&err_msg,
406 "epoll_ctl (epoll_fd: %d) add wakeup fd: %d failed with "
407 "error: %d (%s)",
Craig Tillerc3571792017-05-02 12:33:38 -0700408 pi->epoll_fd, GRPC_WAKEUP_FD_GET_READ_FD(wakeup_fd), errno,
409 strerror(errno));
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -0700410 append_error(error, GRPC_OS_ERROR(errno, err_msg), err_desc);
411 gpr_free(err_msg);
Sree Kuchibhotla24b10622016-06-08 15:20:17 -0700412 }
413}
414
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700415/* The caller is expected to hold pi->mu lock before calling this function */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700416static void polling_island_remove_all_fds_locked(polling_island* pi,
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -0700417 bool remove_fd_refs,
Craig Tillerbaa14a92017-11-03 09:09:36 -0700418 grpc_error** error) {
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700419 int err;
420 size_t i;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700421 char* err_msg;
422 const char* err_desc = "polling_island_remove_fds";
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700423
424 for (i = 0; i < pi->fd_cnt; i++) {
Craig Tiller4782d922017-11-10 09:53:21 -0800425 err = epoll_ctl(pi->epoll_fd, EPOLL_CTL_DEL, pi->fds[i]->fd, nullptr);
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700426 if (err < 0 && errno != ENOENT) {
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -0700427 gpr_asprintf(&err_msg,
428 "epoll_ctl (epoll_fd: %d) delete fds[%zu]: %d failed with "
429 "error: %d (%s)",
430 pi->epoll_fd, i, pi->fds[i]->fd, errno, strerror(errno));
431 append_error(error, GRPC_OS_ERROR(errno, err_msg), err_desc);
432 gpr_free(err_msg);
Sree Kuchibhotlaad162ba2016-06-06 16:23:37 -0700433 }
434
435 if (remove_fd_refs) {
436 GRPC_FD_UNREF(pi->fds[i], "polling_island");
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700437 }
438 }
439
440 pi->fd_cnt = 0;
441}
442
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700443/* The caller is expected to hold pi->mu lock before calling this function */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700444static void polling_island_remove_fd_locked(polling_island* pi, grpc_fd* fd,
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -0700445 bool is_fd_closed,
Craig Tillerbaa14a92017-11-03 09:09:36 -0700446 grpc_error** error) {
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700447 int err;
448 size_t i;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700449 char* err_msg;
450 const char* err_desc = "polling_island_remove_fd";
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700451
Sree Kuchibhotla79a62332016-06-04 14:01:03 -0700452 /* If fd is already closed, then it would have been automatically been removed
453 from the epoll set */
454 if (!is_fd_closed) {
Craig Tiller4782d922017-11-10 09:53:21 -0800455 err = epoll_ctl(pi->epoll_fd, EPOLL_CTL_DEL, fd->fd, nullptr);
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700456 if (err < 0 && errno != ENOENT) {
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -0700457 gpr_asprintf(
458 &err_msg,
459 "epoll_ctl (epoll_fd: %d) del fd: %d failed with error: %d (%s)",
460 pi->epoll_fd, fd->fd, errno, strerror(errno));
461 append_error(error, GRPC_OS_ERROR(errno, err_msg), err_desc);
462 gpr_free(err_msg);
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700463 }
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700464 }
465
466 for (i = 0; i < pi->fd_cnt; i++) {
467 if (pi->fds[i] == fd) {
468 pi->fds[i] = pi->fds[--pi->fd_cnt];
Sree Kuchibhotla79a62332016-06-04 14:01:03 -0700469 GRPC_FD_UNREF(fd, "polling_island");
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700470 break;
471 }
472 }
473}
474
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -0700475/* Might return NULL in case of an error */
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800476static polling_island* polling_island_create(grpc_fd* initial_fd,
Craig Tillerbaa14a92017-11-03 09:09:36 -0700477 grpc_error** error) {
Craig Tiller4782d922017-11-10 09:53:21 -0800478 polling_island* pi = nullptr;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700479 const char* err_desc = "polling_island_create";
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700480
Craig Tillerb39307d2016-06-30 15:39:13 -0700481 *error = GRPC_ERROR_NONE;
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700482
Noah Eisenbe82e642018-02-09 09:16:55 -0800483 pi = static_cast<polling_island*>(gpr_malloc(sizeof(*pi)));
Craig Tillerb39307d2016-06-30 15:39:13 -0700484 gpr_mu_init(&pi->mu);
485 pi->fd_cnt = 0;
486 pi->fd_capacity = 0;
Craig Tiller4782d922017-11-10 09:53:21 -0800487 pi->fds = nullptr;
Craig Tillerb39307d2016-06-30 15:39:13 -0700488 pi->epoll_fd = -1;
Craig Tillerd8a3c042016-09-09 12:42:37 -0700489
Craig Tiller15007612016-07-06 09:36:16 -0700490 gpr_atm_rel_store(&pi->ref_count, 0);
Craig Tillerd8a3c042016-09-09 12:42:37 -0700491 gpr_atm_rel_store(&pi->poller_count, 0);
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800492 gpr_atm_rel_store(&pi->merged_to, (gpr_atm) nullptr);
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700493
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700494 pi->epoll_fd = epoll_create1(EPOLL_CLOEXEC);
Sree Kuchibhotla41622a82016-06-13 16:43:14 -0700495
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700496 if (pi->epoll_fd < 0) {
Craig Tillerb39307d2016-06-30 15:39:13 -0700497 append_error(error, GRPC_OS_ERROR(errno, "epoll_create1"), err_desc);
498 goto done;
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700499 }
500
Craig Tiller4782d922017-11-10 09:53:21 -0800501 if (initial_fd != nullptr) {
Craig Tillerb39307d2016-06-30 15:39:13 -0700502 polling_island_add_fds_locked(pi, &initial_fd, 1, true, error);
Craig Tillerb39307d2016-06-30 15:39:13 -0700503 }
504
Craig Tillerb39307d2016-06-30 15:39:13 -0700505done:
506 if (*error != GRPC_ERROR_NONE) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800507 polling_island_delete(pi);
Craig Tiller4782d922017-11-10 09:53:21 -0800508 pi = nullptr;
Craig Tillerb39307d2016-06-30 15:39:13 -0700509 }
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700510 return pi;
511}
512
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800513static void polling_island_delete(polling_island* pi) {
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700514 GPR_ASSERT(pi->fd_cnt == 0);
515
Craig Tiller0a06cd72016-07-14 13:21:24 -0700516 if (pi->epoll_fd >= 0) {
517 close(pi->epoll_fd);
518 }
Craig Tillerb39307d2016-06-30 15:39:13 -0700519 gpr_mu_destroy(&pi->mu);
520 gpr_free(pi->fds);
521 gpr_free(pi);
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700522}
523
Sree Kuchibhotla229533b12016-06-21 20:42:52 -0700524/* Attempts to gets the last polling island in the linked list (liked by the
525 * 'merged_to' field). Since this does not lock the polling island, there are no
526 * guarantees that the island returned is the last island */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700527static polling_island* polling_island_maybe_get_latest(polling_island* pi) {
528 polling_island* next = (polling_island*)gpr_atm_acq_load(&pi->merged_to);
Craig Tiller4782d922017-11-10 09:53:21 -0800529 while (next != nullptr) {
Sree Kuchibhotla229533b12016-06-21 20:42:52 -0700530 pi = next;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700531 next = (polling_island*)gpr_atm_acq_load(&pi->merged_to);
Sree Kuchibhotla229533b12016-06-21 20:42:52 -0700532 }
533
534 return pi;
535}
536
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700537/* Gets the lock on the *latest* polling island i.e the last polling island in
Sree Kuchibhotla229533b12016-06-21 20:42:52 -0700538 the linked list (linked by the 'merged_to' field). Call gpr_mu_unlock on the
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700539 returned polling island's mu.
540 Usage: To lock/unlock polling island "pi", do the following:
541 polling_island *pi_latest = polling_island_lock(pi);
542 ...
543 ... critical section ..
544 ...
Sree Kuchibhotla229533b12016-06-21 20:42:52 -0700545 gpr_mu_unlock(&pi_latest->mu); // NOTE: use pi_latest->mu. NOT pi->mu */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700546static polling_island* polling_island_lock(polling_island* pi) {
Craig Tiller4782d922017-11-10 09:53:21 -0800547 polling_island* next = nullptr;
Sree Kuchibhotla229533b12016-06-21 20:42:52 -0700548
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700549 while (true) {
Craig Tillerbaa14a92017-11-03 09:09:36 -0700550 next = (polling_island*)gpr_atm_acq_load(&pi->merged_to);
Craig Tiller4782d922017-11-10 09:53:21 -0800551 if (next == nullptr) {
Sree Kuchibhotla229533b12016-06-21 20:42:52 -0700552 /* Looks like 'pi' is the last node in the linked list but unless we check
553 this by holding the pi->mu lock, we cannot be sure (i.e without the
554 pi->mu lock, we don't prevent island merges).
555 To be absolutely sure, check once more by holding the pi->mu lock */
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700556 gpr_mu_lock(&pi->mu);
Craig Tillerbaa14a92017-11-03 09:09:36 -0700557 next = (polling_island*)gpr_atm_acq_load(&pi->merged_to);
Craig Tiller4782d922017-11-10 09:53:21 -0800558 if (next == nullptr) {
Sree Kuchibhotla229533b12016-06-21 20:42:52 -0700559 /* pi is infact the last node and we have the pi->mu lock. we're done */
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700560 break;
561 }
Sree Kuchibhotla229533b12016-06-21 20:42:52 -0700562
563 /* pi->merged_to is not NULL i.e pi isn't the last node anymore. pi->mu
564 * isn't the lock we are interested in. Continue traversing the list */
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700565 gpr_mu_unlock(&pi->mu);
566 }
567
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700568 pi = next;
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700569 }
570
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700571 return pi;
572}
573
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -0700574/* Gets the lock on the *latest* polling islands in the linked lists pointed by
575 *p and *q (and also updates *p and *q to point to the latest polling islands)
576
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700577 This function is needed because calling the following block of code to obtain
578 locks on polling islands (*p and *q) is prone to deadlocks.
579 {
Sree Kuchibhotla229533b12016-06-21 20:42:52 -0700580 polling_island_lock(*p, true);
581 polling_island_lock(*q, true);
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700582 }
583
Sree Kuchibhotla229533b12016-06-21 20:42:52 -0700584 Usage/example:
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700585 polling_island *p1;
586 polling_island *p2;
587 ..
588 polling_island_lock_pair(&p1, &p2);
589 ..
590 .. Critical section with both p1 and p2 locked
591 ..
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -0700592 // Release locks: Always call polling_island_unlock_pair() to release locks
593 polling_island_unlock_pair(p1, p2);
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700594*/
Craig Tillerbaa14a92017-11-03 09:09:36 -0700595static void polling_island_lock_pair(polling_island** p, polling_island** q) {
596 polling_island* pi_1 = *p;
597 polling_island* pi_2 = *q;
Craig Tiller4782d922017-11-10 09:53:21 -0800598 polling_island* next_1 = nullptr;
599 polling_island* next_2 = nullptr;
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700600
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700601 /* The algorithm is simple:
602 - Go to the last polling islands in the linked lists *pi_1 and *pi_2 (and
603 keep updating pi_1 and pi_2)
604 - Then obtain locks on the islands by following a lock order rule of
605 locking polling_island with lower address first
606 Special case: Before obtaining the locks, check if pi_1 and pi_2 are
607 pointing to the same island. If that is the case, we can just call
608 polling_island_lock()
609 - After obtaining both the locks, double check that the polling islands
610 are still the last polling islands in their respective linked lists
611 (this is because there might have been polling island merges before
612 we got the lock)
613 - If the polling islands are the last islands, we are done. If not,
614 release the locks and continue the process from the first step */
615 while (true) {
Craig Tillerbaa14a92017-11-03 09:09:36 -0700616 next_1 = (polling_island*)gpr_atm_acq_load(&pi_1->merged_to);
Craig Tiller4782d922017-11-10 09:53:21 -0800617 while (next_1 != nullptr) {
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700618 pi_1 = next_1;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700619 next_1 = (polling_island*)gpr_atm_acq_load(&pi_1->merged_to);
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700620 }
621
Craig Tillerbaa14a92017-11-03 09:09:36 -0700622 next_2 = (polling_island*)gpr_atm_acq_load(&pi_2->merged_to);
Craig Tiller4782d922017-11-10 09:53:21 -0800623 while (next_2 != nullptr) {
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700624 pi_2 = next_2;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700625 next_2 = (polling_island*)gpr_atm_acq_load(&pi_2->merged_to);
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700626 }
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700627
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700628 if (pi_1 == pi_2) {
629 pi_1 = pi_2 = polling_island_lock(pi_1);
630 break;
631 }
632
633 if (pi_1 < pi_2) {
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700634 gpr_mu_lock(&pi_1->mu);
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700635 gpr_mu_lock(&pi_2->mu);
636 } else {
637 gpr_mu_lock(&pi_2->mu);
638 gpr_mu_lock(&pi_1->mu);
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700639 }
640
Craig Tillerbaa14a92017-11-03 09:09:36 -0700641 next_1 = (polling_island*)gpr_atm_acq_load(&pi_1->merged_to);
642 next_2 = (polling_island*)gpr_atm_acq_load(&pi_2->merged_to);
Craig Tiller4782d922017-11-10 09:53:21 -0800643 if (next_1 == nullptr && next_2 == nullptr) {
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700644 break;
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700645 }
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700646
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700647 gpr_mu_unlock(&pi_1->mu);
648 gpr_mu_unlock(&pi_2->mu);
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700649 }
650
651 *p = pi_1;
652 *q = pi_2;
653}
654
Craig Tillerbaa14a92017-11-03 09:09:36 -0700655static void polling_island_unlock_pair(polling_island* p, polling_island* q) {
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -0700656 if (p == q) {
657 gpr_mu_unlock(&p->mu);
658 } else {
659 gpr_mu_unlock(&p->mu);
660 gpr_mu_unlock(&q->mu);
661 }
662}
663
Craig Tillerbaa14a92017-11-03 09:09:36 -0700664static polling_island* polling_island_merge(polling_island* p,
665 polling_island* q,
666 grpc_error** error) {
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700667 /* Get locks on both the polling islands */
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700668 polling_island_lock_pair(&p, &q);
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700669
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -0700670 if (p != q) {
671 /* Make sure that p points to the polling island with fewer fds than q */
672 if (p->fd_cnt > q->fd_cnt) {
Craig Tillerbaa14a92017-11-03 09:09:36 -0700673 GPR_SWAP(polling_island*, p, q);
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -0700674 }
675
676 /* Merge p with q i.e move all the fds from p (The one with fewer fds) to q
677 Note that the refcounts on the fds being moved will not change here.
678 This is why the last param in the following two functions is 'false') */
679 polling_island_add_fds_locked(q, p->fds, p->fd_cnt, false, error);
680 polling_island_remove_all_fds_locked(p, false, error);
681
682 /* Wakeup all the pollers (if any) on p so that they pickup this change */
683 polling_island_add_wakeup_fd_locked(p, &polling_island_wakeup_fd, error);
684
685 /* Add the 'merged_to' link from p --> q */
686 gpr_atm_rel_store(&p->merged_to, (gpr_atm)q);
687 PI_ADD_REF(q, "pi_merge"); /* To account for the new incoming ref from p */
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700688 }
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -0700689 /* else if p == q, nothing needs to be done */
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700690
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -0700691 polling_island_unlock_pair(p, q);
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700692
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -0700693 /* Return the merged polling island (Note that no merge would have happened
694 if p == q which is ok) */
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700695 return q;
Sree Kuchibhotla9442bab2016-05-20 17:54:06 -0700696}
697
Craig Tillerbaa14a92017-11-03 09:09:36 -0700698static grpc_error* polling_island_global_init() {
699 grpc_error* error = GRPC_ERROR_NONE;
Sree Kuchibhotla3131c262016-06-21 17:28:28 -0700700
Sree Kuchibhotla3131c262016-06-21 17:28:28 -0700701 error = grpc_wakeup_fd_init(&polling_island_wakeup_fd);
702 if (error == GRPC_ERROR_NONE) {
703 error = grpc_wakeup_fd_wakeup(&polling_island_wakeup_fd);
704 }
705
706 return error;
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700707}
708
Sree Kuchibhotlad627c102016-06-06 15:49:32 -0700709static void polling_island_global_shutdown() {
Sree Kuchibhotla24b10622016-06-08 15:20:17 -0700710 grpc_wakeup_fd_destroy(&polling_island_wakeup_fd);
Sree Kuchibhotlad627c102016-06-06 15:49:32 -0700711}
712
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700713/*******************************************************************************
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700714 * Fd Definitions
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700715 */
716
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700717/* We need to keep a freelist not because of any concerns of malloc performance
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700718 * but instead so that implementations with multiple threads in (for example)
719 * epoll_wait deal with the race between pollset removal and incoming poll
720 * notifications.
721 *
722 * The problem is that the poller ultimately holds a reference to this
723 * object, so it is very difficult to know when is safe to free it, at least
724 * without some expensive synchronization.
725 *
726 * If we keep the object freelisted, in the worst case losing this race just
727 * becomes a spurious read notification on a reused fd.
728 */
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700729
730/* The alarm system needs to be able to wakeup 'some poller' sometimes
731 * (specifically when a new alarm needs to be triggered earlier than the next
732 * alarm 'epoch'). This wakeup_fd gives us something to alert on when such a
733 * case occurs. */
Sree Kuchibhotla9bc3d2d2016-06-06 10:27:56 -0700734
Craig Tiller4782d922017-11-10 09:53:21 -0800735static grpc_fd* fd_freelist = nullptr;
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700736static gpr_mu fd_freelist_mu;
737
ncteisend39010e2017-06-08 17:08:07 -0700738#ifndef NDEBUG
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700739#define REF_BY(fd, n, reason) ref_by(fd, n, reason, __FILE__, __LINE__)
740#define UNREF_BY(fd, n, reason) unref_by(fd, n, reason, __FILE__, __LINE__)
Craig Tillerbaa14a92017-11-03 09:09:36 -0700741static void ref_by(grpc_fd* fd, int n, const char* reason, const char* file,
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700742 int line) {
ncteisen9ffb1492017-11-10 14:00:49 -0800743 if (grpc_trace_fd_refcount.enabled()) {
ncteisen973863d2017-06-12 10:28:50 -0700744 gpr_log(GPR_DEBUG,
745 "FD %d %p ref %d %" PRIdPTR " -> %" PRIdPTR " [%s; %s:%d]",
ncteisen3ac64f82017-06-19 17:35:44 -0700746 fd->fd, fd, n, gpr_atm_no_barrier_load(&fd->refst),
ncteisend39010e2017-06-08 17:08:07 -0700747 gpr_atm_no_barrier_load(&fd->refst) + n, reason, file, line);
748 }
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700749#else
750#define REF_BY(fd, n, reason) ref_by(fd, n)
751#define UNREF_BY(fd, n, reason) unref_by(fd, n)
Craig Tillerbaa14a92017-11-03 09:09:36 -0700752static void ref_by(grpc_fd* fd, int n) {
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700753#endif
754 GPR_ASSERT(gpr_atm_no_barrier_fetch_add(&fd->refst, n) > 0);
755}
756
ncteisend39010e2017-06-08 17:08:07 -0700757#ifndef NDEBUG
Craig Tillerbaa14a92017-11-03 09:09:36 -0700758static void unref_by(grpc_fd* fd, int n, const char* reason, const char* file,
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700759 int line) {
ncteisen9ffb1492017-11-10 14:00:49 -0800760 if (grpc_trace_fd_refcount.enabled()) {
ncteisen973863d2017-06-12 10:28:50 -0700761 gpr_log(GPR_DEBUG,
762 "FD %d %p unref %d %" PRIdPTR " -> %" PRIdPTR " [%s; %s:%d]",
ncteisen3ac64f82017-06-19 17:35:44 -0700763 fd->fd, fd, n, gpr_atm_no_barrier_load(&fd->refst),
ncteisend39010e2017-06-08 17:08:07 -0700764 gpr_atm_no_barrier_load(&fd->refst) - n, reason, file, line);
765 }
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700766#else
Craig Tillerbaa14a92017-11-03 09:09:36 -0700767static void unref_by(grpc_fd* fd, int n) {
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700768#endif
Noah Eisen264879f2017-06-20 17:14:47 -0700769 gpr_atm old = gpr_atm_full_fetch_add(&fd->refst, -n);
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700770 if (old == n) {
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700771 /* Add the fd to the freelist */
772 gpr_mu_lock(&fd_freelist_mu);
773 fd->freelist_next = fd_freelist;
774 fd_freelist = fd;
775 grpc_iomgr_unregister_object(&fd->iomgr_object);
Sree Kuchibhotla82d73412017-02-09 18:27:45 -0800776
yang-ged49fe52017-11-20 13:49:54 -0800777 fd->read_closure->DestroyEvent();
778 fd->write_closure->DestroyEvent();
Yash Tibrewaladc733f2018-04-02 18:32:06 -0700779 fd->error_closure->DestroyEvent();
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700780
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700781 gpr_mu_unlock(&fd_freelist_mu);
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700782 } else {
783 GPR_ASSERT(old > n);
784 }
785}
786
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700787/* Increment refcount by two to avoid changing the orphan bit */
ncteisend39010e2017-06-08 17:08:07 -0700788#ifndef NDEBUG
Craig Tillerbaa14a92017-11-03 09:09:36 -0700789static void fd_ref(grpc_fd* fd, const char* reason, const char* file,
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700790 int line) {
791 ref_by(fd, 2, reason, file, line);
792}
793
Craig Tillerbaa14a92017-11-03 09:09:36 -0700794static void fd_unref(grpc_fd* fd, const char* reason, const char* file,
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700795 int line) {
796 unref_by(fd, 2, reason, file, line);
797}
798#else
Craig Tillerbaa14a92017-11-03 09:09:36 -0700799static void fd_ref(grpc_fd* fd) { ref_by(fd, 2); }
800static void fd_unref(grpc_fd* fd) { unref_by(fd, 2); }
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700801#endif
802
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700803static void fd_global_init(void) { gpr_mu_init(&fd_freelist_mu); }
804
805static void fd_global_shutdown(void) {
806 gpr_mu_lock(&fd_freelist_mu);
807 gpr_mu_unlock(&fd_freelist_mu);
Craig Tiller4782d922017-11-10 09:53:21 -0800808 while (fd_freelist != nullptr) {
Craig Tillerbaa14a92017-11-03 09:09:36 -0700809 grpc_fd* fd = fd_freelist;
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700810 fd_freelist = fd_freelist->freelist_next;
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -0800811 gpr_mu_destroy(&fd->po.mu);
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700812 gpr_free(fd);
813 }
814 gpr_mu_destroy(&fd_freelist_mu);
815}
816
Yash Tibrewaladc733f2018-04-02 18:32:06 -0700817static grpc_fd* fd_create(int fd, const char* name, bool track_err) {
Craig Tiller4782d922017-11-10 09:53:21 -0800818 grpc_fd* new_fd = nullptr;
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700819
820 gpr_mu_lock(&fd_freelist_mu);
Craig Tiller4782d922017-11-10 09:53:21 -0800821 if (fd_freelist != nullptr) {
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700822 new_fd = fd_freelist;
823 fd_freelist = fd_freelist->freelist_next;
824 }
825 gpr_mu_unlock(&fd_freelist_mu);
826
Craig Tiller4782d922017-11-10 09:53:21 -0800827 if (new_fd == nullptr) {
Noah Eisenbe82e642018-02-09 09:16:55 -0800828 new_fd = static_cast<grpc_fd*>(gpr_malloc(sizeof(grpc_fd)));
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -0800829 gpr_mu_init(&new_fd->po.mu);
yang-g26521b32017-11-17 17:15:37 -0800830 new_fd->read_closure.Init();
831 new_fd->write_closure.Init();
Yash Tibrewaladc733f2018-04-02 18:32:06 -0700832 new_fd->error_closure.Init();
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700833 }
834
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -0800835 /* Note: It is not really needed to get the new_fd->po.mu lock here. If this
836 * is a newly created fd (or an fd we got from the freelist), no one else
837 * would be holding a lock to it anyway. */
838 gpr_mu_lock(&new_fd->po.mu);
Craig Tiller4782d922017-11-10 09:53:21 -0800839 new_fd->po.pi = nullptr;
ncteisene9cd8a82017-06-29 06:03:52 -0400840#ifndef NDEBUG
Sree Kuchibhotlaa0749a62016-11-18 20:22:09 -0800841 new_fd->po.obj_type = POLL_OBJ_FD;
842#endif
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700843
Sree Kuchibhotla0224dcc2016-06-22 18:04:00 -0700844 gpr_atm_rel_store(&new_fd->refst, (gpr_atm)1);
Sree Kuchibhotla5855c472016-06-08 12:56:56 -0700845 new_fd->fd = fd;
Sree Kuchibhotla5855c472016-06-08 12:56:56 -0700846 new_fd->orphaned = false;
yang-ged49fe52017-11-20 13:49:54 -0800847 new_fd->read_closure->InitEvent();
848 new_fd->write_closure->InitEvent();
Yash Tibrewaladc733f2018-04-02 18:32:06 -0700849 new_fd->error_closure->InitEvent();
850 new_fd->track_err = track_err;
Sree Kuchibhotla61fe0942017-02-14 12:26:56 -0800851 gpr_atm_no_barrier_store(&new_fd->read_notifier_pollset, (gpr_atm)NULL);
Sree Kuchibhotla82d73412017-02-09 18:27:45 -0800852
Craig Tiller4782d922017-11-10 09:53:21 -0800853 new_fd->freelist_next = nullptr;
854 new_fd->on_done_closure = nullptr;
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700855
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -0800856 gpr_mu_unlock(&new_fd->po.mu);
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700857
Craig Tillerbaa14a92017-11-03 09:09:36 -0700858 char* fd_name;
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700859 gpr_asprintf(&fd_name, "%s fd=%d", name, fd);
860 grpc_iomgr_register_object(&new_fd->iomgr_object, fd_name);
Sree Kuchibhotla6a295452016-06-23 15:53:10 -0700861 gpr_free(fd_name);
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700862 return new_fd;
863}
864
Craig Tillerbaa14a92017-11-03 09:09:36 -0700865static int fd_wrapped_fd(grpc_fd* fd) {
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700866 int ret_fd = -1;
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -0800867 gpr_mu_lock(&fd->po.mu);
Sree Kuchibhotla79a62332016-06-04 14:01:03 -0700868 if (!fd->orphaned) {
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700869 ret_fd = fd->fd;
870 }
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -0800871 gpr_mu_unlock(&fd->po.mu);
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700872
873 return ret_fd;
874}
875
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800876static void fd_orphan(grpc_fd* fd, grpc_closure* on_done, int* release_fd,
Craig Tillerbaa14a92017-11-03 09:09:36 -0700877 bool already_closed, const char* reason) {
878 grpc_error* error = GRPC_ERROR_NONE;
Craig Tiller4782d922017-11-10 09:53:21 -0800879 polling_island* unref_pi = nullptr;
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -0700880
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -0800881 gpr_mu_lock(&fd->po.mu);
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700882 fd->on_done_closure = on_done;
883
Sree Kuchibhotla79a62332016-06-04 14:01:03 -0700884 /* Remove the active status but keep referenced. We want this grpc_fd struct
885 to be alive (and not added to freelist) until the end of this function */
886 REF_BY(fd, 1, reason);
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700887
888 /* Remove the fd from the polling island:
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700889 - Get a lock on the latest polling island (i.e the last island in the
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -0800890 linked list pointed by fd->po.pi). This is the island that
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700891 would actually contain the fd
892 - Remove the fd from the latest polling island
893 - Unlock the latest polling island
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -0800894 - Set fd->po.pi to NULL (but remove the ref on the polling island
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700895 before doing this.) */
Craig Tiller4782d922017-11-10 09:53:21 -0800896 if (fd->po.pi != nullptr) {
Craig Tillerbaa14a92017-11-03 09:09:36 -0700897 polling_island* pi_latest = polling_island_lock(fd->po.pi);
Yuchen Zengd40a7ae2017-07-12 15:59:56 -0700898 polling_island_remove_fd_locked(pi_latest, fd, already_closed, &error);
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -0700899 gpr_mu_unlock(&pi_latest->mu);
Sree Kuchibhotla79a62332016-06-04 14:01:03 -0700900
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -0800901 unref_pi = fd->po.pi;
Craig Tiller4782d922017-11-10 09:53:21 -0800902 fd->po.pi = nullptr;
Sree Kuchibhotla88ee12f2016-06-03 19:26:48 -0700903 }
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700904
Yuchen Zeng5b48dea2017-07-12 19:00:35 -0700905 /* If release_fd is not NULL, we should be relinquishing control of the file
906 descriptor fd->fd (but we still own the grpc_fd structure). */
Craig Tiller4782d922017-11-10 09:53:21 -0800907 if (release_fd != nullptr) {
Yuchen Zeng5b48dea2017-07-12 19:00:35 -0700908 *release_fd = fd->fd;
909 } else {
910 close(fd->fd);
911 }
912
913 fd->orphaned = true;
914
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800915 GRPC_CLOSURE_SCHED(fd->on_done_closure, GRPC_ERROR_REF(error));
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700916
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -0800917 gpr_mu_unlock(&fd->po.mu);
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700918 UNREF_BY(fd, 2, reason); /* Drop the reference */
Craig Tiller4782d922017-11-10 09:53:21 -0800919 if (unref_pi != nullptr) {
Craig Tiller0a06cd72016-07-14 13:21:24 -0700920 /* Unref stale polling island here, outside the fd lock above.
921 The polling island owns a workqueue which owns an fd, and unreffing
922 inside the lock can cause an eventual lock loop that makes TSAN very
923 unhappy. */
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800924 PI_UNREF(unref_pi, "fd_orphan");
Craig Tiller15007612016-07-06 09:36:16 -0700925 }
Yuchen Zeng4ebace72017-06-05 17:24:06 -0700926 if (error != GRPC_ERROR_NONE) {
Craig Tillerbaa14a92017-11-03 09:09:36 -0700927 const char* msg = grpc_error_string(error);
Yuchen Zeng4ebace72017-06-05 17:24:06 -0700928 gpr_log(GPR_DEBUG, "fd_orphan: %s", msg);
929 }
Yuchen Zenga0399f22016-08-04 17:52:53 -0700930 GRPC_ERROR_UNREF(error);
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700931}
932
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800933static grpc_pollset* fd_get_read_notifier_pollset(grpc_fd* fd) {
Sree Kuchibhotlafb7ced62017-02-15 18:31:57 -0800934 gpr_atm notifier = gpr_atm_acq_load(&fd->read_notifier_pollset);
Craig Tillerbaa14a92017-11-03 09:09:36 -0700935 return (grpc_pollset*)notifier;
Sree Kuchibhotla5855c472016-06-08 12:56:56 -0700936}
937
Craig Tillerbaa14a92017-11-03 09:09:36 -0700938static bool fd_is_shutdown(grpc_fd* fd) {
Craig Tillerfbf61bb2017-11-08 11:50:14 -0800939 return fd->read_closure->IsShutdown();
Sree Kuchibhotla24b6eae2016-06-21 18:01:14 -0700940}
941
Sree Kuchibhotla0100b2f2016-06-21 17:38:13 -0700942/* Might be called multiple times */
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800943static void fd_shutdown(grpc_fd* fd, grpc_error* why) {
944 if (fd->read_closure->SetShutdown(GRPC_ERROR_REF(why))) {
Sree Kuchibhotla0100b2f2016-06-21 17:38:13 -0700945 shutdown(fd->fd, SHUT_RDWR);
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800946 fd->write_closure->SetShutdown(GRPC_ERROR_REF(why));
Yash Tibrewaladc733f2018-04-02 18:32:06 -0700947 fd->error_closure->SetShutdown(GRPC_ERROR_REF(why));
Sree Kuchibhotla0100b2f2016-06-21 17:38:13 -0700948 }
Craig Tiller376887d2017-04-06 08:27:03 -0700949 GRPC_ERROR_UNREF(why);
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700950}
951
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800952static void fd_notify_on_read(grpc_fd* fd, grpc_closure* closure) {
953 fd->read_closure->NotifyOn(closure);
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700954}
955
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800956static void fd_notify_on_write(grpc_fd* fd, grpc_closure* closure) {
957 fd->write_closure->NotifyOn(closure);
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700958}
959
Yash Tibrewaladc733f2018-04-02 18:32:06 -0700960static void fd_notify_on_error(grpc_fd* fd, grpc_closure* closure) {
961 fd->error_closure->NotifyOn(closure);
962}
963
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700964/*******************************************************************************
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700965 * Pollset Definitions
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700966 */
Sree Kuchibhotla8e4926c2016-06-08 20:33:19 -0700967GPR_TLS_DECL(g_current_thread_pollset);
968GPR_TLS_DECL(g_current_thread_worker);
Craig Tiller19196992016-06-27 18:45:56 -0700969static __thread bool g_initialized_sigmask;
970static __thread sigset_t g_orig_sigmask;
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700971
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700972static void sig_handler(int sig_num) {
Sree Kuchibhotlad627c102016-06-06 15:49:32 -0700973#ifdef GRPC_EPOLL_DEBUG
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700974 gpr_log(GPR_INFO, "Received signal %d", sig_num);
Sree Kuchibhotla9bc3d2d2016-06-06 10:27:56 -0700975#endif
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700976}
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -0700977
Sree Kuchibhotlac7be7c62016-06-09 17:08:50 -0700978static void poller_kick_init() { signal(grpc_wakeup_signal, sig_handler); }
Sree Kuchibhotla5855c472016-06-08 12:56:56 -0700979
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700980/* Global state management */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700981static grpc_error* pollset_global_init(void) {
Sree Kuchibhotla8e4926c2016-06-08 20:33:19 -0700982 gpr_tls_init(&g_current_thread_pollset);
983 gpr_tls_init(&g_current_thread_worker);
Sree Kuchibhotla5855c472016-06-08 12:56:56 -0700984 poller_kick_init();
Craig Tillerc3571792017-05-02 12:33:38 -0700985 return GRPC_ERROR_NONE;
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700986}
987
988static void pollset_global_shutdown(void) {
Sree Kuchibhotla8e4926c2016-06-08 20:33:19 -0700989 gpr_tls_destroy(&g_current_thread_pollset);
990 gpr_tls_destroy(&g_current_thread_worker);
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -0700991}
992
Craig Tillerbaa14a92017-11-03 09:09:36 -0700993static grpc_error* pollset_worker_kick(grpc_pollset_worker* worker) {
994 grpc_error* err = GRPC_ERROR_NONE;
Sree Kuchibhotla34217242016-06-29 00:19:07 -0700995
996 /* Kick the worker only if it was not already kicked */
Noah Eisen4d20a662018-02-09 09:34:04 -0800997 if (gpr_atm_no_barrier_cas(&worker->is_kicked, static_cast<gpr_atm>(0),
998 static_cast<gpr_atm>(1))) {
Sree Kuchibhotla34217242016-06-29 00:19:07 -0700999 GRPC_POLLING_TRACE(
1000 "pollset_worker_kick: Kicking worker: %p (thread id: %ld)",
Craig Tillerbaa14a92017-11-03 09:09:36 -07001001 (void*)worker, (long int)worker->pt_id);
Sree Kuchibhotla34217242016-06-29 00:19:07 -07001002 int err_num = pthread_kill(worker->pt_id, grpc_wakeup_signal);
1003 if (err_num != 0) {
1004 err = GRPC_OS_ERROR(err_num, "pthread_kill");
1005 }
Sree Kuchibhotla3131c262016-06-21 17:28:28 -07001006 }
1007 return err;
Sree Kuchibhotla5855c472016-06-08 12:56:56 -07001008}
1009
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001010/* Return 1 if the pollset has active threads in pollset_work (pollset must
1011 * be locked) */
Craig Tillerbaa14a92017-11-03 09:09:36 -07001012static int pollset_has_workers(grpc_pollset* p) {
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001013 return p->root_worker.next != &p->root_worker;
1014}
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001015
Craig Tillerbaa14a92017-11-03 09:09:36 -07001016static void remove_worker(grpc_pollset* p, grpc_pollset_worker* worker) {
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001017 worker->prev->next = worker->next;
1018 worker->next->prev = worker->prev;
1019}
1020
Craig Tillerbaa14a92017-11-03 09:09:36 -07001021static grpc_pollset_worker* pop_front_worker(grpc_pollset* p) {
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001022 if (pollset_has_workers(p)) {
Craig Tillerbaa14a92017-11-03 09:09:36 -07001023 grpc_pollset_worker* w = p->root_worker.next;
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001024 remove_worker(p, w);
1025 return w;
1026 } else {
Craig Tiller4782d922017-11-10 09:53:21 -08001027 return nullptr;
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001028 }
1029}
1030
Craig Tillerbaa14a92017-11-03 09:09:36 -07001031static void push_back_worker(grpc_pollset* p, grpc_pollset_worker* worker) {
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001032 worker->next = &p->root_worker;
1033 worker->prev = worker->next->prev;
1034 worker->prev->next = worker->next->prev = worker;
1035}
1036
Craig Tillerbaa14a92017-11-03 09:09:36 -07001037static void push_front_worker(grpc_pollset* p, grpc_pollset_worker* worker) {
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001038 worker->prev = &p->root_worker;
1039 worker->next = worker->prev->next;
1040 worker->prev->next = worker->next->prev = worker;
1041}
1042
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001043/* p->mu must be held before calling this function */
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001044static grpc_error* pollset_kick(grpc_pollset* p,
Craig Tillerbaa14a92017-11-03 09:09:36 -07001045 grpc_pollset_worker* specific_worker) {
yang-gce1cfea2018-01-31 15:59:50 -08001046 GPR_TIMER_SCOPE("pollset_kick", 0);
Craig Tillerbaa14a92017-11-03 09:09:36 -07001047 grpc_error* error = GRPC_ERROR_NONE;
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001048 GRPC_STATS_INC_POLLSET_KICK();
Craig Tillerbaa14a92017-11-03 09:09:36 -07001049 const char* err_desc = "Kick Failure";
1050 grpc_pollset_worker* worker = specific_worker;
Craig Tiller4782d922017-11-10 09:53:21 -08001051 if (worker != nullptr) {
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001052 if (worker == GRPC_POLLSET_KICK_BROADCAST) {
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001053 if (pollset_has_workers(p)) {
yang-gce1cfea2018-01-31 15:59:50 -08001054 GPR_TIMER_SCOPE("pollset_kick.broadcast", 0);
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001055 for (worker = p->root_worker.next; worker != &p->root_worker;
1056 worker = worker->next) {
Sree Kuchibhotla8e4926c2016-06-08 20:33:19 -07001057 if (gpr_tls_get(&g_current_thread_worker) != (intptr_t)worker) {
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -07001058 append_error(&error, pollset_worker_kick(worker), err_desc);
Sree Kuchibhotla8e4926c2016-06-08 20:33:19 -07001059 }
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001060 }
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001061 } else {
1062 p->kicked_without_pollers = true;
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001063 }
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001064 } else {
1065 GPR_TIMER_MARK("kicked_specifically", 0);
Sree Kuchibhotla8e4926c2016-06-08 20:33:19 -07001066 if (gpr_tls_get(&g_current_thread_worker) != (intptr_t)worker) {
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -07001067 append_error(&error, pollset_worker_kick(worker), err_desc);
Sree Kuchibhotla8e4926c2016-06-08 20:33:19 -07001068 }
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001069 }
Sree Kuchibhotla8e4926c2016-06-08 20:33:19 -07001070 } else if (gpr_tls_get(&g_current_thread_pollset) != (intptr_t)p) {
1071 /* Since worker == NULL, it means that we can kick "any" worker on this
1072 pollset 'p'. If 'p' happens to be the same pollset this thread is
1073 currently polling (i.e in pollset_work() function), then there is no need
1074 to kick any other worker since the current thread can just absorb the
1075 kick. This is the reason why we enter this case only when
1076 g_current_thread_pollset is != p */
1077
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001078 GPR_TIMER_MARK("kick_anonymous", 0);
1079 worker = pop_front_worker(p);
Craig Tiller4782d922017-11-10 09:53:21 -08001080 if (worker != nullptr) {
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001081 GPR_TIMER_MARK("finally_kick", 0);
1082 push_back_worker(p, worker);
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -07001083 append_error(&error, pollset_worker_kick(worker), err_desc);
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001084 } else {
1085 GPR_TIMER_MARK("kicked_no_pollers", 0);
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001086 p->kicked_without_pollers = true;
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001087 }
1088 }
1089
Sree Kuchibhotla3131c262016-06-21 17:28:28 -07001090 GRPC_LOG_IF_ERROR("pollset_kick", GRPC_ERROR_REF(error));
1091 return error;
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001092}
1093
Craig Tillerbaa14a92017-11-03 09:09:36 -07001094static void pollset_init(grpc_pollset* pollset, gpr_mu** mu) {
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001095 gpr_mu_init(&pollset->po.mu);
1096 *mu = &pollset->po.mu;
Craig Tiller4782d922017-11-10 09:53:21 -08001097 pollset->po.pi = nullptr;
ncteisene9cd8a82017-06-29 06:03:52 -04001098#ifndef NDEBUG
Sree Kuchibhotlaa0749a62016-11-18 20:22:09 -08001099 pollset->po.obj_type = POLL_OBJ_POLLSET;
1100#endif
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001101
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001102 pollset->root_worker.next = pollset->root_worker.prev = &pollset->root_worker;
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001103 pollset->kicked_without_pollers = false;
1104
1105 pollset->shutting_down = false;
1106 pollset->finish_shutdown_called = false;
Craig Tiller4782d922017-11-10 09:53:21 -08001107 pollset->shutdown_done = nullptr;
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001108}
1109
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001110static int poll_deadline_to_millis_timeout(grpc_millis millis) {
Craig Tiller20397792017-07-18 11:35:27 -07001111 if (millis == GRPC_MILLIS_INF_FUTURE) return -1;
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001112 grpc_millis delta = millis - grpc_core::ExecCtx::Get()->Now();
Craig Tiller20397792017-07-18 11:35:27 -07001113 if (delta > INT_MAX)
1114 return INT_MAX;
1115 else if (delta < 0)
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001116 return 0;
Craig Tiller20397792017-07-18 11:35:27 -07001117 else
Noah Eisenbe82e642018-02-09 09:16:55 -08001118 return static_cast<int>(delta);
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001119}
1120
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001121static void fd_become_readable(grpc_fd* fd, grpc_pollset* notifier) {
1122 fd->read_closure->SetReady();
Sree Kuchibhotla82d73412017-02-09 18:27:45 -08001123
Sree Kuchibhotla4db8c822017-02-12 17:07:31 -08001124 /* Note, it is possible that fd_become_readable might be called twice with
Sree Kuchibhotla4c60d0d2017-02-12 17:09:08 -08001125 different 'notifier's when an fd becomes readable and it is in two epoll
1126 sets (This can happen briefly during polling island merges). In such cases
1127 it does not really matter which notifer is set as the read_notifier_pollset
1128 (They would both point to the same polling island anyway) */
Sree Kuchibhotlafb7ced62017-02-15 18:31:57 -08001129 /* Use release store to match with acquire load in fd_get_read_notifier */
1130 gpr_atm_rel_store(&fd->read_notifier_pollset, (gpr_atm)notifier);
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001131}
1132
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001133static void fd_become_writable(grpc_fd* fd) { fd->write_closure->SetReady(); }
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001134
Yash Tibrewaladc733f2018-04-02 18:32:06 -07001135static void fd_has_errors(grpc_fd* fd) { fd->error_closure->SetReady(); }
1136
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001137static void pollset_release_polling_island(grpc_pollset* ps,
Craig Tillerbaa14a92017-11-03 09:09:36 -07001138 const char* reason) {
Craig Tiller4782d922017-11-10 09:53:21 -08001139 if (ps->po.pi != nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001140 PI_UNREF(ps->po.pi, reason);
Sree Kuchibhotla8e4926c2016-06-08 20:33:19 -07001141 }
Craig Tiller4782d922017-11-10 09:53:21 -08001142 ps->po.pi = nullptr;
Sree Kuchibhotla8e4926c2016-06-08 20:33:19 -07001143}
1144
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001145static void finish_shutdown_locked(grpc_pollset* pollset) {
Sree Kuchibhotla8e4926c2016-06-08 20:33:19 -07001146 /* The pollset cannot have any workers if we are at this stage */
1147 GPR_ASSERT(!pollset_has_workers(pollset));
1148
1149 pollset->finish_shutdown_called = true;
Sree Kuchibhotla8e4926c2016-06-08 20:33:19 -07001150
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001151 /* Release the ref and set pollset->po.pi to NULL */
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001152 pollset_release_polling_island(pollset, "ps_shutdown");
1153 GRPC_CLOSURE_SCHED(pollset->shutdown_done, GRPC_ERROR_NONE);
Sree Kuchibhotla8e4926c2016-06-08 20:33:19 -07001154}
1155
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001156/* pollset->po.mu lock must be held by the caller before calling this */
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001157static void pollset_shutdown(grpc_pollset* pollset, grpc_closure* closure) {
yang-gce1cfea2018-01-31 15:59:50 -08001158 GPR_TIMER_SCOPE("pollset_shutdown", 0);
Sree Kuchibhotla8e4926c2016-06-08 20:33:19 -07001159 GPR_ASSERT(!pollset->shutting_down);
1160 pollset->shutting_down = true;
1161 pollset->shutdown_done = closure;
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001162 pollset_kick(pollset, GRPC_POLLSET_KICK_BROADCAST);
Sree Kuchibhotla8e4926c2016-06-08 20:33:19 -07001163
1164 /* If the pollset has any workers, we cannot call finish_shutdown_locked()
1165 because it would release the underlying polling island. In such a case, we
1166 let the last worker call finish_shutdown_locked() from pollset_work() */
1167 if (!pollset_has_workers(pollset)) {
1168 GPR_ASSERT(!pollset->finish_shutdown_called);
1169 GPR_TIMER_MARK("pollset_shutdown.finish_shutdown_locked", 0);
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001170 finish_shutdown_locked(pollset);
Sree Kuchibhotla8e4926c2016-06-08 20:33:19 -07001171 }
Sree Kuchibhotla8e4926c2016-06-08 20:33:19 -07001172}
1173
1174/* pollset_shutdown is guaranteed to be called before pollset_destroy. So other
1175 * than destroying the mutexes, there is nothing special that needs to be done
1176 * here */
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001177static void pollset_destroy(grpc_pollset* pollset) {
Sree Kuchibhotla8e4926c2016-06-08 20:33:19 -07001178 GPR_ASSERT(!pollset_has_workers(pollset));
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001179 gpr_mu_destroy(&pollset->po.mu);
Sree Kuchibhotla8e4926c2016-06-08 20:33:19 -07001180}
1181
Craig Tiller84ea3412016-09-08 14:57:56 -07001182#define GRPC_EPOLL_MAX_EVENTS 100
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -07001183/* Note: sig_mask contains the signal mask to use *during* epoll_wait() */
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001184static void pollset_work_and_unlock(grpc_pollset* pollset,
Craig Tillerbaa14a92017-11-03 09:09:36 -07001185 grpc_pollset_worker* worker, int timeout_ms,
1186 sigset_t* sig_mask, grpc_error** error) {
yang-gce1cfea2018-01-31 15:59:50 -08001187 GPR_TIMER_SCOPE("pollset_work_and_unlock", 0);
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001188 struct epoll_event ep_ev[GRPC_EPOLL_MAX_EVENTS];
Sree Kuchibhotla88ee12f2016-06-03 19:26:48 -07001189 int epoll_fd = -1;
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001190 int ep_rv;
Craig Tiller4782d922017-11-10 09:53:21 -08001191 polling_island* pi = nullptr;
Craig Tillerbaa14a92017-11-03 09:09:36 -07001192 char* err_msg;
1193 const char* err_desc = "pollset_work_and_unlock";
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001194
1195 /* We need to get the epoll_fd to wait on. The epoll_fd is in inside the
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001196 latest polling island pointed by pollset->po.pi
Sree Kuchibhotla229533b12016-06-21 20:42:52 -07001197
1198 Since epoll_fd is immutable, we can read it without obtaining the polling
1199 island lock. There is however a possibility that the polling island (from
1200 which we got the epoll_fd) got merged with another island while we are
1201 in this function. This is still okay because in such a case, we will wakeup
1202 right-away from epoll_wait() and pick up the latest polling_island the next
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -07001203 this function (i.e pollset_work_and_unlock()) is called */
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001204
Craig Tiller4782d922017-11-10 09:53:21 -08001205 if (pollset->po.pi == nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001206 pollset->po.pi = polling_island_create(nullptr, error);
Craig Tiller4782d922017-11-10 09:53:21 -08001207 if (pollset->po.pi == nullptr) {
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -07001208 return; /* Fatal error. We cannot continue */
1209 }
1210
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001211 PI_ADD_REF(pollset->po.pi, "ps");
Sree Kuchibhotla1e776682016-06-28 14:09:26 -07001212 GRPC_POLLING_TRACE("pollset_work: pollset: %p created new pi: %p",
Craig Tillerbaa14a92017-11-03 09:09:36 -07001213 (void*)pollset, (void*)pollset->po.pi);
Sree Kuchibhotla88ee12f2016-06-03 19:26:48 -07001214 }
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001215
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001216 pi = polling_island_maybe_get_latest(pollset->po.pi);
Sree Kuchibhotla24b10622016-06-08 15:20:17 -07001217 epoll_fd = pi->epoll_fd;
1218
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001219 /* Update the pollset->po.pi since the island being pointed by
1220 pollset->po.pi maybe older than the one pointed by pi) */
1221 if (pollset->po.pi != pi) {
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -07001222 /* Always do PI_ADD_REF before PI_UNREF because PI_UNREF may cause the
1223 polling island to be deleted */
1224 PI_ADD_REF(pi, "ps");
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001225 PI_UNREF(pollset->po.pi, "ps");
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001226 pollset->po.pi = pi;
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -07001227 }
Sree Kuchibhotlad627c102016-06-06 15:49:32 -07001228
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -07001229 /* Add an extra ref so that the island does not get destroyed (which means
1230 the epoll_fd won't be closed) while we are are doing an epoll_wait() on the
1231 epoll_fd */
1232 PI_ADD_REF(pi, "ps_work");
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001233 gpr_mu_unlock(&pollset->po.mu);
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001234
Craig Tiller61f96c12017-05-12 13:36:39 -07001235 gpr_atm_no_barrier_fetch_add(&pi->poller_count, 1);
1236 g_current_thread_polling_island = pi;
Craig Tillerd8a3c042016-09-09 12:42:37 -07001237
Craig Tiller61f96c12017-05-12 13:36:39 -07001238 GRPC_SCHEDULING_START_BLOCKING_REGION;
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001239 GRPC_STATS_INC_SYSCALL_POLL();
Craig Tiller61f96c12017-05-12 13:36:39 -07001240 ep_rv =
1241 epoll_pwait(epoll_fd, ep_ev, GRPC_EPOLL_MAX_EVENTS, timeout_ms, sig_mask);
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001242 GRPC_SCHEDULING_END_BLOCKING_REGION;
Craig Tiller61f96c12017-05-12 13:36:39 -07001243 if (ep_rv < 0) {
1244 if (errno != EINTR) {
1245 gpr_asprintf(&err_msg,
1246 "epoll_wait() epoll fd: %d failed with error: %d (%s)",
1247 epoll_fd, errno, strerror(errno));
1248 append_error(error, GRPC_OS_ERROR(errno, err_msg), err_desc);
1249 } else {
1250 /* We were interrupted. Save an interation by doing a zero timeout
1251 epoll_wait to see if there are any other events of interest */
1252 GRPC_POLLING_TRACE("pollset_work: pollset: %p, worker: %p received kick",
Craig Tillerbaa14a92017-11-03 09:09:36 -07001253 (void*)pollset, (void*)worker);
Craig Tiller61f96c12017-05-12 13:36:39 -07001254 ep_rv = epoll_wait(epoll_fd, ep_ev, GRPC_EPOLL_MAX_EVENTS, 0);
Sree Kuchibhotlae5012ba2016-06-06 16:01:45 -07001255 }
Craig Tiller61f96c12017-05-12 13:36:39 -07001256 }
Sree Kuchibhotla79a62332016-06-04 14:01:03 -07001257
Sree Kuchibhotlaad2c4772016-06-13 19:06:54 -07001258#ifdef GRPC_TSAN
Craig Tiller61f96c12017-05-12 13:36:39 -07001259 /* See the definition of g_poll_sync for more details */
1260 gpr_atm_acq_load(&g_epoll_sync);
Sree Kuchibhotlaad2c4772016-06-13 19:06:54 -07001261#endif /* defined(GRPC_TSAN) */
Sree Kuchibhotla41622a82016-06-13 16:43:14 -07001262
Craig Tiller61f96c12017-05-12 13:36:39 -07001263 for (int i = 0; i < ep_rv; ++i) {
Craig Tillerbaa14a92017-11-03 09:09:36 -07001264 void* data_ptr = ep_ev[i].data.ptr;
Craig Tiller61f96c12017-05-12 13:36:39 -07001265 if (data_ptr == &polling_island_wakeup_fd) {
1266 GRPC_POLLING_TRACE(
1267 "pollset_work: pollset: %p, worker: %p polling island (epoll_fd: "
1268 "%d) got merged",
Craig Tillerbaa14a92017-11-03 09:09:36 -07001269 (void*)pollset, (void*)worker, epoll_fd);
Craig Tiller61f96c12017-05-12 13:36:39 -07001270 /* This means that our polling island is merged with a different
1271 island. We do not have to do anything here since the subsequent call
1272 to the function pollset_work_and_unlock() will pick up the correct
1273 epoll_fd */
1274 } else {
Yash Tibrewaladc733f2018-04-02 18:32:06 -07001275 grpc_fd* fd = reinterpret_cast<grpc_fd*>(
1276 reinterpret_cast<intptr_t>(data_ptr) & ~static_cast<intptr_t>(1));
1277 bool track_err =
1278 reinterpret_cast<intptr_t>(data_ptr) & ~static_cast<intptr_t>(1);
1279 bool cancel = (ep_ev[i].events & EPOLLHUP) != 0;
1280 bool error = (ep_ev[i].events & EPOLLERR) != 0;
1281 bool read_ev = (ep_ev[i].events & (EPOLLIN | EPOLLPRI)) != 0;
1282 bool write_ev = (ep_ev[i].events & EPOLLOUT) != 0;
1283 bool err_fallback = error && track_err;
1284
1285 if (error && !err_fallback) {
1286 fd_has_errors(fd);
1287 }
1288 if (read_ev || cancel || err_fallback) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001289 fd_become_readable(fd, pollset);
Craig Tiller61f96c12017-05-12 13:36:39 -07001290 }
Yash Tibrewaladc733f2018-04-02 18:32:06 -07001291 if (write_ev || cancel || err_fallback) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001292 fd_become_writable(fd);
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001293 }
Sree Kuchibhotlae5012ba2016-06-06 16:01:45 -07001294 }
Craig Tillerd8a3c042016-09-09 12:42:37 -07001295 }
Sree Kuchibhotla24b10622016-06-08 15:20:17 -07001296
Craig Tiller4782d922017-11-10 09:53:21 -08001297 g_current_thread_polling_island = nullptr;
Craig Tiller61f96c12017-05-12 13:36:39 -07001298 gpr_atm_no_barrier_fetch_add(&pi->poller_count, -1);
1299
Craig Tiller4782d922017-11-10 09:53:21 -08001300 GPR_ASSERT(pi != nullptr);
Sree Kuchibhotla24b10622016-06-08 15:20:17 -07001301
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -07001302 /* Before leaving, release the extra ref we added to the polling island. It
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001303 is important to use "pi" here (i.e our old copy of pollset->po.pi
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -07001304 that we got before releasing the polling island lock). This is because
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001305 pollset->po.pi pointer might get udpated in other parts of the
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -07001306 code when there is an island merge while we are doing epoll_wait() above */
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001307 PI_UNREF(pi, "ps_work");
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001308}
1309
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001310/* pollset->po.mu lock must be held by the caller before calling this.
1311 The function pollset_work() may temporarily release the lock (pollset->po.mu)
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001312 during the course of its execution but it will always re-acquire the lock and
1313 ensure that it is held by the time the function returns */
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001314static grpc_error* pollset_work(grpc_pollset* pollset,
Craig Tillerbaa14a92017-11-03 09:09:36 -07001315 grpc_pollset_worker** worker_hdl,
Craig Tiller20397792017-07-18 11:35:27 -07001316 grpc_millis deadline) {
yang-gce1cfea2018-01-31 15:59:50 -08001317 GPR_TIMER_SCOPE("pollset_work", 0);
Craig Tillerbaa14a92017-11-03 09:09:36 -07001318 grpc_error* error = GRPC_ERROR_NONE;
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001319 int timeout_ms = poll_deadline_to_millis_timeout(deadline);
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001320
1321 sigset_t new_mask;
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001322
1323 grpc_pollset_worker worker;
Craig Tiller4782d922017-11-10 09:53:21 -08001324 worker.next = worker.prev = nullptr;
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001325 worker.pt_id = pthread_self();
Sree Kuchibhotla34217242016-06-29 00:19:07 -07001326 gpr_atm_no_barrier_store(&worker.is_kicked, (gpr_atm)0);
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001327
Craig Tiller557c88c2017-04-05 17:20:18 -07001328 if (worker_hdl) *worker_hdl = &worker;
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -07001329
Sree Kuchibhotla8e4926c2016-06-08 20:33:19 -07001330 gpr_tls_set(&g_current_thread_pollset, (intptr_t)pollset);
1331 gpr_tls_set(&g_current_thread_worker, (intptr_t)&worker);
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001332
1333 if (pollset->kicked_without_pollers) {
1334 /* If the pollset was kicked without pollers, pretend that the current
1335 worker got the kick and skip polling. A kick indicates that there is some
1336 work that needs attention like an event on the completion queue or an
1337 alarm */
1338 GPR_TIMER_MARK("pollset_work.kicked_without_pollers", 0);
1339 pollset->kicked_without_pollers = 0;
1340 } else if (!pollset->shutting_down) {
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -07001341 /* We use the posix-signal with number 'grpc_wakeup_signal' for waking up
Sree Kuchibhotla34217242016-06-29 00:19:07 -07001342 (i.e 'kicking') a worker in the pollset. A 'kick' is a way to inform the
1343 worker that there is some pending work that needs immediate attention
1344 (like an event on the completion queue, or a polling island merge that
1345 results in a new epoll-fd to wait on) and that the worker should not
1346 spend time waiting in epoll_pwait().
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -07001347
Sree Kuchibhotla34217242016-06-29 00:19:07 -07001348 A worker can be kicked anytime from the point it is added to the pollset
1349 via push_front_worker() (or push_back_worker()) to the point it is
1350 removed via remove_worker().
1351 If the worker is kicked before/during it calls epoll_pwait(), it should
1352 immediately exit from epoll_wait(). If the worker is kicked after it
1353 returns from epoll_wait(), then nothing really needs to be done.
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -07001354
Sree Kuchibhotla34217242016-06-29 00:19:07 -07001355 To accomplish this, we mask 'grpc_wakeup_signal' on this thread at all
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -07001356 times *except* when it is in epoll_pwait(). This way, the worker never
1357 misses acting on a kick */
1358
Craig Tiller19196992016-06-27 18:45:56 -07001359 if (!g_initialized_sigmask) {
1360 sigemptyset(&new_mask);
1361 sigaddset(&new_mask, grpc_wakeup_signal);
1362 pthread_sigmask(SIG_BLOCK, &new_mask, &g_orig_sigmask);
1363 sigdelset(&g_orig_sigmask, grpc_wakeup_signal);
1364 g_initialized_sigmask = true;
1365 /* new_mask: The new thread mask which blocks 'grpc_wakeup_signal'.
1366 This is the mask used at all times *except during
1367 epoll_wait()*"
1368 g_orig_sigmask: The thread mask which allows 'grpc_wakeup_signal' and
Craig Tiller510ff692016-06-27 20:31:49 -07001369 this is the mask to use *during epoll_wait()*
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001370
Craig Tiller19196992016-06-27 18:45:56 -07001371 The new_mask is set on the worker before it is added to the pollset
Craig Tiller510ff692016-06-27 20:31:49 -07001372 (i.e before it can be kicked) */
Craig Tiller19196992016-06-27 18:45:56 -07001373 }
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001374
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -07001375 push_front_worker(pollset, &worker); /* Add worker to pollset */
1376
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001377 pollset_work_and_unlock(pollset, &worker, timeout_ms, &g_orig_sigmask,
1378 &error);
1379 grpc_core::ExecCtx::Get()->Flush();
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001380
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001381 gpr_mu_lock(&pollset->po.mu);
Sree Kuchibhotla34217242016-06-29 00:19:07 -07001382
1383 /* Note: There is no need to reset worker.is_kicked to 0 since we are no
1384 longer going to use this worker */
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001385 remove_worker(pollset, &worker);
1386 }
1387
1388 /* If we are the last worker on the pollset (i.e pollset_has_workers() is
1389 false at this point) and the pollset is shutting down, we may have to
1390 finish the shutdown process by calling finish_shutdown_locked().
1391 See pollset_shutdown() for more details.
1392
1393 Note: Continuing to access pollset here is safe; it is the caller's
1394 responsibility to not destroy a pollset when it has outstanding calls to
1395 pollset_work() */
1396 if (pollset->shutting_down && !pollset_has_workers(pollset) &&
1397 !pollset->finish_shutdown_called) {
1398 GPR_TIMER_MARK("pollset_work.finish_shutdown_locked", 0);
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001399 finish_shutdown_locked(pollset);
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001400
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001401 gpr_mu_unlock(&pollset->po.mu);
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001402 grpc_core::ExecCtx::Get()->Flush();
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001403 gpr_mu_lock(&pollset->po.mu);
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001404 }
1405
Craig Tiller4782d922017-11-10 09:53:21 -08001406 if (worker_hdl) *worker_hdl = nullptr;
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -07001407
Sree Kuchibhotla8e4926c2016-06-08 20:33:19 -07001408 gpr_tls_set(&g_current_thread_pollset, (intptr_t)0);
1409 gpr_tls_set(&g_current_thread_worker, (intptr_t)0);
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -07001410
Sree Kuchibhotla3131c262016-06-21 17:28:28 -07001411 GRPC_LOG_IF_ERROR("pollset_work", GRPC_ERROR_REF(error));
1412 return error;
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001413}
1414
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001415static void add_poll_object(poll_obj* bag, poll_obj_type bag_type,
1416 poll_obj* item, poll_obj_type item_type) {
yang-gce1cfea2018-01-31 15:59:50 -08001417 GPR_TIMER_SCOPE("add_poll_object", 0);
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001418
ncteisene9cd8a82017-06-29 06:03:52 -04001419#ifndef NDEBUG
Sree Kuchibhotlaa0749a62016-11-18 20:22:09 -08001420 GPR_ASSERT(item->obj_type == item_type);
1421 GPR_ASSERT(bag->obj_type == bag_type);
1422#endif
Craig Tiller57726ca2016-09-12 11:59:45 -07001423
Craig Tillerbaa14a92017-11-03 09:09:36 -07001424 grpc_error* error = GRPC_ERROR_NONE;
Craig Tiller4782d922017-11-10 09:53:21 -08001425 polling_island* pi_new = nullptr;
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001426
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001427 gpr_mu_lock(&bag->mu);
1428 gpr_mu_lock(&item->mu);
1429
Craig Tiller7212c232016-07-06 13:11:09 -07001430retry:
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001431 /*
1432 * 1) If item->pi and bag->pi are both non-NULL and equal, do nothing
1433 * 2) If item->pi and bag->pi are both NULL, create a new polling island (with
1434 * a refcount of 2) and point item->pi and bag->pi to the new island
1435 * 3) If exactly one of item->pi or bag->pi is NULL, update it to point to
1436 * the other's non-NULL pi
1437 * 4) Finally if item->pi and bag-pi are non-NULL and not-equal, merge the
1438 * polling islands and update item->pi and bag->pi to point to the new
1439 * island
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001440 */
Craig Tiller8e8027b2016-07-07 10:42:09 -07001441
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001442 /* Early out if we are trying to add an 'fd' to a 'bag' but the fd is already
1443 * orphaned */
1444 if (item_type == POLL_OBJ_FD && (FD_FROM_PO(item))->orphaned) {
1445 gpr_mu_unlock(&item->mu);
1446 gpr_mu_unlock(&bag->mu);
Craig Tiller42ac6db2016-07-06 17:13:56 -07001447 return;
1448 }
1449
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001450 if (item->pi == bag->pi) {
1451 pi_new = item->pi;
Craig Tiller4782d922017-11-10 09:53:21 -08001452 if (pi_new == nullptr) {
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001453 /* GPR_ASSERT(item->pi == bag->pi == NULL) */
Sree Kuchibhotlaa129adf2016-10-26 16:44:44 -07001454
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001455 /* If we are adding an fd to a bag (i.e pollset or pollset_set), then
1456 * we need to do some extra work to make TSAN happy */
1457 if (item_type == POLL_OBJ_FD) {
1458 /* Unlock before creating a new polling island: the polling island will
1459 create a workqueue which creates a file descriptor, and holding an fd
1460 lock here can eventually cause a loop to appear to TSAN (making it
1461 unhappy). We don't think it's a real loop (there's an epoch point
1462 where that loop possibility disappears), but the advantages of
1463 keeping TSAN happy outweigh any performance advantage we might have
1464 by keeping the lock held. */
1465 gpr_mu_unlock(&item->mu);
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001466 pi_new = polling_island_create(FD_FROM_PO(item), &error);
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001467 gpr_mu_lock(&item->mu);
Sree Kuchibhotlaa129adf2016-10-26 16:44:44 -07001468
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001469 /* Need to reverify any assumptions made between the initial lock and
1470 getting to this branch: if they've changed, we need to throw away our
1471 work and figure things out again. */
Craig Tiller4782d922017-11-10 09:53:21 -08001472 if (item->pi != nullptr) {
Sree Kuchibhotla499b94b2016-11-18 14:35:47 -08001473 GRPC_POLLING_TRACE(
1474 "add_poll_object: Raced creating new polling island. pi_new: %p "
1475 "(fd: %d, %s: %p)",
Craig Tillerbaa14a92017-11-03 09:09:36 -07001476 (void*)pi_new, FD_FROM_PO(item)->fd, poll_obj_string(bag_type),
1477 (void*)bag);
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001478 /* No need to lock 'pi_new' here since this is a new polling island
Sree Kuchibhotla8214b872017-02-23 12:49:48 -08001479 and no one has a reference to it yet */
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001480 polling_island_remove_all_fds_locked(pi_new, true, &error);
1481
1482 /* Ref and unref so that the polling island gets deleted during unref
1483 */
1484 PI_ADD_REF(pi_new, "dance_of_destruction");
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001485 PI_UNREF(pi_new, "dance_of_destruction");
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001486 goto retry;
1487 }
Craig Tiller27da6422016-07-06 13:14:46 -07001488 } else {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001489 pi_new = polling_island_create(nullptr, &error);
Craig Tiller7212c232016-07-06 13:11:09 -07001490 }
Sree Kuchibhotla499b94b2016-11-18 14:35:47 -08001491
1492 GRPC_POLLING_TRACE(
1493 "add_poll_object: Created new polling island. pi_new: %p (%s: %p, "
1494 "%s: %p)",
Craig Tillerbaa14a92017-11-03 09:09:36 -07001495 (void*)pi_new, poll_obj_string(item_type), (void*)item,
1496 poll_obj_string(bag_type), (void*)bag);
Sree Kuchibhotla499b94b2016-11-18 14:35:47 -08001497 } else {
1498 GRPC_POLLING_TRACE(
1499 "add_poll_object: Same polling island. pi: %p (%s, %s)",
Craig Tillerbaa14a92017-11-03 09:09:36 -07001500 (void*)pi_new, poll_obj_string(item_type), poll_obj_string(bag_type));
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001501 }
Craig Tiller4782d922017-11-10 09:53:21 -08001502 } else if (item->pi == nullptr) {
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001503 /* GPR_ASSERT(bag->pi != NULL) */
1504 /* Make pi_new point to latest pi*/
1505 pi_new = polling_island_lock(bag->pi);
Sree Kuchibhotla1e776682016-06-28 14:09:26 -07001506
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001507 if (item_type == POLL_OBJ_FD) {
Craig Tillerbaa14a92017-11-03 09:09:36 -07001508 grpc_fd* fd = FD_FROM_PO(item);
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001509 polling_island_add_fds_locked(pi_new, &fd, 1, true, &error);
1510 }
Sree Kuchibhotla1e776682016-06-28 14:09:26 -07001511
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001512 gpr_mu_unlock(&pi_new->mu);
Sree Kuchibhotla1e776682016-06-28 14:09:26 -07001513 GRPC_POLLING_TRACE(
Sree Kuchibhotla499b94b2016-11-18 14:35:47 -08001514 "add_poll_obj: item->pi was NULL. pi_new: %p (item(%s): %p, "
1515 "bag(%s): %p)",
Craig Tillerbaa14a92017-11-03 09:09:36 -07001516 (void*)pi_new, poll_obj_string(item_type), (void*)item,
1517 poll_obj_string(bag_type), (void*)bag);
Craig Tiller4782d922017-11-10 09:53:21 -08001518 } else if (bag->pi == nullptr) {
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001519 /* GPR_ASSERT(item->pi != NULL) */
1520 /* Make pi_new to point to latest pi */
1521 pi_new = polling_island_lock(item->pi);
1522 gpr_mu_unlock(&pi_new->mu);
Sree Kuchibhotla499b94b2016-11-18 14:35:47 -08001523 GRPC_POLLING_TRACE(
1524 "add_poll_obj: bag->pi was NULL. pi_new: %p (item(%s): %p, "
1525 "bag(%s): %p)",
Craig Tillerbaa14a92017-11-03 09:09:36 -07001526 (void*)pi_new, poll_obj_string(item_type), (void*)item,
1527 poll_obj_string(bag_type), (void*)bag);
Sree Kuchibhotla5098f912016-05-31 10:58:17 -07001528 } else {
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001529 pi_new = polling_island_merge(item->pi, bag->pi, &error);
Sree Kuchibhotla1e776682016-06-28 14:09:26 -07001530 GRPC_POLLING_TRACE(
Sree Kuchibhotla499b94b2016-11-18 14:35:47 -08001531 "add_poll_obj: polling islands merged. pi_new: %p (item(%s): %p, "
1532 "bag(%s): %p)",
Craig Tillerbaa14a92017-11-03 09:09:36 -07001533 (void*)pi_new, poll_obj_string(item_type), (void*)item,
1534 poll_obj_string(bag_type), (void*)bag);
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001535 }
1536
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001537 /* At this point, pi_new is the polling island that both item->pi and bag->pi
1538 MUST be pointing to */
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -07001539
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001540 if (item->pi != pi_new) {
1541 PI_ADD_REF(pi_new, poll_obj_string(item_type));
Craig Tiller4782d922017-11-10 09:53:21 -08001542 if (item->pi != nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001543 PI_UNREF(item->pi, poll_obj_string(item_type));
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -07001544 }
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001545 item->pi = pi_new;
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -07001546 }
1547
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001548 if (bag->pi != pi_new) {
1549 PI_ADD_REF(pi_new, poll_obj_string(bag_type));
Craig Tiller4782d922017-11-10 09:53:21 -08001550 if (bag->pi != nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001551 PI_UNREF(bag->pi, poll_obj_string(bag_type));
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -07001552 }
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001553 bag->pi = pi_new;
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -07001554 }
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001555
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001556 gpr_mu_unlock(&item->mu);
1557 gpr_mu_unlock(&bag->mu);
Craig Tiller15007612016-07-06 09:36:16 -07001558
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001559 GRPC_LOG_IF_ERROR("add_poll_object", error);
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001560}
Craig Tiller57726ca2016-09-12 11:59:45 -07001561
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001562static void pollset_add_fd(grpc_pollset* pollset, grpc_fd* fd) {
1563 add_poll_object(&pollset->po, POLL_OBJ_POLLSET, &fd->po, POLL_OBJ_FD);
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001564}
1565
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001566/*******************************************************************************
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001567 * Pollset-set Definitions
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001568 */
1569
Craig Tillerbaa14a92017-11-03 09:09:36 -07001570static grpc_pollset_set* pollset_set_create(void) {
Noah Eisen4d20a662018-02-09 09:34:04 -08001571 grpc_pollset_set* pss =
1572 static_cast<grpc_pollset_set*>(gpr_malloc(sizeof(*pss)));
Sree Kuchibhotla2385fd72016-11-18 16:30:41 -08001573 gpr_mu_init(&pss->po.mu);
Craig Tiller4782d922017-11-10 09:53:21 -08001574 pss->po.pi = nullptr;
ncteisene9cd8a82017-06-29 06:03:52 -04001575#ifndef NDEBUG
Sree Kuchibhotlaa0749a62016-11-18 20:22:09 -08001576 pss->po.obj_type = POLL_OBJ_POLLSET_SET;
1577#endif
Sree Kuchibhotla2385fd72016-11-18 16:30:41 -08001578 return pss;
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001579}
1580
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001581static void pollset_set_destroy(grpc_pollset_set* pss) {
Sree Kuchibhotla2385fd72016-11-18 16:30:41 -08001582 gpr_mu_destroy(&pss->po.mu);
1583
Craig Tiller4782d922017-11-10 09:53:21 -08001584 if (pss->po.pi != nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001585 PI_UNREF(pss->po.pi, "pss_destroy");
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001586 }
Sree Kuchibhotla2385fd72016-11-18 16:30:41 -08001587
1588 gpr_free(pss);
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001589}
1590
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001591static void pollset_set_add_fd(grpc_pollset_set* pss, grpc_fd* fd) {
1592 add_poll_object(&pss->po, POLL_OBJ_POLLSET_SET, &fd->po, POLL_OBJ_FD);
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001593}
1594
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001595static void pollset_set_del_fd(grpc_pollset_set* pss, grpc_fd* fd) {
Sree Kuchibhotla2385fd72016-11-18 16:30:41 -08001596 /* Nothing to do */
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001597}
1598
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001599static void pollset_set_add_pollset(grpc_pollset_set* pss, grpc_pollset* ps) {
1600 add_poll_object(&pss->po, POLL_OBJ_POLLSET_SET, &ps->po, POLL_OBJ_POLLSET);
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001601}
1602
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001603static void pollset_set_del_pollset(grpc_pollset_set* pss, grpc_pollset* ps) {
Sree Kuchibhotla2385fd72016-11-18 16:30:41 -08001604 /* Nothing to do */
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001605}
1606
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001607static void pollset_set_add_pollset_set(grpc_pollset_set* bag,
Craig Tillerbaa14a92017-11-03 09:09:36 -07001608 grpc_pollset_set* item) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001609 add_poll_object(&bag->po, POLL_OBJ_POLLSET_SET, &item->po,
Sree Kuchibhotla2385fd72016-11-18 16:30:41 -08001610 POLL_OBJ_POLLSET_SET);
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001611}
1612
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001613static void pollset_set_del_pollset_set(grpc_pollset_set* bag,
Craig Tillerbaa14a92017-11-03 09:09:36 -07001614 grpc_pollset_set* item) {
Sree Kuchibhotla2385fd72016-11-18 16:30:41 -08001615 /* Nothing to do */
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001616}
1617
Sree Kuchibhotla2e12db92016-06-16 16:53:59 -07001618/* Test helper functions
1619 * */
Craig Tillerbaa14a92017-11-03 09:09:36 -07001620void* grpc_fd_get_polling_island(grpc_fd* fd) {
1621 polling_island* pi;
Sree Kuchibhotla2e12db92016-06-16 16:53:59 -07001622
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001623 gpr_mu_lock(&fd->po.mu);
1624 pi = fd->po.pi;
1625 gpr_mu_unlock(&fd->po.mu);
Sree Kuchibhotla2e12db92016-06-16 16:53:59 -07001626
1627 return pi;
1628}
1629
Craig Tillerbaa14a92017-11-03 09:09:36 -07001630void* grpc_pollset_get_polling_island(grpc_pollset* ps) {
1631 polling_island* pi;
Sree Kuchibhotla2e12db92016-06-16 16:53:59 -07001632
Sree Kuchibhotlaf6f33d72016-11-18 11:38:52 -08001633 gpr_mu_lock(&ps->po.mu);
1634 pi = ps->po.pi;
1635 gpr_mu_unlock(&ps->po.mu);
Sree Kuchibhotla2e12db92016-06-16 16:53:59 -07001636
1637 return pi;
1638}
1639
Craig Tillerbaa14a92017-11-03 09:09:36 -07001640bool grpc_are_polling_islands_equal(void* p, void* q) {
Noah Eisenbe82e642018-02-09 09:16:55 -08001641 polling_island* p1 = static_cast<polling_island*>(p);
1642 polling_island* p2 = static_cast<polling_island*>(q);
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -07001643
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -07001644 /* Note: polling_island_lock_pair() may change p1 and p2 to point to the
1645 latest polling islands in their respective linked lists */
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -07001646 polling_island_lock_pair(&p1, &p2);
Sree Kuchibhotla20d0a162016-06-23 15:14:03 -07001647 polling_island_unlock_pair(p1, p2);
Sree Kuchibhotla2f8ade02016-06-17 13:28:38 -07001648
1649 return p1 == p2;
Sree Kuchibhotla2e12db92016-06-16 16:53:59 -07001650}
1651
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001652/*******************************************************************************
Sree Kuchibhotla0bcbd792016-06-01 15:43:03 -07001653 * Event engine binding
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001654 */
1655
1656static void shutdown_engine(void) {
1657 fd_global_shutdown();
1658 pollset_global_shutdown();
Sree Kuchibhotlad627c102016-06-06 15:49:32 -07001659 polling_island_global_shutdown();
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001660}
1661
1662static const grpc_event_engine_vtable vtable = {
Yash Tibrewal533d1182017-09-18 10:48:22 -07001663 sizeof(grpc_pollset),
Yash Tibrewaladc733f2018-04-02 18:32:06 -07001664 true,
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001665
Yash Tibrewal533d1182017-09-18 10:48:22 -07001666 fd_create,
1667 fd_wrapped_fd,
1668 fd_orphan,
1669 fd_shutdown,
1670 fd_notify_on_read,
1671 fd_notify_on_write,
Yash Tibrewaladc733f2018-04-02 18:32:06 -07001672 fd_notify_on_error,
Yash Tibrewal533d1182017-09-18 10:48:22 -07001673 fd_is_shutdown,
1674 fd_get_read_notifier_pollset,
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001675
Yash Tibrewal533d1182017-09-18 10:48:22 -07001676 pollset_init,
1677 pollset_shutdown,
1678 pollset_destroy,
1679 pollset_work,
1680 pollset_kick,
1681 pollset_add_fd,
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001682
Yash Tibrewal533d1182017-09-18 10:48:22 -07001683 pollset_set_create,
1684 pollset_set_destroy,
1685 pollset_set_add_pollset,
1686 pollset_set_del_pollset,
1687 pollset_set_add_pollset_set,
1688 pollset_set_del_pollset_set,
1689 pollset_set_add_fd,
1690 pollset_set_del_fd,
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001691
Yash Tibrewal533d1182017-09-18 10:48:22 -07001692 shutdown_engine,
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001693};
1694
Sree Kuchibhotla72744022016-06-09 09:42:06 -07001695/* It is possible that GLIBC has epoll but the underlying kernel doesn't.
1696 * Create a dummy epoll_fd to make sure epoll support is available */
1697static bool is_epoll_available() {
1698 int fd = epoll_create1(EPOLL_CLOEXEC);
1699 if (fd < 0) {
1700 gpr_log(
1701 GPR_ERROR,
1702 "epoll_create1 failed with error: %d. Not using epoll polling engine",
1703 fd);
1704 return false;
1705 }
1706 close(fd);
1707 return true;
1708}
1709
Craig Tillerbaa14a92017-11-03 09:09:36 -07001710const grpc_event_engine_vtable* grpc_init_epollsig_linux(
Craig Tillerf8382b82017-04-27 15:09:48 -07001711 bool explicit_request) {
Sree Kuchibhotlac7be7c62016-06-09 17:08:50 -07001712 /* If use of signals is disabled, we cannot use epoll engine*/
1713 if (is_grpc_wakeup_signal_initialized && grpc_wakeup_signal < 0) {
yang-g30101b02017-11-06 14:35:30 -08001714 gpr_log(GPR_ERROR, "Skipping epollsig because use of signals is disabled.");
Craig Tiller4782d922017-11-10 09:53:21 -08001715 return nullptr;
Sree Kuchibhotlac7be7c62016-06-09 17:08:50 -07001716 }
1717
Ken Paysoncd7d0472016-10-11 12:24:20 -07001718 if (!grpc_has_wakeup_fd()) {
yang-g30101b02017-11-06 14:35:30 -08001719 gpr_log(GPR_ERROR, "Skipping epollsig because of no wakeup fd.");
Craig Tiller4782d922017-11-10 09:53:21 -08001720 return nullptr;
Ken Paysonbc544be2016-10-06 19:23:47 -07001721 }
1722
Sree Kuchibhotla72744022016-06-09 09:42:06 -07001723 if (!is_epoll_available()) {
yang-g30101b02017-11-06 14:35:30 -08001724 gpr_log(GPR_ERROR, "Skipping epollsig because epoll is unavailable.");
Craig Tiller4782d922017-11-10 09:53:21 -08001725 return nullptr;
Sree Kuchibhotla72744022016-06-09 09:42:06 -07001726 }
1727
Sree Kuchibhotlac7be7c62016-06-09 17:08:50 -07001728 if (!is_grpc_wakeup_signal_initialized) {
Sree Kuchibhotla0fda8802017-08-30 20:34:51 -07001729 if (explicit_request) {
Craig Tillerf8382b82017-04-27 15:09:48 -07001730 grpc_use_signal(SIGRTMIN + 6);
1731 } else {
yang-g30101b02017-11-06 14:35:30 -08001732 gpr_log(GPR_ERROR,
1733 "Skipping epollsig because uninitialized wakeup signal.");
Craig Tiller4782d922017-11-10 09:53:21 -08001734 return nullptr;
Craig Tillerf8382b82017-04-27 15:09:48 -07001735 }
Sree Kuchibhotlac7be7c62016-06-09 17:08:50 -07001736 }
1737
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001738 fd_global_init();
Sree Kuchibhotla3131c262016-06-21 17:28:28 -07001739
1740 if (!GRPC_LOG_IF_ERROR("pollset_global_init", pollset_global_init())) {
Craig Tiller4782d922017-11-10 09:53:21 -08001741 return nullptr;
Sree Kuchibhotla3131c262016-06-21 17:28:28 -07001742 }
1743
1744 if (!GRPC_LOG_IF_ERROR("polling_island_global_init",
1745 polling_island_global_init())) {
Craig Tiller4782d922017-11-10 09:53:21 -08001746 return nullptr;
Sree Kuchibhotla3131c262016-06-21 17:28:28 -07001747 }
1748
Sree Kuchibhotlaf448c342016-05-19 10:51:24 -07001749 return &vtable;
1750}
1751
Mehrdad Afsharifb669002018-01-17 15:37:56 -08001752#else /* defined(GRPC_LINUX_EPOLL_CREATE1) */
murgatroid99623dd4f2016-08-08 17:31:27 -07001753#if defined(GRPC_POSIX_SOCKET)
Yash Tibrewal1cac2232017-09-26 11:31:11 -07001754#include "src/core/lib/iomgr/ev_epollsig_linux.h"
Mehrdad Afsharifb669002018-01-17 15:37:56 -08001755/* If GRPC_LINUX_EPOLL_CREATE1 is not defined, it means
1756 epoll_create1 is not available. Return NULL */
Craig Tillerbaa14a92017-11-03 09:09:36 -07001757const grpc_event_engine_vtable* grpc_init_epollsig_linux(
Craig Tillerf8382b82017-04-27 15:09:48 -07001758 bool explicit_request) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001759 return nullptr;
Craig Tillerf8382b82017-04-27 15:09:48 -07001760}
murgatroid99623dd4f2016-08-08 17:31:27 -07001761#endif /* defined(GRPC_POSIX_SOCKET) */
Sree Kuchibhotla5855c472016-06-08 12:56:56 -07001762
Sree Kuchibhotlac7be7c62016-06-09 17:08:50 -07001763void grpc_use_signal(int signum) {}
Mehrdad Afsharifb669002018-01-17 15:37:56 -08001764#endif /* !defined(GRPC_LINUX_EPOLL_CREATE1) */