| | |
|---|
| | | import org.dromara.common.encrypt.annotation.ApiEncrypt; |
|---|
| | | import org.dromara.common.encrypt.properties.ApiDecryptProperties; |
|---|
| | | import org.springframework.http.HttpMethod; |
|---|
| | | import org.springframework.http.MediaType; |
|---|
| | | import org.springframework.web.method.HandlerMethod; |
|---|
| | | import org.springframework.web.servlet.HandlerExceptionResolver; |
|---|
| | | import org.springframework.web.servlet.HandlerExecutionChain; |
|---|
| | | import org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerMapping; |
|---|
| | | |
|---|
| | | import java.io.IOException; |
|---|
| | | import java.io.PrintWriter; |
|---|
| | | |
|---|
| | | |
|---|
| | | /** |
|---|
| | |
|---|
| | | public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException { |
|---|
| | | HttpServletRequest servletRequest = (HttpServletRequest) request; |
|---|
| | | HttpServletResponse servletResponse = (HttpServletResponse) response; |
|---|
| | | |
|---|
| | | boolean responseFlag = false; |
|---|
| | | // 获取加密注解 |
|---|
| | | ApiEncrypt apiEncrypt = this.getApiEncryptAnnotation(servletRequest); |
|---|
| | | boolean responseFlag = apiEncrypt != null && apiEncrypt.response(); |
|---|
| | | ServletRequest requestWrapper = null; |
|---|
| | | ServletResponse responseWrapper = null; |
|---|
| | | EncryptResponseBodyWrapper responseBodyWrapper = null; |
|---|
| | | |
|---|
| | | // 是否为 json 请求 |
|---|
| | | if (StringUtils.startsWithIgnoreCase(request.getContentType(), MediaType.APPLICATION_JSON_VALUE)) { |
|---|
| | | // 是否为 put 或者 post 请求 |
|---|
| | | if (HttpMethod.PUT.matches(servletRequest.getMethod()) || HttpMethod.POST.matches(servletRequest.getMethod())) { |
|---|
| | | // 是否存在加密标头 |
|---|
| | | String headerValue = servletRequest.getHeader(properties.getHeaderFlag()); |
|---|
| | | // 获取加密注解 |
|---|
| | | ApiEncrypt apiEncrypt = this.getApiEncryptAnnotation(servletRequest); |
|---|
| | | responseFlag = apiEncrypt != null && apiEncrypt.response(); |
|---|
| | | if (StringUtils.isNotBlank(headerValue)) { |
|---|
| | | // 请求解密 |
|---|
| | | requestWrapper = new DecryptRequestBodyWrapper(servletRequest, properties.getPrivateKey(), properties.getHeaderFlag()); |
|---|
| | |
|---|
| | | return; |
|---|
| | | } |
|---|
| | | } |
|---|
| | | } |
|---|
| | | |
|---|
| | | // 判断是否响应加密 |
|---|
| | | if (responseFlag) { |
|---|
| | | responseBodyWrapper = new EncryptResponseBodyWrapper(servletResponse); |
|---|
| | | responseWrapper = responseBodyWrapper; |
|---|
| | | } |
|---|
| | | } |
|---|
| | | } |
|---|
| | | |
|---|
| | | chain.doFilter( |
|---|
