/* * Copyright (c) 1994, 2019, Oracle and/or its affiliates. All rights reserved. * ORACLE PROPRIETARY/CONFIDENTIAL. Use is subject to license terms. * * * * * * * * * * * * * * * * * * * * */ package it.cavallium.stream; import java.nio.charset.Charset; import java.nio.charset.StandardCharsets; import it.cavallium.buffer.IgnoreCoverage; import org.jetbrains.annotations.NotNull; public class SafeDataInputStream extends SafeFilterInputStream implements SafeDataInput { /** * Creates a DataInputStream that uses the specified * underlying InputStream. * * @param in the specified input stream */ public SafeDataInputStream(SafeInputStream in) { super(in); } @Override public final int read(byte @NotNull[] b) { return in.read(b, 0, b.length); } /** * See the general contract of the {@code readFully} * method of {@code DataInput}. *
* Bytes * for this operation are read from the contained * input stream. * * @param b the buffer into which the data is read. * @throws NullPointerException if {@code b} is {@code null}. * @see SafeFilterInputStream#in */ @Override public final void readFully(byte @NotNull [] b) { readFully(b, 0, b.length); } /** * See the general contract of the {@code readFully} * method of {@code DataInput}. *
* Bytes * for this operation are read from the contained * input stream. * * @param b the buffer into which the data is read. * @param off the start offset in the data array {@code b}. * @param len the number of bytes to read. * @throws NullPointerException if {@code b} is {@code null}. * @throws IndexOutOfBoundsException if {@code off} is negative, * {@code len} is negative, or {@code len} is greater than * {@code b.length - off}. * @see SafeFilterInputStream#in */ @Override public final void readFully(byte @NotNull [] b, int off, int len) { if (len < 0) throw new IndexOutOfBoundsException(); int n = 0; while (n < len) { int count = in.read(b, off + n, len - n); if (count < 0) throw new IndexOutOfBoundsException(); n += count; } } /** * See the general contract of the {@code skipBytes} * method of {@code DataInput}. *
* Bytes for this operation are read from the contained
* input stream.
*
* @param n the number of bytes to be skipped.
* @return the actual number of bytes skipped.
*/
@Override
public final int skipBytes(int n) {
int total = 0;
int cur;
while ((total
* Bytes for this operation are read from the contained
* input stream.
*
* @return the {@code boolean} value read.
* @see SafeFilterInputStream#in
*/
@Override
public final boolean readBoolean() {
int ch = in.read();
if (ch < 0)
throw new IndexOutOfBoundsException();
return (ch != 0);
}
/**
* See the general contract of the {@code readByte}
* method of {@code DataInput}.
*
* Bytes
* for this operation are read from the contained
* input stream.
*
* @return the next byte of this input stream as a signed 8-bit
* {@code byte}.
* @see SafeFilterInputStream#in
*/
@Override
public final byte readByte() {
int ch = in.read();
if (ch < 0)
throw new IndexOutOfBoundsException();
return (byte)(ch);
}
/**
* See the general contract of the {@code readUnsignedByte}
* method of {@code DataInput}.
*
* Bytes
* for this operation are read from the contained
* input stream.
*
* @return the next byte of this input stream, interpreted as an
* unsigned 8-bit number.
* @see SafeFilterInputStream#in
*/
@Override
public final int readUnsignedByte() {
int ch = in.read();
if (ch < 0)
throw new IndexOutOfBoundsException();
return ch;
}
/**
* See the general contract of the {@code readShort}
* method of {@code DataInput}.
*
* Bytes
* for this operation are read from the contained
* input stream.
*
* @return the next two bytes of this input stream, interpreted as a
* signed 16-bit number.
* @see SafeFilterInputStream#in
*/
@Override
public final short readShort() {
int ch1 = in.read();
int ch2 = in.read();
if ((ch1 | ch2) < 0)
throw new IndexOutOfBoundsException();
return (short)((ch1 << 8) + (ch2));
}
/**
* See the general contract of the {@code readUnsignedShort}
* method of {@code DataInput}.
*
* Bytes
* for this operation are read from the contained
* input stream.
*
* @return the next two bytes of this input stream, interpreted as an
* unsigned 16-bit integer.
* @see SafeFilterInputStream#in
*/
@Override
public final int readUnsignedShort() {
int ch1 = in.read();
int ch2 = in.read();
if ((ch1 | ch2) < 0)
throw new IndexOutOfBoundsException();
return (ch1 << 8) + (ch2);
}
/**
* See the general contract of the {@code readChar}
* method of {@code DataInput}.
*
* Bytes
* for this operation are read from the contained
* input stream.
*
* @return the next two bytes of this input stream, interpreted as a
* {@code char}.
* @see SafeFilterInputStream#in
*/
@Override
public final char readChar() {
int ch1 = in.read();
int ch2 = in.read();
if ((ch1 | ch2) < 0)
throw new IndexOutOfBoundsException();
return (char)((ch1 << 8) + (ch2));
}
/**
* See the general contract of the {@code readInt}
* method of {@code DataInput}.
*
* Bytes
* for this operation are read from the contained
* input stream.
*
* @return the next four bytes of this input stream, interpreted as an
* {@code int}.
* @see SafeFilterInputStream#in
*/
@Override
public final int readInt() {
int ch1 = in.read();
int ch2 = in.read();
int ch3 = in.read();
int ch4 = in.read();
if ((ch1 | ch2 | ch3 | ch4) < 0)
throw new IndexOutOfBoundsException();
return ((ch1 << 24) + (ch2 << 16) + (ch3 << 8) + (ch4));
}
private final byte[] readBuffer = new byte[8];
/**
* See the general contract of the {@code readLong}
* method of {@code DataInput}.
*
* Bytes
* for this operation are read from the contained
* input stream.
*
* @return the next eight bytes of this input stream, interpreted as a
* {@code long}.
* @see SafeFilterInputStream#in
*/
@Override
public final long readLong() {
readFully(readBuffer, 0, 8);
return (((long)readBuffer[0] << 56) +
((long)(readBuffer[1] & 255) << 48) +
((long)(readBuffer[2] & 255) << 40) +
((long)(readBuffer[3] & 255) << 32) +
((long)(readBuffer[4] & 255) << 24) +
((readBuffer[5] & 255) << 16) +
((readBuffer[6] & 255) << 8) +
((readBuffer[7] & 255)));
}
@Override
public final long readInt52() {
readFully(readBuffer, 0, 7);
return ((long) (readBuffer[0] & 0xf) << 48)
+ ((long) (readBuffer[1] & 0xff) << 40)
+ ((long) (readBuffer[2] & 0xff) << 32)
+ ((long) (readBuffer[3] & 0xff) << 24)
+ ((long) (readBuffer[4] & 0xff) << 16)
+ ((long) (readBuffer[5] & 0xff) << 8)
+ (long) (readBuffer[6] & 0xff);
}
/**
* See the general contract of the {@code readFloat}
* method of {@code DataInput}.
*
* Bytes
* for this operation are read from the contained
* input stream.
*
* @return the next four bytes of this input stream, interpreted as a
* {@code float}.
* @see SafeDataInputStream#readInt()
* @see Float#intBitsToFloat(int)
*/
@Override
public final float readFloat() {
return Float.intBitsToFloat(readInt());
}
/**
* See the general contract of the {@code readDouble}
* method of {@code DataInput}.
*
* Bytes
* for this operation are read from the contained
* input stream.
*
* @return the next eight bytes of this input stream, interpreted as a
* {@code double}.
* @see SafeDataInputStream#readLong()
* @see Double#longBitsToDouble(long)
*/
@Override
public final double readDouble() {
return Double.longBitsToDouble(readLong());
}
@IgnoreCoverage
@Override
@Deprecated
public final String readLine() {
throw new UnsupportedOperationException();
}
/**
* See the general contract of the {@code readUTF}
* method of {@code DataInput}.
*
* Bytes
* for this operation are read from the contained
* input stream.
*
* @return a Unicode string.
*/
@Deprecated
@IgnoreCoverage
@Override
public @NotNull String readUTF() {
return readShortText(StandardCharsets.UTF_8);
}
@IgnoreCoverage
@Override
public @NotNull String readShortText(Charset charset) {
int utfLength = this.readUnsignedShort();
return in.readString(utfLength, charset);
}
@IgnoreCoverage
@Override
public @NotNull String readMediumText(Charset charset) {
int utfLength = this.readInt();
return in.readString(utfLength, charset);
}
}