sochix/TLSharp
1
0
Fork 0
mirror of https://github.com/sochix/TLSharp.git synced 2025-12-06 08:02:00 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Free TCP network stream

Browse source 
- Use single TCP network stream fore read and write

- Cleanup TCP stream on disposing as microsoft note : https://docs.microsoft.com/en-us/dotnet/api/system.net.sockets.tcpclient.getstream?view=netframework-4.8
This commit is contained in:
Felony 2019-10-02 12:06:18 +03:30 committed by Andres G. Aragoneses
parent 0dd3992997
commit 460ec6fb32
1 changed files with 15 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
TLSharp.Core/Network/TcpTransport.cs
View file

@ -10,6 +10,7 @@ namespace TLSharp.Core.Network
public class TcpTransport : IDisposable public class TcpTransport : IDisposable
{ {
private readonly TcpClient _tcpClient; private readonly TcpClient _tcpClient;
private readonly NetworkStream _stream;
private int sendCounter = 0; private int sendCounter = 0;
public TcpTransport(string address, int port, TcpClientConnectionHandler handler = null) public TcpTransport(string address, int port, TcpClientConnectionHandler handler = null)
@ -24,6 +25,11 @@ namespace TLSharp.Core.Network
} }
else else
_tcpClient = handler(address, port); _tcpClient = handler(address, port);
if (_tcpClient.Connected)
{
_stream = _tcpClient.GetStream();
}
} }
public async Task Send(byte[] packet) public async Task Send(byte[] packet)
@ -33,21 +39,19 @@ namespace TLSharp.Core.Network
var tcpMessage = new TcpMessage(sendCounter, packet); var tcpMessage = new TcpMessage(sendCounter, packet);
await _tcpClient.GetStream().WriteAsync(tcpMessage.Encode(), 0, tcpMessage.Encode().Length); await _stream.WriteAsync(tcpMessage.Encode(), 0, tcpMessage.Encode().Length);
sendCounter++; sendCounter++;
} }
public async Task<TcpMessage> Receieve() public async Task<TcpMessage> Receieve()
{ {
var stream = _tcpClient.GetStream();
var packetLengthBytes = new byte[4]; var packetLengthBytes = new byte[4];
if (await stream.ReadAsync(packetLengthBytes, 0, 4) != 4) if (await _stream.ReadAsync(packetLengthBytes, 0, 4) != 4)
throw new InvalidOperationException("Couldn't read the packet length"); throw new InvalidOperationException("Couldn't read the packet length");
int packetLength = BitConverter.ToInt32(packetLengthBytes, 0); int packetLength = BitConverter.ToInt32(packetLengthBytes, 0);
var seqBytes = new byte[4]; var seqBytes = new byte[4];
if (await stream.ReadAsync(seqBytes, 0, 4) != 4) if (await _stream.ReadAsync(seqBytes, 0, 4) != 4)
throw new InvalidOperationException("Couldn't read the sequence"); throw new InvalidOperationException("Couldn't read the sequence");
int seq = BitConverter.ToInt32(seqBytes, 0); int seq = BitConverter.ToInt32(seqBytes, 0);
@ -58,7 +62,7 @@ namespace TLSharp.Core.Network
do do
{ {
var bodyByte = new byte[packetLength - 12]; var bodyByte = new byte[packetLength - 12];
var availableBytes = await stream.ReadAsync(bodyByte, 0, neededToRead); var availableBytes = await _stream.ReadAsync(bodyByte, 0, neededToRead);
neededToRead -= availableBytes; neededToRead -= availableBytes;
Buffer.BlockCopy(bodyByte, 0, body, readBytes, availableBytes); Buffer.BlockCopy(bodyByte, 0, body, readBytes, availableBytes);
readBytes += availableBytes; readBytes += availableBytes;
@ -66,7 +70,7 @@ namespace TLSharp.Core.Network
while (readBytes != packetLength - 12); while (readBytes != packetLength - 12);
var crcBytes = new byte[4]; var crcBytes = new byte[4];
if (await stream.ReadAsync(crcBytes, 0, 4) != 4) if (await _stream.ReadAsync(crcBytes, 0, 4) != 4)
throw new InvalidOperationException("Couldn't read the crc"); throw new InvalidOperationException("Couldn't read the crc");
int checksum = BitConverter.ToInt32(crcBytes, 0); int checksum = BitConverter.ToInt32(crcBytes, 0);
@ -99,7 +103,10 @@ namespace TLSharp.Core.Network
public void Dispose() public void Dispose()
{ {
if (_tcpClient.Connected) if (_tcpClient.Connected)
{
_stream.Close();
_tcpClient.Close(); _tcpClient.Close();
}
} }
} }
} }