murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 1 | /* |
| 2 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 3 | * Copyright 2016 gRPC authors. |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 4 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 5 | * 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 |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 8 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 9 | * http://www.apache.org/licenses/LICENSE-2.0 |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 10 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 11 | * 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. |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 16 | * |
| 17 | */ |
| 18 | |
Nicolas "Pixel" Noble | d51d121 | 2016-01-31 11:33:19 +0100 | [diff] [blame] | 19 | #include <ruby/ruby.h> |
Nicolas "Pixel" Noble | 9fcdc87 | 2016-05-05 06:15:34 +0200 | [diff] [blame] | 20 | |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 21 | #include "rb_event_thread.h" |
Craig Tiller | 5b1c5f2 | 2017-04-19 09:52:18 -0700 | [diff] [blame] | 22 | #include "rb_grpc_imports.generated.h" |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 23 | |
| 24 | #include <stdbool.h> |
| 25 | |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 26 | #include <grpc/support/alloc.h> |
Craig Tiller | 5b1c5f2 | 2017-04-19 09:52:18 -0700 | [diff] [blame] | 27 | #include <grpc/support/log.h> |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 28 | #include <grpc/support/sync.h> |
| 29 | #include <grpc/support/time.h> |
Craig Tiller | 5b1c5f2 | 2017-04-19 09:52:18 -0700 | [diff] [blame] | 30 | #include <ruby/thread.h> |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 31 | |
| 32 | typedef struct grpc_rb_event { |
| 33 | // callback will be called with argument while holding the GVL |
Craig Tiller | baa14a9 | 2017-11-03 09:09:36 -0700 | [diff] [blame] | 34 | void (*callback)(void*); |
| 35 | void* argument; |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 36 | |
Craig Tiller | baa14a9 | 2017-11-03 09:09:36 -0700 | [diff] [blame] | 37 | struct grpc_rb_event* next; |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 38 | } grpc_rb_event; |
| 39 | |
| 40 | typedef struct grpc_rb_event_queue { |
Craig Tiller | baa14a9 | 2017-11-03 09:09:36 -0700 | [diff] [blame] | 41 | grpc_rb_event* head; |
| 42 | grpc_rb_event* tail; |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 43 | |
murgatroid99 | cc2b8d4 | 2016-01-07 17:32:34 -0800 | [diff] [blame] | 44 | gpr_mu mu; |
| 45 | gpr_cv cv; |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 46 | |
| 47 | // Indicates that the thread should stop waiting |
| 48 | bool abort; |
| 49 | } grpc_rb_event_queue; |
| 50 | |
| 51 | static grpc_rb_event_queue event_queue; |
| 52 | |
Craig Tiller | baa14a9 | 2017-11-03 09:09:36 -0700 | [diff] [blame] | 53 | void grpc_rb_event_queue_enqueue(void (*callback)(void*), void* argument) { |
| 54 | grpc_rb_event* event = gpr_malloc(sizeof(grpc_rb_event)); |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 55 | event->callback = callback; |
| 56 | event->argument = argument; |
| 57 | event->next = NULL; |
murgatroid99 | cc2b8d4 | 2016-01-07 17:32:34 -0800 | [diff] [blame] | 58 | gpr_mu_lock(&event_queue.mu); |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 59 | if (event_queue.tail == NULL) { |
| 60 | event_queue.head = event_queue.tail = event; |
| 61 | } else { |
| 62 | event_queue.tail->next = event; |
| 63 | event_queue.tail = event; |
| 64 | } |
murgatroid99 | cc2b8d4 | 2016-01-07 17:32:34 -0800 | [diff] [blame] | 65 | gpr_cv_signal(&event_queue.cv); |
| 66 | gpr_mu_unlock(&event_queue.mu); |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 67 | } |
| 68 | |
Craig Tiller | baa14a9 | 2017-11-03 09:09:36 -0700 | [diff] [blame] | 69 | static grpc_rb_event* grpc_rb_event_queue_dequeue() { |
| 70 | grpc_rb_event* event; |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 71 | if (event_queue.head == NULL) { |
| 72 | event = NULL; |
| 73 | } else { |
| 74 | event = event_queue.head; |
| 75 | if (event_queue.head->next == NULL) { |
| 76 | event_queue.head = event_queue.tail = NULL; |
| 77 | } else { |
| 78 | event_queue.head = event_queue.head->next; |
| 79 | } |
| 80 | } |
| 81 | return event; |
| 82 | } |
| 83 | |
| 84 | static void grpc_rb_event_queue_destroy() { |
murgatroid99 | cc2b8d4 | 2016-01-07 17:32:34 -0800 | [diff] [blame] | 85 | gpr_mu_destroy(&event_queue.mu); |
| 86 | gpr_cv_destroy(&event_queue.cv); |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 87 | } |
| 88 | |
Craig Tiller | baa14a9 | 2017-11-03 09:09:36 -0700 | [diff] [blame] | 89 | static void* grpc_rb_wait_for_event_no_gil(void* param) { |
| 90 | grpc_rb_event* event = NULL; |
Nicolas "Pixel" Noble | 7b92db6 | 2016-01-26 22:46:34 +0100 | [diff] [blame] | 91 | (void)param; |
murgatroid99 | cc2b8d4 | 2016-01-07 17:32:34 -0800 | [diff] [blame] | 92 | gpr_mu_lock(&event_queue.mu); |
Alexander Polcyn | b2c0b7b | 2017-04-27 00:26:25 -0700 | [diff] [blame] | 93 | while (!event_queue.abort) { |
| 94 | if ((event = grpc_rb_event_queue_dequeue()) != NULL) { |
| 95 | gpr_mu_unlock(&event_queue.mu); |
| 96 | return event; |
| 97 | } |
Craig Tiller | 5b1c5f2 | 2017-04-19 09:52:18 -0700 | [diff] [blame] | 98 | gpr_cv_wait(&event_queue.cv, &event_queue.mu, |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 99 | gpr_inf_future(GPR_CLOCK_REALTIME)); |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 100 | } |
murgatroid99 | cc2b8d4 | 2016-01-07 17:32:34 -0800 | [diff] [blame] | 101 | gpr_mu_unlock(&event_queue.mu); |
Alexander Polcyn | b2c0b7b | 2017-04-27 00:26:25 -0700 | [diff] [blame] | 102 | return NULL; |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 103 | } |
| 104 | |
Craig Tiller | baa14a9 | 2017-11-03 09:09:36 -0700 | [diff] [blame] | 105 | static void grpc_rb_event_unblocking_func(void* arg) { |
Nicolas "Pixel" Noble | 7b92db6 | 2016-01-26 22:46:34 +0100 | [diff] [blame] | 106 | (void)arg; |
murgatroid99 | cc2b8d4 | 2016-01-07 17:32:34 -0800 | [diff] [blame] | 107 | gpr_mu_lock(&event_queue.mu); |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 108 | event_queue.abort = true; |
murgatroid99 | cc2b8d4 | 2016-01-07 17:32:34 -0800 | [diff] [blame] | 109 | gpr_cv_signal(&event_queue.cv); |
| 110 | gpr_mu_unlock(&event_queue.mu); |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 111 | } |
| 112 | |
| 113 | /* This is the implementation of the thread that handles auth metadata plugin |
| 114 | * events */ |
| 115 | static VALUE grpc_rb_event_thread(VALUE arg) { |
Craig Tiller | baa14a9 | 2017-11-03 09:09:36 -0700 | [diff] [blame] | 116 | grpc_rb_event* event; |
Nicolas "Pixel" Noble | 7b92db6 | 2016-01-26 22:46:34 +0100 | [diff] [blame] | 117 | (void)arg; |
Craig Tiller | 5b1c5f2 | 2017-04-19 09:52:18 -0700 | [diff] [blame] | 118 | while (true) { |
Craig Tiller | baa14a9 | 2017-11-03 09:09:36 -0700 | [diff] [blame] | 119 | event = (grpc_rb_event*)rb_thread_call_without_gvl( |
Craig Tiller | 5b1c5f2 | 2017-04-19 09:52:18 -0700 | [diff] [blame] | 120 | grpc_rb_wait_for_event_no_gil, NULL, grpc_rb_event_unblocking_func, |
| 121 | NULL); |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 122 | if (event == NULL) { |
| 123 | // Indicates that the thread needs to shut down |
| 124 | break; |
| 125 | } else { |
| 126 | event->callback(event->argument); |
| 127 | gpr_free(event); |
| 128 | } |
| 129 | } |
| 130 | grpc_rb_event_queue_destroy(); |
| 131 | return Qnil; |
| 132 | } |
| 133 | |
| 134 | void grpc_rb_event_queue_thread_start() { |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 135 | event_queue.head = event_queue.tail = NULL; |
| 136 | event_queue.abort = false; |
murgatroid99 | cc2b8d4 | 2016-01-07 17:32:34 -0800 | [diff] [blame] | 137 | gpr_mu_init(&event_queue.mu); |
| 138 | gpr_cv_init(&event_queue.cv); |
murgatroid99 | 9b83bd7 | 2016-01-05 09:50:04 -0800 | [diff] [blame] | 139 | |
| 140 | rb_thread_create(grpc_rb_event_thread, NULL); |
| 141 | } |