74 lines
2.7 KiB
Java
74 lines
2.7 KiB
Java
package com.boydti.fawe.util;
|
|
|
|
import com.sk89q.worldedit.internal.util.LogManagerCompat;
|
|
import org.apache.logging.log4j.Logger;
|
|
|
|
import java.io.DataInputStream;
|
|
import java.io.IOException;
|
|
import java.net.URL;
|
|
import java.security.MessageDigest;
|
|
import java.security.NoSuchAlgorithmException;
|
|
import java.util.Base64;
|
|
|
|
/** An internal FAWE class not meant for public use. **/
|
|
public enum ThirdPartyManager {
|
|
|
|
MapManager(
|
|
"https://github.com/InventivetalentDev/MapManager/releases/download/1.7.8-SNAPSHOT/MapManager_v1.7.8-SNAPSHOT.jar",
|
|
"m3YLUqZz66k2DmvdcYLeu38u3zKRKhrAXqGGpVKfO6g=", 544000),
|
|
|
|
PacketListenerAPI(
|
|
"https://github.com/InventivetalentDev/PacketListenerAPI/releases/download/3.7.6-SNAPSHOT/PacketListenerAPI_v3.7.6-SNAPSHOT.jar",
|
|
"etdBRzLn5pRVDfr/mSQdPm6Jjer3wQOKhcn8fUxo5zM=", 143000),
|
|
|
|
;
|
|
|
|
private static final Logger LOGGER = LogManagerCompat.getLogger();
|
|
|
|
public final String url;
|
|
public final int fileSize;
|
|
public final String digest;
|
|
|
|
/**
|
|
* @param url Where this jar can be found and downloaded
|
|
* @param digest The Base64-encoded SHA-256 digest
|
|
* @param fileSize Size of this jar in bytes
|
|
*/
|
|
ThirdPartyManager(String url, String digest, int fileSize) {
|
|
this.url = url;
|
|
this.digest = digest;
|
|
this.fileSize = fileSize;
|
|
}
|
|
|
|
/**
|
|
* Download a jar, verify hash, return byte[] containing the jar
|
|
*/
|
|
public byte[] download() throws IOException {
|
|
byte[] jarBytes = new byte[this.fileSize];
|
|
URL url = new URL(this.url);
|
|
try (DataInputStream dis = new DataInputStream(url.openConnection().getInputStream())) {
|
|
dis.readFully(jarBytes);
|
|
if (dis.read() != -1) { // assert that we've read everything
|
|
throw new IllegalStateException("Downloaded jar is longer than expected");
|
|
}
|
|
MessageDigest md = MessageDigest.getInstance("SHA-256");
|
|
byte[] jarDigestBytes = md.digest(jarBytes);
|
|
|
|
String jarDigest = Base64.getEncoder().encodeToString(jarDigestBytes);
|
|
|
|
if (this.digest.equals(jarDigest)) {
|
|
LOGGER.debug("++++ HASH CHECK ++++");
|
|
LOGGER.debug(this.url);
|
|
LOGGER.debug(this.digest);
|
|
return jarBytes;
|
|
} else {
|
|
LOGGER.debug(jarDigest + " | " + url);
|
|
throw new IllegalStateException("The downloaded jar does not match the hash");
|
|
}
|
|
} catch (NoSuchAlgorithmException e) {
|
|
// Shouldn't ever happen, Minecraft won't even run on such a JRE
|
|
throw new IllegalStateException("Your JRE does not support SHA-256");
|
|
}
|
|
}
|
|
}
|