springboot實現(xiàn)增加黑名單和白名單功能

更新時間：2024年05月08日 10:39:48 作者：nanshaws

本文主要介紹了springboot實現(xiàn)增加黑名單和白名單功能,就是單純的實現(xiàn)filter,然后注冊到springboot里面,在filter里面進行黑白名單的篩選,感興趣的可以了解一下

題外話

關(guān)于黑名單和白名單功能，我覺得可以直接用linux服務(wù)器的iptables或nftables來實現(xiàn)黑名單和白名單功能。這兩個工具都是Linux系統(tǒng)上用于配置防火墻規(guī)則的命令行工具。

iptables：

描述： iptables 是一個用于配置IPv4數(shù)據(jù)包過濾規(guī)則的工具。它是Linux系統(tǒng)上最常用的防火墻工具之一。

命令示例：

添加黑名單規(guī)則：sudo iptables -A INPUT -s <黑名單IP地址> -j DROP
添加白名單規(guī)則：sudo iptables -A INPUT -s <白名單IP地址> -j ACCEPT

nftables：

描述： nftables 是一個用于配置網(wǎng)絡(luò)過濾和分類規(guī)則的框架，可以替代iptables。它支持IPv4和IPv6，并提供更靈活的語法。

命令示例：

添加黑名單規(guī)則：sudo nft add rule ip filter input ip saddr <黑名單IP地址> drop
添加白名單規(guī)則：sudo nft add rule ip filter input ip saddr <白名單IP地址> accept

在選擇使用哪個工具時，您可以根據(jù)您的偏好以及系統(tǒng)是否已經(jīng)采用nftables來進行選擇。新er版本的Linux系統(tǒng)通常更傾向于使用nftables。請注意，上述命令中的<黑名單IP地址>和<白名單IP地址>需要替換為實際的IP地址。

無論您選擇使用iptables還是nftables，都需要小心配置規(guī)則，以確保不會阻止您訪問服務(wù)器。在配置之前，請確保您了解防火墻規(guī)則的工作原理和影響。

或者像openai一樣使用cf防護也就是CloudFlare作為其服務(wù)的安全防護層。

這里就不講那么多題外話，開始我們的springboot實現(xiàn)黑名單和白名單功能

先講一下springboot實現(xiàn)黑白名單的原理，很簡單，就是單純的實現(xiàn)filter,然后注冊到springboot里面，在filter里面進行黑白名單的篩選

第一步：創(chuàng)建一個springboot項目

pom文件的依賴：

<dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>

        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
            <scope>test</scope>
        </dependency>
    </dependencies>

注意：由于我使用的是springboot3，它內(nèi)置的tomcat是10及其以上的。所以引入的依賴是jakarta，可能你們版本比較低的，可以使用javax

第二步：創(chuàng)建一個過濾器類

package org.cyl.test.fliter;
import jakarta.servlet.*;
import java.io.IOException;
import java.util.Arrays;
import java.util.List;

public class BlackWhiteListFilter implements Filter {

    private List<String> blackList = Arrays.asList("blocked-ip1", "blocked-ip2");
    private List<String> whiteList = Arrays.asList("127.0.0.1", "0:0:0:0:0:0:0:1");

    @Override
    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain)
            throws IOException, ServletException {
        String clientIP = request.getRemoteAddr();

        // 檢查黑名單
        if (blackList.contains(clientIP)) {
            response.getWriter().write("Your IP is blacklisted.");
            return;
        }

        // 檢查白名單
        if (!whiteList.contains(clientIP)) {
            response.getWriter().write("Your IP is not whitelisted.");
            return;
        }

        chain.doFilter(request, response);
    }

    // 可以實現(xiàn) init 和 destroy 方法，根據(jù)需要進行處理
}

第三步：注冊config類：

package org.cyl.test.config;
import org.cyl.test.fliter.BlackWhiteListFilter;
import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class WebConfig {

    @Bean
    public FilterRegistrationBean<BlackWhiteListFilter> myFilter() {
        FilterRegistrationBean<BlackWhiteListFilter> registrationBean = new FilterRegistrationBean<>();
        registrationBean.setFilter(new BlackWhiteListFilter());
        registrationBean.addUrlPatterns("/*");
        return registrationBean;
    }
}

第四步：開始測試

在這里面編寫白名單和黑名單。由于每次請求都會經(jīng)過Filter，所以在這里面可以實現(xiàn)除了我設(shè)置的127.0.0.1和0:0:0:0:0:0:0:1之外，其他ip無法訪問的情況。

編寫controller類

package org.cyl.test.controller;

import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
public class HelloController {

    @GetMapping("/allowed-endpoint")
    public String sayHello(){
        return "Hello";
    }
}

訪問localhost:8080/allowed-endpoint

結(jié)果如下：