lsm/sa-token
1
0
Fork 0
mirror of https://gitee.com/dromara/sa-token.git synced 2025-09-18 17:48:03 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

在登录时强制性检查账号 id 是否为异常值，如果是则登录失败。

Browse Source
This commit is contained in:
click33
2023-05-24 16:17:42 +08:00
parent 4087751fcc
commit 48ba6da91a
2 changed files with 13 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
sa-token-core/src/main/java/cn/dev33/satoken/stp/StpLogic.java
View File

@@ -521,17 +521,22 @@ public class StpLogic {
*/ */
protected void checkLoginArgs(Object id, SaLoginModel loginModel) { protected void checkLoginArgs(Object id, SaLoginModel loginModel) {
// 1、先检查一下，传入的账号id是否可用 // 1、账号 id 不能为空
if(SaFoxUtil.isEmpty(id)) { if(SaFoxUtil.isEmpty(id)) {
throw new SaTokenException("loginId 不能为空").setCode(SaErrorCode.CODE_11002); throw new SaTokenException("loginId 不能为空").setCode(SaErrorCode.CODE_11002);
} }
// 2、判断账号id是否为简单类型 // 2、账号 id 不能是异常标记值
if(NotLoginException.ABNORMAL_LIST.contains(id.toString())) {
throw new SaTokenException("loginId 不能为以下值：" + NotLoginException.ABNORMAL_LIST);
}
// 3、账号 id 不能是简单类型
if( ! SaFoxUtil.isBasicType(id.getClass())) { if( ! SaFoxUtil.isBasicType(id.getClass())) {
SaManager.log.warn("loginId 应该为简单类型，例如：String | int | long，不推荐使用复杂类型：" + id.getClass()); SaManager.log.warn("loginId 应该为简单类型，例如：String | int | long，不推荐使用复杂类型：" + id.getClass());
} }
// 3、判断当前 StpLogic 是否支持 extra 扩展参数 // 4、判断当前 StpLogic 是否支持 extra 扩展参数
if( ! isSupportExtra()) { if( ! isSupportExtra()) {
// 如果不支持，开发者却传入了 extra 扩展参数，那么就打印警告信息 // 如果不支持，开发者却传入了 extra 扩展参数，那么就打印警告信息
Map<String, Object> extraData = loginModel.getExtraData(); Map<String, Object> extraData = loginModel.getExtraData();
@@ -1040,8 +1045,8 @@ public class StpLogic {
* @return / * @return /
*/ */
public boolean isValidLoginId(Object loginId) { public boolean isValidLoginId(Object loginId) {
// 判断标准：不为null，且不在异常标记项里面 // 判断标准：不为 null、空字符串，且不在异常标记项里面
return loginId != null && !NotLoginException.ABNORMAL_LIST.contains(loginId.toString()); return SaFoxUtil.isNotEmpty(loginId) && !NotLoginException.ABNORMAL_LIST.contains(loginId.toString());
} }
/** /**

6
sa-token-demo/sa-token-demo-test/src/main/java/com/pj/test/TestController.java
View File

@@ -1,7 +1,7 @@
package com.pj.test; package com.pj.test;
import cn.dev33.satoken.stp.StpUtil;
import cn.dev33.satoken.util.SaResult; import cn.dev33.satoken.util.SaResult;
import com.pj.satoken.StpUserUtil;
import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam; import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController; import org.springframework.web.bind.annotation.RestController;
@@ -18,7 +18,7 @@ public class TestController {
// 测试登录 ---- http://localhost:8081/test/login // 测试登录 ---- http://localhost:8081/test/login
@RequestMapping("login") @RequestMapping("login")
public SaResult login(@RequestParam(defaultValue = "10001") long id) { public SaResult login(@RequestParam(defaultValue = "10001") long id) {
StpUserUtil.login(id); StpUtil.login(id);
return SaResult.ok("登录成功"); return SaResult.ok("登录成功");
} }
@@ -27,7 +27,7 @@ public class TestController {
public SaResult test() { public SaResult test() {
System.out.println("------------进来了"); System.out.println("------------进来了");
// 返回 // 返回
return SaResult.data(""); return SaResult.data(null);
} }
// 测试 浏览器访问： http://localhost:8081/test/test2 // 测试 浏览器访问： http://localhost:8081/test/test2