xref: /freebsd/sys/fs/cuse/cuse.c (revision 2c28cd09d9f7ac266a8c7b283384712aba0af93a)
1fa0f6e62SHans Petter Selasky /* $FreeBSD$ */
2fa0f6e62SHans Petter Selasky /*-
3*2c28cd09SHans Petter Selasky  * Copyright (c) 2010-2022 Hans Petter Selasky. All rights reserved.
4fa0f6e62SHans Petter Selasky  *
5fa0f6e62SHans Petter Selasky  * Redistribution and use in source and binary forms, with or without
6fa0f6e62SHans Petter Selasky  * modification, are permitted provided that the following conditions
7fa0f6e62SHans Petter Selasky  * are met:
8fa0f6e62SHans Petter Selasky  * 1. Redistributions of source code must retain the above copyright
9fa0f6e62SHans Petter Selasky  *    notice, this list of conditions and the following disclaimer.
10fa0f6e62SHans Petter Selasky  * 2. Redistributions in binary form must reproduce the above copyright
11fa0f6e62SHans Petter Selasky  *    notice, this list of conditions and the following disclaimer in the
12fa0f6e62SHans Petter Selasky  *    documentation and/or other materials provided with the distribution.
13fa0f6e62SHans Petter Selasky  *
14fa0f6e62SHans Petter Selasky  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15fa0f6e62SHans Petter Selasky  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16fa0f6e62SHans Petter Selasky  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17fa0f6e62SHans Petter Selasky  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18fa0f6e62SHans Petter Selasky  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19fa0f6e62SHans Petter Selasky  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20fa0f6e62SHans Petter Selasky  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21fa0f6e62SHans Petter Selasky  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22fa0f6e62SHans Petter Selasky  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23fa0f6e62SHans Petter Selasky  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24fa0f6e62SHans Petter Selasky  * SUCH DAMAGE.
25fa0f6e62SHans Petter Selasky  */
26fa0f6e62SHans Petter Selasky 
27fa0f6e62SHans Petter Selasky #include <sys/stdint.h>
28fa0f6e62SHans Petter Selasky #include <sys/stddef.h>
29fa0f6e62SHans Petter Selasky #include <sys/param.h>
30fa0f6e62SHans Petter Selasky #include <sys/types.h>
31fa0f6e62SHans Petter Selasky #include <sys/systm.h>
32fa0f6e62SHans Petter Selasky #include <sys/conf.h>
33fa0f6e62SHans Petter Selasky #include <sys/kernel.h>
34fa0f6e62SHans Petter Selasky #include <sys/bus.h>
35fa0f6e62SHans Petter Selasky #include <sys/linker_set.h>
36fa0f6e62SHans Petter Selasky #include <sys/module.h>
37fa0f6e62SHans Petter Selasky #include <sys/lock.h>
38fa0f6e62SHans Petter Selasky #include <sys/mutex.h>
39fa0f6e62SHans Petter Selasky #include <sys/condvar.h>
40fa0f6e62SHans Petter Selasky #include <sys/sysctl.h>
41fa0f6e62SHans Petter Selasky #include <sys/unistd.h>
42fa0f6e62SHans Petter Selasky #include <sys/malloc.h>
43fa0f6e62SHans Petter Selasky #include <sys/priv.h>
44fa0f6e62SHans Petter Selasky #include <sys/uio.h>
45fa0f6e62SHans Petter Selasky #include <sys/poll.h>
46fa0f6e62SHans Petter Selasky #include <sys/sx.h>
4756c49c63SHans Petter Selasky #include <sys/rwlock.h>
48fa0f6e62SHans Petter Selasky #include <sys/queue.h>
49fa0f6e62SHans Petter Selasky #include <sys/fcntl.h>
50fa0f6e62SHans Petter Selasky #include <sys/proc.h>
51fa0f6e62SHans Petter Selasky #include <sys/vnode.h>
52fa0f6e62SHans Petter Selasky #include <sys/selinfo.h>
53fa0f6e62SHans Petter Selasky #include <sys/ptrace.h>
54db92a6cdSHans Petter Selasky #include <sys/sysent.h>
55fa0f6e62SHans Petter Selasky 
56fa0f6e62SHans Petter Selasky #include <machine/bus.h>
57fa0f6e62SHans Petter Selasky 
58fa0f6e62SHans Petter Selasky #include <vm/vm.h>
59fa0f6e62SHans Petter Selasky #include <vm/pmap.h>
6056c49c63SHans Petter Selasky #include <vm/vm_object.h>
6156c49c63SHans Petter Selasky #include <vm/vm_page.h>
6256c49c63SHans Petter Selasky #include <vm/vm_pager.h>
63fa0f6e62SHans Petter Selasky 
64fa0f6e62SHans Petter Selasky #include <fs/cuse/cuse_defs.h>
65fa0f6e62SHans Petter Selasky #include <fs/cuse/cuse_ioctl.h>
66fa0f6e62SHans Petter Selasky 
67*2c28cd09SHans Petter Selasky #define	CUSE_ALLOC_PAGES_MAX \
68*2c28cd09SHans Petter Selasky 	(CUSE_ALLOC_BYTES_MAX / PAGE_SIZE)
69*2c28cd09SHans Petter Selasky 
70*2c28cd09SHans Petter Selasky #if (CUSE_ALLOC_PAGES_MAX == 0)
71*2c28cd09SHans Petter Selasky #error "PAGE_SIZE is too big!"
72*2c28cd09SHans Petter Selasky #endif
73*2c28cd09SHans Petter Selasky 
74a71074e0SHans Petter Selasky static int
75a71074e0SHans Petter Selasky cuse_modevent(module_t mod, int type, void *data)
76a71074e0SHans Petter Selasky {
77a71074e0SHans Petter Selasky 	switch (type) {
78a71074e0SHans Petter Selasky 	case MOD_LOAD:
79a71074e0SHans Petter Selasky 	case MOD_UNLOAD:
80a71074e0SHans Petter Selasky 		return (0);
81a71074e0SHans Petter Selasky 	default:
82a71074e0SHans Petter Selasky 		return (EOPNOTSUPP);
83a71074e0SHans Petter Selasky 	}
84a71074e0SHans Petter Selasky }
85a71074e0SHans Petter Selasky 
86a71074e0SHans Petter Selasky static moduledata_t cuse_mod = {
87a71074e0SHans Petter Selasky 	.name = "cuse",
88a71074e0SHans Petter Selasky 	.evhand = &cuse_modevent,
89a71074e0SHans Petter Selasky };
90a71074e0SHans Petter Selasky 
91a71074e0SHans Petter Selasky DECLARE_MODULE(cuse, cuse_mod, SI_SUB_DEVFS, SI_ORDER_FIRST);
92fa0f6e62SHans Petter Selasky MODULE_VERSION(cuse, 1);
93fa0f6e62SHans Petter Selasky 
9419f1b6fbSHans Petter Selasky /*
9519f1b6fbSHans Petter Selasky  * Prevent cuse4bsd.ko and cuse.ko from loading at the same time by
9619f1b6fbSHans Petter Selasky  * declaring support for the cuse4bsd interface in cuse.ko:
9719f1b6fbSHans Petter Selasky  */
9819f1b6fbSHans Petter Selasky MODULE_VERSION(cuse4bsd, 1);
9919f1b6fbSHans Petter Selasky 
100fa0f6e62SHans Petter Selasky #ifdef FEATURE
101fa0f6e62SHans Petter Selasky FEATURE(cuse, "Userspace character devices");
102fa0f6e62SHans Petter Selasky #endif
103fa0f6e62SHans Petter Selasky 
104fa0f6e62SHans Petter Selasky struct cuse_command;
105fa0f6e62SHans Petter Selasky struct cuse_server;
106fa0f6e62SHans Petter Selasky struct cuse_client;
107fa0f6e62SHans Petter Selasky 
108fa0f6e62SHans Petter Selasky struct cuse_client_command {
109fa0f6e62SHans Petter Selasky 	TAILQ_ENTRY(cuse_client_command) entry;
110fa0f6e62SHans Petter Selasky 	struct cuse_command sub;
111fa0f6e62SHans Petter Selasky 	struct sx sx;
112fa0f6e62SHans Petter Selasky 	struct cv cv;
113fa0f6e62SHans Petter Selasky 	struct thread *entered;
114fa0f6e62SHans Petter Selasky 	struct cuse_client *client;
115fa0f6e62SHans Petter Selasky 	struct proc *proc_curr;
116fa0f6e62SHans Petter Selasky 	int	proc_refs;
117fa0f6e62SHans Petter Selasky 	int	got_signal;
118fa0f6e62SHans Petter Selasky 	int	error;
119fa0f6e62SHans Petter Selasky 	int	command;
120fa0f6e62SHans Petter Selasky };
121fa0f6e62SHans Petter Selasky 
122fa0f6e62SHans Petter Selasky struct cuse_memory {
12356c49c63SHans Petter Selasky 	TAILQ_ENTRY(cuse_memory) entry;
12456c49c63SHans Petter Selasky 	vm_object_t object;
125fa0f6e62SHans Petter Selasky 	uint32_t page_count;
12656c49c63SHans Petter Selasky 	uint32_t alloc_nr;
127fa0f6e62SHans Petter Selasky };
128fa0f6e62SHans Petter Selasky 
129fa0f6e62SHans Petter Selasky struct cuse_server_dev {
130fa0f6e62SHans Petter Selasky 	TAILQ_ENTRY(cuse_server_dev) entry;
131fa0f6e62SHans Petter Selasky 	struct cuse_server *server;
132fa0f6e62SHans Petter Selasky 	struct cdev *kern_dev;
133fa0f6e62SHans Petter Selasky 	struct cuse_dev *user_dev;
134fa0f6e62SHans Petter Selasky };
135fa0f6e62SHans Petter Selasky 
136fa0f6e62SHans Petter Selasky struct cuse_server {
137fa0f6e62SHans Petter Selasky 	TAILQ_ENTRY(cuse_server) entry;
138fa0f6e62SHans Petter Selasky 	TAILQ_HEAD(, cuse_client_command) head;
139fa0f6e62SHans Petter Selasky 	TAILQ_HEAD(, cuse_server_dev) hdev;
140fa0f6e62SHans Petter Selasky 	TAILQ_HEAD(, cuse_client) hcli;
14156c49c63SHans Petter Selasky 	TAILQ_HEAD(, cuse_memory) hmem;
14298029019SHans Petter Selasky 	struct mtx mtx;
143fa0f6e62SHans Petter Selasky 	struct cv cv;
144fa0f6e62SHans Petter Selasky 	struct selinfo selinfo;
145ac60e801SHans Petter Selasky 	pid_t	pid;
146fa0f6e62SHans Petter Selasky 	int	is_closing;
147fa0f6e62SHans Petter Selasky 	int	refs;
148fa0f6e62SHans Petter Selasky };
149fa0f6e62SHans Petter Selasky 
150fa0f6e62SHans Petter Selasky struct cuse_client {
151fa0f6e62SHans Petter Selasky 	TAILQ_ENTRY(cuse_client) entry;
152fa0f6e62SHans Petter Selasky 	TAILQ_ENTRY(cuse_client) entry_ref;
153fa0f6e62SHans Petter Selasky 	struct cuse_client_command cmds[CUSE_CMD_MAX];
154fa0f6e62SHans Petter Selasky 	struct cuse_server *server;
155fa0f6e62SHans Petter Selasky 	struct cuse_server_dev *server_dev;
156fa0f6e62SHans Petter Selasky 
157fa0f6e62SHans Petter Selasky 	uint8_t	ioctl_buffer[CUSE_BUFFER_MAX] __aligned(4);
158fa0f6e62SHans Petter Selasky 
159fa0f6e62SHans Petter Selasky 	int	fflags;			/* file flags */
160fa0f6e62SHans Petter Selasky 	int	cflags;			/* client flags */
161fa0f6e62SHans Petter Selasky #define	CUSE_CLI_IS_CLOSING 0x01
162fa0f6e62SHans Petter Selasky #define	CUSE_CLI_KNOTE_NEED_READ 0x02
163fa0f6e62SHans Petter Selasky #define	CUSE_CLI_KNOTE_NEED_WRITE 0x04
164fa0f6e62SHans Petter Selasky #define	CUSE_CLI_KNOTE_HAS_READ 0x08
165fa0f6e62SHans Petter Selasky #define	CUSE_CLI_KNOTE_HAS_WRITE 0x10
166fa0f6e62SHans Petter Selasky };
167fa0f6e62SHans Petter Selasky 
168fa0f6e62SHans Petter Selasky #define	CUSE_CLIENT_CLOSING(pcc) \
169fa0f6e62SHans Petter Selasky     ((pcc)->cflags & CUSE_CLI_IS_CLOSING)
170fa0f6e62SHans Petter Selasky 
171fa0f6e62SHans Petter Selasky static	MALLOC_DEFINE(M_CUSE, "cuse", "CUSE memory");
172fa0f6e62SHans Petter Selasky 
173fa0f6e62SHans Petter Selasky static TAILQ_HEAD(, cuse_server) cuse_server_head;
17498029019SHans Petter Selasky static struct mtx cuse_global_mtx;
175fa0f6e62SHans Petter Selasky static struct cdev *cuse_dev;
176fa0f6e62SHans Petter Selasky static struct cuse_server *cuse_alloc_unit[CUSE_DEVICES_MAX];
177fa0f6e62SHans Petter Selasky static int cuse_alloc_unit_id[CUSE_DEVICES_MAX];
178fa0f6e62SHans Petter Selasky 
17995700043SHans Petter Selasky static void cuse_server_wakeup_all_client_locked(struct cuse_server *pcs);
180fa0f6e62SHans Petter Selasky static void cuse_client_kqfilter_read_detach(struct knote *kn);
181fa0f6e62SHans Petter Selasky static void cuse_client_kqfilter_write_detach(struct knote *kn);
182fa0f6e62SHans Petter Selasky static int cuse_client_kqfilter_read_event(struct knote *kn, long hint);
183fa0f6e62SHans Petter Selasky static int cuse_client_kqfilter_write_event(struct knote *kn, long hint);
184fa0f6e62SHans Petter Selasky 
185fa0f6e62SHans Petter Selasky static struct filterops cuse_client_kqfilter_read_ops = {
186fa0f6e62SHans Petter Selasky 	.f_isfd = 1,
187fa0f6e62SHans Petter Selasky 	.f_detach = cuse_client_kqfilter_read_detach,
188fa0f6e62SHans Petter Selasky 	.f_event = cuse_client_kqfilter_read_event,
189fa0f6e62SHans Petter Selasky };
190fa0f6e62SHans Petter Selasky 
191fa0f6e62SHans Petter Selasky static struct filterops cuse_client_kqfilter_write_ops = {
192fa0f6e62SHans Petter Selasky 	.f_isfd = 1,
193fa0f6e62SHans Petter Selasky 	.f_detach = cuse_client_kqfilter_write_detach,
194fa0f6e62SHans Petter Selasky 	.f_event = cuse_client_kqfilter_write_event,
195fa0f6e62SHans Petter Selasky };
196fa0f6e62SHans Petter Selasky 
197fa0f6e62SHans Petter Selasky static d_open_t cuse_client_open;
198fa0f6e62SHans Petter Selasky static d_close_t cuse_client_close;
199fa0f6e62SHans Petter Selasky static d_ioctl_t cuse_client_ioctl;
200fa0f6e62SHans Petter Selasky static d_read_t cuse_client_read;
201fa0f6e62SHans Petter Selasky static d_write_t cuse_client_write;
202fa0f6e62SHans Petter Selasky static d_poll_t cuse_client_poll;
20356c49c63SHans Petter Selasky static d_mmap_single_t cuse_client_mmap_single;
204fa0f6e62SHans Petter Selasky static d_kqfilter_t cuse_client_kqfilter;
205fa0f6e62SHans Petter Selasky 
206fa0f6e62SHans Petter Selasky static struct cdevsw cuse_client_devsw = {
207fa0f6e62SHans Petter Selasky 	.d_version = D_VERSION,
208fa0f6e62SHans Petter Selasky 	.d_open = cuse_client_open,
209fa0f6e62SHans Petter Selasky 	.d_close = cuse_client_close,
210fa0f6e62SHans Petter Selasky 	.d_ioctl = cuse_client_ioctl,
211fa0f6e62SHans Petter Selasky 	.d_name = "cuse_client",
212fa0f6e62SHans Petter Selasky 	.d_flags = D_TRACKCLOSE,
213fa0f6e62SHans Petter Selasky 	.d_read = cuse_client_read,
214fa0f6e62SHans Petter Selasky 	.d_write = cuse_client_write,
215fa0f6e62SHans Petter Selasky 	.d_poll = cuse_client_poll,
21656c49c63SHans Petter Selasky 	.d_mmap_single = cuse_client_mmap_single,
217fa0f6e62SHans Petter Selasky 	.d_kqfilter = cuse_client_kqfilter,
218fa0f6e62SHans Petter Selasky };
219fa0f6e62SHans Petter Selasky 
220fa0f6e62SHans Petter Selasky static d_open_t cuse_server_open;
221fa0f6e62SHans Petter Selasky static d_close_t cuse_server_close;
222fa0f6e62SHans Petter Selasky static d_ioctl_t cuse_server_ioctl;
223fa0f6e62SHans Petter Selasky static d_read_t cuse_server_read;
224fa0f6e62SHans Petter Selasky static d_write_t cuse_server_write;
225fa0f6e62SHans Petter Selasky static d_poll_t cuse_server_poll;
22656c49c63SHans Petter Selasky static d_mmap_single_t cuse_server_mmap_single;
227fa0f6e62SHans Petter Selasky 
228fa0f6e62SHans Petter Selasky static struct cdevsw cuse_server_devsw = {
229fa0f6e62SHans Petter Selasky 	.d_version = D_VERSION,
230fa0f6e62SHans Petter Selasky 	.d_open = cuse_server_open,
231fa0f6e62SHans Petter Selasky 	.d_close = cuse_server_close,
232fa0f6e62SHans Petter Selasky 	.d_ioctl = cuse_server_ioctl,
233fa0f6e62SHans Petter Selasky 	.d_name = "cuse_server",
234fa0f6e62SHans Petter Selasky 	.d_flags = D_TRACKCLOSE,
235fa0f6e62SHans Petter Selasky 	.d_read = cuse_server_read,
236fa0f6e62SHans Petter Selasky 	.d_write = cuse_server_write,
237fa0f6e62SHans Petter Selasky 	.d_poll = cuse_server_poll,
23856c49c63SHans Petter Selasky 	.d_mmap_single = cuse_server_mmap_single,
239fa0f6e62SHans Petter Selasky };
240fa0f6e62SHans Petter Selasky 
241fa0f6e62SHans Petter Selasky static void cuse_client_is_closing(struct cuse_client *);
242fa0f6e62SHans Petter Selasky static int cuse_free_unit_by_id_locked(struct cuse_server *, int);
243fa0f6e62SHans Petter Selasky 
244fa0f6e62SHans Petter Selasky static void
24598029019SHans Petter Selasky cuse_global_lock(void)
246fa0f6e62SHans Petter Selasky {
24798029019SHans Petter Selasky 	mtx_lock(&cuse_global_mtx);
248fa0f6e62SHans Petter Selasky }
249fa0f6e62SHans Petter Selasky 
250fa0f6e62SHans Petter Selasky static void
25198029019SHans Petter Selasky cuse_global_unlock(void)
252fa0f6e62SHans Petter Selasky {
25398029019SHans Petter Selasky 	mtx_unlock(&cuse_global_mtx);
25498029019SHans Petter Selasky }
25598029019SHans Petter Selasky 
25698029019SHans Petter Selasky static void
25798029019SHans Petter Selasky cuse_server_lock(struct cuse_server *pcs)
25898029019SHans Petter Selasky {
25998029019SHans Petter Selasky 	mtx_lock(&pcs->mtx);
26098029019SHans Petter Selasky }
26198029019SHans Petter Selasky 
26298029019SHans Petter Selasky static void
26398029019SHans Petter Selasky cuse_server_unlock(struct cuse_server *pcs)
26498029019SHans Petter Selasky {
26598029019SHans Petter Selasky 	mtx_unlock(&pcs->mtx);
266fa0f6e62SHans Petter Selasky }
267fa0f6e62SHans Petter Selasky 
268fa0f6e62SHans Petter Selasky static void
269fa0f6e62SHans Petter Selasky cuse_cmd_lock(struct cuse_client_command *pccmd)
270fa0f6e62SHans Petter Selasky {
271fa0f6e62SHans Petter Selasky 	sx_xlock(&pccmd->sx);
272fa0f6e62SHans Petter Selasky }
273fa0f6e62SHans Petter Selasky 
274fa0f6e62SHans Petter Selasky static void
275fa0f6e62SHans Petter Selasky cuse_cmd_unlock(struct cuse_client_command *pccmd)
276fa0f6e62SHans Petter Selasky {
277fa0f6e62SHans Petter Selasky 	sx_xunlock(&pccmd->sx);
278fa0f6e62SHans Petter Selasky }
279fa0f6e62SHans Petter Selasky 
280fa0f6e62SHans Petter Selasky static void
281fa0f6e62SHans Petter Selasky cuse_kern_init(void *arg)
282fa0f6e62SHans Petter Selasky {
283fa0f6e62SHans Petter Selasky 	TAILQ_INIT(&cuse_server_head);
284fa0f6e62SHans Petter Selasky 
28598029019SHans Petter Selasky 	mtx_init(&cuse_global_mtx, "cuse-global-mtx", NULL, MTX_DEF);
286fa0f6e62SHans Petter Selasky 
287fa0f6e62SHans Petter Selasky 	cuse_dev = make_dev(&cuse_server_devsw, 0,
288fa0f6e62SHans Petter Selasky 	    UID_ROOT, GID_OPERATOR, 0600, "cuse");
289fa0f6e62SHans Petter Selasky 
290fa0f6e62SHans Petter Selasky 	printf("Cuse v%d.%d.%d @ /dev/cuse\n",
291fa0f6e62SHans Petter Selasky 	    (CUSE_VERSION >> 16) & 0xFF, (CUSE_VERSION >> 8) & 0xFF,
292fa0f6e62SHans Petter Selasky 	    (CUSE_VERSION >> 0) & 0xFF);
293fa0f6e62SHans Petter Selasky }
294891cf3edSEd Maste SYSINIT(cuse_kern_init, SI_SUB_DEVFS, SI_ORDER_ANY, cuse_kern_init, NULL);
295fa0f6e62SHans Petter Selasky 
296fa0f6e62SHans Petter Selasky static void
297fa0f6e62SHans Petter Selasky cuse_kern_uninit(void *arg)
298fa0f6e62SHans Petter Selasky {
299fa0f6e62SHans Petter Selasky 	void *ptr;
300fa0f6e62SHans Petter Selasky 
301fa0f6e62SHans Petter Selasky 	while (1) {
302fa0f6e62SHans Petter Selasky 		printf("Cuse: Please exit all /dev/cuse instances "
303fa0f6e62SHans Petter Selasky 		    "and processes which have used this device.\n");
304fa0f6e62SHans Petter Selasky 
305fa0f6e62SHans Petter Selasky 		pause("DRAIN", 2 * hz);
306fa0f6e62SHans Petter Selasky 
30798029019SHans Petter Selasky 		cuse_global_lock();
308fa0f6e62SHans Petter Selasky 		ptr = TAILQ_FIRST(&cuse_server_head);
30998029019SHans Petter Selasky 		cuse_global_unlock();
310fa0f6e62SHans Petter Selasky 
311fa0f6e62SHans Petter Selasky 		if (ptr == NULL)
312fa0f6e62SHans Petter Selasky 			break;
313fa0f6e62SHans Petter Selasky 	}
314fa0f6e62SHans Petter Selasky 
315fa0f6e62SHans Petter Selasky 	if (cuse_dev != NULL)
316fa0f6e62SHans Petter Selasky 		destroy_dev(cuse_dev);
317fa0f6e62SHans Petter Selasky 
31898029019SHans Petter Selasky 	mtx_destroy(&cuse_global_mtx);
319fa0f6e62SHans Petter Selasky }
320fa0f6e62SHans Petter Selasky SYSUNINIT(cuse_kern_uninit, SI_SUB_DEVFS, SI_ORDER_ANY, cuse_kern_uninit, 0);
321fa0f6e62SHans Petter Selasky 
322fa0f6e62SHans Petter Selasky static int
323fa0f6e62SHans Petter Selasky cuse_server_get(struct cuse_server **ppcs)
324fa0f6e62SHans Petter Selasky {
325fa0f6e62SHans Petter Selasky 	struct cuse_server *pcs;
326fa0f6e62SHans Petter Selasky 	int error;
327fa0f6e62SHans Petter Selasky 
328fa0f6e62SHans Petter Selasky 	error = devfs_get_cdevpriv((void **)&pcs);
329fa0f6e62SHans Petter Selasky 	if (error != 0) {
330fa0f6e62SHans Petter Selasky 		*ppcs = NULL;
331fa0f6e62SHans Petter Selasky 		return (error);
332fa0f6e62SHans Petter Selasky 	}
333fa0f6e62SHans Petter Selasky 	if (pcs->is_closing) {
334fa0f6e62SHans Petter Selasky 		*ppcs = NULL;
335fa0f6e62SHans Petter Selasky 		return (EINVAL);
336fa0f6e62SHans Petter Selasky 	}
337fa0f6e62SHans Petter Selasky 	*ppcs = pcs;
338fa0f6e62SHans Petter Selasky 	return (0);
339fa0f6e62SHans Petter Selasky }
340fa0f6e62SHans Petter Selasky 
341fa0f6e62SHans Petter Selasky static void
342fa0f6e62SHans Petter Selasky cuse_server_is_closing(struct cuse_server *pcs)
343fa0f6e62SHans Petter Selasky {
344fa0f6e62SHans Petter Selasky 	struct cuse_client *pcc;
345fa0f6e62SHans Petter Selasky 
346fa0f6e62SHans Petter Selasky 	if (pcs->is_closing)
347fa0f6e62SHans Petter Selasky 		return;
348fa0f6e62SHans Petter Selasky 
349fa0f6e62SHans Petter Selasky 	pcs->is_closing = 1;
350fa0f6e62SHans Petter Selasky 
351fa0f6e62SHans Petter Selasky 	TAILQ_FOREACH(pcc, &pcs->hcli, entry) {
352fa0f6e62SHans Petter Selasky 		cuse_client_is_closing(pcc);
353fa0f6e62SHans Petter Selasky 	}
354fa0f6e62SHans Petter Selasky }
355fa0f6e62SHans Petter Selasky 
356fa0f6e62SHans Petter Selasky static struct cuse_client_command *
357fa0f6e62SHans Petter Selasky cuse_server_find_command(struct cuse_server *pcs, struct thread *td)
358fa0f6e62SHans Petter Selasky {
359fa0f6e62SHans Petter Selasky 	struct cuse_client *pcc;
360fa0f6e62SHans Petter Selasky 	int n;
361fa0f6e62SHans Petter Selasky 
362fa0f6e62SHans Petter Selasky 	if (pcs->is_closing)
363fa0f6e62SHans Petter Selasky 		goto done;
364fa0f6e62SHans Petter Selasky 
365fa0f6e62SHans Petter Selasky 	TAILQ_FOREACH(pcc, &pcs->hcli, entry) {
366fa0f6e62SHans Petter Selasky 		if (CUSE_CLIENT_CLOSING(pcc))
367fa0f6e62SHans Petter Selasky 			continue;
368fa0f6e62SHans Petter Selasky 		for (n = 0; n != CUSE_CMD_MAX; n++) {
369fa0f6e62SHans Petter Selasky 			if (pcc->cmds[n].entered == td)
370fa0f6e62SHans Petter Selasky 				return (&pcc->cmds[n]);
371fa0f6e62SHans Petter Selasky 		}
372fa0f6e62SHans Petter Selasky 	}
373fa0f6e62SHans Petter Selasky done:
374fa0f6e62SHans Petter Selasky 	return (NULL);
375fa0f6e62SHans Petter Selasky }
376fa0f6e62SHans Petter Selasky 
377fa0f6e62SHans Petter Selasky static void
378fa0f6e62SHans Petter Selasky cuse_str_filter(char *ptr)
379fa0f6e62SHans Petter Selasky {
380fa0f6e62SHans Petter Selasky 	int c;
381fa0f6e62SHans Petter Selasky 
382fa0f6e62SHans Petter Selasky 	while (((c = *ptr) != 0)) {
383fa0f6e62SHans Petter Selasky 		if ((c >= 'a') && (c <= 'z')) {
384fa0f6e62SHans Petter Selasky 			ptr++;
385fa0f6e62SHans Petter Selasky 			continue;
386fa0f6e62SHans Petter Selasky 		}
387fa0f6e62SHans Petter Selasky 		if ((c >= 'A') && (c <= 'Z')) {
388fa0f6e62SHans Petter Selasky 			ptr++;
389fa0f6e62SHans Petter Selasky 			continue;
390fa0f6e62SHans Petter Selasky 		}
391fa0f6e62SHans Petter Selasky 		if ((c >= '0') && (c <= '9')) {
392fa0f6e62SHans Petter Selasky 			ptr++;
393fa0f6e62SHans Petter Selasky 			continue;
394fa0f6e62SHans Petter Selasky 		}
395fa0f6e62SHans Petter Selasky 		if ((c == '.') || (c == '_') || (c == '/')) {
396fa0f6e62SHans Petter Selasky 			ptr++;
397fa0f6e62SHans Petter Selasky 			continue;
398fa0f6e62SHans Petter Selasky 		}
399fa0f6e62SHans Petter Selasky 		*ptr = '_';
400fa0f6e62SHans Petter Selasky 
401fa0f6e62SHans Petter Selasky 		ptr++;
402fa0f6e62SHans Petter Selasky 	}
403fa0f6e62SHans Petter Selasky }
404fa0f6e62SHans Petter Selasky 
405fa0f6e62SHans Petter Selasky static int
406fa0f6e62SHans Petter Selasky cuse_convert_error(int error)
407fa0f6e62SHans Petter Selasky {
408fa0f6e62SHans Petter Selasky 	;				/* indent fix */
409fa0f6e62SHans Petter Selasky 	switch (error) {
410fa0f6e62SHans Petter Selasky 	case CUSE_ERR_NONE:
411fa0f6e62SHans Petter Selasky 		return (0);
412fa0f6e62SHans Petter Selasky 	case CUSE_ERR_BUSY:
413fa0f6e62SHans Petter Selasky 		return (EBUSY);
414fa0f6e62SHans Petter Selasky 	case CUSE_ERR_WOULDBLOCK:
415fa0f6e62SHans Petter Selasky 		return (EWOULDBLOCK);
416fa0f6e62SHans Petter Selasky 	case CUSE_ERR_INVALID:
417fa0f6e62SHans Petter Selasky 		return (EINVAL);
418fa0f6e62SHans Petter Selasky 	case CUSE_ERR_NO_MEMORY:
419fa0f6e62SHans Petter Selasky 		return (ENOMEM);
420fa0f6e62SHans Petter Selasky 	case CUSE_ERR_FAULT:
421fa0f6e62SHans Petter Selasky 		return (EFAULT);
422fa0f6e62SHans Petter Selasky 	case CUSE_ERR_SIGNAL:
423fa0f6e62SHans Petter Selasky 		return (EINTR);
4249f16d9c9SHans Petter Selasky 	case CUSE_ERR_NO_DEVICE:
4259f16d9c9SHans Petter Selasky 		return (ENODEV);
426fa0f6e62SHans Petter Selasky 	default:
427fa0f6e62SHans Petter Selasky 		return (ENXIO);
428fa0f6e62SHans Petter Selasky 	}
429fa0f6e62SHans Petter Selasky }
430fa0f6e62SHans Petter Selasky 
431fa0f6e62SHans Petter Selasky static void
43256c49c63SHans Petter Selasky cuse_vm_memory_free(struct cuse_memory *mem)
433fa0f6e62SHans Petter Selasky {
43456c49c63SHans Petter Selasky 	/* last user is gone - free */
43556c49c63SHans Petter Selasky 	vm_object_deallocate(mem->object);
436fa0f6e62SHans Petter Selasky 
43756c49c63SHans Petter Selasky 	/* free CUSE memory */
43856c49c63SHans Petter Selasky 	free(mem, M_CUSE);
439fa0f6e62SHans Petter Selasky }
440fa0f6e62SHans Petter Selasky 
441fa0f6e62SHans Petter Selasky static int
44256c49c63SHans Petter Selasky cuse_server_alloc_memory(struct cuse_server *pcs, uint32_t alloc_nr,
44356c49c63SHans Petter Selasky     uint32_t page_count)
444fa0f6e62SHans Petter Selasky {
44556c49c63SHans Petter Selasky 	struct cuse_memory *temp;
44656c49c63SHans Petter Selasky 	struct cuse_memory *mem;
44756c49c63SHans Petter Selasky 	vm_object_t object;
448fa0f6e62SHans Petter Selasky 	int error;
449fa0f6e62SHans Petter Selasky 
45056c49c63SHans Petter Selasky 	mem = malloc(sizeof(*mem), M_CUSE, M_WAITOK | M_ZERO);
451fa0f6e62SHans Petter Selasky 
45256c49c63SHans Petter Selasky 	object = vm_pager_allocate(OBJT_SWAP, NULL, PAGE_SIZE * page_count,
45356c49c63SHans Petter Selasky 	    VM_PROT_DEFAULT, 0, curthread->td_ucred);
45456c49c63SHans Petter Selasky 	if (object == NULL) {
455fa0f6e62SHans Petter Selasky 		error = ENOMEM;
45656c49c63SHans Petter Selasky 		goto error_0;
45756c49c63SHans Petter Selasky 	}
458fa0f6e62SHans Petter Selasky 
45998029019SHans Petter Selasky 	cuse_server_lock(pcs);
46056c49c63SHans Petter Selasky 	/* check if allocation number already exists */
46156c49c63SHans Petter Selasky 	TAILQ_FOREACH(temp, &pcs->hmem, entry) {
46256c49c63SHans Petter Selasky 		if (temp->alloc_nr == alloc_nr)
46356c49c63SHans Petter Selasky 			break;
46456c49c63SHans Petter Selasky 	}
46556c49c63SHans Petter Selasky 	if (temp != NULL) {
46698029019SHans Petter Selasky 		cuse_server_unlock(pcs);
46756c49c63SHans Petter Selasky 		error = EBUSY;
46856c49c63SHans Petter Selasky 		goto error_1;
46956c49c63SHans Petter Selasky 	}
47056c49c63SHans Petter Selasky 	mem->object = object;
47156c49c63SHans Petter Selasky 	mem->page_count = page_count;
47256c49c63SHans Petter Selasky 	mem->alloc_nr = alloc_nr;
47356c49c63SHans Petter Selasky 	TAILQ_INSERT_TAIL(&pcs->hmem, mem, entry);
47498029019SHans Petter Selasky 	cuse_server_unlock(pcs);
47556c49c63SHans Petter Selasky 
47656c49c63SHans Petter Selasky 	return (0);
47756c49c63SHans Petter Selasky 
47856c49c63SHans Petter Selasky error_1:
47956c49c63SHans Petter Selasky 	vm_object_deallocate(object);
48056c49c63SHans Petter Selasky error_0:
48156c49c63SHans Petter Selasky 	free(mem, M_CUSE);
482fa0f6e62SHans Petter Selasky 	return (error);
483fa0f6e62SHans Petter Selasky }
48456c49c63SHans Petter Selasky 
48556c49c63SHans Petter Selasky static int
48656c49c63SHans Petter Selasky cuse_server_free_memory(struct cuse_server *pcs, uint32_t alloc_nr)
48756c49c63SHans Petter Selasky {
48856c49c63SHans Petter Selasky 	struct cuse_memory *mem;
48956c49c63SHans Petter Selasky 
49098029019SHans Petter Selasky 	cuse_server_lock(pcs);
49156c49c63SHans Petter Selasky 	TAILQ_FOREACH(mem, &pcs->hmem, entry) {
49256c49c63SHans Petter Selasky 		if (mem->alloc_nr == alloc_nr)
49356c49c63SHans Petter Selasky 			break;
49456c49c63SHans Petter Selasky 	}
49556c49c63SHans Petter Selasky 	if (mem == NULL) {
49698029019SHans Petter Selasky 		cuse_server_unlock(pcs);
49756c49c63SHans Petter Selasky 		return (EINVAL);
49856c49c63SHans Petter Selasky 	}
49956c49c63SHans Petter Selasky 	TAILQ_REMOVE(&pcs->hmem, mem, entry);
50098029019SHans Petter Selasky 	cuse_server_unlock(pcs);
50156c49c63SHans Petter Selasky 
50256c49c63SHans Petter Selasky 	cuse_vm_memory_free(mem);
503fa0f6e62SHans Petter Selasky 
504fa0f6e62SHans Petter Selasky 	return (0);
505fa0f6e62SHans Petter Selasky }
506fa0f6e62SHans Petter Selasky 
507fa0f6e62SHans Petter Selasky static int
508fa0f6e62SHans Petter Selasky cuse_client_get(struct cuse_client **ppcc)
509fa0f6e62SHans Petter Selasky {
510fa0f6e62SHans Petter Selasky 	struct cuse_client *pcc;
511fa0f6e62SHans Petter Selasky 	int error;
512fa0f6e62SHans Petter Selasky 
513fa0f6e62SHans Petter Selasky 	/* try to get private data */
514fa0f6e62SHans Petter Selasky 	error = devfs_get_cdevpriv((void **)&pcc);
515fa0f6e62SHans Petter Selasky 	if (error != 0) {
516fa0f6e62SHans Petter Selasky 		*ppcc = NULL;
517fa0f6e62SHans Petter Selasky 		return (error);
518fa0f6e62SHans Petter Selasky 	}
519fa0f6e62SHans Petter Selasky 	if (CUSE_CLIENT_CLOSING(pcc) || pcc->server->is_closing) {
520fa0f6e62SHans Petter Selasky 		*ppcc = NULL;
521fa0f6e62SHans Petter Selasky 		return (EINVAL);
522fa0f6e62SHans Petter Selasky 	}
523fa0f6e62SHans Petter Selasky 	*ppcc = pcc;
524fa0f6e62SHans Petter Selasky 	return (0);
525fa0f6e62SHans Petter Selasky }
526fa0f6e62SHans Petter Selasky 
527fa0f6e62SHans Petter Selasky static void
528fa0f6e62SHans Petter Selasky cuse_client_is_closing(struct cuse_client *pcc)
529fa0f6e62SHans Petter Selasky {
530fa0f6e62SHans Petter Selasky 	struct cuse_client_command *pccmd;
531fa0f6e62SHans Petter Selasky 	uint32_t n;
532fa0f6e62SHans Petter Selasky 
533fa0f6e62SHans Petter Selasky 	if (CUSE_CLIENT_CLOSING(pcc))
534fa0f6e62SHans Petter Selasky 		return;
535fa0f6e62SHans Petter Selasky 
536fa0f6e62SHans Petter Selasky 	pcc->cflags |= CUSE_CLI_IS_CLOSING;
537fa0f6e62SHans Petter Selasky 	pcc->server_dev = NULL;
538fa0f6e62SHans Petter Selasky 
539fa0f6e62SHans Petter Selasky 	for (n = 0; n != CUSE_CMD_MAX; n++) {
540fa0f6e62SHans Petter Selasky 		pccmd = &pcc->cmds[n];
541fa0f6e62SHans Petter Selasky 
542fa0f6e62SHans Petter Selasky 		if (pccmd->entry.tqe_prev != NULL) {
543fa0f6e62SHans Petter Selasky 			TAILQ_REMOVE(&pcc->server->head, pccmd, entry);
544fa0f6e62SHans Petter Selasky 			pccmd->entry.tqe_prev = NULL;
545fa0f6e62SHans Petter Selasky 		}
546fa0f6e62SHans Petter Selasky 		cv_broadcast(&pccmd->cv);
547fa0f6e62SHans Petter Selasky 	}
548fa0f6e62SHans Petter Selasky }
549fa0f6e62SHans Petter Selasky 
550fa0f6e62SHans Petter Selasky static void
551fa0f6e62SHans Petter Selasky cuse_client_send_command_locked(struct cuse_client_command *pccmd,
552beebd9aaSHans Petter Selasky     uintptr_t data_ptr, unsigned long arg, int fflags, int ioflag)
553fa0f6e62SHans Petter Selasky {
554fa0f6e62SHans Petter Selasky 	unsigned long cuse_fflags = 0;
555fa0f6e62SHans Petter Selasky 	struct cuse_server *pcs;
556fa0f6e62SHans Petter Selasky 
557fa0f6e62SHans Petter Selasky 	if (fflags & FREAD)
558fa0f6e62SHans Petter Selasky 		cuse_fflags |= CUSE_FFLAG_READ;
559fa0f6e62SHans Petter Selasky 
560fa0f6e62SHans Petter Selasky 	if (fflags & FWRITE)
561fa0f6e62SHans Petter Selasky 		cuse_fflags |= CUSE_FFLAG_WRITE;
562fa0f6e62SHans Petter Selasky 
563fa0f6e62SHans Petter Selasky 	if (ioflag & IO_NDELAY)
564fa0f6e62SHans Petter Selasky 		cuse_fflags |= CUSE_FFLAG_NONBLOCK;
565db92a6cdSHans Petter Selasky #if defined(__LP64__)
566db92a6cdSHans Petter Selasky 	if (SV_CURPROC_FLAG(SV_ILP32))
567db92a6cdSHans Petter Selasky 		cuse_fflags |= CUSE_FFLAG_COMPAT32;
568db92a6cdSHans Petter Selasky #endif
569fa0f6e62SHans Petter Selasky 	pccmd->sub.fflags = cuse_fflags;
570fa0f6e62SHans Petter Selasky 	pccmd->sub.data_pointer = data_ptr;
571fa0f6e62SHans Petter Selasky 	pccmd->sub.argument = arg;
572fa0f6e62SHans Petter Selasky 
573fa0f6e62SHans Petter Selasky 	pcs = pccmd->client->server;
574fa0f6e62SHans Petter Selasky 
575fa0f6e62SHans Petter Selasky 	if ((pccmd->entry.tqe_prev == NULL) &&
576fa0f6e62SHans Petter Selasky 	    (CUSE_CLIENT_CLOSING(pccmd->client) == 0) &&
577fa0f6e62SHans Petter Selasky 	    (pcs->is_closing == 0)) {
578fa0f6e62SHans Petter Selasky 		TAILQ_INSERT_TAIL(&pcs->head, pccmd, entry);
579fa0f6e62SHans Petter Selasky 		cv_signal(&pcs->cv);
580fa0f6e62SHans Petter Selasky 	}
581fa0f6e62SHans Petter Selasky }
582fa0f6e62SHans Petter Selasky 
583fa0f6e62SHans Petter Selasky static void
584fa0f6e62SHans Petter Selasky cuse_client_got_signal(struct cuse_client_command *pccmd)
585fa0f6e62SHans Petter Selasky {
586fa0f6e62SHans Petter Selasky 	struct cuse_server *pcs;
587fa0f6e62SHans Petter Selasky 
588fa0f6e62SHans Petter Selasky 	pccmd->got_signal = 1;
589fa0f6e62SHans Petter Selasky 
590fa0f6e62SHans Petter Selasky 	pccmd = &pccmd->client->cmds[CUSE_CMD_SIGNAL];
591fa0f6e62SHans Petter Selasky 
592fa0f6e62SHans Petter Selasky 	pcs = pccmd->client->server;
593fa0f6e62SHans Petter Selasky 
594fa0f6e62SHans Petter Selasky 	if ((pccmd->entry.tqe_prev == NULL) &&
595fa0f6e62SHans Petter Selasky 	    (CUSE_CLIENT_CLOSING(pccmd->client) == 0) &&
596fa0f6e62SHans Petter Selasky 	    (pcs->is_closing == 0)) {
597fa0f6e62SHans Petter Selasky 		TAILQ_INSERT_TAIL(&pcs->head, pccmd, entry);
598fa0f6e62SHans Petter Selasky 		cv_signal(&pcs->cv);
599fa0f6e62SHans Petter Selasky 	}
600fa0f6e62SHans Petter Selasky }
601fa0f6e62SHans Petter Selasky 
602fa0f6e62SHans Petter Selasky static int
603fa0f6e62SHans Petter Selasky cuse_client_receive_command_locked(struct cuse_client_command *pccmd,
604fa0f6e62SHans Petter Selasky     uint8_t *arg_ptr, uint32_t arg_len)
605fa0f6e62SHans Petter Selasky {
60698029019SHans Petter Selasky 	struct cuse_server *pcs;
607fa0f6e62SHans Petter Selasky 	int error;
608fa0f6e62SHans Petter Selasky 
60998029019SHans Petter Selasky 	pcs = pccmd->client->server;
610fa0f6e62SHans Petter Selasky 	error = 0;
611fa0f6e62SHans Petter Selasky 
612fa0f6e62SHans Petter Selasky 	pccmd->proc_curr = curthread->td_proc;
613fa0f6e62SHans Petter Selasky 
61498029019SHans Petter Selasky 	if (CUSE_CLIENT_CLOSING(pccmd->client) || pcs->is_closing) {
615fa0f6e62SHans Petter Selasky 		error = CUSE_ERR_OTHER;
616fa0f6e62SHans Petter Selasky 		goto done;
617fa0f6e62SHans Petter Selasky 	}
618fa0f6e62SHans Petter Selasky 	while (pccmd->command == CUSE_CMD_NONE) {
619fa0f6e62SHans Petter Selasky 		if (error != 0) {
62098029019SHans Petter Selasky 			cv_wait(&pccmd->cv, &pcs->mtx);
621fa0f6e62SHans Petter Selasky 		} else {
62298029019SHans Petter Selasky 			error = cv_wait_sig(&pccmd->cv, &pcs->mtx);
623fa0f6e62SHans Petter Selasky 
624fa0f6e62SHans Petter Selasky 			if (error != 0)
625fa0f6e62SHans Petter Selasky 				cuse_client_got_signal(pccmd);
626fa0f6e62SHans Petter Selasky 		}
62798029019SHans Petter Selasky 		if (CUSE_CLIENT_CLOSING(pccmd->client) || pcs->is_closing) {
628fa0f6e62SHans Petter Selasky 			error = CUSE_ERR_OTHER;
629fa0f6e62SHans Petter Selasky 			goto done;
630fa0f6e62SHans Petter Selasky 		}
631fa0f6e62SHans Petter Selasky 	}
632fa0f6e62SHans Petter Selasky 
633fa0f6e62SHans Petter Selasky 	error = pccmd->error;
634fa0f6e62SHans Petter Selasky 	pccmd->command = CUSE_CMD_NONE;
635fa0f6e62SHans Petter Selasky 	cv_signal(&pccmd->cv);
636fa0f6e62SHans Petter Selasky 
637fa0f6e62SHans Petter Selasky done:
638fa0f6e62SHans Petter Selasky 
639fa0f6e62SHans Petter Selasky 	/* wait until all process references are gone */
640fa0f6e62SHans Petter Selasky 
641fa0f6e62SHans Petter Selasky 	pccmd->proc_curr = NULL;
642fa0f6e62SHans Petter Selasky 
643fa0f6e62SHans Petter Selasky 	while (pccmd->proc_refs != 0)
64498029019SHans Petter Selasky 		cv_wait(&pccmd->cv, &pcs->mtx);
645fa0f6e62SHans Petter Selasky 
646fa0f6e62SHans Petter Selasky 	return (error);
647fa0f6e62SHans Petter Selasky }
648fa0f6e62SHans Petter Selasky 
649fa0f6e62SHans Petter Selasky /*------------------------------------------------------------------------*
650fa0f6e62SHans Petter Selasky  *	CUSE SERVER PART
651fa0f6e62SHans Petter Selasky  *------------------------------------------------------------------------*/
652fa0f6e62SHans Petter Selasky 
653fa0f6e62SHans Petter Selasky static void
654fa0f6e62SHans Petter Selasky cuse_server_free_dev(struct cuse_server_dev *pcsd)
655fa0f6e62SHans Petter Selasky {
656fa0f6e62SHans Petter Selasky 	struct cuse_server *pcs;
657fa0f6e62SHans Petter Selasky 	struct cuse_client *pcc;
658fa0f6e62SHans Petter Selasky 
659fa0f6e62SHans Petter Selasky 	/* get server pointer */
660fa0f6e62SHans Petter Selasky 	pcs = pcsd->server;
661fa0f6e62SHans Petter Selasky 
662fa0f6e62SHans Petter Selasky 	/* prevent creation of more devices */
66398029019SHans Petter Selasky 	cuse_server_lock(pcs);
664fa0f6e62SHans Petter Selasky 	if (pcsd->kern_dev != NULL)
665fa0f6e62SHans Petter Selasky 		pcsd->kern_dev->si_drv1 = NULL;
666fa0f6e62SHans Petter Selasky 
667fa0f6e62SHans Petter Selasky 	TAILQ_FOREACH(pcc, &pcs->hcli, entry) {
668fa0f6e62SHans Petter Selasky 		if (pcc->server_dev == pcsd)
669fa0f6e62SHans Petter Selasky 			cuse_client_is_closing(pcc);
670fa0f6e62SHans Petter Selasky 	}
67198029019SHans Petter Selasky 	cuse_server_unlock(pcs);
672fa0f6e62SHans Petter Selasky 
673fa0f6e62SHans Petter Selasky 	/* destroy device, if any */
674fa0f6e62SHans Petter Selasky 	if (pcsd->kern_dev != NULL) {
675fa0f6e62SHans Petter Selasky 		/* destroy device synchronously */
676fa0f6e62SHans Petter Selasky 		destroy_dev(pcsd->kern_dev);
677fa0f6e62SHans Petter Selasky 	}
678fa0f6e62SHans Petter Selasky 	free(pcsd, M_CUSE);
679fa0f6e62SHans Petter Selasky }
680fa0f6e62SHans Petter Selasky 
681fa0f6e62SHans Petter Selasky static void
68256c49c63SHans Petter Selasky cuse_server_unref(struct cuse_server *pcs)
683fa0f6e62SHans Petter Selasky {
684fa0f6e62SHans Petter Selasky 	struct cuse_server_dev *pcsd;
68556c49c63SHans Petter Selasky 	struct cuse_memory *mem;
686fa0f6e62SHans Petter Selasky 
68798029019SHans Petter Selasky 	cuse_server_lock(pcs);
68898029019SHans Petter Selasky 	if (--(pcs->refs) != 0) {
68998029019SHans Petter Selasky 		cuse_server_unlock(pcs);
690fa0f6e62SHans Petter Selasky 		return;
691fa0f6e62SHans Petter Selasky 	}
692fa0f6e62SHans Petter Selasky 	cuse_server_is_closing(pcs);
69395700043SHans Petter Selasky 	/* final client wakeup, if any */
69495700043SHans Petter Selasky 	cuse_server_wakeup_all_client_locked(pcs);
695fa0f6e62SHans Petter Selasky 
69698029019SHans Petter Selasky 	cuse_global_lock();
697fa0f6e62SHans Petter Selasky 	TAILQ_REMOVE(&cuse_server_head, pcs, entry);
69898029019SHans Petter Selasky 	cuse_global_unlock();
699fa0f6e62SHans Petter Selasky 
700fa0f6e62SHans Petter Selasky 	while ((pcsd = TAILQ_FIRST(&pcs->hdev)) != NULL) {
701fa0f6e62SHans Petter Selasky 		TAILQ_REMOVE(&pcs->hdev, pcsd, entry);
70298029019SHans Petter Selasky 		cuse_server_unlock(pcs);
703fa0f6e62SHans Petter Selasky 		cuse_server_free_dev(pcsd);
70498029019SHans Petter Selasky 		cuse_server_lock(pcs);
705fa0f6e62SHans Petter Selasky 	}
706fa0f6e62SHans Petter Selasky 
70743a9329eSHans Petter Selasky 	cuse_free_unit_by_id_locked(pcs, -1);
70843a9329eSHans Petter Selasky 
70956c49c63SHans Petter Selasky 	while ((mem = TAILQ_FIRST(&pcs->hmem)) != NULL) {
71056c49c63SHans Petter Selasky 		TAILQ_REMOVE(&pcs->hmem, mem, entry);
71198029019SHans Petter Selasky 		cuse_server_unlock(pcs);
71256c49c63SHans Petter Selasky 		cuse_vm_memory_free(mem);
71398029019SHans Petter Selasky 		cuse_server_lock(pcs);
71456c49c63SHans Petter Selasky 	}
715fa0f6e62SHans Petter Selasky 
716fa0f6e62SHans Petter Selasky 	knlist_clear(&pcs->selinfo.si_note, 1);
717fa0f6e62SHans Petter Selasky 	knlist_destroy(&pcs->selinfo.si_note);
718fa0f6e62SHans Petter Selasky 
71998029019SHans Petter Selasky 	cuse_server_unlock(pcs);
720fa0f6e62SHans Petter Selasky 
721fa0f6e62SHans Petter Selasky 	seldrain(&pcs->selinfo);
722fa0f6e62SHans Petter Selasky 
723fa0f6e62SHans Petter Selasky 	cv_destroy(&pcs->cv);
724fa0f6e62SHans Petter Selasky 
72598029019SHans Petter Selasky 	mtx_destroy(&pcs->mtx);
72698029019SHans Petter Selasky 
727fa0f6e62SHans Petter Selasky 	free(pcs, M_CUSE);
728fa0f6e62SHans Petter Selasky }
729fa0f6e62SHans Petter Selasky 
730c7ffaed9SHans Petter Selasky static int
731c7ffaed9SHans Petter Selasky cuse_server_do_close(struct cuse_server *pcs)
732c7ffaed9SHans Petter Selasky {
733c7ffaed9SHans Petter Selasky 	int retval;
734c7ffaed9SHans Petter Selasky 
73598029019SHans Petter Selasky 	cuse_server_lock(pcs);
736c7ffaed9SHans Petter Selasky 	cuse_server_is_closing(pcs);
737c7ffaed9SHans Petter Selasky 	/* final client wakeup, if any */
738c7ffaed9SHans Petter Selasky 	cuse_server_wakeup_all_client_locked(pcs);
739c7ffaed9SHans Petter Selasky 
740c7ffaed9SHans Petter Selasky 	knlist_clear(&pcs->selinfo.si_note, 1);
741c7ffaed9SHans Petter Selasky 
742c7ffaed9SHans Petter Selasky 	retval = pcs->refs;
74398029019SHans Petter Selasky 	cuse_server_unlock(pcs);
744c7ffaed9SHans Petter Selasky 
745c7ffaed9SHans Petter Selasky 	return (retval);
746c7ffaed9SHans Petter Selasky }
747c7ffaed9SHans Petter Selasky 
74856c49c63SHans Petter Selasky static void
74956c49c63SHans Petter Selasky cuse_server_free(void *arg)
75056c49c63SHans Petter Selasky {
75156c49c63SHans Petter Selasky 	struct cuse_server *pcs = arg;
75256c49c63SHans Petter Selasky 
753c7ffaed9SHans Petter Selasky 	/*
754c7ffaed9SHans Petter Selasky 	 * The final server unref should be done by the server thread
755c7ffaed9SHans Petter Selasky 	 * to prevent deadlock in the client cdevpriv destructor,
756c7ffaed9SHans Petter Selasky 	 * which cannot destroy itself.
757c7ffaed9SHans Petter Selasky 	 */
758c7ffaed9SHans Petter Selasky 	while (cuse_server_do_close(pcs) != 1)
759c7ffaed9SHans Petter Selasky 		pause("W", hz);
760c7ffaed9SHans Petter Selasky 
761c7ffaed9SHans Petter Selasky 	/* drop final refcount */
76256c49c63SHans Petter Selasky 	cuse_server_unref(pcs);
76356c49c63SHans Petter Selasky }
76456c49c63SHans Petter Selasky 
765fa0f6e62SHans Petter Selasky static int
766fa0f6e62SHans Petter Selasky cuse_server_open(struct cdev *dev, int fflags, int devtype, struct thread *td)
767fa0f6e62SHans Petter Selasky {
768fa0f6e62SHans Petter Selasky 	struct cuse_server *pcs;
769fa0f6e62SHans Petter Selasky 
770fa0f6e62SHans Petter Selasky 	pcs = malloc(sizeof(*pcs), M_CUSE, M_WAITOK | M_ZERO);
771fa0f6e62SHans Petter Selasky 
772fa0f6e62SHans Petter Selasky 	if (devfs_set_cdevpriv(pcs, &cuse_server_free)) {
773fa0f6e62SHans Petter Selasky 		printf("Cuse: Cannot set cdevpriv.\n");
774fa0f6e62SHans Petter Selasky 		free(pcs, M_CUSE);
775fa0f6e62SHans Petter Selasky 		return (ENOMEM);
776fa0f6e62SHans Petter Selasky 	}
777ac60e801SHans Petter Selasky 	/* store current process ID */
778ac60e801SHans Petter Selasky 	pcs->pid = curproc->p_pid;
779ac60e801SHans Petter Selasky 
780fa0f6e62SHans Petter Selasky 	TAILQ_INIT(&pcs->head);
781fa0f6e62SHans Petter Selasky 	TAILQ_INIT(&pcs->hdev);
782fa0f6e62SHans Petter Selasky 	TAILQ_INIT(&pcs->hcli);
78356c49c63SHans Petter Selasky 	TAILQ_INIT(&pcs->hmem);
784fa0f6e62SHans Petter Selasky 
785fa0f6e62SHans Petter Selasky 	cv_init(&pcs->cv, "cuse-server-cv");
786fa0f6e62SHans Petter Selasky 
78798029019SHans Petter Selasky 	mtx_init(&pcs->mtx, "cuse-server-mtx", NULL, MTX_DEF);
788fa0f6e62SHans Petter Selasky 
78998029019SHans Petter Selasky 	knlist_init_mtx(&pcs->selinfo.si_note, &pcs->mtx);
79098029019SHans Petter Selasky 
79198029019SHans Petter Selasky 	cuse_global_lock();
792fa0f6e62SHans Petter Selasky 	pcs->refs++;
793fa0f6e62SHans Petter Selasky 	TAILQ_INSERT_TAIL(&cuse_server_head, pcs, entry);
79498029019SHans Petter Selasky 	cuse_global_unlock();
795fa0f6e62SHans Petter Selasky 
796fa0f6e62SHans Petter Selasky 	return (0);
797fa0f6e62SHans Petter Selasky }
798fa0f6e62SHans Petter Selasky 
799fa0f6e62SHans Petter Selasky static int
800fa0f6e62SHans Petter Selasky cuse_server_close(struct cdev *dev, int fflag, int devtype, struct thread *td)
801fa0f6e62SHans Petter Selasky {
802fa0f6e62SHans Petter Selasky 	struct cuse_server *pcs;
803fa0f6e62SHans Petter Selasky 
804c7ffaed9SHans Petter Selasky 	if (cuse_server_get(&pcs) == 0)
805c7ffaed9SHans Petter Selasky 		cuse_server_do_close(pcs);
806fa0f6e62SHans Petter Selasky 
807fa0f6e62SHans Petter Selasky 	return (0);
808fa0f6e62SHans Petter Selasky }
809fa0f6e62SHans Petter Selasky 
810fa0f6e62SHans Petter Selasky static int
811fa0f6e62SHans Petter Selasky cuse_server_read(struct cdev *dev, struct uio *uio, int ioflag)
812fa0f6e62SHans Petter Selasky {
813fa0f6e62SHans Petter Selasky 	return (ENXIO);
814fa0f6e62SHans Petter Selasky }
815fa0f6e62SHans Petter Selasky 
816fa0f6e62SHans Petter Selasky static int
817fa0f6e62SHans Petter Selasky cuse_server_write(struct cdev *dev, struct uio *uio, int ioflag)
818fa0f6e62SHans Petter Selasky {
819fa0f6e62SHans Petter Selasky 	return (ENXIO);
820fa0f6e62SHans Petter Selasky }
821fa0f6e62SHans Petter Selasky 
822fa0f6e62SHans Petter Selasky static int
82398029019SHans Petter Selasky cuse_server_ioctl_copy_locked(struct cuse_server *pcs,
82498029019SHans Petter Selasky     struct cuse_client_command *pccmd,
825fa0f6e62SHans Petter Selasky     struct cuse_data_chunk *pchk, int isread)
826fa0f6e62SHans Petter Selasky {
827fa0f6e62SHans Petter Selasky 	struct proc *p_proc;
828fa0f6e62SHans Petter Selasky 	uint32_t offset;
829fa0f6e62SHans Petter Selasky 	int error;
830fa0f6e62SHans Petter Selasky 
831fa0f6e62SHans Petter Selasky 	offset = pchk->peer_ptr - CUSE_BUF_MIN_PTR;
832fa0f6e62SHans Petter Selasky 
833fa0f6e62SHans Petter Selasky 	if (pchk->length > CUSE_BUFFER_MAX)
834fa0f6e62SHans Petter Selasky 		return (EFAULT);
835fa0f6e62SHans Petter Selasky 
836fa0f6e62SHans Petter Selasky 	if (offset >= CUSE_BUFFER_MAX)
837fa0f6e62SHans Petter Selasky 		return (EFAULT);
838fa0f6e62SHans Petter Selasky 
839fa0f6e62SHans Petter Selasky 	if ((offset + pchk->length) > CUSE_BUFFER_MAX)
840fa0f6e62SHans Petter Selasky 		return (EFAULT);
841fa0f6e62SHans Petter Selasky 
842fa0f6e62SHans Petter Selasky 	p_proc = pccmd->proc_curr;
843fa0f6e62SHans Petter Selasky 	if (p_proc == NULL)
844fa0f6e62SHans Petter Selasky 		return (ENXIO);
845fa0f6e62SHans Petter Selasky 
846fa0f6e62SHans Petter Selasky 	if (pccmd->proc_refs < 0)
847fa0f6e62SHans Petter Selasky 		return (ENOMEM);
848fa0f6e62SHans Petter Selasky 
849fa0f6e62SHans Petter Selasky 	pccmd->proc_refs++;
850fa0f6e62SHans Petter Selasky 
85198029019SHans Petter Selasky 	cuse_server_unlock(pcs);
852fa0f6e62SHans Petter Selasky 
853fa0f6e62SHans Petter Selasky 	if (isread == 0) {
854fa0f6e62SHans Petter Selasky 		error = copyin(
855fa0f6e62SHans Petter Selasky 		    (void *)pchk->local_ptr,
856fa0f6e62SHans Petter Selasky 		    pccmd->client->ioctl_buffer + offset,
857fa0f6e62SHans Petter Selasky 		    pchk->length);
858fa0f6e62SHans Petter Selasky 	} else {
859fa0f6e62SHans Petter Selasky 		error = copyout(
860fa0f6e62SHans Petter Selasky 		    pccmd->client->ioctl_buffer + offset,
861fa0f6e62SHans Petter Selasky 		    (void *)pchk->local_ptr,
862fa0f6e62SHans Petter Selasky 		    pchk->length);
863fa0f6e62SHans Petter Selasky 	}
864fa0f6e62SHans Petter Selasky 
86598029019SHans Petter Selasky 	cuse_server_lock(pcs);
866fa0f6e62SHans Petter Selasky 
867fa0f6e62SHans Petter Selasky 	pccmd->proc_refs--;
868fa0f6e62SHans Petter Selasky 
869fa0f6e62SHans Petter Selasky 	if (pccmd->proc_curr == NULL)
870fa0f6e62SHans Petter Selasky 		cv_signal(&pccmd->cv);
871fa0f6e62SHans Petter Selasky 
872fa0f6e62SHans Petter Selasky 	return (error);
873fa0f6e62SHans Petter Selasky }
874fa0f6e62SHans Petter Selasky 
875fa0f6e62SHans Petter Selasky static int
876fa0f6e62SHans Petter Selasky cuse_proc2proc_copy(struct proc *proc_s, vm_offset_t data_s,
877fa0f6e62SHans Petter Selasky     struct proc *proc_d, vm_offset_t data_d, size_t len)
878fa0f6e62SHans Petter Selasky {
879fa0f6e62SHans Petter Selasky 	struct thread *td;
880fa0f6e62SHans Petter Selasky 	struct proc *proc_cur;
881fa0f6e62SHans Petter Selasky 	int error;
882fa0f6e62SHans Petter Selasky 
883fa0f6e62SHans Petter Selasky 	td = curthread;
884fa0f6e62SHans Petter Selasky 	proc_cur = td->td_proc;
885fa0f6e62SHans Petter Selasky 
886fa0f6e62SHans Petter Selasky 	if (proc_cur == proc_d) {
887fa0f6e62SHans Petter Selasky 		struct iovec iov = {
888fa0f6e62SHans Petter Selasky 			.iov_base = (caddr_t)data_d,
889fa0f6e62SHans Petter Selasky 			.iov_len = len,
890fa0f6e62SHans Petter Selasky 		};
891fa0f6e62SHans Petter Selasky 		struct uio uio = {
892fa0f6e62SHans Petter Selasky 			.uio_iov = &iov,
893fa0f6e62SHans Petter Selasky 			.uio_iovcnt = 1,
894fa0f6e62SHans Petter Selasky 			.uio_offset = (off_t)data_s,
895fa0f6e62SHans Petter Selasky 			.uio_resid = len,
896fa0f6e62SHans Petter Selasky 			.uio_segflg = UIO_USERSPACE,
897fa0f6e62SHans Petter Selasky 			.uio_rw = UIO_READ,
898fa0f6e62SHans Petter Selasky 			.uio_td = td,
899fa0f6e62SHans Petter Selasky 		};
900fa0f6e62SHans Petter Selasky 
9015928e3b9SHans Petter Selasky 		PHOLD(proc_s);
902fa0f6e62SHans Petter Selasky 		error = proc_rwmem(proc_s, &uio);
9035928e3b9SHans Petter Selasky 		PRELE(proc_s);
904fa0f6e62SHans Petter Selasky 
905fa0f6e62SHans Petter Selasky 	} else if (proc_cur == proc_s) {
906fa0f6e62SHans Petter Selasky 		struct iovec iov = {
907fa0f6e62SHans Petter Selasky 			.iov_base = (caddr_t)data_s,
908fa0f6e62SHans Petter Selasky 			.iov_len = len,
909fa0f6e62SHans Petter Selasky 		};
910fa0f6e62SHans Petter Selasky 		struct uio uio = {
911fa0f6e62SHans Petter Selasky 			.uio_iov = &iov,
912fa0f6e62SHans Petter Selasky 			.uio_iovcnt = 1,
913fa0f6e62SHans Petter Selasky 			.uio_offset = (off_t)data_d,
914fa0f6e62SHans Petter Selasky 			.uio_resid = len,
915fa0f6e62SHans Petter Selasky 			.uio_segflg = UIO_USERSPACE,
916fa0f6e62SHans Petter Selasky 			.uio_rw = UIO_WRITE,
917fa0f6e62SHans Petter Selasky 			.uio_td = td,
918fa0f6e62SHans Petter Selasky 		};
919fa0f6e62SHans Petter Selasky 
9205928e3b9SHans Petter Selasky 		PHOLD(proc_d);
921fa0f6e62SHans Petter Selasky 		error = proc_rwmem(proc_d, &uio);
9225928e3b9SHans Petter Selasky 		PRELE(proc_d);
923fa0f6e62SHans Petter Selasky 	} else {
924fa0f6e62SHans Petter Selasky 		error = EINVAL;
925fa0f6e62SHans Petter Selasky 	}
926fa0f6e62SHans Petter Selasky 	return (error);
927fa0f6e62SHans Petter Selasky }
928fa0f6e62SHans Petter Selasky 
929fa0f6e62SHans Petter Selasky static int
93098029019SHans Petter Selasky cuse_server_data_copy_locked(struct cuse_server *pcs,
93198029019SHans Petter Selasky     struct cuse_client_command *pccmd,
932fa0f6e62SHans Petter Selasky     struct cuse_data_chunk *pchk, int isread)
933fa0f6e62SHans Petter Selasky {
934fa0f6e62SHans Petter Selasky 	struct proc *p_proc;
935fa0f6e62SHans Petter Selasky 	int error;
936fa0f6e62SHans Petter Selasky 
937fa0f6e62SHans Petter Selasky 	p_proc = pccmd->proc_curr;
938fa0f6e62SHans Petter Selasky 	if (p_proc == NULL)
939fa0f6e62SHans Petter Selasky 		return (ENXIO);
940fa0f6e62SHans Petter Selasky 
941fa0f6e62SHans Petter Selasky 	if (pccmd->proc_refs < 0)
942fa0f6e62SHans Petter Selasky 		return (ENOMEM);
943fa0f6e62SHans Petter Selasky 
944fa0f6e62SHans Petter Selasky 	pccmd->proc_refs++;
945fa0f6e62SHans Petter Selasky 
94698029019SHans Petter Selasky 	cuse_server_unlock(pcs);
947fa0f6e62SHans Petter Selasky 
948fa0f6e62SHans Petter Selasky 	if (isread == 0) {
949fa0f6e62SHans Petter Selasky 		error = cuse_proc2proc_copy(
950fa0f6e62SHans Petter Selasky 		    curthread->td_proc, pchk->local_ptr,
951fa0f6e62SHans Petter Selasky 		    p_proc, pchk->peer_ptr,
952fa0f6e62SHans Petter Selasky 		    pchk->length);
953fa0f6e62SHans Petter Selasky 	} else {
954fa0f6e62SHans Petter Selasky 		error = cuse_proc2proc_copy(
955fa0f6e62SHans Petter Selasky 		    p_proc, pchk->peer_ptr,
956fa0f6e62SHans Petter Selasky 		    curthread->td_proc, pchk->local_ptr,
957fa0f6e62SHans Petter Selasky 		    pchk->length);
958fa0f6e62SHans Petter Selasky 	}
959fa0f6e62SHans Petter Selasky 
96098029019SHans Petter Selasky 	cuse_server_lock(pcs);
961fa0f6e62SHans Petter Selasky 
962fa0f6e62SHans Petter Selasky 	pccmd->proc_refs--;
963fa0f6e62SHans Petter Selasky 
964fa0f6e62SHans Petter Selasky 	if (pccmd->proc_curr == NULL)
965fa0f6e62SHans Petter Selasky 		cv_signal(&pccmd->cv);
966fa0f6e62SHans Petter Selasky 
967fa0f6e62SHans Petter Selasky 	return (error);
968fa0f6e62SHans Petter Selasky }
969fa0f6e62SHans Petter Selasky 
970fa0f6e62SHans Petter Selasky static int
971fa0f6e62SHans Petter Selasky cuse_alloc_unit_by_id_locked(struct cuse_server *pcs, int id)
972fa0f6e62SHans Petter Selasky {
973fa0f6e62SHans Petter Selasky 	int n;
974fa0f6e62SHans Petter Selasky 	int x = 0;
975fa0f6e62SHans Petter Selasky 	int match;
976fa0f6e62SHans Petter Selasky 
977fa0f6e62SHans Petter Selasky 	do {
978fa0f6e62SHans Petter Selasky 		for (match = n = 0; n != CUSE_DEVICES_MAX; n++) {
979fa0f6e62SHans Petter Selasky 			if (cuse_alloc_unit[n] != NULL) {
980fa0f6e62SHans Petter Selasky 				if ((cuse_alloc_unit_id[n] ^ id) & CUSE_ID_MASK)
981fa0f6e62SHans Petter Selasky 					continue;
982fa0f6e62SHans Petter Selasky 				if ((cuse_alloc_unit_id[n] & ~CUSE_ID_MASK) == x) {
983fa0f6e62SHans Petter Selasky 					x++;
984fa0f6e62SHans Petter Selasky 					match = 1;
985fa0f6e62SHans Petter Selasky 				}
986fa0f6e62SHans Petter Selasky 			}
987fa0f6e62SHans Petter Selasky 		}
988fa0f6e62SHans Petter Selasky 	} while (match);
989fa0f6e62SHans Petter Selasky 
990fa0f6e62SHans Petter Selasky 	if (x < 256) {
991fa0f6e62SHans Petter Selasky 		for (n = 0; n != CUSE_DEVICES_MAX; n++) {
992fa0f6e62SHans Petter Selasky 			if (cuse_alloc_unit[n] == NULL) {
993fa0f6e62SHans Petter Selasky 				cuse_alloc_unit[n] = pcs;
994fa0f6e62SHans Petter Selasky 				cuse_alloc_unit_id[n] = id | x;
995fa0f6e62SHans Petter Selasky 				return (x);
996fa0f6e62SHans Petter Selasky 			}
997fa0f6e62SHans Petter Selasky 		}
998fa0f6e62SHans Petter Selasky 	}
999fa0f6e62SHans Petter Selasky 	return (-1);
1000fa0f6e62SHans Petter Selasky }
1001fa0f6e62SHans Petter Selasky 
1002fa0f6e62SHans Petter Selasky static void
1003fa0f6e62SHans Petter Selasky cuse_server_wakeup_locked(struct cuse_server *pcs)
1004fa0f6e62SHans Petter Selasky {
1005fa0f6e62SHans Petter Selasky 	selwakeup(&pcs->selinfo);
1006fa0f6e62SHans Petter Selasky 	KNOTE_LOCKED(&pcs->selinfo.si_note, 0);
1007fa0f6e62SHans Petter Selasky }
1008fa0f6e62SHans Petter Selasky 
100995700043SHans Petter Selasky static void
101095700043SHans Petter Selasky cuse_server_wakeup_all_client_locked(struct cuse_server *pcs)
101195700043SHans Petter Selasky {
101295700043SHans Petter Selasky 	struct cuse_client *pcc;
101395700043SHans Petter Selasky 
101495700043SHans Petter Selasky 	TAILQ_FOREACH(pcc, &pcs->hcli, entry) {
101595700043SHans Petter Selasky 		pcc->cflags |= (CUSE_CLI_KNOTE_NEED_READ |
101695700043SHans Petter Selasky 		    CUSE_CLI_KNOTE_NEED_WRITE);
101795700043SHans Petter Selasky 	}
101895700043SHans Petter Selasky 	cuse_server_wakeup_locked(pcs);
101995700043SHans Petter Selasky }
102095700043SHans Petter Selasky 
1021fa0f6e62SHans Petter Selasky static int
1022fa0f6e62SHans Petter Selasky cuse_free_unit_by_id_locked(struct cuse_server *pcs, int id)
1023fa0f6e62SHans Petter Selasky {
1024fa0f6e62SHans Petter Selasky 	int n;
1025fa0f6e62SHans Petter Selasky 	int found = 0;
1026fa0f6e62SHans Petter Selasky 
1027fa0f6e62SHans Petter Selasky 	for (n = 0; n != CUSE_DEVICES_MAX; n++) {
1028fa0f6e62SHans Petter Selasky 		if (cuse_alloc_unit[n] == pcs) {
1029fa0f6e62SHans Petter Selasky 			if (cuse_alloc_unit_id[n] == id || id == -1) {
1030fa0f6e62SHans Petter Selasky 				cuse_alloc_unit[n] = NULL;
1031fa0f6e62SHans Petter Selasky 				cuse_alloc_unit_id[n] = 0;
1032fa0f6e62SHans Petter Selasky 				found = 1;
1033fa0f6e62SHans Petter Selasky 			}
1034fa0f6e62SHans Petter Selasky 		}
1035fa0f6e62SHans Petter Selasky 	}
1036fa0f6e62SHans Petter Selasky 
1037fa0f6e62SHans Petter Selasky 	return (found ? 0 : EINVAL);
1038fa0f6e62SHans Petter Selasky }
1039fa0f6e62SHans Petter Selasky 
1040fa0f6e62SHans Petter Selasky static int
1041fa0f6e62SHans Petter Selasky cuse_server_ioctl(struct cdev *dev, unsigned long cmd,
1042fa0f6e62SHans Petter Selasky     caddr_t data, int fflag, struct thread *td)
1043fa0f6e62SHans Petter Selasky {
1044fa0f6e62SHans Petter Selasky 	struct cuse_server *pcs;
1045fa0f6e62SHans Petter Selasky 	int error;
1046fa0f6e62SHans Petter Selasky 
1047fa0f6e62SHans Petter Selasky 	error = cuse_server_get(&pcs);
1048fa0f6e62SHans Petter Selasky 	if (error != 0)
1049fa0f6e62SHans Petter Selasky 		return (error);
1050fa0f6e62SHans Petter Selasky 
1051fa0f6e62SHans Petter Selasky 	switch (cmd) {
1052fa0f6e62SHans Petter Selasky 		struct cuse_client_command *pccmd;
1053fa0f6e62SHans Petter Selasky 		struct cuse_client *pcc;
1054fa0f6e62SHans Petter Selasky 		struct cuse_command *pcmd;
1055fa0f6e62SHans Petter Selasky 		struct cuse_alloc_info *pai;
1056fa0f6e62SHans Petter Selasky 		struct cuse_create_dev *pcd;
1057fa0f6e62SHans Petter Selasky 		struct cuse_server_dev *pcsd;
1058fa0f6e62SHans Petter Selasky 		struct cuse_data_chunk *pchk;
1059fa0f6e62SHans Petter Selasky 		int n;
1060fa0f6e62SHans Petter Selasky 
1061fa0f6e62SHans Petter Selasky 	case CUSE_IOCTL_GET_COMMAND:
1062fa0f6e62SHans Petter Selasky 		pcmd = (void *)data;
1063fa0f6e62SHans Petter Selasky 
106498029019SHans Petter Selasky 		cuse_server_lock(pcs);
1065fa0f6e62SHans Petter Selasky 
1066fa0f6e62SHans Petter Selasky 		while ((pccmd = TAILQ_FIRST(&pcs->head)) == NULL) {
106798029019SHans Petter Selasky 			error = cv_wait_sig(&pcs->cv, &pcs->mtx);
1068fa0f6e62SHans Petter Selasky 
1069fa0f6e62SHans Petter Selasky 			if (pcs->is_closing)
1070fa0f6e62SHans Petter Selasky 				error = ENXIO;
1071fa0f6e62SHans Petter Selasky 
1072fa0f6e62SHans Petter Selasky 			if (error) {
107398029019SHans Petter Selasky 				cuse_server_unlock(pcs);
1074fa0f6e62SHans Petter Selasky 				return (error);
1075fa0f6e62SHans Petter Selasky 			}
1076fa0f6e62SHans Petter Selasky 		}
1077fa0f6e62SHans Petter Selasky 
1078fa0f6e62SHans Petter Selasky 		TAILQ_REMOVE(&pcs->head, pccmd, entry);
1079fa0f6e62SHans Petter Selasky 		pccmd->entry.tqe_prev = NULL;
1080fa0f6e62SHans Petter Selasky 
1081fa0f6e62SHans Petter Selasky 		pccmd->entered = curthread;
1082fa0f6e62SHans Petter Selasky 
1083fa0f6e62SHans Petter Selasky 		*pcmd = pccmd->sub;
1084fa0f6e62SHans Petter Selasky 
108598029019SHans Petter Selasky 		cuse_server_unlock(pcs);
1086fa0f6e62SHans Petter Selasky 
1087fa0f6e62SHans Petter Selasky 		break;
1088fa0f6e62SHans Petter Selasky 
1089fa0f6e62SHans Petter Selasky 	case CUSE_IOCTL_SYNC_COMMAND:
1090fa0f6e62SHans Petter Selasky 
109198029019SHans Petter Selasky 		cuse_server_lock(pcs);
1092fa0f6e62SHans Petter Selasky 		while ((pccmd = cuse_server_find_command(pcs, curthread)) != NULL) {
1093fa0f6e62SHans Petter Selasky 			/* send sync command */
1094fa0f6e62SHans Petter Selasky 			pccmd->entered = NULL;
1095fa0f6e62SHans Petter Selasky 			pccmd->error = *(int *)data;
1096fa0f6e62SHans Petter Selasky 			pccmd->command = CUSE_CMD_SYNC;
1097fa0f6e62SHans Petter Selasky 
1098fa0f6e62SHans Petter Selasky 			/* signal peer, if any */
1099fa0f6e62SHans Petter Selasky 			cv_signal(&pccmd->cv);
1100fa0f6e62SHans Petter Selasky 		}
110198029019SHans Petter Selasky 		cuse_server_unlock(pcs);
1102fa0f6e62SHans Petter Selasky 
1103fa0f6e62SHans Petter Selasky 		break;
1104fa0f6e62SHans Petter Selasky 
1105fa0f6e62SHans Petter Selasky 	case CUSE_IOCTL_ALLOC_UNIT:
1106fa0f6e62SHans Petter Selasky 
110798029019SHans Petter Selasky 		cuse_server_lock(pcs);
1108fa0f6e62SHans Petter Selasky 		n = cuse_alloc_unit_by_id_locked(pcs,
1109fa0f6e62SHans Petter Selasky 		    CUSE_ID_DEFAULT(0));
111098029019SHans Petter Selasky 		cuse_server_unlock(pcs);
1111fa0f6e62SHans Petter Selasky 
1112fa0f6e62SHans Petter Selasky 		if (n < 0)
1113fa0f6e62SHans Petter Selasky 			error = ENOMEM;
1114fa0f6e62SHans Petter Selasky 		else
1115fa0f6e62SHans Petter Selasky 			*(int *)data = n;
1116fa0f6e62SHans Petter Selasky 		break;
1117fa0f6e62SHans Petter Selasky 
1118fa0f6e62SHans Petter Selasky 	case CUSE_IOCTL_ALLOC_UNIT_BY_ID:
1119fa0f6e62SHans Petter Selasky 
1120fa0f6e62SHans Petter Selasky 		n = *(int *)data;
1121fa0f6e62SHans Petter Selasky 
1122fa0f6e62SHans Petter Selasky 		n = (n & CUSE_ID_MASK);
1123fa0f6e62SHans Petter Selasky 
112498029019SHans Petter Selasky 		cuse_server_lock(pcs);
1125fa0f6e62SHans Petter Selasky 		n = cuse_alloc_unit_by_id_locked(pcs, n);
112698029019SHans Petter Selasky 		cuse_server_unlock(pcs);
1127fa0f6e62SHans Petter Selasky 
1128fa0f6e62SHans Petter Selasky 		if (n < 0)
1129fa0f6e62SHans Petter Selasky 			error = ENOMEM;
1130fa0f6e62SHans Petter Selasky 		else
1131fa0f6e62SHans Petter Selasky 			*(int *)data = n;
1132fa0f6e62SHans Petter Selasky 		break;
1133fa0f6e62SHans Petter Selasky 
1134fa0f6e62SHans Petter Selasky 	case CUSE_IOCTL_FREE_UNIT:
1135fa0f6e62SHans Petter Selasky 
1136fa0f6e62SHans Petter Selasky 		n = *(int *)data;
1137fa0f6e62SHans Petter Selasky 
1138fa0f6e62SHans Petter Selasky 		n = CUSE_ID_DEFAULT(n);
1139fa0f6e62SHans Petter Selasky 
114098029019SHans Petter Selasky 		cuse_server_lock(pcs);
1141fa0f6e62SHans Petter Selasky 		error = cuse_free_unit_by_id_locked(pcs, n);
114298029019SHans Petter Selasky 		cuse_server_unlock(pcs);
1143fa0f6e62SHans Petter Selasky 		break;
1144fa0f6e62SHans Petter Selasky 
1145fa0f6e62SHans Petter Selasky 	case CUSE_IOCTL_FREE_UNIT_BY_ID:
1146fa0f6e62SHans Petter Selasky 
1147fa0f6e62SHans Petter Selasky 		n = *(int *)data;
1148fa0f6e62SHans Petter Selasky 
114998029019SHans Petter Selasky 		cuse_server_lock(pcs);
1150fa0f6e62SHans Petter Selasky 		error = cuse_free_unit_by_id_locked(pcs, n);
115198029019SHans Petter Selasky 		cuse_server_unlock(pcs);
1152fa0f6e62SHans Petter Selasky 		break;
1153fa0f6e62SHans Petter Selasky 
1154fa0f6e62SHans Petter Selasky 	case CUSE_IOCTL_ALLOC_MEMORY:
1155fa0f6e62SHans Petter Selasky 
1156fa0f6e62SHans Petter Selasky 		pai = (void *)data;
1157fa0f6e62SHans Petter Selasky 
1158fa0f6e62SHans Petter Selasky 		if (pai->alloc_nr >= CUSE_ALLOC_UNIT_MAX) {
1159fa0f6e62SHans Petter Selasky 			error = ENOMEM;
1160fa0f6e62SHans Petter Selasky 			break;
1161fa0f6e62SHans Petter Selasky 		}
116256c49c63SHans Petter Selasky 		if (pai->page_count >= CUSE_ALLOC_PAGES_MAX) {
1163fa0f6e62SHans Petter Selasky 			error = ENOMEM;
1164fa0f6e62SHans Petter Selasky 			break;
1165fa0f6e62SHans Petter Selasky 		}
1166fa0f6e62SHans Petter Selasky 		error = cuse_server_alloc_memory(pcs,
116756c49c63SHans Petter Selasky 		    pai->alloc_nr, pai->page_count);
1168fa0f6e62SHans Petter Selasky 		break;
1169fa0f6e62SHans Petter Selasky 
1170fa0f6e62SHans Petter Selasky 	case CUSE_IOCTL_FREE_MEMORY:
1171fa0f6e62SHans Petter Selasky 		pai = (void *)data;
1172fa0f6e62SHans Petter Selasky 
1173fa0f6e62SHans Petter Selasky 		if (pai->alloc_nr >= CUSE_ALLOC_UNIT_MAX) {
1174fa0f6e62SHans Petter Selasky 			error = ENOMEM;
1175fa0f6e62SHans Petter Selasky 			break;
1176fa0f6e62SHans Petter Selasky 		}
117756c49c63SHans Petter Selasky 		error = cuse_server_free_memory(pcs, pai->alloc_nr);
1178fa0f6e62SHans Petter Selasky 		break;
1179fa0f6e62SHans Petter Selasky 
1180fa0f6e62SHans Petter Selasky 	case CUSE_IOCTL_GET_SIG:
1181fa0f6e62SHans Petter Selasky 
118298029019SHans Petter Selasky 		cuse_server_lock(pcs);
1183fa0f6e62SHans Petter Selasky 		pccmd = cuse_server_find_command(pcs, curthread);
1184fa0f6e62SHans Petter Selasky 
1185fa0f6e62SHans Petter Selasky 		if (pccmd != NULL) {
1186fa0f6e62SHans Petter Selasky 			n = pccmd->got_signal;
1187fa0f6e62SHans Petter Selasky 			pccmd->got_signal = 0;
1188fa0f6e62SHans Petter Selasky 		} else {
1189fa0f6e62SHans Petter Selasky 			n = 0;
1190fa0f6e62SHans Petter Selasky 		}
119198029019SHans Petter Selasky 		cuse_server_unlock(pcs);
1192fa0f6e62SHans Petter Selasky 
1193fa0f6e62SHans Petter Selasky 		*(int *)data = n;
1194fa0f6e62SHans Petter Selasky 
1195fa0f6e62SHans Petter Selasky 		break;
1196fa0f6e62SHans Petter Selasky 
1197fa0f6e62SHans Petter Selasky 	case CUSE_IOCTL_SET_PFH:
1198fa0f6e62SHans Petter Selasky 
119998029019SHans Petter Selasky 		cuse_server_lock(pcs);
1200fa0f6e62SHans Petter Selasky 		pccmd = cuse_server_find_command(pcs, curthread);
1201fa0f6e62SHans Petter Selasky 
1202fa0f6e62SHans Petter Selasky 		if (pccmd != NULL) {
1203fa0f6e62SHans Petter Selasky 			pcc = pccmd->client;
1204fa0f6e62SHans Petter Selasky 			for (n = 0; n != CUSE_CMD_MAX; n++) {
12059fb69c9dSHans Petter Selasky 				pcc->cmds[n].sub.per_file_handle = *(uintptr_t *)data;
1206fa0f6e62SHans Petter Selasky 			}
1207fa0f6e62SHans Petter Selasky 		} else {
1208fa0f6e62SHans Petter Selasky 			error = ENXIO;
1209fa0f6e62SHans Petter Selasky 		}
121098029019SHans Petter Selasky 		cuse_server_unlock(pcs);
1211fa0f6e62SHans Petter Selasky 		break;
1212fa0f6e62SHans Petter Selasky 
1213fa0f6e62SHans Petter Selasky 	case CUSE_IOCTL_CREATE_DEV:
1214fa0f6e62SHans Petter Selasky 
1215fa0f6e62SHans Petter Selasky 		error = priv_check(curthread, PRIV_DRIVER);
1216fa0f6e62SHans Petter Selasky 		if (error)
1217fa0f6e62SHans Petter Selasky 			break;
1218fa0f6e62SHans Petter Selasky 
1219fa0f6e62SHans Petter Selasky 		pcd = (void *)data;
1220fa0f6e62SHans Petter Selasky 
1221fa0f6e62SHans Petter Selasky 		/* filter input */
1222fa0f6e62SHans Petter Selasky 
1223fa0f6e62SHans Petter Selasky 		pcd->devname[sizeof(pcd->devname) - 1] = 0;
1224fa0f6e62SHans Petter Selasky 
1225fa0f6e62SHans Petter Selasky 		if (pcd->devname[0] == 0) {
1226fa0f6e62SHans Petter Selasky 			error = EINVAL;
1227fa0f6e62SHans Petter Selasky 			break;
1228fa0f6e62SHans Petter Selasky 		}
1229fa0f6e62SHans Petter Selasky 		cuse_str_filter(pcd->devname);
1230fa0f6e62SHans Petter Selasky 
1231fa0f6e62SHans Petter Selasky 		pcd->permissions &= 0777;
1232fa0f6e62SHans Petter Selasky 
1233fa0f6e62SHans Petter Selasky 		/* try to allocate a character device */
1234fa0f6e62SHans Petter Selasky 
1235fa0f6e62SHans Petter Selasky 		pcsd = malloc(sizeof(*pcsd), M_CUSE, M_WAITOK | M_ZERO);
1236fa0f6e62SHans Petter Selasky 
1237fa0f6e62SHans Petter Selasky 		pcsd->server = pcs;
1238fa0f6e62SHans Petter Selasky 
1239fa0f6e62SHans Petter Selasky 		pcsd->user_dev = pcd->dev;
1240fa0f6e62SHans Petter Selasky 
1241fa0f6e62SHans Petter Selasky 		pcsd->kern_dev = make_dev_credf(MAKEDEV_CHECKNAME,
1242fa0f6e62SHans Petter Selasky 		    &cuse_client_devsw, 0, NULL, pcd->user_id, pcd->group_id,
1243fa0f6e62SHans Petter Selasky 		    pcd->permissions, "%s", pcd->devname);
1244fa0f6e62SHans Petter Selasky 
1245fa0f6e62SHans Petter Selasky 		if (pcsd->kern_dev == NULL) {
1246fa0f6e62SHans Petter Selasky 			free(pcsd, M_CUSE);
1247fa0f6e62SHans Petter Selasky 			error = ENOMEM;
1248fa0f6e62SHans Petter Selasky 			break;
1249fa0f6e62SHans Petter Selasky 		}
1250fa0f6e62SHans Petter Selasky 		pcsd->kern_dev->si_drv1 = pcsd;
1251fa0f6e62SHans Petter Selasky 
125298029019SHans Petter Selasky 		cuse_server_lock(pcs);
1253fa0f6e62SHans Petter Selasky 		TAILQ_INSERT_TAIL(&pcs->hdev, pcsd, entry);
125498029019SHans Petter Selasky 		cuse_server_unlock(pcs);
1255fa0f6e62SHans Petter Selasky 
1256fa0f6e62SHans Petter Selasky 		break;
1257fa0f6e62SHans Petter Selasky 
1258fa0f6e62SHans Petter Selasky 	case CUSE_IOCTL_DESTROY_DEV:
1259fa0f6e62SHans Petter Selasky 
1260fa0f6e62SHans Petter Selasky 		error = priv_check(curthread, PRIV_DRIVER);
1261fa0f6e62SHans Petter Selasky 		if (error)
1262fa0f6e62SHans Petter Selasky 			break;
1263fa0f6e62SHans Petter Selasky 
126498029019SHans Petter Selasky 		cuse_server_lock(pcs);
1265fa0f6e62SHans Petter Selasky 
1266fa0f6e62SHans Petter Selasky 		error = EINVAL;
1267fa0f6e62SHans Petter Selasky 
1268fa0f6e62SHans Petter Selasky 		pcsd = TAILQ_FIRST(&pcs->hdev);
1269fa0f6e62SHans Petter Selasky 		while (pcsd != NULL) {
1270fa0f6e62SHans Petter Selasky 			if (pcsd->user_dev == *(struct cuse_dev **)data) {
1271fa0f6e62SHans Petter Selasky 				TAILQ_REMOVE(&pcs->hdev, pcsd, entry);
127298029019SHans Petter Selasky 				cuse_server_unlock(pcs);
1273fa0f6e62SHans Petter Selasky 				cuse_server_free_dev(pcsd);
127498029019SHans Petter Selasky 				cuse_server_lock(pcs);
1275fa0f6e62SHans Petter Selasky 				error = 0;
1276fa0f6e62SHans Petter Selasky 				pcsd = TAILQ_FIRST(&pcs->hdev);
1277fa0f6e62SHans Petter Selasky 			} else {
1278fa0f6e62SHans Petter Selasky 				pcsd = TAILQ_NEXT(pcsd, entry);
1279fa0f6e62SHans Petter Selasky 			}
1280fa0f6e62SHans Petter Selasky 		}
1281fa0f6e62SHans Petter Selasky 
128298029019SHans Petter Selasky 		cuse_server_unlock(pcs);
1283fa0f6e62SHans Petter Selasky 		break;
1284fa0f6e62SHans Petter Selasky 
1285fa0f6e62SHans Petter Selasky 	case CUSE_IOCTL_WRITE_DATA:
1286fa0f6e62SHans Petter Selasky 	case CUSE_IOCTL_READ_DATA:
1287fa0f6e62SHans Petter Selasky 
128898029019SHans Petter Selasky 		cuse_server_lock(pcs);
1289fa0f6e62SHans Petter Selasky 		pchk = (struct cuse_data_chunk *)data;
1290fa0f6e62SHans Petter Selasky 
1291fa0f6e62SHans Petter Selasky 		pccmd = cuse_server_find_command(pcs, curthread);
1292fa0f6e62SHans Petter Selasky 
1293fa0f6e62SHans Petter Selasky 		if (pccmd == NULL) {
1294fa0f6e62SHans Petter Selasky 			error = ENXIO;	/* invalid request */
1295fa0f6e62SHans Petter Selasky 		} else if (pchk->peer_ptr < CUSE_BUF_MIN_PTR) {
1296fa0f6e62SHans Petter Selasky 			error = EFAULT;	/* NULL pointer */
1297fa0f6e62SHans Petter Selasky 		} else if (pchk->peer_ptr < CUSE_BUF_MAX_PTR) {
129898029019SHans Petter Selasky 			error = cuse_server_ioctl_copy_locked(pcs, pccmd,
1299fa0f6e62SHans Petter Selasky 			    pchk, cmd == CUSE_IOCTL_READ_DATA);
1300fa0f6e62SHans Petter Selasky 		} else {
130198029019SHans Petter Selasky 			error = cuse_server_data_copy_locked(pcs, pccmd,
1302fa0f6e62SHans Petter Selasky 			    pchk, cmd == CUSE_IOCTL_READ_DATA);
1303fa0f6e62SHans Petter Selasky 		}
130498029019SHans Petter Selasky 		cuse_server_unlock(pcs);
1305fa0f6e62SHans Petter Selasky 		break;
1306fa0f6e62SHans Petter Selasky 
1307fa0f6e62SHans Petter Selasky 	case CUSE_IOCTL_SELWAKEUP:
130898029019SHans Petter Selasky 		cuse_server_lock(pcs);
1309fa0f6e62SHans Petter Selasky 		/*
1310fa0f6e62SHans Petter Selasky 		 * We don't know which direction caused the event.
1311fa0f6e62SHans Petter Selasky 		 * Wakeup both!
1312fa0f6e62SHans Petter Selasky 		 */
131395700043SHans Petter Selasky 		cuse_server_wakeup_all_client_locked(pcs);
131498029019SHans Petter Selasky 		cuse_server_unlock(pcs);
1315fa0f6e62SHans Petter Selasky 		break;
1316fa0f6e62SHans Petter Selasky 
1317fa0f6e62SHans Petter Selasky 	default:
1318fa0f6e62SHans Petter Selasky 		error = ENXIO;
1319fa0f6e62SHans Petter Selasky 		break;
1320fa0f6e62SHans Petter Selasky 	}
1321fa0f6e62SHans Petter Selasky 	return (error);
1322fa0f6e62SHans Petter Selasky }
1323fa0f6e62SHans Petter Selasky 
1324fa0f6e62SHans Petter Selasky static int
1325fa0f6e62SHans Petter Selasky cuse_server_poll(struct cdev *dev, int events, struct thread *td)
1326fa0f6e62SHans Petter Selasky {
1327fa0f6e62SHans Petter Selasky 	return (events & (POLLHUP | POLLPRI | POLLIN |
1328fa0f6e62SHans Petter Selasky 	    POLLRDNORM | POLLOUT | POLLWRNORM));
1329fa0f6e62SHans Petter Selasky }
1330fa0f6e62SHans Petter Selasky 
1331fa0f6e62SHans Petter Selasky static int
133256c49c63SHans Petter Selasky cuse_server_mmap_single(struct cdev *dev, vm_ooffset_t *offset,
133356c49c63SHans Petter Selasky     vm_size_t size, struct vm_object **object, int nprot)
1334fa0f6e62SHans Petter Selasky {
133556c49c63SHans Petter Selasky 	uint32_t page_nr = *offset / PAGE_SIZE;
1336fa0f6e62SHans Petter Selasky 	uint32_t alloc_nr = page_nr / CUSE_ALLOC_PAGES_MAX;
1337fa0f6e62SHans Petter Selasky 	struct cuse_memory *mem;
1338fa0f6e62SHans Petter Selasky 	struct cuse_server *pcs;
1339fa0f6e62SHans Petter Selasky 	int error;
1340fa0f6e62SHans Petter Selasky 
1341fa0f6e62SHans Petter Selasky 	error = cuse_server_get(&pcs);
1342fa0f6e62SHans Petter Selasky 	if (error != 0)
134356c49c63SHans Petter Selasky 		return (error);
1344fa0f6e62SHans Petter Selasky 
134598029019SHans Petter Selasky 	cuse_server_lock(pcs);
134656c49c63SHans Petter Selasky 	/* lookup memory structure */
134756c49c63SHans Petter Selasky 	TAILQ_FOREACH(mem, &pcs->hmem, entry) {
134856c49c63SHans Petter Selasky 		if (mem->alloc_nr == alloc_nr)
134956c49c63SHans Petter Selasky 			break;
1350fa0f6e62SHans Petter Selasky 	}
135156c49c63SHans Petter Selasky 	if (mem == NULL) {
135298029019SHans Petter Selasky 		cuse_server_unlock(pcs);
1353fa0f6e62SHans Petter Selasky 		return (ENOMEM);
1354fa0f6e62SHans Petter Selasky 	}
135556c49c63SHans Petter Selasky 	/* verify page offset */
1356fa0f6e62SHans Petter Selasky 	page_nr %= CUSE_ALLOC_PAGES_MAX;
1357fa0f6e62SHans Petter Selasky 	if (page_nr >= mem->page_count) {
135898029019SHans Petter Selasky 		cuse_server_unlock(pcs);
1359fa0f6e62SHans Petter Selasky 		return (ENXIO);
1360fa0f6e62SHans Petter Selasky 	}
136156c49c63SHans Petter Selasky 	/* verify mmap size */
136256c49c63SHans Petter Selasky 	if ((size % PAGE_SIZE) != 0 || (size < PAGE_SIZE) ||
136356c49c63SHans Petter Selasky 	    (size > ((mem->page_count - page_nr) * PAGE_SIZE))) {
136498029019SHans Petter Selasky 		cuse_server_unlock(pcs);
136556c49c63SHans Petter Selasky 		return (EINVAL);
136656c49c63SHans Petter Selasky 	}
136756c49c63SHans Petter Selasky 	vm_object_reference(mem->object);
136856c49c63SHans Petter Selasky 	*object = mem->object;
136998029019SHans Petter Selasky 	cuse_server_unlock(pcs);
1370fa0f6e62SHans Petter Selasky 
137156c49c63SHans Petter Selasky 	/* set new VM object offset to use */
137256c49c63SHans Petter Selasky 	*offset = page_nr * PAGE_SIZE;
1373fa0f6e62SHans Petter Selasky 
137456c49c63SHans Petter Selasky 	/* success */
1375fa0f6e62SHans Petter Selasky 	return (0);
1376fa0f6e62SHans Petter Selasky }
1377fa0f6e62SHans Petter Selasky 
1378fa0f6e62SHans Petter Selasky /*------------------------------------------------------------------------*
1379fa0f6e62SHans Petter Selasky  *	CUSE CLIENT PART
1380fa0f6e62SHans Petter Selasky  *------------------------------------------------------------------------*/
1381fa0f6e62SHans Petter Selasky static void
1382fa0f6e62SHans Petter Selasky cuse_client_free(void *arg)
1383fa0f6e62SHans Petter Selasky {
1384fa0f6e62SHans Petter Selasky 	struct cuse_client *pcc = arg;
1385fa0f6e62SHans Petter Selasky 	struct cuse_client_command *pccmd;
1386fa0f6e62SHans Petter Selasky 	struct cuse_server *pcs;
1387fa0f6e62SHans Petter Selasky 	int n;
1388fa0f6e62SHans Petter Selasky 
138998029019SHans Petter Selasky 	pcs = pcc->server;
139098029019SHans Petter Selasky 
139198029019SHans Petter Selasky 	cuse_server_lock(pcs);
1392fa0f6e62SHans Petter Selasky 	cuse_client_is_closing(pcc);
139398029019SHans Petter Selasky 	TAILQ_REMOVE(&pcs->hcli, pcc, entry);
139498029019SHans Petter Selasky 	cuse_server_unlock(pcs);
1395fa0f6e62SHans Petter Selasky 
1396fa0f6e62SHans Petter Selasky 	for (n = 0; n != CUSE_CMD_MAX; n++) {
1397fa0f6e62SHans Petter Selasky 		pccmd = &pcc->cmds[n];
1398fa0f6e62SHans Petter Selasky 
1399fa0f6e62SHans Petter Selasky 		sx_destroy(&pccmd->sx);
1400fa0f6e62SHans Petter Selasky 		cv_destroy(&pccmd->cv);
1401fa0f6e62SHans Petter Selasky 	}
1402fa0f6e62SHans Petter Selasky 
1403fa0f6e62SHans Petter Selasky 	free(pcc, M_CUSE);
1404fa0f6e62SHans Petter Selasky 
1405fa0f6e62SHans Petter Selasky 	/* drop reference on server */
140656c49c63SHans Petter Selasky 	cuse_server_unref(pcs);
1407fa0f6e62SHans Petter Selasky }
1408fa0f6e62SHans Petter Selasky 
1409fa0f6e62SHans Petter Selasky static int
1410fa0f6e62SHans Petter Selasky cuse_client_open(struct cdev *dev, int fflags, int devtype, struct thread *td)
1411fa0f6e62SHans Petter Selasky {
1412fa0f6e62SHans Petter Selasky 	struct cuse_client_command *pccmd;
1413fa0f6e62SHans Petter Selasky 	struct cuse_server_dev *pcsd;
1414fa0f6e62SHans Petter Selasky 	struct cuse_client *pcc;
1415fa0f6e62SHans Petter Selasky 	struct cuse_server *pcs;
1416fa0f6e62SHans Petter Selasky 	struct cuse_dev *pcd;
1417fa0f6e62SHans Petter Selasky 	int error;
1418fa0f6e62SHans Petter Selasky 	int n;
1419fa0f6e62SHans Petter Selasky 
1420fa0f6e62SHans Petter Selasky 	pcsd = dev->si_drv1;
1421fa0f6e62SHans Petter Selasky 	if (pcsd != NULL) {
1422fa0f6e62SHans Petter Selasky 		pcs = pcsd->server;
1423fa0f6e62SHans Petter Selasky 		pcd = pcsd->user_dev;
142498029019SHans Petter Selasky 
142598029019SHans Petter Selasky 		cuse_server_lock(pcs);
1426ac60e801SHans Petter Selasky 		/*
1427ac60e801SHans Petter Selasky 		 * Check that the refcount didn't wrap and that the
1428ac60e801SHans Petter Selasky 		 * same process is not both client and server. This
1429ac60e801SHans Petter Selasky 		 * can easily lead to deadlocks when destroying the
1430ac60e801SHans Petter Selasky 		 * CUSE character device nodes:
1431ac60e801SHans Petter Selasky 		 */
1432fa0f6e62SHans Petter Selasky 		pcs->refs++;
1433ac60e801SHans Petter Selasky 		if (pcs->refs < 0 || pcs->pid == curproc->p_pid) {
1434ac60e801SHans Petter Selasky 			/* overflow or wrong PID */
1435fa0f6e62SHans Petter Selasky 			pcs->refs--;
143698029019SHans Petter Selasky 			cuse_server_unlock(pcs);
1437fa0f6e62SHans Petter Selasky 			return (EINVAL);
143898029019SHans Petter Selasky 		}
143998029019SHans Petter Selasky 		cuse_server_unlock(pcs);
144098029019SHans Petter Selasky 	} else {
144198029019SHans Petter Selasky 		return (EINVAL);
144298029019SHans Petter Selasky 	}
1443fa0f6e62SHans Petter Selasky 
1444fa0f6e62SHans Petter Selasky 	pcc = malloc(sizeof(*pcc), M_CUSE, M_WAITOK | M_ZERO);
1445fa0f6e62SHans Petter Selasky 	if (devfs_set_cdevpriv(pcc, &cuse_client_free)) {
1446fa0f6e62SHans Petter Selasky 		printf("Cuse: Cannot set cdevpriv.\n");
1447fa0f6e62SHans Petter Selasky 		/* drop reference on server */
144856c49c63SHans Petter Selasky 		cuse_server_unref(pcs);
1449fa0f6e62SHans Petter Selasky 		free(pcc, M_CUSE);
1450fa0f6e62SHans Petter Selasky 		return (ENOMEM);
1451fa0f6e62SHans Petter Selasky 	}
1452fa0f6e62SHans Petter Selasky 	pcc->fflags = fflags;
1453fa0f6e62SHans Petter Selasky 	pcc->server_dev = pcsd;
1454fa0f6e62SHans Petter Selasky 	pcc->server = pcs;
1455fa0f6e62SHans Petter Selasky 
1456fa0f6e62SHans Petter Selasky 	for (n = 0; n != CUSE_CMD_MAX; n++) {
1457fa0f6e62SHans Petter Selasky 		pccmd = &pcc->cmds[n];
1458fa0f6e62SHans Petter Selasky 
1459fa0f6e62SHans Petter Selasky 		pccmd->sub.dev = pcd;
1460fa0f6e62SHans Petter Selasky 		pccmd->sub.command = n;
1461fa0f6e62SHans Petter Selasky 		pccmd->client = pcc;
1462fa0f6e62SHans Petter Selasky 
1463fa0f6e62SHans Petter Selasky 		sx_init(&pccmd->sx, "cuse-client-sx");
1464fa0f6e62SHans Petter Selasky 		cv_init(&pccmd->cv, "cuse-client-cv");
1465fa0f6e62SHans Petter Selasky 	}
1466fa0f6e62SHans Petter Selasky 
146798029019SHans Petter Selasky 	cuse_server_lock(pcs);
1468fa0f6e62SHans Petter Selasky 
1469fa0f6e62SHans Petter Selasky 	/* cuse_client_free() assumes that the client is listed somewhere! */
1470fa0f6e62SHans Petter Selasky 	/* always enqueue */
1471fa0f6e62SHans Petter Selasky 
1472fa0f6e62SHans Petter Selasky 	TAILQ_INSERT_TAIL(&pcs->hcli, pcc, entry);
1473fa0f6e62SHans Petter Selasky 
1474fa0f6e62SHans Petter Selasky 	/* check if server is closing */
1475fa0f6e62SHans Petter Selasky 	if ((pcs->is_closing != 0) || (dev->si_drv1 == NULL)) {
1476fa0f6e62SHans Petter Selasky 		error = EINVAL;
1477fa0f6e62SHans Petter Selasky 	} else {
1478fa0f6e62SHans Petter Selasky 		error = 0;
1479fa0f6e62SHans Petter Selasky 	}
148098029019SHans Petter Selasky 	cuse_server_unlock(pcs);
1481fa0f6e62SHans Petter Selasky 
1482fa0f6e62SHans Petter Selasky 	if (error) {
1483fa0f6e62SHans Petter Selasky 		devfs_clear_cdevpriv();	/* XXX bugfix */
1484fa0f6e62SHans Petter Selasky 		return (error);
1485fa0f6e62SHans Petter Selasky 	}
1486fa0f6e62SHans Petter Selasky 	pccmd = &pcc->cmds[CUSE_CMD_OPEN];
1487fa0f6e62SHans Petter Selasky 
1488fa0f6e62SHans Petter Selasky 	cuse_cmd_lock(pccmd);
1489fa0f6e62SHans Petter Selasky 
149098029019SHans Petter Selasky 	cuse_server_lock(pcs);
1491fa0f6e62SHans Petter Selasky 	cuse_client_send_command_locked(pccmd, 0, 0, pcc->fflags, 0);
1492fa0f6e62SHans Petter Selasky 
1493fa0f6e62SHans Petter Selasky 	error = cuse_client_receive_command_locked(pccmd, 0, 0);
149498029019SHans Petter Selasky 	cuse_server_unlock(pcs);
1495fa0f6e62SHans Petter Selasky 
1496fa0f6e62SHans Petter Selasky 	if (error < 0) {
1497fa0f6e62SHans Petter Selasky 		error = cuse_convert_error(error);
1498fa0f6e62SHans Petter Selasky 	} else {
1499fa0f6e62SHans Petter Selasky 		error = 0;
1500fa0f6e62SHans Petter Selasky 	}
1501fa0f6e62SHans Petter Selasky 
1502fa0f6e62SHans Petter Selasky 	cuse_cmd_unlock(pccmd);
1503fa0f6e62SHans Petter Selasky 
1504fa0f6e62SHans Petter Selasky 	if (error)
1505fa0f6e62SHans Petter Selasky 		devfs_clear_cdevpriv();	/* XXX bugfix */
1506fa0f6e62SHans Petter Selasky 
1507fa0f6e62SHans Petter Selasky 	return (error);
1508fa0f6e62SHans Petter Selasky }
1509fa0f6e62SHans Petter Selasky 
1510fa0f6e62SHans Petter Selasky static int
1511fa0f6e62SHans Petter Selasky cuse_client_close(struct cdev *dev, int fflag, int devtype, struct thread *td)
1512fa0f6e62SHans Petter Selasky {
1513fa0f6e62SHans Petter Selasky 	struct cuse_client_command *pccmd;
1514fa0f6e62SHans Petter Selasky 	struct cuse_client *pcc;
151598029019SHans Petter Selasky 	struct cuse_server *pcs;
1516fa0f6e62SHans Petter Selasky 	int error;
1517fa0f6e62SHans Petter Selasky 
1518fa0f6e62SHans Petter Selasky 	error = cuse_client_get(&pcc);
1519fa0f6e62SHans Petter Selasky 	if (error != 0)
1520fa0f6e62SHans Petter Selasky 		return (0);
1521fa0f6e62SHans Petter Selasky 
1522fa0f6e62SHans Petter Selasky 	pccmd = &pcc->cmds[CUSE_CMD_CLOSE];
152398029019SHans Petter Selasky 	pcs = pcc->server;
1524fa0f6e62SHans Petter Selasky 
1525fa0f6e62SHans Petter Selasky 	cuse_cmd_lock(pccmd);
1526fa0f6e62SHans Petter Selasky 
152798029019SHans Petter Selasky 	cuse_server_lock(pcs);
1528fa0f6e62SHans Petter Selasky 	cuse_client_send_command_locked(pccmd, 0, 0, pcc->fflags, 0);
1529fa0f6e62SHans Petter Selasky 
1530fa0f6e62SHans Petter Selasky 	error = cuse_client_receive_command_locked(pccmd, 0, 0);
1531fa0f6e62SHans Petter Selasky 	cuse_cmd_unlock(pccmd);
1532fa0f6e62SHans Petter Selasky 
1533fa0f6e62SHans Petter Selasky 	cuse_client_is_closing(pcc);
153498029019SHans Petter Selasky 	cuse_server_unlock(pcs);
1535fa0f6e62SHans Petter Selasky 
1536fa0f6e62SHans Petter Selasky 	return (0);
1537fa0f6e62SHans Petter Selasky }
1538fa0f6e62SHans Petter Selasky 
1539fa0f6e62SHans Petter Selasky static void
1540fa0f6e62SHans Petter Selasky cuse_client_kqfilter_poll(struct cdev *dev, struct cuse_client *pcc)
1541fa0f6e62SHans Petter Selasky {
154298029019SHans Petter Selasky 	struct cuse_server *pcs = pcc->server;
1543fa0f6e62SHans Petter Selasky 	int temp;
1544fa0f6e62SHans Petter Selasky 
154598029019SHans Petter Selasky 	cuse_server_lock(pcs);
1546fa0f6e62SHans Petter Selasky 	temp = (pcc->cflags & (CUSE_CLI_KNOTE_HAS_READ |
1547fa0f6e62SHans Petter Selasky 	    CUSE_CLI_KNOTE_HAS_WRITE));
1548fa0f6e62SHans Petter Selasky 	pcc->cflags &= ~(CUSE_CLI_KNOTE_NEED_READ |
1549fa0f6e62SHans Petter Selasky 	    CUSE_CLI_KNOTE_NEED_WRITE);
155098029019SHans Petter Selasky 	cuse_server_unlock(pcs);
1551fa0f6e62SHans Petter Selasky 
1552fa0f6e62SHans Petter Selasky 	if (temp != 0) {
1553fa0f6e62SHans Petter Selasky 		/* get the latest polling state from the server */
1554fa0f6e62SHans Petter Selasky 		temp = cuse_client_poll(dev, POLLIN | POLLOUT, NULL);
1555fa0f6e62SHans Petter Selasky 
1556fa0f6e62SHans Petter Selasky 		if (temp & (POLLIN | POLLOUT)) {
155798029019SHans Petter Selasky 			cuse_server_lock(pcs);
1558fa0f6e62SHans Petter Selasky 			if (temp & POLLIN)
1559fa0f6e62SHans Petter Selasky 				pcc->cflags |= CUSE_CLI_KNOTE_NEED_READ;
1560fa0f6e62SHans Petter Selasky 			if (temp & POLLOUT)
1561fa0f6e62SHans Petter Selasky 				pcc->cflags |= CUSE_CLI_KNOTE_NEED_WRITE;
1562fa0f6e62SHans Petter Selasky 
1563fa0f6e62SHans Petter Selasky 			/* make sure the "knote" gets woken up */
1564fa0f6e62SHans Petter Selasky 			cuse_server_wakeup_locked(pcc->server);
156598029019SHans Petter Selasky 			cuse_server_unlock(pcs);
1566fa0f6e62SHans Petter Selasky 		}
1567fa0f6e62SHans Petter Selasky 	}
15689a83b097SHans Petter Selasky }
1569fa0f6e62SHans Petter Selasky 
1570fa0f6e62SHans Petter Selasky static int
1571fa0f6e62SHans Petter Selasky cuse_client_read(struct cdev *dev, struct uio *uio, int ioflag)
1572fa0f6e62SHans Petter Selasky {
1573fa0f6e62SHans Petter Selasky 	struct cuse_client_command *pccmd;
1574fa0f6e62SHans Petter Selasky 	struct cuse_client *pcc;
157598029019SHans Petter Selasky 	struct cuse_server *pcs;
1576fa0f6e62SHans Petter Selasky 	int error;
1577fa0f6e62SHans Petter Selasky 	int len;
1578fa0f6e62SHans Petter Selasky 
1579fa0f6e62SHans Petter Selasky 	error = cuse_client_get(&pcc);
1580fa0f6e62SHans Petter Selasky 	if (error != 0)
1581fa0f6e62SHans Petter Selasky 		return (error);
1582fa0f6e62SHans Petter Selasky 
1583fa0f6e62SHans Petter Selasky 	pccmd = &pcc->cmds[CUSE_CMD_READ];
158498029019SHans Petter Selasky 	pcs = pcc->server;
1585fa0f6e62SHans Petter Selasky 
1586fa0f6e62SHans Petter Selasky 	if (uio->uio_segflg != UIO_USERSPACE) {
1587fa0f6e62SHans Petter Selasky 		return (EINVAL);
1588fa0f6e62SHans Petter Selasky 	}
1589fa0f6e62SHans Petter Selasky 	uio->uio_segflg = UIO_NOCOPY;
1590fa0f6e62SHans Petter Selasky 
1591fa0f6e62SHans Petter Selasky 	cuse_cmd_lock(pccmd);
1592fa0f6e62SHans Petter Selasky 
1593fa0f6e62SHans Petter Selasky 	while (uio->uio_resid != 0) {
1594fa0f6e62SHans Petter Selasky 		if (uio->uio_iov->iov_len > CUSE_LENGTH_MAX) {
1595fa0f6e62SHans Petter Selasky 			error = ENOMEM;
1596fa0f6e62SHans Petter Selasky 			break;
1597fa0f6e62SHans Petter Selasky 		}
1598fa0f6e62SHans Petter Selasky 		len = uio->uio_iov->iov_len;
1599fa0f6e62SHans Petter Selasky 
160098029019SHans Petter Selasky 		cuse_server_lock(pcs);
1601fa0f6e62SHans Petter Selasky 		cuse_client_send_command_locked(pccmd,
1602beebd9aaSHans Petter Selasky 		    (uintptr_t)uio->uio_iov->iov_base,
1603fa0f6e62SHans Petter Selasky 		    (unsigned long)(unsigned int)len, pcc->fflags, ioflag);
1604fa0f6e62SHans Petter Selasky 
1605fa0f6e62SHans Petter Selasky 		error = cuse_client_receive_command_locked(pccmd, 0, 0);
160698029019SHans Petter Selasky 		cuse_server_unlock(pcs);
1607fa0f6e62SHans Petter Selasky 
1608fa0f6e62SHans Petter Selasky 		if (error < 0) {
1609fa0f6e62SHans Petter Selasky 			error = cuse_convert_error(error);
1610fa0f6e62SHans Petter Selasky 			break;
1611fa0f6e62SHans Petter Selasky 		} else if (error == len) {
1612fa0f6e62SHans Petter Selasky 			error = uiomove(NULL, error, uio);
1613fa0f6e62SHans Petter Selasky 			if (error)
1614fa0f6e62SHans Petter Selasky 				break;
1615fa0f6e62SHans Petter Selasky 		} else {
1616fa0f6e62SHans Petter Selasky 			error = uiomove(NULL, error, uio);
1617fa0f6e62SHans Petter Selasky 			break;
1618fa0f6e62SHans Petter Selasky 		}
1619fa0f6e62SHans Petter Selasky 	}
1620fa0f6e62SHans Petter Selasky 	cuse_cmd_unlock(pccmd);
1621fa0f6e62SHans Petter Selasky 
1622fa0f6e62SHans Petter Selasky 	uio->uio_segflg = UIO_USERSPACE;/* restore segment flag */
1623fa0f6e62SHans Petter Selasky 
1624fa0f6e62SHans Petter Selasky 	if (error == EWOULDBLOCK)
1625fa0f6e62SHans Petter Selasky 		cuse_client_kqfilter_poll(dev, pcc);
1626fa0f6e62SHans Petter Selasky 
1627fa0f6e62SHans Petter Selasky 	return (error);
1628fa0f6e62SHans Petter Selasky }
1629fa0f6e62SHans Petter Selasky 
1630fa0f6e62SHans Petter Selasky static int
1631fa0f6e62SHans Petter Selasky cuse_client_write(struct cdev *dev, struct uio *uio, int ioflag)
1632fa0f6e62SHans Petter Selasky {
1633fa0f6e62SHans Petter Selasky 	struct cuse_client_command *pccmd;
1634fa0f6e62SHans Petter Selasky 	struct cuse_client *pcc;
163598029019SHans Petter Selasky 	struct cuse_server *pcs;
1636fa0f6e62SHans Petter Selasky 	int error;
1637fa0f6e62SHans Petter Selasky 	int len;
1638fa0f6e62SHans Petter Selasky 
1639fa0f6e62SHans Petter Selasky 	error = cuse_client_get(&pcc);
1640fa0f6e62SHans Petter Selasky 	if (error != 0)
1641fa0f6e62SHans Petter Selasky 		return (error);
1642fa0f6e62SHans Petter Selasky 
1643fa0f6e62SHans Petter Selasky 	pccmd = &pcc->cmds[CUSE_CMD_WRITE];
164498029019SHans Petter Selasky 	pcs = pcc->server;
1645fa0f6e62SHans Petter Selasky 
1646fa0f6e62SHans Petter Selasky 	if (uio->uio_segflg != UIO_USERSPACE) {
1647fa0f6e62SHans Petter Selasky 		return (EINVAL);
1648fa0f6e62SHans Petter Selasky 	}
1649fa0f6e62SHans Petter Selasky 	uio->uio_segflg = UIO_NOCOPY;
1650fa0f6e62SHans Petter Selasky 
1651fa0f6e62SHans Petter Selasky 	cuse_cmd_lock(pccmd);
1652fa0f6e62SHans Petter Selasky 
1653fa0f6e62SHans Petter Selasky 	while (uio->uio_resid != 0) {
1654fa0f6e62SHans Petter Selasky 		if (uio->uio_iov->iov_len > CUSE_LENGTH_MAX) {
1655fa0f6e62SHans Petter Selasky 			error = ENOMEM;
1656fa0f6e62SHans Petter Selasky 			break;
1657fa0f6e62SHans Petter Selasky 		}
1658fa0f6e62SHans Petter Selasky 		len = uio->uio_iov->iov_len;
1659fa0f6e62SHans Petter Selasky 
166098029019SHans Petter Selasky 		cuse_server_lock(pcs);
1661fa0f6e62SHans Petter Selasky 		cuse_client_send_command_locked(pccmd,
1662beebd9aaSHans Petter Selasky 		    (uintptr_t)uio->uio_iov->iov_base,
1663fa0f6e62SHans Petter Selasky 		    (unsigned long)(unsigned int)len, pcc->fflags, ioflag);
1664fa0f6e62SHans Petter Selasky 
1665fa0f6e62SHans Petter Selasky 		error = cuse_client_receive_command_locked(pccmd, 0, 0);
166698029019SHans Petter Selasky 		cuse_server_unlock(pcs);
1667fa0f6e62SHans Petter Selasky 
1668fa0f6e62SHans Petter Selasky 		if (error < 0) {
1669fa0f6e62SHans Petter Selasky 			error = cuse_convert_error(error);
1670fa0f6e62SHans Petter Selasky 			break;
1671fa0f6e62SHans Petter Selasky 		} else if (error == len) {
1672fa0f6e62SHans Petter Selasky 			error = uiomove(NULL, error, uio);
1673fa0f6e62SHans Petter Selasky 			if (error)
1674fa0f6e62SHans Petter Selasky 				break;
1675fa0f6e62SHans Petter Selasky 		} else {
1676fa0f6e62SHans Petter Selasky 			error = uiomove(NULL, error, uio);
1677fa0f6e62SHans Petter Selasky 			break;
1678fa0f6e62SHans Petter Selasky 		}
1679fa0f6e62SHans Petter Selasky 	}
1680fa0f6e62SHans Petter Selasky 	cuse_cmd_unlock(pccmd);
1681fa0f6e62SHans Petter Selasky 
1682fa0f6e62SHans Petter Selasky 	uio->uio_segflg = UIO_USERSPACE;/* restore segment flag */
1683fa0f6e62SHans Petter Selasky 
1684fa0f6e62SHans Petter Selasky 	if (error == EWOULDBLOCK)
1685fa0f6e62SHans Petter Selasky 		cuse_client_kqfilter_poll(dev, pcc);
1686fa0f6e62SHans Petter Selasky 
1687fa0f6e62SHans Petter Selasky 	return (error);
1688fa0f6e62SHans Petter Selasky }
1689fa0f6e62SHans Petter Selasky 
1690fa0f6e62SHans Petter Selasky int
1691fa0f6e62SHans Petter Selasky cuse_client_ioctl(struct cdev *dev, unsigned long cmd,
1692fa0f6e62SHans Petter Selasky     caddr_t data, int fflag, struct thread *td)
1693fa0f6e62SHans Petter Selasky {
1694fa0f6e62SHans Petter Selasky 	struct cuse_client_command *pccmd;
1695fa0f6e62SHans Petter Selasky 	struct cuse_client *pcc;
169698029019SHans Petter Selasky 	struct cuse_server *pcs;
1697fa0f6e62SHans Petter Selasky 	int error;
1698fa0f6e62SHans Petter Selasky 	int len;
1699fa0f6e62SHans Petter Selasky 
1700fa0f6e62SHans Petter Selasky 	error = cuse_client_get(&pcc);
1701fa0f6e62SHans Petter Selasky 	if (error != 0)
1702fa0f6e62SHans Petter Selasky 		return (error);
1703fa0f6e62SHans Petter Selasky 
1704fa0f6e62SHans Petter Selasky 	len = IOCPARM_LEN(cmd);
1705fa0f6e62SHans Petter Selasky 	if (len > CUSE_BUFFER_MAX)
1706fa0f6e62SHans Petter Selasky 		return (ENOMEM);
1707fa0f6e62SHans Petter Selasky 
1708fa0f6e62SHans Petter Selasky 	pccmd = &pcc->cmds[CUSE_CMD_IOCTL];
170998029019SHans Petter Selasky 	pcs = pcc->server;
1710fa0f6e62SHans Petter Selasky 
1711fa0f6e62SHans Petter Selasky 	cuse_cmd_lock(pccmd);
1712fa0f6e62SHans Petter Selasky 
1713010638abSHans Petter Selasky 	if (cmd & (IOC_IN | IOC_VOID))
1714fa0f6e62SHans Petter Selasky 		memcpy(pcc->ioctl_buffer, data, len);
1715fa0f6e62SHans Petter Selasky 
1716fa0f6e62SHans Petter Selasky 	/*
1717fa0f6e62SHans Petter Selasky 	 * When the ioctl-length is zero drivers can pass information
1718fa0f6e62SHans Petter Selasky 	 * through the data pointer of the ioctl. Make sure this information
1719fa0f6e62SHans Petter Selasky 	 * is forwarded to the driver.
1720fa0f6e62SHans Petter Selasky 	 */
1721fa0f6e62SHans Petter Selasky 
172298029019SHans Petter Selasky 	cuse_server_lock(pcs);
1723fa0f6e62SHans Petter Selasky 	cuse_client_send_command_locked(pccmd,
1724fa0f6e62SHans Petter Selasky 	    (len == 0) ? *(long *)data : CUSE_BUF_MIN_PTR,
1725fa0f6e62SHans Petter Selasky 	    (unsigned long)cmd, pcc->fflags,
1726fa0f6e62SHans Petter Selasky 	    (fflag & O_NONBLOCK) ? IO_NDELAY : 0);
1727fa0f6e62SHans Petter Selasky 
1728fa0f6e62SHans Petter Selasky 	error = cuse_client_receive_command_locked(pccmd, data, len);
172998029019SHans Petter Selasky 	cuse_server_unlock(pcs);
1730fa0f6e62SHans Petter Selasky 
1731fa0f6e62SHans Petter Selasky 	if (error < 0) {
1732fa0f6e62SHans Petter Selasky 		error = cuse_convert_error(error);
1733fa0f6e62SHans Petter Selasky 	} else {
1734fa0f6e62SHans Petter Selasky 		error = 0;
1735fa0f6e62SHans Petter Selasky 	}
1736fa0f6e62SHans Petter Selasky 
1737fa0f6e62SHans Petter Selasky 	if (cmd & IOC_OUT)
1738fa0f6e62SHans Petter Selasky 		memcpy(data, pcc->ioctl_buffer, len);
1739fa0f6e62SHans Petter Selasky 
1740fa0f6e62SHans Petter Selasky 	cuse_cmd_unlock(pccmd);
1741fa0f6e62SHans Petter Selasky 
1742fa0f6e62SHans Petter Selasky 	if (error == EWOULDBLOCK)
1743fa0f6e62SHans Petter Selasky 		cuse_client_kqfilter_poll(dev, pcc);
1744fa0f6e62SHans Petter Selasky 
1745fa0f6e62SHans Petter Selasky 	return (error);
1746fa0f6e62SHans Petter Selasky }
1747fa0f6e62SHans Petter Selasky 
1748fa0f6e62SHans Petter Selasky static int
1749fa0f6e62SHans Petter Selasky cuse_client_poll(struct cdev *dev, int events, struct thread *td)
1750fa0f6e62SHans Petter Selasky {
1751fa0f6e62SHans Petter Selasky 	struct cuse_client_command *pccmd;
1752fa0f6e62SHans Petter Selasky 	struct cuse_client *pcc;
175398029019SHans Petter Selasky 	struct cuse_server *pcs;
1754fa0f6e62SHans Petter Selasky 	unsigned long temp;
1755fa0f6e62SHans Petter Selasky 	int error;
1756fa0f6e62SHans Petter Selasky 	int revents;
1757fa0f6e62SHans Petter Selasky 
1758fa0f6e62SHans Petter Selasky 	error = cuse_client_get(&pcc);
1759fa0f6e62SHans Petter Selasky 	if (error != 0)
176095700043SHans Petter Selasky 		goto pollnval;
1761fa0f6e62SHans Petter Selasky 
1762fa0f6e62SHans Petter Selasky 	temp = 0;
176398029019SHans Petter Selasky 	pcs = pcc->server;
1764fa0f6e62SHans Petter Selasky 
1765fa0f6e62SHans Petter Selasky 	if (events & (POLLPRI | POLLIN | POLLRDNORM))
1766fa0f6e62SHans Petter Selasky 		temp |= CUSE_POLL_READ;
1767fa0f6e62SHans Petter Selasky 
1768fa0f6e62SHans Petter Selasky 	if (events & (POLLOUT | POLLWRNORM))
1769fa0f6e62SHans Petter Selasky 		temp |= CUSE_POLL_WRITE;
1770fa0f6e62SHans Petter Selasky 
1771fa0f6e62SHans Petter Selasky 	if (events & POLLHUP)
1772fa0f6e62SHans Petter Selasky 		temp |= CUSE_POLL_ERROR;
1773fa0f6e62SHans Petter Selasky 
1774fa0f6e62SHans Petter Selasky 	pccmd = &pcc->cmds[CUSE_CMD_POLL];
1775fa0f6e62SHans Petter Selasky 
1776fa0f6e62SHans Petter Selasky 	cuse_cmd_lock(pccmd);
1777fa0f6e62SHans Petter Selasky 
1778fa0f6e62SHans Petter Selasky 	/* Need to selrecord() first to not loose any events. */
1779fa0f6e62SHans Petter Selasky 	if (temp != 0 && td != NULL)
178098029019SHans Petter Selasky 		selrecord(td, &pcs->selinfo);
1781fa0f6e62SHans Petter Selasky 
178298029019SHans Petter Selasky 	cuse_server_lock(pcs);
1783fa0f6e62SHans Petter Selasky 	cuse_client_send_command_locked(pccmd,
1784fa0f6e62SHans Petter Selasky 	    0, temp, pcc->fflags, IO_NDELAY);
1785fa0f6e62SHans Petter Selasky 
1786fa0f6e62SHans Petter Selasky 	error = cuse_client_receive_command_locked(pccmd, 0, 0);
178798029019SHans Petter Selasky 	cuse_server_unlock(pcs);
1788fa0f6e62SHans Petter Selasky 
178995700043SHans Petter Selasky 	cuse_cmd_unlock(pccmd);
179095700043SHans Petter Selasky 
1791fa0f6e62SHans Petter Selasky 	if (error < 0) {
179295700043SHans Petter Selasky 		goto pollnval;
1793fa0f6e62SHans Petter Selasky 	} else {
1794fa0f6e62SHans Petter Selasky 		revents = 0;
1795fa0f6e62SHans Petter Selasky 		if (error & CUSE_POLL_READ)
1796fa0f6e62SHans Petter Selasky 			revents |= (events & (POLLPRI | POLLIN | POLLRDNORM));
1797fa0f6e62SHans Petter Selasky 		if (error & CUSE_POLL_WRITE)
1798fa0f6e62SHans Petter Selasky 			revents |= (events & (POLLOUT | POLLWRNORM));
1799fa0f6e62SHans Petter Selasky 		if (error & CUSE_POLL_ERROR)
1800fa0f6e62SHans Petter Selasky 			revents |= (events & POLLHUP);
1801fa0f6e62SHans Petter Selasky 	}
1802fa0f6e62SHans Petter Selasky 	return (revents);
180395700043SHans Petter Selasky 
180495700043SHans Petter Selasky pollnval:
180595700043SHans Petter Selasky 	/* XXX many clients don't understand POLLNVAL */
180695700043SHans Petter Selasky 	return (events & (POLLHUP | POLLPRI | POLLIN |
180795700043SHans Petter Selasky 	    POLLRDNORM | POLLOUT | POLLWRNORM));
1808fa0f6e62SHans Petter Selasky }
1809fa0f6e62SHans Petter Selasky 
1810fa0f6e62SHans Petter Selasky static int
181156c49c63SHans Petter Selasky cuse_client_mmap_single(struct cdev *dev, vm_ooffset_t *offset,
181256c49c63SHans Petter Selasky     vm_size_t size, struct vm_object **object, int nprot)
1813fa0f6e62SHans Petter Selasky {
181456c49c63SHans Petter Selasky 	uint32_t page_nr = *offset / PAGE_SIZE;
1815fa0f6e62SHans Petter Selasky 	uint32_t alloc_nr = page_nr / CUSE_ALLOC_PAGES_MAX;
1816fa0f6e62SHans Petter Selasky 	struct cuse_memory *mem;
1817fa0f6e62SHans Petter Selasky 	struct cuse_client *pcc;
181898029019SHans Petter Selasky 	struct cuse_server *pcs;
1819fa0f6e62SHans Petter Selasky 	int error;
1820fa0f6e62SHans Petter Selasky 
1821fa0f6e62SHans Petter Selasky 	error = cuse_client_get(&pcc);
1822fa0f6e62SHans Petter Selasky 	if (error != 0)
182356c49c63SHans Petter Selasky 		return (error);
1824fa0f6e62SHans Petter Selasky 
182598029019SHans Petter Selasky 	pcs = pcc->server;
182698029019SHans Petter Selasky 
182798029019SHans Petter Selasky 	cuse_server_lock(pcs);
182856c49c63SHans Petter Selasky 	/* lookup memory structure */
182998029019SHans Petter Selasky 	TAILQ_FOREACH(mem, &pcs->hmem, entry) {
183056c49c63SHans Petter Selasky 		if (mem->alloc_nr == alloc_nr)
183156c49c63SHans Petter Selasky 			break;
1832fa0f6e62SHans Petter Selasky 	}
183356c49c63SHans Petter Selasky 	if (mem == NULL) {
183498029019SHans Petter Selasky 		cuse_server_unlock(pcs);
1835fa0f6e62SHans Petter Selasky 		return (ENOMEM);
1836fa0f6e62SHans Petter Selasky 	}
183756c49c63SHans Petter Selasky 	/* verify page offset */
1838fa0f6e62SHans Petter Selasky 	page_nr %= CUSE_ALLOC_PAGES_MAX;
1839fa0f6e62SHans Petter Selasky 	if (page_nr >= mem->page_count) {
184098029019SHans Petter Selasky 		cuse_server_unlock(pcs);
1841fa0f6e62SHans Petter Selasky 		return (ENXIO);
1842fa0f6e62SHans Petter Selasky 	}
184356c49c63SHans Petter Selasky 	/* verify mmap size */
184456c49c63SHans Petter Selasky 	if ((size % PAGE_SIZE) != 0 || (size < PAGE_SIZE) ||
184556c49c63SHans Petter Selasky 	    (size > ((mem->page_count - page_nr) * PAGE_SIZE))) {
184698029019SHans Petter Selasky 		cuse_server_unlock(pcs);
184756c49c63SHans Petter Selasky 		return (EINVAL);
184856c49c63SHans Petter Selasky 	}
184956c49c63SHans Petter Selasky 	vm_object_reference(mem->object);
185056c49c63SHans Petter Selasky 	*object = mem->object;
185198029019SHans Petter Selasky 	cuse_server_unlock(pcs);
1852fa0f6e62SHans Petter Selasky 
185356c49c63SHans Petter Selasky 	/* set new VM object offset to use */
185456c49c63SHans Petter Selasky 	*offset = page_nr * PAGE_SIZE;
1855fa0f6e62SHans Petter Selasky 
185656c49c63SHans Petter Selasky 	/* success */
1857fa0f6e62SHans Petter Selasky 	return (0);
1858fa0f6e62SHans Petter Selasky }
1859fa0f6e62SHans Petter Selasky 
1860fa0f6e62SHans Petter Selasky static void
1861fa0f6e62SHans Petter Selasky cuse_client_kqfilter_read_detach(struct knote *kn)
1862fa0f6e62SHans Petter Selasky {
1863fa0f6e62SHans Petter Selasky 	struct cuse_client *pcc;
186498029019SHans Petter Selasky 	struct cuse_server *pcs;
1865fa0f6e62SHans Petter Selasky 
1866fa0f6e62SHans Petter Selasky 	pcc = kn->kn_hook;
186798029019SHans Petter Selasky 	pcs = pcc->server;
186898029019SHans Petter Selasky 
186998029019SHans Petter Selasky 	cuse_server_lock(pcs);
187098029019SHans Petter Selasky 	knlist_remove(&pcs->selinfo.si_note, kn, 1);
187198029019SHans Petter Selasky 	cuse_server_unlock(pcs);
1872fa0f6e62SHans Petter Selasky }
1873fa0f6e62SHans Petter Selasky 
1874fa0f6e62SHans Petter Selasky static void
1875fa0f6e62SHans Petter Selasky cuse_client_kqfilter_write_detach(struct knote *kn)
1876fa0f6e62SHans Petter Selasky {
1877fa0f6e62SHans Petter Selasky 	struct cuse_client *pcc;
187898029019SHans Petter Selasky 	struct cuse_server *pcs;
1879fa0f6e62SHans Petter Selasky 
1880fa0f6e62SHans Petter Selasky 	pcc = kn->kn_hook;
188198029019SHans Petter Selasky 	pcs = pcc->server;
188298029019SHans Petter Selasky 
188398029019SHans Petter Selasky 	cuse_server_lock(pcs);
188498029019SHans Petter Selasky 	knlist_remove(&pcs->selinfo.si_note, kn, 1);
188598029019SHans Petter Selasky 	cuse_server_unlock(pcs);
1886fa0f6e62SHans Petter Selasky }
1887fa0f6e62SHans Petter Selasky 
1888fa0f6e62SHans Petter Selasky static int
1889fa0f6e62SHans Petter Selasky cuse_client_kqfilter_read_event(struct knote *kn, long hint)
1890fa0f6e62SHans Petter Selasky {
1891fa0f6e62SHans Petter Selasky 	struct cuse_client *pcc;
1892fa0f6e62SHans Petter Selasky 
1893fa0f6e62SHans Petter Selasky 	pcc = kn->kn_hook;
189498029019SHans Petter Selasky 
189598029019SHans Petter Selasky 	mtx_assert(&pcc->server->mtx, MA_OWNED);
189698029019SHans Petter Selasky 
1897fa0f6e62SHans Petter Selasky 	return ((pcc->cflags & CUSE_CLI_KNOTE_NEED_READ) ? 1 : 0);
1898fa0f6e62SHans Petter Selasky }
1899fa0f6e62SHans Petter Selasky 
1900fa0f6e62SHans Petter Selasky static int
1901fa0f6e62SHans Petter Selasky cuse_client_kqfilter_write_event(struct knote *kn, long hint)
1902fa0f6e62SHans Petter Selasky {
1903fa0f6e62SHans Petter Selasky 	struct cuse_client *pcc;
1904fa0f6e62SHans Petter Selasky 
1905fa0f6e62SHans Petter Selasky 	pcc = kn->kn_hook;
190698029019SHans Petter Selasky 
190798029019SHans Petter Selasky 	mtx_assert(&pcc->server->mtx, MA_OWNED);
190898029019SHans Petter Selasky 
1909fa0f6e62SHans Petter Selasky 	return ((pcc->cflags & CUSE_CLI_KNOTE_NEED_WRITE) ? 1 : 0);
1910fa0f6e62SHans Petter Selasky }
1911fa0f6e62SHans Petter Selasky 
1912fa0f6e62SHans Petter Selasky static int
1913fa0f6e62SHans Petter Selasky cuse_client_kqfilter(struct cdev *dev, struct knote *kn)
1914fa0f6e62SHans Petter Selasky {
1915fa0f6e62SHans Petter Selasky 	struct cuse_client *pcc;
1916fa0f6e62SHans Petter Selasky 	struct cuse_server *pcs;
1917fa0f6e62SHans Petter Selasky 	int error;
1918fa0f6e62SHans Petter Selasky 
1919fa0f6e62SHans Petter Selasky 	error = cuse_client_get(&pcc);
1920fa0f6e62SHans Petter Selasky 	if (error != 0)
1921fa0f6e62SHans Petter Selasky 		return (error);
1922fa0f6e62SHans Petter Selasky 
1923fa0f6e62SHans Petter Selasky 	pcs = pcc->server;
192498029019SHans Petter Selasky 
192598029019SHans Petter Selasky 	cuse_server_lock(pcs);
1926fa0f6e62SHans Petter Selasky 	switch (kn->kn_filter) {
1927fa0f6e62SHans Petter Selasky 	case EVFILT_READ:
1928fa0f6e62SHans Petter Selasky 		pcc->cflags |= CUSE_CLI_KNOTE_HAS_READ;
1929fa0f6e62SHans Petter Selasky 		kn->kn_hook = pcc;
1930fa0f6e62SHans Petter Selasky 		kn->kn_fop = &cuse_client_kqfilter_read_ops;
1931fa0f6e62SHans Petter Selasky 		knlist_add(&pcs->selinfo.si_note, kn, 1);
1932fa0f6e62SHans Petter Selasky 		break;
1933fa0f6e62SHans Petter Selasky 	case EVFILT_WRITE:
1934fa0f6e62SHans Petter Selasky 		pcc->cflags |= CUSE_CLI_KNOTE_HAS_WRITE;
1935fa0f6e62SHans Petter Selasky 		kn->kn_hook = pcc;
1936fa0f6e62SHans Petter Selasky 		kn->kn_fop = &cuse_client_kqfilter_write_ops;
1937fa0f6e62SHans Petter Selasky 		knlist_add(&pcs->selinfo.si_note, kn, 1);
1938fa0f6e62SHans Petter Selasky 		break;
1939fa0f6e62SHans Petter Selasky 	default:
1940fa0f6e62SHans Petter Selasky 		error = EINVAL;
1941fa0f6e62SHans Petter Selasky 		break;
1942fa0f6e62SHans Petter Selasky 	}
194398029019SHans Petter Selasky 	cuse_server_unlock(pcs);
1944fa0f6e62SHans Petter Selasky 
1945fa0f6e62SHans Petter Selasky 	if (error == 0)
1946fa0f6e62SHans Petter Selasky 		cuse_client_kqfilter_poll(dev, pcc);
1947fa0f6e62SHans Petter Selasky 	return (error);
1948fa0f6e62SHans Petter Selasky }
1949