fix captures nil case

fix #70

Author: naitoh

ext/jruby/org/jruby/ext/strscan/RubyStringScanner.java

@@ -892,6 +892,7 @@ public IRubyObject size(ThreadContext context) {
     public IRubyObject captures(ThreadContext context) {
         int i, numRegs;
         RubyArray newAry;
+        IRubyObject str;
 
         if (!isMatched()) return context.nil;
 
@@ -901,9 +902,13 @@ public IRubyObject captures(ThreadContext context) {
         newAry = RubyArray.newArray(runtime, numRegs);
 
         for (i = 1; i < numRegs; i++) {
-            IRubyObject str = extractRange(runtime,
+            if (REGION_ADAPTER.getBeg(regs, i) == -1) {
+                str = context.nil;
+            } else {
+                str = extractRange(runtime,
                     adjustRegisterPosition(REGION_ADAPTER.getBeg(regs, i)),
                     adjustRegisterPosition(REGION_ADAPTER.getEnd(regs, i)));
+            }
             newAry.push(str);
         }
 

ext/strscan/strscan.c

@@ -1243,17 +1243,18 @@ strscan_size(VALUE self)
  * If nothing was priorly matched, it returns nil.
  *
  *   s = StringScanner.new("Fri Dec 12 1975 14:39")
- *   s.scan(/(\w+) (\w+) (\d+) /)       # -> "Fri Dec 12 "
- *   s.captures                         # -> ["Fri", "Dec", "12"]
- *   s.scan(/(\w+) (\w+) (\d+) /)       # -> nil
- *   s.captures                         # -> nil
+ *   s.scan(/(\w+) (\w+) (\d+) (1980)?/)       # -> "Fri Dec 12 "
+ *   s.captures                                # -> ["Fri", "Dec", "12", nil]
+ *   s.scan(/(\w+) (\w+) (\d+) (1980)?/)       # -> nil
+ *   s.captures                                # -> nil
  */
 static VALUE
 strscan_captures(VALUE self)
 {
     struct strscanner *p;
     int   i, num_regs;
     VALUE new_ary;
+    VALUE str;
 
     GET_SCANNER(self, p);
     if (! MATCHED_P(p))        return Qnil;
@@ -1262,9 +1263,12 @@ strscan_captures(VALUE self)
     new_ary  = rb_ary_new2(num_regs);
 
     for (i = 1; i < num_regs; i++) {
-        VALUE str = extract_range(p,
-                                  adjust_register_position(p, p->regs.beg[i]),
-                                  adjust_register_position(p, p->regs.end[i]));
+        if (p->regs.beg[i] == -1)
+            str = Qnil;
+        else
+            str = extract_range(p,
+                                adjust_register_position(p, p->regs.beg[i]),
+                                adjust_register_position(p, p->regs.end[i]));
         rb_ary_push(new_ary, str);
     }
 

test/strscan/test_stringscanner.rb

@@ -737,8 +737,8 @@ def test_size
   def test_captures
     s = create_string_scanner("Timestamp: Fri Dec 12 1975 14:39")
     s.scan("Timestamp: ")
-    s.scan(/(\w+) (\w+) (\d+) /)
-    assert_equal(["Fri", "Dec", "12"], s.captures)
+    s.scan(/(\w+) (\w+) (\d+) (1980)?/)
+    assert_equal(["Fri", "Dec", "12", nil], s.captures)
     s.scan(/(\w+) (\w+) (\d+) /)
     assert_nil(s.captures)
   end