Search in sources :

Example 1 with LimitExedeedException

use of org.java_websocket.exceptions.LimitExedeedException in project quorrabot by GloriousEggroll.

the class Draft_10 method translateSingleFrame.

public Framedata translateSingleFrame(ByteBuffer buffer) throws IncompleteException, InvalidDataException {
    int maxpacketsize = buffer.remaining();
    int realpacketsize = 2;
    if (maxpacketsize < realpacketsize) {
        throw new IncompleteException(realpacketsize);
    }
    byte b1 = buffer.get();
    boolean FIN = b1 >> 8 != 0;
    byte rsv = (byte) ((b1 & ~(byte) 128) >> 4);
    if (rsv != 0) {
        throw new InvalidFrameException("bad rsv " + rsv);
    }
    byte b2 = buffer.get();
    boolean MASK = (b2 & -128) != 0;
    int payloadlength = (byte) (b2 & ~(byte) 128);
    Opcode optcode = toOpcode((byte) (b1 & 15));
    if (!FIN) {
        if (optcode == Opcode.PING || optcode == Opcode.PONG || optcode == Opcode.CLOSING) {
            throw new InvalidFrameException("control frames may no be fragmented");
        }
    }
    if (payloadlength >= 0 && payloadlength <= 125) {
    } else {
        if (optcode == Opcode.PING || optcode == Opcode.PONG || optcode == Opcode.CLOSING) {
            throw new InvalidFrameException("more than 125 octets");
        }
        if (payloadlength == 126) {
            // additional length bytes
            realpacketsize += 2;
            if (maxpacketsize < realpacketsize) {
                throw new IncompleteException(realpacketsize);
            }
            byte[] sizebytes = new byte[3];
            sizebytes[1] = buffer.get();
            sizebytes[2] = buffer.get();
            payloadlength = new BigInteger(sizebytes).intValue();
        } else {
            // additional length bytes
            realpacketsize += 8;
            if (maxpacketsize < realpacketsize) {
                throw new IncompleteException(realpacketsize);
            }
            byte[] bytes = new byte[8];
            for (int i = 0; i < 8; i++) {
                bytes[i] = buffer.get();
            }
            long length = new BigInteger(bytes).longValue();
            if (length > Integer.MAX_VALUE) {
                throw new LimitExedeedException("Payloadsize is to big...");
            } else {
                payloadlength = (int) length;
            }
        }
    }
    // int maskskeystart = foff + realpacketsize;
    realpacketsize += (MASK ? 4 : 0);
    // int payloadstart = foff + realpacketsize;
    realpacketsize += payloadlength;
    if (maxpacketsize < realpacketsize) {
        throw new IncompleteException(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) ((byte) buffer.get() ^ (byte) maskskey[i % 4]));
        }
    } else {
        payload.put(buffer.array(), buffer.position(), payload.limit());
        buffer.position(buffer.position() + payload.limit());
    }
    FrameBuilder frame;
    if (optcode == Opcode.CLOSING) {
        frame = new CloseFrameBuilder();
    } else {
        frame = new FramedataImpl1();
        frame.setFin(FIN);
        frame.setOptcode(optcode);
    }
    payload.flip();
    frame.setPayload(payload);
    return frame;
}
Also used : Opcode(org.java_websocket.framing.Framedata.Opcode) CloseFrameBuilder(org.java_websocket.framing.CloseFrameBuilder) InvalidFrameException(org.java_websocket.exceptions.InvalidFrameException) ByteBuffer(java.nio.ByteBuffer) LimitExedeedException(org.java_websocket.exceptions.LimitExedeedException) BigInteger(java.math.BigInteger) CloseFrameBuilder(org.java_websocket.framing.CloseFrameBuilder) FrameBuilder(org.java_websocket.framing.FrameBuilder) FramedataImpl1(org.java_websocket.framing.FramedataImpl1)

Aggregations

BigInteger (java.math.BigInteger)1 ByteBuffer (java.nio.ByteBuffer)1 InvalidFrameException (org.java_websocket.exceptions.InvalidFrameException)1 LimitExedeedException (org.java_websocket.exceptions.LimitExedeedException)1 CloseFrameBuilder (org.java_websocket.framing.CloseFrameBuilder)1 FrameBuilder (org.java_websocket.framing.FrameBuilder)1 Opcode (org.java_websocket.framing.Framedata.Opcode)1 FramedataImpl1 (org.java_websocket.framing.FramedataImpl1)1