Draft_6455.java

/*
 * Copyright (c) 2010-2020 Nathan Rajlich
 *
 *  Permission is hereby granted, free of charge, to any person
 *  obtaining a copy of this software and associated documentation
 *  files (the "Software"), to deal in the Software without
 *  restriction, including without limitation the rights to use,
 *  copy, modify, merge, publish, distribute, sublicense, and/or sell
 *  copies of the Software, and to permit persons to whom the
 *  Software is furnished to do so, subject to the following
 *  conditions:
 *
 *  The above copyright notice and this permission notice shall be
 *  included in all copies or substantial portions of the Software.
 *
 *  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 *  EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
 *  OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 *  NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
 *  HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
 *  WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 *  FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 *  OTHER DEALINGS IN THE SOFTWARE.
 */

package org.java_websocket.drafts;

import java.math.BigInteger;
import java.nio.ByteBuffer;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.security.SecureRandom;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.Collections;
import java.util.LinkedList;
import java.util.List;
import java.util.Locale;
import java.util.TimeZone;
import org.java_websocket.WebSocketImpl;
import org.java_websocket.enums.CloseHandshakeType;
import org.java_websocket.enums.HandshakeState;
import org.java_websocket.enums.Opcode;
import org.java_websocket.enums.ReadyState;
import org.java_websocket.enums.Role;
import org.java_websocket.exceptions.IncompleteException;
import org.java_websocket.exceptions.InvalidDataException;
import org.java_websocket.exceptions.InvalidFrameException;
import org.java_websocket.exceptions.InvalidHandshakeException;
import org.java_websocket.exceptions.LimitExceededException;
import org.java_websocket.exceptions.NotSendableException;
import org.java_websocket.extensions.DefaultExtension;
import org.java_websocket.extensions.IExtension;
import org.java_websocket.framing.BinaryFrame;
import org.java_websocket.framing.CloseFrame;
import org.java_websocket.framing.Framedata;
import org.java_websocket.framing.FramedataImpl1;
import org.java_websocket.framing.TextFrame;
import org.java_websocket.handshake.ClientHandshake;
import org.java_websocket.handshake.ClientHandshakeBuilder;
import org.java_websocket.handshake.HandshakeBuilder;
import org.java_websocket.handshake.ServerHandshake;
import org.java_websocket.handshake.ServerHandshakeBuilder;
import org.java_websocket.protocols.IProtocol;
import org.java_websocket.protocols.Protocol;
import org.java_websocket.util.Base64;
import org.java_websocket.util.Charsetfunctions;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Implementation for the RFC 6455 websocket protocol This is the recommended class for your
 * websocket connection
 */
public class Draft_6455 extends Draft {

  /**
   * Handshake specific field for the key
   */
  private static final String SEC_WEB_SOCKET_KEY = "Sec-WebSocket-Key";

  /**
   * Handshake specific field for the protocol
   */
  private static final String SEC_WEB_SOCKET_PROTOCOL = "Sec-WebSocket-Protocol";

  /**
   * Handshake specific field for the extension
   */
  private static final String SEC_WEB_SOCKET_EXTENSIONS = "Sec-WebSocket-Extensions";

  /**
   * Handshake specific field for the accept
   */
  private static final String SEC_WEB_SOCKET_ACCEPT = "Sec-WebSocket-Accept";

  /**
   * Handshake specific field for the upgrade
   */
  private static final String UPGRADE = "Upgrade";

  /**
   * Handshake specific field for the connection
   */
  private static final String CONNECTION = "Connection";

  /**
   * Logger instance
   *
   * @since 1.4.0
   */
  private final Logger log = LoggerFactory.getLogger(Draft_6455.class);

  /**
   * Attribute for the used extension in this draft
   */
  private IExtension negotiatedExtension = new DefaultExtension();

  /**
   * Attribute for the default extension
   */
  private IExtension defaultExtension = new DefaultExtension();

  /**
   * Attribute for all available extension in this draft
   */
  private List<IExtension> knownExtensions;

  /**
   * Current active extension used to decode messages
   */
  private IExtension currentDecodingExtension;

  /**
   * Attribute for the used protocol in this draft
   */
  private IProtocol protocol;

  /**
   * Attribute for all available protocols in this draft
   */
  private List<IProtocol> knownProtocols;

  /**
   * Attribute for the current continuous frame
   */
  private Framedata currentContinuousFrame;

  /**
   * Attribute for the payload of the current continuous frame
   */
  private final List<ByteBuffer> byteBufferList;

  /**
   * Attribute for the current incomplete frame
   */
  private ByteBuffer incompleteframe;

  /**
   * Attribute for the reusable random instance
   */
  private final SecureRandom reuseableRandom = new SecureRandom();

  /**
   * Default maximum payload size for a single frame when not explicitly configured (16 MiB).
   *
   * @since 1.6.1
   */
  public static final int DEFAULT_MAX_FRAME_SIZE = 16 * 1024 * 1024;

  /**
   * Attribute for the maximum allowed size of a frame
   *
   * @since 1.4.0
   */
  private int maxFrameSize;

  /**
   * Constructor for the websocket protocol specified by RFC 6455 with default extensions
   *
   * @since 1.3.5
   */
  public Draft_6455() {
    this(Collections.<IExtension>emptyList());
  }

  /**
   * Constructor for the websocket protocol specified by RFC 6455 with custom extensions
   *
   * @param inputExtension the extension which should be used for this draft
   * @since 1.3.5
   */
  public Draft_6455(IExtension inputExtension) {
    this(Collections.singletonList(inputExtension));
  }

