src/java.base/share/classes/sun/nio/ch/ChannelInputStream.java
author igerasim
Wed, 21 Aug 2019 13:49:16 -0700
changeset 57826 bf4c808a4488
parent 57595 d629735db937
child 58679 9c3209ff7550
permissions -rw-r--r--
8211360: Change #if DEF to #if defined(DEF) Reviewed-by: bpb, vtewari
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
     1
/*
55578
8c19519114e7 8227080: (fs) Files.newInputStream(...).skip(n) is slow
bpb
parents: 47216
diff changeset
     2
 * Copyright (c) 2001, 2019, Oracle and/or its affiliates. All rights reserved.
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
     3
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
90ce3da70b43 Initial load
duke
parents:
diff changeset
     4
 *
90ce3da70b43 Initial load
duke
parents:
diff changeset
     5
 * This code is free software; you can redistribute it and/or modify it
90ce3da70b43 Initial load
duke
parents:
diff changeset
     6
 * under the terms of the GNU General Public License version 2 only, as
5506
202f599c92aa 6943119: Rebrand source copyright notices
ohair
parents: 3955
diff changeset
     7
 * published by the Free Software Foundation.  Oracle designates this
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
     8
 * particular file as subject to the "Classpath" exception as provided
5506
202f599c92aa 6943119: Rebrand source copyright notices
ohair
parents: 3955
diff changeset
     9
 * by Oracle in the LICENSE file that accompanied this code.
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
    10
 *
90ce3da70b43 Initial load
duke
parents:
diff changeset
    11
 * This code is distributed in the hope that it will be useful, but WITHOUT
90ce3da70b43 Initial load
duke
parents:
diff changeset
    12
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
90ce3da70b43 Initial load
duke
parents:
diff changeset
    13
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
90ce3da70b43 Initial load
duke
parents:
diff changeset
    14
 * version 2 for more details (a copy is included in the LICENSE file that
90ce3da70b43 Initial load
duke
parents:
diff changeset
    15
 * accompanied this code).
90ce3da70b43 Initial load
duke
parents:
diff changeset
    16
 *
90ce3da70b43 Initial load
duke
parents:
diff changeset
    17
 * You should have received a copy of the GNU General Public License version
90ce3da70b43 Initial load
duke
parents:
diff changeset
    18
 * 2 along with this work; if not, write to the Free Software Foundation,
90ce3da70b43 Initial load
duke
parents:
diff changeset
    19
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
90ce3da70b43 Initial load
duke
parents:
diff changeset
    20
 *
5506
202f599c92aa 6943119: Rebrand source copyright notices
ohair
parents: 3955
diff changeset
    21
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
202f599c92aa 6943119: Rebrand source copyright notices
ohair
parents: 3955
diff changeset
    22
 * or visit www.oracle.com if you need additional information or have any
202f599c92aa 6943119: Rebrand source copyright notices
ohair
parents: 3955
diff changeset
    23
 * questions.
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
    24
 */
90ce3da70b43 Initial load
duke
parents:
diff changeset
    25
90ce3da70b43 Initial load
duke
parents:
diff changeset
    26
package sun.nio.ch;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    27
90ce3da70b43 Initial load
duke
parents:
diff changeset
    28
import java.io.*;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    29
import java.nio.*;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    30
import java.nio.channels.*;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    31
import java.nio.channels.spi.*;
55578
8c19519114e7 8227080: (fs) Files.newInputStream(...).skip(n) is slow
bpb
parents: 47216
diff changeset
    32
import java.util.Objects;
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
    33
90ce3da70b43 Initial load
duke
parents:
diff changeset
    34
/**
90ce3da70b43 Initial load
duke
parents:
diff changeset
    35
 * This class is defined here rather than in java.nio.channels.Channels
90ce3da70b43 Initial load
duke
parents:
diff changeset
    36
 * so that code can be shared with SocketAdaptor.
90ce3da70b43 Initial load
duke
parents:
diff changeset
    37
 *
90ce3da70b43 Initial load
duke
parents:
diff changeset
    38
 * @author Mike McCloskey
90ce3da70b43 Initial load
duke
parents:
diff changeset
    39
 * @author Mark Reinhold
90ce3da70b43 Initial load
duke
parents:
diff changeset
    40
 * @since 1.4
90ce3da70b43 Initial load
duke
parents:
diff changeset
    41
 */
