Add native BS_GetNumberOfBitsAllocated

katursis · May 6, 2018 · 8c1deff · 8c1deff
1 parent 46740f9
commit 8c1deff
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/src/Natives.h b/src/Natives.h
@@ -338,6 +338,33 @@ namespace Natives {
         return 1;
     }
 
+    // native BS_GetNumberOfBitsAllocated(BitStream:bs, &number);
+    cell AMX_NATIVE_CALL n_BS_GetNumberOfBitsAllocated(AMX *amx, cell *params) {
+        if (!Utils::check_params(__FUNCTION__, 2, params)) {
+            return 0;
+        }
+
+        const auto bs = reinterpret_cast<RakNet::BitStream *>(params[1]);
+
+        if (!bs) {
+            Logger::instance()->Write("[%s] %s: invalid BitStream handle", Settings::kPluginName, __FUNCTION__);
+
+            return 0;
+        }
+
+        cell *cptr{};
+
+        if (amx_GetAddr(amx, params[2], &cptr) != AMX_ERR_NONE) {
+            Logger::instance()->Write("[%s] %s: invalid param reference", Settings::kPluginName, __FUNCTION__);
+
+            return 0;
+        }
+
+        *cptr = static_cast<cell>(bs->GetNumberOfBitsAllocated());
+
+        return 1;
+    }
+
     // native BS_WriteValue(BitStream:bs, {Float,_}:...);
     cell AMX_NATIVE_CALL n_BS_WriteValue(AMX *amx, cell *params) {
         if (params[0] < (sizeof(cell) * 3)) {
@@ -697,6 +724,7 @@ namespace Natives {
             { "BS_GetNumberOfBitsUsed", n_BS_GetNumberOfBitsUsed },
             { "BS_GetNumberOfBytesUsed", n_BS_GetNumberOfBytesUsed },
             { "BS_GetNumberOfUnreadBits", n_BS_GetNumberOfUnreadBits },
+            { "BS_GetNumberOfBitsAllocated", n_BS_GetNumberOfBitsAllocated },
 
             { "BS_WriteValue", n_BS_WriteValue },
             { "BS_ReadValue", n_BS_ReadValue },

diff --git a/src/Pawn.RakNet.inc b/src/Pawn.RakNet.inc
@@ -249,6 +249,7 @@
         native BS_GetNumberOfBitsUsed(BitStream:bs, &number);
         native BS_GetNumberOfBytesUsed(BitStream:bs, &number);
         native BS_GetNumberOfUnreadBits(BitStream:bs, &number);
+        native BS_GetNumberOfBitsAllocated(BitStream:bs, &number);
 
         native BS_WriteValue(BitStream:bs, {PR_ValueType, Float, _}:...);
         native BS_ReadValue(BitStream:bs, {PR_ValueType, Float, _}:...);