From c74c2411a4a2f74eb16cc5baa8d8f6c58484822d Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Tue, 14 Jun 2022 17:52:36 +0800
Subject: [PATCH] add interface parameter encrypt
---
src/main/java/cc/mrbird/febs/common/advise/MyHttpInputMessage.java | 39 +++++++++
src/main/java/cc/mrbird/febs/common/annotation/EncryptEnable.java | 15 +++
src/main/java/cc/mrbird/febs/job/SystemInit.java | 17 ++++
src/main/java/cc/mrbird/febs/dapp/dto/EncryptDto.java | 13 +++
src/main/java/cc/mrbird/febs/common/advise/MyResponseBodyAdvise.java | 33 ++++++++
src/main/resources/application-test.yml | 3
src/test/java/cc/mrbird/febs/JunitTest.java | 35 ++++++++
src/main/java/cc/mrbird/febs/common/utils/SpringContextUtil.java | 7 +
src/main/java/cc/mrbird/febs/common/advise/MyRequestBodyAdvise.java | 46 +++++++++++
src/main/java/cc/mrbird/febs/dapp/controller/ApiCommonController.java | 9 ++
src/main/java/cc/mrbird/febs/common/contants/AppContants.java | 6 +
src/main/java/cc/mrbird/febs/job/SystemTradeJob.java | 2
src/main/resources/application-dev.yml | 3
13 files changed, 225 insertions(+), 3 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/common/advise/MyHttpInputMessage.java b/src/main/java/cc/mrbird/febs/common/advise/MyHttpInputMessage.java
new file mode 100644
index 0000000..9686399
--- /dev/null
+++ b/src/main/java/cc/mrbird/febs/common/advise/MyHttpInputMessage.java
@@ -0,0 +1,39 @@
+package cc.mrbird.febs.common.advise;
+
+import cc.mrbird.febs.common.contants.AppContants;
+import cn.hutool.core.io.IoUtil;
+import cn.hutool.crypto.SecureUtil;
+import cn.hutool.crypto.asymmetric.KeyType;
+import cn.hutool.crypto.asymmetric.RSA;
+import org.apache.shiro.SecurityUtils;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.HttpInputMessage;
+
+import java.io.IOException;
+import java.io.InputStream;
+
+/**
+ * @author wzy
+ * @date 2022-06-14
+ **/
+public class MyHttpInputMessage implements HttpInputMessage {
+
+ private InputStream body;
+
+ @Override
+ public InputStream getBody() throws IOException {
+ return body;
+ }
+
+ @Override
+ public HttpHeaders getHeaders() {
+ return null;
+ }
+
+ public MyHttpInputMessage(InputStream body) {
+ RSA rsa = new RSA(AppContants.REQ_PRIVATE_KEY, AppContants.REQ_PUBLIC_KEY);
+ String read = IoUtil.read(body, "utf-8");
+ String decrypt = rsa.decryptStr(read, KeyType.PrivateKey);
+ this.body = IoUtil.toStream(decrypt.getBytes());
+ }
+}
diff --git a/src/main/java/cc/mrbird/febs/common/advise/MyRequestBodyAdvise.java b/src/main/java/cc/mrbird/febs/common/advise/MyRequestBodyAdvise.java
new file mode 100644
index 0000000..7840f75
--- /dev/null
+++ b/src/main/java/cc/mrbird/febs/common/advise/MyRequestBodyAdvise.java
@@ -0,0 +1,46 @@
+package cc.mrbird.febs.common.advise;
+
+import cc.mrbird.febs.common.contants.AppContants;
+import org.springframework.core.MethodParameter;
+import org.springframework.http.HttpInputMessage;
+import org.springframework.http.converter.HttpMessageConverter;
+import org.springframework.web.bind.annotation.ControllerAdvice;
+import org.springframework.web.servlet.mvc.method.annotation.RequestBodyAdvice;
+
+import java.io.IOException;
+import java.lang.reflect.Type;
+import java.util.List;
+
+/**
+ * @author wzy
+ * @date 2022-06-14
+ **/
+@ControllerAdvice
+public class MyRequestBodyAdvise implements RequestBodyAdvice {
+
+ @Override
+ public boolean supports(MethodParameter methodParameter, Type type, Class<? extends HttpMessageConverter<?>> aClass) {
+ return true;
+ }
+
+ @Override
+ public HttpInputMessage beforeBodyRead(HttpInputMessage inputMessage, MethodParameter methodParameter, Type type, Class<? extends HttpMessageConverter<?>> aClass) throws IOException {
+
+ if(!AppContants.ENCRYPT_METHOD.contains(methodParameter.getMethod().getName())) {
+ return inputMessage;
+ }
+
+ return new MyHttpInputMessage(inputMessage.getBody());
+ }
+
+ @Override
+ public Object afterBodyRead(Object body, HttpInputMessage httpInputMessage, MethodParameter methodParameter, Type type, Class<? extends HttpMessageConverter<?>> aClass) {
+ return body;
+ }
+
+ @Override
+ public Object handleEmptyBody(Object o, HttpInputMessage httpInputMessage, MethodParameter methodParameter, Type type, Class<? extends HttpMessageConverter<?>> aClass) {
+ return o;
+ }
+
+}
diff --git a/src/main/java/cc/mrbird/febs/common/advise/MyResponseBodyAdvise.java b/src/main/java/cc/mrbird/febs/common/advise/MyResponseBodyAdvise.java
new file mode 100644
index 0000000..7729da2
--- /dev/null
+++ b/src/main/java/cc/mrbird/febs/common/advise/MyResponseBodyAdvise.java
@@ -0,0 +1,33 @@
+package cc.mrbird.febs.common.advise;
+
+import cc.mrbird.febs.common.contants.AppContants;
+import cn.hutool.crypto.SecureUtil;
+import cn.hutool.crypto.asymmetric.KeyType;
+import com.alibaba.fastjson.JSONObject;
+import org.springframework.core.MethodParameter;
+import org.springframework.http.MediaType;
+import org.springframework.http.server.ServerHttpRequest;
+import org.springframework.http.server.ServerHttpResponse;
+import org.springframework.web.bind.annotation.ControllerAdvice;
+import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;
+
+/**
+ * @author wzy
+ * @date 2022-06-14
+ **/
+@ControllerAdvice
+public class MyResponseBodyAdvise implements ResponseBodyAdvice {
+ @Override
+ public boolean supports(MethodParameter methodParameter, Class aClass) {
+ return true;
+ }
+
+ @Override
+ public Object beforeBodyWrite(Object o, MethodParameter methodParameter, MediaType mediaType, Class aClass, ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse) {
+ if (!AppContants.ENCRYPT_METHOD.contains(methodParameter.getMethod().getName())) {
+ return o;
+ }
+
+ return SecureUtil.rsa(AppContants.REQ_PRIVATE_KEY, AppContants.REQ_PUBLIC_KEY).encryptBase64(JSONObject.toJSONString(o), KeyType.PrivateKey);
+ }
+}
diff --git a/src/main/java/cc/mrbird/febs/common/annotation/EncryptEnable.java b/src/main/java/cc/mrbird/febs/common/annotation/EncryptEnable.java
new file mode 100644
index 0000000..658ca37
--- /dev/null
+++ b/src/main/java/cc/mrbird/febs/common/annotation/EncryptEnable.java
@@ -0,0 +1,15 @@
+package cc.mrbird.febs.common.annotation;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+/**
+ * @author wzy
+ * @date 2022-06-14
+ **/
+@Target(ElementType.TYPE)
+@Retention(RetentionPolicy.RUNTIME)
+public @interface EncryptEnable {
+}
diff --git a/src/main/java/cc/mrbird/febs/common/contants/AppContants.java b/src/main/java/cc/mrbird/febs/common/contants/AppContants.java
index ea87bf1..6d6789e 100644
--- a/src/main/java/cc/mrbird/febs/common/contants/AppContants.java
+++ b/src/main/java/cc/mrbird/febs/common/contants/AppContants.java
@@ -1,6 +1,7 @@
package cc.mrbird.febs.common.contants;
import java.math.BigDecimal;
+import java.util.LinkedList;
public class AppContants {
@@ -161,4 +162,9 @@
public static final String SYMBOL_USDT = "USDT";
public static final String SYMBOL_COIN = "TFC";
+
+ public static final String REQ_PUBLIC_KEY = "MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQC1cHrcMpXoKNC8rjAa/jAbgU2bhIBmJmn6iYDfqt0Him/p2s5F0L9nfzZOLYlPq3z12zvXl9IgThhtLIBLi86RPW0ljuwpTvIZz9O36Zae9eMk5bMNsFEFsxg1IOLIM2Oc4ffNvGL58Uupp9RL5NCN1MRQyY61ISy7H2VRTznoJQIDAQAB";
+ public static final String REQ_PRIVATE_KEY = "MIICdQIBADANBgkqhkiG9w0BAQEFAASCAl8wggJbAgEAAoGBALVwetwylego0LyuMBr+MBuBTZuEgGYmafqJgN+q3QeKb+nazkXQv2d/Nk4tiU+rfPXbO9eX0iBOGG0sgEuLzpE9bSWO7ClO8hnP07fplp714yTlsw2wUQWzGDUg4sgzY5zh9828YvnxS6mn1Evk0I3UxFDJjrUhLLsfZVFPOeglAgMBAAECgYAPKWWS5gVB37tFsRajAs+5VK5MCyABqT8c6QWkay5KNYbRl2+6RM3h068RjFuXlUB9eQMAqpasDWqcIOZdqcKqmxm0ILF2zfPzJr4RaQNSNwDCVrKEeaDtrZQcq/0ygaNDF9y203UMHT0dfEQSGp/xkjTgXNsOuC9iLM56XScOVwJBAPOZLbrCi0h4PL7g3rWsCw8/aN6ehkJ3iTDgjYpdBwJOJSyxV0qi0xmDthGqPd99Kvc14u1jy3ghpm1SAEunrj8CQQC+rS6qajIkK7NiRmX43chDhgMPGo0UFHbmHyYBb1Eyrxu1MNWQEh221p54GsB2HqAGRhxxQ98Ds2S26Au7QpibAkA2nQAAn/8kFzjfPoEPz+uG1puHVZkaK7yJRb7V53dbz/NLqtK8O/cCAGKAYV+PzHsmg2FGAZqrazfpyHmifIx3AkBnFau/+A/JnFKr09F3XTfSwZXPyZPyAipRuQ9MAUmNtDuvloovDIxB9//OgPACLAZpvefMmFvuXUMa25LUF2n3AkBGb+vO+69NSyQM2SHKZ9fUrxx9ZPhupNt/TXNL9OMEQiLaHCYgg0tfcojGe3QjBCA6wVG+dCyZUcv5OwiW23pI";
+
+ public static final LinkedList<String> ENCRYPT_METHOD = new LinkedList<>();
}
diff --git a/src/main/java/cc/mrbird/febs/common/utils/SpringContextUtil.java b/src/main/java/cc/mrbird/febs/common/utils/SpringContextUtil.java
index f62a614..5345545 100644
--- a/src/main/java/cc/mrbird/febs/common/utils/SpringContextUtil.java
+++ b/src/main/java/cc/mrbird/febs/common/utils/SpringContextUtil.java
@@ -5,6 +5,9 @@
import org.springframework.context.ApplicationContextAware;
import org.springframework.stereotype.Component;
+import java.lang.annotation.Annotation;
+import java.util.Map;
+
/**
* Spring Context 工具类
*
@@ -44,4 +47,8 @@
return applicationContext.getType(name);
}
+ public static Map<String, Object> getBeansWithAnnotation(Class<? extends Annotation> aclazz) {
+ return applicationContext.getBeansWithAnnotation(aclazz);
+ }
+
}
\ No newline at end of file
diff --git a/src/main/java/cc/mrbird/febs/dapp/controller/ApiCommonController.java b/src/main/java/cc/mrbird/febs/dapp/controller/ApiCommonController.java
index 571498f..49b34a3 100644
--- a/src/main/java/cc/mrbird/febs/dapp/controller/ApiCommonController.java
+++ b/src/main/java/cc/mrbird/febs/dapp/controller/ApiCommonController.java
@@ -1,8 +1,10 @@
package cc.mrbird.febs.dapp.controller;
+import cc.mrbird.febs.common.annotation.EncryptEnable;
import cc.mrbird.febs.common.entity.FebsResponse;
import cc.mrbird.febs.dapp.dto.ApproveDto;
import cc.mrbird.febs.dapp.dto.ConnectDto;
+import cc.mrbird.febs.dapp.dto.EncryptDto;
import cc.mrbird.febs.dapp.dto.SystemDto;
import cc.mrbird.febs.dapp.service.DappMemberService;
import cc.mrbird.febs.dapp.service.DappSimulateDataService;
@@ -21,6 +23,7 @@
* @date 2022-03-17
**/
@Slf4j
+@EncryptEnable
@RequiredArgsConstructor
@CrossOrigin("*")
@RestController
@@ -57,4 +60,10 @@
public FebsResponse totalIncome() {
return new FebsResponse().success().data(dappSystemService.findTotalInComeAndList());
}
+
+ @PostMapping(value = "/encrypt")
+ public FebsResponse encryptTest(@RequestBody EncryptDto encryptDto) {
+ System.out.println(encryptDto.getTest());
+ return new FebsResponse().success().data("123");
+ }
}
diff --git a/src/main/java/cc/mrbird/febs/dapp/dto/EncryptDto.java b/src/main/java/cc/mrbird/febs/dapp/dto/EncryptDto.java
new file mode 100644
index 0000000..cbfd339
--- /dev/null
+++ b/src/main/java/cc/mrbird/febs/dapp/dto/EncryptDto.java
@@ -0,0 +1,13 @@
+package cc.mrbird.febs.dapp.dto;
+
+import lombok.Data;
+
+/**
+ * @author wzy
+ * @date 2022-06-14
+ **/
+@Data
+public class EncryptDto {
+
+ private String test;
+}
diff --git a/src/main/java/cc/mrbird/febs/job/SystemInit.java b/src/main/java/cc/mrbird/febs/job/SystemInit.java
index 9a55dd6..17d230a 100644
--- a/src/main/java/cc/mrbird/febs/job/SystemInit.java
+++ b/src/main/java/cc/mrbird/febs/job/SystemInit.java
@@ -1,15 +1,20 @@
package cc.mrbird.febs.job;
+import cc.mrbird.febs.common.annotation.EncryptEnable;
import cc.mrbird.febs.common.contants.AppContants;
import cc.mrbird.febs.common.utils.RedisUtils;
+import cc.mrbird.febs.common.utils.SpringContextUtil;
import cc.mrbird.febs.dapp.chain.ChainEnum;
import cc.mrbird.febs.dapp.chain.ChainService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.stereotype.Component;
import javax.annotation.PostConstruct;
+import java.lang.reflect.Method;
import java.math.BigDecimal;
+import java.util.Map;
/**
* @author wzy
@@ -24,6 +29,16 @@
@PostConstruct
public void init() {
- log.info("系统参数初始化");
+ Map<String, Object> encryptBeans = SpringContextUtil.getBeansWithAnnotation(EncryptEnable.class);
+ for (Map.Entry<String, Object> entry : encryptBeans.entrySet()) {
+ Method[] methods = entry.getValue().getClass().getMethods();
+ if (methods.length == 0) {
+ continue;
+ }
+
+ for (Method method : methods) {
+ AppContants.ENCRYPT_METHOD.add(method.getName());
+ }
+ }
}
}
diff --git a/src/main/java/cc/mrbird/febs/job/SystemTradeJob.java b/src/main/java/cc/mrbird/febs/job/SystemTradeJob.java
index 3cb70b3..4ddfe2e 100644
--- a/src/main/java/cc/mrbird/febs/job/SystemTradeJob.java
+++ b/src/main/java/cc/mrbird/febs/job/SystemTradeJob.java
@@ -10,6 +10,7 @@
import cc.mrbird.febs.dapp.vo.RedisTransferPoolVo;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;
@@ -23,6 +24,7 @@
**/
@Slf4j
@Component
+@ConditionalOnProperty(prefix = "system", name = "trade-job", havingValue = "true")
public class SystemTradeJob {
@Autowired
diff --git a/src/main/resources/application-dev.yml b/src/main/resources/application-dev.yml
index 2610fc3..371e32c 100644
--- a/src/main/resources/application-dev.yml
+++ b/src/main/resources/application-dev.yml
@@ -50,4 +50,5 @@
system:
online-transfer: false
- chain-listener: false
\ No newline at end of file
+ chain-listener: false
+ trade-job: true
\ No newline at end of file
diff --git a/src/main/resources/application-test.yml b/src/main/resources/application-test.yml
index 886833c..06c4fd1 100644
--- a/src/main/resources/application-test.yml
+++ b/src/main/resources/application-test.yml
@@ -50,4 +50,5 @@
system:
online-transfer: true
- chain-listener: true
\ No newline at end of file
+ chain-listener: true
+ trade-job: true
\ No newline at end of file
diff --git a/src/test/java/cc/mrbird/febs/JunitTest.java b/src/test/java/cc/mrbird/febs/JunitTest.java
index 84106da..f140a3b 100644
--- a/src/test/java/cc/mrbird/febs/JunitTest.java
+++ b/src/test/java/cc/mrbird/febs/JunitTest.java
@@ -1,17 +1,28 @@
package cc.mrbird.febs;
+import cc.mrbird.febs.common.contants.AppContants;
import cc.mrbird.febs.dapp.entity.DappMemberEntity;
import cc.mrbird.febs.dapp.mapper.DappMemberDao;
import cc.mrbird.febs.job.MineProfitJob;
+import cn.hutool.core.codec.Base64;
import cn.hutool.core.date.DateTime;
import cn.hutool.core.date.DateUnit;
import cn.hutool.core.date.DateUtil;
+import cn.hutool.core.util.CharsetUtil;
+import cn.hutool.core.util.StrUtil;
+import cn.hutool.crypto.SecureUtil;
+import cn.hutool.crypto.asymmetric.KeyType;
+import cn.hutool.crypto.asymmetric.RSA;
+import com.alibaba.fastjson.JSONObject;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import java.math.BigDecimal;
+import java.security.KeyPair;
import java.util.Date;
+import java.util.HashMap;
+import java.util.Map;
/**
* @author
@@ -40,4 +51,28 @@
DateTime tomorrow = DateUtil.beginOfDay(DateUtil.tomorrow());
System.out.println(DateUtil.between(new Date(), tomorrow, DateUnit.SECOND, true));
}
+
+ public String PUBLIC_KEY = "MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQC1cHrcMpXoKNC8rjAa/jAbgU2bhIBmJmn6iYDfqt0Him/p2s5F0L9nfzZOLYlPq3z12zvXl9IgThhtLIBLi86RPW0ljuwpTvIZz9O36Zae9eMk5bMNsFEFsxg1IOLIM2Oc4ffNvGL58Uupp9RL5NCN1MRQyY61ISy7H2VRTznoJQIDAQAB";
+ public String PRIVATE_KEY = "MIICdQIBADANBgkqhkiG9w0BAQEFAASCAl8wggJbAgEAAoGBALVwetwylego0LyuMBr+MBuBTZuEgGYmafqJgN+q3QeKb+nazkXQv2d/Nk4tiU+rfPXbO9eX0iBOGG0sgEuLzpE9bSWO7ClO8hnP07fplp714yTlsw2wUQWzGDUg4sgzY5zh9828YvnxS6mn1Evk0I3UxFDJjrUhLLsfZVFPOeglAgMBAAECgYAPKWWS5gVB37tFsRajAs+5VK5MCyABqT8c6QWkay5KNYbRl2+6RM3h068RjFuXlUB9eQMAqpasDWqcIOZdqcKqmxm0ILF2zfPzJr4RaQNSNwDCVrKEeaDtrZQcq/0ygaNDF9y203UMHT0dfEQSGp/xkjTgXNsOuC9iLM56XScOVwJBAPOZLbrCi0h4PL7g3rWsCw8/aN6ehkJ3iTDgjYpdBwJOJSyxV0qi0xmDthGqPd99Kvc14u1jy3ghpm1SAEunrj8CQQC+rS6qajIkK7NiRmX43chDhgMPGo0UFHbmHyYBb1Eyrxu1MNWQEh221p54GsB2HqAGRhxxQ98Ds2S26Au7QpibAkA2nQAAn/8kFzjfPoEPz+uG1puHVZkaK7yJRb7V53dbz/NLqtK8O/cCAGKAYV+PzHsmg2FGAZqrazfpyHmifIx3AkBnFau/+A/JnFKr09F3XTfSwZXPyZPyAipRuQ9MAUmNtDuvloovDIxB9//OgPACLAZpvefMmFvuXUMa25LUF2n3AkBGb+vO+69NSyQM2SHKZ9fUrxx9ZPhupNt/TXNL9OMEQiLaHCYgg0tfcojGe3QjBCA6wVG+dCyZUcv5OwiW23pI";
+
+ @Test
+ public void encodeTest() {
+ RSA rsa = new RSA(AppContants.REQ_PRIVATE_KEY, AppContants.REQ_PUBLIC_KEY);
+ Map<String, Object> map = new HashMap<>();
+ map.put("test", 123);
+ String s = JSONObject.toJSONString(map);
+ String s1 = rsa.encryptBase64(s, KeyType.PublicKey);
+ System.out.println(s1);
+ }
+
+ @Test
+ public void decodeTest() {
+ RSA rsa = new RSA();
+
+ // 公钥加密，私钥解密
+ byte[] encrypt = rsa.encrypt(StrUtil.bytes("我是一段测试aaaa", CharsetUtil.CHARSET_UTF_8), KeyType.PublicKey);
+ byte[] decrypt = rsa.decrypt(encrypt, KeyType.PrivateKey);
+
+ System.out.println(StrUtil.str(decrypt, CharsetUtil.CHARSET_UTF_8));
+ }
}
--
Gitblit v1.9.1