Examples with CloseReason jakarta.websocket.CloseReason used on opensource projects

Example 11 with CloseReason

use of jakarta.websocket.CloseReason in project spring-framework by spring-projects.

the class StandardWebSocketHandlerAdapterTests method onClose.

@Test
public void onClose() throws Throwable {
    this.adapter.onClose(this.session, new CloseReason(CloseCodes.NORMAL_CLOSURE, "reason"));
    verify(this.webSocketHandler).afterConnectionClosed(this.webSocketSession, CloseStatus.NORMAL.withReason("reason"));
}

Example 12 with CloseReason

use of jakarta.websocket.CloseReason in project tomcat by apache.

the class WsSession method checkExpiration.

protected void checkExpiration() {
    // Local copies to ensure consistent behaviour during method execution
    long timeout = maxIdleTimeout;
    long timeoutRead = getMaxIdleTimeoutRead();
    long timeoutWrite = getMaxIdleTimeoutWrite();
    long currentTime = System.currentTimeMillis();
    String key = null;
    if (timeoutRead > 0 && (currentTime - lastActiveRead) > timeoutRead) {
        key = "wsSession.timeoutRead";
    } else if (timeoutWrite > 0 && (currentTime - lastActiveWrite) > timeoutWrite) {
        key = "wsSession.timeoutWrite";
    } else if (timeout > 0 && (currentTime - lastActiveRead) > timeout && (currentTime - lastActiveWrite) > timeout) {
        key = "wsSession.timeout";
    }
    if (key != null) {
        String msg = sm.getString(key, getId());
        if (log.isDebugEnabled()) {
            log.debug(msg);
        }
        doClose(new CloseReason(CloseCodes.GOING_AWAY, msg), new CloseReason(CloseCodes.CLOSED_ABNORMALLY, msg));
    }
}

Example 13 with CloseReason

use of jakarta.websocket.CloseReason in project tomcat by apache.

the class WsWebSocketContainer method destroy.

/**
 * Cleans up the resources still in use by WebSocket sessions created from
 * this container. This includes closing sessions and cancelling
 * {@link Future}s associated with blocking read/writes.
 */
public void destroy() {
    CloseReason cr = new CloseReason(CloseCodes.GOING_AWAY, sm.getString("wsWebSocketContainer.shutdown"));
    for (WsSession session : sessions.keySet()) {
        try {
            session.close(cr);
        } catch (IOException ioe) {
            log.debug(sm.getString("wsWebSocketContainer.sessionCloseFail", session.getId()), ioe);
        }
    }
    // registered with it
    if (asynchronousChannelGroup != null) {
        synchronized (asynchronousChannelGroupLock) {
            if (asynchronousChannelGroup != null) {
                AsyncChannelGroupUtil.unregister();
                asynchronousChannelGroup = null;
            }
        }
    }
}

Example 14 with CloseReason

use of jakarta.websocket.CloseReason in project tomcat by apache.

the class WsRemoteEndpointImplBase method sendMessageBlock.

private void sendMessageBlock(byte opCode, ByteBuffer payload, boolean last, long timeoutExpiry) throws IOException {
    wsSession.updateLastActiveWrite();
    BlockingSendHandler bsh = new BlockingSendHandler();
    List<MessagePart> messageParts = new ArrayList<>();
    messageParts.add(new MessagePart(last, 0, opCode, payload, bsh, bsh, timeoutExpiry));
    messageParts = transformation.sendMessagePart(messageParts);
    // return.
    if (messageParts.size() == 0) {
        return;
    }
    long timeout = timeoutExpiry - System.currentTimeMillis();
    try {
        if (!messagePartInProgress.tryAcquire(timeout, TimeUnit.MILLISECONDS)) {
            String msg = sm.getString("wsRemoteEndpoint.acquireTimeout");
            wsSession.doClose(new CloseReason(CloseCodes.GOING_AWAY, msg), new CloseReason(CloseCodes.CLOSED_ABNORMALLY, msg), true);
            throw new SocketTimeoutException(msg);
        }
    } catch (InterruptedException e) {
        String msg = sm.getString("wsRemoteEndpoint.sendInterrupt");
        wsSession.doClose(new CloseReason(CloseCodes.GOING_AWAY, msg), new CloseReason(CloseCodes.CLOSED_ABNORMALLY, msg), true);
        throw new IOException(msg, e);
    }
    for (MessagePart mp : messageParts) {
        try {
            writeMessagePart(mp);
        } catch (Throwable t) {
            ExceptionUtils.handleThrowable(t);
            messagePartInProgress.release();
            wsSession.doClose(new CloseReason(CloseCodes.GOING_AWAY, t.getMessage()), new CloseReason(CloseCodes.CLOSED_ABNORMALLY, t.getMessage()), true);
            throw t;
        }
        if (!bsh.getSendResult().isOK()) {
            messagePartInProgress.release();
            Throwable t = bsh.getSendResult().getException();
            wsSession.doClose(new CloseReason(CloseCodes.GOING_AWAY, t.getMessage()), new CloseReason(CloseCodes.CLOSED_ABNORMALLY, t.getMessage()), true);
            throw new IOException(t);
        }
        // The BlockingSendHandler doesn't call end message so update the
        // flags.
        fragmented = nextFragmented;
        text = nextText;
    }
    if (payload != null) {
        payload.clear();
    }
    endMessage(null, null);
}

Example 15 with CloseReason

use of jakarta.websocket.CloseReason in project tomcat by apache.

the class WsFrameBase method processRemainingHeader.

/**
 * @return <code>true</code> if sufficient data was present to complete the
 *         processing of the header
 */
private boolean processRemainingHeader() throws IOException {
    // Ignore the 2 bytes already read. 4 for the mask
    int headerLength;
    if (isMasked()) {
        headerLength = 4;
    } else {
        headerLength = 0;
    }
    // Add additional bytes depending on length
    if (payloadLength == 126) {
        headerLength += 2;
    } else if (payloadLength == 127) {
        headerLength += 8;
    }
    if (inputBuffer.remaining() < headerLength) {
        return false;
    }
    // Calculate new payload length if necessary
    if (payloadLength == 126) {
        payloadLength = byteArrayToLong(inputBuffer.array(), inputBuffer.arrayOffset() + inputBuffer.position(), 2);
        inputBuffer.position(inputBuffer.position() + 2);
    } else if (payloadLength == 127) {
        payloadLength = byteArrayToLong(inputBuffer.array(), inputBuffer.arrayOffset() + inputBuffer.position(), 8);
        // the resulting payload length will be negative so test for that.
        if (payloadLength < 0) {
            throw new WsIOException(new CloseReason(CloseCodes.PROTOCOL_ERROR, sm.getString("wsFrame.payloadMsbInvalid")));
        }
        inputBuffer.position(inputBuffer.position() + 8);
    }
    if (Util.isControl(opCode)) {
        if (payloadLength > 125) {
            throw new WsIOException(new CloseReason(CloseCodes.PROTOCOL_ERROR, sm.getString("wsFrame.controlPayloadTooBig", Long.valueOf(payloadLength))));
        }
        if (!fin) {
            throw new WsIOException(new CloseReason(CloseCodes.PROTOCOL_ERROR, sm.getString("wsFrame.controlNoFin")));
        }
    }
    if (isMasked()) {
        inputBuffer.get(mask, 0, 4);
    }
    state = State.DATA;
    return true;
}