  /**
   * Constructor for the websocket protocol specified by RFC 6455 with custom extensions
   *
   * @param inputExtensions the extensions which should be used for this draft
   * @since 1.3.5
   */
  public Draft_6455(List<IExtension> inputExtensions) {
    this(inputExtensions, Collections.<IProtocol>singletonList(new Protocol("")));
  }

  /**
   * Constructor for the websocket protocol specified by RFC 6455 with custom extensions and
   * protocols
   *
   * @param inputExtensions the extensions which should be used for this draft
   * @param inputProtocols  the protocols which should be used for this draft
   * @since 1.3.7
   */
  public Draft_6455(List<IExtension> inputExtensions, List<IProtocol> inputProtocols) {
    this(inputExtensions, inputProtocols, DEFAULT_MAX_FRAME_SIZE);
  }

  /**
   * Constructor for the websocket protocol specified by RFC 6455 with custom extensions and
   * protocols
   *
   * @param inputExtensions   the extensions which should be used for this draft
   * @param inputMaxFrameSize the maximum allowed size of a frame (the real payload size, decoded
   *                          frames can be bigger)
   * @since 1.4.0
   */
  public Draft_6455(List<IExtension> inputExtensions, int inputMaxFrameSize) {
    this(inputExtensions, Collections.<IProtocol>singletonList(new Protocol("")),
        inputMaxFrameSize);
  }

  /**
   * Constructor for the websocket protocol specified by RFC 6455 with custom extensions and
   * protocols
   *
   * @param inputExtensions   the extensions which should be used for this draft
   * @param inputProtocols    the protocols which should be used for this draft
   * @param inputMaxFrameSize the maximum allowed size of a frame (the real payload size, decoded
   *                          frames can be bigger)
   * @since 1.4.0
   */
  public Draft_6455(List<IExtension> inputExtensions, List<IProtocol> inputProtocols,
      int inputMaxFrameSize) {
    if (inputExtensions == null || inputProtocols == null || inputMaxFrameSize < 1) {
      throw new IllegalArgumentException();
    }
    knownExtensions = new ArrayList<>(inputExtensions.size());
    knownProtocols = new ArrayList<>(inputProtocols.size());
    boolean hasDefault = false;
    byteBufferList = new ArrayList<>();
    for (IExtension inputExtension : inputExtensions) {
      if (inputExtension.getClass().equals(DefaultExtension.class)) {
        hasDefault = true;
      }
    }
    knownExtensions.addAll(inputExtensions);
    //We always add the DefaultExtension to implement the normal RFC 6455 specification
    if (!hasDefault) {
      knownExtensions.add(this.knownExtensions.size(), negotiatedExtension);
    }
    knownProtocols.addAll(inputProtocols);
    maxFrameSize = inputMaxFrameSize;
    currentDecodingExtension = null;
  }

  @Override
  public HandshakeState acceptHandshakeAsServer(ClientHandshake handshakedata)
      throws InvalidHandshakeException {
    int v = readVersion(handshakedata);
    if (v != 13) {
      log.trace("acceptHandshakeAsServer - Wrong websocket version.");
      return HandshakeState.NOT_MATCHED;
    }
    HandshakeState extensionState = HandshakeState.NOT_MATCHED;
    String requestedExtension = handshakedata.getFieldValue(SEC_WEB_SOCKET_EXTENSIONS);
    for (IExtension knownExtension : knownExtensions) {
      if (knownExtension.acceptProvidedExtensionAsServer(requestedExtension)) {
        negotiatedExtension = knownExtension;
        extensionState = HandshakeState.MATCHED;
        log.trace("acceptHandshakeAsServer - Matching extension found: {}", negotiatedExtension);
        break;
      }
    }
    HandshakeState protocolState = containsRequestedProtocol(
        handshakedata.getFieldValue(SEC_WEB_SOCKET_PROTOCOL));
    if (protocolState == HandshakeState.MATCHED && extensionState == HandshakeState.MATCHED) {
      return HandshakeState.MATCHED;
    }
    log.trace("acceptHandshakeAsServer - No matching extension or protocol found.");
    return HandshakeState.NOT_MATCHED;
  }

  /**
   * Check if the requested protocol is part of this draft
   *
   * @param requestedProtocol the requested protocol
   * @return MATCHED if it is matched, otherwise NOT_MATCHED
   */
  private HandshakeState containsRequestedProtocol(String requestedProtocol) {
    for (IProtocol knownProtocol : knownProtocols) {
      if (knownProtocol.acceptProvidedProtocol(requestedProtocol)) {
        protocol = knownProtocol;
        log.trace("acceptHandshake - Matching protocol found: {}", protocol);
        return HandshakeState.MATCHED;
      }
    }
    return HandshakeState.NOT_MATCHED;
  }

