在涉及跨系统接口调用时,我们容易碰到以下安全问题:
sa-token api-sign
模块将帮你轻松解决以上难题。(此插件是内嵌到 sa-token-core 核心包中的模块,开发者无需再次引入其它依赖,插件直接可用)
假设我们有如下业务需求:
用户在 A 系统参与活动成功后,活动奖励以余额的形式下发到 B 系统。
在不考虑安全问题的情况下,我们很容易完成这个需求:
1、在 B 系统开放一个接口
@RestController
@RequestMapping("/sign")
public class SignController {
@PostMapping("/addMoney")
public String addMoney(Long userId, Long money) {
// TODO 处理业务...
return "ADD SUCCESS";
}
}
2、在 A 系统使用 http 工具类调用这个接口
@RestController
@RequestMapping("/activity")
public class ActivityController {
@PostMapping("/join")
public String join() {
// 参加完活动后,发送余额
Long userId = 1L;
Long money = 100L;
Map<String, Object> params = new HashMap<>();
params.put("userId", userId);
params.put("money", money);
String url = "http://localhost:8079/sign/addMoney";
String result = HttpUtil.post(url, params);
return "join";
}
}
上述代码简单的完成了需求,但是很明显它有一个安全问题:
B 系统开放的接口不仅可以被 A 系统调用,还可以被其它任何人调用,甚至别人可以本地跑一个 for 循环调用这个接口,为自己无限充值金额
为防止 B 系统开放的接口被陌生人任意调用,我们增加一个 secretKey 参数
@PostMapping("/addMoney")
public String addMoney(Long userId, Long money, String secretKey) {
// 校验 secretKey
if (!check(secretKey)) {
throw new RuntimeException("无效 secretKey,无法响应请求");
}
// TODO 处理业务...
return "ADD SUCCESS";
}
由于 A 系统是我们 “自己人”,所以它可以拿着 secretKey 进行合法请求:
@PostMapping("/join")
public String join() {
// 参加完活动后,发送余额
Map<String, Object> params = new HashMap<>();
params.put("userId", userId);
params.put("money", money);
params.put("secretKey", "×××××××××××");
String url = "http://localhost:8079/sign/addMoney";
String result = HttpUtil.post(url, params);
return "join";
}
现在,即使 B 系统的接口被暴露了,也不会被陌生人任意调用了,安全性得到了一定的保证,但是仍然存在一些问题:
首先,在 A 系统不要直接发起请求,而是先计算一个 sign 参数:
@PostMapping("/join")
public String join() {
// 参加完活动后,发送余额
Long userId = 1L;
Long money = 100L;
String secretKey = "×××××××××××";
// 计算 sign
String sign = md5("money=" + money + "&userId=" + userId + "&key=" + secretKey);
Map<String, Object> params = new HashMap<>();
params.put("userId", userId);
params.put("money", money);
params.put("sign", sign);
String url = "http://localhost:8079/sign/addMoney";
String result = HttpUtil.post(url, params);
return "join";
}
注意:此处计算签名时,需要将所有参数按照字典顺序依次排列(key除外,挂在最后面)
然后在 B 系统接收请求时,使用同样的算法、同样的秘钥,生成 sign 字符串,与参数中 sign 值进行比较:
@PostMapping("/addMoney")
public String addMoney(Long userId, Long money, String sign) {
// 在 B 系统,使用同样的算法、同样的密钥,计算出 sign2,与传入的 sign 进行比对
String sign2 = md5("money=" + money + "&userId=" + userId + "&key=" + secretKey);
if (!sign2.equals(sign)) {
return "无效 sign,无法响应请求";
}
// TODO 处理业务...
return "ADD SUCCESS";
}
因为 sign 的值是由 userId、money、secretKey 三个参数共同决定的,所以只要有一个参数不一致,就会造成最终生成 sign 也是不一致的,所以,根据比对结果:
此方案优点:
此方案仍然存在以下缺陷:
@PostMapping("/join")
public String join() {
// 参加完活动后,发送余额
Long userId = 1L;
Long money = 100L;
String nonce = SaFoxUtil.getRandomString(32); // 随机32位字符串
String secretKey = "×××××××××××";
// 计算 sign
String sign = md5("money=" + money + "&nonce=" + nonce + "&userId=" + userId + "&key=" + secretKey);
Map<String, Object> params = new HashMap<>();
params.put("userId", userId);
params.put("money", money);
params.put("nonce", nonce);
params.put("sign", sign);
String url = "http://localhost:8079/sign/addMoney";
String result = HttpUtil.post(url, params);
return "join";
}
首先,在 A 系统发起调用前,追加一个 nonce 参数,一起参与到签名中:
public String join() {
// 参加完活动后,发送余额
Long userId = 1L;
Long money = 100L;
String nonce = SaFoxUtil.getRandomString(32); // 随机32位字符串
String secretKey = "×××××××××××";
// 计算 sign
String sign = md5("money=" + money + "&nonce=" + nonce + "&userId=" + userId + "&key=" + secretKey);
Map<String, Object> params = new HashMap<>();
params.put("userId", userId);
params.put("money", money);
params.put("nonce", nonce);
params.put("sign", sign);
String url = "http://localhost:8079/sign/addMoney";
String result = HttpUtil.post(url, params);
return "join";
}
然后在 B 系统接收请求时,也把 nonce 参数加进去生成 sign 字符串,进行比较:
public String addMoney(Long userId, Long money, String nonce,String sign) {
// 检查此 nonce 是否已被使用过了
if (Objects.nonNull(CacheUtil.get("nonce_" + nonce))) {
return "此 nonce 已被使用过了,请求无效";
}
// 在 B 系统,使用同样的算法、同样的密钥,计算出 sign2,与传入的 sign 进行比对
String sign2 = md5("money=" + money + "&nonce=" + nonce + "&userId=" + userId + "&key=" + secretKey);
if (!sign2.equals(sign)) {
return "无效 sign,无法响应请求";
}
// 存入缓存
CacheUtil.set("nonce_" + nonce, "1");
// TODO 处理业务...
return "ADD SUCCESS";
}
代码分析:
这两步的组合,保证了一个 nonce 随机字符串只能被使用一次,如果请求被抓包后重放,是无法通过 nonce 校验的。
至此,问题似乎已被解决了 …… 吗?
别急,我们还有一个问题没有考虑:这个 nonce 在字符串在缓存应该被保存多久呢?
你可能会想到,那我永久保存吧。这样确实能解决问题,但显然服务器承载不了这么做,即使再微小的数据量,在时间的累加下,也总一天会超出服务器能够承载的上限。
我们可以再追加一个 timestamp 时间戳参数,将请求的有效性限定在一个有限时间范围内,例如 15分钟。
首先,在 A 系统追加 timestamp 参数:
public String join() {
// 参加完活动后,发送余额
Long userId = 1L;
Long money = 100L;
Long timestamp = System.currentTimeMillis();
String nonce = SaFoxUtil.getRandomString(32); // 随机32位字符串
String secretKey = "×××××××××××";
// 计算 sign
String sign = md5("money=" + money + "&nonce=" + nonce + "×tamp=" + timestamp + "&userId=" + userId + "&key=" + secretKey);
Map<String, Object> params = new HashMap<>();
params.put("userId", userId);
params.put("money", money);
params.put("nonce", nonce);
params.put("timestamp", timestamp);
params.put("sign", sign);
String url = "http://localhost:8079/sign/addMoney";
String result = HttpUtil.post(url, params);
return "join";
}
在 B 系统检测这个 timestamp 是否超出了允许的范围
public String addMoney(Long userId, Long money, Long timestamp, String nonce,String sign) {
// 1、检查 timestamp 是否超出允许的范围(此处假定最大允许15分钟差距)
long timestampDisparity = System.currentTimeMillis() - timestamp; // 实际的时间差
if(timestampDisparity > 1000 * 60 * 15) {
return "timestamp 时间差超出允许的范围,请求无效";
}
// 检查此 nonce 是否已被使用过了
if (Objects.nonNull(CacheUtil.get("nonce_" + nonce))) {
return "此 nonce 已被使用过了,请求无效";
}
// 在 B 系统,使用同样的算法、同样的密钥,计算出 sign2,与传入的 sign 进行比对
String sign2 = md5("money=" + money + "&nonce=" + nonce + "&userId=" + userId + "&key=" + secretKey);
if (!sign2.equals(sign)) {
return "无效 sign,无法响应请求";
}
// 将 nonce 记入缓存,ttl 有效期和 allowDisparity 允许时间差一致
CacheUtil.set("nonce_" + nonce, "1", 1000 * 60 * 15);
// TODO 处理业务...
return "ADD SUCCESS";
}
至此,抓包者:
以上的代码,均假设 A 系统服务器与 B 系统服务器的时钟一致,才可以正常完成安全校验,但在实际的开发场景中,有些服务器会存在时钟不准确的问题。
假设 A 服务器与 B 服务器的时钟差异为 10 分钟,即:在 A 服务器为 8:00 的时候,B 服务器为 7:50。
要解决上述问题,有两种方案:
要采用方案一的同学可自行搜索一下同步时钟的方法,在此暂不赘述,此处详细阐述一下方案二。
我们只需简单修改一下,B 系统校验参数的代码即可:
public String addMoney(Long userId, Long money, Long timestamp, String nonce,String sign) {
// 1、检查 timestamp 是否超出允许的范围 (重点一:此处需要取绝对值)
long timestampDisparity = Math.abs(System.currentTimeMillis() - timestamp);
if(timestampDisparity > 1000 * 60 * 15) {
return "timestamp 时间差超出允许的范围,请求无效";
}
// 检查此 nonce 是否已被使用过了
if (Objects.nonNull(CacheUtil.get("nonce_" + nonce))) {
return "此 nonce 已被使用过了,请求无效";
}
// 在 B 系统,使用同样的算法、同样的密钥,计算出 sign2,与传入的 sign 进行比对
String sign2 = md5("money=" + money + "&nonce=" + nonce + "&userId=" + userId + "&key=" + secretKey);
if (!sign2.equals(sign)) {
return "无效 sign,无法响应请求";
}
// 将 nonce 记入缓存,防止重复使用(重点二:此处需要将 ttl 设定为允许 timestamp 时间差的值 x 2 )
CacheUtil.set("nonce_" + nonce, "1", (1000 * 60 * 15) * 2;
// TODO 处理业务...
return "ADD SUCCESS";
}
接下来步入正题,使用 Sa-Token 内置的 sign 模块,方便的完成 API 签名创建、校验等步骤:
api-sign 模块已内嵌到核心包,只需要引入 sa-token 本身依赖即可:(请求发起端和接收端都需要引入)
<dependency>
<groupId>cn.dev33groupId>
<artifactId>sa-token-spring-boot-starterartifactId>
<version>1.35.0.RCversion>
dependency>
请求发起端和接收端需要配置一个相同的秘钥,在 application.yml 中配置:
sa-token:
sign:
# API 接口签名秘钥 (随便乱摁几个字母即可)
secret-key: kQwIOrYvnXmSDkwEiFngrKidMcdrgKor
public String join() {
// 参加完活动后,发送余额
Long userId = 1L;
Long money = 100L;
Map<String, Object> params = new HashMap<>();
params.put("userId", userId);
params.put("money", money);
SaSignUtil.addSignParamsAndJoin(params);
String url = "http://localhost:8079/sign/addMoney";
return HttpUtil.post(url, params);
}
public String addMoney(Long userId, Long money) {
// 1、校验请求中的签名
SaSignUtil.checkRequest(SaHolder.getRequest());
// 2、校验通过,处理业务
System.out.println("userId=" + userId);
System.out.println("money=" + money);
return "ADD SUCCESS";
}
如上代码便可简单方便的完成 API 接口参数签名校验,当请求端的秘钥不对,或者请求参数被篡改、请求被重放时,均无法通过 SaSignUtil.checkRequest 校验
SaSignUtil#addSignParamsAndJoin(params);
:
public static String addSignParamsAndJoin(Map<String, Object> paramsMap) {
return SaManager.getSaSignTemplate().addSignParamsAndJoin(paramsMap);
}
会调用 SaSignTemplate
类中的方法
SaSignTemplate#addSignParamsAndJoin() 方法
public String addSignParamsAndJoin(Map<String, Object> paramsMap) {
// 1.添加参数:timestamp、nonce、sign
paramsMap = this.addSignParams(paramsMap);
// 2.将 map 使用 & 转化为String
return this.joinParams(paramsMap);
}
这个方法有两个逻辑:
SaSignTemplate#addSignParams() 方法
public Map<String, Object> addSignParams(Map<String, Object> paramsMap) {
paramsMap.put(timestamp, String.valueOf(System.currentTimeMillis()));
paramsMap.put(nonce, SaFoxUtil.getRandomString(32));
paramsMap.put(sign, this.createSign(paramsMap));
return paramsMap;
}
SaSignTemplate#createSign() 方法
:生成签名
public String createSign(Map<String, ?> paramsMap) {
String secretKey = this.getSecretKey();
SaSignException.throwByNull(secretKey, "参与参数签名的秘钥不可为空", 12201);
if (((Map)paramsMap).containsKey(sign)) {
paramsMap = new TreeMap((Map)paramsMap);
((Map)paramsMap).remove(sign);
}
// 按照数据字典进行排序,并将 map 使用 & 转化为String
String paramsStr = this.joinParamsDictSort((Map)paramsMap);
String fullStr = paramsStr + "&" + key + "=" + secretKey;
// md5
return this.abstractStr(fullStr);
}
public String abstractStr(String fullStr) {
return SaSecureUtil.md5(fullStr);
}
这个方法有两个逻辑:
SaSignUtil.checkRequest(SaHolder.getRequest());
:
public static void checkRequest(SaRequest request) {
SaManager.getSaSignTemplate().checkRequest(request);
}
还是会调用 SaSignTemplate
类中的方法
SaSignTemplate#checkParamMap() 方法
:校验请求参数
public void checkRequest(SaRequest request) {
this.checkParamMap(request.getParamMap());
}
public void checkParamMap(Map<String, String> paramMap) {
String timestampValue = (String)paramMap.get(timestamp);
String nonceValue = (String)paramMap.get(nonce);
String signValue = (String)paramMap.get(sign);
// 1.校验时间戳
this.checkTimestamp(Long.parseLong(timestampValue));
// 2.校验随机数
if (this.getSignConfigOrGlobal().getIsCheckNonce()) {
this.checkNonce(nonceValue);
}
// 3.校验签名
this.checkSign(paramMap, signValue);
}
这个方法有三个逻辑:
SaSignTemplate#checkNonce() 方法:
校验随机数
public void checkNonce(String nonce) {
if (SaFoxUtil.isEmpty(nonce)) {
throw new SaSignException("nonce 为空,无效");
} else {
String key = this.splicingNonceSaveKey(nonce);
if (SaManager.getSaTokenDao().get(key) != null) {
throw new SaSignException("此 nonce 已被使用过,不可重复使用:" + nonce);
} else {
SaManager.getSaTokenDao().set(key, nonce, this.getSignConfigOrGlobal().getSaveNonceExpire() * 2L + 2L);
}
}
}
SaTokenDao
是存储接口,默认实现是用的是 SaTokenDaoDefaultImpl
。SaTokenDaoDefaultImpl
存储数据,主要是通过 ConcurrentHashMap
存放在本地内存中。
SaManager#getSaTokenDao() 方法:
public static SaTokenDao getSaTokenDao() {
if (saTokenDao == null) {
Class var0 = SaManager.class;
synchronized(SaManager.class) {
if (saTokenDao == null) {
setSaTokenDaoMethod(new SaTokenDaoDefaultImpl());
}
}
}
return saTokenDao;
}
SaTokenDaoDefaultImpl
:
public class SaTokenDaoDefaultImpl implements SaTokenDao {
// 数据集合
public Map<String, Object> dataMap = new ConcurrentHashMap();
// 过期时间集合 (单位: 毫秒) , 记录所有key的到期时间 [注意不是剩余存活时间]
public Map<String, Long> expireMap = new ConcurrentHashMap();
public Thread refreshThread;
public volatile boolean refreshFlag;
public SaTokenDaoDefaultImpl() {
// 定时清理过期数据
this.initRefreshThread();
}
public String get(String key) {
this.clearKeyByTimeout(key);
return (String)this.dataMap.get(key);
}
public void set(String key, String value, long timeout) {
if (timeout != 0L && timeout > -2L) {
this.dataMap.put(key, value);
this.expireMap.put(key, timeout == -1L ? -1L : System.currentTimeMillis() + timeout * 1000L);
}
}
public void initRefreshThread() {
if (SaManager.getConfig().getDataRefreshPeriod() > 0) {
this.refreshFlag = true;
this.refreshThread = new Thread(() -> {
while(true) {
try {
try {
if (!this.refreshFlag) {
return;
}
this.refreshDataMap();
} catch (Exception var2) {
var2.printStackTrace();
}
int dataRefreshPeriod = SaManager.getConfig().getDataRefreshPeriod();
if (dataRefreshPeriod <= 0) {
dataRefreshPeriod = 1;
}
Thread.sleep((long)dataRefreshPeriod * 1000L);
} catch (Exception var3) {
var3.printStackTrace();
}
}
});
this.refreshThread.start();
}
}
}
如果仅仅存放在本地内存中,涉及到多个项目,可能数据无法共享。
引入仓库 sa-token-dao-redis-jackson
<!-- Sa-Token 整合 Redis (使用 jackson 序列化方式) -->
<dependency>
<groupId>cn.dev33</groupId>
<artifactId>sa-token-redis-jackson</artifactId>
<version>1.35.0.RC</version>
</dependency>
<!-- 提供Redis连接池 -->
<dependency>
<groupId>org.apache.commons</groupId>
<artifactId>commons-pool2</artifactId>
</dependency>
SaTokenDaoRedisJackson
使用 Redis 作为存储数据的地方
SaBeanInject#setSaTokenDao
,SaBeanInject
是自动配置的。当系统中存在 SaTokenDao
的 Bean 实例,则设置SaTokenDao
实例
public class SaBeanInject {
@Autowired(
required = false
)
public void setSaTokenDao(SaTokenDao saTokenDao) {
SaManager.setSaTokenDao(saTokenDao);
}
}
参考:
API 接口参数签名
【开源项目】使用Sa-Token框架完成API参数签名