X-Git-Url: http://demsky.eecs.uci.edu/git/?a=blobdiff_plain;f=schedule.cc;h=b1b41c3d88d579a766d5cd45ccf26815bc551a1e;hb=8497408d26002ec1a9d7cfd42458f92f2cdd9864;hp=7a703abab8cc7e229f7218c85add8580c7f3ad53;hpb=20d006a8c581117c6636e835cdbe36dac5d6ef93;p=model-checker.git

diff --git a/schedule.cc b/schedule.cc
index 7a703ab..b1b41c3 100644
--- a/schedule.cc
+++ b/schedule.cc
@@ -18,11 +18,11 @@ Scheduler::Scheduler() :
 void Scheduler::set_enabled(Thread *t, bool enabled_status) {
 	int threadid=id_to_int(t->get_id());
 	if (threadid>=enabled_len) {
-		bool * new_enabled=(bool *)malloc(sizeof(bool)*(threadid+1));
+		bool *new_enabled = (bool *)snapshot_malloc(sizeof(bool) * (threadid + 1));
 		memset(&new_enabled[enabled_len], 0, (threadid+1-enabled_len)*sizeof(bool));
 		if (is_enabled != NULL) {
 			memcpy(new_enabled, is_enabled, enabled_len*sizeof(bool));
-			free(is_enabled);
+			snapshot_free(is_enabled);
 		}
 		is_enabled=new_enabled;
 		enabled_len=threadid+1;