Add scheduler hook Addrinfo.getaddrinfo.

Author: bruno-

ext/socket/raddrinfo.c

@@ -10,6 +10,8 @@
 
 #include "rubysocket.h"
 
+#include "internal/scheduler.h"
+
 #if defined(INET6) && (defined(LOOKUP_ORDER_HACK_INET) || defined(LOOKUP_ORDER_HACK_INET6))
 #define LOOKUP_ORDERS (sizeof(lookup_order_table) / sizeof(lookup_order_table[0]))
 static const int lookup_order_table[] = {
@@ -2614,6 +2616,12 @@ static ID id_timeout;
 static VALUE
 addrinfo_s_getaddrinfo(int argc, VALUE *argv, VALUE self)
 {
+ VALUE scheduler = rb_scheduler_current();
+
+ if (scheduler != Qnil) {
+ return rb_scheduler_address_resolve(scheduler, argc, argv);
+ }
+
  VALUE node, service, family, socktype, protocol, flags, opts, timeout;
 
  rb_scan_args(argc, argv, "24:", &node, &service, &family, &socktype,

internal/scheduler.h

@@ -38,4 +38,6 @@ VALUE rb_scheduler_io_read(VALUE scheduler, VALUE io, VALUE buffer, size_t offse
 int rb_scheduler_supports_io_write(VALUE scheduler);
 VALUE rb_scheduler_io_write(VALUE scheduler, VALUE io, VALUE buffer, size_t offset, size_t length);
 
+VALUE rb_scheduler_address_resolve(VALUE scheduler, int argc, VALUE * argv);
+
 #endif /* RUBY_SCHEDULER_H */

scheduler.c

@@ -23,6 +23,8 @@ static ID id_io_read;
 static ID id_io_write;
 static ID id_io_wait;
 
+static ID id_address_resolve;
+
 void
 Init_Scheduler(void)
 {
@@ -36,6 +38,8 @@ Init_Scheduler(void)
  id_io_read = rb_intern_const("io_read");
  id_io_write = rb_intern_const("io_write");
  id_io_wait = rb_intern_const("io_wait");
+
+ id_address_resolve = rb_intern_const("address_resolve");
 }
 
 VALUE
@@ -172,3 +176,9 @@ rb_scheduler_io_write(VALUE scheduler, VALUE io, VALUE buffer, size_t offset, si
  // We should ensure string has capacity to receive data, and then resize it afterwards.
  return rb_funcall(scheduler, id_io_write, 4, io, buffer, SIZET2NUM(offset), SIZET2NUM(length));
 }
+
+VALUE
+rb_scheduler_address_resolve(VALUE scheduler, int argc, VALUE * argv)
+{
+ return rb_funcallv(scheduler, id_address_resolve, argc, argv);
+}