additional buffer argument

Author: ahorek

src/main/java/org/jruby/ext/openssl/Cipher.java

@@ -1104,6 +1104,11 @@ private String getCipherAlgorithm() {
 
     @JRubyMethod
     public IRubyObject update(final ThreadContext context, final IRubyObject arg) {
+        return update(context, arg, null);
+    }
+
+    @JRubyMethod
+    public IRubyObject update(final ThreadContext context, final IRubyObject arg, IRubyObject buffer) {
         final Ruby runtime = context.runtime;
 
         if ( isDebug(runtime) ) dumpVars( runtime.getOut(), "update()" );
@@ -1143,7 +1148,11 @@ public IRubyObject update(final ThreadContext context, final IRubyObject arg) {
             debugStackTrace( runtime, e );
             throw newCipherError(runtime, e);
         }
-        return RubyString.newString(runtime, str);
+        if( buffer != null ) {
+            return ((RubyString) buffer).replace(RubyString.newString(runtime, str));
+        } else {
+            return RubyString.newString(runtime, str);
+        }
     }
 
     @JRubyMethod(name = "<<")

src/test/ruby/test_cipher.rb

@@ -475,4 +475,21 @@ def test_encrypt_aes_cfb_20_incompatibility
     end
   end
 
+  def test_encrypt_aes_256_cbc_modifies_buffer
+    cipher = OpenSSL::Cipher.new("AES-256-CBC")
+    cipher.key = "a" * 32
+    cipher.encrypt
+    buffer = ''
+    actual = cipher.update('bar' * 10, buffer)
+    if jruby?
+      expected = "\xE6\xD3Y\fc\xEE\xBA\xB2*\x0Fr\xD1\xC2b\x03\xD0"
+      assert_equal actual, expected
+      assert_equal buffer, expected
+    else
+      expected = "8\xA7\xBE\xB1\xAE\x88j\xCB\xA3\xE9j\x00\xD2W_\x91"
+      assert_equal actual, expected
+      assert_equal buffer, expected
+    end
+  end
+
 end