xref: /freebsd/sys/kern/vfs_aio.c (revision 2d2f8ae7ad1674f5e74838d87c6011ffa77b0133)
1ee877a35SJohn Dyson /*
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  *
162d2f8ae7SBruce Evans  * $Id: vfs_aio.c,v 1.32 1998/07/15 06:51:14 bde Exp $
17ee877a35SJohn Dyson  */
18ee877a35SJohn Dyson 
19ee877a35SJohn Dyson /*
208a6472b7SPeter Dufault  * This file contains support for the POSIX 1003.1B AIO/LIO facility.
21ee877a35SJohn Dyson  */
22ee877a35SJohn Dyson 
23ee877a35SJohn Dyson #include <sys/param.h>
24ee877a35SJohn Dyson #include <sys/systm.h>
25ee877a35SJohn Dyson #include <sys/sysproto.h>
26ee877a35SJohn Dyson #include <sys/filedesc.h>
27ee877a35SJohn Dyson #include <sys/kernel.h>
28ee877a35SJohn Dyson #include <sys/fcntl.h>
29ee877a35SJohn Dyson #include <sys/file.h>
30fdebd4f0SBruce Evans #include <sys/lock.h>
31ee877a35SJohn Dyson #include <sys/unistd.h>
32ee877a35SJohn Dyson #include <sys/proc.h>
332d2f8ae7SBruce Evans #include <sys/resourcevar.h>
34ee877a35SJohn Dyson #include <sys/signalvar.h>
35a624e84fSJohn Dyson #include <sys/sysctl.h>
36fd3bf775SJohn Dyson #include <sys/vnode.h>
37fd3bf775SJohn Dyson #include <sys/conf.h>
38fd3bf775SJohn Dyson #include <miscfs/specfs/specdev.h>
39ee877a35SJohn Dyson 
40ee877a35SJohn Dyson #include <vm/vm.h>
41ee877a35SJohn Dyson #include <vm/vm_param.h>
42ee877a35SJohn Dyson #include <vm/vm_extern.h>
432244ea07SJohn Dyson #include <vm/pmap.h>
442244ea07SJohn Dyson #include <vm/vm_map.h>
45fd3bf775SJohn Dyson #include <vm/vm_zone.h>
46ee877a35SJohn Dyson #include <sys/aio.h>
475aaef07cSJohn Dyson #include <sys/shm.h>
485aaef07cSJohn Dyson 
495aaef07cSJohn Dyson #include <machine/cpu.h>
5008637435SBruce Evans #include <machine/limits.h>
51ee877a35SJohn Dyson 
528c12612cSDoug Rabson static	long jobrefid;
532244ea07SJohn Dyson 
542244ea07SJohn Dyson #define JOBST_NULL			0x0
552244ea07SJohn Dyson #define	JOBST_JOBQPROC		0x1
562244ea07SJohn Dyson #define JOBST_JOBQGLOBAL	0x2
572244ea07SJohn Dyson #define JOBST_JOBRUNNING	0x3
582244ea07SJohn Dyson #define JOBST_JOBFINISHED	0x4
59fd3bf775SJohn Dyson #define	JOBST_JOBQBUF		0x5
60fd3bf775SJohn Dyson #define	JOBST_JOBBFINISHED	0x6
612244ea07SJohn Dyson 
6284af4da6SJohn Dyson #ifndef MAX_AIO_PER_PROC
632244ea07SJohn Dyson #define MAX_AIO_PER_PROC	32
6484af4da6SJohn Dyson #endif
6584af4da6SJohn Dyson 
6684af4da6SJohn Dyson #ifndef MAX_AIO_QUEUE_PER_PROC
672244ea07SJohn Dyson #define MAX_AIO_QUEUE_PER_PROC	256 /* Bigger than AIO_LISTIO_MAX */
6884af4da6SJohn Dyson #endif
6984af4da6SJohn Dyson 
7084af4da6SJohn Dyson #ifndef MAX_AIO_PROCS
71fd3bf775SJohn Dyson #define MAX_AIO_PROCS		32
7284af4da6SJohn Dyson #endif
7384af4da6SJohn Dyson 
7484af4da6SJohn Dyson #ifndef MAX_AIO_QUEUE
752244ea07SJohn Dyson #define	MAX_AIO_QUEUE		1024 /* Bigger than AIO_LISTIO_MAX */
7684af4da6SJohn Dyson #endif
7784af4da6SJohn Dyson 
7884af4da6SJohn Dyson #ifndef TARGET_AIO_PROCS
7984af4da6SJohn Dyson #define TARGET_AIO_PROCS	0
8084af4da6SJohn Dyson #endif
8184af4da6SJohn Dyson 
8284af4da6SJohn Dyson #ifndef MAX_BUF_AIO
8384af4da6SJohn Dyson #define MAX_BUF_AIO 16
8484af4da6SJohn Dyson #endif
8584af4da6SJohn Dyson 
8684af4da6SJohn Dyson #ifndef AIOD_TIMEOUT_DEFAULT
8784af4da6SJohn Dyson #define	AIOD_TIMEOUT_DEFAULT (10 * hz)
8884af4da6SJohn Dyson #endif
8984af4da6SJohn Dyson 
9084af4da6SJohn Dyson #ifndef AIOD_LIFETIME_DEFAULT
9184af4da6SJohn Dyson #define AIOD_LIFETIME_DEFAULT (30 * hz)
9284af4da6SJohn Dyson #endif
932244ea07SJohn Dyson 
94303b270bSEivind Eklund static int max_aio_procs = MAX_AIO_PROCS;
95303b270bSEivind Eklund static int num_aio_procs = 0;
96303b270bSEivind Eklund static int target_aio_procs = TARGET_AIO_PROCS;
97303b270bSEivind Eklund static int max_queue_count = MAX_AIO_QUEUE;
98303b270bSEivind Eklund static int num_queue_count = 0;
99303b270bSEivind Eklund static int num_buf_aio = 0;
100303b270bSEivind Eklund static int num_aio_resv_start = 0;
101303b270bSEivind Eklund static int aiod_timeout;
102303b270bSEivind Eklund static int aiod_lifetime;
103a624e84fSJohn Dyson 
104303b270bSEivind Eklund static int max_aio_per_proc = MAX_AIO_PER_PROC,
105a624e84fSJohn Dyson 	max_aio_queue_per_proc=MAX_AIO_QUEUE_PER_PROC;
106a624e84fSJohn Dyson 
107303b270bSEivind Eklund static int max_buf_aio = MAX_BUF_AIO;
108a624e84fSJohn Dyson 
109a624e84fSJohn Dyson SYSCTL_NODE(_vfs, OID_AUTO, aio, CTLFLAG_RW, 0, "AIO mgmt");
110a624e84fSJohn Dyson 
111a624e84fSJohn Dyson SYSCTL_INT(_vfs_aio, OID_AUTO, max_aio_per_proc,
112a624e84fSJohn Dyson 	CTLFLAG_RW, &max_aio_per_proc, 0, "");
113a624e84fSJohn Dyson 
114a624e84fSJohn Dyson SYSCTL_INT(_vfs_aio, OID_AUTO, max_aio_queue_per_proc,
115a624e84fSJohn Dyson 	CTLFLAG_RW, &max_aio_queue_per_proc, 0, "");
116a624e84fSJohn Dyson 
117a624e84fSJohn Dyson SYSCTL_INT(_vfs_aio, OID_AUTO, max_aio_procs,
118a624e84fSJohn Dyson 	CTLFLAG_RW, &max_aio_procs, 0, "");
119a624e84fSJohn Dyson 
120a624e84fSJohn Dyson SYSCTL_INT(_vfs_aio, OID_AUTO, num_aio_procs,
121a624e84fSJohn Dyson 	CTLFLAG_RD, &num_aio_procs, 0, "");
122a624e84fSJohn Dyson 
123a624e84fSJohn Dyson SYSCTL_INT(_vfs_aio, OID_AUTO, num_queue_count,
124a624e84fSJohn Dyson 	CTLFLAG_RD, &num_queue_count, 0, "");
125a624e84fSJohn Dyson 
126a624e84fSJohn Dyson SYSCTL_INT(_vfs_aio, OID_AUTO, max_aio_queue,
127a624e84fSJohn Dyson 	CTLFLAG_RW, &max_queue_count, 0, "");
128a624e84fSJohn Dyson 
129a624e84fSJohn Dyson SYSCTL_INT(_vfs_aio, OID_AUTO, target_aio_procs,
130a624e84fSJohn Dyson 	CTLFLAG_RW, &target_aio_procs, 0, "");
131a624e84fSJohn Dyson 
13284af4da6SJohn Dyson SYSCTL_INT(_vfs_aio, OID_AUTO, max_buf_aio,
13384af4da6SJohn Dyson 	CTLFLAG_RW, &max_buf_aio, 0, "");
134fd3bf775SJohn Dyson 
135fd3bf775SJohn Dyson SYSCTL_INT(_vfs_aio, OID_AUTO, num_buf_aio,
136fd3bf775SJohn Dyson 	CTLFLAG_RD, &num_buf_aio, 0, "");
137fd3bf775SJohn Dyson 
13884af4da6SJohn Dyson SYSCTL_INT(_vfs_aio, OID_AUTO, aiod_lifetime,
13984af4da6SJohn Dyson 	CTLFLAG_RW, &aiod_lifetime, 0, "");
14084af4da6SJohn Dyson 
14184af4da6SJohn Dyson SYSCTL_INT(_vfs_aio, OID_AUTO, aiod_timeout,
14284af4da6SJohn Dyson 	CTLFLAG_RW, &aiod_timeout, 0, "");
14384af4da6SJohn Dyson 
14484af4da6SJohn Dyson 
1452244ea07SJohn Dyson /*
1462244ea07SJohn Dyson  * Job queue item
1472244ea07SJohn Dyson  */
14884af4da6SJohn Dyson 
14984af4da6SJohn Dyson #define AIOCBLIST_CANCELLED	0x1
15084af4da6SJohn Dyson #define AIOCBLIST_RUNDOWN	0x4
15184af4da6SJohn Dyson #define AIOCBLIST_ASYNCFREE	0x8
15284af4da6SJohn Dyson #define AIOCBLIST_DONE		0x10
15384af4da6SJohn Dyson 
1542244ea07SJohn Dyson struct aiocblist {
1552244ea07SJohn Dyson 	TAILQ_ENTRY (aiocblist) list;		/* List of jobs */
1562244ea07SJohn Dyson 	TAILQ_ENTRY (aiocblist) plist;		/* List of jobs for proc */
1572244ea07SJohn Dyson 	int	jobflags;
1582244ea07SJohn Dyson 	int	jobstate;
159fd3bf775SJohn Dyson 	int inputcharge, outputcharge;
160fd3bf775SJohn Dyson 	struct	buf *bp;				/* buffer pointer */
1612244ea07SJohn Dyson 	struct	proc *userproc;			/* User process */
1622244ea07SJohn Dyson 	struct	aioproclist	*jobaioproc;	/* AIO process descriptor */
16384af4da6SJohn Dyson 	struct	aio_liojob	*lio;		/* optional lio job */
16411783b14SJohn Dyson 	struct	aiocb *uuaiocb;			/* pointer in userspace of aiocb */
1652244ea07SJohn Dyson 	struct	aiocb uaiocb;			/* Kernel I/O control block */
1662244ea07SJohn Dyson };
1672244ea07SJohn Dyson 
168fd3bf775SJohn Dyson 
1692244ea07SJohn Dyson /*
1702244ea07SJohn Dyson  * AIO process info
1712244ea07SJohn Dyson  */
17284af4da6SJohn Dyson #define AIOP_FREE	0x1			/* proc on free queue */
17384af4da6SJohn Dyson #define AIOP_SCHED	0x2			/* proc explicitly scheduled */
17484af4da6SJohn Dyson 
1752244ea07SJohn Dyson struct aioproclist {
1762244ea07SJohn Dyson 	int aioprocflags;			/* AIO proc flags */
1772244ea07SJohn Dyson 	TAILQ_ENTRY(aioproclist) list;		/* List of processes */
1782244ea07SJohn Dyson 	struct proc *aioproc;			/* The AIO thread */
1792244ea07SJohn Dyson 	TAILQ_HEAD (,aiocblist) jobtorun;	/* suggested job to run */
1802244ea07SJohn Dyson };
1812244ea07SJohn Dyson 
18284af4da6SJohn Dyson /*
18384af4da6SJohn Dyson  * data-structure for lio signal management
18484af4da6SJohn Dyson  */
18584af4da6SJohn Dyson struct aio_liojob {
18684af4da6SJohn Dyson 	int lioj_flags;
18784af4da6SJohn Dyson 	int	lioj_buffer_count;
18884af4da6SJohn Dyson 	int	lioj_buffer_finished_count;
18984af4da6SJohn Dyson 	int	lioj_queue_count;
19084af4da6SJohn Dyson 	int	lioj_queue_finished_count;
19184af4da6SJohn Dyson 	struct sigevent lioj_signal;	/* signal on all I/O done */
19284af4da6SJohn Dyson 	TAILQ_ENTRY (aio_liojob) lioj_list;
19384af4da6SJohn Dyson 	struct kaioinfo *lioj_ki;
19484af4da6SJohn Dyson };
19584af4da6SJohn Dyson #define	LIOJ_SIGNAL			0x1 /* signal on all done (lio) */
19684af4da6SJohn Dyson #define	LIOJ_SIGNAL_POSTED	0x2	/* signal has been posted */
19784af4da6SJohn Dyson 
19884af4da6SJohn Dyson /*
19984af4da6SJohn Dyson  * per process aio data structure
20084af4da6SJohn Dyson  */
2012244ea07SJohn Dyson struct kaioinfo {
202fd3bf775SJohn Dyson 	int	kaio_flags;			/* per process kaio flags */
2032244ea07SJohn Dyson 	int	kaio_maxactive_count;	/* maximum number of AIOs */
2042244ea07SJohn Dyson 	int	kaio_active_count;	/* number of currently used AIOs */
2052244ea07SJohn Dyson 	int	kaio_qallowed_count;	/* maxiumu size of AIO queue */
2062244ea07SJohn Dyson 	int	kaio_queue_count;	/* size of AIO queue */
207fd3bf775SJohn Dyson 	int	kaio_ballowed_count;	/* maximum number of buffers */
20884af4da6SJohn Dyson 	int	kaio_queue_finished_count;	/* number of daemon jobs finished */
209fd3bf775SJohn Dyson 	int	kaio_buffer_count;	/* number of physio buffers */
21084af4da6SJohn Dyson 	int	kaio_buffer_finished_count;	/* count of I/O done */
21184af4da6SJohn Dyson 	struct proc *kaio_p;			/* process that uses this kaio block */
21284af4da6SJohn Dyson 	TAILQ_HEAD (,aio_liojob) kaio_liojoblist;	/* list of lio jobs */
2132244ea07SJohn Dyson 	TAILQ_HEAD (,aiocblist)	kaio_jobqueue;	/* job queue for process */
2142244ea07SJohn Dyson 	TAILQ_HEAD (,aiocblist)	kaio_jobdone;	/* done queue for process */
215fd3bf775SJohn Dyson 	TAILQ_HEAD (,aiocblist)	kaio_bufqueue;	/* buffer job queue for process */
216fd3bf775SJohn Dyson 	TAILQ_HEAD (,aiocblist)	kaio_bufdone;	/* buffer done queue for process */
2172244ea07SJohn Dyson };
2182244ea07SJohn Dyson 
21984af4da6SJohn Dyson #define KAIO_RUNDOWN 0x1		/* process is being run down */
22084af4da6SJohn Dyson #define KAIO_WAKEUP 0x2			/* wakeup process when there is a significant
22184af4da6SJohn Dyson 								   event */
22284af4da6SJohn Dyson 
223fd3bf775SJohn Dyson 
224303b270bSEivind Eklund static TAILQ_HEAD (,aioproclist) aio_freeproc, aio_activeproc;
225303b270bSEivind Eklund static TAILQ_HEAD(,aiocblist) aio_jobs;			/* Async job list */
226303b270bSEivind Eklund static TAILQ_HEAD(,aiocblist) aio_bufjobs;		/* Phys I/O job list */
227303b270bSEivind Eklund static TAILQ_HEAD(,aiocblist) aio_freejobs;		/* Pool of free jobs */
2282244ea07SJohn Dyson 
229fd3bf775SJohn Dyson static void aio_init_aioinfo(struct proc *p) ;
230fd3bf775SJohn Dyson static void aio_onceonly(void *) ;
231fd3bf775SJohn Dyson static int aio_free_entry(struct aiocblist *aiocbe);
232fd3bf775SJohn Dyson static void aio_process(struct aiocblist *aiocbe);
2332244ea07SJohn Dyson static int aio_newproc(void) ;
2342244ea07SJohn Dyson static int aio_aqueue(struct proc *p, struct aiocb *job, int type) ;
235fd3bf775SJohn Dyson static void aio_physwakeup(struct buf *bp);
236fd3bf775SJohn Dyson static int aio_fphysio(struct proc *p, struct aiocblist *aiocbe, int type);
237fd3bf775SJohn Dyson static int aio_qphysio(struct proc *p, struct aiocblist *iocb);
238fd3bf775SJohn Dyson static void aio_daemon(void *uproc);
2392244ea07SJohn Dyson 
2402244ea07SJohn Dyson SYSINIT(aio, SI_SUB_VFS, SI_ORDER_ANY, aio_onceonly, NULL);
2412244ea07SJohn Dyson 
24284af4da6SJohn Dyson static vm_zone_t kaio_zone=0, aiop_zone=0,
24384af4da6SJohn Dyson 	aiocb_zone=0, aiol_zone=0, aiolio_zone=0;
244fd3bf775SJohn Dyson 
245fd3bf775SJohn Dyson /*
246fd3bf775SJohn Dyson  * Single AIOD vmspace shared amongst all of them
247fd3bf775SJohn Dyson  */
248fd3bf775SJohn Dyson static struct vmspace *aiovmspace = NULL;
249a624e84fSJohn Dyson 
2502244ea07SJohn Dyson /*
2512244ea07SJohn Dyson  * Startup initialization
2522244ea07SJohn Dyson  */
2532244ea07SJohn Dyson void
254fd3bf775SJohn Dyson aio_onceonly(void *na)
255fd3bf775SJohn Dyson {
2562244ea07SJohn Dyson 	TAILQ_INIT(&aio_freeproc);
2572244ea07SJohn Dyson 	TAILQ_INIT(&aio_activeproc);
2582244ea07SJohn Dyson 	TAILQ_INIT(&aio_jobs);
259fd3bf775SJohn Dyson 	TAILQ_INIT(&aio_bufjobs);
2602244ea07SJohn Dyson 	TAILQ_INIT(&aio_freejobs);
261fd3bf775SJohn Dyson 	kaio_zone = zinit("AIO", sizeof (struct kaioinfo), 0, 0, 1);
262fd3bf775SJohn Dyson 	aiop_zone = zinit("AIOP", sizeof (struct aioproclist), 0, 0, 1);
263fd3bf775SJohn Dyson 	aiocb_zone = zinit("AIOCB", sizeof (struct aiocblist), 0, 0, 1);
264fd3bf775SJohn Dyson 	aiol_zone = zinit("AIOL", AIO_LISTIO_MAX * sizeof (int), 0, 0, 1);
26584af4da6SJohn Dyson 	aiolio_zone = zinit("AIOLIO",
26684af4da6SJohn Dyson 		AIO_LISTIO_MAX * sizeof (struct aio_liojob), 0, 0, 1);
26784af4da6SJohn Dyson 	aiod_timeout = AIOD_TIMEOUT_DEFAULT;
26884af4da6SJohn Dyson 	aiod_lifetime = AIOD_LIFETIME_DEFAULT;
269fd3bf775SJohn Dyson 	jobrefid = 1;
2702244ea07SJohn Dyson }
2712244ea07SJohn Dyson 
2722244ea07SJohn Dyson /*
2732244ea07SJohn Dyson  * Init the per-process aioinfo structure.
27484af4da6SJohn Dyson  * The aioinfo limits are set per-process for user limit (resource) management.
2752244ea07SJohn Dyson  */
2762244ea07SJohn Dyson void
277fd3bf775SJohn Dyson aio_init_aioinfo(struct proc *p)
278fd3bf775SJohn Dyson {
2792244ea07SJohn Dyson 	struct kaioinfo *ki;
2802244ea07SJohn Dyson 	if (p->p_aioinfo == NULL) {
281fd3bf775SJohn Dyson 		ki = zalloc(kaio_zone);
2822244ea07SJohn Dyson 		p->p_aioinfo = ki;
28384af4da6SJohn Dyson 		ki->kaio_flags = 0;
284a624e84fSJohn Dyson 		ki->kaio_maxactive_count = max_aio_per_proc;
2852244ea07SJohn Dyson 		ki->kaio_active_count = 0;
286a624e84fSJohn Dyson 		ki->kaio_qallowed_count = max_aio_queue_per_proc;
2872244ea07SJohn Dyson 		ki->kaio_queue_count = 0;
28884af4da6SJohn Dyson 		ki->kaio_ballowed_count = max_buf_aio;
289fd3bf775SJohn Dyson 		ki->kaio_buffer_count = 0;
29084af4da6SJohn Dyson 		ki->kaio_buffer_finished_count = 0;
29184af4da6SJohn Dyson 		ki->kaio_p = p;
2922244ea07SJohn Dyson 		TAILQ_INIT(&ki->kaio_jobdone);
2932244ea07SJohn Dyson 		TAILQ_INIT(&ki->kaio_jobqueue);
294fd3bf775SJohn Dyson 		TAILQ_INIT(&ki->kaio_bufdone);
295fd3bf775SJohn Dyson 		TAILQ_INIT(&ki->kaio_bufqueue);
29684af4da6SJohn Dyson 		TAILQ_INIT(&ki->kaio_liojoblist);
2972244ea07SJohn Dyson 	}
2982244ea07SJohn Dyson }
2992244ea07SJohn Dyson 
3002244ea07SJohn Dyson /*
3012244ea07SJohn Dyson  * Free a job entry.  Wait for completion if it is currently
3022244ea07SJohn Dyson  * active, but don't delay forever.  If we delay, we return
3032244ea07SJohn Dyson  * a flag that says that we have to restart the queue scan.
3042244ea07SJohn Dyson  */
3052244ea07SJohn Dyson int
306fd3bf775SJohn Dyson aio_free_entry(struct aiocblist *aiocbe)
307fd3bf775SJohn Dyson {
3082244ea07SJohn Dyson 	struct kaioinfo *ki;
3092244ea07SJohn Dyson 	struct aioproclist *aiop;
31084af4da6SJohn Dyson 	struct aio_liojob *lj;
3112244ea07SJohn Dyson 	struct proc *p;
312fd3bf775SJohn Dyson 	int error;
31311783b14SJohn Dyson 	int s;
3142244ea07SJohn Dyson 
3152244ea07SJohn Dyson 	if (aiocbe->jobstate == JOBST_NULL)
3162244ea07SJohn Dyson 		panic("aio_free_entry: freeing already free job");
3172244ea07SJohn Dyson 
3182244ea07SJohn Dyson 	p = aiocbe->userproc;
3192244ea07SJohn Dyson 	ki = p->p_aioinfo;
32084af4da6SJohn Dyson 	lj = aiocbe->lio;
3212244ea07SJohn Dyson 	if (ki == NULL)
3222244ea07SJohn Dyson 		panic("aio_free_entry: missing p->p_aioinfo");
3232244ea07SJohn Dyson 
3242244ea07SJohn Dyson 	if (aiocbe->jobstate == JOBST_JOBRUNNING) {
3252244ea07SJohn Dyson 		if (aiocbe->jobflags & AIOCBLIST_ASYNCFREE)
3262244ea07SJohn Dyson 			return 0;
3272244ea07SJohn Dyson 		aiocbe->jobflags |= AIOCBLIST_RUNDOWN;
328a624e84fSJohn Dyson 		tsleep(aiocbe, PRIBIO|PCATCH, "jobwai", 0);
3292244ea07SJohn Dyson 	}
3302244ea07SJohn Dyson 	aiocbe->jobflags &= ~AIOCBLIST_ASYNCFREE;
3312244ea07SJohn Dyson 
332fd3bf775SJohn Dyson 	if (aiocbe->bp == NULL) {
3332244ea07SJohn Dyson 		if (ki->kaio_queue_count <= 0)
3342244ea07SJohn Dyson 			panic("aio_free_entry: process queue size <= 0");
3352244ea07SJohn Dyson 		if (num_queue_count <= 0)
3362244ea07SJohn Dyson 			panic("aio_free_entry: system wide queue size <= 0");
3372244ea07SJohn Dyson 
33884af4da6SJohn Dyson 		if(lj) {
33984af4da6SJohn Dyson 			lj->lioj_queue_count--;
34084af4da6SJohn Dyson 			if (aiocbe->jobflags & AIOCBLIST_DONE)
34184af4da6SJohn Dyson 				lj->lioj_queue_finished_count--;
34284af4da6SJohn Dyson 		}
34384af4da6SJohn Dyson 		ki->kaio_queue_count--;
34484af4da6SJohn Dyson 		if (aiocbe->jobflags & AIOCBLIST_DONE)
34584af4da6SJohn Dyson 			ki->kaio_queue_finished_count--;
34684af4da6SJohn Dyson 		num_queue_count--;
347fd3bf775SJohn Dyson 
348fd3bf775SJohn Dyson 	} else {
34984af4da6SJohn Dyson 		if(lj) {
35084af4da6SJohn Dyson 			lj->lioj_buffer_count--;
35184af4da6SJohn Dyson 			if (aiocbe->jobflags & AIOCBLIST_DONE)
35284af4da6SJohn Dyson 				lj->lioj_buffer_finished_count--;
35384af4da6SJohn Dyson 		}
35484af4da6SJohn Dyson 		if (aiocbe->jobflags & AIOCBLIST_DONE)
35584af4da6SJohn Dyson 			ki->kaio_buffer_finished_count--;
35684af4da6SJohn Dyson 		ki->kaio_buffer_count--;
35784af4da6SJohn Dyson 		num_buf_aio--;
3582244ea07SJohn Dyson 
359fd3bf775SJohn Dyson 	}
360fd3bf775SJohn Dyson 
361fd3bf775SJohn Dyson 	if ((ki->kaio_flags & KAIO_WAKEUP) ||
362fd3bf775SJohn Dyson 		(ki->kaio_flags & KAIO_RUNDOWN) &&
363fd3bf775SJohn Dyson 		((ki->kaio_buffer_count == 0) && (ki->kaio_queue_count == 0))) {
364fd3bf775SJohn Dyson 		ki->kaio_flags &= ~KAIO_WAKEUP;
365fd3bf775SJohn Dyson 		wakeup(p);
366fd3bf775SJohn Dyson 	}
367fd3bf775SJohn Dyson 
368fd3bf775SJohn Dyson 	if ( aiocbe->jobstate == JOBST_JOBQBUF) {
369fd3bf775SJohn Dyson 		if ((error = aio_fphysio(p, aiocbe, 1)) != 0)
370fd3bf775SJohn Dyson 			return error;
371fd3bf775SJohn Dyson 		if (aiocbe->jobstate != JOBST_JOBBFINISHED)
372fd3bf775SJohn Dyson 			panic("aio_free_entry: invalid physio finish-up state");
37311783b14SJohn Dyson 		s = splbio();
374fd3bf775SJohn Dyson 		TAILQ_REMOVE(&ki->kaio_bufdone, aiocbe, plist);
37511783b14SJohn Dyson 		splx(s);
376fd3bf775SJohn Dyson 	} else if ( aiocbe->jobstate == JOBST_JOBQPROC) {
3772244ea07SJohn Dyson 		aiop = aiocbe->jobaioproc;
3782244ea07SJohn Dyson 		TAILQ_REMOVE(&aiop->jobtorun, aiocbe, list);
3792244ea07SJohn Dyson 	} else if ( aiocbe->jobstate == JOBST_JOBQGLOBAL) {
3802244ea07SJohn Dyson 		TAILQ_REMOVE(&aio_jobs, aiocbe, list);
3812244ea07SJohn Dyson 	} else if ( aiocbe->jobstate == JOBST_JOBFINISHED) {
3822244ea07SJohn Dyson 		TAILQ_REMOVE(&ki->kaio_jobdone, aiocbe, plist);
383fd3bf775SJohn Dyson 	} else if ( aiocbe->jobstate == JOBST_JOBBFINISHED) {
38411783b14SJohn Dyson 		s = splbio();
385fd3bf775SJohn Dyson 		TAILQ_REMOVE(&ki->kaio_bufdone, aiocbe, plist);
38611783b14SJohn Dyson 		splx(s);
38784af4da6SJohn Dyson 		if (aiocbe->bp) {
38884af4da6SJohn Dyson 			vunmapbuf(aiocbe->bp);
38984af4da6SJohn Dyson 			relpbuf(aiocbe->bp);
39084af4da6SJohn Dyson 			aiocbe->bp = NULL;
39184af4da6SJohn Dyson 		}
39284af4da6SJohn Dyson 	}
39384af4da6SJohn Dyson 	if (lj && (lj->lioj_buffer_count == 0) && (lj->lioj_queue_count == 0)) {
39484af4da6SJohn Dyson 		TAILQ_REMOVE(&ki->kaio_liojoblist, lj, lioj_list);
39584af4da6SJohn Dyson 		zfree(aiolio_zone, lj);
3962244ea07SJohn Dyson 	}
3972244ea07SJohn Dyson 	TAILQ_INSERT_HEAD(&aio_freejobs, aiocbe, list);
3982244ea07SJohn Dyson 	aiocbe->jobstate = JOBST_NULL;
3992244ea07SJohn Dyson 	return 0;
4002244ea07SJohn Dyson }
4012244ea07SJohn Dyson 
4022244ea07SJohn Dyson /*
4032244ea07SJohn Dyson  * Rundown the jobs for a given process.
4042244ea07SJohn Dyson  */
4052244ea07SJohn Dyson void
406fd3bf775SJohn Dyson aio_proc_rundown(struct proc *p)
407fd3bf775SJohn Dyson {
40884af4da6SJohn Dyson 	int s;
4092244ea07SJohn Dyson 	struct kaioinfo *ki;
41084af4da6SJohn Dyson 	struct aio_liojob *lj, *ljn;
4112244ea07SJohn Dyson 	struct aiocblist *aiocbe, *aiocbn;
4122244ea07SJohn Dyson 
4132244ea07SJohn Dyson 	ki = p->p_aioinfo;
4142244ea07SJohn Dyson 	if (ki == NULL)
4152244ea07SJohn Dyson 		return;
4162244ea07SJohn Dyson 
41784af4da6SJohn Dyson 	ki->kaio_flags |= LIOJ_SIGNAL_POSTED;
41884af4da6SJohn Dyson 	while ((ki->kaio_active_count > 0) ||
41984af4da6SJohn Dyson 		(ki->kaio_buffer_count > ki->kaio_buffer_finished_count)) {
420fd3bf775SJohn Dyson 		ki->kaio_flags |= KAIO_RUNDOWN;
42184af4da6SJohn Dyson 		if (tsleep(p, PRIBIO, "kaiowt", aiod_timeout))
422a624e84fSJohn Dyson 			break;
423a624e84fSJohn Dyson 	}
424a624e84fSJohn Dyson 
4252244ea07SJohn Dyson restart1:
4262244ea07SJohn Dyson 	for ( aiocbe = TAILQ_FIRST(&ki->kaio_jobdone);
4272244ea07SJohn Dyson 		aiocbe;
4282244ea07SJohn Dyson 		aiocbe = aiocbn) {
4292244ea07SJohn Dyson 		aiocbn = TAILQ_NEXT(aiocbe, plist);
4302244ea07SJohn Dyson 		if (aio_free_entry(aiocbe))
4312244ea07SJohn Dyson 			goto restart1;
4322244ea07SJohn Dyson 	}
4332244ea07SJohn Dyson 
4342244ea07SJohn Dyson restart2:
4352244ea07SJohn Dyson 	for ( aiocbe = TAILQ_FIRST(&ki->kaio_jobqueue);
4362244ea07SJohn Dyson 		aiocbe;
4372244ea07SJohn Dyson 		aiocbe = aiocbn) {
4382244ea07SJohn Dyson 		aiocbn = TAILQ_NEXT(aiocbe, plist);
4392244ea07SJohn Dyson 		if (aio_free_entry(aiocbe))
4402244ea07SJohn Dyson 			goto restart2;
4412244ea07SJohn Dyson 	}
44284af4da6SJohn Dyson 
443c5efdcbdSJohn Dyson /*
444c5efdcbdSJohn Dyson  * Note the use of lots of splbio here, trying to avoid
445c5efdcbdSJohn Dyson  * splbio for long chains of I/O.  Probably unnecessary.
446c5efdcbdSJohn Dyson  */
447c5efdcbdSJohn Dyson 
44884af4da6SJohn Dyson restart3:
44984af4da6SJohn Dyson 	s = splbio();
45084af4da6SJohn Dyson 	while (TAILQ_FIRST(&ki->kaio_bufqueue)) {
45184af4da6SJohn Dyson 		ki->kaio_flags |= KAIO_WAKEUP;
45284af4da6SJohn Dyson 		tsleep (p, PRIBIO, "aioprn", 0);
45384af4da6SJohn Dyson 		splx(s);
45484af4da6SJohn Dyson 		goto restart3;
45584af4da6SJohn Dyson 	}
456c5efdcbdSJohn Dyson 	splx(s);
45784af4da6SJohn Dyson 
45884af4da6SJohn Dyson restart4:
45984af4da6SJohn Dyson 	s = splbio();
46084af4da6SJohn Dyson 	for ( aiocbe = TAILQ_FIRST(&ki->kaio_bufdone);
46184af4da6SJohn Dyson 		aiocbe;
46284af4da6SJohn Dyson 		aiocbe = aiocbn) {
46384af4da6SJohn Dyson 		aiocbn = TAILQ_NEXT(aiocbe, plist);
46484af4da6SJohn Dyson 		if (aio_free_entry(aiocbe)) {
46584af4da6SJohn Dyson 			splx(s);
46684af4da6SJohn Dyson 			goto restart4;
46784af4da6SJohn Dyson 		}
46884af4da6SJohn Dyson 	}
46984af4da6SJohn Dyson 	splx(s);
47084af4da6SJohn Dyson 
47184af4da6SJohn Dyson 	for ( lj = TAILQ_FIRST(&ki->kaio_liojoblist);
47284af4da6SJohn Dyson 		  lj;
47384af4da6SJohn Dyson 		  lj = ljn) {
47484af4da6SJohn Dyson 			ljn = TAILQ_NEXT(lj, lioj_list);
47584af4da6SJohn Dyson 			if ((lj->lioj_buffer_count == 0) && (lj->lioj_queue_count == 0)) {
47684af4da6SJohn Dyson 				TAILQ_REMOVE(&ki->kaio_liojoblist, lj, lioj_list);
47784af4da6SJohn Dyson 				zfree(aiolio_zone, lj);
478f4f0ecefSJohn Dyson 			} else {
47911783b14SJohn Dyson #if defined(DIAGNOSTIC)
48084af4da6SJohn Dyson 				printf("LIO job not cleaned up: B:%d, BF:%d, Q:%d, QF:%d\n",
48184af4da6SJohn Dyson 					lj->lioj_buffer_count, lj->lioj_buffer_finished_count,
48284af4da6SJohn Dyson 					lj->lioj_queue_count, lj->lioj_queue_finished_count);
48311783b14SJohn Dyson #endif
48484af4da6SJohn Dyson 			}
485f4f0ecefSJohn Dyson 	}
48684af4da6SJohn Dyson 
487fd3bf775SJohn Dyson 	zfree(kaio_zone, ki);
488a624e84fSJohn Dyson 	p->p_aioinfo = NULL;
4892244ea07SJohn Dyson }
4902244ea07SJohn Dyson 
4912244ea07SJohn Dyson /*
4922244ea07SJohn Dyson  * Select a job to run (called by an AIO daemon)
4932244ea07SJohn Dyson  */
4942244ea07SJohn Dyson static struct aiocblist *
495fd3bf775SJohn Dyson aio_selectjob(struct aioproclist *aiop)
496fd3bf775SJohn Dyson {
4972244ea07SJohn Dyson 
4982244ea07SJohn Dyson 	struct aiocblist *aiocbe;
4992244ea07SJohn Dyson 
5002244ea07SJohn Dyson 	aiocbe = TAILQ_FIRST(&aiop->jobtorun);
5012244ea07SJohn Dyson 	if (aiocbe) {
5022244ea07SJohn Dyson 		TAILQ_REMOVE(&aiop->jobtorun, aiocbe, list);
5032244ea07SJohn Dyson 		return aiocbe;
5042244ea07SJohn Dyson 	}
5052244ea07SJohn Dyson 
5062244ea07SJohn Dyson 	for (aiocbe = TAILQ_FIRST(&aio_jobs);
5072244ea07SJohn Dyson 		aiocbe;
5082244ea07SJohn Dyson 		aiocbe = TAILQ_NEXT(aiocbe, list)) {
5092244ea07SJohn Dyson 		struct kaioinfo *ki;
5102244ea07SJohn Dyson 		struct proc *userp;
5112244ea07SJohn Dyson 
5122244ea07SJohn Dyson 		userp = aiocbe->userproc;
5132244ea07SJohn Dyson 		ki = userp->p_aioinfo;
5142244ea07SJohn Dyson 
5152244ea07SJohn Dyson 		if (ki->kaio_active_count < ki->kaio_maxactive_count) {
5162244ea07SJohn Dyson 			TAILQ_REMOVE(&aio_jobs, aiocbe, list);
5172244ea07SJohn Dyson 			return aiocbe;
5182244ea07SJohn Dyson 		}
5192244ea07SJohn Dyson 	}
5202244ea07SJohn Dyson 
5212244ea07SJohn Dyson 	return NULL;
5222244ea07SJohn Dyson }
5232244ea07SJohn Dyson 
5242244ea07SJohn Dyson /*
525fd3bf775SJohn Dyson  * The AIO processing activity.  This is the code that does the
526fd3bf775SJohn Dyson  * I/O request for the non-physio version of the operations.  The
527fd3bf775SJohn Dyson  * normal vn operations are used, and this code should work in
528fd3bf775SJohn Dyson  * all instances for every type of file, including pipes, sockets,
529fd3bf775SJohn Dyson  * fifos, and regular files.
5302244ea07SJohn Dyson  */
5312244ea07SJohn Dyson void
532fd3bf775SJohn Dyson aio_process(struct aiocblist *aiocbe)
533fd3bf775SJohn Dyson {
5342244ea07SJohn Dyson 	struct filedesc *fdp;
535fd3bf775SJohn Dyson 	struct proc *userp, *mycp;
5362244ea07SJohn Dyson 	struct aiocb *cb;
5372244ea07SJohn Dyson 	struct file *fp;
5382244ea07SJohn Dyson 	struct uio auio;
5392244ea07SJohn Dyson 	struct iovec aiov;
5402244ea07SJohn Dyson 	unsigned int fd;
5412244ea07SJohn Dyson 	int cnt;
5422244ea07SJohn Dyson 	int error;
543a624e84fSJohn Dyson 	off_t offset;
544fd3bf775SJohn Dyson 	int oublock_st, oublock_end;
545fd3bf775SJohn Dyson 	int inblock_st, inblock_end;
5462244ea07SJohn Dyson 
5472244ea07SJohn Dyson 	userp = aiocbe->userproc;
5482244ea07SJohn Dyson 	cb = &aiocbe->uaiocb;
5492244ea07SJohn Dyson 
550fd3bf775SJohn Dyson 	mycp = curproc;
551fd3bf775SJohn Dyson 
552fd3bf775SJohn Dyson 	fdp = mycp->p_fd;
5532244ea07SJohn Dyson 	fd = cb->aio_fildes;
5542244ea07SJohn Dyson 	fp = fdp->fd_ofiles[fd];
5552244ea07SJohn Dyson 
55664889941SJohn Dyson 	aiov.iov_base = (void *) cb->aio_buf;
5572244ea07SJohn Dyson 	aiov.iov_len = cb->aio_nbytes;
5582244ea07SJohn Dyson 
5592244ea07SJohn Dyson 	auio.uio_iov = &aiov;
5602244ea07SJohn Dyson 	auio.uio_iovcnt = 1;
561a624e84fSJohn Dyson 	auio.uio_offset = offset = cb->aio_offset;
5622244ea07SJohn Dyson 	auio.uio_resid = cb->aio_nbytes;
5632244ea07SJohn Dyson 	cnt = cb->aio_nbytes;
5642244ea07SJohn Dyson 	auio.uio_segflg = UIO_USERSPACE;
565fd3bf775SJohn Dyson 	auio.uio_procp = mycp;
5662244ea07SJohn Dyson 
567fd3bf775SJohn Dyson 	inblock_st = mycp->p_stats->p_ru.ru_inblock;
568fd3bf775SJohn Dyson 	oublock_st = mycp->p_stats->p_ru.ru_oublock;
5692244ea07SJohn Dyson 	if (cb->aio_lio_opcode == LIO_READ) {
5702244ea07SJohn Dyson 		auio.uio_rw = UIO_READ;
5712244ea07SJohn Dyson 		error = (*fp->f_ops->fo_read)(fp, &auio, fp->f_cred);
5722244ea07SJohn Dyson 	} else {
5732244ea07SJohn Dyson 		auio.uio_rw = UIO_WRITE;
5742244ea07SJohn Dyson 		error = (*fp->f_ops->fo_write)(fp, &auio, fp->f_cred);
5752244ea07SJohn Dyson 	}
576fd3bf775SJohn Dyson 	inblock_end = mycp->p_stats->p_ru.ru_inblock;
577fd3bf775SJohn Dyson 	oublock_end = mycp->p_stats->p_ru.ru_oublock;
578fd3bf775SJohn Dyson 
579fd3bf775SJohn Dyson 	aiocbe->inputcharge = inblock_end - inblock_st;
580fd3bf775SJohn Dyson 	aiocbe->outputcharge = oublock_end - oublock_st;
5812244ea07SJohn Dyson 
5822244ea07SJohn Dyson 	if (error) {
5832244ea07SJohn Dyson 		if (auio.uio_resid != cnt) {
5842244ea07SJohn Dyson 			if (error == ERESTART || error == EINTR || error == EWOULDBLOCK)
5852244ea07SJohn Dyson 				error = 0;
5862244ea07SJohn Dyson 			if ((error == EPIPE) && (cb->aio_lio_opcode == LIO_WRITE))
5872244ea07SJohn Dyson 				psignal(userp, SIGPIPE);
5882244ea07SJohn Dyson 		}
5892244ea07SJohn Dyson 	}
5902244ea07SJohn Dyson 
5912244ea07SJohn Dyson 	cnt -= auio.uio_resid;
5922244ea07SJohn Dyson 	cb->_aiocb_private.error = error;
5932244ea07SJohn Dyson 	cb->_aiocb_private.status = cnt;
5942244ea07SJohn Dyson 
5952244ea07SJohn Dyson 	return;
5962244ea07SJohn Dyson 
5972244ea07SJohn Dyson }
5982244ea07SJohn Dyson 
5992244ea07SJohn Dyson /*
60084af4da6SJohn Dyson  * The AIO daemon, most of the actual work is done in aio_process,
60184af4da6SJohn Dyson  * but the setup (and address space mgmt) is done in this routine.
6022244ea07SJohn Dyson  */
6032244ea07SJohn Dyson static void
604fd3bf775SJohn Dyson aio_daemon(void *uproc)
6052244ea07SJohn Dyson {
60684af4da6SJohn Dyson 	int s;
6072244ea07SJohn Dyson 	struct aioproclist *aiop;
608fd3bf775SJohn Dyson 	struct vmspace *myvm, *aiovm;
609fd3bf775SJohn Dyson 	struct proc *mycp;
6102244ea07SJohn Dyson 
6112244ea07SJohn Dyson 	/*
612fd3bf775SJohn Dyson 	 * Local copies of curproc (cp) and vmspace (myvm)
6132244ea07SJohn Dyson 	 */
614fd3bf775SJohn Dyson 	mycp = curproc;
615fd3bf775SJohn Dyson 	myvm = mycp->p_vmspace;
616fd3bf775SJohn Dyson 
617fd3bf775SJohn Dyson 	/*
618fd3bf775SJohn Dyson 	 * We manage to create only one VM space for all AIOD processes.
619fd3bf775SJohn Dyson 	 * The VM space for the first AIOD created becomes the shared VM
620fd3bf775SJohn Dyson 	 * space for all of them.  We add an additional reference count,
621fd3bf775SJohn Dyson 	 * even for the first AIOD, so the address space does not go away,
622fd3bf775SJohn Dyson 	 * and we continue to use that original VM space even if the first
623fd3bf775SJohn Dyson 	 * AIOD exits.
624fd3bf775SJohn Dyson 	 */
625fd3bf775SJohn Dyson 	if ((aiovm = aiovmspace) == NULL) {
626fd3bf775SJohn Dyson 		aiovmspace = myvm;
62784af4da6SJohn Dyson 		myvm->vm_refcnt++;
628fd3bf775SJohn Dyson 		/*
629fd3bf775SJohn Dyson 		 * Remove userland cruft from address space.
630fd3bf775SJohn Dyson 		 */
631fd3bf775SJohn Dyson 		if (myvm->vm_shm)
632fd3bf775SJohn Dyson 			shmexit(mycp);
633fd3bf775SJohn Dyson 		pmap_remove_pages(&myvm->vm_pmap, 0, USRSTACK);
634fd3bf775SJohn Dyson 		vm_map_remove(&myvm->vm_map, 0, USRSTACK);
635fd3bf775SJohn Dyson 		myvm->vm_tsize = 0;
636fd3bf775SJohn Dyson 		myvm->vm_dsize = 0;
637fd3bf775SJohn Dyson 		myvm->vm_ssize = 0;
638fd3bf775SJohn Dyson 	} else {
63984af4da6SJohn Dyson 		aiovm->vm_refcnt++;
640fd3bf775SJohn Dyson 		mycp->p_vmspace = aiovm;
641fd3bf775SJohn Dyson 		pmap_activate(mycp);
642fd3bf775SJohn Dyson 		vmspace_free(myvm);
643fd3bf775SJohn Dyson 		myvm = aiovm;
644fd3bf775SJohn Dyson 	}
645fd3bf775SJohn Dyson 
646fd3bf775SJohn Dyson 	if (mycp->p_textvp) {
647fd3bf775SJohn Dyson 		vrele(mycp->p_textvp);
648fd3bf775SJohn Dyson 		mycp->p_textvp = NULL;
649fd3bf775SJohn Dyson 	}
650fd3bf775SJohn Dyson 
651fd3bf775SJohn Dyson 	/*
652fd3bf775SJohn Dyson 	 * Allocate and ready the aio control info.  There is one
653fd3bf775SJohn Dyson 	 * aiop structure per daemon.
654fd3bf775SJohn Dyson 	 */
655fd3bf775SJohn Dyson 	aiop = zalloc(aiop_zone);
656fd3bf775SJohn Dyson 	aiop->aioproc = mycp;
6572244ea07SJohn Dyson 	aiop->aioprocflags |= AIOP_FREE;
6582244ea07SJohn Dyson 	TAILQ_INIT(&aiop->jobtorun);
6592244ea07SJohn Dyson 
6602244ea07SJohn Dyson 	/*
661fd3bf775SJohn Dyson 	 * Place thread (lightweight process) onto the AIO free thread list
6622244ea07SJohn Dyson 	 */
663fd3bf775SJohn Dyson 	if (TAILQ_EMPTY(&aio_freeproc))
664fd3bf775SJohn Dyson 		wakeup(&aio_freeproc);
665fd3bf775SJohn Dyson 	TAILQ_INSERT_HEAD(&aio_freeproc, aiop, list);
6662244ea07SJohn Dyson 
6672244ea07SJohn Dyson 	/*
6682244ea07SJohn Dyson 	 * Make up a name for the daemon
6692244ea07SJohn Dyson 	 */
670fd3bf775SJohn Dyson 	strcpy(mycp->p_comm, "aiod");
6712244ea07SJohn Dyson 
6722244ea07SJohn Dyson 	/*
673fd3bf775SJohn Dyson 	 * Get rid of our current filedescriptors.  AIOD's don't need any
674fd3bf775SJohn Dyson 	 * filedescriptors, except as temporarily inherited from the client.
675fd3bf775SJohn Dyson 	 * Credentials are also cloned, and made equivalent to "root."
6762244ea07SJohn Dyson 	 */
677fd3bf775SJohn Dyson 	fdfree(mycp);
678fd3bf775SJohn Dyson 	mycp->p_fd = NULL;
679fd3bf775SJohn Dyson 	mycp->p_ucred = crcopy(mycp->p_ucred);
680fd3bf775SJohn Dyson 	mycp->p_ucred->cr_uid = 0;
681fd3bf775SJohn Dyson 	mycp->p_ucred->cr_ngroups = 1;
682fd3bf775SJohn Dyson 	mycp->p_ucred->cr_groups[0] = 1;
683fd3bf775SJohn Dyson 
684fd3bf775SJohn Dyson 	/*
685dc733423SDag-Erling Smørgrav 	 * The daemon resides in its own pgrp.
686fd3bf775SJohn Dyson 	 */
687fd3bf775SJohn Dyson 	enterpgrp(mycp, mycp->p_pid, 1);
688fd3bf775SJohn Dyson 
689fd3bf775SJohn Dyson 	/*
690fd3bf775SJohn Dyson 	 * Mark special process type
691fd3bf775SJohn Dyson 	 */
692fd3bf775SJohn Dyson 	mycp->p_flag |= P_SYSTEM|P_KTHREADP;
6932244ea07SJohn Dyson 
694fd3bf775SJohn Dyson 	/*
695fd3bf775SJohn Dyson 	 * Wakeup parent process.  (Parent sleeps to keep from blasting away
696fd3bf775SJohn Dyson 	 * creating to many daemons.)
697fd3bf775SJohn Dyson 	 */
698fd3bf775SJohn Dyson 	wakeup(mycp);
6992244ea07SJohn Dyson 
7002244ea07SJohn Dyson 	while(1) {
701fd3bf775SJohn Dyson 		struct proc *curcp;
7022244ea07SJohn Dyson 		struct	aiocblist *aiocbe;
7032244ea07SJohn Dyson 
704fd3bf775SJohn Dyson 		/*
705fd3bf775SJohn Dyson 		 * curcp is the current daemon process context.
706fd3bf775SJohn Dyson 		 * userp is the current user process context.
707fd3bf775SJohn Dyson 		 */
708fd3bf775SJohn Dyson 		curcp = mycp;
709c4860686SJohn Dyson 
710fd3bf775SJohn Dyson 		/*
711fd3bf775SJohn Dyson 		 * Take daemon off of free queue
712fd3bf775SJohn Dyson 		 */
7132244ea07SJohn Dyson 		if (aiop->aioprocflags & AIOP_FREE) {
7142244ea07SJohn Dyson 			TAILQ_REMOVE(&aio_freeproc, aiop, list);
7152244ea07SJohn Dyson 			TAILQ_INSERT_TAIL(&aio_activeproc, aiop, list);
7162244ea07SJohn Dyson 			aiop->aioprocflags &= ~AIOP_FREE;
7172244ea07SJohn Dyson 		}
718fd3bf775SJohn Dyson 		aiop->aioprocflags &= ~AIOP_SCHED;
7192244ea07SJohn Dyson 
720fd3bf775SJohn Dyson 		/*
721fd3bf775SJohn Dyson 		 * Check for jobs
722fd3bf775SJohn Dyson 		 */
7232244ea07SJohn Dyson 		while ( aiocbe = aio_selectjob(aiop)) {
724fd3bf775SJohn Dyson 			struct proc *userp;
7252244ea07SJohn Dyson 			struct aiocb *cb;
7262244ea07SJohn Dyson 			struct kaioinfo *ki;
72784af4da6SJohn Dyson 			struct aio_liojob *lj;
7282244ea07SJohn Dyson 
7292244ea07SJohn Dyson 			cb = &aiocbe->uaiocb;
7302244ea07SJohn Dyson 			userp = aiocbe->userproc;
7312244ea07SJohn Dyson 
7322244ea07SJohn Dyson 			aiocbe->jobstate = JOBST_JOBRUNNING;
733fd3bf775SJohn Dyson 
734fd3bf775SJohn Dyson 			/*
735fd3bf775SJohn Dyson 			 * Connect to process address space for user program
736fd3bf775SJohn Dyson 			 */
737fd3bf775SJohn Dyson 			if (userp != curcp) {
738fd3bf775SJohn Dyson 				struct vmspace *tmpvm;
739fd3bf775SJohn Dyson 				/*
740fd3bf775SJohn Dyson 				 * Save the current address space that we are connected to.
741fd3bf775SJohn Dyson 				 */
742fd3bf775SJohn Dyson 				tmpvm = mycp->p_vmspace;
743fd3bf775SJohn Dyson 				/*
744fd3bf775SJohn Dyson 				 * Point to the new user address space, and refer to it.
745fd3bf775SJohn Dyson 				 */
746fd3bf775SJohn Dyson 				mycp->p_vmspace = userp->p_vmspace;
74784af4da6SJohn Dyson 				mycp->p_vmspace->vm_refcnt++;
748fd3bf775SJohn Dyson 				/*
749fd3bf775SJohn Dyson 				 * Activate the new mapping.
750fd3bf775SJohn Dyson 				 */
751fd3bf775SJohn Dyson 				pmap_activate(mycp);
752fd3bf775SJohn Dyson 				/*
753fd3bf775SJohn Dyson 				 * If the old address space wasn't the daemons own address
754fd3bf775SJohn Dyson 				 * space, then we need to remove the daemon's reference from
755fd3bf775SJohn Dyson 				 * the other process that it was acting on behalf of.
756fd3bf775SJohn Dyson 				 */
7572244ea07SJohn Dyson 				if (tmpvm != myvm) {
7582244ea07SJohn Dyson 					vmspace_free(tmpvm);
7592244ea07SJohn Dyson 				}
760fd3bf775SJohn Dyson 				/*
761fd3bf775SJohn Dyson 				 * Disassociate from previous clients file descriptors, and
762fd3bf775SJohn Dyson 				 * associate to the new clients descriptors.  Note that
763dc733423SDag-Erling Smørgrav 				 * the daemon doesn't need to worry about its orginal
764fd3bf775SJohn Dyson 				 * descriptors, because they were originally freed.
765fd3bf775SJohn Dyson 				 */
766fd3bf775SJohn Dyson 				if (mycp->p_fd)
767fd3bf775SJohn Dyson 					fdfree(mycp);
768fd3bf775SJohn Dyson 				mycp->p_fd = fdshare(userp);
769fd3bf775SJohn Dyson 				curcp = userp;
7702244ea07SJohn Dyson 			}
7712244ea07SJohn Dyson 
772fd3bf775SJohn Dyson 			ki = userp->p_aioinfo;
77384af4da6SJohn Dyson 			lj = aiocbe->lio;
77484af4da6SJohn Dyson 
77584af4da6SJohn Dyson 			/*
77684af4da6SJohn Dyson 			 * Account for currently active jobs
77784af4da6SJohn Dyson 			 */
7782244ea07SJohn Dyson 			ki->kaio_active_count++;
77911783b14SJohn Dyson 
78084af4da6SJohn Dyson 			/*
78184af4da6SJohn Dyson 			 * Do the I/O function
78284af4da6SJohn Dyson 			 */
7832244ea07SJohn Dyson 			aiocbe->jobaioproc = aiop;
7842244ea07SJohn Dyson 			aio_process(aiocbe);
78584af4da6SJohn Dyson 
78684af4da6SJohn Dyson 			/*
78784af4da6SJohn Dyson 			 * decrement the active job count
78884af4da6SJohn Dyson 			 */
78984af4da6SJohn Dyson 			ki->kaio_active_count--;
79084af4da6SJohn Dyson 
79184af4da6SJohn Dyson 			/*
79284af4da6SJohn Dyson 			 * increment the completion count for wakeup/signal comparisons
79384af4da6SJohn Dyson 			 */
79484af4da6SJohn Dyson 			aiocbe->jobflags |= AIOCBLIST_DONE;
79584af4da6SJohn Dyson 			ki->kaio_queue_finished_count++;
79684af4da6SJohn Dyson 			if (lj) {
79784af4da6SJohn Dyson 				lj->lioj_queue_finished_count++;
79884af4da6SJohn Dyson 			}
799fd3bf775SJohn Dyson 			if ((ki->kaio_flags & KAIO_WAKEUP) ||
80084af4da6SJohn Dyson 				(ki->kaio_flags & KAIO_RUNDOWN) &&
80184af4da6SJohn Dyson 				(ki->kaio_active_count == 0)) {
802fd3bf775SJohn Dyson 				ki->kaio_flags &= ~KAIO_WAKEUP;
803fd3bf775SJohn Dyson 				wakeup(userp);
804fd3bf775SJohn Dyson 			}
8052244ea07SJohn Dyson 
80684af4da6SJohn Dyson 			s = splbio();
80784af4da6SJohn Dyson 			if (lj && (lj->lioj_flags & (LIOJ_SIGNAL|LIOJ_SIGNAL_POSTED)) ==
80884af4da6SJohn Dyson 				LIOJ_SIGNAL) {
80984af4da6SJohn Dyson 				if ((lj->lioj_queue_finished_count == lj->lioj_queue_count) &&
81084af4da6SJohn Dyson 					(lj->lioj_buffer_finished_count == lj->lioj_buffer_count)) {
81184af4da6SJohn Dyson 						psignal(userp, lj->lioj_signal.sigev_signo);
81284af4da6SJohn Dyson 						lj->lioj_flags |= LIOJ_SIGNAL_POSTED;
81384af4da6SJohn Dyson 				}
81484af4da6SJohn Dyson 			}
81584af4da6SJohn Dyson 			splx(s);
81684af4da6SJohn Dyson 
8172244ea07SJohn Dyson 			aiocbe->jobstate = JOBST_JOBFINISHED;
8182244ea07SJohn Dyson 
819fd3bf775SJohn Dyson 			/*
820fd3bf775SJohn Dyson 			 * If the I/O request should be automatically rundown, do the
821fd3bf775SJohn Dyson 			 * needed cleanup.  Otherwise, place the queue entry for
822fd3bf775SJohn Dyson 			 * the just finished I/O request into the done queue for the
823fd3bf775SJohn Dyson 			 * associated client.
824fd3bf775SJohn Dyson 			 */
8252244ea07SJohn Dyson 			if (aiocbe->jobflags & AIOCBLIST_ASYNCFREE) {
8262244ea07SJohn Dyson 				aiocbe->jobflags &= ~AIOCBLIST_ASYNCFREE;
8272244ea07SJohn Dyson 				TAILQ_INSERT_HEAD(&aio_freejobs, aiocbe, list);
8282244ea07SJohn Dyson 			} else {
8292244ea07SJohn Dyson 				TAILQ_REMOVE(&ki->kaio_jobqueue,
8302244ea07SJohn Dyson 					aiocbe, plist);
8312244ea07SJohn Dyson 				TAILQ_INSERT_TAIL(&ki->kaio_jobdone,
8322244ea07SJohn Dyson 					aiocbe, plist);
8332244ea07SJohn Dyson 			}
8342244ea07SJohn Dyson 
8352244ea07SJohn Dyson 			if (aiocbe->jobflags & AIOCBLIST_RUNDOWN) {
8362244ea07SJohn Dyson 				wakeup(aiocbe);
8372244ea07SJohn Dyson 				aiocbe->jobflags &= ~AIOCBLIST_RUNDOWN;
8382244ea07SJohn Dyson 			}
8392244ea07SJohn Dyson 
8402244ea07SJohn Dyson 			if (cb->aio_sigevent.sigev_notify == SIGEV_SIGNAL) {
8412244ea07SJohn Dyson 				psignal(userp, cb->aio_sigevent.sigev_signo);
8422244ea07SJohn Dyson 			}
8432244ea07SJohn Dyson 		}
8442244ea07SJohn Dyson 
845fd3bf775SJohn Dyson 		/*
846fd3bf775SJohn Dyson 		 * Disconnect from user address space
847fd3bf775SJohn Dyson 		 */
848fd3bf775SJohn Dyson 		if (curcp != mycp) {
849fd3bf775SJohn Dyson 			struct vmspace *tmpvm;
850fd3bf775SJohn Dyson 			/*
851fd3bf775SJohn Dyson 			 * Get the user address space to disconnect from.
852fd3bf775SJohn Dyson 			 */
853fd3bf775SJohn Dyson 			tmpvm = mycp->p_vmspace;
854fd3bf775SJohn Dyson 			/*
855fd3bf775SJohn Dyson 			 * Get original address space for daemon.
856fd3bf775SJohn Dyson 			 */
857fd3bf775SJohn Dyson 			mycp->p_vmspace = myvm;
858fd3bf775SJohn Dyson 			/*
859fd3bf775SJohn Dyson 			 * Activate the daemon's address space.
860fd3bf775SJohn Dyson 			 */
861fd3bf775SJohn Dyson 			pmap_activate(mycp);
86211783b14SJohn Dyson #if defined(DIAGNOSTIC)
863fd3bf775SJohn Dyson 			if (tmpvm == myvm)
864fd3bf775SJohn Dyson 				printf("AIOD: vmspace problem -- %d\n", mycp->p_pid);
86511783b14SJohn Dyson #endif
866fd3bf775SJohn Dyson 			/*
867fd3bf775SJohn Dyson 			 * remove our vmspace reference.
868fd3bf775SJohn Dyson 			 */
8692244ea07SJohn Dyson 			vmspace_free(tmpvm);
870fd3bf775SJohn Dyson 			/*
871fd3bf775SJohn Dyson 			 * disassociate from the user process's file descriptors.
872fd3bf775SJohn Dyson 			 */
873fd3bf775SJohn Dyson 			if (mycp->p_fd)
874fd3bf775SJohn Dyson 				fdfree(mycp);
875fd3bf775SJohn Dyson 			mycp->p_fd = NULL;
876fd3bf775SJohn Dyson 			curcp = mycp;
877fd3bf775SJohn Dyson 		}
878fd3bf775SJohn Dyson 
879fd3bf775SJohn Dyson 		/*
880fd3bf775SJohn Dyson 		 * If we are the first to be put onto the free queue, wakeup
881fd3bf775SJohn Dyson 		 * anyone waiting for a daemon.
882fd3bf775SJohn Dyson 		 */
883fd3bf775SJohn Dyson 		TAILQ_REMOVE(&aio_activeproc, aiop, list);
884fd3bf775SJohn Dyson 		if (TAILQ_EMPTY(&aio_freeproc))
885fd3bf775SJohn Dyson 			wakeup(&aio_freeproc);
886fd3bf775SJohn Dyson 		TAILQ_INSERT_HEAD(&aio_freeproc, aiop, list);
887fd3bf775SJohn Dyson 		aiop->aioprocflags |= AIOP_FREE;
888fd3bf775SJohn Dyson 
889fd3bf775SJohn Dyson 		/*
890fd3bf775SJohn Dyson 		 * If daemon is inactive for a long time, allow it to exit, thereby
891fd3bf775SJohn Dyson 		 * freeing resources.
892fd3bf775SJohn Dyson 		 */
893fd3bf775SJohn Dyson 		if (((aiop->aioprocflags & AIOP_SCHED) == 0) &&
89484af4da6SJohn Dyson 			tsleep(mycp, PRIBIO, "aiordy", aiod_lifetime)) {
895fd3bf775SJohn Dyson 			if ((TAILQ_FIRST(&aio_jobs) == NULL) &&
896fd3bf775SJohn Dyson 				(TAILQ_FIRST(&aiop->jobtorun) == NULL)) {
89784af4da6SJohn Dyson 				if ((aiop->aioprocflags & AIOP_FREE) &&
89884af4da6SJohn Dyson 					(num_aio_procs > target_aio_procs)) {
899fd3bf775SJohn Dyson 					TAILQ_REMOVE(&aio_freeproc, aiop, list);
900fd3bf775SJohn Dyson 					zfree(aiop_zone, aiop);
90184af4da6SJohn Dyson 					num_aio_procs--;
90211783b14SJohn Dyson #if defined(DIAGNOSTIC)
903fd3bf775SJohn Dyson 					if (mycp->p_vmspace->vm_refcnt <= 1)
904fd3bf775SJohn Dyson 						printf("AIOD: bad vm refcnt for exiting daemon: %d\n",
905fd3bf775SJohn Dyson 							mycp->p_vmspace->vm_refcnt);
90611783b14SJohn Dyson #endif
907fd3bf775SJohn Dyson 					exit1(mycp, 0);
908fd3bf775SJohn Dyson 				}
909fd3bf775SJohn Dyson 			}
9102244ea07SJohn Dyson 		}
9112244ea07SJohn Dyson 	}
9122244ea07SJohn Dyson }
9132244ea07SJohn Dyson 
9142244ea07SJohn Dyson /*
91584af4da6SJohn Dyson  * Create a new AIO daemon.  This is mostly a kernel-thread fork routine.
916dc733423SDag-Erling Smørgrav  * The AIO daemon modifies its environment itself.
9172244ea07SJohn Dyson  */
9182244ea07SJohn Dyson static int
919fd3bf775SJohn Dyson aio_newproc()
920fd3bf775SJohn Dyson {
9212244ea07SJohn Dyson 	int error;
9222244ea07SJohn Dyson 	struct rfork_args rfa;
923fd3bf775SJohn Dyson 	struct proc *p, *np;
9242244ea07SJohn Dyson 
925fd3bf775SJohn Dyson 	rfa.flags = RFPROC | RFCFDG;
9262244ea07SJohn Dyson 
927cb226aaaSPoul-Henning Kamp 	p = curproc;
928cb226aaaSPoul-Henning Kamp 	if (error = rfork(p, &rfa))
9292244ea07SJohn Dyson 		return error;
930fd3bf775SJohn Dyson 
931fd3bf775SJohn Dyson 	np = pfind(p->p_retval[0]);
932fd3bf775SJohn Dyson 	cpu_set_fork_handler(np, aio_daemon, p);
9332244ea07SJohn Dyson 
934fd3bf775SJohn Dyson 	/*
935fd3bf775SJohn Dyson 	 * Wait until daemon is started, but continue on just in case (to
936fd3bf775SJohn Dyson 	 * handle error conditions.
937fd3bf775SJohn Dyson 	 */
93884af4da6SJohn Dyson 	error = tsleep(np, PZERO, "aiosta", aiod_timeout);
93984af4da6SJohn Dyson 	num_aio_procs++;
9402244ea07SJohn Dyson 
9412244ea07SJohn Dyson 	return error;
9422244ea07SJohn Dyson 
9432244ea07SJohn Dyson }
9442244ea07SJohn Dyson 
9452244ea07SJohn Dyson /*
94684af4da6SJohn Dyson  * Try the high-performance physio method for eligible VCHR devices.  This
94784af4da6SJohn Dyson  * routine doesn't require the use of any additional threads, and have
94884af4da6SJohn Dyson  * overhead.
949fd3bf775SJohn Dyson  */
950fd3bf775SJohn Dyson int
95184af4da6SJohn Dyson aio_qphysio(p, aiocbe)
952fd3bf775SJohn Dyson 	struct proc *p;
95384af4da6SJohn Dyson 	struct aiocblist *aiocbe;
954fd3bf775SJohn Dyson {
955fd3bf775SJohn Dyson 	int error;
956fd3bf775SJohn Dyson 	caddr_t sa;
957fd3bf775SJohn Dyson 	struct aiocb *cb;
958fd3bf775SJohn Dyson 	struct file *fp;
959fd3bf775SJohn Dyson 	struct buf *bp;
960fd3bf775SJohn Dyson 	int bflags;
961fd3bf775SJohn Dyson 	struct vnode *vp;
962fd3bf775SJohn Dyson 	struct kaioinfo *ki;
963fd3bf775SJohn Dyson 	struct filedesc *fdp;
96484af4da6SJohn Dyson 	struct aio_liojob *lj;
965fd3bf775SJohn Dyson 	int fd;
966fd3bf775SJohn Dyson 	int majordev;
967fd3bf775SJohn Dyson 	int s;
968fd3bf775SJohn Dyson 	int cnt;
969fd3bf775SJohn Dyson 	dev_t dev;
970fd3bf775SJohn Dyson 	int rw;
971fd3bf775SJohn Dyson 	d_strategy_t *fstrategy;
97284af4da6SJohn Dyson 	struct cdevsw *cdev;
973f7ea2f55SJulian Elischer 	struct cdevsw *bdev;
974fd3bf775SJohn Dyson 
97584af4da6SJohn Dyson 	cb = &aiocbe->uaiocb;
976fd3bf775SJohn Dyson 	fdp = p->p_fd;
977fd3bf775SJohn Dyson 	fd = cb->aio_fildes;
978fd3bf775SJohn Dyson 	fp = fdp->fd_ofiles[fd];
979fd3bf775SJohn Dyson 
98084af4da6SJohn Dyson 	if (fp->f_type != DTYPE_VNODE) {
981fd3bf775SJohn Dyson 		return -1;
98284af4da6SJohn Dyson 	}
983fd3bf775SJohn Dyson 
984fd3bf775SJohn Dyson 	vp = (struct vnode *)fp->f_data;
98584af4da6SJohn Dyson 	if (vp->v_type != VCHR || ((cb->aio_nbytes & (DEV_BSIZE - 1)) != 0)) {
98611783b14SJohn Dyson 		return -1;
98784af4da6SJohn Dyson 	}
98811783b14SJohn Dyson 
98911783b14SJohn Dyson 	if ((cb->aio_nbytes > MAXPHYS) && (num_buf_aio >= max_buf_aio)) {
990fd3bf775SJohn Dyson 		return -1;
99184af4da6SJohn Dyson 	}
992fd3bf775SJohn Dyson 
99384af4da6SJohn Dyson 	if ((vp->v_specinfo == NULL) || (vp->v_flag & VISTTY)) {
994fd3bf775SJohn Dyson 		return -1;
99584af4da6SJohn Dyson 	}
996fd3bf775SJohn Dyson 
997fd3bf775SJohn Dyson 	majordev = major(vp->v_rdev);
99884af4da6SJohn Dyson 	if (majordev == NODEV) {
999fd3bf775SJohn Dyson 		return -1;
100084af4da6SJohn Dyson 	}
1001fd3bf775SJohn Dyson 
100284af4da6SJohn Dyson 	cdev = cdevsw[major(vp->v_rdev)];
100384af4da6SJohn Dyson 	if (cdev == NULL) {
1004fd3bf775SJohn Dyson 		return -1;
100584af4da6SJohn Dyson 	}
1006f7ea2f55SJulian Elischer 
1007f7ea2f55SJulian Elischer 	if (cdev->d_bmaj == -1) {
100884af4da6SJohn Dyson 		return -1;
100984af4da6SJohn Dyson 	}
1010f7ea2f55SJulian Elischer 	bdev = cdev;
1011fd3bf775SJohn Dyson 
1012fd3bf775SJohn Dyson 	ki = p->p_aioinfo;
101384af4da6SJohn Dyson 	if (ki->kaio_buffer_count >= ki->kaio_ballowed_count) {
1014fd3bf775SJohn Dyson 		return -1;
101584af4da6SJohn Dyson 	}
1016fd3bf775SJohn Dyson 
1017fd3bf775SJohn Dyson 	cnt = cb->aio_nbytes;
101884af4da6SJohn Dyson 	if (cnt > MAXPHYS) {
1019fd3bf775SJohn Dyson 		return -1;
102084af4da6SJohn Dyson 	}
102184af4da6SJohn Dyson 
1022596f8506SJulian Elischer 	dev = makedev(bdev->d_bmaj, minor(vp->v_rdev));
102384af4da6SJohn Dyson 
102484af4da6SJohn Dyson 	/*
102584af4da6SJohn Dyson 	 * Physical I/O is charged directly to the process, so we don't have
102684af4da6SJohn Dyson 	 * to fake it.
102784af4da6SJohn Dyson 	 */
102884af4da6SJohn Dyson 	aiocbe->inputcharge = 0;
102984af4da6SJohn Dyson 	aiocbe->outputcharge = 0;
1030fd3bf775SJohn Dyson 
1031fd3bf775SJohn Dyson 	ki->kaio_buffer_count++;
103211783b14SJohn Dyson 
103311783b14SJohn Dyson 	lj = aiocbe->lio;
103484af4da6SJohn Dyson 	if (lj) {
103584af4da6SJohn Dyson 		lj->lioj_buffer_count++;
103684af4da6SJohn Dyson 	}
1037fd3bf775SJohn Dyson 
1038fd3bf775SJohn Dyson 	/* create and build a buffer header for a transfer */
1039fd3bf775SJohn Dyson 	bp = (struct buf *)getpbuf();
1040fd3bf775SJohn Dyson 
1041fd3bf775SJohn Dyson 	/*
1042fd3bf775SJohn Dyson 	 * get a copy of the kva from the physical buffer
1043fd3bf775SJohn Dyson 	 */
1044fd3bf775SJohn Dyson 	bp->b_proc = p;
1045fd3bf775SJohn Dyson 	bp->b_dev = dev;
1046fd3bf775SJohn Dyson 	error = bp->b_error = 0;
1047fd3bf775SJohn Dyson 
1048fd3bf775SJohn Dyson 	if (cb->aio_lio_opcode == LIO_WRITE) {
1049fd3bf775SJohn Dyson 		rw = 0;
1050fd3bf775SJohn Dyson 		bflags = B_WRITE;
1051fd3bf775SJohn Dyson 	} else {
1052fd3bf775SJohn Dyson 		rw = 1;
1053fd3bf775SJohn Dyson 		bflags = B_READ;
1054fd3bf775SJohn Dyson 	}
1055fd3bf775SJohn Dyson 
1056fd3bf775SJohn Dyson 	bp->b_bcount = cb->aio_nbytes;
1057fd3bf775SJohn Dyson 	bp->b_bufsize = cb->aio_nbytes;
1058fd3bf775SJohn Dyson 	bp->b_flags = B_BUSY | B_PHYS | B_CALL | bflags;
1059fd3bf775SJohn Dyson 	bp->b_iodone = aio_physwakeup;
1060fd3bf775SJohn Dyson 	bp->b_saveaddr = bp->b_data;
106164889941SJohn Dyson 	bp->b_data = (void *) cb->aio_buf;
1062fd3bf775SJohn Dyson 	bp->b_blkno = btodb(cb->aio_offset);
1063fd3bf775SJohn Dyson 
1064fd3bf775SJohn Dyson 	if (rw && !useracc(bp->b_data, bp->b_bufsize, B_WRITE)) {
1065fd3bf775SJohn Dyson 		error = EFAULT;
1066fd3bf775SJohn Dyson 		goto doerror;
1067fd3bf775SJohn Dyson 	}
1068fd3bf775SJohn Dyson 	if (!rw && !useracc(bp->b_data, bp->b_bufsize, B_READ)) {
1069fd3bf775SJohn Dyson 		error = EFAULT;
1070fd3bf775SJohn Dyson 		goto doerror;
1071fd3bf775SJohn Dyson 	}
1072fd3bf775SJohn Dyson 
1073fd3bf775SJohn Dyson 	/* bring buffer into kernel space */
1074fd3bf775SJohn Dyson 	vmapbuf(bp);
1075fd3bf775SJohn Dyson 
107684af4da6SJohn Dyson 	s = splbio();
1077fd3bf775SJohn Dyson 	aiocbe->bp = bp;
1078fd3bf775SJohn Dyson 	bp->b_spc = (void *)aiocbe;
1079fd3bf775SJohn Dyson 	TAILQ_INSERT_TAIL(&aio_bufjobs, aiocbe, list);
108084af4da6SJohn Dyson 	TAILQ_INSERT_TAIL(&ki->kaio_bufqueue, aiocbe, plist);
1081fd3bf775SJohn Dyson 	aiocbe->jobstate = JOBST_JOBQBUF;
108284af4da6SJohn Dyson 	cb->_aiocb_private.status = cb->aio_nbytes;
108384af4da6SJohn Dyson 	num_buf_aio++;
108484af4da6SJohn Dyson 	fstrategy = bdev->d_strategy;
1085fd3bf775SJohn Dyson 	bp->b_error = 0;
1086fd3bf775SJohn Dyson 
108784af4da6SJohn Dyson 	splx(s);
1088fd3bf775SJohn Dyson 	/* perform transfer */
1089fd3bf775SJohn Dyson 	(*fstrategy)(bp);
1090fd3bf775SJohn Dyson 
109184af4da6SJohn Dyson 	s = splbio();
109211783b14SJohn Dyson 	/*
109311783b14SJohn Dyson 	 * If we had an error invoking the request, or an error in processing
109411783b14SJohn Dyson 	 * the request before we have returned, we process it as an error
109511783b14SJohn Dyson 	 * in transfer.  Note that such an I/O error is not indicated immediately,
109611783b14SJohn Dyson 	 * but is returned using the aio_error mechanism.  In this case, aio_suspend
109711783b14SJohn Dyson 	 * will return immediately.
109811783b14SJohn Dyson 	 */
109911783b14SJohn Dyson 	if (bp->b_error || (bp->b_flags & B_ERROR)) {
110011783b14SJohn Dyson 		struct aiocb *job = aiocbe->uuaiocb;
110111783b14SJohn Dyson 
110211783b14SJohn Dyson 		aiocbe->uaiocb._aiocb_private.status = 0;
110311783b14SJohn Dyson 		suword(&job->_aiocb_private.status, 0);
110411783b14SJohn Dyson 		aiocbe->uaiocb._aiocb_private.error = bp->b_error;
110511783b14SJohn Dyson 		suword(&job->_aiocb_private.error, bp->b_error);
110611783b14SJohn Dyson 
110711783b14SJohn Dyson 		ki->kaio_buffer_finished_count++;
110811783b14SJohn Dyson 
110911783b14SJohn Dyson 		if (aiocbe->jobstate != JOBST_JOBBFINISHED) {
111011783b14SJohn Dyson 			aiocbe->jobstate = JOBST_JOBBFINISHED;
111111783b14SJohn Dyson 			aiocbe->jobflags |= AIOCBLIST_DONE;
1112fd3bf775SJohn Dyson 			TAILQ_REMOVE(&aio_bufjobs, aiocbe, list);
111384af4da6SJohn Dyson 			TAILQ_REMOVE(&ki->kaio_bufqueue, aiocbe, plist);
111411783b14SJohn Dyson 			TAILQ_INSERT_TAIL(&ki->kaio_bufdone, aiocbe, plist);
111511783b14SJohn Dyson 		}
111611783b14SJohn Dyson 	}
111784af4da6SJohn Dyson 	splx(s);
1118fd3bf775SJohn Dyson 	return 0;
1119fd3bf775SJohn Dyson 
1120fd3bf775SJohn Dyson doerror:
1121fd3bf775SJohn Dyson 	ki->kaio_buffer_count--;
112284af4da6SJohn Dyson 	if (lj) {
112384af4da6SJohn Dyson 		lj->lioj_buffer_count--;
112484af4da6SJohn Dyson 	}
112584af4da6SJohn Dyson 	aiocbe->bp = NULL;
1126fd3bf775SJohn Dyson 	relpbuf(bp);
1127fd3bf775SJohn Dyson 	return error;
1128fd3bf775SJohn Dyson }
1129fd3bf775SJohn Dyson 
113084af4da6SJohn Dyson /*
113184af4da6SJohn Dyson  * This waits/tests physio completion.
113284af4da6SJohn Dyson  */
1133fd3bf775SJohn Dyson int
1134fd3bf775SJohn Dyson aio_fphysio(p, iocb, flgwait)
1135fd3bf775SJohn Dyson 	struct proc *p;
1136fd3bf775SJohn Dyson 	struct aiocblist *iocb;
1137fd3bf775SJohn Dyson 	int flgwait;
1138fd3bf775SJohn Dyson {
1139fd3bf775SJohn Dyson 	int s;
1140fd3bf775SJohn Dyson 	struct buf *bp;
1141fd3bf775SJohn Dyson 	int error;
1142fd3bf775SJohn Dyson 
1143fd3bf775SJohn Dyson 	bp = iocb->bp;
1144fd3bf775SJohn Dyson 
1145fd3bf775SJohn Dyson 	s = splbio();
1146fd3bf775SJohn Dyson 	if (flgwait == 0) {
1147fd3bf775SJohn Dyson 		if ((bp->b_flags & B_DONE) == 0) {
1148fd3bf775SJohn Dyson 			splx(s);
1149fd3bf775SJohn Dyson 			return EINPROGRESS;
1150fd3bf775SJohn Dyson 		}
1151fd3bf775SJohn Dyson 	}
1152fd3bf775SJohn Dyson 
1153fd3bf775SJohn Dyson 	while ((bp->b_flags & B_DONE) == 0) {
115484af4da6SJohn Dyson 		if (tsleep((caddr_t)bp, PRIBIO, "physstr", aiod_timeout)) {
1155fd3bf775SJohn Dyson 			if ((bp->b_flags & B_DONE) == 0) {
1156fd3bf775SJohn Dyson 				splx(s);
1157fd3bf775SJohn Dyson 				return EINPROGRESS;
1158fd3bf775SJohn Dyson 			} else {
1159fd3bf775SJohn Dyson 				break;
1160fd3bf775SJohn Dyson 			}
1161fd3bf775SJohn Dyson 		}
1162fd3bf775SJohn Dyson 	}
1163fd3bf775SJohn Dyson 
1164fd3bf775SJohn Dyson 	/* release mapping into kernel space */
1165fd3bf775SJohn Dyson 	vunmapbuf(bp);
1166fd3bf775SJohn Dyson 	iocb->bp = 0;
1167fd3bf775SJohn Dyson 
1168fd3bf775SJohn Dyson 	error = 0;
1169fd3bf775SJohn Dyson 	/*
1170fd3bf775SJohn Dyson 	 * check for an error
1171fd3bf775SJohn Dyson 	 */
1172fd3bf775SJohn Dyson 	if (bp->b_flags & B_ERROR) {
1173fd3bf775SJohn Dyson 		error = bp->b_error;
1174fd3bf775SJohn Dyson 	}
1175fd3bf775SJohn Dyson 
1176fd3bf775SJohn Dyson 	relpbuf(bp);
1177fd3bf775SJohn Dyson 	return (error);
1178fd3bf775SJohn Dyson }
1179fd3bf775SJohn Dyson 
1180fd3bf775SJohn Dyson /*
118184af4da6SJohn Dyson  * Queue a new AIO request.  Choosing either the threaded or direct physio
118284af4da6SJohn Dyson  * VCHR technique is done in this code.
11832244ea07SJohn Dyson  */
11842244ea07SJohn Dyson static int
118584af4da6SJohn Dyson _aio_aqueue(struct proc *p, struct aiocb *job, struct aio_liojob *lj, int type)
1186fd3bf775SJohn Dyson {
11872244ea07SJohn Dyson 	struct filedesc *fdp;
11882244ea07SJohn Dyson 	struct file *fp;
11892244ea07SJohn Dyson 	unsigned int fd;
11902244ea07SJohn Dyson 
11912244ea07SJohn Dyson 	int error;
11922244ea07SJohn Dyson 	int opcode;
11932244ea07SJohn Dyson 	struct aiocblist *aiocbe;
11942244ea07SJohn Dyson 	struct aioproclist *aiop;
11952244ea07SJohn Dyson 	struct kaioinfo *ki;
11962244ea07SJohn Dyson 
11972244ea07SJohn Dyson 	if (aiocbe = TAILQ_FIRST(&aio_freejobs)) {
11982244ea07SJohn Dyson 		TAILQ_REMOVE(&aio_freejobs, aiocbe, list);
11992244ea07SJohn Dyson 	} else {
1200fd3bf775SJohn Dyson 		aiocbe = zalloc (aiocb_zone);
12012244ea07SJohn Dyson 	}
12022244ea07SJohn Dyson 
1203fd3bf775SJohn Dyson 	aiocbe->inputcharge = 0;
1204fd3bf775SJohn Dyson 	aiocbe->outputcharge = 0;
1205fd3bf775SJohn Dyson 
1206fd3bf775SJohn Dyson 	suword(&job->_aiocb_private.status, -1);
1207fd3bf775SJohn Dyson 	suword(&job->_aiocb_private.error, 0);
1208fd3bf775SJohn Dyson 	suword(&job->_aiocb_private.kernelinfo, -1);
1209fd3bf775SJohn Dyson 
12102244ea07SJohn Dyson 	error = copyin((caddr_t)job,
12112244ea07SJohn Dyson 		(caddr_t) &aiocbe->uaiocb, sizeof aiocbe->uaiocb);
12122244ea07SJohn Dyson 	if (error) {
1213fd3bf775SJohn Dyson 		suword(&job->_aiocb_private.error, error);
1214fd3bf775SJohn Dyson 
12152244ea07SJohn Dyson 		TAILQ_INSERT_HEAD(&aio_freejobs, aiocbe, list);
12162244ea07SJohn Dyson 		return error;
12172244ea07SJohn Dyson 	}
12182244ea07SJohn Dyson 
1219a624e84fSJohn Dyson 	/*
122011783b14SJohn Dyson 	 * Save userspace address of the job info
122111783b14SJohn Dyson 	 */
122211783b14SJohn Dyson 	aiocbe->uuaiocb = job;
122311783b14SJohn Dyson 
122411783b14SJohn Dyson 	/*
1225a624e84fSJohn Dyson 	 * Get the opcode
1226a624e84fSJohn Dyson 	 */
1227a624e84fSJohn Dyson 	if (type != LIO_NOP) {
1228a624e84fSJohn Dyson 		aiocbe->uaiocb.aio_lio_opcode = type;
1229a624e84fSJohn Dyson 	}
1230a624e84fSJohn Dyson 	opcode = aiocbe->uaiocb.aio_lio_opcode;
12312244ea07SJohn Dyson 
12322244ea07SJohn Dyson 	/*
12332244ea07SJohn Dyson 	 * Get the fd info for process
12342244ea07SJohn Dyson 	 */
12352244ea07SJohn Dyson 	fdp = p->p_fd;
12362244ea07SJohn Dyson 
12372244ea07SJohn Dyson 	/*
12382244ea07SJohn Dyson 	 * Range check file descriptor
12392244ea07SJohn Dyson 	 */
12402244ea07SJohn Dyson 	fd = aiocbe->uaiocb.aio_fildes;
12412244ea07SJohn Dyson 	if (fd >= fdp->fd_nfiles) {
12422244ea07SJohn Dyson 		TAILQ_INSERT_HEAD(&aio_freejobs, aiocbe, list);
12432244ea07SJohn Dyson 		if (type == 0) {
12442244ea07SJohn Dyson 			suword(&job->_aiocb_private.error, EBADF);
12452244ea07SJohn Dyson 		}
12462244ea07SJohn Dyson 		return EBADF;
12472244ea07SJohn Dyson 	}
12482244ea07SJohn Dyson 
12492244ea07SJohn Dyson 	fp = fdp->fd_ofiles[fd];
1250a624e84fSJohn Dyson 	if ((fp == NULL) ||
1251a624e84fSJohn Dyson 		((opcode == LIO_WRITE) && ((fp->f_flag & FWRITE) == 0))) {
12522244ea07SJohn Dyson 		TAILQ_INSERT_HEAD(&aio_freejobs, aiocbe, list);
12532244ea07SJohn Dyson 		if (type == 0) {
12542244ea07SJohn Dyson 			suword(&job->_aiocb_private.error, EBADF);
12552244ea07SJohn Dyson 		}
12562244ea07SJohn Dyson 		return EBADF;
12572244ea07SJohn Dyson 	}
12582244ea07SJohn Dyson 
12592244ea07SJohn Dyson 	if (aiocbe->uaiocb.aio_offset == -1LL) {
12602244ea07SJohn Dyson 		TAILQ_INSERT_HEAD(&aio_freejobs, aiocbe, list);
12612244ea07SJohn Dyson 		if (type == 0) {
12622244ea07SJohn Dyson 			suword(&job->_aiocb_private.error, EINVAL);
12632244ea07SJohn Dyson 		}
12642244ea07SJohn Dyson 		return EINVAL;
12652244ea07SJohn Dyson 	}
12662244ea07SJohn Dyson 
12672244ea07SJohn Dyson 	error = suword(&job->_aiocb_private.kernelinfo, jobrefid);
12682244ea07SJohn Dyson 	if (error) {
12692244ea07SJohn Dyson 		TAILQ_INSERT_HEAD(&aio_freejobs, aiocbe, list);
12702244ea07SJohn Dyson 		if (type == 0) {
12712244ea07SJohn Dyson 			suword(&job->_aiocb_private.error, EINVAL);
12722244ea07SJohn Dyson 		}
12732244ea07SJohn Dyson 		return error;
12742244ea07SJohn Dyson 	}
12752244ea07SJohn Dyson 
127630166fabSBruce Evans 	aiocbe->uaiocb._aiocb_private.kernelinfo = (void *)(intptr_t)jobrefid;
12772d2f8ae7SBruce Evans 	if (jobrefid == LONG_MAX)
1278fd3bf775SJohn Dyson 		jobrefid = 1;
12792d2f8ae7SBruce Evans 	else
12802d2f8ae7SBruce Evans 		jobrefid++;
12812244ea07SJohn Dyson 
12822244ea07SJohn Dyson 	if (opcode == LIO_NOP) {
12832244ea07SJohn Dyson 		TAILQ_INSERT_HEAD(&aio_freejobs, aiocbe, list);
12842244ea07SJohn Dyson 		if (type == 0) {
12852244ea07SJohn Dyson 			suword(&job->_aiocb_private.error, 0);
1286fd3bf775SJohn Dyson 			suword(&job->_aiocb_private.status, 0);
1287fd3bf775SJohn Dyson 			suword(&job->_aiocb_private.kernelinfo, 0);
12882244ea07SJohn Dyson 		}
12892244ea07SJohn Dyson 		return 0;
12902244ea07SJohn Dyson 	}
12912244ea07SJohn Dyson 
1292fd3bf775SJohn Dyson 	if ((opcode != LIO_READ) && (opcode != LIO_WRITE)) {
12932244ea07SJohn Dyson 		TAILQ_INSERT_HEAD(&aio_freejobs, aiocbe, list);
12942244ea07SJohn Dyson 		if (type == 0) {
1295fd3bf775SJohn Dyson 			suword(&job->_aiocb_private.status, 0);
12962244ea07SJohn Dyson 			suword(&job->_aiocb_private.error, EINVAL);
12972244ea07SJohn Dyson 		}
12982244ea07SJohn Dyson 		return EINVAL;
12992244ea07SJohn Dyson 	}
13002244ea07SJohn Dyson 
1301fd3bf775SJohn Dyson 	suword(&job->_aiocb_private.error, EINPROGRESS);
1302fd3bf775SJohn Dyson 	aiocbe->uaiocb._aiocb_private.error = EINPROGRESS;
13032244ea07SJohn Dyson 	aiocbe->userproc = p;
13042244ea07SJohn Dyson 	aiocbe->jobflags = 0;
130584af4da6SJohn Dyson 	aiocbe->lio = lj;
130684af4da6SJohn Dyson 	ki = p->p_aioinfo;
13072244ea07SJohn Dyson 
1308fd3bf775SJohn Dyson 	if ((error = aio_qphysio(p, aiocbe)) == 0) {
1309fd3bf775SJohn Dyson 		return 0;
1310fd3bf775SJohn Dyson 	} else if (error > 0) {
1311fd3bf775SJohn Dyson 		suword(&job->_aiocb_private.status, 0);
1312fd3bf775SJohn Dyson 		aiocbe->uaiocb._aiocb_private.error = error;
1313fd3bf775SJohn Dyson 		suword(&job->_aiocb_private.error, error);
1314fd3bf775SJohn Dyson 		return error;
1315fd3bf775SJohn Dyson 	}
1316fd3bf775SJohn Dyson 
131784af4da6SJohn Dyson 	/*
131884af4da6SJohn Dyson 	 * No buffer for daemon I/O
131984af4da6SJohn Dyson 	 */
132084af4da6SJohn Dyson 	aiocbe->bp = NULL;
132184af4da6SJohn Dyson 
132284af4da6SJohn Dyson 	ki->kaio_queue_count++;
132384af4da6SJohn Dyson 	if (lj) {
132484af4da6SJohn Dyson 		lj->lioj_queue_count++;
132584af4da6SJohn Dyson 	}
1326fd3bf775SJohn Dyson 	TAILQ_INSERT_TAIL(&ki->kaio_jobqueue, aiocbe, plist);
1327fd3bf775SJohn Dyson 	TAILQ_INSERT_TAIL(&aio_jobs, aiocbe, list);
1328fd3bf775SJohn Dyson 	aiocbe->jobstate = JOBST_JOBQGLOBAL;
1329fd3bf775SJohn Dyson 
133084af4da6SJohn Dyson 	num_queue_count++;
1331fd3bf775SJohn Dyson 	error = 0;
1332fd3bf775SJohn Dyson 
1333fd3bf775SJohn Dyson 	/*
1334fd3bf775SJohn Dyson 	 * If we don't have a free AIO process, and we are below our
1335fd3bf775SJohn Dyson 	 * quota, then start one.  Otherwise, depend on the subsequent
1336fd3bf775SJohn Dyson 	 * I/O completions to pick-up this job.  If we don't sucessfully
1337fd3bf775SJohn Dyson 	 * create the new process (thread) due to resource issues, we
1338fd3bf775SJohn Dyson 	 * return an error for now (EAGAIN), which is likely not the
1339fd3bf775SJohn Dyson 	 * correct thing to do.
1340fd3bf775SJohn Dyson 	 */
13412244ea07SJohn Dyson retryproc:
13422244ea07SJohn Dyson 	if (aiop = TAILQ_FIRST(&aio_freeproc)) {
13432244ea07SJohn Dyson 		TAILQ_REMOVE(&aio_freeproc, aiop, list);
13442244ea07SJohn Dyson 		TAILQ_INSERT_TAIL(&aio_activeproc, aiop, list);
13452244ea07SJohn Dyson 		aiop->aioprocflags &= ~AIOP_FREE;
13462244ea07SJohn Dyson 		wakeup(aiop->aioproc);
1347fd3bf775SJohn Dyson 	} else if (((num_aio_resv_start + num_aio_procs) < max_aio_procs) &&
1348fd3bf775SJohn Dyson 			((ki->kaio_active_count + num_aio_resv_start) <
1349fd3bf775SJohn Dyson 				ki->kaio_maxactive_count)) {
1350fd3bf775SJohn Dyson 		num_aio_resv_start++;
1351fd3bf775SJohn Dyson 		if ((error = aio_newproc()) == 0) {
135284af4da6SJohn Dyson 			num_aio_resv_start--;
1353e499ed6fSJohn Dyson 			p->p_retval[0] = 0;
13542244ea07SJohn Dyson 			goto retryproc;
1355fd3bf775SJohn Dyson 		}
135684af4da6SJohn Dyson 		num_aio_resv_start--;
1357fd3bf775SJohn Dyson 	}
1358fd3bf775SJohn Dyson 	return error;
13592244ea07SJohn Dyson }
13602244ea07SJohn Dyson 
1361fd3bf775SJohn Dyson /*
1362fd3bf775SJohn Dyson  * This routine queues an AIO request, checking for quotas.
1363fd3bf775SJohn Dyson  */
13642244ea07SJohn Dyson static int
1365fd3bf775SJohn Dyson aio_aqueue(struct proc *p, struct aiocb *job, int type)
1366fd3bf775SJohn Dyson {
13672244ea07SJohn Dyson 	struct kaioinfo *ki;
13682244ea07SJohn Dyson 
13692244ea07SJohn Dyson 	if (p->p_aioinfo == NULL) {
13702244ea07SJohn Dyson 		aio_init_aioinfo(p);
13712244ea07SJohn Dyson 	}
13722244ea07SJohn Dyson 
13732244ea07SJohn Dyson 	if (num_queue_count >= max_queue_count)
13742244ea07SJohn Dyson 		return EAGAIN;
13752244ea07SJohn Dyson 
13762244ea07SJohn Dyson 	ki = p->p_aioinfo;
13772244ea07SJohn Dyson 	if (ki->kaio_queue_count >= ki->kaio_qallowed_count)
13782244ea07SJohn Dyson 		return EAGAIN;
13792244ea07SJohn Dyson 
138084af4da6SJohn Dyson 	return _aio_aqueue(p, job, NULL, type);
13812244ea07SJohn Dyson }
13822244ea07SJohn Dyson 
13832244ea07SJohn Dyson /*
1384fd3bf775SJohn Dyson  * Support the aio_return system call, as a side-effect, kernel
1385fd3bf775SJohn Dyson  * resources are released.
13862244ea07SJohn Dyson  */
13872244ea07SJohn Dyson int
1388fd3bf775SJohn Dyson aio_return(struct proc *p, struct aio_return_args *uap)
1389fd3bf775SJohn Dyson {
139084af4da6SJohn Dyson 	int s;
13912244ea07SJohn Dyson 	int jobref, status;
139284af4da6SJohn Dyson 	struct aiocblist *cb, *ncb;
139311783b14SJohn Dyson 	struct aiocb *ujob;
13942244ea07SJohn Dyson 	struct kaioinfo *ki;
1395fd3bf775SJohn Dyson 	struct proc *userp;
13962244ea07SJohn Dyson 
13972244ea07SJohn Dyson 	ki = p->p_aioinfo;
13982244ea07SJohn Dyson 	if (ki == NULL) {
13992244ea07SJohn Dyson 		return EINVAL;
14002244ea07SJohn Dyson 	}
14012244ea07SJohn Dyson 
140211783b14SJohn Dyson 	ujob = uap->aiocbp;
140311783b14SJohn Dyson 
140411783b14SJohn Dyson 	jobref = fuword(&ujob->_aiocb_private.kernelinfo);
1405fd3bf775SJohn Dyson 	if (jobref == -1 || jobref == 0)
14062244ea07SJohn Dyson 		return EINVAL;
14072244ea07SJohn Dyson 
14082244ea07SJohn Dyson 	for (cb = TAILQ_FIRST(&ki->kaio_jobdone);
14092244ea07SJohn Dyson 		cb;
14102244ea07SJohn Dyson 		cb = TAILQ_NEXT(cb, plist)) {
141130166fabSBruce Evans 		if (((intptr_t) cb->uaiocb._aiocb_private.kernelinfo) == jobref) {
141211783b14SJohn Dyson 			if (ujob == cb->uuaiocb) {
1413cb226aaaSPoul-Henning Kamp 				p->p_retval[0] = cb->uaiocb._aiocb_private.status;
141411783b14SJohn Dyson 			} else {
141511783b14SJohn Dyson 				p->p_retval[0] = EFAULT;
141611783b14SJohn Dyson 			}
1417fd3bf775SJohn Dyson 			if (cb->uaiocb.aio_lio_opcode == LIO_WRITE) {
1418fd3bf775SJohn Dyson 				curproc->p_stats->p_ru.ru_oublock += cb->outputcharge;
1419fd3bf775SJohn Dyson 				cb->outputcharge = 0;
1420fd3bf775SJohn Dyson 			} else if (cb->uaiocb.aio_lio_opcode == LIO_READ) {
1421fd3bf775SJohn Dyson 				curproc->p_stats->p_ru.ru_inblock += cb->inputcharge;
1422fd3bf775SJohn Dyson 				cb->inputcharge = 0;
1423fd3bf775SJohn Dyson 			}
14242244ea07SJohn Dyson 			aio_free_entry(cb);
14252244ea07SJohn Dyson 			return 0;
14262244ea07SJohn Dyson 		}
14272244ea07SJohn Dyson 	}
14282244ea07SJohn Dyson 
142984af4da6SJohn Dyson 	s = splbio();
143084af4da6SJohn Dyson 	for (cb = TAILQ_FIRST(&ki->kaio_bufdone);
143184af4da6SJohn Dyson 		cb;
143284af4da6SJohn Dyson 		cb = ncb) {
143384af4da6SJohn Dyson 		ncb = TAILQ_NEXT(cb, plist);
143430166fabSBruce Evans 		if (((intptr_t) cb->uaiocb._aiocb_private.kernelinfo) == jobref) {
143584af4da6SJohn Dyson 			splx(s);
143611783b14SJohn Dyson 			if (ujob == cb->uuaiocb) {
143784af4da6SJohn Dyson 				p->p_retval[0] = cb->uaiocb._aiocb_private.status;
143811783b14SJohn Dyson 			} else {
143911783b14SJohn Dyson 				p->p_retval[0] = EFAULT;
144011783b14SJohn Dyson 			}
144184af4da6SJohn Dyson 			aio_free_entry(cb);
144284af4da6SJohn Dyson 			return 0;
144384af4da6SJohn Dyson 		}
144484af4da6SJohn Dyson 	}
144584af4da6SJohn Dyson 	splx(s);
144684af4da6SJohn Dyson 
14472244ea07SJohn Dyson 	return (EINVAL);
14482244ea07SJohn Dyson }
14492244ea07SJohn Dyson 
14502244ea07SJohn Dyson /*
14512244ea07SJohn Dyson  * Allow a process to wakeup when any of the I/O requests are
14522244ea07SJohn Dyson  * completed.
14532244ea07SJohn Dyson  */
14542244ea07SJohn Dyson int
1455fd3bf775SJohn Dyson aio_suspend(struct proc *p, struct aio_suspend_args *uap)
1456fd3bf775SJohn Dyson {
14574a11ca4eSPoul-Henning Kamp 	struct timeval atv;
14582244ea07SJohn Dyson 	struct timespec ts;
14592244ea07SJohn Dyson 	struct aiocb *const *cbptr, *cbp;
14602244ea07SJohn Dyson 	struct kaioinfo *ki;
14612244ea07SJohn Dyson 	struct aiocblist *cb;
14622244ea07SJohn Dyson 	int i;
146384af4da6SJohn Dyson 	int njoblist;
14642244ea07SJohn Dyson 	int error, s, timo;
146511783b14SJohn Dyson 	int *ijoblist;
146611783b14SJohn Dyson 	struct aiocb **ujoblist;
14672244ea07SJohn Dyson 
1468fd3bf775SJohn Dyson 	if (uap->nent >= AIO_LISTIO_MAX)
1469fd3bf775SJohn Dyson 		return EINVAL;
14702244ea07SJohn Dyson 
14712244ea07SJohn Dyson 	timo = 0;
14722244ea07SJohn Dyson 	if (uap->timeout) {
14732244ea07SJohn Dyson 		/*
14742244ea07SJohn Dyson 		 * Get timespec struct
14752244ea07SJohn Dyson 		 */
14762244ea07SJohn Dyson 		if (error = copyin((caddr_t) uap->timeout, (caddr_t) &ts, sizeof ts)) {
14772244ea07SJohn Dyson 			return error;
14782244ea07SJohn Dyson 		}
14792244ea07SJohn Dyson 
14802244ea07SJohn Dyson 		if (ts.tv_nsec < 0 || ts.tv_nsec >= 1000000000)
14812244ea07SJohn Dyson 			return (EINVAL);
14822244ea07SJohn Dyson 
14832244ea07SJohn Dyson 		TIMESPEC_TO_TIMEVAL(&atv, &ts)
14842244ea07SJohn Dyson 		if (itimerfix(&atv))
14852244ea07SJohn Dyson 			return (EINVAL);
1486227ee8a1SPoul-Henning Kamp 		timo = tvtohz(&atv);
14872244ea07SJohn Dyson 	}
14882244ea07SJohn Dyson 
14892244ea07SJohn Dyson 	ki = p->p_aioinfo;
14902244ea07SJohn Dyson 	if (ki == NULL)
14912244ea07SJohn Dyson 		return EAGAIN;
14922244ea07SJohn Dyson 
149384af4da6SJohn Dyson 	njoblist = 0;
149411783b14SJohn Dyson 	ijoblist = zalloc(aiol_zone);
149511783b14SJohn Dyson 	ujoblist = zalloc(aiol_zone);
14962244ea07SJohn Dyson 	cbptr = uap->aiocbp;
14972244ea07SJohn Dyson 
14982244ea07SJohn Dyson 	for(i = 0; i < uap->nent; i++) {
149930166fabSBruce Evans 		cbp = (struct aiocb *) (intptr_t) fuword((caddr_t) &cbptr[i]);
150084af4da6SJohn Dyson 		if (cbp == 0)
150184af4da6SJohn Dyson 			continue;
150211783b14SJohn Dyson 		ujoblist[njoblist] = cbp;
150311783b14SJohn Dyson 		ijoblist[njoblist] = fuword(&cbp->_aiocb_private.kernelinfo);
150484af4da6SJohn Dyson 		njoblist++;
15052244ea07SJohn Dyson 	}
150611783b14SJohn Dyson 	if (njoblist == 0) {
150711783b14SJohn Dyson 		zfree(aiol_zone, ijoblist);
150811783b14SJohn Dyson 		zfree(aiol_zone, ujoblist);
150984af4da6SJohn Dyson 		return 0;
151011783b14SJohn Dyson 	}
15112244ea07SJohn Dyson 
151211783b14SJohn Dyson 	error = 0;
15132244ea07SJohn Dyson 	while (1) {
15142244ea07SJohn Dyson 		for (cb = TAILQ_FIRST(&ki->kaio_jobdone);
151584af4da6SJohn Dyson 			cb; cb = TAILQ_NEXT(cb, plist)) {
151684af4da6SJohn Dyson 			for(i = 0; i < njoblist; i++) {
151730166fabSBruce Evans 				if (((intptr_t) cb->uaiocb._aiocb_private.kernelinfo) ==
151811783b14SJohn Dyson 					ijoblist[i]) {
151911783b14SJohn Dyson 					if (ujoblist[i] != cb->uuaiocb)
152011783b14SJohn Dyson 						error = EINVAL;
152111783b14SJohn Dyson 					zfree(aiol_zone, ijoblist);
152211783b14SJohn Dyson 					zfree(aiol_zone, ujoblist);
152311783b14SJohn Dyson 					return error;
152484af4da6SJohn Dyson 				}
152584af4da6SJohn Dyson 			}
152684af4da6SJohn Dyson 		}
152784af4da6SJohn Dyson 
152884af4da6SJohn Dyson 		s = splbio();
152984af4da6SJohn Dyson 		for (cb = TAILQ_FIRST(&ki->kaio_bufdone);
153084af4da6SJohn Dyson 			cb; cb = TAILQ_NEXT(cb, plist)) {
153184af4da6SJohn Dyson 			for(i = 0; i < njoblist; i++) {
153230166fabSBruce Evans 				if (((intptr_t) cb->uaiocb._aiocb_private.kernelinfo) ==
153311783b14SJohn Dyson 					ijoblist[i]) {
153484af4da6SJohn Dyson 					splx(s);
153511783b14SJohn Dyson 					if (ujoblist[i] != cb->uuaiocb)
153611783b14SJohn Dyson 						error = EINVAL;
153711783b14SJohn Dyson 					zfree(aiol_zone, ijoblist);
153811783b14SJohn Dyson 					zfree(aiol_zone, ujoblist);
153911783b14SJohn Dyson 					return error;
15402244ea07SJohn Dyson 				}
15412244ea07SJohn Dyson 			}
15422244ea07SJohn Dyson 		}
15432244ea07SJohn Dyson 
1544fd3bf775SJohn Dyson 		ki->kaio_flags |= KAIO_WAKEUP;
15452244ea07SJohn Dyson 		error = tsleep(p, PRIBIO|PCATCH, "aiospn", timo);
154684af4da6SJohn Dyson 		splx(s);
15472244ea07SJohn Dyson 
15482244ea07SJohn Dyson 		if (error == EINTR) {
154911783b14SJohn Dyson 			zfree(aiol_zone, ijoblist);
155011783b14SJohn Dyson 			zfree(aiol_zone, ujoblist);
15512244ea07SJohn Dyson 			return EINTR;
15522244ea07SJohn Dyson 		} else if (error == EWOULDBLOCK) {
155311783b14SJohn Dyson 			zfree(aiol_zone, ijoblist);
155411783b14SJohn Dyson 			zfree(aiol_zone, ujoblist);
15552244ea07SJohn Dyson 			return EAGAIN;
15562244ea07SJohn Dyson 		}
15572244ea07SJohn Dyson 	}
15582244ea07SJohn Dyson 
15592244ea07SJohn Dyson /* NOTREACHED */
15602244ea07SJohn Dyson 	return EINVAL;
15612244ea07SJohn Dyson }
1562ee877a35SJohn Dyson 
1563ee877a35SJohn Dyson /*
1564ee877a35SJohn Dyson  * aio_cancel at the kernel level is a NOOP right now.  It
1565ee877a35SJohn Dyson  * might be possible to support it partially in user mode, or
1566ee877a35SJohn Dyson  * in kernel mode later on.
1567ee877a35SJohn Dyson  */
1568ee877a35SJohn Dyson int
1569fd3bf775SJohn Dyson aio_cancel(struct proc *p, struct aio_cancel_args *uap)
1570fd3bf775SJohn Dyson {
157178922e41SJohn Dyson       return ENOSYS;
1572ee877a35SJohn Dyson }
1573ee877a35SJohn Dyson 
1574ee877a35SJohn Dyson /*
1575ee877a35SJohn Dyson  * aio_error is implemented in the kernel level for compatibility
1576ee877a35SJohn Dyson  * purposes only.  For a user mode async implementation, it would be
1577ee877a35SJohn Dyson  * best to do it in a userland subroutine.
1578ee877a35SJohn Dyson  */
1579ee877a35SJohn Dyson int
1580fd3bf775SJohn Dyson aio_error(struct proc *p, struct aio_error_args *uap)
1581fd3bf775SJohn Dyson {
158284af4da6SJohn Dyson 	int s;
15832244ea07SJohn Dyson 	struct aiocblist *cb;
15842244ea07SJohn Dyson 	struct kaioinfo *ki;
15852244ea07SJohn Dyson 	int jobref;
1586fd3bf775SJohn Dyson 	int error, status;
1587ee877a35SJohn Dyson 
15882244ea07SJohn Dyson 	ki = p->p_aioinfo;
15892244ea07SJohn Dyson 	if (ki == NULL)
15902244ea07SJohn Dyson 		return EINVAL;
15912244ea07SJohn Dyson 
15922244ea07SJohn Dyson 	jobref = fuword(&uap->aiocbp->_aiocb_private.kernelinfo);
1593fd3bf775SJohn Dyson 	if ((jobref == -1) || (jobref == 0))
1594fd3bf775SJohn Dyson 		return EINVAL;
1595ee877a35SJohn Dyson 
15962244ea07SJohn Dyson 	for (cb = TAILQ_FIRST(&ki->kaio_jobdone);
15972244ea07SJohn Dyson 		cb;
15982244ea07SJohn Dyson 		cb = TAILQ_NEXT(cb, plist)) {
15992244ea07SJohn Dyson 
160030166fabSBruce Evans 		if (((intptr_t) cb->uaiocb._aiocb_private.kernelinfo) == jobref) {
1601cb226aaaSPoul-Henning Kamp 			p->p_retval[0] = cb->uaiocb._aiocb_private.error;
16022244ea07SJohn Dyson 			return 0;
16032244ea07SJohn Dyson 		}
1604ee877a35SJohn Dyson 	}
1605ee877a35SJohn Dyson 
16062244ea07SJohn Dyson 	for (cb = TAILQ_FIRST(&ki->kaio_jobqueue);
16072244ea07SJohn Dyson 		cb;
16082244ea07SJohn Dyson 		cb = TAILQ_NEXT(cb, plist)) {
16092244ea07SJohn Dyson 
161030166fabSBruce Evans 		if (((intptr_t) cb->uaiocb._aiocb_private.kernelinfo) == jobref) {
1611cb226aaaSPoul-Henning Kamp 			p->p_retval[0] = EINPROGRESS;
16122244ea07SJohn Dyson 			return 0;
16132244ea07SJohn Dyson 		}
16142244ea07SJohn Dyson 	}
16152244ea07SJohn Dyson 
161684af4da6SJohn Dyson 	s = splbio();
161784af4da6SJohn Dyson 	for (cb = TAILQ_FIRST(&ki->kaio_bufdone);
161884af4da6SJohn Dyson 		cb;
161984af4da6SJohn Dyson 		cb = TAILQ_NEXT(cb, plist)) {
162030166fabSBruce Evans 		if (((intptr_t) cb->uaiocb._aiocb_private.kernelinfo) == jobref) {
162184af4da6SJohn Dyson 			p->p_retval[0] = cb->uaiocb._aiocb_private.error;
162284af4da6SJohn Dyson 			splx(s);
162384af4da6SJohn Dyson 			return 0;
162484af4da6SJohn Dyson 		}
162584af4da6SJohn Dyson 	}
162684af4da6SJohn Dyson 
162784af4da6SJohn Dyson 	for (cb = TAILQ_FIRST(&ki->kaio_bufqueue);
162884af4da6SJohn Dyson 		cb;
162984af4da6SJohn Dyson 		cb = TAILQ_NEXT(cb, plist)) {
163030166fabSBruce Evans 		if (((intptr_t) cb->uaiocb._aiocb_private.kernelinfo) == jobref) {
163184af4da6SJohn Dyson 			p->p_retval[0] = EINPROGRESS;
163284af4da6SJohn Dyson 			splx(s);
163384af4da6SJohn Dyson 			return 0;
163484af4da6SJohn Dyson 		}
163584af4da6SJohn Dyson 	}
163684af4da6SJohn Dyson 	splx(s);
163784af4da6SJohn Dyson 
163884af4da6SJohn Dyson 
16392244ea07SJohn Dyson 	/*
16402244ea07SJohn Dyson 	 * Hack for lio
16412244ea07SJohn Dyson 	 */
1642fd3bf775SJohn Dyson /*
16432244ea07SJohn Dyson 	status = fuword(&uap->aiocbp->_aiocb_private.status);
16442244ea07SJohn Dyson 	if (status == -1) {
16452244ea07SJohn Dyson 		return fuword(&uap->aiocbp->_aiocb_private.error);
16462244ea07SJohn Dyson 	}
1647fd3bf775SJohn Dyson */
16482244ea07SJohn Dyson 	return EINVAL;
1649ee877a35SJohn Dyson }
1650ee877a35SJohn Dyson 
1651ee877a35SJohn Dyson int
1652fd3bf775SJohn Dyson aio_read(struct proc *p, struct aio_read_args *uap)
1653fd3bf775SJohn Dyson {
1654ee877a35SJohn Dyson 	struct filedesc *fdp;
1655ee877a35SJohn Dyson 	struct file *fp;
1656ee877a35SJohn Dyson 	struct uio auio;
1657ee877a35SJohn Dyson 	struct iovec aiov;
1658ee877a35SJohn Dyson 	unsigned int fd;
1659ee877a35SJohn Dyson 	int cnt;
1660ee877a35SJohn Dyson 	struct aiocb iocb;
16612244ea07SJohn Dyson 	int error, pmodes;
1662ee877a35SJohn Dyson 
16632244ea07SJohn Dyson 	pmodes = fuword(&uap->aiocbp->_aiocb_private.privatemodes);
16642244ea07SJohn Dyson 	if ((pmodes & AIO_PMODE_SYNC) == 0) {
16652244ea07SJohn Dyson 		return aio_aqueue(p, (struct aiocb *) uap->aiocbp, LIO_READ);
16662244ea07SJohn Dyson 	}
1667ee877a35SJohn Dyson 
1668ee877a35SJohn Dyson 	/*
1669ee877a35SJohn Dyson 	 * Get control block
1670ee877a35SJohn Dyson 	 */
1671ee877a35SJohn Dyson 	if (error = copyin((caddr_t) uap->aiocbp, (caddr_t) &iocb, sizeof iocb))
1672ee877a35SJohn Dyson 		return error;
1673ee877a35SJohn Dyson 
1674ee877a35SJohn Dyson 	/*
1675ee877a35SJohn Dyson 	 * Get the fd info for process
1676ee877a35SJohn Dyson 	 */
1677ee877a35SJohn Dyson 	fdp = p->p_fd;
1678ee877a35SJohn Dyson 
1679ee877a35SJohn Dyson 	/*
1680ee877a35SJohn Dyson 	 * Range check file descriptor
1681ee877a35SJohn Dyson 	 */
1682ee877a35SJohn Dyson 	fd = iocb.aio_fildes;
1683ee877a35SJohn Dyson 	if (fd >= fdp->fd_nfiles)
1684ee877a35SJohn Dyson 		return EBADF;
1685ee877a35SJohn Dyson 	fp = fdp->fd_ofiles[fd];
1686ee877a35SJohn Dyson 	if ((fp == NULL) || ((fp->f_flag & FREAD) == 0))
1687ee877a35SJohn Dyson 		return EBADF;
16882244ea07SJohn Dyson 	if (iocb.aio_offset == -1LL)
1689ee877a35SJohn Dyson 		return EINVAL;
1690ee877a35SJohn Dyson 
1691ee877a35SJohn Dyson 	auio.uio_resid = iocb.aio_nbytes;
1692ee877a35SJohn Dyson 	if (auio.uio_resid < 0)
1693ee877a35SJohn Dyson 		return (EINVAL);
1694ee877a35SJohn Dyson 
16952244ea07SJohn Dyson 	/*
16962244ea07SJohn Dyson 	 * Process sync simply -- queue async request.
16972244ea07SJohn Dyson 	 */
16982244ea07SJohn Dyson 	if ((iocb._aiocb_private.privatemodes & AIO_PMODE_SYNC) == 0) {
16992244ea07SJohn Dyson 		return aio_aqueue(p, (struct aiocb *) uap->aiocbp, LIO_READ);
17002244ea07SJohn Dyson 	}
17012244ea07SJohn Dyson 
170264889941SJohn Dyson 	aiov.iov_base = (void *) iocb.aio_buf;
17032244ea07SJohn Dyson 	aiov.iov_len = iocb.aio_nbytes;
17042244ea07SJohn Dyson 
17052244ea07SJohn Dyson 	auio.uio_iov = &aiov;
17062244ea07SJohn Dyson 	auio.uio_iovcnt = 1;
17072244ea07SJohn Dyson 	auio.uio_offset = iocb.aio_offset;
1708ee877a35SJohn Dyson 	auio.uio_rw = UIO_READ;
1709ee877a35SJohn Dyson 	auio.uio_segflg = UIO_USERSPACE;
1710ee877a35SJohn Dyson 	auio.uio_procp = p;
1711ee877a35SJohn Dyson 
1712ee877a35SJohn Dyson 	cnt = iocb.aio_nbytes;
1713ee877a35SJohn Dyson 	error = (*fp->f_ops->fo_read)(fp, &auio, fp->f_cred);
1714ee877a35SJohn Dyson 	if (error &&
1715ee877a35SJohn Dyson 		(auio.uio_resid != cnt) &&
1716ee877a35SJohn Dyson 		(error == ERESTART || error == EINTR || error == EWOULDBLOCK))
1717ee877a35SJohn Dyson 			error = 0;
1718ee877a35SJohn Dyson 	cnt -= auio.uio_resid;
1719cb226aaaSPoul-Henning Kamp 	p->p_retval[0] = cnt;
1720ee877a35SJohn Dyson 	return error;
1721ee877a35SJohn Dyson }
1722ee877a35SJohn Dyson 
1723ee877a35SJohn Dyson int
1724fd3bf775SJohn Dyson aio_write(struct proc *p, struct aio_write_args *uap)
1725fd3bf775SJohn Dyson {
1726ee877a35SJohn Dyson 	struct filedesc *fdp;
1727ee877a35SJohn Dyson 	struct file *fp;
1728ee877a35SJohn Dyson 	struct uio auio;
1729ee877a35SJohn Dyson 	struct iovec aiov;
1730ee877a35SJohn Dyson 	unsigned int fd;
1731ee877a35SJohn Dyson 	int cnt;
1732ee877a35SJohn Dyson 	struct aiocb iocb;
1733ee877a35SJohn Dyson 	int error;
17342244ea07SJohn Dyson 	int pmodes;
17352244ea07SJohn Dyson 
17362244ea07SJohn Dyson 	/*
17372244ea07SJohn Dyson 	 * Process sync simply -- queue async request.
17382244ea07SJohn Dyson 	 */
17392244ea07SJohn Dyson 	pmodes = fuword(&uap->aiocbp->_aiocb_private.privatemodes);
17402244ea07SJohn Dyson 	if ((pmodes & AIO_PMODE_SYNC) == 0) {
17412244ea07SJohn Dyson 		return aio_aqueue(p, (struct aiocb *) uap->aiocbp, LIO_WRITE);
17422244ea07SJohn Dyson 	}
1743ee877a35SJohn Dyson 
1744ee877a35SJohn Dyson 	if (error = copyin((caddr_t) uap->aiocbp, (caddr_t) &iocb, sizeof iocb))
1745ee877a35SJohn Dyson 		return error;
1746ee877a35SJohn Dyson 
1747ee877a35SJohn Dyson 	/*
1748ee877a35SJohn Dyson 	 * Get the fd info for process
1749ee877a35SJohn Dyson 	 */
1750ee877a35SJohn Dyson 	fdp = p->p_fd;
1751ee877a35SJohn Dyson 
1752ee877a35SJohn Dyson 	/*
1753ee877a35SJohn Dyson 	 * Range check file descriptor
1754ee877a35SJohn Dyson 	 */
1755ee877a35SJohn Dyson 	fd = iocb.aio_fildes;
1756ee877a35SJohn Dyson 	if (fd >= fdp->fd_nfiles)
1757ee877a35SJohn Dyson 		return EBADF;
1758ee877a35SJohn Dyson 	fp = fdp->fd_ofiles[fd];
1759ee877a35SJohn Dyson 	if ((fp == NULL) || ((fp->f_flag & FWRITE) == 0))
1760ee877a35SJohn Dyson 		return EBADF;
17612244ea07SJohn Dyson 	if (iocb.aio_offset == -1LL)
1762ee877a35SJohn Dyson 		return EINVAL;
1763ee877a35SJohn Dyson 
176464889941SJohn Dyson 	aiov.iov_base = (void *) iocb.aio_buf;
1765ee877a35SJohn Dyson 	aiov.iov_len = iocb.aio_nbytes;
1766ee877a35SJohn Dyson 	auio.uio_iov = &aiov;
1767ee877a35SJohn Dyson 	auio.uio_iovcnt = 1;
1768ee877a35SJohn Dyson 	auio.uio_offset = iocb.aio_offset;
1769ee877a35SJohn Dyson 
1770ee877a35SJohn Dyson 	auio.uio_resid = iocb.aio_nbytes;
1771ee877a35SJohn Dyson 	if (auio.uio_resid < 0)
1772ee877a35SJohn Dyson 		return (EINVAL);
1773ee877a35SJohn Dyson 
1774ee877a35SJohn Dyson 	auio.uio_rw = UIO_WRITE;
1775ee877a35SJohn Dyson 	auio.uio_segflg = UIO_USERSPACE;
1776ee877a35SJohn Dyson 	auio.uio_procp = p;
1777ee877a35SJohn Dyson 
1778ee877a35SJohn Dyson 	cnt = iocb.aio_nbytes;
1779ee877a35SJohn Dyson 	error = (*fp->f_ops->fo_write)(fp, &auio, fp->f_cred);
1780ee877a35SJohn Dyson 	if (error) {
1781ee877a35SJohn Dyson 		if (auio.uio_resid != cnt) {
1782ee877a35SJohn Dyson 			if (error == ERESTART || error == EINTR || error == EWOULDBLOCK)
1783ee877a35SJohn Dyson 				error = 0;
1784ee877a35SJohn Dyson 			if (error == EPIPE)
1785ee877a35SJohn Dyson 				psignal(p, SIGPIPE);
1786ee877a35SJohn Dyson 		}
1787ee877a35SJohn Dyson 	}
1788ee877a35SJohn Dyson 	cnt -= auio.uio_resid;
1789cb226aaaSPoul-Henning Kamp 	p->p_retval[0] = cnt;
1790ee877a35SJohn Dyson 	return error;
1791ee877a35SJohn Dyson }
1792ee877a35SJohn Dyson 
1793ee877a35SJohn Dyson int
1794fd3bf775SJohn Dyson lio_listio(struct proc *p, struct lio_listio_args *uap)
1795fd3bf775SJohn Dyson {
17964a11ca4eSPoul-Henning Kamp 	int nent, nentqueued;
17972244ea07SJohn Dyson 	struct aiocb *iocb, * const *cbptr;
17982244ea07SJohn Dyson 	struct aiocblist *cb;
17992244ea07SJohn Dyson 	struct kaioinfo *ki;
180084af4da6SJohn Dyson 	struct aio_liojob *lj;
18012244ea07SJohn Dyson 	int error, runningcode;
1802fd3bf775SJohn Dyson 	int nerror;
1803ee877a35SJohn Dyson 	int i;
180484af4da6SJohn Dyson 	int s;
1805ee877a35SJohn Dyson 
1806a624e84fSJohn Dyson 	if ((uap->mode != LIO_NOWAIT) && (uap->mode != LIO_WAIT)) {
1807ee877a35SJohn Dyson 		return EINVAL;
1808a624e84fSJohn Dyson 	}
18092244ea07SJohn Dyson 
18102244ea07SJohn Dyson 	nent = uap->nent;
1811a624e84fSJohn Dyson 	if (nent > AIO_LISTIO_MAX) {
18122244ea07SJohn Dyson 		return EINVAL;
1813a624e84fSJohn Dyson 	}
18142244ea07SJohn Dyson 
18152244ea07SJohn Dyson 	if (p->p_aioinfo == NULL) {
18162244ea07SJohn Dyson 		aio_init_aioinfo(p);
18172244ea07SJohn Dyson 	}
18182244ea07SJohn Dyson 
1819a624e84fSJohn Dyson 	if ((nent + num_queue_count) > max_queue_count) {
18202244ea07SJohn Dyson 		return EAGAIN;
1821a624e84fSJohn Dyson 	}
18222244ea07SJohn Dyson 
18232244ea07SJohn Dyson 	ki = p->p_aioinfo;
1824a624e84fSJohn Dyson 	if ((nent + ki->kaio_queue_count) > ki->kaio_qallowed_count) {
18252244ea07SJohn Dyson 		return EAGAIN;
1826a624e84fSJohn Dyson 	}
18272244ea07SJohn Dyson 
182884af4da6SJohn Dyson 	lj = zalloc(aiolio_zone);
182984af4da6SJohn Dyson 	if (!lj) {
183084af4da6SJohn Dyson 		return EAGAIN;
183184af4da6SJohn Dyson 	}
183284af4da6SJohn Dyson 
183384af4da6SJohn Dyson 	lj->lioj_flags = 0;
183484af4da6SJohn Dyson 	lj->lioj_buffer_count = 0;
183584af4da6SJohn Dyson 	lj->lioj_buffer_finished_count = 0;
183684af4da6SJohn Dyson 	lj->lioj_queue_count = 0;
183784af4da6SJohn Dyson 	lj->lioj_queue_finished_count = 0;
183884af4da6SJohn Dyson 	lj->lioj_ki = ki;
183984af4da6SJohn Dyson 	TAILQ_INSERT_TAIL(&ki->kaio_liojoblist, lj, lioj_list);
184084af4da6SJohn Dyson 
184184af4da6SJohn Dyson 	/*
184284af4da6SJohn Dyson 	 * Setup signal
184384af4da6SJohn Dyson 	 */
184484af4da6SJohn Dyson 	if (uap->sig && (uap->mode == LIO_NOWAIT)) {
184584af4da6SJohn Dyson 		error = copyin(uap->sig, &lj->lioj_signal, sizeof lj->lioj_signal);
184684af4da6SJohn Dyson 		if (error)
184784af4da6SJohn Dyson 			return error;
184884af4da6SJohn Dyson 		lj->lioj_flags |= LIOJ_SIGNAL;
184984af4da6SJohn Dyson 		lj->lioj_flags &= ~LIOJ_SIGNAL_POSTED;
185084af4da6SJohn Dyson 	} else {
185184af4da6SJohn Dyson 		lj->lioj_flags &= ~LIOJ_SIGNAL;
185284af4da6SJohn Dyson 	}
185384af4da6SJohn Dyson 
18542244ea07SJohn Dyson /*
18552244ea07SJohn Dyson  * get pointers to the list of I/O requests
18562244ea07SJohn Dyson  */
18572244ea07SJohn Dyson 
1858fd3bf775SJohn Dyson 	nerror = 0;
1859fd3bf775SJohn Dyson 	nentqueued = 0;
18602244ea07SJohn Dyson 	cbptr = uap->acb_list;
18612244ea07SJohn Dyson 	for(i = 0; i < uap->nent; i++) {
186230166fabSBruce Evans 		iocb = (struct aiocb *) (intptr_t) fuword((caddr_t) &cbptr[i]);
186330166fabSBruce Evans 		if (((intptr_t) iocb != -1) && ((intptr_t) iocb != NULL)) {
186484af4da6SJohn Dyson 			error = _aio_aqueue(p, iocb, lj, 0);
1865fd3bf775SJohn Dyson 			if (error == 0) {
18662244ea07SJohn Dyson 				nentqueued++;
1867fd3bf775SJohn Dyson 			} else {
1868fd3bf775SJohn Dyson 				nerror++;
1869fd3bf775SJohn Dyson 			}
1870fd3bf775SJohn Dyson 		}
18712244ea07SJohn Dyson 	}
18722244ea07SJohn Dyson 
1873a624e84fSJohn Dyson 	/*
1874a624e84fSJohn Dyson 	 * If we haven't queued any, then just return error
1875a624e84fSJohn Dyson 	 */
1876a624e84fSJohn Dyson 	if (nentqueued == 0) {
1877fd3bf775SJohn Dyson 		return 0;
1878a624e84fSJohn Dyson 	}
18792244ea07SJohn Dyson 
1880a624e84fSJohn Dyson 	/*
1881a624e84fSJohn Dyson 	 * Calculate the appropriate error return
1882a624e84fSJohn Dyson 	 */
18832244ea07SJohn Dyson 	runningcode = 0;
1884fd3bf775SJohn Dyson 	if (nerror)
18852244ea07SJohn Dyson 		runningcode = EIO;
18862244ea07SJohn Dyson 
18872244ea07SJohn Dyson 	if (uap->mode == LIO_WAIT) {
18882244ea07SJohn Dyson 		while (1) {
18892244ea07SJohn Dyson 			int found;
1890fd3bf775SJohn Dyson 			found = 0;
1891fd3bf775SJohn Dyson 			for(i = 0; i < uap->nent; i++) {
1892fd3bf775SJohn Dyson 				int jobref, command;
18932244ea07SJohn Dyson 
1894a624e84fSJohn Dyson 				/*
1895a624e84fSJohn Dyson 				 * Fetch address of the control buf pointer in user space
1896a624e84fSJohn Dyson 				 */
189730166fabSBruce Evans 				iocb = (struct aiocb *) (intptr_t) fuword((caddr_t) &cbptr[i]);
189830166fabSBruce Evans 				if (((intptr_t) iocb == -1) || ((intptr_t) iocb == 0))
1899fd3bf775SJohn Dyson 					continue;
1900a624e84fSJohn Dyson 
1901a624e84fSJohn Dyson 				/*
1902a624e84fSJohn Dyson 				 * Fetch the associated command from user space
1903a624e84fSJohn Dyson 				 */
19042244ea07SJohn Dyson 				command = fuword(&iocb->aio_lio_opcode);
1905fd3bf775SJohn Dyson 				if (command == LIO_NOP) {
1906fd3bf775SJohn Dyson 					found++;
19072244ea07SJohn Dyson 					continue;
1908fd3bf775SJohn Dyson 				}
1909a624e84fSJohn Dyson 
19102244ea07SJohn Dyson 				jobref = fuword(&iocb->_aiocb_private.kernelinfo);
19112244ea07SJohn Dyson 
19122244ea07SJohn Dyson 				for (cb = TAILQ_FIRST(&ki->kaio_jobdone);
19132244ea07SJohn Dyson 					cb;
19142244ea07SJohn Dyson 					cb = TAILQ_NEXT(cb, plist)) {
191530166fabSBruce Evans 					if (((intptr_t) cb->uaiocb._aiocb_private.kernelinfo) ==
1916fd3bf775SJohn Dyson 						jobref) {
191784af4da6SJohn Dyson 						if (cb->uaiocb.aio_lio_opcode == LIO_WRITE) {
191884af4da6SJohn Dyson 							curproc->p_stats->p_ru.ru_oublock +=
191984af4da6SJohn Dyson 								cb->outputcharge;
192084af4da6SJohn Dyson 							cb->outputcharge = 0;
192184af4da6SJohn Dyson 						} else if (cb->uaiocb.aio_lio_opcode == LIO_READ) {
192284af4da6SJohn Dyson 							curproc->p_stats->p_ru.ru_inblock +=
192384af4da6SJohn Dyson 								cb->inputcharge;
192484af4da6SJohn Dyson 							cb->inputcharge = 0;
192584af4da6SJohn Dyson 						}
19262244ea07SJohn Dyson 						found++;
19272244ea07SJohn Dyson 						break;
19282244ea07SJohn Dyson 					}
19292244ea07SJohn Dyson 				}
1930fd3bf775SJohn Dyson 
193184af4da6SJohn Dyson 				s = splbio();
193284af4da6SJohn Dyson 				for (cb = TAILQ_FIRST(&ki->kaio_bufdone);
193384af4da6SJohn Dyson 					cb;
193484af4da6SJohn Dyson 					cb = TAILQ_NEXT(cb, plist)) {
193530166fabSBruce Evans 					if (((intptr_t) cb->uaiocb._aiocb_private.kernelinfo) ==
193684af4da6SJohn Dyson 						jobref) {
193784af4da6SJohn Dyson 						found++;
193884af4da6SJohn Dyson 						break;
1939fd3bf775SJohn Dyson 					}
19402244ea07SJohn Dyson 				}
194184af4da6SJohn Dyson 				splx(s);
194284af4da6SJohn Dyson 
194384af4da6SJohn Dyson 			}
19442244ea07SJohn Dyson 
1945a624e84fSJohn Dyson 			/*
1946a624e84fSJohn Dyson 			 * If all I/Os have been disposed of, then we can return
1947a624e84fSJohn Dyson 			 */
1948fd3bf775SJohn Dyson 			if (found == nentqueued) {
19492244ea07SJohn Dyson 				return runningcode;
19502244ea07SJohn Dyson 			}
19512244ea07SJohn Dyson 
1952fd3bf775SJohn Dyson 			ki->kaio_flags |= KAIO_WAKEUP;
19532244ea07SJohn Dyson 			error = tsleep(p, PRIBIO|PCATCH, "aiospn", 0);
19542244ea07SJohn Dyson 
19552244ea07SJohn Dyson 			if (error == EINTR) {
19562244ea07SJohn Dyson 				return EINTR;
19572244ea07SJohn Dyson 			} else if (error == EWOULDBLOCK) {
19582244ea07SJohn Dyson 				return EAGAIN;
19592244ea07SJohn Dyson 			}
19602244ea07SJohn Dyson 
19612244ea07SJohn Dyson 		}
19622244ea07SJohn Dyson 	}
19632244ea07SJohn Dyson 
19642244ea07SJohn Dyson 	return runningcode;
1965ee877a35SJohn Dyson }
1966fd3bf775SJohn Dyson 
196784af4da6SJohn Dyson /*
196884af4da6SJohn Dyson  * This is a wierd hack so that we can post a signal.  It is safe
196984af4da6SJohn Dyson  * to do so from a timeout routine, but *not* from an interrupt routine.
197084af4da6SJohn Dyson  */
197184af4da6SJohn Dyson static void
197284af4da6SJohn Dyson process_signal(void *ljarg)
197384af4da6SJohn Dyson {
197484af4da6SJohn Dyson 	struct aio_liojob *lj = ljarg;
197584af4da6SJohn Dyson 	if (lj->lioj_signal.sigev_notify == SIGEV_SIGNAL) {
197684af4da6SJohn Dyson 		if (lj->lioj_queue_count == lj->lioj_queue_finished_count) {
197784af4da6SJohn Dyson 			psignal(lj->lioj_ki->kaio_p, lj->lioj_signal.sigev_signo);
197884af4da6SJohn Dyson 			lj->lioj_flags |= LIOJ_SIGNAL_POSTED;
197984af4da6SJohn Dyson 		}
198084af4da6SJohn Dyson 	}
198184af4da6SJohn Dyson }
198284af4da6SJohn Dyson 
198384af4da6SJohn Dyson /*
198484af4da6SJohn Dyson  * Interrupt handler for physio, performs the necessary process wakeups,
198584af4da6SJohn Dyson  * and signals.
198684af4da6SJohn Dyson  */
1987fd3bf775SJohn Dyson static void
1988fd3bf775SJohn Dyson aio_physwakeup(bp)
1989fd3bf775SJohn Dyson 	struct buf *bp;
1990fd3bf775SJohn Dyson {
199184af4da6SJohn Dyson 	struct aiocblist *aiocbe;
1992fd3bf775SJohn Dyson 	struct proc *p;
1993fd3bf775SJohn Dyson 	struct kaioinfo *ki;
199484af4da6SJohn Dyson 	struct aio_liojob *lj;
199511783b14SJohn Dyson 	int s;
199611783b14SJohn Dyson 	s = splbio();
1997fd3bf775SJohn Dyson 
1998fd3bf775SJohn Dyson 	wakeup((caddr_t) bp);
1999fd3bf775SJohn Dyson 	bp->b_flags &= ~B_CALL;
200084af4da6SJohn Dyson 	bp->b_flags |= B_DONE;
2001fd3bf775SJohn Dyson 
200284af4da6SJohn Dyson 	aiocbe = (struct aiocblist *)bp->b_spc;
200384af4da6SJohn Dyson 	if (aiocbe) {
200484af4da6SJohn Dyson 		p = bp->b_proc;
200584af4da6SJohn Dyson 
200684af4da6SJohn Dyson 		aiocbe->jobstate = JOBST_JOBBFINISHED;
200784af4da6SJohn Dyson 		aiocbe->uaiocb._aiocb_private.status -= bp->b_resid;
200884af4da6SJohn Dyson 		aiocbe->uaiocb._aiocb_private.error = 0;
200984af4da6SJohn Dyson 		aiocbe->jobflags |= AIOCBLIST_DONE;
201084af4da6SJohn Dyson 
201184af4da6SJohn Dyson 		if (bp->b_flags & B_ERROR) {
201284af4da6SJohn Dyson 			aiocbe->uaiocb._aiocb_private.error = bp->b_error;
201384af4da6SJohn Dyson 		}
201484af4da6SJohn Dyson 
201584af4da6SJohn Dyson 		lj = aiocbe->lio;
201684af4da6SJohn Dyson 		if (lj) {
201784af4da6SJohn Dyson 			lj->lioj_buffer_finished_count++;
201884af4da6SJohn Dyson 			/*
201984af4da6SJohn Dyson 			 * wakeup/signal if all of the interrupt jobs are done
202084af4da6SJohn Dyson 			 */
202184af4da6SJohn Dyson 			if (lj->lioj_buffer_finished_count == lj->lioj_buffer_count) {
202284af4da6SJohn Dyson 				/*
202384af4da6SJohn Dyson 				 * post a signal if it is called for
202484af4da6SJohn Dyson 				 */
202584af4da6SJohn Dyson 				if ((lj->lioj_flags & (LIOJ_SIGNAL|LIOJ_SIGNAL_POSTED)) ==
202684af4da6SJohn Dyson 					LIOJ_SIGNAL) {
202784af4da6SJohn Dyson 					lj->lioj_flags |= LIOJ_SIGNAL_POSTED;
202884af4da6SJohn Dyson 					timeout(process_signal, lj, 0);
202984af4da6SJohn Dyson 				}
203084af4da6SJohn Dyson 			}
203184af4da6SJohn Dyson 		}
203284af4da6SJohn Dyson 
2033fd3bf775SJohn Dyson 		ki = p->p_aioinfo;
203484af4da6SJohn Dyson 		if (ki) {
203584af4da6SJohn Dyson 			ki->kaio_buffer_finished_count++;
203684af4da6SJohn Dyson 			TAILQ_REMOVE(&aio_bufjobs, aiocbe, list);
203784af4da6SJohn Dyson 			TAILQ_REMOVE(&ki->kaio_bufqueue, aiocbe, plist);
203884af4da6SJohn Dyson 			TAILQ_INSERT_TAIL(&ki->kaio_bufdone, aiocbe, plist);
203984af4da6SJohn Dyson 			/*
204084af4da6SJohn Dyson 			 * and do the wakeup
204184af4da6SJohn Dyson 			 */
204284af4da6SJohn Dyson 			if (ki->kaio_flags & (KAIO_RUNDOWN|KAIO_WAKEUP)) {
2043fd3bf775SJohn Dyson 				ki->kaio_flags &= ~KAIO_WAKEUP;
2044fd3bf775SJohn Dyson 				wakeup(p);
2045fd3bf775SJohn Dyson 			}
2046fd3bf775SJohn Dyson 		}
2047fd3bf775SJohn Dyson 	}
204811783b14SJohn Dyson 	splx(s);
204984af4da6SJohn Dyson }
2050