/*----------------------------------------------------------------------------*- =================================== Y Sever Includes - Malloc Functions =================================== Description: Functions for using malloc/calloc/free type functions in PAWN. Legal: Version: MPL 1.1 The contents of this file are subject to the Mozilla Public License Version 1.1 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.mozilla.org/MPL/ Software distributed under the License is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License for the specific language governing rights and limitations under the License. The Original Code is the YSI malloc include. The Initial Developer of the Original Code is Alex "Y_Less" Cole. Portions created by the Initial Developer are Copyright (C) 2011 the Initial Developer. All Rights Reserved. Contributors: ZeeX, koolk, JoeBullet/Google63, g_aSlice/Slice Thanks: JoeBullet/Google63 - Handy arbitrary ASM jump code using SCTRL. ZeeX - Very productive conversations. koolk - IsPlayerinAreaEx code. TheAlpha - Danish translation. breadfish - German translation. Fireburn - Dutch translation. yom - French translation. 50p - Polish translation. Zamaroht - Spanish translation. Dracoblue, sintax, mabako, Xtreme, other coders - Producing other modes for me to strive to better. Pixels^ - Running XScripters where the idea was born. Matite - Pestering me to release it and using it. Very special thanks to: Thiadmer - PAWN, whose limits continue to amaze me! Kye/Kalcor - SA:MP. SA:MP Team past, present and future - SA:MP. Version: 0.1 Changelog: 02/12/11: Added variable argument functions. 22/12/08: First version. Functions: Public - Core: - Stock: malloc - Allocate a block of memory (may be inline). calloc - Allocate a block of memory and blank. free - Free an allocated block of memory (may be inline). Malloc_Set - Set a value in an allocated array (may be inline). Malloc_Get - Get a value in an allocated array (may be inline). Malloc_SetS - Set a string in an allocated array. Malloc_GetS - Get a string in an allocated array. Malloc_Allocate - Do the memory allocation (may be static). Malloc_Free - Do the memory freeing (may be static). Malloc_SlotSize - Get the size of an allocated block (may be inline). Malloc_NewS - Allocate for and store a given string. Static: Malloc_Allocate - Do the memory allocation (may be stock). Malloc_Free - Do the memory freeing (may be stock). Inline: mget - Get data from an allocation unit. mset - Set data in an allocation unit. mgets - Get a string from an allocation unit. msets - Set a string in an allocation unit. malloc - Allocate a block of memory (may be stock). free - Free an allocated block of memory (may be stock). Malloc_Set - Set a value in an allocated array (may be stock). Malloc_Get - Get a value in an allocated array (may be stock). Malloc_NextSlot - Get the next free data block. Malloc_GetSlotSize - Get the size of a slot. Malloc_SetSlotSize - Set the size of a block. Malloc_GetData - Direct data access getter. Malloc_SetData - Direct data access setter. Malloc_SlotSize - Get the size of an allocated block (may be stock). API: - Callbacks: - Definitions: MALLOC_KB_TO_CELL - Multiplication value to convert kb to cells. NO_ALLOC - A failed allocation (NULL, but YSI already has NULL). Enums: - Macros: - Tags: Alloc - An allocated block handle variable. Variables: Global: YSI_gMallocMemory - Stores the data (may be static). Static: YSI_gMallocMemory - Stores the data (may be global). YSI_g_sUnusedStart - Start of free memory. Commands: - Compile options: MALLOC_MEMORY - Number of cells to reserve. MALLOC_MEMORY_KB - Number of killobytes to reserve. MALLOC_MEMORY_B - Number of bytes to reserve. MALLOC_MEMORY_MB - Number of megabytes to reserve. YSI_MALLOC_SECURE - Use enhanced bounds checking. YSI_MALLOC_NO_SHORT - Avoid conflicts with mget/mset. Operators: - -*----------------------------------------------------------------------------*/ #include "internal\y_version" #include "y_debug" #include "internal\y_funcinc" #define MALLOC_KB_TO_CELL ((1024 * 8) / cellbits) #define NO_ALLOC (Alloc:0) #if !defined MALLOC_MEMORY #if defined MALLOC_MEMORY_KB #define MALLOC_MEMORY ((MALLOC_MEMORY_KB) * MALLOC_KB_TO_CELL) #else #if defined MALLOC_MEMORY_MB #define MALLOC_MEMORY ((MALLOC_MEMORY_MB) * 1024 * MALLOC_KB_TO_CELL) #else #if defined MALLOC_MEMORY_B #define MALLOC_MEMORY (((MALLOC_MEMORY_B) * 8) / cellbits) #else #define MALLOC_MEMORY (1 * 1024 * MALLOC_KB_TO_CELL) #endif #endif #endif #endif static YSI_g_sUnusedStart = 1; #if defined YSI_MALLOC_SECURE static #else new #endif YSI_gMallocMemory[MALLOC_MEMORY] = {MALLOC_MEMORY - 1, 0}; forward Alloc:Malloc_Allocate(size); forward Alloc:calloc(size); #if defined YSI_MALLOC_SECURE forward Alloc:malloc(size); #endif /*----------------------------------------------------------------------------*- Function: Malloc_GetSlotSize Params: slot - Allocation unit to get the size of. Return: The size. Notes: - -*----------------------------------------------------------------------------*/ #define Malloc_GetSlotSize(%1) \ (YSI_gMallocMemory[_:(%1) - 1]) /*----------------------------------------------------------------------------*- Function: Malloc_SlotSize Params: slot - Allocation unit to get the size of. Return: The size. Notes: - -*----------------------------------------------------------------------------*/ #if defined YSI_MALLOC_SECURE stock Malloc_SlotSize(slot) { return Malloc_GetSlotSize(slot); } #else #define Malloc_SlotSize(%1) \ Malloc_GetSlotSize(%1) #endif /*----------------------------------------------------------------------------*- Function: Malloc_NewS Params: string[] - The string to store. pack - Whether or not the string will be packed. Return: 0 on fail or a data handle on sucess. Notes: - -*----------------------------------------------------------------------------*/ stock Alloc:Malloc_NewS(const string[], bool:pack = false) { new size = pack ? ((strlen(string) + 1) char) : (strlen(string) + 1), Alloc:alloc = Malloc_Allocate(size) ; if (alloc) { if (pack) strpack(YSI_gMallocMemory[_:alloc], string, size); else strunpack(YSI_gMallocMemory[_:alloc], string, size); } return alloc; } /*----------------------------------------------------------------------------*- Function: Malloc_SetSlotSize Params: slot - The allocation unit to set the size of. size - The size to set it to. Return: - Notes: - -*----------------------------------------------------------------------------*/ #define Malloc_SetSlotSize(%1,%2) \ YSI_gMallocMemory[_:(%1) - 1] = (%2) /*----------------------------------------------------------------------------*- Function: Malloc_GetData Params: slot - The allocation unit to get data from. index - The location in the unit to get. Return: The data Notes: Basically like Malloc_Get but used internally. -*----------------------------------------------------------------------------*/ #define Malloc_GetData(%1,%2) \ (YSI_gMallocMemory[_:((%1)+Alloc:(%2))]) /*----------------------------------------------------------------------------*- Function: Malloc_SetData Params: slot - The allocation unit to set in. index - Where in the unit to set. value - The value to save. Return: - Notes: - -*----------------------------------------------------------------------------*/ #define Malloc_SetData(%1,%2,%3) \ YSI_gMallocMemory[_:((%1)+Alloc:(%2))]=(%3) /*----------------------------------------------------------------------------*- Function: mget Params: slot - The allocation unit to get data from. index - The location in the unit to get. Return: The data Notes: Shorthand for Malloc_Get. -*----------------------------------------------------------------------------*/ #if !defined YSI_MALLOC_NO_SHORT #define mget(%1) \ Malloc_Get(%1) #endif /*----------------------------------------------------------------------------*- Function: mset Params: slot - The allocation unit to set in. index - Where in the unit to set. value - The value to save. Return: - Notes: Shorthand for Malloc_Set. -*----------------------------------------------------------------------------*/ #if !defined YSI_MALLOC_NO_SHORT #define mset(%1) \ Malloc_Set(%1) #endif /*----------------------------------------------------------------------------*- Function: mgets Params: target[] - Target for the string. len - Length of the target. array - Data unit to put information in. index - Index in the unit. Return: The data Notes: Shorthand for Malloc_GetS. -*----------------------------------------------------------------------------*/ #if !defined YSI_MALLOC_NO_SHORT #define mgets(%1) \ Malloc_GetS(%1) #define mgeta(%1) \ Malloc_GetA(%1) #endif /*----------------------------------------------------------------------------*- Function: msets Params: array - Data unit to put information in. index - Index in the unit. str[] - String to insert Return: - Notes: Shorthand for Malloc_SetS. -*----------------------------------------------------------------------------*/ #if !defined YSI_MALLOC_NO_SHORT #define msets(%1) \ Malloc_SetS(%1) #define mseta(%1) \ Malloc_SetA(%1) #endif /*----------------------------------------------------------------------------*- Function: Malloc_NextSlot Params: slot - The unit to get the one after of. Return: - Notes: Gets the next free block of memory after the current one. -*----------------------------------------------------------------------------*/ #define Malloc_NextSlot(%1) \ (YSI_gMallocMemory[(%1)]) /*----------------------------------------------------------------------------*- Function: Malloc_Get Params: array - Data unit to get information from. index - Index in the unit. Return: Data. Notes: Displays errors in secure mode. -*----------------------------------------------------------------------------*/ #if defined YSI_MALLOC_SECURE stock Malloc_Get(Alloc:array, index) { if (index >= 0 && index < Malloc_GetSlotSize(_:array)) { return Malloc_GetData(_:array, index); } P:C(else P:E("Array read index out of bounds: %d[%d]", _:array, index);); return 0; } #else #define Malloc_Get(%1,%2) \ Malloc_GetData(%1, %2) #endif /*----------------------------------------------------------------------------*- Function: Malloc_Set Params: array - Data unit to put information in. index - Index in the unit. value - Value to insert Return: - Notes: Displays errors in secure mode. -*----------------------------------------------------------------------------*/ #if defined YSI_MALLOC_SECURE stock Malloc_Set(Alloc:array, index, value) { if (index >= 0 && index < Malloc_GetSlotSize(_:array)) { return Malloc_SetData(_:array, index, value); } P:C(else P:E("Array write index out of bounds: %d[%d]", _:array, index);); return 0; } #else #define Malloc_Set(%1,%2,%3) \ Malloc_SetData(%1, %2, %3) #endif /*----------------------------------------------------------------------------*- Function: Malloc_GetS Params: target[] - Target for the string. len - Length of the target. array - Data unit to put information in. index - Index in the unit. Return: - Notes: Displays errors in secure mode. Gets a string. -*----------------------------------------------------------------------------*/ stock Malloc_GetS(target[], length, Alloc:array, index) { P:3("Malloc_GetS: array = %d, index = %d.", _:array, index); if (index >= 0) { new size = Malloc_GetSlotSize(_:array); P:5("Malloc_GetS: size = %d.", size); new i = 0; index += _:array; size += _:array; while (i < length && index < size) { P:6("Malloc_GetS: i = %d.", i); if (!(target[i++] = YSI_gMallocMemory[index++])) { return 1; } } #if defined YSI_MALLOC_SECURE P:C(if(index == size)P:E("Out of data (%d, %d)", _:array, index);); P:C(if(i == length)P:E("Out of buffer space");); #endif } return 0; } /*----------------------------------------------------------------------------*- Function: Malloc_SetS Params: array - Data unit to put information in. index - Index in the unit. str[] - String to insert Return: - Notes: Displays errors in secure mode. Inserts a string. -*----------------------------------------------------------------------------*/ stock Malloc_SetS(Alloc:array, index, const str[]) { P:3("Malloc_SetS called: %d, %d, %s", _:array, index, str); if (index >= 0 && index + strlen(str) + 1 <= Malloc_GetSlotSize(_:array)) { P:5("Malloc_SetS: In check."); index += _:array; YSI_gMallocMemory[index] = 0; strcat(YSI_gMallocMemory[index], str, cellmax); //while ((YSI_gMallocMemory[index++] = str[i++])) {} } #if defined YSI_MALLOC_SECURE P:C(else P:E("String copy failed (%s)", str);); #endif } /*----------------------------------------------------------------------------*- Function: Malloc_SetVAS Params: array - Data unit to put information in. index - Index in the unit. arg - Offset in the stack of the string to store. Return: - Notes: Inserts a string by stack offset for use in vararg functions. -*----------------------------------------------------------------------------*/ stock Malloc_SetVAS(Alloc:array, index, arg) { P:3("Malloc_SetVAS called: %d, %d, %d", _:array, index, arg); new len; #emit LOAD.S.pri arg #emit SMUL.C 4 #emit LOAD.S.alt 0 #emit ADD #emit ADD.C 12 #emit LOAD.I // For later reuse. #emit STOR.S.pri arg #emit PUSH.pri #emit PUSH.C 4 #emit SYSREQ.C strlen #emit STACK 8 #emit STOR.S.pri len if (index >= 0 && index + (++len) <= Malloc_GetSlotSize(_:array)) { P:5("Malloc_SetVAS: In check."); index += _:array; // Blank and copy the string. //#emit LOAD.S.pri arg //#emit SMUL.C 4 //#emit LOAD.S.alt 0 //#emit ADD //#emit ADD.C 12 //#emit LOAD.I // Skip the code above the second time now (store the true source // address in "arg" the first time). #emit PUSH.S len #emit PUSH.S arg #emit CONST.alt YSI_gMallocMemory #emit LOAD.S.pri index #emit IDXADDR #emit MOVE.alt #emit CONST.pri 0 #emit STOR.I #emit PUSH.alt #emit PUSH.C 12 #emit SYSREQ.C strcat #emit STACK 16 //strcat(YSI_gMallocMemory[index], str, cellmax); } #if defined YSI_MALLOC_SECURE P:C(else P:E("String copy failed (%s)", str);); #endif } /*----------------------------------------------------------------------------*- Function: Malloc_GetA Params: target[] - Target for the array. len - Length of the target. array - Data unit to put information in. index - Index in the unit. Return: - Notes: Displays errors in secure mode. Gets an array. -*----------------------------------------------------------------------------*/ stock Malloc_GetA(target[], length, Alloc:array, index) { P:3("Malloc_GetA called: array = %d, index = %d.", _:array, index); if (index >= 0) { new size = Malloc_GetSlotSize(_:array); P:5("Malloc_GetA: size = %d.", size); memcpy(target, YSI_gMallocMemory, index + _:array, size * 4, length); #if defined YSI_MALLOC_SECURE P:C(if(length > size)P:E("Out of buffer space.");); #endif return 1; } return 0; } /*----------------------------------------------------------------------------*- Function: Malloc_SetA Params: array - Data unit to put information in. index - Index in the unit. str[] - Array to insert. len - Length of the array. Return: - Notes: Displays errors in secure mode. Inserts an array. -*----------------------------------------------------------------------------*/ stock Malloc_SetA(Alloc:array, index, const str[], len) { P:3("Malloc_SetA: array = %d, index = %d.", _:array, index); if (index >= 0) { new size = Malloc_GetSlotSize(_:array); P:5("Malloc_SetA: size = %d.", size); memcpy(YSI_gMallocMemory[index + _:array], str, 0, len * 4, size - index); #if defined YSI_MALLOC_SECURE P:C(if(len > size - index)P:E("Out of buffer space.");); #endif } } /*----------------------------------------------------------------------------*- Function: Malloc_SetVAA Params: array - Data unit to put information in. index - Index in the unit. arg - Offset in the stack of the array to store. Return: - Notes: Inserts an array by stack offset for use in vararg functions. -*----------------------------------------------------------------------------*/ stock Malloc_SetVAA(Alloc:array, index, arg) { P:3("Malloc_SetVAA called: %d, %d, %d", _:array, index, arg); new len; #emit LOAD.S.pri arg #emit ADD.C 1 #emit SMUL.C 4 #emit LOAD.S.alt 0 #emit ADD #emit ADD.C 12 #emit LOAD.I #emit LOAD.I // For later reuse. #emit STOR.S.pri len printf("%d", len); if (index >= 0 && index + len <= Malloc_GetSlotSize(_:array)) { P:5("Malloc_SetVAA In check."); index += _:array; // Skip the code above the second time now (store the true source // address in "arg" the first time). #emit LOAD.S.pri len #emit PUSH.pri #emit SMUL.C 4 #emit PUSH.pri #emit PUSH.C 0 // Source. #emit LOAD.S.pri arg #emit SMUL.C 4 #emit LOAD.S.alt 0 #emit ADD #emit ADD.C 12 #emit LOAD.I #emit PUSH.pri // Destination. #emit CONST.alt YSI_gMallocMemory #emit LOAD.S.pri index #emit IDXADDR #emit MOVE.alt #emit CONST.pri 0 #emit STOR.I #emit PUSH.alt // GO! #emit PUSH.C 20 #emit SYSREQ.C memcpy #emit STACK 24 //strcat(YSI_gMallocMemory[index], str, cellmax); } #if defined YSI_MALLOC_SECURE P:C(else P:E("Array copy out of memory.");); #endif } /*----------------------------------------------------------------------------*- Function: malloc Params: size - Size of memory to allocate. Return: 0 on fail or a data handle on sucess. Notes: Displays errors in secure mode. -*----------------------------------------------------------------------------*/ #if defined YSI_MALLOC_SECURE stock Alloc:malloc(size) { new Alloc:slot = Malloc_Allocate(size); P:C(if(!slot)P:E("Allocation failed (%d)", size);); return slot; } #else #define malloc(%1) \ Malloc_Allocate(%1) #endif /*----------------------------------------------------------------------------*- Function: calloc Params: size - Size of memory to allocate. Return: 0 on fail or a data handle on sucess. Notes: Displays errors in secure mode. Blanks allocated mmeory. -*----------------------------------------------------------------------------*/ stock Alloc:calloc(size) { new Alloc:slot = Malloc_Allocate(size); if (slot) { new temp = _:slot; while (size--) { YSI_gMallocMemory[temp++] = 0; } } #if defined YSI_MALLOC_SECURE P:C(if(!slot)P:E("Allocation failed (%d)", size);); #endif return slot; } /*----------------------------------------------------------------------------*- Function: free Params: slot - Slot of memory to free up. Return: - Notes: Displays errors in secure mode. -*----------------------------------------------------------------------------*/ #if defined YSI_MALLOC_SECURE stock free(Alloc:slot) { if (!slot || !Malloc_GetSlotSize(_:slot)) { P:1("Free failed (%d)", _:slot); return 0; } return Malloc_Free(slot); } #else #define free(%1) \ Malloc_Free(%1) #endif /*----------------------------------------------------------------------------*- Function: Malloc_Allocate Params: size - Ammount of memory to allocate. Return: Memory identifier. Notes: The size check should never fail, if there's only 1 cell extra somewhere just sneak it onto the end of an array, if the user does proper bounds checking it shouldn't matter. Implementation code for malloc(). This code will find an area in memory with sufficient space to store the given data and -*----------------------------------------------------------------------------*/ #if defined YSI_MALLOC_SECURE static #endif stock Alloc:Malloc_Allocate(size) { new slot = YSI_g_sUnusedStart, p = 0, cs; while (slot) { cs = Malloc_GetSlotSize(slot); if (!cs) { return NO_ALLOC; } if (cs >= size) { break; } p = slot; slot = Malloc_NextSlot(slot); } if (slot) { if (cs == size + 1) { size++; } if (cs == size) { if (p) { Malloc_SetData(Alloc:p, 0, Malloc_NextSlot(slot)); } else { YSI_g_sUnusedStart = Malloc_NextSlot(slot); } } else { Malloc_SetSlotSize(slot, size); size++; cs -= size; size += slot; if (p) { Malloc_SetData(Alloc:p, 0, size); } else { YSI_g_sUnusedStart = size; } Malloc_SetData(Alloc:size, 0, Malloc_NextSlot(slot)); Malloc_SetSlotSize(size, cs); } return Alloc:slot; } return NO_ALLOC; } /*----------------------------------------------------------------------------*- Function: Malloc_Free Params: slot - Memory allocation unit to release Return: - Notes: Implementation code for free(). -*----------------------------------------------------------------------------*/ #if defined YSI_MALLOC_SECURE static #endif stock Malloc_Free(Alloc:slot) { P:4("Malloc_Free called: %d", _:slot); new size = Malloc_GetSlotSize(_:slot), p = YSI_g_sUnusedStart, l = 0; P:5("Malloc_Free: size = %d", size); if (p) { while (p && p < _:slot) { l = p; p = Malloc_NextSlot(p); } if (p) { if (l) { new tmp = Malloc_GetSlotSize(l); if (l + tmp + 1 == _:slot) { size += tmp + 1; Malloc_SetSlotSize(l, size); slot = Alloc:l; } else { Malloc_SetData(slot, 0, p); Malloc_SetData(Alloc:l, 0, _:slot); } } else { YSI_g_sUnusedStart = _:slot; } if (_:slot + size + 1 == p) { Malloc_SetSlotSize(_:slot, Malloc_GetSlotSize(p) + size + 1); Malloc_SetData(slot, 0, Malloc_NextSlot(p)); } else { Malloc_SetData(slot, 0, p); } } else { new tmp = Malloc_GetSlotSize(l); if (l + tmp + 1 == _:slot) { Malloc_SetSlotSize(l, size + tmp + 1); } else { Malloc_SetData(slot, 0, 0); Malloc_SetData(Alloc:l, 0, _:slot); } } } else { YSI_g_sUnusedStart = _:slot; Malloc_SetData(slot, 0, 0); } return 1; }