【Sa-Token】SpringBoot 整合 Sa-Token 快速实现 API 接口签名安全校验-Toy模板网

这篇具有很好参考价值的文章主要介绍了【Sa-Token】SpringBoot 整合 Sa-Token 快速实现 API 接口签名安全校验。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

在涉及跨系统接口调用时，我们容易碰到以下安全问题：

请求身份被伪造
请求参数被篡改
请求被抓包，然后重放攻击

sa-token api-sign 模块将帮你轻松解决以上难题。（此插件是内嵌到 sa-token-core 核心包中的模块，开发者无需再次引入其它依赖，插件直接可用）

假设我们有如下业务需求：

用户在 A 系统参与活动成功后，活动奖励以余额的形式下发到 B 系统。

1. 初始方案：直接裸奔

在不考虑安全问题的情况下，我们很容易完成这个需求：

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 循环调用这个接口，为自己无限充值金额

2. 方案升级：增加 secretKey 校验

为防止 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 系统的接口被暴露了，也不会被陌生人任意调用了，安全性得到了一定的保证，但是仍然存在一些问题：

如果请求被抓包，secretKey 就会泄露，因为每次请求都在 url 中明文传输了 secretKey 参数。
如果请求被抓包，请求的其它参数就可以被任意修改，例如可以将 money 参数修改为 9999999，B系统无法确定参数是否被修改过。

3.方案再升级：使用摘要算法生成参数签名

首先，在 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 也是不一致的，所以，根据比对结果：

如果 sign 一致，说明这是个合法请求。
如果 sign 不一致，说明发起请求的客户端秘钥不正确，或者请求参数被篡改过，是个不合法请求。

此方案优点：

不在 url 中直接传递 secretKey 参数了，避免了泄露风险。
由于 sign 参数的限制，请求中的参数也不可被篡改，B 系统可放心的使用这些参数。

此方案仍然存在以下缺陷：

被抓包后，请求可以被无限重放，B 系统无法判断请求是真正来自于 A 系统发出的，还是被抓包后重放的。

@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";
}

4. 方案再再升级：追加 nonce 随机字符串

首先，在 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";
}

代码分析：

为方便理解，我们先看第 3 步：此处在校验签名成功后，将 nonce 随机字符串记入缓存中。
再看第 1 步：每次请求进来，先查看一下缓存中是否已经记录了这个随机字符串，如果是，则立即返回：无效请求。

这两步的组合，保证了一个 nonce 随机字符串只能被使用一次，如果请求被抓包后重放，是无法通过 nonce 校验的。

至此，问题似乎已被解决了 …… 吗？

别急，我们还有一个问题没有考虑：这个 nonce 在字符串在缓存应该被保存多久呢？

保存 15 分钟？那抓包的人只需要等待 15 分钟，你的 nonce 记录在缓存中消失，请求就可以被重放了。
那保存 24 小时？保存一周？保存半个月？好像无论保存多久，都无法从根本上解决这个问题。

你可能会想到，那我永久保存吧。这样确实能解决问题，但显然服务器承载不了这么做，即使再微小的数据量，在时间的累加下，也总一天会超出服务器能够承载的上限。

5. 方案再再再升级：追加 timestamp 时间戳

我们可以再追加一个 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 + "&timestamp=" + 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";
}

至此，抓包者：

如果在 15 分钟内重放攻击，nonce 参数不答应：缓存中可以查出 nonce 值，直接拒绝响应请求。
如果在 15 分钟后重放攻击，timestamp 参数不答应：超出了允许的 timestamp 时间差，直接拒绝响应请求。

6. 服务器的时钟差异造成安全问题

以上的代码，均假设 A 系统服务器与 B 系统服务器的时钟一致，才可以正常完成安全校验，但在实际的开发场景中，有些服务器会存在时钟不准确的问题。

假设 A 服务器与 B 服务器的时钟差异为 10 分钟，即：在 A 服务器为 8:00 的时候，B 服务器为 7:50。

A 系统发起请求，其生成的时间戳也是代表 8:00。
B 系统接受到请求后，完成业务处理，此时 nonce 的 ttl 为 15分钟，到期时间为 7:50 + 15分 = 8:05。
8.05 后，nonce 缓存消失，抓包者重放请求攻击：
- timestamp 校验通过：因为时间戳差距仅有 8.05 - 8.00 = 5分钟，小于 15 分钟，校验通过。
- -nonce 校验通过：因为此时 nonce 缓存已经消失，可以通过校验。
- sign 校验通过：因为这本来就是由 A 系统构建的一个合法签名。
- 攻击完成。

要解决上述问题，有两种方案：

方案一：修改服务器时钟，使两个服务器时钟保持一致。
方案二：在代码层面兼容时钟不一致的场景。

要采用方案一的同学可自行搜索一下同步时钟的方法，在此暂不赘述，此处详细阐述一下方案二。

我们只需简单修改一下，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";
}

8. 使用 Sa-Token 框架完成 API 参数签名

接下来步入正题，使用 Sa-Token 内置的 sign 模块，方便的完成 API 签名创建、校验等步骤：

不限制请求的参数数量，方便组织业务需求代码。
自动补全 nonce、timestamp 参数，省时省力。
自动构建签名，并序列化参数为字符串。
一句代码完成 nonce、timestamp、sign 的校验，防伪造请求调用、防参数篡改、防重放攻击。

8.1 引入依赖

api-sign 模块已内嵌到核心包，只需要引入 sa-token 本身依赖即可：（请求发起端和接收端都需要引入）

<dependency>
    <groupId>cn.dev33</groupId>
    <artifactId>sa-token-spring-boot-starter</artifactId>
    <version>1.35.0.RC</version>
</dependency>

8.2 配置密钥

请求发起端和接收端需要配置一个相同的秘钥，在 application.yml 中配置：

sa-token: 
    sign:
        # API 接口签名秘钥 （随便乱摁几个字母即可）
        secret-key: kQwIOrYvnXmSDkwEiFngrKidMcdrgKor

8.3 请求发起端构建签名

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);
}

8.4 请求接受端校验签名

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 校验

8.5 原理分析

8.5.1 构建签名

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);
}

这个方法有两个逻辑：

添加参数：timestamp、nonce、sign
将 map 使用 & 转化为String

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);
}

这个方法有两个逻辑：

按照数据字典进行排序，并将 map 使用 & 转化为String
使用 md5 摘要算法

8.5.2 验证签名

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);
        }
    }
}

SaToken 存储

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);
    }
}