improve sqlwall and parser

This commit is contained in:
Yako
2014-10-24 13:25:54 +08:00
parent 9da2cbc8d3
commit 2c9add76e6
17 changed files with 96 additions and 35 deletions
@@ -64,13 +64,15 @@ public class Lexer {
/*
* anti sql injection
*/
private boolean allowComment = true;
private boolean allowComment = true;
private int varIndex = -1;
private int varIndex = -1;
protected CommentHandler commentHandler;
protected boolean hasComment = false;
protected boolean hasComment = false;
protected boolean endOfComment = false;
public Lexer(String input){
this(input, true);
@@ -1058,4 +1060,9 @@ public class Lexer {
pos = text.length();
this.token = Token.EOF;
}
public boolean isEndOfComment() {
return endOfComment;
}
}
@@ -169,6 +169,9 @@ public class SQLParser {
} else if (lexer.token() == Token.USER) {
alias = lexer.stringVal();
lexer.nextToken();
} else if (lexer.token() == Token.END) {
alias = lexer.stringVal();
lexer.nextToken();
}
switch (lexer.token()) {