Added proper atArray implementation.

BigBaseV2/src/gta/array.hpp

@@ -1,65 +1,174 @@
 #pragma once
 #include <cstdint>
 #include "fwddec.hpp"
+#include "sysMemAllocator.hpp"
 
 namespace rage
 {
-	template <typename T>
+    template <typename T>
-	class atArray
+    class atArray
-	{
+    {
-	public:
+    public:
-		T *begin()
+        atArray()
-		{
+        {
-			return m_data;
+            m_data = nullptr;
-		}
+            m_count = 0;
+            m_size = 0;
+        }
 
-		T *end()
+        atArray(const atArray& right)
-		{
+        {
-			return m_data + m_size;
+            m_count = right.m_count;
-		}
+            m_size = right.m_size;
 
-		const T *begin() const
+            m_data = (T*)rage::GetAllocator()->allocate(m_size * sizeof(T), 16, 0);
-		{
+            std::uninitialized_copy(right.m_data, right.m_data + right.m_count, m_data);
-			return m_data;
+        }
-		}
 
-		const T *end() const
+        atArray(int capacity)
-		{
+        {
-			return m_data + m_size;
+            m_data = (T*)rage::GetAllocator()->allocate(capacity * sizeof(T), 16, 0);
-		}
+            m_count = 0;
+            m_size = capacity;
+        }
 
-		T *data()
+        void clear()
-		{
+        {
-			return m_data;
+            m_count = 0;
-		}
+            m_size = 0;
 
-		const T *data() const
+            if (m_data)
-		{
+            {
-			return m_data;
+                rage::GetAllocator()->free(m_data);
-		}
 
-		std::uint16_t size() const
+                m_data = nullptr;
-		{
+            }
-			return m_size;
+        }
-		}
 
-		std::uint16_t capacity() const
+        void append(const std::initializer_list<T> value_array)
-		{
+        {
-			return m_capacity;
+            auto value_array_size = value_array.size();
-		}
+            auto old_capacity = m_count;
 
-		T &operator[](std::uint16_t index)
+            if ((value_array_size + m_count) > std::numeric_limits<uint16_t>::max())
-		{
+                LOG(FATAL) << "RAGE atArray was given too large of a list to append";
-			return m_data[index];
-		}
 
-		const T &operator[](std::uint16_t index) const
+            auto size = (uint16_t)value_array_size;
-		{
+            expand(m_count + size);
-			return m_data[index];
+            m_size += size;
-		}
+
-	private:
+            auto i = old_capacity;
-		T *m_data;
+            for (const T* it = value_array.begin(); it != value_array.end(); ++it)
-		std::uint16_t m_size;
+            {
-		std::uint16_t m_capacity;
+                m_data[i] = *it;
-	};
+                i++;
+            }
+        }
+
+        void append(T value)
+        {
+            set(m_count, value);
+        }
+
+        void set(uint16_t offset, const T& value)
+        {
+            if (offset >= m_count)
+            {
+                expand(offset + 1);
+            }
+
+            if (offset >= m_size)
+            {
+                m_size = offset + 1;
+            }
+
+            m_data[offset] = value;
+        }
+
+        void expand(uint16_t newSize)
+        {
+            if (m_count >= newSize)
+            {
+                return;
+            }
+
+            T* newOffset = (T*)rage::GetAllocator()->allocate(newSize * sizeof(T), 16, 0);
+
+            // initialize the new entries
+            std::uninitialized_fill(newOffset, newOffset + newSize, T());
+
+            // copy the existing entries
+            if (m_data)
+            {
+                std::copy(m_data, m_data + m_size, newOffset);
+
+                rage::GetAllocator()->free(m_data);
+            }
+
+            m_data = newOffset;
+            m_count = newSize;
+        }
+
+        T* begin()
+        {
+            return m_data;
+        }
+
+        T* end()
+        {
+            return m_data + m_size;
+        }
+
+        const T* begin() const
+        {
+            return m_data;
+        }
+
+        const T* end() const
+        {
+            return m_data + m_size;
+        }
+
+        T* data()
+        {
+            return m_data;
+        }
+
+        const T* data() const
+        {
+            return m_data;
+        }
+
+        std::uint16_t size() const
+        {
+            return m_size;
+        }
+
+        std::uint16_t count() const
+        {
+            return m_count;
+        }
+
+        T& operator[](std::uint16_t index)
+        {
+            return m_data[index];
+        }
+
+        friend std::ostream& operator<<(std::ostream& o, const atArray<T>& j)
+        {
+            o << "Array Size: " << j.size() << std::endl;
+            for (T item : j)
+                o << "\tArray Item: " << item << " [" << HEX_TO_UPPER(item) << "]" << (j.end() == item) ? "" : std::endl;
+            return o;
+        }
+
+        const T& operator[](std::uint16_t index) const
+        {
+            return m_data[index];
+        }
+    private:
+        T* m_data;
+        std::uint16_t m_size;
+        std::uint16_t m_count;
+    };
 }

