Back to project page SocketPlugin.
The source code is released under:
MIT License
If you think the Android project SocketPlugin listed in this page is inappropriate, such as containing malicious code/tools or violating the copyright, please email info at java2s dot com, thanks.
/* * Copyright (C) 2011 The Android Open Source Project *//from w w w. jav a 2 s . c o m * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package com.squareup.okhttp.internal.spdy; import com.squareup.okhttp.internal.Util; import java.io.Closeable; import java.io.DataInputStream; import java.io.IOException; import java.io.InputStream; import java.io.UnsupportedEncodingException; import java.net.ProtocolException; import java.util.ArrayList; import java.util.List; import java.util.logging.Logger; import java.util.zip.DataFormatException; import java.util.zip.Inflater; import java.util.zip.InflaterInputStream; /** Read spdy/3 frames. */ final class SpdyReader implements Closeable { static final byte[] DICTIONARY; static { try { DICTIONARY = ("\u0000\u0000\u0000\u0007options\u0000\u0000\u0000\u0004hea" + "d\u0000\u0000\u0000\u0004post\u0000\u0000\u0000\u0003put\u0000\u0000\u0000\u0006dele" + "te\u0000\u0000\u0000\u0005trace\u0000\u0000\u0000\u0006accept\u0000\u0000\u0000" + "\u000Eaccept-charset\u0000\u0000\u0000\u000Faccept-encoding\u0000\u0000\u0000\u000Fa" + "ccept-language\u0000\u0000\u0000\raccept-ranges\u0000\u0000\u0000\u0003age\u0000" + "\u0000\u0000\u0005allow\u0000\u0000\u0000\rauthorization\u0000\u0000\u0000\rcache-co" + "ntrol\u0000\u0000\u0000\nconnection\u0000\u0000\u0000\fcontent-base\u0000\u0000" + "\u0000\u0010content-encoding\u0000\u0000\u0000\u0010content-language\u0000\u0000" + "\u0000\u000Econtent-length\u0000\u0000\u0000\u0010content-location\u0000\u0000\u0000" + "\u000Bcontent-md5\u0000\u0000\u0000\rcontent-range\u0000\u0000\u0000\fcontent-type" + "\u0000\u0000\u0000\u0004date\u0000\u0000\u0000\u0004etag\u0000\u0000\u0000\u0006expe" + "ct\u0000\u0000\u0000\u0007expires\u0000\u0000\u0000\u0004from\u0000\u0000\u0000" + "\u0004host\u0000\u0000\u0000\bif-match\u0000\u0000\u0000\u0011if-modified-since" + "\u0000\u0000\u0000\rif-none-match\u0000\u0000\u0000\bif-range\u0000\u0000\u0000" + "\u0013if-unmodified-since\u0000\u0000\u0000\rlast-modified\u0000\u0000\u0000\blocati" + "on\u0000\u0000\u0000\fmax-forwards\u0000\u0000\u0000\u0006pragma\u0000\u0000\u0000" + "\u0012proxy-authenticate\u0000\u0000\u0000\u0013proxy-authorization\u0000\u0000" + "\u0000\u0005range\u0000\u0000\u0000\u0007referer\u0000\u0000\u0000\u000Bretry-after" + "\u0000\u0000\u0000\u0006server\u0000\u0000\u0000\u0002te\u0000\u0000\u0000\u0007trai" + "ler\u0000\u0000\u0000\u0011transfer-encoding\u0000\u0000\u0000\u0007upgrade\u0000" + "\u0000\u0000\nuser-agent\u0000\u0000\u0000\u0004vary\u0000\u0000\u0000\u0003via" + "\u0000\u0000\u0000\u0007warning\u0000\u0000\u0000\u0010www-authenticate\u0000\u0000" + "\u0000\u0006method\u0000\u0000\u0000\u0003get\u0000\u0000\u0000\u0006status\u0000" + "\u0000\u0000\u0006200 OK\u0000\u0000\u0000\u0007version\u0000\u0000\u0000\bHTTP/1.1" + "\u0000\u0000\u0000\u0003url\u0000\u0000\u0000\u0006public\u0000\u0000\u0000\nset-coo" + "kie\u0000\u0000\u0000\nkeep-alive\u0000\u0000\u0000\u0006origin100101201202205206300" + "302303304305306307402405406407408409410411412413414415416417502504505203 Non-Authori" + "tative Information204 No Content301 Moved Permanently400 Bad Request401 Unauthorized" + "403 Forbidden404 Not Found500 Internal Server Error501 Not Implemented503 Service Un" + "availableJan Feb Mar Apr May Jun Jul Aug Sept Oct Nov Dec 00:00:00 Mon, Tue, Wed, Th" + "u, Fri, Sat, Sun, GMTchunked,text/html,image/png,image/jpg,image/gif,application/xml" + ",application/xhtml+xml,text/plain,text/javascript,publicprivatemax-age=gzip,deflate," + "sdchcharset=utf-8charset=iso-8859-1,utf-,*,enq=0.").getBytes(Util.UTF_8.name()); } catch (UnsupportedEncodingException e) { throw new AssertionError(); } } private final DataInputStream in; private final DataInputStream nameValueBlockIn; private int compressedLimit; SpdyReader(InputStream in) { this.in = new DataInputStream(in); this.nameValueBlockIn = newNameValueBlockStream(); } /** * Send the next frame to {@code handler}. Returns true unless there are no * more frames on the stream. */ public boolean nextFrame(Handler handler) throws IOException { int w1; try { w1 = in.readInt(); } catch (IOException e) { return false; // This might be a normal socket close. } int w2 = in.readInt(); boolean control = (w1 & 0x80000000) != 0; int flags = (w2 & 0xff000000) >>> 24; int length = (w2 & 0xffffff); if (control) { int version = (w1 & 0x7fff0000) >>> 16; int type = (w1 & 0xffff); if (version != 3) { throw new ProtocolException("version != 3: " + version); } switch (type) { case SpdyConnection.TYPE_SYN_STREAM: readSynStream(handler, flags, length); return true; case SpdyConnection.TYPE_SYN_REPLY: readSynReply(handler, flags, length); return true; case SpdyConnection.TYPE_RST_STREAM: readRstStream(handler, flags, length); return true; case SpdyConnection.TYPE_SETTINGS: readSettings(handler, flags, length); return true; case SpdyConnection.TYPE_NOOP: if (length != 0) throw ioException("TYPE_NOOP length: %d != 0", length); handler.noop(); return true; case SpdyConnection.TYPE_PING: readPing(handler, flags, length); return true; case SpdyConnection.TYPE_GOAWAY: readGoAway(handler, flags, length); return true; case SpdyConnection.TYPE_HEADERS: readHeaders(handler, flags, length); return true; case SpdyConnection.TYPE_WINDOW_UPDATE: readWindowUpdate(handler, flags, length); return true; case SpdyConnection.TYPE_CREDENTIAL: Util.skipByReading(in, length); throw new UnsupportedOperationException("TODO"); // TODO: implement default: throw new IOException("Unexpected frame"); } } else { int streamId = w1 & 0x7fffffff; handler.data(flags, streamId, in, length); return true; } } private void readSynStream(Handler handler, int flags, int length) throws IOException { int w1 = in.readInt(); int w2 = in.readInt(); int s3 = in.readShort(); int streamId = w1 & 0x7fffffff; int associatedStreamId = w2 & 0x7fffffff; int priority = (s3 & 0xe000) >>> 13; int slot = s3 & 0xff; List<String> nameValueBlock = readNameValueBlock(length - 10); handler.synStream(flags, streamId, associatedStreamId, priority, slot, nameValueBlock); } private void readSynReply(Handler handler, int flags, int length) throws IOException { int w1 = in.readInt(); int streamId = w1 & 0x7fffffff; List<String> nameValueBlock = readNameValueBlock(length - 4); handler.synReply(flags, streamId, nameValueBlock); } private void readRstStream(Handler handler, int flags, int length) throws IOException { if (length != 8) throw ioException("TYPE_RST_STREAM length: %d != 8", length); int streamId = in.readInt() & 0x7fffffff; int statusCode = in.readInt(); handler.rstStream(flags, streamId, statusCode); } private void readHeaders(Handler handler, int flags, int length) throws IOException { int w1 = in.readInt(); int streamId = w1 & 0x7fffffff; List<String> nameValueBlock = readNameValueBlock(length - 4); handler.headers(flags, streamId, nameValueBlock); } private void readWindowUpdate(Handler handler, int flags, int length) throws IOException { if (length != 8) throw ioException("TYPE_WINDOW_UPDATE length: %d != 8", length); int w1 = in.readInt(); int w2 = in.readInt(); int streamId = w1 & 0x7fffffff; int deltaWindowSize = w2 & 0x7fffffff; handler.windowUpdate(flags, streamId, deltaWindowSize); } private DataInputStream newNameValueBlockStream() { // Limit the inflater input stream to only those bytes in the Name/Value block. final InputStream throttleStream = new InputStream() { @Override public int read() throws IOException { return Util.readSingleByte(this); } @Override public int read(byte[] buffer, int offset, int byteCount) throws IOException { byteCount = Math.min(byteCount, compressedLimit); int consumed = in.read(buffer, offset, byteCount); compressedLimit -= consumed; return consumed; } @Override public void close() throws IOException { in.close(); } }; // Subclass inflater to install a dictionary when it's needed. Inflater inflater = new Inflater() { @Override public int inflate(byte[] buffer, int offset, int count) throws DataFormatException { int result = super.inflate(buffer, offset, count); if (result == 0 && needsDictionary()) { setDictionary(DICTIONARY); result = super.inflate(buffer, offset, count); } return result; } }; return new DataInputStream(new InflaterInputStream(throttleStream, inflater)); } private List<String> readNameValueBlock(int length) throws IOException { this.compressedLimit += length; try { int numberOfPairs = nameValueBlockIn.readInt(); if (numberOfPairs < 0) { Logger.getLogger(getClass().getName()).warning("numberOfPairs < 0: " + numberOfPairs); throw ioException("numberOfPairs < 0"); } List<String> entries = new ArrayList<String>(numberOfPairs * 2); for (int i = 0; i < numberOfPairs; i++) { String name = readString(); String values = readString(); if (name.length() == 0) throw ioException("name.length == 0"); if (values.length() == 0) throw ioException("values.length == 0"); entries.add(name); entries.add(values); } if (compressedLimit != 0) { Logger.getLogger(getClass().getName()).warning("compressedLimit > 0: " + compressedLimit); } return entries; } catch (DataFormatException e) { throw new IOException(e.getMessage()); } } private String readString() throws DataFormatException, IOException { int length = nameValueBlockIn.readInt(); byte[] bytes = new byte[length]; Util.readFully(nameValueBlockIn, bytes); return new String(bytes, 0, length, "UTF-8"); } private void readPing(Handler handler, int flags, int length) throws IOException { if (length != 4) throw ioException("TYPE_PING length: %d != 4", length); int id = in.readInt(); handler.ping(flags, id); } private void readGoAway(Handler handler, int flags, int length) throws IOException { if (length != 8) throw ioException("TYPE_GOAWAY length: %d != 8", length); int lastGoodStreamId = in.readInt() & 0x7fffffff; int statusCode = in.readInt(); handler.goAway(flags, lastGoodStreamId, statusCode); } private void readSettings(Handler handler, int flags, int length) throws IOException { int numberOfEntries = in.readInt(); if (length != 4 + 8 * numberOfEntries) { throw ioException("TYPE_SETTINGS length: %d != 4 + 8 * %d", length, numberOfEntries); } Settings settings = new Settings(); for (int i = 0; i < numberOfEntries; i++) { int w1 = in.readInt(); int value = in.readInt(); int idFlags = (w1 & 0xff000000) >>> 24; int id = w1 & 0xffffff; settings.set(id, idFlags, value); } handler.settings(flags, settings); } private static IOException ioException(String message, Object... args) throws IOException { throw new IOException(String.format(message, args)); } @Override public void close() throws IOException { Util.closeAll(in, nameValueBlockIn); } public interface Handler { void data(int flags, int streamId, InputStream in, int length) throws IOException; void synStream(int flags, int streamId, int associatedStreamId, int priority, int slot, List<String> nameValueBlock); void synReply(int flags, int streamId, List<String> nameValueBlock) throws IOException; void headers(int flags, int streamId, List<String> nameValueBlock) throws IOException; void rstStream(int flags, int streamId, int statusCode); void settings(int flags, Settings settings); void noop(); void ping(int flags, int streamId); void goAway(int flags, int lastGoodStreamId, int statusCode); void windowUpdate(int flags, int streamId, int deltaWindowSize); } }