1215976Sjmallett/* $NetBSD: snapshot.c,v 1.8 2019/07/09 16:24:01 maya Exp $ */ 2215976Sjmallett 3215976Sjmallett#include <sys/types.h> 4215976Sjmallett#include <sys/ioctl.h> 5215976Sjmallett#include <sys/mount.h> 6215976Sjmallett 7215976Sjmallett#include <dev/fssvar.h> 8215976Sjmallett 9215976Sjmallett#include <atf-c.h> 10215976Sjmallett#include <fcntl.h> 11215976Sjmallett#include <pthread.h> 12215976Sjmallett#include <stdio.h> 13215976Sjmallett#include <stdlib.h> 14215976Sjmallett#include <string.h> 15215976Sjmallett#include <unistd.h> 16215976Sjmallett 17215976SjmallettATF_TC_WITH_CLEANUP(snapshot); 18215976SjmallettATF_TC_HEAD(snapshot, tc) 19215976Sjmallett{ 20215976Sjmallett 21215976Sjmallett atf_tc_set_md_var(tc, "descr", "basic snapshot features"); 22215976Sjmallett} 23215976Sjmallett 24215976Sjmallettstatic void 25215976Sjmallettmakefile(const char *path) 26215976Sjmallett{ 27215976Sjmallett int fd; 28215976Sjmallett 29215976Sjmallett fd = rump_sys_open(path, O_CREAT | O_RDWR, 0777); 30215976Sjmallett if (fd == -1) 31215976Sjmallett atf_tc_fail_errno("create %s", path); 32215976Sjmallett rump_sys_close(fd); 33215976Sjmallett} 34215976Sjmallett 35215976SjmallettATF_TC_BODY(snapshot, tc) 36215976Sjmallett{ 37215976Sjmallett char buf[1024]; 38215976Sjmallett struct fss_set fss; 39215976Sjmallett int fssfd; 40215976Sjmallett int fd, fd2, i; 41215976Sjmallett 42215976Sjmallett if (system(NEWFS) == -1) 43215976Sjmallett atf_tc_fail_errno("cannot create file system"); 44215976Sjmallett 45215976Sjmallett rump_init(); 46215976Sjmallett begin(); 47215976Sjmallett 48215976Sjmallett if (rump_sys_mkdir("/mnt", 0777) == -1) 49215976Sjmallett atf_tc_fail_errno("mount point create"); 50215976Sjmallett if (rump_sys_mkdir("/snap", 0777) == -1) 51215976Sjmallett atf_tc_fail_errno("mount point 2 create"); 52215976Sjmallett 53215976Sjmallett rump_pub_etfs_register("/diskdev", IMGNAME, RUMP_ETFS_BLK); 54215976Sjmallett 55215976Sjmallett mount_diskfs("/diskdev", "/mnt"); 56215976Sjmallett 57215976Sjmallett#define TESTSTR1 "huihai\n" 58215976Sjmallett#define TESTSZ1 (sizeof(TESTSTR1)-1) 59215976Sjmallett#define TESTSTR2 "baana liten\n" 60215976Sjmallett#define TESTSZ2 (sizeof(TESTSTR2)-1) 61215976Sjmallett 62215976Sjmallett fd = rump_sys_open("/mnt/myfile", O_RDWR | O_CREAT, 0777); 63215976Sjmallett if (fd == -1) 64215976Sjmallett atf_tc_fail_errno("create file"); 65215976Sjmallett if (rump_sys_write(fd, TESTSTR1, TESTSZ1) != TESTSZ1) 66215976Sjmallett atf_tc_fail_errno("write fail"); 67215976Sjmallett 68215976Sjmallett fssfd = rump_sys_open("/dev/rfss0", O_RDWR); 69215976Sjmallett if (fssfd == -1) 70215976Sjmallett atf_tc_fail_errno("cannot open fss"); 71215976Sjmallett makefile(BAKNAME); 72215976Sjmallett memset(&fss, 0, sizeof(fss)); 73215976Sjmallett fss.fss_mount = __UNCONST("/mnt"); 74215976Sjmallett fss.fss_bstore = __UNCONST(BAKNAME); 75215976Sjmallett fss.fss_csize = 0; 76215976Sjmallett if (rump_sys_ioctl(fssfd, FSSIOCSET, &fss) == -1) 77215976Sjmallett atf_tc_fail_errno("create snapshot"); 78215976Sjmallett 79215976Sjmallett for (i = 0; i < 10000; i++) { 80215976Sjmallett if (rump_sys_write(fd, TESTSTR2, TESTSZ2) != TESTSZ2) 81215976Sjmallett atf_tc_fail_errno("write fail"); 82215976Sjmallett } 83215976Sjmallett rump_sys_sync(); 84215976Sjmallett 85215976Sjmallett /* technically we should fsck it first? */ 86215976Sjmallett mount_diskfs("/dev/fss0", "/snap"); 87215976Sjmallett 88215976Sjmallett /* check for old contents */ 89215976Sjmallett fd2 = rump_sys_open("/snap/myfile", O_RDONLY); 90215976Sjmallett if (fd2 == -1) 91215976Sjmallett atf_tc_fail_errno("fail"); 92215976Sjmallett memset(buf, 0, sizeof(buf)); 93215976Sjmallett if (rump_sys_read(fd2, buf, sizeof(buf)) == -1) 94215976Sjmallett atf_tc_fail_errno("read snap"); 95215976Sjmallett ATF_CHECK(strcmp(buf, TESTSTR1) == 0); 96215976Sjmallett 97215976Sjmallett /* check that new files are invisible in the snapshot */ 98215976Sjmallett makefile("/mnt/newfile"); 99215976Sjmallett if (rump_sys_open("/snap/newfile", O_RDONLY) != -1) 100215976Sjmallett atf_tc_fail("newfile exists in snapshot"); 101215976Sjmallett if (errno != ENOENT) 102215976Sjmallett atf_tc_fail_errno("newfile open should fail with ENOENT"); 103215976Sjmallett 104215976Sjmallett /* check that removed files are still visible in the snapshot */ 105215976Sjmallett rump_sys_unlink("/mnt/myfile"); 106215976Sjmallett 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 atf_no_error(); 228} 229