mirror of
https://github.com/MariaDB/server.git
synced 2025-04-06 15:25:32 +02:00
23debf214f
In commit 0f56e21efa
only one pfs_malloc() stub was fixed to return aligned memory.
Also, the MSVC _aligned_malloc() pairs with _aligned_free().
106 lines
2.9 KiB
C
106 lines
2.9 KiB
C
/* Copyright (c) 2008, 2023, Oracle and/or its affiliates.
|
|
Copyright (c) 2022, MariaDB Corporation.
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License, version 2.0,
|
|
as published by the Free Software Foundation.
|
|
|
|
This program is also distributed with certain software (including
|
|
but not limited to OpenSSL) that is licensed under separate terms,
|
|
as designated in a particular file or component or in included license
|
|
documentation. The authors of MySQL hereby grant you an additional
|
|
permission to link the program and your derivative works with the
|
|
separately licensed software that they have included with MySQL.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License, version 2.0, for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program; if not, write to the Free Software Foundation,
|
|
51 Franklin Street, Fifth Floor, Boston, MA 02110-1335 USA */
|
|
|
|
#include <my_global.h>
|
|
#include <my_sys.h>
|
|
#include <pfs_global.h>
|
|
#include <string.h>
|
|
#ifdef HAVE_MEMALIGN
|
|
# include <malloc.h>
|
|
#endif
|
|
|
|
bool pfs_initialized= false;
|
|
size_t pfs_allocated_memory_size= 0;
|
|
size_t pfs_allocated_memory_count= 0;
|
|
|
|
bool stub_alloc_always_fails= true;
|
|
int stub_alloc_fails_after_count= 0;
|
|
|
|
void *pfs_malloc(PFS_builtin_memory_class *klass, size_t size, myf)
|
|
{
|
|
/*
|
|
Catch non initialized sizing parameter in the unit tests.
|
|
*/
|
|
assert(size <= 100*1024*1024);
|
|
|
|
if (stub_alloc_always_fails)
|
|
return NULL;
|
|
|
|
if (--stub_alloc_fails_after_count <= 0)
|
|
return NULL;
|
|
|
|
#ifndef PFS_ALIGNEMENT
|
|
void *ptr= malloc(size);
|
|
#elif defined HAVE_MEMALIGN
|
|
void *ptr= memalign(PFS_ALIGNEMENT, size);
|
|
#elif defined HAVE_ALIGNED_MALLOC
|
|
void *ptr= _aligned_malloc(size, PFS_ALIGNEMENT);
|
|
#else
|
|
void *ptr;
|
|
if (posix_memalign(&ptr, PFS_ALIGNEMENT, size))
|
|
ptr= NULL;
|
|
#endif
|
|
if (ptr != NULL)
|
|
memset(ptr, 0, size);
|
|
return ptr;
|
|
}
|
|
|
|
void pfs_free(PFS_builtin_memory_class *, size_t, void *ptr)
|
|
{
|
|
#ifdef HAVE_ALIGNED_MALLOC
|
|
_aligned_free(ptr);
|
|
#else
|
|
free(ptr);
|
|
#endif
|
|
}
|
|
|
|
void *pfs_malloc_array(PFS_builtin_memory_class *klass, size_t n, size_t size, myf flags)
|
|
{
|
|
size_t array_size= n * size;
|
|
/* Check for overflow before allocating. */
|
|
if (is_overflow(array_size, n, size))
|
|
return NULL;
|
|
return pfs_malloc(klass, array_size, flags);
|
|
}
|
|
|
|
void pfs_free_array(PFS_builtin_memory_class *klass, size_t n, size_t size, void *ptr)
|
|
{
|
|
if (ptr == NULL)
|
|
return;
|
|
size_t array_size= n * size;
|
|
return pfs_free(klass, array_size, ptr);
|
|
}
|
|
|
|
bool is_overflow(size_t product, size_t n1, size_t n2)
|
|
{
|
|
if (n1 != 0 && (product / n1 != n2))
|
|
return true;
|
|
else
|
|
return false;
|
|
}
|
|
|
|
void pfs_print_error(const char *format, ...)
|
|
{
|
|
}
|
|
|
|
|