From 3380498042ffe830476bf12e75075cd826b10c4b Mon Sep 17 00:00:00 2001 From: David Bremner Date: Sun, 16 Dec 2012 23:08:49 -0400 Subject: [PATCH] CLI: add talloc leak report, controlled by an environment variable. The argument handling in notmuch.c seems due for an overhaul, but until then use an environment variable to specify a location to write the talloc leak report to. This is only enabled for the (interesting) case where some notmuch subcommand is invoked. --- notmuch.c | 24 ++++++++++++++++++++++-- 1 file changed, 22 insertions(+), 2 deletions(-) diff --git a/notmuch.c b/notmuch.c index 9516dfb7..4fc0973d 100644 --- a/notmuch.c +++ b/notmuch.c @@ -322,8 +322,28 @@ main (int argc, char *argv[]) for (i = 0; i < ARRAY_SIZE (commands); i++) { command = &commands[i]; - if (strcmp (argv[1], command->name) == 0) - return (command->function) (local, argc - 1, &argv[1]); + if (strcmp (argv[1], command->name) == 0) { + int ret; + char *talloc_report; + + ret = (command->function)(local, argc - 1, &argv[1]); + + /* in the future support for this environment variable may + * be supplemented or replaced by command line arguments + * --leak-report and/or --leak-report-full */ + + talloc_report = getenv ("NOTMUCH_TALLOC_REPORT"); + + /* this relies on the previous call to + * talloc_enable_null_tracking */ + + if (talloc_report && strcmp (talloc_report, "") != 0) { + FILE *report = fopen (talloc_report, "w"); + talloc_report_full (NULL, report); + } + + return ret; + } } fprintf (stderr, "Error: Unknown command '%s' (see \"notmuch help\")\n",