Allow access to the built class byte[]

psuter · Jul 19, 2012 · f1d2a58 · f1d2a58
1 parent fd83079
commit f1d2a58
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/scala/cafebabe/ByteStream.scala b/src/main/scala/cafebabe/ByteStream.scala
@@ -10,7 +10,7 @@ class ByteStream {
   import ClassFileTypes._
 
   private var bytes = new ByteArrayOutputStream
-  protected[cafebabe] def getBytes : Array[Byte] = bytes.toByteArray
+  def getBytes : Array[Byte] = { stream.flush() ; bytes.toByteArray }
   private var stream: DataOutputStream = new DataOutputStream(bytes)
 
   // appends bytes to the stream