Showing the changes to java/pkg/sun/rmi/transport/proxy/HttpInputStream.java (1 of 18352 files). Show all
Unified · Split
| @@ -0,0 +1,201 @@ | ||
| 1 | +/* | |
| 2 | + * Copyright (c) 1996, 2001, Oracle and/or its affiliates. All rights reserved. | |
| 3 | + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. | |
| 4 | + * | |
| 5 | + * This code is free software; you can redistribute it and/or modify it | |
| 6 | + * under the terms of the GNU General Public License version 2 only, as | |
| 7 | + * published by the Free Software Foundation. Oracle designates this | |
| 8 | + * particular file as subject to the "Classpath" exception as provided | |
| 9 | + * by Oracle in the LICENSE file that accompanied this code. | |
| 10 | + * | |
| 11 | + * This code is distributed in the hope that it will be useful, but WITHOUT | |
| 12 | + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or | |
| 13 | + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License | |
| 14 | + * version 2 for more details (a copy is included in the LICENSE file that | |
| 15 | + * accompanied this code). | |
| 16 | + * | |
| 17 | + * You should have received a copy of the GNU General Public License version | |
| 18 | + * 2 along with this work; if not, write to the Free Software Foundation, | |
| 19 | + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. | |
| 20 | + * | |
| 21 | + * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA | |
| 22 | + * or visit www.oracle.com if you need additional information or have any | |
| 23 | + * questions. | |
| 24 | + */ | |
| 25 | +package sun.rmi.transport.proxy; | |
| 26 | + | |
| 27 | +import java.io.*; | |
| 28 | + | |
| 29 | +import sun.rmi.runtime.Log; | |
| 30 | + | |
| 31 | +/** | |
| 32 | + * The HttpInputStream class assists the HttpSendSocket and HttpReceiveSocket | |
| 33 | + * classes by filtering out the header for the message as well as any | |
| 34 | + * data after its proper content length. | |
| 35 | + */ | |
| 36 | +class HttpInputStream extends FilterInputStream { | |
| 37 | + | |
| 38 | + /** bytes remaining to be read from proper content of message */ | |
| 39 | + protected int bytesLeft; | |
| 40 | + | |
| 41 | + /** bytes remaining to be read at time of last mark */ | |
| 42 | + protected int bytesLeftAtMark; | |
| 43 | + | |
| 44 | + /** | |
| 45 | + * Create new filter on a given input stream. | |
| 46 | + * @param in the InputStream to filter from | |
| 47 | + */ | |
| 48 | + public HttpInputStream(InputStream in) throws IOException | |
| 49 | + { | |
| 50 | + super(in); | |
| 51 | + | |
| 52 | + if (in.markSupported()) | |
| 53 | + in.mark(0); // prevent resetting back to old marks | |
| 54 | + | |
| 55 | + // pull out header, looking for content length | |
| 56 | + | |
| 57 | + DataInputStream dis = new DataInputStream(in); | |
| 58 | + String key = "Content-length:".toLowerCase(); | |
| 59 | + boolean contentLengthFound = false; | |
| 60 | + String line; | |
| 61 | + do { | |
| 62 | + line = dis.readLine(); | |
| 63 | + | |
| 64 | + if (RMIMasterSocketFactory.proxyLog.isLoggable(Log.VERBOSE)) { | |
| 65 | + RMIMasterSocketFactory.proxyLog.log(Log.VERBOSE, | |
| 66 | + "received header line: \"" + line + "\""); | |
| 67 | + } | |
| 68 | + | |
| 69 | + if (line == null) | |
| 70 | + throw new EOFException(); | |
| 71 | + | |
| 72 | + if (line.toLowerCase().startsWith(key)) { | |
| 73 | + if (contentLengthFound) | |
| 74 | + ; // what would we want to do in this case?? | |
| 75 | + bytesLeft = | |
| 76 | + Integer.parseInt(line.substring(key.length()).trim()); | |
| 77 | + contentLengthFound = true; | |
| 78 | + } | |
| 79 | + | |
| 80 | + // The idea here is to go past the first blank line. | |
| 81 | + // Some DataInputStream.readLine() documentation specifies that | |
| 82 | + // it does include the line-terminating character(s) in the | |
| 83 | + // returned string, but it actually doesn't, so we'll cover | |
| 84 | + // all cases here... | |
| 85 | + } while ((line.length() != 0) && | |
| 86 | + (line.charAt(0) != '\r') && (line.charAt(0) != '\n')); | |
| 87 | + | |
| 88 | + if (!contentLengthFound || bytesLeft < 0) { | |
| 89 | + // This really shouldn't happen, but if it does, shoud we fail?? | |
| 90 | + // For now, just give up and let a whole lot of bytes through... | |
| 91 | + bytesLeft = Integer.MAX_VALUE; | |
| 92 | + } | |
| 93 | + bytesLeftAtMark = bytesLeft; | |
| 94 | + | |
| 95 | + if (RMIMasterSocketFactory.proxyLog.isLoggable(Log.VERBOSE)) { | |
| 96 | + RMIMasterSocketFactory.proxyLog.log(Log.VERBOSE, | |
| 97 | + "content length: " + bytesLeft); | |
| 98 | + } | |
| 99 | + } | |
| 100 | + | |
| 101 | + /** | |
| 102 | + * Returns the number of bytes that can be read with blocking. | |
| 103 | + * Make sure that this does not exceed the number of bytes remaining | |
| 104 | + * in the proper content of the message. | |
| 105 | + */ | |
| 106 | + public int available() throws IOException | |
| 107 | + { | |
| 108 | + int bytesAvailable = in.available(); | |
| 109 | + if (bytesAvailable > bytesLeft) | |
| 110 | + bytesAvailable = bytesLeft; | |
| 111 | + | |
| 112 | + return bytesAvailable; | |
| 113 | + } | |
| 114 | + | |
| 115 | + /** | |
| 116 | + * Read a byte of data from the stream. Make sure that one is available | |
| 117 | + * from the proper content of the message, else -1 is returned to | |
| 118 | + * indicate to the user that the end of the stream has been reached. | |
| 119 | + */ | |
| 120 | + public int read() throws IOException | |
| 121 | + { | |
| 122 | + if (bytesLeft > 0) { | |
| 123 | + int data = in.read(); | |
| 124 | + if (data != -1) | |
| 125 | + -- bytesLeft; | |
| 126 | + | |
| 127 | + if (RMIMasterSocketFactory.proxyLog.isLoggable(Log.VERBOSE)) { | |
| 128 | + RMIMasterSocketFactory.proxyLog.log(Log.VERBOSE, | |
| 129 | + "received byte: '" + | |
| 130 | + ((data & 0x7F) < ' ' ? " " : String.valueOf((char) data)) + | |
| 131 | + "' " + data); | |
| 132 | + } | |
| 133 | + | |
| 134 | + return data; | |
| 135 | + } | |
| 136 | + else { | |
| 137 | + RMIMasterSocketFactory.proxyLog.log(Log.VERBOSE, | |
| 138 | + "read past content length"); | |
| 139 | + | |
| 140 | + return -1; | |
| 141 | + } | |
| 142 | + } | |
| 143 | + | |
| 144 | + public int read(byte b[], int off, int len) throws IOException | |
| 145 | + { | |
| 146 | + if (bytesLeft == 0 && len > 0) { | |
| 147 | + RMIMasterSocketFactory.proxyLog.log(Log.VERBOSE, | |
| 148 | + "read past content length"); | |
| 149 | + | |
| 150 | + return -1; | |
| 151 | + } | |
| 152 | + if (len > bytesLeft) | |
| 153 | + len = bytesLeft; | |
| 154 | + int bytesRead = in.read(b, off, len); | |
| 155 | + bytesLeft -= bytesRead; | |
| 156 | + | |
| 157 | + if (RMIMasterSocketFactory.proxyLog.isLoggable(Log.VERBOSE)) { | |
| 158 | + RMIMasterSocketFactory.proxyLog.log(Log.VERBOSE, | |
| 159 | + "read " + bytesRead + " bytes, " + bytesLeft + " remaining"); | |
| 160 | + } | |
| 161 | + | |
| 162 | + return bytesRead; | |
| 163 | + } | |
| 164 | + | |
| 165 | + /** | |
| 166 | + * Mark the current position in the stream (for future calls to reset). | |
| 167 | + * Remember where we are within the proper content of the message, so | |
| 168 | + * that a reset method call can recreate our state properly. | |
| 169 | + * @param readlimit how many bytes can be read before mark becomes invalid | |
| 170 | + */ | |
| 171 | + public void mark(int readlimit) | |
| 172 | + { | |
| 173 | + in.mark(readlimit); | |
| 174 | + if (in.markSupported()) | |
| 175 | + bytesLeftAtMark = bytesLeft; | |
| 176 | + } | |
| 177 | + | |
| 178 | + /** | |
| 179 | + * Repositions the stream to the last marked position. Make sure to | |
| 180 | + * adjust our position within the proper content accordingly. | |
| 181 | + */ | |
| 182 | + public void reset() throws IOException | |
| 183 | + { | |
| 184 | + in.reset(); | |
| 185 | + bytesLeft = bytesLeftAtMark; | |
| 186 | + } | |
| 187 | + | |
| 188 | + /** | |
| 189 | + * Skips bytes of the stream. Make sure to adjust our | |
| 190 | + * position within the proper content accordingly. | |
| 191 | + * @param n number of bytes to be skipped | |
| 192 | + */ | |
| 193 | + public long skip(long n) throws IOException | |
| 194 | + { | |
| 195 | + if (n > bytesLeft) | |
| 196 | + n = bytesLeft; | |
| 197 | + long bytesSkipped = in.skip(n); | |
| 198 | + bytesLeft -= bytesSkipped; | |
| 199 | + return bytesSkipped; | |
| 200 | + } | |
| 201 | +} | |