mirror of
https://github.com/klzgrad/naiveproxy.git
synced 2024-12-01 01:36:09 +03:00
104 lines
4.0 KiB
C++
104 lines
4.0 KiB
C++
// Copyright 2017 The Chromium Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#ifndef BASE_ALLOCATOR_MALLOC_ZONE_FUNCTIONS_MAC_H_
|
|
#define BASE_ALLOCATOR_MALLOC_ZONE_FUNCTIONS_MAC_H_
|
|
|
|
#include <malloc/malloc.h>
|
|
#include <stddef.h>
|
|
|
|
#include "base/base_export.h"
|
|
#include "base/logging.h"
|
|
#include "third_party/apple_apsl/malloc.h"
|
|
|
|
namespace base {
|
|
namespace allocator {
|
|
|
|
typedef void* (*malloc_type)(struct _malloc_zone_t* zone, size_t size);
|
|
typedef void* (*calloc_type)(struct _malloc_zone_t* zone,
|
|
size_t num_items,
|
|
size_t size);
|
|
typedef void* (*valloc_type)(struct _malloc_zone_t* zone, size_t size);
|
|
typedef void (*free_type)(struct _malloc_zone_t* zone, void* ptr);
|
|
typedef void* (*realloc_type)(struct _malloc_zone_t* zone,
|
|
void* ptr,
|
|
size_t size);
|
|
typedef void* (*memalign_type)(struct _malloc_zone_t* zone,
|
|
size_t alignment,
|
|
size_t size);
|
|
typedef unsigned (*batch_malloc_type)(struct _malloc_zone_t* zone,
|
|
size_t size,
|
|
void** results,
|
|
unsigned num_requested);
|
|
typedef void (*batch_free_type)(struct _malloc_zone_t* zone,
|
|
void** to_be_freed,
|
|
unsigned num_to_be_freed);
|
|
typedef void (*free_definite_size_type)(struct _malloc_zone_t* zone,
|
|
void* ptr,
|
|
size_t size);
|
|
typedef size_t (*size_fn_type)(struct _malloc_zone_t* zone, const void* ptr);
|
|
|
|
struct MallocZoneFunctions {
|
|
malloc_type malloc;
|
|
calloc_type calloc;
|
|
valloc_type valloc;
|
|
free_type free;
|
|
realloc_type realloc;
|
|
memalign_type memalign;
|
|
batch_malloc_type batch_malloc;
|
|
batch_free_type batch_free;
|
|
free_definite_size_type free_definite_size;
|
|
size_fn_type size;
|
|
const ChromeMallocZone* context;
|
|
};
|
|
|
|
BASE_EXPORT void StoreZoneFunctions(const ChromeMallocZone* zone,
|
|
MallocZoneFunctions* functions);
|
|
static constexpr int kMaxZoneCount = 30;
|
|
BASE_EXPORT extern MallocZoneFunctions g_malloc_zones[kMaxZoneCount];
|
|
|
|
// The array g_malloc_zones stores all information about malloc zones before
|
|
// they are shimmed. This information needs to be accessed during dispatch back
|
|
// into the zone, and additional zones may be added later in the execution fo
|
|
// the program, so the array needs to be both thread-safe and high-performance.
|
|
//
|
|
// We begin by creating an array of MallocZoneFunctions of fixed size. We will
|
|
// never modify the container, which provides thread-safety to iterators. When
|
|
// we want to add a MallocZoneFunctions to the container, we:
|
|
// 1. Fill in all the fields.
|
|
// 2. Update the total zone count.
|
|
// 3. Insert a memory barrier.
|
|
// 4. Insert our shim.
|
|
//
|
|
// Each MallocZoneFunctions is uniquely identified by |context|, which is a
|
|
// pointer to the original malloc zone. When we wish to dispatch back to the
|
|
// original malloc zones, we iterate through the array, looking for a matching
|
|
// |context|.
|
|
//
|
|
// Most allocations go through the default allocator. We will ensure that the
|
|
// default allocator is stored as the first MallocZoneFunctions.
|
|
//
|
|
// Returns whether the zone was successfully stored.
|
|
BASE_EXPORT bool StoreMallocZone(ChromeMallocZone* zone);
|
|
BASE_EXPORT bool IsMallocZoneAlreadyStored(ChromeMallocZone* zone);
|
|
BASE_EXPORT bool DoesMallocZoneNeedReplacing(
|
|
ChromeMallocZone* zone,
|
|
const MallocZoneFunctions* functions);
|
|
|
|
BASE_EXPORT int GetMallocZoneCountForTesting();
|
|
BASE_EXPORT void ClearAllMallocZonesForTesting();
|
|
|
|
inline MallocZoneFunctions& GetFunctionsForZone(void* zone) {
|
|
for (unsigned int i = 0; i < kMaxZoneCount; ++i) {
|
|
if (g_malloc_zones[i].context == zone)
|
|
return g_malloc_zones[i];
|
|
}
|
|
IMMEDIATE_CRASH();
|
|
}
|
|
|
|
} // namespace allocator
|
|
} // namespace base
|
|
|
|
#endif // BASE_ALLOCATOR_MALLOC_ZONE_FUNCTIONS_MAC_H_
|