Fixing usage of DataOutputStream

tbuktu · Jan 11, 2016 · 4fd7cb8 · 4fd7cb8
1 parent 8126929
commit 4fd7cb8
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 0 deletions.
diff --git a/src/main/java/net/sf/ntru/encrypt/EncryptionParameters.java b/src/main/java/net/sf/ntru/encrypt/EncryptionParameters.java
@@ -252,6 +252,7 @@ public void writeTo(OutputStream os) throws IOException {
         dos.writeBoolean(fastFp);
         dos.write(polyType.ordinal());
         dos.writeUTF(hashAlg);
+        dos.flush();
     }
 
 

diff --git a/src/main/java/net/sf/ntru/sign/SignatureParameters.java b/src/main/java/net/sf/ntru/sign/SignatureParameters.java
@@ -213,6 +213,7 @@ public void writeTo(OutputStream os) throws IOException {
         dos.write(keyGenAlg.ordinal());
         dos.writeUTF(hashAlg);
         dos.write(polyType.ordinal());
+        dos.flush();
     }
 
     @Override

diff --git a/src/main/java/net/sf/ntru/sign/SignaturePrivateKey.java b/src/main/java/net/sf/ntru/sign/SignaturePrivateKey.java
@@ -147,6 +147,7 @@ public byte[] getEncoded() {
             for (int i=0; i<numBases; i++)
                 // all bases except for the first one contain a public key
                 bases.get(i).encode(os, i!=0);
+            dataStream.close();
         } catch (IOException e) {
             throw new NtruException(e);
         }

diff --git a/src/main/java/net/sf/ntru/sign/SignaturePublicKey.java b/src/main/java/net/sf/ntru/sign/SignaturePublicKey.java
@@ -92,6 +92,7 @@ public byte[] getEncoded() {
             dataStream.writeShort(h.coeffs.length);
             dataStream.writeShort(q);
             dataStream.write(h.toBinary(q));
+            dataStream.close();
         } catch (IOException e) {
             throw new NtruException(e);
         }