1 files changed, 26 insertions, 0 deletions
diff --git a/lib/yahns/http_client.rb b/lib/yahns/http_client.rb
index 1d64e08..1cdaa0f 100644
--- a/lib/yahns/http_client.rb
+++ b/lib/yahns/http_client.rb
@@ -2,6 +2,12 @@
  # Copyright (C) 2013-2016 all contributors <yahns-public@yhbt.net>
  # License: GPL-3.0+ (https://www.gnu.org/licenses/gpl-3.0.txt)
  # frozen_string_literal: true
+begin
+  raise LoadError, 'SENDFILE_BROKEN env set' if ENV['SENDFILE_BROKEN']
+  require 'sendfile'
+rescue LoadError
+end
+
  class Yahns::HttpClient < Kgio::Socket # :nodoc:
    NULL_IO = StringIO.new(''.dup) # :nodoc:
  
@@ -304,4 +310,24 @@ class Yahns::HttpClient < Kgio::Socket # :nodoc:
      body.close if body.respond_to?(:close)
      true
    end
+
+  def trysendio(io, offset, count)
+    return 0 if count == 0
+    count = 0x4000 if count > 0x4000
+    buf = Thread.current[:yahns_sfbuf] ||= ''.dup
+    io.pos = offset
+    str = io.read(count, buf) or return # nil for EOF
+    n = 0
+    case rv = kgio_trywrite(str)
+    when String # partial write, keep trying
+      n += (str.size - rv.size)
+      str = rv
+    when :wait_writable, :wait_readable
+      return n > 0 ? n : rv
+    when nil
+      return n + str.size # yay!
+    end while true
+  end
+
+  alias trysendfile trysendio unless IO.instance_methods.include?(:trysendfile)
  end