Merge pull request #1356 from Barenboim/master

Fix mysql parser.
sogou · Aug 17, 2023 · 3d8b3a8 · 3d8b3a8
2 parents 0caa115 + 5131d07
commit 3d8b3a8
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/protocol/mysql_parser.c b/src/protocol/mysql_parser.c
@@ -206,12 +206,21 @@ static int parse_ok_packet(const void *buf, size_t len, mysql_parser_t *parser)
 	{
 		const unsigned char *tmp_str;
 		unsigned long long tmp_len;
-	
+
 		if (decode_string(&str, &info_len, &p, buf_end) == 0)
 			return -2;
 
 		if (decode_string(&tmp_str, &tmp_len, &p, buf_end) == 0)
 			return -2;
+
+	} else if (p != buf_end &&
+			   *p != MYSQL_PACKET_HEADER_OK &&
+			   *p != MYSQL_PACKET_HEADER_NULL &&
+			   *p != MYSQL_PACKET_HEADER_EOF &&
+			   *p != MYSQL_PACKET_HEADER_ERROR)
+	{
+		if (decode_string(&str, &info_len, &p, buf_end) == 0)
+			return -2;
 	} else {
 		str = p;
 		info_len = 0;