From 642da35ec47710659b0c839527aadc583b66ea14 Mon Sep 17 00:00:00 2001
From: "novice.li" <novice.li@outlook.com>
Date: Fri, 5 Apr 2024 10:40:35 +0800
Subject: [PATCH] hook LicensingFacade

---
 .../java/com/novitechie/jetbra/AgentMain.java |  14 +-
 .../jetbra/LicensingFacadeAdvice.java         |  22 +++
 .../src/main/resources/public/index.html      | 175 ++++++++++++++----
 3 files changed, 176 insertions(+), 35 deletions(-)
 create mode 100644 jetbra-agent/src/main/java/com/novitechie/jetbra/LicensingFacadeAdvice.java

diff --git a/jetbra-agent/src/main/java/com/novitechie/jetbra/AgentMain.java b/jetbra-agent/src/main/java/com/novitechie/jetbra/AgentMain.java
index 0157d5d..3f78f02 100644
--- a/jetbra-agent/src/main/java/com/novitechie/jetbra/AgentMain.java
+++ b/jetbra-agent/src/main/java/com/novitechie/jetbra/AgentMain.java
@@ -30,6 +30,12 @@ public class AgentMain {
                                 .on(ElementMatchers.named("getUserOptionsFile"))))
                 .asTerminalTransformation()
 
+                .type(ElementMatchers.named("com.intellij.ui.LicensingFacade"))
+                .transform((builder, typeDescription, classLoader, javaModule, protectionDomain) ->
+                        builder.visit(Advice.to(LicensingFacadeAdvice.class)
+                                .on(ElementMatchers.named("getLicenseExpirationDate"))))
+                .asTerminalTransformation()
+
                 .installOn(inst);
 
         agentBuilder.installOn(inst);