90ce3da70b43 Initial load
duke
parents:
diff changeset
    42
90ce3da70b43 Initial load
duke
parents:
diff changeset
    43
public class ChannelInputStream
90ce3da70b43 Initial load
duke
parents:
diff changeset
    44
    extends InputStream
90ce3da70b43 Initial load
duke
parents:
diff changeset
    45
{
90ce3da70b43 Initial load
duke
parents:
diff changeset
    46
90ce3da70b43 Initial load
duke
parents:
diff changeset
    47
    public static int read(ReadableByteChannel ch, ByteBuffer bb,
90ce3da70b43 Initial load
duke
parents:
diff changeset
    48
                           boolean block)
90ce3da70b43 Initial load
duke
parents:
diff changeset
    49
        throws IOException
90ce3da70b43 Initial load
duke
parents:
diff changeset
    50
    {
90ce3da70b43 Initial load
duke
parents:
diff changeset
    51
        if (ch instanceof SelectableChannel) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
    52
            SelectableChannel sc = (SelectableChannel)ch;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    53
            synchronized (sc.blockingLock()) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
    54
                boolean bm = sc.isBlocking();
90ce3da70b43 Initial load
duke
parents:
diff changeset
    55
                if (!bm)
90ce3da70b43 Initial load
duke
parents:
diff changeset
    56
                    throw new IllegalBlockingModeException();
90ce3da70b43 Initial load
duke
parents:
diff changeset
    57
                if (bm != block)
90ce3da70b43 Initial load
duke
parents:
diff changeset
    58
                    sc.configureBlocking(block);
90ce3da70b43 Initial load
duke
parents:
diff changeset
    59
                int n = ch.read(bb);
90ce3da70b43 Initial load
duke
parents:
diff changeset
    60
                if (bm != block)
90ce3da70b43 Initial load
duke
parents:
diff changeset
    61
                    sc.configureBlocking(bm);
90ce3da70b43 Initial load
duke
parents:
diff changeset
    62
                return n;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    63
            }
90ce3da70b43 Initial load
duke
parents:
diff changeset
    64
        } else {
90ce3da70b43 Initial load
duke
parents:
diff changeset
    65
            return ch.read(bb);
90ce3da70b43 Initial load
duke
parents:
diff changeset
    66
        }
90ce3da70b43 Initial load
duke
parents:
diff changeset
    67
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
    68
90ce3da70b43 Initial load
duke
parents:
diff changeset
    69
    protected final ReadableByteChannel ch;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    70
    private ByteBuffer bb = null;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    71
    private byte[] bs = null;           // Invoker's previous array
90ce3da70b43 Initial load
duke
parents:
diff changeset
    72
    private byte[] b1 = null;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    73
