Teknikode
/
Teknik
Watch 3
Star 16
Fork 4
Code Issues 38 Pull Requests 0 Releases 3 Wiki Activity
Browse Source

Removed range support as decrypting directly to the output stream didn't work well with ranges.

tags/3.0.0
Teknikode 2 years ago
parent
afe9a7727a
commit
446c74d45b
4 changed files with 49 additions and 99 deletions
Split View Show Diff Stats
  1. 1
    1
      Teknik/Areas/Podcast/Controllers/PodcastController.cs
  2. 7
    71
      Teknik/Areas/Upload/Controllers/UploadController.cs
  3. 7
    6
      Utilities/Utilities/Crypto.cs
  4. 34
    21
      Utilities/Utilities/ResponseHelper.cs

+ 1
- 1
Teknik/Areas/Podcast/Controllers/PodcastController.cs View File

@@ -100,7 +100,7 @@ namespace Teknik.Areas.Podcast.Controllers

Response.AppendHeader("Content-Disposition", cd.ToString());

return new FileGenerateResult(file.FileName, file.ContentType, (response) => ResponseHelper.StreamToOutput(response, fileStream, file.ContentLength, 4 * 1024), false);
return new FileGenerateResult(file.FileName, file.ContentType, (response) => ResponseHelper.StreamToOutput(response, true, fileStream, file.ContentLength, 4 * 1024), false);
//return File(data, file.ContentType);
}
}

+ 7
- 71
Teknik/Areas/Upload/Controllers/UploadController.cs View File