  @Override
  public HandshakeState acceptHandshakeAsClient(ClientHandshake request, ServerHandshake response)
      throws InvalidHandshakeException {
    if (!basicAccept(response)) {
      log.trace("acceptHandshakeAsClient - Missing/wrong upgrade or connection in handshake.");
      return HandshakeState.NOT_MATCHED;
    }
    if (!request.hasFieldValue(SEC_WEB_SOCKET_KEY) || !response
        .hasFieldValue(SEC_WEB_SOCKET_ACCEPT)) {
      log.trace("acceptHandshakeAsClient - Missing Sec-WebSocket-Key or Sec-WebSocket-Accept");
      return HandshakeState.NOT_MATCHED;
    }

    String seckeyAnswer = response.getFieldValue(SEC_WEB_SOCKET_ACCEPT);
    String seckeyChallenge = request.getFieldValue(SEC_WEB_SOCKET_KEY);
    seckeyChallenge = generateFinalKey(seckeyChallenge);

    if (!seckeyChallenge.equals(seckeyAnswer)) {
      log.trace("acceptHandshakeAsClient - Wrong key for Sec-WebSocket-Key.");
      return HandshakeState.NOT_MATCHED;
    }
    HandshakeState extensionState = HandshakeState.NOT_MATCHED;
    String requestedExtension = response.getFieldValue(SEC_WEB_SOCKET_EXTENSIONS);
    for (IExtension knownExtension : knownExtensions) {
      if (knownExtension.acceptProvidedExtensionAsClient(requestedExtension)) {
        negotiatedExtension = knownExtension;
        extensionState = HandshakeState.MATCHED;
        log.trace("acceptHandshakeAsClient - Matching extension found: {}", negotiatedExtension);
        break;
      }
    }
    HandshakeState protocolState = containsRequestedProtocol(
        response.getFieldValue(SEC_WEB_SOCKET_PROTOCOL));
    if (protocolState == HandshakeState.MATCHED && extensionState == HandshakeState.MATCHED) {
      return HandshakeState.MATCHED;
    }
    log.trace("acceptHandshakeAsClient - No matching extension or protocol found.");
    return HandshakeState.NOT_MATCHED;
  }

  /**
   * Getter for the extension which is used by this draft
   *
   * @return the extension which is used or null, if handshake is not yet done
   */
  public IExtension getExtension() {
    return negotiatedExtension;
  }

  /**
   * Getter for all available extensions for this draft
   *
   * @return the extensions which are enabled for this draft
   */
  public List<IExtension> getKnownExtensions() {
    return knownExtensions;
  }

  /**
   * Getter for the protocol which is used by this draft
   *
   * @return the protocol which is used or null, if handshake is not yet done or no valid protocols
   * @since 1.3.7
   */
  public IProtocol getProtocol() {
    return protocol;
  }


  /**
   * Getter for the maximum allowed payload size which is used by this draft
   *
   * @return the size, which is allowed for the payload
   * @since 1.4.0
   */
  public int getMaxFrameSize() {
    return maxFrameSize;
  }

  @Override
  public int checkAlloc(int bytecount) throws InvalidDataException {
    int allowed = super.checkAlloc(bytecount);
    if (allowed > maxFrameSize) {
      throw new LimitExceededException("Payload limit reached.", maxFrameSize);
    }
    return allowed;
  }

  /**
   * Getter for all available protocols for this draft
   *
   * @return the protocols which are enabled for this draft
   * @since 1.3.7
   */
  public List<IProtocol> getKnownProtocols() {
    return knownProtocols;
  }

  @Override
  public ClientHandshakeBuilder postProcessHandshakeRequestAsClient(
      ClientHandshakeBuilder request) {
    request.put(UPGRADE, "websocket");
    request.put(CONNECTION, UPGRADE); // to respond to a Connection keep alives
    byte[] random = new byte[16];
    reuseableRandom.nextBytes(random);
    request.put(SEC_WEB_SOCKET_KEY, Base64.encodeBytes(random));
    request.put("Sec-WebSocket-Version", "13");// overwriting the previous
    StringBuilder requestedExtensions = new StringBuilder();
    for (IExtension knownExtension : knownExtensions) {
      if (knownExtension.getProvidedExtensionAsClient() != null
          && knownExtension.getProvidedExtensionAsClient().length() != 0) {
        if (requestedExtensions.length() > 0) {
          requestedExtensions.append(", ");
        }
        requestedExtensions.append(knownExtension.getProvidedExtensionAsClient());
      }
    }
    if (requestedExtensions.length() != 0) {
      request.put(SEC_WEB_SOCKET_EXTENSIONS, requestedExtensions.toString());
    }
    StringBuilder requestedProtocols = new StringBuilder();
    for (IProtocol knownProtocol : knownProtocols) {
      if (knownProtocol.getProvidedProtocol().length() != 0) {
        if (requestedProtocols.length() > 0) {
          requestedProtocols.append(", ");
        }
        requestedProtocols.append(knownProtocol.getProvidedProtocol());
      }
    }
    if (requestedProtocols.length() != 0) {
      request.put(SEC_WEB_SOCKET_PROTOCOL, requestedProtocols.toString());
    }
    return request;
  }

  @Override
  public HandshakeBuilder postProcessHandshakeResponseAsServer(ClientHandshake request,
      ServerHandshakeBuilder response) throws InvalidHandshakeException {
    response.put(UPGRADE, "websocket");
    response.put(CONNECTION,
        request.getFieldValue(CONNECTION)); // to respond to a Connection keep alives
    String seckey = request.getFieldValue(SEC_WEB_SOCKET_KEY);
    if (seckey == null || "".equals(seckey)) {
      throw new InvalidHandshakeException("missing Sec-WebSocket-Key");
    }
    response.put(SEC_WEB_SOCKET_ACCEPT, generateFinalKey(seckey));
    if (getExtension().getProvidedExtensionAsServer().length() != 0) {
      response.put(SEC_WEB_SOCKET_EXTENSIONS, getExtension().getProvidedExtensionAsServer());
    }
    if (getProtocol() != null && getProtocol().getProvidedProtocol().length() != 0) {
      response.put(SEC_WEB_SOCKET_PROTOCOL, getProtocol().getProvidedProtocol());
    }
    response.setHttpStatusMessage("Web Socket Protocol Handshake");
    response.put("Server", "TooTallNate Java-WebSocket");
    response.put("Date", getServerTime());
    return response;
  }

