增加 proxy例子 移除不必要的文件

Author: chaz6chez

examples/proxy.php

@@ -0,0 +1,39 @@
+<?php
+
+declare(strict_types=1);
+
+use Workbunny\MysqlProtocol\Utils\Binary;
+use Workbunny\MysqlProtocol\Utils\Packet;
+use Workerman\Connection\TcpConnection;
+
+require_once __DIR__ . '/../vendor/autoload.php';
+
+$server = new \Workerman\Worker("MySQL://0.0.0.0:8844");
+$server->name = 'workbunny-mysql-server';
+$server->count = 2;
+$server->onConnect = function (TcpConnection $source) {
+    // 创建与MySQL-server的连接
+    $target = new \Workerman\Connection\AsyncTcpConnection("MySQL://host.docker.internal:3306");
+    // 管道传输
+    $target->pipe($source);
+    $target->onMessage = function (TcpConnection $target, Binary $data) use ($source) {
+        // 客户端的来源信息
+        dump('-------------------------------------------------------------------------------------------------------------------');
+        dump('Server', $data->dump(), Packet::parser(null, $data));
+        dump('Packet-Class: ' . Packet::getPacketClass($data));
+        dump('-------------------------------------------------------------------------------------------------------------------');
+        $source->send($data);
+    };
+
+    $source->pipe($target);
+    $source->onMessage = function (TcpConnection $source, Binary $data) use ($target) {
+        dump('-------------------------------------------------------------------------------------------------------------------');
+        dump('Client', $data->dump(), Packet::parser(null, $data));
+        dump('-------------------------------------------------------------------------------------------------------------------');
+        $target->send($data);
+    };
+
+    $target->connect();
+};
+
+\Workerman\Worker::runAll();

src/Packets/AuthMoreDataRequest.php

@@ -29,7 +29,7 @@ class AuthMoreDataRequest implements PacketInterface
      * @param Binary $binary
      * @return array 包含：
      *               - flag: int，标志字节
-     *               - extra_data: string，附加数据（可能为空）
+     *               - extra_data: array，附加数据字节组
      */
     public static function unpack(Binary $binary): array
     {
@@ -41,9 +41,9 @@ public static function unpack(Binary $binary): array
             }
             // 如果后续有附加数据，则读取之
             $remainingLength = $binary->length() - $binary->getReadCursor();
-            $extraData = '';
+            $extraData = [];
             if ($remainingLength > 0) {
-                $extraData = Binary::BytesToString($binary->readBytes($remainingLength));
+                $extraData = $binary->readBytes($remainingLength);
             }
 
             return [
@@ -60,7 +60,7 @@ public static function unpack(Binary $binary): array
      *   - flag: int，标志字节（默认 0x01 表示请求全认证）
      *
      * 可选键：
-     *   - extra_data: string，附加数据
+     *   - extra_data: array，附加数据
      *
      * @param array $data
      * @return Binary
@@ -69,10 +69,10 @@ public static function pack(array $data): Binary
     {
         $packetId = $data['packet_id'] ?? 0;
         return Packet::binary(function (Binary $binary) use ($data) {
-            $extraData = $data['extra_data'] ?? null;
+            $extraData = $data['extra_data'] ?? [];
             $binary->writeByte(self::PACKET_FLAG);
             if ($extraData) {
-                $binary->writeBytes(Binary::StringToBytes($extraData));
+                $binary->writeBytes($extraData);
             }
         }, $packetId);
     }

src/Plugins/AbstractPlugin.php

@@ -8,6 +8,8 @@
 use Workbunny\MysqlProtocol\Constants\ExceptionCode;
 use Workbunny\MysqlProtocol\Exceptions\Exception;
 use Workbunny\MysqlProtocol\Exceptions\PacketException;
+use Workbunny\MysqlProtocol\Packets\AuthMoreDataRequest;
+use Workbunny\MysqlProtocol\Packets\AuthSwitchRequest;
 use Workbunny\MysqlProtocol\Packets\EOF;
 use Workbunny\MysqlProtocol\Packets\Error;
 use Workbunny\MysqlProtocol\Packets\OK;
@@ -63,19 +65,33 @@ public static function parser(?Closure $closure, Binary $binary): array
      * 根据包头获取包类型
      *
      * @param Binary $binary
+     * @param Closure|null $closure = function(Binary $binary): class-string<PacketInterface>|null {}
      * @return class-string<PacketInterface>|null
      */
-    public static function getPacketClass(Binary $binary): ?string
+    public static function getPacketClass(Binary $binary, ?Closure $closure = null): ?string
     {
+        // 获取当前指针
         $readCursor = $binary->getReadCursor();
+        // 重置读指针
+        $binary->setReadCursor(0);
+        // 包长
+        $packetLength = $binary->readUB(Binary::UB3);
+        // 都指针下移包身
         $binary->setReadCursor(4);
+        // 读取flag
         $header = $binary->readByte();
+        // 恢复指针
         $binary->setReadCursor($readCursor);
-        return match ($header) {
-            Ok::PACKET_FLAG => OK::class,
-            Error::PACKET_FLAG => Error::class,
-            EOF::PACKET_FLAG => EOF::class,
-            default => null,
+        return match (true) {
+            $header === Ok::PACKET_FLAG => OK::class,
+            $header === Error::PACKET_FLAG => Error::class,
+            // EOF 包体长度小于9字节，且flag为0xFE
+            ($packetLength < 9) and $header === EOF::PACKET_FLAG => EOF::class,
+            // AuthSwitchRequest 包体长度通常大于9字节，且flag为0xFE
+            $header === AuthSwitchRequest::PACKET_FLAG => AuthSwitchRequest::class,
+            $header === AuthMoreDataRequest::PACKET_FLAG => AuthMoreDataRequest::class,
+            // 如果传入自定义解析函数，则使用
+            default => $closure ? $closure($binary) : null,
         };
     }