@@ -37,10 +43,10 @@ public class AgentMain {
 
     static AgentBuilder newAgentBuilder() {
         return new AgentBuilder.Default()
-                .with(AgentBuilder.RedefinitionStrategy.RETRANSFORMATION)
-                .with(AgentBuilder.InitializationStrategy.NoOp.INSTANCE)
-                .with(AgentBuilder.TypeStrategy.Default.REDEFINE)
-                .ignore(ElementMatchers.nameStartsWith("net.bytebuddy."));
+                .ignore(ElementMatchers.none())
+                .with(AgentBuilder.RedefinitionStrategy.REDEFINITION)
+                .with(AgentBuilder.Listener.StreamWriting.toSystemError().withErrorsOnly())
+                .with(AgentBuilder.Listener.StreamWriting.toSystemOut().withTransformationsOnly());
     }
 
 
diff --git a/jetbra-agent/src/main/java/com/novitechie/jetbra/LicensingFacadeAdvice.java b/jetbra-agent/src/main/java/com/novitechie/jetbra/LicensingFacadeAdvice.java
new file mode 100644
index 0000000..125b818
--- /dev/null
+++ b/jetbra-agent/src/main/java/com/novitechie/jetbra/LicensingFacadeAdvice.java
@@ -0,0 +1,22 @@
+package com.novitechie.jetbra;
+
+import net.bytebuddy.asm.Advice;
+
+import java.util.Calendar;
+import java.util.Date;
+
+public class LicensingFacadeAdvice {
+
+    @Advice.OnMethodExit
+    public static void intercept(@Advice.Return(readOnly = false) Date expirationDate) {
+        RuntimeException e = new RuntimeException();
+        for (StackTraceElement stackTraceElement : e.getStackTrace()) {
+            if (stackTraceElement.getFileName() == null){
+                Calendar calendar = Calendar.getInstance();
+                calendar.add(Calendar.DAY_OF_MONTH, 50);
+                expirationDate = calendar.getTime();
+                break;
+            }
+        }
+    }
+}
diff --git a/jetbra-server/src/main/resources/public/index.html b/jetbra-server/src/main/resources/public/index.html
index 3334c16..f7209c2 100644
--- a/jetbra-server/src/main/resources/public/index.html
+++ b/jetbra-server/src/main/resources/public/index.html
@@ -795,10 +795,13 @@
 </main>
 <footer class="flex items-center justify-between">
     <div class="text-sm text-grey">Theme by QieTuZai</div>
-    <div class="text-sm text-grey"><button class="jetbra-button" onclick="showLicenseForm()">Refill licensee information</button></div>
+    <div class="text-sm text-grey">
+        <button class="jetbra-button" onclick="showLicenseForm()">Refill licensee information</button>
+    </div>
     <div class="text-sm text-grey">Copy from ja-netfilter</div>
 </footer>
-<div id ="mask" style="position:fixed;top:0;left:0;z-index:998;width:100%;height:100%;display:none;background-color:#000;opacity:0.5;overflow:hidden;"></div>
+<div id="mask"
+     style="position:fixed;top:0;left:0;z-index:998;width:100%;height:100%;display:none;background-color:#000;opacity:0.5;overflow:hidden;"></div>
 <div id="form" style="position:fixed;top:20%;left:40%;width:40%;height:500px;z-index:999;display:none;">
     <div class="form">
         <div class="title">Welcome</div>
@@ -818,7 +821,7 @@
             <div class="cut cut-short"></div>
             <label for="expiryDate" class="placeholder">expiryDate</label>
         </div>
-        <button  class="submit" onclick="submitLicenseInfo(this)">submit</button>
+        <button class="submit" onclick="submitLicenseInfo(this)">submit</button>
     </div>
 </div>
 
@@ -847,7 +850,7 @@
             document.getElementById('licenseeName').value = licenseInfo.licenseeName
             document.getElementById('assigneeName').value = licenseInfo.assigneeName
             document.getElementById('expiryDate').value = licenseInfo.expiryDate
-        }else {
+        } else {
             document.getElementById('licenseeName').value = 'Test only'
             document.getElementById('assigneeName').value = 'Test only'
             document.getElementById('expiryDate').value = '2026-12-31'
@@ -858,26 +861,114 @@
 </script>
 
 <script>
-    // const codes = ["YTD", "QDGO", "MF", "DG", "PS", "QA", "IIE", "YTWE", "FLS", "DLE", "RFU", "PPS", "PCWMP", "II", "TCC", "RSU", "PCC", "RC", "PCE", "FLIJ", "TBA", "DL", "SPP", "QDCLD", "SPA", "DMCLP", "PSW", "GW", "PSI", "IIU", "DMU", "PWS", "HB", "WS", "PCP", "KT", "DCCLT", "RSCLT", "WRS", "RSC", "RRD", "TC", "IIC", "QDPY", "DPK", "DC", "PDB", "DPPS", "QDPHP", "GO", "HCC", "RDCPPP", "QDJVMC", "CL", "DM", "CWML", "FLL", "RR", "QDJS", "RS", "RM", "DS", "MPS", "DPN", "US", "CLN", "DPCLT", "RSV", "MPSIIP", "DB", "QDANDC", "AC", "QDJVM", "PRB", "RD", "CWMR", "SP", "RS0", "DP", "RSF", "PGO", "QDPYC", "PPC", "PC", "EHS", "RSCHB", "FL", "QDNET", "JCD"]
-    window.copyLicense = async function (e) {
+    const pemEncodedKey = `-----BEGIN PRIVATE KEY-----
+MIIJQgIBADANBgkqhkiG9w0BAQEFAASCCSwwggkoAgEAAoICAQC3p6nBj9mcRpGK
+pigPXOB83/PmA9bJr5jsSo3fm5ky67rTP4V79XI9a1t/5asg7XQ5OyulvP0w6tQk
+axLfg6Opd9A8YQIgt+Gh/A5hsIKu+8RKC4prx+S6Xj8X5RfrWwdUWbRYBQziGC3U
+kGihR9iQ4FSsYS4ld0uo54j4ZArVlq07PhOr6uDdeQZtzZzOQCSC6o7VGzozX2sV
+aukazqE3NEdxaqqOsr8aP/iWGtlJxyAvq9nWyrgyzFK7YJ8nRFHSTV9Mx/RbXHRC
+76+PLnPZmNN/E1lLGVCtaZ0G8QNmz8gOKp2CfSL1IDui7S17xhZtd+2EDEtTeNQB
+wwTq8KDSPFKA1/qiN6zPem4hThb5+xHZMu6wcs8m7dx/s8XaI3476S9RNDTvfAU/
+c37nxwGgMWbZZgzruSwyXtwhrq58kTERMW7XPkI5dZlIerRuJWpAMHbKMa8tBnRu
+9smmMm/Yred5GhZLP0/7O2e20Sc1Rc0A1dWOG81LZON7yptr87QVZUJAZGWOX9iW
+3uSIN2/LMEMwZzk9Jqy+Uj1IcJkOiMZBFs7Y+eThLowTJka/dqBErqvWWDGni4nS
+llUd//vhwPAUWWkRhCrUh2QmxRBXYoQ8cSb/V4ejbk/3sCeh2KftDUKRZ1Jye9p/
+MFfrbnaWpu2inHc1Zs+3DcDoRti7OwIDAQABAoICABM991zG6BtmD2ix+P+HESQ0
+SLcgTthJ1CFpvEyh3l7F8QiiHqe6szH5NhiD5TapemRmrS+LyhFegUShjVQq1DJ0
+0bYJyfHIolTY9l7I4iBYU5wYcnPReUcHid/EiomHu5BcZ7dTLCLiOqcLTFMdlnSz
+dFutQOr/AUfcnm67+KChTVwoKGJ6VP4PaJuHj/bSJKEs1zM/y4zHYg5X6b17ycth
+aFzbOqyB0OD8s9xySrLesKIeBNBq4/q6iq6ENJimIVaB8cq3JoSN/sZmm4PKb6vs
+RbiKO/BQ4jGRH7ky9lLG0WSelWsvFkMNkgIDjKDrw7zLdHDB4wCHZ9sZZkIBXTAL
+6ktSBFq2IMuyn3C6hhbYWHADOo7x/RStk88/sGF39TYSsK+76QuRQ8SBdvzlHS+e
+CNCJMIhZUHSUCn10mo17V6MDV/lXMuajSLlxzSsKxjzxFQxswIMCtEaxFMvqokZw
+pyZdsYs5aZGAaRQ0fBbCsVAR1neki5Z7hhChBFOf5DuMbU3djD6/efoqhyhscruD
+Vb8r2bslL244830ZhX3yJRKiyxvKNPvquuzORIG7BHi9kucU60zMrXZ8tGC2W07+
+KtPKxTc6SVe5QiPDUsi5okyM0qQJ/5oLxNaD9vOV2wCIKmfKotgC42svITDNLkvJ
+6nLFYUELHQcqVHlsmwW1AoIBAQD50eyXAk4FwlBxWDO8zJush7J3pRMPndWMDoL1
+5XEPCzENEz8FZoBCNCup6CSUSL8WjnOVrMyBRQgWfZhB7u2T3NBRsMLt66RCu2L5
+BjZHRi6F3nuIS9Xfs7CNi9D2tcI/FT7xPe3AEgJKuGIKaySUip1Q+dAAh3XyPlco
+1EPYpPJAY+FnYNStcBHLXa2v+v212GCMGB9WLhegBWt+FWjn/tMyqJifUIQy7M7k
+5dCLO6kmo0VPWMngspUPcX77JtJLOA/JgeBAO91uDJMWHelPS2zIkPZb3pG8L+yz
+K33Ry+YY0SUqzBLFfQmD6HWz7sivcv2aCHD0PcY8GcCQxAJNAoIBAQC8Mrl8w7vE
+LMDvbiAMoK6iYJr0FbFGoiDfJhLU5MkIRRv5439qtXmF3EVFcQTsXf6km0U/YaYq
+/e4b44YCjIQDiD/LJPjZThHiCyYduNl9RUeVYbAubtBK7MJ2KQxNVfG2XOYJmxg1
+j5/McX5v5JA+bTdtRp0OH0OYPiA/ilM2+Gp1m/qOD85OS+Z1Np+jNQ0UXY7LYZP3
+NbFdBRnil1obeZKxqOxdAuate+cioKrvHRvbHLF6GNWde9+f8q+2cfNZijyJwE6R
+vURwDCwdNUaPCTtc7s9NSP2WNHaOM4pkmlu6mgZl2PLzZimUCeev8EovGH3VAMl3
+i2ytNEJ+56enAoIBAQDGbZaFj6AXdPNeRBe8M6zHCnWYEPcl5VEUUQZ2eAsoTtRk
+NVBOYs8nRrcT2r8LRQj6yqVGUp2RZBp7esDwRe5RDwgsisEaJ5wuIRcJA4UjcbxM
+Op5WcR3s9JYcp9yPyWkDoEWBapYohGVroi7FZbsFfWBdTD+J3A60Hg4u8QL+1m1Q
+9cS4zzG+nRCVPtBRwoO456gwPozNcAj14rgxyqGr/D0WtNGdYV/P70aai2vs27OM
+bA0GbFjVcCNzw8t/g6NveZUYkl9jxekomzZNT+7cO+WpHXOBHzUUi+Bvo/DpLhKS
+zbS+3J9gW+Ot8XtkMxsWOLj0mxXU+ig13qKUmgvVAoIBAB2k08jOP/5HmmBcdVnn
+2XokQ2QdIp5gnVLo+WBlZTETSbPT3NcfHLQ0HQkyIzdkGt8swfyY0gbFlsL31L0E
+CytPQ9UozrXT8UcswGVAH6n2xq7GA21c8RxMLNlV3+Uym29BNM7gijCtndsjKWpQ
+k1Px+iARVl3KGOibKJM5o5/uAz7hQdcssC9vDy75Wq3nhlbl4b8xcJAo+fYP/qLN
+elkHjk7Dr+96rIE5GhA/RI2DhUa/P0lfLg6vW2sjXAAd9Nnux1hfXUDhki0gDbbQ
+FHwlVR9vUmH3FFKbku0VO0BbfAVpi4ZxZNtoBTaXVNJGxDik3/U0OYfGA2lI6Qx6
+StMCggEAV1XytpdVbCAlPitA5mkncFXXW6YhRufmkmzbYeTboPzYlNz9F2xmYjIo
+xNfzwiGepHyG38YdgNJ/h1NNo4a7JCLKRPReRca1V+td9BP7ZKAQEHAtAY6QwHJ1
+aJzZxmcohMWh9LXmUzeSnSIMbG/JNqIwy6W6EMmzC5eXL9FHaWCr3WQs05wE+CJF
+pJkXbmXkg+rbct9hAYKVw7zQjezTbfRPqcHdsHVOJBZCTbCSm44XWnLuu90jQ2Ku
+pTOTmM3h0mKOG8tVTaibJdeNHzk0+SDhUdOI5ORA0Q+iHZaEbPO39/c+sr0n9xLF
+17M9lCizO9o9dONdHsHfNQi6y9Jcnw==
+-----END PRIVATE KEY-----`;
 
-        while (localStorage.getItem('licenseInfo') === null) {
-            document.getElementById('mask').style.display = 'block'
-            document.getElementById('form').style.display = 'block'
-            await new Promise(r => setTimeout(r, 1000));
-        }
-        let licenseInfo =  JSON.parse(localStorage.getItem('licenseInfo'))
-        let codes = e.closest('.card').dataset.productCodes.split(',')
-        let products = Array.from(codes).map((code) => {
-            return {
-                code: code,
-                fallbackDate: licenseInfo.expiryDate,
-                paidUpTo: licenseInfo.expiryDate
-            }
-        })
-        let data = {
-            "licenseeName": licenseInfo.licenseeName,
-            "assigneeName": licenseInfo.assigneeName,
+
+    const pemEncodedCrt = `-----BEGIN CERTIFICATE-----
+MIIEtTCCAp2gAwIBAgIUDyuccmylba71lZQAQic5TJiAhwwwDQYJKoZIhvcNAQEL
+BQAwGDEWMBQGA1UEAwwNSmV0UHJvZmlsZSBDQTAeFw0yMzA5MjkxNDA2MTJaFw0z
+MzA5MjcxNDA2MTJaMBExDzANBgNVBAMMBk5vdmljZTCCAiIwDQYJKoZIhvcNAQEB
+BQADggIPADCCAgoCggIBALenqcGP2ZxGkYqmKA9c4Hzf8+YD1smvmOxKjd+bmTLr
+utM/hXv1cj1rW3/lqyDtdDk7K6W8/TDq1CRrEt+Do6l30DxhAiC34aH8DmGwgq77
+xEoLimvH5LpePxflF+tbB1RZtFgFDOIYLdSQaKFH2JDgVKxhLiV3S6jniPhkCtWW
+rTs+E6vq4N15Bm3NnM5AJILqjtUbOjNfaxVq6RrOoTc0R3Fqqo6yvxo/+JYa2UnH
+IC+r2dbKuDLMUrtgnydEUdJNX0zH9FtcdELvr48uc9mY038TWUsZUK1pnQbxA2bP
+yA4qnYJ9IvUgO6LtLXvGFm137YQMS1N41AHDBOrwoNI8UoDX+qI3rM96biFOFvn7
+Edky7rByzybt3H+zxdojfjvpL1E0NO98BT9zfufHAaAxZtlmDOu5LDJe3CGurnyR
+MRExbtc+Qjl1mUh6tG4lakAwdsoxry0GdG72yaYyb9it53kaFks/T/s7Z7bRJzVF
+zQDV1Y4bzUtk43vKm2vztBVlQkBkZY5f2Jbe5Ig3b8swQzBnOT0mrL5SPUhwmQ6I
+xkEWztj55OEujBMmRr92oESuq9ZYMaeLidKWVR3/++HA8BRZaRGEKtSHZCbFEFdi
+hDxxJv9Xh6NuT/ewJ6HYp+0NQpFnUnJ72n8wV+tudpam7aKcdzVmz7cNwOhG2Ls7
+AgMBAAEwDQYJKoZIhvcNAQELBQADggIBAIdeaQfKni7tXtcywC3zJvGzaaj242pS
+WB1y40HW8jub0uHjTLsBPX27iA/5rb+rNXtUWX/f2K+DU4IgaIiiHhkDrMsw7piv
+azqwA9h7/uA0A5nepmTYf/HY4W6P2stbeqInNsFRZXS7Jg4Q5LgEtHKo/H8USjtV
+w9apmE3BCElkXRuelXMsSllpR/JEVv/8NPLmnHSY02q4KMVW2ozXtaAxSYQmZswy
+P1YnBcnRukoI4igobpcKQXwGoQCIUlec8LbFXYM9V2eNCwgABqd4r67m7QJq31Y/
+1TJysQdMH+hoPFy9rqNCxSq3ptpuzcYAk6qVf58PrrYH/6bHwiYPAayvvdzNPOhM
+9OCwomfcazhK3y7HyS8aBLntTQYFf7vYzZxPMDybYTvJM+ClCNnVD7Q9fttIJ6eM
+XFsXb8YK1uGNjQW8Y4WHk1MCHuD9ZumWu/CtAhBn6tllTQWwNMaPOQvKf1kr1Kt5
+etrONY+B6O+Oi75SZbDuGz7PIF9nMPy4WB/8XgKdVFtKJ7/zLIPHgY8IKgbx/VTz
+6uBhYo8wOf3xzzweMnn06UcfV3JGNvtMuV4vlkZNNxXeifsgzHugCvJX0nybhfBh
+fIqVyfK6t0eKJqrvp54XFEtJGR+lf3pBfTdcOI6QFEPKGZKoQz8Ck+BC/WBDtbjc
+/uYKczZ8DKZu
+-----END CERTIFICATE-----`;
+
+    function pem2base64(pem) {
+        return pem.split('\n').reduce((base64, line) => line.includes("--") ? base64 : base64 + line, '');
+    }
+
+    function arrayBufferToBase64(buffer) {
+        return btoa([...new Uint8Array(buffer)].map(b => String.fromCharCode(b)).join(''));
+    }
+
+    function base64ToArrayBuffer(base64) {
+        return Uint8Array.from(atob(base64), c => c.charCodeAt(0)).buffer;
+    }
+
+    function genLicenseId() {
+        const CHARSET = "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
+        return Array.from({length: 10}, () => {
+            let idx = Math.floor(Math.random() * CHARSET.length);
+            return CHARSET[idx];
+        }).join('');
+    }
+
+    function buildLicensePartJson(products, licenseId) {
+        return JSON.stringify({
+            "licenseId": licenseId,
+            "licenseeName": "reborn",
+            "assigneeName": "reborn",
             "assigneeEmail": "",
             "licenseRestriction": "",
             "checkConcurrentUse": false,
@@ -887,19 +978,41 @@
             "gracePeriodDays": 7,
             "autoProlongated": true,
             "isAutoProlongated": true
+        })
+    }
+
+    window.copyLicense = async function (e) {
+        while (localStorage.getItem('licenseInfo') === null) {
+            document.getElementById('mask').style.display = 'block'
+            document.getElementById('form').style.display = 'block'
+            await new Promise(r => setTimeout(r, 1000));
         }
-        let resp = await fetch('/generateLicense', {
-            method: 'POST',
-            headers: {
-                'Content-Type': 'application/json'
-            },
-            body: JSON.stringify(data)
-        }).then(response => response.json())
-        navigator.clipboard.writeText(resp.license)
+        let licenseInfo = JSON.parse(localStorage.getItem('licenseInfo'))
+        let codes = e.closest('.card').dataset.productCodes.split(',')
+        let products = Array.from(codes).map((code) => {
+            return {
+                code: code,
+                fallbackDate: licenseInfo.expiryDate,
+                paidUpTo: licenseInfo.expiryDate
+            }
+        })
+        let licenseId = genLicenseId()
+        let licensePartJson = buildLicensePartJson(products, licenseId)
+
+        let privateKey = await window.crypto.subtle.importKey("pkcs8", base64ToArrayBuffer(pem2base64(pemEncodedKey)), {
+            name: "RSASSA-PKCS1-v1_5", hash: "SHA-1",
+        }, true, ["sign"]);
+
+        let licensePartBase64 = btoa(unescape(encodeURIComponent(licensePartJson)));
+        let sigResultsBase64 = arrayBufferToBase64(await window.crypto.subtle.sign("RSASSA-PKCS1-v1_5", privateKey, new TextEncoder().encode(licensePartJson)));
+        let cert_base64 = pem2base64(pemEncodedCrt);
+
+        navigator.clipboard.writeText(`${licenseId}-${licensePartBase64}-${sigResultsBase64}-${cert_base64}`)
             .then(() => {
                 alert("The activation code has been copied to your clipboard");
             })
     }
+
 </script>
 </body>
 </html>