From: Martin Willi <martin@revosec.ch>
Date: Wed, 11 Jul 2012 15:08:30 +0000 (+0200)
Subject: Add an external method to disable leak detective temporarly
X-Git-Tag: 5.0.1~394
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=b46776aea159df93264ca7df02a2169a0e57e258;p=thirdparty%2Fstrongswan.git

Add an external method to disable leak detective temporarly
---

diff --git a/src/libstrongswan/utils/leak_detective.c b/src/libstrongswan/utils/leak_detective.c
index 81524b6e74..edf743b118 100644
--- a/src/libstrongswan/utils/leak_detective.c
+++ b/src/libstrongswan/utils/leak_detective.c
@@ -397,6 +397,35 @@ METHOD(leak_detective_t, report, void,
 	}
 }
 
+METHOD(leak_detective_t, set_state, bool,
+	private_leak_detective_t *this, bool enable)
+{
+	static struct sched_param oldparams;
+	static int oldpolicy;
+	struct sched_param params;
+	pthread_t thread_id;
+
+	if (enable == installed)
+	{
+		return installed;
+	}
+	thread_id = pthread_self();
+	if (enable)
+	{
+		install_hooks();
+		pthread_setschedparam(thread_id, oldpolicy, &oldparams);
+	}
+	else
+	{
+		pthread_getschedparam(thread_id, &oldpolicy, &oldparams);
+		params.__sched_priority = sched_get_priority_max(SCHED_FIFO);
+		pthread_setschedparam(thread_id, SCHED_FIFO, &params);
+		uninstall_hooks();
+	}
+	installed = enable;
+	return !installed;
+}
+
 METHOD(leak_detective_t, usage, void,
 	private_leak_detective_t *this, FILE *out)
 {
@@ -620,6 +649,7 @@ leak_detective_t *leak_detective_create()
 		.public = {
 			.report = _report,
 			.usage = _usage,
+			.set_state = _set_state,
 			.destroy = _destroy,
 		},
 	);
diff --git a/src/libstrongswan/utils/leak_detective.h b/src/libstrongswan/utils/leak_detective.h
index 8c80d2532e..55d7e44d9d 100644
--- a/src/libstrongswan/utils/leak_detective.h
+++ b/src/libstrongswan/utils/leak_detective.h
@@ -49,6 +49,14 @@ struct leak_detective_t {
 	 */
 	void (*usage)(leak_detective_t *this, FILE *out);
 
+	/**
+	 * Enable/disable leak detective hooks.
+	 *
+	 * @param				TRUE to enable, FALSE to disable
+	 * @return				state active before calling set_state
+	 */
+	bool (*set_state)(leak_detective_t *this, bool enabled);
+
 	/**
 	 * Destroy a leak_detective instance.
 	 */