  @Override
  public Draft copyInstance() {
    ArrayList<IExtension> newExtensions = new ArrayList<>();
    for (IExtension knownExtension : getKnownExtensions()) {
      newExtensions.add(knownExtension.copyInstance());
    }
    ArrayList<IProtocol> newProtocols = new ArrayList<>();
    for (IProtocol knownProtocol : getKnownProtocols()) {
      newProtocols.add(knownProtocol.copyInstance());
    }
    return new Draft_6455(newExtensions, newProtocols, maxFrameSize);
  }

  @Override
  public ByteBuffer createBinaryFrame(Framedata framedata) {
    getExtension().encodeFrame(framedata);
    if (log.isTraceEnabled()) {
      log.trace("afterEnconding({}): {}", framedata.getPayloadData().remaining(),
          (framedata.getPayloadData().remaining() > 1000 ? "too big to display"
              : new String(framedata.getPayloadData().array())));
    }
    return createByteBufferFromFramedata(framedata);
  }

  private ByteBuffer createByteBufferFromFramedata(Framedata framedata) {
    ByteBuffer mes = framedata.getPayloadData();
    boolean mask = role == Role.CLIENT;
    int sizebytes = getSizeBytes(mes);
    ByteBuffer buf = ByteBuffer.allocate(
        1 + (sizebytes > 1 ? sizebytes + 1 : sizebytes) + (mask ? 4 : 0) + mes.remaining());
    byte optcode = fromOpcode(framedata.getOpcode());
    byte one = (byte) (framedata.isFin() ? -128 : 0);
    one |= optcode;
    if (framedata.isRSV1()) {
      one |= getRSVByte(1);
    }
    if (framedata.isRSV2()) {
      one |= getRSVByte(2);
    }
    if (framedata.isRSV3()) {
      one |= getRSVByte(3);
    }
    buf.put(one);
    byte[] payloadlengthbytes = toByteArray(mes.remaining(), sizebytes);
    assert (payloadlengthbytes.length == sizebytes);

    if (sizebytes == 1) {
      buf.put((byte) (payloadlengthbytes[0] | getMaskByte(mask)));
    } else if (sizebytes == 2) {
      buf.put((byte) ((byte) 126 | getMaskByte(mask)));
      buf.put(payloadlengthbytes);
    } else if (sizebytes == 8) {
      buf.put((byte) ((byte) 127 | getMaskByte(mask)));
      buf.put(payloadlengthbytes);
    } else {
      throw new IllegalStateException("Size representation not supported/specified");
    }
    if (mask) {
      ByteBuffer maskkey = ByteBuffer.allocate(4);
      maskkey.putInt(reuseableRandom.nextInt());
      buf.put(maskkey.array());
      for (int i = 0; mes.hasRemaining(); i++) {
        buf.put((byte) (mes.get() ^ maskkey.get(i % 4)));
      }
    } else {
      buf.put(mes);
      //Reset the position of the bytebuffer e.g. for additional use
      mes.flip();
    }
    assert (buf.remaining() == 0) : buf.remaining();
    buf.flip();
    return buf;
  }

  private Framedata translateSingleFrame(ByteBuffer buffer)
      throws IncompleteException, InvalidDataException {
    if (buffer == null) {
      throw new IllegalArgumentException();
    }
    int maxpacketsize = buffer.remaining();
    int realpacketsize = 2;
    translateSingleFrameCheckPacketSize(maxpacketsize, realpacketsize);
    byte b1 = buffer.get(/*0*/);
    boolean fin = b1 >> 8 != 0;
    boolean rsv1 = (b1 & 0x40) != 0;
    boolean rsv2 = (b1 & 0x20) != 0;
    boolean rsv3 = (b1 & 0x10) != 0;
    byte b2 = buffer.get(/*1*/);
    boolean mask = (b2 & -128) != 0;
    int payloadlength = (byte) (b2 & ~(byte) 128);
    Opcode optcode = toOpcode((byte) (b1 & 15));

    if (!(payloadlength >= 0 && payloadlength <= 125)) {
      TranslatedPayloadMetaData payloadData = translateSingleFramePayloadLength(buffer, optcode,
          payloadlength, maxpacketsize, realpacketsize);
      payloadlength = payloadData.getPayloadLength();
      realpacketsize = payloadData.getRealPackageSize();
    }
    translateSingleFrameCheckLengthLimit(payloadlength);
    realpacketsize += (mask ? 4 : 0);
    realpacketsize += payloadlength;
    translateSingleFrameCheckPacketSize(maxpacketsize, realpacketsize);

    ByteBuffer payload = ByteBuffer.allocate(checkAlloc(payloadlength));
    if (mask) {
      byte[] maskskey = new byte[4];
      buffer.get(maskskey);
      for (int i = 0; i < payloadlength; i++) {
        payload.put((byte) (buffer.get(/*payloadstart + i*/) ^ maskskey[i % 4]));
      }
    } else {
      payload.put(buffer.array(), buffer.position(), payload.limit());
      buffer.position(buffer.position() + payload.limit());
    }

    FramedataImpl1 frame = FramedataImpl1.get(optcode);
    frame.setFin(fin);
    frame.setRSV1(rsv1);
    frame.setRSV2(rsv2);
    frame.setRSV3(rsv3);
    payload.flip();
    frame.setPayload(payload);
    if (frame.getOpcode() != Opcode.CONTINUOUS) {
      // Prioritize the negotiated extension
      if (frame.isRSV1() || frame.isRSV2() || frame.isRSV3()) {
        currentDecodingExtension = getExtension();
      } else {
        // No encoded message, so we can use the default one
        currentDecodingExtension = defaultExtension;
      }
    }
    if (currentDecodingExtension == null) {
      currentDecodingExtension = defaultExtension;
    }
    currentDecodingExtension.isFrameValid(frame);
    currentDecodingExtension.decodeFrame(frame);
    if (log.isTraceEnabled()) {
      log.trace("afterDecoding({}): {}", frame.getPayloadData().remaining(),
          (frame.getPayloadData().remaining() > 1000 ? "too big to display"
              : new String(frame.getPayloadData().array())));
    }
    frame.isValid();
    return frame;
  }

