172 lines
3.0 KiB
C
172 lines
3.0 KiB
C
/*
|
|
Copyright (c) 2009-2018 Roger Light <roger@atchoo.org>
|
|
|
|
All rights reserved. This program and the accompanying materials
|
|
are made available under the terms of the Eclipse Public License v1.0
|
|
and Eclipse Distribution License v1.0 which accompany this distribution.
|
|
|
|
The Eclipse Public License is available at
|
|
http://www.eclipse.org/legal/epl-v10.html
|
|
and the Eclipse Distribution License is available at
|
|
http://www.eclipse.org/org/documents/edl-v10.php.
|
|
|
|
Contributors:
|
|
Roger Light - initial implementation and documentation.
|
|
*/
|
|
|
|
#include "config.h"
|
|
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include "memory_mosq.h"
|
|
|
|
#ifdef REAL_WITH_MEMORY_TRACKING
|
|
# if defined(__APPLE__)
|
|
# include <malloc/malloc.h>
|
|
# define malloc_usable_size malloc_size
|
|
# elif defined(__FreeBSD__)
|
|
# include <malloc_np.h>
|
|
# else
|
|
# include <malloc.h>
|
|
# endif
|
|
#endif
|
|
|
|
#ifdef REAL_WITH_MEMORY_TRACKING
|
|
static unsigned long memcount = 0;
|
|
static unsigned long max_memcount = 0;
|
|
#endif
|
|
|
|
#ifdef WITH_BROKER
|
|
static size_t mem_limit = 0;
|
|
void memory__set_limit(size_t lim)
|
|
{
|
|
#ifdef LINUX
|
|
struct rlimit r;
|
|
|
|
r.rlim_cur = lim;
|
|
r.rlim_max = lim;
|
|
|
|
setrlimit(RLIMIT_CPU, &r);
|
|
|
|
mem_limit = 0;
|
|
#else
|
|
mem_limit = lim;
|
|
#endif
|
|
}
|
|
#endif
|
|
|
|
void *mosquitto__calloc(size_t nmemb, size_t size)
|
|
{
|
|
#ifdef REAL_WITH_MEMORY_TRACKING
|
|
if(mem_limit && memcount + size > mem_limit){
|
|
return NULL;
|
|
}
|
|
#endif
|
|
void *mem = calloc(nmemb, size);
|
|
|
|
#ifdef REAL_WITH_MEMORY_TRACKING
|
|
if(mem){
|
|
memcount += malloc_usable_size(mem);
|
|
if(memcount > max_memcount){
|
|
max_memcount = memcount;
|
|
}
|
|
}
|
|
#endif
|
|
|
|
return mem;
|
|
}
|
|
|
|
void mosquitto__free(void *mem)
|
|
{
|
|
#ifdef REAL_WITH_MEMORY_TRACKING
|
|
if(!mem){
|
|
return;
|
|
}
|
|
memcount -= malloc_usable_size(mem);
|
|
#endif
|
|
free(mem);
|
|
}
|
|
|
|
void *mosquitto__malloc(size_t size)
|
|
{
|
|
#ifdef REAL_WITH_MEMORY_TRACKING
|
|
if(mem_limit && memcount + size > mem_limit){
|
|
return NULL;
|
|
}
|
|
#endif
|
|
void *mem = malloc(size);
|
|
|
|
#ifdef REAL_WITH_MEMORY_TRACKING
|
|
if(mem){
|
|
memcount += malloc_usable_size(mem);
|
|
if(memcount > max_memcount){
|
|
max_memcount = memcount;
|
|
}
|
|
}
|
|
#endif
|
|
|
|
return mem;
|
|
}
|
|
|
|
#ifdef REAL_WITH_MEMORY_TRACKING
|
|
unsigned long mosquitto__memory_used(void)
|
|
{
|
|
return memcount;
|
|
}
|
|
|
|
unsigned long mosquitto__max_memory_used(void)
|
|
{
|
|
return max_memcount;
|
|
}
|
|
#endif
|
|
|
|
void *mosquitto__realloc(void *ptr, size_t size)
|
|
{
|
|
#ifdef REAL_WITH_MEMORY_TRACKING
|
|
if(mem_limit && memcount + size > mem_limit){
|
|
return NULL;
|
|
}
|
|
#endif
|
|
void *mem;
|
|
#ifdef REAL_WITH_MEMORY_TRACKING
|
|
if(ptr){
|
|
memcount -= malloc_usable_size(ptr);
|
|
}
|
|
#endif
|
|
mem = realloc(ptr, size);
|
|
|
|
#ifdef REAL_WITH_MEMORY_TRACKING
|
|
if(mem){
|
|
memcount += malloc_usable_size(mem);
|
|
if(memcount > max_memcount){
|
|
max_memcount = memcount;
|
|
}
|
|
}
|
|
#endif
|
|
|
|
return mem;
|
|
}
|
|
|
|
char *mosquitto__strdup(const char *s)
|
|
{
|
|
#ifdef REAL_WITH_MEMORY_TRACKING
|
|
if(mem_limit && memcount + strlen(s) > mem_limit){
|
|
return NULL;
|
|
}
|
|
#endif
|
|
char *str = strdup(s);
|
|
|
|
#ifdef REAL_WITH_MEMORY_TRACKING
|
|
if(str){
|
|
memcount += malloc_usable_size(str);
|
|
if(memcount > max_memcount){
|
|
max_memcount = memcount;
|
|
}
|
|
}
|
|
#endif
|
|
|
|
return str;
|
|
}
|
|
|