helpy4/Minecraft-Console-Client
1
0
Fork 0
mirror of https://github.com/MCCTeam/Minecraft-Console-Client synced 2025-10-14 21:22:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Use separate threads for decryption and decompression

Browse source
This commit is contained in:
BruceChen 2022-08-30 17:37:46 +08:00
parent c941d086d9
commit 1a90b6d942
8 changed files with 47 additions and 37 deletions
Download patch file Download diff file Expand all files Collapse all files

59
MinecraftClient/Protocol/Handlers/Protocol18.cs
View file

@ -20,6 +20,7 @@ using System.Threading.Tasks;
using MinecraftClient.Protocol.Keys;
using System.Text.RegularExpressions;
using MinecraftClient.Protocol.Session;
using System.Collections.Concurrent;
namespace MinecraftClient.Protocol.Handlers
{
@ -65,6 +66,7 @@ namespace MinecraftClient.Protocol.Handlers
private bool login_phase = true;
private int protocolversion;
private int currentDimension;
private readonly BlockingCollection<Tuple<int, Queue<byte>>> packetQueue = new();
Protocol18Forge pForge;
Protocol18Terrain pTerrain;
@ -211,30 +213,54 @@ namespace MinecraftClient.Protocol.Handlers
private bool Update()
{
handler.OnUpdate();
if (!socketWrapper.IsConnected())
if (packetQueue.IsAddingCompleted)
return false;
try
while (packetQueue.TryTake(out Tuple<int, Queue<byte>>? packetInfo))
{
while (socketWrapper.HasDataAvailable())
(int packetID, Queue<byte> packetData) = packetInfo;
HandlePacket(packetID, packetData);
if (handler.GetNetworkPacketCaptureEnabled())
{
int packetID = 0;
Queue<byte> packetData = ReadNextPacket(ref packetID);
HandlePacket(packetID, packetData);
List<byte> clone = packetData.ToList();
handler.OnNetworkPacket(packetID, clone, login_phase, true);
}
}
catch (System.IO.IOException) { return false; }
catch (SocketException) { return false; }
catch (NullReferenceException) { return false; }
catch (Ionic.Zlib.ZlibException) { return false; }
return true;
}
/// <summary>
/// Read and decompress packets.
/// </summary>
internal void PacketReader()
{
while (socketWrapper.IsConnected())
{
try
{
while (socketWrapper.HasDataAvailable())
packetQueue.Add(ReadNextPacket());
}
catch (System.IO.IOException) { break; }
catch (SocketException) { break; }
catch (NullReferenceException) { break; }
catch (Ionic.Zlib.ZlibException) { break; }
Thread.Sleep(10);
}
packetQueue.CompleteAdding();
}
/// <summary>
/// Read the next packet from the network
/// </summary>
/// <param name="packetID">will contain packet ID</param>
/// <param name="packetData">will contain raw packet Data</param>
internal Queue<byte> ReadNextPacket(ref int packetID)
internal Tuple<int, Queue<byte>> ReadNextPacket()
{
int size = dataTypes.ReadNextVarIntRAW(socketWrapper); //Packet size
Queue<byte> packetData = new(socketWrapper.ReadDataRAW(size)); //Packet contents
@ -252,7 +278,7 @@ namespace MinecraftClient.Protocol.Handlers
}
}
packetID = dataTypes.ReadNextVarInt(packetData); //Packet ID
int packetID = dataTypes.ReadNextVarInt(packetData); //Packet ID
if (handler.GetNetworkPacketCaptureEnabled())
{
@ -260,7 +286,7 @@ namespace MinecraftClient.Protocol.Handlers
handler.OnNetworkPacket(packetID, clone, login_phase, true);
}
return packetData;
return new(packetID, packetData);
}
/// <summary>
@ -1447,6 +1473,7 @@ namespace MinecraftClient.Protocol.Handlers
/// </summary>
private void StartUpdating()
{
new Thread(new ThreadStart(PacketReader)).Start();
netRead = new Tuple<Thread, CancellationTokenSource>(new Thread(new ParameterizedThreadStart(Updater)), new CancellationTokenSource());
netRead.Item1.Name = "ProtocolPacketHandler";
@ -1552,10 +1579,9 @@ namespace MinecraftClient.Protocol.Handlers
}
SendPacket(0x00, fullLoginPacket);
int packetID = -1;
while (true)
{
Queue<byte> packetData = ReadNextPacket(ref packetID);
(int packetID, Queue<byte> packetData) = ReadNextPacket();
if (packetID == 0x00) //Login rejected
{
handler.OnConnectionLost(ChatBot.DisconnectReason.LoginRejected, ChatParser.ParseText(dataTypes.ReadNextString(packetData)));
@ -1660,8 +1686,7 @@ namespace MinecraftClient.Protocol.Handlers
int loopPrevention = UInt16.MaxValue;
while (true)
{
int packetID = -1;
Queue<byte> packetData = ReadNextPacket(ref packetID);
(int packetID, Queue<byte> packetData) = ReadNextPacket();
if (packetID < 0 || loopPrevention-- < 0) // Failed to read packet or too many iterations (issue #1150)
{
handler.OnConnectionLost(ChatBot.DisconnectReason.ConnectionLost, Translations.Get("error.invalid_encrypt"));