1 /* 2 * nghttp2 - HTTP/2 C Library 3 * 4 * Copyright (c) 2012 Tatsuhiro Tsujikawa 5 * 6 * Permission is hereby granted, free of charge, to any person obtaining 7 * a copy of this software and associated documentation files (the 8 * "Software"), to deal in the Software without restriction, including 9 * without limitation the rights to use, copy, modify, merge, publish, 10 * distribute, sublicense, and/or sell copies of the Software, and to 11 * permit persons to whom the Software is furnished to do so, subject to 12 * the following conditions: 13 * 14 * The above copyright notice and this permission notice shall be 15 * included in all copies or substantial portions of the Software. 16 * 17 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, 18 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF 19 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND 20 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE 21 * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION 22 * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION 23 * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. 24 */ 25 #ifndef MALLOC_WRAPPER_H 26 #define MALLOC_WRAPPER_H 27 28 #ifdef HAVE_CONFIG_H 29 # include <config.h> 30 #endif /* HAVE_CONFIG_H */ 31 32 #include <stdlib.h> 33 34 #include "nghttp2_mem.h" 35 36 /* Global variables to control the behavior of malloc() */ 37 38 /* If nonzero, malloc failure mode is on */ 39 extern int nghttp2_failmalloc; 40 /* If nghttp2_failstart <= nghttp2_nmalloc and nghttp2_failmalloc is 41 nonzero, malloc() fails. */ 42 extern int nghttp2_failstart; 43 /* If nonzero, nghttp2_nmalloc is incremented if malloc() succeeds. */ 44 extern int nghttp2_countmalloc; 45 /* The number of successful invocation of malloc(). This value is only 46 incremented if nghttp2_nmalloc is nonzero. */ 47 extern int nghttp2_nmalloc; 48 49 /* Returns pointer to nghttp2_mem, which, when dereferenced, contains 50 specifically instrumented memory allocators for failmalloc 51 tests. */ 52 nghttp2_mem *nghttp2_mem_fm(void); 53 54 /* Copies nghttp2_failmalloc and nghttp2_countmalloc to statically 55 allocated space and sets 0 to them. This will effectively make 56 malloc() work like normal malloc(). This is useful when you want to 57 disable malloc() failure mode temporarily. */ 58 void nghttp2_failmalloc_pause(void); 59 60 /* Restores the values of nghttp2_failmalloc and nghttp2_countmalloc 61 with the values saved by the previous 62 nghttp2_failmalloc_pause(). */ 63 void nghttp2_failmalloc_unpause(void); 64 65 #endif /* MALLOC_WRAPPER_H */ 66