1// Copyright (c) 2005, Google Inc. 2// All rights reserved. 3// 4// Redistribution and use in source and binary forms, with or without 5// modification, are permitted provided that the following conditions are 6// met: 7// 8// * Redistributions of source code must retain the above copyright 9// notice, this list of conditions and the following disclaimer. 10// * Redistributions in binary form must reproduce the above 11// copyright notice, this list of conditions and the following disclaimer 12// in the documentation and/or other materials provided with the 13// distribution. 14// * Neither the name of Google Inc. nor the names of its 15// contributors may be used to endorse or promote products derived from 16// this software without specific prior written permission. 17// 18// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 19// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 20// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 21// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 22// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 23// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 24// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 25// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 26// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 27// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 28// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 29 30// --- 31// Author: Craig Silverstein 32// 33// A small program that just exercises our heap profiler by allocating 34// memory and letting the heap-profiler emit a profile. We don't test 35// threads (TODO). By itself, this unittest tests that the heap-profiler 36// doesn't crash on simple programs, but its output can be analyzed by 37// another testing script to actually verify correctness. See, eg, 38// heap-profiler_unittest.sh. 39 40#include "config_for_unittests.h" 41#include <stdlib.h> 42#include <stdio.h> 43#include <fcntl.h> // for mkdir() 44#include <sys/stat.h> // for mkdir() on freebsd and os x 45#ifdef HAVE_UNISTD_H 46#include <unistd.h> // for fork() 47#endif 48#include <sys/wait.h> // for wait() 49#include <string> 50#include "base/basictypes.h" 51#include "base/logging.h" 52#include <gperftools/heap-profiler.h> 53 54using std::string; 55 56static const int kMaxCount = 100000; 57int* g_array[kMaxCount]; // an array of int-vectors 58 59static ATTRIBUTE_NOINLINE void Allocate(int start, int end, int size) { 60 for (int i = start; i < end; ++i) { 61 if (i < kMaxCount) 62 g_array[i] = new int[size]; 63 } 64} 65 66static ATTRIBUTE_NOINLINE void Allocate2(int start, int end, int size) { 67 for (int i = start; i < end; ++i) { 68 if (i < kMaxCount) 69 g_array[i] = new int[size]; 70 } 71} 72 73static void Deallocate(int start, int end) { 74 for (int i = start; i < end; ++i) { 75 delete[] g_array[i]; 76 g_array[i] = 0; 77 } 78} 79 80static void TestHeapProfilerStartStopIsRunning() { 81 // If you run this with whole-program heap-profiling on, than 82 // IsHeapProfilerRunning should return true. 83 if (!IsHeapProfilerRunning()) { 84 const char* tmpdir = getenv("TMPDIR"); 85 if (tmpdir == NULL) 86 tmpdir = "/tmp"; 87 mkdir(tmpdir, 0755); // if necessary 88 HeapProfilerStart((string(tmpdir) + "/start_stop").c_str()); 89 CHECK(IsHeapProfilerRunning()); 90 91 Allocate(0, 40, 100); 92 Deallocate(0, 40); 93 94 HeapProfilerStop(); 95 CHECK(!IsHeapProfilerRunning()); 96 } 97} 98 99static void TestDumpHeapProfiler() { 100 // If you run this with whole-program heap-profiling on, than 101 // IsHeapProfilerRunning should return true. 102 if (!IsHeapProfilerRunning()) { 103 const char* tmpdir = getenv("TMPDIR"); 104 if (tmpdir == NULL) 105 tmpdir = "/tmp"; 106 mkdir(tmpdir, 0755); // if necessary 107 HeapProfilerStart((string(tmpdir) + "/dump").c_str()); 108 CHECK(IsHeapProfilerRunning()); 109 110 Allocate(0, 40, 100); 111 Deallocate(0, 40); 112 113 char* output = GetHeapProfile(); 114 free(output); 115 HeapProfilerStop(); 116 } 117} 118 119 120int main(int argc, char** argv) { 121 if (argc > 2 || (argc == 2 && argv[1][0] == '-')) { 122 printf("USAGE: %s [number of children to fork]\n", argv[0]); 123 exit(0); 124 } 125 int num_forks = 0; 126 if (argc == 2) { 127 num_forks = atoi(argv[1]); 128 } 129 130 TestHeapProfilerStartStopIsRunning(); 131 TestDumpHeapProfiler(); 132 133 Allocate(0, 40, 100); 134 Deallocate(0, 40); 135 136 Allocate(0, 40, 100); 137 Allocate(0, 40, 100); 138 Allocate2(40, 400, 1000); 139 Allocate2(400, 1000, 10000); 140 Deallocate(0, 1000); 141 142 Allocate(0, 100, 100000); 143 Deallocate(0, 10); 144 Deallocate(10, 20); 145 Deallocate(90, 100); 146 Deallocate(20, 90); 147 148 while (num_forks-- > 0) { 149 switch (fork()) { 150 case -1: 151 printf("FORK failed!\n"); 152 return 1; 153 case 0: // child 154 return execl(argv[0], argv[0], NULL); // run child with no args 155 default: 156 wait(NULL); // we'll let the kids run one at a time 157 } 158 } 159 160 printf("DONE.\n"); 161 162 return 0; 163} 164