logan
a9510235bb
驱动接入（全部本机真实例验收）： - Elasticsearch 原生 SQL API、InfluxDB 原生 HTTP+Flux（官方 JDBC 付费许可，实测弃用） - ZooKeeper/MongoDB JVM agent（连接时自动下载官方驱动 jar，sha256 钉版本到 ~/.opsxdb/drivers/） - 达梦 JDBC agent 桥就绪（待厂商 jar+实例）；TiDB 真实例验收 - 协议族代码审查放行 PolarDB/TDSQL/OceanBase/Kingbase/GaussDB/Oracle - 驱动能力模型：按驱动开关对象树分组/右键菜单/行编辑，NoSQL 树拍平（对象直挂库节点） 驱动管理： - 驱动管理弹框（状态/预装/离线包导入/JDK 检测）+ /api/drivers/* 三接口 - opsxdb-driver 驱动商店仓库（清单/fetch.sh/pack.sh 离线包/import.sh） - 连接时懒加载：DMG 零 Java 产物，首连自动下载（实测 ZK 九件套 1.35s），离线包自动回退 UI/交互： - 设置中心（⚙）：常规/执行/结果/驱动/快捷键/关于，左导航+右面板（对标 opsxshell） - 实例级展开箭头（行首 SVG chevron，替代字体回退失真的 Unicode 三角） - 树内嵌套渲染修复：对象树挂在激活连接下；工具栏窄窗口不断行（按钮 nowrap+整颗换行） - 侧栏连接行精简（图标+连接名）、SQLite SSH 守卫、Oracle 死参数清理 质量： - Go 协议层/离线回退单测、E2E 门控测试（ES/ZK/Influx/Mongo/TiDB/JDBC 桥） - 驱动注册表单一事实来源 connection-types.json，README/BACKLOG/AGENTS 同步
77 lines
3.3 KiB
Go
77 lines
3.3 KiB
Go
package db
|
|
|
|
import "testing"
|
|
|
|
// 危险语句命中/放行矩阵,含四类历史绕过:多语句、/*! */ 可执行注释、
|
|
// 注释吞掉 WHERE、WITH CTE 包裹写操作。
|
|
func TestCheckDangerousSQLMatrix(t *testing.T) {
|
|
dangerous := []string{
|
|
"DROP TABLE users",
|
|
"drop table users", // 大小写
|
|
"TRUNCATE TABLE t", // TRUNCATE
|
|
"ALTER TABLE t ADD COLUMN c int", // ALTER(误报方向维持现状)
|
|
"GRANT ALL ON *.* TO 'u'", // GRANT
|
|
"REVOKE ALL ON *.* FROM 'u'", // REVOKE
|
|
"CREATE USER 'u'@'%' IDENTIFIED BY 'p'", // CREATE USER
|
|
"CREATE ROLE app", // CREATE ROLE
|
|
"RENAME TABLE a TO b", // RENAME TABLE
|
|
"REPLACE INTO t VALUES (1)", // REPLACE INTO
|
|
"LOAD DATA INFILE '/tmp/x' INTO TABLE t", // LOAD DATA
|
|
"SELECT * INTO OUTFILE '/tmp/x' FROM t", // INTO OUTFILE
|
|
"SELECT * INTO DUMPFILE '/tmp/x' FROM t", // INTO DUMPFILE
|
|
"EXEC xp_cmdshell 'dir'", // xp_cmdshell
|
|
"EXEC sp_configure 'x'", // EXEC 开头
|
|
"CALL do_thing()", // CALL 开头
|
|
"COPY t FROM PROGRAM 'cmd'", // COPY FROM PROGRAM
|
|
"DELETE FROM users", // DELETE 无 WHERE
|
|
"UPDATE users SET x=1", // UPDATE 无 WHERE
|
|
"SELECT 1; DROP TABLE t", // 多语句绕过
|
|
"/*!50000 DROP TABLE t*/", // MySQL 可执行注释
|
|
"SELECT 1; /*!40101 SET @x = 1*/; DROP TABLE t", // 可执行注释 + 多语句
|
|
"UPDATE t SET x=1 -- WHERE id=2", // 注释吞掉 WHERE
|
|
"UPDATE t SET x=1 # WHERE id=2", // # 注释吞掉 WHERE
|
|
"UPDATE t SET x=1 /* WHERE id=2 */", // 块注释吞掉 WHERE
|
|
"WITH x AS (SELECT 1) DELETE FROM t", // CTE 包裹 DELETE 无 WHERE
|
|
"WITH x AS (SELECT 1) UPDATE t SET a=1", // CTE 包裹 UPDATE 无 WHERE
|
|
}
|
|
for _, sql := range dangerous {
|
|
if hits := CheckDangerousSQL("mysql", sql); len(hits) == 0 {
|
|
t.Errorf("应命中但未命中: %q", sql)
|
|
}
|
|
}
|
|
|
|
safe := []string{
|
|
"SELECT * FROM users",
|
|
"SELECT * FROM users WHERE id = 1",
|
|
"DELETE FROM users WHERE id = 1",
|
|
"UPDATE users SET x = 1 WHERE id = 2",
|
|
"UPDATE users SET x = 1\nWHERE id = 2",
|
|
"INSERT INTO t VALUES (1)",
|
|
"CREATE TABLE t (id int)",
|
|
"SELECT 'DROP TABLE x'", // 字符串内关键字
|
|
"SELECT * FROM t WHERE note = 'truncate it'", // 字符串内关键字
|
|
"-- DROP TABLE t\nSELECT 1", // 注释内关键字
|
|
"SELECT * FROM t /* DROP TABLE x */ WHERE 1=1", // 块注释内关键字
|
|
"SELECT * FROM updated WHERE id = 1", // 词边界:updated 不算 UPDATE
|
|
"SELECT 1; SELECT 2",
|
|
"WITH x AS (SELECT 1) SELECT * FROM x",
|
|
"SHOW TABLES",
|
|
"EXPLAIN SELECT 1",
|
|
}
|
|
for _, sql := range safe {
|
|
if hits := CheckDangerousSQL("mysql", sql); len(hits) > 0 {
|
|
t.Errorf("误报: %q → %v", sql, hits)
|
|
}
|
|
}
|
|
}
|
|
|
|
// 旧接口兼容:IsDangerousSQL 委托 CheckDangerousSQL。
|
|
func TestIsDangerousSQLCompat(t *testing.T) {
|
|
if !IsDangerousSQL("DROP TABLE users") {
|
|
t.Fatal("drop")
|
|
}
|
|
if IsDangerousSQL("SELECT * FROM users") {
|
|
t.Fatal("select")
|
|
}
|
|
}
|