Closes s1lentq#176

FWGS · Nov 1, 2017 · 026a55e · 026a55e
1 parent 875189f
commit 026a55e
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/regamedll/dlls/weapons.cpp b/regamedll/dlls/weapons.cpp
@@ -789,10 +789,17 @@ BOOL CanAttack(float attack_time, float curtime, BOOL isPredicted)
 
 bool CBasePlayerWeapon::HasSecondaryAttack()
 {
-	if (m_pPlayer->HasShield())
+	if (m_pPlayer && m_pPlayer->HasShield())
+	{
+		return true;
+	}
+
+#ifdef REGAMEDLL_API
+	if (CSPlayerItem()->m_bHasSecondaryAttack)
 	{
 		return true;
 	}
+#endif
 
 	switch (m_iId)
 	{
@@ -1084,6 +1091,8 @@ void CBasePlayerWeapon::Spawn()
 	if (GetItemInfo(&info)) {
 		CSPlayerItem()->SetItemInfo(&info);
 	}
+
+	CSPlayerItem()->m_bHasSecondaryAttack = HasSecondaryAttack();
 }
 
 // CALLED THROUGH the newly-touched weapon's instance. The existing player weapon is pOriginal

diff --git a/regamedll/public/regamedll/API/CSPlayerItem.h b/regamedll/public/regamedll/API/CSPlayerItem.h
@@ -32,7 +32,8 @@ class CBasePlayerItem;
 class CCSPlayerItem: public CCSAnimating
 {
 public:
-	CCSPlayerItem()
+	CCSPlayerItem() :
+		m_bHasSecondaryAttack(false)
 	{
 		Q_memset(&m_ItemInfo, 0, sizeof(m_ItemInfo));
 	}
@@ -43,6 +44,7 @@ class CCSPlayerItem: public CCSAnimating
 
 public:
 	ItemInfo m_ItemInfo;
+	bool m_bHasSecondaryAttack;
 };
 
 // Inlines