package com.walker.security.admin;
|
|
import java.io.File;
|
import java.util.Map;
|
|
public abstract class AbstractLisenceGenerator implements LisenceGenerator {
|
|
// private static final String RECORD_FILE_PREFIX = "d:/logs/walker_sn_";
|
// private static final String RECORD_FILE_SUFFIX = ".txt";
|
|
private Map<String, Object> keyMap = null;
|
|
private String secretKey = null;
|
private String publicKey = null;
|
|
//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
// update code
|
//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
private GenerateCoder generateCoder = new GenerateCoder();
|
|
public AbstractLisenceGenerator(){
|
try {
|
keyMap = RSAUtils.genKeyPair();
|
} catch (Exception e) {
|
e.printStackTrace();
|
throw new Error();
|
}
|
}
|
|
@Override
|
public String getSecretKey() {
|
if(secretKey != null) return secretKey;
|
String s = null;
|
try{
|
s = keyMap == null ? null : RSAUtils.getPrivateKey(keyMap);
|
} catch(Exception e){
|
throw new RuntimeException(null,e);
|
}
|
if(s == null || s.equals(""))
|
throw new RuntimeException("not found my spider!");
|
secretKey = s;
|
return secretKey;
|
}
|
|
@Override
|
public String getPublicKey() {
|
if(publicKey == null){
|
try{
|
publicKey = RSAUtils.getPublicKey(keyMap);
|
} catch(Exception e){
|
throw new RuntimeException(null,e);
|
}
|
if(publicKey == null || publicKey.equals("")){
|
throw new RuntimeException("not found my spider!");
|
}
|
}
|
return publicKey;
|
}
|
|
@Override
|
public byte[] generate(String message) throws Exception {
|
// Assert.isTrue(StringUtils.isNotEmpty(message));
|
if(message == null || message.equals(""))
|
throw new IllegalArgumentException();
|
String[] texts = RSAUtils.toArray(message);
|
if(texts.length != 5)
|
throw new IllegalArgumentException();
|
return generateCoder.getFileContent(Integer.parseInt(texts[1]), texts[2].toString()
|
, Long.parseLong(texts[3]), Long.parseLong(texts[4]));
|
}
|
|
@Override
|
public Object generate(String message, File destFile) throws Exception {
|
if(message == null || message.equals("")){
|
throw new IllegalArgumentException();
|
}
|
// Assert.notNull(destFile);
|
|
// String publicKey = getPublicKey();
|
// assert (publicKey != null && !publicKey.equals(""));
|
//
|
// // 加密内容
|
// byte[] encryptData = RSAUtils.encryptByPrivateKey(message.toString().getBytes(), getSecretKey());
|
// ByteBuffer result = ByteBuffer.allocate(encryptData.length + 2 + publicKey.length());
|
// result.put(encryptData);
|
// result.put("\n".getBytes());
|
// result.put(publicKey.getBytes());
|
// result.flip();
|
|
//
|
if(message == null || message.equals(""))
|
throw new IllegalArgumentException();
|
String[] texts = RSAUtils.toArray(message);
|
if(texts.length != 5)
|
throw new IllegalArgumentException();
|
generateCoder.writeLisenceFile(texts[0].toString()
|
, Integer.parseInt(texts[1]), texts[2].toString()
|
, Long.parseLong(texts[3]), Long.parseLong(texts[4]), destFile);
|
return null;
|
}
|
|
// /**
|
// * 记录生成的协议文件
|
// * @param message
|
// */
|
// void recordLisence(String message){
|
// File destFile = new File(RECORD_FILE_PREFIX + message + RECORD_FILE_SUFFIX);
|
// StringBuilder content = new StringBuilder();
|
// content.append(message);
|
// content.append("\r\n");
|
// content.append("secretKey: ");
|
// content.append(getSecretKey());
|
// content.append("\r\n");
|
// content.append("publicKey: ");
|
// content.append(getPublicKey());
|
// content.append("\r\n");
|
//
|
// try {
|
// FileUtils.writeTxtFile(content.toString(), destFile);
|
// System.out.println("----------------> 生成文件: " + destFile);
|
// } catch (Exception e) {
|
// throw new NestedRuntimeException("write lisence record file failed!", e);
|
// }
|
// }
|
|
@Override
|
public void print() {
|
// TODO Auto-generated method stub
|
try{
|
StringBuilder sb = new StringBuilder();
|
sb.append("[lisenceGenerator: secretKey = ");
|
sb.append(getSecretKey());
|
sb.append(", publicKey = ");
|
sb.append(getPublicKey());
|
sb.append("]");
|
System.out.println(sb);
|
} catch(Exception e){
|
e.printStackTrace();
|
}
|
}
|
|
/**
|
* 产生真正的协议对象，文件、对象、数据库等格式，子类预留实现。
|
* @param encryptData 加密过的协议数据
|
* @return
|
*/
|
protected abstract Object generateObject(byte[] encryptData);
|
|
private class GenerateCoder {
|
// private String pk = "MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQCL4fwI6s3zdK0RIDQ0o9xQ7Lq4CoETtOKbDRSCwdCeQ5PrWys4tlenCGOds5SAkOrXFt7wOt0Z33PlAMXpLxAq1gDp/vlOLPPxTfm7S6mUhGNrYOv/0FXQ3rWa7G/UZaXlybn7WLcr2yFaJbwSxxKWa/WuNc1wyCii1S854ZvGBQIDAQAB";
|
private String sk = "MIICdgIBADANBgkqhkiG9w0BAQEFAASCAmAwggJcAgEAAoGBAIvh/AjqzfN0rREgNDSj3FDsurgKgRO04psNFILB0J5Dk+tbKzi2V6cIY52zlICQ6tcW3vA63Rnfc+UAxekvECrWAOn++U4s8/FN+btLqZSEY2tg6//QVdDetZrsb9RlpeXJuftYtyvbIVolvBLHEpZr9a41zXDIKKLVLznhm8YFAgMBAAECgYBBauEMZosEdSdO4AmqKIurSNqVoUv2JFzpeDeEYxGmQiJWKh/PeO8LDZBQuyAS9DLvfiGqpUyrgeXYTgFAaSMcw+Xw8+MKvQXlPy+6joP4wZgXdU0LqLCY+tQGlGl/k8saWwRgT3VqqqB3NnuNE5eqvXlNWxWiL3hbKklus1jrQQJBAMrQ5GNQwV2DqIgwtPLOi3OBne/dqf7WEh2DlJZBi86+qJfmkJ9u6EiTTrwK+gkavBtG6JxAquhNQeE9LTApnjUCQQCwkFzLV/mrcRm/Hru9QhzwxHWRj2zy8v0+3F1zdMWEdpxfN2DnBgsO5CDSM7PB6arQAozdjqIh9IgUY/ioMcKRAkAqp2oQ9Q/lseXE1rGHiybK4QytQKoa3TysFlsuipRzo6djYusxjKqvcMOTSp+xQH0lI3GSQwtulajwTORvB6GtAkAQ3Xk/oyOiEKeY7eKY7Vu+U8W3JbAOdM9j1cDHanwDbyD4p12GM7mSro/EBFRDTYKXY+b4sssh2y20ZOg3iKlBAkEAiAmNOldhGlqPfvRkm2fCYCMYcTrsi6Qduz8n82ChuJughXUbBWvVZ0YwvpPUAAX2uhteMP6ZdN3+BK7xSrdyAQ==";
|
|
// /**
|
// * 写入ip类型授权信息
|
// * @param customId
|
// * @param address
|
// */
|
// public void writeLisenceFile(String customId, String address){
|
// doWriteLisenceFile(customId, 0, address, 0, 0);
|
// }
|
//
|
// public void writeLisenceFile(String customId, long start, long end){
|
// doWriteLisenceFile(customId, 0, "", start, end);
|
// }
|
public byte[] getFileContent(int type, String address, long start, long end){
|
try {
|
return RSAUtils.encryptByPrivateKey(getFileData(type, address, start, end).getBytes(), sk);
|
} catch (Exception e) {
|
throw new Error(e);
|
}
|
}
|
|
public void writeLisenceFile(String customId, int type
|
, String address, long start, long end, File destFile){
|
doWriteLisenceFile(customId, type, address, start, end, destFile);
|
}
|
|
private void doWriteLisenceFile(String customId, int type
|
, String address, long start, long end, File destFile){
|
// File destFile = new File(RECORD_FILE_PREFIX + customId + ".bin");
|
String content = getFileData(type, address, start, end);
|
try {
|
byte[] encryptData = RSAUtils.encryptByPrivateKey(content.getBytes(), sk);
|
FileCopyUtils.copy(encryptData, destFile);
|
} catch (Exception e) {
|
throw new Error(e);
|
}
|
}
|
|
private String getFileData(int type, String address, long start, long end){
|
StringBuilder content = new StringBuilder();
|
// content.append("license_type=");
|
content.append(type);
|
content.append("\r\n");
|
// content.append("mac_address=");
|
content.append(address);
|
content.append("\r\n");
|
// content.append("start=");
|
content.append(start);
|
content.append("\r\n");
|
// content.append("end=");
|
content.append(end);
|
return content.toString();
|
}
|
}
|
}
|
