码农知识堂 - 1000bd
  •   Python
  •   PHP
  •   JS/TS
  •   JAVA
  •   C/C++
  •   C#
  •   GO
  •   Kotlin
  •   Swift
  • 接口返回字段加密


    package com.lxkj.frame.core.converter;
    
    import com.alibaba.fastjson.JSON;
    import com.alibaba.fastjson.serializer.SerializerFeature;
    import com.lxkj.frame.core.annotation.Encrypt;
    import com.lxkj.frame.core.status.EnumEnabled;
    import com.lxkj.frame.core.status.EnumSysConfig;
    import com.lxkj.frame.core.utils.RSAUtils;
    import com.lxkj.frame.core.utils.StringUtils;
    import com.lxkj.sys.service.SysConfigService;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.core.MethodParameter;
    import org.springframework.http.MediaType;
    import org.springframework.http.converter.HttpMessageConverter;
    import org.springframework.http.server.ServerHttpRequest;
    import org.springframework.http.server.ServerHttpResponse;
    import org.springframework.util.CollectionUtils;
    import org.springframework.web.bind.annotation.RestControllerAdvice;
    import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;
    
    import java.util.ArrayList;
    import java.util.Arrays;
    import java.util.List;
    import java.util.regex.Matcher;
    import java.util.regex.Pattern;
    
    @RestControllerAdvice
    public class ResponseResultAdvice implements ResponseBodyAdvice {
    
        @Autowired
        private SysConfigService sysConfigService;
    
        @Override
        public boolean supports(MethodParameter returnType, Class> converterType) {
            return true;
        }
    
        @Override
        public Object beforeBodyWrite(Object body, MethodParameter returnType, MediaType selectedContentType, Class> selectedConverterType, ServerHttpRequest request, ServerHttpResponse response) {
            //如果加密标识为无效,则不作加密处理
            List encrypt = request.getHeaders().get("encrypt");
            if(!CollectionUtils.isEmpty(encrypt) && encrypt.get(0).equalsIgnoreCase("N")){
                return body;
            }
    
            if(body != null && body.getClass().isAnnotationPresent(Encrypt.class)){
                //检查是否已开启加密功能
                String isEnabled = sysConfigService.selectConfigByKeyCache(EnumSysConfig.SYS_SECURITY_ENABLED.getColumn_name(), EnumSysConfig.SYS_SECURITY_ENABLED.getValue());
                if(!EnumEnabled.ENABLED.getCode().equals(isEnabled)){
                    return body;
                }
                List exceptUrls = Arrays.asList(sysConfigService.selectConfigByKeyCache(EnumSysConfig.SYS_SECURITY_CC_EXCEPT_URLS.getColumn_name(), EnumSysConfig.SYS_SECURITY_CC_EXCEPT_URLS.getValue()).split(","));
                String items = sysConfigService.selectConfigByKeyCache(EnumSysConfig.SYS_SECURITY_TARGET_KEYS.getColumn_name(), EnumSysConfig.SYS_SECURITY_TARGET_KEYS.getValue());
                List matchItems = Arrays.asList(items.split(","));
                String publicKey = sysConfigService.selectConfigByKeyCache(EnumSysConfig.SYS_SECURITY_RSA_KEY_PUBLIC.getColumn_name(), EnumSysConfig.SYS_SECURITY_RSA_KEY_PUBLIC.getValue());
                //确认当前请求地址是否不使用加密
                String requestUrl = request.getURI().getPath();
                for(String url : exceptUrls){
                    if(url.trim().contains(requestUrl) || requestUrl.contains(url.trim())){
                        return body;
                    }
                }
                //将数据对象转化为字符串
                String data = null;
                try {
                    data = JSON.toJSONString(body, SerializerFeature.WriteMapNullValue);
                }catch(Exception e){
                    return body;
                }
                //同时匹配多个敏感信息关键字,如果未找到则返回原始数据
                String regexAny = "";
                for(String item : matchItems){
                    if(!"".equals(regexAny)){
                        regexAny = regexAny + "|";
                    }
                    regexAny = regexAny + "(\"" + item.trim() + "\")";
                }
                Pattern pattern = Pattern.compile(regexAny);
                Matcher matcher = pattern.matcher(data);
                if(!matcher.find()){
                    return body;
                }
                //去除重复的Key
                List noRepeatItems = new ArrayList<>();
                for(String item : matchItems){
                    boolean isRepeat = false;
                    for(String item2 : noRepeatItems){
                        if(item.equals(item2)){
                            isRepeat = true;
                            break;
                        }
                    }
                    if(!isRepeat){
                        noRepeatItems.add(item);
                    }
                }
                matchItems = noRepeatItems;
                //精确匹配敏感信息关键字并加密替换
                List encryptItems = new ArrayList<>();
                for(String item : matchItems){
                    String regex = "(\"" + item.trim() + "\"):\"([^\"]*)\"";
                    Pattern r = Pattern.compile(regex);
                    Matcher m = r.matcher(data);
                    while (m.find()) {
                        if(m.groupCount() == 2) {
                            String dataAll = m.group();
                            String dataKey = m.group(1);
                            String dataValue = m.group(2);
                            if(StringUtils.isNotEmpty(dataValue)) {
                                String encrytedData = RSAUtils.encryptedDataOnJava(dataValue.replace("\"", ""), publicKey);
                                data = data.replace(dataAll, dataKey + ":\"" + encrytedData + "\"");
                                if (!encryptItems.contains(item)) {
                                    encryptItems.add(item);
                                }
                            }
                        }
                    }
                }
                //如果存在替换则处理,则返回替换后的数据
                if(encryptItems.size() > 0) {
                    //检查是否包含关键字
                    if(response != null) {
                        response.getHeaders().add("encrypt", "Y");
                        response.getHeaders().add("encryptItems", StringUtils.join(encryptItems, ","));
                    }
                    return JSON.parseObject(data);
                }
            }
            return body;
        }
    
    }
    
    
    
                    
                        
                    
    
                    
  • 相关阅读:
    对el-table表格的表头操作
    【精选】矩阵加速
    如何使用 chrome Timeline 工具
    C++多线程 task--std::promise和std::future
    Redis解决缓存雪崩和缓存穿透
    华钜同创:亚马逊开店六个日常运营小技巧
    STL map的几种赋值方式
    MindSpore-AttributeError: module ‘mindspore‘ has no attribute ‘value_and_grad‘
    深度学习Tensorflow生产环境部署
    2022-2028年全球与中国工程夯锤市场现状及未来发展趋势分析报告
  • 原文地址:https://blog.csdn.net/qq_39716220/article/details/133822097
    • 最新文章
    • 攻防演习之三天拿下官网站群
      数据安全治理学习——前期安全规划和安全管理体系建设
      企业安全 | 企业内一次钓鱼演练准备过程
      内网渗透测试 | Kerberos协议及其部分攻击手法
      0day的产生 | 不懂代码的"代码审计"
      安装scrcpy-client模块av模块异常,环境问题解决方案
      leetcode hot100【LeetCode 279. 完全平方数】java实现
      OpenWrt下安装Mosquitto
      AnatoMask论文汇总
      【AI日记】24.11.01 LangChain、openai api和github copilot
    • 热门文章
    • 十款代码表白小特效 一个比一个浪漫 赶紧收藏起来吧!!!
      奉劝各位学弟学妹们,该打造你的技术影响力了!
      五年了,我在 CSDN 的两个一百万。
      Java俄罗斯方块,老程序员花了一个周末,连接中学年代!
      面试官都震惊,你这网络基础可以啊!
      你真的会用百度吗?我不信 — 那些不为人知的搜索引擎语法
      心情不好的时候,用 Python 画棵樱花树送给自己吧
      通宵一晚做出来的一款类似CS的第一人称射击游戏Demo!原来做游戏也不是很难,连憨憨学妹都学会了!
      13 万字 C 语言从入门到精通保姆级教程2021 年版
      10行代码集2000张美女图,Python爬虫120例,再上征途
    Copyright © 2022 侵权请联系2656653265@qq.com    京ICP备2022015340号-1
    正则表达式工具 cron表达式工具 密码生成工具

    京公网安备 11010502049817号