Lanmikeman/Launcher
1
1
Fork 0
mirror of https://github.com/GravitLauncher/Launcher synced 2024-12-23 00:51:01 +03:00
Code Issues Projects Releases Packages Wiki Activity

[FEATURE] CertificateService

Browse source
This commit is contained in:
Gravita 2021-03-06 17:18:38 +07:00
parent 8a016b7d46
commit 9d06689f35
2 changed files with 113 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

96
Launcher/src/main/java/pro/gravit/launcher/api/CertificateService.java Normal file
View file

@ -0,0 +1,96 @@
package pro.gravit.launcher.api;
import pro.gravit.launcher.Launcher;
import pro.gravit.launcher.LauncherTrustManager;
import pro.gravit.launcher.utils.ApiBridgeService;
import java.security.cert.X509Certificate;
import static pro.gravit.launcher.LauncherEngine.getCertificates;
public class CertificateService {
private CertificateService() {
throw new UnsupportedOperationException();
}
public static CheckClassResultApi checkClass(Class<?> clazz) throws SecurityException {
X509Certificate[] certificates = getCertificates(clazz);
if (certificates == null) {
return new CheckClassResultApi(CheckClassResultTypeApi.NOT_SIGNED, null, null);
}
try {
return CheckClassResultApi.fromCheckClassResult(ApiBridgeService.checkCertificates(certificates));
} catch (Exception e) {
throw new SecurityException(e);
}
}
public static void checkClassSuccess(Class<?> clazz) {
X509Certificate[] certificates = getCertificates(clazz);
if (certificates == null) {
throw new SecurityException(String.format("Class %s not signed", clazz.getName()));
}
try {
ApiBridgeService.checkCertificatesSuccess(certificates);
} catch (Exception e) {
throw new SecurityException(e);
}
}
public enum CheckClassResultTypeApi {
NOT_SIGNED,
SUCCESS,
UNTRUSTED,
UNVERIFED,
UNCOMPAT,
UNKNOWN
}
public static class CheckClassResultApi {
public final CheckClassResultTypeApi type;
public final X509Certificate endCertificate;
public final X509Certificate rootCertificate;
public final Exception exception;
private CheckClassResultApi(CheckClassResultTypeApi type, X509Certificate endCertificate, X509Certificate rootCertificate) {
this.type = type;
this.endCertificate = endCertificate;
this.rootCertificate = rootCertificate;
exception = null;
}
private CheckClassResultApi(CheckClassResultTypeApi type, X509Certificate endCertificate, X509Certificate rootCertificate, Exception exception) {
this.type = type;
this.endCertificate = endCertificate;
this.rootCertificate = rootCertificate;
this.exception = exception;
}
private CheckClassResultApi(CheckClassResultApi orig) {
this.type = orig.type;
this.exception = orig.exception;
this.rootCertificate = orig.rootCertificate;
this.endCertificate = orig.endCertificate;
}
private static CheckClassResultApi fromCheckClassResult(LauncherTrustManager.CheckClassResult result) {
if(result == null) return null;
return new CheckClassResultApi(fromType(result.type), result.endCertificate, result.rootCertificate, result.exception);
}
private static CheckClassResultTypeApi fromType(LauncherTrustManager.CheckClassResultType type) {
if(type == null) return null;
switch (type) {
case NOT_SIGNED:
return CheckClassResultTypeApi.NOT_SIGNED;
case SUCCESS:
return CheckClassResultTypeApi.SUCCESS;
case UNTRUSTED:
return CheckClassResultTypeApi.UNTRUSTED;
case UNVERIFED:
return CheckClassResultTypeApi.UNVERIFED;
case UNCOMPAT:
return CheckClassResultTypeApi.UNCOMPAT;
default:
return CheckClassResultTypeApi.UNKNOWN;
}
}
}
}

17
Launcher/src/main/java/pro/gravit/launcher/utils/ApiBridgeService.java Normal file
View file

@ -0,0 +1,17 @@
package pro.gravit.launcher.utils;
import pro.gravit.launcher.Launcher;
import pro.gravit.launcher.LauncherTrustManager;
import java.security.cert.X509Certificate;
public class ApiBridgeService {
public static LauncherTrustManager.CheckClassResult checkCertificates(X509Certificate[] certs) {
LauncherTrustManager trustManager = Launcher.getConfig().trustManager;
return trustManager.checkCertificates(certs, trustManager::stdCertificateChecker);
}
public static void checkCertificatesSuccess(X509Certificate[] certs) throws Exception {
LauncherTrustManager trustManager = Launcher.getConfig().trustManager;
trustManager.checkCertificatesSuccess(certs, trustManager::stdCertificateChecker);
}
}