summaryrefslogtreecommitdiffstats
path: root/src/old-stats/global-memory.c
blob: dc04bfb1af2b8502ca69de9b16be8de262054a70 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
/* Copyright (c) 2011-2018 Dovecot authors, see the included COPYING file */

#include "lib.h"
#include "mail-command.h"
#include "mail-session.h"
#include "mail-user.h"
#include "mail-domain.h"
#include "mail-ip.h"
#include "stats-settings.h"
#include "global-memory.h"

size_t global_used_memory = 0;

static bool global_memory_free_something(void)
{
	size_t orig_used_memory = global_used_memory;

	mail_commands_free_memory();
	if (global_used_memory > stats_settings->memory_limit)
		mail_sessions_free_memory();
	if (global_used_memory > stats_settings->memory_limit)
		mail_users_free_memory();
	if (global_used_memory > stats_settings->memory_limit)
		mail_ips_free_memory();
	if (global_used_memory > stats_settings->memory_limit)
		mail_domains_free_memory();

	return global_used_memory < orig_used_memory;
}

void global_memory_alloc(size_t size)
{
	i_assert(size < SIZE_MAX - global_used_memory);
	global_used_memory += size;

	while (global_used_memory > stats_settings->memory_limit) {
		if (!global_memory_free_something())
			break;
	}
}

void global_memory_free(size_t size)
{
	i_assert(size <= global_used_memory);
	global_used_memory -= size;
}