/ Hex Artifact Content
Login

Artifact 4e63c764345ab1df59d4f13a77980c6f3643798210b10d6cdbd785b4b888fda5:


0000: 0a 23 69 6e 63 6c 75 64 65 20 3c 73 74 64 69 6f  .#include <stdio
0010: 2e 68 3e 0a 23 69 6e 63 6c 75 64 65 20 3c 61 73  .h>.#include <as
0020: 73 65 72 74 2e 68 3e 0a 23 69 6e 63 6c 75 64 65  sert.h>.#include
0030: 20 3c 73 74 72 69 6e 67 2e 68 3e 0a 0a 23 64 65   <string.h>..#de
0040: 66 69 6e 65 20 41 72 72 61 79 53 69 7a 65 28 78  fine ArraySize(x
0050: 29 20 28 28 69 6e 74 29 28 73 69 7a 65 6f 66 28  ) ((int)(sizeof(
0060: 78 29 20 2f 20 73 69 7a 65 6f 66 28 28 78 29 5b  x) / sizeof((x)[
0070: 30 5d 29 29 29 0a 0a 23 64 65 66 69 6e 65 20 4d  0])))..#define M
0080: 49 4e 28 78 2c 79 29 20 28 28 78 29 3c 28 79 29  IN(x,y) ((x)<(y)
0090: 20 3f 20 28 78 29 20 3a 20 28 79 29 29 0a 0a 74   ? (x) : (y))..t
00a0: 79 70 65 64 65 66 20 75 6e 73 69 67 6e 65 64 20  ypedef unsigned 
00b0: 69 6e 74 20 20 75 33 32 3b 0a 74 79 70 65 64 65  int  u32;.typede
00c0: 66 20 75 6e 73 69 67 6e 65 64 20 63 68 61 72 20  f unsigned char 
00d0: 75 38 3b 0a 74 79 70 65 64 65 66 20 6c 6f 6e 67  u8;.typedef long
00e0: 20 6c 6f 6e 67 20 69 6e 74 20 69 36 34 3b 0a 74   long int i64;.t
00f0: 79 70 65 64 65 66 20 75 6e 73 69 67 6e 65 64 20  ypedef unsigned 
0100: 6c 6f 6e 67 20 6c 6f 6e 67 20 69 6e 74 20 75 36  long long int u6
0110: 34 3b 0a 0a 23 69 66 20 64 65 66 69 6e 65 64 28  4;..#if defined(
0120: 5f 5f 47 4c 49 42 43 5f 5f 29 20 26 26 20 64 65  __GLIBC__) && de
0130: 66 69 6e 65 64 28 4c 53 4d 5f 44 45 42 55 47 5f  fined(LSM_DEBUG_
0140: 4d 45 4d 29 0a 20 20 65 78 74 65 72 6e 20 69 6e  MEM).  extern in
0150: 74 20 62 61 63 6b 74 72 61 63 65 28 76 6f 69 64  t backtrace(void
0160: 2a 2a 2c 69 6e 74 29 3b 0a 20 20 65 78 74 65 72  **,int);.  exter
0170: 6e 20 76 6f 69 64 20 62 61 63 6b 74 72 61 63 65  n void backtrace
0180: 5f 73 79 6d 62 6f 6c 73 5f 66 64 28 76 6f 69 64  _symbols_fd(void
0190: 2a 63 6f 6e 73 74 2a 2c 69 6e 74 2c 69 6e 74 29  *const*,int,int)
01a0: 3b 0a 23 20 64 65 66 69 6e 65 20 54 4d 5f 42 41  ;.# define TM_BA
01b0: 43 4b 54 52 41 43 45 20 31 32 0a 23 65 6c 73 65  CKTRACE 12.#else
01c0: 0a 23 20 64 65 66 69 6e 65 20 62 61 63 6b 74 72  .# define backtr
01d0: 61 63 65 28 41 2c 42 29 20 31 0a 23 20 64 65 66  ace(A,B) 1.# def
01e0: 69 6e 65 20 62 61 63 6b 74 72 61 63 65 5f 73 79  ine backtrace_sy
01f0: 6d 62 6f 6c 73 5f 66 64 28 41 2c 42 2c 43 29 0a  mbols_fd(A,B,C).
0200: 23 65 6e 64 69 66 0a 0a 0a 74 79 70 65 64 65 66  #endif...typedef
0210: 20 73 74 72 75 63 74 20 54 6d 42 6c 6f 63 6b 48   struct TmBlockH
0220: 64 72 20 54 6d 42 6c 6f 63 6b 48 64 72 3b 0a 74  dr TmBlockHdr;.t
0230: 79 70 65 64 65 66 20 73 74 72 75 63 74 20 54 6d  ypedef struct Tm
0240: 41 67 67 20 54 6d 41 67 67 3b 0a 74 79 70 65 64  Agg TmAgg;.typed
0250: 65 66 20 73 74 72 75 63 74 20 54 6d 47 6c 6f 62  ef struct TmGlob
0260: 61 6c 20 54 6d 47 6c 6f 62 61 6c 3b 0a 0a 73 74  al TmGlobal;..st
0270: 72 75 63 74 20 54 6d 47 6c 6f 62 61 6c 20 7b 0a  ruct TmGlobal {.
0280: 20 20 2f 2a 20 4c 69 6e 6b 65 64 20 6c 69 73 74    /* Linked list
0290: 20 6f 66 20 61 6c 6c 20 63 75 72 72 65 6e 74 6c   of all currentl
02a0: 79 20 6f 75 74 73 74 61 6e 64 69 6e 67 20 61 6c  y outstanding al
02b0: 6c 6f 63 61 74 69 6f 6e 73 2e 20 41 6e 64 20 61  locations. And a
02c0: 20 74 61 62 6c 65 20 6f 66 0a 20 20 2a 2a 20 61   table of.  ** a
02d0: 6c 6c 20 61 6c 6c 6f 63 61 74 69 6f 6e 73 2c 20  ll allocations, 
02e0: 70 61 73 74 20 61 6e 64 20 70 72 65 73 65 6e 74  past and present
02f0: 2c 20 69 6e 64 65 78 65 64 20 62 79 20 62 61 63  , indexed by bac
0300: 6b 74 72 61 63 65 28 29 20 69 6e 66 6f 2e 20 20  ktrace() info.  
0310: 2a 2f 0a 20 20 54 6d 42 6c 6f 63 6b 48 64 72 20  */.  TmBlockHdr 
0320: 2a 70 46 69 72 73 74 3b 0a 23 69 66 64 65 66 20  *pFirst;.#ifdef 
0330: 54 4d 5f 42 41 43 4b 54 52 41 43 45 0a 20 20 54  TM_BACKTRACE.  T
0340: 6d 41 67 67 20 2a 61 48 61 73 68 5b 31 30 30 30  mAgg *aHash[1000
0350: 30 5d 3b 0a 23 65 6e 64 69 66 0a 0a 20 20 2f 2a  0];.#endif..  /*
0360: 20 55 6e 64 65 72 6c 79 69 6e 67 20 6d 61 6c 6c   Underlying mall
0370: 6f 63 2f 72 65 61 6c 6c 6f 63 2f 66 72 65 65 20  oc/realloc/free 
0380: 66 75 6e 63 74 69 6f 6e 73 20 2a 2f 0a 20 20 76  functions */.  v
0390: 6f 69 64 20 2a 28 2a 78 4d 61 6c 6c 6f 63 29 28  oid *(*xMalloc)(
03a0: 69 6e 74 29 3b 20 20 20 20 20 20 20 20 20 20 2f  int);          /
03b0: 2a 20 75 6e 64 65 72 6c 79 69 6e 67 20 6d 61 6c  * underlying mal
03c0: 6c 6f 63 28 33 29 20 66 75 6e 63 74 69 6f 6e 20  loc(3) function 
03d0: 2a 2f 0a 20 20 76 6f 69 64 20 2a 28 2a 78 52 65  */.  void *(*xRe
03e0: 61 6c 6c 6f 63 29 28 76 6f 69 64 20 2a 2c 20 69  alloc)(void *, i
03f0: 6e 74 29 3b 20 2f 2a 20 75 6e 64 65 72 6c 79 69  nt); /* underlyi
0400: 6e 67 20 72 65 61 6c 6c 6f 63 28 33 29 20 66 75  ng realloc(3) fu
0410: 6e 63 74 69 6f 6e 20 2a 2f 0a 20 20 76 6f 69 64  nction */.  void
0420: 20 28 2a 78 46 72 65 65 29 28 76 6f 69 64 20 2a   (*xFree)(void *
0430: 29 3b 20 20 20 20 20 20 20 20 20 20 2f 2a 20 75  );          /* u
0440: 6e 64 65 72 6c 79 69 6e 67 20 66 72 65 65 28 33  nderlying free(3
0450: 29 20 66 75 6e 63 74 69 6f 6e 20 2a 2f 0a 0a 20  ) function */.. 
0460: 20 2f 2a 20 4d 75 74 65 78 20 74 6f 20 70 72 6f   /* Mutex to pro
0470: 74 65 63 74 20 70 46 69 72 73 74 20 61 6e 64 20  tect pFirst and 
0480: 61 48 61 73 68 20 2a 2f 0a 20 20 76 6f 69 64 20  aHash */.  void 
0490: 28 2a 78 45 6e 74 65 72 4d 75 74 65 78 29 28 54  (*xEnterMutex)(T
04a0: 6d 47 6c 6f 62 61 6c 2a 29 3b 20 2f 2a 20 43 61  mGlobal*); /* Ca
04b0: 6c 6c 20 74 68 69 73 20 74 6f 20 65 6e 74 65 72  ll this to enter
04c0: 20 74 68 65 20 6d 75 74 65 78 20 2a 2f 0a 20 20   the mutex */.  
04d0: 76 6f 69 64 20 28 2a 78 4c 65 61 76 65 4d 75 74  void (*xLeaveMut
04e0: 65 78 29 28 54 6d 47 6c 6f 62 61 6c 2a 29 3b 20  ex)(TmGlobal*); 
04f0: 2f 2a 20 43 61 6c 6c 20 74 68 69 73 20 74 6f 20  /* Call this to 
0500: 6c 65 61 76 65 20 6d 75 74 65 78 20 2a 2f 0a 20  leave mutex */. 
0510: 20 76 6f 69 64 20 28 2a 78 44 65 6c 4d 75 74 65   void (*xDelMute
0520: 78 29 28 54 6d 47 6c 6f 62 61 6c 2a 29 3b 20 20  x)(TmGlobal*);  
0530: 20 2f 2a 20 43 61 6c 6c 20 74 68 69 73 20 74 6f   /* Call this to
0540: 20 64 65 6c 65 74 65 20 6d 75 74 65 78 20 2a 2f   delete mutex */
0550: 0a 20 20 76 6f 69 64 20 2a 70 4d 75 74 65 78 3b  .  void *pMutex;
0560: 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20                  
0570: 20 20 20 2f 2a 20 4d 75 74 65 78 20 68 61 6e 64     /* Mutex hand
0580: 6c 65 20 2a 2f 0a 0a 20 20 76 6f 69 64 20 2a 28  le */..  void *(
0590: 2a 78 53 61 76 65 4d 61 6c 6c 6f 63 29 28 76 6f  *xSaveMalloc)(vo
05a0: 69 64 20 2a 2c 20 73 69 7a 65 5f 74 29 3b 0a 20  id *, size_t);. 
05b0: 20 76 6f 69 64 20 2a 28 2a 78 53 61 76 65 52 65   void *(*xSaveRe
05c0: 61 6c 6c 6f 63 29 28 76 6f 69 64 20 2a 2c 20 76  alloc)(void *, v
05d0: 6f 69 64 20 2a 2c 20 73 69 7a 65 5f 74 29 3b 0a  oid *, size_t);.
05e0: 20 20 76 6f 69 64 20 28 2a 78 53 61 76 65 46 72    void (*xSaveFr
05f0: 65 65 29 28 76 6f 69 64 20 2a 2c 20 76 6f 69 64  ee)(void *, void
0600: 20 2a 29 3b 0a 0a 20 20 2f 2a 20 4f 4f 4d 20 69   *);..  /* OOM i
0610: 6e 6a 65 63 74 69 6f 6e 20 73 63 68 65 64 75 6c  njection schedul
0620: 69 6e 67 2e 20 49 66 20 6e 43 6f 75 6e 74 64 6f  ing. If nCountdo
0630: 77 6e 20 69 73 20 67 72 65 61 74 65 72 20 74 68  wn is greater th
0640: 61 6e 20 7a 65 72 6f 20 77 68 65 6e 20 61 20 0a  an zero when a .
0650: 20 20 2a 2a 20 6d 61 6c 6c 6f 63 20 61 74 74 65    ** malloc atte
0660: 6d 70 74 20 69 73 20 6d 61 64 65 2c 20 69 74 20  mpt is made, it 
0670: 69 73 20 64 65 63 72 65 6d 65 6e 74 65 64 2e 20  is decremented. 
0680: 49 66 20 74 68 69 73 20 6d 65 61 6e 73 20 6e 43  If this means nC
0690: 6f 75 6e 74 64 6f 77 6e 20 0a 20 20 2a 2a 20 74  ountdown .  ** t
06a0: 72 61 6e 73 69 74 69 6f 6e 73 20 66 72 6f 6d 20  ransitions from 
06b0: 31 20 74 6f 20 30 2c 20 74 68 65 6e 20 74 68 65  1 to 0, then the
06c0: 20 61 6c 6c 6f 63 61 74 69 6f 6e 20 66 61 69 6c   allocation fail
06d0: 73 2e 20 49 66 20 62 50 65 72 73 69 73 74 20 69  s. If bPersist i
06e0: 73 20 74 72 75 65 20 0a 20 20 2a 2a 20 77 68 65  s true .  ** whe
06f0: 6e 20 74 68 69 73 20 68 61 70 70 65 6e 73 2c 20  n this happens, 
0700: 6e 43 6f 75 6e 74 64 6f 77 6e 20 69 73 20 74 68  nCountdown is th
0710: 65 6e 20 69 6e 63 72 65 6d 65 6e 74 65 64 20 62  en incremented b
0720: 61 63 6b 20 74 6f 20 31 20 28 73 6f 20 74 68 61  ack to 1 (so tha
0730: 74 20 74 68 65 20 0a 20 20 2a 2a 20 6e 65 78 74  t the .  ** next
0740: 20 61 74 74 65 6d 70 74 20 66 61 69 6c 73 20 74   attempt fails t
0750: 6f 6f 29 2e 20 20 0a 20 20 2a 2f 0a 20 20 69 6e  oo).  .  */.  in
0760: 74 20 6e 43 6f 75 6e 74 64 6f 77 6e 3b 0a 20 20  t nCountdown;.  
0770: 69 6e 74 20 62 50 65 72 73 69 73 74 3b 0a 20 20  int bPersist;.  
0780: 69 6e 74 20 62 45 6e 61 62 6c 65 3b 0a 20 20 76  int bEnable;.  v
0790: 6f 69 64 20 28 2a 78 48 6f 6f 6b 29 28 76 6f 69  oid (*xHook)(voi
07a0: 64 20 2a 29 3b 0a 20 20 76 6f 69 64 20 2a 70 48  d *);.  void *pH
07b0: 6f 6f 6b 43 74 78 3b 0a 7d 3b 0a 0a 73 74 72 75  ookCtx;.};..stru
07c0: 63 74 20 54 6d 42 6c 6f 63 6b 48 64 72 20 7b 0a  ct TmBlockHdr {.
07d0: 20 20 54 6d 42 6c 6f 63 6b 48 64 72 20 2a 70 4e    TmBlockHdr *pN
07e0: 65 78 74 3b 0a 20 20 54 6d 42 6c 6f 63 6b 48 64  ext;.  TmBlockHd
07f0: 72 20 2a 70 50 72 65 76 3b 0a 20 20 69 6e 74 20  r *pPrev;.  int 
0800: 6e 42 79 74 65 3b 0a 23 69 66 64 65 66 20 54 4d  nByte;.#ifdef TM
0810: 5f 42 41 43 4b 54 52 41 43 45 0a 20 20 54 6d 41  _BACKTRACE.  TmA
0820: 67 67 20 2a 70 41 67 67 3b 0a 23 65 6e 64 69 66  gg *pAgg;.#endif
0830: 0a 20 20 75 33 32 20 69 46 6f 72 65 47 75 61 72  .  u32 iForeGuar
0840: 64 3b 0a 7d 3b 0a 0a 23 69 66 64 65 66 20 54 4d  d;.};..#ifdef TM
0850: 5f 42 41 43 4b 54 52 41 43 45 0a 73 74 72 75 63  _BACKTRACE.struc
0860: 74 20 54 6d 41 67 67 20 7b 0a 20 20 69 6e 74 20  t TmAgg {.  int 
0870: 6e 41 6c 6c 6f 63 3b 20 20 20 20 20 20 20 20 20  nAlloc;         
0880: 20 20 20 20 20 20 20 20 20 20 20 20 2f 2a 20 4e              /* N
0890: 75 6d 62 65 72 20 6f 66 20 61 6c 6c 6f 63 61 74  umber of allocat
08a0: 69 6f 6e 73 20 61 74 20 74 68 69 73 20 70 61 74  ions at this pat
08b0: 68 20 2a 2f 0a 20 20 69 6e 74 20 6e 42 79 74 65  h */.  int nByte
08c0: 3b 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20  ;               
08d0: 20 20 20 20 20 20 20 2f 2a 20 54 6f 74 61 6c 20         /* Total 
08e0: 6e 75 6d 62 65 72 20 6f 66 20 62 79 74 65 73 20  number of bytes 
08f0: 61 6c 6c 6f 63 61 74 65 64 20 2a 2f 0a 20 20 69  allocated */.  i
0900: 6e 74 20 6e 4f 75 74 41 6c 6c 6f 63 3b 20 20 20  nt nOutAlloc;   
0910: 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20 2f                 /
0920: 2a 20 4e 75 6d 62 65 72 20 6f 66 20 6f 75 74 73  * Number of outs
0930: 74 61 6e 64 69 6e 67 20 61 6c 6c 6f 63 61 74 69  tanding allocati
0940: 6f 6e 73 20 2a 2f 0a 20 20 69 6e 74 20 6e 4f 75  ons */.  int nOu
0950: 74 42 79 74 65 3b 20 20 20 20 20 20 20 20 20 20  tByte;          
0960: 20 20 20 20 20 20 20 20 20 2f 2a 20 4e 75 6d 62           /* Numb
0970: 65 72 20 6f 66 20 6f 75 74 73 74 61 6e 64 69 6e  er of outstandin
0980: 67 20 62 79 74 65 73 20 2a 2f 0a 20 20 76 6f 69  g bytes */.  voi
0990: 64 20 2a 61 46 72 61 6d 65 5b 54 4d 5f 42 41 43  d *aFrame[TM_BAC
09a0: 4b 54 52 41 43 45 5d 3b 20 20 20 20 20 2f 2a 20  KTRACE];     /* 
09b0: 62 61 63 6b 74 72 61 63 65 28 29 20 6f 75 74 70  backtrace() outp
09c0: 75 74 20 2a 2f 0a 20 20 54 6d 41 67 67 20 2a 70  ut */.  TmAgg *p
09d0: 4e 65 78 74 3b 20 20 20 20 20 20 20 20 20 20 20  Next;           
09e0: 20 20 20 20 20 20 20 20 2f 2a 20 4e 65 78 74 20          /* Next 
09f0: 6f 62 6a 65 63 74 20 69 6e 20 68 61 73 68 2d 74  object in hash-t
0a00: 61 62 6c 65 20 63 6f 6c 6c 69 73 69 6f 6e 20 2a  able collision *
0a10: 2f 0a 7d 3b 0a 23 65 6e 64 69 66 0a 0a 23 64 65  /.};.#endif..#de
0a20: 66 69 6e 65 20 46 4f 52 45 47 55 41 52 44 20 30  fine FOREGUARD 0
0a30: 78 38 30 46 35 45 31 35 33 0a 23 64 65 66 69 6e  x80F5E153.#defin
0a40: 65 20 52 45 41 52 47 55 41 52 44 20 30 78 45 34  e REARGUARD 0xE4
0a50: 36 37 36 42 35 33 0a 73 74 61 74 69 63 20 63 6f  676B53.static co
0a60: 6e 73 74 20 75 33 32 20 72 65 61 72 67 75 61 72  nst u32 rearguar
0a70: 64 20 3d 20 52 45 41 52 47 55 41 52 44 3b 0a 0a  d = REARGUARD;..
0a80: 23 64 65 66 69 6e 65 20 52 4f 55 4e 44 38 28 78  #define ROUND8(x
0a90: 29 20 28 28 28 78 29 2b 37 29 26 7e 37 29 0a 0a  ) (((x)+7)&~7)..
0aa0: 23 64 65 66 69 6e 65 20 42 4c 4f 43 4b 5f 48 44  #define BLOCK_HD
0ab0: 52 5f 53 49 5a 45 20 28 52 4f 55 4e 44 38 28 20  R_SIZE (ROUND8( 
0ac0: 73 69 7a 65 6f 66 28 54 6d 42 6c 6f 63 6b 48 64  sizeof(TmBlockHd
0ad0: 72 29 20 29 29 0a 0a 73 74 61 74 69 63 20 76 6f  r) ))..static vo
0ae0: 69 64 20 6c 73 6d 74 65 73 74 5f 6f 6f 6d 5f 65  id lsmtest_oom_e
0af0: 72 72 6f 72 28 76 6f 69 64 29 7b 0a 20 20 73 74  rror(void){.  st
0b00: 61 74 69 63 20 69 6e 74 20 6e 45 72 72 20 3d 20  atic int nErr = 
0b10: 30 3b 0a 20 20 6e 45 72 72 2b 2b 3b 0a 7d 0a 0a  0;.  nErr++;.}..
0b20: 73 74 61 74 69 63 20 76 6f 69 64 20 74 6d 45 6e  static void tmEn
0b30: 74 65 72 4d 75 74 65 78 28 54 6d 47 6c 6f 62 61  terMutex(TmGloba
0b40: 6c 20 2a 70 54 6d 29 7b 0a 20 20 70 54 6d 2d 3e  l *pTm){.  pTm->
0b50: 78 45 6e 74 65 72 4d 75 74 65 78 28 70 54 6d 29  xEnterMutex(pTm)
0b60: 3b 0a 7d 0a 73 74 61 74 69 63 20 76 6f 69 64 20  ;.}.static void 
0b70: 74 6d 4c 65 61 76 65 4d 75 74 65 78 28 54 6d 47  tmLeaveMutex(TmG
0b80: 6c 6f 62 61 6c 20 2a 70 54 6d 29 7b 0a 20 20 70  lobal *pTm){.  p
0b90: 54 6d 2d 3e 78 4c 65 61 76 65 4d 75 74 65 78 28  Tm->xLeaveMutex(
0ba0: 70 54 6d 29 3b 0a 7d 0a 0a 73 74 61 74 69 63 20  pTm);.}..static 
0bb0: 76 6f 69 64 20 2a 74 6d 4d 61 6c 6c 6f 63 28 54  void *tmMalloc(T
0bc0: 6d 47 6c 6f 62 61 6c 20 2a 70 54 6d 2c 20 69 6e  mGlobal *pTm, in
0bd0: 74 20 6e 42 79 74 65 29 7b 0a 20 20 54 6d 42 6c  t nByte){.  TmBl
0be0: 6f 63 6b 48 64 72 20 2a 70 4e 65 77 3b 20 20 20  ockHdr *pNew;   
0bf0: 20 20 20 20 20 20 20 20 20 20 20 20 2f 2a 20 4e              /* N
0c00: 65 77 20 61 6c 6c 6f 63 61 74 69 6f 6e 20 68 65  ew allocation he
0c10: 61 64 65 72 20 62 6c 6f 63 6b 20 2a 2f 0a 20 20  ader block */.  
0c20: 75 38 20 2a 70 55 73 65 72 3b 20 20 20 20 20 20  u8 *pUser;      
0c30: 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20                  
0c40: 2f 2a 20 52 65 74 75 72 6e 20 76 61 6c 75 65 20  /* Return value 
0c50: 2a 2f 0a 20 20 69 6e 74 20 6e 52 65 71 3b 20 20  */.  int nReq;  
0c60: 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20 20                  
0c70: 20 20 20 20 20 2f 2a 20 54 6f 74 61 6c 20 6e 75       /* Total nu
0c80: 6d 62 65 72 20 6f 66 20 62 79 74 65 73 20 72 65  mber of bytes re
0c90: 71 75 65 73 74 65 64 20 2a 2f 0a 0a 20 20 61 73  quested */..  as
0ca0: 73 65 72 74 28 20 73 69 7a 65 6f 66 28 72 65 61  sert( sizeof(rea
0cb0: 72 67 75 61 72 64 29 3d 3d 34 20 29 3b 0a 20 20  rguard)==4 );.  
0cc0: 6e 52 65 71 20 3d 20 42 4c 4f 43 4b 5f 48 44 52  nReq = BLOCK_HDR
0cd0: 5f 53 49 5a 45 20 2b 20 6e 42 79 74 65 20 2b 20  _SIZE + nByte + 
0ce0: 34 3b 0a 20 20 70 4e 65 77 20 3d 20 28 54 6d 42  4;.  pNew = (TmB
0cf0: 6c 6f 63 6b 48 64 72 20 2a 29 70 54 6d 2d 3e 78  lockHdr *)pTm->x
0d00: 4d 61 6c 6c 6f 63 28 6e 52 65 71 29 3b 0a 20 20  Malloc(nReq);.  
0d10: 6d 65 6d 73 65 74 28 70 4e 65 77 2c 20 30 2c 20  memset(pNew, 0, 
0d20: 73 69 7a 65 6f 66 28 54 6d 42 6c 6f 63 6b 48 64  sizeof(TmBlockHd
0d30: 72 29 29 3b 0a 0a 20 20 74 6d 45 6e 74 65 72 4d  r));..  tmEnterM
0d40: 75 74 65 78 28 70 54 6d 29 3b 0a 20 20 61 73 73  utex(pTm);.  ass
0d50: 65 72 74 28 20 70 54 6d 2d 3e 6e 43 6f 75 6e 74  ert( pTm->nCount
0d60: 64 6f 77 6e 3e 3d 30 20 29 3b 0a 20 20 61 73 73  down>=0 );.  ass
0d70: 65 72 74 28 20 70 54 6d 2d 3e 62 50 65 72 73 69  ert( pTm->bPersi
0d80: 73 74 3d 3d 30 20 7c 7c 20 70 54 6d 2d 3e 62 50  st==0 || pTm->bP
0d90: 65 72 73 69 73 74 3d 3d 31 20 29 3b 0a 0a 20 20  ersist==1 );..  
0da0: 69 66 28 20 70 54 6d 2d 3e 62 45 6e 61 62 6c 65  if( pTm->bEnable
0db0: 20 26 26 20 70 54 6d 2d 3e 6e 43 6f 75 6e 74 64   && pTm->nCountd
0dc0: 6f 77 6e 3d 3d 31 20 29 7b 0a 20 20 20 20 2f 2a  own==1 ){.    /*
0dd0: 20 53 69 6d 75 6c 61 74 65 20 61 6e 20 4f 4f 4d   Simulate an OOM
0de0: 20 65 72 72 6f 72 2e 20 2a 2f 0a 20 20 20 20 6c   error. */.    l
0df0: 73 6d 74 65 73 74 5f 6f 6f 6d 5f 65 72 72 6f 72  smtest_oom_error
0e00: 28 29 3b 0a 20 20 20 20 70 54 6d 2d 3e 78 46 72  ();.    pTm->xFr
0e10: 65 65 28 70 4e 65 77 29 3b 0a 20 20 20 20 70 54  ee(pNew);.    pT
0e20: 6d 2d 3e 6e 43 6f 75 6e 74 64 6f 77 6e 20 3d 20  m->nCountdown = 
0e30: 70 54 6d 2d 3e 62 50 65 72 73 69 73 74 3b 0a 20  pTm->bPersist;. 
0e40: 20 20 20 69 66 28 20 70 54 6d 2d 3e 78 48 6f 6f     if( pTm->xHoo
0e50: 6b 20 29 20 70 54 6d 2d 3e 78 48 6f 6f 6b 28 70  k ) pTm->xHook(p
0e60: 54 6d 2d 3e 70 48 6f 6f 6b 43 74 78 29 3b 0a 20  Tm->pHookCtx);. 
0e70: 20 20 20 70 55 73 65 72 20 3d 20 30 3b 0a 20 20     pUser = 0;.  
0e80: 7d 65 6c 73 65 7b 0a 20 20 20 20 69 66 28 20 70  }else{.    if( p
0e90: 54 6d 2d 3e 62 45 6e 61 62 6c 65 20 26 26 20 70  Tm->bEnable && p
0ea0: 54 6d 2d 3e 6e 43 6f 75 6e 74 64 6f 77 6e 20 29  Tm->nCountdown )
0eb0: 20 70 54 6d 2d 3e 6e 43 6f 75 6e 74 64 6f 77 6e   pTm->nCountdown
0ec0: 2d 2d 3b 0a 0a 20 20 20 20 70 4e 65 77 2d 3e 69  --;..    pNew->i
0ed0: 46 6f 72 65 47 75 61 72 64 20 3d 20 46 4f 52 45  ForeGuard = FORE
0ee0: 47 55 41 52 44 3b 0a 20 20 20 20 70 4e 65 77 2d  GUARD;.    pNew-
0ef0: 3e 6e 42 79 74 65 20 3d 20 6e 42 79 74 65 3b 0a  >nByte = nByte;.
0f00: 20 20 20 20 70 4e 65 77 2d 3e 70 4e 65 78 74 20      pNew->pNext 
0f10: 3d 20 70 54 6d 2d 3e 70 46 69 72 73 74 3b 0a 0a  = pTm->pFirst;..
0f20: 20 20 20 20 69 66 28 20 70 54 6d 2d 3e 70 46 69      if( pTm->pFi
0f30: 72 73 74 20 29 7b 0a 20 20 20 20 20 20 70 54 6d  rst ){.      pTm
0f40: 2d 3e 70 46 69 72 73 74 2d 3e 70 50 72 65 76 20  ->pFirst->pPrev 
0f50: 3d 20 70 4e 65 77 3b 0a 20 20 20 20 7d 0a 20 20  = pNew;.    }.  
0f60: 20 20 70 54 6d 2d 3e 70 46 69 72 73 74 20 3d 20    pTm->pFirst = 
0f70: 70 4e 65 77 3b 0a 0a 20 20 20 20 70 55 73 65 72  pNew;..    pUser
0f80: 20 3d 20 26 28 28 75 38 20 2a 29 70 4e 65 77 29   = &((u8 *)pNew)
0f90: 5b 42 4c 4f 43 4b 5f 48 44 52 5f 53 49 5a 45 5d  [BLOCK_HDR_SIZE]
0fa0: 3b 0a 20 20 20 20 6d 65 6d 73 65 74 28 70 55 73  ;.    memset(pUs
0fb0: 65 72 2c 20 30 78 35 36 2c 20 6e 42 79 74 65 29  er, 0x56, nByte)
0fc0: 3b 0a 20 20 20 20 6d 65 6d 63 70 79 28 26 70 55  ;.    memcpy(&pU
0fd0: 73 65 72 5b 6e 42 79 74 65 5d 2c 20 26 72 65 61  ser[nByte], &rea
0fe0: 72 67 75 61 72 64 2c 20 34 29 3b 0a 0a 23 69 66  rguard, 4);..#if
0ff0: 64 65 66 20 54 4d 5f 42 41 43 4b 54 52 41 43 45  def TM_BACKTRACE
1000: 0a 20 20 20 20 7b 0a 20 20 20 20 20 20 54 6d 41  .    {.      TmA
1010: 67 67 20 2a 70 41 67 67 3b 0a 20 20 20 20 20 20  gg *pAgg;.      
1020: 69 6e 74 20 69 3b 0a 20 20 20 20 20 20 75 33 32  int i;.      u32
1030: 20 69 48 61 73 68 20 3d 20 30 3b 0a 20 20 20 20   iHash = 0;.    
1040: 20 20 76 6f 69 64 20 2a 61 46 72 61 6d 65 5b 54    void *aFrame[T
1050: 4d 5f 42 41 43 4b 54 52 41 43 45 5d 3b 0a 20 20  M_BACKTRACE];.  
1060: 20 20 20 20 6d 65 6d 73 65 74 28 61 46 72 61 6d      memset(aFram
1070: 65 2c 20 30 2c 20 73 69 7a 65 6f 66 28 61 46 72  e, 0, sizeof(aFr
1080: 61 6d 65 29 29 3b 0a 20 20 20 20 20 20 62 61 63  ame));.      bac
1090: 6b 74 72 61 63 65 28 61 46 72 61 6d 65 2c 20 54  ktrace(aFrame, T
10a0: 4d 5f 42 41 43 4b 54 52 41 43 45 29 3b 0a 0a 20  M_BACKTRACE);.. 
10b0: 20 20 20 20 20 66 6f 72 28 69 3d 30 3b 20 69 3c       for(i=0; i<
10c0: 41 72 72 61 79 53 69 7a 65 28 61 46 72 61 6d 65  ArraySize(aFrame
10d0: 29 3b 20 69 2b 2b 29 7b 0a 20 20 20 20 20 20 20  ); i++){.       
10e0: 20 69 48 61 73 68 20 2b 3d 20 28 75 36 34 29 28   iHash += (u64)(
10f0: 61 46 72 61 6d 65 5b 69 5d 29 20 2b 20 28 69 48  aFrame[i]) + (iH
1100: 61 73 68 3c 3c 33 29 3b 0a 20 20 20 20 20 20 7d  ash<<3);.      }
1110: 0a 20 20 20 20 20 20 69 48 61 73 68 20 3d 20 69  .      iHash = i
1120: 48 61 73 68 20 25 20 41 72 72 61 79 53 69 7a 65  Hash % ArraySize
1130: 28 70 54 6d 2d 3e 61 48 61 73 68 29 3b 0a 0a 20  (pTm->aHash);.. 
1140: 20 20 20 20 20 66 6f 72 28 70 41 67 67 3d 70 54       for(pAgg=pT
1150: 6d 2d 3e 61 48 61 73 68 5b 69 48 61 73 68 5d 3b  m->aHash[iHash];
1160: 20 70 41 67 67 3b 20 70 41 67 67 3d 70 41 67 67   pAgg; pAgg=pAgg
1170: 2d 3e 70 4e 65 78 74 29 7b 0a 20 20 20 20 20 20  ->pNext){.      
1180: 20 20 69 66 28 20 6d 65 6d 63 6d 70 28 70 41 67    if( memcmp(pAg
1190: 67 2d 3e 61 46 72 61 6d 65 2c 20 61 46 72 61 6d  g->aFrame, aFram
11a0: 65 2c 20 73 69 7a 65 6f 66 28 61 46 72 61 6d 65  e, sizeof(aFrame
11b0: 29 29 3d 3d 30 20 29 20 62 72 65 61 6b 3b 0a 20  ))==0 ) break;. 
11c0: 20 20 20 20 20 7d 0a 20 20 20 20 20 20 69 66 28       }.      if(
11d0: 20 21 70 41 67 67 20 29 7b 0a 20 20 20 20 20 20   !pAgg ){.      
11e0: 20 20 70 41 67 67 20 3d 20 28 54 6d 41 67 67 20    pAgg = (TmAgg 
11f0: 2a 29 70 54 6d 2d 3e 78 4d 61 6c 6c 6f 63 28 73  *)pTm->xMalloc(s
1200: 69 7a 65 6f 66 28 54 6d 41 67 67 29 29 3b 0a 20  izeof(TmAgg));. 
1210: 20 20 20 20 20 20 20 6d 65 6d 73 65 74 28 70 41         memset(pA
1220: 67 67 2c 20 30 2c 20 73 69 7a 65 6f 66 28 54 6d  gg, 0, sizeof(Tm
1230: 41 67 67 29 29 3b 0a 20 20 20 20 20 20 20 20 6d  Agg));.        m
1240: 65 6d 63 70 79 28 70 41 67 67 2d 3e 61 46 72 61  emcpy(pAgg->aFra
1250: 6d 65 2c 20 61 46 72 61 6d 65 2c 20 73 69 7a 65  me, aFrame, size
1260: 6f 66 28 61 46 72 61 6d 65 29 29 3b 0a 20 20 20  of(aFrame));.   
1270: 20 20 20 20 20 70 41 67 67 2d 3e 70 4e 65 78 74       pAgg->pNext
1280: 20 3d 20 70 54 6d 2d 3e 61 48 61 73 68 5b 69 48   = pTm->aHash[iH
1290: 61 73 68 5d 3b 0a 20 20 20 20 20 20 20 20 70 54  ash];.        pT
12a0: 6d 2d 3e 61 48 61 73 68 5b 69 48 61 73 68 5d 20  m->aHash[iHash] 
12b0: 3d 20 70 41 67 67 3b 0a 20 20 20 20 20 20 7d 0a  = pAgg;.      }.
12c0: 20 20 20 20 20 20 70 41 67 67 2d 3e 6e 41 6c 6c        pAgg->nAll
12d0: 6f 63 2b 2b 3b 0a 20 20 20 20 20 20 70 41 67 67  oc++;.      pAgg
12e0: 2d 3e 6e 42 79 74 65 20 2b 3d 20 6e 42 79 74 65  ->nByte += nByte
12f0: 3b 0a 20 20 20 20 20 20 70 41 67 67 2d 3e 6e 4f  ;.      pAgg->nO
1300: 75 74 41 6c 6c 6f 63 2b 2b 3b 0a 20 20 20 20 20  utAlloc++;.     
1310: 20 70 41 67 67 2d 3e 6e 4f 75 74 42 79 74 65 20   pAgg->nOutByte 
1320: 2b 3d 20 6e 42 79 74 65 3b 0a 20 20 20 20 20 20  += nByte;.      
1330: 70 4e 65 77 2d 3e 70 41 67 67 20 3d 20 70 41 67  pNew->pAgg = pAg
1340: 67 3b 0a 20 20 20 20 7d 0a 23 65 6e 64 69 66 0a  g;.    }.#endif.
1350: 20 20 7d 0a 0a 20 20 74 6d 4c 65 61 76 65 4d 75    }..  tmLeaveMu
1360: 74 65 78 28 70 54 6d 29 3b 0a 20 20 72 65 74 75  tex(pTm);.  retu
1370: 72 6e 20 70 55 73 65 72 3b 0a 7d 0a 0a 73 74 61  rn pUser;.}..sta
1380: 74 69 63 20 76 6f 69 64 20 74 6d 46 72 65 65 28  tic void tmFree(
1390: 54 6d 47 6c 6f 62 61 6c 20 2a 70 54 6d 2c 20 76  TmGlobal *pTm, v
13a0: 6f 69 64 20 2a 70 29 7b 0a 20 20 69 66 28 20 70  oid *p){.  if( p
13b0: 20 29 7b 0a 20 20 20 20 54 6d 42 6c 6f 63 6b 48   ){.    TmBlockH
13c0: 64 72 20 2a 70 48 64 72 3b 0a 20 20 20 20 75 38  dr *pHdr;.    u8
13d0: 20 2a 70 55 73 65 72 20 3d 20 28 75 38 20 2a 29   *pUser = (u8 *)
13e0: 70 3b 0a 0a 20 20 20 20 74 6d 45 6e 74 65 72 4d  p;..    tmEnterM
13f0: 75 74 65 78 28 70 54 6d 29 3b 0a 20 20 20 20 70  utex(pTm);.    p
1400: 48 64 72 20 3d 20 28 54 6d 42 6c 6f 63 6b 48 64  Hdr = (TmBlockHd
1410: 72 20 2a 29 28 70 55 73 65 72 20 2d 20 42 4c 4f  r *)(pUser - BLO
1420: 43 4b 5f 48 44 52 5f 53 49 5a 45 29 3b 0a 20 20  CK_HDR_SIZE);.  
1430: 20 20 61 73 73 65 72 74 28 20 70 48 64 72 2d 3e    assert( pHdr->
1440: 69 46 6f 72 65 47 75 61 72 64 3d 3d 46 4f 52 45  iForeGuard==FORE
1450: 47 55 41 52 44 20 29 3b 0a 20 20 20 20 61 73 73  GUARD );.    ass
1460: 65 72 74 28 20 30 3d 3d 6d 65 6d 63 6d 70 28 26  ert( 0==memcmp(&
1470: 70 55 73 65 72 5b 70 48 64 72 2d 3e 6e 42 79 74  pUser[pHdr->nByt
1480: 65 5d 2c 20 26 72 65 61 72 67 75 61 72 64 2c 20  e], &rearguard, 
1490: 34 29 20 29 3b 0a 0a 20 20 20 20 69 66 28 20 70  4) );..    if( p
14a0: 48 64 72 2d 3e 70 50 72 65 76 20 29 7b 0a 20 20  Hdr->pPrev ){.  
14b0: 20 20 20 20 61 73 73 65 72 74 28 20 70 48 64 72      assert( pHdr
14c0: 2d 3e 70 50 72 65 76 2d 3e 70 4e 65 78 74 3d 3d  ->pPrev->pNext==
14d0: 70 48 64 72 20 29 3b 0a 20 20 20 20 20 20 70 48  pHdr );.      pH
14e0: 64 72 2d 3e 70 50 72 65 76 2d 3e 70 4e 65 78 74  dr->pPrev->pNext
14f0: 20 3d 20 70 48 64 72 2d 3e 70 4e 65 78 74 3b 0a   = pHdr->pNext;.
1500: 20 20 20 20 7d 65 6c 73 65 7b 0a 20 20 20 20 20      }else{.     
1510: 20 61 73 73 65 72 74 28 20 70 48 64 72 3d 3d 70   assert( pHdr==p
1520: 54 6d 2d 3e 70 46 69 72 73 74 20 29 3b 0a 20 20  Tm->pFirst );.  
1530: 20 20 20 20 70 54 6d 2d 3e 70 46 69 72 73 74 20      pTm->pFirst 
1540: 3d 20 70 48 64 72 2d 3e 70 4e 65 78 74 3b 0a 20  = pHdr->pNext;. 
1550: 20 20 20 7d 0a 20 20 20 20 69 66 28 20 70 48 64     }.    if( pHd
1560: 72 2d 3e 70 4e 65 78 74 20 29 7b 0a 20 20 20 20  r->pNext ){.    
1570: 20 20 61 73 73 65 72 74 28 20 70 48 64 72 2d 3e    assert( pHdr->
1580: 70 4e 65 78 74 2d 3e 70 50 72 65 76 3d 3d 70 48  pNext->pPrev==pH
1590: 64 72 20 29 3b 0a 20 20 20 20 20 20 70 48 64 72  dr );.      pHdr
15a0: 2d 3e 70 4e 65 78 74 2d 3e 70 50 72 65 76 20 3d  ->pNext->pPrev =
15b0: 20 70 48 64 72 2d 3e 70 50 72 65 76 3b 0a 20 20   pHdr->pPrev;.  
15c0: 20 20 7d 0a 0a 23 69 66 64 65 66 20 54 4d 5f 42    }..#ifdef TM_B
15d0: 41 43 4b 54 52 41 43 45 0a 20 20 20 20 70 48 64  ACKTRACE.    pHd
15e0: 72 2d 3e 70 41 67 67 2d 3e 6e 4f 75 74 41 6c 6c  r->pAgg->nOutAll
15f0: 6f 63 2d 2d 3b 0a 20 20 20 20 70 48 64 72 2d 3e  oc--;.    pHdr->
1600: 70 41 67 67 2d 3e 6e 4f 75 74 42 79 74 65 20 2d  pAgg->nOutByte -
1610: 3d 20 70 48 64 72 2d 3e 6e 42 79 74 65 3b 0a 23  = pHdr->nByte;.#
1620: 65 6e 64 69 66 0a 0a 20 20 20 20 74 6d 4c 65 61  endif..    tmLea
1630: 76 65 4d 75 74 65 78 28 70 54 6d 29 3b 0a 20 20  veMutex(pTm);.  
1640: 20 20 6d 65 6d 73 65 74 28 70 55 73 65 72 2c 20    memset(pUser, 
1650: 30 78 35 38 2c 20 70 48 64 72 2d 3e 6e 42 79 74  0x58, pHdr->nByt
1660: 65 29 3b 0a 20 20 20 20 6d 65 6d 73 65 74 28 70  e);.    memset(p
1670: 48 64 72 2c 20 30 78 35 37 2c 20 73 69 7a 65 6f  Hdr, 0x57, sizeo
1680: 66 28 54 6d 42 6c 6f 63 6b 48 64 72 29 29 3b 0a  f(TmBlockHdr));.
1690: 20 20 20 20 70 54 6d 2d 3e 78 46 72 65 65 28 70      pTm->xFree(p
16a0: 48 64 72 29 3b 0a 20 20 7d 0a 7d 0a 0a 73 74 61  Hdr);.  }.}..sta
16b0: 74 69 63 20 76 6f 69 64 20 2a 74 6d 52 65 61 6c  tic void *tmReal
16c0: 6c 6f 63 28 54 6d 47 6c 6f 62 61 6c 20 2a 70 54  loc(TmGlobal *pT
16d0: 6d 2c 20 76 6f 69 64 20 2a 70 2c 20 69 6e 74 20  m, void *p, int 
16e0: 6e 42 79 74 65 29 7b 0a 20 20 76 6f 69 64 20 2a  nByte){.  void *
16f0: 70 4e 65 77 3b 0a 0a 20 20 70 4e 65 77 20 3d 20  pNew;..  pNew = 
1700: 74 6d 4d 61 6c 6c 6f 63 28 70 54 6d 2c 20 6e 42  tmMalloc(pTm, nB
1710: 79 74 65 29 3b 0a 20 20 69 66 28 20 70 4e 65 77  yte);.  if( pNew
1720: 20 26 26 20 70 20 29 7b 0a 20 20 20 20 54 6d 42   && p ){.    TmB
1730: 6c 6f 63 6b 48 64 72 20 2a 70 48 64 72 3b 0a 20  lockHdr *pHdr;. 
1740: 20 20 20 75 38 20 2a 70 55 73 65 72 20 3d 20 28     u8 *pUser = (
1750: 75 38 20 2a 29 70 3b 0a 20 20 20 20 70 48 64 72  u8 *)p;.    pHdr
1760: 20 3d 20 28 54 6d 42 6c 6f 63 6b 48 64 72 20 2a   = (TmBlockHdr *
1770: 29 28 70 55 73 65 72 20 2d 20 42 4c 4f 43 4b 5f  )(pUser - BLOCK_
1780: 48 44 52 5f 53 49 5a 45 29 3b 0a 20 20 20 20 6d  HDR_SIZE);.    m
1790: 65 6d 63 70 79 28 70 4e 65 77 2c 20 70 2c 20 4d  emcpy(pNew, p, M
17a0: 49 4e 28 6e 42 79 74 65 2c 20 70 48 64 72 2d 3e  IN(nByte, pHdr->
17b0: 6e 42 79 74 65 29 29 3b 0a 20 20 20 20 74 6d 46  nByte));.    tmF
17c0: 72 65 65 28 70 54 6d 2c 20 70 29 3b 0a 20 20 7d  ree(pTm, p);.  }
17d0: 0a 20 20 72 65 74 75 72 6e 20 70 4e 65 77 3b 0a  .  return pNew;.
17e0: 7d 0a 0a 73 74 61 74 69 63 20 76 6f 69 64 20 74  }..static void t
17f0: 6d 4d 61 6c 6c 6f 63 4f 6f 6d 28 0a 20 20 54 6d  mMallocOom(.  Tm
1800: 47 6c 6f 62 61 6c 20 2a 70 54 6d 2c 20 0a 20 20  Global *pTm, .  
1810: 69 6e 74 20 6e 43 6f 75 6e 74 64 6f 77 6e 2c 20  int nCountdown, 
1820: 0a 20 20 69 6e 74 20 62 50 65 72 73 69 73 74 2c  .  int bPersist,
1830: 0a 20 20 76 6f 69 64 20 28 2a 78 48 6f 6f 6b 29  .  void (*xHook)
1840: 28 76 6f 69 64 20 2a 29 2c 0a 20 20 76 6f 69 64  (void *),.  void
1850: 20 2a 70 48 6f 6f 6b 43 74 78 0a 29 7b 0a 20 20   *pHookCtx.){.  
1860: 61 73 73 65 72 74 28 20 6e 43 6f 75 6e 74 64 6f  assert( nCountdo
1870: 77 6e 3e 3d 30 20 29 3b 0a 20 20 61 73 73 65 72  wn>=0 );.  asser
1880: 74 28 20 62 50 65 72 73 69 73 74 3d 3d 30 20 7c  t( bPersist==0 |
1890: 7c 20 62 50 65 72 73 69 73 74 3d 3d 31 20 29 3b  | bPersist==1 );
18a0: 0a 20 20 70 54 6d 2d 3e 6e 43 6f 75 6e 74 64 6f  .  pTm->nCountdo
18b0: 77 6e 20 3d 20 6e 43 6f 75 6e 74 64 6f 77 6e 3b  wn = nCountdown;
18c0: 0a 20 20 70 54 6d 2d 3e 62 50 65 72 73 69 73 74  .  pTm->bPersist
18d0: 20 3d 20 62 50 65 72 73 69 73 74 3b 0a 20 20 70   = bPersist;.  p
18e0: 54 6d 2d 3e 78 48 6f 6f 6b 20 3d 20 78 48 6f 6f  Tm->xHook = xHoo
18f0: 6b 3b 0a 20 20 70 54 6d 2d 3e 70 48 6f 6f 6b 43  k;.  pTm->pHookC
1900: 74 78 20 3d 20 70 48 6f 6f 6b 43 74 78 3b 0a 20  tx = pHookCtx;. 
1910: 20 70 54 6d 2d 3e 62 45 6e 61 62 6c 65 20 3d 20   pTm->bEnable = 
1920: 31 3b 0a 7d 0a 0a 73 74 61 74 69 63 20 76 6f 69  1;.}..static voi
1930: 64 20 74 6d 4d 61 6c 6c 6f 63 4f 6f 6d 45 6e 61  d tmMallocOomEna
1940: 62 6c 65 28 0a 20 20 54 6d 47 6c 6f 62 61 6c 20  ble(.  TmGlobal 
1950: 2a 70 54 6d 2c 20 0a 20 20 69 6e 74 20 62 45 6e  *pTm, .  int bEn
1960: 61 62 6c 65 0a 29 7b 0a 20 20 70 54 6d 2d 3e 62  able.){.  pTm->b
1970: 45 6e 61 62 6c 65 20 3d 20 62 45 6e 61 62 6c 65  Enable = bEnable
1980: 3b 0a 7d 0a 0a 73 74 61 74 69 63 20 76 6f 69 64  ;.}..static void
1990: 20 74 6d 4d 61 6c 6c 6f 63 43 68 65 63 6b 28 0a   tmMallocCheck(.
19a0: 20 20 54 6d 47 6c 6f 62 61 6c 20 2a 70 54 6d 2c    TmGlobal *pTm,
19b0: 0a 20 20 69 6e 74 20 2a 70 6e 4c 65 61 6b 41 6c  .  int *pnLeakAl
19c0: 6c 6f 63 2c 0a 20 20 69 6e 74 20 2a 70 6e 4c 65  loc,.  int *pnLe
19d0: 61 6b 42 79 74 65 2c 0a 20 20 46 49 4c 45 20 2a  akByte,.  FILE *
19e0: 70 46 69 6c 65 0a 29 7b 0a 20 20 54 6d 42 6c 6f  pFile.){.  TmBlo
19f0: 63 6b 48 64 72 20 2a 70 48 64 72 3b 0a 20 20 69  ckHdr *pHdr;.  i
1a00: 6e 74 20 6e 4c 65 61 6b 20 3d 20 30 3b 0a 20 20  nt nLeak = 0;.  
1a10: 69 6e 74 20 6e 42 79 74 65 20 3d 20 30 3b 0a 0a  int nByte = 0;..
1a20: 20 20 69 66 28 20 70 54 6d 3d 3d 30 20 29 20 72    if( pTm==0 ) r
1a30: 65 74 75 72 6e 3b 0a 0a 20 20 66 6f 72 28 70 48  eturn;..  for(pH
1a40: 64 72 3d 70 54 6d 2d 3e 70 46 69 72 73 74 3b 20  dr=pTm->pFirst; 
1a50: 70 48 64 72 3b 20 70 48 64 72 3d 70 48 64 72 2d  pHdr; pHdr=pHdr-
1a60: 3e 70 4e 65 78 74 29 7b 0a 20 20 20 20 6e 4c 65  >pNext){.    nLe
1a70: 61 6b 2b 2b 3b 20 0a 20 20 20 20 6e 42 79 74 65  ak++; .    nByte
1a80: 20 2b 3d 20 70 48 64 72 2d 3e 6e 42 79 74 65 3b   += pHdr->nByte;
1a90: 0a 20 20 7d 0a 20 20 69 66 28 20 70 6e 4c 65 61  .  }.  if( pnLea
1aa0: 6b 41 6c 6c 6f 63 20 29 20 2a 70 6e 4c 65 61 6b  kAlloc ) *pnLeak
1ab0: 41 6c 6c 6f 63 20 3d 20 6e 4c 65 61 6b 3b 0a 20  Alloc = nLeak;. 
1ac0: 20 69 66 28 20 70 6e 4c 65 61 6b 42 79 74 65 20   if( pnLeakByte 
1ad0: 29 20 2a 70 6e 4c 65 61 6b 42 79 74 65 20 3d 20  ) *pnLeakByte = 
1ae0: 6e 42 79 74 65 3b 0a 0a 23 69 66 64 65 66 20 54  nByte;..#ifdef T
1af0: 4d 5f 42 41 43 4b 54 52 41 43 45 0a 20 20 69 66  M_BACKTRACE.  if
1b00: 28 20 70 46 69 6c 65 20 29 7b 0a 20 20 20 20 69  ( pFile ){.    i
1b10: 6e 74 20 69 3b 0a 20 20 20 20 66 70 72 69 6e 74  nt i;.    fprint
1b20: 66 28 70 46 69 6c 65 2c 20 22 4c 45 41 4b 53 5c  f(pFile, "LEAKS\
1b30: 6e 22 29 3b 0a 20 20 20 20 66 6f 72 28 69 3d 30  n");.    for(i=0
1b40: 3b 20 69 3c 41 72 72 61 79 53 69 7a 65 28 70 54  ; i<ArraySize(pT
1b50: 6d 2d 3e 61 48 61 73 68 29 3b 20 69 2b 2b 29 7b  m->aHash); i++){
1b60: 0a 20 20 20 20 20 20 54 6d 41 67 67 20 2a 70 41  .      TmAgg *pA
1b70: 67 67 3b 0a 20 20 20 20 20 20 66 6f 72 28 70 41  gg;.      for(pA
1b80: 67 67 3d 70 54 6d 2d 3e 61 48 61 73 68 5b 69 5d  gg=pTm->aHash[i]
1b90: 3b 20 70 41 67 67 3b 20 70 41 67 67 3d 70 41 67  ; pAgg; pAgg=pAg
1ba0: 67 2d 3e 70 4e 65 78 74 29 7b 0a 20 20 20 20 20  g->pNext){.     
1bb0: 20 20 20 69 66 28 20 70 41 67 67 2d 3e 6e 4f 75     if( pAgg->nOu
1bc0: 74 41 6c 6c 6f 63 20 29 7b 0a 20 20 20 20 20 20  tAlloc ){.      
1bd0: 20 20 20 20 69 6e 74 20 6a 3b 0a 20 20 20 20 20      int j;.     
1be0: 20 20 20 20 20 66 70 72 69 6e 74 66 28 70 46 69       fprintf(pFi
1bf0: 6c 65 2c 20 22 25 64 20 25 64 20 22 2c 20 70 41  le, "%d %d ", pA
1c00: 67 67 2d 3e 6e 4f 75 74 42 79 74 65 2c 20 70 41  gg->nOutByte, pA
1c10: 67 67 2d 3e 6e 4f 75 74 41 6c 6c 6f 63 29 3b 0a  gg->nOutAlloc);.
1c20: 20 20 20 20 20 20 20 20 20 20 66 6f 72 28 6a 3d            for(j=
1c30: 30 3b 20 6a 3c 54 4d 5f 42 41 43 4b 54 52 41 43  0; j<TM_BACKTRAC
1c40: 45 3b 20 6a 2b 2b 29 7b 0a 20 20 20 20 20 20 20  E; j++){.       
1c50: 20 20 20 20 20 66 70 72 69 6e 74 66 28 70 46 69       fprintf(pFi
1c60: 6c 65 2c 20 22 25 70 20 22 2c 20 70 41 67 67 2d  le, "%p ", pAgg-
1c70: 3e 61 46 72 61 6d 65 5b 6a 5d 29 3b 0a 20 20 20  >aFrame[j]);.   
1c80: 20 20 20 20 20 20 20 7d 0a 20 20 20 20 20 20 20         }.       
1c90: 20 20 20 66 70 72 69 6e 74 66 28 70 46 69 6c 65     fprintf(pFile
1ca0: 2c 20 22 5c 6e 22 29 3b 0a 20 20 20 20 20 20 20  , "\n");.       
1cb0: 20 7d 0a 20 20 20 20 20 20 7d 0a 20 20 20 20 7d   }.      }.    }
1cc0: 0a 20 20 20 20 66 70 72 69 6e 74 66 28 70 46 69  .    fprintf(pFi
1cd0: 6c 65 2c 20 22 5c 6e 41 4c 4c 4f 43 41 54 49 4f  le, "\nALLOCATIO
1ce0: 4e 53 5c 6e 22 29 3b 0a 20 20 20 20 66 6f 72 28  NS\n");.    for(
1cf0: 69 3d 30 3b 20 69 3c 41 72 72 61 79 53 69 7a 65  i=0; i<ArraySize
1d00: 28 70 54 6d 2d 3e 61 48 61 73 68 29 3b 20 69 2b  (pTm->aHash); i+
1d10: 2b 29 7b 0a 20 20 20 20 20 20 54 6d 41 67 67 20  +){.      TmAgg 
1d20: 2a 70 41 67 67 3b 0a 20 20 20 20 20 20 66 6f 72  *pAgg;.      for
1d30: 28 70 41 67 67 3d 70 54 6d 2d 3e 61 48 61 73 68  (pAgg=pTm->aHash
1d40: 5b 69 5d 3b 20 70 41 67 67 3b 20 70 41 67 67 3d  [i]; pAgg; pAgg=
1d50: 70 41 67 67 2d 3e 70 4e 65 78 74 29 7b 0a 20 20  pAgg->pNext){.  
1d60: 20 20 20 20 20 20 69 6e 74 20 6a 3b 0a 20 20 20        int j;.   
1d70: 20 20 20 20 20 66 70 72 69 6e 74 66 28 70 46 69       fprintf(pFi
1d80: 6c 65 2c 20 22 25 64 20 25 64 20 22 2c 20 70 41  le, "%d %d ", pA
1d90: 67 67 2d 3e 6e 42 79 74 65 2c 20 70 41 67 67 2d  gg->nByte, pAgg-
1da0: 3e 6e 41 6c 6c 6f 63 29 3b 0a 20 20 20 20 20 20  >nAlloc);.      
1db0: 20 20 66 6f 72 28 6a 3d 30 3b 20 6a 3c 54 4d 5f    for(j=0; j<TM_
1dc0: 42 41 43 4b 54 52 41 43 45 3b 20 6a 2b 2b 29 20  BACKTRACE; j++) 
1dd0: 66 70 72 69 6e 74 66 28 70 46 69 6c 65 2c 20 22  fprintf(pFile, "
1de0: 25 70 20 22 2c 20 70 41 67 67 2d 3e 61 46 72 61  %p ", pAgg->aFra
1df0: 6d 65 5b 6a 5d 29 3b 0a 20 20 20 20 20 20 20 20  me[j]);.        
1e00: 66 70 72 69 6e 74 66 28 70 46 69 6c 65 2c 20 22  fprintf(pFile, "
1e10: 5c 6e 22 29 3b 0a 20 20 20 20 20 20 7d 0a 20 20  \n");.      }.  
1e20: 20 20 7d 0a 20 20 7d 0a 23 65 6c 73 65 0a 20 20    }.  }.#else.  
1e30: 28 76 6f 69 64 29 70 46 69 6c 65 3b 0a 23 65 6e  (void)pFile;.#en
1e40: 64 69 66 0a 7d 0a 0a 0a 23 69 6e 63 6c 75 64 65  dif.}...#include
1e50: 20 22 6c 73 6d 2e 68 22 0a 23 69 6e 63 6c 75 64   "lsm.h".#includ
1e60: 65 20 22 73 74 64 6c 69 62 2e 68 22 0a 0a 74 79  e "stdlib.h"..ty
1e70: 70 65 64 65 66 20 73 74 72 75 63 74 20 4c 73 6d  pedef struct Lsm
1e80: 4d 75 74 65 78 20 4c 73 6d 4d 75 74 65 78 3b 0a  Mutex LsmMutex;.
1e90: 73 74 72 75 63 74 20 4c 73 6d 4d 75 74 65 78 20  struct LsmMutex 
1ea0: 7b 0a 20 20 6c 73 6d 5f 65 6e 76 20 2a 70 45 6e  {.  lsm_env *pEn
1eb0: 76 3b 0a 20 20 6c 73 6d 5f 6d 75 74 65 78 20 2a  v;.  lsm_mutex *
1ec0: 70 4d 75 74 65 78 3b 0a 7d 3b 0a 0a 73 74 61 74  pMutex;.};..stat
1ed0: 69 63 20 76 6f 69 64 20 74 6d 4c 73 6d 4d 75 74  ic void tmLsmMut
1ee0: 65 78 45 6e 74 65 72 28 54 6d 47 6c 6f 62 61 6c  exEnter(TmGlobal
1ef0: 20 2a 70 54 6d 29 7b 0a 20 20 4c 73 6d 4d 75 74   *pTm){.  LsmMut
1f00: 65 78 20 2a 70 20 3d 20 28 4c 73 6d 4d 75 74 65  ex *p = (LsmMute
1f10: 78 20 2a 29 70 54 6d 2d 3e 70 4d 75 74 65 78 3b  x *)pTm->pMutex;
1f20: 0a 20 20 70 2d 3e 70 45 6e 76 2d 3e 78 4d 75 74  .  p->pEnv->xMut
1f30: 65 78 45 6e 74 65 72 28 70 2d 3e 70 4d 75 74 65  exEnter(p->pMute
1f40: 78 29 3b 0a 7d 0a 73 74 61 74 69 63 20 76 6f 69  x);.}.static voi
1f50: 64 20 74 6d 4c 73 6d 4d 75 74 65 78 4c 65 61 76  d tmLsmMutexLeav
1f60: 65 28 54 6d 47 6c 6f 62 61 6c 20 2a 70 54 6d 29  e(TmGlobal *pTm)
1f70: 7b 0a 20 20 4c 73 6d 4d 75 74 65 78 20 2a 70 20  {.  LsmMutex *p 
1f80: 3d 20 28 4c 73 6d 4d 75 74 65 78 20 2a 29 28 70  = (LsmMutex *)(p
1f90: 54 6d 2d 3e 70 4d 75 74 65 78 29 3b 0a 20 20 70  Tm->pMutex);.  p
1fa0: 2d 3e 70 45 6e 76 2d 3e 78 4d 75 74 65 78 4c 65  ->pEnv->xMutexLe
1fb0: 61 76 65 28 70 2d 3e 70 4d 75 74 65 78 29 3b 0a  ave(p->pMutex);.
1fc0: 7d 0a 73 74 61 74 69 63 20 76 6f 69 64 20 74 6d  }.static void tm
1fd0: 4c 73 6d 4d 75 74 65 78 44 65 6c 28 54 6d 47 6c  LsmMutexDel(TmGl
1fe0: 6f 62 61 6c 20 2a 70 54 6d 29 7b 0a 20 20 4c 73  obal *pTm){.  Ls
1ff0: 6d 4d 75 74 65 78 20 2a 70 20 3d 20 28 4c 73 6d  mMutex *p = (Lsm
2000: 4d 75 74 65 78 20 2a 29 70 54 6d 2d 3e 70 4d 75  Mutex *)pTm->pMu
2010: 74 65 78 3b 0a 20 20 70 54 6d 2d 3e 78 46 72 65  tex;.  pTm->xFre
2020: 65 28 70 29 3b 0a 7d 0a 73 74 61 74 69 63 20 76  e(p);.}.static v
2030: 6f 69 64 20 2a 74 6d 4c 73 6d 4d 61 6c 6c 6f 63  oid *tmLsmMalloc
2040: 28 69 6e 74 20 6e 29 7b 20 72 65 74 75 72 6e 20  (int n){ return 
2050: 6d 61 6c 6c 6f 63 28 6e 29 3b 20 7d 0a 73 74 61  malloc(n); }.sta
2060: 74 69 63 20 76 6f 69 64 20 74 6d 4c 73 6d 46 72  tic void tmLsmFr
2070: 65 65 28 76 6f 69 64 20 2a 70 74 72 29 7b 20 66  ee(void *ptr){ f
2080: 72 65 65 28 70 74 72 29 3b 20 7d 0a 73 74 61 74  ree(ptr); }.stat
2090: 69 63 20 76 6f 69 64 20 2a 74 6d 4c 73 6d 52 65  ic void *tmLsmRe
20a0: 61 6c 6c 6f 63 28 76 6f 69 64 20 2a 70 74 72 2c  alloc(void *ptr,
20b0: 20 69 6e 74 20 6e 29 7b 20 72 65 74 75 72 6e 20   int n){ return 
20c0: 72 65 61 6c 6c 6f 63 28 70 74 72 2c 20 6e 29 3b  realloc(ptr, n);
20d0: 20 7d 0a 0a 73 74 61 74 69 63 20 76 6f 69 64 20   }..static void 
20e0: 2a 74 6d 4c 73 6d 45 6e 76 4d 61 6c 6c 6f 63 28  *tmLsmEnvMalloc(
20f0: 6c 73 6d 5f 65 6e 76 20 2a 70 2c 20 73 69 7a 65  lsm_env *p, size
2100: 5f 74 20 6e 29 7b 20 0a 20 20 72 65 74 75 72 6e  _t n){ .  return
2110: 20 74 6d 4d 61 6c 6c 6f 63 28 28 54 6d 47 6c 6f   tmMalloc((TmGlo
2120: 62 61 6c 20 2a 29 28 70 2d 3e 70 4d 65 6d 43 74  bal *)(p->pMemCt
2130: 78 29 2c 20 6e 29 3b 20 0a 7d 0a 73 74 61 74 69  x), n); .}.stati
2140: 63 20 76 6f 69 64 20 74 6d 4c 73 6d 45 6e 76 46  c void tmLsmEnvF
2150: 72 65 65 28 6c 73 6d 5f 65 6e 76 20 2a 70 2c 20  ree(lsm_env *p, 
2160: 76 6f 69 64 20 2a 70 74 72 29 7b 20 0a 20 20 74  void *ptr){ .  t
2170: 6d 46 72 65 65 28 28 54 6d 47 6c 6f 62 61 6c 20  mFree((TmGlobal 
2180: 2a 29 28 70 2d 3e 70 4d 65 6d 43 74 78 29 2c 20  *)(p->pMemCtx), 
2190: 70 74 72 29 3b 20 0a 7d 0a 73 74 61 74 69 63 20  ptr); .}.static 
21a0: 76 6f 69 64 20 2a 74 6d 4c 73 6d 45 6e 76 52 65  void *tmLsmEnvRe
21b0: 61 6c 6c 6f 63 28 6c 73 6d 5f 65 6e 76 20 2a 70  alloc(lsm_env *p
21c0: 2c 20 76 6f 69 64 20 2a 70 74 72 2c 20 73 69 7a  , void *ptr, siz
21d0: 65 5f 74 20 6e 29 7b 20 0a 20 20 72 65 74 75 72  e_t n){ .  retur
21e0: 6e 20 74 6d 52 65 61 6c 6c 6f 63 28 28 54 6d 47  n tmRealloc((TmG
21f0: 6c 6f 62 61 6c 20 2a 29 28 70 2d 3e 70 4d 65 6d  lobal *)(p->pMem
2200: 43 74 78 29 2c 20 70 74 72 2c 20 6e 29 3b 0a 7d  Ctx), ptr, n);.}
2210: 0a 0a 76 6f 69 64 20 74 65 73 74 4d 61 6c 6c 6f  ..void testMallo
2220: 63 49 6e 73 74 61 6c 6c 28 6c 73 6d 5f 65 6e 76  cInstall(lsm_env
2230: 20 2a 70 45 6e 76 29 7b 0a 20 20 54 6d 47 6c 6f   *pEnv){.  TmGlo
2240: 62 61 6c 20 2a 70 47 6c 6f 62 61 6c 3b 0a 20 20  bal *pGlobal;.  
2250: 4c 73 6d 4d 75 74 65 78 20 2a 70 4d 75 74 65 78  LsmMutex *pMutex
2260: 3b 0a 20 20 61 73 73 65 72 74 28 20 70 45 6e 76  ;.  assert( pEnv
2270: 2d 3e 70 4d 65 6d 43 74 78 3d 3d 30 20 29 3b 0a  ->pMemCtx==0 );.
2280: 0a 20 20 2f 2a 20 41 6c 6c 6f 63 61 74 65 20 61  .  /* Allocate a
2290: 6e 64 20 70 6f 70 75 6c 61 74 65 20 61 20 54 6d  nd populate a Tm
22a0: 47 6c 6f 62 61 6c 20 73 74 72 75 63 74 75 72 65  Global structure
22b0: 2e 20 2a 2f 0a 20 20 70 47 6c 6f 62 61 6c 20 3d  . */.  pGlobal =
22c0: 20 28 54 6d 47 6c 6f 62 61 6c 20 2a 29 74 6d 4c   (TmGlobal *)tmL
22d0: 73 6d 4d 61 6c 6c 6f 63 28 73 69 7a 65 6f 66 28  smMalloc(sizeof(
22e0: 54 6d 47 6c 6f 62 61 6c 29 29 3b 0a 20 20 6d 65  TmGlobal));.  me
22f0: 6d 73 65 74 28 70 47 6c 6f 62 61 6c 2c 20 30 2c  mset(pGlobal, 0,
2300: 20 73 69 7a 65 6f 66 28 54 6d 47 6c 6f 62 61 6c   sizeof(TmGlobal
2310: 29 29 3b 0a 20 20 70 47 6c 6f 62 61 6c 2d 3e 78  ));.  pGlobal->x
2320: 4d 61 6c 6c 6f 63 20 3d 20 74 6d 4c 73 6d 4d 61  Malloc = tmLsmMa
2330: 6c 6c 6f 63 3b 0a 20 20 70 47 6c 6f 62 61 6c 2d  lloc;.  pGlobal-
2340: 3e 78 52 65 61 6c 6c 6f 63 20 3d 20 74 6d 4c 73  >xRealloc = tmLs
2350: 6d 52 65 61 6c 6c 6f 63 3b 0a 20 20 70 47 6c 6f  mRealloc;.  pGlo
2360: 62 61 6c 2d 3e 78 46 72 65 65 20 3d 20 74 6d 4c  bal->xFree = tmL
2370: 73 6d 46 72 65 65 3b 0a 20 20 70 4d 75 74 65 78  smFree;.  pMutex
2380: 20 3d 20 28 4c 73 6d 4d 75 74 65 78 20 2a 29 70   = (LsmMutex *)p
2390: 47 6c 6f 62 61 6c 2d 3e 78 4d 61 6c 6c 6f 63 28  Global->xMalloc(
23a0: 73 69 7a 65 6f 66 28 4c 73 6d 4d 75 74 65 78 29  sizeof(LsmMutex)
23b0: 29 3b 0a 20 20 70 4d 75 74 65 78 2d 3e 70 45 6e  );.  pMutex->pEn
23c0: 76 20 3d 20 70 45 6e 76 3b 0a 20 20 70 45 6e 76  v = pEnv;.  pEnv
23d0: 2d 3e 78 4d 75 74 65 78 53 74 61 74 69 63 28 70  ->xMutexStatic(p
23e0: 45 6e 76 2c 20 4c 53 4d 5f 4d 55 54 45 58 5f 48  Env, LSM_MUTEX_H
23f0: 45 41 50 2c 20 26 70 4d 75 74 65 78 2d 3e 70 4d  EAP, &pMutex->pM
2400: 75 74 65 78 29 3b 0a 20 20 70 47 6c 6f 62 61 6c  utex);.  pGlobal
2410: 2d 3e 78 45 6e 74 65 72 4d 75 74 65 78 20 3d 20  ->xEnterMutex = 
2420: 74 6d 4c 73 6d 4d 75 74 65 78 45 6e 74 65 72 3b  tmLsmMutexEnter;
2430: 0a 20 20 70 47 6c 6f 62 61 6c 2d 3e 78 4c 65 61  .  pGlobal->xLea
2440: 76 65 4d 75 74 65 78 20 3d 20 74 6d 4c 73 6d 4d  veMutex = tmLsmM
2450: 75 74 65 78 4c 65 61 76 65 3b 0a 20 20 70 47 6c  utexLeave;.  pGl
2460: 6f 62 61 6c 2d 3e 78 44 65 6c 4d 75 74 65 78 20  obal->xDelMutex 
2470: 3d 20 74 6d 4c 73 6d 4d 75 74 65 78 44 65 6c 3b  = tmLsmMutexDel;
2480: 0a 20 20 70 47 6c 6f 62 61 6c 2d 3e 70 4d 75 74  .  pGlobal->pMut
2490: 65 78 20 3d 20 28 76 6f 69 64 20 2a 29 70 4d 75  ex = (void *)pMu
24a0: 74 65 78 3b 0a 0a 20 20 70 47 6c 6f 62 61 6c 2d  tex;..  pGlobal-
24b0: 3e 78 53 61 76 65 4d 61 6c 6c 6f 63 20 3d 20 70  >xSaveMalloc = p
24c0: 45 6e 76 2d 3e 78 4d 61 6c 6c 6f 63 3b 0a 20 20  Env->xMalloc;.  
24d0: 70 47 6c 6f 62 61 6c 2d 3e 78 53 61 76 65 52 65  pGlobal->xSaveRe
24e0: 61 6c 6c 6f 63 20 3d 20 70 45 6e 76 2d 3e 78 52  alloc = pEnv->xR
24f0: 65 61 6c 6c 6f 63 3b 0a 20 20 70 47 6c 6f 62 61  ealloc;.  pGloba
2500: 6c 2d 3e 78 53 61 76 65 46 72 65 65 20 3d 20 70  l->xSaveFree = p
2510: 45 6e 76 2d 3e 78 46 72 65 65 3b 0a 0a 20 20 2f  Env->xFree;..  /
2520: 2a 20 53 65 74 20 75 70 20 70 45 6e 76 20 74 6f  * Set up pEnv to
2530: 20 74 68 65 20 75 73 65 20 74 68 65 20 6e 65 77   the use the new
2540: 20 54 6d 47 6c 6f 62 61 6c 20 2a 2f 0a 20 20 70   TmGlobal */.  p
2550: 45 6e 76 2d 3e 70 4d 65 6d 43 74 78 20 3d 20 28  Env->pMemCtx = (
2560: 76 6f 69 64 20 2a 29 70 47 6c 6f 62 61 6c 3b 0a  void *)pGlobal;.
2570: 20 20 70 45 6e 76 2d 3e 78 4d 61 6c 6c 6f 63 20    pEnv->xMalloc 
2580: 3d 20 74 6d 4c 73 6d 45 6e 76 4d 61 6c 6c 6f 63  = tmLsmEnvMalloc
2590: 3b 0a 20 20 70 45 6e 76 2d 3e 78 52 65 61 6c 6c  ;.  pEnv->xReall
25a0: 6f 63 20 3d 20 74 6d 4c 73 6d 45 6e 76 52 65 61  oc = tmLsmEnvRea
25b0: 6c 6c 6f 63 3b 0a 20 20 70 45 6e 76 2d 3e 78 46  lloc;.  pEnv->xF
25c0: 72 65 65 20 3d 20 74 6d 4c 73 6d 45 6e 76 46 72  ree = tmLsmEnvFr
25d0: 65 65 3b 0a 7d 0a 0a 76 6f 69 64 20 74 65 73 74  ee;.}..void test
25e0: 4d 61 6c 6c 6f 63 55 6e 69 6e 73 74 61 6c 6c 28  MallocUninstall(
25f0: 6c 73 6d 5f 65 6e 76 20 2a 70 45 6e 76 29 7b 0a  lsm_env *pEnv){.
2600: 20 20 54 6d 47 6c 6f 62 61 6c 20 2a 70 20 3d 20    TmGlobal *p = 
2610: 28 54 6d 47 6c 6f 62 61 6c 20 2a 29 70 45 6e 76  (TmGlobal *)pEnv
2620: 2d 3e 70 4d 65 6d 43 74 78 3b 0a 20 20 70 45 6e  ->pMemCtx;.  pEn
2630: 76 2d 3e 70 4d 65 6d 43 74 78 20 3d 20 30 3b 0a  v->pMemCtx = 0;.
2640: 20 20 69 66 28 20 70 20 29 7b 0a 20 20 20 20 70    if( p ){.    p
2650: 45 6e 76 2d 3e 78 4d 61 6c 6c 6f 63 20 3d 20 70  Env->xMalloc = p
2660: 2d 3e 78 53 61 76 65 4d 61 6c 6c 6f 63 3b 0a 20  ->xSaveMalloc;. 
2670: 20 20 20 70 45 6e 76 2d 3e 78 52 65 61 6c 6c 6f     pEnv->xReallo
2680: 63 20 3d 20 70 2d 3e 78 53 61 76 65 52 65 61 6c  c = p->xSaveReal
2690: 6c 6f 63 3b 0a 20 20 20 20 70 45 6e 76 2d 3e 78  loc;.    pEnv->x
26a0: 46 72 65 65 20 3d 20 70 2d 3e 78 53 61 76 65 46  Free = p->xSaveF
26b0: 72 65 65 3b 0a 20 20 20 20 70 2d 3e 78 44 65 6c  ree;.    p->xDel
26c0: 4d 75 74 65 78 28 70 29 3b 0a 20 20 20 20 74 6d  Mutex(p);.    tm
26d0: 4c 73 6d 46 72 65 65 28 70 29 3b 0a 20 20 7d 0a  LsmFree(p);.  }.
26e0: 7d 0a 0a 76 6f 69 64 20 74 65 73 74 4d 61 6c 6c  }..void testMall
26f0: 6f 63 43 68 65 63 6b 28 0a 20 20 6c 73 6d 5f 65  ocCheck(.  lsm_e
2700: 6e 76 20 2a 70 45 6e 76 2c 0a 20 20 69 6e 74 20  nv *pEnv,.  int 
2710: 2a 70 6e 4c 65 61 6b 41 6c 6c 6f 63 2c 0a 20 20  *pnLeakAlloc,.  
2720: 69 6e 74 20 2a 70 6e 4c 65 61 6b 42 79 74 65 2c  int *pnLeakByte,
2730: 0a 20 20 46 49 4c 45 20 2a 70 46 69 6c 65 0a 29  .  FILE *pFile.)
2740: 7b 0a 20 20 69 66 28 20 70 45 6e 76 2d 3e 70 4d  {.  if( pEnv->pM
2750: 65 6d 43 74 78 3d 3d 30 20 29 7b 0a 20 20 20 20  emCtx==0 ){.    
2760: 2a 70 6e 4c 65 61 6b 41 6c 6c 6f 63 20 3d 20 30  *pnLeakAlloc = 0
2770: 3b 0a 20 20 20 20 2a 70 6e 4c 65 61 6b 42 79 74  ;.    *pnLeakByt
2780: 65 20 3d 20 30 3b 0a 20 20 7d 65 6c 73 65 7b 0a  e = 0;.  }else{.
2790: 20 20 20 20 74 6d 4d 61 6c 6c 6f 63 43 68 65 63      tmMallocChec
27a0: 6b 28 28 54 6d 47 6c 6f 62 61 6c 20 2a 29 28 70  k((TmGlobal *)(p
27b0: 45 6e 76 2d 3e 70 4d 65 6d 43 74 78 29 2c 20 70  Env->pMemCtx), p
27c0: 6e 4c 65 61 6b 41 6c 6c 6f 63 2c 20 70 6e 4c 65  nLeakAlloc, pnLe
27d0: 61 6b 42 79 74 65 2c 20 70 46 69 6c 65 29 3b 0a  akByte, pFile);.
27e0: 20 20 7d 0a 7d 0a 0a 76 6f 69 64 20 74 65 73 74    }.}..void test
27f0: 4d 61 6c 6c 6f 63 4f 6f 6d 28 0a 20 20 6c 73 6d  MallocOom(.  lsm
2800: 5f 65 6e 76 20 2a 70 45 6e 76 2c 20 0a 20 20 69  _env *pEnv, .  i
2810: 6e 74 20 6e 43 6f 75 6e 74 64 6f 77 6e 2c 20 0a  nt nCountdown, .
2820: 20 20 69 6e 74 20 62 50 65 72 73 69 73 74 2c 0a    int bPersist,.
2830: 20 20 76 6f 69 64 20 28 2a 78 48 6f 6f 6b 29 28    void (*xHook)(
2840: 76 6f 69 64 20 2a 29 2c 0a 20 20 76 6f 69 64 20  void *),.  void 
2850: 2a 70 48 6f 6f 6b 43 74 78 0a 29 7b 0a 20 20 54  *pHookCtx.){.  T
2860: 6d 47 6c 6f 62 61 6c 20 2a 70 54 6d 20 3d 20 28  mGlobal *pTm = (
2870: 54 6d 47 6c 6f 62 61 6c 20 2a 29 28 70 45 6e 76  TmGlobal *)(pEnv
2880: 2d 3e 70 4d 65 6d 43 74 78 29 3b 0a 20 20 74 6d  ->pMemCtx);.  tm
2890: 4d 61 6c 6c 6f 63 4f 6f 6d 28 70 54 6d 2c 20 6e  MallocOom(pTm, n
28a0: 43 6f 75 6e 74 64 6f 77 6e 2c 20 62 50 65 72 73  Countdown, bPers
28b0: 69 73 74 2c 20 78 48 6f 6f 6b 2c 20 70 48 6f 6f  ist, xHook, pHoo
28c0: 6b 43 74 78 29 3b 0a 7d 0a 0a 76 6f 69 64 20 74  kCtx);.}..void t
28d0: 65 73 74 4d 61 6c 6c 6f 63 4f 6f 6d 45 6e 61 62  estMallocOomEnab
28e0: 6c 65 28 6c 73 6d 5f 65 6e 76 20 2a 70 45 6e 76  le(lsm_env *pEnv
28f0: 2c 20 69 6e 74 20 62 45 6e 61 62 6c 65 29 7b 0a  , int bEnable){.
2900: 20 20 54 6d 47 6c 6f 62 61 6c 20 2a 70 54 6d 20    TmGlobal *pTm 
2910: 3d 20 28 54 6d 47 6c 6f 62 61 6c 20 2a 29 28 70  = (TmGlobal *)(p
2920: 45 6e 76 2d 3e 70 4d 65 6d 43 74 78 29 3b 0a 20  Env->pMemCtx);. 
2930: 20 74 6d 4d 61 6c 6c 6f 63 4f 6f 6d 45 6e 61 62   tmMallocOomEnab
2940: 6c 65 28 70 54 6d 2c 20 62 45 6e 61 62 6c 65 29  le(pTm, bEnable)
2950: 3b 0a 7d 0a                                      ;.}.