mem.c
上传用户:yisoukefu
上传日期:2020-08-09
资源大小:39506k
文件大小:13k
源码类别:

其他游戏

开发平台:

Visual C++

  1. /* crypto/mem.c */
  2. /* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
  3.  * All rights reserved.
  4.  *
  5.  * This package is an SSL implementation written
  6.  * by Eric Young (eay@cryptsoft.com).
  7.  * The implementation was written so as to conform with Netscapes SSL.
  8.  * 
  9.  * This library is free for commercial and non-commercial use as long as
  10.  * the following conditions are aheared to.  The following conditions
  11.  * apply to all code found in this distribution, be it the RC4, RSA,
  12.  * lhash, DES, etc., code; not just the SSL code.  The SSL documentation
  13.  * included with this distribution is covered by the same copyright terms
  14.  * except that the holder is Tim Hudson (tjh@cryptsoft.com).
  15.  * 
  16.  * Copyright remains Eric Young's, and as such any Copyright notices in
  17.  * the code are not to be removed.
  18.  * If this package is used in a product, Eric Young should be given attribution
  19.  * as the author of the parts of the library used.
  20.  * This can be in the form of a textual message at program startup or
  21.  * in documentation (online or textual) provided with the package.
  22.  * 
  23.  * Redistribution and use in source and binary forms, with or without
  24.  * modification, are permitted provided that the following conditions
  25.  * are met:
  26.  * 1. Redistributions of source code must retain the copyright
  27.  *    notice, this list of conditions and the following disclaimer.
  28.  * 2. Redistributions in binary form must reproduce the above copyright
  29.  *    notice, this list of conditions and the following disclaimer in the
  30.  *    documentation and/or other materials provided with the distribution.
  31.  * 3. All advertising materials mentioning features or use of this software
  32.  *    must display the following acknowledgement:
  33.  *    "This product includes cryptographic software written by
  34.  *     Eric Young (eay@cryptsoft.com)"
  35.  *    The word 'cryptographic' can be left out if the rouines from the library
  36.  *    being used are not cryptographic related :-).
  37.  * 4. If you include any Windows specific code (or a derivative thereof) from 
  38.  *    the apps directory (application code) you must include an acknowledgement:
  39.  *    "This product includes software written by Tim Hudson (tjh@cryptsoft.com)"
  40.  * 
  41.  * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND
  42.  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  43.  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  44.  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  45.  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  46.  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  47.  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  48.  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  49.  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  50.  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  51.  * SUCH DAMAGE.
  52.  * 
  53.  * The licence and distribution terms for any publically available version or
  54.  * derivative of this code cannot be changed.  i.e. this code cannot simply be
  55.  * copied and put under another distribution licence
  56.  * [including the GNU Public Licence.]
  57.  */
  58. #include <stdio.h>
  59. #include <stdlib.h>
  60. #include <openssl/crypto.h>
  61. #include "cryptlib.h"
  62. static int allow_customize = 1;      /* we provide flexible functions for */
  63. static int allow_customize_debug = 1;/* exchanging memory-related functions at
  64.                                       * run-time, but this must be done
  65.                                       * before any blocks are actually
  66.                                       * allocated; or we'll run into huge
  67.                                       * problems when malloc/free pairs
  68.                                       * don't match etc. */
  69. /* the following pointers may be changed as long as 'allow_customize' is set */
  70. static void *(*malloc_func)(size_t)         = malloc;
  71. static void *default_malloc_ex(size_t num, const char *file, int line)
  72. { return malloc_func(num); }
  73. static void *(*malloc_ex_func)(size_t, const char *file, int line)
  74.         = default_malloc_ex;
  75. static void *(*realloc_func)(void *, size_t)= realloc;
  76. static void *default_realloc_ex(void *str, size_t num,
  77.         const char *file, int line)
  78. { return realloc_func(str,num); }
  79. static void *(*realloc_ex_func)(void *, size_t, const char *file, int line)
  80.         = default_realloc_ex;
  81. static void (*free_func)(void *)            = free;
  82. static void *(*malloc_locked_func)(size_t)  = malloc;
  83. static void *default_malloc_locked_ex(size_t num, const char *file, int line)
  84. { return malloc_locked_func(num); }
  85. static void *(*malloc_locked_ex_func)(size_t, const char *file, int line)
  86.         = default_malloc_locked_ex;
  87. static void (*free_locked_func)(void *)     = free;
  88. /* may be changed as long as 'allow_customize_debug' is set */
  89. /* XXX use correct function pointer types */
  90. #ifdef CRYPTO_MDEBUG
  91. /* use default functions from mem_dbg.c */
  92. static void (*malloc_debug_func)(void *,int,const char *,int,int)
  93. = CRYPTO_dbg_malloc;
  94. static void (*realloc_debug_func)(void *,void *,int,const char *,int,int)
  95. = CRYPTO_dbg_realloc;
  96. static void (*free_debug_func)(void *,int) = CRYPTO_dbg_free;
  97. static void (*set_debug_options_func)(long) = CRYPTO_dbg_set_options;
  98. static long (*get_debug_options_func)(void) = CRYPTO_dbg_get_options;
  99. #else
  100. /* applications can use CRYPTO_malloc_debug_init() to select above case
  101.  * at run-time */
  102. static void (*malloc_debug_func)(void *,int,const char *,int,int) = NULL;
  103. static void (*realloc_debug_func)(void *,void *,int,const char *,int,int)
  104. = NULL;
  105. static void (*free_debug_func)(void *,int) = NULL;
  106. static void (*set_debug_options_func)(long) = NULL;
  107. static long (*get_debug_options_func)(void) = NULL;
  108. #endif
  109. int CRYPTO_set_mem_functions(void *(*m)(size_t), void *(*r)(void *, size_t),
  110. void (*f)(void *))
  111. {
  112. if (!allow_customize)
  113. return 0;
  114. if ((m == 0) || (r == 0) || (f == 0))
  115. return 0;
  116. malloc_func=m; malloc_ex_func=default_malloc_ex;
  117. realloc_func=r; realloc_ex_func=default_realloc_ex;
  118. free_func=f;
  119. malloc_locked_func=m; malloc_locked_ex_func=default_malloc_locked_ex;
  120. free_locked_func=f;
  121. return 1;
  122. }
  123. int CRYPTO_set_mem_ex_functions(
  124.         void *(*m)(size_t,const char *,int),
  125.         void *(*r)(void *, size_t,const char *,int),
  126. void (*f)(void *))
  127. {
  128. if (!allow_customize)
  129. return 0;
  130. if ((m == 0) || (r == 0) || (f == 0))
  131. return 0;
  132. malloc_func=0; malloc_ex_func=m;
  133. realloc_func=0; realloc_ex_func=r;
  134. free_func=f;
  135. malloc_locked_func=0; malloc_locked_ex_func=m;
  136. free_locked_func=f;
  137. return 1;
  138. }
  139. int CRYPTO_set_locked_mem_functions(void *(*m)(size_t), void (*f)(void *))
  140. {
  141. if (!allow_customize)
  142. return 0;
  143. if ((m == NULL) || (f == NULL))
  144. return 0;
  145. malloc_locked_func=m; malloc_locked_ex_func=default_malloc_locked_ex;
  146. free_locked_func=f;
  147. return 1;
  148. }
  149. int CRYPTO_set_locked_mem_ex_functions(
  150.         void *(*m)(size_t,const char *,int),
  151.         void (*f)(void *))
  152. {
  153. if (!allow_customize)
  154. return 0;
  155. if ((m == NULL) || (f == NULL))
  156. return 0;
  157. malloc_locked_func=0; malloc_locked_ex_func=m;
  158. free_func=f;
  159. return 1;
  160. }
  161. int CRYPTO_set_mem_debug_functions(void (*m)(void *,int,const char *,int,int),
  162.    void (*r)(void *,void *,int,const char *,int,int),
  163.    void (*f)(void *,int),
  164.    void (*so)(long),
  165.    long (*go)(void))
  166. {
  167. if (!allow_customize_debug)
  168. return 0;
  169. malloc_debug_func=m;
  170. realloc_debug_func=r;
  171. free_debug_func=f;
  172. set_debug_options_func=so;
  173. get_debug_options_func=go;
  174. return 1;
  175. }
  176. void CRYPTO_get_mem_functions(void *(**m)(size_t), void *(**r)(void *, size_t),
  177. void (**f)(void *))
  178. {
  179. if (m != NULL) *m = (malloc_ex_func == default_malloc_ex) ? 
  180.                      malloc_func : 0;
  181. if (r != NULL) *r = (realloc_ex_func == default_realloc_ex) ? 
  182.                      realloc_func : 0;
  183. if (f != NULL) *f=free_func;
  184. }
  185. void CRYPTO_get_mem_ex_functions(
  186.         void *(**m)(size_t,const char *,int),
  187.         void *(**r)(void *, size_t,const char *,int),
  188. void (**f)(void *))
  189. {
  190. if (m != NULL) *m = (malloc_ex_func != default_malloc_ex) ?
  191.                     malloc_ex_func : 0;
  192. if (r != NULL) *r = (realloc_ex_func != default_realloc_ex) ?
  193.                     realloc_ex_func : 0;
  194. if (f != NULL) *f=free_func;
  195. }
  196. void CRYPTO_get_locked_mem_functions(void *(**m)(size_t), void (**f)(void *))
  197. {
  198. if (m != NULL) *m = (malloc_locked_ex_func == default_malloc_locked_ex) ? 
  199.                      malloc_locked_func : 0;
  200. if (f != NULL) *f=free_locked_func;
  201. }
  202. void CRYPTO_get_locked_mem_ex_functions(
  203.         void *(**m)(size_t,const char *,int),
  204.         void (**f)(void *))
  205. {
  206. if (m != NULL) *m = (malloc_locked_ex_func != default_malloc_locked_ex) ?
  207.                     malloc_locked_ex_func : 0;
  208. if (f != NULL) *f=free_locked_func;
  209. }
  210. void CRYPTO_get_mem_debug_functions(void (**m)(void *,int,const char *,int,int),
  211.     void (**r)(void *,void *,int,const char *,int,int),
  212.     void (**f)(void *,int),
  213.     void (**so)(long),
  214.     long (**go)(void))
  215. {
  216. if (m != NULL) *m=malloc_debug_func;
  217. if (r != NULL) *r=realloc_debug_func;
  218. if (f != NULL) *f=free_debug_func;
  219. if (so != NULL) *so=set_debug_options_func;
  220. if (go != NULL) *go=get_debug_options_func;
  221. }
  222. void *CRYPTO_malloc_locked(int num, const char *file, int line)
  223. {
  224. void *ret = NULL;
  225. extern unsigned char cleanse_ctr;
  226. if (num <= 0) return NULL;
  227. allow_customize = 0;
  228. if (malloc_debug_func != NULL)
  229. {
  230. allow_customize_debug = 0;
  231. malloc_debug_func(NULL, num, file, line, 0);
  232. }
  233. ret = malloc_locked_ex_func(num,file,line);
  234. #ifdef LEVITTE_DEBUG_MEM
  235. fprintf(stderr, "LEVITTE_DEBUG_MEM:         > 0x%p (%d)n", ret, num);
  236. #endif
  237. if (malloc_debug_func != NULL)
  238. malloc_debug_func(ret, num, file, line, 1);
  239.         /* Create a dependency on the value of 'cleanse_ctr' so our memory
  240.          * sanitisation function can't be optimised out. NB: We only do
  241.          * this for >2Kb so the overhead doesn't bother us. */
  242.         if(ret && (num > 2048))
  243. ((unsigned char *)ret)[0] = cleanse_ctr;
  244. return ret;
  245. }
  246. void CRYPTO_free_locked(void *str)
  247. {
  248. if (free_debug_func != NULL)
  249. free_debug_func(str, 0);
  250. #ifdef LEVITTE_DEBUG_MEM
  251. fprintf(stderr, "LEVITTE_DEBUG_MEM:         < 0x%pn", str);
  252. #endif
  253. free_locked_func(str);
  254. if (free_debug_func != NULL)
  255. free_debug_func(NULL, 1);
  256. }
  257. void *CRYPTO_malloc(int num, const char *file, int line)
  258. {
  259. void *ret = NULL;
  260. extern unsigned char cleanse_ctr;
  261. if (num <= 0) return NULL;
  262. allow_customize = 0;
  263. if (malloc_debug_func != NULL)
  264. {
  265. allow_customize_debug = 0;
  266. malloc_debug_func(NULL, num, file, line, 0);
  267. }
  268. ret = malloc_ex_func(num,file,line);
  269. #ifdef LEVITTE_DEBUG_MEM
  270. fprintf(stderr, "LEVITTE_DEBUG_MEM:         > 0x%p (%d)n", ret, num);
  271. #endif
  272. if (malloc_debug_func != NULL)
  273. malloc_debug_func(ret, num, file, line, 1);
  274.         /* Create a dependency on the value of 'cleanse_ctr' so our memory
  275.          * sanitisation function can't be optimised out. NB: We only do
  276.          * this for >2Kb so the overhead doesn't bother us. */
  277.         if(ret && (num > 2048))
  278.                 ((unsigned char *)ret)[0] = cleanse_ctr;
  279. return ret;
  280. }
  281. void *CRYPTO_realloc(void *str, int num, const char *file, int line)
  282. {
  283. void *ret = NULL;
  284. if (str == NULL)
  285. return CRYPTO_malloc(num, file, line);
  286. if (num <= 0) return NULL;
  287. if (realloc_debug_func != NULL)
  288. realloc_debug_func(str, NULL, num, file, line, 0);
  289. ret = realloc_ex_func(str,num,file,line);
  290. #ifdef LEVITTE_DEBUG_MEM
  291. fprintf(stderr, "LEVITTE_DEBUG_MEM:         | 0x%p -> 0x%p (%d)n", str, ret, num);
  292. #endif
  293. if (realloc_debug_func != NULL)
  294. realloc_debug_func(str, ret, num, file, line, 1);
  295. return ret;
  296. }
  297. void *CRYPTO_realloc_clean(void *str, int old_len, int num, const char *file,
  298.    int line)
  299. {
  300. void *ret = NULL;
  301. if (str == NULL)
  302. return CRYPTO_malloc(num, file, line);
  303. if (num <= 0) return NULL;
  304. if (realloc_debug_func != NULL)
  305. realloc_debug_func(str, NULL, num, file, line, 0);
  306. ret=malloc_ex_func(num,file,line);
  307. if(ret)
  308. {
  309. memcpy(ret,str,old_len);
  310. OPENSSL_cleanse(str,old_len);
  311. free_func(str);
  312. }
  313. #ifdef LEVITTE_DEBUG_MEM
  314. fprintf(stderr,
  315. "LEVITTE_DEBUG_MEM:         | 0x%p -> 0x%p (%d)n",
  316. str, ret, num);
  317. #endif
  318. if (realloc_debug_func != NULL)
  319. realloc_debug_func(str, ret, num, file, line, 1);
  320. return ret;
  321. }
  322. void CRYPTO_free(void *str)
  323. {
  324. if (free_debug_func != NULL)
  325. free_debug_func(str, 0);
  326. #ifdef LEVITTE_DEBUG_MEM
  327. fprintf(stderr, "LEVITTE_DEBUG_MEM:         < 0x%pn", str);
  328. #endif
  329. free_func(str);
  330. if (free_debug_func != NULL)
  331. free_debug_func(NULL, 1);
  332. }
  333. void *CRYPTO_remalloc(void *a, int num, const char *file, int line)
  334. {
  335. if (a != NULL) OPENSSL_free(a);
  336. a=(char *)OPENSSL_malloc(num);
  337. return(a);
  338. }
  339. void CRYPTO_set_mem_debug_options(long bits)
  340. {
  341. if (set_debug_options_func != NULL)
  342. set_debug_options_func(bits);
  343. }
  344. long CRYPTO_get_mem_debug_options(void)
  345. {
  346. if (get_debug_options_func != NULL)
  347. return get_debug_options_func();
  348. return 0;
  349. }