Track JS allocations as they arrive with no affection on performance when tracking is switched off

test/cctest/cctest.h

 // Copyright 2008 the V8 project authors. 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
 //       copyright notice, this list of conditions and the following
 //       disclaimer in the documentation and/or other materials provided
@@ skipping 53 matching lines @@
 #undef DEFINE_EXTENSION_ID
 
 typedef v8::internal::EnumSet<CcTestExtensionIds> CcTestExtensionFlags;
 #define DEFINE_EXTENSION_FLAG(Name, Ident)                               \
   static const CcTestExtensionFlags Name(1 << Name##_ID);
   static const CcTestExtensionFlags NO_EXTENSIONS(0);
   static const CcTestExtensionFlags ALL_EXTENSIONS((1 << kMaxExtensions) - 1);
   EXTENSION_LIST(DEFINE_EXTENSION_FLAG)
 #undef DEFINE_EXTENSION_FLAG
 
+void RunAndTrackAllocations(void (*test)());
+
+#ifndef TEST_TRACK_ALLOCATIONS
+#define TEST_TRACK_ALLOCATIONS(Name)            \
+  static void Test##Name();                     \
+  TEST(Name##TrackAllocations) {                \
+    RunAndTrackAllocations(&Test##Name);        \
+  }                                             \
+  TEST(Name)
+#endif
+
 class CcTest {
  public:
   typedef void (TestFunction)();
   CcTest(TestFunction* callback, const char* file, const char* name,
          const char* dependency, bool enabled);
   void Run() { callback_(); }
   static CcTest* last() { return last_; }
   CcTest* prev() { return prev_; }
   const char* file() { return file_; }
   const char* name() { return name_; }
@@ skipping 209 matching lines @@
 // Helper function that simulates a full old-space in the heap.
 static inline void SimulateFullSpace(v8::internal::PagedSpace* space) {
   int old_linear_size = static_cast<int>(space->limit() - space->top());
   space->Free(space->top(), old_linear_size);
   space->SetTop(space->limit(), space->limit());
   space->ResetFreeList();
   space->ClearStats();
 }
 
 
+// Helper class for new allocations tracking and checking.
+// To use checking of JS allocations tracking in a test,
+// use TEST_TRACK_ALLOCATIONS macro or just create an instance
+// of this class.
+class HeapObjectsTracker {
+ public:
+  HeapObjectsTracker() {
+    heap_profiler_ = i::Isolate::Current()->heap_profiler();
+    CHECK_NE(NULL, heap_profiler_);
+
+    heap_profiler_->StartHeapAllocationsRecording();
+
+    HEAP->CollectAllAvailableGarbage();
+    HEAP->CollectAllAvailableGarbage();
+    HEAP->CollectAllAvailableGarbage();
+    untracked_on_start_ = heap_profiler_->FindUntrackedObjects();
+  }
+
+  virtual ~HeapObjectsTracker() {
+    CHECK_GE(0, CountUntrackedObjects());
+    heap_profiler_->StopHeapAllocationsRecording();
+  }
+
+ private:
+  int CountUntrackedObjects() {
+    HEAP->CollectAllAvailableGarbage();
+    HEAP->CollectAllAvailableGarbage();
+    HEAP->CollectAllAvailableGarbage();
+
+    return heap_profiler_->FindUntrackedObjects() - untracked_on_start_;
+  }
+
+  i::HeapProfiler* heap_profiler_;
+  int untracked_on_start_;
+};
+
+
 #endif  // ifndef CCTEST_H_