Refactor packet parsing and authentication

Author: JanJakes

packages/wp-mysql-proxy/src/class-mysql-protocol.php

@@ -229,18 +229,20 @@ class MySQL_Protocol {
 	 * @see https://dev.mysql.com/doc/dev/mysql-server/latest/page_protocol_basic_ok_packet.html
 	 *
 	 * @param  int $sequence_id    The sequence ID of the packet.
+	 * @param  int $server_status  The status flags representing the server state.
 	 * @param  int $affected_rows  Number of rows affected by the query.
 	 * @param  int $last_insert_id The last insert ID.
-	 * @param  int $server_status  The status flags representing the server state.
 	 * @param  int $warning_count  The warning count.
+	 * @param  int $packet_header  The packet header, indicating an OK or EOF semantic.
 	 * @return string              The OK packet.
 	 */
 	public static function build_ok_packet(
 		int $sequence_id,
-		int $affected_rows,
-		int $last_insert_id,
 		int $server_status,
-		int $warning_count
+		int $affected_rows = 0,
+		int $last_insert_id = 0,
+		int $warning_count = 0,
+		int $packet_header = self::OK_PACKET_HEADER
 	): string {
 		/**
 		 * Assemble the OK packet payload.
@@ -255,16 +257,47 @@ public static function build_ok_packet(
 		 */
 		$payload = pack(
 			'Ca*a*vv',
-			self::OK_PACKET_HEADER,                             // (C)  OK packet header.
+			$packet_header,                                     // (C)  OK packet header.
 			self::encode_length_encoded_int( $affected_rows ),  // (a*) Affected rows.
 			self::encode_length_encoded_int( $last_insert_id ), // (a*) Last insert ID.
 			$server_status,                                     // (v)  Server status flags.
 			$warning_count,                                     // (v)  Server status flags.
-			// No human-readable message for simplicity
 		);
 		return self::build_packet( $sequence_id, $payload );
 	}
 
+	/**
+	 * Build the OK packet with an EOF header.
+	 *
+	 * When the CLIENT_DEPRECATE_EOF capability is supported, an OK packet with
+	 * an EOF header is used to mark EOF, instead of the deprecated EOF packet.
+	 *
+	 * @see https://dev.mysql.com/doc/dev/mysql-server/latest/page_protocol_basic_ok_packet.html
+	 *
+	 * @param  int $sequence_id    The sequence ID of the packet.
+	 * @param  int $server_status  The status flags representing the server state.
+	 * @param  int $affected_rows  Number of rows affected by the query.
+	 * @param  int $last_insert_id The last insert ID.
+	 * @param  int $warning_count  The warning count.
+	 * @return string              The OK packet.
+	 */
+	public static function build_ok_packet_as_eof(
+		int $sequence_id,
+		int $server_status,
+		int $affected_rows = 0,
+		int $last_insert_id = 0,
+		int $warning_count = 0
+	): string {
+		return self::build_ok_packet(
+			$sequence_id,
+			$server_status,
+			$affected_rows,
+			$last_insert_id,
+			$warning_count,
+			self::EOF_PACKET_HEADER
+		);
+	}
+
 	/**
 	 * Build the ERR packet.
 	 *
@@ -316,7 +349,7 @@ public static function build_err_packet(
 	public static function build_eof_packet(
 		int $sequence_id,
 		int $server_status,
-		int $warning_count
+		int $warning_count = 0
 	): string {
 		/**
 		 * Assemble the EOF packet payload.