diff --git a/wfc-common/wfc-common-license/src/main/java/org/wfc/common/license/domain/LicenseCheckModel.java b/wfc-common/wfc-common-license/src/main/java/org/wfc/common/license/domain/LicenseCheckModel.java
index 5786c69..37a6336 100644
--- a/wfc-common/wfc-common-license/src/main/java/org/wfc/common/license/domain/LicenseCheckModel.java
+++ b/wfc-common/wfc-common-license/src/main/java/org/wfc/common/license/domain/LicenseCheckModel.java
@@ -70,4 +70,29 @@ public class LicenseCheckModel implements Serializable {
", mainBoardSerial='" + mainBoardSerial + '\'' +
'}';
}
+
+ public String toJsonString() {
+ StringBuilder jsonBuilder = new StringBuilder();
+ jsonBuilder.append("{");
+// jsonBuilder.append("\"ipAddress\": [");
+// for (String ip : ipAddress) {
+// jsonBuilder.append("\"").append(ip).append("\", ");
+// }
+// jsonBuilder.deleteCharAt(jsonBuilder.length() - 1);
+// jsonBuilder.deleteCharAt(jsonBuilder.length() - 1);
+// jsonBuilder.append("], ");
+// jsonBuilder.append("\"macAddress\": [");
+// for (String mac : macAddress) {
+// jsonBuilder.append("\"").append(mac).append("\", ");
+// }
+// jsonBuilder.deleteCharAt(jsonBuilder.length() - 1);
+// jsonBuilder.deleteCharAt(jsonBuilder.length() - 1);
+// jsonBuilder.append("], ");
+ jsonBuilder.append("\"cpuSerial\": \"").append(cpuSerial).append("\", ");
+ jsonBuilder.append("\"mainBoardSerial\": \"").append(mainBoardSerial).append("\"");
+// jsonBuilder.append("\"registerAmount\": \"").append(registerAmount).append("\"");
+ jsonBuilder.append("}");
+
+ return jsonBuilder.toString();
+ }
}
diff --git a/wfc-common/wfc-common-license/src/main/java/org/wfc/common/license/runner/LicenseCheckRunner.java b/wfc-common/wfc-common-license/src/main/java/org/wfc/common/license/runner/LicenseCheckRunner.java
index e015321..f5a3ea6 100644
--- a/wfc-common/wfc-common-license/src/main/java/org/wfc/common/license/runner/LicenseCheckRunner.java
+++ b/wfc-common/wfc-common-license/src/main/java/org/wfc/common/license/runner/LicenseCheckRunner.java
@@ -1,6 +1,7 @@
package org.wfc.common.license.runner;
import cn.hutool.core.util.StrUtil;
+import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
@@ -8,13 +9,19 @@ import org.springframework.boot.ApplicationArguments;
import org.springframework.boot.ApplicationRunner;
import org.springframework.stereotype.Component;
import org.wfc.common.license.LicenseVerify;
+import org.wfc.common.license.domain.LicenseCheckModel;
import org.wfc.common.license.domain.LicenseVerifyParam;
+import org.wfc.common.license.serverinfo.AbstractServerInfos;
+import javax.crypto.Cipher;
+import javax.crypto.spec.SecretKeySpec;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
+import java.io.FileWriter;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
+import java.util.Base64;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.ScheduledFuture;
@@ -31,6 +38,8 @@ public class LicenseCheckRunner implements ApplicationRunner {
private final ScheduledExecutorService scheduler = Executors.newSingleThreadScheduledExecutor();
private ScheduledFuture scheduledFuture;
+ private static final String key = "wfcwanfiAdmin6666";
+
// 启动定时任务
public void startTimer() {
@@ -114,6 +123,73 @@ public class LicenseCheckRunner implements ApplicationRunner {
log.info("++++++++ 证书安装结束 ++++++++");
}
+ private void generate() {
+ try {
+ LicenseCheckModel serverInfos = AbstractServerInfos.getServer("").getServerInfos();
+
+ //对象序列化为json
+ String jsonString = serverInfos.toJsonString();
+ System.out.println("原始数据 = " + jsonString);
+ //加密
+ String encryptedJson = encrypt(jsonString);
+ System.out.println("加密后 = " + encryptedJson);
+ // 解密
+ String decryptedJson = decrypt(encryptedJson);
+ System.out.println("解密后 = " + decryptedJson);
+ // 比较解密后的字符串与原始字符串
+ if (jsonString.equals(decryptedJson)) {
+ } else {
+ log.info("解密后的字符串与原始字符串不同");
+ }
+
+ // 断言
+ assert jsonString.equals(decryptedJson) : "解密后的字符串与原始字符串不匹配";
+
+ // 反序列化为对象
+ ObjectMapper mapper = new ObjectMapper();
+ LicenseCheckModel licenseParam = null;
+ licenseParam = mapper.readValue(decryptedJson, LicenseCheckModel.class);
+ // 标记为null的字段需要特殊处理
+// if ("null".equals(licenseParam.getRegisterAmount())) {
+// licenseParam.setRegisterAmount(null);
+// }
+ FileWriter writer = new FileWriter(licenseProperties.getCodePath());
+ writer.write(encryptedJson);
+ writer.close();
+ log.info("licenseParam {}", licenseParam);
+ } catch (Exception e) {
+ log.error("生成激活码失败", e);
+ }
+ }
+
+
+ private String encrypt(String json) throws Exception {
+ byte[] keyBytes = key.getBytes();
+ MessageDigest sha = MessageDigest.getInstance("SHA-256");
+ keyBytes = sha.digest(keyBytes);
+ keyBytes = java.util.Arrays.copyOf(keyBytes, 16);
+ SecretKeySpec secretKey = new SecretKeySpec(keyBytes, "AES");
+
+ Cipher cipher = Cipher.getInstance("AES/ECB/PKCS5PADDING");
+ cipher.init(Cipher.ENCRYPT_MODE, secretKey);
+ byte[] encryptedBytes = cipher.doFinal(json.getBytes());
+ return Base64.getEncoder().encodeToString(encryptedBytes);
+ }
+
+ private String decrypt(String encryptedJson) throws Exception {
+ byte[] keyBytes = key.getBytes();
+ MessageDigest sha = MessageDigest.getInstance("SHA-256");
+ keyBytes = sha.digest(keyBytes);
+ keyBytes = java.util.Arrays.copyOf(keyBytes, 16);
+ SecretKeySpec secretKey = new SecretKeySpec(keyBytes, "AES");
+
+ Cipher cipher = Cipher.getInstance("AES/ECB/PKCS5PADDING");
+ cipher.init(Cipher.DECRYPT_MODE, secretKey);
+ byte[] encryptedBytes = Base64.getDecoder().decode(encryptedJson);
+ byte[] decryptedBytes = cipher.doFinal(encryptedBytes);
+ return new String(decryptedBytes);
+ }
+
/**
* 获取文件的md5
*/
@@ -159,6 +235,7 @@ public class LicenseCheckRunner implements ApplicationRunner {
@Override
public void run(ApplicationArguments args) throws Exception {
+ generate();
install();
startTimer();
}
diff --git a/wfc-common/wfc-common-license/src/main/java/org/wfc/common/license/runner/LicenseProperties.java b/wfc-common/wfc-common-license/src/main/java/org/wfc/common/license/runner/LicenseProperties.java
index cfb5a48..1b286f1 100644
--- a/wfc-common/wfc-common-license/src/main/java/org/wfc/common/license/runner/LicenseProperties.java
+++ b/wfc-common/wfc-common-license/src/main/java/org/wfc/common/license/runner/LicenseProperties.java
@@ -32,6 +32,11 @@ public class LicenseProperties {
*/
private String licensePath = "/opt/wfc/license/license.lic";
+ /**
+ * 激活码生成路径
+ */
+ private String codePath = "/opt/wfc/license/activation_code.txt";
+
/**
* 密钥库存储路径
*/
diff --git a/wfc-common/wfc-common-license/src/main/java/org/wfc/common/license/serverinfo/AbstractServerInfos.java b/wfc-common/wfc-common-license/src/main/java/org/wfc/common/license/serverinfo/AbstractServerInfos.java
index 9d52ade..3fc123c 100644
--- a/wfc-common/wfc-common-license/src/main/java/org/wfc/common/license/serverinfo/AbstractServerInfos.java
+++ b/wfc-common/wfc-common-license/src/main/java/org/wfc/common/license/serverinfo/AbstractServerInfos.java
@@ -20,17 +20,18 @@ public abstract class AbstractServerInfos {
* 组装需要额外校验的License参数
*
* @return demo.LicenseCheckModel
+ * @since 1.0.0
*/
public LicenseCheckModel getServerInfos() {
LicenseCheckModel result = new LicenseCheckModel();
try {
- result.setIpAddress(this.getIpAddress());
- result.setMacAddress(this.getMacAddress());
+// result.setIpAddress(this.getIpAddress());
+// result.setMacAddress(this.getMacAddress());
result.setCpuSerial(this.getCPUSerial());
result.setMainBoardSerial(this.getMainBoardSerial());
} catch (Exception e) {
- log.error("获取服务器硬件信息失败 {}", e.getMessage());
+ log.error("获取服务器硬件信息失败", e);
}
return result;
@@ -86,7 +87,6 @@ public abstract class AbstractServerInfos {
}
}
}
-
return result;
}
@@ -100,12 +100,10 @@ public abstract class AbstractServerInfos {
try {
byte[] mac = NetworkInterface.getByInetAddress(inetAddr).getHardwareAddress();
StringBuffer stringBuffer = new StringBuffer();
-
for (int i = 0; i < mac.length; i++) {
if (i != 0) {
stringBuffer.append("-");
}
-
//将十六进制byte转化为字符串
String temp = Integer.toHexString(mac[i] & 0xff);
if (temp.length() == 1) {
@@ -114,14 +112,33 @@ public abstract class AbstractServerInfos {
stringBuffer.append(temp);
}
}
-
return stringBuffer.toString().toUpperCase();
} catch (SocketException e) {
e.printStackTrace();
log.error("获取某个网络接口的Mac地址异常", e.getMessage());
}
-
return null;
}
+ /**
+ * 获取服务器信息
+ *
+ * @param osName 系统类型
+ * @return AGxServerInfos 服务信息
+ */
+ public static AbstractServerInfos getServer(String osName) {
+ if ("".equals(osName) || osName == null) {
+ osName = System.getProperty("os.name").toLowerCase();
+ }
+ AbstractServerInfos abstractServerInfos;
+ // 根据不同操作系统类型选择不同的数据获取方法
+ if (osName.startsWith("windows")) {
+ abstractServerInfos = new WindowsServerInfos();
+ } else if (osName.startsWith("linux")) {
+ abstractServerInfos = new LinuxServerInfos();
+ } else {// 其他服务器类型
+ abstractServerInfos = new LinuxServerInfos();
+ }
+ return abstractServerInfos;
+ }
}
diff --git a/wfc-common/wfc-common-license/src/main/java/org/wfc/common/license/serverinfo/LinuxServerInfos.java b/wfc-common/wfc-common-license/src/main/java/org/wfc/common/license/serverinfo/LinuxServerInfos.java
index fc2666f..ab62711 100644
--- a/wfc-common/wfc-common-license/src/main/java/org/wfc/common/license/serverinfo/LinuxServerInfos.java
+++ b/wfc-common/wfc-common-license/src/main/java/org/wfc/common/license/serverinfo/LinuxServerInfos.java
@@ -1,7 +1,8 @@
package org.wfc.common.license.serverinfo;
+import org.apache.commons.lang3.StringUtils;
+
import java.io.BufferedReader;
-import java.io.IOException;
import java.io.InputStreamReader;
import java.net.InetAddress;
import java.util.List;
@@ -12,9 +13,6 @@ import java.util.stream.Collectors;
*/
public class LinuxServerInfos extends AbstractServerInfos {
- private final String[] CPU_SHELL = {"/bin/bash", "-c", "dmidecode -t processor | grep 'ID' | awk -F ':' '{print $2}' | head -n 1"};
- private final String[] MAIN_BOARD_SHELL = {"/bin/bash", "-c", "dmidecode | grep 'Serial Number' | awk -F ':' '{print $2}' | head -n 1"};
-
@Override
protected List getIpAddress() throws Exception {
List result = null;
@@ -46,50 +44,43 @@ public class LinuxServerInfos extends AbstractServerInfos {
@Override
protected String getCPUSerial() throws Exception {
- String result = "";
- String CPU_ID_CMD = "dmidecode";
- BufferedReader bufferedReader = null;
- Process p = null;
- try {
- p = Runtime.getRuntime().exec(new String[]{"sh", "-c", CPU_ID_CMD});// 管道
- bufferedReader = new BufferedReader(new InputStreamReader(p.getInputStream()));
- String line = null;
- int index = -1;
- while ((line = bufferedReader.readLine()) != null) {
- // 寻找标示字符串[hwaddr]
- index = line.toLowerCase().indexOf("uuid");
- if (index >= 0) {// 找到了
- // 取出mac地址并去除2边空格
- result = line.substring(index + "uuid".length() + 1).trim();
- break;
- }
- }
- } catch (IOException e) {
- System.out.println("获取cpu硬件信息失败 " + e);
+ //序列号
+ String serialNumber = "";
+
+ //使用dmidecode命令获取CPU序列号
+ String[] shell = {"/bin/bash", "-c", "dmidecode -t processor | grep 'ID' | awk -F ':' '{print $2}' | head -n 1"};
+ Process process = Runtime.getRuntime().exec(shell);
+ process.getOutputStream().close();
+
+ BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream()));
+
+ String line = reader.readLine().trim();
+ if (StringUtils.isNotBlank(line)) {
+ serialNumber = line;
}
- return result.trim();
-// return GxServerSerialHelper.getLinuxSerial(CPU_SHELL);
+
+ reader.close();
+ return serialNumber;
}
@Override
protected String getMainBoardSerial() throws Exception {
- String result = "";
- String maniBord_cmd = "dmidecode | grep 'Serial Number' | awk '{print $3}' | tail -1";
- Process p;
- try {
- p = Runtime.getRuntime().exec(new String[]{"sh", "-c", maniBord_cmd});// 管道
- BufferedReader br = new BufferedReader(new InputStreamReader(p.getInputStream()));
- String line;
- while ((line = br.readLine()) != null) {
- result += line;
- break;
- }
- br.close();
- } catch (IOException e) {
- System.out.println("获取主板信息错误" + e);
- }
- return result;
-// return GxServerSerialHelper.getLinuxSerial(MAIN_BOARD_SHELL);
- }
+ //序列号
+ String serialNumber = "";
+ //使用dmidecode命令获取主板序列号
+ String[] shell = {"/bin/bash", "-c", "dmidecode | grep 'Serial Number' | awk -F ':' '{print $2}' | head -n 1"};
+ Process process = Runtime.getRuntime().exec(shell);
+ process.getOutputStream().close();
+
+ BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream()));
+
+ String line = reader.readLine().trim();
+ if (StringUtils.isNotBlank(line)) {
+ serialNumber = line;
+ }
+
+ reader.close();
+ return serialNumber;
+ }
}