JRebel-License-Server授权服务器 Java-vertx 实现

2020-01-06

项目依赖:pom.xml

<dependency>
    <groupId>junit</groupId>
    <artifactId>junit</artifactId>
    <version>4.12</version>
    <scope>test</scope>
</dependency>
<dependency>
    <groupId>org.apache.commons</groupId>
    <artifactId>commons-lang3</artifactId>
    <version>3.9</version>
</dependency>
<dependency>
    <groupId>com.google.code.gson</groupId>
    <artifactId>gson</artifactId>
    <version>2.8.6</version>
</dependency>
<dependency>
    <groupId>commons-codec</groupId>
    <artifactId>commons-codec</artifactId>
    <version>1.13</version>
</dependency>
<dependency>
    <groupId>org.slf4j</groupId>
    <artifactId>slf4j-api</artifactId>
    <version>1.7.29</version>
</dependency>
<dependency>
    <groupId>org.apache.logging.log4j</groupId>
    <artifactId>log4j-core</artifactId>
    <version>2.12.1</version>
</dependency>
<dependency>
    <groupId>org.apache.logging.log4j</groupId>
    <artifactId>log4j-api</artifactId>
    <version>2.12.1</version>
</dependency>
<dependency>
    <groupId>org.apache.logging.log4j</groupId>
    <artifactId>log4j-slf4j-impl</artifactId>
    <version>2.12.1</version>
</dependency>
<dependency>
    <groupId>org.slf4j</groupId>
    <artifactId>slf4j-log4j12</artifactId>
    <version>1.7.29</version>
</dependency>
<dependency>
    <groupId>io.vertx</groupId>
    <artifactId>vertx-web</artifactId>
    <version>3.8.4</version>
</dependency>
<dependency>
    <groupId>org.projectlombok</groupId>
    <artifactId>lombok</artifactId>
    <version>1.18.10</version>
</dependency>

主类：Server.java

import com.google.gson.JsonObject;
import com.google.gson.JsonParser;
import com.movefeng.utils.JRebelSign;
import io.vertx.core.AbstractVerticle;
import io.vertx.core.Vertx;
import io.vertx.core.http.HttpMethod;
import io.vertx.core.http.HttpServer;
import io.vertx.core.http.HttpServerRequest;
import io.vertx.ext.web.Router;
import io.vertx.ext.web.RoutingContext;
import io.vertx.ext.web.handler.BodyHandler;
import lombok.extern.slf4j.Slf4j;

import java.util.UUID;

@Slf4j
public class Server extends AbstractVerticle {

    private static String[] commandArgs = new String[]{};

    public static void main(String[] args) {
        commandArgs = args;
        System.out.println("默认端口9000, 如需指定其他端口, 增加启动参数: port=9001\n" +
                "jrebel激活时填写的url格式为 http://<ip>:<port>/<随机的uuid> ,比如: http://192.168.5.1:9000/63481187-7cbf-434d-88d2-efa665bed7ec \n" +
                "命令行执行: curl http://<ip>:<port>/uuid 或者浏览器访问 http://<ip>:<port>/uuid 生成一个随机的uuid");
        Vertx.vertx().deployVerticle(new Server());
    }

    @Override
    public void start() {
        HttpServer httpServer = vertx.createHttpServer();
        Router router = Router.router(vertx);
        router.route().handler(BodyHandler.create());
        router.route("/uuid").handler(Server::uuid);
        router.post("/jrebel/leases").handler(Server::leases);
        router.route("/jrebel/leases/1").method(HttpMethod.POST).method(HttpMethod.DELETE).handler(Server::leases1);
        router.get("/jrebel/validate-connection").handler(Server::validateConnection);
        httpServer.requestHandler(router);
        int port = 9000;
        if (commandArgs.length > 0) {
            for (String commandArg : commandArgs) {
                String[] split = commandArg.split("=");
                if (split.length > 0) {
                    if ("port".equals(split[0])) {
                        String portStr = split[1];
                        try {
                            port = Integer.parseInt(portStr);
                        } catch (NumberFormatException e) {
                            e.printStackTrace();
                            log.error("端口号格式不正确！已使用默认端口号 9000");
                        }
                    }
                }
            }
        }
        httpServer.listen(port);
    }

    private static void uuid(RoutingContext context) {
        context.response().end(UUID.randomUUID().toString() + "\n");
    }

