org.spigotmc.LimitStream Maven / Gradle / Ivy

Go to download
package org.spigotmc;

import net.minecraft.server.NBTReadLimiter;

import java.io.FilterInputStream;
import java.io.IOException;
import java.io.InputStream;

public class LimitStream extends FilterInputStream {

  private final NBTReadLimiter limit;

  public LimitStream(InputStream is, NBTReadLimiter limit) {
    super(is);
    this.limit = limit;
  }

  @Override
  public int read() throws IOException {
    limit.a(8);
    return super.read();
  }

  @Override
  public int read(byte[] b) throws IOException {
    limit.a(b.length * 8);
    return super.read(b);
  }

  @Override
  public int read(byte[] b, int off, int len) throws IOException {
    limit.a(len * 8);
    return super.read(b, off, len);
  }
}