  /**
   * Translate the buffer depending when it has an extended payload length (126 or 127)
   *
   * @param buffer            the buffer to read from
   * @param optcode           the decoded optcode
   * @param oldPayloadlength  the old payload length
   * @param maxpacketsize     the max packet size allowed
   * @param oldRealpacketsize the real packet size
   * @return the new payload data containing new payload length and new packet size
   * @throws InvalidFrameException  thrown if a control frame has an invalid length
   * @throws IncompleteException    if the maxpacketsize is smaller than the realpackagesize
   * @throws LimitExceededException if the payload length is to big
   */
  private TranslatedPayloadMetaData translateSingleFramePayloadLength(ByteBuffer buffer,
      Opcode optcode, int oldPayloadlength, int maxpacketsize, int oldRealpacketsize)
      throws InvalidFrameException, IncompleteException, LimitExceededException {
    int payloadlength = oldPayloadlength;
    int realpacketsize = oldRealpacketsize;
    if (optcode == Opcode.PING || optcode == Opcode.PONG || optcode == Opcode.CLOSING) {
      log.trace("Invalid frame: more than 125 octets");
      throw new InvalidFrameException("more than 125 octets");
    }
    if (payloadlength == 126) {
      realpacketsize += 2; // additional length bytes
      translateSingleFrameCheckPacketSize(maxpacketsize, realpacketsize);
      byte[] sizebytes = new byte[3];
      sizebytes[1] = buffer.get(/*1 + 1*/);
      sizebytes[2] = buffer.get(/*1 + 2*/);
      payloadlength = new BigInteger(sizebytes).intValue();
    } else {
      realpacketsize += 8; // additional length bytes
      translateSingleFrameCheckPacketSize(maxpacketsize, realpacketsize);
      byte[] bytes = new byte[8];
      for (int i = 0; i < 8; i++) {
        bytes[i] = buffer.get(/*1 + i*/);
      }
      long length = new BigInteger(bytes).longValue();
      translateSingleFrameCheckLengthLimit(length);
      payloadlength = (int) length;
    }
    return new TranslatedPayloadMetaData(payloadlength, realpacketsize);
  }

  /**
   * Check if the frame size exceeds the allowed limit
   *
   * @param length the current payload length
   * @throws LimitExceededException if the payload length is to big
   */
  private void translateSingleFrameCheckLengthLimit(long length) throws LimitExceededException {
    if (length > Integer.MAX_VALUE) {
      log.trace("Limit exedeed: Payloadsize is to big...");
      throw new LimitExceededException("Payloadsize is to big...");
    }
    if (length > maxFrameSize) {
      log.trace("Payload limit reached. Allowed: {} Current: {}", maxFrameSize, length);
      throw new LimitExceededException("Payload limit reached.", maxFrameSize);
    }
    if (length < 0) {
      log.trace("Limit underflow: Payloadsize is to little...");
      throw new LimitExceededException("Payloadsize is to little...");
    }
  }

  /**
   * Check if the max packet size is smaller than the real packet size
   *
   * @param maxpacketsize  the max packet size
   * @param realpacketsize the real packet size
   * @throws IncompleteException if the maxpacketsize is smaller than the realpackagesize
   */
  private void translateSingleFrameCheckPacketSize(int maxpacketsize, int realpacketsize)
      throws IncompleteException {
    if (maxpacketsize < realpacketsize) {
      log.trace("Incomplete frame: maxpacketsize < realpacketsize");
      throw new IncompleteException(realpacketsize);
    }
  }

  /**
   * Get a byte that can set RSV bits when OR(|)'d. 0 1 2 3 4 5 6 7 +-+-+-+-+-------+ |F|R|R|R|
   * opcode| |I|S|S|S|  (4)  | |N|V|V|V|       | | |1|2|3|       |
   *
   * @param rsv Can only be {0, 1, 2, 3}
   * @return byte that represents which RSV bit is set.
   */
  private byte getRSVByte(int rsv) {
    switch (rsv) {
      case 1 : // 0100 0000
        return 0x40;
      case 2 : // 0010 0000
        return 0x20;
      case 3 : // 0001 0000
        return 0x10;
      default:
        return 0;
    }
  }

  /**
   * Get the mask byte if existing
   *
   * @param mask is mask active or not
   * @return -128 for true, 0 for false
   */
  private byte getMaskByte(boolean mask) {
    return mask ? (byte) -128 : 0;
  }

  /**
   * Get the size bytes for the byte buffer
   *
   * @param mes the current buffer
   * @return the size bytes
   */
  private int getSizeBytes(ByteBuffer mes) {
    if (mes.remaining() <= 125) {
      return 1;
    } else if (mes.remaining() <= 65535) {
      return 2;
    }
    return 8;
  }