@@ -143,7 +143,7 @@ namespace Teknik.Areas.Upload.Controllers
else // We have the key, so that means server side decryption
{
// Are they downloading it by range?
bool byRange = !string.IsNullOrEmpty(Request.ServerVariables["HTTP_RANGE"]);
bool byRange = !string.IsNullOrEmpty(Request.ServerVariables["HTTP_RANGE"]); // We do not support ranges
// Check to see if they have a cache
bool isCached = !string.IsNullOrEmpty(Request.Headers["If-Modified-Since"]);

@@ -159,79 +159,15 @@ namespace Teknik.Areas.Upload.Controllers
{
string subDir = upload.FileName[0].ToString();
string filePath = Path.Combine(Config.UploadConfig.UploadDirectory, subDir, upload.FileName);
long startByte = 0;
long endByte = upload.ContentLength - 1;
long length = upload.ContentLength;
if (System.IO.File.Exists(filePath))
{
// Read in the file
FileStream fs = new FileStream(filePath, FileMode.Open, FileAccess.Read);
// We accept ranges
Response.AddHeader("Accept-Ranges", "0-" + length);

// check to see if we need to pass a specified range
if (byRange)
{
long anotherStart = startByte;
long anotherEnd = endByte;
string[] arr_split = Request.ServerVariables["HTTP_RANGE"].Split(new char[] { '=' });
string range = arr_split[1];

// Make sure the client hasn't sent us a multibyte range
if (range.IndexOf(",") > -1)
{
// (?) Shoud this be issued here, or should the first
// range be used? Or should the header be ignored and
// we output the whole content?
Response.AddHeader("Content-Range", "bytes " + startByte + "-" + endByte + "/" + upload.ContentLength);
throw new HttpException(416, "Requested Range Not Satisfiable");
}

// If the range starts with an '-' we start from the beginning
// If not, we forward the file pointer
// And make sure to get the end byte if spesified
if (range.StartsWith("-"))
{
// The n-number of the last bytes is requested
anotherStart = startByte - Convert.ToInt64(range.Substring(1));
}
else
{
arr_split = range.Split(new char[] { '-' });
anotherStart = Convert.ToInt64(arr_split[0]);
long temp = 0;
anotherEnd = (arr_split.Length > 1 && Int64.TryParse(arr_split[1].ToString(), out temp)) ? Convert.ToInt64(arr_split[1]) : upload.ContentLength;
}

/* Check the range and make sure it's treated according to the specs.
* http://www.w3.org/Protocols/rfc2616/rfc2616-sec14.html
*/
// End bytes can not be larger than $end.
anotherEnd = (anotherEnd > endByte) ? endByte : anotherEnd;
// Validate the requested range and return an error if it's not correct.
if (anotherStart > anotherEnd || anotherStart > upload.ContentLength - 1 || anotherEnd >= upload.ContentLength)
{

Response.AddHeader("Content-Range", "bytes " + startByte + "-" + endByte + "/" + upload.ContentLength);
throw new HttpException(416, "Requested Range Not Satisfiable");
}
startByte = anotherStart;
endByte = anotherEnd;

length = endByte - startByte + 1; // Calculate new content length

// Ranges are response of 206
Response.StatusCode = 206;
}

// Add cache parameters
Response.Cache.SetCacheability(HttpCacheability.Public);
Response.Cache.SetMaxAge(new TimeSpan(365, 0, 0, 0));
Response.Cache.SetLastModified(upload.DateUploaded);

// Notify the client the byte range we'll be outputting
Response.AddHeader("Content-Range", "bytes " + startByte + "-" + endByte + "/" + upload.ContentLength);
Response.AddHeader("Content-Length", length.ToString());
// Notify the client the content length we'll be outputting
Response.AddHeader("Content-Length", upload.ContentLength.ToString());

// Create content disposition
var cd = new System.Net.Mime.ContentDisposition
@@ -249,8 +185,8 @@ namespace Teknik.Areas.Upload.Controllers
contentType = "text/plain";
}

// Reset file stream to starting position (or start of range)
fs.Seek(startByte, SeekOrigin.Begin);
// Read in the file
FileStream fs = new FileStream(filePath, FileMode.Open, FileAccess.Read);

// If the IV is set, and Key is set, then decrypt it while sending
if (!string.IsNullOrEmpty(upload.Key) && !string.IsNullOrEmpty(upload.IV))
@@ -260,7 +196,7 @@ namespace Teknik.Areas.Upload.Controllers

return new FileGenerateResult(upload.Url,
contentType,
(response) => ResponseHelper.DecryptStreamToOutput(response, fs, (int)length, keyBytes, ivBytes, "CTR", "NoPadding", Config.UploadConfig.ChunkSize),
(response) => ResponseHelper.DecryptStreamToOutput(response, true, fs, (int)upload.ContentLength, keyBytes, ivBytes, "CTR", "NoPadding", Config.UploadConfig.ChunkSize),
false);
}
else // Otherwise just send it
@@ -270,7 +206,7 @@ namespace Teknik.Areas.Upload.Controllers
// Send the file
return new FileGenerateResult(upload.Url,
contentType,
(response) => ResponseHelper.StreamToOutput(response, fs, (int)length, Config.UploadConfig.ChunkSize),
(response) => ResponseHelper.StreamToOutput(response, true, fs, (int)upload.ContentLength, Config.UploadConfig.ChunkSize),
false);
}
}

+ 7
- 6
Utilities/Utilities/Crypto.cs View File

@@ -204,15 +204,15 @@ namespace Teknik.Utilities
// Initialize variables
byte[] output = new byte[input.Length];
int cipherOffset = 0;
int processedBytes = 0;
int bytesRead = 0;

// Process the stream and save the bytes to the output
do
{
processedBytes = ProcessCipherBlock(cipher, input, chunkSize, output, cipherOffset);
int processedBytes = ProcessCipherBlock(cipher, input, chunkSize, output, cipherOffset, out bytesRead);
cipherOffset += processedBytes;
}
while (processedBytes > 0);
while (bytesRead > 0);

