mirror of
https://gitlab.winehq.org/wine/wine-gecko.git
synced 2024-09-13 09:24:08 -07:00
d8e503c38b
--HG-- rename : tools/trace-malloc/bloatblame.c => tools/trace-malloc/bloatblame.cpp
321 lines
8.9 KiB
C++
321 lines
8.9 KiB
C++
/* -*- Mode: C++; tab-width: 6; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
|
|
/* ***** BEGIN LICENSE BLOCK *****
|
|
* Version: MPL 1.1/GPL 2.0/LGPL 2.1
|
|
*
|
|
* 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 Mozilla XPCOM.
|
|
*
|
|
* The Initial Developer of the Original Code is
|
|
* Benjamin Smedberg <benjamin@smedbergs.us>
|
|
*
|
|
* Portions created by the Initial Developer are Copyright (C) 2005
|
|
* the Initial Developer. All Rights Reserved.
|
|
*
|
|
* Contributor(s):
|
|
* Mike Hommey <mh@glandium.org>
|
|
*
|
|
* Alternatively, the contents of this file may be used under the terms of
|
|
* either the GNU General Public License Version 2 or later (the "GPL"), or
|
|
* the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
|
|
* in which case the provisions of the GPL or the LGPL are applicable instead
|
|
* of those above. If you wish to allow use of your version of this file only
|
|
* under the terms of either the GPL or the LGPL, and not to allow others to
|
|
* use your version of this file under the terms of the MPL, indicate your
|
|
* decision by deleting the provisions above and replace them with the notice
|
|
* and other provisions required by the GPL or the LGPL. If you do not delete
|
|
* the provisions above, a recipient may use your version of this file under
|
|
* the terms of any one of the MPL, the GPL or the LGPL.
|
|
*
|
|
* ***** END LICENSE BLOCK ***** */
|
|
|
|
#include "nsGlueLinking.h"
|
|
#include "nsXPCOMGlue.h"
|
|
#include "nscore.h"
|
|
|
|
#if defined(LINUX) && !defined(ANDROID)
|
|
#define _GNU_SOURCE
|
|
#include <fcntl.h>
|
|
#include <unistd.h>
|
|
#include <elf.h>
|
|
#include <limits.h>
|
|
#endif
|
|
|
|
#include <errno.h>
|
|
#include <dlfcn.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <stdio.h>
|
|
|
|
#if defined(SUNOS4) || defined(NEXTSTEP) || \
|
|
(defined(OPENBSD) || defined(NETBSD)) && !defined(__ELF__)
|
|
#define LEADING_UNDERSCORE "_"
|
|
#else
|
|
#define LEADING_UNDERSCORE
|
|
#endif
|
|
|
|
#ifdef NS_TRACE_MALLOC
|
|
extern "C" {
|
|
NS_EXPORT_(__ptr_t) __libc_malloc(size_t);
|
|
NS_EXPORT_(__ptr_t) __libc_calloc(size_t, size_t);
|
|
NS_EXPORT_(__ptr_t) __libc_realloc(__ptr_t, size_t);
|
|
NS_EXPORT_(void) __libc_free(__ptr_t);
|
|
NS_EXPORT_(__ptr_t) __libc_memalign(size_t, size_t);
|
|
NS_EXPORT_(__ptr_t) __libc_valloc(size_t);
|
|
}
|
|
|
|
static __ptr_t (*_malloc)(size_t) = __libc_malloc;
|
|
static __ptr_t (*_calloc)(size_t, size_t) = __libc_calloc;
|
|
static __ptr_t (*_realloc)(__ptr_t, size_t) = __libc_realloc;
|
|
static void (*_free)(__ptr_t) = __libc_free;
|
|
static __ptr_t (*_memalign)(size_t, size_t) = __libc_memalign;
|
|
static __ptr_t (*_valloc)(size_t) = __libc_valloc;
|
|
|
|
NS_EXPORT_(__ptr_t) malloc(size_t size)
|
|
{
|
|
return _malloc(size);
|
|
}
|
|
|
|
NS_EXPORT_(__ptr_t) calloc(size_t nmemb, size_t size)
|
|
{
|
|
return _calloc(nmemb, size);
|
|
}
|
|
|
|
NS_EXPORT_(__ptr_t) realloc(__ptr_t ptr, size_t size)
|
|
{
|
|
return _realloc(ptr, size);
|
|
}
|
|
|
|
NS_EXPORT_(void) free(__ptr_t ptr)
|
|
{
|
|
_free(ptr);
|
|
}
|
|
|
|
NS_EXPORT_(void) cfree(__ptr_t ptr)
|
|
{
|
|
_free(ptr);
|
|
}
|
|
|
|
NS_EXPORT_(__ptr_t) memalign(size_t boundary, size_t size)
|
|
{
|
|
return _memalign(boundary, size);
|
|
}
|
|
|
|
NS_EXPORT_(int)
|
|
posix_memalign(void **memptr, size_t alignment, size_t size)
|
|
{
|
|
__ptr_t ptr = _memalign(alignment, size);
|
|
if (!ptr)
|
|
return ENOMEM;
|
|
*memptr = ptr;
|
|
return 0;
|
|
}
|
|
|
|
NS_EXPORT_(__ptr_t) valloc(size_t size)
|
|
{
|
|
return _valloc(size);
|
|
}
|
|
#endif /* NS_TRACE_MALLOC */
|
|
|
|
struct DependentLib
|
|
{
|
|
void *libHandle;
|
|
DependentLib *next;
|
|
};
|
|
|
|
static DependentLib *sTop;
|
|
static void* sXULLibHandle;
|
|
|
|
static void
|
|
AppendDependentLib(void *libHandle)
|
|
{
|
|
DependentLib *d = new DependentLib;
|
|
if (!d)
|
|
return;
|
|
|
|
d->next = sTop;
|
|
d->libHandle = libHandle;
|
|
|
|
sTop = d;
|
|
}
|
|
|
|
#if defined(LINUX) && !defined(ANDROID)
|
|
static const unsigned int bufsize = 4096;
|
|
|
|
#ifdef HAVE_64BIT_OS
|
|
typedef Elf64_Ehdr Elf_Ehdr;
|
|
typedef Elf64_Phdr Elf_Phdr;
|
|
static const unsigned char ELFCLASS = ELFCLASS64;
|
|
typedef Elf64_Off Elf_Off;
|
|
#else
|
|
typedef Elf32_Ehdr Elf_Ehdr;
|
|
typedef Elf32_Phdr Elf_Phdr;
|
|
static const unsigned char ELFCLASS = ELFCLASS32;
|
|
typedef Elf32_Off Elf_Off;
|
|
#endif
|
|
|
|
static void
|
|
preload(const char *file)
|
|
{
|
|
union {
|
|
char buf[bufsize];
|
|
Elf_Ehdr ehdr;
|
|
} elf;
|
|
int fd = open(file, O_RDONLY);
|
|
if (fd < 0)
|
|
return;
|
|
// Read ELF header (ehdr) and program header table (phdr).
|
|
// We check that the ELF magic is found, that the ELF class matches
|
|
// our own, and that the program header table as defined in the ELF
|
|
// headers fits in the buffer we read.
|
|
if ((read(fd, elf.buf, bufsize) <= 0) ||
|
|
(memcmp(elf.buf, ELFMAG, 4)) ||
|
|
(elf.ehdr.e_ident[EI_CLASS] != ELFCLASS) ||
|
|
(elf.ehdr.e_phoff + elf.ehdr.e_phentsize * elf.ehdr.e_phnum >= bufsize)) {
|
|
close(fd);
|
|
return;
|
|
}
|
|
// The program header table contains segment definitions. One such
|
|
// segment type is PT_LOAD, which describes how the dynamic loader
|
|
// is going to map the file in memory. We use that information to
|
|
// find the biggest offset from the library that will be mapped in
|
|
// memory.
|
|
Elf_Phdr *phdr = (Elf_Phdr *)&elf.buf[elf.ehdr.e_phoff];
|
|
Elf_Off end = 0;
|
|
for (int phnum = elf.ehdr.e_phnum; phnum; phdr++, phnum--)
|
|
if ((phdr->p_type == PT_LOAD) &&
|
|
(end < phdr->p_offset + phdr->p_filesz))
|
|
end = phdr->p_offset + phdr->p_filesz;
|
|
// Let the kernel read ahead what the dynamic loader is going to
|
|
// map in memory soon after.
|
|
if (end > 0) {
|
|
readahead(fd, 0, end);
|
|
}
|
|
close(fd);
|
|
}
|
|
#endif
|
|
|
|
static void
|
|
ReadDependentCB(const char *aDependentLib, bool do_preload)
|
|
{
|
|
#if defined(LINUX) && !defined(ANDROID)
|
|
if (do_preload)
|
|
preload(aDependentLib);
|
|
#endif
|
|
void *libHandle = dlopen(aDependentLib, RTLD_GLOBAL | RTLD_LAZY);
|
|
if (!libHandle)
|
|
return;
|
|
|
|
AppendDependentLib(libHandle);
|
|
}
|
|
|
|
nsresult
|
|
XPCOMGlueLoad(const char *xpcomFile, GetFrozenFunctionsFunc *func)
|
|
{
|
|
char xpcomDir[MAXPATHLEN];
|
|
if (realpath(xpcomFile, xpcomDir)) {
|
|
char *lastSlash = strrchr(xpcomDir, '/');
|
|
if (lastSlash) {
|
|
*lastSlash = '\0';
|
|
|
|
XPCOMGlueLoadDependentLibs(xpcomDir, ReadDependentCB);
|
|
|
|
snprintf(lastSlash, MAXPATHLEN - strlen(xpcomDir), "/" XUL_DLL);
|
|
|
|
sXULLibHandle = dlopen(xpcomDir, RTLD_GLOBAL | RTLD_LAZY);
|
|
|
|
#ifdef NS_TRACE_MALLOC
|
|
_malloc = (__ptr_t(*)(size_t)) dlsym(sXULLibHandle, "malloc");
|
|
_calloc = (__ptr_t(*)(size_t, size_t)) dlsym(sXULLibHandle, "calloc");
|
|
_realloc = (__ptr_t(*)(__ptr_t, size_t)) dlsym(sXULLibHandle, "realloc");
|
|
_free = (void(*)(__ptr_t)) dlsym(sXULLibHandle, "free");
|
|
_memalign = (__ptr_t(*)(size_t, size_t)) dlsym(sXULLibHandle, "memalign");
|
|
_valloc = (__ptr_t(*)(size_t)) dlsym(sXULLibHandle, "valloc");
|
|
#endif
|
|
}
|
|
}
|
|
|
|
// RTLD_DEFAULT is not defined in non-GNU toolchains, and it is
|
|
// (void*) 0 in any case.
|
|
|
|
void *libHandle = nsnull;
|
|
|
|
if (xpcomFile[0] != '.' || xpcomFile[1] != '\0') {
|
|
libHandle = dlopen(xpcomFile, RTLD_GLOBAL | RTLD_LAZY);
|
|
if (libHandle) {
|
|
AppendDependentLib(libHandle);
|
|
} else {
|
|
fprintf(stderr, "XPCOMGlueLoad error for file %s:\n%s\n", xpcomFile,
|
|
dlerror());
|
|
}
|
|
}
|
|
|
|
GetFrozenFunctionsFunc sym =
|
|
(GetFrozenFunctionsFunc) dlsym(libHandle,
|
|
LEADING_UNDERSCORE "NS_GetFrozenFunctions");
|
|
|
|
if (!sym) { // No symbol found.
|
|
XPCOMGlueUnload();
|
|
return NS_ERROR_NOT_AVAILABLE;
|
|
}
|
|
|
|
*func = sym;
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
void
|
|
XPCOMGlueUnload()
|
|
{
|
|
while (sTop) {
|
|
dlclose(sTop->libHandle);
|
|
|
|
DependentLib *temp = sTop;
|
|
sTop = sTop->next;
|
|
|
|
delete temp;
|
|
}
|
|
|
|
if (sXULLibHandle) {
|
|
#ifdef NS_TRACE_MALLOC
|
|
_malloc = __libc_malloc;
|
|
_calloc = __libc_calloc;
|
|
_realloc = __libc_realloc;
|
|
_free = __libc_free;
|
|
_memalign = __libc_memalign;
|
|
_valloc = __libc_valloc;
|
|
#endif
|
|
dlclose(sXULLibHandle);
|
|
sXULLibHandle = nsnull;
|
|
}
|
|
}
|
|
|
|
nsresult
|
|
XPCOMGlueLoadXULFunctions(const nsDynamicFunctionLoad *symbols)
|
|
{
|
|
// We don't null-check sXULLibHandle because this might work even
|
|
// if it is null (same as RTLD_DEFAULT)
|
|
|
|
nsresult rv = NS_OK;
|
|
while (symbols->functionName) {
|
|
char buffer[512];
|
|
snprintf(buffer, sizeof(buffer),
|
|
LEADING_UNDERSCORE "%s", symbols->functionName);
|
|
|
|
*symbols->function = (NSFuncPtr) dlsym(sXULLibHandle, buffer);
|
|
if (!*symbols->function)
|
|
rv = NS_ERROR_LOSS_OF_SIGNIFICANT_DATA;
|
|
|
|
++symbols;
|
|
}
|
|
return rv;
|
|
}
|