1 # -*- encoding: binary -*-
2 require 'io_splice_ext'
7 # The maximum default capacity of the pipe in bytes.
8 # Under stock Linux, this is 65536 bytes as of 2.6.11, and 4096 before
9 # We detect this at runtime as it is easy to recompile the kernel
10 # and set a new value.
11 # Starting with Linux 2.6.35, pipe capacity will be tunable
12 # and this will only represent the default capacity of a
19 n += wr.write_nonblock(buf)
28 def self.need_open?(obj) # :nodoc:
29 return false if obj.respond_to?(:to_io)
30 obj.respond_to?(:to_path) || obj.kind_of?(String)
33 # copies the contents of the IO object given by +src+ to +dst+
34 # If +len+ is specified, then only +len+ bytes are copied and
35 # +EOFError+ is raised if fewer than +len+ bytes could be copied.
36 # Otherwise the copy will be until EOF is reached on the +src+.
37 # +src+ and +dst+ must be IO objects or respond to +to_io+
39 # This is nearly a drop-in replacement for IO.copy_stream (in Ruby 1.9)
40 # but does not take into account userspace I/O buffers nor IO-like
41 # objects with no underlying file descriptor (e.g. StringIO).
42 def self.copy_stream(src, dst, len = nil, src_offset = nil)
44 need_open?(src) and close << (src = File.open(src))
45 need_open?(dst) and close << (dst = File.open(dst, "w"))
47 src, dst = src.to_io, dst.to_io
49 if src.stat.pipe? || dst.stat.pipe?
51 len -= full(src, dst, len, src_offset) until len == 0
54 while n = partial(src, dst, PIPE_CAPA, src_offset)
56 src_offset += n if src_offset
63 while len != 0 && n = partial(src, w, len, src_offset)
64 src_offset += n if src_offset
65 len -= full(r, dst, n, nil)
69 while n = partial(src, w, PIPE_CAPA, src_offset)
70 src_offset += n if src_offset
71 rv += full(r, dst, n, nil)
78 close.each { |io| io.close }
81 # splice the full amount specified from +src+ to +dst+
82 # Either +dst+ or +src+ must be a pipe. +dst+ and +src+
83 # may BOTH be pipes in Linux 2.6.31 or later.
84 # This will block and wait for IO completion of +len+
85 # Raises +EOFError+ if end of file is reached.
86 # bytes. Returns the number of bytes actually spliced (always +len+)
87 def self.full(src, dst, len, src_offset)
88 IO.splice(src, src_offset, dst, nil, len, F_MOVE | WAITALL)
91 # splice up to +len+ bytes from +src+ to +dst+.
92 # Either +dst+ or +src+ must be a pipe. +dst+ and +src+
93 # may BOTH be pipes in Linux 2.6.31 or later.
94 # Returns the number of bytes actually spliced.
95 # Like IO#readpartial, this never returns Errno::EAGAIN
96 def self.partial(src, dst, len, src_offset)
97 IO.splice(src, src_offset, dst, nil, len, F_MOVE)
103 IO.select(nil, [dst])
104 rv = IO.trysplice(src, src_offset, dst, nil, len, F_MOVE)
105 end while rv == :EAGAIN
109 if (! defined?(RUBY_ENGINE) || RUBY_ENGINE == "ruby") &&
110 RUBY_VERSION.to_f <= 1.8
111 require "io/splice/mri_18"