// Finalize processing of the cipher
FinalizeCipherBlock(cipher, output, cipherOffset);
@@ -231,9 +231,10 @@ namespace Teknik.Utilities
{
int processedBytes = 0;
byte[] buffer = new byte[chunkSize];
int bytesRead = 0;
do
{
processedBytes = ProcessCipherBlock(cipher, input, chunkSize, buffer, 0);
processedBytes = ProcessCipherBlock(cipher, input, chunkSize, buffer, 0, out bytesRead);
if (processedBytes > 0)
{
// We have bytes, lets write them to the file
@@ -267,13 +268,13 @@ namespace Teknik.Utilities
return cipher;
}

public static int ProcessCipherBlock(IBufferedCipher cipher, Stream input, int chunkSize, byte[] output, int outputOffset)
public static int ProcessCipherBlock(IBufferedCipher cipher, Stream input, int chunkSize, byte[] output, int outputOffset, out int bytesRead)
{
// Initialize buffer
byte[] buffer = new byte[chunkSize];

// Read the next block of data
int bytesRead = input.Read(buffer, 0, chunkSize);
bytesRead = input.Read(buffer, 0, chunkSize);
if (bytesRead > 0)
{
// process the cipher for the read block and add it to the output

+ 34
- 21
Utilities/Utilities/ResponseHelper.cs View File

@@ -11,11 +11,14 @@ namespace Teknik.Utilities
{
public static class ResponseHelper
{
public static void StreamToOutput(HttpResponseBase response, Stream stream, int length, int chunkSize)
public static void StreamToOutput(HttpResponseBase response, bool flush, Stream stream, int length, int chunkSize)
{
try
{
response.Flush();
if (flush)
{
response.Flush();
}

int curByte = 0;
int processedBytes = 0;
@@ -33,7 +36,10 @@ namespace Teknik.Utilities
if (processedBytes > 0)
{
response.OutputStream.Write(buffer, 0, processedBytes);
response.Flush();
if (flush)
{
response.Flush();
}

// Clear the buffer
Array.Clear(buffer, 0, chunkSize);
@@ -52,7 +58,7 @@ namespace Teknik.Utilities
}
else
{
throw httpEx;
//throw httpEx;
}
}
catch (Exception ex)
@@ -66,11 +72,14 @@ namespace Teknik.Utilities
}
}

public static void DecryptStreamToOutput(HttpResponseBase response, Stream stream, int length, byte[] key, byte[] iv, string mode, string padding, int chunkSize)
public static void DecryptStreamToOutput(HttpResponseBase response, bool flush, Stream stream, int length, byte[] key, byte[] iv, string mode, string padding, int chunkSize)
{
try
{
response.Flush();
if (flush)
{
response.Flush();
}
IBufferedCipher cipher = AES.CreateCipher(false, key, iv, mode, padding);

int curByte = 0;
@@ -78,37 +87,41 @@ namespace Teknik.Utilities
byte[] buffer = new byte[chunkSize];
int bytesRemaining = length;
int bytesToRead = chunkSize;
int bytesRead = 0;
do
{
if (chunkSize > bytesRemaining)
{
bytesToRead = bytesRemaining;
}
processedBytes = AES.ProcessCipherBlock(cipher, stream, bytesToRead, buffer, 0);
processedBytes = AES.ProcessCipherBlock(cipher, stream, bytesToRead, buffer, 0, out bytesRead);
if (processedBytes > 0)
{
response.OutputStream.Write(buffer, 0, processedBytes);
response.Flush();
if (flush)
{
response.Flush();
}

// Clear the buffer
Array.Clear(buffer, 0, chunkSize);
}
curByte += processedBytes;
bytesRemaining -= processedBytes;
curByte += bytesRead;
bytesRemaining -= bytesRead;
}
while (processedBytes > 0 && bytesRemaining > 0);
while (bytesRemaining > 0);

if (bytesRemaining > 0)
{
// Clear the buffer
Array.Clear(buffer, 0, chunkSize);
// Clear the buffer
Array.Clear(buffer, 0, chunkSize);

// Finalize processing of the cipher
processedBytes = AES.FinalizeCipherBlock(cipher, buffer, 0);
if (processedBytes > 0)
// Finalize processing of the cipher
processedBytes = AES.FinalizeCipherBlock(cipher, buffer, 0);
if (processedBytes > 0)
{
// We have bytes, lets write them to the output
response.OutputStream.Write(buffer, 0, processedBytes);
if (flush)
{
// We have bytes, lets write them to the output
response.OutputStream.Write(buffer, 0, processedBytes);
response.Flush();
}
}
@@ -122,7 +135,7 @@ namespace Teknik.Utilities
}
else
{
throw httpEx;
//throw httpEx;
}
}
catch (Exception ex)

Write Preview
Loading…
Cancel
Save