bit writing fix

wootguy · Nov 11, 2024 · 3bea052 · 3bea052
1 parent 26174ba
commit 3bea052
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/dlls/mstream.cpp b/dlls/mstream.cpp
@@ -176,8 +176,8 @@ uint8_t mstream::writeBits(uint64_t value, uint8_t bitCount) {
 	return bitCount;
 }
 
-bool mstream::endBitWriting() {
-	if (currentBit == 0 || currentBit == 8) {
+bool mstream::endBitWriting() {	
+	if (currentBit == 0 || eomFlag) {
 		return false;
 	}
 

diff --git a/dlls/util.h b/dlls/util.h
@@ -245,6 +245,7 @@ inline entvars_t* VARS(EOFFSET eoffset)			{ return VARS(ENT(eoffset)); }
 inline int	  ENTINDEX(const edict_t *pEdict)			{ return (*g_engfuncs.pfnIndexOfEdict)(pEdict); }
 inline edict_t* INDEXENT( int iEdictNum )		{ return (*g_engfuncs.pfnPEntityOfEntIndex)(iEdictNum); }
 inline uint32_t PLRBIT(const edict_t* pEdict)			{ return 1 << (ENTINDEX(pEdict) & 31); }
+inline uint32_t PLRBIT(uint32_t idx)			{ return 1 << (idx & 31); }
 inline void MESSAGE_BEGIN( int msg_dest, int msg_type, const float *pOrigin, entvars_t *ent ) {
 	MESSAGE_BEGIN(msg_dest, msg_type, pOrigin, ENT(ent));
 }