tools/footprint/heapdump.cpp

   1 /* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* ***** BEGIN LICENSE BLOCK *****
   3  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
   4  *
   5  * The contents of this file are subject to the Mozilla Public License Version
   6  * 1.1 (the "License"); you may not use this file except in compliance with
   7  * the License. You may obtain a copy of the License at
   8  * http://www.mozilla.org/MPL/
   9  *
  10  * Software distributed under the License is distributed on an "AS IS" basis,
  11  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  12  * for the specific language governing rights and limitations under the
  13  * License.
  14  *
  15  * The Original Code is mozilla.org code.
  16  *
  17  * The Initial Developer of the Original Code is
  18  * Netscape Communications Corporation.
  19  * Portions created by the Initial Developer are Copyright (C) 2001, 2002
  20  * the Initial Developer. All Rights Reserved.
  21  *
  22  * Contributor(s):
  23  *      Suresh Duddi <dp@netscape.com>
  24  *
  25  * Alternatively, the contents of this file may be used under the terms of
  26  * either the GNU General Public License Version 2 or later (the "GPL"), or
  27  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  28  * in which case the provisions of the GPL or the LGPL are applicable instead
  29  * of those above. If you wish to allow use of your version of this file only
  30  * under the terms of either the GPL or the LGPL, and not to allow others to
  31  * use your version of this file under the terms of the MPL, indicate your
  32  * decision by deleting the provisions above and replace them with the notice
  33  * and other provisions required by the GPL or the LGPL. If you do not delete
  34  * the provisions above, a recipient may use your version of this file under
  35  * the terms of any one of the MPL, the GPL or the LGPL.
  36  *
  37  * ***** END LICENSE BLOCK ***** */
  38
  39 /*
  40  * heapdump
  41  *
  42  * Sends a message to netscape 6.3 to dump the heap.
  43  */
  44
  45 #include <windows.h>
  46 #include <stdio.h>
  47
  48 static const char *kMozHeapDumpMessageString = "MOZ_HeapDump";
  49 static const char *kMozAppClassName = "MozillaWindowClass";
  50
  51 BOOL IsMozilla(const char *title)
  52 {
  53     if (!title || !*title)
  54         return FALSE;
  55     // Title containing " - Mozilla" is a mozilla window
  56     if (strstr(title, " - Mozilla"))
  57       return TRUE;
  58
  59     // Title containing "Mozilla {" is a mozilla window
  60     // This form happens when there is no title
  61     if (strstr(title, "Mozilla {"))
  62       return TRUE;
  63
  64     return FALSE;
  65 }
  66
  67 BOOL IsNetscape(const char *title)
  68 {
  69     if (!title || !*title)
  70         return FALSE;
  71     // Skip to where the <space>-<space>Netscape 6 exists
  72     return strstr(title, " - Netscape 6") ? TRUE : FALSE;
  73 }
  74
  75 // Find a mozilla or Netscape top level window in that order
  76 // Pattter we are looking for is
  77 // .* - Netscape 6 [{.*}]
  78 // .* - Mozilla [{.*}]
  79 BOOL CALLBACK EnumWindowsProc(HWND hwnd, LPARAM lParam)
  80 {
  81     HWND * pwnd = (HWND *)lParam;
  82     char buf[1024];
  83
  84     // Make sure we are dealing with a window of our class name
  85     GetClassName(hwnd, buf, sizeof(buf)-1);
  86     if (strcmp(buf, kMozAppClassName))
  87         return TRUE;
  88
  89     GetWindowText(hwnd, buf, sizeof(buf)-1);
  90
  91     if (IsMozilla(buf)) {
  92         // Yeah. Search ends.
  93         *pwnd = hwnd;
  94         return FALSE;
  95     }
  96
  97     // We continue search if we find a Netscape window
  98     // since we might find a mozilla window next
  99     if (IsNetscape(buf))
 100         *pwnd = hwnd;
 101
 102     return TRUE;
 103 }
 104
 105
 106 int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpszCmdLine, int nCmdShow)
 107 {
 108     UINT msgHandle = RegisterWindowMessage(kMozHeapDumpMessageString);
 109
 110     // Find a window to send the message to.
 111     // We look for a mozilla window first and then a netscape window
 112     // logic being that developers run release Netscape (to read mail) and
 113     // mozilla (debug builds).
 114
 115     HWND mozwindow = 0;
 116     EnumWindows(EnumWindowsProc, (LPARAM) &mozwindow);
 117     if (mozwindow == 0) {
 118         printf("Cannot find Mozilla or Netscape 6 window. Exit\n");
 119         exit(-1);
 120     }
 121
 122
 123     char buf[1024];
 124     GetWindowText(mozwindow, buf, sizeof(buf)-1);
 125     if (IsMozilla(buf))
 126         printf("Found Mozilla window with title : %s\n", buf);
 127     else if (IsNetscape(buf))
 128         printf("Found Netscape window with title : %s\n", buf);
 129
 130     SendMessage(mozwindow, msgHandle, 0, 0);
 131
 132     printf("Sending HeapDump Message done.  Heapdump available in c:\\heapdump.txt\n");
 133     return 0;
 134 }
 135