update

Author: chaz6chez

examples/client.php

@@ -0,0 +1,77 @@
+<?php
+
+declare(strict_types=1);
+
+use Workbunny\MysqlProtocol\Packets\HandshakeInitialization;
+use Workbunny\MysqlProtocol\Packets\HandshakeResponse;
+use Workbunny\MysqlProtocol\Utils\Binary;
+use Workbunny\MysqlProtocol\Utils\Packet;
+use Workbunny\MysqlProtocol\Packets\Error;
+use Workbunny\MysqlProtocol\Packets\Command;
+use Workbunny\MysqlProtocol\Packets\Ok;
+use Workerman\Connection\TcpConnection;
+
+require_once __DIR__ . '/../vendor/autoload.php';
+
+$server = new \Workerman\Worker();
+$server->name = 'workbunny-mysql-client';
+$server->onWorkerStart = function () {
+    global $clientMysqlHandshakeStatus;
+    $clientMysqlHandshakeStatus = $clientMysqlHandshakeStatus ?: 0;
+    $client = new \Workerman\Connection\AsyncTcpConnection("MySQL://host.docker.internal:3306");
+
+    $client->onConnect = function (TcpConnection $connection) use (&$clientMysqlHandshakeStatus) {
+        $connection->errorHandler = function (Throwable $exception) {
+            dump($exception);
+        };
+        // 模拟心跳
+        \Workerman\Timer::add(30, function () use ($connection, &$clientMysqlHandshakeStatus) {
+            if ($clientMysqlHandshakeStatus > 1) {
+                $connection->send(Command::pack([
+                    'command'   => Command::COM_QUERY,
+                    'data'      => 'SELECT 1'
+                ]));
+            }
+        });
+    };
+
+    $client->onMessage = function (TcpConnection $connection, Binary $binary) use (&$clientMysqlHandshakeStatus) {
+        dump($binary->dump());
+        // 还未握手
+        if ($clientMysqlHandshakeStatus < 1) {
+            // 解析握手包
+            $handshakeInitialization = HandshakeInitialization::unpack($binary);
+            $clientMysqlHandshakeStatus = 1;
+            $connection->send(HandshakeResponse::pack([
+                'packet_id'         => $handshakeInitialization['packet_id'] + 1,
+                "capability_flags"  => $handshakeInitialization['capability_flags'],
+                "max_packet_size"   => 1073741824,
+                "character_set"     => 33,
+                "username"          => "root",
+                "database"          => null,
+                "auth_plugin"       => $handshakeInitialization['auth_plugin_name'],
+                "auth_response"     => "",
+            ]));
+        }
+        // 握手还未确认
+        elseif ($clientMysqlHandshakeStatus < 2) {
+            if (!$class = Packet::getPacketClass($binary)) {
+                echo "Error!\n";
+                return;
+            }
+            $result = $class::unpack($binary);
+            dump($result);
+            // todo判定是否握手成功
+
+        }
+        // 握手成功
+        else {
+
+            dump($binary->dump());
+        }
+    };
+
+    $client->connect();
+};
+
+\Workerman\Worker::runAll();

examples/server.php

@@ -59,7 +59,7 @@
     if ($connection->mysql_handshake_status < 1) {
         // 握手响应信息获取
         $handshakeResponse = HandshakeResponse::unpack($binary);
-
+        dump($handshakeResponse);
         // todo 可以对 数据信息进行验证，这里暂时不验证用户信息等
 
         // 状态机：1 已经握手

src/Packets/Error.php

@@ -18,7 +18,7 @@ class Error implements PacketInterface
      *
      * @param Binary $binary
      * @return array
-     *         - header: int 固定 0xFF
+     *         - flag: int 固定 0xFF
      *         - error_code: int
      *         - sql_state: string (5字节)
      *         - error_message: string

src/Utils/Packet.php

@@ -8,6 +8,10 @@
 use Workbunny\MysqlProtocol\Constants\ExceptionCode;
 use Workbunny\MysqlProtocol\Exceptions\Exception;
 use Workbunny\MysqlProtocol\Exceptions\PacketException;
+use Workbunny\MysqlProtocol\Packets\EOF;
+use Workbunny\MysqlProtocol\Packets\Error;
+use Workbunny\MysqlProtocol\Packets\OK;
+use Workbunny\MysqlProtocol\Packets\PacketInterface;
 
 class Packet
 {
@@ -55,11 +59,31 @@ public static function parser(?Closure $closure, Binary $binary): array
         ], $result);
     }
 
+    /**
+     * 根据包头获取包类型
+     *
+     * @param Binary $binary
+     * @return class-string<PacketInterface>|null
+     */
+    public static function getPacketClass(Binary $binary): ?string
+    {
+        $readCursor = $binary->getReadCursor();
+        $binary->setReadCursor(4);
+        $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,
+        };
+    }
+
     /**
      * 生成认证数据
      *
      * @param int $bytesCount
-     * @return array
+     * @return array<int>
      */
     public static function authData(int $bytesCount = 8): array
     {