xref: /freebsd/sys/kern/vfs_aio.c (revision f3215a60fd1c3b90904036ba1be82490632ca283)
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 
243858a1f4SJohn Baldwin #include "opt_compat.h"
253858a1f4SJohn Baldwin 
26ee877a35SJohn Dyson #include <sys/param.h>
27ee877a35SJohn Dyson #include <sys/systm.h>
28f591779bSSeigo Tanimura #include <sys/malloc.h>
299626b608SPoul-Henning Kamp #include <sys/bio.h>
30a5c9bce7SBruce Evans #include <sys/buf.h>
31a9d2f8d8SRobert Watson #include <sys/capability.h>
3275b8b3b2SJohn Baldwin #include <sys/eventhandler.h>
33ee877a35SJohn Dyson #include <sys/sysproto.h>
34ee877a35SJohn Dyson #include <sys/filedesc.h>
35ee877a35SJohn Dyson #include <sys/kernel.h>
3677409fe1SPoul-Henning Kamp #include <sys/module.h>
37c9a970a7SAlan Cox #include <sys/kthread.h>
38ee877a35SJohn Dyson #include <sys/fcntl.h>
39ee877a35SJohn Dyson #include <sys/file.h>
40104a9b7eSAlexander Kabaev #include <sys/limits.h>
41fdebd4f0SBruce Evans #include <sys/lock.h>
4235e0e5b3SJohn Baldwin #include <sys/mutex.h>
43ee877a35SJohn Dyson #include <sys/unistd.h>
446aeb05d7STom Rhodes #include <sys/posix4.h>
45ee877a35SJohn Dyson #include <sys/proc.h>
462d2f8ae7SBruce Evans #include <sys/resourcevar.h>
47ee877a35SJohn Dyson #include <sys/signalvar.h>
48bfbbc4aaSJason Evans #include <sys/protosw.h>
4989f6b863SAttilio Rao #include <sys/rwlock.h>
501ce91824SDavid Xu #include <sys/sema.h>
511ce91824SDavid Xu #include <sys/socket.h>
52bfbbc4aaSJason Evans #include <sys/socketvar.h>
5321d56e9cSAlfred Perlstein #include <sys/syscall.h>
5421d56e9cSAlfred Perlstein #include <sys/sysent.h>
55a624e84fSJohn Dyson #include <sys/sysctl.h>
56ee99e978SBruce Evans #include <sys/sx.h>
571ce91824SDavid Xu #include <sys/taskqueue.h>
58fd3bf775SJohn Dyson #include <sys/vnode.h>
59fd3bf775SJohn Dyson #include <sys/conf.h>
60cb679c38SJonathan Lemon #include <sys/event.h>
6199eee864SDavid Xu #include <sys/mount.h>
62ee877a35SJohn Dyson 
631ce91824SDavid Xu #include <machine/atomic.h>
641ce91824SDavid Xu 
65ee877a35SJohn Dyson #include <vm/vm.h>
66ee877a35SJohn Dyson #include <vm/vm_extern.h>
672244ea07SJohn Dyson #include <vm/pmap.h>
682244ea07SJohn Dyson #include <vm/vm_map.h>
6999eee864SDavid Xu #include <vm/vm_object.h>
70c897b813SJeff Roberson #include <vm/uma.h>
71ee877a35SJohn Dyson #include <sys/aio.h>
725aaef07cSJohn Dyson 
73dd85920aSJason Evans #include "opt_vfs_aio.h"
74ee877a35SJohn Dyson 
75eb8e6d52SEivind Eklund /*
76eb8e6d52SEivind Eklund  * Counter for allocating reference ids to new jobs.  Wrapped to 1 on
7799eee864SDavid Xu  * overflow. (XXX will be removed soon.)
78eb8e6d52SEivind Eklund  */
7999eee864SDavid Xu static u_long jobrefid;
802244ea07SJohn Dyson 
8199eee864SDavid Xu /*
8299eee864SDavid Xu  * Counter for aio_fsync.
8399eee864SDavid Xu  */
8499eee864SDavid Xu static uint64_t jobseqno;
8599eee864SDavid Xu 
8699eee864SDavid Xu #define JOBST_NULL		0
8799eee864SDavid Xu #define JOBST_JOBQSOCK		1
8899eee864SDavid Xu #define JOBST_JOBQGLOBAL	2
8999eee864SDavid Xu #define JOBST_JOBRUNNING	3
9099eee864SDavid Xu #define JOBST_JOBFINISHED	4
9199eee864SDavid Xu #define JOBST_JOBQBUF		5
9299eee864SDavid Xu #define JOBST_JOBQSYNC		6
932244ea07SJohn Dyson 
9484af4da6SJohn Dyson #ifndef MAX_AIO_PER_PROC
952244ea07SJohn Dyson #define MAX_AIO_PER_PROC	32
9684af4da6SJohn Dyson #endif
9784af4da6SJohn Dyson 
9884af4da6SJohn Dyson #ifndef MAX_AIO_QUEUE_PER_PROC
992244ea07SJohn Dyson #define MAX_AIO_QUEUE_PER_PROC	256 /* Bigger than AIO_LISTIO_MAX */
10084af4da6SJohn Dyson #endif
10184af4da6SJohn Dyson 
10284af4da6SJohn Dyson #ifndef MAX_AIO_PROCS
103fd3bf775SJohn Dyson #define MAX_AIO_PROCS		32
10484af4da6SJohn Dyson #endif
10584af4da6SJohn Dyson 
10684af4da6SJohn Dyson #ifndef MAX_AIO_QUEUE
1072244ea07SJohn Dyson #define	MAX_AIO_QUEUE		1024 /* Bigger than AIO_LISTIO_MAX */
10884af4da6SJohn Dyson #endif
10984af4da6SJohn Dyson 
11084af4da6SJohn Dyson #ifndef TARGET_AIO_PROCS
111bfbbc4aaSJason Evans #define TARGET_AIO_PROCS	4
11284af4da6SJohn Dyson #endif
11384af4da6SJohn Dyson 
11484af4da6SJohn Dyson #ifndef MAX_BUF_AIO
11584af4da6SJohn Dyson #define MAX_BUF_AIO		16
11684af4da6SJohn Dyson #endif
11784af4da6SJohn Dyson 
11884af4da6SJohn Dyson #ifndef AIOD_TIMEOUT_DEFAULT
11984af4da6SJohn Dyson #define	AIOD_TIMEOUT_DEFAULT	(10 * hz)
12084af4da6SJohn Dyson #endif
12184af4da6SJohn Dyson 
12284af4da6SJohn Dyson #ifndef AIOD_LIFETIME_DEFAULT
12384af4da6SJohn Dyson #define AIOD_LIFETIME_DEFAULT	(30 * hz)
12484af4da6SJohn Dyson #endif
1252244ea07SJohn Dyson 
126e603be7aSRobert Watson FEATURE(aio, "Asynchronous I/O");
127e603be7aSRobert Watson 
1283858a1f4SJohn Baldwin static MALLOC_DEFINE(M_LIO, "lio", "listio aio control block list");
1293858a1f4SJohn Baldwin 
1305ece08f5SPoul-Henning Kamp static SYSCTL_NODE(_vfs, OID_AUTO, aio, CTLFLAG_RW, 0, "Async IO management");
131eb8e6d52SEivind Eklund 
132303b270bSEivind Eklund static int max_aio_procs = MAX_AIO_PROCS;
133a624e84fSJohn Dyson SYSCTL_INT(_vfs_aio, OID_AUTO, max_aio_procs,
134eb8e6d52SEivind Eklund 	CTLFLAG_RW, &max_aio_procs, 0,
135eb8e6d52SEivind Eklund 	"Maximum number of kernel threads to use for handling async IO ");
136a624e84fSJohn Dyson 
137eb8e6d52SEivind Eklund static int num_aio_procs = 0;
138a624e84fSJohn Dyson SYSCTL_INT(_vfs_aio, OID_AUTO, num_aio_procs,
139eb8e6d52SEivind Eklund 	CTLFLAG_RD, &num_aio_procs, 0,
140eb8e6d52SEivind Eklund 	"Number of presently active kernel threads for async IO");
141a624e84fSJohn Dyson 
142eb8e6d52SEivind Eklund /*
143eb8e6d52SEivind Eklund  * The code will adjust the actual number of AIO processes towards this
144eb8e6d52SEivind Eklund  * number when it gets a chance.
145eb8e6d52SEivind Eklund  */
146eb8e6d52SEivind Eklund static int target_aio_procs = TARGET_AIO_PROCS;
147eb8e6d52SEivind Eklund SYSCTL_INT(_vfs_aio, OID_AUTO, target_aio_procs, CTLFLAG_RW, &target_aio_procs,
148eb8e6d52SEivind Eklund 	0, "Preferred number of ready kernel threads for async IO");
149a624e84fSJohn Dyson 
150eb8e6d52SEivind Eklund static int max_queue_count = MAX_AIO_QUEUE;
151eb8e6d52SEivind Eklund SYSCTL_INT(_vfs_aio, OID_AUTO, max_aio_queue, CTLFLAG_RW, &max_queue_count, 0,
152eb8e6d52SEivind Eklund     "Maximum number of aio requests to queue, globally");
153a624e84fSJohn Dyson 
154eb8e6d52SEivind Eklund static int num_queue_count = 0;
155eb8e6d52SEivind Eklund SYSCTL_INT(_vfs_aio, OID_AUTO, num_queue_count, CTLFLAG_RD, &num_queue_count, 0,
156eb8e6d52SEivind Eklund     "Number of queued aio requests");
157a624e84fSJohn Dyson 
158eb8e6d52SEivind Eklund static int num_buf_aio = 0;
159eb8e6d52SEivind Eklund SYSCTL_INT(_vfs_aio, OID_AUTO, num_buf_aio, CTLFLAG_RD, &num_buf_aio, 0,
160eb8e6d52SEivind Eklund     "Number of aio requests presently handled by the buf subsystem");
161fd3bf775SJohn Dyson 
162eb8e6d52SEivind Eklund /* Number of async I/O thread in the process of being started */
163a9bf5e37SDavid Xu /* XXX This should be local to aio_aqueue() */
164eb8e6d52SEivind Eklund static int num_aio_resv_start = 0;
165fd3bf775SJohn Dyson 
166eb8e6d52SEivind Eklund static int aiod_timeout;
167eb8e6d52SEivind Eklund SYSCTL_INT(_vfs_aio, OID_AUTO, aiod_timeout, CTLFLAG_RW, &aiod_timeout, 0,
168eb8e6d52SEivind Eklund     "Timeout value for synchronous aio operations");
16984af4da6SJohn Dyson 
170eb8e6d52SEivind Eklund static int aiod_lifetime;
171eb8e6d52SEivind Eklund SYSCTL_INT(_vfs_aio, OID_AUTO, aiod_lifetime, CTLFLAG_RW, &aiod_lifetime, 0,
172eb8e6d52SEivind Eklund     "Maximum lifetime for idle aiod");
17384af4da6SJohn Dyson 
174eb8e6d52SEivind Eklund static int unloadable = 0;
17521d56e9cSAlfred Perlstein SYSCTL_INT(_vfs_aio, OID_AUTO, unloadable, CTLFLAG_RW, &unloadable, 0,
17621d56e9cSAlfred Perlstein     "Allow unload of aio (not recommended)");
17721d56e9cSAlfred Perlstein 
178eb8e6d52SEivind Eklund 
179eb8e6d52SEivind Eklund static int max_aio_per_proc = MAX_AIO_PER_PROC;
180eb8e6d52SEivind Eklund SYSCTL_INT(_vfs_aio, OID_AUTO, max_aio_per_proc, CTLFLAG_RW, &max_aio_per_proc,
181eb8e6d52SEivind Eklund     0, "Maximum active aio requests per process (stored in the process)");
182eb8e6d52SEivind Eklund 
183eb8e6d52SEivind Eklund static int max_aio_queue_per_proc = MAX_AIO_QUEUE_PER_PROC;
184eb8e6d52SEivind Eklund SYSCTL_INT(_vfs_aio, OID_AUTO, max_aio_queue_per_proc, CTLFLAG_RW,
185eb8e6d52SEivind Eklund     &max_aio_queue_per_proc, 0,
186eb8e6d52SEivind Eklund     "Maximum queued aio requests per process (stored in the process)");
187eb8e6d52SEivind Eklund 
188eb8e6d52SEivind Eklund static int max_buf_aio = MAX_BUF_AIO;
189eb8e6d52SEivind Eklund SYSCTL_INT(_vfs_aio, OID_AUTO, max_buf_aio, CTLFLAG_RW, &max_buf_aio, 0,
190eb8e6d52SEivind Eklund     "Maximum buf aio requests per process (stored in the process)");
191eb8e6d52SEivind Eklund 
1920972628aSDavid Xu typedef struct oaiocb {
1930972628aSDavid Xu 	int	aio_fildes;		/* File descriptor */
1940972628aSDavid Xu 	off_t	aio_offset;		/* File offset for I/O */
1950972628aSDavid Xu 	volatile void *aio_buf;         /* I/O buffer in process space */
1960972628aSDavid Xu 	size_t	aio_nbytes;		/* Number of bytes for I/O */
1970972628aSDavid Xu 	struct	osigevent aio_sigevent;	/* Signal to deliver */
1980972628aSDavid Xu 	int	aio_lio_opcode;		/* LIO opcode */
1990972628aSDavid Xu 	int	aio_reqprio;		/* Request priority -- ignored */
2000972628aSDavid Xu 	struct	__aiocb_private	_aiocb_private;
2010972628aSDavid Xu } oaiocb_t;
2020972628aSDavid Xu 
2031aa4c324SDavid Xu /*
2041aa4c324SDavid Xu  * Below is a key of locks used to protect each member of struct aiocblist
2051aa4c324SDavid Xu  * aioliojob and kaioinfo and any backends.
2061aa4c324SDavid Xu  *
2071aa4c324SDavid Xu  * * - need not protected
208759ccccaSDavid Xu  * a - locked by kaioinfo lock
2091aa4c324SDavid Xu  * b - locked by backend lock, the backend lock can be null in some cases,
2101aa4c324SDavid Xu  *     for example, BIO belongs to this type, in this case, proc lock is
2111aa4c324SDavid Xu  *     reused.
2121aa4c324SDavid Xu  * c - locked by aio_job_mtx, the lock for the generic file I/O backend.
2131aa4c324SDavid Xu  */
2141aa4c324SDavid Xu 
2151aa4c324SDavid Xu /*
2161aa4c324SDavid Xu  * Current, there is only two backends: BIO and generic file I/O.
2171aa4c324SDavid Xu  * socket I/O is served by generic file I/O, this is not a good idea, since
2181aa4c324SDavid Xu  * disk file I/O and any other types without O_NONBLOCK flag can block daemon
2191aa4c324SDavid Xu  * threads, if there is no thread to serve socket I/O, the socket I/O will be
2201aa4c324SDavid Xu  * delayed too long or starved, we should create some threads dedicated to
2211aa4c324SDavid Xu  * sockets to do non-blocking I/O, same for pipe and fifo, for these I/O
2221aa4c324SDavid Xu  * systems we really need non-blocking interface, fiddling O_NONBLOCK in file
2231aa4c324SDavid Xu  * structure is not safe because there is race between userland and aio
2241aa4c324SDavid Xu  * daemons.
2251aa4c324SDavid Xu  */
2261aa4c324SDavid Xu 
22748dac059SAlan Cox struct aiocblist {
2281aa4c324SDavid Xu 	TAILQ_ENTRY(aiocblist) list;	/* (b) internal list of for backend */
2291aa4c324SDavid Xu 	TAILQ_ENTRY(aiocblist) plist;	/* (a) list of jobs for each backend */
2301aa4c324SDavid Xu 	TAILQ_ENTRY(aiocblist) allist;  /* (a) list of all jobs in proc */
2311aa4c324SDavid Xu 	int	jobflags;		/* (a) job flags */
2321aa4c324SDavid Xu 	int	jobstate;		/* (b) job state */
2331aa4c324SDavid Xu 	int	inputcharge;		/* (*) input blockes */
2341aa4c324SDavid Xu 	int	outputcharge;		/* (*) output blockes */
2351aa4c324SDavid Xu 	struct	buf *bp;		/* (*) private to BIO backend,
2361aa4c324SDavid Xu 				  	 * buffer pointer
2371aa4c324SDavid Xu 					 */
2381aa4c324SDavid Xu 	struct	proc *userproc;		/* (*) user process */
2391aa4c324SDavid Xu 	struct  ucred *cred;		/* (*) active credential when created */
2401aa4c324SDavid Xu 	struct	file *fd_file;		/* (*) pointer to file structure */
2411aa4c324SDavid Xu 	struct	aioliojob *lio;		/* (*) optional lio job */
2421aa4c324SDavid Xu 	struct	aiocb *uuaiocb;		/* (*) pointer in userspace of aiocb */
2431aa4c324SDavid Xu 	struct	knlist klist;		/* (a) list of knotes */
2441aa4c324SDavid Xu 	struct	aiocb uaiocb;		/* (*) kernel I/O control block */
2451aa4c324SDavid Xu 	ksiginfo_t ksi;			/* (a) realtime signal info */
2461aa4c324SDavid Xu 	struct	task biotask;		/* (*) private to BIO backend */
24799eee864SDavid Xu 	uint64_t seqno;			/* (*) job number */
24899eee864SDavid Xu 	int	pending;		/* (a) number of pending I/O, aio_fsync only */
24948dac059SAlan Cox };
25048dac059SAlan Cox 
25148dac059SAlan Cox /* jobflags */
25299eee864SDavid Xu #define AIOCBLIST_DONE		0x01
25399eee864SDavid Xu #define AIOCBLIST_BUFDONE	0x02
2541ce91824SDavid Xu #define AIOCBLIST_RUNDOWN	0x04
25599eee864SDavid Xu #define AIOCBLIST_CHECKSYNC	0x08
25648dac059SAlan Cox 
2572244ea07SJohn Dyson /*
2582244ea07SJohn Dyson  * AIO process info
2592244ea07SJohn Dyson  */
26084af4da6SJohn Dyson #define AIOP_FREE	0x1			/* proc on free queue */
26184af4da6SJohn Dyson 
262b40ce416SJulian Elischer struct aiothreadlist {
2631aa4c324SDavid Xu 	int aiothreadflags;			/* (c) AIO proc flags */
2641aa4c324SDavid Xu 	TAILQ_ENTRY(aiothreadlist) list;	/* (c) list of processes */
2651aa4c324SDavid Xu 	struct thread *aiothread;		/* (*) the AIO thread */
2662244ea07SJohn Dyson };
2672244ea07SJohn Dyson 
26884af4da6SJohn Dyson /*
26984af4da6SJohn Dyson  * data-structure for lio signal management
27084af4da6SJohn Dyson  */
2711ce91824SDavid Xu struct aioliojob {
2721aa4c324SDavid Xu 	int	lioj_flags;			/* (a) listio flags */
2731aa4c324SDavid Xu 	int	lioj_count;			/* (a) listio flags */
2741aa4c324SDavid Xu 	int	lioj_finished_count;		/* (a) listio flags */
2751aa4c324SDavid Xu 	struct	sigevent lioj_signal;		/* (a) signal on all I/O done */
2761aa4c324SDavid Xu 	TAILQ_ENTRY(aioliojob) lioj_list;	/* (a) lio list */
2771aa4c324SDavid Xu 	struct  knlist klist;			/* (a) list of knotes */
2781aa4c324SDavid Xu 	ksiginfo_t lioj_ksi;			/* (a) Realtime signal info */
27984af4da6SJohn Dyson };
2801ce91824SDavid Xu 
28184af4da6SJohn Dyson #define	LIOJ_SIGNAL		0x1	/* signal on all done (lio) */
28284af4da6SJohn Dyson #define	LIOJ_SIGNAL_POSTED	0x2	/* signal has been posted */
28369cd28daSDoug Ambrisko #define LIOJ_KEVENT_POSTED	0x4	/* kevent triggered */
28484af4da6SJohn Dyson 
28584af4da6SJohn Dyson /*
28684af4da6SJohn Dyson  * per process aio data structure
28784af4da6SJohn Dyson  */
2882244ea07SJohn Dyson struct kaioinfo {
289759ccccaSDavid Xu 	struct mtx	kaio_mtx;	/* the lock to protect this struct */
2901aa4c324SDavid Xu 	int	kaio_flags;		/* (a) per process kaio flags */
2911aa4c324SDavid Xu 	int	kaio_maxactive_count;	/* (*) maximum number of AIOs */
2921aa4c324SDavid Xu 	int	kaio_active_count;	/* (c) number of currently used AIOs */
2931aa4c324SDavid Xu 	int	kaio_qallowed_count;	/* (*) maxiumu size of AIO queue */
2941aa4c324SDavid Xu 	int	kaio_count;		/* (a) size of AIO queue */
2951aa4c324SDavid Xu 	int	kaio_ballowed_count;	/* (*) maximum number of buffers */
2961aa4c324SDavid Xu 	int	kaio_buffer_count;	/* (a) number of physio buffers */
2971aa4c324SDavid Xu 	TAILQ_HEAD(,aiocblist) kaio_all;	/* (a) all AIOs in the process */
2981aa4c324SDavid Xu 	TAILQ_HEAD(,aiocblist) kaio_done;	/* (a) done queue for process */
2991aa4c324SDavid Xu 	TAILQ_HEAD(,aioliojob) kaio_liojoblist; /* (a) list of lio jobs */
3001aa4c324SDavid Xu 	TAILQ_HEAD(,aiocblist) kaio_jobqueue;	/* (a) job queue for process */
3011aa4c324SDavid Xu 	TAILQ_HEAD(,aiocblist) kaio_bufqueue;	/* (a) buffer job queue for process */
3021aa4c324SDavid Xu 	TAILQ_HEAD(,aiocblist) kaio_sockqueue;  /* (a) queue for aios waiting on sockets,
30399eee864SDavid Xu 						 *  NOT USED YET.
3041aa4c324SDavid Xu 						 */
30599eee864SDavid Xu 	TAILQ_HEAD(,aiocblist) kaio_syncqueue;	/* (a) queue for aio_fsync */
30699eee864SDavid Xu 	struct	task	kaio_task;	/* (*) task to kick aio threads */
3072244ea07SJohn Dyson };
3082244ea07SJohn Dyson 
309759ccccaSDavid Xu #define AIO_LOCK(ki)		mtx_lock(&(ki)->kaio_mtx)
310759ccccaSDavid Xu #define AIO_UNLOCK(ki)		mtx_unlock(&(ki)->kaio_mtx)
311759ccccaSDavid Xu #define AIO_LOCK_ASSERT(ki, f)	mtx_assert(&(ki)->kaio_mtx, (f))
312759ccccaSDavid Xu #define AIO_MTX(ki)		(&(ki)->kaio_mtx)
313759ccccaSDavid Xu 
31484af4da6SJohn Dyson #define KAIO_RUNDOWN	0x1	/* process is being run down */
315bfbbc4aaSJason Evans #define KAIO_WAKEUP	0x2	/* wakeup process when there is a significant event */
316fd3bf775SJohn Dyson 
3173858a1f4SJohn Baldwin /*
3183858a1f4SJohn Baldwin  * Operations used to interact with userland aio control blocks.
3193858a1f4SJohn Baldwin  * Different ABIs provide their own operations.
3203858a1f4SJohn Baldwin  */
3213858a1f4SJohn Baldwin struct aiocb_ops {
3223858a1f4SJohn Baldwin 	int	(*copyin)(struct aiocb *ujob, struct aiocb *kjob);
3233858a1f4SJohn Baldwin 	long	(*fetch_status)(struct aiocb *ujob);
3243858a1f4SJohn Baldwin 	long	(*fetch_error)(struct aiocb *ujob);
3253858a1f4SJohn Baldwin 	int	(*store_status)(struct aiocb *ujob, long status);
3263858a1f4SJohn Baldwin 	int	(*store_error)(struct aiocb *ujob, long error);
3273858a1f4SJohn Baldwin 	int	(*store_kernelinfo)(struct aiocb *ujob, long jobref);
3283858a1f4SJohn Baldwin 	int	(*store_aiocb)(struct aiocb **ujobp, struct aiocb *ujob);
3293858a1f4SJohn Baldwin };
3303858a1f4SJohn Baldwin 
3311aa4c324SDavid Xu static TAILQ_HEAD(,aiothreadlist) aio_freeproc;		/* (c) Idle daemons */
3321ce91824SDavid Xu static struct sema aio_newproc_sem;
3331ce91824SDavid Xu static struct mtx aio_job_mtx;
3341ce91824SDavid Xu static struct mtx aio_sock_mtx;
3351aa4c324SDavid Xu static TAILQ_HEAD(,aiocblist) aio_jobs;			/* (c) Async job list */
3361ce91824SDavid Xu static struct unrhdr *aiod_unr;
3372244ea07SJohn Dyson 
3386a1162d4SAlexander Leidinger void		aio_init_aioinfo(struct proc *p);
339723d37c0SKonstantin Belousov static int	aio_onceonly(void);
340fd3bf775SJohn Dyson static int	aio_free_entry(struct aiocblist *aiocbe);
341fd3bf775SJohn Dyson static void	aio_process(struct aiocblist *aiocbe);
3421ce91824SDavid Xu static int	aio_newproc(int *);
3436a1162d4SAlexander Leidinger int		aio_aqueue(struct thread *td, struct aiocb *job,
3443858a1f4SJohn Baldwin 			struct aioliojob *lio, int type, struct aiocb_ops *ops);
345fd3bf775SJohn Dyson static void	aio_physwakeup(struct buf *bp);
34675b8b3b2SJohn Baldwin static void	aio_proc_rundown(void *arg, struct proc *p);
347993182e5SAlexander Leidinger static void	aio_proc_rundown_exec(void *arg, struct proc *p, struct image_params *imgp);
348fd3bf775SJohn Dyson static int	aio_qphysio(struct proc *p, struct aiocblist *iocb);
3491ce91824SDavid Xu static void	biohelper(void *, int);
3501ce91824SDavid Xu static void	aio_daemon(void *param);
35148dac059SAlan Cox static void	aio_swake_cb(struct socket *, struct sockbuf *);
35221d56e9cSAlfred Perlstein static int	aio_unload(void);
35399eee864SDavid Xu static void	aio_bio_done_notify(struct proc *userp, struct aiocblist *aiocbe, int type);
35499eee864SDavid Xu #define DONE_BUF	1
35599eee864SDavid Xu #define DONE_QUEUE	2
356dbbccfe9SDavid Xu static int	aio_kick(struct proc *userp);
35799eee864SDavid Xu static void	aio_kick_nowait(struct proc *userp);
35899eee864SDavid Xu static void	aio_kick_helper(void *context, int pending);
35921d56e9cSAlfred Perlstein static int	filt_aioattach(struct knote *kn);
36021d56e9cSAlfred Perlstein static void	filt_aiodetach(struct knote *kn);
36121d56e9cSAlfred Perlstein static int	filt_aio(struct knote *kn, long hint);
36269cd28daSDoug Ambrisko static int	filt_lioattach(struct knote *kn);
36369cd28daSDoug Ambrisko static void	filt_liodetach(struct knote *kn);
36469cd28daSDoug Ambrisko static int	filt_lio(struct knote *kn, long hint);
3652244ea07SJohn Dyson 
366eb8e6d52SEivind Eklund /*
367eb8e6d52SEivind Eklund  * Zones for:
368eb8e6d52SEivind Eklund  * 	kaio	Per process async io info
369eb8e6d52SEivind Eklund  *	aiop	async io thread data
370eb8e6d52SEivind Eklund  *	aiocb	async io jobs
371eb8e6d52SEivind Eklund  *	aiol	list io job pointer - internal to aio_suspend XXX
372eb8e6d52SEivind Eklund  *	aiolio	list io jobs
373eb8e6d52SEivind Eklund  */
374c897b813SJeff Roberson static uma_zone_t kaio_zone, aiop_zone, aiocb_zone, aiol_zone, aiolio_zone;
375fd3bf775SJohn Dyson 
376eb8e6d52SEivind Eklund /* kqueue filters for aio */
377e76d823bSRobert Watson static struct filterops aio_filtops = {
378e76d823bSRobert Watson 	.f_isfd = 0,
379e76d823bSRobert Watson 	.f_attach = filt_aioattach,
380e76d823bSRobert Watson 	.f_detach = filt_aiodetach,
381e76d823bSRobert Watson 	.f_event = filt_aio,
382e76d823bSRobert Watson };
383e76d823bSRobert Watson static struct filterops lio_filtops = {
384e76d823bSRobert Watson 	.f_isfd = 0,
385e76d823bSRobert Watson 	.f_attach = filt_lioattach,
386e76d823bSRobert Watson 	.f_detach = filt_liodetach,
387e76d823bSRobert Watson 	.f_event = filt_lio
388e76d823bSRobert Watson };
38921d56e9cSAlfred Perlstein 
39075b8b3b2SJohn Baldwin static eventhandler_tag exit_tag, exec_tag;
39175b8b3b2SJohn Baldwin 
3921ce91824SDavid Xu TASKQUEUE_DEFINE_THREAD(aiod_bio);
3931ce91824SDavid Xu 
394eb8e6d52SEivind Eklund /*
395eb8e6d52SEivind Eklund  * Main operations function for use as a kernel module.
396eb8e6d52SEivind Eklund  */
39721d56e9cSAlfred Perlstein static int
39821d56e9cSAlfred Perlstein aio_modload(struct module *module, int cmd, void *arg)
39921d56e9cSAlfred Perlstein {
40021d56e9cSAlfred Perlstein 	int error = 0;
40121d56e9cSAlfred Perlstein 
40221d56e9cSAlfred Perlstein 	switch (cmd) {
40321d56e9cSAlfred Perlstein 	case MOD_LOAD:
40421d56e9cSAlfred Perlstein 		aio_onceonly();
40521d56e9cSAlfred Perlstein 		break;
40621d56e9cSAlfred Perlstein 	case MOD_UNLOAD:
40721d56e9cSAlfred Perlstein 		error = aio_unload();
40821d56e9cSAlfred Perlstein 		break;
40921d56e9cSAlfred Perlstein 	case MOD_SHUTDOWN:
41021d56e9cSAlfred Perlstein 		break;
41121d56e9cSAlfred Perlstein 	default:
41221d56e9cSAlfred Perlstein 		error = EINVAL;
41321d56e9cSAlfred Perlstein 		break;
41421d56e9cSAlfred Perlstein 	}
41521d56e9cSAlfred Perlstein 	return (error);
41621d56e9cSAlfred Perlstein }
41721d56e9cSAlfred Perlstein 
41821d56e9cSAlfred Perlstein static moduledata_t aio_mod = {
41921d56e9cSAlfred Perlstein 	"aio",
42021d56e9cSAlfred Perlstein 	&aio_modload,
42121d56e9cSAlfred Perlstein 	NULL
42221d56e9cSAlfred Perlstein };
42321d56e9cSAlfred Perlstein 
424723d37c0SKonstantin Belousov static struct syscall_helper_data aio_syscalls[] = {
425723d37c0SKonstantin Belousov 	SYSCALL_INIT_HELPER(aio_cancel),
426723d37c0SKonstantin Belousov 	SYSCALL_INIT_HELPER(aio_error),
427723d37c0SKonstantin Belousov 	SYSCALL_INIT_HELPER(aio_fsync),
428723d37c0SKonstantin Belousov 	SYSCALL_INIT_HELPER(aio_read),
429723d37c0SKonstantin Belousov 	SYSCALL_INIT_HELPER(aio_return),
430723d37c0SKonstantin Belousov 	SYSCALL_INIT_HELPER(aio_suspend),
431723d37c0SKonstantin Belousov 	SYSCALL_INIT_HELPER(aio_waitcomplete),
432723d37c0SKonstantin Belousov 	SYSCALL_INIT_HELPER(aio_write),
433723d37c0SKonstantin Belousov 	SYSCALL_INIT_HELPER(lio_listio),
434723d37c0SKonstantin Belousov 	SYSCALL_INIT_HELPER(oaio_read),
435723d37c0SKonstantin Belousov 	SYSCALL_INIT_HELPER(oaio_write),
436723d37c0SKonstantin Belousov 	SYSCALL_INIT_HELPER(olio_listio),
437723d37c0SKonstantin Belousov 	SYSCALL_INIT_LAST
438723d37c0SKonstantin Belousov };
439723d37c0SKonstantin Belousov 
440723d37c0SKonstantin Belousov #ifdef COMPAT_FREEBSD32
441723d37c0SKonstantin Belousov #include <sys/mount.h>
442723d37c0SKonstantin Belousov #include <sys/socket.h>
443723d37c0SKonstantin Belousov #include <compat/freebsd32/freebsd32.h>
444723d37c0SKonstantin Belousov #include <compat/freebsd32/freebsd32_proto.h>
445723d37c0SKonstantin Belousov #include <compat/freebsd32/freebsd32_signal.h>
446723d37c0SKonstantin Belousov #include <compat/freebsd32/freebsd32_syscall.h>
447723d37c0SKonstantin Belousov #include <compat/freebsd32/freebsd32_util.h>
448723d37c0SKonstantin Belousov 
449723d37c0SKonstantin Belousov static struct syscall_helper_data aio32_syscalls[] = {
450723d37c0SKonstantin Belousov 	SYSCALL32_INIT_HELPER(freebsd32_aio_return),
451723d37c0SKonstantin Belousov 	SYSCALL32_INIT_HELPER(freebsd32_aio_suspend),
452723d37c0SKonstantin Belousov 	SYSCALL32_INIT_HELPER(freebsd32_aio_cancel),
453723d37c0SKonstantin Belousov 	SYSCALL32_INIT_HELPER(freebsd32_aio_error),
454723d37c0SKonstantin Belousov 	SYSCALL32_INIT_HELPER(freebsd32_aio_fsync),
455723d37c0SKonstantin Belousov 	SYSCALL32_INIT_HELPER(freebsd32_aio_read),
456723d37c0SKonstantin Belousov 	SYSCALL32_INIT_HELPER(freebsd32_aio_write),
457723d37c0SKonstantin Belousov 	SYSCALL32_INIT_HELPER(freebsd32_aio_waitcomplete),
458723d37c0SKonstantin Belousov 	SYSCALL32_INIT_HELPER(freebsd32_lio_listio),
459723d37c0SKonstantin Belousov 	SYSCALL32_INIT_HELPER(freebsd32_oaio_read),
460723d37c0SKonstantin Belousov 	SYSCALL32_INIT_HELPER(freebsd32_oaio_write),
461723d37c0SKonstantin Belousov 	SYSCALL32_INIT_HELPER(freebsd32_olio_listio),
462723d37c0SKonstantin Belousov 	SYSCALL_INIT_LAST
463723d37c0SKonstantin Belousov };
464723d37c0SKonstantin Belousov #endif
46521d56e9cSAlfred Perlstein 
46621d56e9cSAlfred Perlstein DECLARE_MODULE(aio, aio_mod,
46721d56e9cSAlfred Perlstein 	SI_SUB_VFS, SI_ORDER_ANY);
46821d56e9cSAlfred Perlstein MODULE_VERSION(aio, 1);
46921d56e9cSAlfred Perlstein 
470fd3bf775SJohn Dyson /*
4712244ea07SJohn Dyson  * Startup initialization
4722244ea07SJohn Dyson  */
473723d37c0SKonstantin Belousov static int
47421d56e9cSAlfred Perlstein aio_onceonly(void)
475fd3bf775SJohn Dyson {
476723d37c0SKonstantin Belousov 	int error;
47721d56e9cSAlfred Perlstein 
47821d56e9cSAlfred Perlstein 	/* XXX: should probably just use so->callback */
47921d56e9cSAlfred Perlstein 	aio_swake = &aio_swake_cb;
48075b8b3b2SJohn Baldwin 	exit_tag = EVENTHANDLER_REGISTER(process_exit, aio_proc_rundown, NULL,
48175b8b3b2SJohn Baldwin 	    EVENTHANDLER_PRI_ANY);
482993182e5SAlexander Leidinger 	exec_tag = EVENTHANDLER_REGISTER(process_exec, aio_proc_rundown_exec, NULL,
48375b8b3b2SJohn Baldwin 	    EVENTHANDLER_PRI_ANY);
48421d56e9cSAlfred Perlstein 	kqueue_add_filteropts(EVFILT_AIO, &aio_filtops);
48569cd28daSDoug Ambrisko 	kqueue_add_filteropts(EVFILT_LIO, &lio_filtops);
4862244ea07SJohn Dyson 	TAILQ_INIT(&aio_freeproc);
4871ce91824SDavid Xu 	sema_init(&aio_newproc_sem, 0, "aio_new_proc");
4881ce91824SDavid Xu 	mtx_init(&aio_job_mtx, "aio_job", NULL, MTX_DEF);
4891ce91824SDavid Xu 	mtx_init(&aio_sock_mtx, "aio_sock", NULL, MTX_DEF);
4902244ea07SJohn Dyson 	TAILQ_INIT(&aio_jobs);
4911ce91824SDavid Xu 	aiod_unr = new_unrhdr(1, INT_MAX, NULL);
492c897b813SJeff Roberson 	kaio_zone = uma_zcreate("AIO", sizeof(struct kaioinfo), NULL, NULL,
493c897b813SJeff Roberson 	    NULL, NULL, UMA_ALIGN_PTR, UMA_ZONE_NOFREE);
494c897b813SJeff Roberson 	aiop_zone = uma_zcreate("AIOP", sizeof(struct aiothreadlist), NULL,
495c897b813SJeff Roberson 	    NULL, NULL, NULL, UMA_ALIGN_PTR, UMA_ZONE_NOFREE);
496c897b813SJeff Roberson 	aiocb_zone = uma_zcreate("AIOCB", sizeof(struct aiocblist), NULL, NULL,
497c897b813SJeff Roberson 	    NULL, NULL, UMA_ALIGN_PTR, UMA_ZONE_NOFREE);
498c897b813SJeff Roberson 	aiol_zone = uma_zcreate("AIOL", AIO_LISTIO_MAX*sizeof(intptr_t) , NULL,
499c897b813SJeff Roberson 	    NULL, NULL, NULL, UMA_ALIGN_PTR, UMA_ZONE_NOFREE);
5001ce91824SDavid Xu 	aiolio_zone = uma_zcreate("AIOLIO", sizeof(struct aioliojob), NULL,
501c897b813SJeff Roberson 	    NULL, NULL, NULL, UMA_ALIGN_PTR, UMA_ZONE_NOFREE);
50284af4da6SJohn Dyson 	aiod_timeout = AIOD_TIMEOUT_DEFAULT;
50384af4da6SJohn Dyson 	aiod_lifetime = AIOD_LIFETIME_DEFAULT;
504fd3bf775SJohn Dyson 	jobrefid = 1;
505c7047e52SGarrett Wollman 	async_io_version = _POSIX_VERSION;
506c844abc9SAlfred Perlstein 	p31b_setcfg(CTL_P1003_1B_AIO_LISTIO_MAX, AIO_LISTIO_MAX);
50786d52125SAlfred Perlstein 	p31b_setcfg(CTL_P1003_1B_AIO_MAX, MAX_AIO_QUEUE);
50886d52125SAlfred Perlstein 	p31b_setcfg(CTL_P1003_1B_AIO_PRIO_DELTA_MAX, 0);
509723d37c0SKonstantin Belousov 
510723d37c0SKonstantin Belousov 	error = syscall_helper_register(aio_syscalls);
511723d37c0SKonstantin Belousov 	if (error)
512723d37c0SKonstantin Belousov 		return (error);
513723d37c0SKonstantin Belousov #ifdef COMPAT_FREEBSD32
514723d37c0SKonstantin Belousov 	error = syscall32_helper_register(aio32_syscalls);
515723d37c0SKonstantin Belousov 	if (error)
516723d37c0SKonstantin Belousov 		return (error);
517723d37c0SKonstantin Belousov #endif
518723d37c0SKonstantin Belousov 	return (0);
5192244ea07SJohn Dyson }
5202244ea07SJohn Dyson 
521eb8e6d52SEivind Eklund /*
522eb8e6d52SEivind Eklund  * Callback for unload of AIO when used as a module.
523eb8e6d52SEivind Eklund  */
52421d56e9cSAlfred Perlstein static int
52521d56e9cSAlfred Perlstein aio_unload(void)
52621d56e9cSAlfred Perlstein {
527ad3b9257SJohn-Mark Gurney 	int error;
52821d56e9cSAlfred Perlstein 
52921d56e9cSAlfred Perlstein 	/*
53021d56e9cSAlfred Perlstein 	 * XXX: no unloads by default, it's too dangerous.
53121d56e9cSAlfred Perlstein 	 * perhaps we could do it if locked out callers and then
53221d56e9cSAlfred Perlstein 	 * did an aio_proc_rundown() on each process.
5332a522eb9SJohn Baldwin 	 *
5342a522eb9SJohn Baldwin 	 * jhb: aio_proc_rundown() needs to run on curproc though,
5352a522eb9SJohn Baldwin 	 * so I don't think that would fly.
53621d56e9cSAlfred Perlstein 	 */
53721d56e9cSAlfred Perlstein 	if (!unloadable)
53821d56e9cSAlfred Perlstein 		return (EOPNOTSUPP);
53921d56e9cSAlfred Perlstein 
540723d37c0SKonstantin Belousov #ifdef COMPAT_FREEBSD32
541723d37c0SKonstantin Belousov 	syscall32_helper_unregister(aio32_syscalls);
542723d37c0SKonstantin Belousov #endif
543723d37c0SKonstantin Belousov 	syscall_helper_unregister(aio_syscalls);
544723d37c0SKonstantin Belousov 
545ad3b9257SJohn-Mark Gurney 	error = kqueue_del_filteropts(EVFILT_AIO);
546ad3b9257SJohn-Mark Gurney 	if (error)
547ad3b9257SJohn-Mark Gurney 		return error;
548bd793be3SDavid Xu 	error = kqueue_del_filteropts(EVFILT_LIO);
549bd793be3SDavid Xu 	if (error)
550bd793be3SDavid Xu 		return error;
551c7047e52SGarrett Wollman 	async_io_version = 0;
55221d56e9cSAlfred Perlstein 	aio_swake = NULL;
5531ce91824SDavid Xu 	taskqueue_free(taskqueue_aiod_bio);
5541ce91824SDavid Xu 	delete_unrhdr(aiod_unr);
555bd793be3SDavid Xu 	uma_zdestroy(kaio_zone);
556bd793be3SDavid Xu 	uma_zdestroy(aiop_zone);
557bd793be3SDavid Xu 	uma_zdestroy(aiocb_zone);
558bd793be3SDavid Xu 	uma_zdestroy(aiol_zone);
559bd793be3SDavid Xu 	uma_zdestroy(aiolio_zone);
56075b8b3b2SJohn Baldwin 	EVENTHANDLER_DEREGISTER(process_exit, exit_tag);
56175b8b3b2SJohn Baldwin 	EVENTHANDLER_DEREGISTER(process_exec, exec_tag);
5621ce91824SDavid Xu 	mtx_destroy(&aio_job_mtx);
5631ce91824SDavid Xu 	mtx_destroy(&aio_sock_mtx);
5641ce91824SDavid Xu 	sema_destroy(&aio_newproc_sem);
565f51c1e89SAlfred Perlstein 	p31b_setcfg(CTL_P1003_1B_AIO_LISTIO_MAX, -1);
566f51c1e89SAlfred Perlstein 	p31b_setcfg(CTL_P1003_1B_AIO_MAX, -1);
567f51c1e89SAlfred Perlstein 	p31b_setcfg(CTL_P1003_1B_AIO_PRIO_DELTA_MAX, -1);
56821d56e9cSAlfred Perlstein 	return (0);
56921d56e9cSAlfred Perlstein }
57021d56e9cSAlfred Perlstein 
5712244ea07SJohn Dyson /*
572bfbbc4aaSJason Evans  * Init the per-process aioinfo structure.  The aioinfo limits are set
573bfbbc4aaSJason Evans  * per-process for user limit (resource) management.
5742244ea07SJohn Dyson  */
5756a1162d4SAlexander Leidinger void
576fd3bf775SJohn Dyson aio_init_aioinfo(struct proc *p)
577fd3bf775SJohn Dyson {
5782244ea07SJohn Dyson 	struct kaioinfo *ki;
579ac41f2efSAlfred Perlstein 
580a163d034SWarner Losh 	ki = uma_zalloc(kaio_zone, M_WAITOK);
581759ccccaSDavid Xu 	mtx_init(&ki->kaio_mtx, "aiomtx", NULL, MTX_DEF);
58284af4da6SJohn Dyson 	ki->kaio_flags = 0;
583a624e84fSJohn Dyson 	ki->kaio_maxactive_count = max_aio_per_proc;
5842244ea07SJohn Dyson 	ki->kaio_active_count = 0;
585a624e84fSJohn Dyson 	ki->kaio_qallowed_count = max_aio_queue_per_proc;
5861ce91824SDavid Xu 	ki->kaio_count = 0;
58784af4da6SJohn Dyson 	ki->kaio_ballowed_count = max_buf_aio;
588fd3bf775SJohn Dyson 	ki->kaio_buffer_count = 0;
5891ce91824SDavid Xu 	TAILQ_INIT(&ki->kaio_all);
5901ce91824SDavid Xu 	TAILQ_INIT(&ki->kaio_done);
5912244ea07SJohn Dyson 	TAILQ_INIT(&ki->kaio_jobqueue);
592fd3bf775SJohn Dyson 	TAILQ_INIT(&ki->kaio_bufqueue);
59384af4da6SJohn Dyson 	TAILQ_INIT(&ki->kaio_liojoblist);
594bfbbc4aaSJason Evans 	TAILQ_INIT(&ki->kaio_sockqueue);
59599eee864SDavid Xu 	TAILQ_INIT(&ki->kaio_syncqueue);
59699eee864SDavid Xu 	TASK_INIT(&ki->kaio_task, 0, aio_kick_helper, p);
5973999ebe3SAlan Cox 	PROC_LOCK(p);
5983999ebe3SAlan Cox 	if (p->p_aioinfo == NULL) {
5993999ebe3SAlan Cox 		p->p_aioinfo = ki;
6003999ebe3SAlan Cox 		PROC_UNLOCK(p);
6013999ebe3SAlan Cox 	} else {
6023999ebe3SAlan Cox 		PROC_UNLOCK(p);
603759ccccaSDavid Xu 		mtx_destroy(&ki->kaio_mtx);
6043999ebe3SAlan Cox 		uma_zfree(kaio_zone, ki);
6052244ea07SJohn Dyson 	}
606bfbbc4aaSJason Evans 
60722035f47SOleksandr Tymoshenko 	while (num_aio_procs < MIN(target_aio_procs, max_aio_procs))
6081ce91824SDavid Xu 		aio_newproc(NULL);
6092244ea07SJohn Dyson }
6102244ea07SJohn Dyson 
6114c0fb2cfSDavid Xu static int
6124c0fb2cfSDavid Xu aio_sendsig(struct proc *p, struct sigevent *sigev, ksiginfo_t *ksi)
6134c0fb2cfSDavid Xu {
614cf7d9a8cSDavid Xu 	struct thread *td;
615cf7d9a8cSDavid Xu 	int error;
616759ccccaSDavid Xu 
617cf7d9a8cSDavid Xu 	error = sigev_findtd(p, sigev, &td);
618cf7d9a8cSDavid Xu 	if (error)
619cf7d9a8cSDavid Xu 		return (error);
6204c0fb2cfSDavid Xu 	if (!KSI_ONQ(ksi)) {
621cf7d9a8cSDavid Xu 		ksiginfo_set_sigev(ksi, sigev);
6224c0fb2cfSDavid Xu 		ksi->ksi_code = SI_ASYNCIO;
6234c0fb2cfSDavid Xu 		ksi->ksi_flags |= KSI_EXT | KSI_INS;
624cf7d9a8cSDavid Xu 		tdsendsignal(p, td, ksi->ksi_signo, ksi);
6254c0fb2cfSDavid Xu 	}
626759ccccaSDavid Xu 	PROC_UNLOCK(p);
627cf7d9a8cSDavid Xu 	return (error);
6284c0fb2cfSDavid Xu }
6294c0fb2cfSDavid Xu 
6302244ea07SJohn Dyson /*
631bfbbc4aaSJason Evans  * Free a job entry.  Wait for completion if it is currently active, but don't
632bfbbc4aaSJason Evans  * delay forever.  If we delay, we return a flag that says that we have to
633bfbbc4aaSJason Evans  * restart the queue scan.
6342244ea07SJohn Dyson  */
63588ed460eSAlan Cox static int
636fd3bf775SJohn Dyson aio_free_entry(struct aiocblist *aiocbe)
637fd3bf775SJohn Dyson {
6382244ea07SJohn Dyson 	struct kaioinfo *ki;
6391ce91824SDavid Xu 	struct aioliojob *lj;
6402244ea07SJohn Dyson 	struct proc *p;
6412244ea07SJohn Dyson 
6422244ea07SJohn Dyson 	p = aiocbe->userproc;
6431ce91824SDavid Xu 	MPASS(curproc == p);
6442244ea07SJohn Dyson 	ki = p->p_aioinfo;
6451ce91824SDavid Xu 	MPASS(ki != NULL);
6461ce91824SDavid Xu 
647759ccccaSDavid Xu 	AIO_LOCK_ASSERT(ki, MA_OWNED);
648759ccccaSDavid Xu 	MPASS(aiocbe->jobstate == JOBST_JOBFINISHED);
649759ccccaSDavid Xu 
6501ce91824SDavid Xu 	atomic_subtract_int(&num_queue_count, 1);
6511ce91824SDavid Xu 
6521ce91824SDavid Xu 	ki->kaio_count--;
6531ce91824SDavid Xu 	MPASS(ki->kaio_count >= 0);
6541ce91824SDavid Xu 
65527b8220dSDavid Xu 	TAILQ_REMOVE(&ki->kaio_done, aiocbe, plist);
65627b8220dSDavid Xu 	TAILQ_REMOVE(&ki->kaio_all, aiocbe, allist);
65727b8220dSDavid Xu 
65884af4da6SJohn Dyson 	lj = aiocbe->lio;
65984af4da6SJohn Dyson 	if (lj) {
6601ce91824SDavid Xu 		lj->lioj_count--;
6611ce91824SDavid Xu 		lj->lioj_finished_count--;
6621ce91824SDavid Xu 
663a9bf5e37SDavid Xu 		if (lj->lioj_count == 0) {
6641ce91824SDavid Xu 			TAILQ_REMOVE(&ki->kaio_liojoblist, lj, lioj_list);
6651ce91824SDavid Xu 			/* lio is going away, we need to destroy any knotes */
6661ce91824SDavid Xu 			knlist_delete(&lj->klist, curthread, 1);
667759ccccaSDavid Xu 			PROC_LOCK(p);
6681ce91824SDavid Xu 			sigqueue_take(&lj->lioj_ksi);
669759ccccaSDavid Xu 			PROC_UNLOCK(p);
6701ce91824SDavid Xu 			uma_zfree(aiolio_zone, lj);
67184af4da6SJohn Dyson 		}
67284af4da6SJohn Dyson 	}
6731ce91824SDavid Xu 
674cb679c38SJonathan Lemon 	/* aiocbe is going away, we need to destroy any knotes */
6751ce91824SDavid Xu 	knlist_delete(&aiocbe->klist, curthread, 1);
676759ccccaSDavid Xu 	PROC_LOCK(p);
6771ce91824SDavid Xu 	sigqueue_take(&aiocbe->ksi);
678759ccccaSDavid Xu 	PROC_UNLOCK(p);
6791ce91824SDavid Xu 
6801ce91824SDavid Xu 	MPASS(aiocbe->bp == NULL);
6811ce91824SDavid Xu 	aiocbe->jobstate = JOBST_NULL;
682759ccccaSDavid Xu 	AIO_UNLOCK(ki);
6832a522eb9SJohn Baldwin 
6842a522eb9SJohn Baldwin 	/*
6852a522eb9SJohn Baldwin 	 * The thread argument here is used to find the owning process
6862a522eb9SJohn Baldwin 	 * and is also passed to fo_close() which may pass it to various
6872a522eb9SJohn Baldwin 	 * places such as devsw close() routines.  Because of that, we
6882a522eb9SJohn Baldwin 	 * need a thread pointer from the process owning the job that is
6892a522eb9SJohn Baldwin 	 * persistent and won't disappear out from under us or move to
6902a522eb9SJohn Baldwin 	 * another process.
6912a522eb9SJohn Baldwin 	 *
6922a522eb9SJohn Baldwin 	 * Currently, all the callers of this function call it to remove
6932a522eb9SJohn Baldwin 	 * an aiocblist from the current process' job list either via a
6942a522eb9SJohn Baldwin 	 * syscall or due to the current process calling exit() or
6952a522eb9SJohn Baldwin 	 * execve().  Thus, we know that p == curproc.  We also know that
6962a522eb9SJohn Baldwin 	 * curthread can't exit since we are curthread.
6972a522eb9SJohn Baldwin 	 *
6982a522eb9SJohn Baldwin 	 * Therefore, we use curthread as the thread to pass to
6992a522eb9SJohn Baldwin 	 * knlist_delete().  This does mean that it is possible for the
7002a522eb9SJohn Baldwin 	 * thread pointer at close time to differ from the thread pointer
7012a522eb9SJohn Baldwin 	 * at open time, but this is already true of file descriptors in
7022a522eb9SJohn Baldwin 	 * a multithreaded process.
703b40ce416SJulian Elischer 	 */
704a5c0b1c0SAlan Cox 	fdrop(aiocbe->fd_file, curthread);
705f8f750c5SRobert Watson 	crfree(aiocbe->cred);
706c897b813SJeff Roberson 	uma_zfree(aiocb_zone, aiocbe);
707759ccccaSDavid Xu 	AIO_LOCK(ki);
7081ce91824SDavid Xu 
709ac41f2efSAlfred Perlstein 	return (0);
7102244ea07SJohn Dyson }
7112244ea07SJohn Dyson 
712993182e5SAlexander Leidinger static void
713993182e5SAlexander Leidinger aio_proc_rundown_exec(void *arg, struct proc *p, struct image_params *imgp __unused)
714993182e5SAlexander Leidinger {
715993182e5SAlexander Leidinger    	aio_proc_rundown(arg, p);
716993182e5SAlexander Leidinger }
717993182e5SAlexander Leidinger 
7182244ea07SJohn Dyson /*
7192244ea07SJohn Dyson  * Rundown the jobs for a given process.
7202244ea07SJohn Dyson  */
72121d56e9cSAlfred Perlstein static void
72275b8b3b2SJohn Baldwin aio_proc_rundown(void *arg, struct proc *p)
723fd3bf775SJohn Dyson {
7242244ea07SJohn Dyson 	struct kaioinfo *ki;
7251ce91824SDavid Xu 	struct aioliojob *lj;
7261ce91824SDavid Xu 	struct aiocblist *cbe, *cbn;
727bfbbc4aaSJason Evans 	struct file *fp;
728bfbbc4aaSJason Evans 	struct socket *so;
7291aa4c324SDavid Xu 	int remove;
7302244ea07SJohn Dyson 
7312a522eb9SJohn Baldwin 	KASSERT(curthread->td_proc == p,
7322a522eb9SJohn Baldwin 	    ("%s: called on non-curproc", __func__));
7332244ea07SJohn Dyson 	ki = p->p_aioinfo;
7342244ea07SJohn Dyson 	if (ki == NULL)
7352244ea07SJohn Dyson 		return;
7362244ea07SJohn Dyson 
737759ccccaSDavid Xu 	AIO_LOCK(ki);
73827b8220dSDavid Xu 	ki->kaio_flags |= KAIO_RUNDOWN;
7391ce91824SDavid Xu 
7401ce91824SDavid Xu restart:
741a624e84fSJohn Dyson 
742bfbbc4aaSJason Evans 	/*
7431ce91824SDavid Xu 	 * Try to cancel all pending requests. This code simulates
7441ce91824SDavid Xu 	 * aio_cancel on all pending I/O requests.
745bfbbc4aaSJason Evans 	 */
7461ce91824SDavid Xu 	TAILQ_FOREACH_SAFE(cbe, &ki->kaio_jobqueue, plist, cbn) {
7471aa4c324SDavid Xu 		remove = 0;
7481ce91824SDavid Xu 		mtx_lock(&aio_job_mtx);
7491ce91824SDavid Xu 		if (cbe->jobstate == JOBST_JOBQGLOBAL) {
7501ce91824SDavid Xu 			TAILQ_REMOVE(&aio_jobs, cbe, list);
7511aa4c324SDavid Xu 			remove = 1;
7521aa4c324SDavid Xu 		} else if (cbe->jobstate == JOBST_JOBQSOCK) {
7531aa4c324SDavid Xu 			fp = cbe->fd_file;
7541aa4c324SDavid Xu 			MPASS(fp->f_type == DTYPE_SOCKET);
7551aa4c324SDavid Xu 			so = fp->f_data;
7561aa4c324SDavid Xu 			TAILQ_REMOVE(&so->so_aiojobq, cbe, list);
7571aa4c324SDavid Xu 			remove = 1;
75899eee864SDavid Xu 		} else if (cbe->jobstate == JOBST_JOBQSYNC) {
75999eee864SDavid Xu 			TAILQ_REMOVE(&ki->kaio_syncqueue, cbe, list);
76099eee864SDavid Xu 			remove = 1;
7611aa4c324SDavid Xu 		}
7621ce91824SDavid Xu 		mtx_unlock(&aio_job_mtx);
7631aa4c324SDavid Xu 
7641aa4c324SDavid Xu 		if (remove) {
7651ce91824SDavid Xu 			cbe->jobstate = JOBST_JOBFINISHED;
7661ce91824SDavid Xu 			cbe->uaiocb._aiocb_private.status = -1;
7671ce91824SDavid Xu 			cbe->uaiocb._aiocb_private.error = ECANCELED;
7681ce91824SDavid Xu 			TAILQ_REMOVE(&ki->kaio_jobqueue, cbe, plist);
7691ce91824SDavid Xu 			aio_bio_done_notify(p, cbe, DONE_QUEUE);
7701ce91824SDavid Xu 		}
7712244ea07SJohn Dyson 	}
77284af4da6SJohn Dyson 
7731ce91824SDavid Xu 	/* Wait for all running I/O to be finished */
7741ce91824SDavid Xu 	if (TAILQ_FIRST(&ki->kaio_bufqueue) ||
7751ce91824SDavid Xu 	    TAILQ_FIRST(&ki->kaio_jobqueue)) {
77684af4da6SJohn Dyson 		ki->kaio_flags |= KAIO_WAKEUP;
777759ccccaSDavid Xu 		msleep(&p->p_aioinfo, AIO_MTX(ki), PRIBIO, "aioprn", hz);
7781ce91824SDavid Xu 		goto restart;
77984af4da6SJohn Dyson 	}
78084af4da6SJohn Dyson 
7811ce91824SDavid Xu 	/* Free all completed I/O requests. */
7821ce91824SDavid Xu 	while ((cbe = TAILQ_FIRST(&ki->kaio_done)) != NULL)
7831ce91824SDavid Xu 		aio_free_entry(cbe);
78484af4da6SJohn Dyson 
7851ce91824SDavid Xu 	while ((lj = TAILQ_FIRST(&ki->kaio_liojoblist)) != NULL) {
786a9bf5e37SDavid Xu 		if (lj->lioj_count == 0) {
78784af4da6SJohn Dyson 			TAILQ_REMOVE(&ki->kaio_liojoblist, lj, lioj_list);
7881ce91824SDavid Xu 			knlist_delete(&lj->klist, curthread, 1);
789759ccccaSDavid Xu 			PROC_LOCK(p);
7901ce91824SDavid Xu 			sigqueue_take(&lj->lioj_ksi);
791759ccccaSDavid Xu 			PROC_UNLOCK(p);
792c897b813SJeff Roberson 			uma_zfree(aiolio_zone, lj);
793f4f0ecefSJohn Dyson 		} else {
794a9bf5e37SDavid Xu 			panic("LIO job not cleaned up: C:%d, FC:%d\n",
795a9bf5e37SDavid Xu 			    lj->lioj_count, lj->lioj_finished_count);
79684af4da6SJohn Dyson 		}
797f4f0ecefSJohn Dyson 	}
798759ccccaSDavid Xu 	AIO_UNLOCK(ki);
79999eee864SDavid Xu 	taskqueue_drain(taskqueue_aiod_bio, &ki->kaio_task);
8005114048bSKonstantin Belousov 	mtx_destroy(&ki->kaio_mtx);
801c897b813SJeff Roberson 	uma_zfree(kaio_zone, ki);
802a624e84fSJohn Dyson 	p->p_aioinfo = NULL;
8032244ea07SJohn Dyson }
8042244ea07SJohn Dyson 
8052244ea07SJohn Dyson /*
806bfbbc4aaSJason Evans  * Select a job to run (called by an AIO daemon).
8072244ea07SJohn Dyson  */
8082244ea07SJohn Dyson static struct aiocblist *
809b40ce416SJulian Elischer aio_selectjob(struct aiothreadlist *aiop)
810fd3bf775SJohn Dyson {
8112244ea07SJohn Dyson 	struct aiocblist *aiocbe;
812bfbbc4aaSJason Evans 	struct kaioinfo *ki;
813bfbbc4aaSJason Evans 	struct proc *userp;
8142244ea07SJohn Dyson 
8151ce91824SDavid Xu 	mtx_assert(&aio_job_mtx, MA_OWNED);
8162a522eb9SJohn Baldwin 	TAILQ_FOREACH(aiocbe, &aio_jobs, list) {
8172244ea07SJohn Dyson 		userp = aiocbe->userproc;
8182244ea07SJohn Dyson 		ki = userp->p_aioinfo;
8192244ea07SJohn Dyson 
8202244ea07SJohn Dyson 		if (ki->kaio_active_count < ki->kaio_maxactive_count) {
8212244ea07SJohn Dyson 			TAILQ_REMOVE(&aio_jobs, aiocbe, list);
8221ce91824SDavid Xu 			/* Account for currently active jobs. */
8231ce91824SDavid Xu 			ki->kaio_active_count++;
8241ce91824SDavid Xu 			aiocbe->jobstate = JOBST_JOBRUNNING;
8251ce91824SDavid Xu 			break;
8261ce91824SDavid Xu 		}
8271ce91824SDavid Xu 	}
828ac41f2efSAlfred Perlstein 	return (aiocbe);
8292244ea07SJohn Dyson }
8302244ea07SJohn Dyson 
8312244ea07SJohn Dyson /*
83299eee864SDavid Xu  *  Move all data to a permanent storage device, this code
83399eee864SDavid Xu  *  simulates fsync syscall.
83499eee864SDavid Xu  */
83599eee864SDavid Xu static int
83699eee864SDavid Xu aio_fsync_vnode(struct thread *td, struct vnode *vp)
83799eee864SDavid Xu {
83899eee864SDavid Xu 	struct mount *mp;
83999eee864SDavid Xu 	int error;
84099eee864SDavid Xu 
84199eee864SDavid Xu 	if ((error = vn_start_write(vp, &mp, V_WAIT | PCATCH)) != 0)
84299eee864SDavid Xu 		goto drop;
843cb05b60aSAttilio Rao 	vn_lock(vp, LK_EXCLUSIVE | LK_RETRY);
84499eee864SDavid Xu 	if (vp->v_object != NULL) {
84589f6b863SAttilio Rao 		VM_OBJECT_WLOCK(vp->v_object);
84699eee864SDavid Xu 		vm_object_page_clean(vp->v_object, 0, 0, 0);
84789f6b863SAttilio Rao 		VM_OBJECT_WUNLOCK(vp->v_object);
84899eee864SDavid Xu 	}
84999eee864SDavid Xu 	error = VOP_FSYNC(vp, MNT_WAIT, td);
85099eee864SDavid Xu 
85122db15c0SAttilio Rao 	VOP_UNLOCK(vp, 0);
85299eee864SDavid Xu 	vn_finished_write(mp);
85399eee864SDavid Xu drop:
85499eee864SDavid Xu 	return (error);
85599eee864SDavid Xu }
85699eee864SDavid Xu 
85799eee864SDavid Xu /*
858bfbbc4aaSJason Evans  * The AIO processing activity.  This is the code that does the I/O request for
859bfbbc4aaSJason Evans  * the non-physio version of the operations.  The normal vn operations are used,
860bfbbc4aaSJason Evans  * and this code should work in all instances for every type of file, including
861bfbbc4aaSJason Evans  * pipes, sockets, fifos, and regular files.
8621ce91824SDavid Xu  *
8631aa4c324SDavid Xu  * XXX I don't think it works well for socket, pipe, and fifo.
8642244ea07SJohn Dyson  */
86588ed460eSAlan Cox static void
866fd3bf775SJohn Dyson aio_process(struct aiocblist *aiocbe)
867fd3bf775SJohn Dyson {
868f8f750c5SRobert Watson 	struct ucred *td_savedcred;
869b40ce416SJulian Elischer 	struct thread *td;
8702244ea07SJohn Dyson 	struct aiocb *cb;
8712244ea07SJohn Dyson 	struct file *fp;
8721ce91824SDavid Xu 	struct socket *so;
8732244ea07SJohn Dyson 	struct uio auio;
8742244ea07SJohn Dyson 	struct iovec aiov;
8752244ea07SJohn Dyson 	int cnt;
8762244ea07SJohn Dyson 	int error;
877fd3bf775SJohn Dyson 	int oublock_st, oublock_end;
878fd3bf775SJohn Dyson 	int inblock_st, inblock_end;
8792244ea07SJohn Dyson 
880b40ce416SJulian Elischer 	td = curthread;
881f8f750c5SRobert Watson 	td_savedcred = td->td_ucred;
882f8f750c5SRobert Watson 	td->td_ucred = aiocbe->cred;
8832244ea07SJohn Dyson 	cb = &aiocbe->uaiocb;
88400e73160SAlan Cox 	fp = aiocbe->fd_file;
885bfbbc4aaSJason Evans 
88699eee864SDavid Xu 	if (cb->aio_lio_opcode == LIO_SYNC) {
88799eee864SDavid Xu 		error = 0;
88899eee864SDavid Xu 		cnt = 0;
88999eee864SDavid Xu 		if (fp->f_vnode != NULL)
89099eee864SDavid Xu 			error = aio_fsync_vnode(td, fp->f_vnode);
89199eee864SDavid Xu 		cb->_aiocb_private.error = error;
89299eee864SDavid Xu 		cb->_aiocb_private.status = 0;
89399eee864SDavid Xu 		td->td_ucred = td_savedcred;
89499eee864SDavid Xu 		return;
89599eee864SDavid Xu 	}
89699eee864SDavid Xu 
89791369fc7SAlan Cox 	aiov.iov_base = (void *)(uintptr_t)cb->aio_buf;
8982244ea07SJohn Dyson 	aiov.iov_len = cb->aio_nbytes;
8992244ea07SJohn Dyson 
9002244ea07SJohn Dyson 	auio.uio_iov = &aiov;
9012244ea07SJohn Dyson 	auio.uio_iovcnt = 1;
9029b16adc1SAlan Cox 	auio.uio_offset = cb->aio_offset;
9032244ea07SJohn Dyson 	auio.uio_resid = cb->aio_nbytes;
9042244ea07SJohn Dyson 	cnt = cb->aio_nbytes;
9052244ea07SJohn Dyson 	auio.uio_segflg = UIO_USERSPACE;
906b40ce416SJulian Elischer 	auio.uio_td = td;
9072244ea07SJohn Dyson 
9081c4bcd05SJeff Roberson 	inblock_st = td->td_ru.ru_inblock;
9091c4bcd05SJeff Roberson 	oublock_st = td->td_ru.ru_oublock;
910279d7226SMatthew Dillon 	/*
911a9bf5e37SDavid Xu 	 * aio_aqueue() acquires a reference to the file that is
9129b16adc1SAlan Cox 	 * released in aio_free_entry().
913279d7226SMatthew Dillon 	 */
9142244ea07SJohn Dyson 	if (cb->aio_lio_opcode == LIO_READ) {
9152244ea07SJohn Dyson 		auio.uio_rw = UIO_READ;
9165114048bSKonstantin Belousov 		if (auio.uio_resid == 0)
9175114048bSKonstantin Belousov 			error = 0;
9185114048bSKonstantin Belousov 		else
919b40ce416SJulian Elischer 			error = fo_read(fp, &auio, fp->f_cred, FOF_OFFSET, td);
9202244ea07SJohn Dyson 	} else {
9216d53aa62SDavid Xu 		if (fp->f_type == DTYPE_VNODE)
9226d53aa62SDavid Xu 			bwillwrite();
9232244ea07SJohn Dyson 		auio.uio_rw = UIO_WRITE;
924b40ce416SJulian Elischer 		error = fo_write(fp, &auio, fp->f_cred, FOF_OFFSET, td);
9252244ea07SJohn Dyson 	}
9261c4bcd05SJeff Roberson 	inblock_end = td->td_ru.ru_inblock;
9271c4bcd05SJeff Roberson 	oublock_end = td->td_ru.ru_oublock;
928fd3bf775SJohn Dyson 
929fd3bf775SJohn Dyson 	aiocbe->inputcharge = inblock_end - inblock_st;
930fd3bf775SJohn Dyson 	aiocbe->outputcharge = oublock_end - oublock_st;
9312244ea07SJohn Dyson 
932bfbbc4aaSJason Evans 	if ((error) && (auio.uio_resid != cnt)) {
9332244ea07SJohn Dyson 		if (error == ERESTART || error == EINTR || error == EWOULDBLOCK)
9342244ea07SJohn Dyson 			error = 0;
93519eb87d2SJohn Baldwin 		if ((error == EPIPE) && (cb->aio_lio_opcode == LIO_WRITE)) {
9361ce91824SDavid Xu 			int sigpipe = 1;
9371ce91824SDavid Xu 			if (fp->f_type == DTYPE_SOCKET) {
9381ce91824SDavid Xu 				so = fp->f_data;
9391ce91824SDavid Xu 				if (so->so_options & SO_NOSIGPIPE)
9401ce91824SDavid Xu 					sigpipe = 0;
9411ce91824SDavid Xu 			}
9421ce91824SDavid Xu 			if (sigpipe) {
9439b16adc1SAlan Cox 				PROC_LOCK(aiocbe->userproc);
9448451d0ddSKip Macy 				kern_psignal(aiocbe->userproc, SIGPIPE);
9459b16adc1SAlan Cox 				PROC_UNLOCK(aiocbe->userproc);
94619eb87d2SJohn Baldwin 			}
9472244ea07SJohn Dyson 		}
9481ce91824SDavid Xu 	}
9492244ea07SJohn Dyson 
9502244ea07SJohn Dyson 	cnt -= auio.uio_resid;
9512244ea07SJohn Dyson 	cb->_aiocb_private.error = error;
9522244ea07SJohn Dyson 	cb->_aiocb_private.status = cnt;
953f8f750c5SRobert Watson 	td->td_ucred = td_savedcred;
9542244ea07SJohn Dyson }
9552244ea07SJohn Dyson 
95669cd28daSDoug Ambrisko static void
9571ce91824SDavid Xu aio_bio_done_notify(struct proc *userp, struct aiocblist *aiocbe, int type)
9581ce91824SDavid Xu {
9591ce91824SDavid Xu 	struct aioliojob *lj;
96069cd28daSDoug Ambrisko 	struct kaioinfo *ki;
96199eee864SDavid Xu 	struct aiocblist *scb, *scbn;
9621ce91824SDavid Xu 	int lj_done;
96369cd28daSDoug Ambrisko 
96469cd28daSDoug Ambrisko 	ki = userp->p_aioinfo;
965759ccccaSDavid Xu 	AIO_LOCK_ASSERT(ki, MA_OWNED);
96669cd28daSDoug Ambrisko 	lj = aiocbe->lio;
96769cd28daSDoug Ambrisko 	lj_done = 0;
96869cd28daSDoug Ambrisko 	if (lj) {
9691ce91824SDavid Xu 		lj->lioj_finished_count++;
9701ce91824SDavid Xu 		if (lj->lioj_count == lj->lioj_finished_count)
97169cd28daSDoug Ambrisko 			lj_done = 1;
97269cd28daSDoug Ambrisko 	}
97369cd28daSDoug Ambrisko 	if (type == DONE_QUEUE) {
9741ce91824SDavid Xu 		aiocbe->jobflags |= AIOCBLIST_DONE;
97569cd28daSDoug Ambrisko 	} else {
9761ce91824SDavid Xu 		aiocbe->jobflags |= AIOCBLIST_BUFDONE;
97769cd28daSDoug Ambrisko 	}
9781ce91824SDavid Xu 	TAILQ_INSERT_TAIL(&ki->kaio_done, aiocbe, plist);
9791ce91824SDavid Xu 	aiocbe->jobstate = JOBST_JOBFINISHED;
98027b8220dSDavid Xu 
98127b8220dSDavid Xu 	if (ki->kaio_flags & KAIO_RUNDOWN)
98227b8220dSDavid Xu 		goto notification_done;
98327b8220dSDavid Xu 
9841ce91824SDavid Xu 	if (aiocbe->uaiocb.aio_sigevent.sigev_notify == SIGEV_SIGNAL ||
9851ce91824SDavid Xu 	    aiocbe->uaiocb.aio_sigevent.sigev_notify == SIGEV_THREAD_ID)
9861ce91824SDavid Xu 		aio_sendsig(userp, &aiocbe->uaiocb.aio_sigevent, &aiocbe->ksi);
9871ce91824SDavid Xu 
9881ce91824SDavid Xu 	KNOTE_LOCKED(&aiocbe->klist, 1);
9891ce91824SDavid Xu 
99069cd28daSDoug Ambrisko 	if (lj_done) {
9911ce91824SDavid Xu 		if (lj->lioj_signal.sigev_notify == SIGEV_KEVENT) {
99269cd28daSDoug Ambrisko 			lj->lioj_flags |= LIOJ_KEVENT_POSTED;
9931ce91824SDavid Xu 			KNOTE_LOCKED(&lj->klist, 1);
99469cd28daSDoug Ambrisko 		}
9951ce91824SDavid Xu 		if ((lj->lioj_flags & (LIOJ_SIGNAL|LIOJ_SIGNAL_POSTED))
99669cd28daSDoug Ambrisko 		    == LIOJ_SIGNAL
9974c0fb2cfSDavid Xu 		    && (lj->lioj_signal.sigev_notify == SIGEV_SIGNAL ||
9984c0fb2cfSDavid Xu 		        lj->lioj_signal.sigev_notify == SIGEV_THREAD_ID)) {
9994c0fb2cfSDavid Xu 			aio_sendsig(userp, &lj->lioj_signal, &lj->lioj_ksi);
100069cd28daSDoug Ambrisko 			lj->lioj_flags |= LIOJ_SIGNAL_POSTED;
100169cd28daSDoug Ambrisko 		}
100269cd28daSDoug Ambrisko 	}
100327b8220dSDavid Xu 
100427b8220dSDavid Xu notification_done:
100599eee864SDavid Xu 	if (aiocbe->jobflags & AIOCBLIST_CHECKSYNC) {
100699eee864SDavid Xu 		TAILQ_FOREACH_SAFE(scb, &ki->kaio_syncqueue, list, scbn) {
1007dbbccfe9SDavid Xu 			if (aiocbe->fd_file == scb->fd_file &&
100899eee864SDavid Xu 			    aiocbe->seqno < scb->seqno) {
100999eee864SDavid Xu 				if (--scb->pending == 0) {
101099eee864SDavid Xu 					mtx_lock(&aio_job_mtx);
101199eee864SDavid Xu 					scb->jobstate = JOBST_JOBQGLOBAL;
101299eee864SDavid Xu 					TAILQ_REMOVE(&ki->kaio_syncqueue, scb, list);
101399eee864SDavid Xu 					TAILQ_INSERT_TAIL(&aio_jobs, scb, list);
101499eee864SDavid Xu 					aio_kick_nowait(userp);
101599eee864SDavid Xu 					mtx_unlock(&aio_job_mtx);
101699eee864SDavid Xu 				}
101799eee864SDavid Xu 			}
101899eee864SDavid Xu 		}
101999eee864SDavid Xu 	}
102027b8220dSDavid Xu 	if (ki->kaio_flags & KAIO_WAKEUP) {
102169cd28daSDoug Ambrisko 		ki->kaio_flags &= ~KAIO_WAKEUP;
10221ce91824SDavid Xu 		wakeup(&userp->p_aioinfo);
102369cd28daSDoug Ambrisko 	}
102469cd28daSDoug Ambrisko }
102569cd28daSDoug Ambrisko 
10262244ea07SJohn Dyson /*
102784af4da6SJohn Dyson  * The AIO daemon, most of the actual work is done in aio_process,
102884af4da6SJohn Dyson  * but the setup (and address space mgmt) is done in this routine.
10292244ea07SJohn Dyson  */
10302244ea07SJohn Dyson static void
10311ce91824SDavid Xu aio_daemon(void *_id)
10322244ea07SJohn Dyson {
1033bfbbc4aaSJason Evans 	struct aiocblist *aiocbe;
1034b40ce416SJulian Elischer 	struct aiothreadlist *aiop;
1035bfbbc4aaSJason Evans 	struct kaioinfo *ki;
1036bfbbc4aaSJason Evans 	struct proc *curcp, *mycp, *userp;
1037bfbbc4aaSJason Evans 	struct vmspace *myvm, *tmpvm;
1038b40ce416SJulian Elischer 	struct thread *td = curthread;
10391ce91824SDavid Xu 	int id = (intptr_t)_id;
10402244ea07SJohn Dyson 
10412244ea07SJohn Dyson 	/*
1042fd3bf775SJohn Dyson 	 * Local copies of curproc (cp) and vmspace (myvm)
10432244ea07SJohn Dyson 	 */
1044b40ce416SJulian Elischer 	mycp = td->td_proc;
1045fd3bf775SJohn Dyson 	myvm = mycp->p_vmspace;
1046fd3bf775SJohn Dyson 
1047cd4ed3b5SJohn Baldwin 	KASSERT(mycp->p_textvp == NULL, ("kthread has a textvp"));
1048fd3bf775SJohn Dyson 
1049fd3bf775SJohn Dyson 	/*
1050bfbbc4aaSJason Evans 	 * Allocate and ready the aio control info.  There is one aiop structure
1051bfbbc4aaSJason Evans 	 * per daemon.
1052fd3bf775SJohn Dyson 	 */
1053a163d034SWarner Losh 	aiop = uma_zalloc(aiop_zone, M_WAITOK);
1054b40ce416SJulian Elischer 	aiop->aiothread = td;
105599eee864SDavid Xu 	aiop->aiothreadflags = 0;
1056bfbbc4aaSJason Evans 
1057e6bdc05fSDavid Xu 	/* The daemon resides in its own pgrp. */
10588451d0ddSKip Macy 	sys_setsid(td, NULL);
1059fd3bf775SJohn Dyson 
1060fd3bf775SJohn Dyson 	/*
1061fd3bf775SJohn Dyson 	 * Wakeup parent process.  (Parent sleeps to keep from blasting away
1062b40ce416SJulian Elischer 	 * and creating too many daemons.)
1063fd3bf775SJohn Dyson 	 */
10641ce91824SDavid Xu 	sema_post(&aio_newproc_sem);
10652244ea07SJohn Dyson 
10661ce91824SDavid Xu 	mtx_lock(&aio_job_mtx);
1067bfbbc4aaSJason Evans 	for (;;) {
1068fd3bf775SJohn Dyson 		/*
1069fd3bf775SJohn Dyson 		 * curcp is the current daemon process context.
1070fd3bf775SJohn Dyson 		 * userp is the current user process context.
1071fd3bf775SJohn Dyson 		 */
1072fd3bf775SJohn Dyson 		curcp = mycp;
1073c4860686SJohn Dyson 
1074fd3bf775SJohn Dyson 		/*
1075fd3bf775SJohn Dyson 		 * Take daemon off of free queue
1076fd3bf775SJohn Dyson 		 */
1077b40ce416SJulian Elischer 		if (aiop->aiothreadflags & AIOP_FREE) {
10782244ea07SJohn Dyson 			TAILQ_REMOVE(&aio_freeproc, aiop, list);
1079b40ce416SJulian Elischer 			aiop->aiothreadflags &= ~AIOP_FREE;
10802244ea07SJohn Dyson 		}
10812244ea07SJohn Dyson 
1082fd3bf775SJohn Dyson 		/*
1083bfbbc4aaSJason Evans 		 * Check for jobs.
1084fd3bf775SJohn Dyson 		 */
1085d254af07SMatthew Dillon 		while ((aiocbe = aio_selectjob(aiop)) != NULL) {
10861ce91824SDavid Xu 			mtx_unlock(&aio_job_mtx);
10872244ea07SJohn Dyson 			userp = aiocbe->userproc;
10882244ea07SJohn Dyson 
1089fd3bf775SJohn Dyson 			/*
1090bfbbc4aaSJason Evans 			 * Connect to process address space for user program.
1091fd3bf775SJohn Dyson 			 */
1092fd3bf775SJohn Dyson 			if (userp != curcp) {
1093fd3bf775SJohn Dyson 				/*
1094bfbbc4aaSJason Evans 				 * Save the current address space that we are
1095bfbbc4aaSJason Evans 				 * connected to.
1096fd3bf775SJohn Dyson 				 */
1097fd3bf775SJohn Dyson 				tmpvm = mycp->p_vmspace;
1098bfbbc4aaSJason Evans 
1099fd3bf775SJohn Dyson 				/*
1100bfbbc4aaSJason Evans 				 * Point to the new user address space, and
1101bfbbc4aaSJason Evans 				 * refer to it.
1102fd3bf775SJohn Dyson 				 */
1103fd3bf775SJohn Dyson 				mycp->p_vmspace = userp->p_vmspace;
11041a276a3fSAlan Cox 				atomic_add_int(&mycp->p_vmspace->vm_refcnt, 1);
1105bfbbc4aaSJason Evans 
1106bfbbc4aaSJason Evans 				/* Activate the new mapping. */
1107079b7badSJulian Elischer 				pmap_activate(FIRST_THREAD_IN_PROC(mycp));
1108bfbbc4aaSJason Evans 
1109fd3bf775SJohn Dyson 				/*
1110bfbbc4aaSJason Evans 				 * If the old address space wasn't the daemons
1111bfbbc4aaSJason Evans 				 * own address space, then we need to remove the
1112bfbbc4aaSJason Evans 				 * daemon's reference from the other process
1113bfbbc4aaSJason Evans 				 * that it was acting on behalf of.
1114fd3bf775SJohn Dyson 				 */
11152244ea07SJohn Dyson 				if (tmpvm != myvm) {
11162244ea07SJohn Dyson 					vmspace_free(tmpvm);
11172244ea07SJohn Dyson 				}
1118fd3bf775SJohn Dyson 				curcp = userp;
11192244ea07SJohn Dyson 			}
11202244ea07SJohn Dyson 
1121fd3bf775SJohn Dyson 			ki = userp->p_aioinfo;
112284af4da6SJohn Dyson 
1123bfbbc4aaSJason Evans 			/* Do the I/O function. */
11242244ea07SJohn Dyson 			aio_process(aiocbe);
112584af4da6SJohn Dyson 
11269b84335cSDavid Xu 			mtx_lock(&aio_job_mtx);
11279b84335cSDavid Xu 			/* Decrement the active job count. */
11289b84335cSDavid Xu 			ki->kaio_active_count--;
11299b84335cSDavid Xu 			mtx_unlock(&aio_job_mtx);
11309b84335cSDavid Xu 
1131759ccccaSDavid Xu 			AIO_LOCK(ki);
11321ce91824SDavid Xu 			TAILQ_REMOVE(&ki->kaio_jobqueue, aiocbe, plist);
113369cd28daSDoug Ambrisko 			aio_bio_done_notify(userp, aiocbe, DONE_QUEUE);
1134759ccccaSDavid Xu 			AIO_UNLOCK(ki);
11351ce91824SDavid Xu 
11361ce91824SDavid Xu 			mtx_lock(&aio_job_mtx);
11372244ea07SJohn Dyson 		}
11382244ea07SJohn Dyson 
1139fd3bf775SJohn Dyson 		/*
1140bfbbc4aaSJason Evans 		 * Disconnect from user address space.
1141fd3bf775SJohn Dyson 		 */
1142fd3bf775SJohn Dyson 		if (curcp != mycp) {
11431ce91824SDavid Xu 
11441ce91824SDavid Xu 			mtx_unlock(&aio_job_mtx);
11451ce91824SDavid Xu 
1146bfbbc4aaSJason Evans 			/* Get the user address space to disconnect from. */
1147fd3bf775SJohn Dyson 			tmpvm = mycp->p_vmspace;
1148bfbbc4aaSJason Evans 
1149bfbbc4aaSJason Evans 			/* Get original address space for daemon. */
1150fd3bf775SJohn Dyson 			mycp->p_vmspace = myvm;
1151bfbbc4aaSJason Evans 
1152bfbbc4aaSJason Evans 			/* Activate the daemon's address space. */
1153079b7badSJulian Elischer 			pmap_activate(FIRST_THREAD_IN_PROC(mycp));
1154bfbbc4aaSJason Evans #ifdef DIAGNOSTIC
1155bfbbc4aaSJason Evans 			if (tmpvm == myvm) {
1156bfbbc4aaSJason Evans 				printf("AIOD: vmspace problem -- %d\n",
1157bfbbc4aaSJason Evans 				    mycp->p_pid);
1158bfbbc4aaSJason Evans 			}
115911783b14SJohn Dyson #endif
1160bfbbc4aaSJason Evans 			/* Remove our vmspace reference. */
11612244ea07SJohn Dyson 			vmspace_free(tmpvm);
1162bfbbc4aaSJason Evans 
1163fd3bf775SJohn Dyson 			curcp = mycp;
11641ce91824SDavid Xu 
11651ce91824SDavid Xu 			mtx_lock(&aio_job_mtx);
11661ce91824SDavid Xu 			/*
11671ce91824SDavid Xu 			 * We have to restart to avoid race, we only sleep if
11681ce91824SDavid Xu 			 * no job can be selected, that should be
11691ce91824SDavid Xu 			 * curcp == mycp.
11701ce91824SDavid Xu 			 */
11711ce91824SDavid Xu 			continue;
1172fd3bf775SJohn Dyson 		}
1173fd3bf775SJohn Dyson 
11741ce91824SDavid Xu 		mtx_assert(&aio_job_mtx, MA_OWNED);
11751ce91824SDavid Xu 
1176fd3bf775SJohn Dyson 		TAILQ_INSERT_HEAD(&aio_freeproc, aiop, list);
1177b40ce416SJulian Elischer 		aiop->aiothreadflags |= AIOP_FREE;
1178fd3bf775SJohn Dyson 
1179fd3bf775SJohn Dyson 		/*
1180bfbbc4aaSJason Evans 		 * If daemon is inactive for a long time, allow it to exit,
1181bfbbc4aaSJason Evans 		 * thereby freeing resources.
1182fd3bf775SJohn Dyson 		 */
11831ce91824SDavid Xu 		if (msleep(aiop->aiothread, &aio_job_mtx, PRIBIO, "aiordy",
11841ce91824SDavid Xu 		    aiod_lifetime)) {
1185c3869e4bSAlan Cox 			if (TAILQ_EMPTY(&aio_jobs)) {
1186b40ce416SJulian Elischer 				if ((aiop->aiothreadflags & AIOP_FREE) &&
118784af4da6SJohn Dyson 				    (num_aio_procs > target_aio_procs)) {
1188fd3bf775SJohn Dyson 					TAILQ_REMOVE(&aio_freeproc, aiop, list);
118984af4da6SJohn Dyson 					num_aio_procs--;
11901ce91824SDavid Xu 					mtx_unlock(&aio_job_mtx);
11911ce91824SDavid Xu 					uma_zfree(aiop_zone, aiop);
11921ce91824SDavid Xu 					free_unr(aiod_unr, id);
1193bfbbc4aaSJason Evans #ifdef DIAGNOSTIC
1194bfbbc4aaSJason Evans 					if (mycp->p_vmspace->vm_refcnt <= 1) {
1195bfbbc4aaSJason Evans 						printf("AIOD: bad vm refcnt for"
1196bfbbc4aaSJason Evans 						    " exiting daemon: %d\n",
1197fd3bf775SJohn Dyson 						    mycp->p_vmspace->vm_refcnt);
1198bfbbc4aaSJason Evans 					}
119911783b14SJohn Dyson #endif
12003745c395SJulian Elischer 					kproc_exit(0);
1201fd3bf775SJohn Dyson 				}
12022244ea07SJohn Dyson 			}
12032244ea07SJohn Dyson 		}
12042244ea07SJohn Dyson 	}
12051ce91824SDavid Xu 	mtx_unlock(&aio_job_mtx);
12061ce91824SDavid Xu 	panic("shouldn't be here\n");
12071ce91824SDavid Xu }
12082244ea07SJohn Dyson 
12092244ea07SJohn Dyson /*
1210bfbbc4aaSJason Evans  * Create a new AIO daemon. This is mostly a kernel-thread fork routine. The
1211bfbbc4aaSJason Evans  * AIO daemon modifies its environment itself.
12122244ea07SJohn Dyson  */
12132244ea07SJohn Dyson static int
12141ce91824SDavid Xu aio_newproc(int *start)
1215fd3bf775SJohn Dyson {
12162244ea07SJohn Dyson 	int error;
1217c9a970a7SAlan Cox 	struct proc *p;
12181ce91824SDavid Xu 	int id;
12192244ea07SJohn Dyson 
12201ce91824SDavid Xu 	id = alloc_unr(aiod_unr);
12213745c395SJulian Elischer 	error = kproc_create(aio_daemon, (void *)(intptr_t)id, &p,
12221ce91824SDavid Xu 		RFNOWAIT, 0, "aiod%d", id);
12231ce91824SDavid Xu 	if (error == 0) {
1224fd3bf775SJohn Dyson 		/*
12251ce91824SDavid Xu 		 * Wait until daemon is started.
1226fd3bf775SJohn Dyson 		 */
12271ce91824SDavid Xu 		sema_wait(&aio_newproc_sem);
12281ce91824SDavid Xu 		mtx_lock(&aio_job_mtx);
122984af4da6SJohn Dyson 		num_aio_procs++;
12301ce91824SDavid Xu 		if (start != NULL)
12317f34b521SDavid Xu 			(*start)--;
12321ce91824SDavid Xu 		mtx_unlock(&aio_job_mtx);
12331ce91824SDavid Xu 	} else {
12341ce91824SDavid Xu 		free_unr(aiod_unr, id);
12351ce91824SDavid Xu 	}
1236ac41f2efSAlfred Perlstein 	return (error);
12372244ea07SJohn Dyson }
12382244ea07SJohn Dyson 
12392244ea07SJohn Dyson /*
124088ed460eSAlan Cox  * Try the high-performance, low-overhead physio method for eligible
124188ed460eSAlan Cox  * VCHR devices.  This method doesn't use an aio helper thread, and
124288ed460eSAlan Cox  * thus has very low overhead.
124388ed460eSAlan Cox  *
1244a9bf5e37SDavid Xu  * Assumes that the caller, aio_aqueue(), has incremented the file
124588ed460eSAlan Cox  * structure's reference count, preventing its deallocation for the
124688ed460eSAlan Cox  * duration of this call.
1247fd3bf775SJohn Dyson  */
124888ed460eSAlan Cox static int
1249bfbbc4aaSJason Evans aio_qphysio(struct proc *p, struct aiocblist *aiocbe)
1250fd3bf775SJohn Dyson {
1251fd3bf775SJohn Dyson 	struct aiocb *cb;
1252fd3bf775SJohn Dyson 	struct file *fp;
1253fd3bf775SJohn Dyson 	struct buf *bp;
1254fd3bf775SJohn Dyson 	struct vnode *vp;
1255*f3215a60SKonstantin Belousov 	struct cdevsw *csw;
1256*f3215a60SKonstantin Belousov 	struct cdev *dev;
1257fd3bf775SJohn Dyson 	struct kaioinfo *ki;
12581ce91824SDavid Xu 	struct aioliojob *lj;
1259*f3215a60SKonstantin Belousov 	int error, ref;
1260fd3bf775SJohn Dyson 
126184af4da6SJohn Dyson 	cb = &aiocbe->uaiocb;
12629fbd7ccfSAlan Cox 	fp = aiocbe->fd_file;
1263fd3bf775SJohn Dyson 
1264008626c3SPoul-Henning Kamp 	if (fp->f_type != DTYPE_VNODE)
1265008626c3SPoul-Henning Kamp 		return (-1);
1266fd3bf775SJohn Dyson 
12673b6d9652SPoul-Henning Kamp 	vp = fp->f_vnode;
126811783b14SJohn Dyson 
1269f582ac06SBrian Feldman 	/*
1270f582ac06SBrian Feldman 	 * If its not a disk, we don't want to return a positive error.
1271f582ac06SBrian Feldman 	 * It causes the aio code to not fall through to try the thread
1272f582ac06SBrian Feldman 	 * way when you're talking to a regular file.
1273f582ac06SBrian Feldman 	 */
1274f582ac06SBrian Feldman 	if (!vn_isdisk(vp, &error)) {
1275f582ac06SBrian Feldman 		if (error == ENOTBLK)
1276f582ac06SBrian Feldman 			return (-1);
1277f582ac06SBrian Feldman 		else
1278ba4ad1fcSPoul-Henning Kamp 			return (error);
1279f582ac06SBrian Feldman 	}
1280fd3bf775SJohn Dyson 
1281ad8de0f2SDavid Xu 	if (vp->v_bufobj.bo_bsize == 0)
1282ad8de0f2SDavid Xu 		return (-1);
1283ad8de0f2SDavid Xu 
12845d9d81e7SPoul-Henning Kamp  	if (cb->aio_nbytes % vp->v_bufobj.bo_bsize)
1285008626c3SPoul-Henning Kamp 		return (-1);
1286fd3bf775SJohn Dyson 
128713644654SAlan Cox 	if (cb->aio_nbytes >
128813644654SAlan Cox 	    MAXPHYS - (((vm_offset_t) cb->aio_buf) & PAGE_MASK))
1289008626c3SPoul-Henning Kamp 		return (-1);
1290fd3bf775SJohn Dyson 
1291fd3bf775SJohn Dyson 	ki = p->p_aioinfo;
1292008626c3SPoul-Henning Kamp 	if (ki->kaio_buffer_count >= ki->kaio_ballowed_count)
1293008626c3SPoul-Henning Kamp 		return (-1);
1294fd3bf775SJohn Dyson 
1295*f3215a60SKonstantin Belousov 	ref = 0;
1296*f3215a60SKonstantin Belousov 	csw = devvn_refthread(vp, &dev, &ref);
1297*f3215a60SKonstantin Belousov 	if (csw == NULL)
1298*f3215a60SKonstantin Belousov 		return (ENXIO);
1299*f3215a60SKonstantin Belousov 	if (cb->aio_nbytes > dev->si_iosize_max) {
1300*f3215a60SKonstantin Belousov 		error = -1;
1301*f3215a60SKonstantin Belousov 		goto unref;
1302*f3215a60SKonstantin Belousov 	}
1303*f3215a60SKonstantin Belousov 
1304bfbbc4aaSJason Evans 	/* Create and build a buffer header for a transfer. */
13051c7c3c6aSMatthew Dillon 	bp = (struct buf *)getpbuf(NULL);
130613644654SAlan Cox 	BUF_KERNPROC(bp);
1307fd3bf775SJohn Dyson 
1308759ccccaSDavid Xu 	AIO_LOCK(ki);
13091ce91824SDavid Xu 	ki->kaio_count++;
13101ce91824SDavid Xu 	ki->kaio_buffer_count++;
13111ce91824SDavid Xu 	lj = aiocbe->lio;
13121ce91824SDavid Xu 	if (lj)
13131ce91824SDavid Xu 		lj->lioj_count++;
1314759ccccaSDavid Xu 	AIO_UNLOCK(ki);
13151ce91824SDavid Xu 
1316fd3bf775SJohn Dyson 	/*
1317bfbbc4aaSJason Evans 	 * Get a copy of the kva from the physical buffer.
1318fd3bf775SJohn Dyson 	 */
1319ef38cda1SAlan Cox 	error = 0;
1320fd3bf775SJohn Dyson 
1321fd3bf775SJohn Dyson 	bp->b_bcount = cb->aio_nbytes;
1322fd3bf775SJohn Dyson 	bp->b_bufsize = cb->aio_nbytes;
1323fd3bf775SJohn Dyson 	bp->b_iodone = aio_physwakeup;
1324fd3bf775SJohn Dyson 	bp->b_saveaddr = bp->b_data;
132591369fc7SAlan Cox 	bp->b_data = (void *)(uintptr_t)cb->aio_buf;
1326a44ca4f0SHidetoshi Shimokawa 	bp->b_offset = cb->aio_offset;
1327a44ca4f0SHidetoshi Shimokawa 	bp->b_iooffset = cb->aio_offset;
1328fd3bf775SJohn Dyson 	bp->b_blkno = btodb(cb->aio_offset);
132906363906SAlan Cox 	bp->b_iocmd = cb->aio_lio_opcode == LIO_WRITE ? BIO_WRITE : BIO_READ;
1330fd3bf775SJohn Dyson 
13312d5c7e45SMatthew Dillon 	/*
13322d5c7e45SMatthew Dillon 	 * Bring buffer into kernel space.
13332d5c7e45SMatthew Dillon 	 */
1334*f3215a60SKonstantin Belousov 	if (vmapbuf(bp, (csw->d_flags & D_UNMAPPED_IO) == 0) < 0) {
13352d5c7e45SMatthew Dillon 		error = EFAULT;
13362d5c7e45SMatthew Dillon 		goto doerror;
13372d5c7e45SMatthew Dillon 	}
1338fd3bf775SJohn Dyson 
1339759ccccaSDavid Xu 	AIO_LOCK(ki);
1340fd3bf775SJohn Dyson 	aiocbe->bp = bp;
13418edbaf85SHidetoshi Shimokawa 	bp->b_caller1 = (void *)aiocbe;
134284af4da6SJohn Dyson 	TAILQ_INSERT_TAIL(&ki->kaio_bufqueue, aiocbe, plist);
13431ce91824SDavid Xu 	TAILQ_INSERT_TAIL(&ki->kaio_all, aiocbe, allist);
1344fd3bf775SJohn Dyson 	aiocbe->jobstate = JOBST_JOBQBUF;
134584af4da6SJohn Dyson 	cb->_aiocb_private.status = cb->aio_nbytes;
1346759ccccaSDavid Xu 	AIO_UNLOCK(ki);
13471ce91824SDavid Xu 
13481ce91824SDavid Xu 	atomic_add_int(&num_queue_count, 1);
13491ce91824SDavid Xu 	atomic_add_int(&num_buf_aio, 1);
13501ce91824SDavid Xu 
1351fd3bf775SJohn Dyson 	bp->b_error = 0;
1352fd3bf775SJohn Dyson 
13531ce91824SDavid Xu 	TASK_INIT(&aiocbe->biotask, 0, biohelper, aiocbe);
1354bfbbc4aaSJason Evans 
1355bfbbc4aaSJason Evans 	/* Perform transfer. */
1356*f3215a60SKonstantin Belousov 	dev_strategy_csw(dev, csw, bp);
1357*f3215a60SKonstantin Belousov 	dev_relthread(dev, ref);
1358ac41f2efSAlfred Perlstein 	return (0);
1359fd3bf775SJohn Dyson 
1360fd3bf775SJohn Dyson doerror:
1361759ccccaSDavid Xu 	AIO_LOCK(ki);
13621ce91824SDavid Xu 	ki->kaio_count--;
1363fd3bf775SJohn Dyson 	ki->kaio_buffer_count--;
1364bfbbc4aaSJason Evans 	if (lj)
13651ce91824SDavid Xu 		lj->lioj_count--;
136684af4da6SJohn Dyson 	aiocbe->bp = NULL;
1367759ccccaSDavid Xu 	AIO_UNLOCK(ki);
13681c7c3c6aSMatthew Dillon 	relpbuf(bp, NULL);
1369*f3215a60SKonstantin Belousov unref:
1370*f3215a60SKonstantin Belousov 	dev_relthread(dev, ref);
1371fd3bf775SJohn Dyson 	return (error);
1372fd3bf775SJohn Dyson }
1373fd3bf775SJohn Dyson 
1374fd3bf775SJohn Dyson /*
1375bfbbc4aaSJason Evans  * Wake up aio requests that may be serviceable now.
1376bfbbc4aaSJason Evans  */
137748dac059SAlan Cox static void
137821d56e9cSAlfred Perlstein aio_swake_cb(struct socket *so, struct sockbuf *sb)
1379bfbbc4aaSJason Evans {
1380bfbbc4aaSJason Evans 	struct aiocblist *cb, *cbn;
138199eee864SDavid Xu 	int opcode;
1382bfbbc4aaSJason Evans 
138374fb0ba7SJohn Baldwin 	SOCKBUF_LOCK_ASSERT(sb);
13841aa4c324SDavid Xu 	if (sb == &so->so_snd)
1385bfbbc4aaSJason Evans 		opcode = LIO_WRITE;
13861aa4c324SDavid Xu 	else
1387bfbbc4aaSJason Evans 		opcode = LIO_READ;
1388bfbbc4aaSJason Evans 
13891aa4c324SDavid Xu 	sb->sb_flags &= ~SB_AIO;
13901aa4c324SDavid Xu 	mtx_lock(&aio_job_mtx);
13912a522eb9SJohn Baldwin 	TAILQ_FOREACH_SAFE(cb, &so->so_aiojobq, list, cbn) {
1392bfbbc4aaSJason Evans 		if (opcode == cb->uaiocb.aio_lio_opcode) {
13938c0d9af5SDavid Xu 			if (cb->jobstate != JOBST_JOBQSOCK)
13941ce91824SDavid Xu 				panic("invalid queue value");
13951aa4c324SDavid Xu 			/* XXX
13961aa4c324SDavid Xu 			 * We don't have actual sockets backend yet,
13971aa4c324SDavid Xu 			 * so we simply move the requests to the generic
13981aa4c324SDavid Xu 			 * file I/O backend.
13991ce91824SDavid Xu 			 */
14001aa4c324SDavid Xu 			TAILQ_REMOVE(&so->so_aiojobq, cb, list);
14011ce91824SDavid Xu 			TAILQ_INSERT_TAIL(&aio_jobs, cb, list);
140299eee864SDavid Xu 			aio_kick_nowait(cb->userproc);
1403bfbbc4aaSJason Evans 		}
1404bfbbc4aaSJason Evans 	}
14051aa4c324SDavid Xu 	mtx_unlock(&aio_job_mtx);
1406bfbbc4aaSJason Evans }
1407bfbbc4aaSJason Evans 
14083858a1f4SJohn Baldwin static int
14093858a1f4SJohn Baldwin convert_old_sigevent(struct osigevent *osig, struct sigevent *nsig)
14103858a1f4SJohn Baldwin {
14113858a1f4SJohn Baldwin 
14123858a1f4SJohn Baldwin 	/*
14133858a1f4SJohn Baldwin 	 * Only SIGEV_NONE, SIGEV_SIGNAL, and SIGEV_KEVENT are
14143858a1f4SJohn Baldwin 	 * supported by AIO with the old sigevent structure.
14153858a1f4SJohn Baldwin 	 */
14163858a1f4SJohn Baldwin 	nsig->sigev_notify = osig->sigev_notify;
14173858a1f4SJohn Baldwin 	switch (nsig->sigev_notify) {
14183858a1f4SJohn Baldwin 	case SIGEV_NONE:
14193858a1f4SJohn Baldwin 		break;
14203858a1f4SJohn Baldwin 	case SIGEV_SIGNAL:
14213858a1f4SJohn Baldwin 		nsig->sigev_signo = osig->__sigev_u.__sigev_signo;
14223858a1f4SJohn Baldwin 		break;
14233858a1f4SJohn Baldwin 	case SIGEV_KEVENT:
14243858a1f4SJohn Baldwin 		nsig->sigev_notify_kqueue =
14253858a1f4SJohn Baldwin 		    osig->__sigev_u.__sigev_notify_kqueue;
14263858a1f4SJohn Baldwin 		nsig->sigev_value.sival_ptr = osig->sigev_value.sival_ptr;
14273858a1f4SJohn Baldwin 		break;
14283858a1f4SJohn Baldwin 	default:
14293858a1f4SJohn Baldwin 		return (EINVAL);
14303858a1f4SJohn Baldwin 	}
14313858a1f4SJohn Baldwin 	return (0);
14323858a1f4SJohn Baldwin }
14333858a1f4SJohn Baldwin 
14343858a1f4SJohn Baldwin static int
14353858a1f4SJohn Baldwin aiocb_copyin_old_sigevent(struct aiocb *ujob, struct aiocb *kjob)
14363858a1f4SJohn Baldwin {
14373858a1f4SJohn Baldwin 	struct oaiocb *ojob;
14383858a1f4SJohn Baldwin 	int error;
14393858a1f4SJohn Baldwin 
14403858a1f4SJohn Baldwin 	bzero(kjob, sizeof(struct aiocb));
14413858a1f4SJohn Baldwin 	error = copyin(ujob, kjob, sizeof(struct oaiocb));
14423858a1f4SJohn Baldwin 	if (error)
14433858a1f4SJohn Baldwin 		return (error);
14443858a1f4SJohn Baldwin 	ojob = (struct oaiocb *)kjob;
14453858a1f4SJohn Baldwin 	return (convert_old_sigevent(&ojob->aio_sigevent, &kjob->aio_sigevent));
14463858a1f4SJohn Baldwin }
14473858a1f4SJohn Baldwin 
14483858a1f4SJohn Baldwin static int
14493858a1f4SJohn Baldwin aiocb_copyin(struct aiocb *ujob, struct aiocb *kjob)
14503858a1f4SJohn Baldwin {
14513858a1f4SJohn Baldwin 
14523858a1f4SJohn Baldwin 	return (copyin(ujob, kjob, sizeof(struct aiocb)));
14533858a1f4SJohn Baldwin }
14543858a1f4SJohn Baldwin 
14553858a1f4SJohn Baldwin static long
14563858a1f4SJohn Baldwin aiocb_fetch_status(struct aiocb *ujob)
14573858a1f4SJohn Baldwin {
14583858a1f4SJohn Baldwin 
14593858a1f4SJohn Baldwin 	return (fuword(&ujob->_aiocb_private.status));
14603858a1f4SJohn Baldwin }
14613858a1f4SJohn Baldwin 
14623858a1f4SJohn Baldwin static long
14633858a1f4SJohn Baldwin aiocb_fetch_error(struct aiocb *ujob)
14643858a1f4SJohn Baldwin {
14653858a1f4SJohn Baldwin 
14663858a1f4SJohn Baldwin 	return (fuword(&ujob->_aiocb_private.error));
14673858a1f4SJohn Baldwin }
14683858a1f4SJohn Baldwin 
14693858a1f4SJohn Baldwin static int
14703858a1f4SJohn Baldwin aiocb_store_status(struct aiocb *ujob, long status)
14713858a1f4SJohn Baldwin {
14723858a1f4SJohn Baldwin 
14733858a1f4SJohn Baldwin 	return (suword(&ujob->_aiocb_private.status, status));
14743858a1f4SJohn Baldwin }
14753858a1f4SJohn Baldwin 
14763858a1f4SJohn Baldwin static int
14773858a1f4SJohn Baldwin aiocb_store_error(struct aiocb *ujob, long error)
14783858a1f4SJohn Baldwin {
14793858a1f4SJohn Baldwin 
14803858a1f4SJohn Baldwin 	return (suword(&ujob->_aiocb_private.error, error));
14813858a1f4SJohn Baldwin }
14823858a1f4SJohn Baldwin 
14833858a1f4SJohn Baldwin static int
14843858a1f4SJohn Baldwin aiocb_store_kernelinfo(struct aiocb *ujob, long jobref)
14853858a1f4SJohn Baldwin {
14863858a1f4SJohn Baldwin 
14873858a1f4SJohn Baldwin 	return (suword(&ujob->_aiocb_private.kernelinfo, jobref));
14883858a1f4SJohn Baldwin }
14893858a1f4SJohn Baldwin 
14903858a1f4SJohn Baldwin static int
14913858a1f4SJohn Baldwin aiocb_store_aiocb(struct aiocb **ujobp, struct aiocb *ujob)
14923858a1f4SJohn Baldwin {
14933858a1f4SJohn Baldwin 
14943858a1f4SJohn Baldwin 	return (suword(ujobp, (long)ujob));
14953858a1f4SJohn Baldwin }
14963858a1f4SJohn Baldwin 
14973858a1f4SJohn Baldwin static struct aiocb_ops aiocb_ops = {
14983858a1f4SJohn Baldwin 	.copyin = aiocb_copyin,
14993858a1f4SJohn Baldwin 	.fetch_status = aiocb_fetch_status,
15003858a1f4SJohn Baldwin 	.fetch_error = aiocb_fetch_error,
15013858a1f4SJohn Baldwin 	.store_status = aiocb_store_status,
15023858a1f4SJohn Baldwin 	.store_error = aiocb_store_error,
15033858a1f4SJohn Baldwin 	.store_kernelinfo = aiocb_store_kernelinfo,
15043858a1f4SJohn Baldwin 	.store_aiocb = aiocb_store_aiocb,
15053858a1f4SJohn Baldwin };
15063858a1f4SJohn Baldwin 
15073858a1f4SJohn Baldwin static struct aiocb_ops aiocb_ops_osigevent = {
15083858a1f4SJohn Baldwin 	.copyin = aiocb_copyin_old_sigevent,
15093858a1f4SJohn Baldwin 	.fetch_status = aiocb_fetch_status,
15103858a1f4SJohn Baldwin 	.fetch_error = aiocb_fetch_error,
15113858a1f4SJohn Baldwin 	.store_status = aiocb_store_status,
15123858a1f4SJohn Baldwin 	.store_error = aiocb_store_error,
15133858a1f4SJohn Baldwin 	.store_kernelinfo = aiocb_store_kernelinfo,
15143858a1f4SJohn Baldwin 	.store_aiocb = aiocb_store_aiocb,
15153858a1f4SJohn Baldwin };
15163858a1f4SJohn Baldwin 
1517bfbbc4aaSJason Evans /*
1518bfbbc4aaSJason Evans  * Queue a new AIO request.  Choosing either the threaded or direct physio VCHR
1519bfbbc4aaSJason Evans  * technique is done in this code.
15202244ea07SJohn Dyson  */
15216a1162d4SAlexander Leidinger int
1522a9bf5e37SDavid Xu aio_aqueue(struct thread *td, struct aiocb *job, struct aioliojob *lj,
15233858a1f4SJohn Baldwin 	int type, struct aiocb_ops *ops)
1524fd3bf775SJohn Dyson {
1525b40ce416SJulian Elischer 	struct proc *p = td->td_proc;
15262244ea07SJohn Dyson 	struct file *fp;
1527bfbbc4aaSJason Evans 	struct socket *so;
1528dbbccfe9SDavid Xu 	struct aiocblist *aiocbe, *cb;
15292244ea07SJohn Dyson 	struct kaioinfo *ki;
1530c6fa9f78SAlan Cox 	struct kevent kev;
1531576c004fSAlfred Perlstein 	struct sockbuf *sb;
15321ce91824SDavid Xu 	int opcode;
15331ce91824SDavid Xu 	int error;
15344db71d27SJohn-Mark Gurney 	int fd, kqfd;
15351ce91824SDavid Xu 	int jid;
1536fde80935SDavid Xu 	u_short evflags;
15372244ea07SJohn Dyson 
1538a9bf5e37SDavid Xu 	if (p->p_aioinfo == NULL)
1539a9bf5e37SDavid Xu 		aio_init_aioinfo(p);
1540a9bf5e37SDavid Xu 
15411ce91824SDavid Xu 	ki = p->p_aioinfo;
15421ce91824SDavid Xu 
15433858a1f4SJohn Baldwin 	ops->store_status(job, -1);
15443858a1f4SJohn Baldwin 	ops->store_error(job, 0);
15453858a1f4SJohn Baldwin 	ops->store_kernelinfo(job, -1);
1546a9bf5e37SDavid Xu 
1547a9bf5e37SDavid Xu 	if (num_queue_count >= max_queue_count ||
1548a9bf5e37SDavid Xu 	    ki->kaio_count >= ki->kaio_qallowed_count) {
15493858a1f4SJohn Baldwin 		ops->store_error(job, EAGAIN);
1550a9bf5e37SDavid Xu 		return (EAGAIN);
1551a9bf5e37SDavid Xu 	}
1552a9bf5e37SDavid Xu 
15531ce91824SDavid Xu 	aiocbe = uma_zalloc(aiocb_zone, M_WAITOK | M_ZERO);
1554fd3bf775SJohn Dyson 	aiocbe->inputcharge = 0;
1555fd3bf775SJohn Dyson 	aiocbe->outputcharge = 0;
1556d8b0556cSKonstantin Belousov 	knlist_init_mtx(&aiocbe->klist, AIO_MTX(ki));
1557fd3bf775SJohn Dyson 
15583858a1f4SJohn Baldwin 	error = ops->copyin(job, &aiocbe->uaiocb);
15592244ea07SJohn Dyson 	if (error) {
15603858a1f4SJohn Baldwin 		ops->store_error(job, error);
1561c897b813SJeff Roberson 		uma_zfree(aiocb_zone, aiocbe);
1562ac41f2efSAlfred Perlstein 		return (error);
15632244ea07SJohn Dyson 	}
156468d71118SDavid Xu 
1565434ea137SGleb Smirnoff 	/* XXX: aio_nbytes is later casted to signed types. */
156694fce847SGleb Smirnoff 	if (aiocbe->uaiocb.aio_nbytes > INT_MAX) {
1567434ea137SGleb Smirnoff 		uma_zfree(aiocb_zone, aiocbe);
1568434ea137SGleb Smirnoff 		return (EINVAL);
1569434ea137SGleb Smirnoff 	}
1570434ea137SGleb Smirnoff 
157168d71118SDavid Xu 	if (aiocbe->uaiocb.aio_sigevent.sigev_notify != SIGEV_KEVENT &&
157268d71118SDavid Xu 	    aiocbe->uaiocb.aio_sigevent.sigev_notify != SIGEV_SIGNAL &&
157368d71118SDavid Xu 	    aiocbe->uaiocb.aio_sigevent.sigev_notify != SIGEV_THREAD_ID &&
157468d71118SDavid Xu 	    aiocbe->uaiocb.aio_sigevent.sigev_notify != SIGEV_NONE) {
15753858a1f4SJohn Baldwin 		ops->store_error(job, EINVAL);
157668d71118SDavid Xu 		uma_zfree(aiocb_zone, aiocbe);
157768d71118SDavid Xu 		return (EINVAL);
157868d71118SDavid Xu 	}
157968d71118SDavid Xu 
15804c0fb2cfSDavid Xu 	if ((aiocbe->uaiocb.aio_sigevent.sigev_notify == SIGEV_SIGNAL ||
15814c0fb2cfSDavid Xu 	     aiocbe->uaiocb.aio_sigevent.sigev_notify == SIGEV_THREAD_ID) &&
15822f3cf918SAlfred Perlstein 		!_SIG_VALID(aiocbe->uaiocb.aio_sigevent.sigev_signo)) {
1583c897b813SJeff Roberson 		uma_zfree(aiocb_zone, aiocbe);
1584ac41f2efSAlfred Perlstein 		return (EINVAL);
15852f3cf918SAlfred Perlstein 	}
15862244ea07SJohn Dyson 
15874c0fb2cfSDavid Xu 	ksiginfo_init(&aiocbe->ksi);
15884c0fb2cfSDavid Xu 
1589bfbbc4aaSJason Evans 	/* Save userspace address of the job info. */
159011783b14SJohn Dyson 	aiocbe->uuaiocb = job;
159111783b14SJohn Dyson 
1592bfbbc4aaSJason Evans 	/* Get the opcode. */
1593bfbbc4aaSJason Evans 	if (type != LIO_NOP)
1594a624e84fSJohn Dyson 		aiocbe->uaiocb.aio_lio_opcode = type;
1595a624e84fSJohn Dyson 	opcode = aiocbe->uaiocb.aio_lio_opcode;
15962244ea07SJohn Dyson 
1597a9d2f8d8SRobert Watson 	/*
1598a9d2f8d8SRobert Watson 	 * Validate the opcode and fetch the file object for the specified
1599a9d2f8d8SRobert Watson 	 * file descriptor.
1600a9d2f8d8SRobert Watson 	 *
1601a9d2f8d8SRobert Watson 	 * XXXRW: Moved the opcode validation up here so that we don't
1602a9d2f8d8SRobert Watson 	 * retrieve a file descriptor without knowing what the capabiltity
1603a9d2f8d8SRobert Watson 	 * should be.
1604a9d2f8d8SRobert Watson 	 */
16052244ea07SJohn Dyson 	fd = aiocbe->uaiocb.aio_fildes;
16062a522eb9SJohn Baldwin 	switch (opcode) {
16072a522eb9SJohn Baldwin 	case LIO_WRITE:
16082609222aSPawel Jakub Dawidek 		error = fget_write(td, fd, CAP_PWRITE, &fp);
16092a522eb9SJohn Baldwin 		break;
16102a522eb9SJohn Baldwin 	case LIO_READ:
16112609222aSPawel Jakub Dawidek 		error = fget_read(td, fd, CAP_PREAD, &fp);
1612a9d2f8d8SRobert Watson 		break;
1613a9d2f8d8SRobert Watson 	case LIO_SYNC:
1614a9d2f8d8SRobert Watson 		error = fget(td, fd, CAP_FSYNC, &fp);
1615a9d2f8d8SRobert Watson 		break;
1616a9d2f8d8SRobert Watson 	case LIO_NOP:
16172609222aSPawel Jakub Dawidek 		error = fget(td, fd, CAP_NONE, &fp);
16182a522eb9SJohn Baldwin 		break;
16192a522eb9SJohn Baldwin 	default:
1620a9d2f8d8SRobert Watson 		error = EINVAL;
16212a522eb9SJohn Baldwin 	}
16222a522eb9SJohn Baldwin 	if (error) {
1623c897b813SJeff Roberson 		uma_zfree(aiocb_zone, aiocbe);
16243858a1f4SJohn Baldwin 		ops->store_error(job, error);
1625af56abaaSJohn Baldwin 		return (error);
16262244ea07SJohn Dyson 	}
162799eee864SDavid Xu 
162899eee864SDavid Xu 	if (opcode == LIO_SYNC && fp->f_vnode == NULL) {
162999eee864SDavid Xu 		error = EINVAL;
163099eee864SDavid Xu 		goto aqueue_fail;
163199eee864SDavid Xu 	}
16322244ea07SJohn Dyson 
1633dbbccfe9SDavid Xu 	if (opcode != LIO_SYNC && aiocbe->uaiocb.aio_offset == -1LL) {
1634ae124fc4SAlan Cox 		error = EINVAL;
1635ae124fc4SAlan Cox 		goto aqueue_fail;
16362244ea07SJohn Dyson 	}
16371ce91824SDavid Xu 
163899eee864SDavid Xu 	aiocbe->fd_file = fp;
16391ce91824SDavid Xu 
164099eee864SDavid Xu 	mtx_lock(&aio_job_mtx);
164199eee864SDavid Xu 	jid = jobrefid++;
164299eee864SDavid Xu 	aiocbe->seqno = jobseqno++;
164399eee864SDavid Xu 	mtx_unlock(&aio_job_mtx);
16443858a1f4SJohn Baldwin 	error = ops->store_kernelinfo(job, jid);
16451ce91824SDavid Xu 	if (error) {
16461ce91824SDavid Xu 		error = EINVAL;
16471ce91824SDavid Xu 		goto aqueue_fail;
16481ce91824SDavid Xu 	}
16491ce91824SDavid Xu 	aiocbe->uaiocb._aiocb_private.kernelinfo = (void *)(intptr_t)jid;
16502244ea07SJohn Dyson 
16512244ea07SJohn Dyson 	if (opcode == LIO_NOP) {
1652a5c0b1c0SAlan Cox 		fdrop(fp, td);
1653c897b813SJeff Roberson 		uma_zfree(aiocb_zone, aiocbe);
1654ac41f2efSAlfred Perlstein 		return (0);
16552244ea07SJohn Dyson 	}
16562244ea07SJohn Dyson 
16574db71d27SJohn-Mark Gurney 	if (aiocbe->uaiocb.aio_sigevent.sigev_notify != SIGEV_KEVENT)
1658cb679c38SJonathan Lemon 		goto no_kqueue;
1659fde80935SDavid Xu 	evflags = aiocbe->uaiocb.aio_sigevent.sigev_notify_kevent_flags;
1660fde80935SDavid Xu 	if ((evflags & ~(EV_CLEAR | EV_DISPATCH | EV_ONESHOT)) != 0) {
1661fde80935SDavid Xu 		error = EINVAL;
1662fde80935SDavid Xu 		goto aqueue_fail;
1663fde80935SDavid Xu 	}
16644db71d27SJohn-Mark Gurney 	kqfd = aiocbe->uaiocb.aio_sigevent.sigev_notify_kqueue;
1665b46f1c55SAlan Cox 	kev.ident = (uintptr_t)aiocbe->uuaiocb;
1666cb679c38SJonathan Lemon 	kev.filter = EVFILT_AIO;
1667fde80935SDavid Xu 	kev.flags = EV_ADD | EV_ENABLE | EV_FLAG1 | evflags;
1668b46f1c55SAlan Cox 	kev.data = (intptr_t)aiocbe;
16691ce91824SDavid Xu 	kev.udata = aiocbe->uaiocb.aio_sigevent.sigev_value.sival_ptr;
16704db71d27SJohn-Mark Gurney 	error = kqfd_register(kqfd, &kev, td, 1);
1671cb679c38SJonathan Lemon aqueue_fail:
1672cb679c38SJonathan Lemon 	if (error) {
1673a5c0b1c0SAlan Cox 		fdrop(fp, td);
1674c897b813SJeff Roberson 		uma_zfree(aiocb_zone, aiocbe);
16753858a1f4SJohn Baldwin 		ops->store_error(job, error);
1676279d7226SMatthew Dillon 		goto done;
1677cb679c38SJonathan Lemon 	}
1678cb679c38SJonathan Lemon no_kqueue:
1679cb679c38SJonathan Lemon 
16803858a1f4SJohn Baldwin 	ops->store_error(job, EINPROGRESS);
1681fd3bf775SJohn Dyson 	aiocbe->uaiocb._aiocb_private.error = EINPROGRESS;
16822244ea07SJohn Dyson 	aiocbe->userproc = p;
1683f8f750c5SRobert Watson 	aiocbe->cred = crhold(td->td_ucred);
16842244ea07SJohn Dyson 	aiocbe->jobflags = 0;
168584af4da6SJohn Dyson 	aiocbe->lio = lj;
16862244ea07SJohn Dyson 
168799eee864SDavid Xu 	if (opcode == LIO_SYNC)
168899eee864SDavid Xu 		goto queueit;
168999eee864SDavid Xu 
1690bfbbc4aaSJason Evans 	if (fp->f_type == DTYPE_SOCKET) {
1691bfbbc4aaSJason Evans 		/*
1692bfbbc4aaSJason Evans 		 * Alternate queueing for socket ops: Reach down into the
1693bfbbc4aaSJason Evans 		 * descriptor to get the socket data.  Then check to see if the
1694bfbbc4aaSJason Evans 		 * socket is ready to be read or written (based on the requested
1695bfbbc4aaSJason Evans 		 * operation).
1696bfbbc4aaSJason Evans 		 *
1697bfbbc4aaSJason Evans 		 * If it is not ready for io, then queue the aiocbe on the
1698bfbbc4aaSJason Evans 		 * socket, and set the flags so we get a call when sbnotify()
1699bfbbc4aaSJason Evans 		 * happens.
1700576c004fSAlfred Perlstein 		 *
1701576c004fSAlfred Perlstein 		 * Note if opcode is neither LIO_WRITE nor LIO_READ we lock
1702576c004fSAlfred Perlstein 		 * and unlock the snd sockbuf for no reason.
1703bfbbc4aaSJason Evans 		 */
170448e3128bSMatthew Dillon 		so = fp->f_data;
1705576c004fSAlfred Perlstein 		sb = (opcode == LIO_READ) ? &so->so_rcv : &so->so_snd;
1706576c004fSAlfred Perlstein 		SOCKBUF_LOCK(sb);
1707bfbbc4aaSJason Evans 		if (((opcode == LIO_READ) && (!soreadable(so))) || ((opcode ==
1708bfbbc4aaSJason Evans 		    LIO_WRITE) && (!sowriteable(so)))) {
1709576c004fSAlfred Perlstein 			sb->sb_flags |= SB_AIO;
17101aa4c324SDavid Xu 
17111aa4c324SDavid Xu 			mtx_lock(&aio_job_mtx);
17121aa4c324SDavid Xu 			TAILQ_INSERT_TAIL(&so->so_aiojobq, aiocbe, list);
17131aa4c324SDavid Xu 			mtx_unlock(&aio_job_mtx);
17141aa4c324SDavid Xu 
1715759ccccaSDavid Xu 			AIO_LOCK(ki);
17161ce91824SDavid Xu 			TAILQ_INSERT_TAIL(&ki->kaio_all, aiocbe, allist);
17171aa4c324SDavid Xu 			TAILQ_INSERT_TAIL(&ki->kaio_jobqueue, aiocbe, plist);
17181ce91824SDavid Xu 			aiocbe->jobstate = JOBST_JOBQSOCK;
17191ce91824SDavid Xu 			ki->kaio_count++;
17201ce91824SDavid Xu 			if (lj)
17211ce91824SDavid Xu 				lj->lioj_count++;
1722759ccccaSDavid Xu 			AIO_UNLOCK(ki);
1723576c004fSAlfred Perlstein 			SOCKBUF_UNLOCK(sb);
17241ce91824SDavid Xu 			atomic_add_int(&num_queue_count, 1);
1725279d7226SMatthew Dillon 			error = 0;
1726279d7226SMatthew Dillon 			goto done;
1727bfbbc4aaSJason Evans 		}
1728576c004fSAlfred Perlstein 		SOCKBUF_UNLOCK(sb);
1729bfbbc4aaSJason Evans 	}
1730bfbbc4aaSJason Evans 
1731bfbbc4aaSJason Evans 	if ((error = aio_qphysio(p, aiocbe)) == 0)
1732279d7226SMatthew Dillon 		goto done;
17331ce91824SDavid Xu #if 0
1734279d7226SMatthew Dillon 	if (error > 0) {
1735fd3bf775SJohn Dyson 		aiocbe->uaiocb._aiocb_private.error = error;
17363858a1f4SJohn Baldwin 		ops->store_error(job, error);
1737279d7226SMatthew Dillon 		goto done;
1738fd3bf775SJohn Dyson 	}
17391ce91824SDavid Xu #endif
174099eee864SDavid Xu queueit:
1741bfbbc4aaSJason Evans 	/* No buffer for daemon I/O. */
174284af4da6SJohn Dyson 	aiocbe->bp = NULL;
174399eee864SDavid Xu 	atomic_add_int(&num_queue_count, 1);
174484af4da6SJohn Dyson 
1745759ccccaSDavid Xu 	AIO_LOCK(ki);
17461ce91824SDavid Xu 	ki->kaio_count++;
1747bfbbc4aaSJason Evans 	if (lj)
17481ce91824SDavid Xu 		lj->lioj_count++;
1749fd3bf775SJohn Dyson 	TAILQ_INSERT_TAIL(&ki->kaio_jobqueue, aiocbe, plist);
17501ce91824SDavid Xu 	TAILQ_INSERT_TAIL(&ki->kaio_all, aiocbe, allist);
175199eee864SDavid Xu 	if (opcode == LIO_SYNC) {
1752dbbccfe9SDavid Xu 		TAILQ_FOREACH(cb, &ki->kaio_jobqueue, plist) {
1753dbbccfe9SDavid Xu 			if (cb->fd_file == aiocbe->fd_file &&
1754dbbccfe9SDavid Xu 			    cb->uaiocb.aio_lio_opcode != LIO_SYNC &&
1755dbbccfe9SDavid Xu 			    cb->seqno < aiocbe->seqno) {
1756dbbccfe9SDavid Xu 				cb->jobflags |= AIOCBLIST_CHECKSYNC;
1757dbbccfe9SDavid Xu 				aiocbe->pending++;
1758dbbccfe9SDavid Xu 			}
1759dbbccfe9SDavid Xu 		}
1760dbbccfe9SDavid Xu 		TAILQ_FOREACH(cb, &ki->kaio_bufqueue, plist) {
1761dbbccfe9SDavid Xu 			if (cb->fd_file == aiocbe->fd_file &&
1762dbbccfe9SDavid Xu 			    cb->uaiocb.aio_lio_opcode != LIO_SYNC &&
1763dbbccfe9SDavid Xu 			    cb->seqno < aiocbe->seqno) {
1764dbbccfe9SDavid Xu 				cb->jobflags |= AIOCBLIST_CHECKSYNC;
1765dbbccfe9SDavid Xu 				aiocbe->pending++;
1766dbbccfe9SDavid Xu 			}
1767dbbccfe9SDavid Xu 		}
1768dbbccfe9SDavid Xu 		if (aiocbe->pending != 0) {
176999eee864SDavid Xu 			TAILQ_INSERT_TAIL(&ki->kaio_syncqueue, aiocbe, list);
177099eee864SDavid Xu 			aiocbe->jobstate = JOBST_JOBQSYNC;
1771759ccccaSDavid Xu 			AIO_UNLOCK(ki);
1772dbbccfe9SDavid Xu 			goto done;
1773dbbccfe9SDavid Xu 		}
1774dbbccfe9SDavid Xu 	}
17751ce91824SDavid Xu 	mtx_lock(&aio_job_mtx);
17761ce91824SDavid Xu 	TAILQ_INSERT_TAIL(&aio_jobs, aiocbe, list);
17771ce91824SDavid Xu 	aiocbe->jobstate = JOBST_JOBQGLOBAL;
177899eee864SDavid Xu 	aio_kick_nowait(p);
177999eee864SDavid Xu 	mtx_unlock(&aio_job_mtx);
1780759ccccaSDavid Xu 	AIO_UNLOCK(ki);
17811ce91824SDavid Xu 	error = 0;
178299eee864SDavid Xu done:
178399eee864SDavid Xu 	return (error);
178499eee864SDavid Xu }
178599eee864SDavid Xu 
178699eee864SDavid Xu static void
178799eee864SDavid Xu aio_kick_nowait(struct proc *userp)
178899eee864SDavid Xu {
178999eee864SDavid Xu 	struct kaioinfo *ki = userp->p_aioinfo;
179099eee864SDavid Xu 	struct aiothreadlist *aiop;
179199eee864SDavid Xu 
179299eee864SDavid Xu 	mtx_assert(&aio_job_mtx, MA_OWNED);
179399eee864SDavid Xu 	if ((aiop = TAILQ_FIRST(&aio_freeproc)) != NULL) {
179499eee864SDavid Xu 		TAILQ_REMOVE(&aio_freeproc, aiop, list);
179599eee864SDavid Xu 		aiop->aiothreadflags &= ~AIOP_FREE;
179699eee864SDavid Xu 		wakeup(aiop->aiothread);
1797dbbccfe9SDavid Xu 	} else if (((num_aio_resv_start + num_aio_procs) < max_aio_procs) &&
1798dbbccfe9SDavid Xu 	    ((ki->kaio_active_count + num_aio_resv_start) <
1799dbbccfe9SDavid Xu 	    ki->kaio_maxactive_count)) {
180099eee864SDavid Xu 		taskqueue_enqueue(taskqueue_aiod_bio, &ki->kaio_task);
180199eee864SDavid Xu 	}
180299eee864SDavid Xu }
180399eee864SDavid Xu 
1804dbbccfe9SDavid Xu static int
180599eee864SDavid Xu aio_kick(struct proc *userp)
180699eee864SDavid Xu {
180799eee864SDavid Xu 	struct kaioinfo *ki = userp->p_aioinfo;
180899eee864SDavid Xu 	struct aiothreadlist *aiop;
1809dbbccfe9SDavid Xu 	int error, ret = 0;
181099eee864SDavid Xu 
181199eee864SDavid Xu 	mtx_assert(&aio_job_mtx, MA_OWNED);
181299eee864SDavid Xu retryproc:
1813d254af07SMatthew Dillon 	if ((aiop = TAILQ_FIRST(&aio_freeproc)) != NULL) {
18142244ea07SJohn Dyson 		TAILQ_REMOVE(&aio_freeproc, aiop, list);
1815b40ce416SJulian Elischer 		aiop->aiothreadflags &= ~AIOP_FREE;
1816b40ce416SJulian Elischer 		wakeup(aiop->aiothread);
1817fd3bf775SJohn Dyson 	} else if (((num_aio_resv_start + num_aio_procs) < max_aio_procs) &&
1818fd3bf775SJohn Dyson 	    ((ki->kaio_active_count + num_aio_resv_start) <
1819fd3bf775SJohn Dyson 	    ki->kaio_maxactive_count)) {
1820fd3bf775SJohn Dyson 		num_aio_resv_start++;
18211ce91824SDavid Xu 		mtx_unlock(&aio_job_mtx);
18221ce91824SDavid Xu 		error = aio_newproc(&num_aio_resv_start);
18231ce91824SDavid Xu 		mtx_lock(&aio_job_mtx);
18241ce91824SDavid Xu 		if (error) {
182584af4da6SJohn Dyson 			num_aio_resv_start--;
18262244ea07SJohn Dyson 			goto retryproc;
1827fd3bf775SJohn Dyson 		}
1828dbbccfe9SDavid Xu 	} else {
1829dbbccfe9SDavid Xu 		ret = -1;
18301ce91824SDavid Xu 	}
1831dbbccfe9SDavid Xu 	return (ret);
183299eee864SDavid Xu }
18331ce91824SDavid Xu 
183499eee864SDavid Xu static void
183599eee864SDavid Xu aio_kick_helper(void *context, int pending)
183699eee864SDavid Xu {
183799eee864SDavid Xu 	struct proc *userp = context;
183899eee864SDavid Xu 
183999eee864SDavid Xu 	mtx_lock(&aio_job_mtx);
1840dbbccfe9SDavid Xu 	while (--pending >= 0) {
1841dbbccfe9SDavid Xu 		if (aio_kick(userp))
1842dbbccfe9SDavid Xu 			break;
1843dbbccfe9SDavid Xu 	}
184499eee864SDavid Xu 	mtx_unlock(&aio_job_mtx);
18452244ea07SJohn Dyson }
18462244ea07SJohn Dyson 
1847fd3bf775SJohn Dyson /*
1848bfbbc4aaSJason Evans  * Support the aio_return system call, as a side-effect, kernel resources are
1849bfbbc4aaSJason Evans  * released.
18502244ea07SJohn Dyson  */
18513858a1f4SJohn Baldwin static int
18523858a1f4SJohn Baldwin kern_aio_return(struct thread *td, struct aiocb *uaiocb, struct aiocb_ops *ops)
1853fd3bf775SJohn Dyson {
1854b40ce416SJulian Elischer 	struct proc *p = td->td_proc;
18551ce91824SDavid Xu 	struct aiocblist *cb;
18562244ea07SJohn Dyson 	struct kaioinfo *ki;
18571ce91824SDavid Xu 	int status, error;
18582244ea07SJohn Dyson 
1859c0bf5caaSAlan Cox 	ki = p->p_aioinfo;
1860c0bf5caaSAlan Cox 	if (ki == NULL)
1861ac41f2efSAlfred Perlstein 		return (EINVAL);
1862759ccccaSDavid Xu 	AIO_LOCK(ki);
18631ce91824SDavid Xu 	TAILQ_FOREACH(cb, &ki->kaio_done, plist) {
18641ce91824SDavid Xu 		if (cb->uuaiocb == uaiocb)
1865c0bf5caaSAlan Cox 			break;
1866c0bf5caaSAlan Cox 	}
1867c0bf5caaSAlan Cox 	if (cb != NULL) {
18681ce91824SDavid Xu 		MPASS(cb->jobstate == JOBST_JOBFINISHED);
18691ce91824SDavid Xu 		status = cb->uaiocb._aiocb_private.status;
18701ce91824SDavid Xu 		error = cb->uaiocb._aiocb_private.error;
18711ce91824SDavid Xu 		td->td_retval[0] = status;
187269cd28daSDoug Ambrisko 		if (cb->uaiocb.aio_lio_opcode == LIO_WRITE) {
18731c4bcd05SJeff Roberson 			td->td_ru.ru_oublock += cb->outputcharge;
187469cd28daSDoug Ambrisko 			cb->outputcharge = 0;
187569cd28daSDoug Ambrisko 		} else if (cb->uaiocb.aio_lio_opcode == LIO_READ) {
18761c4bcd05SJeff Roberson 			td->td_ru.ru_inblock += cb->inputcharge;
187769cd28daSDoug Ambrisko 			cb->inputcharge = 0;
187869cd28daSDoug Ambrisko 		}
187984af4da6SJohn Dyson 		aio_free_entry(cb);
1880759ccccaSDavid Xu 		AIO_UNLOCK(ki);
18813858a1f4SJohn Baldwin 		ops->store_error(uaiocb, error);
18823858a1f4SJohn Baldwin 		ops->store_status(uaiocb, status);
188355a122bfSDavid Xu 	} else {
18841ce91824SDavid Xu 		error = EINVAL;
1885759ccccaSDavid Xu 		AIO_UNLOCK(ki);
188655a122bfSDavid Xu 	}
18871ce91824SDavid Xu 	return (error);
18882244ea07SJohn Dyson }
18892244ea07SJohn Dyson 
18903858a1f4SJohn Baldwin int
18918451d0ddSKip Macy sys_aio_return(struct thread *td, struct aio_return_args *uap)
18923858a1f4SJohn Baldwin {
18933858a1f4SJohn Baldwin 
18943858a1f4SJohn Baldwin 	return (kern_aio_return(td, uap->aiocbp, &aiocb_ops));
18953858a1f4SJohn Baldwin }
18963858a1f4SJohn Baldwin 
18972244ea07SJohn Dyson /*
1898bfbbc4aaSJason Evans  * Allow a process to wakeup when any of the I/O requests are completed.
18992244ea07SJohn Dyson  */
19003858a1f4SJohn Baldwin static int
19013858a1f4SJohn Baldwin kern_aio_suspend(struct thread *td, int njoblist, struct aiocb **ujoblist,
19023858a1f4SJohn Baldwin     struct timespec *ts)
1903fd3bf775SJohn Dyson {
1904b40ce416SJulian Elischer 	struct proc *p = td->td_proc;
19054a11ca4eSPoul-Henning Kamp 	struct timeval atv;
19062244ea07SJohn Dyson 	struct kaioinfo *ki;
19071ce91824SDavid Xu 	struct aiocblist *cb, *cbfirst;
19083858a1f4SJohn Baldwin 	int error, i, timo;
19092244ea07SJohn Dyson 
19102244ea07SJohn Dyson 	timo = 0;
19113858a1f4SJohn Baldwin 	if (ts) {
19123858a1f4SJohn Baldwin 		if (ts->tv_nsec < 0 || ts->tv_nsec >= 1000000000)
19132244ea07SJohn Dyson 			return (EINVAL);
19142244ea07SJohn Dyson 
19153858a1f4SJohn Baldwin 		TIMESPEC_TO_TIMEVAL(&atv, ts);
19162244ea07SJohn Dyson 		if (itimerfix(&atv))
19172244ea07SJohn Dyson 			return (EINVAL);
1918227ee8a1SPoul-Henning Kamp 		timo = tvtohz(&atv);
19192244ea07SJohn Dyson 	}
19202244ea07SJohn Dyson 
19212244ea07SJohn Dyson 	ki = p->p_aioinfo;
19222244ea07SJohn Dyson 	if (ki == NULL)
1923ac41f2efSAlfred Perlstein 		return (EAGAIN);
19242244ea07SJohn Dyson 
19253858a1f4SJohn Baldwin 	if (njoblist == 0)
1926ac41f2efSAlfred Perlstein 		return (0);
19272244ea07SJohn Dyson 
1928759ccccaSDavid Xu 	AIO_LOCK(ki);
19291ce91824SDavid Xu 	for (;;) {
19301ce91824SDavid Xu 		cbfirst = NULL;
19311ce91824SDavid Xu 		error = 0;
19321ce91824SDavid Xu 		TAILQ_FOREACH(cb, &ki->kaio_all, allist) {
193384af4da6SJohn Dyson 			for (i = 0; i < njoblist; i++) {
19341ce91824SDavid Xu 				if (cb->uuaiocb == ujoblist[i]) {
19351ce91824SDavid Xu 					if (cbfirst == NULL)
19361ce91824SDavid Xu 						cbfirst = cb;
19371ce91824SDavid Xu 					if (cb->jobstate == JOBST_JOBFINISHED)
19381ce91824SDavid Xu 						goto RETURN;
193984af4da6SJohn Dyson 				}
194084af4da6SJohn Dyson 			}
194184af4da6SJohn Dyson 		}
19421ce91824SDavid Xu 		/* All tasks were finished. */
19431ce91824SDavid Xu 		if (cbfirst == NULL)
19441ce91824SDavid Xu 			break;
19452244ea07SJohn Dyson 
1946fd3bf775SJohn Dyson 		ki->kaio_flags |= KAIO_WAKEUP;
1947759ccccaSDavid Xu 		error = msleep(&p->p_aioinfo, AIO_MTX(ki), PRIBIO | PCATCH,
19481ce91824SDavid Xu 		    "aiospn", timo);
19491ce91824SDavid Xu 		if (error == ERESTART)
19501ce91824SDavid Xu 			error = EINTR;
19511ce91824SDavid Xu 		if (error)
19521ce91824SDavid Xu 			break;
19532244ea07SJohn Dyson 	}
19541ce91824SDavid Xu RETURN:
1955759ccccaSDavid Xu 	AIO_UNLOCK(ki);
19563858a1f4SJohn Baldwin 	return (error);
19573858a1f4SJohn Baldwin }
19583858a1f4SJohn Baldwin 
19593858a1f4SJohn Baldwin int
19608451d0ddSKip Macy sys_aio_suspend(struct thread *td, struct aio_suspend_args *uap)
19613858a1f4SJohn Baldwin {
19623858a1f4SJohn Baldwin 	struct timespec ts, *tsp;
19633858a1f4SJohn Baldwin 	struct aiocb **ujoblist;
19643858a1f4SJohn Baldwin 	int error;
19653858a1f4SJohn Baldwin 
19663858a1f4SJohn Baldwin 	if (uap->nent < 0 || uap->nent > AIO_LISTIO_MAX)
19673858a1f4SJohn Baldwin 		return (EINVAL);
19683858a1f4SJohn Baldwin 
19693858a1f4SJohn Baldwin 	if (uap->timeout) {
19703858a1f4SJohn Baldwin 		/* Get timespec struct. */
19713858a1f4SJohn Baldwin 		if ((error = copyin(uap->timeout, &ts, sizeof(ts))) != 0)
19723858a1f4SJohn Baldwin 			return (error);
19733858a1f4SJohn Baldwin 		tsp = &ts;
19743858a1f4SJohn Baldwin 	} else
19753858a1f4SJohn Baldwin 		tsp = NULL;
19763858a1f4SJohn Baldwin 
19773858a1f4SJohn Baldwin 	ujoblist = uma_zalloc(aiol_zone, M_WAITOK);
19783858a1f4SJohn Baldwin 	error = copyin(uap->aiocbp, ujoblist, uap->nent * sizeof(ujoblist[0]));
19793858a1f4SJohn Baldwin 	if (error == 0)
19803858a1f4SJohn Baldwin 		error = kern_aio_suspend(td, uap->nent, ujoblist, tsp);
19811ce91824SDavid Xu 	uma_zfree(aiol_zone, ujoblist);
19821ce91824SDavid Xu 	return (error);
19832244ea07SJohn Dyson }
1984ee877a35SJohn Dyson 
1985ee877a35SJohn Dyson /*
1986dd85920aSJason Evans  * aio_cancel cancels any non-physio aio operations not currently in
1987dd85920aSJason Evans  * progress.
1988ee877a35SJohn Dyson  */
1989ee877a35SJohn Dyson int
19908451d0ddSKip Macy sys_aio_cancel(struct thread *td, struct aio_cancel_args *uap)
1991fd3bf775SJohn Dyson {
1992b40ce416SJulian Elischer 	struct proc *p = td->td_proc;
1993dd85920aSJason Evans 	struct kaioinfo *ki;
1994dd85920aSJason Evans 	struct aiocblist *cbe, *cbn;
1995dd85920aSJason Evans 	struct file *fp;
1996dd85920aSJason Evans 	struct socket *so;
19971ce91824SDavid Xu 	int error;
19981aa4c324SDavid Xu 	int remove;
1999dd85920aSJason Evans 	int cancelled = 0;
2000dd85920aSJason Evans 	int notcancelled = 0;
2001dd85920aSJason Evans 	struct vnode *vp;
2002dd85920aSJason Evans 
20032a522eb9SJohn Baldwin 	/* Lookup file object. */
2004a9d2f8d8SRobert Watson 	error = fget(td, uap->fd, 0, &fp);
20052a522eb9SJohn Baldwin 	if (error)
20062a522eb9SJohn Baldwin 		return (error);
2007dd85920aSJason Evans 
20081ce91824SDavid Xu 	ki = p->p_aioinfo;
20091ce91824SDavid Xu 	if (ki == NULL)
20101ce91824SDavid Xu 		goto done;
20111ce91824SDavid Xu 
2012dd85920aSJason Evans 	if (fp->f_type == DTYPE_VNODE) {
20133b6d9652SPoul-Henning Kamp 		vp = fp->f_vnode;
2014dd85920aSJason Evans 		if (vn_isdisk(vp, &error)) {
20152a522eb9SJohn Baldwin 			fdrop(fp, td);
2016b40ce416SJulian Elischer 			td->td_retval[0] = AIO_NOTCANCELED;
2017ac41f2efSAlfred Perlstein 			return (0);
2018dd85920aSJason Evans 		}
2019dd85920aSJason Evans 	}
2020dd85920aSJason Evans 
2021759ccccaSDavid Xu 	AIO_LOCK(ki);
20222a522eb9SJohn Baldwin 	TAILQ_FOREACH_SAFE(cbe, &ki->kaio_jobqueue, plist, cbn) {
2023dd85920aSJason Evans 		if ((uap->fd == cbe->uaiocb.aio_fildes) &&
2024dd85920aSJason Evans 		    ((uap->aiocbp == NULL) ||
2025dd85920aSJason Evans 		     (uap->aiocbp == cbe->uuaiocb))) {
20261aa4c324SDavid Xu 			remove = 0;
20271aa4c324SDavid Xu 
20281ce91824SDavid Xu 			mtx_lock(&aio_job_mtx);
2029dd85920aSJason Evans 			if (cbe->jobstate == JOBST_JOBQGLOBAL) {
2030dd85920aSJason Evans 				TAILQ_REMOVE(&aio_jobs, cbe, list);
20311aa4c324SDavid Xu 				remove = 1;
20321aa4c324SDavid Xu 			} else if (cbe->jobstate == JOBST_JOBQSOCK) {
20331aa4c324SDavid Xu 				MPASS(fp->f_type == DTYPE_SOCKET);
20341aa4c324SDavid Xu 				so = fp->f_data;
20351aa4c324SDavid Xu 				TAILQ_REMOVE(&so->so_aiojobq, cbe, list);
20361aa4c324SDavid Xu 				remove = 1;
203799eee864SDavid Xu 			} else if (cbe->jobstate == JOBST_JOBQSYNC) {
203899eee864SDavid Xu 				TAILQ_REMOVE(&ki->kaio_syncqueue, cbe, list);
203999eee864SDavid Xu 				remove = 1;
20401aa4c324SDavid Xu 			}
20411ce91824SDavid Xu 			mtx_unlock(&aio_job_mtx);
20421aa4c324SDavid Xu 
20431aa4c324SDavid Xu 			if (remove) {
20441ce91824SDavid Xu 				TAILQ_REMOVE(&ki->kaio_jobqueue, cbe, plist);
2045dd85920aSJason Evans 				cbe->uaiocb._aiocb_private.status = -1;
2046dd85920aSJason Evans 				cbe->uaiocb._aiocb_private.error = ECANCELED;
20471ce91824SDavid Xu 				aio_bio_done_notify(p, cbe, DONE_QUEUE);
20481ce91824SDavid Xu 				cancelled++;
2049dd85920aSJason Evans 			} else {
2050dd85920aSJason Evans 				notcancelled++;
2051dd85920aSJason Evans 			}
20521aa4c324SDavid Xu 			if (uap->aiocbp != NULL)
20531aa4c324SDavid Xu 				break;
2054dd85920aSJason Evans 		}
2055dd85920aSJason Evans 	}
2056759ccccaSDavid Xu 	AIO_UNLOCK(ki);
20571ce91824SDavid Xu 
2058ad49abc0SAlan Cox done:
20592a522eb9SJohn Baldwin 	fdrop(fp, td);
20601aa4c324SDavid Xu 
20611aa4c324SDavid Xu 	if (uap->aiocbp != NULL) {
2062dd85920aSJason Evans 		if (cancelled) {
2063b40ce416SJulian Elischer 			td->td_retval[0] = AIO_CANCELED;
2064ac41f2efSAlfred Perlstein 			return (0);
2065dd85920aSJason Evans 		}
20661aa4c324SDavid Xu 	}
20671aa4c324SDavid Xu 
20681aa4c324SDavid Xu 	if (notcancelled) {
20691aa4c324SDavid Xu 		td->td_retval[0] = AIO_NOTCANCELED;
20701aa4c324SDavid Xu 		return (0);
20711aa4c324SDavid Xu 	}
20721aa4c324SDavid Xu 
20731aa4c324SDavid Xu 	if (cancelled) {
20741aa4c324SDavid Xu 		td->td_retval[0] = AIO_CANCELED;
20751aa4c324SDavid Xu 		return (0);
20761aa4c324SDavid Xu 	}
20771aa4c324SDavid Xu 
2078b40ce416SJulian Elischer 	td->td_retval[0] = AIO_ALLDONE;
2079dd85920aSJason Evans 
2080ac41f2efSAlfred Perlstein 	return (0);
2081ee877a35SJohn Dyson }
2082ee877a35SJohn Dyson 
2083ee877a35SJohn Dyson /*
2084873fbcd7SRobert Watson  * aio_error is implemented in the kernel level for compatibility purposes
2085873fbcd7SRobert Watson  * only.  For a user mode async implementation, it would be best to do it in
2086873fbcd7SRobert Watson  * a userland subroutine.
2087ee877a35SJohn Dyson  */
20883858a1f4SJohn Baldwin static int
20893858a1f4SJohn Baldwin kern_aio_error(struct thread *td, struct aiocb *aiocbp, struct aiocb_ops *ops)
2090fd3bf775SJohn Dyson {
2091b40ce416SJulian Elischer 	struct proc *p = td->td_proc;
20922244ea07SJohn Dyson 	struct aiocblist *cb;
20932244ea07SJohn Dyson 	struct kaioinfo *ki;
20941ce91824SDavid Xu 	int status;
2095ee877a35SJohn Dyson 
20962244ea07SJohn Dyson 	ki = p->p_aioinfo;
20971ce91824SDavid Xu 	if (ki == NULL) {
20981ce91824SDavid Xu 		td->td_retval[0] = EINVAL;
20991ce91824SDavid Xu 		return (0);
21001ce91824SDavid Xu 	}
2101ee877a35SJohn Dyson 
2102759ccccaSDavid Xu 	AIO_LOCK(ki);
21031ce91824SDavid Xu 	TAILQ_FOREACH(cb, &ki->kaio_all, allist) {
21043858a1f4SJohn Baldwin 		if (cb->uuaiocb == aiocbp) {
21051ce91824SDavid Xu 			if (cb->jobstate == JOBST_JOBFINISHED)
21061ce91824SDavid Xu 				td->td_retval[0] =
21071ce91824SDavid Xu 					cb->uaiocb._aiocb_private.error;
21081ce91824SDavid Xu 			else
2109b40ce416SJulian Elischer 				td->td_retval[0] = EINPROGRESS;
2110759ccccaSDavid Xu 			AIO_UNLOCK(ki);
2111ac41f2efSAlfred Perlstein 			return (0);
21122244ea07SJohn Dyson 		}
21132244ea07SJohn Dyson 	}
2114759ccccaSDavid Xu 	AIO_UNLOCK(ki);
211584af4da6SJohn Dyson 
21162244ea07SJohn Dyson 	/*
2117a9bf5e37SDavid Xu 	 * Hack for failure of aio_aqueue.
21182244ea07SJohn Dyson 	 */
21193858a1f4SJohn Baldwin 	status = ops->fetch_status(aiocbp);
21201ce91824SDavid Xu 	if (status == -1) {
21213858a1f4SJohn Baldwin 		td->td_retval[0] = ops->fetch_error(aiocbp);
21221ce91824SDavid Xu 		return (0);
21231ce91824SDavid Xu 	}
21241ce91824SDavid Xu 
21251ce91824SDavid Xu 	td->td_retval[0] = EINVAL;
21261ce91824SDavid Xu 	return (0);
2127ee877a35SJohn Dyson }
2128ee877a35SJohn Dyson 
21293858a1f4SJohn Baldwin int
21308451d0ddSKip Macy sys_aio_error(struct thread *td, struct aio_error_args *uap)
21313858a1f4SJohn Baldwin {
21323858a1f4SJohn Baldwin 
21333858a1f4SJohn Baldwin 	return (kern_aio_error(td, uap->aiocbp, &aiocb_ops));
21343858a1f4SJohn Baldwin }
21353858a1f4SJohn Baldwin 
2136eb8e6d52SEivind Eklund /* syscall - asynchronous read from a file (REALTIME) */
2137ee877a35SJohn Dyson int
21388451d0ddSKip Macy sys_oaio_read(struct thread *td, struct oaio_read_args *uap)
21390972628aSDavid Xu {
21400972628aSDavid Xu 
21413858a1f4SJohn Baldwin 	return (aio_aqueue(td, (struct aiocb *)uap->aiocbp, NULL, LIO_READ,
21423858a1f4SJohn Baldwin 	    &aiocb_ops_osigevent));
21430972628aSDavid Xu }
21440972628aSDavid Xu 
21450972628aSDavid Xu int
21468451d0ddSKip Macy sys_aio_read(struct thread *td, struct aio_read_args *uap)
2147fd3bf775SJohn Dyson {
214821d56e9cSAlfred Perlstein 
21493858a1f4SJohn Baldwin 	return (aio_aqueue(td, uap->aiocbp, NULL, LIO_READ, &aiocb_ops));
2150ee877a35SJohn Dyson }
2151ee877a35SJohn Dyson 
2152eb8e6d52SEivind Eklund /* syscall - asynchronous write to a file (REALTIME) */
2153ee877a35SJohn Dyson int
21548451d0ddSKip Macy sys_oaio_write(struct thread *td, struct oaio_write_args *uap)
21550972628aSDavid Xu {
21560972628aSDavid Xu 
21573858a1f4SJohn Baldwin 	return (aio_aqueue(td, (struct aiocb *)uap->aiocbp, NULL, LIO_WRITE,
21583858a1f4SJohn Baldwin 	    &aiocb_ops_osigevent));
21590972628aSDavid Xu }
21600972628aSDavid Xu 
21610972628aSDavid Xu int
21628451d0ddSKip Macy sys_aio_write(struct thread *td, struct aio_write_args *uap)
2163fd3bf775SJohn Dyson {
216421d56e9cSAlfred Perlstein 
21653858a1f4SJohn Baldwin 	return (aio_aqueue(td, uap->aiocbp, NULL, LIO_WRITE, &aiocb_ops));
21660972628aSDavid Xu }
21670972628aSDavid Xu 
21680972628aSDavid Xu static int
21693858a1f4SJohn Baldwin kern_lio_listio(struct thread *td, int mode, struct aiocb * const *uacb_list,
21703858a1f4SJohn Baldwin     struct aiocb **acb_list, int nent, struct sigevent *sig,
21713858a1f4SJohn Baldwin     struct aiocb_ops *ops)
21720972628aSDavid Xu {
2173b40ce416SJulian Elischer 	struct proc *p = td->td_proc;
21743858a1f4SJohn Baldwin 	struct aiocb *iocb;
21752244ea07SJohn Dyson 	struct kaioinfo *ki;
21761ce91824SDavid Xu 	struct aioliojob *lj;
217769cd28daSDoug Ambrisko 	struct kevent kev;
21781ce91824SDavid Xu 	int error;
2179fd3bf775SJohn Dyson 	int nerror;
2180ee877a35SJohn Dyson 	int i;
2181ee877a35SJohn Dyson 
21823858a1f4SJohn Baldwin 	if ((mode != LIO_NOWAIT) && (mode != LIO_WAIT))
2183ac41f2efSAlfred Perlstein 		return (EINVAL);
21842244ea07SJohn Dyson 
2185ae3b195fSTim J. Robbins 	if (nent < 0 || nent > AIO_LISTIO_MAX)
2186ac41f2efSAlfred Perlstein 		return (EINVAL);
21872244ea07SJohn Dyson 
2188bfbbc4aaSJason Evans 	if (p->p_aioinfo == NULL)
21892244ea07SJohn Dyson 		aio_init_aioinfo(p);
21902244ea07SJohn Dyson 
21912244ea07SJohn Dyson 	ki = p->p_aioinfo;
21922244ea07SJohn Dyson 
2193a163d034SWarner Losh 	lj = uma_zalloc(aiolio_zone, M_WAITOK);
219484af4da6SJohn Dyson 	lj->lioj_flags = 0;
21951ce91824SDavid Xu 	lj->lioj_count = 0;
21961ce91824SDavid Xu 	lj->lioj_finished_count = 0;
2197d8b0556cSKonstantin Belousov 	knlist_init_mtx(&lj->klist, AIO_MTX(ki));
21984c0fb2cfSDavid Xu 	ksiginfo_init(&lj->lioj_ksi);
219969cd28daSDoug Ambrisko 
220084af4da6SJohn Dyson 	/*
2201bfbbc4aaSJason Evans 	 * Setup signal.
220284af4da6SJohn Dyson 	 */
22033858a1f4SJohn Baldwin 	if (sig && (mode == LIO_NOWAIT)) {
22043858a1f4SJohn Baldwin 		bcopy(sig, &lj->lioj_signal, sizeof(lj->lioj_signal));
220569cd28daSDoug Ambrisko 		if (lj->lioj_signal.sigev_notify == SIGEV_KEVENT) {
220669cd28daSDoug Ambrisko 			/* Assume only new style KEVENT */
220769cd28daSDoug Ambrisko 			kev.filter = EVFILT_LIO;
220869cd28daSDoug Ambrisko 			kev.flags = EV_ADD | EV_ENABLE | EV_FLAG1;
22093858a1f4SJohn Baldwin 			kev.ident = (uintptr_t)uacb_list; /* something unique */
221069cd28daSDoug Ambrisko 			kev.data = (intptr_t)lj;
22111ce91824SDavid Xu 			/* pass user defined sigval data */
22121ce91824SDavid Xu 			kev.udata = lj->lioj_signal.sigev_value.sival_ptr;
22134db71d27SJohn-Mark Gurney 			error = kqfd_register(
22144db71d27SJohn-Mark Gurney 			    lj->lioj_signal.sigev_notify_kqueue, &kev, td, 1);
221569cd28daSDoug Ambrisko 			if (error) {
221669cd28daSDoug Ambrisko 				uma_zfree(aiolio_zone, lj);
221769cd28daSDoug Ambrisko 				return (error);
221869cd28daSDoug Ambrisko 			}
22191ce91824SDavid Xu 		} else if (lj->lioj_signal.sigev_notify == SIGEV_NONE) {
22201ce91824SDavid Xu 			;
222168d71118SDavid Xu 		} else if (lj->lioj_signal.sigev_notify == SIGEV_SIGNAL ||
222268d71118SDavid Xu 			   lj->lioj_signal.sigev_notify == SIGEV_THREAD_ID) {
222368d71118SDavid Xu 				if (!_SIG_VALID(lj->lioj_signal.sigev_signo)) {
222469cd28daSDoug Ambrisko 					uma_zfree(aiolio_zone, lj);
222569cd28daSDoug Ambrisko 					return EINVAL;
222668d71118SDavid Xu 				}
222784af4da6SJohn Dyson 				lj->lioj_flags |= LIOJ_SIGNAL;
222868d71118SDavid Xu 		} else {
222968d71118SDavid Xu 			uma_zfree(aiolio_zone, lj);
223068d71118SDavid Xu 			return EINVAL;
22314d752b01SAlan Cox 		}
22321ce91824SDavid Xu 	}
223369cd28daSDoug Ambrisko 
2234759ccccaSDavid Xu 	AIO_LOCK(ki);
22352f3cf918SAlfred Perlstein 	TAILQ_INSERT_TAIL(&ki->kaio_liojoblist, lj, lioj_list);
22362244ea07SJohn Dyson 	/*
22371ce91824SDavid Xu 	 * Add extra aiocb count to avoid the lio to be freed
22381ce91824SDavid Xu 	 * by other threads doing aio_waitcomplete or aio_return,
22391ce91824SDavid Xu 	 * and prevent event from being sent until we have queued
22401ce91824SDavid Xu 	 * all tasks.
22411ce91824SDavid Xu 	 */
22421ce91824SDavid Xu 	lj->lioj_count = 1;
2243759ccccaSDavid Xu 	AIO_UNLOCK(ki);
22441ce91824SDavid Xu 
22451ce91824SDavid Xu 	/*
2246bfbbc4aaSJason Evans 	 * Get pointers to the list of I/O requests.
22472244ea07SJohn Dyson 	 */
2248fd3bf775SJohn Dyson 	nerror = 0;
22493858a1f4SJohn Baldwin 	for (i = 0; i < nent; i++) {
22503858a1f4SJohn Baldwin 		iocb = acb_list[i];
22513858a1f4SJohn Baldwin 		if (iocb != NULL) {
22523858a1f4SJohn Baldwin 			error = aio_aqueue(td, iocb, lj, LIO_NOP, ops);
22531ce91824SDavid Xu 			if (error != 0)
2254fd3bf775SJohn Dyson 				nerror++;
2255fd3bf775SJohn Dyson 		}
2256fd3bf775SJohn Dyson 	}
22572244ea07SJohn Dyson 
22581ce91824SDavid Xu 	error = 0;
2259759ccccaSDavid Xu 	AIO_LOCK(ki);
22603858a1f4SJohn Baldwin 	if (mode == LIO_WAIT) {
22611ce91824SDavid Xu 		while (lj->lioj_count - 1 != lj->lioj_finished_count) {
2262fd3bf775SJohn Dyson 			ki->kaio_flags |= KAIO_WAKEUP;
2263759ccccaSDavid Xu 			error = msleep(&p->p_aioinfo, AIO_MTX(ki),
22641ce91824SDavid Xu 			    PRIBIO | PCATCH, "aiospn", 0);
22651ce91824SDavid Xu 			if (error == ERESTART)
22661ce91824SDavid Xu 				error = EINTR;
22671ce91824SDavid Xu 			if (error)
22681ce91824SDavid Xu 				break;
22691ce91824SDavid Xu 		}
22701ce91824SDavid Xu 	} else {
22711ce91824SDavid Xu 		if (lj->lioj_count - 1 == lj->lioj_finished_count) {
22721ce91824SDavid Xu 			if (lj->lioj_signal.sigev_notify == SIGEV_KEVENT) {
22731ce91824SDavid Xu 				lj->lioj_flags |= LIOJ_KEVENT_POSTED;
22741ce91824SDavid Xu 				KNOTE_LOCKED(&lj->klist, 1);
22751ce91824SDavid Xu 			}
22761ce91824SDavid Xu 			if ((lj->lioj_flags & (LIOJ_SIGNAL|LIOJ_SIGNAL_POSTED))
22771ce91824SDavid Xu 			    == LIOJ_SIGNAL
22781ce91824SDavid Xu 			    && (lj->lioj_signal.sigev_notify == SIGEV_SIGNAL ||
22791ce91824SDavid Xu 			    lj->lioj_signal.sigev_notify == SIGEV_THREAD_ID)) {
22801ce91824SDavid Xu 				aio_sendsig(p, &lj->lioj_signal,
22811ce91824SDavid Xu 					    &lj->lioj_ksi);
22821ce91824SDavid Xu 				lj->lioj_flags |= LIOJ_SIGNAL_POSTED;
22832244ea07SJohn Dyson 			}
22842244ea07SJohn Dyson 		}
22851ce91824SDavid Xu 	}
22861ce91824SDavid Xu 	lj->lioj_count--;
22871ce91824SDavid Xu 	if (lj->lioj_count == 0) {
22881ce91824SDavid Xu 		TAILQ_REMOVE(&ki->kaio_liojoblist, lj, lioj_list);
22891ce91824SDavid Xu 		knlist_delete(&lj->klist, curthread, 1);
2290759ccccaSDavid Xu 		PROC_LOCK(p);
22911ce91824SDavid Xu 		sigqueue_take(&lj->lioj_ksi);
22921ce91824SDavid Xu 		PROC_UNLOCK(p);
2293759ccccaSDavid Xu 		AIO_UNLOCK(ki);
22941ce91824SDavid Xu 		uma_zfree(aiolio_zone, lj);
22951ce91824SDavid Xu 	} else
2296759ccccaSDavid Xu 		AIO_UNLOCK(ki);
22972244ea07SJohn Dyson 
22981ce91824SDavid Xu 	if (nerror)
22991ce91824SDavid Xu 		return (EIO);
23001ce91824SDavid Xu 	return (error);
2301ee877a35SJohn Dyson }
2302fd3bf775SJohn Dyson 
23033858a1f4SJohn Baldwin /* syscall - list directed I/O (REALTIME) */
23043858a1f4SJohn Baldwin int
23058451d0ddSKip Macy sys_olio_listio(struct thread *td, struct olio_listio_args *uap)
23063858a1f4SJohn Baldwin {
23073858a1f4SJohn Baldwin 	struct aiocb **acb_list;
23083858a1f4SJohn Baldwin 	struct sigevent *sigp, sig;
23093858a1f4SJohn Baldwin 	struct osigevent osig;
23103858a1f4SJohn Baldwin 	int error, nent;
23113858a1f4SJohn Baldwin 
23123858a1f4SJohn Baldwin 	if ((uap->mode != LIO_NOWAIT) && (uap->mode != LIO_WAIT))
23133858a1f4SJohn Baldwin 		return (EINVAL);
23143858a1f4SJohn Baldwin 
23153858a1f4SJohn Baldwin 	nent = uap->nent;
23163858a1f4SJohn Baldwin 	if (nent < 0 || nent > AIO_LISTIO_MAX)
23173858a1f4SJohn Baldwin 		return (EINVAL);
23183858a1f4SJohn Baldwin 
23193858a1f4SJohn Baldwin 	if (uap->sig && (uap->mode == LIO_NOWAIT)) {
23203858a1f4SJohn Baldwin 		error = copyin(uap->sig, &osig, sizeof(osig));
23213858a1f4SJohn Baldwin 		if (error)
23223858a1f4SJohn Baldwin 			return (error);
23233858a1f4SJohn Baldwin 		error = convert_old_sigevent(&osig, &sig);
23243858a1f4SJohn Baldwin 		if (error)
23253858a1f4SJohn Baldwin 			return (error);
23263858a1f4SJohn Baldwin 		sigp = &sig;
23273858a1f4SJohn Baldwin 	} else
23283858a1f4SJohn Baldwin 		sigp = NULL;
23293858a1f4SJohn Baldwin 
23303858a1f4SJohn Baldwin 	acb_list = malloc(sizeof(struct aiocb *) * nent, M_LIO, M_WAITOK);
23313858a1f4SJohn Baldwin 	error = copyin(uap->acb_list, acb_list, nent * sizeof(acb_list[0]));
23323858a1f4SJohn Baldwin 	if (error == 0)
23333858a1f4SJohn Baldwin 		error = kern_lio_listio(td, uap->mode,
23343858a1f4SJohn Baldwin 		    (struct aiocb * const *)uap->acb_list, acb_list, nent, sigp,
23353858a1f4SJohn Baldwin 		    &aiocb_ops_osigevent);
23363858a1f4SJohn Baldwin 	free(acb_list, M_LIO);
23373858a1f4SJohn Baldwin 	return (error);
23383858a1f4SJohn Baldwin }
23393858a1f4SJohn Baldwin 
23403858a1f4SJohn Baldwin /* syscall - list directed I/O (REALTIME) */
23413858a1f4SJohn Baldwin int
23428451d0ddSKip Macy sys_lio_listio(struct thread *td, struct lio_listio_args *uap)
23433858a1f4SJohn Baldwin {
23443858a1f4SJohn Baldwin 	struct aiocb **acb_list;
23453858a1f4SJohn Baldwin 	struct sigevent *sigp, sig;
23463858a1f4SJohn Baldwin 	int error, nent;
23473858a1f4SJohn Baldwin 
23483858a1f4SJohn Baldwin 	if ((uap->mode != LIO_NOWAIT) && (uap->mode != LIO_WAIT))
23493858a1f4SJohn Baldwin 		return (EINVAL);
23503858a1f4SJohn Baldwin 
23513858a1f4SJohn Baldwin 	nent = uap->nent;
23523858a1f4SJohn Baldwin 	if (nent < 0 || nent > AIO_LISTIO_MAX)
23533858a1f4SJohn Baldwin 		return (EINVAL);
23543858a1f4SJohn Baldwin 
23553858a1f4SJohn Baldwin 	if (uap->sig && (uap->mode == LIO_NOWAIT)) {
23563858a1f4SJohn Baldwin 		error = copyin(uap->sig, &sig, sizeof(sig));
23573858a1f4SJohn Baldwin 		if (error)
23583858a1f4SJohn Baldwin 			return (error);
23593858a1f4SJohn Baldwin 		sigp = &sig;
23603858a1f4SJohn Baldwin 	} else
23613858a1f4SJohn Baldwin 		sigp = NULL;
23623858a1f4SJohn Baldwin 
23633858a1f4SJohn Baldwin 	acb_list = malloc(sizeof(struct aiocb *) * nent, M_LIO, M_WAITOK);
23643858a1f4SJohn Baldwin 	error = copyin(uap->acb_list, acb_list, nent * sizeof(acb_list[0]));
23653858a1f4SJohn Baldwin 	if (error == 0)
23663858a1f4SJohn Baldwin 		error = kern_lio_listio(td, uap->mode, uap->acb_list, acb_list,
23673858a1f4SJohn Baldwin 		    nent, sigp, &aiocb_ops);
23683858a1f4SJohn Baldwin 	free(acb_list, M_LIO);
23693858a1f4SJohn Baldwin 	return (error);
23703858a1f4SJohn Baldwin }
23713858a1f4SJohn Baldwin 
237284af4da6SJohn Dyson /*
23731ce91824SDavid Xu  * Called from interrupt thread for physio, we should return as fast
23741ce91824SDavid Xu  * as possible, so we schedule a biohelper task.
237584af4da6SJohn Dyson  */
2376fd3bf775SJohn Dyson static void
2377bfbbc4aaSJason Evans aio_physwakeup(struct buf *bp)
2378fd3bf775SJohn Dyson {
237984af4da6SJohn Dyson 	struct aiocblist *aiocbe;
2380fd3bf775SJohn Dyson 
23818edbaf85SHidetoshi Shimokawa 	aiocbe = (struct aiocblist *)bp->b_caller1;
23821ce91824SDavid Xu 	taskqueue_enqueue(taskqueue_aiod_bio, &aiocbe->biotask);
23831ce91824SDavid Xu }
238484af4da6SJohn Dyson 
23851ce91824SDavid Xu /*
23861ce91824SDavid Xu  * Task routine to perform heavy tasks, process wakeup, and signals.
23871ce91824SDavid Xu  */
23881ce91824SDavid Xu static void
23891ce91824SDavid Xu biohelper(void *context, int pending)
23901ce91824SDavid Xu {
23911ce91824SDavid Xu 	struct aiocblist *aiocbe = context;
23921ce91824SDavid Xu 	struct buf *bp;
23931ce91824SDavid Xu 	struct proc *userp;
239427b8220dSDavid Xu 	struct kaioinfo *ki;
23951ce91824SDavid Xu 	int nblks;
23961ce91824SDavid Xu 
23971ce91824SDavid Xu 	bp = aiocbe->bp;
23981ce91824SDavid Xu 	userp = aiocbe->userproc;
239927b8220dSDavid Xu 	ki = userp->p_aioinfo;
2400759ccccaSDavid Xu 	AIO_LOCK(ki);
240184af4da6SJohn Dyson 	aiocbe->uaiocb._aiocb_private.status -= bp->b_resid;
240284af4da6SJohn Dyson 	aiocbe->uaiocb._aiocb_private.error = 0;
2403c244d2deSPoul-Henning Kamp 	if (bp->b_ioflags & BIO_ERROR)
240484af4da6SJohn Dyson 		aiocbe->uaiocb._aiocb_private.error = bp->b_error;
24051ce91824SDavid Xu 	nblks = btodb(aiocbe->uaiocb.aio_nbytes);
24061ce91824SDavid Xu 	if (aiocbe->uaiocb.aio_lio_opcode == LIO_WRITE)
24071ce91824SDavid Xu 		aiocbe->outputcharge += nblks;
24081ce91824SDavid Xu 	else
24091ce91824SDavid Xu 		aiocbe->inputcharge += nblks;
24101ce91824SDavid Xu 	aiocbe->bp = NULL;
24111ce91824SDavid Xu 	TAILQ_REMOVE(&userp->p_aioinfo->kaio_bufqueue, aiocbe, plist);
241227b8220dSDavid Xu 	ki->kaio_buffer_count--;
241369cd28daSDoug Ambrisko 	aio_bio_done_notify(userp, aiocbe, DONE_BUF);
2414759ccccaSDavid Xu 	AIO_UNLOCK(ki);
24151ce91824SDavid Xu 
24161ce91824SDavid Xu 	/* Release mapping into kernel space. */
24171ce91824SDavid Xu 	vunmapbuf(bp);
24181ce91824SDavid Xu 	relpbuf(bp, NULL);
24191ce91824SDavid Xu 	atomic_subtract_int(&num_buf_aio, 1);
242084af4da6SJohn Dyson }
2421bfbbc4aaSJason Evans 
2422eb8e6d52SEivind Eklund /* syscall - wait for the next completion of an aio request */
24233858a1f4SJohn Baldwin static int
24243858a1f4SJohn Baldwin kern_aio_waitcomplete(struct thread *td, struct aiocb **aiocbp,
24253858a1f4SJohn Baldwin     struct timespec *ts, struct aiocb_ops *ops)
2426bfbbc4aaSJason Evans {
2427b40ce416SJulian Elischer 	struct proc *p = td->td_proc;
2428bfbbc4aaSJason Evans 	struct timeval atv;
2429bfbbc4aaSJason Evans 	struct kaioinfo *ki;
24301ce91824SDavid Xu 	struct aiocblist *cb;
24311ce91824SDavid Xu 	struct aiocb *uuaiocb;
24321ce91824SDavid Xu 	int error, status, timo;
2433bfbbc4aaSJason Evans 
24343858a1f4SJohn Baldwin 	ops->store_aiocb(aiocbp, NULL);
2435dd85920aSJason Evans 
2436bfbbc4aaSJason Evans 	timo = 0;
24373858a1f4SJohn Baldwin 	if (ts) {
24383858a1f4SJohn Baldwin 		if ((ts->tv_nsec < 0) || (ts->tv_nsec >= 1000000000))
2439bfbbc4aaSJason Evans 			return (EINVAL);
2440bfbbc4aaSJason Evans 
24413858a1f4SJohn Baldwin 		TIMESPEC_TO_TIMEVAL(&atv, ts);
2442bfbbc4aaSJason Evans 		if (itimerfix(&atv))
2443bfbbc4aaSJason Evans 			return (EINVAL);
2444bfbbc4aaSJason Evans 		timo = tvtohz(&atv);
2445bfbbc4aaSJason Evans 	}
2446bfbbc4aaSJason Evans 
24478213baf0SChristian S.J. Peron 	if (p->p_aioinfo == NULL)
2448323fe565SDavid Xu 		aio_init_aioinfo(p);
24498213baf0SChristian S.J. Peron 	ki = p->p_aioinfo;
2450bfbbc4aaSJason Evans 
24511ce91824SDavid Xu 	error = 0;
24521ce91824SDavid Xu 	cb = NULL;
2453759ccccaSDavid Xu 	AIO_LOCK(ki);
24541ce91824SDavid Xu 	while ((cb = TAILQ_FIRST(&ki->kaio_done)) == NULL) {
24551ce91824SDavid Xu 		ki->kaio_flags |= KAIO_WAKEUP;
2456759ccccaSDavid Xu 		error = msleep(&p->p_aioinfo, AIO_MTX(ki), PRIBIO | PCATCH,
24571ce91824SDavid Xu 		    "aiowc", timo);
245827b8220dSDavid Xu 		if (timo && error == ERESTART)
24591ce91824SDavid Xu 			error = EINTR;
24601ce91824SDavid Xu 		if (error)
24611ce91824SDavid Xu 			break;
24621ce91824SDavid Xu 	}
24631ce91824SDavid Xu 
24641ce91824SDavid Xu 	if (cb != NULL) {
24651ce91824SDavid Xu 		MPASS(cb->jobstate == JOBST_JOBFINISHED);
24661ce91824SDavid Xu 		uuaiocb = cb->uuaiocb;
24671ce91824SDavid Xu 		status = cb->uaiocb._aiocb_private.status;
24681ce91824SDavid Xu 		error = cb->uaiocb._aiocb_private.error;
24691ce91824SDavid Xu 		td->td_retval[0] = status;
2470bfbbc4aaSJason Evans 		if (cb->uaiocb.aio_lio_opcode == LIO_WRITE) {
24711c4bcd05SJeff Roberson 			td->td_ru.ru_oublock += cb->outputcharge;
2472bfbbc4aaSJason Evans 			cb->outputcharge = 0;
2473bfbbc4aaSJason Evans 		} else if (cb->uaiocb.aio_lio_opcode == LIO_READ) {
24741c4bcd05SJeff Roberson 			td->td_ru.ru_inblock += cb->inputcharge;
2475bfbbc4aaSJason Evans 			cb->inputcharge = 0;
2476bfbbc4aaSJason Evans 		}
2477bfbbc4aaSJason Evans 		aio_free_entry(cb);
2478759ccccaSDavid Xu 		AIO_UNLOCK(ki);
24793858a1f4SJohn Baldwin 		ops->store_aiocb(aiocbp, uuaiocb);
24803858a1f4SJohn Baldwin 		ops->store_error(uuaiocb, error);
24813858a1f4SJohn Baldwin 		ops->store_status(uuaiocb, status);
24821ce91824SDavid Xu 	} else
2483759ccccaSDavid Xu 		AIO_UNLOCK(ki);
2484bfbbc4aaSJason Evans 
2485ac41f2efSAlfred Perlstein 	return (error);
2486bfbbc4aaSJason Evans }
2487cb679c38SJonathan Lemon 
248899eee864SDavid Xu int
24898451d0ddSKip Macy sys_aio_waitcomplete(struct thread *td, struct aio_waitcomplete_args *uap)
24903858a1f4SJohn Baldwin {
24913858a1f4SJohn Baldwin 	struct timespec ts, *tsp;
24923858a1f4SJohn Baldwin 	int error;
24933858a1f4SJohn Baldwin 
24943858a1f4SJohn Baldwin 	if (uap->timeout) {
24953858a1f4SJohn Baldwin 		/* Get timespec struct. */
24963858a1f4SJohn Baldwin 		error = copyin(uap->timeout, &ts, sizeof(ts));
24973858a1f4SJohn Baldwin 		if (error)
24983858a1f4SJohn Baldwin 			return (error);
24993858a1f4SJohn Baldwin 		tsp = &ts;
25003858a1f4SJohn Baldwin 	} else
25013858a1f4SJohn Baldwin 		tsp = NULL;
25023858a1f4SJohn Baldwin 
25033858a1f4SJohn Baldwin 	return (kern_aio_waitcomplete(td, uap->aiocbp, tsp, &aiocb_ops));
25043858a1f4SJohn Baldwin }
25053858a1f4SJohn Baldwin 
25063858a1f4SJohn Baldwin static int
25073858a1f4SJohn Baldwin kern_aio_fsync(struct thread *td, int op, struct aiocb *aiocbp,
25083858a1f4SJohn Baldwin     struct aiocb_ops *ops)
250999eee864SDavid Xu {
251099eee864SDavid Xu 	struct proc *p = td->td_proc;
251199eee864SDavid Xu 	struct kaioinfo *ki;
251299eee864SDavid Xu 
25133858a1f4SJohn Baldwin 	if (op != O_SYNC) /* XXX lack of O_DSYNC */
251499eee864SDavid Xu 		return (EINVAL);
251599eee864SDavid Xu 	ki = p->p_aioinfo;
251699eee864SDavid Xu 	if (ki == NULL)
251799eee864SDavid Xu 		aio_init_aioinfo(p);
25183858a1f4SJohn Baldwin 	return (aio_aqueue(td, aiocbp, NULL, LIO_SYNC, ops));
25193858a1f4SJohn Baldwin }
25203858a1f4SJohn Baldwin 
25213858a1f4SJohn Baldwin int
25228451d0ddSKip Macy sys_aio_fsync(struct thread *td, struct aio_fsync_args *uap)
25233858a1f4SJohn Baldwin {
25243858a1f4SJohn Baldwin 
25253858a1f4SJohn Baldwin 	return (kern_aio_fsync(td, uap->op, uap->aiocbp, &aiocb_ops));
252699eee864SDavid Xu }
252799eee864SDavid Xu 
2528eb8e6d52SEivind Eklund /* kqueue attach function */
2529cb679c38SJonathan Lemon static int
2530cb679c38SJonathan Lemon filt_aioattach(struct knote *kn)
2531cb679c38SJonathan Lemon {
2532b46f1c55SAlan Cox 	struct aiocblist *aiocbe = (struct aiocblist *)kn->kn_sdata;
2533cb679c38SJonathan Lemon 
2534cb679c38SJonathan Lemon 	/*
2535cb679c38SJonathan Lemon 	 * The aiocbe pointer must be validated before using it, so
2536cb679c38SJonathan Lemon 	 * registration is restricted to the kernel; the user cannot
2537cb679c38SJonathan Lemon 	 * set EV_FLAG1.
2538cb679c38SJonathan Lemon 	 */
2539cb679c38SJonathan Lemon 	if ((kn->kn_flags & EV_FLAG1) == 0)
2540cb679c38SJonathan Lemon 		return (EPERM);
2541a8afa221SJean-Sébastien Pédron 	kn->kn_ptr.p_aio = aiocbe;
2542cb679c38SJonathan Lemon 	kn->kn_flags &= ~EV_FLAG1;
2543cb679c38SJonathan Lemon 
2544ad3b9257SJohn-Mark Gurney 	knlist_add(&aiocbe->klist, kn, 0);
2545cb679c38SJonathan Lemon 
2546cb679c38SJonathan Lemon 	return (0);
2547cb679c38SJonathan Lemon }
2548cb679c38SJonathan Lemon 
2549eb8e6d52SEivind Eklund /* kqueue detach function */
2550cb679c38SJonathan Lemon static void
2551cb679c38SJonathan Lemon filt_aiodetach(struct knote *kn)
2552cb679c38SJonathan Lemon {
25538e9fc278SDoug Ambrisko 	struct knlist *knl;
2554cb679c38SJonathan Lemon 
25558e9fc278SDoug Ambrisko 	knl = &kn->kn_ptr.p_aio->klist;
25568e9fc278SDoug Ambrisko 	knl->kl_lock(knl->kl_lockarg);
25578e9fc278SDoug Ambrisko 	if (!knlist_empty(knl))
25588e9fc278SDoug Ambrisko 		knlist_remove(knl, kn, 1);
25598e9fc278SDoug Ambrisko 	knl->kl_unlock(knl->kl_lockarg);
2560cb679c38SJonathan Lemon }
2561cb679c38SJonathan Lemon 
2562eb8e6d52SEivind Eklund /* kqueue filter function */
2563cb679c38SJonathan Lemon /*ARGSUSED*/
2564cb679c38SJonathan Lemon static int
2565cb679c38SJonathan Lemon filt_aio(struct knote *kn, long hint)
2566cb679c38SJonathan Lemon {
2567a8afa221SJean-Sébastien Pédron 	struct aiocblist *aiocbe = kn->kn_ptr.p_aio;
2568cb679c38SJonathan Lemon 
256991369fc7SAlan Cox 	kn->kn_data = aiocbe->uaiocb._aiocb_private.error;
25701ce91824SDavid Xu 	if (aiocbe->jobstate != JOBST_JOBFINISHED)
2571cb679c38SJonathan Lemon 		return (0);
2572cb679c38SJonathan Lemon 	kn->kn_flags |= EV_EOF;
2573cb679c38SJonathan Lemon 	return (1);
2574cb679c38SJonathan Lemon }
257569cd28daSDoug Ambrisko 
257669cd28daSDoug Ambrisko /* kqueue attach function */
257769cd28daSDoug Ambrisko static int
257869cd28daSDoug Ambrisko filt_lioattach(struct knote *kn)
257969cd28daSDoug Ambrisko {
25801ce91824SDavid Xu 	struct aioliojob * lj = (struct aioliojob *)kn->kn_sdata;
258169cd28daSDoug Ambrisko 
258269cd28daSDoug Ambrisko 	/*
25831ce91824SDavid Xu 	 * The aioliojob pointer must be validated before using it, so
258469cd28daSDoug Ambrisko 	 * registration is restricted to the kernel; the user cannot
258569cd28daSDoug Ambrisko 	 * set EV_FLAG1.
258669cd28daSDoug Ambrisko 	 */
258769cd28daSDoug Ambrisko 	if ((kn->kn_flags & EV_FLAG1) == 0)
258869cd28daSDoug Ambrisko 		return (EPERM);
2589a8afa221SJean-Sébastien Pédron 	kn->kn_ptr.p_lio = lj;
259069cd28daSDoug Ambrisko 	kn->kn_flags &= ~EV_FLAG1;
259169cd28daSDoug Ambrisko 
259269cd28daSDoug Ambrisko 	knlist_add(&lj->klist, kn, 0);
259369cd28daSDoug Ambrisko 
259469cd28daSDoug Ambrisko 	return (0);
259569cd28daSDoug Ambrisko }
259669cd28daSDoug Ambrisko 
259769cd28daSDoug Ambrisko /* kqueue detach function */
259869cd28daSDoug Ambrisko static void
259969cd28daSDoug Ambrisko filt_liodetach(struct knote *kn)
260069cd28daSDoug Ambrisko {
26018e9fc278SDoug Ambrisko 	struct knlist *knl;
260269cd28daSDoug Ambrisko 
26038e9fc278SDoug Ambrisko 	knl = &kn->kn_ptr.p_lio->klist;
26048e9fc278SDoug Ambrisko 	knl->kl_lock(knl->kl_lockarg);
26058e9fc278SDoug Ambrisko 	if (!knlist_empty(knl))
26068e9fc278SDoug Ambrisko 		knlist_remove(knl, kn, 1);
26078e9fc278SDoug Ambrisko 	knl->kl_unlock(knl->kl_lockarg);
260869cd28daSDoug Ambrisko }
260969cd28daSDoug Ambrisko 
261069cd28daSDoug Ambrisko /* kqueue filter function */
261169cd28daSDoug Ambrisko /*ARGSUSED*/
261269cd28daSDoug Ambrisko static int
261369cd28daSDoug Ambrisko filt_lio(struct knote *kn, long hint)
261469cd28daSDoug Ambrisko {
2615a8afa221SJean-Sébastien Pédron 	struct aioliojob * lj = kn->kn_ptr.p_lio;
26161ce91824SDavid Xu 
261769cd28daSDoug Ambrisko 	return (lj->lioj_flags & LIOJ_KEVENT_POSTED);
261869cd28daSDoug Ambrisko }
26193858a1f4SJohn Baldwin 
2620841c0c7eSNathan Whitehorn #ifdef COMPAT_FREEBSD32
26213858a1f4SJohn Baldwin 
26223858a1f4SJohn Baldwin struct __aiocb_private32 {
26233858a1f4SJohn Baldwin 	int32_t	status;
26243858a1f4SJohn Baldwin 	int32_t	error;
26253858a1f4SJohn Baldwin 	uint32_t kernelinfo;
26263858a1f4SJohn Baldwin };
26273858a1f4SJohn Baldwin 
26283858a1f4SJohn Baldwin typedef struct oaiocb32 {
26293858a1f4SJohn Baldwin 	int	aio_fildes;		/* File descriptor */
26303858a1f4SJohn Baldwin 	uint64_t aio_offset __packed;	/* File offset for I/O */
26313858a1f4SJohn Baldwin 	uint32_t aio_buf;		/* I/O buffer in process space */
26323858a1f4SJohn Baldwin 	uint32_t aio_nbytes;		/* Number of bytes for I/O */
26333858a1f4SJohn Baldwin 	struct	osigevent32 aio_sigevent; /* Signal to deliver */
26343858a1f4SJohn Baldwin 	int	aio_lio_opcode;		/* LIO opcode */
26353858a1f4SJohn Baldwin 	int	aio_reqprio;		/* Request priority -- ignored */
26363858a1f4SJohn Baldwin 	struct	__aiocb_private32 _aiocb_private;
26373858a1f4SJohn Baldwin } oaiocb32_t;
26383858a1f4SJohn Baldwin 
26393858a1f4SJohn Baldwin typedef struct aiocb32 {
26403858a1f4SJohn Baldwin 	int32_t	aio_fildes;		/* File descriptor */
26413858a1f4SJohn Baldwin 	uint64_t aio_offset __packed;	/* File offset for I/O */
26423858a1f4SJohn Baldwin 	uint32_t aio_buf;		/* I/O buffer in process space */
26433858a1f4SJohn Baldwin 	uint32_t aio_nbytes;		/* Number of bytes for I/O */
26443858a1f4SJohn Baldwin 	int	__spare__[2];
26453858a1f4SJohn Baldwin 	uint32_t __spare2__;
26463858a1f4SJohn Baldwin 	int	aio_lio_opcode;		/* LIO opcode */
26473858a1f4SJohn Baldwin 	int	aio_reqprio;		/* Request priority -- ignored */
26483858a1f4SJohn Baldwin 	struct __aiocb_private32 _aiocb_private;
26493858a1f4SJohn Baldwin 	struct sigevent32 aio_sigevent;	/* Signal to deliver */
26503858a1f4SJohn Baldwin } aiocb32_t;
26513858a1f4SJohn Baldwin 
26523858a1f4SJohn Baldwin static int
26533858a1f4SJohn Baldwin convert_old_sigevent32(struct osigevent32 *osig, struct sigevent *nsig)
26543858a1f4SJohn Baldwin {
26553858a1f4SJohn Baldwin 
26563858a1f4SJohn Baldwin 	/*
26573858a1f4SJohn Baldwin 	 * Only SIGEV_NONE, SIGEV_SIGNAL, and SIGEV_KEVENT are
26583858a1f4SJohn Baldwin 	 * supported by AIO with the old sigevent structure.
26593858a1f4SJohn Baldwin 	 */
26603858a1f4SJohn Baldwin 	CP(*osig, *nsig, sigev_notify);
26613858a1f4SJohn Baldwin 	switch (nsig->sigev_notify) {
26623858a1f4SJohn Baldwin 	case SIGEV_NONE:
26633858a1f4SJohn Baldwin 		break;
26643858a1f4SJohn Baldwin 	case SIGEV_SIGNAL:
26653858a1f4SJohn Baldwin 		nsig->sigev_signo = osig->__sigev_u.__sigev_signo;
26663858a1f4SJohn Baldwin 		break;
26673858a1f4SJohn Baldwin 	case SIGEV_KEVENT:
26683858a1f4SJohn Baldwin 		nsig->sigev_notify_kqueue =
26693858a1f4SJohn Baldwin 		    osig->__sigev_u.__sigev_notify_kqueue;
26703858a1f4SJohn Baldwin 		PTRIN_CP(*osig, *nsig, sigev_value.sival_ptr);
26713858a1f4SJohn Baldwin 		break;
26723858a1f4SJohn Baldwin 	default:
26733858a1f4SJohn Baldwin 		return (EINVAL);
26743858a1f4SJohn Baldwin 	}
26753858a1f4SJohn Baldwin 	return (0);
26763858a1f4SJohn Baldwin }
26773858a1f4SJohn Baldwin 
26783858a1f4SJohn Baldwin static int
26793858a1f4SJohn Baldwin aiocb32_copyin_old_sigevent(struct aiocb *ujob, struct aiocb *kjob)
26803858a1f4SJohn Baldwin {
26813858a1f4SJohn Baldwin 	struct oaiocb32 job32;
26823858a1f4SJohn Baldwin 	int error;
26833858a1f4SJohn Baldwin 
26843858a1f4SJohn Baldwin 	bzero(kjob, sizeof(struct aiocb));
26853858a1f4SJohn Baldwin 	error = copyin(ujob, &job32, sizeof(job32));
26863858a1f4SJohn Baldwin 	if (error)
26873858a1f4SJohn Baldwin 		return (error);
26883858a1f4SJohn Baldwin 
26893858a1f4SJohn Baldwin 	CP(job32, *kjob, aio_fildes);
26903858a1f4SJohn Baldwin 	CP(job32, *kjob, aio_offset);
26913858a1f4SJohn Baldwin 	PTRIN_CP(job32, *kjob, aio_buf);
26923858a1f4SJohn Baldwin 	CP(job32, *kjob, aio_nbytes);
26933858a1f4SJohn Baldwin 	CP(job32, *kjob, aio_lio_opcode);
26943858a1f4SJohn Baldwin 	CP(job32, *kjob, aio_reqprio);
26953858a1f4SJohn Baldwin 	CP(job32, *kjob, _aiocb_private.status);
26963858a1f4SJohn Baldwin 	CP(job32, *kjob, _aiocb_private.error);
26973858a1f4SJohn Baldwin 	PTRIN_CP(job32, *kjob, _aiocb_private.kernelinfo);
26983858a1f4SJohn Baldwin 	return (convert_old_sigevent32(&job32.aio_sigevent,
26993858a1f4SJohn Baldwin 	    &kjob->aio_sigevent));
27003858a1f4SJohn Baldwin }
27013858a1f4SJohn Baldwin 
27023858a1f4SJohn Baldwin static int
27033858a1f4SJohn Baldwin convert_sigevent32(struct sigevent32 *sig32, struct sigevent *sig)
27043858a1f4SJohn Baldwin {
27053858a1f4SJohn Baldwin 
27063858a1f4SJohn Baldwin 	CP(*sig32, *sig, sigev_notify);
27073858a1f4SJohn Baldwin 	switch (sig->sigev_notify) {
27083858a1f4SJohn Baldwin 	case SIGEV_NONE:
27093858a1f4SJohn Baldwin 		break;
27103858a1f4SJohn Baldwin 	case SIGEV_THREAD_ID:
27113858a1f4SJohn Baldwin 		CP(*sig32, *sig, sigev_notify_thread_id);
27123858a1f4SJohn Baldwin 		/* FALLTHROUGH */
27133858a1f4SJohn Baldwin 	case SIGEV_SIGNAL:
27143858a1f4SJohn Baldwin 		CP(*sig32, *sig, sigev_signo);
27153858a1f4SJohn Baldwin 		break;
27163858a1f4SJohn Baldwin 	case SIGEV_KEVENT:
27173858a1f4SJohn Baldwin 		CP(*sig32, *sig, sigev_notify_kqueue);
2718d56e058aSDavid Xu 		CP(*sig32, *sig, sigev_notify_kevent_flags);
27193858a1f4SJohn Baldwin 		PTRIN_CP(*sig32, *sig, sigev_value.sival_ptr);
27203858a1f4SJohn Baldwin 		break;
27213858a1f4SJohn Baldwin 	default:
27223858a1f4SJohn Baldwin 		return (EINVAL);
27233858a1f4SJohn Baldwin 	}
27243858a1f4SJohn Baldwin 	return (0);
27253858a1f4SJohn Baldwin }
27263858a1f4SJohn Baldwin 
27273858a1f4SJohn Baldwin static int
27283858a1f4SJohn Baldwin aiocb32_copyin(struct aiocb *ujob, struct aiocb *kjob)
27293858a1f4SJohn Baldwin {
27303858a1f4SJohn Baldwin 	struct aiocb32 job32;
27313858a1f4SJohn Baldwin 	int error;
27323858a1f4SJohn Baldwin 
27333858a1f4SJohn Baldwin 	error = copyin(ujob, &job32, sizeof(job32));
27343858a1f4SJohn Baldwin 	if (error)
27353858a1f4SJohn Baldwin 		return (error);
27363858a1f4SJohn Baldwin 	CP(job32, *kjob, aio_fildes);
27373858a1f4SJohn Baldwin 	CP(job32, *kjob, aio_offset);
27383858a1f4SJohn Baldwin 	PTRIN_CP(job32, *kjob, aio_buf);
27393858a1f4SJohn Baldwin 	CP(job32, *kjob, aio_nbytes);
27403858a1f4SJohn Baldwin 	CP(job32, *kjob, aio_lio_opcode);
27413858a1f4SJohn Baldwin 	CP(job32, *kjob, aio_reqprio);
27423858a1f4SJohn Baldwin 	CP(job32, *kjob, _aiocb_private.status);
27433858a1f4SJohn Baldwin 	CP(job32, *kjob, _aiocb_private.error);
27443858a1f4SJohn Baldwin 	PTRIN_CP(job32, *kjob, _aiocb_private.kernelinfo);
27453858a1f4SJohn Baldwin 	return (convert_sigevent32(&job32.aio_sigevent, &kjob->aio_sigevent));
27463858a1f4SJohn Baldwin }
27473858a1f4SJohn Baldwin 
27483858a1f4SJohn Baldwin static long
27493858a1f4SJohn Baldwin aiocb32_fetch_status(struct aiocb *ujob)
27503858a1f4SJohn Baldwin {
27513858a1f4SJohn Baldwin 	struct aiocb32 *ujob32;
27523858a1f4SJohn Baldwin 
27533858a1f4SJohn Baldwin 	ujob32 = (struct aiocb32 *)ujob;
27543858a1f4SJohn Baldwin 	return (fuword32(&ujob32->_aiocb_private.status));
27553858a1f4SJohn Baldwin }
27563858a1f4SJohn Baldwin 
27573858a1f4SJohn Baldwin static long
27583858a1f4SJohn Baldwin aiocb32_fetch_error(struct aiocb *ujob)
27593858a1f4SJohn Baldwin {
27603858a1f4SJohn Baldwin 	struct aiocb32 *ujob32;
27613858a1f4SJohn Baldwin 
27623858a1f4SJohn Baldwin 	ujob32 = (struct aiocb32 *)ujob;
27633858a1f4SJohn Baldwin 	return (fuword32(&ujob32->_aiocb_private.error));
27643858a1f4SJohn Baldwin }
27653858a1f4SJohn Baldwin 
27663858a1f4SJohn Baldwin static int
27673858a1f4SJohn Baldwin aiocb32_store_status(struct aiocb *ujob, long status)
27683858a1f4SJohn Baldwin {
27693858a1f4SJohn Baldwin 	struct aiocb32 *ujob32;
27703858a1f4SJohn Baldwin 
27713858a1f4SJohn Baldwin 	ujob32 = (struct aiocb32 *)ujob;
27723858a1f4SJohn Baldwin 	return (suword32(&ujob32->_aiocb_private.status, status));
27733858a1f4SJohn Baldwin }
27743858a1f4SJohn Baldwin 
27753858a1f4SJohn Baldwin static int
27763858a1f4SJohn Baldwin aiocb32_store_error(struct aiocb *ujob, long error)
27773858a1f4SJohn Baldwin {
27783858a1f4SJohn Baldwin 	struct aiocb32 *ujob32;
27793858a1f4SJohn Baldwin 
27803858a1f4SJohn Baldwin 	ujob32 = (struct aiocb32 *)ujob;
27813858a1f4SJohn Baldwin 	return (suword32(&ujob32->_aiocb_private.error, error));
27823858a1f4SJohn Baldwin }
27833858a1f4SJohn Baldwin 
27843858a1f4SJohn Baldwin static int
27853858a1f4SJohn Baldwin aiocb32_store_kernelinfo(struct aiocb *ujob, long jobref)
27863858a1f4SJohn Baldwin {
27873858a1f4SJohn Baldwin 	struct aiocb32 *ujob32;
27883858a1f4SJohn Baldwin 
27893858a1f4SJohn Baldwin 	ujob32 = (struct aiocb32 *)ujob;
27903858a1f4SJohn Baldwin 	return (suword32(&ujob32->_aiocb_private.kernelinfo, jobref));
27913858a1f4SJohn Baldwin }
27923858a1f4SJohn Baldwin 
27933858a1f4SJohn Baldwin static int
27943858a1f4SJohn Baldwin aiocb32_store_aiocb(struct aiocb **ujobp, struct aiocb *ujob)
27953858a1f4SJohn Baldwin {
27963858a1f4SJohn Baldwin 
27973858a1f4SJohn Baldwin 	return (suword32(ujobp, (long)ujob));
27983858a1f4SJohn Baldwin }
27993858a1f4SJohn Baldwin 
28003858a1f4SJohn Baldwin static struct aiocb_ops aiocb32_ops = {
28013858a1f4SJohn Baldwin 	.copyin = aiocb32_copyin,
28023858a1f4SJohn Baldwin 	.fetch_status = aiocb32_fetch_status,
28033858a1f4SJohn Baldwin 	.fetch_error = aiocb32_fetch_error,
28043858a1f4SJohn Baldwin 	.store_status = aiocb32_store_status,
28053858a1f4SJohn Baldwin 	.store_error = aiocb32_store_error,
28063858a1f4SJohn Baldwin 	.store_kernelinfo = aiocb32_store_kernelinfo,
28073858a1f4SJohn Baldwin 	.store_aiocb = aiocb32_store_aiocb,
28083858a1f4SJohn Baldwin };
28093858a1f4SJohn Baldwin 
28103858a1f4SJohn Baldwin static struct aiocb_ops aiocb32_ops_osigevent = {
28113858a1f4SJohn Baldwin 	.copyin = aiocb32_copyin_old_sigevent,
28123858a1f4SJohn Baldwin 	.fetch_status = aiocb32_fetch_status,
28133858a1f4SJohn Baldwin 	.fetch_error = aiocb32_fetch_error,
28143858a1f4SJohn Baldwin 	.store_status = aiocb32_store_status,
28153858a1f4SJohn Baldwin 	.store_error = aiocb32_store_error,
28163858a1f4SJohn Baldwin 	.store_kernelinfo = aiocb32_store_kernelinfo,
28173858a1f4SJohn Baldwin 	.store_aiocb = aiocb32_store_aiocb,
28183858a1f4SJohn Baldwin };
28193858a1f4SJohn Baldwin 
28203858a1f4SJohn Baldwin int
28213858a1f4SJohn Baldwin freebsd32_aio_return(struct thread *td, struct freebsd32_aio_return_args *uap)
28223858a1f4SJohn Baldwin {
28233858a1f4SJohn Baldwin 
28243858a1f4SJohn Baldwin 	return (kern_aio_return(td, (struct aiocb *)uap->aiocbp, &aiocb32_ops));
28253858a1f4SJohn Baldwin }
28263858a1f4SJohn Baldwin 
28273858a1f4SJohn Baldwin int
28283858a1f4SJohn Baldwin freebsd32_aio_suspend(struct thread *td, struct freebsd32_aio_suspend_args *uap)
28293858a1f4SJohn Baldwin {
28303858a1f4SJohn Baldwin 	struct timespec32 ts32;
28313858a1f4SJohn Baldwin 	struct timespec ts, *tsp;
28323858a1f4SJohn Baldwin 	struct aiocb **ujoblist;
28333858a1f4SJohn Baldwin 	uint32_t *ujoblist32;
28343858a1f4SJohn Baldwin 	int error, i;
28353858a1f4SJohn Baldwin 
28363858a1f4SJohn Baldwin 	if (uap->nent < 0 || uap->nent > AIO_LISTIO_MAX)
28373858a1f4SJohn Baldwin 		return (EINVAL);
28383858a1f4SJohn Baldwin 
28393858a1f4SJohn Baldwin 	if (uap->timeout) {
28403858a1f4SJohn Baldwin 		/* Get timespec struct. */
28413858a1f4SJohn Baldwin 		if ((error = copyin(uap->timeout, &ts32, sizeof(ts32))) != 0)
28423858a1f4SJohn Baldwin 			return (error);
28433858a1f4SJohn Baldwin 		CP(ts32, ts, tv_sec);
28443858a1f4SJohn Baldwin 		CP(ts32, ts, tv_nsec);
28453858a1f4SJohn Baldwin 		tsp = &ts;
28463858a1f4SJohn Baldwin 	} else
28473858a1f4SJohn Baldwin 		tsp = NULL;
28483858a1f4SJohn Baldwin 
28493858a1f4SJohn Baldwin 	ujoblist = uma_zalloc(aiol_zone, M_WAITOK);
28503858a1f4SJohn Baldwin 	ujoblist32 = (uint32_t *)ujoblist;
28513858a1f4SJohn Baldwin 	error = copyin(uap->aiocbp, ujoblist32, uap->nent *
28523858a1f4SJohn Baldwin 	    sizeof(ujoblist32[0]));
28533858a1f4SJohn Baldwin 	if (error == 0) {
28543858a1f4SJohn Baldwin 		for (i = uap->nent; i > 0; i--)
28553858a1f4SJohn Baldwin 			ujoblist[i] = PTRIN(ujoblist32[i]);
28563858a1f4SJohn Baldwin 
28573858a1f4SJohn Baldwin 		error = kern_aio_suspend(td, uap->nent, ujoblist, tsp);
28583858a1f4SJohn Baldwin 	}
28593858a1f4SJohn Baldwin 	uma_zfree(aiol_zone, ujoblist);
28603858a1f4SJohn Baldwin 	return (error);
28613858a1f4SJohn Baldwin }
28623858a1f4SJohn Baldwin 
28633858a1f4SJohn Baldwin int
28643858a1f4SJohn Baldwin freebsd32_aio_cancel(struct thread *td, struct freebsd32_aio_cancel_args *uap)
28653858a1f4SJohn Baldwin {
28663858a1f4SJohn Baldwin 
28678451d0ddSKip Macy 	return (sys_aio_cancel(td, (struct aio_cancel_args *)uap));
28683858a1f4SJohn Baldwin }
28693858a1f4SJohn Baldwin 
28703858a1f4SJohn Baldwin int
28713858a1f4SJohn Baldwin freebsd32_aio_error(struct thread *td, struct freebsd32_aio_error_args *uap)
28723858a1f4SJohn Baldwin {
28733858a1f4SJohn Baldwin 
28743858a1f4SJohn Baldwin 	return (kern_aio_error(td, (struct aiocb *)uap->aiocbp, &aiocb32_ops));
28753858a1f4SJohn Baldwin }
28763858a1f4SJohn Baldwin 
28773858a1f4SJohn Baldwin int
28783858a1f4SJohn Baldwin freebsd32_oaio_read(struct thread *td, struct freebsd32_oaio_read_args *uap)
28793858a1f4SJohn Baldwin {
28803858a1f4SJohn Baldwin 
28813858a1f4SJohn Baldwin 	return (aio_aqueue(td, (struct aiocb *)uap->aiocbp, NULL, LIO_READ,
28823858a1f4SJohn Baldwin 	    &aiocb32_ops_osigevent));
28833858a1f4SJohn Baldwin }
28843858a1f4SJohn Baldwin 
28853858a1f4SJohn Baldwin int
28863858a1f4SJohn Baldwin freebsd32_aio_read(struct thread *td, struct freebsd32_aio_read_args *uap)
28873858a1f4SJohn Baldwin {
28883858a1f4SJohn Baldwin 
28893858a1f4SJohn Baldwin 	return (aio_aqueue(td, (struct aiocb *)uap->aiocbp, NULL, LIO_READ,
28903858a1f4SJohn Baldwin 	    &aiocb32_ops));
28913858a1f4SJohn Baldwin }
28923858a1f4SJohn Baldwin 
28933858a1f4SJohn Baldwin int
28943858a1f4SJohn Baldwin freebsd32_oaio_write(struct thread *td, struct freebsd32_oaio_write_args *uap)
28953858a1f4SJohn Baldwin {
28963858a1f4SJohn Baldwin 
28973858a1f4SJohn Baldwin 	return (aio_aqueue(td, (struct aiocb *)uap->aiocbp, NULL, LIO_WRITE,
28983858a1f4SJohn Baldwin 	    &aiocb32_ops_osigevent));
28993858a1f4SJohn Baldwin }
29003858a1f4SJohn Baldwin 
29013858a1f4SJohn Baldwin int
29023858a1f4SJohn Baldwin freebsd32_aio_write(struct thread *td, struct freebsd32_aio_write_args *uap)
29033858a1f4SJohn Baldwin {
29043858a1f4SJohn Baldwin 
29053858a1f4SJohn Baldwin 	return (aio_aqueue(td, (struct aiocb *)uap->aiocbp, NULL, LIO_WRITE,
29063858a1f4SJohn Baldwin 	    &aiocb32_ops));
29073858a1f4SJohn Baldwin }
29083858a1f4SJohn Baldwin 
29093858a1f4SJohn Baldwin int
29103858a1f4SJohn Baldwin freebsd32_aio_waitcomplete(struct thread *td,
29113858a1f4SJohn Baldwin     struct freebsd32_aio_waitcomplete_args *uap)
29123858a1f4SJohn Baldwin {
2913e588eeb1SJohn Baldwin 	struct timespec32 ts32;
29143858a1f4SJohn Baldwin 	struct timespec ts, *tsp;
29153858a1f4SJohn Baldwin 	int error;
29163858a1f4SJohn Baldwin 
29173858a1f4SJohn Baldwin 	if (uap->timeout) {
29183858a1f4SJohn Baldwin 		/* Get timespec struct. */
29193858a1f4SJohn Baldwin 		error = copyin(uap->timeout, &ts32, sizeof(ts32));
29203858a1f4SJohn Baldwin 		if (error)
29213858a1f4SJohn Baldwin 			return (error);
29223858a1f4SJohn Baldwin 		CP(ts32, ts, tv_sec);
29233858a1f4SJohn Baldwin 		CP(ts32, ts, tv_nsec);
29243858a1f4SJohn Baldwin 		tsp = &ts;
29253858a1f4SJohn Baldwin 	} else
29263858a1f4SJohn Baldwin 		tsp = NULL;
29273858a1f4SJohn Baldwin 
29283858a1f4SJohn Baldwin 	return (kern_aio_waitcomplete(td, (struct aiocb **)uap->aiocbp, tsp,
29293858a1f4SJohn Baldwin 	    &aiocb32_ops));
29303858a1f4SJohn Baldwin }
29313858a1f4SJohn Baldwin 
29323858a1f4SJohn Baldwin int
29333858a1f4SJohn Baldwin freebsd32_aio_fsync(struct thread *td, struct freebsd32_aio_fsync_args *uap)
29343858a1f4SJohn Baldwin {
29353858a1f4SJohn Baldwin 
29363858a1f4SJohn Baldwin 	return (kern_aio_fsync(td, uap->op, (struct aiocb *)uap->aiocbp,
29373858a1f4SJohn Baldwin 	    &aiocb32_ops));
29383858a1f4SJohn Baldwin }
29393858a1f4SJohn Baldwin 
29403858a1f4SJohn Baldwin int
29413858a1f4SJohn Baldwin freebsd32_olio_listio(struct thread *td, struct freebsd32_olio_listio_args *uap)
29423858a1f4SJohn Baldwin {
29433858a1f4SJohn Baldwin 	struct aiocb **acb_list;
29443858a1f4SJohn Baldwin 	struct sigevent *sigp, sig;
29453858a1f4SJohn Baldwin 	struct osigevent32 osig;
29463858a1f4SJohn Baldwin 	uint32_t *acb_list32;
29473858a1f4SJohn Baldwin 	int error, i, nent;
29483858a1f4SJohn Baldwin 
29493858a1f4SJohn Baldwin 	if ((uap->mode != LIO_NOWAIT) && (uap->mode != LIO_WAIT))
29503858a1f4SJohn Baldwin 		return (EINVAL);
29513858a1f4SJohn Baldwin 
29523858a1f4SJohn Baldwin 	nent = uap->nent;
29533858a1f4SJohn Baldwin 	if (nent < 0 || nent > AIO_LISTIO_MAX)
29543858a1f4SJohn Baldwin 		return (EINVAL);
29553858a1f4SJohn Baldwin 
29563858a1f4SJohn Baldwin 	if (uap->sig && (uap->mode == LIO_NOWAIT)) {
29573858a1f4SJohn Baldwin 		error = copyin(uap->sig, &osig, sizeof(osig));
29583858a1f4SJohn Baldwin 		if (error)
29593858a1f4SJohn Baldwin 			return (error);
29603858a1f4SJohn Baldwin 		error = convert_old_sigevent32(&osig, &sig);
29613858a1f4SJohn Baldwin 		if (error)
29623858a1f4SJohn Baldwin 			return (error);
29633858a1f4SJohn Baldwin 		sigp = &sig;
29643858a1f4SJohn Baldwin 	} else
29653858a1f4SJohn Baldwin 		sigp = NULL;
29663858a1f4SJohn Baldwin 
29673858a1f4SJohn Baldwin 	acb_list32 = malloc(sizeof(uint32_t) * nent, M_LIO, M_WAITOK);
29683858a1f4SJohn Baldwin 	error = copyin(uap->acb_list, acb_list32, nent * sizeof(uint32_t));
29693858a1f4SJohn Baldwin 	if (error) {
29703858a1f4SJohn Baldwin 		free(acb_list32, M_LIO);
29713858a1f4SJohn Baldwin 		return (error);
29723858a1f4SJohn Baldwin 	}
29733858a1f4SJohn Baldwin 	acb_list = malloc(sizeof(struct aiocb *) * nent, M_LIO, M_WAITOK);
29743858a1f4SJohn Baldwin 	for (i = 0; i < nent; i++)
29753858a1f4SJohn Baldwin 		acb_list[i] = PTRIN(acb_list32[i]);
29763858a1f4SJohn Baldwin 	free(acb_list32, M_LIO);
29773858a1f4SJohn Baldwin 
29783858a1f4SJohn Baldwin 	error = kern_lio_listio(td, uap->mode,
29793858a1f4SJohn Baldwin 	    (struct aiocb * const *)uap->acb_list, acb_list, nent, sigp,
29803858a1f4SJohn Baldwin 	    &aiocb32_ops_osigevent);
29813858a1f4SJohn Baldwin 	free(acb_list, M_LIO);
29823858a1f4SJohn Baldwin 	return (error);
29833858a1f4SJohn Baldwin }
29843858a1f4SJohn Baldwin 
29853858a1f4SJohn Baldwin int
29863858a1f4SJohn Baldwin freebsd32_lio_listio(struct thread *td, struct freebsd32_lio_listio_args *uap)
29873858a1f4SJohn Baldwin {
29883858a1f4SJohn Baldwin 	struct aiocb **acb_list;
29893858a1f4SJohn Baldwin 	struct sigevent *sigp, sig;
29903858a1f4SJohn Baldwin 	struct sigevent32 sig32;
29913858a1f4SJohn Baldwin 	uint32_t *acb_list32;
29923858a1f4SJohn Baldwin 	int error, i, nent;
29933858a1f4SJohn Baldwin 
29943858a1f4SJohn Baldwin 	if ((uap->mode != LIO_NOWAIT) && (uap->mode != LIO_WAIT))
29953858a1f4SJohn Baldwin 		return (EINVAL);
29963858a1f4SJohn Baldwin 
29973858a1f4SJohn Baldwin 	nent = uap->nent;
29983858a1f4SJohn Baldwin 	if (nent < 0 || nent > AIO_LISTIO_MAX)
29993858a1f4SJohn Baldwin 		return (EINVAL);
30003858a1f4SJohn Baldwin 
30013858a1f4SJohn Baldwin 	if (uap->sig && (uap->mode == LIO_NOWAIT)) {
30023858a1f4SJohn Baldwin 		error = copyin(uap->sig, &sig32, sizeof(sig32));
30033858a1f4SJohn Baldwin 		if (error)
30043858a1f4SJohn Baldwin 			return (error);
30053858a1f4SJohn Baldwin 		error = convert_sigevent32(&sig32, &sig);
30063858a1f4SJohn Baldwin 		if (error)
30073858a1f4SJohn Baldwin 			return (error);
30083858a1f4SJohn Baldwin 		sigp = &sig;
30093858a1f4SJohn Baldwin 	} else
30103858a1f4SJohn Baldwin 		sigp = NULL;
30113858a1f4SJohn Baldwin 
30123858a1f4SJohn Baldwin 	acb_list32 = malloc(sizeof(uint32_t) * nent, M_LIO, M_WAITOK);
30133858a1f4SJohn Baldwin 	error = copyin(uap->acb_list, acb_list32, nent * sizeof(uint32_t));
30143858a1f4SJohn Baldwin 	if (error) {
30153858a1f4SJohn Baldwin 		free(acb_list32, M_LIO);
30163858a1f4SJohn Baldwin 		return (error);
30173858a1f4SJohn Baldwin 	}
30183858a1f4SJohn Baldwin 	acb_list = malloc(sizeof(struct aiocb *) * nent, M_LIO, M_WAITOK);
30193858a1f4SJohn Baldwin 	for (i = 0; i < nent; i++)
30203858a1f4SJohn Baldwin 		acb_list[i] = PTRIN(acb_list32[i]);
30213858a1f4SJohn Baldwin 	free(acb_list32, M_LIO);
30223858a1f4SJohn Baldwin 
30233858a1f4SJohn Baldwin 	error = kern_lio_listio(td, uap->mode,
30243858a1f4SJohn Baldwin 	    (struct aiocb * const *)uap->acb_list, acb_list, nent, sigp,
30253858a1f4SJohn Baldwin 	    &aiocb32_ops);
30263858a1f4SJohn Baldwin 	free(acb_list, M_LIO);
30273858a1f4SJohn Baldwin 	return (error);
30283858a1f4SJohn Baldwin }
30293858a1f4SJohn Baldwin 
30303858a1f4SJohn Baldwin #endif
3031