From 653ca4f697dadcd75401e86a892ca771d677b697 Mon Sep 17 00:00:00 2001
From: Kijin Sung <kijin@kijinsung.com>
Date: Tue, 30 Sep 2025 22:27:36 +0900
Subject: [PATCH] Add "TABLE" to list of keywords to recognize in addPrefixes()

---
 common/framework/DB.php         |  6 +++---
 tests/unit/framework/DBTest.php | 16 ++++++++++++++++
 2 files changed, 19 insertions(+), 3 deletions(-)

diff --git a/common/framework/DB.php b/common/framework/DB.php
index b436ce9e8..0f30246e8 100644
--- a/common/framework/DB.php
+++ b/common/framework/DB.php
@@ -1126,17 +1126,17 @@ class DB
 		}
 		else
 		{
-			$exceptions = [];
+			$exceptions = ['TABLE'];
 		}
 
 		// Add prefixes to all other table names in the query string.
-		return preg_replace_callback('/\b((?:DELETE\s+)?FROM|JOIN|INTO|(?<!KEY\s)UPDATE)(?i)\s+((?:`?\w+`?)(?:\s+AS\s+`?\w+`?)?(?:\s*,\s*(?:`?\w+\`?)(?:\s+AS\s+`?\w+`?)?)*)/', function($m) use($exceptions) {
+		return preg_replace_callback('/\b((?:DELETE\s+)?FROM|JOIN|INTO(?: TABLE)?|TABLE|(?<!KEY\s)UPDATE)(?i)\s+((?:`?\w+`?)(?:\s+AS\s+`?\w+`?)?(?:\s*,\s*(?:`?\w+\`?)(?:\s+AS\s+`?\w+`?)?)*)/', function($m) use($exceptions) {
 			$type = strtoupper($m[1]);
 			$tables = array_map(function($str) use($type, $exceptions) {
 				return preg_replace_callback('/`?(\w+)`?(?:\s+AS\s+`?(\w+)`?)?/i', function($m) use($type, $exceptions) {
 					if (count($exceptions) && in_array($m[1], $exceptions))
 					{
-						return isset($m[2]) ? sprintf('`%s` AS `%s`',  $m[1], $m[2]) : sprintf('`%s`', $m[1]);
+						return isset($m[2]) ? sprintf('`%s` AS `%s`',  $m[1], $m[2]) : (ctype_upper($m[1]) ? $m[1] : sprintf('`%s`', $m[1]));
 					}
 					elseif ($type === 'FROM' || $type === 'JOIN')
 					{
diff --git a/tests/unit/framework/DBTest.php b/tests/unit/framework/DBTest.php
index 6fe855180..730b25aed 100644
--- a/tests/unit/framework/DBTest.php
+++ b/tests/unit/framework/DBTest.php
@@ -175,6 +175,22 @@ class DBTest extends \Codeception\Test\Unit
 		$target = 'INSERT INTO `' . $prefix . 'documents` (a, b, c) VALUES (?, ?, ?) ON DUPLICATE KEY UPDATE b = ?, c = ?';
 		$this->assertEquals($target, $oDB->addPrefixes($source));
 
+		$source = "LOAD DATA LOCAL INFILE '/tmp/foo.csv' INTO TABLE foo_table FIELDS TERMINATED BY ',' ENCLOSED BY '\"' LINES TERMINATED BY '\\n' (a, b, c)";
+		$target = "LOAD DATA LOCAL INFILE '/tmp/foo.csv' INTO TABLE `" . $prefix . "foo_table` FIELDS TERMINATED BY ',' ENCLOSED BY '\"' LINES TERMINATED BY '\\n' (a, b, c)";
+		$this->assertEquals($target, $oDB->addPrefixes($source));
+
+		$source = 'ALTER TABLE documents ADD INDEX idx_foo (a, b)';
+		$target = 'ALTER TABLE `' . $prefix . 'documents` ADD INDEX idx_foo (a, b)';
+		$this->assertEquals($target, $oDB->addPrefixes($source));
+
+		$source = 'TRUNCATE TABLE documents';
+		$target = 'TRUNCATE TABLE `' . $prefix . 'documents`';
+		$this->assertEquals($target, $oDB->addPrefixes($source));
+
+		$source = 'DROP TABLE documents';
+		$target = 'DROP TABLE `' . $prefix . 'documents`';
+		$this->assertEquals($target, $oDB->addPrefixes($source));
+
 		$source = 'update documents set a = ?, b = ? where c = ?';
 		$this->assertEquals($source, $oDB->addPrefixes($source));