xref: /freebsd/tests/sys/fs/fusefs/utils.cc (revision 474ba6fa3b81b13b183a47f8591e83fa8ba499f4)
19821f1d3SAlan Somers /*-
29821f1d3SAlan Somers  * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
39821f1d3SAlan Somers  *
49821f1d3SAlan Somers  * Copyright (c) 2019 The FreeBSD Foundation
59821f1d3SAlan Somers  *
69821f1d3SAlan Somers  * This software was developed by BFF Storage Systems, LLC under sponsorship
79821f1d3SAlan Somers  * from the FreeBSD Foundation.
89821f1d3SAlan Somers  *
99821f1d3SAlan Somers  * Redistribution and use in source and binary forms, with or without
109821f1d3SAlan Somers  * modification, are permitted provided that the following conditions
119821f1d3SAlan Somers  * are met:
129821f1d3SAlan Somers  * 1. Redistributions of source code must retain the above copyright
139821f1d3SAlan Somers  *    notice, this list of conditions and the following disclaimer.
149821f1d3SAlan Somers  * 2. Redistributions in binary form must reproduce the above copyright
159821f1d3SAlan Somers  *    notice, this list of conditions and the following disclaimer in the
169821f1d3SAlan Somers  *    documentation and/or other materials provided with the distribution.
179821f1d3SAlan Somers  *
189821f1d3SAlan Somers  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
199821f1d3SAlan Somers  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
209821f1d3SAlan Somers  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
219821f1d3SAlan Somers  * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
229821f1d3SAlan Somers  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
239821f1d3SAlan Somers  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
249821f1d3SAlan Somers  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
259821f1d3SAlan Somers  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
269821f1d3SAlan Somers  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
279821f1d3SAlan Somers  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
289821f1d3SAlan Somers  * SUCH DAMAGE.
299821f1d3SAlan Somers  */
309821f1d3SAlan Somers 
3109c01e67SAlan Somers extern "C" {
329821f1d3SAlan Somers #include <sys/param.h>
3309c01e67SAlan Somers #include <sys/mman.h>
349821f1d3SAlan Somers #include <sys/module.h>
359821f1d3SAlan Somers #include <sys/sysctl.h>
3609c01e67SAlan Somers #include <sys/wait.h>
3709c01e67SAlan Somers 
3809c01e67SAlan Somers #include <pwd.h>
3909c01e67SAlan Somers #include <semaphore.h>
4009c01e67SAlan Somers #include <unistd.h>
4109c01e67SAlan Somers }
429821f1d3SAlan Somers 
439821f1d3SAlan Somers #include <gtest/gtest.h>
449821f1d3SAlan Somers 
459821f1d3SAlan Somers #include "mockfs.hh"
469821f1d3SAlan Somers #include "utils.hh"
479821f1d3SAlan Somers 
489821f1d3SAlan Somers using namespace testing;
499821f1d3SAlan Somers 
509821f1d3SAlan Somers /* Check that fusefs(4) is accessible and the current user can mount(2) */
519821f1d3SAlan Somers void check_environment()
529821f1d3SAlan Somers {
539821f1d3SAlan Somers 	const char *devnode = "/dev/fuse";
549821f1d3SAlan Somers 	const char *usermount_node = "vfs.usermount";
559821f1d3SAlan Somers 	int usermount_val = 0;
569821f1d3SAlan Somers 	size_t usermount_size = sizeof(usermount_val);
579821f1d3SAlan Somers 	if (eaccess(devnode, R_OK | W_OK)) {
589821f1d3SAlan Somers 		if (errno == ENOENT) {
599821f1d3SAlan Somers 			GTEST_SKIP() << devnode << " does not exist";
609821f1d3SAlan Somers 		} else if (errno == EACCES) {
619821f1d3SAlan Somers 			GTEST_SKIP() << devnode <<
629821f1d3SAlan Somers 			    " is not accessible by the current user";
639821f1d3SAlan Somers 		} else {
649821f1d3SAlan Somers 			GTEST_SKIP() << strerror(errno);
659821f1d3SAlan Somers 		}
669821f1d3SAlan Somers 	}
679821f1d3SAlan Somers 	sysctlbyname(usermount_node, &usermount_val, &usermount_size,
689821f1d3SAlan Somers 		     NULL, 0);
699821f1d3SAlan Somers 	if (geteuid() != 0 && !usermount_val)
709821f1d3SAlan Somers 		GTEST_SKIP() << "current user is not allowed to mount";
719821f1d3SAlan Somers }
729821f1d3SAlan Somers 
739821f1d3SAlan Somers class FuseEnv: public Environment {
749821f1d3SAlan Somers 	virtual void SetUp() {
759821f1d3SAlan Somers 	}
769821f1d3SAlan Somers };
779821f1d3SAlan Somers 
789821f1d3SAlan Somers void FuseTest::SetUp() {
799821f1d3SAlan Somers 	const char *node = "vfs.maxbcachebuf";
809821f1d3SAlan Somers 	int val = 0;
819821f1d3SAlan Somers 	size_t size = sizeof(val);
829821f1d3SAlan Somers 
839821f1d3SAlan Somers 	/*
849821f1d3SAlan Somers 	 * XXX check_environment should be called from FuseEnv::SetUp, but
859821f1d3SAlan Somers 	 * can't due to https://github.com/google/googletest/issues/2189
869821f1d3SAlan Somers 	 */
879821f1d3SAlan Somers 	check_environment();
889821f1d3SAlan Somers 	if (IsSkipped())
899821f1d3SAlan Somers 		return;
909821f1d3SAlan Somers 
919821f1d3SAlan Somers 	ASSERT_EQ(0, sysctlbyname(node, &val, &size, NULL, 0))
929821f1d3SAlan Somers 		<< strerror(errno);
939821f1d3SAlan Somers 	m_maxbcachebuf = val;
949821f1d3SAlan Somers 
959821f1d3SAlan Somers 	try {
9691ff3a0dSAlan Somers 		m_mock = new MockFS(m_maxreadahead, m_allow_other,
97140bb492SAlan Somers 			m_default_permissions, m_push_symlinks_in, m_ro,
9891ff3a0dSAlan Somers 			m_init_flags);
99caf5f57dSAlan Somers 		/*
100caf5f57dSAlan Somers 		 * FUSE_ACCESS is called almost universally.  Expecting it in
101caf5f57dSAlan Somers 		 * each test case would be super-annoying.  Instead, set a
102caf5f57dSAlan Somers 		 * default expectation for FUSE_ACCESS and return ENOSYS.
103caf5f57dSAlan Somers 		 *
104caf5f57dSAlan Somers 		 * Individual test cases can override this expectation since
105caf5f57dSAlan Somers 		 * googlemock evaluates expectations in LIFO order.
106caf5f57dSAlan Somers 		 */
107caf5f57dSAlan Somers 		EXPECT_CALL(*m_mock, process(
108caf5f57dSAlan Somers 			ResultOf([=](auto in) {
109caf5f57dSAlan Somers 				return (in->header.opcode == FUSE_ACCESS);
110caf5f57dSAlan Somers 			}, Eq(true)),
111caf5f57dSAlan Somers 			_)
112caf5f57dSAlan Somers 		).Times(AnyNumber())
113caf5f57dSAlan Somers 		.WillRepeatedly(Invoke(ReturnErrno(ENOSYS)));
1149821f1d3SAlan Somers 	} catch (std::system_error err) {
1159821f1d3SAlan Somers 		FAIL() << err.what();
1169821f1d3SAlan Somers 	}
1179821f1d3SAlan Somers }
1189821f1d3SAlan Somers 
11991ff3a0dSAlan Somers void
12091ff3a0dSAlan Somers FuseTest::expect_access(uint64_t ino, mode_t access_mode, int error)
12191ff3a0dSAlan Somers {
12291ff3a0dSAlan Somers 	EXPECT_CALL(*m_mock, process(
12391ff3a0dSAlan Somers 		ResultOf([=](auto in) {
12491ff3a0dSAlan Somers 			return (in->header.opcode == FUSE_ACCESS &&
12591ff3a0dSAlan Somers 				in->header.nodeid == ino &&
12691ff3a0dSAlan Somers 				in->body.access.mask == access_mode);
12791ff3a0dSAlan Somers 		}, Eq(true)),
12891ff3a0dSAlan Somers 		_)
12991ff3a0dSAlan Somers 	).WillOnce(Invoke(ReturnErrno(error)));
13091ff3a0dSAlan Somers }
13191ff3a0dSAlan Somers 
1329f10f423SAlan Somers void
1339f10f423SAlan Somers FuseTest::expect_flush(uint64_t ino, int times, ProcessMockerT r)
1349f10f423SAlan Somers {
1359f10f423SAlan Somers 	EXPECT_CALL(*m_mock, process(
1369f10f423SAlan Somers 		ResultOf([=](auto in) {
1379f10f423SAlan Somers 			return (in->header.opcode == FUSE_FLUSH &&
1389f10f423SAlan Somers 				in->header.nodeid == ino);
1399f10f423SAlan Somers 		}, Eq(true)),
1409f10f423SAlan Somers 		_)
1419f10f423SAlan Somers 	).Times(times)
1429f10f423SAlan Somers 	.WillRepeatedly(Invoke(r));
1439f10f423SAlan Somers }
1449f10f423SAlan Somers 
145ff4fbdf5SAlan Somers void
146ff4fbdf5SAlan Somers FuseTest::expect_forget(uint64_t ino, uint64_t nlookup)
147ff4fbdf5SAlan Somers {
148ff4fbdf5SAlan Somers 	EXPECT_CALL(*m_mock, process(
149ff4fbdf5SAlan Somers 		ResultOf([=](auto in) {
150ff4fbdf5SAlan Somers 			return (in->header.opcode == FUSE_FORGET &&
151ff4fbdf5SAlan Somers 				in->header.nodeid == ino &&
152ff4fbdf5SAlan Somers 				in->body.forget.nlookup == nlookup);
153ff4fbdf5SAlan Somers 		}, Eq(true)),
154ff4fbdf5SAlan Somers 		_)
155ff4fbdf5SAlan Somers 	).WillOnce(Invoke([](auto in __unused, auto &out __unused) {
156ff4fbdf5SAlan Somers 		/* FUSE_FORGET has no response! */
157ff4fbdf5SAlan Somers 	}));
158ff4fbdf5SAlan Somers }
159ff4fbdf5SAlan Somers 
1609821f1d3SAlan Somers void FuseTest::expect_getattr(uint64_t ino, uint64_t size)
1619821f1d3SAlan Somers {
1629821f1d3SAlan Somers 	EXPECT_CALL(*m_mock, process(
1639821f1d3SAlan Somers 		ResultOf([=](auto in) {
1649821f1d3SAlan Somers 			return (in->header.opcode == FUSE_GETATTR &&
1659821f1d3SAlan Somers 				in->header.nodeid == ino);
1669821f1d3SAlan Somers 		}, Eq(true)),
1679821f1d3SAlan Somers 		_)
168cad67791SAlan Somers 	).WillOnce(Invoke(ReturnImmediate([=](auto i __unused, auto out) {
1699821f1d3SAlan Somers 		SET_OUT_HEADER_LEN(out, attr);
1709821f1d3SAlan Somers 		out->body.attr.attr.ino = ino;	// Must match nodeid
1719821f1d3SAlan Somers 		out->body.attr.attr.mode = S_IFREG | 0644;
1729821f1d3SAlan Somers 		out->body.attr.attr.size = size;
1739821f1d3SAlan Somers 		out->body.attr.attr_valid = UINT64_MAX;
1749821f1d3SAlan Somers 	})));
1759821f1d3SAlan Somers }
1769821f1d3SAlan Somers 
1779821f1d3SAlan Somers void FuseTest::expect_lookup(const char *relpath, uint64_t ino, mode_t mode,
178*474ba6faSAlan Somers 	uint64_t size, int times, uint64_t attr_valid, uid_t uid, gid_t gid)
1799821f1d3SAlan Somers {
1809821f1d3SAlan Somers 	EXPECT_LOOKUP(1, relpath)
1819821f1d3SAlan Somers 	.Times(times)
1829821f1d3SAlan Somers 	.WillRepeatedly(Invoke(ReturnImmediate([=](auto in __unused, auto out) {
1839821f1d3SAlan Somers 		SET_OUT_HEADER_LEN(out, entry);
1849821f1d3SAlan Somers 		out->body.entry.attr.mode = mode;
1859821f1d3SAlan Somers 		out->body.entry.nodeid = ino;
1869821f1d3SAlan Somers 		out->body.entry.attr.nlink = 1;
187ff4fbdf5SAlan Somers 		out->body.entry.attr_valid = attr_valid;
1889821f1d3SAlan Somers 		out->body.entry.attr.size = size;
189ff4fbdf5SAlan Somers 		out->body.entry.attr.uid = uid;
190*474ba6faSAlan Somers 		out->body.entry.attr.gid = gid;
1919821f1d3SAlan Somers 	})));
1929821f1d3SAlan Somers }
1939821f1d3SAlan Somers 
1949821f1d3SAlan Somers void FuseTest::expect_open(uint64_t ino, uint32_t flags, int times)
1959821f1d3SAlan Somers {
1969821f1d3SAlan Somers 	EXPECT_CALL(*m_mock, process(
1979821f1d3SAlan Somers 		ResultOf([=](auto in) {
1989821f1d3SAlan Somers 			return (in->header.opcode == FUSE_OPEN &&
1999821f1d3SAlan Somers 				in->header.nodeid == ino);
2009821f1d3SAlan Somers 		}, Eq(true)),
2019821f1d3SAlan Somers 		_)
2029821f1d3SAlan Somers 	).Times(times)
2039821f1d3SAlan Somers 	.WillRepeatedly(Invoke(ReturnImmediate([=](auto in __unused, auto out) {
2049821f1d3SAlan Somers 		out->header.len = sizeof(out->header);
2059821f1d3SAlan Somers 		SET_OUT_HEADER_LEN(out, open);
2069821f1d3SAlan Somers 		out->body.open.fh = FH;
2079821f1d3SAlan Somers 		out->body.open.open_flags = flags;
2089821f1d3SAlan Somers 	})));
2099821f1d3SAlan Somers }
2109821f1d3SAlan Somers 
2119821f1d3SAlan Somers void FuseTest::expect_opendir(uint64_t ino)
2129821f1d3SAlan Somers {
213363a7416SAlan Somers 	/* opendir(3) calls fstatfs */
2149821f1d3SAlan Somers 	EXPECT_CALL(*m_mock, process(
2159821f1d3SAlan Somers 		ResultOf([](auto in) {
2169821f1d3SAlan Somers 			return (in->header.opcode == FUSE_STATFS);
2179821f1d3SAlan Somers 		}, Eq(true)),
2189821f1d3SAlan Somers 		_)
2199821f1d3SAlan Somers 	).WillRepeatedly(Invoke(ReturnImmediate([=](auto i __unused, auto out) {
2209821f1d3SAlan Somers 		SET_OUT_HEADER_LEN(out, statfs);
2219821f1d3SAlan Somers 	})));
2229821f1d3SAlan Somers 
2239821f1d3SAlan Somers 	EXPECT_CALL(*m_mock, process(
2249821f1d3SAlan Somers 		ResultOf([=](auto in) {
2259821f1d3SAlan Somers 			return (in->header.opcode == FUSE_OPENDIR &&
2269821f1d3SAlan Somers 				in->header.nodeid == ino);
2279821f1d3SAlan Somers 		}, Eq(true)),
2289821f1d3SAlan Somers 		_)
2299821f1d3SAlan Somers 	).WillOnce(Invoke(ReturnImmediate([=](auto in __unused, auto out) {
2309821f1d3SAlan Somers 		out->header.len = sizeof(out->header);
2319821f1d3SAlan Somers 		SET_OUT_HEADER_LEN(out, open);
2329821f1d3SAlan Somers 		out->body.open.fh = FH;
2339821f1d3SAlan Somers 	})));
2349821f1d3SAlan Somers }
2359821f1d3SAlan Somers 
2369821f1d3SAlan Somers void FuseTest::expect_read(uint64_t ino, uint64_t offset, uint64_t isize,
2379821f1d3SAlan Somers 	uint64_t osize, const void *contents)
2389821f1d3SAlan Somers {
2399821f1d3SAlan Somers 	EXPECT_CALL(*m_mock, process(
2409821f1d3SAlan Somers 		ResultOf([=](auto in) {
2419821f1d3SAlan Somers 			return (in->header.opcode == FUSE_READ &&
2429821f1d3SAlan Somers 				in->header.nodeid == ino &&
2439821f1d3SAlan Somers 				in->body.read.fh == FH &&
2449821f1d3SAlan Somers 				in->body.read.offset == offset &&
2459821f1d3SAlan Somers 				in->body.read.size == isize);
2469821f1d3SAlan Somers 		}, Eq(true)),
2479821f1d3SAlan Somers 		_)
2489821f1d3SAlan Somers 	).WillOnce(Invoke(ReturnImmediate([=](auto in __unused, auto out) {
2499821f1d3SAlan Somers 		out->header.len = sizeof(struct fuse_out_header) + osize;
2509821f1d3SAlan Somers 		memmove(out->body.bytes, contents, osize);
2519821f1d3SAlan Somers 	}))).RetiresOnSaturation();
2529821f1d3SAlan Somers }
2539821f1d3SAlan Somers 
25442d50d16SAlan Somers void FuseTest::expect_release(uint64_t ino, uint64_t fh)
2559821f1d3SAlan Somers {
2569821f1d3SAlan Somers 	EXPECT_CALL(*m_mock, process(
2579821f1d3SAlan Somers 		ResultOf([=](auto in) {
2589821f1d3SAlan Somers 			return (in->header.opcode == FUSE_RELEASE &&
2599821f1d3SAlan Somers 				in->header.nodeid == ino &&
26042d50d16SAlan Somers 				in->body.release.fh == fh);
2619821f1d3SAlan Somers 		}, Eq(true)),
2629821f1d3SAlan Somers 		_)
263e0bec057SAlan Somers 	).WillOnce(Invoke(ReturnErrno(0)));
2649821f1d3SAlan Somers }
265e0bec057SAlan Somers 
26635cf0e7eSAlan Somers void FuseTest::expect_releasedir(uint64_t ino, ProcessMockerT r)
26735cf0e7eSAlan Somers {
26835cf0e7eSAlan Somers 	EXPECT_CALL(*m_mock, process(
26935cf0e7eSAlan Somers 		ResultOf([=](auto in) {
27035cf0e7eSAlan Somers 			return (in->header.opcode == FUSE_RELEASEDIR &&
27135cf0e7eSAlan Somers 				in->header.nodeid == ino &&
27235cf0e7eSAlan Somers 				in->body.release.fh == FH);
27335cf0e7eSAlan Somers 		}, Eq(true)),
27435cf0e7eSAlan Somers 		_)
27535cf0e7eSAlan Somers 	).WillOnce(Invoke(r));
27635cf0e7eSAlan Somers }
27735cf0e7eSAlan Somers 
278ff4fbdf5SAlan Somers void FuseTest::expect_unlink(uint64_t parent, const char *path, int error)
279ff4fbdf5SAlan Somers {
280ff4fbdf5SAlan Somers 	EXPECT_CALL(*m_mock, process(
281ff4fbdf5SAlan Somers 		ResultOf([=](auto in) {
282ff4fbdf5SAlan Somers 			return (in->header.opcode == FUSE_UNLINK &&
283ff4fbdf5SAlan Somers 				0 == strcmp(path, in->body.unlink) &&
284ff4fbdf5SAlan Somers 				in->header.nodeid == parent);
285ff4fbdf5SAlan Somers 		}, Eq(true)),
286ff4fbdf5SAlan Somers 		_)
287ff4fbdf5SAlan Somers 	).WillOnce(Invoke(ReturnErrno(error)));
288ff4fbdf5SAlan Somers }
289ff4fbdf5SAlan Somers 
2909821f1d3SAlan Somers void FuseTest::expect_write(uint64_t ino, uint64_t offset, uint64_t isize,
2919821f1d3SAlan Somers 	uint64_t osize, uint32_t flags, const void *contents)
2929821f1d3SAlan Somers {
2939821f1d3SAlan Somers 	EXPECT_CALL(*m_mock, process(
2949821f1d3SAlan Somers 		ResultOf([=](auto in) {
2959821f1d3SAlan Somers 			const char *buf = (const char*)in->body.bytes +
2969821f1d3SAlan Somers 				sizeof(struct fuse_write_in);
2979821f1d3SAlan Somers 			bool pid_ok;
2989821f1d3SAlan Somers 
2999821f1d3SAlan Somers 			if (in->body.write.write_flags & FUSE_WRITE_CACHE)
3009821f1d3SAlan Somers 				pid_ok = true;
3019821f1d3SAlan Somers 			else
3029821f1d3SAlan Somers 				pid_ok = (pid_t)in->header.pid == getpid();
3039821f1d3SAlan Somers 
3049821f1d3SAlan Somers 			return (in->header.opcode == FUSE_WRITE &&
3059821f1d3SAlan Somers 				in->header.nodeid == ino &&
3069821f1d3SAlan Somers 				in->body.write.fh == FH &&
3079821f1d3SAlan Somers 				in->body.write.offset == offset  &&
3089821f1d3SAlan Somers 				in->body.write.size == isize &&
3099821f1d3SAlan Somers 				pid_ok &&
3109821f1d3SAlan Somers 				in->body.write.write_flags == flags &&
3119821f1d3SAlan Somers 				0 == bcmp(buf, contents, isize));
3129821f1d3SAlan Somers 		}, Eq(true)),
3139821f1d3SAlan Somers 		_)
3149821f1d3SAlan Somers 	).WillOnce(Invoke(ReturnImmediate([=](auto in __unused, auto out) {
3159821f1d3SAlan Somers 		SET_OUT_HEADER_LEN(out, write);
3169821f1d3SAlan Somers 		out->body.write.size = osize;
3179821f1d3SAlan Somers 	})));
3189821f1d3SAlan Somers }
3199821f1d3SAlan Somers 
32009c01e67SAlan Somers static void
32109c01e67SAlan Somers get_unprivileged_uid(uid_t *uid)
32209c01e67SAlan Somers {
32309c01e67SAlan Somers 	struct passwd *pw;
32409c01e67SAlan Somers 
32509c01e67SAlan Somers 	/*
32609c01e67SAlan Somers 	 * First try "tests", Kyua's default unprivileged user.  XXX after
32709c01e67SAlan Somers 	 * GoogleTest gains a proper Kyua wrapper, get this with the Kyua API
32809c01e67SAlan Somers 	 */
32909c01e67SAlan Somers 	pw = getpwnam("tests");
33009c01e67SAlan Somers 	if (pw == NULL) {
33109c01e67SAlan Somers 		/* Fall back to "nobody" */
33209c01e67SAlan Somers 		pw = getpwnam("nobody");
33309c01e67SAlan Somers 	}
33409c01e67SAlan Somers 	if (pw == NULL)
33509c01e67SAlan Somers 		GTEST_SKIP() << "Test requires an unprivileged user";
33609c01e67SAlan Somers 	*uid = pw->pw_uid;
33709c01e67SAlan Somers }
33809c01e67SAlan Somers 
33909c01e67SAlan Somers void
340a1542146SAlan Somers FuseTest::fork(bool drop_privs, int *child_status,
341a1542146SAlan Somers 	std::function<void()> parent_func,
34209c01e67SAlan Somers 	std::function<int()> child_func)
34309c01e67SAlan Somers {
34409c01e67SAlan Somers 	sem_t *sem;
34509c01e67SAlan Somers 	int mprot = PROT_READ | PROT_WRITE;
34609c01e67SAlan Somers 	int mflags = MAP_ANON | MAP_SHARED;
34709c01e67SAlan Somers 	pid_t child;
34809c01e67SAlan Somers 	uid_t uid;
34909c01e67SAlan Somers 
35009c01e67SAlan Somers 	if (drop_privs) {
35109c01e67SAlan Somers 		get_unprivileged_uid(&uid);
35209c01e67SAlan Somers 		if (IsSkipped())
35309c01e67SAlan Somers 			return;
35409c01e67SAlan Somers 	}
35509c01e67SAlan Somers 
35609c01e67SAlan Somers 	sem = (sem_t*)mmap(NULL, sizeof(*sem), mprot, mflags, -1, 0);
35709c01e67SAlan Somers 	ASSERT_NE(MAP_FAILED, sem) << strerror(errno);
35809c01e67SAlan Somers 	ASSERT_EQ(0, sem_init(sem, 1, 0)) << strerror(errno);
35909c01e67SAlan Somers 
36009c01e67SAlan Somers 	if ((child = ::fork()) == 0) {
36109c01e67SAlan Somers 		/* In child */
36209c01e67SAlan Somers 		int err = 0;
36309c01e67SAlan Somers 
36409c01e67SAlan Somers 		if (sem_wait(sem)) {
36509c01e67SAlan Somers 			perror("sem_wait");
36609c01e67SAlan Somers 			err = 1;
36709c01e67SAlan Somers 			goto out;
36809c01e67SAlan Somers 		}
36909c01e67SAlan Somers 
37009c01e67SAlan Somers 		if (drop_privs && 0 != setreuid(-1, uid)) {
37109c01e67SAlan Somers 			perror("setreuid");
37209c01e67SAlan Somers 			err = 1;
37309c01e67SAlan Somers 			goto out;
37409c01e67SAlan Somers 		}
37509c01e67SAlan Somers 		err = child_func();
37609c01e67SAlan Somers 
37709c01e67SAlan Somers out:
37809c01e67SAlan Somers 		sem_destroy(sem);
37909c01e67SAlan Somers 		_exit(err);
38009c01e67SAlan Somers 	} else if (child > 0) {
38109c01e67SAlan Somers 		/*
38209c01e67SAlan Somers 		 * In parent.  Cleanup must happen here, because it's still
38309c01e67SAlan Somers 		 * privileged.
38409c01e67SAlan Somers 		 */
38509c01e67SAlan Somers 		m_mock->m_child_pid = child;
38609c01e67SAlan Somers 		ASSERT_NO_FATAL_FAILURE(parent_func());
38709c01e67SAlan Somers 
38809c01e67SAlan Somers 		/* Signal the child process to go */
38909c01e67SAlan Somers 		ASSERT_EQ(0, sem_post(sem)) << strerror(errno);
39009c01e67SAlan Somers 
391a1542146SAlan Somers 		ASSERT_LE(0, wait(child_status)) << strerror(errno);
39209c01e67SAlan Somers 	} else {
39309c01e67SAlan Somers 		FAIL() << strerror(errno);
39409c01e67SAlan Somers 	}
39509c01e67SAlan Somers 	munmap(sem, sizeof(*sem));
396a1542146SAlan Somers 	return;
39709c01e67SAlan Somers }
39809c01e67SAlan Somers 
3999821f1d3SAlan Somers static void usage(char* progname) {
4009821f1d3SAlan Somers 	fprintf(stderr, "Usage: %s [-v]\n\t-v increase verbosity\n", progname);
4019821f1d3SAlan Somers 	exit(2);
4029821f1d3SAlan Somers }
4039821f1d3SAlan Somers 
4049821f1d3SAlan Somers int main(int argc, char **argv) {
4059821f1d3SAlan Somers 	int ch;
4069821f1d3SAlan Somers 	FuseEnv *fuse_env = new FuseEnv;
4079821f1d3SAlan Somers 
4089821f1d3SAlan Somers 	InitGoogleTest(&argc, argv);
4099821f1d3SAlan Somers 	AddGlobalTestEnvironment(fuse_env);
4109821f1d3SAlan Somers 
4119821f1d3SAlan Somers 	while ((ch = getopt(argc, argv, "v")) != -1) {
4129821f1d3SAlan Somers 		switch (ch) {
4139821f1d3SAlan Somers 			case 'v':
4149821f1d3SAlan Somers 				verbosity++;
4159821f1d3SAlan Somers 				break;
4169821f1d3SAlan Somers 			default:
4179821f1d3SAlan Somers 				usage(argv[0]);
4189821f1d3SAlan Somers 				break;
4199821f1d3SAlan Somers 		}
4209821f1d3SAlan Somers 	}
4219821f1d3SAlan Somers 
4229821f1d3SAlan Somers 	return (RUN_ALL_TESTS());
4239821f1d3SAlan Somers }
424