Validate instanceof ManaKey (#36)

ramidzkh · Feb 18, 2023 · ace92c8 · ace92c8
1 parent ea6f346
commit ace92c8
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/src/main/java/appbot/ae2/ManaExternalStorageStrategy.java b/src/main/java/appbot/ae2/ManaExternalStorageStrategy.java
@@ -47,6 +47,10 @@ private record ManaStorageAdapter(ManaReceiver receiver, Runnable injectOrExtrac
 
         @Override
         public long insert(AEKey what, long amount, Actionable mode, IActionSource source) {
+            if (!(what instanceof ManaKey)) {
+                return 0;
+            }
+
             var inserted = (int) Math.min(amount,
                     ManaHelper.getCapacity(receiver) - receiver.getCurrentMana());
 
@@ -60,6 +64,10 @@ public long insert(AEKey what, long amount, Actionable mode, IActionSource sourc
 
         @Override
         public long extract(AEKey what, long amount, Actionable mode, IActionSource source) {
+            if (!(what instanceof ManaKey)) {
+                return 0;
+            }
+
             var extracted = (int) Math.min(amount, receiver.getCurrentMana());
 
             if (extracted > 0 && mode == Actionable.MODULATE) {

diff --git a/src/main/java/appbot/ae2/ManaStorageExportStrategy.java b/src/main/java/appbot/ae2/ManaStorageExportStrategy.java
@@ -29,6 +29,10 @@ public ManaStorageExportStrategy(ServerLevel level,
 
     @Override
     public long transfer(StackTransferContext context, AEKey what, long amount) {
+        if (!(what instanceof ManaKey)) {
+            return 0;
+        }
+
         var receiver = apiCache.find(fromSide);
 
         if (receiver == null) {
@@ -50,6 +54,10 @@ public long transfer(StackTransferContext context, AEKey what, long amount) {
 
     @Override
     public long push(AEKey what, long amount, Actionable mode) {
+        if (!(what instanceof ManaKey)) {
+            return 0;
+        }
+
         var receiver = apiCache.find(fromSide);
 
         if (receiver == null) {