[7769] Prevent access to deleted aura data in Aura::HandleModMechanicImmunity

This commit is contained in:
VladimirMangos 2009-05-05 11:54:50 +04:00
parent 881cceff8c
commit 565afd4d57
2 changed files with 26 additions and 21 deletions

View file

@ -3986,21 +3986,26 @@ void Aura::HandleModMechanicImmunity(bool apply, bool /*Real*/)
if(GetId()==42292 || GetId()==59752) if(GetId()==42292 || GetId()==59752)
mechanic=IMMUNE_TO_MOVEMENT_IMPAIRMENT_AND_LOSS_CONTROL_MASK; mechanic=IMMUNE_TO_MOVEMENT_IMPAIRMENT_AND_LOSS_CONTROL_MASK;
if(apply && GetSpellProto()->AttributesEx & SPELL_ATTR_EX_DISPEL_AURAS_ON_IMMUNITY) // cache values in local vars for prevent access to possible deleted aura data
SpellEntry const* spellInfo = GetSpellProto();
uint32 misc = m_modifier.m_miscvalue;
Unit* target = m_target;
if(apply && spellInfo->AttributesEx & SPELL_ATTR_EX_DISPEL_AURAS_ON_IMMUNITY)
{ {
Unit::AuraMap& Auras = m_target->GetAuras(); Unit::AuraMap& Auras = target->GetAuras();
for(Unit::AuraMap::iterator iter = Auras.begin(), next; iter != Auras.end(); iter = next) for(Unit::AuraMap::iterator iter = Auras.begin(), next; iter != Auras.end(); iter = next)
{ {
next = iter; next = iter;
++next; ++next;
SpellEntry const *spell = iter->second->GetSpellProto(); SpellEntry const *spell = iter->second->GetSpellProto();
if (!( spell->Attributes & SPELL_ATTR_UNAFFECTED_BY_INVULNERABILITY) && // spells unaffected by invulnerability if (!( spell->Attributes & SPELL_ATTR_UNAFFECTED_BY_INVULNERABILITY) && // spells unaffected by invulnerability
spell->Id != GetId()) spell->Id != spellInfo->Id)
{ {
//check for mechanic mask //check for mechanic mask
if(GetSpellMechanicMask(spell, iter->second->GetEffIndex()) & mechanic) if(GetSpellMechanicMask(spell, iter->second->GetEffIndex()) & mechanic)
{ {
m_target->RemoveAurasDueToSpell(spell->Id); target->RemoveAurasDueToSpell(spell->Id);
if(Auras.empty()) if(Auras.empty())
break; break;
else else
@ -4010,19 +4015,19 @@ void Aura::HandleModMechanicImmunity(bool apply, bool /*Real*/)
} }
} }
m_target->ApplySpellImmune(GetId(),IMMUNITY_MECHANIC,m_modifier.m_miscvalue,apply); target->ApplySpellImmune(spellInfo->Id,IMMUNITY_MECHANIC,misc,apply);
// Bestial Wrath // Bestial Wrath
if ( GetSpellProto()->SpellFamilyName == SPELLFAMILY_HUNTER && GetSpellProto()->SpellIconID == 1680) if (spellInfo->SpellFamilyName == SPELLFAMILY_HUNTER && spellInfo->SpellIconID == 1680)
{ {
// The Beast Within cast on owner if talent present // The Beast Within cast on owner if talent present
if ( Unit* owner = m_target->GetOwner() ) if (Unit* owner = target->GetOwner())
{ {
// Search talent // Search talent
Unit::AuraList const& m_dummyAuras = owner->GetAurasByType(SPELL_AURA_DUMMY); Unit::AuraList const& dummyAuras = owner->GetAurasByType(SPELL_AURA_DUMMY);
for(Unit::AuraList::const_iterator i = m_dummyAuras.begin(); i != m_dummyAuras.end(); ++i) for(Unit::AuraList::const_iterator i = dummyAuras.begin(); i != dummyAuras.end(); ++i)
{ {
if ( (*i)->GetSpellProto()->SpellIconID == 2229 ) if ((*i)->GetSpellProto()->SpellIconID == 2229)
{ {
if (apply) if (apply)
owner->CastSpell(owner, 34471, true, 0, this); owner->CastSpell(owner, 34471, true, 0, this);
@ -4035,21 +4040,21 @@ void Aura::HandleModMechanicImmunity(bool apply, bool /*Real*/)
} }
// The Beast Within and Bestial Wrath - immunity // The Beast Within and Bestial Wrath - immunity
if(GetId() == 19574 || GetId() == 34471) if (spellInfo->Id == 19574 || spellInfo->Id == 34471)
{ {
if(apply) if (apply)
{ {
m_target->CastSpell(m_target,24395,true); target->CastSpell(target,24395,true);
m_target->CastSpell(m_target,24396,true); target->CastSpell(target,24396,true);
m_target->CastSpell(m_target,24397,true); target->CastSpell(target,24397,true);
m_target->CastSpell(m_target,26592,true); target->CastSpell(target,26592,true);
} }
else else
{ {
m_target->RemoveAurasDueToSpell(24395); target->RemoveAurasDueToSpell(24395);
m_target->RemoveAurasDueToSpell(24396); target->RemoveAurasDueToSpell(24396);
m_target->RemoveAurasDueToSpell(24397); target->RemoveAurasDueToSpell(24397);
m_target->RemoveAurasDueToSpell(26592); target->RemoveAurasDueToSpell(26592);
} }
} }
} }

View file

@ -1,4 +1,4 @@
#ifndef __REVISION_NR_H__ #ifndef __REVISION_NR_H__
#define __REVISION_NR_H__ #define __REVISION_NR_H__
#define REVISION_NR "7768" #define REVISION_NR "7769"
#endif // __REVISION_NR_H__ #endif // __REVISION_NR_H__