Closing/Flushing DataOutputStream before calling toByteArray on the underlying ByteArrayOutputStream

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();
     }
 
 

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

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);
         }

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);
         }