iou 0.1

data/ext/iou/iou.h

@@ -0,0 +1,66 @@
+#ifndef IOU_H
+#define IOU_H
+
+#include "ruby.h"
+#include <liburing.h>
+
+// debugging
+#define OBJ_ID(obj) (NUM2LONG(rb_funcall(obj, rb_intern("object_id"), 0)))
+#define INSPECT(str, obj) { printf(str); VALUE s = rb_funcall(obj, rb_intern("inspect"), 0); printf(": %s\n", StringValueCStr(s)); }
+#define CALLER() rb_funcall(rb_mKernel, rb_intern("caller"), 0)
+#define TRACE_CALLER() INSPECT("caller: ", CALLER())
+#define TRACE_FREE(ptr) //printf("Free %p %s:%d\n", ptr, __FILE__, __LINE__)
+
+// branching
+#ifndef unlikely
+#define unlikely(cond)	__builtin_expect(!!(cond), 0)
+#endif
+
+#ifndef likely
+#define likely(cond)	__builtin_expect(!!(cond), 1)
+#endif
+
+struct buf_ring_descriptor {
+  struct io_uring_buf_ring *br;
+  size_t br_size;
+  // struct io_uring_buf_ring *buf_ring;
+  unsigned buf_count;
+  unsigned buf_size;
+	char *buf_base;
+  // size_t buf_ring_size;
+};
+
+#define BUFFER_RING_MAX_COUNT 10
+
+typedef struct IOU_t {
+  struct io_uring ring;
+  unsigned int    ring_initialized;
+  unsigned int    op_counter;
+  unsigned int    unsubmitted_sqes;
+  VALUE           pending_ops;
+
+  struct buf_ring_descriptor brs[BUFFER_RING_MAX_COUNT];
+  unsigned int br_counter;
+} IOU_t;
+
+struct sa_data {
+  struct sockaddr addr;
+  socklen_t len;
+};
+
+typedef struct OpSpecData_t {
+  union {
+    struct __kernel_timespec ts;
+    struct sa_data sa;
+  } data;
+} OpSpecData_t;
+
+extern VALUE mIOU;
+extern VALUE cOpSpecData;
+
+struct __kernel_timespec *OpSpecData_ts_get(VALUE self);
+void OpSpecData_ts_set(VALUE self, VALUE value);
+
+struct sa_data *OpSpecData_sa_get(VALUE self);
+
+#endif // IOU_H

data/ext/iou/iou_ext.c

@@ -0,0 +1,9 @@
+#include "iou.h"
+
+void Init_IOU();
+void Init_OpSpecData();
+
+void Init_iou_ext(void) {
+  Init_IOU();
+  Init_OpSpecData();
+}

data/ext/iou/op_spec_data.c

@@ -0,0 +1,61 @@
+#include "iou.h"
+
+VALUE cOpSpecData;
+
+static size_t OpSpecData_size(const void *ptr) {
+  return sizeof(OpSpecData_t);
+}
+
+static const rb_data_type_t OpSpecData_type = {
+    "OpSpecData",
+    {0, 0, OpSpecData_size, 0},
+    0, 0, RUBY_TYPED_FREE_IMMEDIATELY | RUBY_TYPED_WB_PROTECTED
+};
+
+static VALUE OpSpecData_allocate(VALUE klass) {
+  OpSpecData_t *osd = ALLOC(OpSpecData_t);
+
+  return TypedData_Wrap_Struct(klass, &OpSpecData_type, osd);
+}
+
+VALUE OpSpecData_initialize(VALUE self) {
+  OpSpecData_t *osd = RTYPEDDATA_DATA(self);
+  memset(&osd->data, 0, sizeof(osd->data));
+  return self;
+}
+
+struct __kernel_timespec *OpSpecData_ts_get(VALUE self) {
+  OpSpecData_t *osd = RTYPEDDATA_DATA(self);
+  return &osd->data.ts;
+}
+
+inline struct __kernel_timespec double_to_timespec(double value) {
+  double integral;
+  double fraction = modf(value, &integral);
+  struct __kernel_timespec ts;
+  ts.tv_sec = integral;
+  ts.tv_nsec = floor(fraction * 1000000000);
+  return ts;
+}
+
+inline struct __kernel_timespec value_to_timespec(VALUE value) {
+  return double_to_timespec(NUM2DBL(value));
+}
+
+void OpSpecData_ts_set(VALUE self, VALUE value) {
+  OpSpecData_t *osd = RTYPEDDATA_DATA(self);
+  osd->data.ts = value_to_timespec(value);
+}
+
+struct sa_data *OpSpecData_sa_get(VALUE self) {
+  OpSpecData_t *osd = RTYPEDDATA_DATA(self);
+  return &osd->data.sa;
+}
+
+void Init_OpSpecData(void) {
+  mIOU = rb_define_module("IOU");
+  cOpSpecData = rb_define_class_under(mIOU, "OpSpecData", rb_cObject);
+  rb_define_alloc_func(cOpSpecData, OpSpecData_allocate);
+
+  rb_define_method(cOpSpecData, "initialize", OpSpecData_initialize, 0);
+}

