package com.license;


import de.schlichtherle.license.*;

import javax.security.auth.x500.X500Principal;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.text.DateFormat;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.Properties;
import java.util.prefs.Preferences;

/**
 * 生成证书
 */
public class CreateLicense {
    public static void main(String[] args) {
        CreateLicense cLicense = new CreateLicense();
        cLicense.setParam("/param.properties");
        //生成证书
        boolean b = cLicense.create();
    }

    private static String PRIVATE_ALIAS = "";
    private static String KEY_PWD = "";
    private static String STORE_PWD = "";
    private static String SUB_JECT = "";
    private static String LIC_PATH = "";
    private static String PRI_PATH = "";
    private static String ISSUED_TIME = "";
    private static String NOT_BEFORE = "";
    private static String NOT_AFTER = "";
    private static String CONSUMER_TYPE = "";
    private static int CONSUMER_AMOUNT = 0;
    private static String INFO = "";
    // X500 Principal
    private final static X500Principal DEFAULT_HOLDER_AND_ISSUER = new X500Principal("CN=maxzhao, OU=maxzhao, O=maxzhao_org, L=nanjing, ST=jiangsu, C=China");

    public void setParam(String propertiesPath) { // 获取参数
        Properties prop = new Properties();
        InputStream in = getClass().getResourceAsStream(propertiesPath);
        try {
            prop.load(in);
        } catch (IOException e) { // TODO Auto-generated catch block
            e.printStackTrace();
        }
        PRIVATE_ALIAS = prop.getProperty("privateAlias");
        KEY_PWD = prop.getProperty("keyPwd");
        STORE_PWD = prop.getProperty("storePwd");
        SUB_JECT = prop.getProperty("subJect");
        KEY_PWD = prop.getProperty("keyPwd");
        LIC_PATH = prop.getProperty("licPath");
        PRI_PATH = prop.getProperty("priPath"); //license content
        ISSUED_TIME = prop.getProperty("issuedTime");
        NOT_BEFORE = prop.getProperty("notBefore");
        NOT_AFTER = prop.getProperty("notAfter");
        CONSUMER_TYPE = prop.getProperty("consumerType");
        CONSUMER_AMOUNT = Integer.valueOf(prop.getProperty("consumerAmount"));
        INFO = prop.getProperty("info");
    }

    /**
     * 证书发布者端执行
     */
    public boolean create() {
        try {
            LicenseManager licenseManager = new LicenseManager(initLicenseParams());
            licenseManager.store((createLicenseContent()), new File(LIC_PATH));
        } catch (Exception e) {
            e.printStackTrace();
            System.out.println("客户端证书生成失败!");
            return false;
        }
        System.out.println("服务器端生成证书成功!");
        return true;
    }

    /**
     * 返回生成证书时需要的参数
     */
    private static LicenseParam initLicenseParams() {
        Preferences preference = Preferences.userNodeForPackage(LicenseCreator.class);
        // 设置对证书内容加密的对称密码
        CipherParam cipherParam = new DefaultCipherParam(STORE_PWD);
        KeyStoreParam privateStoreParam = new DefaultKeyStoreParam(LicenseCreator.class, PRI_PATH, PRIVATE_ALIAS, STORE_PWD, KEY_PWD);
        return new DefaultLicenseParam(SUB_JECT, preference, privateStoreParam, cipherParam);
    }

    /**
     * 从外部表单拿到证书的内容
     */
    public final static LicenseContent createLicenseContent() {
        DateFormat format = new SimpleDateFormat("yyyy-MM-dd");
        LicenseContent content = null;
        content = new LicenseContent();
        content.setSubject(SUB_JECT);
        content.setHolder(DEFAULT_HOLDER_AND_ISSUER);
        content.setIssuer(DEFAULT_HOLDER_AND_ISSUER);
        try {
            content.setIssued(format.parse(ISSUED_TIME));
            content.setNotBefore(format.parse(NOT_BEFORE));
            content.setNotAfter(format.parse(NOT_AFTER));
        } catch (ParseException e) { // TODO Auto-generated catch block
            e.printStackTrace();
        }
        content.setConsumerType(CONSUMER_TYPE);
        content.setConsumerAmount(CONSUMER_AMOUNT);
        content.setInfo(INFO); // 扩展
        content.setExtra(new Object());
        return content;
    }
}

package com.license;

import de.schlichtherle.license.*;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.util.Properties;
import java.util.prefs.Preferences;

/**
 * 校验证书
 */
public class VerifyLicense {

    public static void main(String[] args) {
        VerifyLicense license = new VerifyLicense();
        license.setParam("/public-param.properties");
        license.verify();
    }

    private static String PUBLIC_ALIAS = "";
    private static String STORE_PWD = "";
    private static String SUB_JECT = "";
    private static String LIC_PATH = "";
    private static String PUB_PATH = "";

    public void setParam(String propertiesPath) {
        // 获取参数
        Properties prop = new Properties();
        InputStream in = getClass().getResourceAsStream(propertiesPath);
        try {
            prop.load(in);
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        PUBLIC_ALIAS = prop.getProperty("publicAlias");
        STORE_PWD = prop.getProperty("storePwd");
        SUB_JECT = prop.getProperty("subJect");
        LIC_PATH = prop.getProperty("licPath");
        PUB_PATH = prop.getProperty("pubPath");
    }

    /**
     * 证书使用者端执行
     */
    public boolean verify() {
        LicenseManager licenseManager = new LicenseManager(initLicenseParams());
        // 安装证书
        try {
            licenseManager.install(new File(LIC_PATH));
            System.out.println("客户端安装证书成功!");
        } catch (Exception e) {
            e.printStackTrace();
            System.out.println("客户端证书安装失败!");
            return false;
        }
        // 验证证书
        try {
            licenseManager.verify();
            System.out.println("客户端验证证书成功!");
        } catch (Exception e) {
            e.printStackTrace();
            System.out.println("客户端证书验证失效!");
            return false;
        }
        return true;
    }

    /**
     * 返回验证证书需要的参数
     */
    private static LicenseParam initLicenseParams() {
        Preferences preference = Preferences.userNodeForPackage(VerifyLicense.class);
        CipherParam cipherParam = new DefaultCipherParam(STORE_PWD);
        KeyStoreParam privateStoreParam = new DefaultKeyStoreParam(VerifyLicense.class, PUB_PATH, PUBLIC_ALIAS, STORE_PWD, null);
        return new DefaultLicenseParam(SUB_JECT, preference, privateStoreParam, cipherParam);
    }
}

/**
 * 默认日志输出路径
 *
 * @author maxzhao
 */
@Component
public class LoggerListener implements ApplicationListener<ApplicationEvent>, Ordered {
    @Override
    public void onApplicationEvent(ApplicationEvent event) {
        if (event instanceof ApplicationEnvironmentPreparedEvent) {
            ConfigurableEnvironment environment = ((ApplicationEnvironmentPreparedEvent) event).getEnvironment();
            String property = environment.getProperty("spring.application.name");
            System.setProperty("empower.log.path", property);
        }
    }

    @Override
    public int getOrder() {
        return LoggingApplicationListener.DEFAULT_ORDER - 1;
    }
}

public class DataHubApplication {


    public static void main(String[] args) {
        SpringApplication application = new SpringApplication(DataHubApplication.class);
        application.addListeners(new LoggerListener());
        ConfigurableApplicationContext context = application.run(args);
        SpringBootUtils.setApplicationContent(context);
    }
}