1 |
|
#include <stdlib.h> |
2 |
|
#include <stdio.h> |
3 |
|
#include "mem_align.h" |
4 |
|
|
5 |
|
void * |
6 |
|
xvid_malloc(size_t size, |
7 |
|
uint8_t alignment) |
8 |
|
{ |
9 |
|
uint8_t *mem_ptr; |
10 |
|
|
11 |
|
if (!alignment) { |
12 |
|
|
13 |
|
/* We have not to satisfy any alignment */ |
14 |
|
if ((mem_ptr = (uint8_t *) malloc(size + 1)) != NULL) { |
15 |
|
|
16 |
|
/* Store (mem_ptr - "real allocated memory") in *(mem_ptr-1) */ |
17 |
|
*mem_ptr = 0; |
18 |
|
|
19 |
|
/* Return the mem_ptr pointer */ |
20 |
|
return (void *) mem_ptr++; |
21 |
|
|
22 |
|
} |
23 |
|
|
24 |
|
} else { |
25 |
|
uint8_t *tmp; |
26 |
|
|
27 |
|
/* |
28 |
|
* Allocate the required size memory + alignment so we |
29 |
|
* can realign the data if necessary |
30 |
|
*/ |
31 |
|
|
32 |
|
if ((tmp = (uint8_t *) malloc(size + alignment)) != NULL) { |
33 |
|
|
34 |
|
/* Align the tmp pointer */ |
35 |
|
mem_ptr = |
36 |
|
(uint8_t *) ((uint32_t) (tmp + alignment - 1) & |
37 |
|
(~(uint32_t) (alignment - 1))); |
38 |
|
|
39 |
|
/* |
40 |
|
* Special case where malloc have already satisfied the alignment |
41 |
|
* We must add alignment to mem_ptr because we must store |
42 |
|
* (mem_ptr - tmp) in *(mem_ptr-1) |
43 |
|
* If we do not add alignment to mem_ptr then *(mem_ptr-1) points |
44 |
|
* to a forbidden memory space |
45 |
|
*/ |
46 |
|
if (mem_ptr == tmp) |
47 |
|
mem_ptr += alignment; |
48 |
|
|
49 |
|
/* |
50 |
|
* (mem_ptr - tmp) is stored in *(mem_ptr-1) so we are able to retrieve |
51 |
|
* the real malloc block allocated and free it in xvid_free |
52 |
|
*/ |
53 |
|
*(mem_ptr - 1) = (uint8_t) (mem_ptr - tmp); |
54 |
|
|
55 |
|
/* Return the aligned pointer */ |
56 |
|
return (void *) mem_ptr; |
57 |
|
|
58 |
|
} |
59 |
|
} |
60 |
|
|
61 |
|
return NULL; |
62 |
|
|
63 |
|
} |
64 |
|
|
65 |
|
void |
66 |
|
xvid_free(void *mem_ptr) |
67 |
|
{ |
68 |
|
|
69 |
|
/* *(mem_ptr - 1) give us the offset to the real malloc block */ |
70 |
|
if (mem_ptr) |
71 |
|
free((uint8_t *) mem_ptr - *((uint8_t *) mem_ptr - 1)); |
72 |
|
|
73 |
|
} |