diff --git a/Source/AaxDecrypter/NetworkFileStream.cs b/Source/AaxDecrypter/NetworkFileStream.cs
index 2bd4863b..bebde73d 100644
--- a/Source/AaxDecrypter/NetworkFileStream.cs
+++ b/Source/AaxDecrypter/NetworkFileStream.cs
@@ -39,13 +39,14 @@ namespace AaxDecrypter
public long ContentLength { get; private set; }
[JsonIgnore]
- public bool IsCancelled { get; private set; }
+ public bool IsCancelled => _cancellationSource.IsCancellationRequested;
#endregion
#region Private Properties
private FileStream _writeFile { get; }
private FileStream _readFile { get; }
+ private CancellationTokenSource _cancellationSource { get; } = new();
private EventWaitHandle _downloadedPiece { get; set; }
private Task _backgroundDownloadTask { get; set; }
@@ -126,7 +127,6 @@ namespace AaxDecrypter
RequestHeaders["Range"] = $"bytes={WritePosition}-";
}
-
/// Begins downloading to in a background thread.
/// The downloader
private Task BeginDownloading()
@@ -137,13 +137,12 @@ namespace AaxDecrypter
if (ContentLength != 0 && WritePosition > ContentLength)
throw new WebException($"Specified write position (0x{WritePosition:X10}) is larger than {nameof(ContentLength)} (0x{ContentLength:X10}).");
-
var request = new HttpRequestMessage(HttpMethod.Get, Uri);
foreach (var header in RequestHeaders)
request.Headers.Add(header.Key, header.Value);
- var response = new HttpClient().Send(request, HttpCompletionOption.ResponseHeadersRead);
+ var response = new HttpClient().Send(request, HttpCompletionOption.ResponseHeadersRead, _cancellationSource.Token);
if (response.StatusCode != HttpStatusCode.PartialContent)
throw new WebException($"Server at {Uri.Host} responded with unexpected status code: {response.StatusCode}.");
@@ -153,15 +152,15 @@ namespace AaxDecrypter
if (WritePosition == 0)
ContentLength = response.Content.Headers.ContentLength.GetValueOrDefault();
- var networkStream = response.Content.ReadAsStream();
+ var networkStream = response.Content.ReadAsStream(_cancellationSource.Token);
_downloadedPiece = new EventWaitHandle(false, EventResetMode.AutoReset);
//Download the file in the background.
- return Task.Run(() => DownloadFile(networkStream));
+ return DownloadFile(networkStream);
}
/// Download to .
- private void DownloadFile(Stream networkStream)
+ private async Task DownloadFile(Stream networkStream)
{
var downloadPosition = WritePosition;
var nextFlush = downloadPosition + DATA_FLUSH_SZ;
@@ -172,14 +171,14 @@ namespace AaxDecrypter
int bytesRead;
do
{
- bytesRead = networkStream.Read(buff, 0, DOWNLOAD_BUFF_SZ);
- _writeFile.Write(buff, 0, bytesRead);
+ bytesRead = await networkStream.ReadAsync(buff, 0, DOWNLOAD_BUFF_SZ, _cancellationSource.Token);
+ await _writeFile.WriteAsync(buff, 0, bytesRead, _cancellationSource.Token);
downloadPosition += bytesRead;
if (downloadPosition > nextFlush)
{
- _writeFile.Flush();
+ await _writeFile.FlushAsync(_cancellationSource.Token);
WritePosition = downloadPosition;
Update();
nextFlush = downloadPosition + DATA_FLUSH_SZ;
@@ -261,7 +260,7 @@ namespace AaxDecrypter
var toRead = Math.Min(count, Length - Position);
WaitToPosition(Position + toRead);
- return _readFile.Read(buffer, offset, count);
+ return IsCancelled ? 0: _readFile.Read(buffer, offset, count);
}
public override long Seek(long offset, SeekOrigin origin)
@@ -274,7 +273,7 @@ namespace AaxDecrypter
};
WaitToPosition(newPosition);
- return _readFile.Position = newPosition;
+ return IsCancelled ? 0 : (_readFile.Position = newPosition);
}
/// Blocks until the file has downloaded to at least , then returns.
@@ -285,13 +284,13 @@ namespace AaxDecrypter
&& _backgroundDownloadTask?.IsCompleted is false
&& !IsCancelled)
{
- _downloadedPiece.WaitOne(100);
+ _downloadedPiece.WaitOne(50);
}
}
public override void Close()
{
- IsCancelled = true;
+ _cancellationSource.Cancel();
_backgroundDownloadTask?.Wait();
_readFile.Close();