uringmachine 0.3 → 0.5

data/examples/write_dev_null.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+require_relative '../lib/uringmachine'
+
+ITERATIONS = 1000
+DEV_NULL = File.open('/dev/null', 'w')
+FD = DEV_NULL.fileno
+BUF = ('*' * 8192).freeze
+
+$machine = UringMachine.new
+
+def run_um_write
+  $machine.write(FD, BUF)
+end
+
+1000.times { run_um_write }

data/ext/um/extconf.rb

@@ -8,8 +8,9 @@ dir_config 'um_ext'
 
 KERNEL_INFO_RE = /Linux (\d)\.(\d+)(?:\.)?((?:\d+\.?)*)(?:\-)?([\w\-]+)?/
 def get_config
-  config = { linux: !!(RUBY_PLATFORM =~ /linux/) }
-  raise "UringMachine only works on Linux!" if !config[:linux]
+  if RUBY_PLATFORM !~ /linux/
+    raise "UringMachine only works on Linux!"
+  end
 
   kernel_info = `uname -sr`
   m = kernel_info.match(KERNEL_INFO_RE)
@@ -20,17 +21,18 @@ def get_config
   combined_version = version.to_i * 100 + major_revision.to_i
   raise "UringMachine requires kernel version 6.4 or newer!" if combined_version < 604
 
-  config[:kernel_version]     = combined_version
-  config[:pidfd_open]         = combined_version > 503
-  config[:multishot_accept]   = combined_version >= 519
-  config[:multishot_recv]     = combined_version >= 600
-  config[:multishot_recvmsg]  = combined_version >= 600
-  config[:multishot_timeout]  = combined_version >= 604
-  config[:submit_all_flag]    = combined_version >= 518
-  config[:coop_taskrun_flag]  = combined_version >= 519
-  config[:single_issuer_flag] = combined_version >= 600
-
-  config
+  {
+    kernel_version:       combined_version,
+    submit_all_flag:      combined_version >= 518,
+    coop_taskrun_flag:    combined_version >= 519,
+    single_issuer_flag:   combined_version >= 600,
+    prep_bind:            combined_version >= 611,
+    prep_listen:          combined_version >= 611,
+    prep_cmd_sock:        combined_version >= 607,
+    prep_futex:           combined_version >= 607,
+    prep_waitid:          combined_version >= 607,
+    prep_read_multishot:  combined_version >= 607
+  }
 end
 
 config = get_config
@@ -54,16 +56,15 @@ if !find_library('uring', nil, File.join(liburing_path, 'src'))
   raise "Couldn't find liburing.a"
 end
 
-def define_bool(name, value)
-  $defs << "-D#{name}=#{value ? 1 : 0 }"
-end
-
-$defs << '-DHAVE_IO_URING_PREP_MULTISHOT_ACCEPT'  if config[:multishot_accept]
-$defs << '-DHAVE_IO_URING_PREP_RECV_MULTISHOT'    if config[:multishot_recv]
-$defs << '-DHAVE_IO_URING_PREP_RECVMSG_MULTISHOT' if config[:multishot_recvmsg]
-$defs << '-DHAVE_IO_URING_TIMEOUT_MULTISHOT'      if config[:multishot_timeout]
-$defs << '-DHAVE_IORING_SETUP_SUBMIT_ALL'         if config[:submit_all_flag]
-$defs << '-DHAVE_IORING_SETUP_COOP_TASKRUN'       if config[:coop_taskrun_flag]
+$defs << "-DUM_KERNEL_VERSION=#{config[:kernel_version]}"
+$defs << '-DHAVE_IORING_SETUP_SUBMIT_ALL'       if config[:submit_all_flag]
+$defs << '-DHAVE_IORING_SETUP_COOP_TASKRUN'     if config[:coop_taskrun_flag]
+$defs << '-DHAVE_IO_URING_PREP_BIND'            if config[:prep_bind]
+$defs << '-DHAVE_IO_URING_PREP_LISTEN'          if config[:prep_listen]
+$defs << '-DHAVE_IO_URING_PREP_CMD_SOCK'        if config[:prep_cmd_sock]
+$defs << '-DHAVE_IO_URING_PREP_FUTEX'           if config[:prep_futex]
+$defs << '-DHAVE_IO_URING_PREP_WAITID'          if config[:prep_waitid]
+$defs << '-DHAVE_IO_URING_PREP_READ_MULTISHOT'  if config[:prep_read_multishot]
 $CFLAGS << ' -Wno-pointer-arith'
 
 CONFIG['optflags'] << ' -fno-strict-aliasing'