    private static void leases(RoutingContext context) {
        HttpServerRequest request = context.request();
        log.info("{} {}", request.method(), request.uri());
        String randomness = request.getParam("randomness");
        String username = request.getParam("username");
        String clientTime = request.getParam("clientTime");
        String guid = request.getParam("guid");
        boolean offline = Boolean.parseBoolean(request.getParam("offline"));
        String validFrom = "null";
        String validUntil = "null";
        if (offline) {
            long clientTimeUntil = Long.parseLong(clientTime) + 180L * 24 * 60 * 60 * 1000;
            validFrom = clientTime;
            validUntil = String.valueOf(clientTimeUntil);
        }
        String jsonStr = "{\n" +
                "    \"serverVersion\": \"3.2.4\",\n" +
                "    \"serverProtocolVersion\": \"1.1\",\n" +
                "    \"serverGuid\": \"a1b4aea8-b031-4302-b602-670a990272cb\",\n" +
                "    \"groupType\": \"managed\",\n" +
                "    \"id\": 1,\n" +
                "    \"licenseType\": 1,\n" +
                "    \"evaluationLicense\": false,\n" +
                "    \"signature\": \"OJE9wGg2xncSb+VgnYT+9HGCFaLOk28tneMFhCbpVMKoC/Iq4LuaDKPirBjG4o394/UjCDGgTBpIrzcXNPdVxVr8PnQzpy7ZSToGO8wv/KIWZT9/ba7bDbA8/RZ4B37YkCeXhjaixpmoyz/CIZMnei4q7oWR7DYUOlOcEWDQhiY=\",\n" +
                "    \"serverRandomness\": \"H2ulzLlh7E0=\",\n" +
                "    \"seatPoolType\": \"standalone\",\n" +
                "    \"statusCode\": \"SUCCESS\",\n" +
                "    \"offline\": " + offline + ",\n" +
                "    \"validFrom\": " + validFrom + ",\n" +
                "    \"validUntil\": " + validUntil + ",\n" +
                "    \"company\": \"Administrator\",\n" +
                "    \"orderId\": \"\",\n" +
                "    \"zeroIds\": [\n" +
                "        \n" +
                "    ],\n" +
                "    \"licenseValidFrom\": 1490544001000,\n" +
                "    \"licenseValidUntil\": 1691839999000\n" +
                "}";
        JsonObject jsonObject = JsonParser.parseString(jsonStr).getAsJsonObject();
        if (randomness == null || username == null || guid == null) {
            context.response().setStatusCode(403);
        } else {
            JRebelSign jRebelSign = new JRebelSign();
            jRebelSign.sign(randomness, guid, offline, validFrom, validUntil);
            String signature = jRebelSign.getSignature();
            jsonObject.addProperty("signature", signature);
            jsonObject.addProperty("company", username);
            String body = jsonObject.toString();
            context.response().putHeader("content-type", "application/json;charset=utf-8");
            context.response().end(body);
        }
    }

    private static void validateConnection(RoutingContext context) {
        HttpServerRequest request = context.request();
        log.info("{} {}", request.method(), request.uri());
        String jsonStr = "{\n" +
                "    \"serverVersion\": \"3.2.4\",\n" +
                "    \"serverProtocolVersion\": \"1.1\",\n" +
                "    \"serverGuid\": \"a1b4aea8-b031-4302-b602-670a990272cb\",\n" +
                "    \"groupType\": \"managed\",\n" +
                "    \"statusCode\": \"SUCCESS\",\n" +
                "    \"company\": \"Administrator\",\n" +
                "    \"canGetLease\": true,\n" +
                "    \"licenseType\": 1,\n" +
                "    \"evaluationLicense\": false,\n" +
                "    \"seatPoolType\": \"standalone\"\n" +
                "}\n";
        JsonObject jsonObject = JsonParser.parseString(jsonStr).getAsJsonObject();
        String body = jsonObject.toString();
        context.response().putHeader("content-type", "application/json;charset=utf-8");
        context.response().end(body);
    }

    private static void leases1(RoutingContext context) {
        HttpServerRequest request = context.request();
        log.info("{} {}", request.method(), request.uri());
        String username = context.request().getParam("username");
        String jsonStr = "{\n" +
                "    \"serverVersion\": \"3.2.4\",\n" +
                "    \"serverProtocolVersion\": \"1.1\",\n" +
                "    \"serverGuid\": \"a1b4aea8-b031-4302-b602-670a990272cb\",\n" +
                "    \"groupType\": \"managed\",\n" +
                "    \"statusCode\": \"SUCCESS\",\n" +
                "    \"msg\": null,\n" +
                "    \"statusMessage\": null\n" +
                "}\n";
        JsonObject jsonObject = JsonParser.parseString(jsonStr).getAsJsonObject();
        if (username != null) {
            jsonObject.addProperty("company", username);
        }
        String body = jsonObject.toString();
        context.response().putHeader("content-type", "application/json;charset=utf-8");
        context.response().end(body);
    }

}

