1 /* $NetBSD: snapshot.c,v 1.7 2013/02/06 09:05:01 hannken Exp $ */
2
3 #include <sys/types.h>
4 #include <sys/ioctl.h>
5 #include <sys/mount.h>
6
7 #include <dev/fssvar.h>
8
9 #include <atf-c.h>
10 #include <fcntl.h>
11 #include <pthread.h>
12 #include <stdio.h>
13 #include <stdlib.h>
14 #include <string.h>
15 #include <unistd.h>
16
17 ATF_TC_WITH_CLEANUP(snapshot);
ATF_TC_HEAD(snapshot,tc)18 ATF_TC_HEAD(snapshot, tc)
19 {
20
21 atf_tc_set_md_var(tc, "descr", "basic snapshot features");
22 }
23
24 static void
makefile(const char * path)25 makefile(const char *path)
26 {
27 int fd;
28
29 fd = rump_sys_open(path, O_CREAT | O_RDWR, 0777);
30 if (fd == -1)
31 atf_tc_fail_errno("create %s", path);
32 rump_sys_close(fd);
33 }
34
ATF_TC_BODY(snapshot,tc)35 ATF_TC_BODY(snapshot, tc)
36 {
37 char buf[1024];
38 struct fss_set fss;
39 int fssfd;
40 int fd, fd2, i;
41
42 if (system(NEWFS) == -1)
43 atf_tc_fail_errno("cannot create file system");
44
45 rump_init();
46 begin();
47
48 if (rump_sys_mkdir("/mnt", 0777) == -1)
49 atf_tc_fail_errno("mount point create");
50 if (rump_sys_mkdir("/snap", 0777) == -1)
51 atf_tc_fail_errno("mount point 2 create");
52
53 rump_pub_etfs_register("/diskdev", IMGNAME, RUMP_ETFS_BLK);
54
55 mount_diskfs("/diskdev", "/mnt");
56
57 #define TESTSTR1 "huihai\n"
58 #define TESTSZ1 (sizeof(TESTSTR1)-1)
59 #define TESTSTR2 "baana liten\n"
60 #define TESTSZ2 (sizeof(TESTSTR2)-1)
61
62 fd = rump_sys_open("/mnt/myfile", O_RDWR | O_CREAT, 0777);
63 if (fd == -1)
64 atf_tc_fail_errno("create file");
65 if (rump_sys_write(fd, TESTSTR1, TESTSZ1) != TESTSZ1)
66 atf_tc_fail_errno("write fail");
67
68 fssfd = rump_sys_open("/dev/rfss0", O_RDWR);
69 if (fssfd == -1)
70 atf_tc_fail_errno("cannot open fss");
71 makefile(BAKNAME);
72 memset(&fss, 0, sizeof(fss));
73 fss.fss_mount = __UNCONST("/mnt");
74 fss.fss_bstore = __UNCONST(BAKNAME);
75 fss.fss_csize = 0;
76 if (rump_sys_ioctl(fssfd, FSSIOCSET, &fss) == -1)
77 atf_tc_fail_errno("create snapshot");
78
79 for (i = 0; i < 10000; i++) {
80 if (rump_sys_write(fd, TESTSTR2, TESTSZ2) != TESTSZ2)
81 atf_tc_fail_errno("write fail");
82 }
83 rump_sys_sync();
84
85 /* technically we should fsck it first? */
86 mount_diskfs("/dev/fss0", "/snap");
87
88 /* check for old contents */
89 fd2 = rump_sys_open("/snap/myfile", O_RDONLY);
90 if (fd2 == -1)
91 atf_tc_fail_errno("fail");
92 memset(buf, 0, sizeof(buf));
93 if (rump_sys_read(fd2, buf, sizeof(buf)) == -1)
94 atf_tc_fail_errno("read snap");
95 ATF_CHECK(strcmp(buf, TESTSTR1) == 0);
96
97 /* check that new files are invisible in the snapshot */
98 makefile("/mnt/newfile");
99 if (rump_sys_open("/snap/newfile", O_RDONLY) != -1)
100 atf_tc_fail("newfile exists in snapshot");
101 if (errno != ENOENT)
102 atf_tc_fail_errno("newfile open should fail with ENOENT");
103
104 /* check that removed files are still visible in the snapshot */
105 rump_sys_unlink("/mnt/myfile");
106 if (rump_sys_open("/snap/myfile", O_RDONLY) == -1)
107 atf_tc_fail_errno("unlinked file no longer in snapshot");
108
109 /* done for now */
110 }
111
ATF_TC_CLEANUP(snapshot,tc)112 ATF_TC_CLEANUP(snapshot, tc)
113 {
114
115 unlink(IMGNAME);
116 }
117
118 ATF_TC_WITH_CLEANUP(snapshotstress);
ATF_TC_HEAD(snapshotstress,tc)119 ATF_TC_HEAD(snapshotstress, tc)
120 {
121
122 atf_tc_set_md_var(tc, "descr", "snapshot on active file system");
123 }
124
125 #define NACTIVITY 4
126
127 static bool activity_stop = false;
128 static pid_t wrkpid;
129
130 static void *
fs_activity(void * arg)131 fs_activity(void *arg)
132 {
133 int di, fi;
134 char *prefix = arg, path[128];
135
136 rump_pub_lwproc_newlwp(wrkpid);
137
138 RL(rump_sys_mkdir(prefix, 0777));
139 while (! activity_stop) {
140 for (di = 0; di < 5; di++) {
141 snprintf(path, sizeof(path), "%s/d%d", prefix, di);
142 RL(rump_sys_mkdir(path, 0777));
143 for (fi = 0; fi < 5; fi++) {
144 snprintf(path, sizeof(path), "%s/d%d/f%d",
145 prefix, di, fi);
146 makefile(path);
147 }
148 }
149 for (di = 0; di < 5; di++) {
150 for (fi = 0; fi < 5; fi++) {
151 snprintf(path, sizeof(path), "%s/d%d/f%d",
152 prefix, di, fi);
153 RL(rump_sys_unlink(path));
154 }
155 snprintf(path, sizeof(path), "%s/d%d", prefix, di);
156 RL(rump_sys_rmdir(path));
157 }
158 }
159 RL(rump_sys_rmdir(prefix));
160
161 rump_pub_lwproc_releaselwp();
162
163 return NULL;
164 }
165
ATF_TC_BODY(snapshotstress,tc)166 ATF_TC_BODY(snapshotstress, tc)
167 {
168 pthread_t at[NACTIVITY];
169 struct fss_set fss;
170 char prefix[NACTIVITY][128];
171 int i, fssfd;
172
173 if (system(NEWFS) == -1)
174 atf_tc_fail_errno("cannot create file system");
175 /* Force SMP so the stress makes sense. */
176 RL(setenv("RUMP_NCPU", "4", 1));
177 RZ(rump_init());
178 /* Prepare for fsck to use the RUMP /dev/fss0. */
179 RL(rump_init_server("unix://commsock"));
180 RL(setenv("LD_PRELOAD", "/usr/lib/librumphijack.so", 1));
181 RL(setenv("RUMP_SERVER", "unix://commsock", 1));
182 RL(setenv("RUMPHIJACK", "blanket=/dev/rfss0", 1));
183 begin();
184
185 RL(rump_sys_mkdir("/mnt", 0777));
186
187 rump_pub_etfs_register("/diskdev", IMGNAME, RUMP_ETFS_BLK);
188
189 mount_diskfs("/diskdev", "/mnt");
190
191 /* Start file system activity. */
192 RL(wrkpid = rump_sys_getpid());
193 for (i = 0; i < NACTIVITY; i++) {
194 snprintf(prefix[i], sizeof(prefix[i]), "/mnt/a%d", i);
195 RL(pthread_create(&at[i], NULL, fs_activity, prefix[i]));
196 sleep(1);
197 }
198
199 fssfd = rump_sys_open("/dev/rfss0", O_RDWR);
200 if (fssfd == -1)
201 atf_tc_fail_errno("cannot open fss");
202 makefile(BAKNAME);
203 memset(&fss, 0, sizeof(fss));
204 fss.fss_mount = __UNCONST("/mnt");
205 fss.fss_bstore = __UNCONST(BAKNAME);
206 fss.fss_csize = 0;
207 if (rump_sys_ioctl(fssfd, FSSIOCSET, &fss) == -1)
208 atf_tc_fail_errno("create snapshot");
209
210 activity_stop = true;
211 for (i = 0; i < NACTIVITY; i++)
212 RL(pthread_join(at[i], NULL));
213
214 RL(system(FSCK " /dev/rfss0"));
215 }
216
ATF_TC_CLEANUP(snapshotstress,tc)217 ATF_TC_CLEANUP(snapshotstress, tc)
218 {
219
220 unlink(IMGNAME);
221 }
222
ATF_TP_ADD_TCS(tp)223 ATF_TP_ADD_TCS(tp)
224 {
225 ATF_TP_ADD_TC(tp, snapshot);
226 ATF_TP_ADD_TC(tp, snapshotstress);
227 return 0;
228 }
229