xZero707/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2024-11-23 19:33:59 +01:00
Code Issues Packages Projects Releases Wiki Activity

[hls] Decrypt fragment when reading from disk

Browse Source 
Closes #373
This commit is contained in:
pukkandan 2021-06-05 18:09:11 +05:30
parent eb03899192
commit e06ca6ddac
No known key found for this signature in database
GPG Key ID: 0F00D95A001F4698
1 changed files with 16 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
yt_dlp/downloader/hls.py
View File

@ -272,12 +272,24 @@ def is_ad_fragment_end(s):
if not success: if not success:
return False return False
else: else:
def decrypt_fragment(fragment, frag_content):
decrypt_info = fragment['decrypt_info']
if decrypt_info['METHOD'] != 'AES-128':
return frag_content
iv = decrypt_info.get('IV') or compat_struct_pack('>8xq', fragment['media_sequence'])
decrypt_info['KEY'] = decrypt_info.get('KEY') or self.ydl.urlopen(
self._prepare_url(info_dict, info_dict.get('_decryption_key_url') or decrypt_info['URI'])).read()
# Don't decrypt the content in tests since the data is explicitly truncated and it's not to a valid block
# size (see https://github.com/ytdl-org/youtube-dl/pull/27660). Tests only care that the correct data downloaded,
# not what it decrypts to.
if test:
return frag_content
return AES.new(decrypt_info['KEY'], AES.MODE_CBC, iv).decrypt(frag_content)
def download_fragment(fragment): def download_fragment(fragment):
frag_index = fragment['frag_index'] frag_index = fragment['frag_index']
frag_url = fragment['url'] frag_url = fragment['url']
decrypt_info = fragment['decrypt_info']
byte_range = fragment['byte_range'] byte_range = fragment['byte_range']
media_sequence = fragment['media_sequence']
ctx['fragment_index'] = frag_index ctx['fragment_index'] = frag_index
@ -305,18 +317,7 @@ def download_fragment(fragment):
self.report_error('Giving up after %s fragment retries' % fragment_retries) self.report_error('Giving up after %s fragment retries' % fragment_retries)
return False, frag_index return False, frag_index
if decrypt_info['METHOD'] == 'AES-128': return decrypt_fragment(fragment, frag_content), frag_index
iv = decrypt_info.get('IV') or compat_struct_pack('>8xq', media_sequence)
decrypt_info['KEY'] = decrypt_info.get('KEY') or self.ydl.urlopen(
self._prepare_url(info_dict, info_dict.get('_decryption_key_url') or decrypt_info['URI'])).read()
# Don't decrypt the content in tests since the data is explicitly truncated and it's not to a valid block
# size (see https://github.com/ytdl-org/youtube-dl/pull/27660). Tests only care that the correct data downloaded,
# not what it decrypts to.
if not test:
frag_content = AES.new(
decrypt_info['KEY'], AES.MODE_CBC, iv).decrypt(frag_content)
return frag_content, frag_index
pack_fragment = lambda frag_content, _: frag_content pack_fragment = lambda frag_content, _: frag_content
@ -447,7 +448,7 @@ def append_fragment(frag_content, frag_index):
fragment['fragment_filename_sanitized'] = frag_sanitized fragment['fragment_filename_sanitized'] = frag_sanitized
frag_content = down.read() frag_content = down.read()
down.close() down.close()
result = append_fragment(frag_content, frag_index) result = append_fragment(decrypt_fragment(fragment, frag_content), frag_index)
if not result: if not result:
return False return False
else: else: