iv-org-mirror/invidious-copy-2022-03-16
1
0
Fork 0
mirror of https://gitea.invidious.io/iv-org/invidious-copy-2022-03-16.git synced 2024-08-15 00:53:18 +00:00
Code Issues Projects Releases Packages Wiki Activity

Catch connection reset when proxying files

Browse source
This commit is contained in:
Omar Roth 2019-05-26 09:41:12 -05:00
parent 267bf289c4
commit 208bb2d72f
No known key found for this signature in database
GPG key ID: B8254FB7EC3D37F2
2 changed files with 25 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

21
src/invidious/helpers/helpers.cr
View file

@ -639,19 +639,16 @@ def proxy_file(response, env)
return
end
begin
if response.headers.includes_word?("Content-Encoding", "gzip")
Gzip::Writer.open(env.response) do |deflate|
copy_in_chunks(response.body_io, deflate)
end
elsif response.headers.includes_word?("Content-Encoding", "deflate")
Flate::Writer.open(env.response) do |deflate|
copy_in_chunks(response.body_io, deflate)
end
else
copy_in_chunks(response.body_io, env.response)
if response.headers.includes_word?("Content-Encoding", "gzip")
Gzip::Writer.open(env.response) do |deflate|
copy_in_chunks(response.body_io, deflate)
end
rescue ex
elsif response.headers.includes_word?("Content-Encoding", "deflate")
Flate::Writer.open(env.response) do |deflate|
copy_in_chunks(response.body_io, deflate)
end
else
copy_in_chunks(response.body_io, env.response)
end
end