alloctracker.h

Go to the documentation of this file.

#ifndef ION_BASE_ALLOCTRACKER_H_
#define ION_BASE_ALLOCTRACKER_H_


#define ION_ALLOC_TRACKER_DEFINED 1

#if defined(__cplusplus)

#include <memory>
#include <new>

#include "base/integral_types.h"

namespace ion {
namespace base {

class ION_API AllocTracker {
 public:
  enum AllocType {
    kNonArrayAlloc,  // Allocations using new.
    kArrayAlloc,     // Allocations using new[].
    kInternalAlloc,  // Allocations for internal AllocTracker use.
  };
  static const int kNumAllocTypes = kInternalAlloc + 1;

  struct TypeCounts {
    TypeCounts() : allocs(0), bytes(0) {}
    uint64 allocs;
    uint64 bytes;
  };

  struct Counts {
    TypeCounts counts[kNumAllocTypes];
  };

  static AllocTracker* GetMutableInstance();
  static const AllocTracker& GetInstance();

  AllocTracker();

  void SetBaseline();

  const Counts& GetBaselineCounts() const { return baseline_counts_; }

  const Counts& GetAllCounts() const { return all_counts_; }

  const Counts& GetOpenCounts() const { return open_counts_; }

  void* New(std::size_t size, AllocType type);
  void Delete(void* ptr, AllocType type);

 private:
  class Helper;

  virtual ~AllocTracker();

  Counts baseline_counts_;   // Allocations when baseline is set.
  Counts all_counts_;        // All allocations.
  Counts open_counts_;       // Allocations for which delete was not called.

  std::unique_ptr<Helper> helper_;
};

}  // namespace base
}  // namespace ion

void* operator new(std::size_t size);
void* operator new[](std::size_t size);
void operator delete(void* ptr);
void operator delete[](void* ptr);

#endif  // __cplusplus

#endif  // ION_BASE_ALLOCTRACKER_H_