[Truffle] Finished implementing MatchData#pre_match.

jruby · Jun 1, 2015 · f084a7b · f084a7b
1 parent 3d202ba
commit f084a7b
Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/spec/truffle/tags/core/matchdata/pre_match_tags.txt b/spec/truffle/tags/core/matchdata/pre_match_tags.txt
diff --git a/truffle/src/main/java/org/jruby/truffle/nodes/core/MatchDataNodes.java b/truffle/src/main/java/org/jruby/truffle/nodes/core/MatchDataNodes.java
@@ -207,13 +207,16 @@ public int length(RubyMatchData matchData) {
     @CoreMethod(names = "pre_match")
     public abstract static class PreMatchNode extends CoreMethodArrayArgumentsNode {
 
+        @Child private TaintResultNode taintResultNode;
+
         public PreMatchNode(RubyContext context, SourceSection sourceSection) {
             super(context, sourceSection);
+            taintResultNode = new TaintResultNode(getContext(), getSourceSection());
         }
 
         @Specialization
-        public RubyBasicObject preMatch(RubyMatchData matchData) {
-            return matchData.getPre();
+        public Object preMatch(RubyMatchData matchData) {
+            return taintResultNode.maybeTaint(matchData.getSource(), matchData.getPre());
         }
 
     }