fix(protection): 修复HTTP接口签名 API重复请求问题

- 在 ApiSignatureAspect 中添加了对重复请求的检查逻辑
- 修改 ApiSignatureRedisDAO 中的 setNonce 方法,使用 setIfAbsent 代替 set
- 优化了日志记录,增加了重复请求的相关信息
pull/171/head
1351515658@qq.com 2025-02-24 19:33:12 +08:00
parent c2de5d9c8c
commit e9ae4196e6
2 changed files with 13 additions and 8 deletions

View File

@ -6,6 +6,7 @@ import cn.hutool.core.util.ObjUtil;
import cn.hutool.core.util.StrUtil; import cn.hutool.core.util.StrUtil;
import cn.hutool.crypto.digest.DigestUtil; import cn.hutool.crypto.digest.DigestUtil;
import cn.iocoder.yudao.framework.common.exception.ServiceException; import cn.iocoder.yudao.framework.common.exception.ServiceException;
import cn.iocoder.yudao.framework.common.exception.enums.GlobalErrorCodeConstants;
import cn.iocoder.yudao.framework.common.util.servlet.ServletUtils; import cn.iocoder.yudao.framework.common.util.servlet.ServletUtils;
import cn.iocoder.yudao.framework.signature.core.annotation.ApiSignature; import cn.iocoder.yudao.framework.signature.core.annotation.ApiSignature;
import cn.iocoder.yudao.framework.signature.core.redis.ApiSignatureRedisDAO; import cn.iocoder.yudao.framework.signature.core.redis.ApiSignatureRedisDAO;
@ -69,13 +70,17 @@ public class ApiSignatureAspect {
// 3. 将 nonce 记入缓存,防止重复使用(重点二:此处需要将 ttl 设定为允许 timestamp 时间差的值 x 2 // 3. 将 nonce 记入缓存,防止重复使用(重点二:此处需要将 ttl 设定为允许 timestamp 时间差的值 x 2
String nonce = request.getHeader(signature.nonce()); String nonce = request.getHeader(signature.nonce());
signatureRedisDAO.setNonce(appId, nonce, signature.timeout() * 2, signature.timeUnit()); if (!signatureRedisDAO.setNonce(appId, nonce, signature.timeout() * 2, signature.timeUnit())) {
String timestamp = request.getHeader(signature.timestamp());
log.info("[verifySignature][appId({}) timestamp({}) nonce({}) sign({}) 存在重复请求]", appId, timestamp, nonce, clientSignature);
throw new ServiceException(GlobalErrorCodeConstants.REPEATED_REQUESTS.getCode(), "存在重复请求");
}
return true; return true;
} }
/** /**
* *
* * <p>
* 1. appId * 1. appId
* 2. timestamp 10 * 2. timestamp 10
* 3. nonce 10 访 * 3. nonce 10 访
@ -118,7 +123,7 @@ public class ApiSignatureAspect {
/** /**
* *
* * <p>
* = + + + * = + + +
* *
* @param signature signature * @param signature signature
@ -139,7 +144,7 @@ public class ApiSignatureAspect {
/** /**
* Map * Map
* *
* @param request * @param request
* @param signature * @param signature
* @return signature params * @return signature params
*/ */

View File

@ -17,7 +17,7 @@ public class ApiSignatureRedisDAO {
/** /**
* *
* * <p>
* KEY signature_nonce:%s // 参数为 随机数 * KEY signature_nonce:%s // 参数为 随机数
* VALUE String * VALUE String
* *
@ -26,7 +26,7 @@ public class ApiSignatureRedisDAO {
/** /**
* *
* * <p>
* HASH * HASH
* KEY %s // 参数为 appid * KEY %s // 参数为 appid
* VALUE String * VALUE String
@ -40,8 +40,8 @@ public class ApiSignatureRedisDAO {
return stringRedisTemplate.opsForValue().get(formatNonceKey(appId, nonce)); return stringRedisTemplate.opsForValue().get(formatNonceKey(appId, nonce));
} }
public void setNonce(String appId, String nonce, int time, TimeUnit timeUnit) { public Boolean setNonce(String appId, String nonce, int time, TimeUnit timeUnit) {
stringRedisTemplate.opsForValue().set(formatNonceKey(appId, nonce), "", time, timeUnit); return stringRedisTemplate.opsForValue().setIfAbsent(formatNonceKey(appId, nonce), "", time, timeUnit);
} }
private static String formatNonceKey(String appId, String nonce) { private static String formatNonceKey(String appId, String nonce) {