| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131 |
- /* Copyright (C) 2001-2023 Artifex Software, Inc.
- All Rights Reserved.
- This software is provided AS-IS with no warranty, either express or
- implied.
- This software is distributed under license and may not be copied,
- modified or distributed except as expressly authorized under the terms
- of the license contained in the file LICENSE in this distribution.
- Refer to licensing information at http://www.artifex.com or contact
- Artifex Software, Inc., 39 Mesa Street, Suite 108A, San Francisco,
- CA 94129, USA, for further information.
- */
- /* Inspired by Fortify by Simon P Bullen. */
- /* Set the following if you're only looking for leaks, not memory overwrites
- * to speed the operation */
- /* #define MEMENTO_LEAKONLY */
- /* Set the following to keep extra details about the history of blocks */
- #define MEMENTO_DETAILS
- /* Don't keep blocks around if they'd mean losing more than a quarter of
- * the freelist. */
- #define MEMENTO_FREELIST_MAX_SINGLE_BLOCK (MEMENTO_FREELIST_MAX/4)
- #define COMPILING_MEMENTO_C
- /* SHUT UP, MSVC. I KNOW WHAT I AM DOING. */
- #define _CRT_SECURE_NO_WARNINGS
- /* We have some GS specific tweaks; more for the GS build environment than
- * anything else. */
- /* #define MEMENTO_GS_HACKS */
- #ifdef MEMENTO_GS_HACKS
- /* For GS we include malloc_.h. Anyone else would just include memento.h */
- #include "malloc_.h"
- #include "memory_.h"
- int atexit(void (*)(void));
- #else
- #include "memento.h"
- #include <stdio.h>
- #endif
- #ifndef _MSC_VER
- #include <stdint.h>
- #include <limits.h>
- #include <unistd.h>
- #endif
- #include <stdlib.h>
- #include <stdarg.h>
- #include <string.h>
- #ifdef __ANDROID__
- #define MEMENTO_ANDROID
- #include <stdio.h>
- #endif
- /* Hacks to portably print large sizes */
- #ifdef _MSC_VER
- #define FMTZ "%llu"
- #define FMTZ_CAST _int64
- #define FMTP "0x%p"
- #else
- #define FMTZ "%zu"
- #define FMTZ_CAST size_t
- #define FMTP "%p"
- #endif
- #define UB(x) ((intptr_t)((x) & 0xFF))
- #define B2I(x) (UB(x) | (UB(x)<<8) | (UB(x)<<16) | (UB(x)<<24))
- #define B2P(x) ((void *)(B2I(x) | ((B2I(x)<<16)<<16)))
- #define MEMENTO_PREFILL_UBYTE ((unsigned char)(MEMENTO_PREFILL))
- #define MEMENTO_PREFILL_USHORT (((unsigned short)MEMENTO_PREFILL_UBYTE) | (((unsigned short)MEMENTO_PREFILL_UBYTE)<<8))
- #define MEMENTO_PREFILL_UINT (((unsigned int)MEMENTO_PREFILL_USHORT) | (((unsigned int)MEMENTO_PREFILL_USHORT)<<16))
- #define MEMENTO_PREFILL_PTR (void *)(((uintptr_t)MEMENTO_PREFILL_UINT) | ((((uintptr_t)MEMENTO_PREFILL_UINT)<<16)<<16))
- #define MEMENTO_POSTFILL_UBYTE ((unsigned char)(MEMENTO_POSTFILL))
- #define MEMENTO_POSTFILL_USHORT (((unsigned short)MEMENTO_POSTFILL_UBYTE) | (((unsigned short)MEMENTO_POSTFILL_UBYTE)<<8))
- #define MEMENTO_POSTFILL_UINT (((unsigned int)MEMENTO_POSTFILL_USHORT) | (((unsigned int)MEMENTO_POSTFILL_USHORT)<<16))
- #define MEMENTO_POSTFILL_PTR (void *)(((uintptr_t)MEMENTO_POSTFILL_UINT) | ((((uintptr_t)MEMENTO_POSTFILL_UINT)<<16)<<16))
- #define MEMENTO_ALLOCFILL_UBYTE ((unsigned char)(MEMENTO_ALLOCFILL))
- #define MEMENTO_ALLOCFILL_USHORT (((unsigned short)MEMENTO_ALLOCFILL_UBYTE) | (((unsigned short)MEMENTO_ALLOCFILL_UBYTE)<<8))
- #define MEMENTO_ALLOCFILL_UINT (((unsigned int)MEMENTO_ALLOCFILL_USHORT) | (((unsigned int)MEMENTO_ALLOCFILL_USHORT)<<16))
- #define MEMENTO_ALLOCFILL_PTR (void *)(((uintptr_t)MEMENTO_ALLOCFILL_UINT) | ((((uintptr_t)MEMENTO_ALLOCFILL_UINT)<<16)<<16))
- #define MEMENTO_FREEFILL_UBYTE ((unsigned char)(MEMENTO_FREEFILL))
- #define MEMENTO_FREEFILL_USHORT (((unsigned short)MEMENTO_FREEFILL_UBYTE) | (((unsigned short)MEMENTO_FREEFILL_UBYTE)<<8))
- #define MEMENTO_FREEFILL_UINT (((unsigned int)MEMENTO_FREEFILL_USHORT) | (((unsigned int)MEMENTO_FREEFILL_USHORT)<<16))
- #define MEMENTO_FREEFILL_PTR (void *)(((uintptr_t)MEMENTO_FREEFILL_UINT) | ((((uintptr_t)MEMENTO_FREEFILL_UINT)<<16)<<16))
- #ifdef MEMENTO
- #ifndef MEMENTO_CPP_EXTRAS_ONLY
- #ifdef MEMENTO_ANDROID
- #include <android/log.h>
- static char log_buffer[4096];
- static int log_fill = 0;
- static char log_buffer2[4096];
- static int
- android_fprintf(FILE *file, const char *fmt, ...)
- {
- va_list args;
- char *p, *q;
- va_start(args, fmt);
- vsnprintf(log_buffer2, sizeof(log_buffer2)-1, fmt, args);
- va_end(args);
- /* Ensure we are always null terminated */
- log_buffer2[sizeof(log_buffer2)-1] = 0;
- p = log_buffer2;
- q = p;
- do
- {
- /* Find the end of the string, or the next \n */
- while (*p && *p != '\n')
- p++;
- /* We need to output from q to p. Limit ourselves to what
- * will fit in the existing */
- if (p - q >= sizeof(log_buffer)-1 - log_fill)
- p = q + sizeof(log_buffer)-1 - log_fill;
- memcpy(&log_buffer[log_fill], q, p-q);
- log_fill += p-q;
- if (*p == '\n')
- {
- log_buffer[log_fill] = 0;
- __android_log_print(ANDROID_LOG_ERROR, "memento", "%s", log_buffer);
- usleep(1);
- log_fill = 0;
- p++; /* Skip over the \n */
- }
- else if (log_fill >= sizeof(log_buffer)-1)
- {
- log_buffer[sizeof(log_buffer2)-1] = 0;
- __android_log_print(ANDROID_LOG_ERROR, "memento", "%s", log_buffer);
- usleep(1);
- log_fill = 0;
- }
- q = p;
- }
- while (*p);
- return 0;
- }
- #define fprintf android_fprintf
- #define MEMENTO_STACKTRACE_METHOD 3
- #endif
- /* _WIN64 defined implies _WIN32 will be */
- #ifdef _WIN32
- #include <windows.h>
- static int
- windows_fprintf(FILE *file, const char *fmt, ...)
- {
- va_list args;
- char text[4096];
- int ret;
- va_start(args, fmt);
- ret = vfprintf(file, fmt, args);
- va_end(args);
- va_start(args, fmt);
- vsnprintf(text, 4096, fmt, args);
- OutputDebugStringA(text);
- va_end(args);
- return ret;
- }
- #define fprintf windows_fprintf
- #endif
- #ifndef MEMENTO_STACKTRACE_METHOD
- #ifdef __GNUC__
- #define MEMENTO_STACKTRACE_METHOD 1
- #endif
- #ifdef _WIN32
- #define MEMENTO_STACKTRACE_METHOD 2
- #endif
- #endif
- #if defined(__linux__)
- #define MEMENTO_HAS_FORK
- #elif defined(__APPLE__) && defined(__MACH__)
- #define MEMENTO_HAS_FORK
- #endif
- /* Define the underlying allocators, just in case */
- void *MEMENTO_UNDERLYING_MALLOC(size_t);
- void MEMENTO_UNDERLYING_FREE(void *);
- void *MEMENTO_UNDERLYING_REALLOC(void *,size_t);
- void *MEMENTO_UNDERLYING_CALLOC(size_t,size_t);
- /* And some other standard functions we use. We don't include the header
- * files, just in case they pull in unexpected others. */
- int atoi(const char *);
- char *getenv(const char *);
- /* How far to search for pointers in each block when calculating nestings */
- /* mupdf needs at least 34000ish (sizeof(fz_shade))/ */
- #define MEMENTO_PTRSEARCH 65536
- #ifndef MEMENTO_MAXPATTERN
- #define MEMENTO_MAXPATTERN 0
- #endif
- #ifdef MEMENTO_GS_HACKS
- #include "valgrind.h"
- #else
- #ifdef HAVE_VALGRIND
- #include "valgrind/memcheck.h"
- #else
- #define VALGRIND_MAKE_MEM_NOACCESS(p,s) do { } while (0==1)
- #define VALGRIND_MAKE_MEM_UNDEFINED(p,s) do { } while (0==1)
- #define VALGRIND_MAKE_MEM_DEFINED(p,s) do { } while (0==1)
- #endif
- #endif
- enum {
- Memento_PreSize = 16,
- Memento_PostSize = 16
- };
- /* Some compile time checks */
- typedef struct
- {
- char MEMENTO_PRESIZE_MUST_BE_A_MULTIPLE_OF_4[Memento_PreSize & 3 ? -1 : 1];
- char MEMENTO_POSTSIZE_MUST_BE_A_MULTIPLE_OF_4[Memento_PostSize & 3 ? -1 : 1];
- char MEMENTO_POSTSIZE_MUST_BE_AT_LEAST_4[Memento_PostSize >= 4 ? 1 : -1];
- char MEMENTO_PRESIZE_MUST_BE_AT_LEAST_4[Memento_PreSize >= 4 ? 1 : -1];
- } MEMENTO_SANITY_CHECK_STRUCT;
- #define MEMENTO_UINT32 unsigned int
- #define MEMENTO_UINT16 unsigned short
- #define MEMENTO_PREFILL_UINT32 ((MEMENTO_UINT32)(MEMENTO_PREFILL | (MEMENTO_PREFILL <<8) | (MEMENTO_PREFILL <<16) |(MEMENTO_PREFILL <<24)))
- #define MEMENTO_POSTFILL_UINT16 ((MEMENTO_UINT16)(MEMENTO_POSTFILL | (MEMENTO_POSTFILL<<8)))
- #define MEMENTO_POSTFILL_UINT32 ((MEMENTO_UINT32)(MEMENTO_POSTFILL | (MEMENTO_POSTFILL<<8) | (MEMENTO_POSTFILL<<16) |(MEMENTO_POSTFILL<<24)))
- #define MEMENTO_FREEFILL_UINT16 ((MEMENTO_UINT16)(MEMENTO_FREEFILL | (MEMENTO_FREEFILL<<8)))
- #define MEMENTO_FREEFILL_UINT32 ((MEMENTO_UINT32)(MEMENTO_FREEFILL | (MEMENTO_FREEFILL<<8) | (MEMENTO_FREEFILL<<16) |(MEMENTO_FREEFILL<<24)))
- enum {
- Memento_Flag_OldBlock = 1,
- Memento_Flag_HasParent = 2,
- Memento_Flag_BreakOnFree = 4,
- Memento_Flag_BreakOnRealloc = 8,
- Memento_Flag_Freed = 16,
- Memento_Flag_KnownLeak = 32,
- Memento_Flag_Reported = 64
- };
- enum {
- Memento_EventType_malloc = 0,
- Memento_EventType_calloc = 1,
- Memento_EventType_realloc = 2,
- Memento_EventType_free = 3,
- Memento_EventType_new = 4,
- Memento_EventType_delete = 5,
- Memento_EventType_newArray = 6,
- Memento_EventType_deleteArray = 7,
- Memento_EventType_takeRef = 8,
- Memento_EventType_dropRef = 9,
- Memento_EventType_reference = 10
- };
- static const char *eventType[] =
- {
- "malloc",
- "calloc",
- "realloc",
- "free",
- "new",
- "delete",
- "new[]",
- "delete[]",
- "takeRef",
- "dropRef",
- "reference"
- };
- /* When we list leaked blocks at the end of execution, we search for pointers
- * between blocks in order to be able to give a nice nested view.
- * Unfortunately, if you have are running your own allocator (such as
- * postscript's chunk allocator) you can often find that the header of the
- * block always contains pointers to next or previous blocks. This tends to
- * mean the nesting displayed is "uninteresting" at best :)
- *
- * As a hack to get around this, we have a define MEMENTO_SKIP_SEARCH that
- * indicates how many bytes to skip over at the start of the chunk.
- * This may cause us to miss true nestings, but such is life...
- */
- #ifndef MEMENTO_SEARCH_SKIP
- #ifdef MEMENTO_GS_HACKS
- #define MEMENTO_SEARCH_SKIP (2*sizeof(void *))
- #else
- #define MEMENTO_SEARCH_SKIP 0
- #endif
- #endif
- #define MEMENTO_CHILD_MAGIC ((Memento_BlkHeader *)('M' | ('3' << 8) | ('m' << 16) | ('3' << 24)))
- #define MEMENTO_SIBLING_MAGIC ((Memento_BlkHeader *)('n' | ('t' << 8) | ('0' << 16) | ('!' << 24)))
- #ifdef MEMENTO_DETAILS
- typedef struct Memento_BlkDetails Memento_BlkDetails;
- struct Memento_BlkDetails
- {
- Memento_BlkDetails *next;
- char type;
- char count;
- int sequence;
- void *stack[1];
- };
- #endif /* MEMENTO_DETAILS */
- typedef struct Memento_BlkHeader Memento_BlkHeader;
- struct Memento_BlkHeader
- {
- size_t rawsize;
- int sequence;
- int lastCheckedOK;
- int flags;
- Memento_BlkHeader *next;
- Memento_BlkHeader *prev; /* Reused as 'parent' when printing nested list */
- const char *label;
- /* Entries for nesting display calculations. Set to magic
- * values at all other time. */
- Memento_BlkHeader *child;
- Memento_BlkHeader *sibling;
- #ifdef MEMENTO_DETAILS
- Memento_BlkDetails *details;
- Memento_BlkDetails **details_tail;
- #endif
- char preblk[Memento_PreSize];
- };
- /* In future this could (should) be a smarter data structure, like, say,
- * splay trees. For now, we use a list.
- */
- typedef struct Memento_Blocks
- {
- Memento_BlkHeader *head;
- Memento_BlkHeader *tail;
- } Memento_Blocks;
- /* What sort of Mutex should we use? */
- #ifdef MEMENTO_LOCKLESS
- typedef int Memento_mutex;
- static void Memento_initMutex(Memento_mutex *m)
- {
- (void)m;
- }
- #define MEMENTO_DO_LOCK() do { } while (0)
- #define MEMENTO_DO_UNLOCK() do { } while (0)
- #else
- #if defined(_WIN32) || defined(_WIN64)
- /* Windows */
- typedef CRITICAL_SECTION Memento_mutex;
- static void Memento_initMutex(Memento_mutex *m)
- {
- InitializeCriticalSection(m);
- }
- #define MEMENTO_DO_LOCK() \
- EnterCriticalSection(&memento.mutex)
- #define MEMENTO_DO_UNLOCK() \
- LeaveCriticalSection(&memento.mutex)
- #else
- #include <pthread.h>
- typedef pthread_mutex_t Memento_mutex;
- static void Memento_initMutex(Memento_mutex *m)
- {
- pthread_mutex_init(m, NULL);
- }
- #define MEMENTO_DO_LOCK() \
- pthread_mutex_lock(&memento.mutex)
- #define MEMENTO_DO_UNLOCK() \
- pthread_mutex_unlock(&memento.mutex)
- #endif
- #endif
- /* And our global structure */
- static struct {
- int inited;
- Memento_Blocks used;
- Memento_Blocks free;
- size_t freeListSize;
- int sequence;
- int paranoia;
- int paranoidAt;
- int countdown;
- int lastChecked;
- int breakAt;
- int failAt;
- int failing;
- int nextFailAt;
- int squeezeAt;
- int squeezing;
- int segv;
- int pattern;
- int nextPattern;
- int patternBit;
- int leaking;
- size_t maxMemory;
- size_t alloc;
- size_t peakAlloc;
- size_t totalAlloc;
- size_t numMallocs;
- size_t numFrees;
- size_t numReallocs;
- Memento_mutex mutex;
- } memento;
- #define MEMENTO_EXTRASIZE (sizeof(Memento_BlkHeader) + Memento_PostSize)
- /* Round up size S to the next multiple of N (where N is a power of 2) */
- #define MEMENTO_ROUNDUP(S,N) ((S + N-1)&~(N-1))
- #define MEMBLK_SIZE(s) MEMENTO_ROUNDUP(s + MEMENTO_EXTRASIZE, MEMENTO_MAXALIGN)
- #define MEMBLK_FROMBLK(B) (&((Memento_BlkHeader*)(void *)(B))[-1])
- #define MEMBLK_TOBLK(B) ((void*)(&((Memento_BlkHeader*)(void*)(B))[1]))
- #define MEMBLK_POSTPTR(B) \
- (&((unsigned char *)(void *)(B))[(B)->rawsize + sizeof(Memento_BlkHeader)])
- enum
- {
- SkipStackBackTraceLevels = 4
- };
- #if defined(MEMENTO_STACKTRACE_METHOD) && MEMENTO_STACKTRACE_METHOD == 1
- extern size_t backtrace(void **, int);
- extern void backtrace_symbols_fd(void **, size_t, int);
- extern char **backtrace_symbols(void **, size_t);
- #define MEMENTO_BACKTRACE_MAX 256
- static void (*print_stack_value)(void *address);
- /* Libbacktrace gubbins - relies on us having libdl to load the .so */
- #ifdef HAVE_LIBDL
- #include <dlfcn.h>
- typedef void (*backtrace_error_callback) (void *data, const char *msg, int errnum);
- typedef struct backtrace_state *(*backtrace_create_state_type)(
- const char *filename, int threaded,
- backtrace_error_callback error_callback, void *data);
- typedef int (*backtrace_full_callback) (void *data, uintptr_t pc,
- const char *filename, int lineno,
- const char *function);
- typedef int (*backtrace_pcinfo_type)(struct backtrace_state *state,
- uintptr_t pc,
- backtrace_full_callback callback,
- backtrace_error_callback error_callback,
- void *data);
- typedef void (*backtrace_syminfo_callback) (void *data, uintptr_t pc,
- const char *symname,
- uintptr_t symval,
- uintptr_t symsize);
- typedef int (*backtrace_syminfo_type)(struct backtrace_state *state,
- uintptr_t addr,
- backtrace_syminfo_callback callback,
- backtrace_error_callback error_callback,
- void *data);
- static backtrace_syminfo_type backtrace_syminfo;
- static backtrace_create_state_type backtrace_create_state;
- static backtrace_pcinfo_type backtrace_pcinfo;
- static struct backtrace_state *my_backtrace_state;
- static void *libbt;
- static char backtrace_exe[4096];
- static void *current_addr;
- static void error2_cb(void *data, const char *msg, int errnum)
- {
- }
- static void syminfo_cb(void *data, uintptr_t pc, const char *symname, uintptr_t symval, uintptr_t symsize)
- {
- if (sizeof(void *) == 4)
- fprintf(stderr, " 0x%08lx %s\n", pc, symname?symname:"?");
- else
- fprintf(stderr, " 0x%016lx %s\n", pc, symname?symname:"?");
- }
- static void error_cb(void *data, const char *msg, int errnum)
- {
- backtrace_syminfo(my_backtrace_state,
- (uintptr_t)current_addr,
- syminfo_cb,
- error2_cb,
- NULL);
- }
- static int full_cb(void *data, uintptr_t pc, const char *fname, int line, const char *fn)
- {
- if (sizeof(void *) == 4)
- fprintf(stderr, " 0x%08lx %s(%s:%d)\n", pc, fn?fn:"?", fname?fname:"?", line);
- else
- fprintf(stderr, " 0x%016lx %s(%s:%d)\n", pc, fn?fn:"?", fname?fname:"?", line);
- return 0;
- }
- static void print_stack_libbt(void *addr)
- {
- current_addr = addr;
- backtrace_pcinfo(my_backtrace_state,
- (uintptr_t)addr,
- full_cb,
- error_cb,
- NULL);
- }
- static void print_stack_libbt_failed(void *addr)
- {
- char **strings;
- #if 0
- /* Let's use a hack from Julian Smith to call gdb to extract the information */
- /* Disabled for now, as I can't make this work. */
- static char command[1024];
- int e;
- static int gdb_invocation_failed = 0;
- if (gdb_invocation_failed == 0)
- {
- snprintf(command, sizeof(command),
- //"gdb -q --batch -p=%i -ex 'info line *%p' -ex quit 2>/dev/null",
- "gdb -q --batch -p=%i -ex 'info line *%p' -ex quit 2>/dev/null| egrep -v '(Thread debugging using)|(Using host libthread_db library)|(A debugging session is active)|(will be detached)|(Quit anyway)|(No such file or directory)|(^0x)|(^$)'",
- getpid(), addr);
- printf("%s\n", command);
- e = system(command);
- if (e == 0)
- return; /* That'll do! */
- gdb_invocation_failed = 1; /* If it's failed once, it'll probably keep failing. */
- }
- #endif
- /* We couldn't even get gdb! Make do. */
- strings = backtrace_symbols(&addr, 1);
- if (strings == NULL || strings[0] == NULL)
- {
- if (sizeof(void *) == 4)
- fprintf(stderr, " [0x%08lx]\n", (uintptr_t)addr);
- else
- fprintf(stderr, " [0x%016lx]\n", (uintptr_t)addr);
- }
- else
- {
- fprintf(stderr, " %s\n", strings[0]);
- }
- (free)(strings);
- }
- static int init_libbt(void)
- {
- static int libbt_inited = 0;
- if (libbt_inited)
- return 0;
- libbt_inited = 1;
- libbt = dlopen("libbacktrace.so", RTLD_LAZY);
- if (libbt == NULL)
- libbt = dlopen("/opt/lib/libbacktrace.so", RTLD_LAZY);
- if (libbt == NULL)
- libbt = dlopen("/lib/libbacktrace.so", RTLD_LAZY);
- if (libbt == NULL)
- libbt = dlopen("/usr/lib/libbacktrace.so", RTLD_LAZY);
- if (libbt == NULL)
- libbt = dlopen("/usr/local/lib/libbacktrace.so", RTLD_LAZY);
- if (libbt == NULL)
- goto fail;
- backtrace_create_state = dlsym(libbt, "backtrace_create_state");
- backtrace_syminfo = dlsym(libbt, "backtrace_syminfo");
- backtrace_pcinfo = dlsym(libbt, "backtrace_pcinfo");
- if (backtrace_create_state == NULL ||
- backtrace_syminfo == NULL ||
- backtrace_pcinfo == NULL)
- {
- goto fail;
- }
- my_backtrace_state = backtrace_create_state(backtrace_exe,
- 1 /*BACKTRACE_SUPPORTS_THREADS*/,
- error_cb,
- NULL);
- if (my_backtrace_state == NULL)
- goto fail;
- print_stack_value = print_stack_libbt;
- return 1;
- fail:
- fprintf(stderr,
- "MEMENTO: libbacktrace.so failed to load; backtraces will be sparse.\n"
- "MEMENTO: See memento.h for how to rectify this.\n");
- libbt = NULL;
- backtrace_create_state = NULL;
- backtrace_syminfo = NULL;
- print_stack_value = print_stack_libbt_failed;
- return 0;
- }
- #endif
- static void print_stack_default(void *addr)
- {
- char **strings = backtrace_symbols(&addr, 1);
- if (strings == NULL || strings[0] == NULL)
- {
- fprintf(stderr, " ["FMTP"]\n", addr);
- }
- #ifdef HAVE_LIBDL
- else if (strchr(strings[0], ':') == NULL)
- {
- /* Probably a "path [address]" format string */
- char *s = strchr(strings[0], ' ');
- if (s != strings[0])
- {
- memcpy(backtrace_exe, strings[0], s - strings[0]);
- backtrace_exe[s-strings[0]] = 0;
- init_libbt();
- print_stack_value(addr);
- }
- }
- #endif
- else
- {
- fprintf(stderr, " %s\n", strings[0]);
- }
- free(strings);
- }
- static void Memento_initStacktracer(void)
- {
- print_stack_value = print_stack_default;
- }
- static int Memento_getStacktrace(void **stack, int *skip)
- {
- size_t num;
- num = backtrace(&stack[0], MEMENTO_BACKTRACE_MAX);
- *skip = SkipStackBackTraceLevels;
- if (num <= SkipStackBackTraceLevels)
- return 0;
- return (int)(num-SkipStackBackTraceLevels);
- }
- static void Memento_showStacktrace(void **stack, int numberOfFrames)
- {
- int i;
- for (i = 0; i < numberOfFrames; i++)
- {
- print_stack_value(stack[i]);
- }
- }
- #elif defined(MEMENTO_STACKTRACE_METHOD) && MEMENTO_STACKTRACE_METHOD == 2
- #include <Windows.h>
- /* We use DbgHelp.dll rather than DbgHelp.lib. This avoids us needing
- * extra link time complications, and enables us to fall back gracefully
- * if the DLL cannot be found.
- *
- * To achieve this we have our own potted versions of the required types
- * inline here.
- */
- #ifdef _WIN64
- typedef DWORD64 DWORD_NATIVESIZED;
- #else
- typedef DWORD DWORD_NATIVESIZED;
- #endif
- #define MEMENTO_BACKTRACE_MAX 64
- typedef USHORT (__stdcall *My_CaptureStackBackTraceType)(__in ULONG, __in ULONG, __out PVOID*, __out_opt PULONG);
- typedef struct MY_IMAGEHLP_LINE {
- DWORD SizeOfStruct;
- PVOID Key;
- DWORD LineNumber;
- PCHAR FileName;
- DWORD_NATIVESIZED Address;
- } MY_IMAGEHLP_LINE, *MY_PIMAGEHLP_LINE;
- typedef BOOL (__stdcall *My_SymGetLineFromAddrType)(HANDLE hProcess, DWORD_NATIVESIZED dwAddr, PDWORD pdwDisplacement, MY_PIMAGEHLP_LINE Line);
- typedef struct MY_SYMBOL_INFO {
- ULONG SizeOfStruct;
- ULONG TypeIndex; // Type Index of symbol
- ULONG64 Reserved[2];
- ULONG info;
- ULONG Size;
- ULONG64 ModBase; // Base Address of module containing this symbol
- ULONG Flags;
- ULONG64 Value; // Value of symbol, ValuePresent should be 1
- ULONG64 Address; // Address of symbol including base address of module
- ULONG Register; // register holding value or pointer to value
- ULONG Scope; // scope of the symbol
- ULONG Tag; // pdb classification
- ULONG NameLen; // Actual length of name
- ULONG MaxNameLen;
- CHAR Name[1]; // Name of symbol
- } MY_SYMBOL_INFO, *MY_PSYMBOL_INFO;
- typedef BOOL (__stdcall *My_SymFromAddrType)(HANDLE hProcess, DWORD64 Address, PDWORD64 Displacement, MY_PSYMBOL_INFO Symbol);
- typedef BOOL (__stdcall *My_SymInitializeType)(HANDLE hProcess, PSTR UserSearchPath, BOOL fInvadeProcess);
- static My_CaptureStackBackTraceType Memento_CaptureStackBackTrace;
- static My_SymGetLineFromAddrType Memento_SymGetLineFromAddr;
- static My_SymFromAddrType Memento_SymFromAddr;
- static My_SymInitializeType Memento_SymInitialize;
- static HANDLE Memento_process;
- static void Memento_initStacktracer(void)
- {
- HMODULE mod = LoadLibrary("kernel32.dll");
- if (mod == NULL)
- return;
- Memento_CaptureStackBackTrace = (My_CaptureStackBackTraceType)(GetProcAddress(mod, "RtlCaptureStackBackTrace"));
- if (Memento_CaptureStackBackTrace == NULL)
- return;
- mod = LoadLibrary("Dbghelp.dll");
- if (mod == NULL) {
- Memento_CaptureStackBackTrace = NULL;
- return;
- }
- Memento_SymGetLineFromAddr =
- (My_SymGetLineFromAddrType)(GetProcAddress(mod,
- #ifdef _WIN64
- "SymGetLineFromAddr64"
- #else
- "SymGetLineFromAddr"
- #endif
- ));
- if (Memento_SymGetLineFromAddr == NULL) {
- Memento_CaptureStackBackTrace = NULL;
- return;
- }
- Memento_SymFromAddr = (My_SymFromAddrType)(GetProcAddress(mod, "SymFromAddr"));
- if (Memento_SymFromAddr == NULL) {
- Memento_CaptureStackBackTrace = NULL;
- return;
- }
- Memento_SymInitialize = (My_SymInitializeType)(GetProcAddress(mod, "SymInitialize"));
- if (Memento_SymInitialize == NULL) {
- Memento_CaptureStackBackTrace = NULL;
- return;
- }
- Memento_process = GetCurrentProcess();
- Memento_SymInitialize(Memento_process, NULL, TRUE);
- }
- static int Memento_getStacktrace(void **stack, int *skip)
- {
- if (Memento_CaptureStackBackTrace == NULL)
- return 0;
- *skip = 0;
- /* Limit us to 63 levels due to windows bug */
- return Memento_CaptureStackBackTrace(SkipStackBackTraceLevels, 63-SkipStackBackTraceLevels, stack, NULL);
- }
- static void Memento_showStacktrace(void **stack, int numberOfFrames)
- {
- MY_IMAGEHLP_LINE line;
- int i;
- char symbol_buffer[sizeof(MY_SYMBOL_INFO) + 1024 + 1];
- MY_SYMBOL_INFO *symbol = (MY_SYMBOL_INFO *)symbol_buffer;
- symbol->MaxNameLen = 1024;
- symbol->SizeOfStruct = sizeof(MY_SYMBOL_INFO);
- line.SizeOfStruct = sizeof(MY_IMAGEHLP_LINE);
- for (i = 0; i < numberOfFrames; i++)
- {
- DWORD64 dwDisplacement64;
- DWORD dwDisplacement;
- Memento_SymFromAddr(Memento_process, (DWORD64)(stack[i]), &dwDisplacement64, symbol);
- Memento_SymGetLineFromAddr(Memento_process, (DWORD_NATIVESIZED)(stack[i]), &dwDisplacement, &line);
- fprintf(stderr, " %s in %s:%d\n", symbol->Name, line.FileName, line.LineNumber);
- }
- }
- #elif defined(MEMENTO_STACKTRACE_METHOD) && MEMENTO_STACKTRACE_METHOD == 3
- #include <unwind.h>
- #include <dlfcn.h>
- /* From cxxabi.h */
- extern char* __cxa_demangle(const char* mangled_name,
- char* output_buffer,
- size_t* length,
- int* status);
- static void Memento_initStacktracer(void)
- {
- }
- #define MEMENTO_BACKTRACE_MAX 256
- typedef struct
- {
- int count;
- void **addr;
- } my_unwind_details;
- static _Unwind_Reason_Code unwind_populate_callback(struct _Unwind_Context *context,
- void *arg)
- {
- my_unwind_details *uw = (my_unwind_details *)arg;
- int count = uw->count;
- if (count >= MEMENTO_BACKTRACE_MAX)
- return _URC_END_OF_STACK;
- uw->addr[count] = (void *)_Unwind_GetIP(context);
- uw->count++;
- return _URC_NO_REASON;
- }
- static int Memento_getStacktrace(void **stack, int *skip)
- {
- my_unwind_details uw = { 0, stack };
- *skip = 0;
- /* Collect the backtrace. Deliberately only unwind once,
- * and avoid using malloc etc until this completes just
- * in case. */
- _Unwind_Backtrace(unwind_populate_callback, &uw);
- if (uw.count <= SkipStackBackTraceLevels)
- return 0;
- *skip = SkipStackBackTraceLevels;
- return uw.count-SkipStackBackTraceLevels;
- }
- static void Memento_showStacktrace(void **stack, int numberOfFrames)
- {
- int i;
- for (i = 0; i < numberOfFrames; i++)
- {
- Dl_info info;
- if (dladdr(stack[i], &info))
- {
- int status = 0;
- const char *sym = info.dli_sname ? info.dli_sname : "<unknown>";
- char *demangled = __cxa_demangle(sym, NULL, 0, &status);
- int offset = stack[i] - info.dli_saddr;
- fprintf(stderr, " ["FMTP"]%s(+0x%x)\n", stack[i], demangled && status == 0 ? demangled : sym, offset);
- free(demangled);
- }
- else
- {
- fprintf(stderr, " ["FMTP"]\n", stack[i]);
- }
- }
- }
- #else
- static void Memento_initStacktracer(void)
- {
- }
- static int Memento_getStacktrace(void **stack, int *skip)
- {
- *skip = 0;
- return 0;
- }
- static void Memento_showStacktrace(void **stack, int numberOfFrames)
- {
- }
- #endif /* MEMENTO_STACKTRACE_METHOD */
- #ifdef MEMENTO_DETAILS
- static void Memento_storeDetails(Memento_BlkHeader *head, int type)
- {
- void *stack[MEMENTO_BACKTRACE_MAX];
- Memento_BlkDetails *details;
- int count;
- int skip;
- if (head == NULL)
- return;
- #ifdef MEMENTO_STACKTRACE_METHOD
- count = Memento_getStacktrace(stack, &skip);
- #else
- skip = 0;
- count = 0;
- #endif
- details = MEMENTO_UNDERLYING_MALLOC(sizeof(*details) + (count-1) * sizeof(void *));
- if (details == NULL)
- return;
- if (count)
- memcpy(&details->stack, &stack[skip], count * sizeof(void *));
- details->type = type;
- details->count = count;
- details->sequence = memento.sequence;
- details->next = NULL;
- VALGRIND_MAKE_MEM_DEFINED(&head->details_tail, sizeof(head->details_tail));
- *head->details_tail = details;
- head->details_tail = &details->next;
- VALGRIND_MAKE_MEM_NOACCESS(&head->details_tail, sizeof(head->details_tail));
- }
- #endif
- void (Memento_bt)(void)
- {
- #ifdef MEMENTO_STACKTRACE_METHOD
- void *stack[MEMENTO_BACKTRACE_MAX];
- int count;
- int skip;
- count = Memento_getStacktrace(stack, &skip);
- Memento_showStacktrace(&stack[skip-2], count-skip+2);
- #endif
- }
- static void Memento_bt_internal(int skip2)
- {
- #ifdef MEMENTO_STACKTRACE_METHOD
- void *stack[MEMENTO_BACKTRACE_MAX];
- int count;
- int skip;
- count = Memento_getStacktrace(stack, &skip);
- Memento_showStacktrace(&stack[skip+skip2], count-skip-skip2);
- #endif
- }
- static int Memento_checkAllMemoryLocked(void);
- void Memento_breakpoint(void)
- {
- /* A handy externally visible function for breakpointing */
- #if 0 /* Enable this to force automatic breakpointing */
- #ifndef NDEBUG
- #ifdef _MSC_VER
- __asm int 3;
- #endif
- #endif
- #endif
- }
- static void Memento_init(void);
- #define MEMENTO_LOCK() \
- do { if (!memento.inited) Memento_init(); MEMENTO_DO_LOCK(); } while (0)
- #define MEMENTO_UNLOCK() \
- do { MEMENTO_DO_UNLOCK(); } while (0)
- /* Do this as a macro to prevent another level in the callstack,
- * which is annoying while stepping. */
- #define Memento_breakpointLocked() \
- do { MEMENTO_UNLOCK(); Memento_breakpoint(); MEMENTO_LOCK(); } while (0)
- static void Memento_addBlockHead(Memento_Blocks *blks,
- Memento_BlkHeader *b,
- int type)
- {
- if (blks->tail == NULL)
- blks->tail = b;
- b->next = blks->head;
- b->prev = NULL;
- if (blks->head)
- {
- VALGRIND_MAKE_MEM_DEFINED(&blks->head->prev, sizeof(blks->head->prev));
- blks->head->prev = b;
- VALGRIND_MAKE_MEM_NOACCESS(&blks->head->prev, sizeof(blks->head->prev));
- }
- blks->head = b;
- #ifndef MEMENTO_LEAKONLY
- memset(b->preblk, MEMENTO_PREFILL, Memento_PreSize);
- memset(MEMBLK_POSTPTR(b), MEMENTO_POSTFILL, Memento_PostSize);
- #endif
- VALGRIND_MAKE_MEM_NOACCESS(MEMBLK_POSTPTR(b), Memento_PostSize);
- if (type == 0) { /* malloc */
- VALGRIND_MAKE_MEM_UNDEFINED(MEMBLK_TOBLK(b), b->rawsize);
- } else if (type == 1) { /* free */
- VALGRIND_MAKE_MEM_NOACCESS(MEMBLK_TOBLK(b), b->rawsize);
- }
- VALGRIND_MAKE_MEM_NOACCESS(b, sizeof(Memento_BlkHeader));
- }
- static void Memento_addBlockTail(Memento_Blocks *blks,
- Memento_BlkHeader *b,
- int type)
- {
- VALGRIND_MAKE_MEM_DEFINED(&blks->tail, sizeof(Memento_BlkHeader *));
- if (blks->head == NULL)
- blks->head = b;
- b->prev = blks->tail;
- b->next = NULL;
- if (blks->tail) {
- VALGRIND_MAKE_MEM_DEFINED(&blks->tail->next, sizeof(blks->tail->next));
- blks->tail->next = b;
- VALGRIND_MAKE_MEM_NOACCESS(&blks->tail->next, sizeof(blks->tail->next));
- }
- blks->tail = b;
- #ifndef MEMENTO_LEAKONLY
- memset(b->preblk, MEMENTO_PREFILL, Memento_PreSize);
- memset(MEMBLK_POSTPTR(b), MEMENTO_POSTFILL, Memento_PostSize);
- #endif
- VALGRIND_MAKE_MEM_NOACCESS(MEMBLK_POSTPTR(b), Memento_PostSize);
- if (type == 0) { /* malloc */
- VALGRIND_MAKE_MEM_UNDEFINED(MEMBLK_TOBLK(b), b->rawsize);
- } else if (type == 1) { /* free */
- VALGRIND_MAKE_MEM_NOACCESS(MEMBLK_TOBLK(b), b->rawsize);
- }
- VALGRIND_MAKE_MEM_NOACCESS(b, sizeof(Memento_BlkHeader));
- VALGRIND_MAKE_MEM_NOACCESS(&blks->tail, sizeof(Memento_BlkHeader *));
- }
- typedef struct BlkCheckData {
- int found;
- int preCorrupt;
- int postCorrupt;
- int freeCorrupt;
- size_t index;
- } BlkCheckData;
- #ifndef MEMENTO_LEAKONLY
- static int Memento_Internal_checkAllocedBlock(Memento_BlkHeader *b, void *arg)
- {
- int i;
- MEMENTO_UINT32 *ip;
- unsigned char *p;
- BlkCheckData *data = (BlkCheckData *)arg;
- ip = (MEMENTO_UINT32 *)(void *)(b->preblk);
- i = Memento_PreSize>>2;
- do {
- if (*ip++ != MEMENTO_PREFILL_UINT32)
- goto pre_corrupt;
- } while (--i);
- if (0) {
- pre_corrupt:
- data->preCorrupt = 1;
- }
- /* Postfill may not be aligned, so have to be slower */
- p = MEMBLK_POSTPTR(b);
- i = Memento_PostSize-4;
- if ((intptr_t)p & 1)
- {
- if (*p++ != MEMENTO_POSTFILL)
- goto post_corrupt;
- i--;
- }
- if ((intptr_t)p & 2)
- {
- if (*(MEMENTO_UINT16 *)p != MEMENTO_POSTFILL_UINT16)
- goto post_corrupt;
- p += 2;
- i -= 2;
- }
- do {
- if (*(MEMENTO_UINT32 *)p != MEMENTO_POSTFILL_UINT32)
- goto post_corrupt;
- p += 4;
- i -= 4;
- } while (i >= 0);
- if (i & 2)
- {
- if (*(MEMENTO_UINT16 *)p != MEMENTO_POSTFILL_UINT16)
- goto post_corrupt;
- p += 2;
- }
- if (i & 1)
- {
- if (*p != MEMENTO_POSTFILL)
- goto post_corrupt;
- }
- if (0) {
- post_corrupt:
- data->postCorrupt = 1;
- }
- if ((data->freeCorrupt | data->preCorrupt | data->postCorrupt) == 0) {
- b->lastCheckedOK = memento.sequence;
- }
- data->found |= 1;
- return 0;
- }
- static int Memento_Internal_checkFreedBlock(Memento_BlkHeader *b, void *arg)
- {
- size_t i;
- unsigned char *p;
- BlkCheckData *data = (BlkCheckData *)arg;
- p = MEMBLK_TOBLK(b); /* p will always be aligned */
- i = b->rawsize;
- /* Attempt to speed this up by checking an (aligned) int at a time */
- if (i >= 4) {
- i -= 4;
- do {
- if (*(MEMENTO_UINT32 *)p != MEMENTO_FREEFILL_UINT32)
- goto mismatch4;
- p += 4;
- i -= 4;
- } while (i > 0);
- i += 4;
- }
- if (i & 2) {
- if (*(MEMENTO_UINT16 *)p != MEMENTO_FREEFILL_UINT16)
- goto mismatch;
- p += 2;
- i -= 2;
- }
- if (0) {
- mismatch4:
- i += 4;
- }
- mismatch:
- while (i) {
- if (*p++ != (unsigned char)MEMENTO_FREEFILL)
- break;
- i--;
- }
- if (i) {
- data->freeCorrupt = 1;
- data->index = b->rawsize-i;
- }
- return Memento_Internal_checkAllocedBlock(b, arg);
- }
- #endif /* MEMENTO_LEAKONLY */
- static void Memento_removeBlock(Memento_Blocks *blks,
- Memento_BlkHeader *b)
- {
- VALGRIND_MAKE_MEM_DEFINED(b, sizeof(*b));
- if (b->next) {
- VALGRIND_MAKE_MEM_DEFINED(&b->next->prev, sizeof(b->next->prev));
- b->next->prev = b->prev;
- VALGRIND_MAKE_MEM_NOACCESS(&b->next->prev, sizeof(b->next->prev));
- }
- if (b->prev) {
- VALGRIND_MAKE_MEM_DEFINED(&b->prev->next, sizeof(b->prev->next));
- b->prev->next = b->next;
- VALGRIND_MAKE_MEM_NOACCESS(&b->prev->next, sizeof(b->prev->next));
- }
- if (blks->tail == b)
- blks->tail = b->prev;
- if (blks->head == b)
- blks->head = b->next;
- }
- static void free_block(Memento_BlkHeader *head)
- {
- #ifdef MEMENTO_DETAILS
- Memento_BlkDetails *details = head->details;
- while (details)
- {
- Memento_BlkDetails *next = details->next;
- MEMENTO_UNDERLYING_FREE(details);
- details = next;
- }
- #endif
- MEMENTO_UNDERLYING_FREE(head);
- }
- static int Memento_Internal_makeSpace(size_t space)
- {
- /* If too big, it can never go on the freelist */
- if (space > MEMENTO_FREELIST_MAX_SINGLE_BLOCK)
- return 0;
- /* Pretend we added it on. */
- memento.freeListSize += space;
- /* Ditch blocks until it fits within our limit */
- while (memento.freeListSize > MEMENTO_FREELIST_MAX) {
- Memento_BlkHeader *head = memento.free.head;
- VALGRIND_MAKE_MEM_DEFINED(head, sizeof(*head));
- memento.free.head = head->next;
- memento.freeListSize -= MEMBLK_SIZE(head->rawsize);
- free_block(head);
- }
- /* Make sure we haven't just completely emptied the free list */
- /* (This should never happen, but belt and braces... */
- if (memento.free.head == NULL)
- memento.free.tail = NULL;
- return 1;
- }
- static int Memento_appBlocks(Memento_Blocks *blks,
- int (*app)(Memento_BlkHeader *,
- void *),
- void *arg)
- {
- Memento_BlkHeader *head = blks->head;
- Memento_BlkHeader *next;
- int result;
- while (head) {
- VALGRIND_MAKE_MEM_DEFINED(head, sizeof(Memento_BlkHeader));
- VALGRIND_MAKE_MEM_DEFINED(MEMBLK_TOBLK(head),
- head->rawsize + Memento_PostSize);
- result = app(head, arg);
- next = head->next;
- VALGRIND_MAKE_MEM_NOACCESS(MEMBLK_POSTPTR(head), Memento_PostSize);
- VALGRIND_MAKE_MEM_NOACCESS(head, sizeof(Memento_BlkHeader));
- if (result)
- return result;
- head = next;
- }
- return 0;
- }
- #ifndef MEMENTO_LEAKONLY
- /* Distrustful - check the block is a real one */
- static int Memento_appBlockUser(Memento_Blocks *blks,
- int (*app)(Memento_BlkHeader *,
- void *),
- void *arg,
- Memento_BlkHeader *b)
- {
- Memento_BlkHeader *head = blks->head;
- Memento_BlkHeader *next;
- int result;
- while (head && head != b) {
- VALGRIND_MAKE_MEM_DEFINED(head, sizeof(Memento_BlkHeader));
- next = head->next;
- VALGRIND_MAKE_MEM_NOACCESS(MEMBLK_POSTPTR(head), Memento_PostSize);
- head = next;
- }
- if (head == b) {
- VALGRIND_MAKE_MEM_DEFINED(head, sizeof(Memento_BlkHeader));
- VALGRIND_MAKE_MEM_DEFINED(MEMBLK_TOBLK(head),
- head->rawsize + Memento_PostSize);
- result = app(head, arg);
- VALGRIND_MAKE_MEM_NOACCESS(MEMBLK_POSTPTR(head), Memento_PostSize);
- VALGRIND_MAKE_MEM_NOACCESS(head, sizeof(Memento_BlkHeader));
- return result;
- }
- return 0;
- }
- static int Memento_appBlock(Memento_Blocks *blks,
- int (*app)(Memento_BlkHeader *,
- void *),
- void *arg,
- Memento_BlkHeader *b)
- {
- int result;
- VALGRIND_MAKE_MEM_DEFINED(b, sizeof(Memento_BlkHeader));
- VALGRIND_MAKE_MEM_DEFINED(MEMBLK_TOBLK(b),
- b->rawsize + Memento_PostSize);
- result = app(b, arg);
- VALGRIND_MAKE_MEM_NOACCESS(MEMBLK_POSTPTR(b), Memento_PostSize);
- VALGRIND_MAKE_MEM_NOACCESS(b, sizeof(Memento_BlkHeader));
- return result;
- }
- #endif /* MEMENTO_LEAKONLY */
- static int showBlock(Memento_BlkHeader *b, int space)
- {
- int seq;
- VALGRIND_MAKE_MEM_DEFINED(b, sizeof(Memento_BlkHeader));
- fprintf(stderr, FMTP":(size=" FMTZ ",num=%d)",
- MEMBLK_TOBLK(b), (FMTZ_CAST)b->rawsize, b->sequence);
- if (b->label)
- fprintf(stderr, "%c(%s)", space, b->label);
- if (b->flags & Memento_Flag_KnownLeak)
- fprintf(stderr, "(Known Leak)");
- seq = b->sequence;
- VALGRIND_MAKE_MEM_NOACCESS(b, sizeof(Memento_BlkHeader));
- return seq;
- }
- static void blockDisplay(Memento_BlkHeader *b, int n)
- {
- n++;
- while (n > 40)
- {
- fprintf(stderr, "*");
- n -= 40;
- }
- while(n > 0)
- {
- int i = n;
- if (i > 32)
- i = 32;
- n -= i;
- fprintf(stderr, "%s", &" "[32-i]);
- }
- showBlock(b, '\t');
- fprintf(stderr, "\n");
- }
- static int Memento_listBlock(Memento_BlkHeader *b,
- void *arg)
- {
- size_t *counts = (size_t *)arg;
- blockDisplay(b, 0);
- counts[0]++;
- VALGRIND_MAKE_MEM_DEFINED(b, sizeof(Memento_BlkHeader));
- counts[1]+= b->rawsize;
- VALGRIND_MAKE_MEM_NOACCESS(b, sizeof(Memento_BlkHeader));
- return 0;
- }
- static void doNestedDisplay(Memento_BlkHeader *b,
- int depth)
- {
- /* Try and avoid recursion if we can help it */
- do {
- Memento_BlkHeader *c = NULL;
- blockDisplay(b, depth);
- VALGRIND_MAKE_MEM_DEFINED(b, sizeof(Memento_BlkHeader));
- if (b->sibling) {
- c = b->child;
- b = b->sibling;
- } else {
- b = b->child;
- depth++;
- }
- VALGRIND_MAKE_MEM_NOACCESS(b, sizeof(Memento_BlkHeader));
- if (c)
- doNestedDisplay(c, depth+1);
- } while (b);
- }
- static int ptrcmp(const void *a_, const void *b_)
- {
- const char **a = (const char **)a_;
- const char **b = (const char **)b_;
- return (int)(*a-*b);
- }
- static
- int Memento_listBlocksNested(void)
- {
- int count, i;
- size_t size;
- Memento_BlkHeader *b, *prev;
- void **blocks, *minptr, *maxptr;
- intptr_t mask;
- /* Count the blocks */
- count = 0;
- size = 0;
- for (b = memento.used.head; b; b = b->next) {
- VALGRIND_MAKE_MEM_DEFINED(b, sizeof(*b));
- size += b->rawsize;
- count++;
- }
- /* Make our block list */
- blocks = MEMENTO_UNDERLYING_MALLOC(sizeof(void *) * count);
- if (blocks == NULL)
- return 1;
- /* Populate our block list */
- b = memento.used.head;
- minptr = maxptr = MEMBLK_TOBLK(b);
- mask = (intptr_t)minptr;
- for (i = 0; b; b = b->next, i++) {
- void *p = MEMBLK_TOBLK(b);
- mask &= (intptr_t)p;
- if (p < minptr)
- minptr = p;
- if (p > maxptr)
- maxptr = p;
- blocks[i] = p;
- b->flags &= ~Memento_Flag_HasParent;
- b->child = NULL;
- b->sibling = NULL;
- b->prev = NULL; /* parent */
- }
- qsort(blocks, count, sizeof(void *), ptrcmp);
- /* Now, calculate tree */
- for (b = memento.used.head; b; b = b->next) {
- char *p = MEMBLK_TOBLK(b);
- int end = (b->rawsize < MEMENTO_PTRSEARCH ? b->rawsize : MEMENTO_PTRSEARCH);
- for (i = MEMENTO_SEARCH_SKIP; i < end; i += sizeof(void *)) {
- void *q = *(void **)(&p[i]);
- void **r;
- /* Do trivial checks on pointer */
- if ((mask & (intptr_t)q) != mask || q < minptr || q > maxptr)
- continue;
- /* Search for pointer */
- r = bsearch(&q, blocks, count, sizeof(void *), ptrcmp);
- if (r) {
- /* Found child */
- Memento_BlkHeader *child = MEMBLK_FROMBLK(*r);
- Memento_BlkHeader *parent;
- /* We're assuming tree structure, not graph - ignore second
- * and subsequent pointers. */
- if (child->prev != NULL) /* parent */
- continue;
- if (child->flags & Memento_Flag_HasParent)
- continue;
- /* Not interested in pointers to ourself! */
- if (child == b)
- continue;
- /* We're also assuming acyclicness here. If this is one of
- * our parents, ignore it. */
- parent = b->prev; /* parent */
- while (parent != NULL && parent != child)
- parent = parent->prev; /* parent */
- if (parent == child)
- continue;
- child->sibling = b->child;
- b->child = child;
- child->prev = b; /* parent */
- child->flags |= Memento_Flag_HasParent;
- }
- }
- }
- /* Now display with nesting */
- for (b = memento.used.head; b; b = b->next) {
- if ((b->flags & Memento_Flag_HasParent) == 0)
- doNestedDisplay(b, 0);
- }
- fprintf(stderr, " Total number of blocks = %d\n", count);
- fprintf(stderr, " Total size of blocks = "FMTZ"\n", (FMTZ_CAST)size);
- MEMENTO_UNDERLYING_FREE(blocks);
- /* Now put the blocks back for valgrind, and restore the prev
- * and magic values. */
- prev = NULL;
- for (b = memento.used.head; b;) {
- Memento_BlkHeader *next = b->next;
- b->prev = prev;
- b->child = MEMENTO_CHILD_MAGIC;
- b->sibling = MEMENTO_SIBLING_MAGIC;
- prev = b;
- VALGRIND_MAKE_MEM_NOACCESS(b, sizeof(*b));
- b = next;
- }
- return 0;
- }
- void Memento_listBlocks(void)
- {
- MEMENTO_LOCK();
- fprintf(stderr, "Allocated blocks:\n");
- if (Memento_listBlocksNested())
- {
- size_t counts[2];
- counts[0] = 0;
- counts[1] = 0;
- Memento_appBlocks(&memento.used, Memento_listBlock, &counts[0]);
- fprintf(stderr, " Total number of blocks = "FMTZ"\n", (FMTZ_CAST)counts[0]);
- fprintf(stderr, " Total size of blocks = "FMTZ"\n", (FMTZ_CAST)counts[1]);
- }
- MEMENTO_UNLOCK();
- }
- static int Memento_listNewBlock(Memento_BlkHeader *b,
- void *arg)
- {
- if (b->flags & Memento_Flag_OldBlock)
- return 0;
- b->flags |= Memento_Flag_OldBlock;
- return Memento_listBlock(b, arg);
- }
- void Memento_listNewBlocks(void)
- {
- size_t counts[2];
- MEMENTO_LOCK();
- counts[0] = 0;
- counts[1] = 0;
- fprintf(stderr, "Blocks allocated and still extant since last list:\n");
- Memento_appBlocks(&memento.used, Memento_listNewBlock, &counts[0]);
- fprintf(stderr, " Total number of blocks = "FMTZ"\n", (FMTZ_CAST)counts[0]);
- fprintf(stderr, " Total size of blocks = "FMTZ"\n", (FMTZ_CAST)counts[1]);
- MEMENTO_UNLOCK();
- }
- static void Memento_endStats(void)
- {
- fprintf(stderr, "Total memory malloced = "FMTZ" bytes\n", (FMTZ_CAST)memento.totalAlloc);
- fprintf(stderr, "Peak memory malloced = "FMTZ" bytes\n", (FMTZ_CAST)memento.peakAlloc);
- fprintf(stderr, FMTZ" mallocs, "FMTZ" frees, "FMTZ" reallocs\n", (FMTZ_CAST)memento.numMallocs,
- (FMTZ_CAST)memento.numFrees, (FMTZ_CAST)memento.numReallocs);
- fprintf(stderr, "Average allocation size "FMTZ" bytes\n", (FMTZ_CAST)
- (memento.numMallocs != 0 ? memento.totalAlloc/memento.numMallocs: 0));
- }
- void Memento_stats(void)
- {
- MEMENTO_LOCK();
- fprintf(stderr, "Current memory malloced = "FMTZ" bytes\n", (FMTZ_CAST)memento.alloc);
- Memento_endStats();
- MEMENTO_UNLOCK();
- }
- #ifdef MEMENTO_DETAILS
- static int showInfo(Memento_BlkHeader *b, void *arg)
- {
- Memento_BlkDetails *details;
- fprintf(stderr, FMTP":(size="FMTZ",num=%d)",
- MEMBLK_TOBLK(b), (FMTZ_CAST)b->rawsize, b->sequence);
- if (b->label)
- fprintf(stderr, " (%s)", b->label);
- fprintf(stderr, "\nEvents:\n");
- details = b->details;
- while (details)
- {
- fprintf(stderr, " Event %d (%s)\n", details->sequence, eventType[(int)details->type]);
- Memento_showStacktrace(details->stack, details->count);
- details = details->next;
- }
- return 0;
- }
- #endif
- void Memento_listBlockInfo(void)
- {
- #ifdef MEMENTO_DETAILS
- MEMENTO_LOCK();
- fprintf(stderr, "Details of allocated blocks:\n");
- Memento_appBlocks(&memento.used, showInfo, NULL);
- MEMENTO_UNLOCK();
- #endif
- }
- static int Memento_nonLeakBlocksLeaked(void)
- {
- Memento_BlkHeader *blk = memento.used.head;
- while (blk)
- {
- Memento_BlkHeader *next;
- int leaked;
- VALGRIND_MAKE_MEM_DEFINED(blk, sizeof(*blk));
- leaked = ((blk->flags & Memento_Flag_KnownLeak) == 0);
- next = blk->next;
- VALGRIND_MAKE_MEM_DEFINED(blk, sizeof(*blk));
- if (leaked)
- return 1;
- blk = next;
- }
- return 0;
- }
- void Memento_fin(void)
- {
- Memento_checkAllMemory();
- if (!memento.segv)
- {
- Memento_endStats();
- if (Memento_nonLeakBlocksLeaked()) {
- Memento_listBlocks();
- #ifdef MEMENTO_DETAILS
- fprintf(stderr, "\n");
- Memento_listBlockInfo();
- #endif
- Memento_breakpoint();
- }
- }
- if (memento.squeezing) {
- if (memento.pattern == 0)
- fprintf(stderr, "Memory squeezing @ %d complete%s\n", memento.squeezeAt, memento.segv ? " (with SEGV)" : "");
- else
- fprintf(stderr, "Memory squeezing @ %d (%d) complete%s\n", memento.squeezeAt, memento.pattern, memento.segv ? " (with SEGV)" : "");
- } else if (memento.segv) {
- fprintf(stderr, "Memento completed (with SEGV)\n");
- }
- if (memento.failing)
- {
- fprintf(stderr, "MEMENTO_FAILAT=%d\n", memento.failAt);
- fprintf(stderr, "MEMENTO_PATTERN=%d\n", memento.pattern);
- }
- if (memento.nextFailAt != 0)
- {
- fprintf(stderr, "MEMENTO_NEXTFAILAT=%d\n", memento.nextFailAt);
- fprintf(stderr, "MEMENTO_NEXTPATTERN=%d\n", memento.nextPattern);
- }
- }
- static void Memento_init(void)
- {
- char *env;
- memset(&memento, 0, sizeof(memento));
- memento.inited = 1;
- memento.used.head = NULL;
- memento.used.tail = NULL;
- memento.free.head = NULL;
- memento.free.tail = NULL;
- memento.sequence = 0;
- memento.countdown = 1024;
- env = getenv("MEMENTO_FAILAT");
- memento.failAt = (env ? atoi(env) : 0);
- env = getenv("MEMENTO_BREAKAT");
- memento.breakAt = (env ? atoi(env) : 0);
- env = getenv("MEMENTO_PARANOIA");
- memento.paranoia = (env ? atoi(env) : 0);
- if (memento.paranoia == 0)
- memento.paranoia = -1024;
- env = getenv("MEMENTO_PARANOIDAT");
- memento.paranoidAt = (env ? atoi(env) : 0);
- env = getenv("MEMENTO_SQUEEZEAT");
- memento.squeezeAt = (env ? atoi(env) : 0);
- env = getenv("MEMENTO_PATTERN");
- memento.pattern = (env ? atoi(env) : 0);
- env = getenv("MEMENTO_MAXMEMORY");
- memento.maxMemory = (env ? atoi(env) : 0);
- atexit(Memento_fin);
- Memento_initMutex(&memento.mutex);
- Memento_initStacktracer();
- Memento_breakpoint();
- }
- typedef struct findBlkData {
- void *addr;
- Memento_BlkHeader *blk;
- int flags;
- } findBlkData;
- static int Memento_containsAddr(Memento_BlkHeader *b,
- void *arg)
- {
- findBlkData *data = (findBlkData *)arg;
- char *blkend = &((char *)MEMBLK_TOBLK(b))[b->rawsize];
- if ((MEMBLK_TOBLK(b) <= data->addr) &&
- ((void *)blkend > data->addr)) {
- data->blk = b;
- data->flags = 1;
- return 1;
- }
- if (((void *)b <= data->addr) &&
- (MEMBLK_TOBLK(b) > data->addr)) {
- data->blk = b;
- data->flags = 2;
- return 1;
- }
- if (((void *)blkend <= data->addr) &&
- ((void *)(blkend + Memento_PostSize) > data->addr)) {
- data->blk = b;
- data->flags = 3;
- return 1;
- }
- return 0;
- }
- void Memento_info(void *addr)
- {
- #ifdef MEMENTO_DETAILS
- findBlkData data;
- MEMENTO_LOCK();
- data.addr = addr;
- data.blk = NULL;
- data.flags = 0;
- Memento_appBlocks(&memento.used, Memento_containsAddr, &data);
- if (data.blk != NULL)
- showInfo(data.blk, NULL);
- data.blk = NULL;
- data.flags = 0;
- Memento_appBlocks(&memento.free, Memento_containsAddr, &data);
- if (data.blk != NULL)
- showInfo(data.blk, NULL);
- MEMENTO_UNLOCK();
- #else
- printf("Memento not compiled with details support\n");
- #endif
- }
- #ifdef MEMENTO_HAS_FORK
- #include <unistd.h>
- #include <sys/wait.h>
- #include <time.h>
- #ifdef MEMENTO_STACKTRACE_METHOD
- #if MEMENTO_STACKTRACE_METHOD == 1
- #include <signal.h>
- #endif
- #endif
- /* FIXME: Find some portable way of getting this */
- /* MacOSX has 10240, Ubuntu seems to have 256 */
- #ifndef OPEN_MAX
- #define OPEN_MAX 10240
- #endif
- /* stashed_map[j] = i means that file descriptor i-1 was duplicated to j */
- int stashed_map[OPEN_MAX];
- static void Memento_signal(int sig)
- {
- (void)sig;
- fprintf(stderr, "SEGV at:\n");
- memento.segv = 1;
- Memento_bt_internal(0);
- exit(1);
- }
- static int squeeze(void)
- {
- pid_t pid;
- int i, status;
- if (memento.patternBit < 0)
- return 1;
- if (memento.squeezing && memento.patternBit >= MEMENTO_MAXPATTERN)
- return 1;
- if (memento.patternBit == 0)
- memento.squeezeAt = memento.sequence;
- if (!memento.squeezing) {
- fprintf(stderr, "Memory squeezing @ %d\n", memento.squeezeAt);
- } else
- fprintf(stderr, "Memory squeezing @ %d (%x,%x)\n", memento.squeezeAt, memento.pattern, memento.patternBit);
- /* When we fork below, the child is going to snaffle all our file pointers
- * and potentially corrupt them. Let's make copies of all of them before
- * we fork, so we can restore them when we restart. */
- for (i = 0; i < OPEN_MAX; i++) {
- if (stashed_map[i] == 0) {
- int j = dup(i);
- stashed_map[j] = i+1;
- }
- }
- fprintf(stderr, "Failing at:\n");
- Memento_bt_internal(2);
- pid = fork();
- if (pid == 0) {
- /* Child */
- signal(SIGSEGV, Memento_signal);
- /* In the child, we always fail the next allocation. */
- if (memento.patternBit == 0) {
- memento.patternBit = 1;
- } else
- memento.patternBit <<= 1;
- memento.squeezing = 1;
- return 1;
- }
- /* In the parent if we hit another allocation, pass it (and record the
- * fact we passed it in the pattern. */
- memento.pattern |= memento.patternBit;
- memento.patternBit <<= 1;
- /* Wait for pid to finish, with a timeout. */
- {
- struct timespec tm = { 0, 10 * 1000 * 1000 }; /* 10ms = 100th sec */
- int timeout = 30 * 1000 * 1000; /* time out in microseconds! */
- while (waitpid(pid, &status, WNOHANG) == 0) {
- nanosleep(&tm, NULL);
- timeout -= (tm.tv_nsec/1000);
- tm.tv_nsec *= 2;
- if (tm.tv_nsec > 999999999)
- tm.tv_nsec = 999999999;
- if (timeout <= 0) {
- char text[32];
- fprintf(stderr, "Child is taking a long time to die. Killing it.\n");
- sprintf(text, "kill %d", pid);
- system(text);
- break;
- }
- }
- }
- if (status != 0) {
- fprintf(stderr, "Child status=%d\n", status);
- }
- /* Put the files back */
- for (i = 0; i < OPEN_MAX; i++) {
- if (stashed_map[i] != 0) {
- dup2(i, stashed_map[i]-1);
- close(i);
- stashed_map[i] = 0;
- }
- }
- return 0;
- }
- #else
- #include <signal.h>
- static void Memento_signal(int sig)
- {
- (void)sig;
- memento.segv = 1;
- /* If we just return from this function the SEGV will be unhandled, and
- * we'll launch into whatever JIT debugging system the OS provides. At
- * least fprintf(stderr, something useful first. If MEMENTO_NOJIT is set, then
- * just exit to avoid the JIT (and get the usual atexit handling). */
- if (getenv("MEMENTO_NOJIT"))
- exit(1);
- else
- Memento_fin();
- }
- static int squeeze(void)
- {
- fprintf(stderr, "Memento memory squeezing disabled as no fork!\n");
- return 0;
- }
- #endif
- static void Memento_startFailing(void)
- {
- if (!memento.failing) {
- fprintf(stderr, "Starting to fail...\n");
- Memento_bt();
- fflush(stderr);
- memento.failing = 1;
- memento.failAt = memento.sequence;
- memento.nextFailAt = memento.sequence+1;
- memento.pattern = 0;
- memento.patternBit = 0;
- signal(SIGSEGV, Memento_signal);
- signal(SIGABRT, Memento_signal);
- Memento_breakpointLocked();
- }
- }
- static int Memento_event(void)
- {
- memento.sequence++;
- if ((memento.sequence >= memento.paranoidAt) && (memento.paranoidAt != 0)) {
- memento.paranoia = 1;
- memento.countdown = 1;
- }
- if (--memento.countdown == 0) {
- Memento_checkAllMemoryLocked();
- if (memento.paranoia > 0)
- memento.countdown = memento.paranoia;
- else
- {
- memento.countdown = -memento.paranoia;
- if (memento.paranoia > INT_MIN/2)
- memento.paranoia *= 2;
- }
- }
- if (memento.sequence == memento.breakAt) {
- fprintf(stderr, "Breaking at event %d\n", memento.breakAt);
- return 1;
- }
- return 0;
- }
- int Memento_sequence(void)
- {
- return memento.sequence;
- }
- int Memento_breakAt(int event)
- {
- MEMENTO_LOCK();
- memento.breakAt = event;
- MEMENTO_UNLOCK();
- return event;
- }
- static void *safe_find_block(void *ptr)
- {
- Memento_BlkHeader *block;
- int valid;
- if (ptr == NULL)
- return NULL;
- block = MEMBLK_FROMBLK(ptr);
- /* Sometimes wrapping allocators can mean Memento_label
- * is called with a value within the block, rather than
- * at the start of the block. If we detect this, find it
- * the slow way. */
- VALGRIND_MAKE_MEM_DEFINED(&block->child, sizeof(block->child));
- VALGRIND_MAKE_MEM_DEFINED(&block->sibling, sizeof(block->sibling));
- valid = (block->child == MEMENTO_CHILD_MAGIC &&
- block->sibling == MEMENTO_SIBLING_MAGIC);
- VALGRIND_MAKE_MEM_NOACCESS(&block->child, sizeof(block->child));
- VALGRIND_MAKE_MEM_NOACCESS(&block->sibling, sizeof(block->sibling));
- if (!valid)
- {
- findBlkData data;
- data.addr = ptr;
- data.blk = NULL;
- data.flags = 0;
- Memento_appBlocks(&memento.used, Memento_containsAddr, &data);
- if (data.blk == NULL)
- return NULL;
- block = data.blk;
- }
- return block;
- }
- void *Memento_label(void *ptr, const char *label)
- {
- Memento_BlkHeader *block;
- if (ptr == NULL)
- return NULL;
- MEMENTO_LOCK();
- block = safe_find_block(ptr);
- if (block != NULL)
- {
- VALGRIND_MAKE_MEM_DEFINED(&block->label, sizeof(block->label));
- block->label = label;
- VALGRIND_MAKE_MEM_NOACCESS(&block->label, sizeof(block->label));
- }
- MEMENTO_UNLOCK();
- return ptr;
- }
- void Memento_tick(void)
- {
- MEMENTO_LOCK();
- if (Memento_event()) Memento_breakpointLocked();
- MEMENTO_UNLOCK();
- }
- static int Memento_failThisEventLocked(void)
- {
- int failThisOne;
- if (Memento_event()) Memento_breakpointLocked();
- if ((memento.sequence >= memento.failAt) && (memento.failAt != 0))
- Memento_startFailing();
- if ((memento.sequence >= memento.squeezeAt) && (memento.squeezeAt != 0)) {
- return squeeze();
- }
- if (!memento.failing)
- return 0;
- failThisOne = ((memento.patternBit & memento.pattern) == 0);
- /* If we are failing, and we've reached the end of the pattern and we've
- * still got bits available in the pattern word, and we haven't already
- * set a nextPattern, then extend the pattern. */
- if (memento.failing &&
- ((~(memento.patternBit-1) & memento.pattern) == 0) &&
- (memento.patternBit != 0) &&
- memento.nextPattern == 0)
- {
- /* We'll fail this one, and set the 'next' one to pass it. */
- memento.nextFailAt = memento.failAt;
- memento.nextPattern = memento.pattern | memento.patternBit;
- }
- memento.patternBit = (memento.patternBit ? memento.patternBit << 1 : 1);
- return failThisOne;
- }
- int Memento_failThisEvent(void)
- {
- int ret;
- if (!memento.inited)
- Memento_init();
- MEMENTO_LOCK();
- ret = Memento_failThisEventLocked();
- MEMENTO_UNLOCK();
- return ret;
- }
- static void *do_malloc(size_t s, int eventType)
- {
- Memento_BlkHeader *memblk;
- size_t smem = MEMBLK_SIZE(s);
- if (Memento_failThisEventLocked())
- return NULL;
- if (s == 0)
- return NULL;
- memento.numMallocs++;
- if (memento.maxMemory != 0 && memento.alloc + s > memento.maxMemory)
- return NULL;
- memblk = MEMENTO_UNDERLYING_MALLOC(smem);
- if (memblk == NULL)
- return NULL;
- memento.alloc += s;
- memento.totalAlloc += s;
- if (memento.peakAlloc < memento.alloc)
- memento.peakAlloc = memento.alloc;
- #ifndef MEMENTO_LEAKONLY
- memset(MEMBLK_TOBLK(memblk), MEMENTO_ALLOCFILL, s);
- #endif
- memblk->rawsize = s;
- memblk->sequence = memento.sequence;
- memblk->lastCheckedOK = memblk->sequence;
- memblk->flags = 0;
- memblk->label = 0;
- memblk->child = MEMENTO_CHILD_MAGIC;
- memblk->sibling = MEMENTO_SIBLING_MAGIC;
- #ifdef MEMENTO_DETAILS
- memblk->details = NULL;
- memblk->details_tail = &memblk->details;
- Memento_storeDetails(memblk, Memento_EventType_malloc);
- #endif /* MEMENTO_DETAILS */
- Memento_addBlockHead(&memento.used, memblk, 0);
- if (memento.leaking > 0)
- memblk->flags |= Memento_Flag_KnownLeak;
- return MEMBLK_TOBLK(memblk);
- }
- void *Memento_malloc(size_t s)
- {
- void *ret;
- if (!memento.inited)
- Memento_init();
- MEMENTO_LOCK();
- ret = do_malloc(s, Memento_EventType_malloc);
- MEMENTO_UNLOCK();
- return ret;
- }
- void *Memento_calloc(size_t n, size_t s)
- {
- void *block;
- if (!memento.inited)
- Memento_init();
- MEMENTO_LOCK();
- block = do_malloc(n*s, Memento_EventType_calloc);
- if (block)
- memset(block, 0, n*s);
- MEMENTO_UNLOCK();
- return block;
- }
- static void do_reference(Memento_BlkHeader *blk, int event)
- {
- #ifdef MEMENTO_DETAILS
- Memento_storeDetails(blk, event);
- #endif /* MEMENTO_DETAILS */
- }
- int Memento_checkPointerOrNull(void *blk)
- {
- if (blk == NULL)
- return 0;
- if (blk == MEMENTO_PREFILL_PTR)
- fprintf(stderr, "Prefill value found as pointer - buffer underrun?\n");
- else if (blk == MEMENTO_POSTFILL_PTR)
- fprintf(stderr, "Postfill value found as pointer - buffer overrun?\n");
- else if (blk == MEMENTO_ALLOCFILL_PTR)
- fprintf(stderr, "Allocfill value found as pointer - use of uninitialised value?\n");
- else if (blk == MEMENTO_FREEFILL_PTR)
- fprintf(stderr, "Allocfill value found as pointer - use after free?\n");
- else
- return 0;
- #ifdef MEMENTO_DETAILS
- fprintf(stderr, "Current backtrace:\n");
- Memento_bt();
- fprintf(stderr, "History:\n");
- Memento_info(blk);
- #endif
- return 1;
- }
- int Memento_checkBytePointerOrNull(void *blk)
- {
- unsigned char i;
- if (blk == NULL)
- return 0;
- Memento_checkPointerOrNull(blk);
- i = *(unsigned int *)blk;
- if (i == MEMENTO_PREFILL_UBYTE)
- fprintf(stderr, "Prefill value found - buffer underrun?\n");
- else if (i == MEMENTO_POSTFILL_UBYTE)
- fprintf(stderr, "Postfill value found - buffer overrun?\n");
- else if (i == MEMENTO_ALLOCFILL_UBYTE)
- fprintf(stderr, "Allocfill value found - use of uninitialised value?\n");
- else if (i == MEMENTO_FREEFILL_UBYTE)
- fprintf(stderr, "Allocfill value found - use after free?\n");
- else
- return 0;
- #ifdef MEMENTO_DETAILS
- fprintf(stderr, "Current backtrace:\n");
- Memento_bt();
- fprintf(stderr, "History:\n");
- Memento_info(blk);
- #endif
- Memento_breakpoint();
- return 1;
- }
- int Memento_checkShortPointerOrNull(void *blk)
- {
- unsigned short i;
- if (blk == NULL)
- return 0;
- Memento_checkPointerOrNull(blk);
- i = *(unsigned short *)blk;
- if (i == MEMENTO_PREFILL_USHORT)
- fprintf(stderr, "Prefill value found - buffer underrun?\n");
- else if (i == MEMENTO_POSTFILL_USHORT)
- fprintf(stderr, "Postfill value found - buffer overrun?\n");
- else if (i == MEMENTO_ALLOCFILL_USHORT)
- fprintf(stderr, "Allocfill value found - use of uninitialised value?\n");
- else if (i == MEMENTO_FREEFILL_USHORT)
- fprintf(stderr, "Allocfill value found - use after free?\n");
- else
- return 0;
- #ifdef MEMENTO_DETAILS
- fprintf(stderr, "Current backtrace:\n");
- Memento_bt();
- fprintf(stderr, "History:\n");
- Memento_info(blk);
- #endif
- Memento_breakpoint();
- return 1;
- }
- int Memento_checkIntPointerOrNull(void *blk)
- {
- unsigned int i;
- if (blk == NULL)
- return 0;
- Memento_checkPointerOrNull(blk);
- i = *(unsigned int *)blk;
- if (i == MEMENTO_PREFILL_UINT)
- fprintf(stderr, "Prefill value found - buffer underrun?\n");
- else if (i == MEMENTO_POSTFILL_UINT)
- fprintf(stderr, "Postfill value found - buffer overrun?\n");
- else if (i == MEMENTO_ALLOCFILL_UINT)
- fprintf(stderr, "Allocfill value found - use of uninitialised value?\n");
- else if (i == MEMENTO_FREEFILL_UINT)
- fprintf(stderr, "Allocfill value found - use after free?\n");
- else
- return 0;
- #ifdef MEMENTO_DETAILS
- fprintf(stderr, "Current backtrace:\n");
- Memento_bt();
- fprintf(stderr, "History:\n");
- Memento_info(blk);
- #endif
- Memento_breakpoint();
- return 1;
- }
- static void *do_takeRef(void *blk)
- {
- MEMENTO_LOCK();
- do_reference(safe_find_block(blk), Memento_EventType_takeRef);
- MEMENTO_UNLOCK();
- return blk;
- }
- void *Memento_takeByteRef(void *blk)
- {
- if (!memento.inited)
- Memento_init();
- if (Memento_event()) Memento_breakpoint();
- if (!blk)
- return NULL;
- (void)Memento_checkBytePointerOrNull(blk);
- return do_takeRef(blk);
- }
- void *Memento_takeShortRef(void *blk)
- {
- if (!memento.inited)
- Memento_init();
- if (Memento_event()) Memento_breakpoint();
- if (!blk)
- return NULL;
- (void)Memento_checkShortPointerOrNull(blk);
- return do_takeRef(blk);
- }
- void *Memento_takeIntRef(void *blk)
- {
- if (!memento.inited)
- Memento_init();
- if (Memento_event()) Memento_breakpoint();
- if (!blk)
- return NULL;
- (void)Memento_checkIntPointerOrNull(blk);
- return do_takeRef(blk);
- }
- void *Memento_takeRef(void *blk)
- {
- if (!memento.inited)
- Memento_init();
- if (Memento_event()) Memento_breakpoint();
- if (!blk)
- return NULL;
- return do_takeRef(blk);
- }
- static void *do_dropRef(void *blk)
- {
- MEMENTO_LOCK();
- do_reference(safe_find_block(blk), Memento_EventType_dropRef);
- MEMENTO_UNLOCK();
- return blk;
- }
- void *Memento_dropByteRef(void *blk)
- {
- if (!memento.inited)
- Memento_init();
- if (Memento_event()) Memento_breakpoint();
- if (!blk)
- return NULL;
- Memento_checkBytePointerOrNull(blk);
- return do_dropRef(blk);
- }
- void *Memento_dropShortRef(void *blk)
- {
- if (!memento.inited)
- Memento_init();
- if (Memento_event()) Memento_breakpoint();
- if (!blk)
- return NULL;
- Memento_checkShortPointerOrNull(blk);
- return do_dropRef(blk);
- }
- void *Memento_dropIntRef(void *blk)
- {
- if (!memento.inited)
- Memento_init();
- if (Memento_event()) Memento_breakpoint();
- if (!blk)
- return NULL;
- Memento_checkIntPointerOrNull(blk);
- return do_dropRef(blk);
- }
- void *Memento_dropRef(void *blk)
- {
- if (!memento.inited)
- Memento_init();
- if (Memento_event()) Memento_breakpoint();
- if (!blk)
- return NULL;
- return do_dropRef(blk);
- }
- void *Memento_adjustRef(void *blk, int adjust)
- {
- if (Memento_event()) Memento_breakpoint();
- if (blk == NULL)
- return NULL;
- while (adjust > 0)
- {
- do_takeRef(blk);
- adjust--;
- }
- while (adjust < 0)
- {
- do_dropRef(blk);
- adjust++;
- }
- return blk;
- }
- void *Memento_reference(void *blk)
- {
- if (!blk)
- return NULL;
- if (!memento.inited)
- Memento_init();
- MEMENTO_LOCK();
- do_reference(safe_find_block(blk), Memento_EventType_reference);
- MEMENTO_UNLOCK();
- return blk;
- }
- /* Treat blocks from the user with suspicion, and check them the slow
- * but safe way. */
- static int checkBlockUser(Memento_BlkHeader *memblk, const char *action)
- {
- #ifndef MEMENTO_LEAKONLY
- BlkCheckData data;
- memset(&data, 0, sizeof(data));
- Memento_appBlockUser(&memento.used, Memento_Internal_checkAllocedBlock,
- &data, memblk);
- if (!data.found) {
- /* Failure! */
- fprintf(stderr, "Attempt to %s block ", action);
- showBlock(memblk, 32);
- fprintf(stderr, "\n");
- Memento_breakpointLocked();
- return 1;
- } else if (data.preCorrupt || data.postCorrupt) {
- fprintf(stderr, "Block ");
- showBlock(memblk, ' ');
- fprintf(stderr, " found to be corrupted on %s!\n", action);
- if (data.preCorrupt) {
- fprintf(stderr, "Preguard corrupted\n");
- }
- if (data.postCorrupt) {
- fprintf(stderr, "Postguard corrupted\n");
- }
- fprintf(stderr, "Block last checked OK at allocation %d. Now %d.\n",
- memblk->lastCheckedOK, memento.sequence);
- if ((memblk->flags & Memento_Flag_Reported) == 0)
- {
- memblk->flags |= Memento_Flag_Reported;
- Memento_breakpointLocked();
- }
- return 1;
- }
- #endif
- return 0;
- }
- static int checkBlock(Memento_BlkHeader *memblk, const char *action)
- {
- #ifndef MEMENTO_LEAKONLY
- BlkCheckData data;
- #endif
- if (memblk->child != MEMENTO_CHILD_MAGIC ||
- memblk->sibling != MEMENTO_SIBLING_MAGIC)
- {
- /* Failure! */
- fprintf(stderr, "Attempt to %s invalid block ", action);
- showBlock(memblk, 32);
- fprintf(stderr, "\n");
- Memento_breakpointLocked();
- return 1;
- }
- #ifndef MEMENTO_LEAKONLY
- memset(&data, 0, sizeof(data));
- Memento_appBlock(&memento.used, Memento_Internal_checkAllocedBlock,
- &data, memblk);
- if (!data.found) {
- /* Failure! */
- fprintf(stderr, "Attempt to %s block ", action);
- showBlock(memblk, 32);
- fprintf(stderr, "\n");
- Memento_breakpointLocked();
- return 1;
- } else if (data.preCorrupt || data.postCorrupt) {
- fprintf(stderr, "Block ");
- showBlock(memblk, ' ');
- fprintf(stderr, " found to be corrupted on %s!\n", action);
- if (data.preCorrupt) {
- fprintf(stderr, "Preguard corrupted\n");
- }
- if (data.postCorrupt) {
- fprintf(stderr, "Postguard corrupted\n");
- }
- fprintf(stderr, "Block last checked OK at allocation %d. Now %d.\n",
- memblk->lastCheckedOK, memento.sequence);
- if ((memblk->flags & Memento_Flag_Reported) == 0)
- {
- memblk->flags |= Memento_Flag_Reported;
- Memento_breakpointLocked();
- }
- return 1;
- }
- #endif
- return 0;
- }
- static void do_free(void *blk, int eventType)
- {
- Memento_BlkHeader *memblk;
- if (Memento_event()) Memento_breakpointLocked();
- if (blk == NULL)
- return;
- memblk = MEMBLK_FROMBLK(blk);
- VALGRIND_MAKE_MEM_DEFINED(memblk, sizeof(*memblk));
- if (checkBlock(memblk, "free"))
- return;
- #ifdef MEMENTO_DETAILS
- Memento_storeDetails(memblk, Memento_EventType_free);
- #endif
- VALGRIND_MAKE_MEM_DEFINED(memblk, sizeof(*memblk));
- if (memblk->flags & Memento_Flag_BreakOnFree)
- Memento_breakpointLocked();
- memento.alloc -= memblk->rawsize;
- memento.numFrees++;
- Memento_removeBlock(&memento.used, memblk);
- VALGRIND_MAKE_MEM_DEFINED(memblk, sizeof(*memblk));
- if (Memento_Internal_makeSpace(MEMBLK_SIZE(memblk->rawsize))) {
- VALGRIND_MAKE_MEM_DEFINED(memblk, sizeof(*memblk));
- VALGRIND_MAKE_MEM_DEFINED(MEMBLK_TOBLK(memblk),
- memblk->rawsize + Memento_PostSize);
- #ifndef MEMENTO_LEAKONLY
- memset(MEMBLK_TOBLK(memblk), MEMENTO_FREEFILL, memblk->rawsize);
- #endif
- memblk->flags |= Memento_Flag_Freed;
- Memento_addBlockTail(&memento.free, memblk, 1);
- } else {
- free_block(memblk);
- }
- }
- void Memento_free(void *blk)
- {
- if (!memento.inited)
- Memento_init();
- MEMENTO_LOCK();
- do_free(blk, Memento_EventType_free);
- MEMENTO_UNLOCK();
- }
- static void *do_realloc(void *blk, size_t newsize, int type)
- {
- Memento_BlkHeader *memblk, *newmemblk;
- size_t newsizemem;
- int flags;
- if (Memento_failThisEventLocked())
- return NULL;
- memblk = MEMBLK_FROMBLK(blk);
- VALGRIND_MAKE_MEM_DEFINED(memblk, sizeof(*memblk));
- if (checkBlock(memblk, "realloc"))
- return NULL;
- #ifdef MEMENTO_DETAILS
- Memento_storeDetails(memblk, type);
- #endif
- VALGRIND_MAKE_MEM_DEFINED(memblk, sizeof(*memblk));
- if (memblk->flags & Memento_Flag_BreakOnRealloc)
- Memento_breakpointLocked();
- VALGRIND_MAKE_MEM_DEFINED(memblk, sizeof(*memblk));
- if (memento.maxMemory != 0 && memento.alloc - memblk->rawsize + newsize > memento.maxMemory)
- return NULL;
- newsizemem = MEMBLK_SIZE(newsize);
- Memento_removeBlock(&memento.used, memblk);
- VALGRIND_MAKE_MEM_DEFINED(memblk, sizeof(*memblk));
- flags = memblk->flags;
- newmemblk = MEMENTO_UNDERLYING_REALLOC(memblk, newsizemem);
- if (newmemblk == NULL)
- {
- Memento_addBlockHead(&memento.used, memblk, 2);
- return NULL;
- }
- memento.numReallocs++;
- memento.totalAlloc += newsize;
- memento.alloc -= newmemblk->rawsize;
- memento.alloc += newsize;
- if (memento.peakAlloc < memento.alloc)
- memento.peakAlloc = memento.alloc;
- newmemblk->flags = flags;
- #ifndef MEMENTO_LEAKONLY
- if (newmemblk->rawsize < newsize) {
- char *newbytes = ((char *)MEMBLK_TOBLK(newmemblk))+newmemblk->rawsize;
- VALGRIND_MAKE_MEM_DEFINED(newbytes, newsize - newmemblk->rawsize);
- memset(newbytes, MEMENTO_ALLOCFILL, newsize - newmemblk->rawsize);
- VALGRIND_MAKE_MEM_UNDEFINED(newbytes, newsize - newmemblk->rawsize);
- }
- #endif
- newmemblk->rawsize = newsize;
- #ifndef MEMENTO_LEAKONLY
- VALGRIND_MAKE_MEM_DEFINED(newmemblk->preblk, Memento_PreSize);
- memset(newmemblk->preblk, MEMENTO_PREFILL, Memento_PreSize);
- VALGRIND_MAKE_MEM_UNDEFINED(newmemblk->preblk, Memento_PreSize);
- VALGRIND_MAKE_MEM_DEFINED(MEMBLK_POSTPTR(newmemblk), Memento_PostSize);
- memset(MEMBLK_POSTPTR(newmemblk), MEMENTO_POSTFILL, Memento_PostSize);
- VALGRIND_MAKE_MEM_UNDEFINED(MEMBLK_POSTPTR(newmemblk), Memento_PostSize);
- #endif
- Memento_addBlockHead(&memento.used, newmemblk, 2);
- return MEMBLK_TOBLK(newmemblk);
- }
- void *Memento_realloc(void *blk, size_t newsize)
- {
- void *ret;
- if (!memento.inited)
- Memento_init();
- if (blk == NULL)
- {
- MEMENTO_LOCK();
- ret = do_malloc(newsize, Memento_EventType_realloc);
- MEMENTO_UNLOCK();
- return ret;
- }
- if (newsize == 0) {
- MEMENTO_LOCK();
- do_free(blk, Memento_EventType_realloc);
- MEMENTO_UNLOCK();
- return NULL;
- }
- MEMENTO_LOCK();
- ret = do_realloc(blk, newsize, Memento_EventType_realloc);
- MEMENTO_UNLOCK();
- return ret;
- }
- int Memento_checkBlock(void *blk)
- {
- Memento_BlkHeader *memblk;
- int ret;
- if (blk == NULL)
- return 0;
- MEMENTO_LOCK();
- memblk = MEMBLK_FROMBLK(blk);
- ret = checkBlockUser(memblk, "check");
- MEMENTO_UNLOCK();
- return ret;
- }
- #ifndef MEMENTO_LEAKONLY
- static int Memento_Internal_checkAllAlloced(Memento_BlkHeader *memblk, void *arg)
- {
- BlkCheckData *data = (BlkCheckData *)arg;
- Memento_Internal_checkAllocedBlock(memblk, data);
- if (data->preCorrupt || data->postCorrupt) {
- if ((data->found & 2) == 0) {
- fprintf(stderr, "Allocated blocks:\n");
- data->found |= 2;
- }
- fprintf(stderr, " Block ");
- showBlock(memblk, ' ');
- if (data->preCorrupt) {
- fprintf(stderr, " Preguard ");
- }
- if (data->postCorrupt) {
- fprintf(stderr, "%s Postguard ",
- (data->preCorrupt ? "&" : ""));
- }
- fprintf(stderr, "corrupted.\n "
- "Block last checked OK at allocation %d. Now %d.\n",
- memblk->lastCheckedOK, memento.sequence);
- data->preCorrupt = 0;
- data->postCorrupt = 0;
- data->freeCorrupt = 0;
- if ((memblk->flags & Memento_Flag_Reported) == 0)
- {
- memblk->flags |= Memento_Flag_Reported;
- Memento_breakpointLocked();
- }
- }
- else
- memblk->lastCheckedOK = memento.sequence;
- return 0;
- }
- static int Memento_Internal_checkAllFreed(Memento_BlkHeader *memblk, void *arg)
- {
- BlkCheckData *data = (BlkCheckData *)arg;
- Memento_Internal_checkFreedBlock(memblk, data);
- if (data->preCorrupt || data->postCorrupt || data->freeCorrupt) {
- if ((data->found & 4) == 0) {
- fprintf(stderr, "Freed blocks:\n");
- data->found |= 4;
- }
- fprintf(stderr, " ");
- showBlock(memblk, ' ');
- if (data->freeCorrupt) {
- fprintf(stderr, " index %d (address "FMTP") onwards", (int)data->index,
- &((char *)MEMBLK_TOBLK(memblk))[data->index]);
- if (data->preCorrupt) {
- fprintf(stderr, "+ preguard");
- }
- if (data->postCorrupt) {
- fprintf(stderr, "+ postguard");
- }
- } else {
- if (data->preCorrupt) {
- fprintf(stderr, " preguard");
- }
- if (data->postCorrupt) {
- fprintf(stderr, "%s Postguard",
- (data->preCorrupt ? "+" : ""));
- }
- }
- VALGRIND_MAKE_MEM_DEFINED(memblk, sizeof(Memento_BlkHeader));
- fprintf(stderr, " corrupted.\n"
- " Block last checked OK at allocation %d. Now %d.\n",
- memblk->lastCheckedOK, memento.sequence);
- if ((memblk->flags & Memento_Flag_Reported) == 0)
- {
- memblk->flags |= Memento_Flag_Reported;
- Memento_breakpointLocked();
- }
- VALGRIND_MAKE_MEM_NOACCESS(memblk, sizeof(Memento_BlkHeader));
- data->preCorrupt = 0;
- data->postCorrupt = 0;
- data->freeCorrupt = 0;
- }
- else
- memblk->lastCheckedOK = memento.sequence;
- return 0;
- }
- #endif /* MEMENTO_LEAKONLY */
- static int Memento_checkAllMemoryLocked(void)
- {
- #ifndef MEMENTO_LEAKONLY
- BlkCheckData data;
- memset(&data, 0, sizeof(data));
- Memento_appBlocks(&memento.used, Memento_Internal_checkAllAlloced, &data);
- Memento_appBlocks(&memento.free, Memento_Internal_checkAllFreed, &data);
- return data.found;
- #else
- return 0;
- #endif
- }
- int Memento_checkAllMemory(void)
- {
- #ifndef MEMENTO_LEAKONLY
- int ret;
- MEMENTO_LOCK();
- ret = Memento_checkAllMemoryLocked();
- MEMENTO_UNLOCK();
- if (ret & 6) {
- Memento_breakpoint();
- return 1;
- }
- return 0;
- #endif
- }
- int Memento_setParanoia(int i)
- {
- memento.paranoia = i;
- if (memento.paranoia > 0)
- memento.countdown = memento.paranoia;
- else
- memento.countdown = -memento.paranoia;
- return i;
- }
- int Memento_paranoidAt(int i)
- {
- memento.paranoidAt = i;
- return i;
- }
- int Memento_getBlockNum(void *b)
- {
- Memento_BlkHeader *memblk;
- if (b == NULL)
- return 0;
- memblk = MEMBLK_FROMBLK(b);
- return (memblk->sequence);
- }
- int Memento_check(void)
- {
- int result;
- fprintf(stderr, "Checking memory\n");
- result = Memento_checkAllMemory();
- fprintf(stderr, "Memory checked!\n");
- return result;
- }
- int Memento_find(void *a)
- {
- findBlkData data;
- int s;
- MEMENTO_LOCK();
- data.addr = a;
- data.blk = NULL;
- data.flags = 0;
- Memento_appBlocks(&memento.used, Memento_containsAddr, &data);
- if (data.blk != NULL) {
- fprintf(stderr, "Address "FMTP" is in %sallocated block ",
- data.addr,
- (data.flags == 1 ? "" : (data.flags == 2 ?
- "preguard of " : "postguard of ")));
- s = showBlock(data.blk, ' ');
- fprintf(stderr, "\n");
- MEMENTO_UNLOCK();
- return s;
- }
- data.blk = NULL;
- data.flags = 0;
- Memento_appBlocks(&memento.free, Memento_containsAddr, &data);
- if (data.blk != NULL) {
- fprintf(stderr, "Address "FMTP" is in %sfreed block ",
- data.addr,
- (data.flags == 1 ? "" : (data.flags == 2 ?
- "preguard of " : "postguard of ")));
- s = showBlock(data.blk, ' ');
- fprintf(stderr, "\n");
- MEMENTO_UNLOCK();
- return s;
- }
- MEMENTO_UNLOCK();
- return 0;
- }
- void Memento_breakOnFree(void *a)
- {
- findBlkData data;
- MEMENTO_LOCK();
- data.addr = a;
- data.blk = NULL;
- data.flags = 0;
- Memento_appBlocks(&memento.used, Memento_containsAddr, &data);
- if (data.blk != NULL) {
- fprintf(stderr, "Will stop when address "FMTP" (in %sallocated block ",
- data.addr,
- (data.flags == 1 ? "" : (data.flags == 2 ?
- "preguard of " : "postguard of ")));
- showBlock(data.blk, ' ');
- fprintf(stderr, ") is freed\n");
- VALGRIND_MAKE_MEM_DEFINED(data.blk, sizeof(Memento_BlkHeader));
- data.blk->flags |= Memento_Flag_BreakOnFree;
- VALGRIND_MAKE_MEM_NOACCESS(data.blk, sizeof(Memento_BlkHeader));
- MEMENTO_UNLOCK();
- return;
- }
- data.blk = NULL;
- data.flags = 0;
- Memento_appBlocks(&memento.free, Memento_containsAddr, &data);
- if (data.blk != NULL) {
- fprintf(stderr, "Can't stop on free; address "FMTP" is in %sfreed block ",
- data.addr,
- (data.flags == 1 ? "" : (data.flags == 2 ?
- "preguard of " : "postguard of ")));
- showBlock(data.blk, ' ');
- fprintf(stderr, "\n");
- MEMENTO_UNLOCK();
- return;
- }
- fprintf(stderr, "Can't stop on free; address "FMTP" is not in a known block.\n", a);
- MEMENTO_UNLOCK();
- }
- void Memento_breakOnRealloc(void *a)
- {
- findBlkData data;
- MEMENTO_LOCK();
- data.addr = a;
- data.blk = NULL;
- data.flags = 0;
- Memento_appBlocks(&memento.used, Memento_containsAddr, &data);
- if (data.blk != NULL) {
- fprintf(stderr, "Will stop when address "FMTP" (in %sallocated block ",
- data.addr,
- (data.flags == 1 ? "" : (data.flags == 2 ?
- "preguard of " : "postguard of ")));
- showBlock(data.blk, ' ');
- fprintf(stderr, ") is freed (or realloced)\n");
- VALGRIND_MAKE_MEM_DEFINED(data.blk, sizeof(Memento_BlkHeader));
- data.blk->flags |= Memento_Flag_BreakOnFree | Memento_Flag_BreakOnRealloc;
- VALGRIND_MAKE_MEM_NOACCESS(data.blk, sizeof(Memento_BlkHeader));
- MEMENTO_UNLOCK();
- return;
- }
- data.blk = NULL;
- data.flags = 0;
- Memento_appBlocks(&memento.free, Memento_containsAddr, &data);
- if (data.blk != NULL) {
- fprintf(stderr, "Can't stop on free/realloc; address "FMTP" is in %sfreed block ",
- data.addr,
- (data.flags == 1 ? "" : (data.flags == 2 ?
- "preguard of " : "postguard of ")));
- showBlock(data.blk, ' ');
- fprintf(stderr, "\n");
- MEMENTO_UNLOCK();
- return;
- }
- fprintf(stderr, "Can't stop on free/realloc; address "FMTP" is not in a known block.\n", a);
- MEMENTO_UNLOCK();
- }
- int Memento_failAt(int i)
- {
- memento.failAt = i;
- if ((memento.sequence > memento.failAt) &&
- (memento.failing != 0))
- Memento_startFailing();
- return i;
- }
- size_t Memento_setMax(size_t max)
- {
- memento.maxMemory = max;
- return max;
- }
- void Memento_startLeaking(void)
- {
- memento.leaking++;
- }
- void Memento_stopLeaking(void)
- {
- memento.leaking--;
- }
- int Memento_squeezing(void)
- {
- return memento.squeezing;
- }
- #endif /* MEMENTO_CPP_EXTRAS_ONLY */
- #ifdef __cplusplus
- /* Dumb overrides for the new and delete operators */
- void *operator new(size_t size)
- {
- void *ret;
- if (!memento.inited)
- Memento_init();
- if (size == 0)
- size = 1;
- MEMENTO_LOCK();
- ret = do_malloc(size, Memento_EventType_new);
- MEMENTO_UNLOCK();
- return ret;
- }
- void operator delete(void *pointer)
- {
- if (!pointer)
- return;
- MEMENTO_LOCK();
- do_free(pointer, Memento_EventType_delete);
- MEMENTO_UNLOCK();
- }
- /* Some C++ systems (apparently) don't provide new[] or delete[]
- * operators. Provide a way to cope with this */
- #ifndef MEMENTO_CPP_NO_ARRAY_CONSTRUCTORS
- void *operator new[](size_t size)
- {
- void *ret;
- if (!memento.inited)
- Memento_init();
- if (size == 0)
- size = 1;
- MEMENTO_LOCK();
- ret = do_malloc(size, Memento_EventType_newArray);
- MEMENTO_UNLOCK();
- return ret;
- }
- void operator delete[](void *pointer)
- {
- MEMENTO_LOCK();
- do_free(pointer, Memento_EventType_deleteArray);
- MEMENTO_UNLOCK();
- }
- #endif /* MEMENTO_CPP_NO_ARRAY_CONSTRUCTORS */
- #endif /* __cplusplus */
- #else
- /* Just in case anyone has left some debugging code in... */
- void (Memento_breakpoint)(void)
- {
- }
- int (Memento_checkBlock)(void *b)
- {
- return 0;
- }
- int (Memento_checkAllMemory)(void)
- {
- return 0;
- }
- int (Memento_check)(void)
- {
- return 0;
- }
- int (Memento_setParanoia)(int i)
- {
- return 0;
- }
- int (Memento_paranoidAt)(int i)
- {
- return 0;
- }
- int (Memento_breakAt)(int i)
- {
- return 0;
- }
- int (Memento_getBlockNum)(void *i)
- {
- return 0;
- }
- int (Memento_find)(void *a)
- {
- return 0;
- }
- int (Memento_failAt)(int i)
- {
- return 0;
- }
- void (Memento_breakOnFree)(void *a)
- {
- }
- void (Memento_breakOnRealloc)(void *a)
- {
- }
- void *(Memento_takeRef)(void *a)
- {
- return a;
- }
- void *(Memento_dropRef)(void *a)
- {
- return a;
- }
- void *(Memento_adjustRef)(void *a, int adjust)
- {
- return a;
- }
- void *(Memento_reference)(void *a)
- {
- return a;
- }
- #undef Memento_malloc
- #undef Memento_free
- #undef Memento_realloc
- #undef Memento_calloc
- void *Memento_malloc(size_t size)
- {
- return MEMENTO_UNDERLYING_MALLOC(size);
- }
- void Memento_free(void *b)
- {
- MEMENTO_UNDERLYING_FREE(b);
- }
- void *Memento_realloc(void *b, size_t s)
- {
- return MEMENTO_UNDERLYING_REALLOC(b, s);
- }
- void *Memento_calloc(size_t n, size_t s)
- {
- return MEMENTO_UNDERLYING_CALLOC(n, s);
- }
- void (Memento_listBlocks)(void)
- {
- }
- void (Memento_listNewBlocks)(void)
- {
- }
- size_t (Memento_setMax)(size_t max)
- {
- return 0;
- }
- void (Memento_stats)(void)
- {
- }
- void *(Memento_label)(void *ptr, const char *label)
- {
- return ptr;
- }
- void (Memento_info)(void *addr)
- {
- }
- void (Memento_listBlockInfo)(void)
- {
- }
- void (Memento_startLeaking)(void)
- {
- }
- void (Memento_stopLeaking)(void)
- {
- }
- int (Memento_squeezing)(void)
- {
- return 0;
- }
- #endif
|