Mirrors/Teknik
1
0
Fork 0
mirror of https://git.teknik.io/Teknikode/Teknik.git synced 2023-08-02 14:16:22 +02:00
Code Releases Activity

Fixed length for limiting amount to write to response

Browse Source
This commit is contained in:
Uncled1023 2017-02-09 01:05:34 -08:00
parent b7a9f2a740
commit 72d9c7a115
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
Teknik/Areas/Upload/Controllers/UploadController.cs
View File

@ -178,7 +178,7 @@ namespace Teknik.Areas.Upload.Controllers
// Read in the file
FileStream fs = new FileStream(filePath, FileMode.Open, FileAccess.Read);
// We accept ranges
Response.AddHeader("Accept-Ranges", "0-" + upload.ContentLength);
Response.AddHeader("Accept-Ranges", "0-" + length);
// check to see if we need to pass a specified range
if (byRange)
@ -296,13 +296,14 @@ namespace Teknik.Areas.Upload.Controllers
int curByte = 0;
int processedBytes = 0;
byte[] buffer = new byte[chunkSize];
int bytesToRead = chunkSize;
do
{
if (curByte + chunkSize > length)
{
chunkSize = length - curByte;
bytesToRead = chunkSize - ((curByte + bytesToRead) - length);
}
processedBytes = AES.ProcessCipherBlock(cipher, fileStream, chunkSize, buffer, 0);
processedBytes = AES.ProcessCipherBlock(cipher, fileStream, bytesToRead, buffer, 0);
if (processedBytes > 0)
{
response.OutputStream.Write(buffer, 0, processedBytes);
@ -313,7 +314,7 @@ namespace Teknik.Areas.Upload.Controllers
}
curByte += processedBytes;
}
while (processedBytes > 0 && curByte < length);
while (processedBytes > 0 && curByte <= length);
// Clear the buffer
Array.Clear(buffer, 0, chunkSize);