binaryen 1.1.6.7 → 1.1.6.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e208c99114153cd5edae8ca1125d26eeecb8f2b5f91e218ccc63f807827ee4fe
-  data.tar.gz: 6839fedb4b0c710fc1aa852909f3a24daf1729a2f1531e6c3c47b7594f2819d3
+  metadata.gz: 4217d4168d453c8e5316e9c66929d5e09057ec356dc6f80802c41521de6fd612
+  data.tar.gz: 2cb4b068c0456d72dcc55cf0d945a5357df0d3ad8c82166fca8a57dcd2908ec4
 SHA512:
-  metadata.gz: 21ec4f01eacfa0d38b08acae723f5f6bf662f42047950354c3260b9d461b774ba91749b82b162a74b83ef0e70aa4b6541fcbbbb70111cd30b11b969955a2cfec
-  data.tar.gz: 5dbbf03fe6f8325cdaa1e78c81bfb7feef07d1887d53a647202f56dc58e5ac1b628a2057f0e5cce083bd2868894e622054f1a999d476a568ef04a5c1558a2d72
+  metadata.gz: 9e3fd947316b466a8d592014bc57b1c34148da5b3042e0630b2bc5f56e54c287ab4c6416e78c1ef392e88429cbbcba0c0c3549bdd8b99d7068972c1adeb0c698
+  data.tar.gz: 2cdfe8e57ab692f16e911e7a819cb6cfecbf9b31027a04c8863eb919fd952cb6d92832d3884768cf05704507c7c408bcfb19eed3c47b90dfcbb6f1e7eaefc679

data/lib/binaryen/command.rb

@@ -1,50 +1,108 @@
 # frozen_string_literal: true
 
-require "English"
-require "shellwords"
-require "timeout"
 require "posix/spawn"
+require "timeout"
+require "tempfile"
 
 module Binaryen
-  # Wrapper around a binaryen executable command with a timeout and streaming IO.
-  #
-  # @example Running wasm-opt
-  #
-  #   ```ruby
-  #   command = Binaryen::Command.new("wasm-opt", timeout: 10)
-  #   optimized_wasm = command.run("-O4", stdin: "(module)")
-  #   ```
   class Command
-    MAX_OUTPUT_SIZE = 256 * 1024 * 1024
+    include POSIX::Spawn
+    DEFAULT_MAX_OUTPUT_SIZE = 256 * 1024 * 1024
+    DEFAULT_TIMEOUT = 10
     DEFAULT_ARGS_FOR_COMMAND = {
       "wasm-opt" => ["--output=-"],
     }.freeze
 
-    def initialize(cmd, timeout: 10, ignore_missing: false)
+    def initialize(cmd, timeout: DEFAULT_TIMEOUT, max_output_size: DEFAULT_MAX_OUTPUT_SIZE, ignore_missing: false)
       @cmd = command_path(cmd, ignore_missing) || raise(ArgumentError, "command not found: #{cmd}")
       @timeout = timeout
       @default_args = DEFAULT_ARGS_FOR_COMMAND.fetch(cmd, [])
+      @max_output_size = max_output_size
     end
 
     def run(*arguments, stdin: nil, stderr: nil)
       args = [@cmd] + arguments + @default_args
-      child = POSIX::Spawn::Child.new(*args, input: stdin, timeout: @timeout, max: MAX_OUTPUT_SIZE)
-      stderr&.write(child.err)
-      status = child.status
 
-      raise Binaryen::NonZeroExitStatus, "command exited with status #{status.exitstatus}" unless status.success?
-
-      child.out
-    rescue POSIX::Spawn::MaximumOutputExceeded => e
-      raise Binaryen::MaximumOutputExceeded, e.message
-    rescue POSIX::Spawn::TimeoutExceeded => e
-      raise Timeout::Error, e.message
+      if stdin
+        with_stdin_tempfile(stdin) { |path| spawn_command(*args, path, stderr: stderr) }
+      else
+        spawn_command(*args, stderr: stderr)
+      end
     end
 
     private
 
+    def with_stdin_tempfile(content)
+      Tempfile.open("binaryen-stdin") do |f|
+        f.binmode
+        f.write(content)
+        f.close
+        yield f.path
+      end
+    end
+
     def command_path(cmd, ignore_missing)
       Dir[File.join(Binaryen.bindir, cmd)].first || (ignore_missing && cmd)
     end
+
+    def spawn_command(*args, stderr: nil)
+      out = "".b
+      data_buffer = "".b
+      start_time = Process.clock_gettime(Process::CLOCK_MONOTONIC)
+      pid, stdin, stdout, stderr_stream = popen4(*args)
+      stdin.close
+      @pid = pid
+      readers = [stdout, stderr_stream]
+
+      while readers.any?
+        elapsed_time = Process.clock_gettime(Process::CLOCK_MONOTONIC) - start_time
+        remaining_time = @timeout - elapsed_time
+        ready = IO.select(readers, nil, nil, remaining_time)
+        raise Timeout::Error, "command timed out after #{@timeout} seconds" if ready.nil?
+
+        ready[0].each do |io|
+          max_amount_to_read = @max_output_size - out.bytesize + 1
+          data = io.read_nonblock(max_amount_to_read, data_buffer, exception: false)
+
+          if data == :wait_readable
+            # If the IO object is not ready for reading, read_nonblock returns :wait_readable.
+            # This isn't an error, but a notification.
+            next
+          elsif data.nil?
+            # At EOF, read_nonblock returns nil instead of raising EOFError.
+            readers.delete(io)
+          elsif io == stdout
+            out << data_buffer
+          elsif io == stderr_stream && stderr
+            stderr << data_buffer
+          end
+        rescue Errno::EPIPE, Errno::EINTR
+          # Handle EPIPE and EINTR errors
+          readers.delete(io)
+        end
+
+        if out.bytesize > @max_output_size
+          Process.kill("TERM", @pid)
+          raise Binaryen::MaximumOutputExceeded, "maximum output size exceeded (#{@max_output_size} bytes)"
+        end
+
+        if remaining_time < 0
+          Process.kill("TERM", @pid)
+          raise Timeout::Error, "command timed out after #{@timeout} seconds"
+        end
+      end
+
+      _, status = Process.waitpid2(pid, Process::WUNTRACED)
+
+      raise Binaryen::NonZeroExitStatus, "command exited with status #{status.exitstatus}" unless status.success?
+
+      out
+    ensure
+      [stdin, stdout, stderr_stream].each do |io|
+        io.close
+      rescue
+        nil
+      end
+    end
   end
 end

data/lib/binaryen/version.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
 module Binaryen
-  VERSION = "1.1.6.7"
+  VERSION = "1.1.6.8"
   BINARYEN_VERSION = "version_116"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: binaryen
 version: !ruby/object:Gem::Version
-  version: 1.1.6.7
+  version: 1.1.6.8
 platform: ruby
 authors:
 - Shopify Inc.
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-12-18 00:00:00.000000000 Z
+date: 2023-12-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: posix-spawn