cyclegraph: template for addEdge()
[c11tester.git] / snapshot-interface.cc
1 #include <stdlib.h>
2 #include <unistd.h>
3 #include <cstring>
4 #include <inttypes.h>
5 #include <vector>
6
7 #include "snapshot-interface.h"
8 #include "snapshot.h"
9 #include "common.h"
10 #include "mymemory.h"
11
12 /* MYBINARYNAME only works because our pathname usually includes 'model' (e.g.,
13  * /.../model-checker/test/userprog.o) */
14 #define MYBINARYNAME "model"
15 #define MYLIBRARYNAME "libmodel.so"
16 #define MAPFILE "/proc/self/maps"
17
18 struct snapshot_entry {
19         snapshot_entry(snapshot_id id, int idx) : snapshotid(id), index(idx) { }
20         snapshot_id snapshotid;
21         int index;
22         MEMALLOC
23 };
24
25 class SnapshotStack {
26  public:
27         int backTrackBeforeStep(int seq_index);
28         void snapshotStep(int seq_index);
29
30         MEMALLOC
31  private:
32         std::vector<struct snapshot_entry, ModelAlloc<struct snapshot_entry> > stack;
33 };
34
35 static SnapshotStack *snap_stack;
36
37 #ifdef MAC
38 /** The SnapshotGlobalSegments function computes the memory regions
39  *      that may contain globals and then configures the snapshotting
40  *      library to snapshot them.
41  */
42 static void SnapshotGlobalSegments()
43 {
44         int pid = getpid();
45         char buf[9000], execname[100];
46         FILE *map;
47
48         sprintf(execname, "vmmap -interleaved %d", pid);
49         map = popen(execname, "r");
50
51         if (!map) {
52                 perror("popen");
53                 exit(EXIT_FAILURE);
54         }
55
56         /* Wait for correct part */
57         while (fgets(buf, sizeof(buf), map)) {
58                 if (strstr(buf, "==== regions for process"))
59                         break;
60         }
61
62         while (fgets(buf, sizeof(buf), map)) {
63                 char regionname[200] = "";
64                 char type[23];
65                 char smstr[23];
66                 char r, w, x;
67                 char mr, mw, mx;
68                 int size;
69                 void *begin, *end;
70
71                 //Skip out at the end of the section
72                 if (buf[0] == '\n')
73                         break;
74
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);
76
77                 if (w == 'w' && (strstr(regionname, MYBINARYNAME) || strstr(regionname, MYLIBRARYNAME))) {
78                         size_t len = ((uintptr_t)end - (uintptr_t)begin) / PAGESIZE;
79                         if (len != 0)
80                                 snapshot_add_memory_region(begin, len);
81                 }
82         }
83         pclose(map);
84 }
85 #else
86
87 static void get_binary_name(char *buf, size_t len)
88 {
89         if (readlink("/proc/self/exe", buf, len) == -1) {
90                 perror("readlink");
91                 exit(EXIT_FAILURE);
92         }
93 }
94
95 /** The SnapshotGlobalSegments function computes the memory regions
96  *      that may contain globals and then configures the snapshotting
97  *      library to snapshot them.
98  */
99 static void SnapshotGlobalSegments()
100 {
101         char buf[9000];
102         char binary_name[800];
103         FILE *map;
104
105         map = fopen(MAPFILE, "r");
106         if (!map) {
107                 perror("fopen");
108                 exit(EXIT_FAILURE);
109         }
110         get_binary_name(binary_name, sizeof(binary_name));
111         while (fgets(buf, sizeof(buf), map)) {
112                 char regionname[200] = "";
113                 char r, w, x, p;
114                 void *begin, *end;
115
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;
119                         if (len != 0)
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);
122                 }
123         }
124         fclose(map);
125 }
126 #endif
127
128 /** This method returns to the last snapshot before the inputted
129  * sequence number.  This function must be called from the model
130  * checking thread and not from a snapshotted stack.
131  * @param seqindex is the sequence number to rollback before.
132  * @return is the sequence number we actually rolled back to.
133  */
134 int SnapshotStack::backTrackBeforeStep(int seqindex)
135 {
136         int i;
137         for (i = (int)stack.size() - 1; i >= 0; i++)
138                 if (stack[i].index <= seqindex)
139                         break;
140                 else
141                         stack.pop_back();
142
143         ASSERT(i >= 0);
144         snapshot_roll_back(stack[i].snapshotid);
145         return stack[i].index;
146 }
147
148 /** This method takes a snapshot at the given sequence number. */
149 void SnapshotStack::snapshotStep(int seqindex)
150 {
151         stack.push_back(snapshot_entry(take_snapshot(), seqindex));
152 }
153
154 void snapshot_stack_init()
155 {
156         snap_stack = new SnapshotStack();
157         SnapshotGlobalSegments();
158 }
159
160 void snapshot_record(int seq_index)
161 {
162         snap_stack->snapshotStep(seq_index);
163 }
164
165 int snapshot_backtrack_before(int seq_index)
166 {
167         return snap_stack->backTrackBeforeStep(seq_index);
168 }