1 files changed, 16 insertions, 7 deletions
diff --git a/src/main/java/at/gv/egiz/pdfas/impl/input/ByteArrayPdfDataSourceImpl.java b/src/main/java/at/gv/egiz/pdfas/impl/input/ByteArrayPdfDataSourceImpl.java
index 0d27781..e4a5649 100644
--- a/src/main/java/at/gv/egiz/pdfas/impl/input/ByteArrayPdfDataSourceImpl.java
+++ b/src/main/java/at/gv/egiz/pdfas/impl/input/ByteArrayPdfDataSourceImpl.java
@@ -23,34 +23,43 @@ public class ByteArrayPdfDataSourceImpl implements PdfDataSource
 {
   protected byte[] pdf = null;
   
-  protected int length = -1;
-
   public ByteArrayPdfDataSourceImpl(byte[] pdf)
   {
     PerformanceCounters.byteArrays.increment();
 
     this.pdf = pdf;
-    this.length = pdf.length;
   }
 
   public ByteArrayPdfDataSourceImpl(byte[] pdf, int length)
   {
     PerformanceCounters.byteArrays.increment();
 
-    this.pdf = pdf;
-    this.length = length;
+    if (pdf.length == length)
+    {
+      this.pdf = pdf;
+    }
+    else
+    {
+      this.pdf = new byte [length];
+      System.arraycopy(pdf, 0, this.pdf, 0, length);
+    }
   }
 
 
   public InputStream createInputStream()
   {
-    ByteArrayInputStream bais = new ByteArrayInputStream(this.pdf, 0, this.length);
+    ByteArrayInputStream bais = new ByteArrayInputStream(this.pdf);
     return bais;
   }
 
   public int getLength()
   {
-    return this.length;
+    return this.pdf.length;
   }
 
+  public byte[] getAsByteArray()
+  {
+    return this.pdf;
+  }
+  
 }