90ce3da70b43 Initial load
duke
parents:
diff changeset
    74
    public ChannelInputStream(ReadableByteChannel ch) {
90ce3da70b43 Initial load
duke
parents:
diff changeset
    75
        this.ch = ch;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    76
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
    77
90ce3da70b43 Initial load
duke
parents:
diff changeset
    78
    public synchronized int read() throws IOException {
90ce3da70b43 Initial load
duke
parents:
diff changeset
    79
        if (b1 == null)
90ce3da70b43 Initial load
duke
parents:
diff changeset
    80
            b1 = new byte[1];
90ce3da70b43 Initial load
duke
parents:
diff changeset
    81
        int n = this.read(b1);
90ce3da70b43 Initial load
duke
parents:
diff changeset
    82
        if (n == 1)
90ce3da70b43 Initial load
duke
parents:
diff changeset
    83
            return b1[0] & 0xff;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    84
        return -1;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    85
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
    86
90ce3da70b43 Initial load
duke
parents:
diff changeset
    87
    public synchronized int read(byte[] bs, int off, int len)
90ce3da70b43 Initial load
duke
parents:
diff changeset
    88
        throws IOException
90ce3da70b43 Initial load
duke
parents:
diff changeset
    89
    {
55578
8c19519114e7 8227080: (fs) Files.newInputStream(...).skip(n) is slow
bpb
parents: 47216
diff changeset
    90
        Objects.checkFromIndexSize(off, len, bs.length);
8c19519114e7 8227080: (fs) Files.newInputStream(...).skip(n) is slow
bpb
parents: 47216
diff changeset
    91
        if (len == 0)
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
    92
            return 0;
90ce3da70b43 Initial load
duke
parents:
diff changeset
    93
90ce3da70b43 Initial load
duke
parents:
diff changeset
    94
        ByteBuffer bb = ((this.bs == bs)
90ce3da70b43 Initial load
duke
parents:
diff changeset
    95
                         ? this.bb
90ce3da70b43 Initial load
duke
parents:
diff changeset
    96
                         : ByteBuffer.wrap(bs));
90ce3da70b43 Initial load
duke
parents:
diff changeset
    97
        bb.limit(Math.min(off + len, bb.capacity()));
90ce3da70b43 Initial load
duke
parents:
diff changeset
    98
        bb.position(off);
90ce3da70b43 Initial load
duke
parents:
diff changeset
    99
        this.bb = bb;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   100
        this.bs = bs;
90ce3da70b43 Initial load
duke
parents:
diff changeset
   101
        return read(bb);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   102
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   103
90ce3da70b43 Initial load
duke
parents:
diff changeset
   104
    protected int read(ByteBuffer bb)
90ce3da70b43 Initial load
duke
parents:
diff changeset
   105
        throws IOException
90ce3da70b43 Initial load
duke
parents:
diff changeset
   106
    {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   107
        return ChannelInputStream.read(ch, bb, true);
90ce3da70b43 Initial load
duke
parents:
diff changeset
   108
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   109
3955
a585e5ca5e4a 6884800: (file) Path.newInputStream does not usefully implement available()
alanb
parents: 2
diff changeset
   110
    public int available() throws IOException {
a585e5ca5e4a 6884800: (file) Path.newInputStream does not usefully implement available()
alanb
parents: 2
diff changeset
   111
        // special case where the channel is to a file
a585e5ca5e4a 6884800: (file) Path.newInputStream does not usefully implement available()
alanb
parents: 2
diff changeset
   112
        if (ch instanceof SeekableByteChannel) {
a585e5ca5e4a 6884800: (file) Path.newInputStream does not usefully implement available()
alanb
parents: 2
diff changeset
   113
            SeekableByteChannel sbc = (SeekableByteChannel)ch;
a585e5ca5e4a 6884800: (file) Path.newInputStream does not usefully implement available()
alanb
parents: 2
diff changeset
   114
            long rem = Math.max(0, sbc.size() - sbc.position());
a585e5ca5e4a 6884800: (file) Path.newInputStream does not usefully implement available()
alanb
parents: 2
diff changeset
   115
            return (rem > Integer.MAX_VALUE) ? Integer.MAX_VALUE : (int)rem;
a585e5ca5e4a 6884800: (file) Path.newInputStream does not usefully implement available()
alanb
parents: 2
diff changeset
   116
        }
a585e5ca5e4a 6884800: (file) Path.newInputStream does not usefully implement available()
alanb
parents: 2
diff changeset
   117
        return 0;
a585e5ca5e4a 6884800: (file) Path.newInputStream does not usefully implement available()
alanb
parents: 2
diff changeset
   118
    }
a585e5ca5e4a 6884800: (file) Path.newInputStream does not usefully implement available()
alanb
parents: 2
diff changeset
   119
55578
8c19519114e7 8227080: (fs) Files.newInputStream(...).skip(n) is slow
bpb
parents: 47216
diff changeset
   120
    public synchronized long skip(long n) throws IOException {
8c19519114e7 8227080: (fs) Files.newInputStream(...).skip(n) is slow
bpb
parents: 47216
diff changeset
   121
        // special case where the channel is to a file
57595
d629735db937 8227609: (fs) Files.newInputStream(...).skip(n) should allow skipping beyond file size
bpb
parents: 55578
diff changeset
   122
        if (ch instanceof SeekableByteChannel) {
55578
8c19519114e7 8227080: (fs) Files.newInputStream(...).skip(n) is slow
bpb
parents: 47216
diff changeset
   123
            SeekableByteChannel sbc = (SeekableByteChannel)ch;
57595
d629735db937 8227609: (fs) Files.newInputStream(...).skip(n) should allow skipping beyond file size
bpb
parents: 55578
diff changeset
   124
            long pos = sbc.position();
d629735db937 8227609: (fs) Files.newInputStream(...).skip(n) should allow skipping beyond file size
bpb
parents: 55578
diff changeset
   125
            long newPos;
d629735db937 8227609: (fs) Files.newInputStream(...).skip(n) should allow skipping beyond file size
bpb
parents: 55578
diff changeset
   126
            if (n > 0) {
d629735db937 8227609: (fs) Files.newInputStream(...).skip(n) should allow skipping beyond file size
bpb
parents: 55578
diff changeset
   127
                newPos = pos + n;
55578
8c19519114e7 8227080: (fs) Files.newInputStream(...).skip(n) is slow
bpb
parents: 47216
diff changeset
   128
                long size = sbc.size();
57595
d629735db937 8227609: (fs) Files.newInputStream(...).skip(n) should allow skipping beyond file size
bpb
parents: 55578
diff changeset
   129
                if (newPos < 0 || newPos > size) {
d629735db937 8227609: (fs) Files.newInputStream(...).skip(n) should allow skipping beyond file size
bpb
parents: 55578
diff changeset
   130
                    newPos = size;
55578
8c19519114e7 8227080: (fs) Files.newInputStream(...).skip(n) is slow
bpb
parents: 47216
diff changeset
   131
                }
57595
d629735db937 8227609: (fs) Files.newInputStream(...).skip(n) should allow skipping beyond file size
bpb
parents: 55578
diff changeset
   132
            } else {
d629735db937 8227609: (fs) Files.newInputStream(...).skip(n) should allow skipping beyond file size
bpb
parents: 55578
diff changeset
   133
                newPos = Long.max(pos + n, 0);
55578
8c19519114e7 8227080: (fs) Files.newInputStream(...).skip(n) is slow
bpb
parents: 47216
diff changeset
   134
            }
57595
d629735db937 8227609: (fs) Files.newInputStream(...).skip(n) should allow skipping beyond file size
bpb
parents: 55578
diff changeset
   135
            sbc.position(newPos);
d629735db937 8227609: (fs) Files.newInputStream(...).skip(n) should allow skipping beyond file size
bpb
parents: 55578
diff changeset
   136
            return newPos - pos;
55578
8c19519114e7 8227080: (fs) Files.newInputStream(...).skip(n) is slow
bpb
parents: 47216
diff changeset
   137
        }
8c19519114e7 8227080: (fs) Files.newInputStream(...).skip(n) is slow
bpb
parents: 47216
diff changeset
   138
        return super.skip(n);
8c19519114e7 8227080: (fs) Files.newInputStream(...).skip(n) is slow
bpb
parents: 47216
diff changeset
   139
    }
8c19519114e7 8227080: (fs) Files.newInputStream(...).skip(n) is slow
bpb
parents: 47216
diff changeset
   140
2
90ce3da70b43 Initial load
duke
parents:
diff changeset
   141
    public void close() throws IOException {
90ce3da70b43 Initial load
duke
parents:
diff changeset
   142
        ch.close();
90ce3da70b43 Initial load
duke
parents:
diff changeset
   143
    }
90ce3da70b43 Initial load
duke
parents:
diff changeset
   144
90ce3da70b43 Initial load
duke
parents:
diff changeset
   145
}