Stop NetworkPacket methods from producing bloated packets

jayarndt · est31 · commit 2923eafaca92 · 2015-05-05T01:47:41.000+02:00
diff --git a/src/network/networkpacket.cpp b/src/network/networkpacket.cpp
@@ -72,8 +72,8 @@ char* NetworkPacket::getString(u32 from_offset)
 
 void NetworkPacket::putRawString(const char* src, u32 len)
 {
-	if (m_read_offset + len * sizeof(char) >= m_datasize) {
-		m_datasize += len * sizeof(char);
+	if (m_read_offset + len > m_datasize) {
+		m_datasize = m_read_offset + len;
 		m_data.resize(m_datasize);
 	}
 
@@ -95,7 +95,7 @@ NetworkPacket& NetworkPacket::operator>>(std::string& dst)
 	dst.reserve(strLen);
 	dst.append((char*)&m_data[m_read_offset], strLen);
 
-	m_read_offset += strLen * sizeof(char);
+	m_read_offset += strLen;
 	return *this;
 }
 
@@ -108,13 +108,7 @@ NetworkPacket& NetworkPacket::operator<<(std::string src)
 
 	*this << msgsize;
 
-	if (m_read_offset + msgsize * sizeof(char) >= m_datasize) {
-		m_datasize += msgsize * sizeof(char);
-		m_data.resize(m_datasize);
-	}
-
-	memcpy(&m_data[m_read_offset], src.c_str(), msgsize);
-	m_read_offset += msgsize;
+	putRawString(src.c_str(), (u32)msgsize);
 
 	return *this;
 }
@@ -128,13 +122,7 @@ void NetworkPacket::putLongString(std::string src)
 
 	*this << msgsize;
 
-	if (m_read_offset + msgsize * sizeof(char) >= m_datasize) {
-		m_datasize += msgsize * sizeof(char);
-		m_data.resize(m_datasize);
-	}
-
-	memcpy(&m_data[m_read_offset], src.c_str(), msgsize);
-	m_read_offset += msgsize;
+	putRawString(src.c_str(), msgsize);
 }
 
 NetworkPacket& NetworkPacket::operator>>(std::wstring& dst)
@@ -189,7 +177,7 @@ std::string NetworkPacket::readLongString()
 	dst.reserve(strLen);
 	dst.append((char*)&m_data[m_read_offset], strLen);
 
-	m_read_offset += strLen*sizeof(char);
+	m_read_offset += strLen;
 
 	return dst;
 }
diff --git a/src/network/networkpacket.h b/src/network/networkpacket.h
@@ -114,7 +114,7 @@ class NetworkPacket
 		template<typename T> void checkDataSize()
 		{
 			if (m_read_offset + sizeof(T) > m_datasize) {
-				m_datasize += sizeof(T);
+				m_datasize = m_read_offset + sizeof(T);
 				m_data.resize(m_datasize);
 			}
 		}

Original file line number	Diff line number	Diff line change
`@@ -114,7 +114,7 @@ class NetworkPacket`
`114`	`114`	`template<typename T> void checkDataSize()`
`115`	`115`	`{`
`116`	`116`	`if (m_read_offset + sizeof(T) > m_datasize) {`
`117`		`- m_datasize += sizeof(T);`
	`117`	`+ m_datasize = m_read_offset + sizeof(T);`
`118`	`118`	`m_data.resize(m_datasize);`
`119`	`119`	`}`
`120`	`120`	`}`