Battle fix 2

P3D-Legacy · Jun 5, 2023 · b0d0630 · b0d0630
1 parent 7d723e0
commit b0d0630
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/src/P3D.Legacy.Server.Client.P3D/P3DConnectionContextHandler.Notifications.cs b/src/P3D.Legacy.Server.Client.P3D/P3DConnectionContextHandler.Notifications.cs
@@ -160,9 +160,22 @@ public async Task HandleAsync(IReceiveContext<PlayerSentRawP3DPacketEvent> conte
 
             if (p3dPacket is BattleOfferFromClientPacket battleOfferFromClientPacket)
             {
+                if (battleOfferFromClientPacket.DestinationPlayerOrigin != Origin) return;
                 await SendPacketAsync(new BattleOfferToClientPacket { Origin = battleOfferFromClientPacket.Origin, BattleData = battleOfferFromClientPacket.BattleData }, ct);
                 return;
             }
+            if (p3dPacket is BattleHostDataFromClientPacket battleHostDataFromClientPacket)
+            {
+                if (battleHostDataFromClientPacket.DestinationPlayerOrigin != Origin) return;
+                await SendPacketAsync(new BattleHostDataToClientPacket { Origin = battleHostDataFromClientPacket.Origin, BattleData = battleHostDataFromClientPacket.BattleData }, ct);
+                return;
+            }
+            if (p3dPacket is BattleEndRoundDataFromClientPacket battleEndRoundDataFromClientPacket)
+            {
+                if (battleEndRoundDataFromClientPacket.DestinationPlayerOrigin != Origin) return;
+                await SendPacketAsync(new BattleEndRoundDataToClientPacket { Origin = battleEndRoundDataFromClientPacket.Origin, BattleData = battleEndRoundDataFromClientPacket.BattleData }, ct);
+                return;
+            }
 
             await SendPacketAsync(p3dPacket, ct);
         }

diff --git a/src/P3D.Legacy.Server.Client.P3D/P3DConnectionContextHandler.Packets.cs b/src/P3D.Legacy.Server.Client.P3D/P3DConnectionContextHandler.Packets.cs
@@ -444,7 +444,7 @@ private async Task HandleBattleHostDataAsync(BattleHostDataFromClientPacket pack
             if (State != PlayerState.Initialized)
                 return;
 
-            await _eventDispatcher.DispatchAsync(new PlayerSentRawP3DPacketEvent(this, new BattleHostDataToClientPacket { Origin = packet.Origin, BattleData = packet.BattleData }), ct);
+            await _eventDispatcher.DispatchAsync(new PlayerSentRawP3DPacketEvent(this, packet), ct);
         }
         private async Task HandleBattleJoinAsync(BattleJoinPacket packet, CancellationToken ct)
         {
@@ -492,7 +492,7 @@ private async Task HandleBattlePokemonDataAsync(BattleEndRoundDataFromClientPack
             if (State != PlayerState.Initialized)
                 return;
 
-            await _eventDispatcher.DispatchAsync(new PlayerSentRawP3DPacketEvent(this, new BattleEndRoundDataToClientPacket { Origin = packet.Origin, BattleData = packet.BattleData }), ct);
+            await _eventDispatcher.DispatchAsync(new PlayerSentRawP3DPacketEvent(this, packet), ct);
         }
         private async Task HandleBattleQuitAsync(BattleQuitPacket packet, CancellationToken ct)
         {