Remove never-working caching and fix bugs in region code (#2716)

Fixes #2684

Author: octylFractal

worldedit-core/src/main/java/com/sk89q/worldedit/world/snapshot/SnapshotRestore.java

@@ -32,6 +32,8 @@
 import com.sk89q.worldedit.world.chunk.Chunk;
 import com.sk89q.worldedit.world.storage.ChunkStore;
 import com.sk89q.worldedit.world.storage.MissingChunkException;
+import org.apache.logging.log4j.LogManager;
+import org.apache.logging.log4j.Logger;
 import org.enginehub.linbus.tree.LinCompoundTag;
 import org.enginehub.linbus.tree.LinDoubleTag;
 import org.enginehub.linbus.tree.LinFloatTag;
@@ -50,6 +52,8 @@
  */
 public class SnapshotRestore {
 
+    private static final Logger LOGGER = LogManager.getLogger();
+
     //FAWE start - Set instead of ArrayList
     private final Map<BlockVector2, Set<BlockVector3>> neededChunks = new LinkedHashMap<>();
     //FAWE end
@@ -223,6 +227,7 @@ public void restore() throws MaxChangedBlocksException {
             } catch (MissingChunkException me) {
                 missingChunks.add(chunkPos);
             } catch (IOException | DataException me) {
+                LOGGER.info(() -> "Failed to load chunk at " + chunkPos, me);
                 errorChunks.add(chunkPos);
                 lastErrorMessage = me.getMessage();
             }

worldedit-core/src/main/java/com/sk89q/worldedit/world/storage/McRegionChunkStore.java

@@ -23,16 +23,17 @@
 import com.sk89q.worldedit.math.BlockVector2;
 import com.sk89q.worldedit.world.DataException;
 import com.sk89q.worldedit.world.World;
+import org.enginehub.linbus.stream.LinBinaryIO;
+import org.enginehub.linbus.tree.LinCompoundTag;
+import org.enginehub.linbus.tree.LinRootEntry;
 
 import javax.annotation.Nullable;
+import java.io.DataInputStream;
 import java.io.IOException;
 import java.io.InputStream;
 
 public abstract class McRegionChunkStore extends ChunkStore {
 
-    protected String curFilename = null;
-    protected McRegionReader cachedReader = null;
-
     /**
      * Get the filename of a region file.
      *
@@ -51,22 +52,11 @@ protected McRegionReader getReader(BlockVector2 pos, String worldname, @Nullable
             IOException {
         //FAWE end
         String filename = getFilename(pos);
-        if (curFilename != null) {
-            if (curFilename.equals(filename)) {
-                return cachedReader;
-            } else {
-                try {
-                    cachedReader.close();
-                } catch (IOException ignored) {
-                }
-            }
-        }
         //FAWE start - biome and entity restore
         InputStream stream = getInputStream(filename, worldname, folderOverride);
         //FAWE end
-        cachedReader = new McRegionReader(stream);
         //curFilename = filename;
-        return cachedReader;
+        return new McRegionReader(stream);
     }
 
     @Override
@@ -107,11 +97,4 @@ protected abstract InputStream getInputStream(String name, String worldName, @Nu
             IOException, DataException;
     //FAWE end
 
-    @Override
-    public void close() throws IOException {
-        if (cachedReader != null) {
-            cachedReader.close();
-        }
-    }
-
 }

worldedit-core/src/main/java/com/sk89q/worldedit/world/storage/McRegionReader.java

@@ -60,7 +60,9 @@ counts. The chunk offset for a chunk (x, z) begins at byte 4*(x+z*32) in the
 import com.sk89q.worldedit.world.DataException;
 
 import java.io.ByteArrayInputStream;
+import java.io.Closeable;
 import java.io.DataInputStream;
+import java.io.EOFException;
 import java.io.IOException;
 import java.io.InputStream;
 import java.util.zip.GZIPInputStream;
@@ -70,7 +72,7 @@ counts. The chunk offset for a chunk (x, z) begins at byte 4*(x+z*32) in the
  * Reader for a MCRegion file. This reader works on input streams, meaning
  * that it can be used to read files from non-file based sources.
  */
-public class McRegionReader {
+public class McRegionReader implements Closeable {
 
     protected static final int VERSION_GZIP = 1;
     protected static final int VERSION_DEFLATE = 2;
@@ -129,10 +131,10 @@ public synchronized InputStream getChunkInputStream(BlockVector2 position) throw
             throw new DataException("The chunk at " + position + " is not generated");
         }
 
-        int sectorNumber = offset >> 8;
+        long sectorNumber = offset >> 8;
         int numSectors = offset & 0xFF;
 
-        stream.seek((long) sectorNumber * SECTOR_BYTES);
+        stream.seek(sectorNumber * SECTOR_BYTES);
         int length = dataStream.readInt();
 
         if (length > SECTOR_BYTES * numSectors) {
@@ -142,19 +144,17 @@ public synchronized InputStream getChunkInputStream(BlockVector2 position) throw
 
         byte version = dataStream.readByte();
 
+        byte[] data = new byte[length - 1];
+        try {
+            dataStream.readFully(data);
+        } catch (EOFException e) {
+            throw new DataException("MCRegion file does not contain "
+                + x + "," + z + " in full");
+        }
+
         if (version == VERSION_GZIP) {
-            byte[] data = new byte[length - 1];
-            if (dataStream.read(data) < length - 1) {
-                throw new DataException("MCRegion file does not contain "
-                        + x + "," + z + " in full");
-            }
             return new GZIPInputStream(new ByteArrayInputStream(data));
         } else if (version == VERSION_DEFLATE) {
-            byte[] data = new byte[length - 1];
-            if (dataStream.read(data) < length - 1) {
-                throw new DataException("MCRegion file does not contain "
-                        + x + "," + z + " in full");
-            }
             return new InflaterInputStream(new ByteArrayInputStream(data));
         } else {
             throw new DataException("MCRegion chunk at "
@@ -187,6 +187,7 @@ public boolean hasChunk(int x, int z) {
     /**
      * Close the stream.
      */
+    @Override
     public void close() throws IOException {
         stream.close();
     }