  @Override
  public List<Framedata> translateFrame(ByteBuffer buffer) throws InvalidDataException {
    while (true) {
      List<Framedata> frames = new LinkedList<>();
      Framedata cur;
      if (incompleteframe != null) {
        // complete an incomplete frame
        try {
          buffer.mark();
          int availableNextByteCount = buffer.remaining();// The number of bytes received
          int expectedNextByteCount = incompleteframe
              .remaining();// The number of bytes to complete the incomplete frame

          if (expectedNextByteCount > availableNextByteCount) {
            // did not receive enough bytes to complete the frame
            incompleteframe.put(buffer.array(), buffer.position(), availableNextByteCount);
            buffer.position(buffer.position() + availableNextByteCount);
            return Collections.emptyList();
          }
          incompleteframe.put(buffer.array(), buffer.position(), expectedNextByteCount);
          buffer.position(buffer.position() + expectedNextByteCount);
          cur = translateSingleFrame((ByteBuffer) incompleteframe.duplicate().position(0));
          frames.add(cur);
          incompleteframe = null;
        } catch (IncompleteException e) {
          // extending as much as suggested
          ByteBuffer extendedframe = ByteBuffer.allocate(checkAlloc(e.getPreferredSize()));
          assert (extendedframe.limit() > incompleteframe.limit());
          incompleteframe.rewind();
          extendedframe.put(incompleteframe);
          incompleteframe = extendedframe;
          continue;
        }
      }

      // Read as much as possible full frames
      while (buffer.hasRemaining()) {
        buffer.mark();
        try {
          cur = translateSingleFrame(buffer);
          frames.add(cur);
        } catch (IncompleteException e) {
          // remember the incomplete data
          buffer.reset();
          int pref = e.getPreferredSize();
          incompleteframe = ByteBuffer.allocate(checkAlloc(pref));
          incompleteframe.put(buffer);
          break;
        }
      }
      return frames;
    }
  }

  @Override
  public List<Framedata> createFrames(ByteBuffer binary, boolean mask) {
    BinaryFrame curframe = new BinaryFrame();
    curframe.setPayload(binary);
    curframe.setTransferemasked(mask);
    try {
      curframe.isValid();
    } catch (InvalidDataException e) {
      throw new NotSendableException(e);
    }
    return Collections.singletonList((Framedata) curframe);
  }

  @Override
  public List<Framedata> createFrames(String text, boolean mask) {
    TextFrame curframe = new TextFrame();
    curframe.setPayload(ByteBuffer.wrap(Charsetfunctions.utf8Bytes(text)));
    curframe.setTransferemasked(mask);
    try {
      curframe.isValid();
    } catch (InvalidDataException e) {
      throw new NotSendableException(e);
    }
    return Collections.singletonList((Framedata) curframe);
  }

  @Override
  public void reset() {
    incompleteframe = null;
    if (negotiatedExtension != null) {
      negotiatedExtension.reset();
    }
    negotiatedExtension = new DefaultExtension();
    protocol = null;
  }

  /**
   * Generate a date for for the date-header
   *
   * @return the server time
   */
  private String getServerTime() {
    Calendar calendar = Calendar.getInstance();
    SimpleDateFormat dateFormat = new SimpleDateFormat(
        "EEE, dd MMM yyyy HH:mm:ss z", Locale.US);
    dateFormat.setTimeZone(TimeZone.getTimeZone("GMT"));
    return dateFormat.format(calendar.getTime());
  }

  /**
   * Generate a final key from a input string
   *
   * @param in the input string
   * @return a final key
   */
  private String generateFinalKey(String in) {
    String seckey = in.trim();
    String acc = seckey + "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
    MessageDigest sh1;
    try {
      sh1 = MessageDigest.getInstance("SHA1");
    } catch (NoSuchAlgorithmException e) {
      throw new IllegalStateException(e);
    }
    return Base64.encodeBytes(sh1.digest(acc.getBytes()));
  }

  private byte[] toByteArray(long val, int bytecount) {
    byte[] buffer = new byte[bytecount];
    int highest = 8 * bytecount - 8;
    for (int i = 0; i < bytecount; i++) {
      buffer[i] = (byte) (val >>> (highest - 8 * i));
    }
    return buffer;
  }


  private byte fromOpcode(Opcode opcode) {
    if (opcode == Opcode.CONTINUOUS) {
      return 0;
    } else if (opcode == Opcode.TEXT) {
      return 1;
    } else if (opcode == Opcode.BINARY) {
      return 2;
    } else if (opcode == Opcode.CLOSING) {
      return 8;
    } else if (opcode == Opcode.PING) {
      return 9;
    } else if (opcode == Opcode.PONG) {
      return 10;
    }
    throw new IllegalArgumentException("Don't know how to handle " + opcode.toString());
  }

  private Opcode toOpcode(byte opcode) throws InvalidFrameException {
    switch (opcode) {
      case 0:
        return Opcode.CONTINUOUS;
      case 1:
        return Opcode.TEXT;
      case 2:
        return Opcode.BINARY;
      // 3-7 are not yet defined
      case 8:
        return Opcode.CLOSING;
      case 9:
        return Opcode.PING;
      case 10:
        return Opcode.PONG;
      // 11-15 are not yet defined
      default:
        throw new InvalidFrameException("Unknown opcode " + (short) opcode);
    }
  }