辅助类: JRebelSign.java

import org.apache.commons.codec.binary.Base64;
import org.apache.commons.lang3.StringUtils;

import java.nio.charset.StandardCharsets;
import java.security.KeyFactory;
import java.security.PrivateKey;
import java.security.Signature;
import java.security.spec.PKCS8EncodedKeySpec;


public class JRebelSign {
    private String signature;
    private static final String PKCS_8_PRIVATE_KEY =
            "MIICdgIBADANBgkqhkiG9w0BAQEFAASCAmAwggJcAgEAAoGBAND3cI/pKMSd4OLM\n" +
                    "IXU/8xoEZ/nza+g00Vy7ygyGB1Nn83qpro7tckOvUVILJoN0pKw8J3E8rtjhSyr9\n" +
                    "849qzaQKBhxFL+J5uu08QVn/tMt+Tf0cu5MSPOjT8I2+NWyBZ6H0FjOcVrEUMvHt\n" +
                    "8sqoJDrDU4pJyex2rCOlpfBeqK6XAgMBAAECgYBM5C+8FIxWxM1CRuCs1yop0aM8\n" +
                    "2vBC0mSTXdo7/3lknGSAJz2/A+o+s50Vtlqmll4drkjJJw4jacsR974OcLtXzQrZ\n" +
                    "0G1ohCM55lC3kehNEbgQdBpagOHbsFa4miKnlYys537Wp+Q61mhGM1weXzosgCH/\n" +
                    "7e/FjJ5uS6DhQc0Y+QJBAP43hlSSEo1BbuanFfp55yK2Y503ti3Rgf1SbE+JbUvI\n" +
                    "IRsvB24xHa1/IZ+ttkAuIbOUomLN7fyyEYLWphIy9kUCQQDSbqmxZaJNRa1o4ozG\n" +
                    "RORxR2KBqVn3EVISXqNcUH3gAP52U9LcnmA3NMSZs8tzXhUhYkWQ75Q6umXvvDm4\n" +
                    "XZ0rAkBoymyWGeyJy8oyS/fUW0G63mIroZZ4Rp+F098P3j9ueJ2k/frbImXwabJr\n" +
                    "hwjUZe/Afel+PxL2ElUDkQW+BMHdAkEAk/U7W4Aanjpfs1+Xm9DUztFicciheRa0\n" +
                    "njXspvvxhY8tXAWUPYseG7L+iRPh+Twtn0t5nm7VynVFN0shSoCIAQJALjo7A6bz\n" +
                    "svfnJpV+lQiOqD/WCw3A2yPwe+1d0X/13fQkgzcbB3K0K81Euo/fkKKiBv0A7yR7\n" +
                    "wvrNjzefE9sKUw==";

    public void sign(String clientRandomness, String guid, boolean offline, String validFrom, String validUntil) {
        try {
            String serverRandomness = "H2ulzLlh7E0=";
            String s2;
            if (offline) {
                s2 = StringUtils.join(new String[]{clientRandomness, serverRandomness, guid, String.valueOf(true), validFrom, validUntil}, ';');
            } else {
                s2 = StringUtils.join(new String[]{clientRandomness, serverRandomness, guid, String.valueOf(false)}, ';');
            }
            PKCS8EncodedKeySpec pkcs8EncodedKeySpec = new PKCS8EncodedKeySpec(Base64.decodeBase64(PKCS_8_PRIVATE_KEY.getBytes(StandardCharsets.UTF_8)));
            PrivateKey privateKey = KeyFactory.getInstance("RSA").generatePrivate(pkcs8EncodedKeySpec);
            this.signature = getSha1Sign(s2, privateKey);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public static String getSha1Sign(String content, PrivateKey privateKey) throws Exception {
        byte[] contentBytes = content.getBytes(StandardCharsets.UTF_8);
        Signature signature = Signature.getInstance("SHA1withRSA");
        signature.initSign(privateKey);
        signature.update(contentBytes);
        byte[] signs = signature.sign();
        return Base64.encodeBase64String(signs);
    }

    public String getSignature() {
        return signature;
    }

}