Merge pull request #542 from lostsnow/feature/add-range-command-overwrite

add taint range command OVERWRITE

Author: lostsnow

dongtai-core/src/main/java/io/dongtai/iast/core/handler/hookpoint/models/taint/range/TaintCommand.java

@@ -13,6 +13,7 @@ public enum TaintCommand {
     REMOVE,
     REPLACE,
     CONCAT,
+    OVERWRITE,
     TRIM,
     TRIM_RIGHT,
     TRIM_LEFT,

dongtai-core/src/main/java/io/dongtai/iast/core/handler/hookpoint/models/taint/range/TaintCommandRunner.java

@@ -133,6 +133,8 @@ public TaintRanges run(PropagatorNode propagatorNode, Object source, Object targ
                 case CONCAT:
                     this.builder.concat(tr, target, oldTaintRanges, source, srcTaintRanges, params);
                     break;
+                case OVERWRITE:
+                    this.builder.overwrite(tr, oldTaintRanges, source, srcTaintRanges, p1, this.paramsCount);
                 case TRIM:
                 case TRIM_LEFT:
                 case TRIM_RIGHT:

dongtai-core/src/main/java/io/dongtai/iast/core/handler/hookpoint/models/taint/range/TaintRangesBuilder.java

@@ -144,6 +144,16 @@ public void concat(TaintRanges taintRanges, Object target, TaintRanges oldTaintR
         taintRanges.merge();
     }
 
+    public void overwrite(TaintRanges taintRanges, TaintRanges oldTagRanges, Object source, TaintRanges srcTagRanges, int p1, int argC) {
+        if (argC == 1) {
+            int length = getLength(source);
+            srcTagRanges.shift(p1);
+            oldTagRanges.clear(p1, length + p1);
+            taintRanges.addAll(oldTagRanges);
+            taintRanges.addAll(srcTagRanges);
+        }
+    }
+
     public void trim(TaintCommand command, TaintRanges taintRanges, Object source, TaintRanges srcTaintRanges, int argC) {
         if (argC > 0) {
             return;