1/*	$NetBSD: fstest_lfs.c,v 1.4 2010/07/30 16:15:05 pooka Exp $	*/
2
3/*-
4 * Copyright (c) 2010 The NetBSD Foundation, Inc.
5 * All rights reserved.
6 *
7 * This code is derived from software contributed to The NetBSD Foundation
8 * by Nicolas Joly.
9 *
10 * Redistribution and use in source and binary forms, with or without
11 * modification, are permitted provided that the following conditions
12 * are met:
13 * 1. Redistributions of source code must retain the above copyright
14 *    notice, this list of conditions and the following disclaimer.
15 * 2. Redistributions in binary form must reproduce the above copyright
16 *    notice, this list of conditions and the following disclaimer in the
17 *    documentation and/or other materials provided with the distribution.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29 * POSSIBILITY OF SUCH DAMAGE.
30 */
31
32#include <sys/mount.h>
33#include <sys/stat.h>
34
35#include <atf-c.h>
36#include <pthread.h>
37#include <semaphore.h>
38#include <stdio.h>
39#include <stdlib.h>
40#include <string.h>
41#include <unistd.h>
42
43#include <ufs/ufs/ufsmount.h>
44
45#include <rump/rump.h>
46#include <rump/rump_syscalls.h>
47
48#include "h_fsmacros.h"
49#include "mount_lfs.h"
50
51sem_t lfs_clearnerloop;
52
53struct lfstestargs {
54	struct ufs_args ta_uargs;
55	pthread_t ta_cleanerthread;
56	sem_t ta_cleanerloop;
57	char ta_devpath[MAXPATHLEN];
58	char ta_imgpath[MAXPATHLEN];
59	char ta_mntpath[MAXPATHLEN];
60	char ta_hostpath[MAXPATHLEN];
61};
62
63int
64lfs_fstest_newfs(const atf_tc_t *tc, void **buf, const char *image, off_t size,
65	void *fspriv)
66{
67	char cmd[1024];
68	int res;
69	static unsigned int num = 0;
70	struct lfstestargs *args;
71
72	size /= 512;
73	snprintf(cmd, 1024, "newfs_lfs -D -F -s %"PRId64" ./%s >/dev/null",
74	    size, image);
75	res = system(cmd);
76	if (res != 0)
77		return res;
78
79	res = rump_init();
80	if (res != 0)
81		return res;
82
83	args = calloc(1, sizeof(*args));
84	if (args == NULL)
85		return -1;
86
87	strcpy(args->ta_hostpath, image);
88	snprintf(args->ta_devpath, MAXPATHLEN, "/dev/device%d.lfs", num);
89	snprintf(args->ta_imgpath, MAXPATHLEN, "%s", image);
90	args->ta_uargs.fspec = args->ta_devpath;
91	sem_init(&args->ta_cleanerloop, 0, 0);
92
93	res = rump_pub_etfs_register(args->ta_devpath, image, RUMP_ETFS_BLK);
94	if (res != 0) {
95		free(args);
96		return res;
97	}
98
99	*buf = args;
100	num++;
101
102	return 0;
103}
104
105int
106lfs_fstest_delfs(const atf_tc_t *tc, void *buf)
107{
108	int res;
109	struct lfstestargs *args = buf;
110
111	res = rump_pub_etfs_remove(args->ta_devpath);
112	if (res != 0)
113		return res;
114
115	res = unlink(args->ta_imgpath);
116	if (res != 0)
117		return res;
118
119	pthread_join(args->ta_cleanerthread, NULL);
120	free(args);
121
122	return 0;
123}
124
125static void *
126cleaner(void *arg)
127{
128	char thepath[MAXPATHLEN];
129	struct lfstestargs *args = arg;
130	const char *the_argv[7];
131	char buf[64];
132
133	/* this inspired by the cleaner code.  fixme */
134	sprintf(thepath, "/dev/r%s", args->ta_devpath+5);
135	rump_pub_etfs_register(thepath, args->ta_hostpath, RUMP_ETFS_CHR);
136	sprintf(buf, "%p", &args->ta_cleanerloop);
137
138	the_argv[0] = "megamaid";
139	the_argv[1] = "-D"; /* don't fork() & detach */
140	the_argv[2] = "-S";
141	the_argv[3] = buf;
142	the_argv[4] = args->ta_mntpath;
143	the_argv[5] = NULL;
144
145	/* xxxatf */
146	optind = 1;
147	opterr = 1;
148
149	lfs_cleaner_main(5, __UNCONST(the_argv));
150
151	return NULL;
152}
153
154int
155lfs_fstest_mount(const atf_tc_t *tc, void *buf, const char *path, int flags)
156{
157	struct lfstestargs *args = buf;
158	int res;
159
160	res = rump_sys_mkdir(path, 0777);
161	if (res == -1)
162		return res;
163
164	res = rump_sys_mount(MOUNT_LFS, path, flags, &args->ta_uargs,
165	    sizeof(args->ta_uargs));
166	if (res == -1)
167		return res;
168
169	strcpy(args->ta_mntpath, path);
170	res = pthread_create(&args->ta_cleanerthread, NULL, cleaner, args);
171	if (res)
172		return res;
173
174	/* wait for cleaner to initialize */
175	sem_wait(&args->ta_cleanerloop);
176
177	return 0;
178}
179
180int
181lfs_fstest_unmount(const atf_tc_t *tc, const char *path, int flags)
182{
183	int res;
184
185	res = rump_sys_unmount(path, flags);
186	if (res == -1) {
187		return res;
188	}
189
190	res = rump_sys_rmdir(path);
191	return res;
192}
193