<!--JSR-303/JSR-380用于驗證的注解 -->
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-validation</artifactId>
    <version>2.6.7</version>
</dependency>

implementation 'org.springframework.boot:spring-boot-starter-validation:2.6.7'

package com.morris.spring.boot.module.xss;
import javax.validation.Constraint;
import javax.validation.Payload;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
@Target(value = {ElementType.METHOD, ElementType.FIELD, ElementType.CONSTRUCTOR, ElementType.PARAMETER})
@Retention(RetentionPolicy.RUNTIME)
@Constraint(validatedBy = XssValidator.class)
public @interface Xss {
    String message() default "非法輸入, 檢測到潛在的XSS";
    Class<?>[] groups() default {};
    Class<? extends Payload>[] payload() default {};
}

package com.morris.spring.boot.module.xss;
import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;
import org.jsoup.safety.Whitelist;
import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
/**
 * xss注解校驗
 */
public class XssValidator implements ConstraintValidator<Xss, String> {
    /**
     * 使用jsoup自帶的relaxed白名單
     */
    private static final Whitelist WHITE_LIST = Whitelist.relaxed();
    /**
     * 定義輸出設置，關閉prettyPrint（prettyPrint=false），目的是避免在清理過程中對代碼進行格式化
     * 從而保持輸入和輸出內容的一致性。
     */
    private static final Document.OutputSettings OUTPUT_SETTINGS = new Document.OutputSettings().prettyPrint(false);
    /**
     * 驗證輸入值是否有效，即是否包含潛在的XSS攻擊腳本。
     * 
     * @param value 輸入值，需要進行XSS攻擊腳本清理。
     * @param context 上下文對象，提供關于驗證環(huán)境的信息，如驗證失敗時的錯誤消息定制。
     * @return 如果清理后的值與原始值相同，則返回true，表示輸入值有效；否則返回false，表示輸入值無效。
     */
    @Override
    public boolean isValid(String value, ConstraintValidatorContext context) {
        // 這里先對value進行一次解析，對其中的單個<、>字符進行編碼
        String oldBody = Jsoup.parse(value).body().html();
        // 使用Jsoup庫對輸入值進行清理，以移除潛在的XSS攻擊腳本。
        // 使用預定義的白名單和輸出設置來確保只保留安全的HTML元素和屬性。
        String newBody = Jsoup.clean(value, "", WHITE_LIST, OUTPUT_SETTINGS);
        // 比較清理后的值與解析后的值是否相同，不相同說明過濾了xss腳本
        // 這里不使用value與newBody比，因為jsoup會對value中的單個<、>字符進行編碼，如果value中還有單個的<，value與newBody也會不一樣
        return oldBody.equals(newBody);
    }
}

package com.morris.spring.boot.module.xss;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import javax.validation.Valid;
/**
 * Xss局部防御post請求
 */
@RestController
@RequestMapping("/xss/local")
public class XssLocalPostController {
    /**
     * 使用注解攔截POST請求中的xss，在實體類需要攔截xss的屬性上面加上@Xss或者@Validated注解
     *
     * @param userLocalLoginPojo 實體類
     * @return 實體類
     */
    @PostMapping("/test")
    public UserLocalLoginPojo test(@Valid @RequestBody UserLocalLoginPojo userLocalLoginPojo) {
        return userLocalLoginPojo;
    }
}

package com.morris.spring.boot.module.xss;
import lombok.Data;
@Data
public class UserLocalLoginPojo {
    @Xss
    private String userAccount;
}

{
    "userAccount": "<iframe οnlοad='alert(0)'>demoData</iframe>"
}

package com.morris.spring.boot.module.xss;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
/**
 * Xss局部防御get請求
 */
@RestController
@RequestMapping("/xss/local")
@Validated
public class XssLocalGetController {
    /**
     * 使用注解攔截get請求中的xss，在方法參數前面加上@Xss，注意類上面要加上@Validated注解
     *
     * @param userAccount 請求參數
     * @return 請求參數
     */
    @GetMapping("/test")
    public String test(@Xss String userAccount) {
        return userAccount;
    }
}