diff options
Diffstat (limited to 'src/main/java/kr')
5 files changed, 321 insertions, 8 deletions
diff --git a/src/main/java/kr/syeyoung/dungeonsguide/Authenticator.java b/src/main/java/kr/syeyoung/dungeonsguide/Authenticator.java new file mode 100644 index 00000000..15b238c6 --- /dev/null +++ b/src/main/java/kr/syeyoung/dungeonsguide/Authenticator.java @@ -0,0 +1,133 @@ +package kr.syeyoung.dungeonsguide; + +import com.auth0.jwt.JWT; +import com.auth0.jwt.interfaces.DecodedJWT; +import com.google.gson.JsonObject; +import com.google.gson.JsonParser; +import com.mojang.authlib.GameProfile; +import com.mojang.authlib.exceptions.AuthenticationException; +import com.mojang.authlib.minecraft.MinecraftSessionService; +import lombok.Getter; +import net.minecraft.client.Minecraft; +import net.minecraft.util.Session; + +import javax.crypto.*; +import javax.crypto.spec.IvParameterSpec; +import javax.crypto.spec.SecretKeySpec; +import javax.xml.bind.DatatypeConverter; +import java.io.IOException; +import java.io.InputStream; +import java.io.InputStreamReader; +import java.math.BigInteger; +import java.net.*; +import java.security.*; +import java.util.UUID; + +public class Authenticator { + @Getter + private KeyPair keyPair; + @Getter + private String token; + private KeyPair generate1024RSAKey() { + KeyPairGenerator generator = null; + try { + generator = KeyPairGenerator.getInstance("RSA"); + } catch (NoSuchAlgorithmException e) { + e.printStackTrace(); + } + generator.initialize(1024); + keyPair = generator.generateKeyPair(); + return keyPair; + } + + public Authenticator() { + generate1024RSAKey(); + } + + private static final String DOMAIN = "http://localhost:8080/"; + + public String authenticate() throws IOException, AuthenticationException, NoSuchAlgorithmException { + Session session = Minecraft.getMinecraft().getSession(); + String token = session.getToken(); + + String jwt = requestAuth(session.getProfile()); + MinecraftSessionService yggdrasilMinecraftSessionService = Minecraft.getMinecraft().getSessionService(); + DecodedJWT jwt2 = JWT.decode(jwt); + String hash = calculateAuthHash(DatatypeConverter.parseBase64Binary(jwt2.getClaim("sharedSecret").asString()), + DatatypeConverter.parseBase64Binary(jwt2.getClaim("publicKey").asString())); + yggdrasilMinecraftSessionService.joinServer(session.getProfile(), token, hash); + token = requestAuth2(jwt, keyPair.getPublic()); + return token; + } + + private String requestAuth(GameProfile profile) throws IOException { + HttpURLConnection huc = (HttpURLConnection) new URL(DOMAIN+"auth/requestAuth").openConnection(); + huc.setRequestProperty("User-Agent", "DungeonsGuide/1.0"); + huc.setRequestProperty("Content-Type", "application/json"); + huc.setRequestMethod("POST"); + huc.setDoInput(true); + huc.setDoOutput(true); + + huc.getOutputStream().write(("{\"uuid\":\""+profile.getId().toString()+"\",\"nickname\":\""+profile.getName()+"\"}").getBytes()); + InputStreamReader inputStreamReader = new InputStreamReader(huc.getInputStream()); + JsonObject object = (JsonObject) new JsonParser().parse(inputStreamReader); + if (!"ok".equals(object.get("status").getAsString())) { + return null; + } + return object.get("data").getAsString(); + } + private String requestAuth2(String token, PublicKey publicKey) throws IOException { + HttpURLConnection huc = (HttpURLConnection) new URL(DOMAIN+"auth/authenticate").openConnection(); + huc.setRequestMethod("POST"); + huc.setRequestProperty("User-Agent", "DungeonsGuide/1.0"); + huc.setRequestProperty("Content-Type", "application/json"); + huc.setDoInput(true); + huc.setDoOutput(true); + + huc.getOutputStream().write(("{\"jwt\":\""+token+"\",\"publicKey\":\""+DatatypeConverter.printBase64Binary(publicKey.getEncoded())+"\"}").getBytes()); + InputStreamReader inputStreamReader = new InputStreamReader(huc.getInputStream()); + JsonObject object = (JsonObject) new JsonParser().parse(inputStreamReader); + if (!"ok".equals(object.get("status").getAsString())) { + return null; + } + return object.get("data").getAsString(); + } + public String calculateAuthHash(byte[] sharedSecret, byte[] pk) throws NoSuchAlgorithmException { + MessageDigest md = MessageDigest.getInstance("SHA-1"); + md.update("".getBytes()); + md.update(sharedSecret); + md.update(pk); + byte[] result = md.digest(); + return new BigInteger(result).toString(16); + } + public InputStream getInputStream(String resource) throws IOException, BadPaddingException, IllegalBlockSizeException, NoSuchPaddingException, NoSuchAlgorithmException, InvalidKeyException, InvalidAlgorithmParameterException { + HttpURLConnection huc = (HttpURLConnection) new URL(DOMAIN+"resource/resource?class="+ URLEncoder.encode(resource)).openConnection(); + huc.setRequestProperty("User-Agent", "DungeonsGuide/1.0"); + huc.setRequestProperty("Content-Type", "application/json"); + huc.setRequestProperty("Authorization", token); + huc.setDoInput(true); + huc.setDoOutput(true); + + InputStream inputStream = huc.getInputStream(); + byte[] bytes = new byte[4]; + inputStream.read(bytes); + int len = ((bytes[0] & 0xFF) << 24) | + ((bytes[1] & 0xFF) << 16) | + ((bytes[2] & 0xFF) << 8 ) | + ((bytes[3] & 0xFF)); + + byte[] pubKey = new byte[len]; + inputStream.read(pubKey); + + Cipher cipher = Cipher.getInstance("RSA"); + byte[] byteEncrypted = pubKey; + cipher.init(Cipher.DECRYPT_MODE, keyPair.getPrivate()); + byte[] bytePlain = cipher.doFinal(byteEncrypted); + + cipher = Cipher.getInstance("AES/CBC/PKCS5Padding"); + SecretKeySpec keySpec = new SecretKeySpec(bytePlain, "AES"); + IvParameterSpec ivSpec = new IvParameterSpec(bytePlain); + cipher.init(Cipher.DECRYPT_MODE,keySpec,ivSpec); + return new CipherInputStream(inputStream, cipher); + } +} diff --git a/src/main/java/kr/syeyoung/dungeonsguide/DungeonsGuide.java b/src/main/java/kr/syeyoung/dungeonsguide/DungeonsGuide.java index bcbcbe0a..de1b0067 100644 --- a/src/main/java/kr/syeyoung/dungeonsguide/DungeonsGuide.java +++ b/src/main/java/kr/syeyoung/dungeonsguide/DungeonsGuide.java @@ -1,21 +1,32 @@ package kr.syeyoung.dungeonsguide; +import com.mojang.authlib.exceptions.AuthenticationException; import kr.syeyoung.dungeonsguide.commands.*; import kr.syeyoung.dungeonsguide.dungeon.roomfinder.DungeonRoomInfoRegistry; import lombok.Getter; import net.minecraft.client.Minecraft; -import net.minecraft.command.CommandDebug; -import net.minecraft.init.Blocks; +import net.minecraft.client.gui.FontRenderer; +import net.minecraft.client.gui.GuiErrorScreen; +import net.minecraft.client.gui.GuiScreen; import net.minecraft.util.IChatComponent; import net.minecraftforge.client.ClientCommandHandler; +import net.minecraftforge.client.event.GuiOpenEvent; import net.minecraftforge.common.MinecraftForge; +import net.minecraftforge.fml.client.CustomModLoadingErrorDisplayException; import net.minecraftforge.fml.common.Mod; import net.minecraftforge.fml.common.Mod.EventHandler; import net.minecraftforge.fml.common.event.FMLInitializationEvent; import net.minecraftforge.fml.common.event.FMLPreInitializationEvent; +import net.minecraftforge.fml.common.eventhandler.SubscribeEvent; import org.apache.commons.io.IOUtils; +import javax.crypto.BadPaddingException; +import javax.crypto.IllegalBlockSizeException; +import javax.crypto.NoSuchPaddingException; import java.io.*; +import java.security.InvalidAlgorithmParameterException; +import java.security.InvalidKeyException; +import java.security.NoSuchAlgorithmException; @Mod(modid = DungeonsGuide.MODID, version = DungeonsGuide.VERSION) public class DungeonsGuide @@ -72,7 +83,23 @@ public class DungeonsGuide } } } - DungeonRoomInfoRegistry.loadAll(configDir); + try { + DungeonRoomInfoRegistry.loadAll(); + } catch (BadPaddingException e) { + e.printStackTrace(); + } catch (InvalidAlgorithmParameterException e) { + e.printStackTrace(); + } catch (NoSuchAlgorithmException e) { + e.printStackTrace(); + } catch (IOException e) { + e.printStackTrace(); + } catch (IllegalBlockSizeException e) { + e.printStackTrace(); + } catch (NoSuchPaddingException e) { + e.printStackTrace(); + } catch (InvalidKeyException e) { + e.printStackTrace(); + } Keybinds.register(); } @@ -88,9 +115,66 @@ public class DungeonsGuide @Getter private File configDir; + @Getter + private Authenticator authenticator; + private NetworkClassLoader classLoader; @EventHandler public void pre(FMLPreInitializationEvent event) { configDir = new File(event.getModConfigurationDirectory(),"dungeonsguide"); + + authenticator = new Authenticator(); + String token = null; + try { + token = authenticator.authenticate(); + System.out.println(token); + if (token != null) { + classLoader = new NetworkClassLoader(authenticator); + Thread.currentThread().setContextClassLoader(classLoader); + return; + } + } catch (IOException e) { + e.printStackTrace(); + } catch (AuthenticationException e) { + e.printStackTrace(); + } catch (NoSuchAlgorithmException e) { + e.printStackTrace(); + } + + final GuiScreen errorGui = new GuiErrorScreen(null, null) { + + @Override + public void handleMouseInput() { + } + + @Override + public void handleKeyboardInput() { + } + + @Override + public void drawScreen(int par1, int par2, float par3) { + drawDefaultBackground(); + String[] s_msg = new String[] { + "Can't validate current installation of Dungeons Guide", + "Please contact mod author if you purchased this mod and getting this error", + "And if you haven't purchased the mod, please consider doing so" + }; + for (int i = 0; i < s_msg.length; ++i) { + drawCenteredString(fontRendererObj, s_msg[i], width / 2, height / 3 + 12 * i, 0xFFFFFFFF); + } + } + }; + @SuppressWarnings("serial") CustomModLoadingErrorDisplayException e = new CustomModLoadingErrorDisplayException() { + + @Override + public void initGui(GuiErrorScreen errorScreen, FontRenderer fontRenderer) { + Minecraft.getMinecraft().displayGuiScreen(errorGui); + } + + @Override + public void drawScreen(GuiErrorScreen errorScreen, FontRenderer fontRenderer, int mouseRelX, int mouseRelY, float tickTime) { + } + }; + throw e; } public SkyblockStatus getSkyblockStatus() { diff --git a/src/main/java/kr/syeyoung/dungeonsguide/NetworkClassLoader.java b/src/main/java/kr/syeyoung/dungeonsguide/NetworkClassLoader.java new file mode 100644 index 00000000..564eeb11 --- /dev/null +++ b/src/main/java/kr/syeyoung/dungeonsguide/NetworkClassLoader.java @@ -0,0 +1,57 @@ +package kr.syeyoung.dungeonsguide; + +import javax.crypto.BadPaddingException; +import javax.crypto.IllegalBlockSizeException; +import javax.crypto.NoSuchPaddingException; +import java.io.ByteArrayOutputStream; +import java.io.File; +import java.io.IOException; +import java.io.InputStream; +import java.security.InvalidAlgorithmParameterException; +import java.security.InvalidKeyException; +import java.security.NoSuchAlgorithmException; + +public class NetworkClassLoader extends ClassLoader { + Authenticator authenticator; + + public NetworkClassLoader(Authenticator authenticator) { + super(); + this.authenticator = authenticator; + } + + @Override + public Class findClass(String name) throws ClassNotFoundException { + byte[] b = new byte[0]; + try { + b = loadClassFromFile(name); + return defineClass(name, b, 0, b.length); + } catch (BadPaddingException e) { + e.printStackTrace(); + } catch (InvalidAlgorithmParameterException e) { + e.printStackTrace(); + } catch (NoSuchAlgorithmException e) { + e.printStackTrace(); + } catch (IOException e) { + e.printStackTrace(); + } catch (IllegalBlockSizeException e) { + e.printStackTrace(); + } catch (NoSuchPaddingException e) { + e.printStackTrace(); + } catch (InvalidKeyException e) { + e.printStackTrace(); + } + throw new ClassNotFoundException(); + } + + private byte[] loadClassFromFile(String fileName) throws BadPaddingException, InvalidAlgorithmParameterException, NoSuchAlgorithmException, IOException, IllegalBlockSizeException, NoSuchPaddingException, InvalidKeyException { + byte[] buffer; + InputStream inputStream = authenticator.getInputStream(fileName.replace('.', '/')+ ".class"); + ByteArrayOutputStream byteStream = new ByteArrayOutputStream(); + int nextValue = 0; + while ( (nextValue = inputStream.read()) != -1 ) { + byteStream.write(nextValue); + } + buffer = byteStream.toByteArray(); + return buffer; + } +} diff --git a/src/main/java/kr/syeyoung/dungeonsguide/commands/CommandLoadData.java b/src/main/java/kr/syeyoung/dungeonsguide/commands/CommandLoadData.java index c1d3512e..637ec398 100644 --- a/src/main/java/kr/syeyoung/dungeonsguide/commands/CommandLoadData.java +++ b/src/main/java/kr/syeyoung/dungeonsguide/commands/CommandLoadData.java @@ -5,6 +5,14 @@ import kr.syeyoung.dungeonsguide.dungeon.roomfinder.DungeonRoomInfoRegistry; import net.minecraft.command.CommandBase; import net.minecraft.command.ICommandSender; +import javax.crypto.BadPaddingException; +import javax.crypto.IllegalBlockSizeException; +import javax.crypto.NoSuchPaddingException; +import java.io.IOException; +import java.security.InvalidAlgorithmParameterException; +import java.security.InvalidKeyException; +import java.security.NoSuchAlgorithmException; + public class CommandLoadData extends CommandBase { @Override public String getCommandName() { @@ -18,7 +26,23 @@ public class CommandLoadData extends CommandBase { @Override public void processCommand(ICommandSender sender, String[] args) { - DungeonRoomInfoRegistry.loadAll(DungeonsGuide.getDungeonsGuide().getConfigDir()); + try { + DungeonRoomInfoRegistry.loadAll(); + } catch (BadPaddingException e) { + e.printStackTrace(); + } catch (InvalidAlgorithmParameterException e) { + e.printStackTrace(); + } catch (NoSuchAlgorithmException e) { + e.printStackTrace(); + } catch (IOException e) { + e.printStackTrace(); + } catch (IllegalBlockSizeException e) { + e.printStackTrace(); + } catch (NoSuchPaddingException e) { + e.printStackTrace(); + } catch (InvalidKeyException e) { + e.printStackTrace(); + } } @Override public int getRequiredPermissionLevel() { diff --git a/src/main/java/kr/syeyoung/dungeonsguide/dungeon/roomfinder/DungeonRoomInfoRegistry.java b/src/main/java/kr/syeyoung/dungeonsguide/dungeon/roomfinder/DungeonRoomInfoRegistry.java index d9df6a8d..383aea08 100644 --- a/src/main/java/kr/syeyoung/dungeonsguide/dungeon/roomfinder/DungeonRoomInfoRegistry.java +++ b/src/main/java/kr/syeyoung/dungeonsguide/dungeon/roomfinder/DungeonRoomInfoRegistry.java @@ -1,8 +1,17 @@ package kr.syeyoung.dungeonsguide.dungeon.roomfinder; +import jdk.internal.util.xml.impl.Input; +import kr.syeyoung.dungeonsguide.Authenticator; +import kr.syeyoung.dungeonsguide.DungeonsGuide; import kr.syeyoung.dungeonsguide.dungeon.data.DungeonRoomInfo; +import javax.crypto.BadPaddingException; +import javax.crypto.IllegalBlockSizeException; +import javax.crypto.NoSuchPaddingException; import java.io.*; +import java.security.InvalidAlgorithmParameterException; +import java.security.InvalidKeyException; +import java.security.NoSuchAlgorithmException; import java.util.*; public class DungeonRoomInfoRegistry { @@ -41,6 +50,7 @@ public class DungeonRoomInfoRegistry { } public static void saveAll(File dir) { + if (!DungeonsGuide.DEBUG) return; dir.mkdirs(); for (DungeonRoomInfo dungeonRoomInfo : registered) { try { @@ -53,14 +63,19 @@ public class DungeonRoomInfoRegistry { } } - public static void loadAll(File dir) { + public static void loadAll() throws BadPaddingException, InvalidAlgorithmParameterException, NoSuchAlgorithmException, IOException, IllegalBlockSizeException, NoSuchPaddingException, InvalidKeyException { registered.clear(); shapeMap.clear(); uuidMap.clear(); - for (File f: dir.listFiles()) { - if (!f.isFile() || !f.getName().endsWith(".roomdata")) continue; + Authenticator authenticator = DungeonsGuide.getDungeonsGuide().getAuthenticator(); + InputStream inputStream = authenticator.getInputStream("roomdata/datas.txt"); + BufferedReader br = new BufferedReader(new InputStreamReader(inputStream)); + String name; + while ((name = br.readLine()) != null) { + System.out.println("Trying to load "+name); + if (name.endsWith(".roomdata")) continue; try { - FileInputStream fis = new FileInputStream(f); + InputStream fis = authenticator.getInputStream(name); ObjectInputStream ois = new ObjectInputStream(fis); DungeonRoomInfo dri = (DungeonRoomInfo) ois.readObject(); ois.close(); |