First pass ext-encoding support (high-level network I/O and read-only data) (#6799)

This implements ext-encoding only in high-level network I/O (only BedrockProtocol and stuff implemented in PM) and read-only data.
This should net a significant performance advantage while being low-risk in the case of critical issues with the extension. Any problems affecting protocol won't do permanent damage while being fairly easy to debug.

Next passes will integrate ext-encoding versions of RakLib, RakLibIpc and NBT, as well as generally using ext-encoding for writeable data.
This commit is contained in:
Dylan T.
2025-09-25 17:32:38 +01:00
committed by GitHub
parent 6fede8a67f
commit 3336cda34a
18 changed files with 177 additions and 156 deletions

View File

@@ -12,6 +12,7 @@
"ext-ctype": "*", "ext-ctype": "*",
"ext-curl": "*", "ext-curl": "*",
"ext-date": "*", "ext-date": "*",
"ext-encoding": "~1.0.0",
"ext-gmp": "*", "ext-gmp": "*",
"ext-hash": "*", "ext-hash": "*",
"ext-igbinary": "^3.0.1", "ext-igbinary": "^3.0.1",
@@ -36,7 +37,7 @@
"pocketmine/bedrock-block-upgrade-schema": "~5.1.0+bedrock-1.21.60", "pocketmine/bedrock-block-upgrade-schema": "~5.1.0+bedrock-1.21.60",
"pocketmine/bedrock-data": "~6.0.0+bedrock-1.21.100", "pocketmine/bedrock-data": "~6.0.0+bedrock-1.21.100",
"pocketmine/bedrock-item-upgrade-schema": "~1.15.0+bedrock-1.21.100", "pocketmine/bedrock-item-upgrade-schema": "~1.15.0+bedrock-1.21.100",
"pocketmine/bedrock-protocol": "~41.0.0+bedrock-1.21.100", "pocketmine/bedrock-protocol": "~50.0.0+bedrock-1.21.100",
"pocketmine/binaryutils": "^0.2.1", "pocketmine/binaryutils": "^0.2.1",
"pocketmine/callback-validator": "^1.0.2", "pocketmine/callback-validator": "^1.0.2",
"pocketmine/color": "^0.3.0", "pocketmine/color": "^0.3.0",

16
composer.lock generated
View File

@@ -4,7 +4,7 @@
"Read more about it at https://getcomposer.org/doc/01-basic-usage.md#installing-dependencies", "Read more about it at https://getcomposer.org/doc/01-basic-usage.md#installing-dependencies",
"This file is @generated automatically" "This file is @generated automatically"
], ],
"content-hash": "1e7545f6cc226b31d54238602143ba78", "content-hash": "0d71d3fba23ba8c4734cac59b9e10129",
"packages": [ "packages": [
{ {
"name": "adhocore/json-comment", "name": "adhocore/json-comment",
@@ -256,19 +256,20 @@
}, },
{ {
"name": "pocketmine/bedrock-protocol", "name": "pocketmine/bedrock-protocol",
"version": "41.0.0+bedrock-1.21.100", "version": "50.0.0+bedrock-1.21.100",
"source": { "source": {
"type": "git", "type": "git",
"url": "https://github.com/pmmp/BedrockProtocol.git", "url": "https://github.com/pmmp/BedrockProtocol.git",
"reference": "920ac291fe1b0143b2ebc90b3374ddab0b8531bf" "reference": "2d7aa27a5537ae593fb1c39158648ea462fef72a"
}, },
"dist": { "dist": {
"type": "zip", "type": "zip",
"url": "https://api.github.com/repos/pmmp/BedrockProtocol/zipball/920ac291fe1b0143b2ebc90b3374ddab0b8531bf", "url": "https://api.github.com/repos/pmmp/BedrockProtocol/zipball/2d7aa27a5537ae593fb1c39158648ea462fef72a",
"reference": "920ac291fe1b0143b2ebc90b3374ddab0b8531bf", "reference": "2d7aa27a5537ae593fb1c39158648ea462fef72a",
"shasum": "" "shasum": ""
}, },
"require": { "require": {
"ext-encoding": "~1.0.0",
"ext-json": "*", "ext-json": "*",
"php": "^8.1", "php": "^8.1",
"pocketmine/binaryutils": "^0.2.0", "pocketmine/binaryutils": "^0.2.0",
@@ -296,9 +297,9 @@
"description": "An implementation of the Minecraft: Bedrock Edition protocol in PHP", "description": "An implementation of the Minecraft: Bedrock Edition protocol in PHP",
"support": { "support": {
"issues": "https://github.com/pmmp/BedrockProtocol/issues", "issues": "https://github.com/pmmp/BedrockProtocol/issues",
"source": "https://github.com/pmmp/BedrockProtocol/tree/41.0.0+bedrock-1.21.100" "source": "https://github.com/pmmp/BedrockProtocol/tree/50.0.0+bedrock-1.21.100"
}, },
"time": "2025-09-09T20:52:18+00:00" "time": "2025-09-20T23:09:19+00:00"
}, },
{ {
"name": "pocketmine/binaryutils", "name": "pocketmine/binaryutils",
@@ -2797,6 +2798,7 @@
"ext-ctype": "*", "ext-ctype": "*",
"ext-curl": "*", "ext-curl": "*",
"ext-date": "*", "ext-date": "*",
"ext-encoding": "~1.0.0",
"ext-gmp": "*", "ext-gmp": "*",
"ext-hash": "*", "ext-hash": "*",
"ext-igbinary": "^3.0.1", "ext-igbinary": "^3.0.1",

View File

@@ -98,6 +98,7 @@ namespace pocketmine {
"crypto" => "php-crypto", "crypto" => "php-crypto",
"ctype" => "ctype", "ctype" => "ctype",
"date" => "Date", "date" => "Date",
"encoding" => "pmmp/ext-encoding",
"gmp" => "GMP", "gmp" => "GMP",
"hash" => "Hash", "hash" => "Hash",
"igbinary" => "igbinary", "igbinary" => "igbinary",
@@ -155,6 +156,12 @@ namespace pocketmine {
} }
} }
if(($encoding_version = phpversion("encoding")) !== false){
if(version_compare($encoding_version, "1.0.0") < 0 || version_compare($encoding_version, "2.0.0") >= 0){
$messages[] = "pmmp/ext-encoding ^1.0.0 is required, while you have $encoding_version.";
}
}
if(extension_loaded("pocketmine")){ if(extension_loaded("pocketmine")){
$messages[] = "The native PocketMine extension is no longer supported."; $messages[] = "The native PocketMine extension is no longer supported.";
} }

View File

@@ -26,6 +26,8 @@ declare(strict_types=1);
*/ */
namespace pocketmine\block; namespace pocketmine\block;
use pmmp\encoding\BE;
use pmmp\encoding\LE;
use pocketmine\block\tile\Spawnable; use pocketmine\block\tile\Spawnable;
use pocketmine\block\tile\Tile; use pocketmine\block\tile\Tile;
use pocketmine\block\utils\SupportType; use pocketmine\block\utils\SupportType;
@@ -49,7 +51,6 @@ use pocketmine\math\Vector3;
use pocketmine\nbt\tag\CompoundTag; use pocketmine\nbt\tag\CompoundTag;
use pocketmine\player\Player; use pocketmine\player\Player;
use pocketmine\utils\AssumptionFailedError; use pocketmine\utils\AssumptionFailedError;
use pocketmine\utils\Binary;
use pocketmine\world\BlockTransaction; use pocketmine\world\BlockTransaction;
use pocketmine\world\format\Chunk; use pocketmine\world\format\Chunk;
use pocketmine\world\Position; use pocketmine\world\Position;
@@ -98,9 +99,10 @@ class Block{
* of operations required to compute the state ID (micro optimization). * of operations required to compute the state ID (micro optimization).
*/ */
private static function computeStateIdXorMask(int $typeId) : int{ private static function computeStateIdXorMask(int $typeId) : int{
//TODO: the mixed byte order here is probably a mistake, but it doesn't break anything for now
return return
$typeId << self::INTERNAL_STATE_DATA_BITS | $typeId << self::INTERNAL_STATE_DATA_BITS |
(Binary::readLong(hash('xxh3', Binary::writeLLong($typeId), binary: true)) & self::INTERNAL_STATE_DATA_MASK); (BE::unpackSignedLong(hash('xxh3', LE::packSignedLong($typeId), binary: true)) & self::INTERNAL_STATE_DATA_MASK);
} }
/** /**

View File

@@ -23,10 +23,11 @@ declare(strict_types=1);
namespace pocketmine\crafting; namespace pocketmine\crafting;
use pmmp\encoding\ByteBufferWriter;
use pmmp\encoding\VarInt;
use pocketmine\item\Item; use pocketmine\item\Item;
use pocketmine\nbt\LittleEndianNbtSerializer; use pocketmine\nbt\LittleEndianNbtSerializer;
use pocketmine\nbt\TreeRoot; use pocketmine\nbt\TreeRoot;
use pocketmine\utils\BinaryStream;
use pocketmine\utils\DestructorCallbackTrait; use pocketmine\utils\DestructorCallbackTrait;
use pocketmine\utils\ObjectSet; use pocketmine\utils\ObjectSet;
use function array_shift; use function array_shift;
@@ -114,11 +115,13 @@ class CraftingManager{
} }
private static function hashOutput(Item $output) : string{ private static function hashOutput(Item $output) : string{
$write = new BinaryStream(); $write = new ByteBufferWriter();
$write->putVarInt($output->getStateId()); VarInt::writeSignedInt($write, $output->getStateId());
$write->put((new LittleEndianNbtSerializer())->write(new TreeRoot($output->getNamedTag()))); //TODO: the NBT serializer allocates its own ByteBufferWriter, we should change the API in the future to
//allow passing our own to avoid this extra allocation
$write->writeByteArray((new LittleEndianNbtSerializer())->write(new TreeRoot($output->getNamedTag())));
return $write->getBuffer(); return $write->getData();
} }
/** /**

View File

@@ -23,11 +23,12 @@ declare(strict_types=1);
namespace pocketmine\data\bedrock\block\upgrade; namespace pocketmine\data\bedrock\block\upgrade;
use pmmp\encoding\ByteBufferReader;
use pmmp\encoding\DataDecodeException;
use pmmp\encoding\VarInt;
use pocketmine\data\bedrock\block\BlockStateData; use pocketmine\data\bedrock\block\BlockStateData;
use pocketmine\data\bedrock\block\BlockStateDeserializeException; use pocketmine\data\bedrock\block\BlockStateDeserializeException;
use pocketmine\nbt\LittleEndianNbtSerializer; use pocketmine\nbt\LittleEndianNbtSerializer;
use pocketmine\utils\BinaryDataException;
use pocketmine\utils\BinaryStream;
/** /**
* Handles translating legacy 1.12 block ID/meta into modern blockstates. * Handles translating legacy 1.12 block ID/meta into modern blockstates.
@@ -84,25 +85,25 @@ final class BlockIdMetaUpgrader{
public static function loadFromString(string $data, LegacyBlockIdToStringIdMap $idMap, BlockStateUpgrader $blockStateUpgrader) : self{ public static function loadFromString(string $data, LegacyBlockIdToStringIdMap $idMap, BlockStateUpgrader $blockStateUpgrader) : self{
$mappingTable = []; $mappingTable = [];
$legacyStateMapReader = new BinaryStream($data); $legacyStateMapReader = new ByteBufferReader($data);
$nbtReader = new LittleEndianNbtSerializer(); $nbtReader = new LittleEndianNbtSerializer();
$idCount = $legacyStateMapReader->getUnsignedVarInt(); $idCount = VarInt::readUnsignedInt($legacyStateMapReader);
for($idIndex = 0; $idIndex < $idCount; $idIndex++){ for($idIndex = 0; $idIndex < $idCount; $idIndex++){
$id = $legacyStateMapReader->get($legacyStateMapReader->getUnsignedVarInt()); $id = $legacyStateMapReader->readByteArray(VarInt::readUnsignedInt($legacyStateMapReader));
$metaCount = $legacyStateMapReader->getUnsignedVarInt(); $metaCount = VarInt::readUnsignedInt($legacyStateMapReader);
for($metaIndex = 0; $metaIndex < $metaCount; $metaIndex++){ for($metaIndex = 0; $metaIndex < $metaCount; $metaIndex++){
$meta = $legacyStateMapReader->getUnsignedVarInt(); $meta = VarInt::readUnsignedInt($legacyStateMapReader);
$offset = $legacyStateMapReader->getOffset(); $offset = $legacyStateMapReader->getOffset();
$state = $nbtReader->read($legacyStateMapReader->getBuffer(), $offset)->mustGetCompoundTag(); $state = $nbtReader->read($legacyStateMapReader->getData(), $offset)->mustGetCompoundTag();
$legacyStateMapReader->setOffset($offset); $legacyStateMapReader->setOffset($offset);
$mappingTable[$id][$meta] = $blockStateUpgrader->upgrade(BlockStateData::fromNbt($state)); $mappingTable[$id][$meta] = $blockStateUpgrader->upgrade(BlockStateData::fromNbt($state));
} }
} }
if(!$legacyStateMapReader->feof()){ if($legacyStateMapReader->getUnreadLength() > 0){
throw new BinaryDataException("Unexpected trailing data in legacy state map data"); throw new DataDecodeException("Unexpected trailing data in legacy state map data");
} }
return new self($mappingTable, $idMap); return new self($mappingTable, $idMap);

View File

@@ -23,6 +23,7 @@ declare(strict_types=1);
namespace pocketmine\network\mcpe; namespace pocketmine\network\mcpe;
use pmmp\encoding\ByteBufferWriter;
use pocketmine\network\mcpe\compression\CompressBatchPromise; use pocketmine\network\mcpe\compression\CompressBatchPromise;
use pocketmine\network\mcpe\compression\Compressor; use pocketmine\network\mcpe\compression\Compressor;
use pocketmine\network\mcpe\convert\TypeConverter; use pocketmine\network\mcpe\convert\TypeConverter;
@@ -33,7 +34,6 @@ use pocketmine\network\mcpe\protocol\types\DimensionIds;
use pocketmine\network\mcpe\serializer\ChunkSerializer; use pocketmine\network\mcpe\serializer\ChunkSerializer;
use pocketmine\scheduler\AsyncTask; use pocketmine\scheduler\AsyncTask;
use pocketmine\thread\NonThreadSafeValue; use pocketmine\thread\NonThreadSafeValue;
use pocketmine\utils\BinaryStream;
use pocketmine\world\format\Chunk; use pocketmine\world\format\Chunk;
use pocketmine\world\format\io\FastChunkSerializer; use pocketmine\world\format\io\FastChunkSerializer;
use function chr; use function chr;
@@ -73,11 +73,11 @@ class ChunkRequestTask extends AsyncTask{
$converter = TypeConverter::getInstance(); $converter = TypeConverter::getInstance();
$payload = ChunkSerializer::serializeFullChunk($chunk, $dimensionId, $converter->getBlockTranslator(), $this->tiles); $payload = ChunkSerializer::serializeFullChunk($chunk, $dimensionId, $converter->getBlockTranslator(), $this->tiles);
$stream = new BinaryStream(); $stream = new ByteBufferWriter();
PacketBatch::encodePackets($stream, [LevelChunkPacket::create(new ChunkPosition($this->chunkX, $this->chunkZ), $dimensionId, $subCount, false, null, $payload)]); PacketBatch::encodePackets($stream, [LevelChunkPacket::create(new ChunkPosition($this->chunkX, $this->chunkZ), $dimensionId, $subCount, false, null, $payload)]);
$compressor = $this->compressor->deserialize(); $compressor = $this->compressor->deserialize();
$this->setResult(chr($compressor->getNetworkId()) . $compressor->compress($stream->getBuffer())); $this->setResult(chr($compressor->getNetworkId()) . $compressor->compress($stream->getData()));
} }
public function onCompletion() : void{ public function onCompletion() : void{

View File

@@ -23,9 +23,10 @@ declare(strict_types=1);
namespace pocketmine\network\mcpe; namespace pocketmine\network\mcpe;
use pmmp\encoding\BE;
use pmmp\encoding\Byte;
use pmmp\encoding\ByteBufferReader;
use pocketmine\utils\AssumptionFailedError; use pocketmine\utils\AssumptionFailedError;
use pocketmine\utils\Binary;
use pocketmine\utils\BinaryStream;
use pocketmine\utils\Utils; use pocketmine\utils\Utils;
use function base64_decode; use function base64_decode;
use function base64_encode; use function base64_encode;
@@ -133,17 +134,17 @@ final class JwtUtils{
return self::ASN1_SEQUENCE_TAG . chr(strlen($sequence)) . $sequence; return self::ASN1_SEQUENCE_TAG . chr(strlen($sequence)) . $sequence;
} }
private static function signaturePartFromAsn1(BinaryStream $stream) : string{ private static function signaturePartFromAsn1(ByteBufferReader $stream) : string{
$prefix = $stream->get(1); $prefix = $stream->readByteArray(1);
if($prefix !== self::ASN1_INTEGER_TAG){ if($prefix !== self::ASN1_INTEGER_TAG){
throw new \InvalidArgumentException("Expected an ASN.1 INTEGER tag, got " . bin2hex($prefix)); throw new \InvalidArgumentException("Expected an ASN.1 INTEGER tag, got " . bin2hex($prefix));
} }
//we can assume the length is 1 byte here - if it were larger than 127, more complex logic would be needed //we can assume the length is 1 byte here - if it were larger than 127, more complex logic would be needed
$length = $stream->getByte(); $length = Byte::readUnsigned($stream);
if($length > self::SIGNATURE_PART_LENGTH + 1){ //each part may have an extra leading 0 byte to prevent it being interpreted as a negative number if($length > self::SIGNATURE_PART_LENGTH + 1){ //each part may have an extra leading 0 byte to prevent it being interpreted as a negative number
throw new \InvalidArgumentException("Expected at most 49 bytes for signature R or S, got $length"); throw new \InvalidArgumentException("Expected at most 49 bytes for signature R or S, got $length");
} }
$part = $stream->get($length); $part = $stream->readByteArray($length);
return str_pad(ltrim($part, "\x00"), self::SIGNATURE_PART_LENGTH, "\x00", STR_PAD_LEFT); return str_pad(ltrim($part, "\x00"), self::SIGNATURE_PART_LENGTH, "\x00", STR_PAD_LEFT);
} }
@@ -159,11 +160,11 @@ final class JwtUtils{
throw new \InvalidArgumentException("Invalid DER signature, expected $length sequence bytes, got " . strlen($parts)); throw new \InvalidArgumentException("Invalid DER signature, expected $length sequence bytes, got " . strlen($parts));
} }
$stream = new BinaryStream($parts); $stream = new ByteBufferReader($parts);
$rRaw = self::signaturePartFromAsn1($stream); $rRaw = self::signaturePartFromAsn1($stream);
$sRaw = self::signaturePartFromAsn1($stream); $sRaw = self::signaturePartFromAsn1($stream);
if(!$stream->feof()){ if($stream->getUnreadLength() > 0){
throw new \InvalidArgumentException("Invalid DER signature, unexpected trailing sequence data"); throw new \InvalidArgumentException("Invalid DER signature, unexpected trailing sequence data");
} }
@@ -250,7 +251,7 @@ final class JwtUtils{
return chr($length); return chr($length);
} }
$lengthBytes = ltrim(Binary::writeInt($length), "\x00"); $lengthBytes = ltrim(BE::packUnsignedInt($length), "\x00");
return chr(0x80 | strlen($lengthBytes)) . $lengthBytes; return chr(0x80 | strlen($lengthBytes)) . $lengthBytes;
} }

View File

@@ -23,6 +23,9 @@ declare(strict_types=1);
namespace pocketmine\network\mcpe; namespace pocketmine\network\mcpe;
use pmmp\encoding\ByteBufferReader;
use pmmp\encoding\ByteBufferWriter;
use pmmp\encoding\DataDecodeException;
use pocketmine\entity\effect\EffectInstance; use pocketmine\entity\effect\EffectInstance;
use pocketmine\event\player\PlayerDuplicateLoginEvent; use pocketmine\event\player\PlayerDuplicateLoginEvent;
use pocketmine\event\player\PlayerResourcePackOfferEvent; use pocketmine\event\player\PlayerResourcePackOfferEvent;
@@ -70,7 +73,6 @@ use pocketmine\network\mcpe\protocol\PlayerStartItemCooldownPacket;
use pocketmine\network\mcpe\protocol\PlayStatusPacket; use pocketmine\network\mcpe\protocol\PlayStatusPacket;
use pocketmine\network\mcpe\protocol\ProtocolInfo; use pocketmine\network\mcpe\protocol\ProtocolInfo;
use pocketmine\network\mcpe\protocol\serializer\PacketBatch; use pocketmine\network\mcpe\protocol\serializer\PacketBatch;
use pocketmine\network\mcpe\protocol\serializer\PacketSerializer;
use pocketmine\network\mcpe\protocol\ServerboundPacket; use pocketmine\network\mcpe\protocol\ServerboundPacket;
use pocketmine\network\mcpe\protocol\ServerToClientHandshakePacket; use pocketmine\network\mcpe\protocol\ServerToClientHandshakePacket;
use pocketmine\network\mcpe\protocol\SetDifficultyPacket; use pocketmine\network\mcpe\protocol\SetDifficultyPacket;
@@ -109,8 +111,6 @@ use pocketmine\promise\PromiseResolver;
use pocketmine\Server; use pocketmine\Server;
use pocketmine\timings\Timings; use pocketmine\timings\Timings;
use pocketmine\utils\AssumptionFailedError; use pocketmine\utils\AssumptionFailedError;
use pocketmine\utils\BinaryDataException;
use pocketmine\utils\BinaryStream;
use pocketmine\utils\ObjectSet; use pocketmine\utils\ObjectSet;
use pocketmine\utils\TextFormat; use pocketmine\utils\TextFormat;
use pocketmine\world\format\io\GlobalItemDataHandlers; use pocketmine\world\format\io\GlobalItemDataHandlers;
@@ -401,7 +401,7 @@ class NetworkSession{
} }
try{ try{
$stream = new BinaryStream($decompressed); $stream = new ByteBufferReader($decompressed);
foreach(PacketBatch::decodeRaw($stream) as $buffer){ foreach(PacketBatch::decodeRaw($stream) as $buffer){
$this->gamePacketLimiter->decrement(); $this->gamePacketLimiter->decrement();
$packet = $this->packetPool->getPacket($buffer); $packet = $this->packetPool->getPacket($buffer);
@@ -421,7 +421,7 @@ class NetworkSession{
break; break;
} }
} }
}catch(PacketDecodeException|BinaryDataException $e){ }catch(PacketDecodeException|DataDecodeException $e){
$this->logger->logException($e); $this->logger->logException($e);
throw PacketHandlingException::wrap($e, "Packet batch decode error"); throw PacketHandlingException::wrap($e, "Packet batch decode error");
} }
@@ -453,14 +453,14 @@ class NetworkSession{
$decodeTimings = Timings::getDecodeDataPacketTimings($packet); $decodeTimings = Timings::getDecodeDataPacketTimings($packet);
$decodeTimings->startTiming(); $decodeTimings->startTiming();
try{ try{
$stream = PacketSerializer::decoder($buffer, 0); $stream = new ByteBufferReader($buffer);
try{ try{
$packet->decode($stream); $packet->decode($stream);
}catch(PacketDecodeException $e){ }catch(PacketDecodeException $e){
throw PacketHandlingException::wrap($e); throw PacketHandlingException::wrap($e);
} }
if(!$stream->feof()){ if($stream->getUnreadLength() > 0){
$remains = substr($stream->getBuffer(), $stream->getOffset()); $remains = substr($stream->getData(), $stream->getOffset());
$this->logger->debug("Still " . strlen($remains) . " bytes unread in " . $packet->getName() . ": " . bin2hex($remains)); $this->logger->debug("Still " . strlen($remains) . " bytes unread in " . $packet->getName() . ": " . bin2hex($remains));
} }
}finally{ }finally{
@@ -478,7 +478,7 @@ class NetworkSession{
$handlerTimings->startTiming(); $handlerTimings->startTiming();
try{ try{
if($this->handler === null || !$packet->handle($this->handler)){ if($this->handler === null || !$packet->handle($this->handler)){
$this->logger->debug("Unhandled " . $packet->getName() . ": " . base64_encode($stream->getBuffer())); $this->logger->debug("Unhandled " . $packet->getName() . ": " . base64_encode($stream->getData()));
} }
}finally{ }finally{
$handlerTimings->stopTiming(); $handlerTimings->stopTiming();
@@ -530,8 +530,10 @@ class NetworkSession{
if($ackReceiptResolver !== null){ if($ackReceiptResolver !== null){
$this->sendBufferAckPromises[] = $ackReceiptResolver; $this->sendBufferAckPromises[] = $ackReceiptResolver;
} }
$writer = new ByteBufferWriter();
foreach($packets as $evPacket){ foreach($packets as $evPacket){
$this->addToSendBuffer(self::encodePacketTimed(PacketSerializer::encoder(), $evPacket)); $writer->clear(); //memory reuse let's gooooo
$this->addToSendBuffer(self::encodePacketTimed($writer, $evPacket));
} }
if($immediate){ if($immediate){
$this->flushGamePacketQueue(); $this->flushGamePacketQueue();
@@ -564,12 +566,12 @@ class NetworkSession{
/** /**
* @internal * @internal
*/ */
public static function encodePacketTimed(PacketSerializer $serializer, ClientboundPacket $packet) : string{ public static function encodePacketTimed(ByteBufferWriter $serializer, ClientboundPacket $packet) : string{
$timings = Timings::getEncodeDataPacketTimings($packet); $timings = Timings::getEncodeDataPacketTimings($packet);
$timings->startTiming(); $timings->startTiming();
try{ try{
$packet->encode($serializer); $packet->encode($serializer);
return $serializer->getBuffer(); return $serializer->getData();
}finally{ }finally{
$timings->stopTiming(); $timings->stopTiming();
} }
@@ -591,13 +593,13 @@ class NetworkSession{
$syncMode = false; $syncMode = false;
} }
$stream = new BinaryStream(); $stream = new ByteBufferWriter();
PacketBatch::encodeRaw($stream, $this->sendBuffer); PacketBatch::encodeRaw($stream, $this->sendBuffer);
if($this->enableCompression){ if($this->enableCompression){
$batch = $this->server->prepareBatch($stream->getBuffer(), $this->compressor, $syncMode, Timings::$playerNetworkSendCompressSessionBuffer); $batch = $this->server->prepareBatch($stream->getData(), $this->compressor, $syncMode, Timings::$playerNetworkSendCompressSessionBuffer);
}else{ }else{
$batch = $stream->getBuffer(); $batch = $stream->getData();
} }
$this->sendBuffer = []; $this->sendBuffer = [];
$ackPromises = $this->sendBufferAckPromises; $ackPromises = $this->sendBufferAckPromises;

View File

@@ -23,12 +23,11 @@ declare(strict_types=1);
namespace pocketmine\network\mcpe; namespace pocketmine\network\mcpe;
use pmmp\encoding\ByteBufferWriter;
use pocketmine\event\server\DataPacketSendEvent; use pocketmine\event\server\DataPacketSendEvent;
use pocketmine\network\mcpe\protocol\serializer\PacketBatch; use pocketmine\network\mcpe\protocol\serializer\PacketBatch;
use pocketmine\network\mcpe\protocol\serializer\PacketSerializer;
use pocketmine\Server; use pocketmine\Server;
use pocketmine\timings\Timings; use pocketmine\timings\Timings;
use pocketmine\utils\BinaryStream;
use function count; use function count;
use function log; use function log;
use function spl_object_id; use function spl_object_id;
@@ -64,8 +63,10 @@ final class StandardPacketBroadcaster implements PacketBroadcaster{
$totalLength = 0; $totalLength = 0;
$packetBuffers = []; $packetBuffers = [];
$writer = new ByteBufferWriter();
foreach($packets as $packet){ foreach($packets as $packet){
$buffer = NetworkSession::encodePacketTimed(PacketSerializer::encoder(), $packet); $writer->clear(); //memory reuse let's gooooo
$buffer = NetworkSession::encodePacketTimed($writer, $packet);
//varint length prefix + packet buffer //varint length prefix + packet buffer
$totalLength += (((int) log(strlen($buffer), 128)) + 1) + strlen($buffer); $totalLength += (((int) log(strlen($buffer), 128)) + 1) + strlen($buffer);
$packetBuffers[] = $buffer; $packetBuffers[] = $buffer;
@@ -77,9 +78,9 @@ final class StandardPacketBroadcaster implements PacketBroadcaster{
$threshold = $compressor->getCompressionThreshold(); $threshold = $compressor->getCompressionThreshold();
if(count($compressorTargets) > 1 && $threshold !== null && $totalLength >= $threshold){ if(count($compressorTargets) > 1 && $threshold !== null && $totalLength >= $threshold){
//do not prepare shared batch unless we're sure it will be compressed //do not prepare shared batch unless we're sure it will be compressed
$stream = new BinaryStream(); $stream = new ByteBufferWriter();
PacketBatch::encodeRaw($stream, $packetBuffers); PacketBatch::encodeRaw($stream, $packetBuffers);
$batchBuffer = $stream->getBuffer(); $batchBuffer = $stream->getData();
$batch = $this->server->prepareBatch($batchBuffer, $compressor, timings: Timings::$playerNetworkSendCompressBroadcast); $batch = $this->server->prepareBatch($batchBuffer, $compressor, timings: Timings::$playerNetworkSendCompressBroadcast);
foreach($compressorTargets as $target){ foreach($compressorTargets as $target){

View File

@@ -23,6 +23,7 @@ declare(strict_types=1);
namespace pocketmine\network\mcpe\cache; namespace pocketmine\network\mcpe\cache;
use pmmp\encoding\BE;
use pocketmine\crafting\CraftingManager; use pocketmine\crafting\CraftingManager;
use pocketmine\crafting\FurnaceType; use pocketmine\crafting\FurnaceType;
use pocketmine\crafting\ShapedRecipe; use pocketmine\crafting\ShapedRecipe;
@@ -41,7 +42,6 @@ use pocketmine\network\mcpe\protocol\types\recipe\ShapedRecipe as ProtocolShaped
use pocketmine\network\mcpe\protocol\types\recipe\ShapelessRecipe as ProtocolShapelessRecipe; use pocketmine\network\mcpe\protocol\types\recipe\ShapelessRecipe as ProtocolShapelessRecipe;
use pocketmine\timings\Timings; use pocketmine\timings\Timings;
use pocketmine\utils\AssumptionFailedError; use pocketmine\utils\AssumptionFailedError;
use pocketmine\utils\Binary;
use pocketmine\utils\SingletonTrait; use pocketmine\utils\SingletonTrait;
use Ramsey\Uuid\Uuid; use Ramsey\Uuid\Uuid;
use function array_map; use function array_map;
@@ -99,7 +99,7 @@ final class CraftingDataCache{
}; };
$recipesWithTypeIds[] = new ProtocolShapelessRecipe( $recipesWithTypeIds[] = new ProtocolShapelessRecipe(
CraftingDataPacket::ENTRY_SHAPELESS, CraftingDataPacket::ENTRY_SHAPELESS,
Binary::writeInt($recipeNetId), BE::packUnsignedInt($recipeNetId), //TODO: this should probably be changed to something human-readable
array_map($converter->coreRecipeIngredientToNet(...), $recipe->getIngredientList()), array_map($converter->coreRecipeIngredientToNet(...), $recipe->getIngredientList()),
array_map($converter->coreItemStackToNet(...), $recipe->getResults()), array_map($converter->coreItemStackToNet(...), $recipe->getResults()),
$nullUUID, $nullUUID,
@@ -118,7 +118,7 @@ final class CraftingDataCache{
} }
$recipesWithTypeIds[] = $r = new ProtocolShapedRecipe( $recipesWithTypeIds[] = $r = new ProtocolShapedRecipe(
CraftingDataPacket::ENTRY_SHAPED, CraftingDataPacket::ENTRY_SHAPED,
Binary::writeInt($recipeNetId), BE::packUnsignedInt($recipeNetId), //TODO: this should probably be changed to something human-readable
$inputs, $inputs,
array_map($converter->coreItemStackToNet(...), $recipe->getResults()), array_map($converter->coreItemStackToNet(...), $recipe->getResults()),
$nullUUID, $nullUUID,

View File

@@ -23,6 +23,8 @@ declare(strict_types=1);
namespace pocketmine\network\mcpe\convert; namespace pocketmine\network\mcpe\convert;
use pmmp\encoding\ByteBufferReader;
use pmmp\encoding\ByteBufferWriter;
use pocketmine\block\tile\Container; use pocketmine\block\tile\Container;
use pocketmine\block\VanillaBlocks; use pocketmine\block\VanillaBlocks;
use pocketmine\crafting\ExactRecipeIngredient; use pocketmine\crafting\ExactRecipeIngredient;
@@ -44,7 +46,6 @@ use pocketmine\nbt\tag\Tag;
use pocketmine\nbt\TreeRoot; use pocketmine\nbt\TreeRoot;
use pocketmine\nbt\UnexpectedTagTypeException; use pocketmine\nbt\UnexpectedTagTypeException;
use pocketmine\network\mcpe\protocol\serializer\ItemTypeDictionary; use pocketmine\network\mcpe\protocol\serializer\ItemTypeDictionary;
use pocketmine\network\mcpe\protocol\serializer\PacketSerializer;
use pocketmine\network\mcpe\protocol\types\GameMode as ProtocolGameMode; use pocketmine\network\mcpe\protocol\types\GameMode as ProtocolGameMode;
use pocketmine\network\mcpe\protocol\types\inventory\ItemStack; use pocketmine\network\mcpe\protocol\types\inventory\ItemStack;
use pocketmine\network\mcpe\protocol\types\inventory\ItemStackExtraData; use pocketmine\network\mcpe\protocol\types\inventory\ItemStackExtraData;
@@ -312,7 +313,7 @@ class TypeConverter{
$extraData = $id === $this->shieldRuntimeId ? $extraData = $id === $this->shieldRuntimeId ?
new ItemStackExtraDataShield($nbt, canPlaceOn: [], canDestroy: [], blockingTick: 0) : new ItemStackExtraDataShield($nbt, canPlaceOn: [], canDestroy: [], blockingTick: 0) :
new ItemStackExtraData($nbt, canPlaceOn: [], canDestroy: []); new ItemStackExtraData($nbt, canPlaceOn: [], canDestroy: []);
$extraDataSerializer = PacketSerializer::encoder(); $extraDataSerializer = new ByteBufferWriter();
$extraData->write($extraDataSerializer); $extraData->write($extraDataSerializer);
return new ItemStack( return new ItemStack(
@@ -320,7 +321,7 @@ class TypeConverter{
$meta, $meta,
$itemStack->getCount(), $itemStack->getCount(),
$blockRuntimeId ?? ItemTranslator::NO_BLOCK_RUNTIME_ID, $blockRuntimeId ?? ItemTranslator::NO_BLOCK_RUNTIME_ID,
$extraDataSerializer->getBuffer(), $extraDataSerializer->getData(),
); );
} }
@@ -359,7 +360,7 @@ class TypeConverter{
} }
public function deserializeItemStackExtraData(string $extraData, int $id) : ItemStackExtraData{ public function deserializeItemStackExtraData(string $extraData, int $id) : ItemStackExtraData{
$extraDataDeserializer = PacketSerializer::decoder($extraData, 0); $extraDataDeserializer = new ByteBufferReader($extraData);
return $id === $this->shieldRuntimeId ? return $id === $this->shieldRuntimeId ?
ItemStackExtraDataShield::read($extraDataDeserializer) : ItemStackExtraDataShield::read($extraDataDeserializer) :
ItemStackExtraData::read($extraDataDeserializer); ItemStackExtraData::read($extraDataDeserializer);

View File

@@ -24,7 +24,7 @@ declare(strict_types=1);
namespace pocketmine\network\mcpe\encryption; namespace pocketmine\network\mcpe\encryption;
use Crypto\Cipher; use Crypto\Cipher;
use pocketmine\utils\Binary; use pmmp\encoding\LE;
use function bin2hex; use function bin2hex;
use function openssl_digest; use function openssl_digest;
use function openssl_error_string; use function openssl_error_string;
@@ -104,7 +104,7 @@ class EncryptionContext{
} }
private function calculateChecksum(int $counter, string $payload) : string{ private function calculateChecksum(int $counter, string $payload) : string{
$hash = openssl_digest(Binary::writeLLong($counter) . $payload . $this->key, self::CHECKSUM_ALGO, true); $hash = openssl_digest(LE::packUnsignedLong($counter) . $payload . $this->key, self::CHECKSUM_ALGO, true);
if($hash === false){ if($hash === false){
throw new \RuntimeException("openssl_digest() error: " . openssl_error_string()); throw new \RuntimeException("openssl_digest() error: " . openssl_error_string());
} }

View File

@@ -23,16 +23,16 @@ declare(strict_types=1);
namespace pocketmine\network\mcpe\serializer; namespace pocketmine\network\mcpe\serializer;
use pmmp\encoding\Byte;
use pmmp\encoding\ByteBufferWriter;
use pmmp\encoding\VarInt;
use pocketmine\block\tile\Spawnable; use pocketmine\block\tile\Spawnable;
use pocketmine\data\bedrock\BiomeIds; use pocketmine\data\bedrock\BiomeIds;
use pocketmine\data\bedrock\LegacyBiomeIdToStringIdMap; use pocketmine\data\bedrock\LegacyBiomeIdToStringIdMap;
use pocketmine\nbt\TreeRoot; use pocketmine\nbt\TreeRoot;
use pocketmine\network\mcpe\convert\BlockTranslator; use pocketmine\network\mcpe\convert\BlockTranslator;
use pocketmine\network\mcpe\protocol\serializer\NetworkNbtSerializer; use pocketmine\network\mcpe\protocol\serializer\NetworkNbtSerializer;
use pocketmine\network\mcpe\protocol\serializer\PacketSerializer;
use pocketmine\network\mcpe\protocol\types\DimensionIds; use pocketmine\network\mcpe\protocol\types\DimensionIds;
use pocketmine\utils\Binary;
use pocketmine\utils\BinaryStream;
use pocketmine\world\format\Chunk; use pocketmine\world\format\Chunk;
use pocketmine\world\format\PalettedBlockArray; use pocketmine\world\format\PalettedBlockArray;
use pocketmine\world\format\SubChunk; use pocketmine\world\format\SubChunk;
@@ -84,7 +84,7 @@ final class ChunkSerializer{
* @phpstan-param DimensionIds::* $dimensionId * @phpstan-param DimensionIds::* $dimensionId
*/ */
public static function serializeFullChunk(Chunk $chunk, int $dimensionId, BlockTranslator $blockTranslator, ?string $tiles = null) : string{ public static function serializeFullChunk(Chunk $chunk, int $dimensionId, BlockTranslator $blockTranslator, ?string $tiles = null) : string{
$stream = PacketSerializer::encoder(); $stream = new ByteBufferWriter();
$subChunkCount = self::getSubChunkCount($chunk, $dimensionId); $subChunkCount = self::getSubChunkCount($chunk, $dimensionId);
$writtenCount = 0; $writtenCount = 0;
@@ -100,37 +100,34 @@ final class ChunkSerializer{
self::serializeBiomePalette($chunk->getSubChunk($y)->getBiomeArray(), $biomeIdMap, $stream); self::serializeBiomePalette($chunk->getSubChunk($y)->getBiomeArray(), $biomeIdMap, $stream);
} }
$stream->putByte(0); //border block array count Byte::writeUnsigned($stream, 0); //border block array count
//Border block entry format: 1 byte (4 bits X, 4 bits Z). These are however useless since they crash the regular client. //Border block entry format: 1 byte (4 bits X, 4 bits Z). These are however useless since they crash the regular client.
if($tiles !== null){ if($tiles !== null){
$stream->put($tiles); $stream->writeByteArray($tiles);
}else{ }else{
$stream->put(self::serializeTiles($chunk)); $stream->writeByteArray(self::serializeTiles($chunk));
} }
return $stream->getBuffer(); return $stream->getData();
} }
public static function serializeSubChunk(SubChunk $subChunk, BlockTranslator $blockTranslator, PacketSerializer $stream, bool $persistentBlockStates) : void{ public static function serializeSubChunk(SubChunk $subChunk, BlockTranslator $blockTranslator, ByteBufferWriter $stream, bool $persistentBlockStates) : void{
$layers = $subChunk->getBlockLayers(); $layers = $subChunk->getBlockLayers();
$stream->putByte(8); //version Byte::writeUnsigned($stream, 8); //version
$stream->putByte(count($layers)); Byte::writeUnsigned($stream, count($layers));
$blockStateDictionary = $blockTranslator->getBlockStateDictionary(); $blockStateDictionary = $blockTranslator->getBlockStateDictionary();
foreach($layers as $blocks){ foreach($layers as $blocks){
$bitsPerBlock = $blocks->getBitsPerBlock(); $bitsPerBlock = $blocks->getBitsPerBlock();
$words = $blocks->getWordArray(); $words = $blocks->getWordArray();
$stream->putByte(($bitsPerBlock << 1) | ($persistentBlockStates ? 0 : 1)); Byte::writeUnsigned($stream, ($bitsPerBlock << 1) | ($persistentBlockStates ? 0 : 1));
$stream->put($words); $stream->writeByteArray($words);
$palette = $blocks->getPalette(); $palette = $blocks->getPalette();
if($bitsPerBlock !== 0){ if($bitsPerBlock !== 0){
//these LSHIFT by 1 uvarints are optimizations: the client expects zigzag varints here VarInt::writeSignedInt($stream, count($palette)); //yes, this is intentionally zigzag
//but since we know they are always unsigned, we can avoid the extra fcall overhead of
//zigzag and just shift directly.
$stream->putUnsignedVarInt(count($palette) << 1); //yes, this is intentionally zigzag
} }
if($persistentBlockStates){ if($persistentBlockStates){
$nbtSerializer = new NetworkNbtSerializer(); $nbtSerializer = new NetworkNbtSerializer();
@@ -141,46 +138,43 @@ final class ChunkSerializer{
$state = $blockTranslator->getFallbackStateData(); $state = $blockTranslator->getFallbackStateData();
} }
$stream->put($nbtSerializer->write(new TreeRoot($state->toNbt()))); $stream->writeByteArray($nbtSerializer->write(new TreeRoot($state->toNbt())));
} }
}else{ }else{
//we would use writeSignedIntArray() here, but the gains of writing in batch are negated by the cost of
//allocating a temporary array for the mapped palette IDs, especially for small palettes
foreach($palette as $p){ foreach($palette as $p){
$stream->put(Binary::writeUnsignedVarInt($blockTranslator->internalIdToNetworkId($p) << 1)); VarInt::writeSignedInt($stream, $blockTranslator->internalIdToNetworkId($p));
} }
} }
} }
} }
private static function serializeBiomePalette(PalettedBlockArray $biomePalette, LegacyBiomeIdToStringIdMap $biomeIdMap, PacketSerializer $stream) : void{ private static function serializeBiomePalette(PalettedBlockArray $biomePalette, LegacyBiomeIdToStringIdMap $biomeIdMap, ByteBufferWriter $stream) : void{
$biomePaletteBitsPerBlock = $biomePalette->getBitsPerBlock(); $biomePaletteBitsPerBlock = $biomePalette->getBitsPerBlock();
$stream->putByte(($biomePaletteBitsPerBlock << 1) | 1); //the last bit is non-persistence (like for blocks), though it has no effect on biomes since they always use integer IDs Byte::writeUnsigned($stream, ($biomePaletteBitsPerBlock << 1) | 1); //the last bit is non-persistence (like for blocks), though it has no effect on biomes since they always use integer IDs
$stream->put($biomePalette->getWordArray()); $stream->writeByteArray($biomePalette->getWordArray());
//these LSHIFT by 1 uvarints are optimizations: the client expects zigzag varints here
//but since we know they are always unsigned, we can avoid the extra fcall overhead of
//zigzag and just shift directly.
$biomePaletteArray = $biomePalette->getPalette(); $biomePaletteArray = $biomePalette->getPalette();
if($biomePaletteBitsPerBlock !== 0){ if($biomePaletteBitsPerBlock !== 0){
$stream->putUnsignedVarInt(count($biomePaletteArray) << 1); VarInt::writeSignedInt($stream, count($biomePaletteArray));
} }
foreach($biomePaletteArray as $p){ foreach($biomePaletteArray as $p){
if($biomeIdMap->legacyToString($p) === null){ //we would use writeSignedIntArray() here, but the gains of writing in batch are negated by the cost of
//make sure we aren't sending bogus biomes - the 1.18.0 client crashes if we do this //allocating a temporary array for the mapped palette IDs, especially for small palettes
$p = BiomeIds::OCEAN; VarInt::writeSignedInt($stream, $biomeIdMap->legacyToString($p) !== null ? $p : BiomeIds::OCEAN);
}
$stream->put(Binary::writeUnsignedVarInt($p << 1));
} }
} }
public static function serializeTiles(Chunk $chunk) : string{ public static function serializeTiles(Chunk $chunk) : string{
$stream = new BinaryStream(); $stream = new ByteBufferWriter();
foreach($chunk->getTiles() as $tile){ foreach($chunk->getTiles() as $tile){
if($tile instanceof Spawnable){ if($tile instanceof Spawnable){
$stream->put($tile->getSerializedSpawnCompound()->getEncodedNbt()); $stream->writeByteArray($tile->getSerializedSpawnCompound()->getEncodedNbt());
} }
} }
return $stream->getBuffer(); return $stream->getData();
} }
} }

View File

@@ -27,16 +27,16 @@ declare(strict_types=1);
*/ */
namespace pocketmine\network\query; namespace pocketmine\network\query;
use pmmp\encoding\BE;
use pmmp\encoding\Byte;
use pmmp\encoding\ByteBufferReader;
use pmmp\encoding\ByteBufferWriter;
use pmmp\encoding\DataDecodeException;
use pocketmine\network\AdvancedNetworkInterface; use pocketmine\network\AdvancedNetworkInterface;
use pocketmine\network\RawPacketHandler; use pocketmine\network\RawPacketHandler;
use pocketmine\Server; use pocketmine\Server;
use pocketmine\utils\Binary;
use pocketmine\utils\BinaryDataException;
use pocketmine\utils\BinaryStream;
use function chr;
use function hash; use function hash;
use function random_bytes; use function random_bytes;
use function strlen;
use function substr; use function substr;
class QueryHandler implements RawPacketHandler{ class QueryHandler implements RawPacketHandler{
@@ -80,51 +80,53 @@ class QueryHandler implements RawPacketHandler{
} }
public static function getTokenString(string $token, string $salt) : int{ public static function getTokenString(string $token, string $salt) : int{
return Binary::readInt(substr(hash("sha512", $salt . ":" . $token, true), 7, 4)); return BE::unpackSignedInt(substr(hash("sha512", $salt . ":" . $token, true), 7, 4));
} }
public function handle(AdvancedNetworkInterface $interface, string $address, int $port, string $packet) : bool{ public function handle(AdvancedNetworkInterface $interface, string $address, int $port, string $packet) : bool{
try{ try{
$stream = new BinaryStream($packet); $stream = new ByteBufferReader($packet);
$header = $stream->get(2); $header = $stream->readByteArray(2);
if($header !== "\xfe\xfd"){ //TODO: have this filtered by the regex filter we installed above if($header !== "\xfe\xfd"){ //TODO: have this filtered by the regex filter we installed above
return false; return false;
} }
$packetType = $stream->getByte(); $packetType = Byte::readUnsigned($stream);
$sessionID = $stream->getInt(); $sessionID = BE::readUnsignedInt($stream);
switch($packetType){ switch($packetType){
case self::HANDSHAKE: //Handshake case self::HANDSHAKE: //Handshake
$reply = chr(self::HANDSHAKE); $writer = new ByteBufferWriter();
$reply .= Binary::writeInt($sessionID); Byte::writeUnsigned($writer, self::HANDSHAKE);
$reply .= self::getTokenString($this->token, $address) . "\x00"; BE::writeUnsignedInt($writer, $sessionID);
$writer->writeByteArray(self::getTokenString($this->token, $address) . "\x00");
$interface->sendRawPacket($address, $port, $reply); $interface->sendRawPacket($address, $port, $writer->getData());
return true; return true;
case self::STATISTICS: //Stat case self::STATISTICS: //Stat
$token = $stream->getInt(); $token = BE::readUnsignedInt($stream);
if($token !== ($t1 = self::getTokenString($this->token, $address)) && $token !== ($t2 = self::getTokenString($this->lastToken, $address))){ if($token !== ($t1 = self::getTokenString($this->token, $address)) && $token !== ($t2 = self::getTokenString($this->lastToken, $address))){
$this->logger->debug("Bad token $token from $address $port, expected $t1 or $t2"); $this->logger->debug("Bad token $token from $address $port, expected $t1 or $t2");
return true; return true;
} }
$reply = chr(self::STATISTICS); $writer = new ByteBufferWriter();
$reply .= Binary::writeInt($sessionID); Byte::writeUnsigned($writer, self::STATISTICS);
BE::writeUnsignedInt($writer, $sessionID);
$remaining = $stream->getRemaining(); $remaining = $stream->getUnreadLength();
if(strlen($remaining) === 4){ //TODO: check this! according to the spec, this should always be here and always be FF FF FF 01 if($remaining === 4){ //TODO: check this! according to the spec, this should always be here and always be FF FF FF 01
$reply .= $this->server->getQueryInformation()->getLongQuery(); $writer->writeByteArray($this->server->getQueryInformation()->getLongQuery());
}else{ }else{
$reply .= $this->server->getQueryInformation()->getShortQuery(); $writer->writeByteArray($this->server->getQueryInformation()->getShortQuery());
} }
$interface->sendRawPacket($address, $port, $reply); $interface->sendRawPacket($address, $port, $writer->getData());
return true; return true;
default: default:
return false; return false;
} }
}catch(BinaryDataException $e){ }catch(DataDecodeException $e){
$this->logger->debug("Bad packet from $address $port: " . $e->getMessage()); $this->logger->debug("Bad packet from $address $port: " . $e->getMessage());
return false; return false;
} }

View File

@@ -23,11 +23,11 @@ declare(strict_types=1);
namespace pocketmine\network\query; namespace pocketmine\network\query;
use pmmp\encoding\LE;
use pocketmine\player\GameMode; use pocketmine\player\GameMode;
use pocketmine\player\Player; use pocketmine\player\Player;
use pocketmine\plugin\Plugin; use pocketmine\plugin\Plugin;
use pocketmine\Server; use pocketmine\Server;
use pocketmine\utils\Binary;
use pocketmine\utils\Utils; use pocketmine\utils\Utils;
use pocketmine\YmlServerProperties; use pocketmine\YmlServerProperties;
use function array_map; use function array_map;
@@ -236,6 +236,6 @@ final class QueryInfo{
} }
public function getShortQuery() : string{ public function getShortQuery() : string{
return $this->shortQueryCache ?? ($this->shortQueryCache = $this->serverName . "\x00" . $this->gametype . "\x00" . $this->map . "\x00" . $this->numPlayers . "\x00" . $this->maxPlayers . "\x00" . Binary::writeLShort($this->port) . $this->ip . "\x00"); return $this->shortQueryCache ?? ($this->shortQueryCache = $this->serverName . "\x00" . $this->gametype . "\x00" . $this->map . "\x00" . $this->numPlayers . "\x00" . $this->maxPlayers . "\x00" . LE::packUnsignedShort($this->port) . $this->ip . "\x00");
} }
} }

View File

@@ -23,8 +23,10 @@ declare(strict_types=1);
namespace pocketmine\world\format\io; namespace pocketmine\world\format\io;
use pocketmine\utils\Binary; use pmmp\encoding\BE;
use pocketmine\utils\BinaryStream; use pmmp\encoding\Byte;
use pmmp\encoding\ByteBufferReader;
use pmmp\encoding\ByteBufferWriter;
use pocketmine\world\format\Chunk; use pocketmine\world\format\Chunk;
use pocketmine\world\format\PalettedBlockArray; use pocketmine\world\format\PalettedBlockArray;
use pocketmine\world\format\SubChunk; use pocketmine\world\format\SubChunk;
@@ -45,15 +47,15 @@ final class FastChunkSerializer{
//NOOP //NOOP
} }
private static function serializePalettedArray(BinaryStream $stream, PalettedBlockArray $array) : void{ private static function serializePalettedArray(ByteBufferWriter $stream, PalettedBlockArray $array) : void{
$wordArray = $array->getWordArray(); $wordArray = $array->getWordArray();
$palette = $array->getPalette(); $palette = $array->getPalette();
$stream->putByte($array->getBitsPerBlock()); Byte::writeUnsigned($stream, $array->getBitsPerBlock());
$stream->put($wordArray); $stream->writeByteArray($wordArray);
$serialPalette = pack("L*", ...$palette); $serialPalette = pack("L*", ...$palette);
$stream->putInt(strlen($serialPalette)); BE::writeUnsignedInt($stream, strlen($serialPalette));
$stream->put($serialPalette); $stream->writeByteArray($serialPalette);
} }
/** /**
@@ -61,21 +63,20 @@ final class FastChunkSerializer{
* TODO: tiles and entities * TODO: tiles and entities
*/ */
public static function serializeTerrain(Chunk $chunk) : string{ public static function serializeTerrain(Chunk $chunk) : string{
$stream = new BinaryStream(); $stream = new ByteBufferWriter();
$stream->putByte( Byte::writeUnsigned($stream, ($chunk->isPopulated() ? self::FLAG_POPULATED : 0));
($chunk->isPopulated() ? self::FLAG_POPULATED : 0)
);
//subchunks //subchunks
$subChunks = $chunk->getSubChunks(); $subChunks = $chunk->getSubChunks();
$count = count($subChunks); $count = count($subChunks);
$stream->putByte($count); Byte::writeUnsigned($stream, $count);
foreach($subChunks as $y => $subChunk){ foreach($subChunks as $y => $subChunk){
$stream->putByte($y); Byte::writeSigned($stream, $y);
$stream->putInt($subChunk->getEmptyBlockId()); BE::writeUnsignedInt($stream, $subChunk->getEmptyBlockId());
$layers = $subChunk->getBlockLayers(); $layers = $subChunk->getBlockLayers();
$stream->putByte(count($layers)); Byte::writeUnsigned($stream, count($layers));
foreach($layers as $blocks){ foreach($layers as $blocks){
self::serializePalettedArray($stream, $blocks); self::serializePalettedArray($stream, $blocks);
} }
@@ -83,14 +84,15 @@ final class FastChunkSerializer{
} }
return $stream->getBuffer(); return $stream->getData();
} }
private static function deserializePalettedArray(BinaryStream $stream) : PalettedBlockArray{ private static function deserializePalettedArray(ByteBufferReader $stream) : PalettedBlockArray{
$bitsPerBlock = $stream->getByte(); $bitsPerBlock = Byte::readUnsigned($stream);
$words = $stream->get(PalettedBlockArray::getExpectedWordArraySize($bitsPerBlock)); $words = $stream->readByteArray(PalettedBlockArray::getExpectedWordArraySize($bitsPerBlock));
$paletteSize = BE::readUnsignedInt($stream);
/** @var int[] $unpackedPalette */ /** @var int[] $unpackedPalette */
$unpackedPalette = unpack("L*", $stream->get($stream->getInt())); //unpack() will never fail here $unpackedPalette = unpack("L*", $stream->readByteArray($paletteSize)); //unpack() will never fail here
$palette = array_values($unpackedPalette); $palette = array_values($unpackedPalette);
return PalettedBlockArray::fromData($bitsPerBlock, $words, $palette); return PalettedBlockArray::fromData($bitsPerBlock, $words, $palette);
@@ -100,20 +102,21 @@ final class FastChunkSerializer{
* Deserializes a fast-serialized chunk * Deserializes a fast-serialized chunk
*/ */
public static function deserializeTerrain(string $data) : Chunk{ public static function deserializeTerrain(string $data) : Chunk{
$stream = new BinaryStream($data); $stream = new ByteBufferReader($data);
$flags = $stream->getByte(); $flags = Byte::readUnsigned($stream);
$terrainPopulated = (bool) ($flags & self::FLAG_POPULATED); $terrainPopulated = (bool) ($flags & self::FLAG_POPULATED);
$subChunks = []; $subChunks = [];
$count = $stream->getByte(); $count = Byte::readUnsigned($stream);
for($subCount = 0; $subCount < $count; ++$subCount){ for($subCount = 0; $subCount < $count; ++$subCount){
$y = Binary::signByte($stream->getByte()); $y = Byte::readSigned($stream);
$airBlockId = $stream->getInt(); //TODO: why the heck are we using big-endian here?
$airBlockId = BE::readUnsignedInt($stream);
$layers = []; $layers = [];
for($i = 0, $layerCount = $stream->getByte(); $i < $layerCount; ++$i){ for($i = 0, $layerCount = Byte::readUnsigned($stream); $i < $layerCount; ++$i){
$layers[] = self::deserializePalettedArray($stream); $layers[] = self::deserializePalettedArray($stream);
} }
$biomeArray = self::deserializePalettedArray($stream); $biomeArray = self::deserializePalettedArray($stream);

View File

@@ -23,6 +23,7 @@ declare(strict_types=1);
namespace pocketmine\tools\generate_bedrock_data_from_packets; namespace pocketmine\tools\generate_bedrock_data_from_packets;
use pmmp\encoding\ByteBufferReader;
use pocketmine\crafting\json\FurnaceRecipeData; use pocketmine\crafting\json\FurnaceRecipeData;
use pocketmine\crafting\json\ItemStackData; use pocketmine\crafting\json\ItemStackData;
use pocketmine\crafting\json\PotionContainerChangeRecipeData; use pocketmine\crafting\json\PotionContainerChangeRecipeData;
@@ -51,7 +52,6 @@ use pocketmine\network\mcpe\protocol\CreativeContentPacket;
use pocketmine\network\mcpe\protocol\ItemRegistryPacket; use pocketmine\network\mcpe\protocol\ItemRegistryPacket;
use pocketmine\network\mcpe\protocol\PacketPool; use pocketmine\network\mcpe\protocol\PacketPool;
use pocketmine\network\mcpe\protocol\serializer\ItemTypeDictionary; use pocketmine\network\mcpe\protocol\serializer\ItemTypeDictionary;
use pocketmine\network\mcpe\protocol\serializer\PacketSerializer;
use pocketmine\network\mcpe\protocol\StartGamePacket; use pocketmine\network\mcpe\protocol\StartGamePacket;
use pocketmine\network\mcpe\protocol\types\inventory\CreativeGroupEntry; use pocketmine\network\mcpe\protocol\types\inventory\CreativeGroupEntry;
use pocketmine\network\mcpe\protocol\types\inventory\ItemStack; use pocketmine\network\mcpe\protocol\types\inventory\ItemStack;
@@ -190,7 +190,7 @@ class ParserPacketHandler extends PacketHandler{
$rawExtraData = $itemStack->getRawExtraData(); $rawExtraData = $itemStack->getRawExtraData();
if($rawExtraData !== ""){ if($rawExtraData !== ""){
$decoder = PacketSerializer::decoder($rawExtraData, 0); $decoder = new ByteBufferReader($rawExtraData);
$extraData = $itemStringId === ItemTypeNames::SHIELD ? ItemStackExtraDataShield::read($decoder) : ItemStackExtraData::read($decoder); $extraData = $itemStringId === ItemTypeNames::SHIELD ? ItemStackExtraDataShield::read($decoder) : ItemStackExtraData::read($decoder);
$nbt = $extraData->getNbt(); $nbt = $extraData->getNbt();
if($nbt !== null && count($nbt) > 0){ if($nbt !== null && count($nbt) > 0){
@@ -645,12 +645,13 @@ function main(array $argv) : int{
fwrite(STDERR, "Unknown packet on line " . ($lineNum + 1) . ": " . $parts[1]); fwrite(STDERR, "Unknown packet on line " . ($lineNum + 1) . ": " . $parts[1]);
continue; continue;
} }
$serializer = PacketSerializer::decoder($raw, 0); $serializer = new ByteBufferReader($raw);
$pk->decode($serializer); $pk->decode($serializer);
$pk->handle($handler); $pk->handle($handler);
if(!$serializer->feof()){ $remaining = strlen($serializer->getData()) - $serializer->getOffset();
echo "Packet on line " . ($lineNum + 1) . ": didn't read all data from " . get_class($pk) . " (stopped at offset " . $serializer->getOffset() . " of " . strlen($serializer->getBuffer()) . " bytes): " . bin2hex($serializer->getRemaining()) . "\n"; if($remaining > 0){
echo "Packet on line " . ($lineNum + 1) . ": didn't read all data from " . get_class($pk) . " (stopped at offset " . $serializer->getOffset() . " of " . strlen($serializer->getData()) . " bytes): " . bin2hex($serializer->readByteArray($remaining)) . "\n";
} }
} }
return 0; return 0;