public static Buffer?Compress(Buffer buf, int headerSize, int compressionLevel)
{
Debug.Assert(Supported());
//
// Compress the message body, but not the header.
//
int uncompressedLen = buf.Size() - headerSize;
byte[] data = buf.B.RawBytes(headerSize, uncompressedLen);
int compressedLen = (int)((uncompressedLen * 1.01) + 600);
byte[] compressed = new byte[compressedLen];
int rc = _compressBuffer(compressed, ref compressedLen, data, uncompressedLen, compressionLevel, 0, 0);
if (rc == BZ_OUTBUFF_FULL)
{
return(null);
}
else if (rc < 0)
{
var ex = new Ice.CompressionException("BZ2_bzBuffToBuffCompress failed");
ex.Reason = GetBZ2Error(rc);
throw ex;
}
//
// Don't bother if the compressed data is larger than the
// uncompressed data.
//
if (compressedLen >= uncompressedLen)
{
return(null);
}
var r = new Buffer();
r.Resize(headerSize + 4 + compressedLen, false);
r.B.Position(0);
//
// Copy the header from the uncompressed stream to the
// compressed one.
//
r.B.Put(buf.B.RawBytes(0, headerSize));
//
// Add the size of the uncompressed stream before the
// message body.
//
r.B.PutInt(buf.Size());
//
// Add the compressed message body.
//
r.B.Put(compressed, 0, compressedLen);
return(r);
}
//
// Only for use by Connector, Acceptor
//
internal Transceiver(IceInternal.ITransceiver transceiver)
{
_transceiver = transceiver;
_configuration = Configuration.getInstance();
_initialized = false;
_readBuffer = new IceInternal.Buffer();
_readBuffer.Resize(1024 * 8, true); // 8KB buffer
_readBuffer.B.Position(0);
_readBufferPos = 0;
_buffered = _configuration.buffered();
}
public bool StartRead(Buffer buf, AsyncCallback callback, object state)
{
Debug.Assert(_fd != null);
Debug.Assert(buf.B.Position() == 0);
int packetSize = Math.Min(MaxPacketSize, _rcvSize - UdpOverhead);
buf.Resize(packetSize, true);
buf.B.Position(0);
try
{
// TODO: Workaround for https://github.com/dotnet/corefx/issues/31182
if (_state == StateConnected ||
(AssemblyUtil.IsMacOS && _fd.AddressFamily == AddressFamily.InterNetworkV6 && _fd.DualMode))
{
_readCallback = callback;
_readEventArgs.UserToken = state;
_readEventArgs.SetBuffer(buf.B.RawBytes(), buf.B.Position(), packetSize);
return(!_fd.ReceiveAsync(_readEventArgs));
}
else
{
Debug.Assert(_incoming);
_readCallback = callback;
_readEventArgs.UserToken = state;
_readEventArgs.SetBuffer(buf.B.RawBytes(), 0, buf.B.Limit());
return(!_fd.ReceiveFromAsync(_readEventArgs));
}
}
catch (SocketException ex)
{
if (Network.RecvTruncated(ex))
{
// Nothing todo
return(true);
}
else
{
if (Network.ConnectionLost(ex))
{
throw new Ice.ConnectionLostException(ex);
}
else
{
throw new Ice.SocketException(ex);
}
}
}
}
public static Buffer Uncompress(Buffer buf, int headerSize, int messageSizeMax)
{
Debug.Assert(Supported());
buf.B.Position(headerSize);
int uncompressedSize = buf.B.GetInt();
if (uncompressedSize <= headerSize)
{
throw new Ice.IllegalMessageSizeException("compressed size <= header size");
}
if (uncompressedSize > messageSizeMax)
{
Ex.ThrowMemoryLimitException(uncompressedSize, messageSizeMax);
}
int compressedLen = buf.Size() - headerSize - 4;
byte[] compressed = buf.B.RawBytes(headerSize + 4, compressedLen);
int uncompressedLen = uncompressedSize - headerSize;
byte[] uncompressed = new byte[uncompressedLen];
int rc = _decompressBuffer(uncompressed, ref uncompressedLen, compressed, compressedLen, 0, 0);
if (rc < 0)
{
throw new Ice.CompressionException($"BZ2_bzBuffToBuffDecompress failed\n{GetBZ2Error(rc)}");
}
var r = new Buffer();
r.Resize(uncompressedSize, false);
//
// Copy the header from the compressed buffer to the uncompressed one.
//
r.B.Position(0);
r.B.Put(buf.B.RawBytes(), 0, headerSize);
r.B.Put(uncompressed);
return(r);
}
public void FinishRead(Buffer buf)
{
if (_fd == null)
{
return;
}
int ret;
try
{
if (_readEventArgs.SocketError != SocketError.Success)
{
throw new SocketException((int)_readEventArgs.SocketError);
}
ret = _readEventArgs.BytesTransferred;
// TODO: Workaround for https://github.com/dotnet/corefx/issues/31182
if (_state != StateConnected &&
!(AssemblyUtil.IsMacOS && _fd.AddressFamily == AddressFamily.InterNetworkV6 && _fd.DualMode))
{
_peerAddr = _readEventArgs.RemoteEndPoint;
}
}
catch (SocketException ex)
{
if (Network.RecvTruncated(ex))
{
// The message was truncated and the whole buffer is filled. We ignore
// this error here, it will be detected at the connection level when
// the Ice message size is checked against the buffer size.
ret = buf.Size();
}
else
{
if (Network.ConnectionLost(ex))
{
throw new Ice.ConnectionLostException(ex);
}
if (Network.ConnectionRefused(ex))
{
throw new Ice.ConnectionRefusedException(ex);
}
else
{
throw new Ice.SocketException(ex);
}
}
}
if (ret == 0)
{
throw new Ice.ConnectionLostException();
}
Debug.Assert(ret > 0);
if (_state == StateNeedConnect)
{
Debug.Assert(_incoming);
//
// If we must connect, then we connect to the first peer that
// sends us a packet.
//
bool connected = !_fd.ConnectAsync(_readEventArgs);
Debug.Assert(connected);
_state = StateConnected; // We're connected now
if (_instance.TraceLevel >= 1)
{
_instance.Logger.Trace(_instance.TraceCategory, $"connected {Transport()} socket\n{this}");
}
}
buf.Resize(ret, true);
buf.B.Position(ret);
}
public int Read(Buffer buf, ref bool hasMoreData)
{
if (!buf.B.HasRemaining())
{
return(SocketOperation.None);
}
Debug.Assert(buf.B.Position() == 0);
Debug.Assert(_fd != null);
int packetSize = Math.Min(MaxPacketSize, _rcvSize - UdpOverhead);
buf.Resize(packetSize, true);
buf.B.Position(0);
int ret;
while (true)
{
try
{
EndPoint?peerAddr = _peerAddr;
if (peerAddr == null)
{
if (_addr.AddressFamily == AddressFamily.InterNetwork)
{
peerAddr = new IPEndPoint(IPAddress.Any, 0);
}
else
{
Debug.Assert(_addr.AddressFamily == AddressFamily.InterNetworkV6);
peerAddr = new IPEndPoint(IPAddress.IPv6Any, 0);
}
}
// TODO: Workaround for https://github.com/dotnet/corefx/issues/31182
if (_state == StateConnected ||
(AssemblyUtil.IsMacOS && _fd.AddressFamily == AddressFamily.InterNetworkV6 && _fd.DualMode))
{
ret = _fd.Receive(buf.B.RawBytes(), 0, buf.B.Limit(), SocketFlags.None);
}
else
{
ret = _fd.ReceiveFrom(buf.B.RawBytes(), 0, buf.B.Limit(), SocketFlags.None, ref peerAddr);
_peerAddr = (IPEndPoint)peerAddr;
}
break;
}
catch (SocketException e)
{
if (Network.RecvTruncated(e))
{
// The message was truncated and the whole buffer is filled. We ignore
// this error here, it will be detected at the connection level when
// the Ice message size is checked against the buffer size.
ret = buf.Size();
break;
}
if (Network.Interrupted(e))
{
continue;
}
if (Network.WouldBlock(e))
{
return(SocketOperation.Read);
}
if (Network.ConnectionLost(e))
{
throw new Ice.ConnectionLostException();
}
else
{
throw new Ice.SocketException(e);
}
}
catch (Exception e)
{
throw new Ice.SyscallException(e);
}
}
if (ret == 0)
{
throw new Ice.ConnectionLostException();
}
if (_state == StateNeedConnect)
{
Debug.Assert(_incoming);
//
// If we must connect, then we connect to the first peer that sends us a packet.
//
Debug.Assert(_peerAddr != null);
bool connected = Network.DoConnect(_fd, _peerAddr, null);
Debug.Assert(connected);
_state = StateConnected; // We're connected now
if (_instance.TraceLevel >= 1)
{
_instance.Logger.Trace(_instance.TraceCategory, $"connected {Transport()} socket\n{this}");
}
}
buf.Resize(ret, true);
buf.B.Position(ret);
return(SocketOperation.None);
}