Examples with HttpUpgradeHandshake io.undertow.server.handlers.HttpUpgradeHandshake used on opensource projects

Example 1 with HttpUpgradeHandshake

use of io.undertow.server.handlers.HttpUpgradeHandshake in project wildfly by wildfly.

the class HTTPUpgradeService method start.

@Override
public void start(StartContext context) throws StartException {
    ListenerRegistry listenerRegistry = listenerRegistrySupplier.get();
    ListenerRegistry.Listener listenerInfo = listenerRegistry.getListener(httpListenerName);
    assert listenerInfo != null;
    httpUpgradeMetadata = new ListenerRegistry.HttpUpgradeMetadata(getProtocol(), CORE);
    listenerInfo.addHttpUpgradeMetadata(httpUpgradeMetadata);
    MessagingLogger.ROOT_LOGGER.registeredHTTPUpgradeHandler(ACTIVEMQ_REMOTING, acceptorName);
    ServiceController<?> activeMQService = context.getController().getServiceContainer().getService(MessagingServices.getActiveMQServiceName(activeMQServerName));
    ActiveMQServer activeMQServer = ActiveMQServer.class.cast(activeMQService.getValue());
    httpUpgradeListener = switchToMessagingProtocol(activeMQServer, acceptorName, getProtocol());
    upgradeSupplier.get().addProtocol(getProtocol(), httpUpgradeListener, new HttpUpgradeHandshake() {

        /**
         * override the default upgrade handshake to take into account the
         * {@code TransportConstants.HTTP_UPGRADE_ENDPOINT_PROP_NAME} header
         * to select the correct acceptors among all that are configured in ActiveMQ.
         *
         * If the request does not have this header, the first acceptor will be used.
         */
        @Override
        public boolean handleUpgrade(HttpServerExchange exchange) throws IOException {
            String secretKey = exchange.getRequestHeaders().getFirst(getSecKeyHeader());
            if (secretKey == null) {
                throw MessagingLogger.ROOT_LOGGER.upgradeRequestMissingKey();
            }
            ActiveMQServer server = selectServer(exchange, activeMQServer);
            if (server == null) {
                return false;
            }
            // If ActiveMQ remoting service is stopped (eg during shutdown), refuse
            // the handshake so that the ActiveMQ client will detect the connection has failed
            RemotingService remotingService = server.getRemotingService();
            if (!server.isActive() || !remotingService.isStarted()) {
                return false;
            }
            final String endpoint = exchange.getRequestHeaders().getFirst(getHttpUpgradeEndpointKey());
            if (endpoint == null || endpoint.equals(acceptorName)) {
                String response = createExpectedResponse(MAGIC_NUMBER, secretKey);
                exchange.getResponseHeaders().put(HttpString.tryFromString(getSecAcceptHeader()), response);
                return true;
            }
            return false;
        }

        private String createExpectedResponse(final String magicNumber, final String secretKey) throws IOException {
            try {
                final String concat = secretKey + magicNumber;
                final MessageDigest digest = MessageDigest.getInstance("SHA1");
                digest.update(concat.getBytes(StandardCharsets.UTF_8));
                final byte[] bytes = digest.digest();
                return FlexBase64.encodeString(bytes, false);
            } catch (NoSuchAlgorithmException e) {
                throw new IOException(e);
            }
        }
    });
}