Mirror-Tools
/
N_m3u8DL-RE
mirror of https://github.com/nilaoda/N_m3u8DL-RE.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[bug fix] 本地文件被多次解密导致报错 The input data is not a complete block

This commit is contained in:
BlueHtml 2024-01-09 22:05:57 +08:00 committed by GitHub
parent 8fdb6bc90b
commit 21f2e59653
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 48 additions and 47 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

93
src/N_m3u8DL-RE/Downloader/SimpleDownloader.cs
View File

@ -1,4 +1,4 @@
using N_m3u8DL_RE.Common.Entity; using N_m3u8DL_RE.Common.Entity;
using N_m3u8DL_RE.Common.Enum; using N_m3u8DL_RE.Common.Enum;
using N_m3u8DL_RE.Common.Log; using N_m3u8DL_RE.Common.Log;
using N_m3u8DL_RE.Config; using N_m3u8DL_RE.Config;
@ -33,50 +33,57 @@ namespace N_m3u8DL_RE.Downloader
public async Task<DownloadResult?> DownloadSegmentAsync(MediaSegment segment, string savePath, SpeedContainer speedContainer, Dictionary<string, string>? headers = null) public async Task<DownloadResult?> DownloadSegmentAsync(MediaSegment segment, string savePath, SpeedContainer speedContainer, Dictionary<string, string>? headers = null)
{ {
var url = segment.Url; var url = segment.Url;
var dResult = await DownClipAsync(url, savePath, speedContainer, segment.StartRange, segment.StopRange, headers, DownloaderConfig.MyOptions.DownloadRetryCount); var (des, dResult) = await DownClipAsync(url, savePath, speedContainer, segment.StartRange, segment.StopRange, headers, DownloaderConfig.MyOptions.DownloadRetryCount);
if (dResult != null && dResult.Success && segment.EncryptInfo != null) if (dResult != null && dResult.Success && dResult.ActualFilePath != des)
{ {
if (segment.EncryptInfo.Method == EncryptMethod.AES_128) if (segment.EncryptInfo != null)
{ {
var key = segment.EncryptInfo.Key; if (segment.EncryptInfo.Method == EncryptMethod.AES_128)
var iv = segment.EncryptInfo.IV; {
AESUtil.AES128Decrypt(dResult.ActualFilePath, key!, iv!); var key = segment.EncryptInfo.Key;
} var iv = segment.EncryptInfo.IV;
else if (segment.EncryptInfo.Method == EncryptMethod.AES_128_ECB) AESUtil.AES128Decrypt(dResult.ActualFilePath, key!, iv!);
{ }
var key = segment.EncryptInfo.Key; else if (segment.EncryptInfo.Method == EncryptMethod.AES_128_ECB)
var iv = segment.EncryptInfo.IV; {
AESUtil.AES128Decrypt(dResult.ActualFilePath, key!, iv!, System.Security.Cryptography.CipherMode.ECB); var key = segment.EncryptInfo.Key;
} var iv = segment.EncryptInfo.IV;
else if (segment.EncryptInfo.Method == EncryptMethod.CHACHA20) AESUtil.AES128Decrypt(dResult.ActualFilePath, key!, iv!, System.Security.Cryptography.CipherMode.ECB);
{ }
var key = segment.EncryptInfo.Key; else if (segment.EncryptInfo.Method == EncryptMethod.CHACHA20)
var nonce = segment.EncryptInfo.IV; {
var key = segment.EncryptInfo.Key;
var nonce = segment.EncryptInfo.IV;
var fileBytes = File.ReadAllBytes(dResult.ActualFilePath); var fileBytes = File.ReadAllBytes(dResult.ActualFilePath);
var decrypted = ChaCha20Util.DecryptPer1024Bytes(fileBytes, key!, nonce!); var decrypted = ChaCha20Util.DecryptPer1024Bytes(fileBytes, key!, nonce!);
await File.WriteAllBytesAsync(dResult.ActualFilePath, decrypted); await File.WriteAllBytesAsync(dResult.ActualFilePath, decrypted);
} }
else if (segment.EncryptInfo.Method == EncryptMethod.SAMPLE_AES_CTR) else if (segment.EncryptInfo.Method == EncryptMethod.SAMPLE_AES_CTR)
{ {
//throw new NotSupportedException("SAMPLE-AES-CTR"); //throw new NotSupportedException("SAMPLE-AES-CTR");
}
//Image头处理
if (dResult.ImageHeader)
{
await ImageHeaderUtil.ProcessAsync(dResult.ActualFilePath);
}
//Gzip解压
if (dResult.GzipHeader)
{
await OtherUtil.DeGzipFileAsync(dResult.ActualFilePath);
}
} }
//Image头处理 //处理完成后改名
if (dResult.ImageHeader) File.Move(dResult.ActualFilePath, des);
{ dResult.ActualFilePath = des;
await ImageHeaderUtil.ProcessAsync(dResult.ActualFilePath);
}
//Gzip解压
if (dResult.GzipHeader)
{
await OtherUtil.DeGzipFileAsync(dResult.ActualFilePath);
}
} }
return dResult; return dResult;
} }
private async Task<DownloadResult?> DownClipAsync(string url, string path, SpeedContainer speedContainer, long? fromPosition, long? toPosition, Dictionary<string, string>? headers = null, int retryCount = 3) private async Task<(string des, DownloadResult? dResult)> DownClipAsync(string url, string path, SpeedContainer speedContainer, long? fromPosition, long? toPosition, Dictionary<string, string>? headers = null, int retryCount = 3)
{ {
CancellationTokenSource? cancellationTokenSource = null; CancellationTokenSource? cancellationTokenSource = null;
retry: retry:
@ -89,7 +96,7 @@ namespace N_m3u8DL_RE.Downloader
if (File.Exists(des)) if (File.Exists(des))
{ {
speedContainer.Add(new FileInfo(des).Length); speedContainer.Add(new FileInfo(des).Length);
return new DownloadResult() { ActualContentLength = 0, ActualFilePath = des }; return (des, new DownloadResult() { ActualContentLength = 0, ActualFilePath = des });
} }
//已解密跳过 //已解密跳过
@ -97,7 +104,7 @@ namespace N_m3u8DL_RE.Downloader
if (File.Exists(dec)) if (File.Exists(dec))
{ {
speedContainer.Add(new FileInfo(dec).Length); speedContainer.Add(new FileInfo(dec).Length);
return new DownloadResult() { ActualContentLength = 0, ActualFilePath = dec }; return (dec, new DownloadResult() { ActualContentLength = 0, ActualFilePath = dec });
} }
//另起线程进行监控 //另起线程进行监控
@ -118,14 +125,8 @@ namespace N_m3u8DL_RE.Downloader
//调用下载 //调用下载
var result = await DownloadUtil.DownloadToFileAsync(url, path, speedContainer, cancellationTokenSource, headers, fromPosition, toPosition); var result = await DownloadUtil.DownloadToFileAsync(url, path, speedContainer, cancellationTokenSource, headers, fromPosition, toPosition);
return (des, result);
//下载完成后改名
if (result.Success || !DownloaderConfig.CheckContentLength)
{
File.Move(path, des);
result.ActualFilePath = des;
return result;
}
throw new Exception("please retry"); throw new Exception("please retry");
} }
catch (Exception ex) catch (Exception ex)
@ -144,7 +145,7 @@ namespace N_m3u8DL_RE.Downloader
Logger.WarnMarkUp($"[grey]{ex.Message.EscapeMarkup()}[/]"); Logger.WarnMarkUp($"[grey]{ex.Message.EscapeMarkup()}[/]");
} }
//throw new Exception("download failed", ex); //throw new Exception("download failed", ex);
return null; return default;
} }
finally finally
{ {