gecko-dev/xpcom/base/nsLeakDetector.cpp

203 lines
4.9 KiB
C++

/* -*- Mode: IDL; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-
*
* The contents of this file are subject to the Mozilla Public
* License Version 1.1 (the "License"); you may not use this file
* except in compliance with the License. You may obtain a copy of
* the License at http://www.mozilla.org/MPL/
*
* Software distributed under the License is distributed on an "AS
* IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
* implied. See the License for the specific language governing
* rights and limitations under the License.
*
* The Original Code is the Mozilla browser.
*
* The Initial Developer of the Original Code is Netscape
* Communications, Inc. Portions created by Netscape are
* Copyright (C) 1999, Mozilla. All Rights Reserved.
*
* Contributor(s):
* Patrick C. Beard <beard@netscape.com>
*/
#if defined(GC_LEAK_DETECTOR)
#include "nsLeakDetector.h"
#include "nsCOMPtr.h"
#include "nsIComponentManager.h"
#include "nsIServiceManager.h"
#include "nsIGenericFactory.h"
#include "nsILeakDetector.h"
#include "nsICollection.h"
#include <stdio.h>
#include <time.h>
#include "gc.h"
extern "C" {
extern FILE *GC_stdout, *GC_stderr;
extern void GC_trace_object(GC_PTR object, int verbose);
extern void GC_mark_object(GC_PTR object, GC_word mark);
}
static nsresult nextLeakFile()
{
if (GC_stderr != NULL)
fclose(GC_stderr);
// generate a time stamped report name.
time_t timer;
time(&timer);
tm* now = localtime(&timer);
char reportName[256];
sprintf(reportName, "Leaks%02d%02d%02d",
now->tm_hour, now->tm_min, now->tm_sec);
GC_stderr = fopen(reportName, "w");
return NS_OK;
}
static FILE* openTraceFile()
{
// generate a time stamped report name.
time_t timer;
time(&timer);
tm* now = localtime(&timer);
char reportName[256];
sprintf(reportName, "Trace%02d%02d%02d",
now->tm_hour, now->tm_min, now->tm_sec);
return fopen(reportName, "w");
}
class nsLeakDetector : public nsILeakDetector {
public:
nsLeakDetector();
virtual ~nsLeakDetector();
NS_DECL_ISUPPORTS
NS_DECL_NSILEAKDETECTOR
};
NS_IMPL_ISUPPORTS1(nsLeakDetector, nsILeakDetector)
nsLeakDetector::nsLeakDetector() {
NS_INIT_REFCNT();
}
nsLeakDetector::~nsLeakDetector() {}
NS_METHOD nsLeakDetector::DumpLeaks()
{
GC_gcollect();
return nextLeakFile();
}
NS_METHOD nsLeakDetector::TraceObject(nsISupports* object, PRBool verbose)
{
FILE* trace = openTraceFile();
if (trace != NULL) {
FILE* old_stderr = GC_stderr;
GC_stderr = trace;
GC_trace_object(object, (verbose ? 1 : 0));
GC_stderr = old_stderr;
fclose(trace);
return NS_OK;
}
return NS_ERROR_FAILURE;
}
NS_METHOD nsLeakDetector::TraceCollection(nsICollection* objects, PRBool verbose)
{
PRUint32 count;
if (NS_FAILED(objects->Count(&count)))
return NS_ERROR_FAILURE;
nsCOMPtr<nsISupports>* elements = new nsCOMPtr<nsISupports>[count];
if (elements == nsnull)
return NS_ERROR_OUT_OF_MEMORY;
for (PRUint32 i = 0; i < count; ++i)
objects->GetElementAt(i, getter_AddRefs(elements[i]));
nsresult rv = NS_ERROR_FAILURE;
FILE* trace = openTraceFile();
if (trace != NULL) {
FILE* old_stderr = GC_stderr;
GC_stderr = trace;
GC_trace_object(elements, (verbose ? 1 : 0));
GC_stderr = old_stderr;
fclose(trace);
rv = NS_OK;
}
delete[] elements;
return rv;
}
NS_METHOD nsLeakDetector::MarkObject(nsISupports* object, PRBool marked)
{
GC_mark_object(object, (marked ? 1 : 0));
return NS_OK;
}
NS_METHOD nsLeakDetector::GetServices(nsISupports* *result)
{
nsIServiceManager* serviceManager = nsnull;
nsresult rv = nsServiceManager::GetGlobalServiceManager(&serviceManager);
if (rv == NS_OK) {
*result = serviceManager;
NS_ADDREF(*result);
}
return rv;
}
#define NS_CLEAKDETECTOR_CID_STR "bb1ba360-1dd1-11b2-b30e-aa2314429f54"
#define NS_CLEAKDETECTOR_CID {0xbb1ba360, 0x1dd1, 0x11b2, {0xb3, 0x0e, 0xaa, 0x23, 0x14, 0x42, 0x9f, 0x54}}
#define NS_CLEAKDETECTOR_CONTRACTID "@mozilla.org/xpcom/leakdetector;1"
NS_GENERIC_FACTORY_CONSTRUCTOR(nsLeakDetector)
static NS_DEFINE_CID(kCLeakDetectorCID, NS_CLEAKDETECTOR_CID);
nsresult NS_InitLeakDetector()
{
nsresult rv;
// open the first leak file.
rv = nextLeakFile();
if (rv != NS_OK)
return rv;
// create a generic factory for the leak detector.
nsCOMPtr<nsIGenericFactory> factory;
rv = NS_NewGenericFactory(getter_AddRefs(factory), &nsLeakDetectorConstructor);
if (rv != NS_OK)
return rv;
// register this factory with the component manager.
rv = nsComponentManager::RegisterFactory(kCLeakDetectorCID, "LeakDetector", NS_CLEAKDETECTOR_CONTRACTID, factory, PR_TRUE);
return rv;
}
nsresult NS_ShutdownLeakDetector()
{
GC_gcollect();
#if 0
nextLeakFile();
if (GC_stdout != NULL) {
fprintf(GC_stdout, "ShutDown Leaks\n");
GC_clear_roots();
GC_gcollect();
}
#endif
return NS_OK;
}
#endif /* defined(GC_LEAK_DETECTOR) */