From 4fd7cb8b4bb97dbf7be98e2fa55f36f5028807d2 Mon Sep 17 00:00:00 2001
From: emopers <emopers@gmail.com>
Date: Mon, 11 Jan 2016 02:45:03 -0600
Subject: [PATCH] Fixing usage of DataOutputStream

---
 src/main/java/net/sf/ntru/encrypt/EncryptionParameters.java | 1 +
 src/main/java/net/sf/ntru/sign/SignatureParameters.java     | 1 +
 src/main/java/net/sf/ntru/sign/SignaturePrivateKey.java     | 1 +
 src/main/java/net/sf/ntru/sign/SignaturePublicKey.java      | 1 +
 4 files changed, 4 insertions(+)

diff --git a/src/main/java/net/sf/ntru/encrypt/EncryptionParameters.java b/src/main/java/net/sf/ntru/encrypt/EncryptionParameters.java
index 03627cb..1e91e02 100644
--- 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
index 804c021..385e5e2 100644
--- 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
index 34feb91..ac2e157 100644
--- 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
index 825ded9..c65da31 100644
--- 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);
         }