1fa0f6e62SHans Petter Selasky /* $FreeBSD$ */ 2fa0f6e62SHans Petter Selasky /*- 32c28cd09SHans 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 672c28cd09SHans Petter Selasky #define CUSE_ALLOC_PAGES_MAX \ 682c28cd09SHans Petter Selasky (CUSE_ALLOC_BYTES_MAX / PAGE_SIZE) 692c28cd09SHans Petter Selasky 702c28cd09SHans Petter Selasky #if (CUSE_ALLOC_PAGES_MAX == 0) 712c28cd09SHans Petter Selasky #error "PAGE_SIZE is too big!" 722c28cd09SHans Petter Selasky #endif 732c28cd09SHans 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 } 1162*0996dd7dSHans 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