1 /* libFLAC - Free Lossless Audio Codec library
2  * Copyright (C) 2001,2002,2003,2004,2005,2006,2007  Josh Coalson
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  *
8  * - Redistributions of source code must retain the above copyright
9  * notice, this list of conditions and the following disclaimer.
10  *
11  * - Redistributions in binary form must reproduce the above copyright
12  * notice, this list of conditions and the following disclaimer in the
13  * documentation and/or other materials provided with the distribution.
14  *
15  * - Neither the name of the Xiph.org Foundation nor the names of its
16  * contributors may be used to endorse or promote products derived from
17  * this software without specific prior written permission.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22  * A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR
23  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
24  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
25  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
26  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
27  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
28  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
29  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30  */
31 
32 #ifdef HAVE_CONFIG_H
33 #  include <config.h>
34 #endif
35 
36 #include "private/memory.h"
37 #include "FLAC/assert.h"
38 #include "share/alloc.h"
39 
FLAC__memory_alloc_aligned(size_t bytes,void ** aligned_address)40 void *FLAC__memory_alloc_aligned(size_t bytes, void **aligned_address)
41 {
42 	void *x;
43 
44 	FLAC__ASSERT(0 != aligned_address);
45 
46 #ifdef FLAC__ALIGN_MALLOC_DATA
47 	/* align on 32-byte (256-bit) boundary */
48 	x = safe_malloc_add_2op_(bytes, /*+*/31);
49 #ifdef SIZEOF_VOIDP
50 #if SIZEOF_VOIDP == 4
51 		/* could do  *aligned_address = x + ((unsigned) (32 - (((unsigned)x) & 31))) & 31; */
52 		*aligned_address = (void*)(((unsigned)x + 31) & -32);
53 #elif SIZEOF_VOIDP == 8
54 		*aligned_address = (void*)(((FLAC__uint64)x + 31) & (FLAC__uint64)(-((FLAC__int64)32)));
55 #else
56 # error  Unsupported sizeof(void*)
57 #endif
58 #else
59 	/* there's got to be a better way to do this right for all archs */
60 	if(sizeof(void*) == sizeof(unsigned))
61 		*aligned_address = (void*)(((unsigned)x + 31) & -32);
62 	else if(sizeof(void*) == sizeof(FLAC__uint64))
63 		*aligned_address = (void*)(((FLAC__uint64)x + 31) & (FLAC__uint64)(-((FLAC__int64)32)));
64 	else
65 		return 0;
66 #endif
67 #else
68 	x = safe_malloc_(bytes);
69 	*aligned_address = x;
70 #endif
71 	return x;
72 }
73 
FLAC__memory_alloc_aligned_int32_array(size_t elements,FLAC__int32 ** unaligned_pointer,FLAC__int32 ** aligned_pointer)74 FLAC__bool FLAC__memory_alloc_aligned_int32_array(size_t elements, FLAC__int32 **unaligned_pointer, FLAC__int32 **aligned_pointer)
75 {
76 	FLAC__int32 *pu; /* unaligned pointer */
77 	union { /* union needed to comply with C99 pointer aliasing rules */
78 		FLAC__int32 *pa; /* aligned pointer */
79 		void        *pv; /* aligned pointer alias */
80 	} u;
81 
82 	FLAC__ASSERT(elements > 0);
83 	FLAC__ASSERT(0 != unaligned_pointer);
84 	FLAC__ASSERT(0 != aligned_pointer);
85 	FLAC__ASSERT(unaligned_pointer != aligned_pointer);
86 
87 	if(elements > SIZE_MAX / sizeof(*pu)) /* overflow check */
88 		return false;
89 
90 	pu = (FLAC__int32*)FLAC__memory_alloc_aligned(sizeof(*pu) * elements, &u.pv);
91 	if(0 == pu) {
92 		return false;
93 	}
94 	else {
95 		if(*unaligned_pointer != 0)
96 			free(*unaligned_pointer);
97 		*unaligned_pointer = pu;
98 		*aligned_pointer = u.pa;
99 		return true;
100 	}
101 }
102 
103 
FLAC__memory_alloc_aligned_uint32_array(size_t elements,FLAC__uint32 ** unaligned_pointer,FLAC__uint32 ** aligned_pointer)104 FLAC__bool FLAC__memory_alloc_aligned_uint32_array(size_t elements, FLAC__uint32 **unaligned_pointer, FLAC__uint32 **aligned_pointer)
105 {
106 	FLAC__uint32 *pu; /* unaligned pointer */
107 	union { /* union needed to comply with C99 pointer aliasing rules */
108 		FLAC__uint32 *pa; /* aligned pointer */
109 		void         *pv; /* aligned pointer alias */
110 	} u;
111 
112 	FLAC__ASSERT(elements > 0);
113 	FLAC__ASSERT(0 != unaligned_pointer);
114 	FLAC__ASSERT(0 != aligned_pointer);
115 	FLAC__ASSERT(unaligned_pointer != aligned_pointer);
116 
117 	if(elements > SIZE_MAX / sizeof(*pu)) /* overflow check */
118 		return false;
119 
120 	pu = (FLAC__uint32*)FLAC__memory_alloc_aligned(sizeof(*pu) * elements, &u.pv);
121 	if(0 == pu) {
122 		return false;
123 	}
124 	else {
125 		if(*unaligned_pointer != 0)
126 			free(*unaligned_pointer);
127 		*unaligned_pointer = pu;
128 		*aligned_pointer = u.pa;
129 		return true;
130 	}
131 }
132 
FLAC__memory_alloc_aligned_uint64_array(size_t elements,FLAC__uint64 ** unaligned_pointer,FLAC__uint64 ** aligned_pointer)133 FLAC__bool FLAC__memory_alloc_aligned_uint64_array(size_t elements, FLAC__uint64 **unaligned_pointer, FLAC__uint64 **aligned_pointer)
134 {
135 	FLAC__uint64 *pu; /* unaligned pointer */
136 	union { /* union needed to comply with C99 pointer aliasing rules */
137 		FLAC__uint64 *pa; /* aligned pointer */
138 		void         *pv; /* aligned pointer alias */
139 	} u;
140 
141 	FLAC__ASSERT(elements > 0);
142 	FLAC__ASSERT(0 != unaligned_pointer);
143 	FLAC__ASSERT(0 != aligned_pointer);
144 	FLAC__ASSERT(unaligned_pointer != aligned_pointer);
145 
146 	if(elements > SIZE_MAX / sizeof(*pu)) /* overflow check */
147 		return false;
148 
149 	pu = (FLAC__uint64*)FLAC__memory_alloc_aligned(sizeof(*pu) * elements, &u.pv);
150 	if(0 == pu) {
151 		return false;
152 	}
153 	else {
154 		if(*unaligned_pointer != 0)
155 			free(*unaligned_pointer);
156 		*unaligned_pointer = pu;
157 		*aligned_pointer = u.pa;
158 		return true;
159 	}
160 }
161 
FLAC__memory_alloc_aligned_unsigned_array(size_t elements,unsigned ** unaligned_pointer,unsigned ** aligned_pointer)162 FLAC__bool FLAC__memory_alloc_aligned_unsigned_array(size_t elements, unsigned **unaligned_pointer, unsigned **aligned_pointer)
163 {
164 	unsigned *pu; /* unaligned pointer */
165 	union { /* union needed to comply with C99 pointer aliasing rules */
166 		unsigned *pa; /* aligned pointer */
167 		void     *pv; /* aligned pointer alias */
168 	} u;
169 
170 	FLAC__ASSERT(elements > 0);
171 	FLAC__ASSERT(0 != unaligned_pointer);
172 	FLAC__ASSERT(0 != aligned_pointer);
173 	FLAC__ASSERT(unaligned_pointer != aligned_pointer);
174 
175 	if(elements > SIZE_MAX / sizeof(*pu)) /* overflow check */
176 		return false;
177 
178 	pu = (unsigned*)FLAC__memory_alloc_aligned(sizeof(*pu) * elements, &u.pv);
179 	if(0 == pu) {
180 		return false;
181 	}
182 	else {
183 		if(*unaligned_pointer != 0)
184 			free(*unaligned_pointer);
185 		*unaligned_pointer = pu;
186 		*aligned_pointer = u.pa;
187 		return true;
188 	}
189 }
190 
191 #ifndef FLAC__INTEGER_ONLY_LIBRARY
192 
FLAC__memory_alloc_aligned_real_array(size_t elements,FLAC__real ** unaligned_pointer,FLAC__real ** aligned_pointer)193 FLAC__bool FLAC__memory_alloc_aligned_real_array(size_t elements, FLAC__real **unaligned_pointer, FLAC__real **aligned_pointer)
194 {
195 	FLAC__real *pu; /* unaligned pointer */
196 	union { /* union needed to comply with C99 pointer aliasing rules */
197 		FLAC__real *pa; /* aligned pointer */
198 		void       *pv; /* aligned pointer alias */
199 	} u;
200 
201 	FLAC__ASSERT(elements > 0);
202 	FLAC__ASSERT(0 != unaligned_pointer);
203 	FLAC__ASSERT(0 != aligned_pointer);
204 	FLAC__ASSERT(unaligned_pointer != aligned_pointer);
205 
206 	if(elements > SIZE_MAX / sizeof(*pu)) /* overflow check */
207 		return false;
208 
209 	pu = (FLAC__real*)FLAC__memory_alloc_aligned(sizeof(*pu) * elements, &u.pv);
210 	if(0 == pu) {
211 		return false;
212 	}
213 	else {
214 		if(*unaligned_pointer != 0)
215 			free(*unaligned_pointer);
216 		*unaligned_pointer = pu;
217 		*aligned_pointer = u.pa;
218 		return true;
219 	}
220 }
221 
222 #endif
223