Alawn/Druid
1
0
Fork 0
mirror of https://github.com/alibaba/druid.git synced 2023-03-04 19:17:16 +00:00
Code Issues Releases Wiki Activity

add wall test

Browse Source
This commit is contained in:
温少
2012-03-15 23:55:29 +08:00
parent 99ca885c12 b73a9b3c56
commit e90422fea1
6 changed files with 106 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/alibaba/druid/filter/wall/WallProvider.java
+16 -3
View File
@@ -1,9 +1,11 @@
package com.alibaba.druid.filter.wall;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import com.alibaba.druid.sql.ast.SQLStatement;
import com.alibaba.druid.sql.parser.NotAllowCommentException;
import com.alibaba.druid.sql.parser.SQLStatementParser;
import com.alibaba.druid.sql.parser.Token;
@@ -30,7 +32,7 @@ public abstract class WallProvider {
if (whiteList == null) {
return false;
}
return whiteList.get(sql) != null;
}
@@ -48,8 +50,19 @@ public abstract class WallProvider {
SQLStatementParser parser = createParser(sql);
parser.getLexer().setAllowComment(false); // permit comment
List<SQLStatement> statementList = parser.parseStatementList();
if(parser.getLexer().token() != Token.EOF) {
List<SQLStatement> statementList = new ArrayList<SQLStatement>();
try {
parser.parseStatementList(statementList);
} catch (NotAllowCommentException e) {
if (throwException) {
throw new WallRuntimeException("not allow comment : " + sql);
}
return false;
}
if (parser.getLexer().token() != Token.EOF) {
if (throwException) {
throw new WallRuntimeException("illegal statement : " + sql);
}
src/main/java/com/alibaba/druid/sql/dialect/hive/parser/HiveLexer.java
+9
View File
@@ -5,6 +5,7 @@ import java.util.Map;
import com.alibaba.druid.sql.parser.Keywords;
import com.alibaba.druid.sql.parser.Lexer;
import com.alibaba.druid.sql.parser.NotAllowCommentException;
import com.alibaba.druid.sql.parser.Token;
public class HiveLexer extends Lexer {
@@ -147,9 +148,17 @@ public class HiveLexer extends Lexer {
stringVal = new String(buf, np, sp);
token = Token.MULTI_LINE_COMMENT;
}
if (token != Token.HINT && !isAllowComment()) {
throw new NotAllowCommentException();
}
return;
}
if (!isAllowComment()) {
throw new NotAllowCommentException();
}
if (ch == '/') {
scanChar();
src/main/java/com/alibaba/druid/sql/dialect/oracle/parser/OracleLexer.java
+9
View File
@@ -7,6 +7,7 @@ import java.util.Map;
import com.alibaba.druid.sql.parser.Keywords;
import com.alibaba.druid.sql.parser.Lexer;
import com.alibaba.druid.sql.parser.NotAllowCommentException;
import com.alibaba.druid.sql.parser.SQLParseException;
import com.alibaba.druid.sql.parser.Token;
@@ -257,9 +258,17 @@ public class OracleLexer extends Lexer {
stringVal = new String(buf, np, sp);
token = Token.MULTI_LINE_COMMENT;
}
if (token != Token.HINT && !isAllowComment()) {
throw new NotAllowCommentException();
}
return;
}
if (!isAllowComment()) {
throw new NotAllowCommentException();
}
if (ch == '/' || ch == '-') {
scanChar();