diff --git a/amxmodx/CDataPack.cpp b/amxmodx/CDataPack.cpp index 658e8484..790faa3a 100644 --- a/amxmodx/CDataPack.cpp +++ b/amxmodx/CDataPack.cpp @@ -8,7 +8,7 @@ * This program is free software; you can redistribute it and/or modify it under * the terms of the GNU General Public License, version 3.0, as published by the * Free Software Foundation. - * + * * This program is distributed in the hope that it will be useful, but WITHOUT * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more @@ -74,9 +74,12 @@ void CDataPack::ResetSize() size_t CDataPack::CreateMemory(size_t size, void **addr) { - CheckSize(sizeof(size_t) + size); + CheckSize(sizeof(char) + sizeof(size_t) + size); size_t pos = m_curptr - m_pBase; + *(char *)m_curptr = Raw; + m_curptr += sizeof(char); + *(size_t *)m_curptr = size; m_curptr += sizeof(size_t); @@ -86,14 +89,17 @@ size_t CDataPack::CreateMemory(size_t size, void **addr) } m_curptr += size; - m_size += sizeof(size_t) + size; + m_size += sizeof(char) + sizeof(size_t) + size; return pos; } void CDataPack::PackCell(cell cells) { - CheckSize(sizeof(size_t) + sizeof(cell)); + CheckSize(sizeof(char) + sizeof(size_t) + sizeof(cell)); + + *(char *)m_curptr = DataPackType::Cell; + m_curptr += sizeof(char); *(size_t *)m_curptr = sizeof(cell); m_curptr += sizeof(size_t); @@ -101,12 +107,15 @@ void CDataPack::PackCell(cell cells) *(cell *)m_curptr = cells; m_curptr += sizeof(cell); - m_size += sizeof(size_t) + sizeof(cell); + m_size += sizeof(char) + sizeof(size_t) + sizeof(cell); } void CDataPack::PackFloat(float val) { - CheckSize(sizeof(size_t) + sizeof(float)); + CheckSize(sizeof(char) + sizeof(size_t) + sizeof(float)); + + *(char *)m_curptr = DataPackType::Float; + m_curptr += sizeof(char); *(size_t *)m_curptr = sizeof(float); m_curptr += sizeof(size_t); @@ -114,15 +123,18 @@ void CDataPack::PackFloat(float val) *(float *)m_curptr = val; m_curptr += sizeof(float); - m_size += sizeof(size_t) + sizeof(float); + m_size += sizeof(char) + sizeof(size_t) + sizeof(float); } void CDataPack::PackString(const char *string) { size_t len = strlen(string); - size_t maxsize = sizeof(size_t) + len + 1; + size_t maxsize = sizeof(char) + sizeof(size_t) + len + 1; CheckSize(maxsize); + *(char *)m_curptr = DataPackType::String; + m_curptr += sizeof(char); + // Pack the string length first for buffer overrun checking. *(size_t *)m_curptr = len; m_curptr += sizeof(size_t); @@ -156,17 +168,32 @@ bool CDataPack::SetPosition(size_t pos) const return true; } +bool CDataPack::CanReadCell() const +{ + if (!IsReadable(sizeof(char) + sizeof(size_t) + sizeof(cell))) + { + return false; + } + if (*reinterpret_cast(m_curptr) != DataPackType::Cell) + { + return false; + } + if (*reinterpret_cast(m_curptr + sizeof(char)) != sizeof(cell)) + { + return false; + } + + return true; +} + cell CDataPack::ReadCell() const { - if (!IsReadable(sizeof(size_t) + sizeof(cell))) - { - return 0; - } - if (*reinterpret_cast(m_curptr) != sizeof(cell)) + if (!CanReadCell()) { return 0; } + m_curptr += sizeof(char); m_curptr += sizeof(size_t); cell val = *reinterpret_cast(m_curptr); @@ -174,17 +201,32 @@ cell CDataPack::ReadCell() const return val; } +bool CDataPack::CanReadFloat() const +{ + if (!IsReadable(sizeof(char) + sizeof(size_t) + sizeof(float))) + { + return false; + } + if (*reinterpret_cast(m_curptr) != DataPackType::Float) + { + return false; + } + if (*reinterpret_cast(m_curptr + sizeof(char)) != sizeof(float)) + { + return false; + } + + return true; +} + float CDataPack::ReadFloat() const { - if (!IsReadable(sizeof(size_t) + sizeof(float))) - { - return 0; - } - if (*reinterpret_cast(m_curptr) != sizeof(float)) + if (!CanReadFloat()) { return 0; } + m_curptr += sizeof(char); m_curptr += sizeof(size_t); float val = *reinterpret_cast(m_curptr); @@ -197,21 +239,23 @@ bool CDataPack::IsReadable(size_t bytes) const return (bytes + (m_curptr - m_pBase) > m_size) ? false : true; } -const char *CDataPack::ReadString(size_t *len) const +bool CDataPack::CanReadString(size_t *len) const { - if (!IsReadable(sizeof(size_t))) + if (!IsReadable(sizeof(char) + sizeof(size_t))) { - return NULL; + return false; + } + if (*reinterpret_cast(m_curptr) != DataPackType::String) + { + return false; } - size_t real_len = *(size_t *)m_curptr; + size_t real_len = *(size_t *)(m_curptr + sizeof(char)); + char *str = (char *)(m_curptr + sizeof(char) + sizeof(size_t)); - m_curptr += sizeof(size_t); - char *str = (char *)m_curptr; - - if ((strlen(str) != real_len) || !(IsReadable(real_len+1))) + if ((strlen(str) != real_len) || !(IsReadable(sizeof(char) + sizeof(size_t) + real_len + 1))) { - return NULL; + return false; } if (len) @@ -219,8 +263,28 @@ const char *CDataPack::ReadString(size_t *len) const *len = real_len; } + return true; +} + +const char *CDataPack::ReadString(size_t *len) const +{ + size_t real_len; + if (!CanReadString(&real_len)) + { + return NULL; + } + + m_curptr += sizeof(char); + m_curptr += sizeof(size_t); + + char *str = (char *)m_curptr; m_curptr += real_len + 1; + if (len) + { + *len = real_len; + } + return str; } @@ -229,29 +293,49 @@ void *CDataPack::GetMemory() const return m_curptr; } -void *CDataPack::ReadMemory(size_t *size) const +bool CDataPack::CanReadMemory(size_t *size) const { - if (!IsReadable(sizeof(size_t))) + if (!IsReadable(sizeof(char) + sizeof(size_t))) { - return NULL; + return false; + } + if (*reinterpret_cast(m_curptr) != DataPackType::Raw) + { + return false; } - size_t bytecount = *(size_t *)m_curptr; - m_curptr += sizeof(size_t); - - if (!IsReadable(bytecount)) + size_t bytecount = *(size_t *)(m_curptr + sizeof(char)); + if (!IsReadable(sizeof(char) + sizeof(size_t) + bytecount)) { - return NULL; + return false; } - void *ptr = m_curptr; - if (size) { *size = bytecount; } + return true; +} + +void *CDataPack::ReadMemory(size_t *size) const +{ + size_t bytecount; + if (!CanReadMemory(&bytecount)) + { + return NULL; + } + + m_curptr += sizeof(char); + m_curptr += sizeof(size_t); + + void *ptr = m_curptr; m_curptr += bytecount; + if (size) + { + *size = bytecount; + } + return ptr; } diff --git a/amxmodx/CDataPack.h b/amxmodx/CDataPack.h index 618680d7..9c91cea2 100644 --- a/amxmodx/CDataPack.h +++ b/amxmodx/CDataPack.h @@ -8,7 +8,7 @@ * This program is free software; you can redistribute it and/or modify it under * the terms of the GNU General Public License, version 3.0, as published by the * Free Software Foundation. - * + * * This program is distributed in the hope that it will be useful, but WITHOUT * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more @@ -79,7 +79,7 @@ public: /** * @brief Returns whether or not a specified number of bytes from the current stream * position to the end can be read. - * + * * @param bytes Number of bytes to simulate reading. * @return True if can be read, false otherwise. */ @@ -108,6 +108,11 @@ public: */ void *ReadMemory(size_t *size) const; + bool CanReadCell() const; + bool CanReadFloat() const; + bool CanReadString(size_t *len) const; + bool CanReadMemory(size_t *size) const; + public: /** * @brief Resets the used size of the stream back to zero. @@ -160,6 +165,13 @@ private: mutable char *m_curptr; size_t m_capacity; size_t m_size; + + enum DataPackType { + Raw, + Cell, + Float, + String, + }; }; class CDataPackHandles diff --git a/amxmodx/datapacks.cpp b/amxmodx/datapacks.cpp index 82a82034..7da0c0ff 100644 --- a/amxmodx/datapacks.cpp +++ b/amxmodx/datapacks.cpp @@ -94,9 +94,9 @@ static cell AMX_NATIVE_CALL ReadPackCell(AMX* amx, cell* params) return 0; } - if (!d->IsReadable(sizeof(size_t) + sizeof(cell))) + if (!d->CanReadCell()) { - LogError(amx, AMX_ERR_NATIVE, "DataPack operation is out of bounds."); + LogError(amx, AMX_ERR_NATIVE, "Datapack operation is invalid."); return 0; } @@ -113,9 +113,9 @@ static cell AMX_NATIVE_CALL ReadPackFloat(AMX* amx, cell* params) return 0; } - if (!d->IsReadable(sizeof(size_t) + sizeof(float))) + if (!d->CanReadFloat()) { - LogError(amx, AMX_ERR_NATIVE, "DataPack operation is out of bounds."); + LogError(amx, AMX_ERR_NATIVE, "Datapack operation is invalid."); return 0; } @@ -134,14 +134,15 @@ static cell AMX_NATIVE_CALL ReadPackString(AMX* amx, cell* params) return 0; } - const char *str; - size_t len; - if (!(str = d->ReadString(&len))) + if (!d->CanReadString(NULL)) { - LogError(amx, AMX_ERR_NATIVE, "DataPack operation is out of bounds."); + LogError(amx, AMX_ERR_NATIVE, "Datapack operation is invalid."); return 0; } + size_t len; + const char *str = d->ReadString(&len); + return set_amxstring_utf8(amx, params[2], str, len, params[3] + 1); // + EOS } @@ -197,7 +198,7 @@ static cell AMX_NATIVE_CALL SetPackPosition(AMX* amx, cell* params) return 1; } -static cell AMX_NATIVE_CALL IsPackReadable(AMX* amx, cell* params) +static cell AMX_NATIVE_CALL IsPackEnded(AMX* amx, cell* params) { CDataPack *d = g_DataPackHandles.lookup(params[1]); @@ -207,7 +208,7 @@ static cell AMX_NATIVE_CALL IsPackReadable(AMX* amx, cell* params) return 0; } - return d->IsReadable(params[2]) ? 1 : 0; + return d->IsReadable(1) ? false : true; } static cell AMX_NATIVE_CALL DestroyDataPack(AMX* amx, cell* params) @@ -243,7 +244,7 @@ AMX_NATIVE_INFO g_DatapackNatives[] = { "ResetPack", ResetPack }, { "GetPackPosition", GetPackPosition }, { "SetPackPosition", SetPackPosition }, - { "IsPackReadable", IsPackReadable }, + { "IsPackEnded", IsPackEnded }, { "DestroyDataPack", DestroyDataPack }, {NULL, NULL} }; diff --git a/plugins/include/datapack.inc b/plugins/include/datapack.inc index fd2743b3..1b6894fd 100644 --- a/plugins/include/datapack.inc +++ b/plugins/include/datapack.inc @@ -141,6 +141,17 @@ native GetPackPosition(DataPack:pack); */ native SetPackPosition(DataPack:pack, position); +/** + * Returns if the datapack has reached its end and no more data can be read. + * + * @param pack Datapack handle + * + * @return True if datapack has reached the end, false otherwise + * @error If an invalid handle is provided, or the new position is + * out of datapack bounds, an error will be thrown. + */ + native bool:IsPackEnded(DataPack:pack); + /** * Destroys the datapack and frees its memory. * diff --git a/plugins/testsuite/datapack_test.sma b/plugins/testsuite/datapack_test.sma index bba11d63..c076e8ff 100644 --- a/plugins/testsuite/datapack_test.sma +++ b/plugins/testsuite/datapack_test.sma @@ -41,51 +41,57 @@ public datapacktest() { failcount = 0; passcount = 0; - + new DataPack:pack = CreateDataPack(); new DataPack:oldPack = pack; // Makes sure that the trie handle system recycles old handles - + new refCell = 23; new Float:refFloat = 42.42; new refString[] = "I'm a little teapot."; - + // Write - WritePackCell(pack, refCell); // 8 - WritePackString(pack, refString); // 25 (sizeof string + 4) - WritePackFloat(pack, refFloat); // 8 - - test("Position #1 test", .pass = GetPackPosition(pack) == 41); - test("Readable #1 test", .pass = !IsPackReadable(pack, 41)); - + new cellPos = GetPackPosition(pack); + WritePackCell(pack, refCell); + new floatPos = GetPackPosition(pack); + WritePackFloat(pack, refFloat); + new strPos = GetPackPosition(pack); + WritePackString(pack, refString); + new endPos = GetPackPosition(pack); + + test("Write position test", + .pass = (cellPos != floatPos && cellPos != strPos && cellPos != endPos + && floatPos != strPos && floatPos != endPos && strPos != endPos)); + //resets the index to the beginning, necessary for read. ResetPack(pack); - - test("Position #2 test", .pass = GetPackPosition(pack) == 0 ); - test("Readable #2 test", .pass = IsPackReadable(pack, 15)); - - // Read + + test("Position #1 test", .pass = (GetPackPosition(pack) == cellPos)); + test("Readable #1 test", .pass = !IsPackEnded(pack)); + new cellValue = ReadPackCell(pack); + test("Cell test", .pass = (cellValue == refCell)); + + test("Position #2 test", .pass = (GetPackPosition(pack) == floatPos)); + test("Readable #2 test", .pass = !IsPackEnded(pack)); + + new Float:floatValue = ReadPackFloat(pack); + test("Float test", .pass = (floatValue == refFloat)); + + test("Position #3 test", .pass = (GetPackPosition(pack) == strPos)); + test("Readable #3 test", .pass = !IsPackEnded(pack)); + new buffer[1024]; ReadPackString(pack, buffer, 1024); - new Float:floatvalue = ReadPackFloat(pack); - - test("Cell test", .pass = cellValue == refCell); - test("String test", .pass = bool:equal(buffer, refString)); - test("Float test #1", .pass = floatvalue == refFloat); - - SetPackPosition(pack, 33); - test("Set Position test", .pass = GetPackPosition(pack) == 33); - - WritePackFloat(pack, refFloat + 1); - SetPackPosition(pack, 33); - test("Float test #2", .pass = ReadPackFloat(pack) == refFloat + 1); + test("String test #1", .pass = bool:equal(buffer, refString)); + + test("End test", .pass = IsPackEnded(pack)); ResetPack(pack, .clear = true); - test("Clear test", .pass = !IsPackReadable(pack, 15)); - + test("Clear test", .pass = IsPackEnded(pack)); + DestroyDataPack(pack); - + test("Recycle handles", CreateDataPack() == oldPack); - + done(); }