DRMTalks
/
devine
mirror of https://github.com/devine-dl/devine.git
1
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

Fix infinite loop in Track.get_init_segment 

If the Server returns a Content-Length Header with a value of 0, then the code near-after it would end up looping response streamed chunks of 0-length size, which would go on forever.
This commit is contained in:
rlaphoenix 2024-01-09 02:45:10 +00:00
parent 58cb00b18b
commit a3ab971132
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
devine/core/tracks/track.py
View File

@ -186,9 +186,10 @@ class Track:
else: else:
size_test = session.head(url) size_test = session.head(url)
if "Content-Length" in size_test.headers: if "Content-Length" in size_test.headers:
content_length = int(size_test.headers["Content-Length"]) content_length_header = int(size_test.headers["Content-Length"])
# use whichever is smaller in case this is a full file if content_length_header > 0:
content_length = min(content_length, maximum_size) # use whichever is smaller in case this is a large file
content_length = min(content_length_header, maximum_size)
range_test = session.head(url, headers={"Range": "bytes=0-1"}) range_test = session.head(url, headers={"Range": "bytes=0-1"})
if range_test.status_code == 206: if range_test.status_code == 206:
byte_range = f"0-{content_length-1}" byte_range = f"0-{content_length-1}"