public override System.IO.Stream Receive(TimeSpan timeout, Information options)
{
if (_disposed) throw new ObjectDisposedException(this.GetType().FullName);
if (!_connect) throw new ConnectionException();
lock (_receiveLock)
{
Stream stream = null;
try
{
stream = _connection.Receive(timeout, options);
var version = (byte)stream.ReadByte();
Stream dataStream = null;
try
{
dataStream = new RangeStream(stream, stream.Position, stream.Length - stream.Position);
if (version == (byte)0)
{
return dataStream;
}
else if (version == (byte)1)
{
BufferStream deflateBufferStream = null;
try
{
deflateBufferStream = new BufferStream(_bufferManager);
using (DeflateStream deflateStream = new DeflateStream(dataStream, CompressionMode.Decompress, true))
using (var safeBuffer = _bufferManager.CreateSafeBuffer(1024 * 4))
{
int length;
while ((length = deflateStream.Read(safeBuffer.Value, 0, safeBuffer.Value.Length)) > 0)
{
deflateBufferStream.Write(safeBuffer.Value, 0, length);
if (deflateBufferStream.Length > _maxReceiveCount) throw new ConnectionException();
}
}
}
catch (Exception e)
{
if (deflateBufferStream != null)
{
deflateBufferStream.Dispose();
}
throw e;
}
#if DEBUG
Debug.WriteLine("Receive : {0}→{1} {2}",
NetworkConverter.ToSizeString(stream.Length),
NetworkConverter.ToSizeString(deflateBufferStream.Length),
NetworkConverter.ToSizeString(stream.Length - deflateBufferStream.Length));
#endif
deflateBufferStream.Seek(0, SeekOrigin.Begin);
dataStream.Dispose();
return deflateBufferStream;
}
else
{
throw new ArgumentException("ArgumentException");
}
}
catch (ConnectionException e)
{
if (dataStream != null) dataStream.Dispose();
throw e;
}
catch (Exception e)
{
if (dataStream != null) dataStream.Dispose();
throw new ConnectionException(e.Message, e);
}
}
catch (ConnectionException e)
{
if (stream != null) stream.Dispose();
throw e;
}
catch (Exception e)
{
if (stream != null) stream.Dispose();
throw new ConnectionException(e.Message, e);
}
}
}
public override void Send(System.IO.Stream stream, TimeSpan timeout, Information options)
{
if (_disposed) throw new ObjectDisposedException(this.GetType().FullName);
if (!_connect) throw new ConnectionException();
if (stream == null) throw new ArgumentNullException(nameof(stream));
if (stream.Length == 0) throw new ArgumentOutOfRangeException(nameof(stream));
bool isCompress = true;
if (options != null)
{
if (options.Contains("IsCompress")) isCompress = (bool)options["IsCompress"];
}
lock (_sendLock)
{
using (RangeStream targetStream = new RangeStream(stream, stream.Position, stream.Length - stream.Position, true))
{
try
{
var list = new List<KeyValuePair<byte, Stream>>();
if (isCompress)
{
if (_otherCompressAlgorithm.HasFlag(CompressAlgorithm.Deflate))
{
BufferStream deflateBufferStream = null;
try
{
deflateBufferStream = new BufferStream(_bufferManager);
using (DeflateStream deflateStream = new DeflateStream(deflateBufferStream, CompressionMode.Compress, true))
using (var safeBuffer = _bufferManager.CreateSafeBuffer(1024 * 4))
{
int length;
while ((length = targetStream.Read(safeBuffer.Value, 0, safeBuffer.Value.Length)) > 0)
{
deflateStream.Write(safeBuffer.Value, 0, length);
}
}
deflateBufferStream.Seek(0, SeekOrigin.Begin);
list.Add(new KeyValuePair<byte, Stream>((byte)1, deflateBufferStream));
}
catch (Exception e)
{
if (deflateBufferStream != null)
{
deflateBufferStream.Dispose();
}
throw e;
}
}
}
list.Add(new KeyValuePair<byte, Stream>((byte)0, new WrapperStream(targetStream, true)));
list.Sort((x, y) =>
{
int c = x.Value.Length.CompareTo(y.Value.Length);
if (c != 0) return c;
return x.Key.CompareTo(y.Key);
});
#if DEBUG
if (list[0].Value.Length != targetStream.Length)
{
Debug.WriteLine("Send : {0}→{1} {2}",
NetworkConverter.ToSizeString(targetStream.Length),
NetworkConverter.ToSizeString(list[0].Value.Length),
NetworkConverter.ToSizeString(list[0].Value.Length - targetStream.Length));
}
#endif
for (int i = 1; i < list.Count; i++)
{
list[i].Value.Dispose();
}
var headerStream = new BufferStream(_bufferManager);
headerStream.WriteByte((byte)list[0].Key);
using (var dataStream = new UniteStream(headerStream, list[0].Value))
{
_connection.Send(dataStream, timeout, options);
}
}
catch (ConnectionException e)
{
throw e;
}
catch (Exception e)
{
throw new ConnectionException(e.Message, e);
}
}
}
}
public override Stream Receive(TimeSpan timeout, Information options)
{
if (_disposed) throw new ObjectDisposedException(this.GetType().FullName);
if (!_connect) throw new ConnectionException();
lock (_receiveLock)
{
try
{
_receiveStopwatch.Restart();
Restart:;
int length = 0;
{
byte[] lengthBuffer = new byte[4];
var time = Connection.CheckTimeout(_receiveStopwatch.Elapsed, timeout);
time = (time < _receiveTimeSpan) ? time : _receiveTimeSpan;
_cap.Receive(lengthBuffer, time);
_receivedByteCount.Add(4);
length = NetworkConverter.ToInt32(lengthBuffer);
}
if (length == 0)
{
Thread.Sleep(300);
goto Restart;
}
else if (length > _maxReceiveCount)
{
throw new ConnectionException();
}
BufferStream bufferStream = null;
try
{
bufferStream = new BufferStream(_bufferManager);
using (var safeBuffer = _bufferManager.CreateSafeBuffer(1024 * 32))
{
do
{
int receiveLength = Math.Min(safeBuffer.Value.Length, length);
var time = Connection.CheckTimeout(_receiveStopwatch.Elapsed, timeout);
time = (time < _receiveTimeSpan) ? time : _receiveTimeSpan;
if (_bandwidthLimit != null && _bandwidthLimit.In != 0)
{
if (_cap.Available == 0)
{
Thread.Sleep(300);
continue;
}
receiveLength = _bandwidthLimit.GetInBandwidth(this, Math.Min(_cap.Available, receiveLength));
if (receiveLength < 0) throw new ConnectionException();
}
_cap.Receive(safeBuffer.Value, 0, receiveLength, time);
_receivedByteCount.Add(receiveLength);
bufferStream.Write(safeBuffer.Value, 0, receiveLength);
length -= receiveLength;
} while (length > 0);
}
}
catch (Exception e)
{
if (bufferStream != null)
{
bufferStream.Dispose();
}
throw e;
}
bufferStream.Seek(0, SeekOrigin.Begin);
return bufferStream;
}
catch (ConnectionException e)
{
throw e;
}
catch (Exception e)
{
throw new ConnectionException(e.Message, e);
}
}
}
private static Stream Encrypt(Stream stream, IExchangeEncrypt publicKey)
{
if (stream == null) throw new ArgumentNullException("stream");
if (publicKey == null) throw new ArgumentNullException("publicKey");
try
{
BufferStream outStream = null;
try
{
outStream = new BufferStream(_bufferManager);
outStream.WriteByte((byte)ConvertCryptoAlgorithm.Aes256);
byte[] cryptoKey = new byte[32];
_random.GetBytes(cryptoKey);
{
var encryptedBuffer = Exchange.Encrypt(publicKey, cryptoKey);
outStream.Write(NetworkConverter.GetBytes((int)encryptedBuffer.Length), 0, 4);
outStream.Write(encryptedBuffer, 0, encryptedBuffer.Length);
}
byte[] iv = new byte[32];
_random.GetBytes(iv);
outStream.Write(iv, 0, iv.Length);
using (Stream inStream = new WrapperStream(stream, true))
{
using (var rijndael = new RijndaelManaged() { KeySize = 256, BlockSize = 256, Mode = CipherMode.CBC, Padding = PaddingMode.PKCS7 })
using (CryptoStream cs = new CryptoStream(inStream, rijndael.CreateEncryptor(cryptoKey, iv), CryptoStreamMode.Read))
{
byte[] buffer = null;
try
{
buffer = _bufferManager.TakeBuffer(1024 * 4);
int i = -1;
while ((i = cs.Read(buffer, 0, buffer.Length)) > 0)
{
outStream.Write(buffer, 0, i);
}
}
finally
{
if (buffer != null)
{
_bufferManager.ReturnBuffer(buffer);
}
}
}
}
outStream.Seek(0, SeekOrigin.Begin);
}
catch (Exception)
{
if (outStream != null)
{
outStream.Dispose();
}
throw;
}
return outStream;
}
catch (Exception e)
{
throw new ArgumentException(e.Message, e);
}
}
private static Stream Decrypt(Stream stream, IExchangeDecrypt privateKey)
{
if (stream == null) throw new ArgumentNullException("stream");
if (privateKey == null) throw new ArgumentNullException("privateKey");
try
{
byte type = (byte)stream.ReadByte();
if (type == (byte)ConvertCryptoAlgorithm.Aes256)
{
byte[] cryptoKey;
{
byte[] lengthBuffer = new byte[4];
if (stream.Read(lengthBuffer, 0, lengthBuffer.Length) != lengthBuffer.Length) throw new ArgumentException();
int length = NetworkConverter.ToInt32(lengthBuffer);
byte[] encryptedBuffer = new byte[length];
if (stream.Read(encryptedBuffer, 0, encryptedBuffer.Length) != encryptedBuffer.Length) throw new ArgumentException();
cryptoKey = Exchange.Decrypt(privateKey, encryptedBuffer);
}
BufferStream outStream = null;
try
{
outStream = new BufferStream(_bufferManager);
using (Stream dataStream = new WrapperStream(stream, true))
{
var iv = new byte[32];
dataStream.Read(iv, 0, iv.Length);
using (var rijndael = new RijndaelManaged() { KeySize = 256, BlockSize = 256, Mode = CipherMode.CBC, Padding = PaddingMode.PKCS7 })
using (var inStream = new RangeStream(dataStream, dataStream.Position, dataStream.Length - dataStream.Position))
using (CryptoStream cs = new CryptoStream(inStream, rijndael.CreateDecryptor(cryptoKey, iv), CryptoStreamMode.Read))
{
byte[] buffer = null;
try
{
buffer = _bufferManager.TakeBuffer(1024 * 4);
int i = -1;
while ((i = cs.Read(buffer, 0, buffer.Length)) > 0)
{
outStream.Write(buffer, 0, i);
}
}
finally
{
_bufferManager.ReturnBuffer(buffer);
}
}
}
outStream.Seek(0, SeekOrigin.Begin);
}
catch (Exception)
{
if (outStream != null)
{
outStream.Dispose();
}
throw;
}
return outStream;
}
throw new NotSupportedException();
}
catch (Exception e)
{
throw new ArgumentException(e.Message, e);
}
}
private static Stream Decompress(Stream stream)
{
if (stream == null) throw new ArgumentNullException("stream");
try
{
var targetStream = new RangeStream(stream, true);
byte type = (byte)targetStream.ReadByte();
if (type == (byte)ConvertCompressionAlgorithm.None)
{
return new RangeStream(targetStream);
}
else if (type == (byte)ConvertCompressionAlgorithm.Deflate)
{
using (Stream dataStream = new WrapperStream(targetStream, true))
{
BufferStream deflateBufferStream = null;
try
{
deflateBufferStream = new BufferStream(_bufferManager);
using (DeflateStream deflateStream = new DeflateStream(dataStream, CompressionMode.Decompress, true))
{
byte[] decompressBuffer = null;
try
{
decompressBuffer = _bufferManager.TakeBuffer(1024 * 4);
int i = -1;
while ((i = deflateStream.Read(decompressBuffer, 0, decompressBuffer.Length)) > 0)
{
deflateBufferStream.Write(decompressBuffer, 0, i);
}
}
finally
{
if (decompressBuffer != null)
{
_bufferManager.ReturnBuffer(decompressBuffer);
}
}
}
deflateBufferStream.Seek(0, SeekOrigin.Begin);
#if DEBUG
Debug.WriteLine("ContentConverter Decompress {3} : {0}→{1} {2}",
NetworkConverter.ToSizeString(dataStream.Length),
NetworkConverter.ToSizeString(deflateBufferStream.Length),
NetworkConverter.ToSizeString(dataStream.Length - deflateBufferStream.Length),
ConvertCompressionAlgorithm.Deflate);
#endif
return deflateBufferStream;
}
catch (Exception)
{
if (deflateBufferStream != null)
{
deflateBufferStream.Dispose();
}
}
}
}
throw new ArgumentException("ArgumentException");
}
catch (Exception e)
{
throw new ArgumentException(e.Message, e);
}
}
private static Stream Compress(Stream stream)
{
if (stream == null) throw new ArgumentNullException("stream");
var targetStream = new RangeStream(stream, true);
List<KeyValuePair<byte, Stream>> list = new List<KeyValuePair<byte, Stream>>();
try
{
targetStream.Seek(0, SeekOrigin.Begin);
BufferStream deflateBufferStream = null;
try
{
deflateBufferStream = new BufferStream(_bufferManager);
using (DeflateStream deflateStream = new DeflateStream(deflateBufferStream, CompressionMode.Compress, true))
{
byte[] compressBuffer = null;
try
{
compressBuffer = _bufferManager.TakeBuffer(1024 * 4);
int i = -1;
while ((i = targetStream.Read(compressBuffer, 0, compressBuffer.Length)) > 0)
{
deflateStream.Write(compressBuffer, 0, i);
}
}
finally
{
if (compressBuffer != null)
{
_bufferManager.ReturnBuffer(compressBuffer);
}
}
}
deflateBufferStream.Seek(0, SeekOrigin.Begin);
list.Add(new KeyValuePair<byte, Stream>((byte)ConvertCompressionAlgorithm.Deflate, deflateBufferStream));
}
catch (Exception)
{
if (deflateBufferStream != null)
{
deflateBufferStream.Dispose();
}
}
}
catch (Exception)
{
}
list.Add(new KeyValuePair<byte, Stream>((byte)ConvertCompressionAlgorithm.None, targetStream));
list.Sort((x, y) =>
{
int c = x.Value.Length.CompareTo(y.Value.Length);
if (c != 0) return c;
return x.Key.CompareTo(y.Key);
});
#if DEBUG
if (list[0].Value.Length != targetStream.Length)
{
Debug.WriteLine("ContentConverter Compress {3} : {0}→{1} {2}",
NetworkConverter.ToSizeString(targetStream.Length),
NetworkConverter.ToSizeString(list[0].Value.Length),
NetworkConverter.ToSizeString(list[0].Value.Length - targetStream.Length),
(ConvertCompressionAlgorithm)list[0].Key);
}
#endif
for (int i = 1; i < list.Count; i++)
{
list[i].Value.Dispose();
}
BufferStream metadataStream = new BufferStream(_bufferManager);
metadataStream.WriteByte((byte)list[0].Key);
return new UniteStream(metadataStream, list[0].Value);
}
