112 lines
4.6 KiB
Java
112 lines
4.6 KiB
Java
package ru.serega6531.packmate.configuration;
|
|
|
|
import lombok.extern.slf4j.Slf4j;
|
|
import org.pcap4j.core.PcapNativeException;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.beans.factory.annotation.Value;
|
|
import org.springframework.context.annotation.Bean;
|
|
import org.springframework.context.annotation.Configuration;
|
|
import org.springframework.context.event.EventListener;
|
|
import org.springframework.scheduling.annotation.EnableAsync;
|
|
import org.springframework.scheduling.annotation.EnableScheduling;
|
|
import org.springframework.security.authentication.event.AuthenticationFailureBadCredentialsEvent;
|
|
import org.springframework.security.config.annotation.authentication.builders.AuthenticationManagerBuilder;
|
|
import org.springframework.security.config.annotation.web.builders.HttpSecurity;
|
|
import org.springframework.security.config.annotation.web.configuration.EnableWebSecurity;
|
|
import org.springframework.security.config.annotation.web.configuration.WebSecurityConfigurerAdapter;
|
|
import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;
|
|
import org.springframework.security.crypto.password.PasswordEncoder;
|
|
import org.springframework.web.socket.config.annotation.EnableWebSocket;
|
|
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
|
|
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;
|
|
import ru.serega6531.packmate.WebSocketHandler;
|
|
import ru.serega6531.packmate.model.enums.CaptureMode;
|
|
import ru.serega6531.packmate.pcap.FilePcapWorker;
|
|
import ru.serega6531.packmate.pcap.LivePcapWorker;
|
|
import ru.serega6531.packmate.pcap.NoOpPcapWorker;
|
|
import ru.serega6531.packmate.pcap.PcapWorker;
|
|
import ru.serega6531.packmate.service.ServicesService;
|
|
import ru.serega6531.packmate.service.StreamService;
|
|
import ru.serega6531.packmate.service.SubscriptionService;
|
|
|
|
import java.net.UnknownHostException;
|
|
|
|
@Configuration
|
|
@EnableWebSecurity
|
|
@EnableScheduling
|
|
@EnableWebSocket
|
|
@EnableAsync
|
|
@Slf4j
|
|
public class ApplicationConfiguration extends WebSecurityConfigurerAdapter implements WebSocketConfigurer {
|
|
|
|
@Value("${account-login}")
|
|
private String login;
|
|
|
|
@Value("${account-password}")
|
|
private String password;
|
|
|
|
private final WebSocketHandler webSocketHandler;
|
|
|
|
@Autowired
|
|
public ApplicationConfiguration(WebSocketHandler webSocketHandler) {
|
|
this.webSocketHandler = webSocketHandler;
|
|
}
|
|
|
|
@Bean(destroyMethod = "stop")
|
|
@Autowired
|
|
public PcapWorker pcapWorker(ServicesService servicesService,
|
|
StreamService streamService,
|
|
SubscriptionService subscriptionService,
|
|
@Value("${local-ip}") String localIpString,
|
|
@Value("${interface-name}") String interfaceName,
|
|
@Value("${pcap-file}") String filename,
|
|
@Value("${capture-mode}") CaptureMode captureMode) throws PcapNativeException, UnknownHostException {
|
|
return switch (captureMode) {
|
|
case LIVE -> new LivePcapWorker(servicesService, streamService, localIpString, interfaceName);
|
|
case FILE -> new FilePcapWorker(servicesService, streamService, subscriptionService, localIpString, filename);
|
|
default -> new NoOpPcapWorker();
|
|
};
|
|
}
|
|
|
|
@Autowired
|
|
public void configureGlobal(AuthenticationManagerBuilder auth) throws Exception {
|
|
auth.inMemoryAuthentication()
|
|
.withUser(login)
|
|
.password(passwordEncoder().encode(password))
|
|
.authorities("ROLE_USER");
|
|
}
|
|
|
|
@Override
|
|
protected void configure(HttpSecurity http) throws Exception {
|
|
http.csrf()
|
|
.disable()
|
|
.authorizeRequests()
|
|
.antMatchers("/site.webmanifest")
|
|
.permitAll()
|
|
.anyRequest().authenticated()
|
|
.and()
|
|
.httpBasic()
|
|
.and()
|
|
.headers()
|
|
.frameOptions()
|
|
.sameOrigin();
|
|
}
|
|
|
|
@Bean
|
|
public PasswordEncoder passwordEncoder() {
|
|
return new BCryptPasswordEncoder();
|
|
}
|
|
|
|
@EventListener
|
|
public void authenticationFailed(AuthenticationFailureBadCredentialsEvent e) {
|
|
log.info("Login failed for user {}, password {}",
|
|
e.getAuthentication().getPrincipal(), e.getAuthentication().getCredentials());
|
|
}
|
|
|
|
@Override
|
|
public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
|
|
registry.addHandler(webSocketHandler, "/api/ws")
|
|
.withSockJS();
|
|
}
|
|
}
|