test/jdk/java/io/InputStream/Skip.java

/*
 * Copyright (c) 1997, 2018, Oracle and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */

/**
 * @test
 * @bug 4016710 6516099
 * @summary check for correct implementation of InputStream.skip{NBytes}
 */

import java.io.EOFException;
import java.io.InputStream;
import java.io.IOException;

public class Skip {
    private static final int EOF = -1;

    private static void dotest(InputStream in, int curpos, long total,
                               long toskip, long expected) throws Exception {
        try {
            System.err.println("\n\nCurrently at pos = " + curpos +
                               "\nTotal bytes in the Stream = " + total +
                               "\nNumber of bytes to skip = " + toskip +
                               "\nNumber of bytes that should be skipped = " +
                               expected);

            long skipped = in.skip(toskip);

            System.err.println("actual number skipped: "+ skipped);

            if ((skipped < 0) || (skipped > expected)) {
                throw new RuntimeException("Unexpected byte count skipped");
            }
        } catch (IOException e) {
            System.err.println("IOException is thrown: " + e);
        } catch (Throwable e) {
            throw new RuntimeException("Unexpected " + e + " is thrown!");
        }
    }

    private static void dotestExact(MyInputStream in, long curpos, long total,
        long toskip, boolean expectIOE, boolean expectEOFE) {

        System.err.println("\n\nCurrently at pos = " + curpos +
                           "\nTotal bytes in the Stream = " + total +
                           "\nNumber of bytes to skip = " + toskip);

        try {
            long pos = in.position();
            assert pos == curpos : pos + " != " + curpos;
            in.skipNBytes(toskip);
            if (in.position() != pos + (toskip < 0 ? 0 : toskip)) {
                throw new RuntimeException((in.position() - pos) +
                    " bytes skipped; expected " + toskip);
            }
        } catch (EOFException eofe) {
            if (!expectEOFE) {
                throw new RuntimeException("Unexpected EOFException", eofe);
            }
            System.err.println("Caught expected EOFException");
        } catch (IOException ioe) {
            if (!expectIOE) {
                throw new RuntimeException("Unexpected IOException", ioe);
            }
            System.err.println("Caught expected IOException");
        }
    }

    public static void main( String argv[] ) throws Exception {
        MyInputStream in = new MyInputStream(11);

        // test for negative skip
        dotest(in,  0, 11, -23,  0);

        // check for skip beyond EOF starting from before EOF
        dotest(in,  0, 11,  20, 11);

        // check for skip after EOF
        dotest(in, EOF, 11,  20,  0);

        in = new MyInputStream(9000);

        // check for skip equal to the read chunk size in InputStream.java
        dotest(in,  0, 9000, 2048, 2048);

        // check for skip larger than the read chunk size in InputStream.java
        dotest(in, 2048, 9000, 5000, 5000);

        // check for skip beyond EOF starting from before EOF
        dotest(in, 7048, 9000, 5000, 1952);

        in = new MyInputStream(5000);

        // check for multiple chunk reads
        dotest(in, 0, 5000, 6000, 5000);

        /*
         * check for skip larger than Integer.MAX_VALUE
         * (Takes about 2 hrs on a sparc ultra-1)
         * long total = (long)Integer.MAX_VALUE + (long)10;
         * long toskip = total - (long)6;
         * in = new MyInputStream(total);
         * dotest(in, 0, total, toskip, toskip);
         */

        // tests for skipping an exact number of bytes

        final long streamLength = Long.MAX_VALUE;
        in = new MyInputStream(streamLength);

        // negative skip: OK
        dotestExact(in, 0, streamLength, -1, false, false);

        // negative skip at EOF: OK
        in.position(streamLength);
        dotestExact(in, streamLength, streamLength, -1, false, false);
        in.position(0);

        // zero skip: OK
        dotestExact(in, 0, streamLength, 0, false, false);

        // zero skip at EOF: OK
        in.position(streamLength);
        dotestExact(in, streamLength, streamLength, 0, false, false);

        // skip(1) at EOF: EOFE
        dotestExact(in, streamLength, streamLength, 1, false, true);
        in.position(0);

        final long n = 31; // skip count
        long pos = 0;

        // skip(n) returns negative value: IOE
        in.setState(-1, 100);
        dotestExact(in, pos, streamLength, n, true, false);

        // skip(n) returns n + 1: IOE
        in.setState(n + 1, 100);
        dotestExact(in, pos, streamLength, n, true, false);
        pos += n + 1;

        // skip(n) returns n/2 but only n/4 subsequent reads succeed: EOFE
        in.setState(n/2, n/2 + n/4);
        dotestExact(in, pos, streamLength, n, false, true);
        pos += n/2 + n/4;

        // skip(n) returns n/2 but n - n/2 subsequent reads succeed: OK
        in.setState(n/2, n);
        dotestExact(in, pos, streamLength, n, false, false);
        pos += n;
    }
}

class MyInputStream extends InputStream {
    private static final int EOF = -1;

    private final long endoffile;

    private long readctr = 0;

    private boolean isStateSet = false;
    private long skipReturn;
    private long readLimit;

    public MyInputStream(long endoffile) {
        this.endoffile = endoffile;
    }

    /**
     * Limits the behavior of skip() and read().
     *
     * @param skipReturn the value to be returned by skip()
     * @param maxReads   the maximum number of reads past the current position
     *                   before EOF is reached
     */
    public void setState(long skipReturn, long maxReads) {
        this.skipReturn = skipReturn;
        this.readLimit = readctr + maxReads;
        isStateSet = true;
    }

    public int read() {
        if (readctr == endoffile ||
            (isStateSet && readctr >= readLimit)) {
            return EOF;
        }
        else {
            readctr++;
            return 0;
        }
    }

    public int available() { return 0; }

    public long position() { return readctr; }

    public void position(long pos) {
        readctr = pos < 0 ? 0 : Math.min(pos, endoffile);
    }

    public long skip(long n) throws IOException {
        if (isStateSet) {
            return skipReturn < 0 ? skipReturn : super.skip(skipReturn);
        }

        // InputStream skip implementation.
        return super.skip(n); // readctr is implicitly incremented
    }
}