squash: Fix seeking the original file.

Author: damencho

src/main/java/org/ebml/matroska/MatroskaFileTags.java

@@ -19,7 +19,8 @@ public class MatroskaFileTags
 
   private final ArrayList<MatroskaFileTagEntry> tags = new ArrayList<>();
 
-  private long myPosition;
+  private long myStartPosition;
+  private long myEndPosition;
 
   public void addTag(final MatroskaFileTagEntry tag)
   {
@@ -28,15 +29,18 @@ public void addTag(final MatroskaFileTagEntry tag)
 
   public long writeTags(final DataWriter ioDW)
   {
-    myPosition = ioDW.getFilePointer();
+    myStartPosition = ioDW.getFilePointer();
     final MasterElement tagsElem = MatroskaDocTypes.Tags.getInstance();
 
     for (final MatroskaFileTagEntry tag : tags)
     {
       tagsElem.addChildElement(tag.toElement());
     }
 
-    if (BLOCK_SIZE < tagsElem.getTotalSize() && ioDW.isSeekable())
+    if (BLOCK_SIZE < tagsElem.getTotalSize() && ioDW.isSeekable()
+        // do the shuffling the data only if the file is big enough to contain the data
+        // if it is not it means we are writing the file for the first time and we don't need to shuffle the data
+        && ioDW.length() > myStartPosition + tagsElem.getTotalSize())
     {
       long len;
 
@@ -48,10 +52,13 @@ public long writeTags(final DataWriter ioDW)
         len = tagsElem.writeElement(dw);
 
         // now let's copy the rest of the original file by first setting the position after the tags
-        ioDW.seek(myPosition + BLOCK_SIZE);
+        ioDW.seek(myEndPosition);
 
         // copy the rest of the original file
         ((FileDataWriter)ioDW).copyEndOfFile(dw);
+        myEndPosition = myStartPosition + len;
+
+        ioDW.seek(myEndPosition);
       }
       catch (IOException ex)
       {
@@ -62,7 +69,7 @@ public long writeTags(final DataWriter ioDW)
     }
 
     long len = tagsElem.writeElement(ioDW);
-
+    myEndPosition = ioDW.getFilePointer();
     if (BLOCK_SIZE > tagsElem.getTotalSize() && ioDW.isSeekable())
     {
       new VoidElement(BLOCK_SIZE - tagsElem.getTotalSize()).writeElement(ioDW);
@@ -76,7 +83,7 @@ public long update(final DataWriter ioDW)
   {
     LOG.info("Updating tags list!");
     final long start = ioDW.getFilePointer();
-    ioDW.seek(myPosition);
+    ioDW.seek(myStartPosition);
     long len = writeTags(ioDW);
     ioDW.seek(start);
     return len;
@@ -87,7 +94,9 @@ public void propertyChange(PropertyChangeEvent evt)
   {
     if (evt.getPropertyName().equals(MatroskaFileTracks.RESIZED))
     {
-      myPosition = myPosition + ((long) evt.getNewValue() - (long) evt.getOldValue());
+      long increase = (long) evt.getNewValue() - (long) evt.getOldValue();
+      myStartPosition += increase;
+      myEndPosition += increase;
     }
   }
 }

src/main/java/org/ebml/matroska/MatroskaFileTracks.java

@@ -22,7 +22,8 @@ public class MatroskaFileTracks
 
   private final ArrayList<MatroskaFileTrack> tracks = new ArrayList<>();
 
-  private long myPosition;
+  private long myStartPosition;
+  private long myEndPosition;
 
   public void addTrack(final MatroskaFileTrack track)
   {
@@ -31,15 +32,18 @@ public void addTrack(final MatroskaFileTrack track)
 
   public long writeTracks(final DataWriter ioDW)
   {
-    myPosition = ioDW.getFilePointer();
+    myStartPosition = ioDW.getFilePointer();
     final MasterElement tracksElem = MatroskaDocTypes.Tracks.getInstance();
 
     for (final MatroskaFileTrack track : tracks)
     {
       tracksElem.addChildElement(track.toElement());
     }
 
-    if (BLOCK_SIZE < tracksElem.getTotalSize() && ioDW.isSeekable())
+    if (BLOCK_SIZE < tracksElem.getTotalSize() && ioDW.isSeekable()
+        // do the shuffling the data only if the file is big enough to contain the data
+        // if it is not it means we are writing the file for the first time and we don't need to shuffle the data
+        && ioDW.length() > myStartPosition + tracksElem.getTotalSize())
     {
       long len;
 
@@ -51,10 +55,14 @@ public long writeTracks(final DataWriter ioDW)
         len = tracksElem.writeElement(dw);
 
         // now let's copy the rest of the original file by first setting the position after the tracks
-        ioDW.seek(myPosition + BLOCK_SIZE);
+        ioDW.seek(myEndPosition);
 
         // copy the rest of the original file
         ((FileDataWriter)ioDW).copyEndOfFile(dw);
+
+        myEndPosition = myStartPosition + len;
+
+        ioDW.seek(myEndPosition);
       }
       catch (IOException ex)
       {
@@ -70,6 +78,7 @@ public long writeTracks(final DataWriter ioDW)
 
     long size = tracksElem.writeElement(ioDW);
 
+    myEndPosition = ioDW.getFilePointer();
     if (BLOCK_SIZE > tracksElem.getTotalSize() && ioDW.isSeekable())
     {
       new VoidElement(BLOCK_SIZE - tracksElem.getTotalSize()).writeElement(ioDW);
@@ -83,7 +92,7 @@ public long update(final DataWriter ioDW)
   {
     LOG.info("Updating tracks list!");
     final long start = ioDW.getFilePointer();
-    ioDW.seek(myPosition);
+    ioDW.seek(myStartPosition);
     long len = writeTracks(ioDW);
     ioDW.seek(start);
     return len;