improve wall filter

src/main/java/com/alibaba/druid/sql/ast/expr/SQLIdentifierExpr.java

@@ -25,15 +25,16 @@ public class SQLIdentifierExpr extends SQLExprImpl implements SQLName {
 
     private String            name;
 
+    private transient String  lowerName;
+
     public SQLIdentifierExpr(){
 
     }
 
     public SQLIdentifierExpr(String name){
-
         this.name = name;
     }
-    
+
     public String getSimleName() {
         return name;
     }
@@ -44,6 +45,18 @@ public class SQLIdentifierExpr extends SQLExprImpl implements SQLName {
 
     public void setName(String name) {
         this.name = name;
+        this.lowerName = null;
+    }
+
+    public String getLowerName() {
+        if (lowerName == null && name != null) {
+            lowerName = name.toLowerCase();
+        }
+        return lowerName;
+    }
+
+    public void setLowerName(String lowerName) {
+        this.lowerName = lowerName;
     }
 
     public void output(StringBuffer buf) {

src/main/java/com/alibaba/druid/sql/dialect/mysql/parser/MySqlLexer.java

@@ -357,6 +357,10 @@ public class MySqlLexer extends Lexer {
             }
 
             for (;;) {
+                if (ch == EOI) {
+                    this.token = Token.ERROR;
+                    return;
+                }
                 if (ch == '*' && charAt(pos + 1) == '/') {
                     bufPos += 2;
                     scanChar();

src/main/java/com/alibaba/druid/sql/dialect/mysql/parser/MySqlStatementParser.java

@@ -189,7 +189,8 @@ public class MySqlStatementParser extends SQLStatementParser {
     }
 
     public SQLSelectStatement parseSelect() {
-        return new SQLSelectStatement(new MySqlSelectParser(this.exprParser).select());
+        MySqlSelectParser selectParser = new MySqlSelectParser(this.exprParser);
+        return new SQLSelectStatement(selectParser.select());
     }
 
     public SQLUpdateStatement parseUpdateStatement() {

src/main/java/com/alibaba/druid/wall/WallFilter.java

@@ -21,7 +21,6 @@ import java.sql.Wrapper;
 import java.util.Collections;
 import java.util.List;
 import java.util.Set;
-import java.util.concurrent.atomic.AtomicLong;
 
 import com.alibaba.druid.DruidRuntimeException;
 import com.alibaba.druid.filter.FilterAdapter;
@@ -56,9 +55,6 @@ public class WallFilter extends FilterAdapter implements WallFilterMBean {
     private volatile boolean logViolation   = false;
     private volatile boolean throwException = true;
 
-    // stats
-    private final AtomicLong violationCount = new AtomicLong();
-
     @Override
     public void init(DataSourceProxy dataSource) {
         if (this.dbType == null || this.dbType.trim().length() == 0) {
@@ -421,8 +417,6 @@ public class WallFilter extends FilterAdapter implements WallFilterMBean {
         List<Violation> violations = checkResult.getViolations();
 
         if (violations.size() > 0) {
-            violationCount.incrementAndGet();
-
             Violation firstViolation = violations.get(0);
             if (isLogViolation()) {
                 LOG.error("sql injection violation, " + firstViolation.getMessage() + " : " + sql);
@@ -461,7 +455,6 @@ public class WallFilter extends FilterAdapter implements WallFilterMBean {
         }
 
         if (!this.provider.getConfig().isWrapAllow()) {
-            violationCount.incrementAndGet();
             return null;
         }
 
@@ -475,7 +468,6 @@ public class WallFilter extends FilterAdapter implements WallFilterMBean {
         }
 
         if (!this.provider.getConfig().isMetadataAllow()) {
-            violationCount.incrementAndGet();
             if (isLogViolation()) {
                 LOG.error("not support method : Connection.getMetdataData");
             }
@@ -491,11 +483,11 @@ public class WallFilter extends FilterAdapter implements WallFilterMBean {
     }
 
     public long getViolationCount() {
-        return this.violationCount.get();
+        return this.provider.getViolationCount();
     }
 
     public void resetViolationCount() {
-        this.violationCount.set(0);
+        this.provider.reset();
     }
 
     public void clearWhiteList() {

src/main/java/com/alibaba/druid/wall/WallProvider.java

@@ -94,7 +94,10 @@ public abstract class WallProvider {
 
     public WallTableStat getTableStat(String tableName) {
         String lowerCaseName = tableName.toLowerCase();
-
+        return getTableStatWithLowerName(lowerCaseName);
+    }
+    
+    public WallTableStat getTableStatWithLowerName(String lowerCaseName) {
         WallTableStat stat = tableStats.get(lowerCaseName);
         if (stat == null) {
             if (tableStats.size() > 10000) {
@@ -385,7 +388,7 @@ public abstract class WallProvider {
                     for (Map.Entry<String, WallSqlTableStat> entry : sqlStat.getTableStats().entrySet()) {
                         String tableName = entry.getKey();
                         WallSqlTableStat sqlTableStat = entry.getValue();
-                        WallTableStat tableStat = getTableStat(tableName);
+                        WallTableStat tableStat = getTableStatWithLowerName(tableName);
                         if (tableStat != null) {
                             tableStat.addSqlTableStat(sqlTableStat);
                         }