  @Override
  public void processFrame(WebSocketImpl webSocketImpl, Framedata frame)
      throws InvalidDataException {
    Opcode curop = frame.getOpcode();
    if (curop == Opcode.CLOSING) {
      processFrameClosing(webSocketImpl, frame);
    } else if (curop == Opcode.PING) {
      webSocketImpl.getWebSocketListener().onWebsocketPing(webSocketImpl, frame);
    } else if (curop == Opcode.PONG) {
      webSocketImpl.updateLastPong();
      webSocketImpl.getWebSocketListener().onWebsocketPong(webSocketImpl, frame);
    } else if (!frame.isFin() || curop == Opcode.CONTINUOUS) {
      processFrameContinuousAndNonFin(webSocketImpl, frame, curop);
    } else if (currentContinuousFrame != null) {
      log.error("Protocol error: Continuous frame sequence not completed.");
      throw new InvalidDataException(CloseFrame.PROTOCOL_ERROR,
          "Continuous frame sequence not completed.");
    } else if (curop == Opcode.TEXT) {
      processFrameText(webSocketImpl, frame);
    } else if (curop == Opcode.BINARY) {
      processFrameBinary(webSocketImpl, frame);
    } else {
      log.error("non control or continious frame expected");
      throw new InvalidDataException(CloseFrame.PROTOCOL_ERROR,
          "non control or continious frame expected");
    }
  }

  /**
   * Process the frame if it is a continuous frame or the fin bit is not set
   *
   * @param webSocketImpl the websocket implementation to use
   * @param frame         the current frame
   * @param curop         the current Opcode
   * @throws InvalidDataException if there is a protocol error
   */
  private void processFrameContinuousAndNonFin(WebSocketImpl webSocketImpl, Framedata frame,
      Opcode curop) throws InvalidDataException {
    if (curop != Opcode.CONTINUOUS) {
      processFrameIsNotFin(frame);
    } else if (frame.isFin()) {
      processFrameIsFin(webSocketImpl, frame);
    } else if (currentContinuousFrame == null) {
      log.error("Protocol error: Continuous frame sequence was not started.");
      throw new InvalidDataException(CloseFrame.PROTOCOL_ERROR,
          "Continuous frame sequence was not started.");
    }
    //Check if the whole payload is valid utf8, when the opcode indicates a text
    if (curop == Opcode.TEXT && !Charsetfunctions.isValidUTF8(frame.getPayloadData())) {
      log.error("Protocol error: Payload is not UTF8");
      throw new InvalidDataException(CloseFrame.NO_UTF8);
    }
    //Checking if the current continuous frame contains a correct payload with the other frames combined
    if (curop == Opcode.CONTINUOUS && currentContinuousFrame != null) {
      addToBufferList(frame.getPayloadData());
    }
  }

  /**
   * Process the frame if it is a binary frame
   *
   * @param webSocketImpl the websocket impl
   * @param frame         the frame
   */
  private void processFrameBinary(WebSocketImpl webSocketImpl, Framedata frame) {
    try {
      webSocketImpl.getWebSocketListener()
          .onWebsocketMessage(webSocketImpl, frame.getPayloadData());
    } catch (RuntimeException e) {
      logRuntimeException(webSocketImpl, e);
    }
  }

  /**
   * Log the runtime exception to the specific WebSocketImpl
   *
   * @param webSocketImpl the implementation of the websocket
   * @param e             the runtime exception
   */
  private void logRuntimeException(WebSocketImpl webSocketImpl, RuntimeException e) {
    log.error("Runtime exception during onWebsocketMessage", e);
    webSocketImpl.getWebSocketListener().onWebsocketError(webSocketImpl, e);
  }

  /**
   * Process the frame if it is a text frame
   *
   * @param webSocketImpl the websocket impl
   * @param frame         the frame
   */
  private void processFrameText(WebSocketImpl webSocketImpl, Framedata frame)
      throws InvalidDataException {
    try {
      webSocketImpl.getWebSocketListener()
          .onWebsocketMessage(webSocketImpl, Charsetfunctions.stringUtf8(frame.getPayloadData()));
    } catch (RuntimeException e) {
      logRuntimeException(webSocketImpl, e);
    }
  }

  /**
   * Process the frame if it is the last frame
   *
   * @param webSocketImpl the websocket impl
   * @param frame         the frame
   * @throws InvalidDataException if there is a protocol error
   */
  private void processFrameIsFin(WebSocketImpl webSocketImpl, Framedata frame)
      throws InvalidDataException {
    if (currentContinuousFrame == null) {
      log.trace("Protocol error: Previous continuous frame sequence not completed.");
      throw new InvalidDataException(CloseFrame.PROTOCOL_ERROR,
          "Continuous frame sequence was not started.");
    }
    addToBufferList(frame.getPayloadData());
    checkBufferLimit();
    if (currentContinuousFrame.getOpcode() == Opcode.TEXT) {
      ((FramedataImpl1) currentContinuousFrame).setPayload(getPayloadFromByteBufferList());
      ((FramedataImpl1) currentContinuousFrame).isValid();
      try {
        webSocketImpl.getWebSocketListener().onWebsocketMessage(webSocketImpl,
            Charsetfunctions.stringUtf8(currentContinuousFrame.getPayloadData()));
      } catch (RuntimeException e) {
        logRuntimeException(webSocketImpl, e);
      }
    } else if (currentContinuousFrame.getOpcode() == Opcode.BINARY) {
      ((FramedataImpl1) currentContinuousFrame).setPayload(getPayloadFromByteBufferList());
      ((FramedataImpl1) currentContinuousFrame).isValid();
      try {
        webSocketImpl.getWebSocketListener()
            .onWebsocketMessage(webSocketImpl, currentContinuousFrame.getPayloadData());
      } catch (RuntimeException e) {
        logRuntimeException(webSocketImpl, e);
      }
    }
    currentContinuousFrame = null;
    clearBufferList();
  }

