code refine

apache · May 31, 2017 · 22c98a3 · 22c98a3
1 parent 9be5d80
commit 22c98a3
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/main/java/org/apache/commons/crypto/cipher/OpenSslGaloisCounterMode.java b/src/main/java/org/apache/commons/crypto/cipher/OpenSslGaloisCounterMode.java
@@ -54,6 +54,12 @@ public OpenSslGaloisCounterMode(long context, int algorithmMode, int padding) {
     public void init(int mode, byte[] key, AlgorithmParameterSpec params)
             throws InvalidAlgorithmParameterException {
 
+        if (aadBuffer == null) {
+            aadBuffer = new ByteArrayOutputStream();
+        } else {
+            aadBuffer.reset();
+        }
+
         this.cipherMode = mode;
         byte[] iv;
         if (params instanceof GCMParameterSpec) {
@@ -255,6 +261,11 @@ public int doFinal(ByteBuffer input, ByteBuffer output)
         return totalLen;
     }
 
+    public void clean() {
+        super.clean();
+        aadBuffer = null;
+    }
+
     @Override
     public void updateAAD(byte[] aad) {
         // must be called after initialized.