Sunyard/sydapi-java-sge-database
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

短数据加密生成mac，短数据校验mac并解密：短链接的性能提升

Browse Source
This commit is contained in:
jif2.zhang 2025-07-21 17:41:59 +08:00
parent 3ecb230cf7
commit 654e7617cb
6 changed files with 1056 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
config.properties
View File

@ -1,6 +1,12 @@
# 应用配置
# 调试模式 (true/false)
debug=true
debug=false
# 最大线程数
maxThread=20
# 总数
total=100000
# 地址列表 (格式: addX=ip:port)
addr1=192.168.110.240:8889
addr1=172.1.41.129:8889

209
src/main/java/com/sunyard/sge/database/SydApiBaseFunction.java
View File

@ -1,17 +1,22 @@
package com.sunyard.sge.database;
import com.sun.jna.ptr.IntByReference;
import com.sun.jna.ptr.PointerByReference;
import com.sunyard.SydApiException;
import com.sunyard.log.ILogFactory;
import com.sunyard.log.ILogger;
import com.sunyard.proto.Util;
import com.sunyard.sge.bytes.BytesUtil;
import com.sunyard.sge.database.pool.HsmLinkInfo;
import com.sunyard.sge.log.LogbackFactory;
import com.sunyard.sge.plus.SydLibrary;
import com.sunyard.sge.pool.ObjectInPool;
import com.sunyard.util.SYMUtil;
import racal.sunyard.main.SydApi4j;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
/**
@ -22,6 +27,10 @@ public class SydApiBaseFunction implements SydApi {
private HsmLinkInfo[] linkInfos = null;
private ObjectInPool oip = null;
private boolean isShortLinkMode = false;
// c 加速句柄，短链接。
private PointerByReference phHandle = null;
private ILogFactory iLogFactory = new LogbackFactory();
private ILogger log = iLogFactory.getLogger(SydApiBaseFunction.class);
@ -85,10 +94,33 @@ public class SydApiBaseFunction implements SydApi {
hsms = apis;
isShortLinkMode = true;
// c 加速链接
if ( Env.isLinux() ) {
PointerByReference phHandle = new PointerByReference();
int connectResult = SydLibrary.INSTANCE.SYD_Connect_Ex(
pcIpList,
iPortList,
pcIpList.length,
iConnectTimeOut,
iDealTimeOut,
0,
phHandle);
if ( 0 != connectResult ){
throw new SydApiException("连接错误", -1);
}
this.phHandle = phHandle;
}
return this;
}
public void SYD_Disconnect() {
if ( null != phHandle && Env.isLinux()) {
SydLibrary.INSTANCE.SYD_DisConnect_Ex(phHandle.getValue());
phHandle = null;
}
if (null == hsms) {
return;
}
@ -494,9 +526,6 @@ public class SydApiBaseFunction implements SydApi {
}
} catch (SydApiException e) {
if (null == e) {
}
le = e;
if (e.getRetCode() < 0) {
log.error("HA 重试");
@ -869,9 +898,48 @@ public class SydApiBaseFunction implements SydApi {
return Arrays.equals(pcMac, mac);
}
DataAndMac SYD_SM4_EncryptAndMac_ShortData_c(int iEncKeyIndex, int iMacKeyIndex, byte[] pcData) {
log.info("SYD_SM4_EncryptAndMac_ShortData_c {} {}", iEncKeyIndex, iMacKeyIndex);
try {
int enLen = BytesUtil.calculatePaddedLength(pcData.length);
byte[] encData = new byte[enLen];
byte[] mac = new byte[16];
IntByReference encDataLen = new IntByReference(encData.length);
int i = SydLibrary.INSTANCE.SYD_SM4_Encrypt_Mac_ShortData(
phHandle.getValue(),
iEncKeyIndex,
iMacKeyIndex,
pcData,
pcData.length,
encData,
encDataLen,
mac
);
if (0 == i) {
byte[] data = new byte[encDataLen.getValue()];
if (data.length == encData.length) {
data = encData;
} else {
System.arraycopy(encData, 0, data, 0, data.length);
}
return new DataAndMac(data, mac);
} else {
throw new SydApiException(i);
}
} catch (SydApiException e) {
throw e;
} catch (Exception e) {
if (e.getCause() instanceof SydApiException) {
throw (SydApiException) e.getCause();
}
throw new SydApiException("并行任务执行失败", 0xC101, e);
}
}
/**
* 非依赖 C 实现，仅长连接依赖 C
*
* @param iEncKeyIndex：输入。加密所需要的密钥索引值
* @param iMacKeyIndex：输入。计算mac所需要的密钥索引值
@ -884,11 +952,58 @@ public class SydApiBaseFunction implements SydApi {
if (log.isDebugEnabled()) {
log.debug("SYD_SM4_EncryptAndMac_ShortData：k1={} k2={}", iEncKeyIndex, iMacKeyIndex);
}
// 参数检查在函数内部
byte[] enData = SYD_SM4_ShortData(iEncKeyIndex, pcData, Consts.ECB_ENC);
// mac 长度放宽到 1040，因为密文填充
byte[] mac = SYD_SM4Mac_ShortData(iMacKeyIndex, enData, 1040);
return new DataAndMac(enData, mac);
if ( null != phHandle ) {
return SYD_SM4_EncryptAndMac_ShortData_c(iEncKeyIndex, iMacKeyIndex, pcData);
} else {
// 参数检查在函数内部
byte[] enData = SYD_SM4_ShortData(iEncKeyIndex, pcData, Consts.ECB_ENC);
// mac 长度放宽到 1040，因为密文填充
byte[] mac = SYD_SM4Mac_ShortData(iMacKeyIndex, enData, 1040);
return new DataAndMac(enData, mac);
}
}
public DataAndMacCheck SYD_SM4_CheckMacAndDecrypt_ShortData_c(int iEncKeyIndex, int iMacKeyIndex, byte[] pcData, byte[] macs) {
log.info("SYD_SM4_CheckMacAndDecrypt_ShortData_c {} {}", iEncKeyIndex, iMacKeyIndex);
try {
byte[] decryptedData = new byte[pcData.length];
IntByReference decryptedDataLen = new IntByReference(decryptedData.length);
int i = SydLibrary.INSTANCE.SYD_SM4_Decrypt_Mac_ShortData(
phHandle.getValue(),
iEncKeyIndex,
iMacKeyIndex,
pcData,
pcData.length,
macs,
decryptedData,
decryptedDataLen
);
byte[] data = new byte[decryptedDataLen.getValue()];
System.arraycopy(decryptedData, 0, data, 0, data.length);
if (0 == i) {
return new DataAndMacCheck(data, true);
}
if (0x01000010 == i) {
return new DataAndMacCheck(data, false);
} else {
throw new SydApiException(i);
}
} catch (SydApiException e) {
throw e;
} catch (Exception e) {
if (e.getCause() instanceof SydApiException) {
throw (SydApiException) e.getCause();
}
throw new SydApiException("并行任务执行失败", 0xC101, e);
}
}
@ -899,14 +1014,21 @@ public class SydApiBaseFunction implements SydApi {
log.debug("SYD_SM4_CheckMacAndDecrypt_ShortData：k1={} k2={}", iEncKeyIndex, iMacKeyIndex);
}
boolean ret = SYD_SM4Mac_ShortData(iMacKeyIndex, pcData, 1040, mac);
byte[] data = null;
try {
data = SYD_SM4_ShortData(iEncKeyIndex, pcData, Consts.ECB_DEC);
} catch (Exception e) {
return new DataAndMacCheck(null, ret);
if( null != phHandle ) {
return SYD_SM4_CheckMacAndDecrypt_ShortData_c(iEncKeyIndex, iMacKeyIndex, pcData, mac);
} else {
boolean ret = SYD_SM4Mac_ShortData(iMacKeyIndex, pcData, 1040, mac);
byte[] data = null;
if (!ret){
return new DataAndMacCheck(data, ret);
}
try {
data = SYD_SM4_ShortData(iEncKeyIndex, pcData, Consts.ECB_DEC);
} catch (Exception e) {
return new DataAndMacCheck(null, ret);
}
return new DataAndMacCheck(data, ret);
}
return new DataAndMacCheck(data, ret);
}
@ -968,13 +1090,56 @@ public class SydApiBaseFunction implements SydApi {
datas.add(pcDatum.getData());
}
boolean[] macCheck = SYD_SM4Mac_BatchData(iMacKeyIndex, datas, macs);
List<byte[]> enData = SYD_SM4_BatchData(iEncKeyIndex, datas, Consts.ECB_DEC);
// boolean[] macCheck = SYD_SM4Mac_BatchData(iMacKeyIndex, datas, macs);
// List<byte[]> enData = SYD_SM4_BatchData(iEncKeyIndex, datas, Consts.ECB_DEC);
//
// List<DataAndMacCheck> ret = new ArrayList<>(enData.size());
// for (int i = 0; i < enData.size(); i++) {
// DataAndMacCheck damc = new DataAndMacCheck(enData.get(i), macCheck[i]);
// ret.add(damc);
// }
List<DataAndMacCheck> ret = new ArrayList<>(enData.size());
for (int i = 0; i < enData.size(); i++) {
DataAndMacCheck damc = new DataAndMacCheck(enData.get(i), macCheck[i]);
ret.add(damc);
// 批量执行MAC校验
boolean[] macCheck = SYD_SM4Mac_BatchData(iMacKeyIndex, datas, macs);
// 分离通过和未通过MAC校验的数据
List<Integer> validIndices = new ArrayList<>();
List<byte[]> validData = new ArrayList<>();
List<Integer> validKeyIndices = new ArrayList<>();
for (int i = 0; i < macCheck.length; i++) {
if (macCheck[i]) {
validIndices.add(i);
validData.add(datas.get(i));
validKeyIndices.add(iEncKeyIndex[i]);
}
}
// 仅对通过MAC校验的数据进行批量解密
List<byte[]> decryptedData;
if (validData.isEmpty()) {
decryptedData = Collections.emptyList();
} else {
// 转换为数组（根据SYD_SM4_BatchData的接口要求）
int[] validKeysArray = validKeyIndices.stream()
.mapToInt(Integer::intValue)
.toArray();
// 使用过滤后的密钥索引和数据
decryptedData = SYD_SM4_BatchData(validKeysArray, validData, Consts.ECB_DEC);
}
// 构建结果集
List<DataAndMacCheck> ret = new ArrayList<>(pcData.size());
int decryptedIndex = 0;
for (int i = 0; i < pcData.size(); i++) {
if (macCheck[i]) {
// 对于通过校验的条目，使用解密数据
ret.add(new DataAndMacCheck(decryptedData.get(decryptedIndex++), true));
} else {
// 对于未通过校验的条目，返回null
ret.add(new DataAndMacCheck(null, false));
}
}
return ret;

BIN
src/main/resources/linux-x86-64/libsunyard.so
View File

Binary file not shown.

16
src/test/java/test/FuncTest.java
View File

@ -355,6 +355,10 @@ public class FuncTest {
DataAndMacCheck de_data1 = api.SYD_SM4_CheckMacAndDecrypt_ShortData(keyEnc, keyMac, en_data1.getData(), en_data1.getMac());
System.out.println("step3 " + de_data1.getMacCheck());
//错误测试
System.out.println("MAC校验失败，不继续进行解密测试");
DataAndMacCheck de_data_err = api.SYD_SM4_CheckMacAndDecrypt_ShortData(keyEnc, keyMac, en_data1.getData(), new byte[16]);
System.out.println("错误测试返回MAC校验结果："+de_data_err.getMacCheck());
try {
DataAndMac en_data2 = api.SYD_SM4_EncryptAndMac_ShortData(keyEnc, keyMac, data2);
@ -436,6 +440,18 @@ public class FuncTest {
System.out.println("short link");
}
System.out.println("批量数据mac校验并解密，mac校验失败测试");
en_data.get(1).setMac(new byte[16]);
List<DataAndMacCheck> de_data1 = api.SYD_SM4_CheckMacAndDecrypt_BatchData(keyEnc, keyMac, en_data);
int index = 0;
for (DataAndMacCheck item : de_data1) {
boolean macCheck = item.getMacCheck();
System.out.println("索引 " + (index++) + ":"+"MAC校验结果为："+macCheck);
}
System.out.println("-------- 批量数据加密计算MAC，校验MAC解密功能测试结束 --------\n");
}

23
src/test/java/test/TestAll.java
View File

@ -456,17 +456,26 @@ public class TestAll {
int keyEnc = 0;
int keyMac = 0;
SydApiBuilder builder = new SydApiBuilder(
config.getIps(),
config.getPorts(),
1000,
2000
);
// SydApiBuilder builder = new SydApiBuilder(
// config.getIps(),
// config.getPorts(),
// 1000,
// 2000
// );
//
// ThreadLocal<SydApi> Api = new ThreadLocal<SydApi>() {
// @Override
// protected SydApi initialValue() {
// return builder.build();
// }
// };
ThreadLocal<SydApi> Api = new ThreadLocal<SydApi>() {
@Override
protected SydApi initialValue() {
return builder.build();
SydApi api = new SydApi4Database();
api.SYD_Short_Connect_Ex(config.getIps(),config.getPorts(),1000,2000);
return api;
}
};

829
src/test/java/test/TestMAC.java Normal file
View File

@ -0,0 +1,829 @@
package test;
import com.sunyard.sge.database.*;
import java.io.BufferedWriter;
import java.io.FileWriter;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Date;
import java.util.List;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
public class TestMAC {
private static final String[] ipListElec = new String[]{"127.0.0.1", "172.1.41.129"};
private static final String[] ipListLight = new String[]{"192.168.110.240", "192.168.100.45"};
private static final String[] ipList = ipListLight;
private static final String[] ipListErr = new String[]{"172.2.41.216", "172.2.41.129"};
public static final int ECB_DEC = 0;
public static final int ECB_ENC = 1;
public static void setLog() {
System.setProperty("com.sunyard.sydapi4j.debug", "false");
}
private static void startPerformanceFile() {
try (FileWriter fw = new FileWriter(PERF_RESULT_FILE, true);
BufferedWriter bw = new BufferedWriter(fw)) {
Date now = new Date();
SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
String timeStr = sdf.format(now);
bw.write("==================== Performance Test Start: " + timeStr + " ====================");
bw.newLine();
} catch (IOException e) {
e.printStackTrace();
}
}
private static void stopPerformanceFile() {
try (FileWriter fw = new FileWriter(PERF_RESULT_FILE, true);
BufferedWriter bw = new BufferedWriter(fw)) {
Date now = new Date();
SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
String timeStr = sdf.format(now);
bw.write("==================== Performance Test End: " + timeStr + " ====================");
bw.newLine();
} catch (IOException e) {
e.printStackTrace();
}
}
private void writeToPerformanceFile(String interfaceName, int dataLen, int machineCount, int threadNum, int totalRequests, double tps, double avgLatency) {
try (FileWriter fw = new FileWriter(PERF_RESULT_FILE, true);
BufferedWriter bw = new BufferedWriter(fw)) {
String line = String.format("%s\t%d\t%d\t%d\t%d\t%.2f\t%.2f",
interfaceName, dataLen, machineCount, threadNum, totalRequests, tps, avgLatency);
bw.write(line);
bw.newLine();
} catch (IOException e) {
e.printStackTrace();
}
}
private void writeToPerformanceFile(String interfaceName, int dataLen, int machineCount, int threadNum, int totalRequests, String tps, double avgLatency) {
try (FileWriter fw = new FileWriter(PERF_RESULT_FILE, true);
BufferedWriter bw = new BufferedWriter(fw)) {
String line = String.format("%s\t%d\t%d\t%d\t%d\t%s\t%.2f",
interfaceName, dataLen, machineCount, threadNum, totalRequests, tps, avgLatency);
bw.write(line);
bw.newLine();
} catch (IOException e) {
e.printStackTrace();
}
}
private void writeToPerformanceFile2(String interfaceName, String dataLen, int machineCount, int threadNum, double totalRequests) {
try (FileWriter fw = new FileWriter(PERF_RESULT_FILE, true);
BufferedWriter bw = new BufferedWriter(fw)) {
String line = String.format("%s\t%s\t%d\t%d\t%.2f",
interfaceName, dataLen, machineCount, threadNum, totalRequests);
bw.write(line);
bw.newLine();
} catch (IOException e) {
e.printStackTrace();
}
}
public static void menu1(Config config, Choise choise, TestMAC test) throws Exception {
FuncTest funcTest = new FuncTest(config);
while (true) {
System.out.println("请选择测试项：");
// System.out.println("1. 功能单项测试");
// System.out.println("2. 功能全量测试");
System.out.println("1. 性能单项测试");
System.out.println("2. 性能全量测试");
System.out.println("-1. 退出");
switch (choise.getAsInt()) {
// case 1:
// SydApi.SYD_SetLogConfig("./logs", 3);
// funcTest.funcMenu(config, choise, funcTest);
// break;
// case 2:
// funcTest.funcAll();
// break;
case 1:
SydApi.SYD_SetLogConfig("./logs", 0);
perfMenu(config, choise, test);
break;
case 2:
SydApi.SYD_SetLogConfig("./logs", 0);
startPerformanceFile();
test.perfAll();
stopPerformanceFile();
break;
default:
System.exit(0);
}
}
}
public void sm4ShortEndes() throws InterruptedException {
System.out.println("-------- SM4短数据加解密性能测试开始 --------");
sm4ShortEnde(1, config.getTotal(), 1024);
sm4ShortEnde(40, config.getTotal(), 1024);
System.out.println("-------- SM4短数据加解密性能测试结束 --------\n");
}
private static final String PERF_RESULT_FILE = "performance_results.txt";
public void sm4ShortEnde(int numberOfThreads, int totalRequest, int plainLen) throws InterruptedException {
byte[] data = new byte[plainLen];
SydApiBuilder builder = new SydApiBuilder(
config.getIps(),
config.getPorts(),
1000,
2000
);
ThreadLocal<SydApi> Api = new ThreadLocal<SydApi>() {
@Override
protected SydApi initialValue() {
return builder.build();
}
};
ExecutorService poolEnc = Executors.newFixedThreadPool(numberOfThreads);
long start = System.currentTimeMillis();
for (int i = 0; i < totalRequest; i++) {
poolEnc.submit(() -> {
SydApi api = Api.get();
api.SYD_SM4_ShortData(0, data, ECB_ENC);
});
}
poolEnc.shutdown();
poolEnc.awaitTermination(Long.MAX_VALUE, TimeUnit.SECONDS);
long end = System.currentTimeMillis();
long duration = end - start;
double tps = totalRequest * 1000.0 / duration;
double throughput = totalRequest * 1000.0 / duration * plainLen / 1024 / 1024;
System.out.println("并发数: " + numberOfThreads);
System.out.println("加密共花费 " + String.format("%.2f", duration / 1000.0) + " 秒");
System.out.println("TPS: " + String.format("%.2f", tps));
System.out.println("吞吐量: " + String.format("%.2f", throughput) + " MB/S");
double avgLatencyEnc = numberOfThreads * duration * 1.0 / totalRequest;
System.out.println("平均延时（加密）: " + String.format("%.2f", avgLatencyEnc) + " ms");
writeToPerformanceFile("SM4短数据加密", 1024, 1, numberOfThreads, totalRequest, tps, avgLatencyEnc);
SydApi apiTest = Api.get();
byte[] res = apiTest.SYD_SM4_ShortData(0, data, ECB_ENC);
ExecutorService poolDec = Executors.newFixedThreadPool(numberOfThreads);
start = System.currentTimeMillis();
for (int i = 0; i < totalRequest; i++) {
poolDec.submit(() -> {
SydApi api = Api.get();
byte[] plain = api.SYD_SM4_ShortData(0, res, ECB_DEC);
});
}
poolDec.shutdown();
poolDec.awaitTermination(Long.MAX_VALUE, TimeUnit.SECONDS);
end = System.currentTimeMillis();
duration = end - start;
tps = totalRequest * 1000.0 / duration;
throughput = totalRequest * 1000.0 / duration * plainLen / 1024 / 1024;
System.out.println("并发数: " + numberOfThreads);
System.out.println("解密共花费 " + String.format("%.2f", duration / 1000.0) + " 秒");
System.out.println("TPS: " + String.format("%.2f", tps));
System.out.println("吞吐量: " + String.format("%.2f", throughput) + " MB/S");
double avgLatencyDec = numberOfThreads * duration * 1.0 / totalRequest;
System.out.println("平均延时（解密）: " + String.format("%.2f", avgLatencyDec) + " ms");
writeToPerformanceFile("SM4短数据解密", 1024, 1, numberOfThreads, totalRequest, tps, avgLatencyDec);
}
public void sm4BatchEndes() throws InterruptedException {
System.out.println("-------- SM4批量数据加解密性能测试开始 --------");
sm4BatchEnde(1, config.getTotal(), 1024);
sm4BatchEnde(40, config.getTotal(), 1024);
System.out.println("-------- SM4批量数据加解密性能测试结束 --------\n");
}
public void sm4BatchEnde(int numberOfThreads, int totalRequest, int plainLen) throws InterruptedException {
int[] keyIdx = new int[]{0, 0, 0, 0};
List<byte[]> data = new ArrayList<>();
data.add(new byte[1024]);
data.add(new byte[1024]);
data.add(new byte[1024]);
data.add(new byte[1024]);
SydApiBuilder builder = new SydApiBuilder(
config.getIps(),
config.getPorts(),
1000,
2000
);
ThreadLocal<SydApi> Api = new ThreadLocal<SydApi>() {
@Override
protected SydApi initialValue() {
return builder.build();
}
};
ExecutorService poolEnc = Executors.newFixedThreadPool(numberOfThreads);
long start = System.currentTimeMillis();
for (int i = 0; i < totalRequest; i++) {
poolEnc.submit(() -> {
SydApi api = Api.get();
api.SYD_SM4_BatchData(keyIdx, data, ECB_ENC);
});
}
poolEnc.shutdown();
poolEnc.awaitTermination(Long.MAX_VALUE, TimeUnit.SECONDS);
long end = System.currentTimeMillis();
long duration = end - start;
double tps = totalRequest * 1000.0 / duration;
double throughput = totalRequest * 1000.0 / duration * plainLen / 1024 / 1024;
System.out.println("并发数: " + numberOfThreads);
System.out.println("加密共花费 " + String.format("%.2f", duration / 1000.0) + " 秒");
System.out.println("TPS: " + String.format("%.2f", tps));
System.out.println("吞吐量: " + String.format("%.2f", throughput) + " MB/S");
double avgLatency = numberOfThreads * duration * 1.0 / totalRequest;
writeToPerformanceFile("SM4批量数据加密", 1024, 1, numberOfThreads, totalRequest, String.format("%.2f", tps) + " x 4", avgLatency);
SydApi apiTest = Api.get();
List<byte[]> res = apiTest.SYD_SM4_BatchData(keyIdx, data, ECB_ENC);
ExecutorService poolDec = Executors.newFixedThreadPool(numberOfThreads);
start = System.currentTimeMillis();
for (int i = 0; i < totalRequest; i++) {
poolDec.submit(() -> {
SydApi api = Api.get();
List<byte[]> plain = api.SYD_SM4_BatchData(keyIdx, res, ECB_DEC);
});
}
poolDec.shutdown();
poolDec.awaitTermination(Long.MAX_VALUE, TimeUnit.SECONDS);
end = System.currentTimeMillis();
duration = end - start;
tps = totalRequest * 1000.0 / duration;
throughput = totalRequest * 1000.0 / duration * plainLen / 1024 / 1024;
System.out.println("并发数: " + numberOfThreads);
System.out.println("解密共花费 " + String.format("%.2f", duration / 1000.0) + " 秒");
System.out.println("TPS: " + String.format("%.2f", tps));
System.out.println("吞吐量: " + String.format("%.2f", throughput) + " MB/S");
avgLatency = numberOfThreads * duration * 1.0 / totalRequest;
writeToPerformanceFile("SM4批量数据解密", 1024, 1, numberOfThreads, totalRequest, tps, avgLatency);
}
public void sm4macShortEndes() throws InterruptedException {
System.out.println("-------- SM4MAC短数据性能测试开始 --------");
sm4macShortEnde(1, config.getTotal(), 1024);
sm4macShortEnde(40, config.getTotal(), 1024);
System.out.println("-------- SM4MAC短数据性能测试结束 --------\n");
}
public void sm4macShortEnde(int numberOfThreads, int totalRequest, int plainLen) throws InterruptedException {
byte[] data = new byte[plainLen];
SydApiBuilder builder = new SydApiBuilder(
config.getIps(),
config.getPorts(),
1000,
2000
);
ThreadLocal<SydApi> Api = new ThreadLocal<SydApi>() {
@Override
protected SydApi initialValue() {
return builder.build();
}
};
ExecutorService poolEnc = Executors.newFixedThreadPool(numberOfThreads);
long start = System.currentTimeMillis();
for (int i = 0; i < totalRequest; i++) {
poolEnc.submit(() -> {
SydApi api = Api.get();
api.SYD_SM4Mac_ShortData(0, data);
});
}
poolEnc.shutdown();
poolEnc.awaitTermination(Long.MAX_VALUE, TimeUnit.SECONDS);
long end = System.currentTimeMillis();
long duration = end - start;
double tps = totalRequest * 1000.0 / duration;
double throughput = totalRequest * 1000.0 / duration * plainLen / 1024 / 1024;
System.out.println("并发数: " + numberOfThreads);
System.out.println("共花费 " + String.format("%.2f", duration / 1000.0) + " 秒");
System.out.println("TPS: " + String.format("%.2f", tps));
System.out.println("吞吐量: " + String.format("%.2f", throughput) + " MB/S");
double avgLatency = numberOfThreads * duration * 1.0 / totalRequest;
writeToPerformanceFile("SM4MAC 短数据", 1024, 1, numberOfThreads, totalRequest, tps, avgLatency);
}
public void sm4macBatchEndes() throws InterruptedException {
System.out.println("-------- SM4MAC批量数据性能测试开始 --------");
sm4macBatchEnde(1, config.getTotal(), 1024);
sm4macBatchEnde(40, config.getTotal(), 1024);
System.out.println("-------- SM4MAC 批量数据性能测试结束 --------\n");
}
public void sm4macBatchEnde(int numberOfThreads, int totalRequest, int plainLen) throws InterruptedException {
int[] keyIdx = new int[]{0, 0, 0, 0};
SydApiBuilder builder = new SydApiBuilder(
config.getIps(),
config.getPorts(),
1000,
2000
);
ThreadLocal<SydApi> Api = new ThreadLocal<SydApi>() {
@Override
protected SydApi initialValue() {
return builder.build();
}
};
List<byte[]> data = new ArrayList<>();
data.add(new byte[1024]);
data.add(new byte[1024]);
data.add(new byte[1024]);
data.add(new byte[1024]);
ExecutorService poolEnc = Executors.newFixedThreadPool(numberOfThreads);
long start = System.currentTimeMillis();
for (int i = 0; i < totalRequest; i++) {
poolEnc.submit(() -> {
SydApi api = Api.get();
List<byte[]> s = api.SYD_SM4Mac_BatchData(keyIdx, data);
});
}
poolEnc.shutdown();
poolEnc.awaitTermination(Long.MAX_VALUE, TimeUnit.SECONDS);
long end = System.currentTimeMillis();
long duration = end - start;
double tps = totalRequest * 1000.0 / duration;
double throughput = totalRequest * 1000.0 / duration * plainLen / 1024 / 1024;
System.out.println("并发数: " + numberOfThreads);
System.out.println("共花费 " + String.format("%.2f", duration / 1000.0) + " 秒");
System.out.println("TPS: " + String.format("%.2f", tps));
System.out.println("吞吐量: " + String.format("%.2f", throughput) + " MB/S");
double avgLatency = numberOfThreads * duration * 1.0 / totalRequest;
writeToPerformanceFile("SM4MAC 批量数据", 1024, 1, numberOfThreads, totalRequest, String.format("%.2f", tps) + " x 4", avgLatency);
}
public void sm3ShortEndes() throws InterruptedException {
System.out.println("-------- SM3短数据性能测试开始 --------");
sm3ShortEnde(1, config.getTotal(), 1024);
sm3ShortEnde(40, config.getTotal(), 1024);
System.out.println("-------- SM3短数据性能测试结束 --------\n");
}
public void sm3ShortEnde(int numberOfThreads, int totalRequest, int plainLen) throws InterruptedException {
byte[] data = new byte[plainLen];
SydApiBuilder builder = new SydApiBuilder(
config.getIps(),
config.getPorts(),
1000,
2000
);
ThreadLocal<SydApi> Api = new ThreadLocal<SydApi>() {
@Override
protected SydApi initialValue() {
return builder.build();
}
};
ExecutorService poolEnc = Executors.newFixedThreadPool(numberOfThreads);
long start = System.currentTimeMillis();
for (int i = 0; i < totalRequest; i++) {
poolEnc.submit(() -> {
SydApi api = Api.get();
api.SYD_SM3_Hash_ShortData(data);
});
}
poolEnc.shutdown();
poolEnc.awaitTermination(Long.MAX_VALUE, TimeUnit.SECONDS);
long end = System.currentTimeMillis();
long duration = end - start;
double tps = totalRequest * 1000.0 / duration;
double throughput = totalRequest * 1000.0 / duration * plainLen / 1024 / 1024;
System.out.println("并发数: " + numberOfThreads);
System.out.println("共花费 " + String.format("%.2f", duration / 1000.0) + " 秒");
System.out.println("TPS: " + String.format("%.2f", tps));
System.out.println("吞吐量: " + String.format("%.2f", throughput) + " MB/S");
double avgLatency = numberOfThreads * duration * 1.0 / totalRequest;
writeToPerformanceFile("SM3 短数据", 1024, 1, numberOfThreads, totalRequest, tps, avgLatency);
}
public void sm4ShortMacEndes() throws InterruptedException {
System.out.println("-------- SM4短数据加密生成MAC,验证MAC解密性能测试开始 --------");
System.out.println("单线程");
sm4ShortMacEnde(1, config.getTotal(), 1024);
System.out.println("40 线程");
sm4ShortMacEnde(40, config.getTotal(), 1024);
System.out.println("-------- SM4短数据加密生成MAC,验证MAC解密性能测试结束 --------\n");
}
public void sm4ShortMacEnde(int numberOfThreads, int totalRequest, int plainLen) throws InterruptedException {
byte[] data = new byte[plainLen];
int keyEnc = 0;
int keyMac = 0;
SydApiBuilder builder = new SydApiBuilder(
config.getIps(),
config.getPorts(),
1000,
2000
);
ThreadLocal<SydApi> Api = new ThreadLocal<SydApi>() {
@Override
protected SydApi initialValue() {
return builder.build();
}
};
SydApi apiTest = Api.get();
DataAndMac ret = apiTest.SYD_SM4_EncryptAndMac_ShortData(keyEnc, keyMac, data);
ExecutorService poolEnc = Executors.newFixedThreadPool(numberOfThreads);
long start = System.currentTimeMillis();
for (int i = 0; i < totalRequest; i++) {
poolEnc.submit(() -> {
SydApi api = Api.get();
api.SYD_SM4_EncryptAndMac_ShortData(keyEnc, keyMac, data);
});
}
poolEnc.shutdown();
poolEnc.awaitTermination(Long.MAX_VALUE, TimeUnit.SECONDS);
long end = System.currentTimeMillis();
long duration = end - start;
double tps = totalRequest * 1000.0 / duration;
double throughput = totalRequest * 1000.0 / duration * plainLen / 1024 / 1024;
System.out.println("并发数: " + numberOfThreads);
System.out.println("加密共花费 " + String.format("%.2f", duration / 1000.0) + " 秒");
System.out.println("TPS: " + String.format("%.2f", tps));
System.out.println("吞吐量: " + String.format("%.2f", throughput) + " MB/S");
double avgLatency = numberOfThreads * duration * 1.0 / totalRequest;
writeToPerformanceFile("SM4短数据加密并生成MAC", 1024, 1, numberOfThreads, totalRequest, tps, avgLatency);
ExecutorService poolDec = Executors.newFixedThreadPool(numberOfThreads);
start = System.currentTimeMillis();
for (int i = 0; i < totalRequest; i++) {
poolDec.submit(() -> {
SydApi api = Api.get();
api.SYD_SM4_CheckMacAndDecrypt_ShortData(keyEnc, keyMac, ret.getData(), ret.getMac());
});
}
poolDec.shutdown();
poolDec.awaitTermination(Long.MAX_VALUE, TimeUnit.SECONDS);
end = System.currentTimeMillis();
duration = end - start;
tps = totalRequest * 1000.0 / duration;
throughput = totalRequest * 1000.0 / duration * plainLen / 1024 / 1024;
System.out.println("并发数: " + numberOfThreads);
System.out.println("解密共花费 " + String.format("%.2f", duration / 1000.0) + " 秒");
System.out.println("TPS: " + String.format("%.2f", tps));
System.out.println("吞吐量: " + String.format("%.2f", throughput) + " MB/S");
avgLatency = numberOfThreads * duration * 1.0 / totalRequest;
writeToPerformanceFile("SM4短数据解密并校验MAC", 1024, 1, numberOfThreads, totalRequest, tps, avgLatency);
}
public void sm4BatchMacEndes() throws InterruptedException {
System.out.println("-------- SM4批量数据加密生成MAC,验证MAC解密性能测试开始 --------");
System.out.println("单线程");
sm4BatchMacEnde(1, config.getTotal(), 1024);
System.out.println("40 线程");
sm4BatchMacEnde(40, config.getTotal(), 1024);
System.out.println("-------- SM4批量数据加密生成MAC,验证MAC解密性能测试结束 --------\n");
}
public void sm4BatchMacEnde(int numberOfThreads, int totalRequest, int plainLen) throws InterruptedException {
int[] keyEnc = new int[]{0, 0, 0, 0};
int[] keyMac = new int[]{0, 0, 0, 0};
List<byte[]> data = new ArrayList<>();
data.add(new byte[plainLen]);
data.add(new byte[plainLen]);
data.add(new byte[plainLen]);
data.add(new byte[plainLen]);
SydApiBuilder builder = new SydApiBuilder(
config.getIps(),
config.getPorts(),
1000,
2000
);
ThreadLocal<SydApi> Api = new ThreadLocal<SydApi>() {
@Override
protected SydApi initialValue() {
return builder.build();
}
};
SydApi apiTest = Api.get();
List<DataAndMac> ret = apiTest.SYD_SM4_EncryptAndMac_BatchData(keyEnc, keyMac, data);
ExecutorService poolEnc = Executors.newFixedThreadPool(numberOfThreads);
long start = System.currentTimeMillis();
for (int i = 0; i < totalRequest; i++) {
poolEnc.submit(() -> {
SydApi api = Api.get();
api.SYD_SM4_EncryptAndMac_BatchData(keyEnc, keyMac, data);
});
}
poolEnc.shutdown();
poolEnc.awaitTermination(Long.MAX_VALUE, TimeUnit.SECONDS);
long end = System.currentTimeMillis();
long duration = end - start;
double tps = totalRequest * 1000.0 / duration;
double throughput = totalRequest * 1000.0 / duration * plainLen / 1024 / 1024;
System.out.println("并发数: " + numberOfThreads);
System.out.println("加密共花费 " + String.format("%.2f", duration / 1000.0) + " 秒");
System.out.println("TPS: " + String.format("%.2f", tps));
System.out.println("吞吐量: " + String.format("%.2f", throughput) + " MB/S");
double avgLatency = numberOfThreads * duration * 1.0 / totalRequest;
writeToPerformanceFile("SM4批量数据加密并生成MAC", 1024, 1, numberOfThreads, totalRequest, String.format("%.2f", tps) + " x 4", avgLatency);
ExecutorService poolDec = Executors.newFixedThreadPool(numberOfThreads);
start = System.currentTimeMillis();
for (int i = 0; i < totalRequest; i++) {
poolDec.submit(() -> {
SydApi api = Api.get();
api.SYD_SM4_CheckMacAndDecrypt_BatchData(keyEnc, keyMac, ret);
});
}
poolDec.shutdown();
poolDec.awaitTermination(Long.MAX_VALUE, TimeUnit.SECONDS);
end = System.currentTimeMillis();
duration = end - start;
tps = totalRequest * 1000.0 / duration;
throughput = totalRequest * 1000.0 / duration * plainLen / 1024 / 1024;
System.out.println("并发数: " + numberOfThreads);
System.out.println("解密共花费 " + String.format("%.2f", duration / 1000.0) + " 秒");
System.out.println("TPS: " + String.format("%.2f", tps));
System.out.println("吞吐量: " + String.format("%.2f", throughput) + " MB/S");
avgLatency = numberOfThreads * duration * 1.0 / totalRequest;
writeToPerformanceFile("SM4批量数据解密并校验MAC", 1024, 1, numberOfThreads, totalRequest, String.format("%.2f", tps) + " x 4", avgLatency);
}
public void sm4LongEnde() {
System.out.println("-------- SM4长数据加解密性能测试开始 --------");
byte[] data = new byte[2048];
int key = 0;
// ByteBuffer enDataBuff = ByteBuffer.allocate(100 * 1024 * 1024 + 16);
SydApiBuilder builder = new SydApiBuilder(
config.getIps(),
config.getPorts(),
1000,
2000
);
long start = System.currentTimeMillis();
SydApi api = builder.build();
ISM4 sm4 = api.initSM4(key, ECB_ENC);
for (int i = 0; i < 100 * 1024 * 1024 / data.length; i++) {
sm4.update(data);
}
byte[] res = sm4.finish();
long end = System.currentTimeMillis();
long duration = end - start;
double throughput = 100.0 * 1000 / duration;
System.out.println("加密共花费 " + String.format("%.2f", duration / 1000.0) + " 秒");
System.out.println("吞吐量: " + String.format("%.2f", throughput) + " MB/S");
writeToPerformanceFile2("SM4长数据加密", "100M", 1, 1, throughput);
start = System.currentTimeMillis();
sm4 = api.initSM4(key, ECB_DEC);
for (int i = 0; i < 100 * 1024 * 1024 / data.length; i++) {
sm4.update(data);
}
// sm4.finish();
end = System.currentTimeMillis();
duration = end - start;
throughput = 100.0 * 1000 / duration;
System.out.println("解密共花费 " + String.format("%.2f", duration / 1000.0) + " 秒");
System.out.println("吞吐量: " + String.format("%.2f", throughput) + " MB/S");
writeToPerformanceFile2("SM4长数据解密", "100M", 1, 1, throughput);
System.out.println("-------- SM4长数据加解密性能测试结束 --------\\n");
}
public void sm4MacLongEnde() {
System.out.println("-------- SM4MAC长数据性能测试开始 --------");
byte[] data = new byte[2048];
SydApiBuilder builder = new SydApiBuilder(
config.getIps(),
config.getPorts(),
1000,
2000
);
long start = System.currentTimeMillis();
SydApi api = builder.build();
ISM4Mac sm4mac = api.initSM4Mac(0);
for (int i = 0; i < 100 * 1024 * 1024 / data.length; i++) {
sm4mac.update(data);
}
sm4mac.digest();
long end = System.currentTimeMillis();
long duration = end - start;
double throughput = 100.0 * 1000 / duration;
System.out.println("共花费 " + String.format("%.2f", duration / 1000.0) + " 秒");
System.out.println("吞吐量: " + String.format("%.2f", throughput) + " MB/S");
writeToPerformanceFile2("SM4长数据MAC", "100M", 1, 1, throughput);
System.out.println("-------- SM4MAC长数据性能测试结束 --------");
}
public void sm3LongEnde() {
System.out.println("-------- SM3长数据性能测试开始 --------");
byte[] data = new byte[2048];
SydApiBuilder builder = new SydApiBuilder(
config.getIps(),
config.getPorts(),
1000,
2000
);
long start = System.currentTimeMillis();
SydApi api = builder.build();
ISM3 sm3 = api.initSM3();
for (int i = 0; i < 100 * 1024 * 1024 / data.length; i++) {
sm3.update(data);
}
sm3.digest();
long end = System.currentTimeMillis();
long duration = end - start;
double throughput = 100.0 * 1000 / duration;
System.out.println("共花费 " + String.format("%.2f", duration / 1000.0) + " 秒");
System.out.println("吞吐量: " + String.format("%.2f", throughput) + " MB/S");
writeToPerformanceFile2("SM3长数据", "100MB", 1, 1, throughput);
System.out.println("-------- SM3长数据性能测试结束 --------");
}
public void perfAll() throws InterruptedException {
this.sm4ShortEndes();
this.sm4BatchEndes();
// this.sm4macShortEndes();
// this.sm4macBatchEndes();
// this.sm3ShortEndes();
this.sm4ShortMacEndes();
this.sm4BatchMacEndes();
// this.sm4LongEnde();
// this.sm4MacLongEnde();
// this.sm3LongEnde();
}
public static void showmenu() {
System.out.println("请选择测试项：");
System.out.println("1.SM4 短数据加解密");
System.out.println("2.SM4 批量数据加解密");
// System.out.println("3.SM4MAC 短数据");
// System.out.println("4.SM4MAC 批量数据");
// System.out.println("5.SM3 短数据");
System.out.println("6.SM4 短数据加密后生成MAC，解密验证MAC");
System.out.println("7.SM4 批量数据加密后生成MAC，解密验证MAC");
// System.out.println("8.SM4 长数据加解密");
// System.out.println("9.SM4 长数据MAC");
// System.out.println("10.SM3 长数据");
// System.out.println("11. 短数据 SM4+MAC性能测试");
// System.out.println("12. 批量数据 SM4+MAC性能测试");
System.out.println("-1.退出");
}
public static void perfMenu(Config config, Choise choise, TestMAC test) throws Exception {
boolean flag = true;
while (flag) {
showmenu();
switch (choise.getAsInt()) {
case 1:
test.sm4ShortEndes();
break;
case 2:
test.sm4BatchEndes();
break;
// case 3:
// test.sm4macShortEndes();
// break;
// case 4:
// test.sm4macBatchEndes();
// break;
// case 5:
// test.sm3ShortEndes();
// break;
case 6:
test.sm4ShortMacEndes();
break;
case 7:
test.sm4BatchMacEndes();
break;
// case 8:
// test.sm4LongEnde();
// break;
// case 9:
// test.sm4MacLongEnde();
// break;
// case 10:
// test.sm3LongEnde();
// break;
// case 11: {
// test.sm4ShortMacEndes();
// break;
// }
// case 12: {
// test.sm4BatchMacEndes();
// break;
// }
case -1:
flag = false;
break;
}
}
}
private static Config config = null;
public static void main(String[] args) throws Exception {
// 载入配置文件
config = Config.loadConfig();
if (config == null || config.isDebug()) {
System.setProperty("com.sunyard.sydapi4j.debug", "true");
}
System.out.println("ips=" + Arrays.toString(config.getIps()));
System.out.println("ports=" + Arrays.toString(config.getPorts()));
System.out.println("debug=" + config.isDebug());
TestMAC test = new TestMAC();
Choise choise = new Choise(args);
menu1(config, choise, test);
}
}