  /**
   * Process the frame if it is not the last frame
   *
   * @param frame the frame
   * @throws InvalidDataException if there is a protocol error
   */
  private void processFrameIsNotFin(Framedata frame) throws InvalidDataException {
    if (currentContinuousFrame != null) {
      log.trace("Protocol error: Previous continuous frame sequence not completed.");
      throw new InvalidDataException(CloseFrame.PROTOCOL_ERROR,
          "Previous continuous frame sequence not completed.");
    }
    currentContinuousFrame = frame;
    addToBufferList(frame.getPayloadData());
    checkBufferLimit();
  }

  /**
   * Process the frame if it is a closing frame
   *
   * @param webSocketImpl the websocket impl
   * @param frame         the frame
   */
  private void processFrameClosing(WebSocketImpl webSocketImpl, Framedata frame) {
    int code = CloseFrame.NOCODE;
    String reason = "";
    if (frame instanceof CloseFrame) {
      CloseFrame cf = (CloseFrame) frame;
      code = cf.getCloseCode();
      reason = cf.getMessage();
    }
    if (webSocketImpl.getReadyState() == ReadyState.CLOSING) {
      // complete the close handshake by disconnecting
      webSocketImpl.closeConnection(code, reason, true);
    } else {
      // echo close handshake
      if (getCloseHandshakeType() == CloseHandshakeType.TWOWAY) {
        webSocketImpl.close(code, reason, true);
      } else {
        webSocketImpl.flushAndClose(code, reason, false);
      }
    }
  }

  /**
   * Clear the current bytebuffer list
   */
  private void clearBufferList() {
    synchronized (byteBufferList) {
      byteBufferList.clear();
    }
  }

  /**
   * Add a payload to the current bytebuffer list
   *
   * @param payloadData the new payload
   */
  private void addToBufferList(ByteBuffer payloadData) {
    synchronized (byteBufferList) {
      byteBufferList.add(payloadData);
    }
  }

  /**
   * Check the current size of the buffer and throw an exception if the size is bigger than the max
   * allowed frame size
   *
   * @throws LimitExceededException if the current size is bigger than the allowed size
   */
  private void checkBufferLimit() throws LimitExceededException {
    long totalSize = getByteBufferListSize();
    if (totalSize > maxFrameSize) {
      clearBufferList();
      log.trace("Payload limit reached. Allowed: {} Current: {}", maxFrameSize, totalSize);
      throw new LimitExceededException(maxFrameSize);
    }
  }

  @Override
  public CloseHandshakeType getCloseHandshakeType() {
    return CloseHandshakeType.TWOWAY;
  }

  @Override
  public String toString() {
    String result = super.toString();
    if (getExtension() != null) {
      result += " extension: " + getExtension().toString();
    }
    if (getProtocol() != null) {
      result += " protocol: " + getProtocol().toString();
    }
    result += " max frame size: " + this.maxFrameSize;
    return result;
  }

  @Override
  public boolean equals(Object o) {
    if (this == o) {
      return true;
    }
    if (o == null || getClass() != o.getClass()) {
      return false;
    }

    Draft_6455 that = (Draft_6455) o;

    if (maxFrameSize != that.getMaxFrameSize()) {
      return false;
    }
    if (negotiatedExtension != null ? !negotiatedExtension.equals(that.getExtension()) : that.getExtension() != null) {
      return false;
    }
    return protocol != null ? protocol.equals(that.getProtocol()) : that.getProtocol() == null;
  }

  @Override
  public int hashCode() {
    int result = negotiatedExtension != null ? negotiatedExtension.hashCode() : 0;
    result = 31 * result + (protocol != null ? protocol.hashCode() : 0);
    result = 31 * result + (maxFrameSize ^ (maxFrameSize >>> 32));
    return result;
  }

  /**
   * Method to generate a full bytebuffer out of all the fragmented frame payload
   *
   * @return a bytebuffer containing all the data
   * @throws LimitExceededException will be thrown when the totalSize is bigger then
   *                                Integer.MAX_VALUE due to not being able to allocate more
   */
  private ByteBuffer getPayloadFromByteBufferList() throws LimitExceededException {
    long totalSize = 0;
    ByteBuffer resultingByteBuffer;
    synchronized (byteBufferList) {
      for (ByteBuffer buffer : byteBufferList) {
        totalSize += buffer.limit();
      }
      checkBufferLimit();
      resultingByteBuffer = ByteBuffer.allocate((int) totalSize);
      for (ByteBuffer buffer : byteBufferList) {
        resultingByteBuffer.put(buffer);
      }
    }
    resultingByteBuffer.flip();
    return resultingByteBuffer;
  }

  /**
   * Get the current size of the resulting bytebuffer in the bytebuffer list
   *
   * @return the size as long (to not get an integer overflow)
   */
  private long getByteBufferListSize() {
    long totalSize = 0;
    synchronized (byteBufferList) {
      for (ByteBuffer buffer : byteBufferList) {
        totalSize += buffer.limit();
      }
    }
    return totalSize;
  }

  private class TranslatedPayloadMetaData {

    private int payloadLength;
    private int realPackageSize;

    private int getPayloadLength() {
      return payloadLength;
    }

    private int getRealPackageSize() {
      return realPackageSize;
    }

    TranslatedPayloadMetaData(int newPayloadLength, int newRealPackageSize) {
      this.payloadLength = newPayloadLength;
      this.realPackageSize = newRealPackageSize;
    }
  }
}