+/* alloc.c
+ *
+ * Copyright (C) 2000-2006 Erik Andersen <andersen@uclibc.org>
+ *
+ * Licensed under the LGPL v2.1, see the file COPYING.LIB in this tarball.
+ */
+/*
+ * Parts of the memalign code were stolen from malloc-930716.
+ */
+
+#include <features.h>
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
-#include <unistd.h>
+#include <errno.h>
#include <sys/mman.h>
+#include <malloc.h>
+extern int weak_function __libc_free_aligned(void *ptr) attribute_hidden;
-#ifdef L_calloc_dbg
-
-void *calloc_dbg(size_t num, size_t size, char *function, char *file,
- int line)
-{
- void *ptr;
-
- fprintf(stderr, "calloc of %d bytes at %s @%s:%d = ", num * size,
- function, file, line);
- ptr = calloc(num, size);
- fprintf(stderr, "%p\n", ptr);
- return ptr;
-}
-
-#endif
-
-#ifdef L_malloc_dbg
-
-void *malloc_dbg(size_t len, char *function, char *file, int line)
+#ifdef L_malloc
+void *malloc(size_t size)
{
void *result;
- fprintf(stderr, "malloc of %d bytes at %s @%s:%d = ", len, function,
- file, line);
- result = malloc(len);
- fprintf(stderr, "%p\n", result);
- return result;
-}
-
+ if (unlikely(size == 0)) {
+#if defined(__MALLOC_GLIBC_COMPAT__)
+ size++;
+#else
+ /* Some programs will call malloc (0). Lets be strict and return NULL */
+ __set_errno(ENOMEM);
+ return NULL;
#endif
+ }
-#ifdef L_free_dbg
+#ifdef __ARCH_USE_MMU__
+# define MMAP_FLAGS MAP_PRIVATE | MAP_ANONYMOUS
+#else
+# define MMAP_FLAGS MAP_SHARED | MAP_ANONYMOUS | MAP_UNINITIALIZED
+#endif
-void free_dbg(void *ptr, char *function, char *file, int line)
-{
- fprintf(stderr, "free of %p at %s @%s:%d\n", ptr, function, file,
- line);
- free(ptr);
+ result = mmap((void *) 0, size + sizeof(size_t), PROT_READ | PROT_WRITE,
+ MMAP_FLAGS, 0, 0);
+ if (result == MAP_FAILED) {
+ __set_errno(ENOMEM);
+ return 0;
+ }
+ * (size_t *) result = size;
+ return(result + sizeof(size_t));
}
-
#endif
-
#ifdef L_calloc
-
-void *calloc(size_t num, size_t size)
+void * calloc(size_t nmemb, size_t lsize)
{
- void *ptr = malloc(num * size);
+ void *result;
+ size_t size=lsize * nmemb;
- if (ptr)
- memset(ptr, 0, num * size);
- return ptr;
-}
+ /* guard vs integer overflow, but allow nmemb
+ * to fall through and call malloc(0) */
+ if (nmemb && lsize != (size / nmemb)) {
+ __set_errno(ENOMEM);
+ return NULL;
+ }
+ result = malloc(size);
+#ifndef __ARCH_USE_MMU__
+ /* mmap'd with MAP_UNINITIALIZED, we have to blank memory ourselves */
+ if (result != NULL) {
+ memset(result, 0, size);
+ }
+#endif
+ return result;
+}
#endif
-#ifdef L_malloc
-
-void *malloc(size_t len)
+#ifdef L_realloc
+void *realloc(void *ptr, size_t size)
{
- void *result = mmap((void *) 0, len, PROT_READ | PROT_WRITE,
- //MAP_SHARED | MAP_ANONYMOUS, 0, 0);
- MAP_PRIVATE | MAP_ANONYMOUS, 0, 0);
+ void *newptr = NULL;
- if (result == (void *) -1)
- return 0;
+ if (!ptr)
+ return malloc(size);
+ if (!size) {
+ free(ptr);
+ return malloc(0);
+ }
- return result;
+ newptr = malloc(size);
+ if (newptr) {
+ size_t old_size = *((size_t *) (ptr - sizeof(size_t)));
+ memcpy(newptr, ptr, (old_size < size ? old_size : size));
+ free(ptr);
+ }
+ return newptr;
}
-
#endif
#ifdef L_free
-
void free(void *ptr)
{
- munmap(ptr, 0);
+ if (unlikely(ptr == NULL))
+ return;
+ if (unlikely(__libc_free_aligned != NULL)) {
+ if (__libc_free_aligned(ptr))
+ return;
+ }
+ ptr -= sizeof(size_t);
+ munmap(ptr, * (size_t *) ptr + sizeof(size_t));
}
-
#endif
-#ifdef L_realloc
+#ifdef L_memalign
-void *realloc(void *ptr, size_t size)
+#include <bits/uClibc_mutex.h>
+__UCLIBC_MUTEX_INIT(__malloc_lock, PTHREAD_RECURSIVE_MUTEX_INITIALIZER_NP);
+#define __MALLOC_LOCK __UCLIBC_MUTEX_LOCK(__malloc_lock)
+#define __MALLOC_UNLOCK __UCLIBC_MUTEX_UNLOCK(__malloc_lock)
+
+/* List of blocks allocated with memalign or valloc */
+struct alignlist
{
- void *newptr = NULL;
+ struct alignlist *next;
+ __ptr_t aligned; /* The address that memaligned returned. */
+ __ptr_t exact; /* The address that malloc returned. */
+};
+static struct alignlist *_aligned_blocks;
+
+/* Return memory to the heap. */
+int __libc_free_aligned(void *ptr)
+{
+ struct alignlist *l;
+
+ if (ptr == NULL)
+ return 0;
- if (size > 0) {
- newptr = malloc(size);
- if (newptr && ptr)
- memcpy(newptr, ptr, size);
+ __MALLOC_LOCK;
+ for (l = _aligned_blocks; l != NULL; l = l->next) {
+ if (l->aligned == ptr) {
+ /* Mark the block as free */
+ l->aligned = NULL;
+ ptr = l->exact;
+ ptr -= sizeof(size_t);
+ munmap(ptr, * (size_t *) ptr + sizeof(size_t));
+ return 1;
+ }
}
- if (ptr)
- free(ptr);
- return newptr;
+ __MALLOC_UNLOCK;
+ return 0;
}
+void * memalign (size_t alignment, size_t size)
+{
+ void * result;
+ unsigned long int adj;
+
+ result = malloc (size + alignment - 1);
+ if (result == NULL)
+ return NULL;
+
+ adj = (unsigned long int) ((unsigned long int) ((char *) result - (char *) NULL)) % alignment;
+ if (adj != 0) {
+ struct alignlist *l;
+ __MALLOC_LOCK;
+ for (l = _aligned_blocks; l != NULL; l = l->next)
+ if (l->aligned == NULL)
+ /* This slot is free. Use it. */
+ break;
+ if (l == NULL) {
+ l = (struct alignlist *) malloc (sizeof (struct alignlist));
+ if (l == NULL) {
+ free(result);
+ result = NULL;
+ goto DONE;
+ }
+ l->next = _aligned_blocks;
+ _aligned_blocks = l;
+ }
+ l->exact = result;
+ result = l->aligned = (char *) result + alignment - adj;
+DONE:
+ __MALLOC_UNLOCK;
+ }
+ return result;
+}
+libc_hidden_def(memalign)
#endif
OSDN Git Service
