/src/lzma-fuzz/common-alloc.h
Line  | Count  | Source  | 
1  |  | /**  | 
2  |  |  *  | 
3  |  |  * @copyright Copyright (c) 2019 Joachim Bauch <mail@joachim-bauch.de>  | 
4  |  |  *  | 
5  |  |  * @license GNU GPL version 3 or any later version  | 
6  |  |  *  | 
7  |  |  * This program is free software: you can redistribute it and/or modify  | 
8  |  |  * it under the terms of the GNU General Public License as published by  | 
9  |  |  * the Free Software Foundation, either version 3 of the License, or  | 
10  |  |  * (at your option) any later version.  | 
11  |  |  *  | 
12  |  |  * This program is distributed in the hope that it will be useful,  | 
13  |  |  * but WITHOUT ANY WARRANTY; without even the implied warranty of  | 
14  |  |  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the  | 
15  |  |  * GNU General Public License for more details.  | 
16  |  |  *  | 
17  |  |  * You should have received a copy of the GNU General Public License  | 
18  |  |  * along with this program.  If not, see <https://www.gnu.org/licenses/>.  | 
19  |  |  *  | 
20  |  |  */  | 
21  |  |  | 
22  |  | #include <stdint.h>  | 
23  |  | #include <stdlib.h>  | 
24  |  |  | 
25  |  | #include "7zAlloc.h"  | 
26  |  | #include "7zTypes.h"  | 
27  |  |  | 
28  |  | #if defined(FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION)  | 
29  |  |  | 
30  |  | // Arbitrary limit of 2 GByte for memory allocations. Used to prevent the  | 
31  |  | // MemorySanitizer from aborting with errors like "requested allocation size  | 
32  |  | // 0xffffffffffffffff exceeds maximum supported size of 0x200000000".  | 
33  |  | static const size_t kMaxAllowedMemory = 2 * 1024 * 1024 * 1024L;  | 
34  |  |  | 
35  | 111k  | static void *LzmaAlloc(ISzAllocPtr p, size_t size) { | 
36  | 111k  |   if (size > kMaxAllowedMemory) { | 
37  | 0  |     return nullptr;  | 
38  | 0  |   }  | 
39  |  |  | 
40  | 111k  |   return SzAlloc(nullptr, size);  | 
41  | 111k  | }  | 
42  |  |  | 
43  | 149k  | static void LzmaFree(ISzAllocPtr p, void *address) { | 
44  | 149k  |   SzFree(nullptr, address);  | 
45  | 149k  | }  | 
46  |  |  | 
47  |  | static ISzAlloc CommonAlloc = {LzmaAlloc, LzmaFree}; | 
48  |  |  | 
49  |  | #else  | 
50  |  |  | 
51  |  | #include "7zAlloc.h"  | 
52  |  |  | 
53  |  | static ISzAlloc CommonAlloc = {SzAlloc, SzFree}; | 
54  |  |  | 
55  |  | #endif  |