BigBaseV2/src/gta/sysMemAllocator.hpp

@@ -0,0 +1,72 @@
+#pragma once
+#include "fwddec.hpp"
+
+namespace rage
+{
+	class sysMemAllocator
+	{
+	public:
+		virtual ~sysMemAllocator() = 0;
+
+		virtual void SetQuitOnFail(bool) = 0;
+		virtual void* Allocate(size_t size, size_t align, int subAllocator) = 0;
+
+		inline void* allocate(size_t size, size_t align, int subAllocator)
+		{
+			return Allocate(size, align, subAllocator);
+		}
+
+		virtual void* TryAllocate(size_t size, size_t align, int subAllocator) = 0;
+
+		virtual void Free(void* pointer) = 0;
+
+		virtual void free(void* pointer)
+		{
+			return Free(pointer);
+		}
+
+		virtual void TryFree(void* pointer) = 0;
+
+		virtual void Resize(void* pointer, size_t size) = 0;
+
+		virtual sysMemAllocator* GetAllocator(int allocator) const = 0;
+
+		virtual sysMemAllocator* GetAllocator(int allocator) = 0;
+
+		virtual sysMemAllocator* GetPointerOwner(void* pointer) = 0;
+
+		virtual size_t GetSize(void* pointer) const = 0;
+
+		virtual size_t GetMemoryUsed(int memoryBucket) = 0;
+
+		virtual size_t GetMemoryAvailable() = 0;
+
+	public:
+
+		static sysMemAllocator* sysMemAllocator::UpdateAllocatorValue()
+		{
+			//B9 ? ? ? ? 48 8B 0C 01 45 33 C9 49 8B D2 48
+			auto g_gtaTlsEntry = *(sysMemAllocator**)(*(uintptr_t*)(__readgsqword(88)) + 0xC8); //This has been 0xC8 since 323, I'm not adding this signature to pointers...
+
+			if (g_gtaTlsEntry == nullptr)
+				LOG(FATAL) << "Failed to find tlsEntry within GTA5.exe via __readgsqword";
+
+			*(sysMemAllocator**)(*(uintptr_t*)(__readgsqword(88)) + 0xC8) = g_gtaTlsEntry;
+			*(sysMemAllocator**)(*(uintptr_t*)(__readgsqword(88)) + 0xC8 - 8) = g_gtaTlsEntry;
+
+			return g_gtaTlsEntry;
+		}
+	};
+
+	inline sysMemAllocator* GetAllocator()
+	{
+		sysMemAllocator* allocator = *(sysMemAllocator**)(*(uintptr_t*)(__readgsqword(88)) + 0xC8);
+
+		if (!allocator)
+		{
+			return sysMemAllocator::UpdateAllocatorValue();
+		}
+
+		return allocator;
+	}
+}