Add partitionAllocGetSize() for determining actual size of allocation

Source/wtf/PartitionAlloc.h

 /*
  * Copyright (C) 2013 Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are
  * met:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 547 matching lines @@
 
     ptr = partitionCookieFreePointerAdjust(ptr);
     ASSERT(partitionPointerIsValid(ptr));
     PartitionPage* page = partitionPointerToPage(ptr);
     spinLockLock(&root->lock);
     partitionFreeWithPage(ptr, page);
     spinLockUnlock(&root->lock);
 #endif
 }
 
+ALWAYS_INLINE bool partitionAllocSupportsGetSize()
+{
+#if defined(MEMORY_TOOL_REPLACES_ALLOCATOR)
+    return false;
+#else
+    return true;
+#endif
+}
+
+ALWAYS_INLINE size_t partitionAllocGetSize(void* ptr)
+{
+    // No need to lock here. Only 'ptr' being freed by another thread could
+    // cause trouble, and the caller is responsible for that not happening.
+    ASSERT(partitionAllocSupportsGetSize());
+    ptr = partitionCookieFreePointerAdjust(ptr);
+    ASSERT(partitionPointerIsValid(ptr));
+    PartitionPage* page = partitionPointerToPage(ptr);
+    size_t size = page->bucket->slotSize;
+    return partitionCookieSizeAdjustSubtract(size);
+}
+
 // N (or more accurately, N - sizeof(void*)) represents the largest size in
 // bytes that will be handled by a SizeSpecificPartitionAllocator.
 // Attempts to partitionAlloc() more than this amount will fail.
 template <size_t N>
 class SizeSpecificPartitionAllocator {
 public:
     static const size_t kMaxAllocation = N - kAllocationGranularity;
     static const size_t kNumBuckets = N / kAllocationGranularity;
     void init() { partitionAllocInit(&m_partitionRoot, kNumBuckets, kMaxAllocation); }
     bool shutdown() { return partitionAllocShutdown(&m_partitionRoot); }
@@ skipping 17 matching lines @@
 using WTF::SizeSpecificPartitionAllocator;
 using WTF::PartitionAllocatorGeneric;
 using WTF::PartitionRoot;
 using WTF::partitionAllocInit;
 using WTF::partitionAllocShutdown;
 using WTF::partitionAlloc;
 using WTF::partitionFree;
 using WTF::partitionAllocGeneric;
 using WTF::partitionFreeGeneric;
 using WTF::partitionReallocGeneric;
+using WTF::partitionAllocSupportsGetSize;
+using WTF::partitionAllocGetSize;
 
 #endif // WTF_PartitionAlloc_h