1 /*        $NetBSD: snapshot.c,v 1.8 2019/07/09 16:24:01 maya 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 atf_no_error();
228 }
229