xref: /freebsd/sys/kern/vfs_aio.c (revision ad8de0f2435d575c10b3849e70dce850e071fc88)
19454b2d8SWarner Losh /*-
2ee877a35SJohn Dyson  * Copyright (c) 1997 John S. Dyson.  All rights reserved.
3ee877a35SJohn Dyson  *
4ee877a35SJohn Dyson  * Redistribution and use in source and binary forms, with or without
5ee877a35SJohn Dyson  * modification, are permitted provided that the following conditions
6ee877a35SJohn Dyson  * are met:
7ee877a35SJohn Dyson  * 1. Redistributions of source code must retain the above copyright
8ee877a35SJohn Dyson  *    notice, this list of conditions and the following disclaimer.
9ee877a35SJohn Dyson  * 2. John S. Dyson's name may not be used to endorse or promote products
10ee877a35SJohn Dyson  *    derived from this software without specific prior written permission.
11ee877a35SJohn Dyson  *
12ee877a35SJohn Dyson  * DISCLAIMER:  This code isn't warranted to do anything useful.  Anything
13ee877a35SJohn Dyson  * bad that happens because of using this software isn't the responsibility
14ee877a35SJohn Dyson  * of the author.  This software is distributed AS-IS.
15ee877a35SJohn Dyson  */
16ee877a35SJohn Dyson 
17ee877a35SJohn Dyson /*
188a6472b7SPeter Dufault  * This file contains support for the POSIX 1003.1B AIO/LIO facility.
19ee877a35SJohn Dyson  */
20ee877a35SJohn Dyson 
21677b542eSDavid E. O'Brien #include <sys/cdefs.h>
22677b542eSDavid E. O'Brien __FBSDID("$FreeBSD$");
23677b542eSDavid E. O'Brien 
24ee877a35SJohn Dyson #include <sys/param.h>
25ee877a35SJohn Dyson #include <sys/systm.h>
26f591779bSSeigo Tanimura #include <sys/malloc.h>
279626b608SPoul-Henning Kamp #include <sys/bio.h>
28a5c9bce7SBruce Evans #include <sys/buf.h>
2975b8b3b2SJohn Baldwin #include <sys/eventhandler.h>
30ee877a35SJohn Dyson #include <sys/sysproto.h>
31ee877a35SJohn Dyson #include <sys/filedesc.h>
32ee877a35SJohn Dyson #include <sys/kernel.h>
3377409fe1SPoul-Henning Kamp #include <sys/module.h>
34c9a970a7SAlan Cox #include <sys/kthread.h>
35ee877a35SJohn Dyson #include <sys/fcntl.h>
36ee877a35SJohn Dyson #include <sys/file.h>
37104a9b7eSAlexander Kabaev #include <sys/limits.h>
38fdebd4f0SBruce Evans #include <sys/lock.h>
3935e0e5b3SJohn Baldwin #include <sys/mutex.h>
40ee877a35SJohn Dyson #include <sys/unistd.h>
41ee877a35SJohn Dyson #include <sys/proc.h>
422d2f8ae7SBruce Evans #include <sys/resourcevar.h>
43ee877a35SJohn Dyson #include <sys/signalvar.h>
44bfbbc4aaSJason Evans #include <sys/protosw.h>
451ce91824SDavid Xu #include <sys/sema.h>
461ce91824SDavid Xu #include <sys/socket.h>
47bfbbc4aaSJason Evans #include <sys/socketvar.h>
4821d56e9cSAlfred Perlstein #include <sys/syscall.h>
4921d56e9cSAlfred Perlstein #include <sys/sysent.h>
50a624e84fSJohn Dyson #include <sys/sysctl.h>
51ee99e978SBruce Evans #include <sys/sx.h>
521ce91824SDavid Xu #include <sys/taskqueue.h>
53fd3bf775SJohn Dyson #include <sys/vnode.h>
54fd3bf775SJohn Dyson #include <sys/conf.h>
55cb679c38SJonathan Lemon #include <sys/event.h>
56ee877a35SJohn Dyson 
571ce91824SDavid Xu #include <machine/atomic.h>
581ce91824SDavid Xu 
59c844abc9SAlfred Perlstein #include <posix4/posix4.h>
60ee877a35SJohn Dyson #include <vm/vm.h>
61ee877a35SJohn Dyson #include <vm/vm_extern.h>
622244ea07SJohn Dyson #include <vm/pmap.h>
632244ea07SJohn Dyson #include <vm/vm_map.h>
64c897b813SJeff Roberson #include <vm/uma.h>
65ee877a35SJohn Dyson #include <sys/aio.h>
665aaef07cSJohn Dyson 
67dd85920aSJason Evans #include "opt_vfs_aio.h"
68ee877a35SJohn Dyson 
69eb8e6d52SEivind Eklund /*
70eb8e6d52SEivind Eklund  * Counter for allocating reference ids to new jobs.  Wrapped to 1 on
71eb8e6d52SEivind Eklund  * overflow.
72eb8e6d52SEivind Eklund  */
738c12612cSDoug Rabson static	long jobrefid;
742244ea07SJohn Dyson 
752244ea07SJohn Dyson #define JOBST_NULL		0x0
761ce91824SDavid Xu #define JOBST_JOBQSOCK		0x1
772244ea07SJohn Dyson #define JOBST_JOBQGLOBAL	0x2
782244ea07SJohn Dyson #define JOBST_JOBRUNNING	0x3
792244ea07SJohn Dyson #define JOBST_JOBFINISHED	0x4
80fd3bf775SJohn Dyson #define	JOBST_JOBQBUF		0x5
812244ea07SJohn Dyson 
8284af4da6SJohn Dyson #ifndef MAX_AIO_PER_PROC
832244ea07SJohn Dyson #define MAX_AIO_PER_PROC	32
8484af4da6SJohn Dyson #endif
8584af4da6SJohn Dyson 
8684af4da6SJohn Dyson #ifndef MAX_AIO_QUEUE_PER_PROC
872244ea07SJohn Dyson #define MAX_AIO_QUEUE_PER_PROC	256 /* Bigger than AIO_LISTIO_MAX */
8884af4da6SJohn Dyson #endif
8984af4da6SJohn Dyson 
9084af4da6SJohn Dyson #ifndef MAX_AIO_PROCS
91fd3bf775SJohn Dyson #define MAX_AIO_PROCS		32
9284af4da6SJohn Dyson #endif
9384af4da6SJohn Dyson 
9484af4da6SJohn Dyson #ifndef MAX_AIO_QUEUE
952244ea07SJohn Dyson #define	MAX_AIO_QUEUE		1024 /* Bigger than AIO_LISTIO_MAX */
9684af4da6SJohn Dyson #endif
9784af4da6SJohn Dyson 
9884af4da6SJohn Dyson #ifndef TARGET_AIO_PROCS
99bfbbc4aaSJason Evans #define TARGET_AIO_PROCS	4
10084af4da6SJohn Dyson #endif
10184af4da6SJohn Dyson 
10284af4da6SJohn Dyson #ifndef MAX_BUF_AIO
10384af4da6SJohn Dyson #define MAX_BUF_AIO		16
10484af4da6SJohn Dyson #endif
10584af4da6SJohn Dyson 
10684af4da6SJohn Dyson #ifndef AIOD_TIMEOUT_DEFAULT
10784af4da6SJohn Dyson #define	AIOD_TIMEOUT_DEFAULT	(10 * hz)
10884af4da6SJohn Dyson #endif
10984af4da6SJohn Dyson 
11084af4da6SJohn Dyson #ifndef AIOD_LIFETIME_DEFAULT
11184af4da6SJohn Dyson #define AIOD_LIFETIME_DEFAULT	(30 * hz)
11284af4da6SJohn Dyson #endif
1132244ea07SJohn Dyson 
1145ece08f5SPoul-Henning Kamp static SYSCTL_NODE(_vfs, OID_AUTO, aio, CTLFLAG_RW, 0, "Async IO management");
115eb8e6d52SEivind Eklund 
116303b270bSEivind Eklund static int max_aio_procs = MAX_AIO_PROCS;
117a624e84fSJohn Dyson SYSCTL_INT(_vfs_aio, OID_AUTO, max_aio_procs,
118eb8e6d52SEivind Eklund 	CTLFLAG_RW, &max_aio_procs, 0,
119eb8e6d52SEivind Eklund 	"Maximum number of kernel threads to use for handling async IO ");
120a624e84fSJohn Dyson 
121eb8e6d52SEivind Eklund static int num_aio_procs = 0;
122a624e84fSJohn Dyson SYSCTL_INT(_vfs_aio, OID_AUTO, num_aio_procs,
123eb8e6d52SEivind Eklund 	CTLFLAG_RD, &num_aio_procs, 0,
124eb8e6d52SEivind Eklund 	"Number of presently active kernel threads for async IO");
125a624e84fSJohn Dyson 
126eb8e6d52SEivind Eklund /*
127eb8e6d52SEivind Eklund  * The code will adjust the actual number of AIO processes towards this
128eb8e6d52SEivind Eklund  * number when it gets a chance.
129eb8e6d52SEivind Eklund  */
130eb8e6d52SEivind Eklund static int target_aio_procs = TARGET_AIO_PROCS;
131eb8e6d52SEivind Eklund SYSCTL_INT(_vfs_aio, OID_AUTO, target_aio_procs, CTLFLAG_RW, &target_aio_procs,
132eb8e6d52SEivind Eklund 	0, "Preferred number of ready kernel threads for async IO");
133a624e84fSJohn Dyson 
134eb8e6d52SEivind Eklund static int max_queue_count = MAX_AIO_QUEUE;
135eb8e6d52SEivind Eklund SYSCTL_INT(_vfs_aio, OID_AUTO, max_aio_queue, CTLFLAG_RW, &max_queue_count, 0,
136eb8e6d52SEivind Eklund     "Maximum number of aio requests to queue, globally");
137a624e84fSJohn Dyson 
138eb8e6d52SEivind Eklund static int num_queue_count = 0;
139eb8e6d52SEivind Eklund SYSCTL_INT(_vfs_aio, OID_AUTO, num_queue_count, CTLFLAG_RD, &num_queue_count, 0,
140eb8e6d52SEivind Eklund     "Number of queued aio requests");
141a624e84fSJohn Dyson 
142eb8e6d52SEivind Eklund static int num_buf_aio = 0;
143eb8e6d52SEivind Eklund SYSCTL_INT(_vfs_aio, OID_AUTO, num_buf_aio, CTLFLAG_RD, &num_buf_aio, 0,
144eb8e6d52SEivind Eklund     "Number of aio requests presently handled by the buf subsystem");
145fd3bf775SJohn Dyson 
146eb8e6d52SEivind Eklund /* Number of async I/O thread in the process of being started */
147a9bf5e37SDavid Xu /* XXX This should be local to aio_aqueue() */
148eb8e6d52SEivind Eklund static int num_aio_resv_start = 0;
149fd3bf775SJohn Dyson 
150eb8e6d52SEivind Eklund static int aiod_timeout;
151eb8e6d52SEivind Eklund SYSCTL_INT(_vfs_aio, OID_AUTO, aiod_timeout, CTLFLAG_RW, &aiod_timeout, 0,
152eb8e6d52SEivind Eklund     "Timeout value for synchronous aio operations");
15384af4da6SJohn Dyson 
154eb8e6d52SEivind Eklund static int aiod_lifetime;
155eb8e6d52SEivind Eklund SYSCTL_INT(_vfs_aio, OID_AUTO, aiod_lifetime, CTLFLAG_RW, &aiod_lifetime, 0,
156eb8e6d52SEivind Eklund     "Maximum lifetime for idle aiod");
15784af4da6SJohn Dyson 
158eb8e6d52SEivind Eklund static int unloadable = 0;
15921d56e9cSAlfred Perlstein SYSCTL_INT(_vfs_aio, OID_AUTO, unloadable, CTLFLAG_RW, &unloadable, 0,
16021d56e9cSAlfred Perlstein     "Allow unload of aio (not recommended)");
16121d56e9cSAlfred Perlstein 
162eb8e6d52SEivind Eklund 
163eb8e6d52SEivind Eklund static int max_aio_per_proc = MAX_AIO_PER_PROC;
164eb8e6d52SEivind Eklund SYSCTL_INT(_vfs_aio, OID_AUTO, max_aio_per_proc, CTLFLAG_RW, &max_aio_per_proc,
165eb8e6d52SEivind Eklund     0, "Maximum active aio requests per process (stored in the process)");
166eb8e6d52SEivind Eklund 
167eb8e6d52SEivind Eklund static int max_aio_queue_per_proc = MAX_AIO_QUEUE_PER_PROC;
168eb8e6d52SEivind Eklund SYSCTL_INT(_vfs_aio, OID_AUTO, max_aio_queue_per_proc, CTLFLAG_RW,
169eb8e6d52SEivind Eklund     &max_aio_queue_per_proc, 0,
170eb8e6d52SEivind Eklund     "Maximum queued aio requests per process (stored in the process)");
171eb8e6d52SEivind Eklund 
172eb8e6d52SEivind Eklund static int max_buf_aio = MAX_BUF_AIO;
173eb8e6d52SEivind Eklund SYSCTL_INT(_vfs_aio, OID_AUTO, max_buf_aio, CTLFLAG_RW, &max_buf_aio, 0,
174eb8e6d52SEivind Eklund     "Maximum buf aio requests per process (stored in the process)");
175eb8e6d52SEivind Eklund 
1760972628aSDavid Xu typedef struct oaiocb {
1770972628aSDavid Xu 	int	aio_fildes;		/* File descriptor */
1780972628aSDavid Xu 	off_t	aio_offset;		/* File offset for I/O */
1790972628aSDavid Xu 	volatile void *aio_buf;         /* I/O buffer in process space */
1800972628aSDavid Xu 	size_t	aio_nbytes;		/* Number of bytes for I/O */
1810972628aSDavid Xu 	struct	osigevent aio_sigevent;	/* Signal to deliver */
1820972628aSDavid Xu 	int	aio_lio_opcode;		/* LIO opcode */
1830972628aSDavid Xu 	int	aio_reqprio;		/* Request priority -- ignored */
1840972628aSDavid Xu 	struct	__aiocb_private	_aiocb_private;
1850972628aSDavid Xu } oaiocb_t;
1860972628aSDavid Xu 
1871aa4c324SDavid Xu /*
1881aa4c324SDavid Xu  * Below is a key of locks used to protect each member of struct aiocblist
1891aa4c324SDavid Xu  * aioliojob and kaioinfo and any backends.
1901aa4c324SDavid Xu  *
1911aa4c324SDavid Xu  * * - need not protected
1921aa4c324SDavid Xu  * a - locked by proc mtx
1931aa4c324SDavid Xu  * b - locked by backend lock, the backend lock can be null in some cases,
1941aa4c324SDavid Xu  *     for example, BIO belongs to this type, in this case, proc lock is
1951aa4c324SDavid Xu  *     reused.
1961aa4c324SDavid Xu  * c - locked by aio_job_mtx, the lock for the generic file I/O backend.
1971aa4c324SDavid Xu  */
1981aa4c324SDavid Xu 
1991aa4c324SDavid Xu /*
2001aa4c324SDavid Xu  * Current, there is only two backends: BIO and generic file I/O.
2011aa4c324SDavid Xu  * socket I/O is served by generic file I/O, this is not a good idea, since
2021aa4c324SDavid Xu  * disk file I/O and any other types without O_NONBLOCK flag can block daemon
2031aa4c324SDavid Xu  * threads, if there is no thread to serve socket I/O, the socket I/O will be
2041aa4c324SDavid Xu  * delayed too long or starved, we should create some threads dedicated to
2051aa4c324SDavid Xu  * sockets to do non-blocking I/O, same for pipe and fifo, for these I/O
2061aa4c324SDavid Xu  * systems we really need non-blocking interface, fiddling O_NONBLOCK in file
2071aa4c324SDavid Xu  * structure is not safe because there is race between userland and aio
2081aa4c324SDavid Xu  * daemons.
2091aa4c324SDavid Xu  */
2101aa4c324SDavid Xu 
21148dac059SAlan Cox struct aiocblist {
2121aa4c324SDavid Xu 	TAILQ_ENTRY(aiocblist) list;	/* (b) internal list of for backend */
2131aa4c324SDavid Xu 	TAILQ_ENTRY(aiocblist) plist;	/* (a) list of jobs for each backend */
2141aa4c324SDavid Xu 	TAILQ_ENTRY(aiocblist) allist;  /* (a) list of all jobs in proc */
2151aa4c324SDavid Xu 	int	jobflags;		/* (a) job flags */
2161aa4c324SDavid Xu 	int	jobstate;		/* (b) job state */
2171aa4c324SDavid Xu 	int	inputcharge;		/* (*) input blockes */
2181aa4c324SDavid Xu 	int	outputcharge;		/* (*) output blockes */
2191aa4c324SDavid Xu 	struct	buf *bp;		/* (*) private to BIO backend,
2201aa4c324SDavid Xu 				  	 * buffer pointer
2211aa4c324SDavid Xu 					 */
2221aa4c324SDavid Xu 	struct	proc *userproc;		/* (*) user process */
2231aa4c324SDavid Xu 	struct  ucred *cred;		/* (*) active credential when created */
2241aa4c324SDavid Xu 	struct	file *fd_file;		/* (*) pointer to file structure */
2251aa4c324SDavid Xu 	struct	aioliojob *lio;		/* (*) optional lio job */
2261aa4c324SDavid Xu 	struct	aiocb *uuaiocb;		/* (*) pointer in userspace of aiocb */
2271aa4c324SDavid Xu 	struct	knlist klist;		/* (a) list of knotes */
2281aa4c324SDavid Xu 	struct	aiocb uaiocb;		/* (*) kernel I/O control block */
2291aa4c324SDavid Xu 	ksiginfo_t ksi;			/* (a) realtime signal info */
2301aa4c324SDavid Xu 	struct task	biotask;	/* (*) private to BIO backend */
23148dac059SAlan Cox };
23248dac059SAlan Cox 
23348dac059SAlan Cox /* jobflags */
2341ce91824SDavid Xu #define AIOCBLIST_RUNDOWN	0x04
23548dac059SAlan Cox #define AIOCBLIST_DONE		0x10
2361ce91824SDavid Xu #define AIOCBLIST_BUFDONE	0x20
23748dac059SAlan Cox 
2382244ea07SJohn Dyson /*
2392244ea07SJohn Dyson  * AIO process info
2402244ea07SJohn Dyson  */
24184af4da6SJohn Dyson #define AIOP_FREE	0x1			/* proc on free queue */
24284af4da6SJohn Dyson 
243b40ce416SJulian Elischer struct aiothreadlist {
2441aa4c324SDavid Xu 	int aiothreadflags;			/* (c) AIO proc flags */
2451aa4c324SDavid Xu 	TAILQ_ENTRY(aiothreadlist) list;	/* (c) list of processes */
2461aa4c324SDavid Xu 	struct thread *aiothread;		/* (*) the AIO thread */
2472244ea07SJohn Dyson };
2482244ea07SJohn Dyson 
24984af4da6SJohn Dyson /*
25084af4da6SJohn Dyson  * data-structure for lio signal management
25184af4da6SJohn Dyson  */
2521ce91824SDavid Xu struct aioliojob {
2531aa4c324SDavid Xu 	int	lioj_flags;			/* (a) listio flags */
2541aa4c324SDavid Xu 	int	lioj_count;			/* (a) listio flags */
2551aa4c324SDavid Xu 	int	lioj_finished_count;		/* (a) listio flags */
2561aa4c324SDavid Xu 	struct	sigevent lioj_signal;		/* (a) signal on all I/O done */
2571aa4c324SDavid Xu 	TAILQ_ENTRY(aioliojob) lioj_list;	/* (a) lio list */
2581aa4c324SDavid Xu 	struct  knlist klist;			/* (a) list of knotes */
2591aa4c324SDavid Xu 	ksiginfo_t lioj_ksi;			/* (a) Realtime signal info */
26084af4da6SJohn Dyson };
2611ce91824SDavid Xu 
26284af4da6SJohn Dyson #define	LIOJ_SIGNAL		0x1	/* signal on all done (lio) */
26384af4da6SJohn Dyson #define	LIOJ_SIGNAL_POSTED	0x2	/* signal has been posted */
26469cd28daSDoug Ambrisko #define LIOJ_KEVENT_POSTED	0x4	/* kevent triggered */
26584af4da6SJohn Dyson 
26684af4da6SJohn Dyson /*
26784af4da6SJohn Dyson  * per process aio data structure
26884af4da6SJohn Dyson  */
2692244ea07SJohn Dyson struct kaioinfo {
2701aa4c324SDavid Xu 	int	kaio_flags;		/* (a) per process kaio flags */
2711aa4c324SDavid Xu 	int	kaio_maxactive_count;	/* (*) maximum number of AIOs */
2721aa4c324SDavid Xu 	int	kaio_active_count;	/* (c) number of currently used AIOs */
2731aa4c324SDavid Xu 	int	kaio_qallowed_count;	/* (*) maxiumu size of AIO queue */
2741aa4c324SDavid Xu 	int	kaio_count;		/* (a) size of AIO queue */
2751aa4c324SDavid Xu 	int	kaio_ballowed_count;	/* (*) maximum number of buffers */
2761aa4c324SDavid Xu 	int	kaio_buffer_count;	/* (a) number of physio buffers */
2771aa4c324SDavid Xu 	TAILQ_HEAD(,aiocblist) kaio_all;	/* (a) all AIOs in the process */
2781aa4c324SDavid Xu 	TAILQ_HEAD(,aiocblist) kaio_done;	/* (a) done queue for process */
2791aa4c324SDavid Xu 	TAILQ_HEAD(,aioliojob) kaio_liojoblist; /* (a) list of lio jobs */
2801aa4c324SDavid Xu 	TAILQ_HEAD(,aiocblist) kaio_jobqueue;	/* (a) job queue for process */
2811aa4c324SDavid Xu 	TAILQ_HEAD(,aiocblist) kaio_bufqueue;	/* (a) buffer job queue for process */
2821aa4c324SDavid Xu 	TAILQ_HEAD(,aiocblist) kaio_sockqueue;  /* (a) queue for aios waiting on sockets,
2831aa4c324SDavid Xu 						 *  not used yet.
2841aa4c324SDavid Xu 						 */
2852244ea07SJohn Dyson };
2862244ea07SJohn Dyson 
28784af4da6SJohn Dyson #define KAIO_RUNDOWN	0x1	/* process is being run down */
288bfbbc4aaSJason Evans #define KAIO_WAKEUP	0x2	/* wakeup process when there is a significant event */
289fd3bf775SJohn Dyson 
2901aa4c324SDavid Xu static TAILQ_HEAD(,aiothreadlist) aio_freeproc;		/* (c) Idle daemons */
2911ce91824SDavid Xu static struct sema aio_newproc_sem;
2921ce91824SDavid Xu static struct mtx aio_job_mtx;
2931ce91824SDavid Xu static struct mtx aio_sock_mtx;
2941aa4c324SDavid Xu static TAILQ_HEAD(,aiocblist) aio_jobs;			/* (c) Async job list */
2951ce91824SDavid Xu static struct unrhdr *aiod_unr;
2962244ea07SJohn Dyson 
297fd3bf775SJohn Dyson static void	aio_init_aioinfo(struct proc *p);
29821d56e9cSAlfred Perlstein static void	aio_onceonly(void);
299fd3bf775SJohn Dyson static int	aio_free_entry(struct aiocblist *aiocbe);
300fd3bf775SJohn Dyson static void	aio_process(struct aiocblist *aiocbe);
3011ce91824SDavid Xu static int	aio_newproc(int *);
302a9bf5e37SDavid Xu static int	aio_aqueue(struct thread *td, struct aiocb *job,
303a9bf5e37SDavid Xu 			struct aioliojob *lio, int type, int osigev);
304fd3bf775SJohn Dyson static void	aio_physwakeup(struct buf *bp);
30575b8b3b2SJohn Baldwin static void	aio_proc_rundown(void *arg, struct proc *p);
306fd3bf775SJohn Dyson static int	aio_qphysio(struct proc *p, struct aiocblist *iocb);
3071ce91824SDavid Xu static void	biohelper(void *, int);
3081ce91824SDavid Xu static void	aio_daemon(void *param);
30948dac059SAlan Cox static void	aio_swake_cb(struct socket *, struct sockbuf *);
31021d56e9cSAlfred Perlstein static int	aio_unload(void);
31121d56e9cSAlfred Perlstein static int	filt_aioattach(struct knote *kn);
31221d56e9cSAlfred Perlstein static void	filt_aiodetach(struct knote *kn);
31321d56e9cSAlfred Perlstein static int	filt_aio(struct knote *kn, long hint);
31469cd28daSDoug Ambrisko static int	filt_lioattach(struct knote *kn);
31569cd28daSDoug Ambrisko static void	filt_liodetach(struct knote *kn);
31669cd28daSDoug Ambrisko static int	filt_lio(struct knote *kn, long hint);
31769cd28daSDoug Ambrisko #define DONE_BUF 1
31869cd28daSDoug Ambrisko #define DONE_QUEUE 2
31969cd28daSDoug Ambrisko static void	aio_bio_done_notify( struct proc *userp, struct aiocblist *aiocbe, int type);
3200972628aSDavid Xu static int	do_lio_listio(struct thread *td, struct lio_listio_args *uap,
3210972628aSDavid Xu 			int oldsigev);
3222244ea07SJohn Dyson 
323eb8e6d52SEivind Eklund /*
324eb8e6d52SEivind Eklund  * Zones for:
325eb8e6d52SEivind Eklund  * 	kaio	Per process async io info
326eb8e6d52SEivind Eklund  *	aiop	async io thread data
327eb8e6d52SEivind Eklund  *	aiocb	async io jobs
328eb8e6d52SEivind Eklund  *	aiol	list io job pointer - internal to aio_suspend XXX
329eb8e6d52SEivind Eklund  *	aiolio	list io jobs
330eb8e6d52SEivind Eklund  */
331c897b813SJeff Roberson static uma_zone_t kaio_zone, aiop_zone, aiocb_zone, aiol_zone, aiolio_zone;
332fd3bf775SJohn Dyson 
333eb8e6d52SEivind Eklund /* kqueue filters for aio */
33421d56e9cSAlfred Perlstein static struct filterops aio_filtops =
33521d56e9cSAlfred Perlstein 	{ 0, filt_aioattach, filt_aiodetach, filt_aio };
33669cd28daSDoug Ambrisko static struct filterops lio_filtops =
33769cd28daSDoug Ambrisko 	{ 0, filt_lioattach, filt_liodetach, filt_lio };
33821d56e9cSAlfred Perlstein 
33975b8b3b2SJohn Baldwin static eventhandler_tag exit_tag, exec_tag;
34075b8b3b2SJohn Baldwin 
3411ce91824SDavid Xu TASKQUEUE_DEFINE_THREAD(aiod_bio);
3421ce91824SDavid Xu 
343eb8e6d52SEivind Eklund /*
344eb8e6d52SEivind Eklund  * Main operations function for use as a kernel module.
345eb8e6d52SEivind Eklund  */
34621d56e9cSAlfred Perlstein static int
34721d56e9cSAlfred Perlstein aio_modload(struct module *module, int cmd, void *arg)
34821d56e9cSAlfred Perlstein {
34921d56e9cSAlfred Perlstein 	int error = 0;
35021d56e9cSAlfred Perlstein 
35121d56e9cSAlfred Perlstein 	switch (cmd) {
35221d56e9cSAlfred Perlstein 	case MOD_LOAD:
35321d56e9cSAlfred Perlstein 		aio_onceonly();
35421d56e9cSAlfred Perlstein 		break;
35521d56e9cSAlfred Perlstein 	case MOD_UNLOAD:
35621d56e9cSAlfred Perlstein 		error = aio_unload();
35721d56e9cSAlfred Perlstein 		break;
35821d56e9cSAlfred Perlstein 	case MOD_SHUTDOWN:
35921d56e9cSAlfred Perlstein 		break;
36021d56e9cSAlfred Perlstein 	default:
36121d56e9cSAlfred Perlstein 		error = EINVAL;
36221d56e9cSAlfred Perlstein 		break;
36321d56e9cSAlfred Perlstein 	}
36421d56e9cSAlfred Perlstein 	return (error);
36521d56e9cSAlfred Perlstein }
36621d56e9cSAlfred Perlstein 
36721d56e9cSAlfred Perlstein static moduledata_t aio_mod = {
36821d56e9cSAlfred Perlstein 	"aio",
36921d56e9cSAlfred Perlstein 	&aio_modload,
37021d56e9cSAlfred Perlstein 	NULL
37121d56e9cSAlfred Perlstein };
37221d56e9cSAlfred Perlstein 
37321d56e9cSAlfred Perlstein SYSCALL_MODULE_HELPER(aio_return);
37421d56e9cSAlfred Perlstein SYSCALL_MODULE_HELPER(aio_suspend);
37521d56e9cSAlfred Perlstein SYSCALL_MODULE_HELPER(aio_cancel);
37621d56e9cSAlfred Perlstein SYSCALL_MODULE_HELPER(aio_error);
37721d56e9cSAlfred Perlstein SYSCALL_MODULE_HELPER(aio_read);
37821d56e9cSAlfred Perlstein SYSCALL_MODULE_HELPER(aio_write);
37921d56e9cSAlfred Perlstein SYSCALL_MODULE_HELPER(aio_waitcomplete);
38021d56e9cSAlfred Perlstein SYSCALL_MODULE_HELPER(lio_listio);
3810972628aSDavid Xu SYSCALL_MODULE_HELPER(oaio_read);
3820972628aSDavid Xu SYSCALL_MODULE_HELPER(oaio_write);
3830972628aSDavid Xu SYSCALL_MODULE_HELPER(olio_listio);
38421d56e9cSAlfred Perlstein 
38521d56e9cSAlfred Perlstein DECLARE_MODULE(aio, aio_mod,
38621d56e9cSAlfred Perlstein 	SI_SUB_VFS, SI_ORDER_ANY);
38721d56e9cSAlfred Perlstein MODULE_VERSION(aio, 1);
38821d56e9cSAlfred Perlstein 
389fd3bf775SJohn Dyson /*
3902244ea07SJohn Dyson  * Startup initialization
3912244ea07SJohn Dyson  */
39288ed460eSAlan Cox static void
39321d56e9cSAlfred Perlstein aio_onceonly(void)
394fd3bf775SJohn Dyson {
39521d56e9cSAlfred Perlstein 
39621d56e9cSAlfred Perlstein 	/* XXX: should probably just use so->callback */
39721d56e9cSAlfred Perlstein 	aio_swake = &aio_swake_cb;
39875b8b3b2SJohn Baldwin 	exit_tag = EVENTHANDLER_REGISTER(process_exit, aio_proc_rundown, NULL,
39975b8b3b2SJohn Baldwin 	    EVENTHANDLER_PRI_ANY);
40075b8b3b2SJohn Baldwin 	exec_tag = EVENTHANDLER_REGISTER(process_exec, aio_proc_rundown, NULL,
40175b8b3b2SJohn Baldwin 	    EVENTHANDLER_PRI_ANY);
40221d56e9cSAlfred Perlstein 	kqueue_add_filteropts(EVFILT_AIO, &aio_filtops);
40369cd28daSDoug Ambrisko 	kqueue_add_filteropts(EVFILT_LIO, &lio_filtops);
4042244ea07SJohn Dyson 	TAILQ_INIT(&aio_freeproc);
4051ce91824SDavid Xu 	sema_init(&aio_newproc_sem, 0, "aio_new_proc");
4061ce91824SDavid Xu 	mtx_init(&aio_job_mtx, "aio_job", NULL, MTX_DEF);
4071ce91824SDavid Xu 	mtx_init(&aio_sock_mtx, "aio_sock", NULL, MTX_DEF);
4082244ea07SJohn Dyson 	TAILQ_INIT(&aio_jobs);
4091ce91824SDavid Xu 	aiod_unr = new_unrhdr(1, INT_MAX, NULL);
410c897b813SJeff Roberson 	kaio_zone = uma_zcreate("AIO", sizeof(struct kaioinfo), NULL, NULL,
411c897b813SJeff Roberson 	    NULL, NULL, UMA_ALIGN_PTR, UMA_ZONE_NOFREE);
412c897b813SJeff Roberson 	aiop_zone = uma_zcreate("AIOP", sizeof(struct aiothreadlist), NULL,
413c897b813SJeff Roberson 	    NULL, NULL, NULL, UMA_ALIGN_PTR, UMA_ZONE_NOFREE);
414c897b813SJeff Roberson 	aiocb_zone = uma_zcreate("AIOCB", sizeof(struct aiocblist), NULL, NULL,
415c897b813SJeff Roberson 	    NULL, NULL, UMA_ALIGN_PTR, UMA_ZONE_NOFREE);
416c897b813SJeff Roberson 	aiol_zone = uma_zcreate("AIOL", AIO_LISTIO_MAX*sizeof(intptr_t) , NULL,
417c897b813SJeff Roberson 	    NULL, NULL, NULL, UMA_ALIGN_PTR, UMA_ZONE_NOFREE);
4181ce91824SDavid Xu 	aiolio_zone = uma_zcreate("AIOLIO", sizeof(struct aioliojob), NULL,
419c897b813SJeff Roberson 	    NULL, NULL, NULL, UMA_ALIGN_PTR, UMA_ZONE_NOFREE);
42084af4da6SJohn Dyson 	aiod_timeout = AIOD_TIMEOUT_DEFAULT;
42184af4da6SJohn Dyson 	aiod_lifetime = AIOD_LIFETIME_DEFAULT;
422fd3bf775SJohn Dyson 	jobrefid = 1;
423c7047e52SGarrett Wollman 	async_io_version = _POSIX_VERSION;
424c844abc9SAlfred Perlstein 	p31b_setcfg(CTL_P1003_1B_AIO_LISTIO_MAX, AIO_LISTIO_MAX);
42586d52125SAlfred Perlstein 	p31b_setcfg(CTL_P1003_1B_AIO_MAX, MAX_AIO_QUEUE);
42686d52125SAlfred Perlstein 	p31b_setcfg(CTL_P1003_1B_AIO_PRIO_DELTA_MAX, 0);
4272244ea07SJohn Dyson }
4282244ea07SJohn Dyson 
429eb8e6d52SEivind Eklund /*
430eb8e6d52SEivind Eklund  * Callback for unload of AIO when used as a module.
431eb8e6d52SEivind Eklund  */
43221d56e9cSAlfred Perlstein static int
43321d56e9cSAlfred Perlstein aio_unload(void)
43421d56e9cSAlfred Perlstein {
435ad3b9257SJohn-Mark Gurney 	int error;
43621d56e9cSAlfred Perlstein 
43721d56e9cSAlfred Perlstein 	/*
43821d56e9cSAlfred Perlstein 	 * XXX: no unloads by default, it's too dangerous.
43921d56e9cSAlfred Perlstein 	 * perhaps we could do it if locked out callers and then
44021d56e9cSAlfred Perlstein 	 * did an aio_proc_rundown() on each process.
4412a522eb9SJohn Baldwin 	 *
4422a522eb9SJohn Baldwin 	 * jhb: aio_proc_rundown() needs to run on curproc though,
4432a522eb9SJohn Baldwin 	 * so I don't think that would fly.
44421d56e9cSAlfred Perlstein 	 */
44521d56e9cSAlfred Perlstein 	if (!unloadable)
44621d56e9cSAlfred Perlstein 		return (EOPNOTSUPP);
44721d56e9cSAlfred Perlstein 
448ad3b9257SJohn-Mark Gurney 	error = kqueue_del_filteropts(EVFILT_AIO);
449ad3b9257SJohn-Mark Gurney 	if (error)
450ad3b9257SJohn-Mark Gurney 		return error;
451bd793be3SDavid Xu 	error = kqueue_del_filteropts(EVFILT_LIO);
452bd793be3SDavid Xu 	if (error)
453bd793be3SDavid Xu 		return error;
454c7047e52SGarrett Wollman 	async_io_version = 0;
45521d56e9cSAlfred Perlstein 	aio_swake = NULL;
4561ce91824SDavid Xu 	taskqueue_free(taskqueue_aiod_bio);
4571ce91824SDavid Xu 	delete_unrhdr(aiod_unr);
458bd793be3SDavid Xu 	uma_zdestroy(kaio_zone);
459bd793be3SDavid Xu 	uma_zdestroy(aiop_zone);
460bd793be3SDavid Xu 	uma_zdestroy(aiocb_zone);
461bd793be3SDavid Xu 	uma_zdestroy(aiol_zone);
462bd793be3SDavid Xu 	uma_zdestroy(aiolio_zone);
46375b8b3b2SJohn Baldwin 	EVENTHANDLER_DEREGISTER(process_exit, exit_tag);
46475b8b3b2SJohn Baldwin 	EVENTHANDLER_DEREGISTER(process_exec, exec_tag);
4651ce91824SDavid Xu 	mtx_destroy(&aio_job_mtx);
4661ce91824SDavid Xu 	mtx_destroy(&aio_sock_mtx);
4671ce91824SDavid Xu 	sema_destroy(&aio_newproc_sem);
468f51c1e89SAlfred Perlstein 	p31b_setcfg(CTL_P1003_1B_AIO_LISTIO_MAX, -1);
469f51c1e89SAlfred Perlstein 	p31b_setcfg(CTL_P1003_1B_AIO_MAX, -1);
470f51c1e89SAlfred Perlstein 	p31b_setcfg(CTL_P1003_1B_AIO_PRIO_DELTA_MAX, -1);
47121d56e9cSAlfred Perlstein 	return (0);
47221d56e9cSAlfred Perlstein }
47321d56e9cSAlfred Perlstein 
4742244ea07SJohn Dyson /*
475bfbbc4aaSJason Evans  * Init the per-process aioinfo structure.  The aioinfo limits are set
476bfbbc4aaSJason Evans  * per-process for user limit (resource) management.
4772244ea07SJohn Dyson  */
47888ed460eSAlan Cox static void
479fd3bf775SJohn Dyson aio_init_aioinfo(struct proc *p)
480fd3bf775SJohn Dyson {
4812244ea07SJohn Dyson 	struct kaioinfo *ki;
482ac41f2efSAlfred Perlstein 
483a163d034SWarner Losh 	ki = uma_zalloc(kaio_zone, M_WAITOK);
48484af4da6SJohn Dyson 	ki->kaio_flags = 0;
485a624e84fSJohn Dyson 	ki->kaio_maxactive_count = max_aio_per_proc;
4862244ea07SJohn Dyson 	ki->kaio_active_count = 0;
487a624e84fSJohn Dyson 	ki->kaio_qallowed_count = max_aio_queue_per_proc;
4881ce91824SDavid Xu 	ki->kaio_count = 0;
48984af4da6SJohn Dyson 	ki->kaio_ballowed_count = max_buf_aio;
490fd3bf775SJohn Dyson 	ki->kaio_buffer_count = 0;
4911ce91824SDavid Xu 	TAILQ_INIT(&ki->kaio_all);
4921ce91824SDavid Xu 	TAILQ_INIT(&ki->kaio_done);
4932244ea07SJohn Dyson 	TAILQ_INIT(&ki->kaio_jobqueue);
494fd3bf775SJohn Dyson 	TAILQ_INIT(&ki->kaio_bufqueue);
49584af4da6SJohn Dyson 	TAILQ_INIT(&ki->kaio_liojoblist);
496bfbbc4aaSJason Evans 	TAILQ_INIT(&ki->kaio_sockqueue);
4973999ebe3SAlan Cox 	PROC_LOCK(p);
4983999ebe3SAlan Cox 	if (p->p_aioinfo == NULL) {
4993999ebe3SAlan Cox 		p->p_aioinfo = ki;
5003999ebe3SAlan Cox 		PROC_UNLOCK(p);
5013999ebe3SAlan Cox 	} else {
5023999ebe3SAlan Cox 		PROC_UNLOCK(p);
5033999ebe3SAlan Cox 		uma_zfree(kaio_zone, ki);
5042244ea07SJohn Dyson 	}
505bfbbc4aaSJason Evans 
506bfbbc4aaSJason Evans 	while (num_aio_procs < target_aio_procs)
5071ce91824SDavid Xu 		aio_newproc(NULL);
5082244ea07SJohn Dyson }
5092244ea07SJohn Dyson 
5104c0fb2cfSDavid Xu static int
5114c0fb2cfSDavid Xu aio_sendsig(struct proc *p, struct sigevent *sigev, ksiginfo_t *ksi)
5124c0fb2cfSDavid Xu {
5134c0fb2cfSDavid Xu 	PROC_LOCK_ASSERT(p, MA_OWNED);
5144c0fb2cfSDavid Xu 	if (!KSI_ONQ(ksi)) {
5154c0fb2cfSDavid Xu 		ksi->ksi_code = SI_ASYNCIO;
5164c0fb2cfSDavid Xu 		ksi->ksi_flags |= KSI_EXT | KSI_INS;
5174c0fb2cfSDavid Xu 		return (psignal_event(p, sigev, ksi));
5184c0fb2cfSDavid Xu 	}
5194c0fb2cfSDavid Xu 	return (0);
5204c0fb2cfSDavid Xu }
5214c0fb2cfSDavid Xu 
5222244ea07SJohn Dyson /*
523bfbbc4aaSJason Evans  * Free a job entry.  Wait for completion if it is currently active, but don't
524bfbbc4aaSJason Evans  * delay forever.  If we delay, we return a flag that says that we have to
525bfbbc4aaSJason Evans  * restart the queue scan.
5262244ea07SJohn Dyson  */
52788ed460eSAlan Cox static int
528fd3bf775SJohn Dyson aio_free_entry(struct aiocblist *aiocbe)
529fd3bf775SJohn Dyson {
5302244ea07SJohn Dyson 	struct kaioinfo *ki;
5311ce91824SDavid Xu 	struct aioliojob *lj;
5322244ea07SJohn Dyson 	struct proc *p;
5332244ea07SJohn Dyson 
5342244ea07SJohn Dyson 	p = aiocbe->userproc;
5351ce91824SDavid Xu 
5361ce91824SDavid Xu 	PROC_LOCK_ASSERT(p, MA_OWNED);
5371ce91824SDavid Xu 	MPASS(curproc == p);
5381ce91824SDavid Xu 	MPASS(aiocbe->jobstate == JOBST_JOBFINISHED);
5391ce91824SDavid Xu 
5402244ea07SJohn Dyson 	ki = p->p_aioinfo;
5411ce91824SDavid Xu 	MPASS(ki != NULL);
5421ce91824SDavid Xu 
5431ce91824SDavid Xu 	atomic_subtract_int(&num_queue_count, 1);
5441ce91824SDavid Xu 
5451ce91824SDavid Xu 	ki->kaio_count--;
5461ce91824SDavid Xu 	MPASS(ki->kaio_count >= 0);
5471ce91824SDavid Xu 
54884af4da6SJohn Dyson 	lj = aiocbe->lio;
54984af4da6SJohn Dyson 	if (lj) {
5501ce91824SDavid Xu 		lj->lioj_count--;
5511ce91824SDavid Xu 		lj->lioj_finished_count--;
5521ce91824SDavid Xu 
553a9bf5e37SDavid Xu 		if (lj->lioj_count == 0) {
5541ce91824SDavid Xu 			TAILQ_REMOVE(&ki->kaio_liojoblist, lj, lioj_list);
5551ce91824SDavid Xu 			/* lio is going away, we need to destroy any knotes */
5561ce91824SDavid Xu 			knlist_delete(&lj->klist, curthread, 1);
5571ce91824SDavid Xu 			sigqueue_take(&lj->lioj_ksi);
5581ce91824SDavid Xu 			uma_zfree(aiolio_zone, lj);
55984af4da6SJohn Dyson 		}
56084af4da6SJohn Dyson 	}
5611ce91824SDavid Xu 
5621ce91824SDavid Xu 	TAILQ_REMOVE(&ki->kaio_done, aiocbe, plist);
5631ce91824SDavid Xu 	TAILQ_REMOVE(&ki->kaio_all, aiocbe, allist);
564fd3bf775SJohn Dyson 
565cb679c38SJonathan Lemon 	/* aiocbe is going away, we need to destroy any knotes */
5661ce91824SDavid Xu 	knlist_delete(&aiocbe->klist, curthread, 1);
5671ce91824SDavid Xu 	sigqueue_take(&aiocbe->ksi);
5681ce91824SDavid Xu 
5691ce91824SDavid Xu  	MPASS(aiocbe->bp == NULL);
5701ce91824SDavid Xu 	aiocbe->jobstate = JOBST_NULL;
5711ce91824SDavid Xu 
5721ce91824SDavid Xu 	/* Wake up anyone who has interest to do cleanup work. */
5731ce91824SDavid Xu 	if (ki->kaio_flags & (KAIO_WAKEUP | KAIO_RUNDOWN)) {
5741ce91824SDavid Xu 		ki->kaio_flags &= ~KAIO_WAKEUP;
5751ce91824SDavid Xu 		wakeup(&p->p_aioinfo);
5761ce91824SDavid Xu 	}
5771ce91824SDavid Xu 	PROC_UNLOCK(p);
5782a522eb9SJohn Baldwin 
5792a522eb9SJohn Baldwin 	/*
5802a522eb9SJohn Baldwin 	 * The thread argument here is used to find the owning process
5812a522eb9SJohn Baldwin 	 * and is also passed to fo_close() which may pass it to various
5822a522eb9SJohn Baldwin 	 * places such as devsw close() routines.  Because of that, we
5832a522eb9SJohn Baldwin 	 * need a thread pointer from the process owning the job that is
5842a522eb9SJohn Baldwin 	 * persistent and won't disappear out from under us or move to
5852a522eb9SJohn Baldwin 	 * another process.
5862a522eb9SJohn Baldwin 	 *
5872a522eb9SJohn Baldwin 	 * Currently, all the callers of this function call it to remove
5882a522eb9SJohn Baldwin 	 * an aiocblist from the current process' job list either via a
5892a522eb9SJohn Baldwin 	 * syscall or due to the current process calling exit() or
5902a522eb9SJohn Baldwin 	 * execve().  Thus, we know that p == curproc.  We also know that
5912a522eb9SJohn Baldwin 	 * curthread can't exit since we are curthread.
5922a522eb9SJohn Baldwin 	 *
5932a522eb9SJohn Baldwin 	 * Therefore, we use curthread as the thread to pass to
5942a522eb9SJohn Baldwin 	 * knlist_delete().  This does mean that it is possible for the
5952a522eb9SJohn Baldwin 	 * thread pointer at close time to differ from the thread pointer
5962a522eb9SJohn Baldwin 	 * at open time, but this is already true of file descriptors in
5972a522eb9SJohn Baldwin 	 * a multithreaded process.
598b40ce416SJulian Elischer 	 */
599a5c0b1c0SAlan Cox 	fdrop(aiocbe->fd_file, curthread);
600f8f750c5SRobert Watson 	crfree(aiocbe->cred);
601c897b813SJeff Roberson 	uma_zfree(aiocb_zone, aiocbe);
6021ce91824SDavid Xu 	PROC_LOCK(p);
6031ce91824SDavid Xu 
604ac41f2efSAlfred Perlstein 	return (0);
6052244ea07SJohn Dyson }
6062244ea07SJohn Dyson 
6072244ea07SJohn Dyson /*
6082244ea07SJohn Dyson  * Rundown the jobs for a given process.
6092244ea07SJohn Dyson  */
61021d56e9cSAlfred Perlstein static void
61175b8b3b2SJohn Baldwin aio_proc_rundown(void *arg, struct proc *p)
612fd3bf775SJohn Dyson {
6132244ea07SJohn Dyson 	struct kaioinfo *ki;
6141ce91824SDavid Xu 	struct aioliojob *lj;
6151ce91824SDavid Xu 	struct aiocblist *cbe, *cbn;
616bfbbc4aaSJason Evans 	struct file *fp;
617bfbbc4aaSJason Evans 	struct socket *so;
6181aa4c324SDavid Xu 	int remove;
6192244ea07SJohn Dyson 
6202a522eb9SJohn Baldwin 	KASSERT(curthread->td_proc == p,
6212a522eb9SJohn Baldwin 	    ("%s: called on non-curproc", __func__));
6222244ea07SJohn Dyson 	ki = p->p_aioinfo;
6232244ea07SJohn Dyson 	if (ki == NULL)
6242244ea07SJohn Dyson 		return;
6252244ea07SJohn Dyson 
6261ce91824SDavid Xu 	PROC_LOCK(p);
6271ce91824SDavid Xu 
6281ce91824SDavid Xu restart:
629fd3bf775SJohn Dyson 	ki->kaio_flags |= KAIO_RUNDOWN;
630a624e84fSJohn Dyson 
631bfbbc4aaSJason Evans 	/*
6321ce91824SDavid Xu 	 * Try to cancel all pending requests. This code simulates
6331ce91824SDavid Xu 	 * aio_cancel on all pending I/O requests.
634bfbbc4aaSJason Evans 	 */
6351ce91824SDavid Xu 	TAILQ_FOREACH_SAFE(cbe, &ki->kaio_jobqueue, plist, cbn) {
6361aa4c324SDavid Xu 		remove = 0;
6371ce91824SDavid Xu 		mtx_lock(&aio_job_mtx);
6381ce91824SDavid Xu 		if (cbe->jobstate == JOBST_JOBQGLOBAL) {
6391ce91824SDavid Xu 			TAILQ_REMOVE(&aio_jobs, cbe, list);
6401aa4c324SDavid Xu 			remove = 1;
6411aa4c324SDavid Xu 		} else if (cbe->jobstate == JOBST_JOBQSOCK) {
6421aa4c324SDavid Xu 			fp = cbe->fd_file;
6431aa4c324SDavid Xu 			MPASS(fp->f_type == DTYPE_SOCKET);
6441aa4c324SDavid Xu 			so = fp->f_data;
6451aa4c324SDavid Xu 			TAILQ_REMOVE(&so->so_aiojobq, cbe, list);
6461aa4c324SDavid Xu 			remove = 1;
6471aa4c324SDavid Xu 		}
6481ce91824SDavid Xu 		mtx_unlock(&aio_job_mtx);
6491aa4c324SDavid Xu 
6501aa4c324SDavid Xu 		if (remove) {
6511ce91824SDavid Xu 			cbe->jobstate = JOBST_JOBFINISHED;
6521ce91824SDavid Xu 			cbe->uaiocb._aiocb_private.status = -1;
6531ce91824SDavid Xu 			cbe->uaiocb._aiocb_private.error = ECANCELED;
6541ce91824SDavid Xu 			TAILQ_REMOVE(&ki->kaio_jobqueue, cbe, plist);
6551ce91824SDavid Xu 			aio_bio_done_notify(p, cbe, DONE_QUEUE);
6561ce91824SDavid Xu 		}
6572244ea07SJohn Dyson 	}
65884af4da6SJohn Dyson 
6591ce91824SDavid Xu 	/* Wait for all running I/O to be finished */
6601ce91824SDavid Xu 	if (TAILQ_FIRST(&ki->kaio_bufqueue) ||
6611ce91824SDavid Xu 	    TAILQ_FIRST(&ki->kaio_jobqueue)) {
66284af4da6SJohn Dyson 		ki->kaio_flags |= KAIO_WAKEUP;
6631ce91824SDavid Xu 		msleep(&p->p_aioinfo, &p->p_mtx, PRIBIO, "aioprn", hz);
6641ce91824SDavid Xu 		goto restart;
66584af4da6SJohn Dyson 	}
66684af4da6SJohn Dyson 
6671ce91824SDavid Xu 	/* Free all completed I/O requests. */
6681ce91824SDavid Xu 	while ((cbe = TAILQ_FIRST(&ki->kaio_done)) != NULL)
6691ce91824SDavid Xu 		aio_free_entry(cbe);
67084af4da6SJohn Dyson 
6711ce91824SDavid Xu 	while ((lj = TAILQ_FIRST(&ki->kaio_liojoblist)) != NULL) {
672a9bf5e37SDavid Xu 		if (lj->lioj_count == 0) {
67384af4da6SJohn Dyson 			TAILQ_REMOVE(&ki->kaio_liojoblist, lj, lioj_list);
6741ce91824SDavid Xu 			knlist_delete(&lj->klist, curthread, 1);
6751ce91824SDavid Xu 			sigqueue_take(&lj->lioj_ksi);
676c897b813SJeff Roberson 			uma_zfree(aiolio_zone, lj);
677f4f0ecefSJohn Dyson 		} else {
678a9bf5e37SDavid Xu 			panic("LIO job not cleaned up: C:%d, FC:%d\n",
679a9bf5e37SDavid Xu 			    lj->lioj_count, lj->lioj_finished_count);
68084af4da6SJohn Dyson 		}
681f4f0ecefSJohn Dyson 	}
68284af4da6SJohn Dyson 
683c897b813SJeff Roberson 	uma_zfree(kaio_zone, ki);
684a624e84fSJohn Dyson 	p->p_aioinfo = NULL;
6851ce91824SDavid Xu 	PROC_UNLOCK(p);
6862244ea07SJohn Dyson }
6872244ea07SJohn Dyson 
6882244ea07SJohn Dyson /*
689bfbbc4aaSJason Evans  * Select a job to run (called by an AIO daemon).
6902244ea07SJohn Dyson  */
6912244ea07SJohn Dyson static struct aiocblist *
692b40ce416SJulian Elischer aio_selectjob(struct aiothreadlist *aiop)
693fd3bf775SJohn Dyson {
6942244ea07SJohn Dyson 	struct aiocblist *aiocbe;
695bfbbc4aaSJason Evans 	struct kaioinfo *ki;
696bfbbc4aaSJason Evans 	struct proc *userp;
6972244ea07SJohn Dyson 
6981ce91824SDavid Xu 	mtx_assert(&aio_job_mtx, MA_OWNED);
6992a522eb9SJohn Baldwin 	TAILQ_FOREACH(aiocbe, &aio_jobs, list) {
7002244ea07SJohn Dyson 		userp = aiocbe->userproc;
7012244ea07SJohn Dyson 		ki = userp->p_aioinfo;
7022244ea07SJohn Dyson 
7032244ea07SJohn Dyson 		if (ki->kaio_active_count < ki->kaio_maxactive_count) {
7042244ea07SJohn Dyson 			TAILQ_REMOVE(&aio_jobs, aiocbe, list);
7051ce91824SDavid Xu 			/* Account for currently active jobs. */
7061ce91824SDavid Xu 			ki->kaio_active_count++;
7071ce91824SDavid Xu 			aiocbe->jobstate = JOBST_JOBRUNNING;
7081ce91824SDavid Xu 			break;
7091ce91824SDavid Xu 		}
7101ce91824SDavid Xu 	}
711ac41f2efSAlfred Perlstein 	return (aiocbe);
7122244ea07SJohn Dyson }
7132244ea07SJohn Dyson 
7142244ea07SJohn Dyson /*
715bfbbc4aaSJason Evans  * The AIO processing activity.  This is the code that does the I/O request for
716bfbbc4aaSJason Evans  * the non-physio version of the operations.  The normal vn operations are used,
717bfbbc4aaSJason Evans  * and this code should work in all instances for every type of file, including
718bfbbc4aaSJason Evans  * pipes, sockets, fifos, and regular files.
7191ce91824SDavid Xu  *
7201aa4c324SDavid Xu  * XXX I don't think it works well for socket, pipe, and fifo.
7212244ea07SJohn Dyson  */
72288ed460eSAlan Cox static void
723fd3bf775SJohn Dyson aio_process(struct aiocblist *aiocbe)
724fd3bf775SJohn Dyson {
725f8f750c5SRobert Watson 	struct ucred *td_savedcred;
726b40ce416SJulian Elischer 	struct thread *td;
727b40ce416SJulian Elischer 	struct proc *mycp;
7282244ea07SJohn Dyson 	struct aiocb *cb;
7292244ea07SJohn Dyson 	struct file *fp;
7301ce91824SDavid Xu 	struct socket *so;
7312244ea07SJohn Dyson 	struct uio auio;
7322244ea07SJohn Dyson 	struct iovec aiov;
7332244ea07SJohn Dyson 	int cnt;
7342244ea07SJohn Dyson 	int error;
735fd3bf775SJohn Dyson 	int oublock_st, oublock_end;
736fd3bf775SJohn Dyson 	int inblock_st, inblock_end;
7372244ea07SJohn Dyson 
738b40ce416SJulian Elischer 	td = curthread;
739f8f750c5SRobert Watson 	td_savedcred = td->td_ucred;
740f8f750c5SRobert Watson 	td->td_ucred = aiocbe->cred;
741b40ce416SJulian Elischer 	mycp = td->td_proc;
7422244ea07SJohn Dyson 	cb = &aiocbe->uaiocb;
74300e73160SAlan Cox 	fp = aiocbe->fd_file;
744bfbbc4aaSJason Evans 
74591369fc7SAlan Cox 	aiov.iov_base = (void *)(uintptr_t)cb->aio_buf;
7462244ea07SJohn Dyson 	aiov.iov_len = cb->aio_nbytes;
7472244ea07SJohn Dyson 
7482244ea07SJohn Dyson 	auio.uio_iov = &aiov;
7492244ea07SJohn Dyson 	auio.uio_iovcnt = 1;
7509b16adc1SAlan Cox 	auio.uio_offset = cb->aio_offset;
7512244ea07SJohn Dyson 	auio.uio_resid = cb->aio_nbytes;
7522244ea07SJohn Dyson 	cnt = cb->aio_nbytes;
7532244ea07SJohn Dyson 	auio.uio_segflg = UIO_USERSPACE;
754b40ce416SJulian Elischer 	auio.uio_td = td;
7552244ea07SJohn Dyson 
756fd3bf775SJohn Dyson 	inblock_st = mycp->p_stats->p_ru.ru_inblock;
757fd3bf775SJohn Dyson 	oublock_st = mycp->p_stats->p_ru.ru_oublock;
758279d7226SMatthew Dillon 	/*
759a9bf5e37SDavid Xu 	 * aio_aqueue() acquires a reference to the file that is
7609b16adc1SAlan Cox 	 * released in aio_free_entry().
761279d7226SMatthew Dillon 	 */
7622244ea07SJohn Dyson 	if (cb->aio_lio_opcode == LIO_READ) {
7632244ea07SJohn Dyson 		auio.uio_rw = UIO_READ;
764b40ce416SJulian Elischer 		error = fo_read(fp, &auio, fp->f_cred, FOF_OFFSET, td);
7652244ea07SJohn Dyson 	} else {
7666d53aa62SDavid Xu 		if (fp->f_type == DTYPE_VNODE)
7676d53aa62SDavid Xu 			bwillwrite();
7682244ea07SJohn Dyson 		auio.uio_rw = UIO_WRITE;
769b40ce416SJulian Elischer 		error = fo_write(fp, &auio, fp->f_cred, FOF_OFFSET, td);
7702244ea07SJohn Dyson 	}
771fd3bf775SJohn Dyson 	inblock_end = mycp->p_stats->p_ru.ru_inblock;
772fd3bf775SJohn Dyson 	oublock_end = mycp->p_stats->p_ru.ru_oublock;
773fd3bf775SJohn Dyson 
774fd3bf775SJohn Dyson 	aiocbe->inputcharge = inblock_end - inblock_st;
775fd3bf775SJohn Dyson 	aiocbe->outputcharge = oublock_end - oublock_st;
7762244ea07SJohn Dyson 
777bfbbc4aaSJason Evans 	if ((error) && (auio.uio_resid != cnt)) {
7782244ea07SJohn Dyson 		if (error == ERESTART || error == EINTR || error == EWOULDBLOCK)
7792244ea07SJohn Dyson 			error = 0;
78019eb87d2SJohn Baldwin 		if ((error == EPIPE) && (cb->aio_lio_opcode == LIO_WRITE)) {
7811ce91824SDavid Xu 			int sigpipe = 1;
7821ce91824SDavid Xu 			if (fp->f_type == DTYPE_SOCKET) {
7831ce91824SDavid Xu 				so = fp->f_data;
7841ce91824SDavid Xu 				if (so->so_options & SO_NOSIGPIPE)
7851ce91824SDavid Xu 					sigpipe = 0;
7861ce91824SDavid Xu 			}
7871ce91824SDavid Xu 			if (sigpipe) {
7889b16adc1SAlan Cox 				PROC_LOCK(aiocbe->userproc);
7899b16adc1SAlan Cox 				psignal(aiocbe->userproc, SIGPIPE);
7909b16adc1SAlan Cox 				PROC_UNLOCK(aiocbe->userproc);
79119eb87d2SJohn Baldwin 			}
7922244ea07SJohn Dyson 		}
7931ce91824SDavid Xu 	}
7942244ea07SJohn Dyson 
7952244ea07SJohn Dyson 	cnt -= auio.uio_resid;
7962244ea07SJohn Dyson 	cb->_aiocb_private.error = error;
7972244ea07SJohn Dyson 	cb->_aiocb_private.status = cnt;
798f8f750c5SRobert Watson 	td->td_ucred = td_savedcred;
7992244ea07SJohn Dyson }
8002244ea07SJohn Dyson 
80169cd28daSDoug Ambrisko static void
8021ce91824SDavid Xu aio_bio_done_notify(struct proc *userp, struct aiocblist *aiocbe, int type)
8031ce91824SDavid Xu {
8041ce91824SDavid Xu 	struct aioliojob *lj;
80569cd28daSDoug Ambrisko 	struct kaioinfo *ki;
8061ce91824SDavid Xu 	int lj_done;
80769cd28daSDoug Ambrisko 
8081ce91824SDavid Xu 	PROC_LOCK_ASSERT(userp, MA_OWNED);
80969cd28daSDoug Ambrisko 	ki = userp->p_aioinfo;
81069cd28daSDoug Ambrisko 	lj = aiocbe->lio;
81169cd28daSDoug Ambrisko 	lj_done = 0;
81269cd28daSDoug Ambrisko 	if (lj) {
8131ce91824SDavid Xu 		lj->lioj_finished_count++;
8141ce91824SDavid Xu 		if (lj->lioj_count == lj->lioj_finished_count)
81569cd28daSDoug Ambrisko 			lj_done = 1;
81669cd28daSDoug Ambrisko 	}
81769cd28daSDoug Ambrisko 	if (type == DONE_QUEUE) {
8181ce91824SDavid Xu 		aiocbe->jobflags |= AIOCBLIST_DONE;
81969cd28daSDoug Ambrisko 	} else {
8201ce91824SDavid Xu 		aiocbe->jobflags |= AIOCBLIST_BUFDONE;
8211ce91824SDavid Xu 		ki->kaio_buffer_count--;
82269cd28daSDoug Ambrisko 	}
8231ce91824SDavid Xu 	TAILQ_INSERT_TAIL(&ki->kaio_done, aiocbe, plist);
8241ce91824SDavid Xu 	aiocbe->jobstate = JOBST_JOBFINISHED;
8251ce91824SDavid Xu 	if (aiocbe->uaiocb.aio_sigevent.sigev_notify == SIGEV_SIGNAL ||
8261ce91824SDavid Xu 	    aiocbe->uaiocb.aio_sigevent.sigev_notify == SIGEV_THREAD_ID)
8271ce91824SDavid Xu 		aio_sendsig(userp, &aiocbe->uaiocb.aio_sigevent, &aiocbe->ksi);
8281ce91824SDavid Xu 
8291ce91824SDavid Xu 	KNOTE_LOCKED(&aiocbe->klist, 1);
8301ce91824SDavid Xu 
83169cd28daSDoug Ambrisko 	if (lj_done) {
8321ce91824SDavid Xu 		if (lj->lioj_signal.sigev_notify == SIGEV_KEVENT) {
83369cd28daSDoug Ambrisko 			lj->lioj_flags |= LIOJ_KEVENT_POSTED;
8341ce91824SDavid Xu 			KNOTE_LOCKED(&lj->klist, 1);
83569cd28daSDoug Ambrisko 		}
8361ce91824SDavid Xu 		if ((lj->lioj_flags & (LIOJ_SIGNAL|LIOJ_SIGNAL_POSTED))
83769cd28daSDoug Ambrisko 		    == LIOJ_SIGNAL
8384c0fb2cfSDavid Xu 		    && (lj->lioj_signal.sigev_notify == SIGEV_SIGNAL ||
8394c0fb2cfSDavid Xu 		        lj->lioj_signal.sigev_notify == SIGEV_THREAD_ID)) {
8404c0fb2cfSDavid Xu 			aio_sendsig(userp, &lj->lioj_signal, &lj->lioj_ksi);
84169cd28daSDoug Ambrisko 			lj->lioj_flags |= LIOJ_SIGNAL_POSTED;
84269cd28daSDoug Ambrisko 		}
84369cd28daSDoug Ambrisko 	}
84469cd28daSDoug Ambrisko 	if (ki->kaio_flags & (KAIO_RUNDOWN|KAIO_WAKEUP)) {
84569cd28daSDoug Ambrisko 		ki->kaio_flags &= ~KAIO_WAKEUP;
8461ce91824SDavid Xu 		wakeup(&userp->p_aioinfo);
84769cd28daSDoug Ambrisko 	}
84869cd28daSDoug Ambrisko }
84969cd28daSDoug Ambrisko 
8502244ea07SJohn Dyson /*
85184af4da6SJohn Dyson  * The AIO daemon, most of the actual work is done in aio_process,
85284af4da6SJohn Dyson  * but the setup (and address space mgmt) is done in this routine.
8532244ea07SJohn Dyson  */
8542244ea07SJohn Dyson static void
8551ce91824SDavid Xu aio_daemon(void *_id)
8562244ea07SJohn Dyson {
857bfbbc4aaSJason Evans 	struct aiocblist *aiocbe;
858b40ce416SJulian Elischer 	struct aiothreadlist *aiop;
859bfbbc4aaSJason Evans 	struct kaioinfo *ki;
860bfbbc4aaSJason Evans 	struct proc *curcp, *mycp, *userp;
861bfbbc4aaSJason Evans 	struct vmspace *myvm, *tmpvm;
862b40ce416SJulian Elischer 	struct thread *td = curthread;
8631ce91824SDavid Xu 	int id = (intptr_t)_id;
8642244ea07SJohn Dyson 
8652244ea07SJohn Dyson 	/*
866fd3bf775SJohn Dyson 	 * Local copies of curproc (cp) and vmspace (myvm)
8672244ea07SJohn Dyson 	 */
868b40ce416SJulian Elischer 	mycp = td->td_proc;
869fd3bf775SJohn Dyson 	myvm = mycp->p_vmspace;
870fd3bf775SJohn Dyson 
871cd4ed3b5SJohn Baldwin 	KASSERT(mycp->p_textvp == NULL, ("kthread has a textvp"));
872fd3bf775SJohn Dyson 
873fd3bf775SJohn Dyson 	/*
874bfbbc4aaSJason Evans 	 * Allocate and ready the aio control info.  There is one aiop structure
875bfbbc4aaSJason Evans 	 * per daemon.
876fd3bf775SJohn Dyson 	 */
877a163d034SWarner Losh 	aiop = uma_zalloc(aiop_zone, M_WAITOK);
878b40ce416SJulian Elischer 	aiop->aiothread = td;
879b40ce416SJulian Elischer 	aiop->aiothreadflags |= AIOP_FREE;
8802244ea07SJohn Dyson 
8812244ea07SJohn Dyson 	/*
882bfbbc4aaSJason Evans 	 * Place thread (lightweight process) onto the AIO free thread list.
8832244ea07SJohn Dyson 	 */
8841ce91824SDavid Xu 	mtx_lock(&aio_job_mtx);
885fd3bf775SJohn Dyson 	TAILQ_INSERT_HEAD(&aio_freeproc, aiop, list);
8861ce91824SDavid Xu 	mtx_unlock(&aio_job_mtx);
887bfbbc4aaSJason Evans 
8882244ea07SJohn Dyson 	/*
889fd3bf775SJohn Dyson 	 * Get rid of our current filedescriptors.  AIOD's don't need any
890fd3bf775SJohn Dyson 	 * filedescriptors, except as temporarily inherited from the client.
8912244ea07SJohn Dyson 	 */
892b40ce416SJulian Elischer 	fdfree(td);
893fd3bf775SJohn Dyson 
894e6bdc05fSDavid Xu 	/* The daemon resides in its own pgrp. */
895bd793be3SDavid Xu 	setsid(td, NULL);
896fd3bf775SJohn Dyson 
897fd3bf775SJohn Dyson 	/*
898fd3bf775SJohn Dyson 	 * Wakeup parent process.  (Parent sleeps to keep from blasting away
899b40ce416SJulian Elischer 	 * and creating too many daemons.)
900fd3bf775SJohn Dyson 	 */
9011ce91824SDavid Xu 	sema_post(&aio_newproc_sem);
9022244ea07SJohn Dyson 
9031ce91824SDavid Xu 	mtx_lock(&aio_job_mtx);
904bfbbc4aaSJason Evans 	for (;;) {
905fd3bf775SJohn Dyson 		/*
906fd3bf775SJohn Dyson 		 * curcp is the current daemon process context.
907fd3bf775SJohn Dyson 		 * userp is the current user process context.
908fd3bf775SJohn Dyson 		 */
909fd3bf775SJohn Dyson 		curcp = mycp;
910c4860686SJohn Dyson 
911fd3bf775SJohn Dyson 		/*
912fd3bf775SJohn Dyson 		 * Take daemon off of free queue
913fd3bf775SJohn Dyson 		 */
914b40ce416SJulian Elischer 		if (aiop->aiothreadflags & AIOP_FREE) {
9152244ea07SJohn Dyson 			TAILQ_REMOVE(&aio_freeproc, aiop, list);
916b40ce416SJulian Elischer 			aiop->aiothreadflags &= ~AIOP_FREE;
9172244ea07SJohn Dyson 		}
9182244ea07SJohn Dyson 
919fd3bf775SJohn Dyson 		/*
920bfbbc4aaSJason Evans 		 * Check for jobs.
921fd3bf775SJohn Dyson 		 */
922d254af07SMatthew Dillon 		while ((aiocbe = aio_selectjob(aiop)) != NULL) {
9231ce91824SDavid Xu 			mtx_unlock(&aio_job_mtx);
9242244ea07SJohn Dyson 			userp = aiocbe->userproc;
9252244ea07SJohn Dyson 
926fd3bf775SJohn Dyson 			/*
927bfbbc4aaSJason Evans 			 * Connect to process address space for user program.
928fd3bf775SJohn Dyson 			 */
929fd3bf775SJohn Dyson 			if (userp != curcp) {
930fd3bf775SJohn Dyson 				/*
931bfbbc4aaSJason Evans 				 * Save the current address space that we are
932bfbbc4aaSJason Evans 				 * connected to.
933fd3bf775SJohn Dyson 				 */
934fd3bf775SJohn Dyson 				tmpvm = mycp->p_vmspace;
935bfbbc4aaSJason Evans 
936fd3bf775SJohn Dyson 				/*
937bfbbc4aaSJason Evans 				 * Point to the new user address space, and
938bfbbc4aaSJason Evans 				 * refer to it.
939fd3bf775SJohn Dyson 				 */
940fd3bf775SJohn Dyson 				mycp->p_vmspace = userp->p_vmspace;
9411a276a3fSAlan Cox 				atomic_add_int(&mycp->p_vmspace->vm_refcnt, 1);
942bfbbc4aaSJason Evans 
943bfbbc4aaSJason Evans 				/* Activate the new mapping. */
944079b7badSJulian Elischer 				pmap_activate(FIRST_THREAD_IN_PROC(mycp));
945bfbbc4aaSJason Evans 
946fd3bf775SJohn Dyson 				/*
947bfbbc4aaSJason Evans 				 * If the old address space wasn't the daemons
948bfbbc4aaSJason Evans 				 * own address space, then we need to remove the
949bfbbc4aaSJason Evans 				 * daemon's reference from the other process
950bfbbc4aaSJason Evans 				 * that it was acting on behalf of.
951fd3bf775SJohn Dyson 				 */
9522244ea07SJohn Dyson 				if (tmpvm != myvm) {
9532244ea07SJohn Dyson 					vmspace_free(tmpvm);
9542244ea07SJohn Dyson 				}
955fd3bf775SJohn Dyson 				curcp = userp;
9562244ea07SJohn Dyson 			}
9572244ea07SJohn Dyson 
958fd3bf775SJohn Dyson 			ki = userp->p_aioinfo;
95984af4da6SJohn Dyson 
960bfbbc4aaSJason Evans 			/* Do the I/O function. */
9612244ea07SJohn Dyson 			aio_process(aiocbe);
96284af4da6SJohn Dyson 
9639b84335cSDavid Xu 			mtx_lock(&aio_job_mtx);
9649b84335cSDavid Xu 			/* Decrement the active job count. */
9659b84335cSDavid Xu 			ki->kaio_active_count--;
9669b84335cSDavid Xu 			mtx_unlock(&aio_job_mtx);
9679b84335cSDavid Xu 
9681ce91824SDavid Xu 			PROC_LOCK(userp);
9691ce91824SDavid Xu 			TAILQ_REMOVE(&ki->kaio_jobqueue, aiocbe, plist);
97069cd28daSDoug Ambrisko 			aio_bio_done_notify(userp, aiocbe, DONE_QUEUE);
9712244ea07SJohn Dyson 			if (aiocbe->jobflags & AIOCBLIST_RUNDOWN) {
9722244ea07SJohn Dyson 				wakeup(aiocbe);
9732244ea07SJohn Dyson 				aiocbe->jobflags &= ~AIOCBLIST_RUNDOWN;
9742244ea07SJohn Dyson 			}
9751ce91824SDavid Xu 			PROC_UNLOCK(userp);
9761ce91824SDavid Xu 
9771ce91824SDavid Xu 			mtx_lock(&aio_job_mtx);
9782244ea07SJohn Dyson 		}
9792244ea07SJohn Dyson 
980fd3bf775SJohn Dyson 		/*
981bfbbc4aaSJason Evans 		 * Disconnect from user address space.
982fd3bf775SJohn Dyson 		 */
983fd3bf775SJohn Dyson 		if (curcp != mycp) {
9841ce91824SDavid Xu 
9851ce91824SDavid Xu 			mtx_unlock(&aio_job_mtx);
9861ce91824SDavid Xu 
987bfbbc4aaSJason Evans 			/* Get the user address space to disconnect from. */
988fd3bf775SJohn Dyson 			tmpvm = mycp->p_vmspace;
989bfbbc4aaSJason Evans 
990bfbbc4aaSJason Evans 			/* Get original address space for daemon. */
991fd3bf775SJohn Dyson 			mycp->p_vmspace = myvm;
992bfbbc4aaSJason Evans 
993bfbbc4aaSJason Evans 			/* Activate the daemon's address space. */
994079b7badSJulian Elischer 			pmap_activate(FIRST_THREAD_IN_PROC(mycp));
995bfbbc4aaSJason Evans #ifdef DIAGNOSTIC
996bfbbc4aaSJason Evans 			if (tmpvm == myvm) {
997bfbbc4aaSJason Evans 				printf("AIOD: vmspace problem -- %d\n",
998bfbbc4aaSJason Evans 				    mycp->p_pid);
999bfbbc4aaSJason Evans 			}
100011783b14SJohn Dyson #endif
1001bfbbc4aaSJason Evans 			/* Remove our vmspace reference. */
10022244ea07SJohn Dyson 			vmspace_free(tmpvm);
1003bfbbc4aaSJason Evans 
1004fd3bf775SJohn Dyson 			curcp = mycp;
10051ce91824SDavid Xu 
10061ce91824SDavid Xu 			mtx_lock(&aio_job_mtx);
10071ce91824SDavid Xu 			/*
10081ce91824SDavid Xu 			 * We have to restart to avoid race, we only sleep if
10091ce91824SDavid Xu 			 * no job can be selected, that should be
10101ce91824SDavid Xu 			 * curcp == mycp.
10111ce91824SDavid Xu 			 */
10121ce91824SDavid Xu 			continue;
1013fd3bf775SJohn Dyson 		}
1014fd3bf775SJohn Dyson 
10151ce91824SDavid Xu 		mtx_assert(&aio_job_mtx, MA_OWNED);
10161ce91824SDavid Xu 
1017fd3bf775SJohn Dyson 		TAILQ_INSERT_HEAD(&aio_freeproc, aiop, list);
1018b40ce416SJulian Elischer 		aiop->aiothreadflags |= AIOP_FREE;
1019fd3bf775SJohn Dyson 
1020fd3bf775SJohn Dyson 		/*
1021bfbbc4aaSJason Evans 		 * If daemon is inactive for a long time, allow it to exit,
1022bfbbc4aaSJason Evans 		 * thereby freeing resources.
1023fd3bf775SJohn Dyson 		 */
10241ce91824SDavid Xu 		if (msleep(aiop->aiothread, &aio_job_mtx, PRIBIO, "aiordy",
10251ce91824SDavid Xu 		    aiod_lifetime)) {
1026c3869e4bSAlan Cox 			if (TAILQ_EMPTY(&aio_jobs)) {
1027b40ce416SJulian Elischer 				if ((aiop->aiothreadflags & AIOP_FREE) &&
102884af4da6SJohn Dyson 				    (num_aio_procs > target_aio_procs)) {
1029fd3bf775SJohn Dyson 					TAILQ_REMOVE(&aio_freeproc, aiop, list);
103084af4da6SJohn Dyson 					num_aio_procs--;
10311ce91824SDavid Xu 					mtx_unlock(&aio_job_mtx);
10321ce91824SDavid Xu 					uma_zfree(aiop_zone, aiop);
10331ce91824SDavid Xu 					free_unr(aiod_unr, id);
1034bfbbc4aaSJason Evans #ifdef DIAGNOSTIC
1035bfbbc4aaSJason Evans 					if (mycp->p_vmspace->vm_refcnt <= 1) {
1036bfbbc4aaSJason Evans 						printf("AIOD: bad vm refcnt for"
1037bfbbc4aaSJason Evans 						    " exiting daemon: %d\n",
1038fd3bf775SJohn Dyson 						    mycp->p_vmspace->vm_refcnt);
1039bfbbc4aaSJason Evans 					}
104011783b14SJohn Dyson #endif
1041c9a970a7SAlan Cox 					kthread_exit(0);
1042fd3bf775SJohn Dyson 				}
10432244ea07SJohn Dyson 			}
10442244ea07SJohn Dyson 		}
10452244ea07SJohn Dyson 	}
10461ce91824SDavid Xu 	mtx_unlock(&aio_job_mtx);
10471ce91824SDavid Xu 	panic("shouldn't be here\n");
10481ce91824SDavid Xu }
10492244ea07SJohn Dyson 
10502244ea07SJohn Dyson /*
1051bfbbc4aaSJason Evans  * Create a new AIO daemon. This is mostly a kernel-thread fork routine. The
1052bfbbc4aaSJason Evans  * AIO daemon modifies its environment itself.
10532244ea07SJohn Dyson  */
10542244ea07SJohn Dyson static int
10551ce91824SDavid Xu aio_newproc(int *start)
1056fd3bf775SJohn Dyson {
10572244ea07SJohn Dyson 	int error;
1058c9a970a7SAlan Cox 	struct proc *p;
10591ce91824SDavid Xu 	int id;
10602244ea07SJohn Dyson 
10611ce91824SDavid Xu 	id = alloc_unr(aiod_unr);
10621ce91824SDavid Xu 	error = kthread_create(aio_daemon, (void *)(intptr_t)id, &p,
10631ce91824SDavid Xu 		RFNOWAIT, 0, "aiod%d", id);
10641ce91824SDavid Xu 	if (error == 0) {
1065fd3bf775SJohn Dyson 		/*
10661ce91824SDavid Xu 		 * Wait until daemon is started.
1067fd3bf775SJohn Dyson 		 */
10681ce91824SDavid Xu 		sema_wait(&aio_newproc_sem);
10691ce91824SDavid Xu 		mtx_lock(&aio_job_mtx);
107084af4da6SJohn Dyson 		num_aio_procs++;
10711ce91824SDavid Xu 		if (start != NULL)
10727f34b521SDavid Xu 			(*start)--;
10731ce91824SDavid Xu 		mtx_unlock(&aio_job_mtx);
10741ce91824SDavid Xu 	} else {
10751ce91824SDavid Xu 		free_unr(aiod_unr, id);
10761ce91824SDavid Xu 	}
1077ac41f2efSAlfred Perlstein 	return (error);
10782244ea07SJohn Dyson }
10792244ea07SJohn Dyson 
10802244ea07SJohn Dyson /*
108188ed460eSAlan Cox  * Try the high-performance, low-overhead physio method for eligible
108288ed460eSAlan Cox  * VCHR devices.  This method doesn't use an aio helper thread, and
108388ed460eSAlan Cox  * thus has very low overhead.
108488ed460eSAlan Cox  *
1085a9bf5e37SDavid Xu  * Assumes that the caller, aio_aqueue(), has incremented the file
108688ed460eSAlan Cox  * structure's reference count, preventing its deallocation for the
108788ed460eSAlan Cox  * duration of this call.
1088fd3bf775SJohn Dyson  */
108988ed460eSAlan Cox static int
1090bfbbc4aaSJason Evans aio_qphysio(struct proc *p, struct aiocblist *aiocbe)
1091fd3bf775SJohn Dyson {
1092fd3bf775SJohn Dyson 	struct aiocb *cb;
1093fd3bf775SJohn Dyson 	struct file *fp;
1094fd3bf775SJohn Dyson 	struct buf *bp;
1095fd3bf775SJohn Dyson 	struct vnode *vp;
1096fd3bf775SJohn Dyson 	struct kaioinfo *ki;
10971ce91824SDavid Xu 	struct aioliojob *lj;
10981ce91824SDavid Xu 	int error;
1099fd3bf775SJohn Dyson 
110084af4da6SJohn Dyson 	cb = &aiocbe->uaiocb;
11019fbd7ccfSAlan Cox 	fp = aiocbe->fd_file;
1102fd3bf775SJohn Dyson 
1103008626c3SPoul-Henning Kamp 	if (fp->f_type != DTYPE_VNODE)
1104008626c3SPoul-Henning Kamp 		return (-1);
1105fd3bf775SJohn Dyson 
11063b6d9652SPoul-Henning Kamp 	vp = fp->f_vnode;
110711783b14SJohn Dyson 
1108f582ac06SBrian Feldman 	/*
1109f582ac06SBrian Feldman 	 * If its not a disk, we don't want to return a positive error.
1110f582ac06SBrian Feldman 	 * It causes the aio code to not fall through to try the thread
1111f582ac06SBrian Feldman 	 * way when you're talking to a regular file.
1112f582ac06SBrian Feldman 	 */
1113f582ac06SBrian Feldman 	if (!vn_isdisk(vp, &error)) {
1114f582ac06SBrian Feldman 		if (error == ENOTBLK)
1115f582ac06SBrian Feldman 			return (-1);
1116f582ac06SBrian Feldman 		else
1117ba4ad1fcSPoul-Henning Kamp 			return (error);
1118f582ac06SBrian Feldman 	}
1119fd3bf775SJohn Dyson 
1120ad8de0f2SDavid Xu 	if (vp->v_bufobj.bo_bsize == 0)
1121ad8de0f2SDavid Xu 		return (-1);
1122ad8de0f2SDavid Xu 
11235d9d81e7SPoul-Henning Kamp  	if (cb->aio_nbytes % vp->v_bufobj.bo_bsize)
1124008626c3SPoul-Henning Kamp 		return (-1);
1125fd3bf775SJohn Dyson 
112669cd28daSDoug Ambrisko 	if (cb->aio_nbytes > vp->v_rdev->si_iosize_max)
112769cd28daSDoug Ambrisko 		return (-1);
112869cd28daSDoug Ambrisko 
112913644654SAlan Cox 	if (cb->aio_nbytes >
113013644654SAlan Cox 	    MAXPHYS - (((vm_offset_t) cb->aio_buf) & PAGE_MASK))
1131008626c3SPoul-Henning Kamp 		return (-1);
1132fd3bf775SJohn Dyson 
1133fd3bf775SJohn Dyson 	ki = p->p_aioinfo;
1134008626c3SPoul-Henning Kamp 	if (ki->kaio_buffer_count >= ki->kaio_ballowed_count)
1135008626c3SPoul-Henning Kamp 		return (-1);
1136fd3bf775SJohn Dyson 
1137bfbbc4aaSJason Evans 	/* Create and build a buffer header for a transfer. */
11381c7c3c6aSMatthew Dillon 	bp = (struct buf *)getpbuf(NULL);
113913644654SAlan Cox 	BUF_KERNPROC(bp);
1140fd3bf775SJohn Dyson 
11411ce91824SDavid Xu 	PROC_LOCK(p);
11421ce91824SDavid Xu 	ki->kaio_count++;
11431ce91824SDavid Xu 	ki->kaio_buffer_count++;
11441ce91824SDavid Xu 	lj = aiocbe->lio;
11451ce91824SDavid Xu 	if (lj)
11461ce91824SDavid Xu 		lj->lioj_count++;
11471ce91824SDavid Xu 	PROC_UNLOCK(p);
11481ce91824SDavid Xu 
1149fd3bf775SJohn Dyson 	/*
1150bfbbc4aaSJason Evans 	 * Get a copy of the kva from the physical buffer.
1151fd3bf775SJohn Dyson 	 */
1152ef38cda1SAlan Cox 	error = 0;
1153fd3bf775SJohn Dyson 
1154fd3bf775SJohn Dyson 	bp->b_bcount = cb->aio_nbytes;
1155fd3bf775SJohn Dyson 	bp->b_bufsize = cb->aio_nbytes;
1156fd3bf775SJohn Dyson 	bp->b_iodone = aio_physwakeup;
1157fd3bf775SJohn Dyson 	bp->b_saveaddr = bp->b_data;
115891369fc7SAlan Cox 	bp->b_data = (void *)(uintptr_t)cb->aio_buf;
1159a44ca4f0SHidetoshi Shimokawa 	bp->b_offset = cb->aio_offset;
1160a44ca4f0SHidetoshi Shimokawa 	bp->b_iooffset = cb->aio_offset;
1161fd3bf775SJohn Dyson 	bp->b_blkno = btodb(cb->aio_offset);
116206363906SAlan Cox 	bp->b_iocmd = cb->aio_lio_opcode == LIO_WRITE ? BIO_WRITE : BIO_READ;
1163fd3bf775SJohn Dyson 
11642d5c7e45SMatthew Dillon 	/*
11652d5c7e45SMatthew Dillon 	 * Bring buffer into kernel space.
11662d5c7e45SMatthew Dillon 	 */
11672d5c7e45SMatthew Dillon 	if (vmapbuf(bp) < 0) {
11682d5c7e45SMatthew Dillon 		error = EFAULT;
11692d5c7e45SMatthew Dillon 		goto doerror;
11702d5c7e45SMatthew Dillon 	}
1171fd3bf775SJohn Dyson 
11721ce91824SDavid Xu 	PROC_LOCK(p);
1173fd3bf775SJohn Dyson 	aiocbe->bp = bp;
11748edbaf85SHidetoshi Shimokawa 	bp->b_caller1 = (void *)aiocbe;
117584af4da6SJohn Dyson 	TAILQ_INSERT_TAIL(&ki->kaio_bufqueue, aiocbe, plist);
11761ce91824SDavid Xu 	TAILQ_INSERT_TAIL(&ki->kaio_all, aiocbe, allist);
1177fd3bf775SJohn Dyson 	aiocbe->jobstate = JOBST_JOBQBUF;
117884af4da6SJohn Dyson 	cb->_aiocb_private.status = cb->aio_nbytes;
11791ce91824SDavid Xu 	PROC_UNLOCK(p);
11801ce91824SDavid Xu 
11811ce91824SDavid Xu 	atomic_add_int(&num_queue_count, 1);
11821ce91824SDavid Xu 	atomic_add_int(&num_buf_aio, 1);
11831ce91824SDavid Xu 
1184fd3bf775SJohn Dyson 	bp->b_error = 0;
1185fd3bf775SJohn Dyson 
11861ce91824SDavid Xu 	TASK_INIT(&aiocbe->biotask, 0, biohelper, aiocbe);
1187bfbbc4aaSJason Evans 
1188bfbbc4aaSJason Evans 	/* Perform transfer. */
11896afb3b1cSPoul-Henning Kamp 	dev_strategy(vp->v_rdev, bp);
1190ac41f2efSAlfred Perlstein 	return (0);
1191fd3bf775SJohn Dyson 
1192fd3bf775SJohn Dyson doerror:
11931ce91824SDavid Xu 	PROC_LOCK(p);
11941ce91824SDavid Xu 	ki->kaio_count--;
1195fd3bf775SJohn Dyson 	ki->kaio_buffer_count--;
1196bfbbc4aaSJason Evans 	if (lj)
11971ce91824SDavid Xu 		lj->lioj_count--;
119884af4da6SJohn Dyson 	aiocbe->bp = NULL;
11991ce91824SDavid Xu 	PROC_UNLOCK(p);
12001c7c3c6aSMatthew Dillon 	relpbuf(bp, NULL);
1201fd3bf775SJohn Dyson 	return (error);
1202fd3bf775SJohn Dyson }
1203fd3bf775SJohn Dyson 
1204fd3bf775SJohn Dyson /*
1205bfbbc4aaSJason Evans  * Wake up aio requests that may be serviceable now.
1206bfbbc4aaSJason Evans  */
120748dac059SAlan Cox static void
120821d56e9cSAlfred Perlstein aio_swake_cb(struct socket *so, struct sockbuf *sb)
1209bfbbc4aaSJason Evans {
1210bfbbc4aaSJason Evans 	struct aiocblist *cb, *cbn;
1211bfbbc4aaSJason Evans 	int opcode, wakecount = 0;
1212b40ce416SJulian Elischer 	struct aiothreadlist *aiop;
1213bfbbc4aaSJason Evans 
12141aa4c324SDavid Xu 	if (sb == &so->so_snd)
1215bfbbc4aaSJason Evans 		opcode = LIO_WRITE;
12161aa4c324SDavid Xu 	else
1217bfbbc4aaSJason Evans 		opcode = LIO_READ;
1218bfbbc4aaSJason Evans 
12191aa4c324SDavid Xu 	SOCKBUF_LOCK(sb);
12201aa4c324SDavid Xu 	sb->sb_flags &= ~SB_AIO;
12211aa4c324SDavid Xu 	mtx_lock(&aio_job_mtx);
12222a522eb9SJohn Baldwin 	TAILQ_FOREACH_SAFE(cb, &so->so_aiojobq, list, cbn) {
1223bfbbc4aaSJason Evans 		if (opcode == cb->uaiocb.aio_lio_opcode) {
12248c0d9af5SDavid Xu 			if (cb->jobstate != JOBST_JOBQSOCK)
12251ce91824SDavid Xu 				panic("invalid queue value");
12261aa4c324SDavid Xu 			/* XXX
12271aa4c324SDavid Xu 			 * We don't have actual sockets backend yet,
12281aa4c324SDavid Xu 			 * so we simply move the requests to the generic
12291aa4c324SDavid Xu 			 * file I/O backend.
12301ce91824SDavid Xu 			 */
12311aa4c324SDavid Xu 			TAILQ_REMOVE(&so->so_aiojobq, cb, list);
12321ce91824SDavid Xu 			TAILQ_INSERT_TAIL(&aio_jobs, cb, list);
1233bfbbc4aaSJason Evans 			wakecount++;
1234bfbbc4aaSJason Evans 		}
1235bfbbc4aaSJason Evans 	}
12361aa4c324SDavid Xu 	mtx_unlock(&aio_job_mtx);
12371aa4c324SDavid Xu 	SOCKBUF_UNLOCK(sb);
1238bfbbc4aaSJason Evans 
1239bfbbc4aaSJason Evans 	while (wakecount--) {
12401ce91824SDavid Xu 		mtx_lock(&aio_job_mtx);
12412a522eb9SJohn Baldwin 		if ((aiop = TAILQ_FIRST(&aio_freeproc)) != NULL) {
1242bfbbc4aaSJason Evans 			TAILQ_REMOVE(&aio_freeproc, aiop, list);
1243b40ce416SJulian Elischer 			aiop->aiothreadflags &= ~AIOP_FREE;
1244b40ce416SJulian Elischer 			wakeup(aiop->aiothread);
1245bfbbc4aaSJason Evans 		}
12461ce91824SDavid Xu 		mtx_unlock(&aio_job_mtx);
1247bfbbc4aaSJason Evans 	}
1248bfbbc4aaSJason Evans }
1249bfbbc4aaSJason Evans 
1250bfbbc4aaSJason Evans /*
1251bfbbc4aaSJason Evans  * Queue a new AIO request.  Choosing either the threaded or direct physio VCHR
1252bfbbc4aaSJason Evans  * technique is done in this code.
12532244ea07SJohn Dyson  */
12542244ea07SJohn Dyson static int
1255a9bf5e37SDavid Xu aio_aqueue(struct thread *td, struct aiocb *job, struct aioliojob *lj,
12560972628aSDavid Xu 	int type, int oldsigev)
1257fd3bf775SJohn Dyson {
1258b40ce416SJulian Elischer 	struct proc *p = td->td_proc;
12592244ea07SJohn Dyson 	struct file *fp;
1260bfbbc4aaSJason Evans 	struct socket *so;
12612244ea07SJohn Dyson 	struct aiocblist *aiocbe;
1262b40ce416SJulian Elischer 	struct aiothreadlist *aiop;
12632244ea07SJohn Dyson 	struct kaioinfo *ki;
1264c6fa9f78SAlan Cox 	struct kevent kev;
1265c6fa9f78SAlan Cox 	struct kqueue *kq;
1266c6fa9f78SAlan Cox 	struct file *kq_fp;
1267576c004fSAlfred Perlstein 	struct sockbuf *sb;
12681ce91824SDavid Xu 	int opcode;
12691ce91824SDavid Xu 	int error;
12701ce91824SDavid Xu 	int fd;
12711ce91824SDavid Xu 	int jid;
12722244ea07SJohn Dyson 
1273a9bf5e37SDavid Xu 	if (p->p_aioinfo == NULL)
1274a9bf5e37SDavid Xu 		aio_init_aioinfo(p);
1275a9bf5e37SDavid Xu 
12761ce91824SDavid Xu 	ki = p->p_aioinfo;
12771ce91824SDavid Xu 
1278a9bf5e37SDavid Xu 	suword(&job->_aiocb_private.status, -1);
1279a9bf5e37SDavid Xu 	suword(&job->_aiocb_private.error, 0);
1280a9bf5e37SDavid Xu 	suword(&job->_aiocb_private.kernelinfo, -1);
1281a9bf5e37SDavid Xu 
1282a9bf5e37SDavid Xu 	if (num_queue_count >= max_queue_count ||
1283a9bf5e37SDavid Xu 	    ki->kaio_count >= ki->kaio_qallowed_count) {
1284a9bf5e37SDavid Xu 		suword(&job->_aiocb_private.error, EAGAIN);
1285a9bf5e37SDavid Xu 		return (EAGAIN);
1286a9bf5e37SDavid Xu 	}
1287a9bf5e37SDavid Xu 
12881ce91824SDavid Xu 	aiocbe = uma_zalloc(aiocb_zone, M_WAITOK | M_ZERO);
1289fd3bf775SJohn Dyson 	aiocbe->inputcharge = 0;
1290fd3bf775SJohn Dyson 	aiocbe->outputcharge = 0;
12911ce91824SDavid Xu 	knlist_init(&aiocbe->klist, &p->p_mtx, NULL, NULL, NULL);
1292fd3bf775SJohn Dyson 
12930972628aSDavid Xu 	if (oldsigev) {
12940972628aSDavid Xu 		bzero(&aiocbe->uaiocb, sizeof(struct aiocb));
12950972628aSDavid Xu 		error = copyin(job, &aiocbe->uaiocb, sizeof(struct oaiocb));
12960972628aSDavid Xu 		bcopy(&aiocbe->uaiocb.__spare__, &aiocbe->uaiocb.aio_sigevent,
12970972628aSDavid Xu 			sizeof(struct osigevent));
12980972628aSDavid Xu 	} else {
12990972628aSDavid Xu 		error = copyin(job, &aiocbe->uaiocb, sizeof(struct aiocb));
13000972628aSDavid Xu 	}
13012244ea07SJohn Dyson 	if (error) {
1302fd3bf775SJohn Dyson 		suword(&job->_aiocb_private.error, error);
1303c897b813SJeff Roberson 		uma_zfree(aiocb_zone, aiocbe);
1304ac41f2efSAlfred Perlstein 		return (error);
13052244ea07SJohn Dyson 	}
130668d71118SDavid Xu 
130768d71118SDavid Xu 	if (aiocbe->uaiocb.aio_sigevent.sigev_notify != SIGEV_KEVENT &&
130868d71118SDavid Xu 	    aiocbe->uaiocb.aio_sigevent.sigev_notify != SIGEV_SIGNAL &&
130968d71118SDavid Xu 	    aiocbe->uaiocb.aio_sigevent.sigev_notify != SIGEV_THREAD_ID &&
131068d71118SDavid Xu 	    aiocbe->uaiocb.aio_sigevent.sigev_notify != SIGEV_NONE) {
131168d71118SDavid Xu 		suword(&job->_aiocb_private.error, EINVAL);
131268d71118SDavid Xu 		uma_zfree(aiocb_zone, aiocbe);
131368d71118SDavid Xu 		return (EINVAL);
131468d71118SDavid Xu 	}
131568d71118SDavid Xu 
13164c0fb2cfSDavid Xu 	if ((aiocbe->uaiocb.aio_sigevent.sigev_notify == SIGEV_SIGNAL ||
13174c0fb2cfSDavid Xu 	     aiocbe->uaiocb.aio_sigevent.sigev_notify == SIGEV_THREAD_ID) &&
13182f3cf918SAlfred Perlstein 		!_SIG_VALID(aiocbe->uaiocb.aio_sigevent.sigev_signo)) {
1319c897b813SJeff Roberson 		uma_zfree(aiocb_zone, aiocbe);
1320ac41f2efSAlfred Perlstein 		return (EINVAL);
13212f3cf918SAlfred Perlstein 	}
13222244ea07SJohn Dyson 
13234c0fb2cfSDavid Xu 	ksiginfo_init(&aiocbe->ksi);
13244c0fb2cfSDavid Xu 
1325bfbbc4aaSJason Evans 	/* Save userspace address of the job info. */
132611783b14SJohn Dyson 	aiocbe->uuaiocb = job;
132711783b14SJohn Dyson 
1328bfbbc4aaSJason Evans 	/* Get the opcode. */
1329bfbbc4aaSJason Evans 	if (type != LIO_NOP)
1330a624e84fSJohn Dyson 		aiocbe->uaiocb.aio_lio_opcode = type;
1331a624e84fSJohn Dyson 	opcode = aiocbe->uaiocb.aio_lio_opcode;
13322244ea07SJohn Dyson 
13332a522eb9SJohn Baldwin 	/* Fetch the file object for the specified file descriptor. */
13342244ea07SJohn Dyson 	fd = aiocbe->uaiocb.aio_fildes;
13352a522eb9SJohn Baldwin 	switch (opcode) {
13362a522eb9SJohn Baldwin 	case LIO_WRITE:
13372a522eb9SJohn Baldwin 		error = fget_write(td, fd, &fp);
13382a522eb9SJohn Baldwin 		break;
13392a522eb9SJohn Baldwin 	case LIO_READ:
13402a522eb9SJohn Baldwin 		error = fget_read(td, fd, &fp);
13412a522eb9SJohn Baldwin 		break;
13422a522eb9SJohn Baldwin 	default:
13432a522eb9SJohn Baldwin 		error = fget(td, fd, &fp);
13442a522eb9SJohn Baldwin 	}
13452a522eb9SJohn Baldwin 	if (error) {
1346c897b813SJeff Roberson 		uma_zfree(aiocb_zone, aiocbe);
134755a122bfSDavid Xu 		suword(&job->_aiocb_private.error, error);
1348af56abaaSJohn Baldwin 		return (error);
13492244ea07SJohn Dyson 	}
13502a522eb9SJohn Baldwin 	aiocbe->fd_file = fp;
13512244ea07SJohn Dyson 
13522244ea07SJohn Dyson 	if (aiocbe->uaiocb.aio_offset == -1LL) {
1353ae124fc4SAlan Cox 		error = EINVAL;
1354ae124fc4SAlan Cox 		goto aqueue_fail;
13552244ea07SJohn Dyson 	}
13561ce91824SDavid Xu 
13571ce91824SDavid Xu 	mtx_lock(&aio_job_mtx);
13581ce91824SDavid Xu 	jid = jobrefid;
13592d2f8ae7SBruce Evans 	if (jobrefid == LONG_MAX)
1360fd3bf775SJohn Dyson 		jobrefid = 1;
13612d2f8ae7SBruce Evans 	else
13622d2f8ae7SBruce Evans 		jobrefid++;
13631ce91824SDavid Xu 	mtx_unlock(&aio_job_mtx);
13641ce91824SDavid Xu 
13651ce91824SDavid Xu 	error = suword(&job->_aiocb_private.kernelinfo, jid);
13661ce91824SDavid Xu 	if (error) {
13671ce91824SDavid Xu 		error = EINVAL;
13681ce91824SDavid Xu 		goto aqueue_fail;
13691ce91824SDavid Xu 	}
13701ce91824SDavid Xu 	aiocbe->uaiocb._aiocb_private.kernelinfo = (void *)(intptr_t)jid;
13712244ea07SJohn Dyson 
13722244ea07SJohn Dyson 	if (opcode == LIO_NOP) {
1373a5c0b1c0SAlan Cox 		fdrop(fp, td);
1374c897b813SJeff Roberson 		uma_zfree(aiocb_zone, aiocbe);
1375ac41f2efSAlfred Perlstein 		return (0);
13762244ea07SJohn Dyson 	}
1377fd3bf775SJohn Dyson 	if ((opcode != LIO_READ) && (opcode != LIO_WRITE)) {
1378ae124fc4SAlan Cox 		error = EINVAL;
1379ae124fc4SAlan Cox 		goto aqueue_fail;
13802244ea07SJohn Dyson 	}
13812244ea07SJohn Dyson 
1382c6fa9f78SAlan Cox 	if (aiocbe->uaiocb.aio_sigevent.sigev_notify == SIGEV_KEVENT) {
1383c6fa9f78SAlan Cox 		kev.ident = aiocbe->uaiocb.aio_sigevent.sigev_notify_kqueue;
1384bbe7bbdfSAlan Cox 	} else
1385cb679c38SJonathan Lemon 		goto no_kqueue;
13862a522eb9SJohn Baldwin 	error = fget(td, (u_int)kev.ident, &kq_fp);
13872a522eb9SJohn Baldwin 	if (error)
13882a522eb9SJohn Baldwin 		goto aqueue_fail;
13892a522eb9SJohn Baldwin 	if (kq_fp->f_type != DTYPE_KQUEUE) {
13902a522eb9SJohn Baldwin 		fdrop(kq_fp, td);
1391cb679c38SJonathan Lemon 		error = EBADF;
1392cb679c38SJonathan Lemon 		goto aqueue_fail;
1393cb679c38SJonathan Lemon 	}
139448e3128bSMatthew Dillon 	kq = kq_fp->f_data;
1395b46f1c55SAlan Cox 	kev.ident = (uintptr_t)aiocbe->uuaiocb;
1396cb679c38SJonathan Lemon 	kev.filter = EVFILT_AIO;
1397cb679c38SJonathan Lemon 	kev.flags = EV_ADD | EV_ENABLE | EV_FLAG1;
1398b46f1c55SAlan Cox 	kev.data = (intptr_t)aiocbe;
13991ce91824SDavid Xu 	kev.udata = aiocbe->uaiocb.aio_sigevent.sigev_value.sival_ptr;
1400ad3b9257SJohn-Mark Gurney 	error = kqueue_register(kq, &kev, td, 1);
14012a522eb9SJohn Baldwin 	fdrop(kq_fp, td);
1402cb679c38SJonathan Lemon aqueue_fail:
1403cb679c38SJonathan Lemon 	if (error) {
1404a5c0b1c0SAlan Cox 		fdrop(fp, td);
1405c897b813SJeff Roberson 		uma_zfree(aiocb_zone, aiocbe);
1406cb679c38SJonathan Lemon 		suword(&job->_aiocb_private.error, error);
1407279d7226SMatthew Dillon 		goto done;
1408cb679c38SJonathan Lemon 	}
1409cb679c38SJonathan Lemon no_kqueue:
1410cb679c38SJonathan Lemon 
1411fd3bf775SJohn Dyson 	suword(&job->_aiocb_private.error, EINPROGRESS);
1412fd3bf775SJohn Dyson 	aiocbe->uaiocb._aiocb_private.error = EINPROGRESS;
14132244ea07SJohn Dyson 	aiocbe->userproc = p;
1414f8f750c5SRobert Watson 	aiocbe->cred = crhold(td->td_ucred);
14152244ea07SJohn Dyson 	aiocbe->jobflags = 0;
141684af4da6SJohn Dyson 	aiocbe->lio = lj;
14172244ea07SJohn Dyson 
1418bfbbc4aaSJason Evans 	if (fp->f_type == DTYPE_SOCKET) {
1419bfbbc4aaSJason Evans 		/*
1420bfbbc4aaSJason Evans 		 * Alternate queueing for socket ops: Reach down into the
1421bfbbc4aaSJason Evans 		 * descriptor to get the socket data.  Then check to see if the
1422bfbbc4aaSJason Evans 		 * socket is ready to be read or written (based on the requested
1423bfbbc4aaSJason Evans 		 * operation).
1424bfbbc4aaSJason Evans 		 *
1425bfbbc4aaSJason Evans 		 * If it is not ready for io, then queue the aiocbe on the
1426bfbbc4aaSJason Evans 		 * socket, and set the flags so we get a call when sbnotify()
1427bfbbc4aaSJason Evans 		 * happens.
1428576c004fSAlfred Perlstein 		 *
1429576c004fSAlfred Perlstein 		 * Note if opcode is neither LIO_WRITE nor LIO_READ we lock
1430576c004fSAlfred Perlstein 		 * and unlock the snd sockbuf for no reason.
1431bfbbc4aaSJason Evans 		 */
143248e3128bSMatthew Dillon 		so = fp->f_data;
1433576c004fSAlfred Perlstein 		sb = (opcode == LIO_READ) ? &so->so_rcv : &so->so_snd;
1434576c004fSAlfred Perlstein 		SOCKBUF_LOCK(sb);
1435bfbbc4aaSJason Evans 		if (((opcode == LIO_READ) && (!soreadable(so))) || ((opcode ==
1436bfbbc4aaSJason Evans 		    LIO_WRITE) && (!sowriteable(so)))) {
1437576c004fSAlfred Perlstein 			sb->sb_flags |= SB_AIO;
14381aa4c324SDavid Xu 
14391aa4c324SDavid Xu 			mtx_lock(&aio_job_mtx);
14401aa4c324SDavid Xu 			TAILQ_INSERT_TAIL(&so->so_aiojobq, aiocbe, list);
14411aa4c324SDavid Xu 			mtx_unlock(&aio_job_mtx);
14421aa4c324SDavid Xu 
14431ce91824SDavid Xu 			PROC_LOCK(p);
14441ce91824SDavid Xu 			TAILQ_INSERT_TAIL(&ki->kaio_all, aiocbe, allist);
14451aa4c324SDavid Xu 			TAILQ_INSERT_TAIL(&ki->kaio_jobqueue, aiocbe, plist);
14461ce91824SDavid Xu 			aiocbe->jobstate = JOBST_JOBQSOCK;
14471ce91824SDavid Xu 			ki->kaio_count++;
14481ce91824SDavid Xu 			if (lj)
14491ce91824SDavid Xu 				lj->lioj_count++;
14501ce91824SDavid Xu 			PROC_UNLOCK(p);
1451576c004fSAlfred Perlstein 			SOCKBUF_UNLOCK(sb);
14521ce91824SDavid Xu 			atomic_add_int(&num_queue_count, 1);
1453279d7226SMatthew Dillon 			error = 0;
1454279d7226SMatthew Dillon 			goto done;
1455bfbbc4aaSJason Evans 		}
1456576c004fSAlfred Perlstein 		SOCKBUF_UNLOCK(sb);
1457bfbbc4aaSJason Evans 	}
1458bfbbc4aaSJason Evans 
1459bfbbc4aaSJason Evans 	if ((error = aio_qphysio(p, aiocbe)) == 0)
1460279d7226SMatthew Dillon 		goto done;
14611ce91824SDavid Xu #if 0
1462279d7226SMatthew Dillon 	if (error > 0) {
1463fd3bf775SJohn Dyson 		aiocbe->uaiocb._aiocb_private.error = error;
1464fd3bf775SJohn Dyson 		suword(&job->_aiocb_private.error, error);
1465279d7226SMatthew Dillon 		goto done;
1466fd3bf775SJohn Dyson 	}
14671ce91824SDavid Xu #endif
1468bfbbc4aaSJason Evans 	/* No buffer for daemon I/O. */
146984af4da6SJohn Dyson 	aiocbe->bp = NULL;
147084af4da6SJohn Dyson 
14711ce91824SDavid Xu 	PROC_LOCK(p);
14721ce91824SDavid Xu 	ki->kaio_count++;
1473bfbbc4aaSJason Evans 	if (lj)
14741ce91824SDavid Xu 		lj->lioj_count++;
1475fd3bf775SJohn Dyson 	TAILQ_INSERT_TAIL(&ki->kaio_jobqueue, aiocbe, plist);
14761ce91824SDavid Xu 	TAILQ_INSERT_TAIL(&ki->kaio_all, aiocbe, allist);
1477fd3bf775SJohn Dyson 
14781ce91824SDavid Xu 	mtx_lock(&aio_job_mtx);
14791ce91824SDavid Xu 	TAILQ_INSERT_TAIL(&aio_jobs, aiocbe, list);
14801ce91824SDavid Xu 	aiocbe->jobstate = JOBST_JOBQGLOBAL;
14811ce91824SDavid Xu 	PROC_UNLOCK(p);
14821ce91824SDavid Xu 
14831ce91824SDavid Xu 	atomic_add_int(&num_queue_count, 1);
1484fd3bf775SJohn Dyson 
1485fd3bf775SJohn Dyson 	/*
1486bfbbc4aaSJason Evans 	 * If we don't have a free AIO process, and we are below our quota, then
1487bfbbc4aaSJason Evans 	 * start one.  Otherwise, depend on the subsequent I/O completions to
1488bfbbc4aaSJason Evans 	 * pick-up this job.  If we don't sucessfully create the new process
1489bfbbc4aaSJason Evans 	 * (thread) due to resource issues, we return an error for now (EAGAIN),
1490bfbbc4aaSJason Evans 	 * which is likely not the correct thing to do.
1491fd3bf775SJohn Dyson 	 */
1492c6c191b2SAlan Cox retryproc:
14931ce91824SDavid Xu 	error = 0;
1494d254af07SMatthew Dillon 	if ((aiop = TAILQ_FIRST(&aio_freeproc)) != NULL) {
14952244ea07SJohn Dyson 		TAILQ_REMOVE(&aio_freeproc, aiop, list);
1496b40ce416SJulian Elischer 		aiop->aiothreadflags &= ~AIOP_FREE;
1497b40ce416SJulian Elischer 		wakeup(aiop->aiothread);
1498fd3bf775SJohn Dyson 	} else if (((num_aio_resv_start + num_aio_procs) < max_aio_procs) &&
1499fd3bf775SJohn Dyson 	    ((ki->kaio_active_count + num_aio_resv_start) <
1500fd3bf775SJohn Dyson 	    ki->kaio_maxactive_count)) {
1501fd3bf775SJohn Dyson 		num_aio_resv_start++;
15021ce91824SDavid Xu 		mtx_unlock(&aio_job_mtx);
15031ce91824SDavid Xu 		error = aio_newproc(&num_aio_resv_start);
15041ce91824SDavid Xu 		mtx_lock(&aio_job_mtx);
15051ce91824SDavid Xu 		if (error) {
150684af4da6SJohn Dyson 			num_aio_resv_start--;
15072244ea07SJohn Dyson 			goto retryproc;
1508fd3bf775SJohn Dyson 		}
15091ce91824SDavid Xu 	}
15101ce91824SDavid Xu 	mtx_unlock(&aio_job_mtx);
15111ce91824SDavid Xu 
1512279d7226SMatthew Dillon done:
1513ac41f2efSAlfred Perlstein 	return (error);
15142244ea07SJohn Dyson }
15152244ea07SJohn Dyson 
1516fd3bf775SJohn Dyson /*
1517bfbbc4aaSJason Evans  * Support the aio_return system call, as a side-effect, kernel resources are
1518bfbbc4aaSJason Evans  * released.
15192244ea07SJohn Dyson  */
15202244ea07SJohn Dyson int
1521b40ce416SJulian Elischer aio_return(struct thread *td, struct aio_return_args *uap)
1522fd3bf775SJohn Dyson {
1523b40ce416SJulian Elischer 	struct proc *p = td->td_proc;
15241ce91824SDavid Xu 	struct aiocblist *cb;
15251ce91824SDavid Xu 	struct aiocb *uaiocb;
15262244ea07SJohn Dyson 	struct kaioinfo *ki;
15271ce91824SDavid Xu 	int status, error;
15282244ea07SJohn Dyson 
1529c0bf5caaSAlan Cox 	ki = p->p_aioinfo;
1530c0bf5caaSAlan Cox 	if (ki == NULL)
1531ac41f2efSAlfred Perlstein 		return (EINVAL);
15321ce91824SDavid Xu 	uaiocb = uap->aiocbp;
15333769f562SAlan Cox 	PROC_LOCK(p);
15341ce91824SDavid Xu 	TAILQ_FOREACH(cb, &ki->kaio_done, plist) {
15351ce91824SDavid Xu 		if (cb->uuaiocb == uaiocb)
1536c0bf5caaSAlan Cox 			break;
1537c0bf5caaSAlan Cox 	}
1538c0bf5caaSAlan Cox 	if (cb != NULL) {
15391ce91824SDavid Xu 		MPASS(cb->jobstate == JOBST_JOBFINISHED);
15401ce91824SDavid Xu 		status = cb->uaiocb._aiocb_private.status;
15411ce91824SDavid Xu 		error = cb->uaiocb._aiocb_private.error;
15421ce91824SDavid Xu 		td->td_retval[0] = status;
154369cd28daSDoug Ambrisko 		if (cb->uaiocb.aio_lio_opcode == LIO_WRITE) {
154469cd28daSDoug Ambrisko 			p->p_stats->p_ru.ru_oublock +=
154569cd28daSDoug Ambrisko 			    cb->outputcharge;
154669cd28daSDoug Ambrisko 			cb->outputcharge = 0;
154769cd28daSDoug Ambrisko 		} else if (cb->uaiocb.aio_lio_opcode == LIO_READ) {
154869cd28daSDoug Ambrisko 			p->p_stats->p_ru.ru_inblock += cb->inputcharge;
154969cd28daSDoug Ambrisko 			cb->inputcharge = 0;
155069cd28daSDoug Ambrisko 		}
155184af4da6SJohn Dyson 		aio_free_entry(cb);
155255a122bfSDavid Xu 		PROC_UNLOCK(p);
15531ce91824SDavid Xu 		suword(&uaiocb->_aiocb_private.error, error);
15541ce91824SDavid Xu 		suword(&uaiocb->_aiocb_private.status, status);
155555a122bfSDavid Xu 	} else {
15561ce91824SDavid Xu 		error = EINVAL;
15571ce91824SDavid Xu 		PROC_UNLOCK(p);
155855a122bfSDavid Xu 	}
15591ce91824SDavid Xu 	return (error);
15602244ea07SJohn Dyson }
15612244ea07SJohn Dyson 
15622244ea07SJohn Dyson /*
1563bfbbc4aaSJason Evans  * Allow a process to wakeup when any of the I/O requests are completed.
15642244ea07SJohn Dyson  */
15652244ea07SJohn Dyson int
1566b40ce416SJulian Elischer aio_suspend(struct thread *td, struct aio_suspend_args *uap)
1567fd3bf775SJohn Dyson {
1568b40ce416SJulian Elischer 	struct proc *p = td->td_proc;
15694a11ca4eSPoul-Henning Kamp 	struct timeval atv;
15702244ea07SJohn Dyson 	struct timespec ts;
15712244ea07SJohn Dyson 	struct aiocb *const *cbptr, *cbp;
15722244ea07SJohn Dyson 	struct kaioinfo *ki;
15731ce91824SDavid Xu 	struct aiocblist *cb, *cbfirst;
157411783b14SJohn Dyson 	struct aiocb **ujoblist;
15751ce91824SDavid Xu 	int njoblist;
15761ce91824SDavid Xu 	int error;
15771ce91824SDavid Xu 	int timo;
15781ce91824SDavid Xu 	int i;
15792244ea07SJohn Dyson 
1580ae3b195fSTim J. Robbins 	if (uap->nent < 0 || uap->nent > AIO_LISTIO_MAX)
1581ac41f2efSAlfred Perlstein 		return (EINVAL);
15822244ea07SJohn Dyson 
15832244ea07SJohn Dyson 	timo = 0;
15842244ea07SJohn Dyson 	if (uap->timeout) {
1585bfbbc4aaSJason Evans 		/* Get timespec struct. */
1586bfbbc4aaSJason Evans 		if ((error = copyin(uap->timeout, &ts, sizeof(ts))) != 0)
1587ac41f2efSAlfred Perlstein 			return (error);
15882244ea07SJohn Dyson 
15892244ea07SJohn Dyson 		if (ts.tv_nsec < 0 || ts.tv_nsec >= 1000000000)
15902244ea07SJohn Dyson 			return (EINVAL);
15912244ea07SJohn Dyson 
1592e3b3ba2dSDag-Erling Smørgrav 		TIMESPEC_TO_TIMEVAL(&atv, &ts);
15932244ea07SJohn Dyson 		if (itimerfix(&atv))
15942244ea07SJohn Dyson 			return (EINVAL);
1595227ee8a1SPoul-Henning Kamp 		timo = tvtohz(&atv);
15962244ea07SJohn Dyson 	}
15972244ea07SJohn Dyson 
15982244ea07SJohn Dyson 	ki = p->p_aioinfo;
15992244ea07SJohn Dyson 	if (ki == NULL)
1600ac41f2efSAlfred Perlstein 		return (EAGAIN);
16012244ea07SJohn Dyson 
160284af4da6SJohn Dyson 	njoblist = 0;
1603a163d034SWarner Losh 	ujoblist = uma_zalloc(aiol_zone, M_WAITOK);
16042244ea07SJohn Dyson 	cbptr = uap->aiocbp;
16052244ea07SJohn Dyson 
16062244ea07SJohn Dyson 	for (i = 0; i < uap->nent; i++) {
1607a739e09cSAlan Cox 		cbp = (struct aiocb *)(intptr_t)fuword(&cbptr[i]);
160884af4da6SJohn Dyson 		if (cbp == 0)
160984af4da6SJohn Dyson 			continue;
161011783b14SJohn Dyson 		ujoblist[njoblist] = cbp;
161184af4da6SJohn Dyson 		njoblist++;
16122244ea07SJohn Dyson 	}
1613bfbbc4aaSJason Evans 
161411783b14SJohn Dyson 	if (njoblist == 0) {
1615c897b813SJeff Roberson 		uma_zfree(aiol_zone, ujoblist);
1616ac41f2efSAlfred Perlstein 		return (0);
161711783b14SJohn Dyson 	}
16182244ea07SJohn Dyson 
16193769f562SAlan Cox 	PROC_LOCK(p);
16201ce91824SDavid Xu 	for (;;) {
16211ce91824SDavid Xu 		cbfirst = NULL;
16221ce91824SDavid Xu 		error = 0;
16231ce91824SDavid Xu 		TAILQ_FOREACH(cb, &ki->kaio_all, allist) {
162484af4da6SJohn Dyson 			for (i = 0; i < njoblist; i++) {
16251ce91824SDavid Xu 				if (cb->uuaiocb == ujoblist[i]) {
16261ce91824SDavid Xu 					if (cbfirst == NULL)
16271ce91824SDavid Xu 						cbfirst = cb;
16281ce91824SDavid Xu 					if (cb->jobstate == JOBST_JOBFINISHED)
16291ce91824SDavid Xu 						goto RETURN;
163084af4da6SJohn Dyson 				}
163184af4da6SJohn Dyson 			}
163284af4da6SJohn Dyson 		}
16331ce91824SDavid Xu 		/* All tasks were finished. */
16341ce91824SDavid Xu 		if (cbfirst == NULL)
16351ce91824SDavid Xu 			break;
16362244ea07SJohn Dyson 
1637fd3bf775SJohn Dyson 		ki->kaio_flags |= KAIO_WAKEUP;
16381ce91824SDavid Xu 		error = msleep(&p->p_aioinfo, &p->p_mtx, PRIBIO | PCATCH,
16391ce91824SDavid Xu 		    "aiospn", timo);
16401ce91824SDavid Xu 		if (error == ERESTART)
16411ce91824SDavid Xu 			error = EINTR;
16421ce91824SDavid Xu 		if (error)
16431ce91824SDavid Xu 			break;
16442244ea07SJohn Dyson 	}
16451ce91824SDavid Xu RETURN:
16461ce91824SDavid Xu 	PROC_UNLOCK(p);
16471ce91824SDavid Xu 	uma_zfree(aiol_zone, ujoblist);
16481ce91824SDavid Xu 	return (error);
16492244ea07SJohn Dyson }
1650ee877a35SJohn Dyson 
1651ee877a35SJohn Dyson /*
1652dd85920aSJason Evans  * aio_cancel cancels any non-physio aio operations not currently in
1653dd85920aSJason Evans  * progress.
1654ee877a35SJohn Dyson  */
1655ee877a35SJohn Dyson int
1656b40ce416SJulian Elischer aio_cancel(struct thread *td, struct aio_cancel_args *uap)
1657fd3bf775SJohn Dyson {
1658b40ce416SJulian Elischer 	struct proc *p = td->td_proc;
1659dd85920aSJason Evans 	struct kaioinfo *ki;
1660dd85920aSJason Evans 	struct aiocblist *cbe, *cbn;
1661dd85920aSJason Evans 	struct file *fp;
1662dd85920aSJason Evans 	struct socket *so;
16631ce91824SDavid Xu 	int error;
16641aa4c324SDavid Xu 	int remove;
1665dd85920aSJason Evans 	int cancelled = 0;
1666dd85920aSJason Evans 	int notcancelled = 0;
1667dd85920aSJason Evans 	struct vnode *vp;
1668dd85920aSJason Evans 
16692a522eb9SJohn Baldwin 	/* Lookup file object. */
16701ce91824SDavid Xu 	error = fget(td, uap->fd, &fp);
16712a522eb9SJohn Baldwin 	if (error)
16722a522eb9SJohn Baldwin 		return (error);
1673dd85920aSJason Evans 
16741ce91824SDavid Xu 	ki = p->p_aioinfo;
16751ce91824SDavid Xu 	if (ki == NULL)
16761ce91824SDavid Xu 		goto done;
16771ce91824SDavid Xu 
1678dd85920aSJason Evans 	if (fp->f_type == DTYPE_VNODE) {
16793b6d9652SPoul-Henning Kamp 		vp = fp->f_vnode;
1680dd85920aSJason Evans 		if (vn_isdisk(vp, &error)) {
16812a522eb9SJohn Baldwin 			fdrop(fp, td);
1682b40ce416SJulian Elischer 			td->td_retval[0] = AIO_NOTCANCELED;
1683ac41f2efSAlfred Perlstein 			return (0);
1684dd85920aSJason Evans 		}
1685dd85920aSJason Evans 	}
1686dd85920aSJason Evans 
16871ce91824SDavid Xu 	PROC_LOCK(p);
16882a522eb9SJohn Baldwin 	TAILQ_FOREACH_SAFE(cbe, &ki->kaio_jobqueue, plist, cbn) {
1689dd85920aSJason Evans 		if ((uap->fd == cbe->uaiocb.aio_fildes) &&
1690dd85920aSJason Evans 		    ((uap->aiocbp == NULL) ||
1691dd85920aSJason Evans 		     (uap->aiocbp == cbe->uuaiocb))) {
16921aa4c324SDavid Xu 			remove = 0;
16931aa4c324SDavid Xu 
16941ce91824SDavid Xu 			mtx_lock(&aio_job_mtx);
1695dd85920aSJason Evans 			if (cbe->jobstate == JOBST_JOBQGLOBAL) {
1696dd85920aSJason Evans 				TAILQ_REMOVE(&aio_jobs, cbe, list);
16971aa4c324SDavid Xu 				remove = 1;
16981aa4c324SDavid Xu 			} else if (cbe->jobstate == JOBST_JOBQSOCK) {
16991aa4c324SDavid Xu 				MPASS(fp->f_type == DTYPE_SOCKET);
17001aa4c324SDavid Xu 				so = fp->f_data;
17011aa4c324SDavid Xu 				TAILQ_REMOVE(&so->so_aiojobq, cbe, list);
17021aa4c324SDavid Xu 				remove = 1;
17031aa4c324SDavid Xu 			}
17041ce91824SDavid Xu 			mtx_unlock(&aio_job_mtx);
17051aa4c324SDavid Xu 
17061aa4c324SDavid Xu 			if (remove) {
17071ce91824SDavid Xu 				TAILQ_REMOVE(&ki->kaio_jobqueue, cbe, plist);
1708dd85920aSJason Evans 				cbe->uaiocb._aiocb_private.status = -1;
1709dd85920aSJason Evans 				cbe->uaiocb._aiocb_private.error = ECANCELED;
17101ce91824SDavid Xu 				aio_bio_done_notify(p, cbe, DONE_QUEUE);
17111ce91824SDavid Xu 				cancelled++;
1712dd85920aSJason Evans 			} else {
1713dd85920aSJason Evans 				notcancelled++;
1714dd85920aSJason Evans 			}
17151aa4c324SDavid Xu 			if (uap->aiocbp != NULL)
17161aa4c324SDavid Xu 				break;
1717dd85920aSJason Evans 		}
1718dd85920aSJason Evans 	}
17191ce91824SDavid Xu 	PROC_UNLOCK(p);
17201ce91824SDavid Xu 
1721ad49abc0SAlan Cox done:
17222a522eb9SJohn Baldwin 	fdrop(fp, td);
17231aa4c324SDavid Xu 
17241aa4c324SDavid Xu 	if (uap->aiocbp != NULL) {
1725dd85920aSJason Evans 		if (cancelled) {
1726b40ce416SJulian Elischer 			td->td_retval[0] = AIO_CANCELED;
1727ac41f2efSAlfred Perlstein 			return (0);
1728dd85920aSJason Evans 		}
17291aa4c324SDavid Xu 	}
17301aa4c324SDavid Xu 
17311aa4c324SDavid Xu 	if (notcancelled) {
17321aa4c324SDavid Xu 		td->td_retval[0] = AIO_NOTCANCELED;
17331aa4c324SDavid Xu 		return (0);
17341aa4c324SDavid Xu 	}
17351aa4c324SDavid Xu 
17361aa4c324SDavid Xu 	if (cancelled) {
17371aa4c324SDavid Xu 		td->td_retval[0] = AIO_CANCELED;
17381aa4c324SDavid Xu 		return (0);
17391aa4c324SDavid Xu 	}
17401aa4c324SDavid Xu 
1741b40ce416SJulian Elischer 	td->td_retval[0] = AIO_ALLDONE;
1742dd85920aSJason Evans 
1743ac41f2efSAlfred Perlstein 	return (0);
1744ee877a35SJohn Dyson }
1745ee877a35SJohn Dyson 
1746ee877a35SJohn Dyson /*
1747bfbbc4aaSJason Evans  * aio_error is implemented in the kernel level for compatibility purposes only.
1748bfbbc4aaSJason Evans  * For a user mode async implementation, it would be best to do it in a userland
1749bfbbc4aaSJason Evans  * subroutine.
1750ee877a35SJohn Dyson  */
1751ee877a35SJohn Dyson int
1752b40ce416SJulian Elischer aio_error(struct thread *td, struct aio_error_args *uap)
1753fd3bf775SJohn Dyson {
1754b40ce416SJulian Elischer 	struct proc *p = td->td_proc;
17552244ea07SJohn Dyson 	struct aiocblist *cb;
17562244ea07SJohn Dyson 	struct kaioinfo *ki;
17571ce91824SDavid Xu 	int status;
1758ee877a35SJohn Dyson 
17592244ea07SJohn Dyson 	ki = p->p_aioinfo;
17601ce91824SDavid Xu 	if (ki == NULL) {
17611ce91824SDavid Xu 		td->td_retval[0] = EINVAL;
17621ce91824SDavid Xu 		return (0);
17631ce91824SDavid Xu 	}
1764ee877a35SJohn Dyson 
17653769f562SAlan Cox 	PROC_LOCK(p);
17661ce91824SDavid Xu 	TAILQ_FOREACH(cb, &ki->kaio_all, allist) {
17671ce91824SDavid Xu 		if (cb->uuaiocb == uap->aiocbp) {
17681ce91824SDavid Xu 			if (cb->jobstate == JOBST_JOBFINISHED)
17691ce91824SDavid Xu 				td->td_retval[0] =
17701ce91824SDavid Xu 					cb->uaiocb._aiocb_private.error;
17711ce91824SDavid Xu 			else
1772b40ce416SJulian Elischer 				td->td_retval[0] = EINPROGRESS;
17731ce91824SDavid Xu 			PROC_UNLOCK(p);
1774ac41f2efSAlfred Perlstein 			return (0);
17752244ea07SJohn Dyson 		}
17762244ea07SJohn Dyson 	}
17773769f562SAlan Cox 	PROC_UNLOCK(p);
177884af4da6SJohn Dyson 
17792244ea07SJohn Dyson 	/*
1780a9bf5e37SDavid Xu 	 * Hack for failure of aio_aqueue.
17812244ea07SJohn Dyson 	 */
17822244ea07SJohn Dyson 	status = fuword(&uap->aiocbp->_aiocb_private.status);
17831ce91824SDavid Xu 	if (status == -1) {
17841ce91824SDavid Xu 		td->td_retval[0] = fuword(&uap->aiocbp->_aiocb_private.error);
17851ce91824SDavid Xu 		return (0);
17861ce91824SDavid Xu 	}
17871ce91824SDavid Xu 
17881ce91824SDavid Xu 	td->td_retval[0] = EINVAL;
17891ce91824SDavid Xu 	return (0);
1790ee877a35SJohn Dyson }
1791ee877a35SJohn Dyson 
1792eb8e6d52SEivind Eklund /* syscall - asynchronous read from a file (REALTIME) */
1793ee877a35SJohn Dyson int
17940972628aSDavid Xu oaio_read(struct thread *td, struct oaio_read_args *uap)
17950972628aSDavid Xu {
17960972628aSDavid Xu 
1797a9bf5e37SDavid Xu 	return aio_aqueue(td, (struct aiocb *)uap->aiocbp, NULL, LIO_READ, 1);
17980972628aSDavid Xu }
17990972628aSDavid Xu 
18000972628aSDavid Xu int
1801b40ce416SJulian Elischer aio_read(struct thread *td, struct aio_read_args *uap)
1802fd3bf775SJohn Dyson {
180321d56e9cSAlfred Perlstein 
1804a9bf5e37SDavid Xu 	return aio_aqueue(td, uap->aiocbp, NULL, LIO_READ, 0);
1805ee877a35SJohn Dyson }
1806ee877a35SJohn Dyson 
1807eb8e6d52SEivind Eklund /* syscall - asynchronous write to a file (REALTIME) */
1808ee877a35SJohn Dyson int
18090972628aSDavid Xu oaio_write(struct thread *td, struct oaio_write_args *uap)
18100972628aSDavid Xu {
18110972628aSDavid Xu 
1812a9bf5e37SDavid Xu 	return aio_aqueue(td, (struct aiocb *)uap->aiocbp, NULL, LIO_WRITE, 1);
18130972628aSDavid Xu }
18140972628aSDavid Xu 
18150972628aSDavid Xu int
1816b40ce416SJulian Elischer aio_write(struct thread *td, struct aio_write_args *uap)
1817fd3bf775SJohn Dyson {
181821d56e9cSAlfred Perlstein 
1819a9bf5e37SDavid Xu 	return aio_aqueue(td, uap->aiocbp, NULL, LIO_WRITE, 0);
18200972628aSDavid Xu }
18210972628aSDavid Xu 
18220972628aSDavid Xu /* syscall - list directed I/O (REALTIME) */
18230972628aSDavid Xu int
18240972628aSDavid Xu olio_listio(struct thread *td, struct olio_listio_args *uap)
18250972628aSDavid Xu {
18260972628aSDavid Xu 	return do_lio_listio(td, (struct lio_listio_args *)uap, 1);
1827ee877a35SJohn Dyson }
1828ee877a35SJohn Dyson 
182944a2c818STim J. Robbins /* syscall - list directed I/O (REALTIME) */
1830ee877a35SJohn Dyson int
1831b40ce416SJulian Elischer lio_listio(struct thread *td, struct lio_listio_args *uap)
1832fd3bf775SJohn Dyson {
18330972628aSDavid Xu 	return do_lio_listio(td, uap, 0);
18340972628aSDavid Xu }
18350972628aSDavid Xu 
18360972628aSDavid Xu static int
18370972628aSDavid Xu do_lio_listio(struct thread *td, struct lio_listio_args *uap, int oldsigev)
18380972628aSDavid Xu {
1839b40ce416SJulian Elischer 	struct proc *p = td->td_proc;
18402244ea07SJohn Dyson 	struct aiocb *iocb, * const *cbptr;
18412244ea07SJohn Dyson 	struct kaioinfo *ki;
18421ce91824SDavid Xu 	struct aioliojob *lj;
184369cd28daSDoug Ambrisko 	struct kevent kev;
184469cd28daSDoug Ambrisko 	struct kqueue * kq;
184569cd28daSDoug Ambrisko 	struct file *kq_fp;
18461ce91824SDavid Xu 	int nent;
18471ce91824SDavid Xu 	int error;
1848fd3bf775SJohn Dyson 	int nerror;
1849ee877a35SJohn Dyson 	int i;
1850ee877a35SJohn Dyson 
1851bfbbc4aaSJason Evans 	if ((uap->mode != LIO_NOWAIT) && (uap->mode != LIO_WAIT))
1852ac41f2efSAlfred Perlstein 		return (EINVAL);
18532244ea07SJohn Dyson 
18542244ea07SJohn Dyson 	nent = uap->nent;
1855ae3b195fSTim J. Robbins 	if (nent < 0 || nent > AIO_LISTIO_MAX)
1856ac41f2efSAlfred Perlstein 		return (EINVAL);
18572244ea07SJohn Dyson 
1858bfbbc4aaSJason Evans 	if (p->p_aioinfo == NULL)
18592244ea07SJohn Dyson 		aio_init_aioinfo(p);
18602244ea07SJohn Dyson 
18612244ea07SJohn Dyson 	ki = p->p_aioinfo;
18622244ea07SJohn Dyson 
1863a163d034SWarner Losh 	lj = uma_zalloc(aiolio_zone, M_WAITOK);
186484af4da6SJohn Dyson 	lj->lioj_flags = 0;
18651ce91824SDavid Xu 	lj->lioj_count = 0;
18661ce91824SDavid Xu 	lj->lioj_finished_count = 0;
18671ce91824SDavid Xu 	knlist_init(&lj->klist, &p->p_mtx, NULL, NULL, NULL);
18684c0fb2cfSDavid Xu 	ksiginfo_init(&lj->lioj_ksi);
186969cd28daSDoug Ambrisko 
187084af4da6SJohn Dyson 	/*
1871bfbbc4aaSJason Evans 	 * Setup signal.
187284af4da6SJohn Dyson 	 */
187384af4da6SJohn Dyson 	if (uap->sig && (uap->mode == LIO_NOWAIT)) {
18740972628aSDavid Xu 		bzero(&lj->lioj_signal, sizeof(&lj->lioj_signal));
1875bfbbc4aaSJason Evans 		error = copyin(uap->sig, &lj->lioj_signal,
18760972628aSDavid Xu 				oldsigev ? sizeof(struct osigevent) :
18770972628aSDavid Xu 					   sizeof(struct sigevent));
18782f3cf918SAlfred Perlstein 		if (error) {
1879c897b813SJeff Roberson 			uma_zfree(aiolio_zone, lj);
1880ac41f2efSAlfred Perlstein 			return (error);
18812f3cf918SAlfred Perlstein 		}
188269cd28daSDoug Ambrisko 
188369cd28daSDoug Ambrisko 		if (lj->lioj_signal.sigev_notify == SIGEV_KEVENT) {
188469cd28daSDoug Ambrisko 			/* Assume only new style KEVENT */
18851ce91824SDavid Xu 			error = fget(td, lj->lioj_signal.sigev_notify_kqueue,
18861ce91824SDavid Xu 				&kq_fp);
18872a522eb9SJohn Baldwin 			if (error) {
18882a522eb9SJohn Baldwin 				uma_zfree(aiolio_zone, lj);
18892a522eb9SJohn Baldwin 				return (error);
18902a522eb9SJohn Baldwin 			}
18912a522eb9SJohn Baldwin 			if (kq_fp->f_type != DTYPE_KQUEUE) {
18922a522eb9SJohn Baldwin 				fdrop(kq_fp, td);
1893c897b813SJeff Roberson 				uma_zfree(aiolio_zone, lj);
189469cd28daSDoug Ambrisko 				return (EBADF);
18952f3cf918SAlfred Perlstein 			}
189669cd28daSDoug Ambrisko 			kq = (struct kqueue *)kq_fp->f_data;
189769cd28daSDoug Ambrisko 			kev.filter = EVFILT_LIO;
189869cd28daSDoug Ambrisko 			kev.flags = EV_ADD | EV_ENABLE | EV_FLAG1;
189969cd28daSDoug Ambrisko 			kev.ident = (uintptr_t)lj; /* something unique */
190069cd28daSDoug Ambrisko 			kev.data = (intptr_t)lj;
19011ce91824SDavid Xu 			/* pass user defined sigval data */
19021ce91824SDavid Xu 			kev.udata = lj->lioj_signal.sigev_value.sival_ptr;
190369cd28daSDoug Ambrisko 			error = kqueue_register(kq, &kev, td, 1);
19042a522eb9SJohn Baldwin 			fdrop(kq_fp, td);
190569cd28daSDoug Ambrisko 			if (error) {
190669cd28daSDoug Ambrisko 				uma_zfree(aiolio_zone, lj);
190769cd28daSDoug Ambrisko 				return (error);
190869cd28daSDoug Ambrisko 			}
19091ce91824SDavid Xu 		} else if (lj->lioj_signal.sigev_notify == SIGEV_NONE) {
19101ce91824SDavid Xu 			;
191168d71118SDavid Xu 		} else if (lj->lioj_signal.sigev_notify == SIGEV_SIGNAL ||
191268d71118SDavid Xu 			   lj->lioj_signal.sigev_notify == SIGEV_THREAD_ID) {
191368d71118SDavid Xu 				if (!_SIG_VALID(lj->lioj_signal.sigev_signo)) {
191469cd28daSDoug Ambrisko 					uma_zfree(aiolio_zone, lj);
191569cd28daSDoug Ambrisko 					return EINVAL;
191668d71118SDavid Xu 				}
191784af4da6SJohn Dyson 				lj->lioj_flags |= LIOJ_SIGNAL;
191868d71118SDavid Xu 		} else {
191968d71118SDavid Xu 			uma_zfree(aiolio_zone, lj);
192068d71118SDavid Xu 			return EINVAL;
19214d752b01SAlan Cox 		}
19221ce91824SDavid Xu 	}
192369cd28daSDoug Ambrisko 
19241ce91824SDavid Xu 	PROC_LOCK(p);
19252f3cf918SAlfred Perlstein 	TAILQ_INSERT_TAIL(&ki->kaio_liojoblist, lj, lioj_list);
19262244ea07SJohn Dyson 	/*
19271ce91824SDavid Xu 	 * Add extra aiocb count to avoid the lio to be freed
19281ce91824SDavid Xu 	 * by other threads doing aio_waitcomplete or aio_return,
19291ce91824SDavid Xu 	 * and prevent event from being sent until we have queued
19301ce91824SDavid Xu 	 * all tasks.
19311ce91824SDavid Xu 	 */
19321ce91824SDavid Xu 	lj->lioj_count = 1;
19331ce91824SDavid Xu 	PROC_UNLOCK(p);
19341ce91824SDavid Xu 
19351ce91824SDavid Xu 	/*
1936bfbbc4aaSJason Evans 	 * Get pointers to the list of I/O requests.
19372244ea07SJohn Dyson 	 */
1938fd3bf775SJohn Dyson 	nerror = 0;
19392244ea07SJohn Dyson 	cbptr = uap->acb_list;
19402244ea07SJohn Dyson 	for (i = 0; i < uap->nent; i++) {
1941a739e09cSAlan Cox 		iocb = (struct aiocb *)(intptr_t)fuword(&cbptr[i]);
19424a6a94d8SArchie Cobbs 		if (((intptr_t)iocb != -1) && ((intptr_t)iocb != 0)) {
1943a9bf5e37SDavid Xu 			error = aio_aqueue(td, iocb, lj, 0, oldsigev);
19441ce91824SDavid Xu 			if (error != 0)
1945fd3bf775SJohn Dyson 				nerror++;
1946fd3bf775SJohn Dyson 		}
1947fd3bf775SJohn Dyson 	}
19482244ea07SJohn Dyson 
19491ce91824SDavid Xu 	error = 0;
19501ce91824SDavid Xu 	PROC_LOCK(p);
19512244ea07SJohn Dyson 	if (uap->mode == LIO_WAIT) {
19521ce91824SDavid Xu 		while (lj->lioj_count - 1 != lj->lioj_finished_count) {
1953fd3bf775SJohn Dyson 			ki->kaio_flags |= KAIO_WAKEUP;
19541ce91824SDavid Xu 			error = msleep(&p->p_aioinfo, &p->p_mtx,
19551ce91824SDavid Xu 			    PRIBIO | PCATCH, "aiospn", 0);
19561ce91824SDavid Xu 			if (error == ERESTART)
19571ce91824SDavid Xu 				error = EINTR;
19581ce91824SDavid Xu 			if (error)
19591ce91824SDavid Xu 				break;
19601ce91824SDavid Xu 		}
19611ce91824SDavid Xu 	} else {
19621ce91824SDavid Xu 		if (lj->lioj_count - 1 == lj->lioj_finished_count) {
19631ce91824SDavid Xu 			if (lj->lioj_signal.sigev_notify == SIGEV_KEVENT) {
19641ce91824SDavid Xu 				lj->lioj_flags |= LIOJ_KEVENT_POSTED;
19651ce91824SDavid Xu 				KNOTE_LOCKED(&lj->klist, 1);
19661ce91824SDavid Xu 			}
19671ce91824SDavid Xu 			if ((lj->lioj_flags & (LIOJ_SIGNAL|LIOJ_SIGNAL_POSTED))
19681ce91824SDavid Xu 			    == LIOJ_SIGNAL
19691ce91824SDavid Xu 			    && (lj->lioj_signal.sigev_notify == SIGEV_SIGNAL ||
19701ce91824SDavid Xu 			    lj->lioj_signal.sigev_notify == SIGEV_THREAD_ID)) {
19711ce91824SDavid Xu 				aio_sendsig(p, &lj->lioj_signal,
19721ce91824SDavid Xu 					    &lj->lioj_ksi);
19731ce91824SDavid Xu 				lj->lioj_flags |= LIOJ_SIGNAL_POSTED;
19742244ea07SJohn Dyson 			}
19752244ea07SJohn Dyson 		}
19761ce91824SDavid Xu 	}
19771ce91824SDavid Xu 	lj->lioj_count--;
19781ce91824SDavid Xu 	if (lj->lioj_count == 0) {
19791ce91824SDavid Xu 		TAILQ_REMOVE(&ki->kaio_liojoblist, lj, lioj_list);
19801ce91824SDavid Xu 		knlist_delete(&lj->klist, curthread, 1);
19811ce91824SDavid Xu 		sigqueue_take(&lj->lioj_ksi);
19821ce91824SDavid Xu 		PROC_UNLOCK(p);
19831ce91824SDavid Xu 		uma_zfree(aiolio_zone, lj);
19841ce91824SDavid Xu 	} else
19851ce91824SDavid Xu 		PROC_UNLOCK(p);
19862244ea07SJohn Dyson 
19871ce91824SDavid Xu 	if (nerror)
19881ce91824SDavid Xu 		return (EIO);
19891ce91824SDavid Xu 	return (error);
1990ee877a35SJohn Dyson }
1991fd3bf775SJohn Dyson 
199284af4da6SJohn Dyson /*
19931ce91824SDavid Xu  * Called from interrupt thread for physio, we should return as fast
19941ce91824SDavid Xu  * as possible, so we schedule a biohelper task.
199584af4da6SJohn Dyson  */
1996fd3bf775SJohn Dyson static void
1997bfbbc4aaSJason Evans aio_physwakeup(struct buf *bp)
1998fd3bf775SJohn Dyson {
199984af4da6SJohn Dyson 	struct aiocblist *aiocbe;
2000fd3bf775SJohn Dyson 
20018edbaf85SHidetoshi Shimokawa 	aiocbe = (struct aiocblist *)bp->b_caller1;
20021ce91824SDavid Xu 	taskqueue_enqueue(taskqueue_aiod_bio, &aiocbe->biotask);
20031ce91824SDavid Xu }
200484af4da6SJohn Dyson 
20051ce91824SDavid Xu /*
20061ce91824SDavid Xu  * Task routine to perform heavy tasks, process wakeup, and signals.
20071ce91824SDavid Xu  */
20081ce91824SDavid Xu static void
20091ce91824SDavid Xu biohelper(void *context, int pending)
20101ce91824SDavid Xu {
20111ce91824SDavid Xu 	struct aiocblist *aiocbe = context;
20121ce91824SDavid Xu 	struct buf *bp;
20131ce91824SDavid Xu 	struct proc *userp;
20141ce91824SDavid Xu 	int nblks;
20151ce91824SDavid Xu 
20161ce91824SDavid Xu 	bp = aiocbe->bp;
20171ce91824SDavid Xu 	userp = aiocbe->userproc;
20181ce91824SDavid Xu 	PROC_LOCK(userp);
201984af4da6SJohn Dyson 	aiocbe->uaiocb._aiocb_private.status -= bp->b_resid;
202084af4da6SJohn Dyson 	aiocbe->uaiocb._aiocb_private.error = 0;
2021c244d2deSPoul-Henning Kamp 	if (bp->b_ioflags & BIO_ERROR)
202284af4da6SJohn Dyson 		aiocbe->uaiocb._aiocb_private.error = bp->b_error;
20231ce91824SDavid Xu 	nblks = btodb(aiocbe->uaiocb.aio_nbytes);
20241ce91824SDavid Xu 	if (aiocbe->uaiocb.aio_lio_opcode == LIO_WRITE)
20251ce91824SDavid Xu 		aiocbe->outputcharge += nblks;
20261ce91824SDavid Xu 	else
20271ce91824SDavid Xu 		aiocbe->inputcharge += nblks;
20281ce91824SDavid Xu 	aiocbe->bp = NULL;
20291ce91824SDavid Xu 	TAILQ_REMOVE(&userp->p_aioinfo->kaio_bufqueue, aiocbe, plist);
203069cd28daSDoug Ambrisko 	aio_bio_done_notify(userp, aiocbe, DONE_BUF);
20311ce91824SDavid Xu 	PROC_UNLOCK(userp);
20321ce91824SDavid Xu 
20331ce91824SDavid Xu 	/* Release mapping into kernel space. */
20341ce91824SDavid Xu 	vunmapbuf(bp);
20351ce91824SDavid Xu 	relpbuf(bp, NULL);
20361ce91824SDavid Xu 	atomic_subtract_int(&num_buf_aio, 1);
203784af4da6SJohn Dyson }
2038bfbbc4aaSJason Evans 
2039eb8e6d52SEivind Eklund /* syscall - wait for the next completion of an aio request */
2040bfbbc4aaSJason Evans int
2041b40ce416SJulian Elischer aio_waitcomplete(struct thread *td, struct aio_waitcomplete_args *uap)
2042bfbbc4aaSJason Evans {
2043b40ce416SJulian Elischer 	struct proc *p = td->td_proc;
2044bfbbc4aaSJason Evans 	struct timeval atv;
2045bfbbc4aaSJason Evans 	struct timespec ts;
2046bfbbc4aaSJason Evans 	struct kaioinfo *ki;
20471ce91824SDavid Xu 	struct aiocblist *cb;
20481ce91824SDavid Xu 	struct aiocb *uuaiocb;
20491ce91824SDavid Xu 	int error, status, timo;
2050bfbbc4aaSJason Evans 
20511ce91824SDavid Xu 	suword(uap->aiocbp, (long)NULL);
2052dd85920aSJason Evans 
2053bfbbc4aaSJason Evans 	timo = 0;
2054bfbbc4aaSJason Evans 	if (uap->timeout) {
2055bfbbc4aaSJason Evans 		/* Get timespec struct. */
205688ed460eSAlan Cox 		error = copyin(uap->timeout, &ts, sizeof(ts));
2057bfbbc4aaSJason Evans 		if (error)
2058ac41f2efSAlfred Perlstein 			return (error);
2059bfbbc4aaSJason Evans 
2060bfbbc4aaSJason Evans 		if ((ts.tv_nsec < 0) || (ts.tv_nsec >= 1000000000))
2061bfbbc4aaSJason Evans 			return (EINVAL);
2062bfbbc4aaSJason Evans 
2063bfbbc4aaSJason Evans 		TIMESPEC_TO_TIMEVAL(&atv, &ts);
2064bfbbc4aaSJason Evans 		if (itimerfix(&atv))
2065bfbbc4aaSJason Evans 			return (EINVAL);
2066bfbbc4aaSJason Evans 		timo = tvtohz(&atv);
2067bfbbc4aaSJason Evans 	}
2068bfbbc4aaSJason Evans 
20698213baf0SChristian S.J. Peron 	if (p->p_aioinfo == NULL)
2070323fe565SDavid Xu 		aio_init_aioinfo(p);
20718213baf0SChristian S.J. Peron 	ki = p->p_aioinfo;
2072bfbbc4aaSJason Evans 
20731ce91824SDavid Xu 	error = 0;
20741ce91824SDavid Xu 	cb = NULL;
20753769f562SAlan Cox 	PROC_LOCK(p);
20761ce91824SDavid Xu 	while ((cb = TAILQ_FIRST(&ki->kaio_done)) == NULL) {
20771ce91824SDavid Xu 		ki->kaio_flags |= KAIO_WAKEUP;
20781ce91824SDavid Xu 		error = msleep(&p->p_aioinfo, &p->p_mtx, PRIBIO | PCATCH,
20791ce91824SDavid Xu 		    "aiowc", timo);
20801ce91824SDavid Xu 		if (error == ERESTART)
20811ce91824SDavid Xu 			error = EINTR;
20821ce91824SDavid Xu 		if (error)
20831ce91824SDavid Xu 			break;
20841ce91824SDavid Xu 	}
20851ce91824SDavid Xu 
20861ce91824SDavid Xu 	if (cb != NULL) {
20871ce91824SDavid Xu 		MPASS(cb->jobstate == JOBST_JOBFINISHED);
20881ce91824SDavid Xu 		uuaiocb = cb->uuaiocb;
20891ce91824SDavid Xu 		status = cb->uaiocb._aiocb_private.status;
20901ce91824SDavid Xu 		error = cb->uaiocb._aiocb_private.error;
20911ce91824SDavid Xu 		td->td_retval[0] = status;
2092bfbbc4aaSJason Evans 		if (cb->uaiocb.aio_lio_opcode == LIO_WRITE) {
20931ce91824SDavid Xu 			p->p_stats->p_ru.ru_oublock += cb->outputcharge;
2094bfbbc4aaSJason Evans 			cb->outputcharge = 0;
2095bfbbc4aaSJason Evans 		} else if (cb->uaiocb.aio_lio_opcode == LIO_READ) {
2096516d2564SAlan Cox 			p->p_stats->p_ru.ru_inblock += cb->inputcharge;
2097bfbbc4aaSJason Evans 			cb->inputcharge = 0;
2098bfbbc4aaSJason Evans 		}
2099bfbbc4aaSJason Evans 		aio_free_entry(cb);
21003769f562SAlan Cox 		PROC_UNLOCK(p);
21011ce91824SDavid Xu 		suword(uap->aiocbp, (long)uuaiocb);
21021ce91824SDavid Xu 		suword(&uuaiocb->_aiocb_private.error, error);
21031ce91824SDavid Xu 		suword(&uuaiocb->_aiocb_private.status, status);
21041ce91824SDavid Xu 	} else
21051ce91824SDavid Xu 		PROC_UNLOCK(p);
2106bfbbc4aaSJason Evans 
2107ac41f2efSAlfred Perlstein 	return (error);
2108bfbbc4aaSJason Evans }
2109cb679c38SJonathan Lemon 
2110eb8e6d52SEivind Eklund /* kqueue attach function */
2111cb679c38SJonathan Lemon static int
2112cb679c38SJonathan Lemon filt_aioattach(struct knote *kn)
2113cb679c38SJonathan Lemon {
2114b46f1c55SAlan Cox 	struct aiocblist *aiocbe = (struct aiocblist *)kn->kn_sdata;
2115cb679c38SJonathan Lemon 
2116cb679c38SJonathan Lemon 	/*
2117cb679c38SJonathan Lemon 	 * The aiocbe pointer must be validated before using it, so
2118cb679c38SJonathan Lemon 	 * registration is restricted to the kernel; the user cannot
2119cb679c38SJonathan Lemon 	 * set EV_FLAG1.
2120cb679c38SJonathan Lemon 	 */
2121cb679c38SJonathan Lemon 	if ((kn->kn_flags & EV_FLAG1) == 0)
2122cb679c38SJonathan Lemon 		return (EPERM);
2123cb679c38SJonathan Lemon 	kn->kn_flags &= ~EV_FLAG1;
2124cb679c38SJonathan Lemon 
2125ad3b9257SJohn-Mark Gurney 	knlist_add(&aiocbe->klist, kn, 0);
2126cb679c38SJonathan Lemon 
2127cb679c38SJonathan Lemon 	return (0);
2128cb679c38SJonathan Lemon }
2129cb679c38SJonathan Lemon 
2130eb8e6d52SEivind Eklund /* kqueue detach function */
2131cb679c38SJonathan Lemon static void
2132cb679c38SJonathan Lemon filt_aiodetach(struct knote *kn)
2133cb679c38SJonathan Lemon {
2134b46f1c55SAlan Cox 	struct aiocblist *aiocbe = (struct aiocblist *)kn->kn_sdata;
2135cb679c38SJonathan Lemon 
213669cd28daSDoug Ambrisko 	if (!knlist_empty(&aiocbe->klist))
2137ad3b9257SJohn-Mark Gurney 		knlist_remove(&aiocbe->klist, kn, 0);
2138cb679c38SJonathan Lemon }
2139cb679c38SJonathan Lemon 
2140eb8e6d52SEivind Eklund /* kqueue filter function */
2141cb679c38SJonathan Lemon /*ARGSUSED*/
2142cb679c38SJonathan Lemon static int
2143cb679c38SJonathan Lemon filt_aio(struct knote *kn, long hint)
2144cb679c38SJonathan Lemon {
2145b46f1c55SAlan Cox 	struct aiocblist *aiocbe = (struct aiocblist *)kn->kn_sdata;
2146cb679c38SJonathan Lemon 
214791369fc7SAlan Cox 	kn->kn_data = aiocbe->uaiocb._aiocb_private.error;
21481ce91824SDavid Xu 	if (aiocbe->jobstate != JOBST_JOBFINISHED)
2149cb679c38SJonathan Lemon 		return (0);
2150cb679c38SJonathan Lemon 	kn->kn_flags |= EV_EOF;
2151cb679c38SJonathan Lemon 	return (1);
2152cb679c38SJonathan Lemon }
215369cd28daSDoug Ambrisko 
215469cd28daSDoug Ambrisko /* kqueue attach function */
215569cd28daSDoug Ambrisko static int
215669cd28daSDoug Ambrisko filt_lioattach(struct knote *kn)
215769cd28daSDoug Ambrisko {
21581ce91824SDavid Xu 	struct aioliojob * lj = (struct aioliojob *)kn->kn_sdata;
215969cd28daSDoug Ambrisko 
216069cd28daSDoug Ambrisko 	/*
21611ce91824SDavid Xu 	 * The aioliojob pointer must be validated before using it, so
216269cd28daSDoug Ambrisko 	 * registration is restricted to the kernel; the user cannot
216369cd28daSDoug Ambrisko 	 * set EV_FLAG1.
216469cd28daSDoug Ambrisko 	 */
216569cd28daSDoug Ambrisko 	if ((kn->kn_flags & EV_FLAG1) == 0)
216669cd28daSDoug Ambrisko 		return (EPERM);
216769cd28daSDoug Ambrisko 	kn->kn_flags &= ~EV_FLAG1;
216869cd28daSDoug Ambrisko 
216969cd28daSDoug Ambrisko 	knlist_add(&lj->klist, kn, 0);
217069cd28daSDoug Ambrisko 
217169cd28daSDoug Ambrisko 	return (0);
217269cd28daSDoug Ambrisko }
217369cd28daSDoug Ambrisko 
217469cd28daSDoug Ambrisko /* kqueue detach function */
217569cd28daSDoug Ambrisko static void
217669cd28daSDoug Ambrisko filt_liodetach(struct knote *kn)
217769cd28daSDoug Ambrisko {
21781ce91824SDavid Xu 	struct aioliojob * lj = (struct aioliojob *)kn->kn_sdata;
217969cd28daSDoug Ambrisko 
218069cd28daSDoug Ambrisko 	if (!knlist_empty(&lj->klist))
218169cd28daSDoug Ambrisko 		knlist_remove(&lj->klist, kn, 0);
218269cd28daSDoug Ambrisko }
218369cd28daSDoug Ambrisko 
218469cd28daSDoug Ambrisko /* kqueue filter function */
218569cd28daSDoug Ambrisko /*ARGSUSED*/
218669cd28daSDoug Ambrisko static int
218769cd28daSDoug Ambrisko filt_lio(struct knote *kn, long hint)
218869cd28daSDoug Ambrisko {
21891ce91824SDavid Xu 	struct aioliojob * lj = (struct aioliojob *)kn->kn_sdata;
21901ce91824SDavid Xu 
219169cd28daSDoug Ambrisko 	return (lj->lioj_flags & LIOJ_KEVENT_POSTED);
219269cd28daSDoug Ambrisko }
2193