/* * Copyright (c) 1998-2011 Caucho Technology -- all rights reserved * * This file is part of Resin(R) Open Source * * Each copy or derived work must preserve the copyright notice and this * notice unmodified. * * Resin Open Source is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * Resin Open Source 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, or any warranty * of NON-INFRINGEMENT. See the GNU General Public License for more * details. * * You should have received a copy of the GNU General Public License * along with Resin Open Source; if not, write to the * * Free Software Foundation, Inc. * 59 Temple Place, Suite 330 * Boston, MA 02111-1307 USA * * @author Scott Ferguson */ package com.caucho.remote.websocket; import java.io.IOException; import java.io.InputStream; /** * WebSocketInputStream reads a single WebSocket packet. * * <code><pre> * +-+------+---------+-+---------+ * |F|xxx(3)|opcode(4)|R|len(7) | * +-+------+---------+-+---------+ * * OPCODES * 0 - cont * 1 - close * 2 - ping * 3 - pong * 4 - text * 5 - binary * </pre></code> */ public class UnmaskedFrameInputStream extends FrameInputStream { private InputStream _is; private boolean _isFinal; private int _op; private long _length; public UnmaskedFrameInputStream() { } @Override public void init(InputStream is) { if (is == null) throw new NullPointerException(); _is = is; } @Override public int getOpcode() { return _op; } @Override public boolean isFinal() { return _isFinal; } @Override public long getLength() { return _length; } @Override public int read() throws IOException { long length = _length; if (length > 0) { int ch = _is.read(); _length = length - 1; return ch; } else return -1; } @Override public int read(byte []buffer, int offset, int length) throws IOException { if (_length == 0 && ! _isFinal) { readFrameHeader(); if (_length == 0) return -1; } int sublen = length; if (_length < sublen) sublen = (int) _length; sublen = _is.read(buffer, offset, sublen); if (sublen < 0) return sublen; _length -= sublen; return sublen; } @Override protected boolean readFrameHeaderImpl() throws IOException { InputStream is = _is; int frame1 = is.read(); int frame2 = is.read(); if (frame2 < 0) return false; boolean isFinal = (frame1 & FLAG_FIN) == FLAG_FIN; _op = frame1 & 0xf; _isFinal = isFinal; long length = frame2 & 0x7f; if (length < 0x7e) { } else if (length == 0x7e) { length = ((((long) is.read()) << 8) + (((long) is.read()))); } else { length = ((((long) is.read()) << 56) + (((long) is.read()) << 48) + (((long) is.read()) << 40) + (((long) is.read()) << 32) + (((long) is.read()) << 24) + (((long) is.read()) << 16) + (((long) is.read()) << 8) + (((long) is.read()))); } _length = length; return true; } }