From 031fe61a53cc3a9edd4883ffc0190095eba24905 Mon Sep 17 00:00:00 2001
From: megamage <none@none>
Date: Wed, 8 Apr 2009 23:05:17 -0500
Subject: *Fix a crash when a player with a charmed target removes the pet.

--HG--
branch : trunk
---
 src/game/Unit.cpp | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

(limited to 'src')

diff --git a/src/game/Unit.cpp b/src/game/Unit.cpp
index aba18ca3b9d..eb5bd306511 100644
--- a/src/game/Unit.cpp
+++ b/src/game/Unit.cpp
@@ -8012,7 +8012,7 @@ void Unit::SetGuardian(Guardian* guardian, bool apply)
 
         if(AddUInt64Value(UNIT_FIELD_SUMMON, guardian->GetGUID()))
         {
-            if(GetTypeId() == TYPEID_PLAYER)
+            if(GetTypeId() == TYPEID_PLAYER && !GetCharmGUID())
             {
                 if(guardian->isPet())
                     ((Player*)this)->PetSpellInitialize();
@@ -8047,12 +8047,17 @@ void Unit::SetGuardian(Guardian* guardian, bool apply)
             //Check if there is another guardian
             for(ControlList::iterator itr = m_Controlled.begin(); itr != m_Controlled.end(); ++itr)
             {
+                if(GetCharmGUID() == (*itr)->GetGUID())
+                    continue;
+
                 assert((*itr)->GetOwnerGUID() == GetGUID());
+                assert((*itr)->GetTypeId() == TYPEID_UNIT);
                 if(AddUInt64Value(UNIT_FIELD_SUMMON, (*itr)->GetGUID()))
                 {
-                    if(GetTypeId() == TYPEID_PLAYER)
+                    //show another pet bar if there is no charm bar
+                    if(GetTypeId() == TYPEID_PLAYER && !GetCharmGUID())
                     {
-                        if(guardian->isPet())
+                        if(((Creature*)(*itr))->isPet())
                             ((Player*)this)->PetSpellInitialize();
                         else
                             ((Player*)this)->CharmSpellInitialize();
-- 
cgit v1.2.3