Sample configuration for Security of web sockets In Spring - Tekraze

webSocketsSecurityConfiguration.java

package com.tekraze.kafka.config;

import com.tekraze.kafka.security.AuthoritiesConstants;

import org.springframework.context.annotation.Configuration;
import org.springframework.messaging.simp.SimpMessageType;
import org.springframework.security.config.annotation.web.messaging.MessageSecurityMetadataSourceRegistry;
import org.springframework.security.config.annotation.web.socket.AbstractSecurityWebSocketMessageBrokerConfigurer;

@Configuration
public class WebsocketSecurityConfiguration extends AbstractSecurityWebSocketMessageBrokerConfigurer 

// TO set authentication for endpoints, you can also add endpoint for public also, if you require
  @Override
  protected void configureInbound(MessageSecurityMetadataSourceRegistry messages) {
    messages
    .nullDestMatcher().authenticated()
    .simpDestMatchers("/topic/tracker").hasAuthority(AuthoritiesConstants.ADMIN)
    // matches any destination that starts with /topic/
    // (i.e. cannot send messages directly to /topic/)
    // (i.e. cannot subscribe to /topic/messages/* to get messages sent to
    // /topic/messages-user<id>)
    .simpDestMatchers("/topic/**").authenticated()
    // message types other than MESSAGE and SUBSCRIBE
    .simpTypeMatchers(SimpMessageType.MESSAGE, SimpMessageType.SUBSCRIBE).denyAll()
    // catch all
    .anyMessage().denyAll();
  }

  /**
  * Disables CSRF for Websockets.
  */
  @Override
  protected boolean sameOriginDisabled() {
    return true;
  }
}