data/iou.gemspec

@@ -0,0 +1,27 @@
+require_relative './lib/iou/version'
+
+Gem::Specification.new do |s|
+  s.name        = 'iou'
+  s.version     = IOU::VERSION
+  s.licenses    = ['MIT']
+  s.summary     = 'io_uring for Ruby'
+  s.author      = 'Sharon Rosner'
+  s.email       = 'sharon@noteflakes.com'
+  s.files       = `git ls-files --recurse-submodules`.split.reject { |fn| fn =~ /liburing\/man/ }
+  s.homepage    = 'https://github.com/digital-fabric/iou'
+  s.metadata    = {
+    "source_code_uri" => "https://github.com/digital-fabric/iou",
+    "documentation_uri" => "https://www.rubydoc.info/gems/iou",
+    "changelog_uri" => "https://github.com/digital-fabric/iou/blob/master/CHANGELOG.md"
+  }
+  s.rdoc_options = ["--title", "IOU", "--main", "README.md"]
+  s.extra_rdoc_files = ["README.md"]
+  s.extensions = ["ext/iou/extconf.rb"]
+  s.require_paths = ["lib"]
+  s.required_ruby_version = '>= 3.3'
+
+  s.add_development_dependency  'rake-compiler',        '1.2.7'
+  s.add_development_dependency  'minitest',             '5.25.1'
+  s.add_development_dependency  'http_parser.rb',       '0.8.0'
+  s.add_development_dependency  'benchmark-ips',        '2.10.0'
+end

data/lib/iou/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module IOU
+  VERSION = '0.1'
+end

data/lib/iou.rb

@@ -0,0 +1,3 @@
+# frozen_string_literal: true
+
+require_relative './iou_ext'

data/test/helper.rb

@@ -0,0 +1,59 @@
+# frozen_string_literal: true
+
+require 'bundler/setup'
+require_relative './coverage' if ENV['COVERAGE']
+require 'iou'
+require 'minitest/autorun'
+
+module ::Kernel
+  def debug(**h)
+    k, v = h.first
+    h.delete(k)
+
+    rest = h.inject(+'') { |s, (k, v)| s << "  #{k}: #{v.inspect}\n" }
+    STDOUT.orig_write("#{k}=>#{v} #{caller[0]}\n#{rest}")
+  end
+
+  def trace(*args)
+    STDOUT.orig_write(format_trace(args))
+  end
+
+  def format_trace(args)
+    if args.first.is_a?(String)
+      if args.size > 1
+        format("%s: %p\n", args.shift, args)
+      else
+        format("%s\n", args.first)
+      end
+    else
+      format("%p\n", args.size == 1 ? args.first : args)
+    end
+  end
+
+  def monotonic_clock
+    ::Process.clock_gettime(::Process::CLOCK_MONOTONIC)
+  end
+end
+
+module Minitest::Assertions
+  # def setup
+  #   sleep 0.0001
+  # end
+
+  def assert_in_range exp_range, act
+    msg = message(msg) { "Expected #{mu_pp(act)} to be in range #{mu_pp(exp_range)}" }
+    assert exp_range.include?(act), msg
+  end
+end
+
+class IOURingBaseTest < Minitest::Test
+  attr_accessor :ring
+  
+  def setup
+    @ring = IOU::Ring.new
+  end
+
+  def teardown
+    ring.close
+  end
+end