1e462795ff5d4c7359f9e8637c10544bb2de70107tturney// Tencent is pleased to support the open source community by making RapidJSON available. 2e462795ff5d4c7359f9e8637c10544bb2de70107tturney// 3e462795ff5d4c7359f9e8637c10544bb2de70107tturney// Copyright (C) 2015 THL A29 Limited, a Tencent company, and Milo Yip. All rights reserved. 4e462795ff5d4c7359f9e8637c10544bb2de70107tturney// 5e462795ff5d4c7359f9e8637c10544bb2de70107tturney// Licensed under the MIT License (the "License"); you may not use this file except 6e462795ff5d4c7359f9e8637c10544bb2de70107tturney// in compliance with the License. You may obtain a copy of the License at 7e462795ff5d4c7359f9e8637c10544bb2de70107tturney// 8e462795ff5d4c7359f9e8637c10544bb2de70107tturney// http://opensource.org/licenses/MIT 9e462795ff5d4c7359f9e8637c10544bb2de70107tturney// 10e462795ff5d4c7359f9e8637c10544bb2de70107tturney// Unless required by applicable law or agreed to in writing, software distributed 11e462795ff5d4c7359f9e8637c10544bb2de70107tturney// under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR 12e462795ff5d4c7359f9e8637c10544bb2de70107tturney// CONDITIONS OF ANY KIND, either express or implied. See the License for the 13e462795ff5d4c7359f9e8637c10544bb2de70107tturney// specific language governing permissions and limitations under the License. 14e462795ff5d4c7359f9e8637c10544bb2de70107tturney 15e462795ff5d4c7359f9e8637c10544bb2de70107tturney#include "unittest.h" 16e462795ff5d4c7359f9e8637c10544bb2de70107tturney 17e462795ff5d4c7359f9e8637c10544bb2de70107tturney#include "rapidjson/allocators.h" 18e462795ff5d4c7359f9e8637c10544bb2de70107tturney 19e462795ff5d4c7359f9e8637c10544bb2de70107tturneyusing namespace rapidjson; 20e462795ff5d4c7359f9e8637c10544bb2de70107tturney 21e462795ff5d4c7359f9e8637c10544bb2de70107tturneytemplate <typename Allocator> 22e462795ff5d4c7359f9e8637c10544bb2de70107tturneyvoid TestAllocator(Allocator& a) { 23e462795ff5d4c7359f9e8637c10544bb2de70107tturney EXPECT_TRUE(a.Malloc(0) == 0); 24e462795ff5d4c7359f9e8637c10544bb2de70107tturney 25e462795ff5d4c7359f9e8637c10544bb2de70107tturney uint8_t* p = (uint8_t*)a.Malloc(100); 26e462795ff5d4c7359f9e8637c10544bb2de70107tturney EXPECT_TRUE(p != 0); 27e462795ff5d4c7359f9e8637c10544bb2de70107tturney for (size_t i = 0; i < 100; i++) 28e462795ff5d4c7359f9e8637c10544bb2de70107tturney p[i] = (uint8_t)i; 29e462795ff5d4c7359f9e8637c10544bb2de70107tturney 30e462795ff5d4c7359f9e8637c10544bb2de70107tturney // Expand 31e462795ff5d4c7359f9e8637c10544bb2de70107tturney uint8_t* q = (uint8_t*)a.Realloc(p, 100, 200); 32e462795ff5d4c7359f9e8637c10544bb2de70107tturney EXPECT_TRUE(q != 0); 33e462795ff5d4c7359f9e8637c10544bb2de70107tturney for (size_t i = 0; i < 100; i++) 34e462795ff5d4c7359f9e8637c10544bb2de70107tturney EXPECT_EQ(i, q[i]); 35e462795ff5d4c7359f9e8637c10544bb2de70107tturney for (size_t i = 100; i < 200; i++) 36e462795ff5d4c7359f9e8637c10544bb2de70107tturney q[i] = (uint8_t)i; 37e462795ff5d4c7359f9e8637c10544bb2de70107tturney 38e462795ff5d4c7359f9e8637c10544bb2de70107tturney // Shrink 39e462795ff5d4c7359f9e8637c10544bb2de70107tturney uint8_t *r = (uint8_t*)a.Realloc(q, 200, 150); 40e462795ff5d4c7359f9e8637c10544bb2de70107tturney EXPECT_TRUE(r != 0); 41e462795ff5d4c7359f9e8637c10544bb2de70107tturney for (size_t i = 0; i < 150; i++) 42e462795ff5d4c7359f9e8637c10544bb2de70107tturney EXPECT_EQ(i, r[i]); 43e462795ff5d4c7359f9e8637c10544bb2de70107tturney 44e462795ff5d4c7359f9e8637c10544bb2de70107tturney Allocator::Free(r); 45e462795ff5d4c7359f9e8637c10544bb2de70107tturney 46e462795ff5d4c7359f9e8637c10544bb2de70107tturney // Realloc to zero size 47e462795ff5d4c7359f9e8637c10544bb2de70107tturney EXPECT_TRUE(a.Realloc(a.Malloc(1), 1, 0) == 0); 48e462795ff5d4c7359f9e8637c10544bb2de70107tturney} 49e462795ff5d4c7359f9e8637c10544bb2de70107tturney 50e462795ff5d4c7359f9e8637c10544bb2de70107tturneyTEST(Allocator, CrtAllocator) { 51e462795ff5d4c7359f9e8637c10544bb2de70107tturney CrtAllocator a; 52e462795ff5d4c7359f9e8637c10544bb2de70107tturney TestAllocator(a); 53e462795ff5d4c7359f9e8637c10544bb2de70107tturney} 54e462795ff5d4c7359f9e8637c10544bb2de70107tturney 55e462795ff5d4c7359f9e8637c10544bb2de70107tturneyTEST(Allocator, MemoryPoolAllocator) { 56e462795ff5d4c7359f9e8637c10544bb2de70107tturney MemoryPoolAllocator<> a; 57e462795ff5d4c7359f9e8637c10544bb2de70107tturney TestAllocator(a); 58e462795ff5d4c7359f9e8637c10544bb2de70107tturney 59e462795ff5d4c7359f9e8637c10544bb2de70107tturney for (int i = 1; i < 1000; i++) { 60e462795ff5d4c7359f9e8637c10544bb2de70107tturney EXPECT_TRUE(a.Malloc(i) != 0); 61e462795ff5d4c7359f9e8637c10544bb2de70107tturney EXPECT_LE(a.Size(), a.Capacity()); 62e462795ff5d4c7359f9e8637c10544bb2de70107tturney } 63e462795ff5d4c7359f9e8637c10544bb2de70107tturney} 64e462795ff5d4c7359f9e8637c10544bb2de70107tturney 65e462795ff5d4c7359f9e8637c10544bb2de70107tturneyTEST(Allocator, Alignment) { 66e462795ff5d4c7359f9e8637c10544bb2de70107tturney#if RAPIDJSON_64BIT == 1 67e462795ff5d4c7359f9e8637c10544bb2de70107tturney EXPECT_EQ(RAPIDJSON_UINT64_C2(0x00000000, 0x00000000), RAPIDJSON_ALIGN(0)); 68e462795ff5d4c7359f9e8637c10544bb2de70107tturney for (uint64_t i = 1; i < 8; i++) { 69e462795ff5d4c7359f9e8637c10544bb2de70107tturney EXPECT_EQ(RAPIDJSON_UINT64_C2(0x00000000, 0x00000008), RAPIDJSON_ALIGN(i)); 70e462795ff5d4c7359f9e8637c10544bb2de70107tturney EXPECT_EQ(RAPIDJSON_UINT64_C2(0x00000000, 0x00000010), RAPIDJSON_ALIGN(RAPIDJSON_UINT64_C2(0x00000000, 0x00000008) + i)); 71e462795ff5d4c7359f9e8637c10544bb2de70107tturney EXPECT_EQ(RAPIDJSON_UINT64_C2(0x00000001, 0x00000000), RAPIDJSON_ALIGN(RAPIDJSON_UINT64_C2(0x00000000, 0xFFFFFFF8) + i)); 72e462795ff5d4c7359f9e8637c10544bb2de70107tturney EXPECT_EQ(RAPIDJSON_UINT64_C2(0xFFFFFFFF, 0xFFFFFFF8), RAPIDJSON_ALIGN(RAPIDJSON_UINT64_C2(0xFFFFFFFF, 0xFFFFFFF0) + i)); 73e462795ff5d4c7359f9e8637c10544bb2de70107tturney } 74e462795ff5d4c7359f9e8637c10544bb2de70107tturney#else 75e462795ff5d4c7359f9e8637c10544bb2de70107tturney EXPECT_EQ(0u, RAPIDJSON_ALIGN(0u)); 76e462795ff5d4c7359f9e8637c10544bb2de70107tturney for (uint32_t i = 1; i < 4; i++) { 77e462795ff5d4c7359f9e8637c10544bb2de70107tturney EXPECT_EQ(4u, RAPIDJSON_ALIGN(i)); 78e462795ff5d4c7359f9e8637c10544bb2de70107tturney EXPECT_EQ(8u, RAPIDJSON_ALIGN(4u + i)); 79e462795ff5d4c7359f9e8637c10544bb2de70107tturney EXPECT_EQ(0xFFFFFFF8u, RAPIDJSON_ALIGN(0xFFFFFFF4u + i)); 80e462795ff5d4c7359f9e8637c10544bb2de70107tturney EXPECT_EQ(0xFFFFFFFCu, RAPIDJSON_ALIGN(0xFFFFFFF8u + i)); 81e462795ff5d4c7359f9e8637c10544bb2de70107tturney } 82e462795ff5d4c7359f9e8637c10544bb2de70107tturney#endif 83e462795ff5d4c7359f9e8637c10544bb2de70107tturney} 84