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
17ATF_TC_WITH_CLEANUP(snapshot);
18ATF_TC_HEAD(snapshot, tc)
19{
20
21	atf_tc_set_md_var(tc, "descr", "basic snapshot features");
22}
23
24static void
25makefile(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
35ATF_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
112ATF_TC_CLEANUP(snapshot, tc)
113{
114
115	unlink(IMGNAME);
116}
117
118ATF_TC_WITH_CLEANUP(snapshotstress);
119ATF_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
127static bool activity_stop = false;
128static pid_t wrkpid;
129
130static void *
131fs_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
166ATF_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
217ATF_TC_CLEANUP(snapshotstress, tc)
218{
219
220	unlink(IMGNAME);
221}
222
223ATF_TP_ADD_TCS(tp)
224{
225	ATF_TP_ADD_TC(tp, snapshot);
226	ATF_TP_ADD_TC(tp, snapshotstress);
227	return 0;
228}
229