X-Git-Url: http://demsky.eecs.uci.edu/git/?a=blobdiff_plain;f=snapshotimp.h;h=0560203f1dc8e358312d9b3cfc30040e4a6fb225;hb=fc32611957cecd106751b62bc4de4aeddc9af56c;hp=e5c331a2e5ee765fe7e3ec6d7a4fa189fa6e0b2b;hpb=123c66e0c1b0f58aae2916cc22b2100143a2ceb4;p=model-checker.git

diff --git a/snapshotimp.h b/snapshotimp.h
index e5c331a..0560203 100644
--- a/snapshotimp.h
+++ b/snapshotimp.h
@@ -2,16 +2,14 @@
  *	@brief Snapshotting implementation header file..
  */
 
-#ifndef _SNAPSHOTIMP_H
-#define _SNAPSHOTIMP_H
+#ifndef __SNAPSHOTIMP_H__
+#define __SNAPSHOTIMP_H__
+
+#include <stddef.h>
+
 #include "snapshot.h"
-#include <iostream>
-#include <inttypes.h>
-#include <fcntl.h>
-#include <sys/mman.h>
-#include <sys/types.h>
-#include <csignal>
-#define SHARED_MEMORY_DEFAULT	(100 * ((size_t)1 << 20)) // 100mb for the shared memory
+
+#define SHARED_MEMORY_DEFAULT  (100 * ((size_t)1 << 20)) // 100mb for the shared memory
 #define STACK_SIZE_DEFAULT      (((size_t)1 << 20) * 20)  // 20 mb out of the above 100 mb for my stack
 
 #if USE_MPROTECT_SNAPSHOT
@@ -54,6 +52,9 @@ struct SnapShot {
 };
 
 #else
+
+#include <ucontext.h>
+
 struct SnapShot {
 	void *mSharedMemoryBase;
 	void *mStackBase;
@@ -66,4 +67,5 @@ struct SnapShot {
 
 //Global reference to snapshot data structure
 extern struct SnapShot * snapshotrecord;
-#endif
+
+#endif /* __SNAPSHOTIMP_H__ */