15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2009, Google Inc. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// All rights reserved. 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Redistribution and use in source and binary forms, with or without 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// modification, are permitted provided that the following conditions are 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// met: 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// * Redistributions of source code must retain the above copyright 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// notice, this list of conditions and the following disclaimer. 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// * Redistributions in binary form must reproduce the above 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// copyright notice, this list of conditions and the following disclaimer 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// in the documentation and/or other materials provided with the 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// distribution. 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// * Neither the name of Google Inc. nor the names of its 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// contributors may be used to endorse or promote products derived from 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// this software without specific prior written permission. 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// --- 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Author: Craig Silverstein 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// This forks out to pprof to do the actual symbolizing. We might 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// be better off writing our own in C++. 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "config.h" 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "symbolize.h" 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <stdlib.h> 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifdef HAVE_UNISTD_H 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <unistd.h> // for write() 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifdef HAVE_SYS_SOCKET_H 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <sys/socket.h> // for socketpair() -- needed by Symbolize 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifdef HAVE_SYS_WAIT_H 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <sys/wait.h> // for wait() -- needed by Symbolize 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifdef HAVE_POLL_H 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <poll.h> 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifdef __MACH__ 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <mach-o/dyld.h> // for GetProgramInvocationName() 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <limits.h> // for PATH_MAX 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(__CYGWIN__) || defined(__CYGWIN32__) 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <io.h> // for get_osfhandle() 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <string> 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/commandlineflags.h" 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h" 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/sysinfo.h" 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using std::string; 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using tcmalloc::DumpProcSelfMaps; // from sysinfo.h 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)DEFINE_string(symbolize_pprof, 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EnvToString("PPROF_PATH", "pprof"), 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Path to pprof to call for reporting function names."); 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// heap_profile_table_pprof may be referenced after destructors are 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// called (since that's when leak-checking is done), so we make 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// a more-permanent copy that won't ever get destroyed. 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static string* g_pprof_path = new string(FLAGS_symbolize_pprof); 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Returns NULL if we're on an OS where we can't get the invocation name. 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Using a static var is ok because we're not called from a thread. 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static char* GetProgramInvocationName() { 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(HAVE_PROGRAM_INVOCATION_NAME) 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) extern char* program_invocation_name; // gcc provides this 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return program_invocation_name; 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#elif defined(__MACH__) 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We don't want to allocate memory for this since we may be 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // calculating it when memory is corrupted. 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static char program_invocation_name[PATH_MAX]; 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (program_invocation_name[0] == '\0') { // first time calculating 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) uint32_t length = sizeof(program_invocation_name); 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (_NSGetExecutablePath(program_invocation_name, &length)) 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return program_invocation_name; 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#else 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; // figure out a way to get argv[0] 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Prints an error message when you can't run Symbolize(). 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static void PrintError(const char* reason) { 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RAW_LOG(ERROR, 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "*** WARNING: Cannot convert addresses to symbols in output below.\n" 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "*** Reason: %s\n" 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "*** If you cannot fix this, try running pprof directly.\n", 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) reason); 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void SymbolTable::Add(const void* addr) { 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) symbolization_table_[addr] = ""; 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char* SymbolTable::GetSymbol(const void* addr) { 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return symbolization_table_[addr]; 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Updates symbolization_table with the pointers to symbol names corresponding 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// to its keys. The symbol names are stored in out, which is allocated and 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// freed by the caller of this routine. 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Note that the forking/etc is not thread-safe or re-entrant. That's 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// ok for the purpose we need -- reporting leaks detected by heap-checker 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// -- but be careful if you decide to use this routine for other purposes. 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Returns number of symbols read on error. If can't symbolize, returns 0 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// and emits an error message about why. 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int SymbolTable::Symbolize() { 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if !defined(HAVE_UNISTD_H) || !defined(HAVE_SYS_SOCKET_H) || !defined(HAVE_SYS_WAIT_H) 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PrintError("Perftools does not know how to call a sub-process on this O/S"); 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0; 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#else 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const char* argv0 = GetProgramInvocationName(); 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (argv0 == NULL) { // can't call symbolize if we can't figure out our name 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PrintError("Cannot figure out the name of this executable (argv0)"); 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0; 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (access(g_pprof_path->c_str(), R_OK) != 0) { 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PrintError("Cannot find 'pprof' (is PPROF_PATH set correctly?)"); 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0; 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // All this work is to do two-way communication. ugh. 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int *child_in = NULL; // file descriptors 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int *child_out = NULL; // for now, we don't worry about child_err 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int child_fds[5][2]; // socketpair may be called up to five times below 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The client program may close its stdin and/or stdout and/or stderr 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // thus allowing socketpair to reuse file descriptors 0, 1 or 2. 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // In this case the communication between the forked processes may be broken 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // if either the parent or the child tries to close or duplicate these 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // descriptors. The loop below produces two pairs of file descriptors, each 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // greater than 2 (stderr). 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (int i = 0; i < 5; i++) { 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (socketpair(AF_UNIX, SOCK_STREAM, 0, child_fds[i]) == -1) { 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (int j = 0; j < i; j++) { 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) close(child_fds[j][0]); 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) close(child_fds[j][1]); 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PrintError("Cannot create a socket pair"); 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0; 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if ((child_fds[i][0] > 2) && (child_fds[i][1] > 2)) { 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (child_in == NULL) { 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) child_in = child_fds[i]; 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) child_out = child_fds[i]; 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (int j = 0; j < i; j++) { 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (child_fds[j] == child_in) continue; 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) close(child_fds[j][0]); 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) close(child_fds[j][1]); 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) break; 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) switch (fork()) { 1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) case -1: { // error 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) close(child_in[0]); 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) close(child_in[1]); 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) close(child_out[0]); 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) close(child_out[1]); 1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PrintError("Unknown error calling fork()"); 1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0; 1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) case 0: { // child 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) close(child_in[1]); // child uses the 0's, parent uses the 1's 1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) close(child_out[1]); // child uses the 0's, parent uses the 1's 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) close(0); 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) close(1); 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (dup2(child_in[0], 0) == -1) _exit(1); 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (dup2(child_out[0], 1) == -1) _exit(2); 1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Unset vars that might cause trouble when we fork 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unsetenv("CPUPROFILE"); 1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unsetenv("HEAPPROFILE"); 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unsetenv("HEAPCHECK"); 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) unsetenv("PERFTOOLS_VERBOSE"); 1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) execlp(g_pprof_path->c_str(), g_pprof_path->c_str(), 1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "--symbols", argv0, NULL); 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) _exit(3); // if execvp fails, it's bad news for us 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) default: { // parent 1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) close(child_in[0]); // child uses the 0's, parent uses the 1's 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) close(child_out[0]); // child uses the 0's, parent uses the 1's 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifdef HAVE_POLL_H 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Waiting for 1ms seems to give the OS time to notice any errors. 2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) poll(0, 0, 1); 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // For maximum safety, we check to make sure the execlp 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // succeeded before trying to write. (Otherwise we'll get a 2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // SIGPIPE.) For systems without poll.h, we'll just skip this 2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // check, and trust that the user set PPROF_PATH correctly! 2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) struct pollfd pfd = { child_in[1], POLLOUT, 0 }; 2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!poll(&pfd, 1, 0) || !(pfd.revents & POLLOUT) || 2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) (pfd.revents & (POLLHUP|POLLERR))) { 2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PrintError("Cannot run 'pprof' (is PPROF_PATH set correctly?)"); 2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0; 2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(__CYGWIN__) || defined(__CYGWIN32__) 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // On cygwin, DumpProcSelfMaps() takes a HANDLE, not an fd. Convert. 2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const HANDLE symbols_handle = (HANDLE) get_osfhandle(child_in[1]); 2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DumpProcSelfMaps(symbols_handle); 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#else 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DumpProcSelfMaps(child_in[1]); // what pprof expects on stdin 2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 2225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Allocate 24 bytes = ("0x" + 8 bytes + "\n" + overhead) for each 2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // address to feed to pprof. 2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const int kOutBufSize = 24 * symbolization_table_.size(); 2265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) char *pprof_buffer = new char[kOutBufSize]; 2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int written = 0; 2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (SymbolMap::const_iterator iter = symbolization_table_.begin(); 2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) iter != symbolization_table_.end(); ++iter) { 2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) written += snprintf(pprof_buffer + written, kOutBufSize - written, 2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // pprof expects format to be 0xXXXXXX 2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "0x%"PRIxPTR"\n", reinterpret_cast<uintptr_t>(iter->first)); 2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) write(child_in[1], pprof_buffer, strlen(pprof_buffer)); 2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) close(child_in[1]); // that's all we need to write 2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const int kSymbolBufferSize = kSymbolSize * symbolization_table_.size(); 2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int total_bytes_read = 0; 2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delete[] symbol_buffer_; 2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) symbol_buffer_ = new char[kSymbolBufferSize]; 2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) memset(symbol_buffer_, '\0', kSymbolBufferSize); 2425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) while (1) { 2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int bytes_read = read(child_out[1], symbol_buffer_ + total_bytes_read, 2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kSymbolBufferSize - total_bytes_read); 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (bytes_read < 0) { 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) close(child_out[1]); 2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PrintError("Cannot read data from pprof"); 2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0; 2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else if (bytes_read == 0) { 2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) close(child_out[1]); 2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) wait(NULL); 2525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) break; 2535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 2545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) total_bytes_read += bytes_read; 2555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We have successfully read the output of pprof into out. Make sure 2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // the last symbol is full (we can tell because it ends with a \n). 2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (total_bytes_read == 0 || symbol_buffer_[total_bytes_read - 1] != '\n') 2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0; 2615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // make the symbolization_table_ values point to the output vector 2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SymbolMap::iterator fill = symbolization_table_.begin(); 2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int num_symbols = 0; 2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const char *current_name = symbol_buffer_; 2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (int i = 0; i < total_bytes_read; i++) { 2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (symbol_buffer_[i] == '\n') { 2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fill->second = current_name; 2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) symbol_buffer_[i] = '\0'; 2695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) current_name = symbol_buffer_ + i + 1; 2705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fill++; 2715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) num_symbols++; 2725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return num_symbols; 2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PrintError("Unkown error (should never occur!)"); 2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0; // shouldn't be reachable 2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 281