binaryen 1.1.6.7 → 1.1.6.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e208c99114153cd5edae8ca1125d26eeecb8f2b5f91e218ccc63f807827ee4fe
-  data.tar.gz: 6839fedb4b0c710fc1aa852909f3a24daf1729a2f1531e6c3c47b7594f2819d3
+  metadata.gz: 18c535d2045e224d5ba46b3117e7eced2ff502b0cb338c6e7307113f011cb5fa
+  data.tar.gz: 8d1a5f09b94e86221e142d0773c8b23ea0a1987275a322a175daf84db2b2e01b
 SHA512:
-  metadata.gz: 21ec4f01eacfa0d38b08acae723f5f6bf662f42047950354c3260b9d461b774ba91749b82b162a74b83ef0e70aa4b6541fcbbbb70111cd30b11b969955a2cfec
-  data.tar.gz: 5dbbf03fe6f8325cdaa1e78c81bfb7feef07d1887d53a647202f56dc58e5ac1b628a2057f0e5cce083bd2868894e622054f1a999d476a568ef04a5c1558a2d72
+  metadata.gz: dea3bf6b7f13d19fc83dc18632092306650ae3b077f2221fa62c0f35beaee50a4acf54ff50f114049da1bdf4b4485076f19e3039e9ec714b158468fb07286850
+  data.tar.gz: ac18a83b7221afaa90c2643ccad41af25f1be1bac141011afb056cc16a824e98c7c75832fe03b7f18b703184479faff1fe9bff127eb255cec321d50904a2e35e

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)
+            io.close
+          elsif io == stdout
+            out << data_buffer
+          elsif io == stderr_stream && stderr
+            stderr << data_buffer
+          end
+        rescue Errno::EINTR
+          # This means that the read was interrupted by a signal, which is not an error. So we just retry.
+        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.9"
   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.9
 platform: ruby
 authors:
 - Shopify Inc.
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-12-18 00:00:00.000000000 Z
+date: 2024-01-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: posix-spawn
@@ -106,7 +106,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.22
+rubygems_version: 3.5.4
 signing_key: 
 specification_version: 4
 summary: Vendors binaryen libraries, headers, and executables for use in Ruby