Search in sources :

Example 1 with ListenerRegistry

use of io.undertow.server.ListenerRegistry in project wildfly by wildfly.

the class HTTPUpgradeService method installService.

@SuppressWarnings("unchecked")
public static void installService(final CapabilityServiceTarget target, String activeMQServerName, final String acceptorName, final String httpListenerName) {
    final CapabilityServiceBuilder sb = target.addCapability(HTTPAcceptorDefinition.CAPABILITY);
    sb.provides(MessagingServices.getHttpUpgradeServiceName(activeMQServerName, acceptorName));
    Supplier<ChannelUpgradeHandler> upgradeSupplier = sb.requiresCapability(HTTP_UPGRADE_REGISTRY_CAPABILITY_NAME, ChannelUpgradeHandler.class, httpListenerName);
    Supplier<ListenerRegistry> listenerRegistrySupplier = sb.requiresCapability(HTTP_LISTENER_REGISTRY_CAPABILITY_NAME, ListenerRegistry.class);
    sb.requires(ActiveMQActivationService.getServiceName(MessagingServices.getActiveMQServiceName(activeMQServerName)));
    final HTTPUpgradeService service = new HTTPUpgradeService(activeMQServerName, acceptorName, httpListenerName, upgradeSupplier, listenerRegistrySupplier);
    sb.setInitialMode(ServiceController.Mode.PASSIVE);
    sb.setInstance(service);
    sb.install();
}
Also used : ListenerRegistry(io.undertow.server.ListenerRegistry) CapabilityServiceBuilder(org.jboss.as.controller.CapabilityServiceBuilder) ChannelUpgradeHandler(io.undertow.server.handlers.ChannelUpgradeHandler)

Example 2 with ListenerRegistry

use of io.undertow.server.ListenerRegistry 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);
            }
        }
    });
}
Also used : ListenerRegistry(io.undertow.server.ListenerRegistry) RemotingService(org.apache.activemq.artemis.core.remoting.server.RemotingService) IOException(java.io.IOException) HttpString(io.undertow.util.HttpString) NoSuchAlgorithmException(java.security.NoSuchAlgorithmException) HttpUpgradeHandshake(io.undertow.server.handlers.HttpUpgradeHandshake) HttpServerExchange(io.undertow.server.HttpServerExchange) ActiveMQServer(org.apache.activemq.artemis.core.server.ActiveMQServer) MessageDigest(java.security.MessageDigest)

Aggregations

ListenerRegistry (io.undertow.server.ListenerRegistry)2 HttpServerExchange (io.undertow.server.HttpServerExchange)1 ChannelUpgradeHandler (io.undertow.server.handlers.ChannelUpgradeHandler)1 HttpUpgradeHandshake (io.undertow.server.handlers.HttpUpgradeHandshake)1 HttpString (io.undertow.util.HttpString)1 IOException (java.io.IOException)1 MessageDigest (java.security.MessageDigest)1 NoSuchAlgorithmException (java.security.NoSuchAlgorithmException)1 RemotingService (org.apache.activemq.artemis.core.remoting.server.RemotingService)1 ActiveMQServer (org.apache.activemq.artemis.core.server.ActiveMQServer)1 CapabilityServiceBuilder (org.jboss.as.controller.CapabilityServiceBuilder)1