6 #include "snapshot-interface.h"
11 /* MYBINARYNAME only works because our pathname usually includes 'model' (e.g.,
12 * /.../model-checker/test/userprog.o) */
13 #define MYBINARYNAME "model"
14 #define MYLIBRARYNAME "libmodel.so"
15 #define MAPFILE "/proc/self/maps"
18 struct stackEntry *next;
19 snapshot_id snapshotid;
27 int backTrackBeforeStep(int seq_index);
28 void snapshotStep(int seq_index);
32 struct stackEntry *stack;
35 static SnapshotStack *snapshotObject;
38 /** The SnapshotGlobalSegments function computes the memory regions
39 * that may contain globals and then configures the snapshotting
40 * library to snapshot them.
42 static void SnapshotGlobalSegments()
45 char buf[9000], execname[100];
48 sprintf(execname, "vmmap -interleaved %d", pid);
49 map = popen(execname, "r");
56 /* Wait for correct part */
57 while (fgets(buf, sizeof(buf), map)) {
58 if (strstr(buf, "==== regions for process"))
62 while (fgets(buf, sizeof(buf), map)) {
63 char regionname[200] = "";
71 //Skip out at the end of the section
75 sscanf(buf, "%22s %p-%p [%5dK] %c%c%c/%c%c%c SM=%3s %200s\n", type, &begin, &end, &size, &r, &w, &x, &mr, &mw, &mx, smstr, regionname);
77 if (w == 'w' && (strstr(regionname, MYBINARYNAME) || strstr(regionname, MYLIBRARYNAME))) {
78 size_t len = ((uintptr_t)end - (uintptr_t)begin) / PAGESIZE;
80 snapshot_add_memory_region(begin, len);
87 static void get_binary_name(char *buf, size_t len)
89 if (readlink("/proc/self/exe", buf, len) == -1) {
95 /** The SnapshotGlobalSegments function computes the memory regions
96 * that may contain globals and then configures the snapshotting
97 * library to snapshot them.
99 static void SnapshotGlobalSegments()
102 char binary_name[800];
105 map = fopen(MAPFILE, "r");
110 get_binary_name(binary_name, sizeof(binary_name));
111 while (fgets(buf, sizeof(buf), map)) {
112 char regionname[200] = "";
116 sscanf(buf, "%p-%p %c%c%c%c %*x %*x:%*x %*u %200s\n", &begin, &end, &r, &w, &x, &p, regionname);
117 if (w == 'w' && (strstr(regionname, binary_name) || strstr(regionname, MYLIBRARYNAME))) {
118 size_t len = ((uintptr_t)end - (uintptr_t)begin) / PAGESIZE;
120 snapshot_add_memory_region(begin, len);
121 DEBUG("%55s: %18p - %18p\t%c%c%c%c\n", regionname, begin, end, r, w, x, p);
128 SnapshotStack::SnapshotStack()
130 SnapshotGlobalSegments();
134 SnapshotStack::~SnapshotStack()
138 /** This method returns to the last snapshot before the inputted
139 * sequence number. This function must be called from the model
140 * checking thread and not from a snapshotted stack.
141 * @param seqindex is the sequence number to rollback before.
142 * @return is the sequence number we actually rolled back to.
144 int SnapshotStack::backTrackBeforeStep(int seqindex)
147 if (stack->index <= seqindex) {
149 snapshot_roll_back(stack->snapshotid);
152 struct stackEntry *tmp = stack;
158 /** This method takes a snapshot at the given sequence number. */
159 void SnapshotStack::snapshotStep(int seqindex)
161 struct stackEntry *tmp = (struct stackEntry *)model_malloc(sizeof(struct stackEntry));
163 tmp->index = seqindex;
164 tmp->snapshotid = take_snapshot();
169 void snapshot_stack_init()
171 snapshotObject = new SnapshotStack();
174 void snapshot_record(int seq_index)
176 snapshotObject->snapshotStep(seq_index);
179 int snapshot_backtrack_before(int seq_index)
181 return snapshotObject->backTrackBeforeStep(seq_index);