diff --git a/pom.xml b/pom.xml index 84a9ec6..a5387da 100644 --- a/pom.xml +++ b/pom.xml @@ -115,6 +115,33 @@ 4.0.1 + + org.bouncycastle + bcprov-jdk15to18 + 1.76 + + + commons-codec + commons-codec + 1.15 + + + org.projectlombok + lombok + 1.18.28 + + + + org.apache.httpcomponents + httpcore + 4.4.16 + + + org.apache.httpcomponents + httpclient + 4.5.14 + + zxxy diff --git a/src/main/java/com/hxgk/zxxy/controller/ArchivesTypeController.java b/src/main/java/com/hxgk/zxxy/controller/ArchivesTypeController.java index 59b4775..74286b7 100644 --- a/src/main/java/com/hxgk/zxxy/controller/ArchivesTypeController.java +++ b/src/main/java/com/hxgk/zxxy/controller/ArchivesTypeController.java @@ -4,6 +4,8 @@ import com.hxgk.zxxy.model.entity.ArchivesType; import com.hxgk.zxxy.model.request.ArchivesTypeRequest; import com.hxgk.zxxy.service.ArchivesTypeService; import com.hxgk.zxxy.utils.JsonData; +import com.hxgk.zxxy.utils.sm4.ApiResponse; +import com.hxgk.zxxy.utils.sm4.DecryptedBody; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.*; @@ -19,11 +21,11 @@ public class ArchivesTypeController { @Autowired private ArchivesTypeService archivesTypeService; @PostMapping("zxxy") - public JsonData getZxxyArchivesType(@RequestBody ArchivesTypeRequest archivesTypeRequest){ + public ApiResponse getZxxyArchivesType(@DecryptedBody ArchivesTypeRequest archivesTypeRequest){ Long L = Long.parseLong(archivesTypeRequest.getAtParentId()); String userkey = archivesTypeRequest.getUserkey(); String usertoken = archivesTypeRequest.getUsertoken(); List archivesTypeList = archivesTypeService.getZxxyArchivesType(L,userkey,usertoken); - return JsonData.buildSuccess(archivesTypeList); + return ApiResponse.success(archivesTypeList); } } diff --git a/src/main/java/com/hxgk/zxxy/controller/GraphicformController.java b/src/main/java/com/hxgk/zxxy/controller/GraphicformController.java index 3f240c1..bd3ebe1 100644 --- a/src/main/java/com/hxgk/zxxy/controller/GraphicformController.java +++ b/src/main/java/com/hxgk/zxxy/controller/GraphicformController.java @@ -9,6 +9,8 @@ import com.hxgk.zxxy.model.request.GraphicformRequest; import com.hxgk.zxxy.service.ArchivesTypeService; import com.hxgk.zxxy.service.GraphicformService; import com.hxgk.zxxy.utils.JsonData; +import com.hxgk.zxxy.utils.sm4.ApiResponse; +import com.hxgk.zxxy.utils.sm4.DecryptedBody; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.PostMapping; import org.springframework.web.bind.annotation.RequestBody; @@ -23,15 +25,15 @@ public class GraphicformController { @Autowired private GraphicformService graphicformService; @PostMapping("cardlist") - public JsonData getZxxyArchivesType(@RequestBody GraphicformRequest graphicformRequest) throws Exception { + public ApiResponse getZxxyArchivesType(@DecryptedBody GraphicformRequest graphicformRequest) throws Exception { PageInfo graphicformList = graphicformService.queryListByParam(graphicformRequest); - return JsonData.buildSuccess(graphicformList); + return ApiResponse.success(graphicformList); } @PostMapping("carousel") - public JsonData getCarousel(@RequestBody GraphicformRequest graphicformRequest) throws Exception { + public ApiResponse getCarousel(@DecryptedBody GraphicformRequest graphicformRequest) throws Exception { PageInfo graphicformList = graphicformService.getCarousel(graphicformRequest); - return JsonData.buildSuccess(graphicformList); + return ApiResponse.success(graphicformList); } } diff --git a/src/main/java/com/hxgk/zxxy/controller/LowCodeTransferController.java b/src/main/java/com/hxgk/zxxy/controller/LowCodeTransferController.java index def5c9d..16c339a 100644 --- a/src/main/java/com/hxgk/zxxy/controller/LowCodeTransferController.java +++ b/src/main/java/com/hxgk/zxxy/controller/LowCodeTransferController.java @@ -7,6 +7,8 @@ import com.hxgk.zxxy.service.HrService; import com.hxgk.zxxy.service.LowCodeTransferService; import com.hxgk.zxxy.service.UserService; import com.hxgk.zxxy.utils.JsonData; +import com.hxgk.zxxy.utils.sm4.ApiResponse; +import com.hxgk.zxxy.utils.sm4.DecryptedBody; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.RequestBody; import org.springframework.web.bind.annotation.RequestMapping; @@ -24,16 +26,16 @@ public class LowCodeTransferController { @Autowired private HrService hrService; @RequestMapping(value = "transfer/getCarList") - public JsonData getUserDetailFromRedis(@RequestParam(value = "transferMark", required = false) String transferMark){ + public ApiResponse getUserDetailFromRedis(@DecryptedBody(value = "transferMark", required = false) String transferMark){ ArrayList carList = lowCodeTransferService.getCarList(); - return JsonData.buildSuccess(carList); + return ApiResponse.success(carList); } @RequestMapping(value = "transfer/getOrgAndManTree") - public JsonData getOrgAndManTree(@RequestParam(value = "transferMark", required = false) String transferMark){ + public ApiResponse getOrgAndManTree(@DecryptedBody(value = "transferMark", required = false) String transferMark){ OrgAndManTree orgAndManTree = hrService.getOrgAndManTree(); - return JsonData.buildSuccess(orgAndManTree); + return ApiResponse.success(orgAndManTree); } } diff --git a/src/main/java/com/hxgk/zxxy/controller/UserController.java b/src/main/java/com/hxgk/zxxy/controller/UserController.java index 7013e31..b104718 100644 --- a/src/main/java/com/hxgk/zxxy/controller/UserController.java +++ b/src/main/java/com/hxgk/zxxy/controller/UserController.java @@ -4,6 +4,8 @@ import com.hxgk.zxxy.model.entity.ManCont; import com.hxgk.zxxy.model.entity.UserDetail; import com.hxgk.zxxy.service.UserService; import com.hxgk.zxxy.utils.JsonData; +import com.hxgk.zxxy.utils.sm4.ApiResponse; +import com.hxgk.zxxy.utils.sm4.DecryptedBody; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.*; @@ -15,16 +17,16 @@ public class UserController { @Autowired private UserService userService; @RequestMapping(value = "detail") - public JsonData getUserDetailFromRedis(@RequestBody Map userInfo){ + public ApiResponse getUserDetailFromRedis(@DecryptedBody Map userInfo){ UserDetail userDetail = userService.getUserDetailFromRedis(userInfo); - return JsonData.buildSuccess(userDetail); + return ApiResponse.success(userDetail); } @GetMapping(value = "getByKey") - public JsonData getManContByKey(@RequestParam(value = "key", required = true) String key){ + public ApiResponse getManContByKey(@DecryptedBody(value = "key", required = true) String key){ ManCont manCont = userService.getManContByKey(key); - return JsonData.buildSuccess(manCont); + return ApiResponse.success(manCont); } diff --git a/src/main/java/com/hxgk/zxxy/utils/sm4/ApiResponse.java b/src/main/java/com/hxgk/zxxy/utils/sm4/ApiResponse.java new file mode 100644 index 0000000..fb98d64 --- /dev/null +++ b/src/main/java/com/hxgk/zxxy/utils/sm4/ApiResponse.java @@ -0,0 +1,32 @@ +package com.hxgk.zxxy.utils.sm4; + + +import lombok.Data; + +@Data +public class ApiResponse { + private int code; + private String msg; + private T data; + + public ApiResponse() {} + + public ApiResponse(int code, String msg, T data) { + this.code = code; + this.msg = msg; + this.data = data; + } + + public static ApiResponse success(T data) { + if(null==data){ + return new ApiResponse(0, "成功", (T) "成功,没有相关数据"); + }else{ + return new ApiResponse<>(0, "成功", data); + } + + } + + public static ApiResponse error(int code, String msg) { + return new ApiResponse<>(code, msg, null); + } +} \ No newline at end of file diff --git a/src/main/java/com/hxgk/zxxy/utils/sm4/DecryptedArgumentResolver.java b/src/main/java/com/hxgk/zxxy/utils/sm4/DecryptedArgumentResolver.java new file mode 100644 index 0000000..8420529 --- /dev/null +++ b/src/main/java/com/hxgk/zxxy/utils/sm4/DecryptedArgumentResolver.java @@ -0,0 +1,134 @@ +package com.hxgk.zxxy.utils.sm4; + +import com.fasterxml.jackson.databind.JsonNode; +import com.fasterxml.jackson.databind.ObjectMapper; +import lombok.extern.slf4j.Slf4j; +import org.springframework.core.MethodParameter; +import org.springframework.stereotype.Component; +import org.springframework.util.StringUtils; +import org.springframework.web.bind.support.WebDataBinderFactory; +import org.springframework.web.context.request.NativeWebRequest; +import org.springframework.web.method.support.HandlerMethodArgumentResolver; +import org.springframework.web.method.support.ModelAndViewContainer; + +import javax.servlet.http.HttpServletRequest; + +@Slf4j +@Component +public class DecryptedArgumentResolver implements HandlerMethodArgumentResolver { + + private final ObjectMapper objectMapper = new ObjectMapper(); + + @Override + public boolean supportsParameter(MethodParameter parameter) { + return parameter.hasParameterAnnotation(DecryptedBody.class); + } + + @Override + public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer, + NativeWebRequest webRequest, WebDataBinderFactory binderFactory) throws Exception { + + HttpServletRequest request = webRequest.getNativeRequest(HttpServletRequest.class); + String decryptedBody = (String) request.getAttribute("decryptedBody"); + + if (decryptedBody == null || decryptedBody.trim().isEmpty()) { + log.debug("没有解密数据，尝试直接解析请求体"); + // 如果没有解密数据，尝试直接解析请求体（非加密模式） + try { + String requestBody = getRequestBody(request); + if (requestBody != null && !requestBody.trim().isEmpty()) { + decryptedBody = requestBody; + } + } catch (Exception e) { + log.warn("无法读取请求体", e); + } + } + + DecryptedBody decryptedBodyAnnotation = parameter.getParameterAnnotation(DecryptedBody.class); + String fieldName = decryptedBodyAnnotation.value(); + boolean required = decryptedBodyAnnotation.required(); + + log.debug("解析参数: fieldName={}, required={}, decryptedBody={}", + fieldName, required, decryptedBody != null ? "有数据" : "无数据"); + + // 如果解密数据为空 + if (decryptedBody == null || decryptedBody.trim().isEmpty()) { + if (required) { + throw new IllegalArgumentException("缺少必要参数: " + (StringUtils.hasText(fieldName) ? fieldName : "请求体")); + } + return null; + } + + try { + // 将解密后的字符串解析为JsonNode + JsonNode rootNode = objectMapper.readTree(decryptedBody); + + // 如果指定了字段名，则提取该字段 + if (StringUtils.hasText(fieldName)) { + JsonNode fieldNode = rootNode.get(fieldName); + + if (fieldNode == null || fieldNode.isNull()) { + if (required) { + throw new IllegalArgumentException("缺少必要参数: " + fieldName); + } + return null; + } + + // 根据参数类型转换 + Class parameterType = parameter.getParameterType(); + return convertJsonNodeToType(fieldNode, parameterType); + } else { + // 如果没有指定字段名，解析整个对象 + Class parameterType = parameter.getParameterType(); + return objectMapper.treeToValue(rootNode, parameterType); + } + + } catch (Exception e) { + log.error("解析解密数据失败", e); + if (required) { + throw new IllegalArgumentException("解析参数失败: " + e.getMessage()); + } + return null; + } + } + + /** + * 将JsonNode转换为指定类型 + */ + private Object convertJsonNodeToType(JsonNode node, Class targetType) throws Exception { + if (targetType == String.class) { + if (node.isTextual()) { + return node.asText(); + } else { + return node.toString(); + } + } else if (targetType == Integer.class || targetType == int.class) { + return node.asInt(); + } else if (targetType == Long.class || targetType == long.class) { + return node.asLong(); + } else if (targetType == Double.class || targetType == double.class) { + return node.asDouble(); + } else if (targetType == Boolean.class || targetType == boolean.class) { + return node.asBoolean(); + } else if (targetType == JsonNode.class) { + return node; + } else { + // 对于其他复杂类型，使用ObjectMapper转换 + return objectMapper.treeToValue(node, targetType); + } + } + + /** + * 获取请求体 + */ + private String getRequestBody(HttpServletRequest request) throws Exception { + StringBuilder sb = new StringBuilder(); + try (java.io.BufferedReader reader = request.getReader()) { + String line; + while ((line = reader.readLine()) != null) { + sb.append(line); + } + } + return sb.toString(); + } +} \ No newline at end of file diff --git a/src/main/java/com/hxgk/zxxy/utils/sm4/DecryptedBody.java b/src/main/java/com/hxgk/zxxy/utils/sm4/DecryptedBody.java new file mode 100644 index 0000000..d0f042c --- /dev/null +++ b/src/main/java/com/hxgk/zxxy/utils/sm4/DecryptedBody.java @@ -0,0 +1,18 @@ +package com.hxgk.zxxy.utils.sm4; + +import java.lang.annotation.*; + +@Target(ElementType.PARAMETER) +@Retention(RetentionPolicy.RUNTIME) +@Documented +public @interface DecryptedBody { + /** + * 指定要提取的字段名，如果为空则解析整个对象 + */ + String value() default ""; + + /** + * 是否必须，如果为true且字段不存在则抛出异常 + */ + boolean required() default false; +} \ No newline at end of file diff --git a/src/main/java/com/hxgk/zxxy/utils/sm4/EncryptedRequest.java b/src/main/java/com/hxgk/zxxy/utils/sm4/EncryptedRequest.java new file mode 100644 index 0000000..70b0463 --- /dev/null +++ b/src/main/java/com/hxgk/zxxy/utils/sm4/EncryptedRequest.java @@ -0,0 +1,20 @@ +package com.hxgk.zxxy.utils.sm4; + + + +import lombok.Data; + +@Data +public class EncryptedRequest { + /** + * 加密的数据字段 + * 前端发送的是 "data" 字段，不是 "encryptedFile" + */ + private String data; + + public EncryptedRequest() {} + + public EncryptedRequest(String data) { + this.data = data; + } +} \ No newline at end of file diff --git a/src/main/java/com/hxgk/zxxy/utils/sm4/EncryptedResponseAdvice.java b/src/main/java/com/hxgk/zxxy/utils/sm4/EncryptedResponseAdvice.java new file mode 100644 index 0000000..13b50c9 --- /dev/null +++ b/src/main/java/com/hxgk/zxxy/utils/sm4/EncryptedResponseAdvice.java @@ -0,0 +1,113 @@ +package com.hxgk.zxxy.utils.sm4; + + +import com.fasterxml.jackson.databind.ObjectMapper; +import lombok.extern.slf4j.Slf4j; +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.http.server.ServletServerHttpRequest; +import org.springframework.stereotype.Component; +import org.springframework.web.bind.annotation.ControllerAdvice; +import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice; + +import javax.servlet.http.HttpServletRequest; + +@Slf4j +@Component +@ControllerAdvice +public class EncryptedResponseAdvice implements ResponseBodyAdvice { + + private final ObjectMapper objectMapper = new ObjectMapper(); + + @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) { + + HttpServletRequest servletRequest = ((ServletServerHttpRequest) request).getServletRequest(); + + // 检查是否应该加密（可以根据路径或其他条件判断） + boolean shouldEncrypt = shouldEncryptResponse(servletRequest); + if (!shouldEncrypt) { + return body; + } + + // 获取或生成 Auth-key + String authKey = (String) servletRequest.getAttribute("responseAuthKey"); + if (authKey == null) { + authKey = RandomStringUtil.generateRandomString(16); + servletRequest.setAttribute("responseAuthKey", authKey); + } + + // 设置响应头 + response.getHeaders().set("Auth-key", authKey); + + try { + // 只有 ApiResponse 才需要加密 + if (body instanceof ApiResponse) { + ApiResponse apiResponse = (ApiResponse) body; + + // 如果数据不为空，进行加密 + if (apiResponse.getData() != null) { + // 将数据转换为 JSON 字符串 + String jsonData = objectMapper.writeValueAsString(apiResponse.getData()); + + log.debug("准备加密的数据: {}", jsonData); + log.debug("使用的 Auth-key: {}", authKey); + + // 使用 SM4 加密 + String encryptedData = SM4Utils.encrypt(jsonData, authKey); + + log.debug("加密结果长度: {}", encryptedData.length()); + + // 返回格式：{ code: 0, msg: "成功", data: "加密字符串" } + return new ApiResponse<>(apiResponse.getCode(), apiResponse.getMsg(), encryptedData); + } else { + // 数据为空，直接返回原始响应 + return apiResponse; + } + } + + // 如果不是 ApiResponse，直接返回（如文件下载等） + return body; + + } catch (Exception e) { + log.error("响应加密失败", e); + // 加密失败时返回错误响应 + return ApiResponse.error(500, "服务器加密错误: " + e.getMessage()); + } + } + + /** + * 判断是否应该加密响应 + */ + private boolean shouldEncryptResponse(HttpServletRequest request) { + String path = request.getRequestURI(); + + // 可以根据路径排除某些接口 + if (path.contains("/api/debug/") || + path.contains("/actuator/") || + path.contains("/swagger") || + path.contains("/v2/api-docs") || + path.endsWith(".html") || + path.endsWith(".js") || + path.endsWith(".css")) { + return false; + } + + // 检查是否有特殊的请求头表示不需要加密 + if ("false".equalsIgnoreCase(request.getHeader("X-No-Encrypt"))) { + return false; + } + + return true; + } +} \ No newline at end of file diff --git a/src/main/java/com/hxgk/zxxy/utils/sm4/ErrorCode.java b/src/main/java/com/hxgk/zxxy/utils/sm4/ErrorCode.java new file mode 100644 index 0000000..7b50811 --- /dev/null +++ b/src/main/java/com/hxgk/zxxy/utils/sm4/ErrorCode.java @@ -0,0 +1,33 @@ +package com.hxgk.zxxy.utils.sm4; + + + +public enum ErrorCode { + SUCCESS(0, "成功"), + DATA_FORMAT_ERROR(100, "提交的数据格式错误！"), + DATA_EMPTY(101, "提交的数据不能为空！"), + KEY_PARAM_EMPTY(102, "关键参数不能为空！"), + DATA_EXISTS(103, "该数据已经存在！请不要重复添加"), + WRITE_FAILED(104, "数据写入失败！"), + QUERY_FAILED(105, "数据查询失败！"), + UPDATE_FAILED(106, "编辑失败！"), + NO_DATA(107, "未能查询到数据！"), + DELETE_FAILED(108, "删除失败"), + AUTH_ERROR(2000, "账户或密码错误！"); + + private final int code; + private final String message; + + ErrorCode(int code, String message) { + this.code = code; + this.message = message; + } + + public int getCode() { + return code; + } + + public String getMessage() { + return message; + } +} \ No newline at end of file diff --git a/src/main/java/com/hxgk/zxxy/utils/sm4/GlobalExceptionHandler.java b/src/main/java/com/hxgk/zxxy/utils/sm4/GlobalExceptionHandler.java new file mode 100644 index 0000000..3832a60 --- /dev/null +++ b/src/main/java/com/hxgk/zxxy/utils/sm4/GlobalExceptionHandler.java @@ -0,0 +1,22 @@ +package com.hxgk.zxxy.utils.sm4; + + +import lombok.extern.slf4j.Slf4j; +import org.springframework.web.bind.annotation.ExceptionHandler; +import org.springframework.web.bind.annotation.RestControllerAdvice; + +@Slf4j +@RestControllerAdvice +public class GlobalExceptionHandler { + + @ExceptionHandler(Exception.class) + public ApiResponse handleException(Exception e) { + log.error("系统异常:", e); + return ApiResponse.error(500, "服务器错误"); + } + + @ExceptionHandler(IllegalArgumentException.class) + public ApiResponse handleIllegalArgumentException(IllegalArgumentException e) { + return ApiResponse.error(400, e.getMessage()); + } +} \ No newline at end of file diff --git a/src/main/java/com/hxgk/zxxy/utils/sm4/GoApiClient.java b/src/main/java/com/hxgk/zxxy/utils/sm4/GoApiClient.java new file mode 100644 index 0000000..0d6796d --- /dev/null +++ b/src/main/java/com/hxgk/zxxy/utils/sm4/GoApiClient.java @@ -0,0 +1,160 @@ +package com.hxgk.zxxy.utils.sm4; + + +import com.alibaba.fastjson2.JSON; +import com.alibaba.fastjson2.JSONObject; +import lombok.extern.slf4j.Slf4j; +import org.apache.http.HttpResponse; +import org.apache.http.client.config.RequestConfig; +import org.apache.http.client.methods.HttpPost; +import org.apache.http.entity.StringEntity; +import org.apache.http.impl.client.CloseableHttpClient; +import org.apache.http.impl.client.HttpClients; +import org.apache.http.util.EntityUtils; +import org.springframework.stereotype.Component; + +import java.io.IOException; +import java.nio.charset.StandardCharsets; +import java.util.HashMap; +import java.util.Map; + +@Slf4j +@Component +public class GoApiClient { + + private static final int TIMEOUT = 5000; + + /** + * 调用 Go 接口（使用加密通信） + * @param url Go接口地址 + * @param param 请求参数 + * @param userKey User-Key + * @param userToken User-Token + * @return 解密后的响应字符串 + * @throws IOException + */ + public String callGoApi(String url, Object param, String userKey, String userToken) throws IOException { + // 1. 生成随机 Auth-key（16位） + String authKey = RandomStringUtil.generateRandomString(16); + + // 2. 加密请求数据 + String encryptedData; + try { + String jsonParam = JSON.toJSONString(param); + encryptedData = SM4Utils.encrypt(jsonParam, authKey); + } catch (Exception e) { + log.error("加密请求数据失败", e); + throw new IOException("加密请求数据失败: " + e.getMessage(), e); + } + + // 3. 构建加密请求体 + Map encryptedBody = new HashMap<>(); + encryptedBody.put("data", encryptedData); // Go后端期望data字段 + String requestBody = JSON.toJSONString(encryptedBody); + + // 4. 发送 HTTP 请求 + CloseableHttpClient httpClient = null; + try { + RequestConfig requestConfig = RequestConfig.custom() + .setConnectTimeout(TIMEOUT) + .setSocketTimeout(TIMEOUT) + .setConnectionRequestTimeout(TIMEOUT) + .build(); + + httpClient = HttpClients.custom() + .setDefaultRequestConfig(requestConfig) + .build(); + + HttpPost httpPost = new HttpPost(url); + + // 设置请求头 + httpPost.setHeader("User-Key", userKey); + httpPost.setHeader("User-Token", userToken); + httpPost.setHeader("Auth-key", authKey); + httpPost.setHeader("Content-Type", "application/json; charset=UTF-8"); + + // 设置请求体 + StringEntity entity = new StringEntity(requestBody, StandardCharsets.UTF_8); + httpPost.setEntity(entity); + + // 执行请求 + HttpResponse response = httpClient.execute(httpPost); + int statusCode = response.getStatusLine().getStatusCode(); + + if (statusCode != 200) { + String errorResponse = EntityUtils.toString(response.getEntity(), StandardCharsets.UTF_8); + throw new IOException("HTTP 请求失败，状态码: " + statusCode + ", 响应: " + errorResponse); + } + + // 获取响应体 + String responseBody = EntityUtils.toString(response.getEntity(), StandardCharsets.UTF_8); + + // 获取响应头中的 Auth-key + org.apache.http.Header[] authKeyHeaders = response.getHeaders("Auth-key"); + String responseAuthKey = null; + if (authKeyHeaders != null && authKeyHeaders.length > 0) { + responseAuthKey = authKeyHeaders[0].getValue(); + } + + // 5. 解密响应数据 + if (responseAuthKey != null && !responseAuthKey.isEmpty()) { + return decryptGoResponse(responseBody, responseAuthKey); + } else { + log.warn("响应头中没有找到 Auth-key，返回原始响应"); + return responseBody; + } + + } catch (Exception e) { + log.error("HTTP 请求异常", e); + throw new IOException("HTTP 请求异常: " + e.getMessage(), e); + } finally { + if (httpClient != null) { + try { + httpClient.close(); + } catch (IOException e) { + log.error("关闭 HTTP 客户端失败", e); + } + } + } + } + + /** + * 解密 Go 后端的响应 + */ + private String decryptGoResponse(String responseBody, String authKey) throws Exception { + try { + // 解析 Go 的响应 + JSONObject responseJson = JSON.parseObject(responseBody); + int code = responseJson.getIntValue("code"); + String msg = responseJson.getString("msg"); + String encryptedData = responseJson.getString("data"); + + if (encryptedData != null && !encryptedData.isEmpty()) { + // 解密数据 + String decryptedData = SM4Utils.decrypt(encryptedData, authKey); + + // 重新构建响应 + JSONObject result = new JSONObject(); + result.put("code", code); + result.put("msg", msg); + + try { + // 尝试将解密后的字符串解析为 JSON 对象 + Object dataObj = JSON.parse(decryptedData); + result.put("data", dataObj); + } catch (Exception e) { + // 如果解析失败，直接使用字符串 + result.put("data", decryptedData); + } + + return result.toJSONString(); + } else { + return responseBody; + } + + } catch (Exception e) { + log.error("解密 Go 响应数据失败", e); + throw new Exception("解密 Go 响应数据失败: " + e.getMessage(), e); + } + } +} \ No newline at end of file diff --git a/src/main/java/com/hxgk/zxxy/utils/sm4/RandomStringUtil.java b/src/main/java/com/hxgk/zxxy/utils/sm4/RandomStringUtil.java new file mode 100644 index 0000000..8ed4ff5 --- /dev/null +++ b/src/main/java/com/hxgk/zxxy/utils/sm4/RandomStringUtil.java @@ -0,0 +1,47 @@ +package com.hxgk.zxxy.utils.sm4; + +import org.apache.commons.codec.binary.Hex; +import org.springframework.stereotype.Component; + +import java.security.SecureRandom; + +@Component +public class RandomStringUtil { + + private static final String CHARACTERS = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789"; + private static final SecureRandom SECURE_RANDOM = new SecureRandom(); + + /** + * 生成指定长度的随机字符串 + */ + public static String generateRandomString(int length) { + StringBuilder sb = new StringBuilder(length); + for (int i = 0; i < length; i++) { + int index = SECURE_RANDOM.nextInt(CHARACTERS.length()); + sb.append(CHARACTERS.charAt(index)); + } + return sb.toString(); + } + + /** + * 获取UUID + */ + public static String generateUUID() { + byte[] randomBytes = new byte[16]; + SECURE_RANDOM.nextBytes(randomBytes); + randomBytes[6] &= 0x0f; /* clear version */ + randomBytes[6] |= 0x40; /* set to version 4 */ + randomBytes[8] &= 0x3f; /* clear variant */ + randomBytes[8] |= 0x80; /* set to IETF variant */ + + char[] uuidChars = Hex.encodeHex(randomBytes, false); + StringBuilder uuid = new StringBuilder(36); + for (int i = 0; i < uuidChars.length; i++) { + if (i == 8 || i == 12 || i == 16 || i == 20) { + uuid.append('-'); + } + uuid.append(uuidChars[i]); + } + return uuid.toString(); + } +} diff --git a/src/main/java/com/hxgk/zxxy/utils/sm4/SM4Interceptor.java b/src/main/java/com/hxgk/zxxy/utils/sm4/SM4Interceptor.java new file mode 100644 index 0000000..50cfa3c --- /dev/null +++ b/src/main/java/com/hxgk/zxxy/utils/sm4/SM4Interceptor.java @@ -0,0 +1,98 @@ +package com.hxgk.zxxy.utils.sm4; + + + +import com.fasterxml.jackson.databind.ObjectMapper; +import lombok.extern.slf4j.Slf4j; +import org.springframework.stereotype.Component; +import org.springframework.web.servlet.HandlerInterceptor; +import org.springframework.web.servlet.ModelAndView; + +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; +import java.io.BufferedReader; +import java.io.PrintWriter; + +@Slf4j +@Component +public class SM4Interceptor implements HandlerInterceptor { + + private final ObjectMapper objectMapper = new ObjectMapper(); + + @Override + public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception { + String authKey = request.getHeader("Auth-key"); + String contentType = request.getHeader("Content-Type"); + + log.debug("请求拦截 - Auth-key: {}, Content-Type: {}", authKey, contentType); + + // 如果是JSON请求且需要解密 + if (contentType != null && contentType.contains("application/json") && authKey != null) { + try { + // 读取请求体 + StringBuilder requestBody = new StringBuilder(); + try (BufferedReader reader = request.getReader()) { + String line; + while ((line = reader.readLine()) != null) { + requestBody.append(line); + } + } + + if (requestBody.length() > 0) { + String requestBodyStr = requestBody.toString(); + //log.debug("原始请求体: {}", requestBodyStr); + + // 解析加密请求 + EncryptedRequest encryptedRequest = objectMapper.readValue(requestBodyStr, EncryptedRequest.class); + + if (encryptedRequest.getData() != null && !encryptedRequest.getData().isEmpty()) { + //log.debug("获取到加密数据: {}", encryptedRequest.getData().substring(0, Math.min(50, encryptedRequest.getData().length())) + "..."); + + // 解密数据 + String decryptedData = SM4Utils.decrypt(encryptedRequest.getData(), authKey); + //log.debug("解密后的数据: {}", decryptedData); + + // 将解密后的数据重新放入请求体 + request.setAttribute("decryptedBody", decryptedData); + + // 记录解密成功 + //log.info("请求解密成功，Auth-key: {}", authKey); + } else { + log.warn("加密数据为空，跳过解密"); + } + } + } catch (Exception e) { + log.error("请求解密失败", e); + response.setStatus(HttpServletResponse.SC_BAD_REQUEST); + response.setContentType("application/json;charset=UTF-8"); + + ApiResponse errorResponse = ApiResponse.error(100, "提交的数据格式错误！"); + PrintWriter writer = response.getWriter(); + writer.write(objectMapper.writeValueAsString(errorResponse)); + writer.flush(); + return false; + } + } + + return true; + } + + @Override + public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, + ModelAndView modelAndView) throws Exception { + // 生成新的随机密钥 + String newAuthKey = RandomStringUtil.generateRandomString(16); + response.setHeader("Auth-key", newAuthKey); + request.setAttribute("responseAuthKey", newAuthKey); + + //log.debug("生成新的 Auth-key: {}", newAuthKey); + } + + @Override + public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, + Exception ex) throws Exception { + // 清理 + request.removeAttribute("decryptedBody"); + request.removeAttribute("responseAuthKey"); + } +} \ No newline at end of file diff --git a/src/main/java/com/hxgk/zxxy/utils/sm4/SM4Utils.java b/src/main/java/com/hxgk/zxxy/utils/sm4/SM4Utils.java new file mode 100644 index 0000000..5e97022 --- /dev/null +++ b/src/main/java/com/hxgk/zxxy/utils/sm4/SM4Utils.java @@ -0,0 +1,217 @@ +package com.hxgk.zxxy.utils.sm4; + + + +import org.bouncycastle.crypto.engines.SM4Engine; +import org.bouncycastle.crypto.modes.CBCBlockCipher; +import org.bouncycastle.crypto.paddings.PKCS7Padding; +import org.bouncycastle.crypto.paddings.PaddedBufferedBlockCipher; +import org.bouncycastle.crypto.params.KeyParameter; +import org.bouncycastle.crypto.params.ParametersWithIV; +import org.springframework.beans.factory.annotation.Value; +import org.springframework.stereotype.Component; + +import javax.annotation.PostConstruct; +import java.nio.charset.StandardCharsets; + +@Component +public class SM4Utils { + + // Go 后端的 Sm4Key - 用作加密密钥 + @Value("${sm4.key:hengxingaokeApp1}") + private String sm4Key; + + // Go 后端的 Sm4Token - 默认用作 IV + @Value("${sm4.token:04TzMuvkHm_EZnHm}") + private String sm4Token; + + // 前端使用的 systemKey（仅用于与前端对比验证） + @Value("${sm4.system-key:appKeyQinDongV01}") + private String systemKey; + + private static String SM4_KEY_PROCESSED; + private static String SM4_TOKEN_PROCESSED; + private static String SYSTEM_KEY_HEX; + + @PostConstruct + public void init() { + // Go 的方式：处理为16字节（不是转16进制！） + SM4_KEY_PROCESSED = processString(sm4Key, 16, "s"); + SM4_TOKEN_PROCESSED = processString(sm4Token, 16, "s"); + + // 前端的 systemKey 转16进制（仅用于验证） + SYSTEM_KEY_HEX = stringToHex(systemKey); + + /*System.out.println("=== SM4Utils 初始化 ==="); + System.out.println("Go 的 Sm4Key: " + sm4Key); + System.out.println("Go 的 Sm4Key(处理后16字节): " + SM4_KEY_PROCESSED); + System.out.println("Go 的 Sm4Token: " + sm4Token); + System.out.println("Go 的 Sm4Token(处理后16字节): " + SM4_TOKEN_PROCESSED); + System.out.println("前端的 VITE_APP_SYSTEM_KEY: " + systemKey); + System.out.println("前端的 systemKey(hex): " + SYSTEM_KEY_HEX); + System.out.println("注意：Java 应该使用 Go 的 Sm4Key，而不是前端的 systemKey！");*/ + } + + /** + * 处理字符串为指定长度（与 Go 的 ProcessString 完全一致） + * 超过截断，不足用 pad 补全 + */ + private static String processString(String s, int length, String pad) { + // 如果字符串长度已经符合要求，直接返回 + if (s.length() == length) { + return s; + } + // 如果字符串长度超过了要求，截取指定长度的部分 + if (s.length() > length) { + return s.substring(0, length); + } + // 如果字符串长度不足，使用pad字符串进行补充 + StringBuilder sb = new StringBuilder(s); + while (sb.length() < length) { + sb.append(pad); + } + return sb.substring(0, length); // 确保不会超过长度 + } + + /** + * 字符串转16进制（仅用于与前端对比） + */ + private static String stringToHex(String str) { + byte[] bytes = str.getBytes(StandardCharsets.UTF_8); + StringBuilder hex = new StringBuilder(); + for (byte b : bytes) { + hex.append(String.format("%02x", b)); + } + return hex.toString(); + } + + /** + * 判断是否为32位16进制字符串 + */ + private static boolean isValidHex32(String str) { + if (str == null || str.length() != 32) return false; + return str.matches("[0-9a-fA-F]{32}"); + } + + /** + * 加密方法（与 Go 的 SM4Encrypt 完全一致） + * @param data 要加密的字符串数据 + * @param customKey 自定义密钥（Auth-key），如果提供则用作 IV + * @return 加密后的16进制字符串 + */ + public static String encrypt(String data, String customKey) throws Exception { + // 1. 确定加密密钥（使用 Go 的 Sm4Key） + String keyStr = SM4_KEY_PROCESSED; + byte[] keyBytes = keyStr.getBytes(StandardCharsets.UTF_8); + + // 2. 确定 IV（如果有 customKey 则用 customKey，否则用 Sm4Token） + String ivStr = SM4_TOKEN_PROCESSED; + if (customKey != null && !customKey.isEmpty()) { + // Go 的逻辑：如果有 customKey，处理为16字节 + ivStr = processString(customKey, 16, "s"); + } + byte[] ivBytes = ivStr.getBytes(StandardCharsets.UTF_8); + + /*System.out.println("=== Java 加密参数 ==="); + System.out.println("加密密钥: " + keyStr + " (UTF-8字节: " + bytesToHex(keyBytes) + ")"); + System.out.println("IV: " + ivStr + " (UTF-8字节: " + bytesToHex(ivBytes) + ")"); + System.out.println("原始数据: " + data);*/ + + // 3. 使用 BouncyCastle 进行 SM4-CBC 加密 + PaddedBufferedBlockCipher cipher = new PaddedBufferedBlockCipher( + new CBCBlockCipher(new SM4Engine()), new PKCS7Padding() + ); + + cipher.init(true, new ParametersWithIV(new KeyParameter(keyBytes), ivBytes)); + + byte[] inputBytes = data.getBytes(StandardCharsets.UTF_8); + byte[] outputBytes = new byte[cipher.getOutputSize(inputBytes.length)]; + + int length1 = cipher.processBytes(inputBytes, 0, inputBytes.length, outputBytes, 0); + int length2 = cipher.doFinal(outputBytes, length1); + + byte[] encryptedBytes = new byte[length1 + length2]; + System.arraycopy(outputBytes, 0, encryptedBytes, 0, encryptedBytes.length); + + String result = bytesToHex(encryptedBytes); + //System.out.println("加密结果(hex): " + result); + + return result; + } + + /** + * 解密方法（与 Go 的 SM4Decrypt 完全一致） + * @param encryptedHex 加密后的16进制字符串 + * @param customKey 自定义密钥（Auth-key），如果提供则用作 IV + * @return 解密后的字符串 + */ + public static String decrypt(String encryptedHex, String customKey) throws Exception { + // 1. 确定加密密钥（使用 Go 的 Sm4Key） + String keyStr = SM4_KEY_PROCESSED; + byte[] keyBytes = keyStr.getBytes(StandardCharsets.UTF_8); + + // 2. 确定 IV（如果有 customKey 则用 customKey，否则用 Sm4Token） + String ivStr = SM4_TOKEN_PROCESSED; + if (customKey != null && !customKey.isEmpty()) { + // Go 的逻辑：如果有 customKey，处理为16字节 + ivStr = processString(customKey, 16, "s"); + } + byte[] ivBytes = ivStr.getBytes(StandardCharsets.UTF_8); + + /*System.out.println("=== Java 解密参数 ==="); + System.out.println("解密密钥: " + keyStr + " (UTF-8字节: " + bytesToHex(keyBytes) + ")"); + System.out.println("IV: " + ivStr + " (UTF-8字节: " + bytesToHex(ivBytes) + ")"); + System.out.println("加密数据(hex): " + encryptedHex);*/ + + // 3. 转换加密数据 + byte[] encryptedBytes = hexToBytes(encryptedHex); + + // 4. 使用 BouncyCastle 进行 SM4-CBC 解密 + PaddedBufferedBlockCipher cipher = new PaddedBufferedBlockCipher( + new CBCBlockCipher(new SM4Engine()), new PKCS7Padding() + ); + + cipher.init(false, new ParametersWithIV(new KeyParameter(keyBytes), ivBytes)); + + byte[] outputBytes = new byte[cipher.getOutputSize(encryptedBytes.length)]; + + int length1 = cipher.processBytes(encryptedBytes, 0, encryptedBytes.length, outputBytes, 0); + int length2 = cipher.doFinal(outputBytes, length1); + + byte[] decryptedBytes = new byte[length1 + length2]; + System.arraycopy(outputBytes, 0, decryptedBytes, 0, decryptedBytes.length); + + String result = new String(decryptedBytes, StandardCharsets.UTF_8); + //System.out.println("解密结果: " + result); + + return result; + } + + /** + * 字节数组转16进制字符串 + */ + private static String bytesToHex(byte[] bytes) { + StringBuilder hex = new StringBuilder(); + for (byte b : bytes) { + hex.append(String.format("%02x", b)); + } + return hex.toString(); + } + + /** + * 16进制字符串转字节数组 + */ + private static byte[] hexToBytes(String hex) { + if (hex.length() % 2 != 0) { + throw new IllegalArgumentException("Hex string length must be even"); + } + byte[] bytes = new byte[hex.length() / 2]; + for (int i = 0; i < bytes.length; i++) { + int index = i * 2; + bytes[i] = (byte) Integer.parseInt(hex.substring(index, index + 2), 16); + } + return bytes; + } + + +} \ No newline at end of file diff --git a/src/main/java/com/hxgk/zxxy/utils/sm4/WebConfig.java b/src/main/java/com/hxgk/zxxy/utils/sm4/WebConfig.java new file mode 100644 index 0000000..1f4ed1c --- /dev/null +++ b/src/main/java/com/hxgk/zxxy/utils/sm4/WebConfig.java @@ -0,0 +1,40 @@ +package com.hxgk.zxxy.utils.sm4; + + + +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.context.annotation.Configuration; +import org.springframework.web.method.support.HandlerMethodArgumentResolver; +import org.springframework.web.servlet.config.annotation.InterceptorRegistry; +import org.springframework.web.servlet.config.annotation.WebMvcConfigurer; + +import java.util.List; + +@Configuration +public class WebConfig implements WebMvcConfigurer { + + @Autowired + private SM4Interceptor sm4Interceptor; + + @Autowired + private DecryptedArgumentResolver decryptedArgumentResolver; + + @Override + public void addInterceptors(InterceptorRegistry registry) { + registry.addInterceptor(sm4Interceptor) + .addPathPatterns("/**") + .excludePathPatterns( + "/error", + "/api/sm4-test/**", // 排除测试接口 + "/actuator/**", // 排除监控接口 + "/swagger-ui/**", // 排除 Swagger + "/v2/api-docs", // 排除 API 文档 + "/webjars/**" // 排除静态资源 + ); + } + + @Override + public void addArgumentResolvers(List resolvers) { + resolvers.add(decryptedArgumentResolver); + } +} \ No newline at end of file diff --git a/src/main/resources/application-dev.yml b/src/main/resources/application-dev.yml index ddd4ca8..b7a9a6d 100644 --- a/src/main/resources/application-dev.yml +++ b/src/main/resources/application-dev.yml @@ -9,6 +9,10 @@ mybatis: #配置xml的结果别名 type-aliases-package: com.hxgk.zxxy.model.entity spring: + jackson: + default-property-inclusion: NON_NULL + date-format: yyyy-MM-dd HH:mm:ss + time-zone: GMT+8 autoconfigure: #去除durid配置 exclude: com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceAutoConfigure @@ -115,3 +119,13 @@ spring: port: 6379 password: Redis+brngJ3U19@8_Z2^7a timeout: 10000 + +# Go 后端的 SM4 配置（Java 应该使用这些） +sm4: + key: "hengxingaokeApp1" # Go 的 Sm4Key + token: "04TzMuvkHm_EZnHm" # Go 的 Sm4Token + + + system-key: "appKeyQinDongV01" + app-key: "04TzMuvkHm_EZnHm" + diff --git a/src/main/resources/application-prodout.yml b/src/main/resources/application-prodout.yml index 285fcd1..cd5b58e 100644 --- a/src/main/resources/application-prodout.yml +++ b/src/main/resources/application-prodout.yml @@ -9,6 +9,10 @@ mybatis: #配置xml的结果别名 type-aliases-package: com.hxgk.zxxy.model.entity spring: + jackson: + default-property-inclusion: NON_NULL + date-format: yyyy-MM-dd HH:mm:ss + time-zone: GMT+8 autoconfigure: #去除durid配置 exclude: com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceAutoConfigure @@ -114,3 +118,11 @@ spring: min-idle: 0 port: 6379 timeout: 10000 +# Go 后端的 SM4 配置（Java 应该使用这些） +sm4: + key: "hengxingaokeApp1" # Go 的 Sm4Key + token: "04TzMuvkHm_EZnHm" # Go 的 Sm4Token + + + system-key: "appKeyQinDongV01" + app-key: "04TzMuvkHm_EZnHm" diff --git a/src/main/resources/application.yml b/src/main/resources/application.yml index 63bc755..0f28338 100644 --- a/src/main/resources/application.yml +++ b/src/main/resources/application.yml @@ -1,5 +1,5 @@ spring: profiles: - active: prodout + active: dev server: port: 8111 \ No newline at end of file