Added extra check to proxies

Auxilor · Jul 8, 2024 · 871e084 · 871e084
1 parent a9624fe
commit 871e084
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/core/common/src/main/kotlin/com/willfp/libreforge/proxy/Proxies.kt b/core/common/src/main/kotlin/com/willfp/libreforge/proxy/Proxies.kt
@@ -30,9 +30,14 @@ private fun loadProxyUncached(clazz: Class<*>): Any {
 
     try {
         val found = Class.forName(className)
+
         val constructor = found.getConstructor()
         val instance = constructor.newInstance()
 
+        if (!clazz.isInstance(instance)) {
+            throw InvalidProxyException("Proxy class $className does not implement ${clazz.name}")
+        }
+
         return instance
     } catch (e: ClassNotFoundException) {
         throw InvalidProxyException("Could not find proxy class $className")