xref: /freebsd/usr.sbin/bhyve/pci_ahci.c (revision 1f4bcc459a76b7aa664f3fd557684cd0ba6da352)
1 /*-
2  * Copyright (c) 2013  Zhixiang Yu <zcore@freebsd.org>
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND
15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24  * SUCH DAMAGE.
25  *
26  * $FreeBSD$
27  */
28 
29 #include <sys/cdefs.h>
30 __FBSDID("$FreeBSD$");
31 
32 #include <sys/param.h>
33 #include <sys/linker_set.h>
34 #include <sys/stat.h>
35 #include <sys/uio.h>
36 #include <sys/ioctl.h>
37 #include <sys/disk.h>
38 #include <sys/ata.h>
39 #include <sys/endian.h>
40 
41 #include <errno.h>
42 #include <fcntl.h>
43 #include <stdio.h>
44 #include <stdlib.h>
45 #include <stdint.h>
46 #include <string.h>
47 #include <strings.h>
48 #include <unistd.h>
49 #include <assert.h>
50 #include <pthread.h>
51 #include <pthread_np.h>
52 #include <inttypes.h>
53 #include <md5.h>
54 
55 #include "bhyverun.h"
56 #include "pci_emul.h"
57 #include "ahci.h"
58 #include "block_if.h"
59 
60 #define	MAX_PORTS	6	/* Intel ICH8 AHCI supports 6 ports */
61 
62 #define	PxSIG_ATA	0x00000101 /* ATA drive */
63 #define	PxSIG_ATAPI	0xeb140101 /* ATAPI drive */
64 
65 enum sata_fis_type {
66 	FIS_TYPE_REGH2D		= 0x27,	/* Register FIS - host to device */
67 	FIS_TYPE_REGD2H		= 0x34,	/* Register FIS - device to host */
68 	FIS_TYPE_DMAACT		= 0x39,	/* DMA activate FIS - device to host */
69 	FIS_TYPE_DMASETUP	= 0x41,	/* DMA setup FIS - bidirectional */
70 	FIS_TYPE_DATA		= 0x46,	/* Data FIS - bidirectional */
71 	FIS_TYPE_BIST		= 0x58,	/* BIST activate FIS - bidirectional */
72 	FIS_TYPE_PIOSETUP	= 0x5F,	/* PIO setup FIS - device to host */
73 	FIS_TYPE_SETDEVBITS	= 0xA1,	/* Set dev bits FIS - device to host */
74 };
75 
76 /*
77  * SCSI opcodes
78  */
79 #define	TEST_UNIT_READY		0x00
80 #define	REQUEST_SENSE		0x03
81 #define	INQUIRY			0x12
82 #define	START_STOP_UNIT		0x1B
83 #define	PREVENT_ALLOW		0x1E
84 #define	READ_CAPACITY		0x25
85 #define	READ_10			0x28
86 #define	POSITION_TO_ELEMENT	0x2B
87 #define	READ_TOC		0x43
88 #define	GET_EVENT_STATUS_NOTIFICATION 0x4A
89 #define	MODE_SENSE_10		0x5A
90 #define	REPORT_LUNS		0xA0
91 #define	READ_12			0xA8
92 #define	READ_CD			0xBE
93 
94 /*
95  * SCSI mode page codes
96  */
97 #define	MODEPAGE_RW_ERROR_RECOVERY	0x01
98 #define	MODEPAGE_CD_CAPABILITIES	0x2A
99 
100 /*
101  * ATA commands
102  */
103 #define	ATA_SF_ENAB_SATA_SF		0x10
104 #define		ATA_SATA_SF_AN		0x05
105 #define	ATA_SF_DIS_SATA_SF		0x90
106 
107 /*
108  * Debug printf
109  */
110 #ifdef AHCI_DEBUG
111 static FILE *dbg;
112 #define DPRINTF(format, arg...)	do{fprintf(dbg, format, ##arg);fflush(dbg);}while(0)
113 #else
114 #define DPRINTF(format, arg...)
115 #endif
116 #define WPRINTF(format, arg...) printf(format, ##arg)
117 
118 struct ahci_ioreq {
119 	struct blockif_req io_req;
120 	struct ahci_port *io_pr;
121 	STAILQ_ENTRY(ahci_ioreq) io_flist;
122 	TAILQ_ENTRY(ahci_ioreq) io_blist;
123 	uint8_t *cfis;
124 	uint32_t len;
125 	uint32_t done;
126 	int slot;
127 	int more;
128 };
129 
130 struct ahci_port {
131 	struct blockif_ctxt *bctx;
132 	struct pci_ahci_softc *pr_sc;
133 	uint8_t *cmd_lst;
134 	uint8_t *rfis;
135 	char ident[20 + 1];
136 	int atapi;
137 	int reset;
138 	int waitforclear;
139 	int mult_sectors;
140 	uint8_t xfermode;
141 	uint8_t err_cfis[20];
142 	uint8_t sense_key;
143 	uint8_t asc;
144 	u_int ccs;
145 	uint32_t pending;
146 
147 	uint32_t clb;
148 	uint32_t clbu;
149 	uint32_t fb;
150 	uint32_t fbu;
151 	uint32_t is;
152 	uint32_t ie;
153 	uint32_t cmd;
154 	uint32_t unused0;
155 	uint32_t tfd;
156 	uint32_t sig;
157 	uint32_t ssts;
158 	uint32_t sctl;
159 	uint32_t serr;
160 	uint32_t sact;
161 	uint32_t ci;
162 	uint32_t sntf;
163 	uint32_t fbs;
164 
165 	/*
166 	 * i/o request info
167 	 */
168 	struct ahci_ioreq *ioreq;
169 	int ioqsz;
170 	STAILQ_HEAD(ahci_fhead, ahci_ioreq) iofhd;
171 	TAILQ_HEAD(ahci_bhead, ahci_ioreq) iobhd;
172 };
173 
174 struct ahci_cmd_hdr {
175 	uint16_t flags;
176 	uint16_t prdtl;
177 	uint32_t prdbc;
178 	uint64_t ctba;
179 	uint32_t reserved[4];
180 };
181 
182 struct ahci_prdt_entry {
183 	uint64_t dba;
184 	uint32_t reserved;
185 #define	DBCMASK		0x3fffff
186 	uint32_t dbc;
187 };
188 
189 struct pci_ahci_softc {
190 	struct pci_devinst *asc_pi;
191 	pthread_mutex_t	mtx;
192 	int ports;
193 	uint32_t cap;
194 	uint32_t ghc;
195 	uint32_t is;
196 	uint32_t pi;
197 	uint32_t vs;
198 	uint32_t ccc_ctl;
199 	uint32_t ccc_pts;
200 	uint32_t em_loc;
201 	uint32_t em_ctl;
202 	uint32_t cap2;
203 	uint32_t bohc;
204 	uint32_t lintr;
205 	struct ahci_port port[MAX_PORTS];
206 };
207 #define	ahci_ctx(sc)	((sc)->asc_pi->pi_vmctx)
208 
209 static void ahci_handle_port(struct ahci_port *p);
210 
211 static inline void lba_to_msf(uint8_t *buf, int lba)
212 {
213 	lba += 150;
214 	buf[0] = (lba / 75) / 60;
215 	buf[1] = (lba / 75) % 60;
216 	buf[2] = lba % 75;
217 }
218 
219 /*
220  * generate HBA intr depending on whether or not ports within
221  * the controller have an interrupt pending.
222  */
223 static void
224 ahci_generate_intr(struct pci_ahci_softc *sc)
225 {
226 	struct pci_devinst *pi;
227 	int i;
228 
229 	pi = sc->asc_pi;
230 
231 	for (i = 0; i < sc->ports; i++) {
232 		struct ahci_port *pr;
233 		pr = &sc->port[i];
234 		if (pr->is & pr->ie)
235 			sc->is |= (1 << i);
236 	}
237 
238 	DPRINTF("%s %x\n", __func__, sc->is);
239 
240 	if (sc->is && (sc->ghc & AHCI_GHC_IE)) {
241 		if (pci_msi_enabled(pi)) {
242 			/*
243 			 * Generate an MSI interrupt on every edge
244 			 */
245 			pci_generate_msi(pi, 0);
246 		} else if (!sc->lintr) {
247 			/*
248 			 * Only generate a pin-based interrupt if one wasn't
249 			 * in progress
250 			 */
251 			sc->lintr = 1;
252 			pci_lintr_assert(pi);
253 		}
254 	} else if (sc->lintr) {
255 		/*
256 		 * No interrupts: deassert pin-based signal if it had
257 		 * been asserted
258 		 */
259 		pci_lintr_deassert(pi);
260 		sc->lintr = 0;
261 	}
262 }
263 
264 static void
265 ahci_write_fis(struct ahci_port *p, enum sata_fis_type ft, uint8_t *fis)
266 {
267 	int offset, len, irq;
268 
269 	if (p->rfis == NULL || !(p->cmd & AHCI_P_CMD_FRE))
270 		return;
271 
272 	switch (ft) {
273 	case FIS_TYPE_REGD2H:
274 		offset = 0x40;
275 		len = 20;
276 		irq = (fis[1] & (1 << 6)) ? AHCI_P_IX_DHR : 0;
277 		break;
278 	case FIS_TYPE_SETDEVBITS:
279 		offset = 0x58;
280 		len = 8;
281 		irq = (fis[1] & (1 << 6)) ? AHCI_P_IX_SDB : 0;
282 		break;
283 	case FIS_TYPE_PIOSETUP:
284 		offset = 0x20;
285 		len = 20;
286 		irq = (fis[1] & (1 << 6)) ? AHCI_P_IX_PS : 0;
287 		break;
288 	default:
289 		WPRINTF("unsupported fis type %d\n", ft);
290 		return;
291 	}
292 	if (fis[2] & ATA_S_ERROR) {
293 		p->waitforclear = 1;
294 		irq |= AHCI_P_IX_TFE;
295 	}
296 	memcpy(p->rfis + offset, fis, len);
297 	if (irq) {
298 		p->is |= irq;
299 		ahci_generate_intr(p->pr_sc);
300 	}
301 }
302 
303 static void
304 ahci_write_fis_piosetup(struct ahci_port *p)
305 {
306 	uint8_t fis[20];
307 
308 	memset(fis, 0, sizeof(fis));
309 	fis[0] = FIS_TYPE_PIOSETUP;
310 	ahci_write_fis(p, FIS_TYPE_PIOSETUP, fis);
311 }
312 
313 static void
314 ahci_write_fis_sdb(struct ahci_port *p, int slot, uint8_t *cfis, uint32_t tfd)
315 {
316 	uint8_t fis[8];
317 	uint8_t error;
318 
319 	error = (tfd >> 8) & 0xff;
320 	tfd &= 0x77;
321 	memset(fis, 0, sizeof(fis));
322 	fis[0] = FIS_TYPE_SETDEVBITS;
323 	fis[1] = (1 << 6);
324 	fis[2] = tfd;
325 	fis[3] = error;
326 	if (fis[2] & ATA_S_ERROR) {
327 		p->err_cfis[0] = slot;
328 		p->err_cfis[2] = tfd;
329 		p->err_cfis[3] = error;
330 		memcpy(&p->err_cfis[4], cfis + 4, 16);
331 	} else {
332 		*(uint32_t *)(fis + 4) = (1 << slot);
333 		p->sact &= ~(1 << slot);
334 	}
335 	p->tfd &= ~0x77;
336 	p->tfd |= tfd;
337 	ahci_write_fis(p, FIS_TYPE_SETDEVBITS, fis);
338 }
339 
340 static void
341 ahci_write_fis_d2h(struct ahci_port *p, int slot, uint8_t *cfis, uint32_t tfd)
342 {
343 	uint8_t fis[20];
344 	uint8_t error;
345 
346 	error = (tfd >> 8) & 0xff;
347 	memset(fis, 0, sizeof(fis));
348 	fis[0] = FIS_TYPE_REGD2H;
349 	fis[1] = (1 << 6);
350 	fis[2] = tfd & 0xff;
351 	fis[3] = error;
352 	fis[4] = cfis[4];
353 	fis[5] = cfis[5];
354 	fis[6] = cfis[6];
355 	fis[7] = cfis[7];
356 	fis[8] = cfis[8];
357 	fis[9] = cfis[9];
358 	fis[10] = cfis[10];
359 	fis[11] = cfis[11];
360 	fis[12] = cfis[12];
361 	fis[13] = cfis[13];
362 	if (fis[2] & ATA_S_ERROR) {
363 		p->err_cfis[0] = 0x80;
364 		p->err_cfis[2] = tfd & 0xff;
365 		p->err_cfis[3] = error;
366 		memcpy(&p->err_cfis[4], cfis + 4, 16);
367 	} else
368 		p->ci &= ~(1 << slot);
369 	p->tfd = tfd;
370 	ahci_write_fis(p, FIS_TYPE_REGD2H, fis);
371 }
372 
373 static void
374 ahci_write_fis_d2h_ncq(struct ahci_port *p, int slot)
375 {
376 	uint8_t fis[20];
377 
378 	p->tfd = ATA_S_READY | ATA_S_DSC;
379 	memset(fis, 0, sizeof(fis));
380 	fis[0] = FIS_TYPE_REGD2H;
381 	fis[1] = 0;			/* No interrupt */
382 	fis[2] = p->tfd;		/* Status */
383 	fis[3] = 0;			/* No error */
384 	p->ci &= ~(1 << slot);
385 	ahci_write_fis(p, FIS_TYPE_REGD2H, fis);
386 }
387 
388 static void
389 ahci_write_reset_fis_d2h(struct ahci_port *p)
390 {
391 	uint8_t fis[20];
392 
393 	memset(fis, 0, sizeof(fis));
394 	fis[0] = FIS_TYPE_REGD2H;
395 	fis[3] = 1;
396 	fis[4] = 1;
397 	if (p->atapi) {
398 		fis[5] = 0x14;
399 		fis[6] = 0xeb;
400 	}
401 	fis[12] = 1;
402 	ahci_write_fis(p, FIS_TYPE_REGD2H, fis);
403 }
404 
405 static void
406 ahci_check_stopped(struct ahci_port *p)
407 {
408 	/*
409 	 * If we are no longer processing the command list and nothing
410 	 * is in-flight, clear the running bit, the current command
411 	 * slot, the command issue and active bits.
412 	 */
413 	if (!(p->cmd & AHCI_P_CMD_ST)) {
414 		if (p->pending == 0) {
415 			p->ccs = 0;
416 			p->cmd &= ~(AHCI_P_CMD_CR | AHCI_P_CMD_CCS_MASK);
417 			p->ci = 0;
418 			p->sact = 0;
419 			p->waitforclear = 0;
420 		}
421 	}
422 }
423 
424 static void
425 ahci_port_stop(struct ahci_port *p)
426 {
427 	struct ahci_ioreq *aior;
428 	uint8_t *cfis;
429 	int slot;
430 	int ncq;
431 	int error;
432 
433 	assert(pthread_mutex_isowned_np(&p->pr_sc->mtx));
434 
435 	TAILQ_FOREACH(aior, &p->iobhd, io_blist) {
436 		/*
437 		 * Try to cancel the outstanding blockif request.
438 		 */
439 		error = blockif_cancel(p->bctx, &aior->io_req);
440 		if (error != 0)
441 			continue;
442 
443 		slot = aior->slot;
444 		cfis = aior->cfis;
445 		if (cfis[2] == ATA_WRITE_FPDMA_QUEUED ||
446 		    cfis[2] == ATA_READ_FPDMA_QUEUED ||
447 		    cfis[2] == ATA_SEND_FPDMA_QUEUED)
448 			ncq = 1;
449 
450 		if (ncq)
451 			p->sact &= ~(1 << slot);
452 		else
453 			p->ci &= ~(1 << slot);
454 
455 		/*
456 		 * This command is now done.
457 		 */
458 		p->pending &= ~(1 << slot);
459 
460 		/*
461 		 * Delete the blockif request from the busy list
462 		 */
463 		TAILQ_REMOVE(&p->iobhd, aior, io_blist);
464 
465 		/*
466 		 * Move the blockif request back to the free list
467 		 */
468 		STAILQ_INSERT_TAIL(&p->iofhd, aior, io_flist);
469 	}
470 
471 	ahci_check_stopped(p);
472 }
473 
474 static void
475 ahci_port_reset(struct ahci_port *pr)
476 {
477 	pr->serr = 0;
478 	pr->sact = 0;
479 	pr->xfermode = ATA_UDMA6;
480 	pr->mult_sectors = 128;
481 
482 	if (!pr->bctx) {
483 		pr->ssts = ATA_SS_DET_NO_DEVICE;
484 		pr->sig = 0xFFFFFFFF;
485 		pr->tfd = 0x7F;
486 		return;
487 	}
488 	pr->ssts = ATA_SS_DET_PHY_ONLINE | ATA_SS_IPM_ACTIVE;
489 	if (pr->sctl & ATA_SC_SPD_MASK)
490 		pr->ssts |= (pr->sctl & ATA_SC_SPD_MASK);
491 	else
492 		pr->ssts |= ATA_SS_SPD_GEN3;
493 	pr->tfd = (1 << 8) | ATA_S_DSC | ATA_S_DMA;
494 	if (!pr->atapi) {
495 		pr->sig = PxSIG_ATA;
496 		pr->tfd |= ATA_S_READY;
497 	} else
498 		pr->sig = PxSIG_ATAPI;
499 	ahci_write_reset_fis_d2h(pr);
500 }
501 
502 static void
503 ahci_reset(struct pci_ahci_softc *sc)
504 {
505 	int i;
506 
507 	sc->ghc = AHCI_GHC_AE;
508 	sc->is = 0;
509 
510 	if (sc->lintr) {
511 		pci_lintr_deassert(sc->asc_pi);
512 		sc->lintr = 0;
513 	}
514 
515 	for (i = 0; i < sc->ports; i++) {
516 		sc->port[i].ie = 0;
517 		sc->port[i].is = 0;
518 		sc->port[i].cmd = (AHCI_P_CMD_SUD | AHCI_P_CMD_POD);
519 		if (sc->port[i].bctx)
520 			sc->port[i].cmd |= AHCI_P_CMD_CPS;
521 		sc->port[i].sctl = 0;
522 		ahci_port_reset(&sc->port[i]);
523 	}
524 }
525 
526 static void
527 ata_string(uint8_t *dest, const char *src, int len)
528 {
529 	int i;
530 
531 	for (i = 0; i < len; i++) {
532 		if (*src)
533 			dest[i ^ 1] = *src++;
534 		else
535 			dest[i ^ 1] = ' ';
536 	}
537 }
538 
539 static void
540 atapi_string(uint8_t *dest, const char *src, int len)
541 {
542 	int i;
543 
544 	for (i = 0; i < len; i++) {
545 		if (*src)
546 			dest[i] = *src++;
547 		else
548 			dest[i] = ' ';
549 	}
550 }
551 
552 /*
553  * Build up the iovec based on the PRDT, 'done' and 'len'.
554  */
555 static void
556 ahci_build_iov(struct ahci_port *p, struct ahci_ioreq *aior,
557     struct ahci_prdt_entry *prdt, uint16_t prdtl)
558 {
559 	struct blockif_req *breq = &aior->io_req;
560 	int i, j, skip, todo, left, extra;
561 	uint32_t dbcsz;
562 
563 	/* Copy part of PRDT between 'done' and 'len' bytes into the iov. */
564 	skip = aior->done;
565 	left = aior->len - aior->done;
566 	todo = 0;
567 	for (i = 0, j = 0; i < prdtl && j < BLOCKIF_IOV_MAX && left > 0;
568 	    i++, prdt++) {
569 		dbcsz = (prdt->dbc & DBCMASK) + 1;
570 		/* Skip already done part of the PRDT */
571 		if (dbcsz <= skip) {
572 			skip -= dbcsz;
573 			continue;
574 		}
575 		dbcsz -= skip;
576 		if (dbcsz > left)
577 			dbcsz = left;
578 		breq->br_iov[j].iov_base = paddr_guest2host(ahci_ctx(p->pr_sc),
579 		    prdt->dba + skip, dbcsz);
580 		breq->br_iov[j].iov_len = dbcsz;
581 		todo += dbcsz;
582 		left -= dbcsz;
583 		skip = 0;
584 		j++;
585 	}
586 
587 	/* If we got limited by IOV length, round I/O down to sector size. */
588 	if (j == BLOCKIF_IOV_MAX) {
589 		extra = todo % blockif_sectsz(p->bctx);
590 		todo -= extra;
591 		assert(todo > 0);
592 		while (extra > 0) {
593 			if (breq->br_iov[j - 1].iov_len > extra) {
594 				breq->br_iov[j - 1].iov_len -= extra;
595 				break;
596 			}
597 			extra -= breq->br_iov[j - 1].iov_len;
598 			j--;
599 		}
600 	}
601 
602 	breq->br_iovcnt = j;
603 	breq->br_resid = todo;
604 	aior->done += todo;
605 	aior->more = (aior->done < aior->len && i < prdtl);
606 }
607 
608 static void
609 ahci_handle_rw(struct ahci_port *p, int slot, uint8_t *cfis, uint32_t done)
610 {
611 	struct ahci_ioreq *aior;
612 	struct blockif_req *breq;
613 	struct ahci_prdt_entry *prdt;
614 	struct ahci_cmd_hdr *hdr;
615 	uint64_t lba;
616 	uint32_t len;
617 	int err, first, ncq, readop;
618 
619 	prdt = (struct ahci_prdt_entry *)(cfis + 0x80);
620 	hdr = (struct ahci_cmd_hdr *)(p->cmd_lst + slot * AHCI_CL_SIZE);
621 	ncq = 0;
622 	readop = 1;
623 	first = (done == 0);
624 
625 	if (cfis[2] == ATA_WRITE || cfis[2] == ATA_WRITE48 ||
626 	    cfis[2] == ATA_WRITE_MUL || cfis[2] == ATA_WRITE_MUL48 ||
627 	    cfis[2] == ATA_WRITE_DMA || cfis[2] == ATA_WRITE_DMA48 ||
628 	    cfis[2] == ATA_WRITE_FPDMA_QUEUED)
629 		readop = 0;
630 
631 	if (cfis[2] == ATA_WRITE_FPDMA_QUEUED ||
632 	    cfis[2] == ATA_READ_FPDMA_QUEUED) {
633 		lba = ((uint64_t)cfis[10] << 40) |
634 			((uint64_t)cfis[9] << 32) |
635 			((uint64_t)cfis[8] << 24) |
636 			((uint64_t)cfis[6] << 16) |
637 			((uint64_t)cfis[5] << 8) |
638 			cfis[4];
639 		len = cfis[11] << 8 | cfis[3];
640 		if (!len)
641 			len = 65536;
642 		ncq = 1;
643 	} else if (cfis[2] == ATA_READ48 || cfis[2] == ATA_WRITE48 ||
644 	    cfis[2] == ATA_READ_MUL48 || cfis[2] == ATA_WRITE_MUL48 ||
645 	    cfis[2] == ATA_READ_DMA48 || cfis[2] == ATA_WRITE_DMA48) {
646 		lba = ((uint64_t)cfis[10] << 40) |
647 			((uint64_t)cfis[9] << 32) |
648 			((uint64_t)cfis[8] << 24) |
649 			((uint64_t)cfis[6] << 16) |
650 			((uint64_t)cfis[5] << 8) |
651 			cfis[4];
652 		len = cfis[13] << 8 | cfis[12];
653 		if (!len)
654 			len = 65536;
655 	} else {
656 		lba = ((cfis[7] & 0xf) << 24) | (cfis[6] << 16) |
657 			(cfis[5] << 8) | cfis[4];
658 		len = cfis[12];
659 		if (!len)
660 			len = 256;
661 	}
662 	lba *= blockif_sectsz(p->bctx);
663 	len *= blockif_sectsz(p->bctx);
664 
665 	/* Pull request off free list */
666 	aior = STAILQ_FIRST(&p->iofhd);
667 	assert(aior != NULL);
668 	STAILQ_REMOVE_HEAD(&p->iofhd, io_flist);
669 
670 	aior->cfis = cfis;
671 	aior->slot = slot;
672 	aior->len = len;
673 	aior->done = done;
674 	breq = &aior->io_req;
675 	breq->br_offset = lba + done;
676 	ahci_build_iov(p, aior, prdt, hdr->prdtl);
677 
678 	/* Mark this command in-flight. */
679 	p->pending |= 1 << slot;
680 
681 	/* Stuff request onto busy list. */
682 	TAILQ_INSERT_HEAD(&p->iobhd, aior, io_blist);
683 
684 	if (ncq && first)
685 		ahci_write_fis_d2h_ncq(p, slot);
686 
687 	if (readop)
688 		err = blockif_read(p->bctx, breq);
689 	else
690 		err = blockif_write(p->bctx, breq);
691 	assert(err == 0);
692 }
693 
694 static void
695 ahci_handle_flush(struct ahci_port *p, int slot, uint8_t *cfis)
696 {
697 	struct ahci_ioreq *aior;
698 	struct blockif_req *breq;
699 	int err;
700 
701 	/*
702 	 * Pull request off free list
703 	 */
704 	aior = STAILQ_FIRST(&p->iofhd);
705 	assert(aior != NULL);
706 	STAILQ_REMOVE_HEAD(&p->iofhd, io_flist);
707 	aior->cfis = cfis;
708 	aior->slot = slot;
709 	aior->len = 0;
710 	aior->done = 0;
711 	aior->more = 0;
712 	breq = &aior->io_req;
713 
714 	/*
715 	 * Mark this command in-flight.
716 	 */
717 	p->pending |= 1 << slot;
718 
719 	/*
720 	 * Stuff request onto busy list
721 	 */
722 	TAILQ_INSERT_HEAD(&p->iobhd, aior, io_blist);
723 
724 	err = blockif_flush(p->bctx, breq);
725 	assert(err == 0);
726 }
727 
728 static inline void
729 read_prdt(struct ahci_port *p, int slot, uint8_t *cfis,
730 		void *buf, int size)
731 {
732 	struct ahci_cmd_hdr *hdr;
733 	struct ahci_prdt_entry *prdt;
734 	void *to;
735 	int i, len;
736 
737 	hdr = (struct ahci_cmd_hdr *)(p->cmd_lst + slot * AHCI_CL_SIZE);
738 	len = size;
739 	to = buf;
740 	prdt = (struct ahci_prdt_entry *)(cfis + 0x80);
741 	for (i = 0; i < hdr->prdtl && len; i++) {
742 		uint8_t *ptr;
743 		uint32_t dbcsz;
744 		int sublen;
745 
746 		dbcsz = (prdt->dbc & DBCMASK) + 1;
747 		ptr = paddr_guest2host(ahci_ctx(p->pr_sc), prdt->dba, dbcsz);
748 		sublen = len < dbcsz ? len : dbcsz;
749 		memcpy(to, ptr, sublen);
750 		len -= sublen;
751 		to += sublen;
752 		prdt++;
753 	}
754 }
755 
756 static void
757 ahci_handle_dsm_trim(struct ahci_port *p, int slot, uint8_t *cfis, uint32_t done)
758 {
759 	struct ahci_ioreq *aior;
760 	struct blockif_req *breq;
761 	uint8_t *entry;
762 	uint64_t elba;
763 	uint32_t len, elen;
764 	int err, first, ncq;
765 	uint8_t buf[512];
766 
767 	first = (done == 0);
768 	if (cfis[2] == ATA_DATA_SET_MANAGEMENT) {
769 		len = (uint16_t)cfis[13] << 8 | cfis[12];
770 		len *= 512;
771 		ncq = 0;
772 	} else { /* ATA_SEND_FPDMA_QUEUED */
773 		len = (uint16_t)cfis[11] << 8 | cfis[3];
774 		len *= 512;
775 		ncq = 1;
776 	}
777 	read_prdt(p, slot, cfis, buf, sizeof(buf));
778 
779 next:
780 	entry = &buf[done];
781 	elba = ((uint64_t)entry[5] << 40) |
782 		((uint64_t)entry[4] << 32) |
783 		((uint64_t)entry[3] << 24) |
784 		((uint64_t)entry[2] << 16) |
785 		((uint64_t)entry[1] << 8) |
786 		entry[0];
787 	elen = (uint16_t)entry[7] << 8 | entry[6];
788 	done += 8;
789 	if (elen == 0) {
790 		if (done >= len) {
791 			ahci_write_fis_d2h(p, slot, cfis, ATA_S_READY | ATA_S_DSC);
792 			p->pending &= ~(1 << slot);
793 			ahci_check_stopped(p);
794 			if (!first)
795 				ahci_handle_port(p);
796 			return;
797 		}
798 		goto next;
799 	}
800 
801 	/*
802 	 * Pull request off free list
803 	 */
804 	aior = STAILQ_FIRST(&p->iofhd);
805 	assert(aior != NULL);
806 	STAILQ_REMOVE_HEAD(&p->iofhd, io_flist);
807 	aior->cfis = cfis;
808 	aior->slot = slot;
809 	aior->len = len;
810 	aior->done = done;
811 	aior->more = (len != done);
812 
813 	breq = &aior->io_req;
814 	breq->br_offset = elba * blockif_sectsz(p->bctx);
815 	breq->br_resid = elen * blockif_sectsz(p->bctx);
816 
817 	/*
818 	 * Mark this command in-flight.
819 	 */
820 	p->pending |= 1 << slot;
821 
822 	/*
823 	 * Stuff request onto busy list
824 	 */
825 	TAILQ_INSERT_HEAD(&p->iobhd, aior, io_blist);
826 
827 	if (ncq && first)
828 		ahci_write_fis_d2h_ncq(p, slot);
829 
830 	err = blockif_delete(p->bctx, breq);
831 	assert(err == 0);
832 }
833 
834 static inline void
835 write_prdt(struct ahci_port *p, int slot, uint8_t *cfis,
836 		void *buf, int size)
837 {
838 	struct ahci_cmd_hdr *hdr;
839 	struct ahci_prdt_entry *prdt;
840 	void *from;
841 	int i, len;
842 
843 	hdr = (struct ahci_cmd_hdr *)(p->cmd_lst + slot * AHCI_CL_SIZE);
844 	len = size;
845 	from = buf;
846 	prdt = (struct ahci_prdt_entry *)(cfis + 0x80);
847 	for (i = 0; i < hdr->prdtl && len; i++) {
848 		uint8_t *ptr;
849 		uint32_t dbcsz;
850 		int sublen;
851 
852 		dbcsz = (prdt->dbc & DBCMASK) + 1;
853 		ptr = paddr_guest2host(ahci_ctx(p->pr_sc), prdt->dba, dbcsz);
854 		sublen = len < dbcsz ? len : dbcsz;
855 		memcpy(ptr, from, sublen);
856 		len -= sublen;
857 		from += sublen;
858 		prdt++;
859 	}
860 	hdr->prdbc = size - len;
861 }
862 
863 static void
864 ahci_checksum(uint8_t *buf, int size)
865 {
866 	int i;
867 	uint8_t sum = 0;
868 
869 	for (i = 0; i < size - 1; i++)
870 		sum += buf[i];
871 	buf[size - 1] = 0x100 - sum;
872 }
873 
874 static void
875 ahci_handle_read_log(struct ahci_port *p, int slot, uint8_t *cfis)
876 {
877 	struct ahci_cmd_hdr *hdr;
878 	uint8_t buf[512];
879 
880 	hdr = (struct ahci_cmd_hdr *)(p->cmd_lst + slot * AHCI_CL_SIZE);
881 	if (p->atapi || hdr->prdtl == 0 || cfis[4] != 0x10 ||
882 	    cfis[5] != 0 || cfis[9] != 0 || cfis[12] != 1 || cfis[13] != 0) {
883 		ahci_write_fis_d2h(p, slot, cfis,
884 		    (ATA_E_ABORT << 8) | ATA_S_READY | ATA_S_ERROR);
885 		return;
886 	}
887 
888 	memset(buf, 0, sizeof(buf));
889 	memcpy(buf, p->err_cfis, sizeof(p->err_cfis));
890 	ahci_checksum(buf, sizeof(buf));
891 
892 	if (cfis[2] == ATA_READ_LOG_EXT)
893 		ahci_write_fis_piosetup(p);
894 	write_prdt(p, slot, cfis, (void *)buf, sizeof(buf));
895 	ahci_write_fis_d2h(p, slot, cfis, ATA_S_DSC | ATA_S_READY);
896 }
897 
898 static void
899 handle_identify(struct ahci_port *p, int slot, uint8_t *cfis)
900 {
901 	struct ahci_cmd_hdr *hdr;
902 
903 	hdr = (struct ahci_cmd_hdr *)(p->cmd_lst + slot * AHCI_CL_SIZE);
904 	if (p->atapi || hdr->prdtl == 0) {
905 		ahci_write_fis_d2h(p, slot, cfis,
906 		    (ATA_E_ABORT << 8) | ATA_S_READY | ATA_S_ERROR);
907 	} else {
908 		uint16_t buf[256];
909 		uint64_t sectors;
910 		int sectsz, psectsz, psectoff, candelete, ro;
911 		uint16_t cyl;
912 		uint8_t sech, heads;
913 
914 		ro = blockif_is_ro(p->bctx);
915 		candelete = blockif_candelete(p->bctx);
916 		sectsz = blockif_sectsz(p->bctx);
917 		sectors = blockif_size(p->bctx) / sectsz;
918 		blockif_chs(p->bctx, &cyl, &heads, &sech);
919 		blockif_psectsz(p->bctx, &psectsz, &psectoff);
920 		memset(buf, 0, sizeof(buf));
921 		buf[0] = 0x0040;
922 		buf[1] = cyl;
923 		buf[3] = heads;
924 		buf[6] = sech;
925 		ata_string((uint8_t *)(buf+10), p->ident, 20);
926 		ata_string((uint8_t *)(buf+23), "001", 8);
927 		ata_string((uint8_t *)(buf+27), "BHYVE SATA DISK", 40);
928 		buf[47] = (0x8000 | 128);
929 		buf[48] = 0;
930 		buf[49] = (1 << 8 | 1 << 9 | 1 << 11);
931 		buf[50] = (1 << 14);
932 		buf[53] = (1 << 1 | 1 << 2);
933 		if (p->mult_sectors)
934 			buf[59] = (0x100 | p->mult_sectors);
935 		if (sectors <= 0x0fffffff) {
936 			buf[60] = sectors;
937 			buf[61] = (sectors >> 16);
938 		} else {
939 			buf[60] = 0xffff;
940 			buf[61] = 0x0fff;
941 		}
942 		buf[63] = 0x7;
943 		if (p->xfermode & ATA_WDMA0)
944 			buf[63] |= (1 << ((p->xfermode & 7) + 8));
945 		buf[64] = 0x3;
946 		buf[65] = 120;
947 		buf[66] = 120;
948 		buf[67] = 120;
949 		buf[68] = 120;
950 		buf[69] = 0;
951 		buf[75] = 31;
952 		buf[76] = (ATA_SATA_GEN1 | ATA_SATA_GEN2 | ATA_SATA_GEN3 |
953 			   ATA_SUPPORT_NCQ);
954 		buf[77] = (ATA_SUPPORT_RCVSND_FPDMA_QUEUED |
955 			   (p->ssts & ATA_SS_SPD_MASK) >> 3);
956 		buf[80] = 0x3f0;
957 		buf[81] = 0x28;
958 		buf[82] = (ATA_SUPPORT_POWERMGT | ATA_SUPPORT_WRITECACHE|
959 			   ATA_SUPPORT_LOOKAHEAD | ATA_SUPPORT_NOP);
960 		buf[83] = (ATA_SUPPORT_ADDRESS48 | ATA_SUPPORT_FLUSHCACHE |
961 			   ATA_SUPPORT_FLUSHCACHE48 | 1 << 14);
962 		buf[84] = (1 << 14);
963 		buf[85] = (ATA_SUPPORT_POWERMGT | ATA_SUPPORT_WRITECACHE|
964 			   ATA_SUPPORT_LOOKAHEAD | ATA_SUPPORT_NOP);
965 		buf[86] = (ATA_SUPPORT_ADDRESS48 | ATA_SUPPORT_FLUSHCACHE |
966 			   ATA_SUPPORT_FLUSHCACHE48 | 1 << 15);
967 		buf[87] = (1 << 14);
968 		buf[88] = 0x7f;
969 		if (p->xfermode & ATA_UDMA0)
970 			buf[88] |= (1 << ((p->xfermode & 7) + 8));
971 		buf[100] = sectors;
972 		buf[101] = (sectors >> 16);
973 		buf[102] = (sectors >> 32);
974 		buf[103] = (sectors >> 48);
975 		if (candelete && !ro) {
976 			buf[69] |= ATA_SUPPORT_RZAT | ATA_SUPPORT_DRAT;
977 			buf[105] = 1;
978 			buf[169] = ATA_SUPPORT_DSM_TRIM;
979 		}
980 		buf[106] = 0x4000;
981 		buf[209] = 0x4000;
982 		if (psectsz > sectsz) {
983 			buf[106] |= 0x2000;
984 			buf[106] |= ffsl(psectsz / sectsz) - 1;
985 			buf[209] |= (psectoff / sectsz);
986 		}
987 		if (sectsz > 512) {
988 			buf[106] |= 0x1000;
989 			buf[117] = sectsz / 2;
990 			buf[118] = ((sectsz / 2) >> 16);
991 		}
992 		buf[119] = (ATA_SUPPORT_RWLOGDMAEXT | 1 << 14);
993 		buf[120] = (ATA_SUPPORT_RWLOGDMAEXT | 1 << 14);
994 		buf[222] = 0x1020;
995 		buf[255] = 0x00a5;
996 		ahci_checksum((uint8_t *)buf, sizeof(buf));
997 		ahci_write_fis_piosetup(p);
998 		write_prdt(p, slot, cfis, (void *)buf, sizeof(buf));
999 		ahci_write_fis_d2h(p, slot, cfis, ATA_S_DSC | ATA_S_READY);
1000 	}
1001 }
1002 
1003 static void
1004 handle_atapi_identify(struct ahci_port *p, int slot, uint8_t *cfis)
1005 {
1006 	if (!p->atapi) {
1007 		ahci_write_fis_d2h(p, slot, cfis,
1008 		    (ATA_E_ABORT << 8) | ATA_S_READY | ATA_S_ERROR);
1009 	} else {
1010 		uint16_t buf[256];
1011 
1012 		memset(buf, 0, sizeof(buf));
1013 		buf[0] = (2 << 14 | 5 << 8 | 1 << 7 | 2 << 5);
1014 		ata_string((uint8_t *)(buf+10), p->ident, 20);
1015 		ata_string((uint8_t *)(buf+23), "001", 8);
1016 		ata_string((uint8_t *)(buf+27), "BHYVE SATA DVD ROM", 40);
1017 		buf[49] = (1 << 9 | 1 << 8);
1018 		buf[50] = (1 << 14 | 1);
1019 		buf[53] = (1 << 2 | 1 << 1);
1020 		buf[62] = 0x3f;
1021 		buf[63] = 7;
1022 		if (p->xfermode & ATA_WDMA0)
1023 			buf[63] |= (1 << ((p->xfermode & 7) + 8));
1024 		buf[64] = 3;
1025 		buf[65] = 120;
1026 		buf[66] = 120;
1027 		buf[67] = 120;
1028 		buf[68] = 120;
1029 		buf[76] = (ATA_SATA_GEN1 | ATA_SATA_GEN2 | ATA_SATA_GEN3);
1030 		buf[77] = ((p->ssts & ATA_SS_SPD_MASK) >> 3);
1031 		buf[78] = (1 << 5);
1032 		buf[80] = 0x3f0;
1033 		buf[82] = (ATA_SUPPORT_POWERMGT | ATA_SUPPORT_PACKET |
1034 			   ATA_SUPPORT_RESET | ATA_SUPPORT_NOP);
1035 		buf[83] = (1 << 14);
1036 		buf[84] = (1 << 14);
1037 		buf[85] = (ATA_SUPPORT_POWERMGT | ATA_SUPPORT_PACKET |
1038 			   ATA_SUPPORT_RESET | ATA_SUPPORT_NOP);
1039 		buf[87] = (1 << 14);
1040 		buf[88] = 0x7f;
1041 		if (p->xfermode & ATA_UDMA0)
1042 			buf[88] |= (1 << ((p->xfermode & 7) + 8));
1043 		buf[222] = 0x1020;
1044 		buf[255] = 0x00a5;
1045 		ahci_checksum((uint8_t *)buf, sizeof(buf));
1046 		ahci_write_fis_piosetup(p);
1047 		write_prdt(p, slot, cfis, (void *)buf, sizeof(buf));
1048 		ahci_write_fis_d2h(p, slot, cfis, ATA_S_DSC | ATA_S_READY);
1049 	}
1050 }
1051 
1052 static void
1053 atapi_inquiry(struct ahci_port *p, int slot, uint8_t *cfis)
1054 {
1055 	uint8_t buf[36];
1056 	uint8_t *acmd;
1057 	int len;
1058 	uint32_t tfd;
1059 
1060 	acmd = cfis + 0x40;
1061 
1062 	if (acmd[1] & 1) {		/* VPD */
1063 		if (acmd[2] == 0) {	/* Supported VPD pages */
1064 			buf[0] = 0x05;
1065 			buf[1] = 0;
1066 			buf[2] = 0;
1067 			buf[3] = 1;
1068 			buf[4] = 0;
1069 			len = 4 + buf[3];
1070 		} else {
1071 			p->sense_key = ATA_SENSE_ILLEGAL_REQUEST;
1072 			p->asc = 0x24;
1073 			tfd = (p->sense_key << 12) | ATA_S_READY | ATA_S_ERROR;
1074 			cfis[4] = (cfis[4] & ~7) | ATA_I_CMD | ATA_I_IN;
1075 			ahci_write_fis_d2h(p, slot, cfis, tfd);
1076 			return;
1077 		}
1078 	} else {
1079 		buf[0] = 0x05;
1080 		buf[1] = 0x80;
1081 		buf[2] = 0x00;
1082 		buf[3] = 0x21;
1083 		buf[4] = 31;
1084 		buf[5] = 0;
1085 		buf[6] = 0;
1086 		buf[7] = 0;
1087 		atapi_string(buf + 8, "BHYVE", 8);
1088 		atapi_string(buf + 16, "BHYVE DVD-ROM", 16);
1089 		atapi_string(buf + 32, "001", 4);
1090 		len = sizeof(buf);
1091 	}
1092 
1093 	if (len > acmd[4])
1094 		len = acmd[4];
1095 	cfis[4] = (cfis[4] & ~7) | ATA_I_CMD | ATA_I_IN;
1096 	write_prdt(p, slot, cfis, buf, len);
1097 	ahci_write_fis_d2h(p, slot, cfis, ATA_S_READY | ATA_S_DSC);
1098 }
1099 
1100 static void
1101 atapi_read_capacity(struct ahci_port *p, int slot, uint8_t *cfis)
1102 {
1103 	uint8_t buf[8];
1104 	uint64_t sectors;
1105 
1106 	sectors = blockif_size(p->bctx) / 2048;
1107 	be32enc(buf, sectors - 1);
1108 	be32enc(buf + 4, 2048);
1109 	cfis[4] = (cfis[4] & ~7) | ATA_I_CMD | ATA_I_IN;
1110 	write_prdt(p, slot, cfis, buf, sizeof(buf));
1111 	ahci_write_fis_d2h(p, slot, cfis, ATA_S_READY | ATA_S_DSC);
1112 }
1113 
1114 static void
1115 atapi_read_toc(struct ahci_port *p, int slot, uint8_t *cfis)
1116 {
1117 	uint8_t *acmd;
1118 	uint8_t format;
1119 	int len;
1120 
1121 	acmd = cfis + 0x40;
1122 
1123 	len = be16dec(acmd + 7);
1124 	format = acmd[9] >> 6;
1125 	switch (format) {
1126 	case 0:
1127 	{
1128 		int msf, size;
1129 		uint64_t sectors;
1130 		uint8_t start_track, buf[20], *bp;
1131 
1132 		msf = (acmd[1] >> 1) & 1;
1133 		start_track = acmd[6];
1134 		if (start_track > 1 && start_track != 0xaa) {
1135 			uint32_t tfd;
1136 			p->sense_key = ATA_SENSE_ILLEGAL_REQUEST;
1137 			p->asc = 0x24;
1138 			tfd = (p->sense_key << 12) | ATA_S_READY | ATA_S_ERROR;
1139 			cfis[4] = (cfis[4] & ~7) | ATA_I_CMD | ATA_I_IN;
1140 			ahci_write_fis_d2h(p, slot, cfis, tfd);
1141 			return;
1142 		}
1143 		bp = buf + 2;
1144 		*bp++ = 1;
1145 		*bp++ = 1;
1146 		if (start_track <= 1) {
1147 			*bp++ = 0;
1148 			*bp++ = 0x14;
1149 			*bp++ = 1;
1150 			*bp++ = 0;
1151 			if (msf) {
1152 				*bp++ = 0;
1153 				lba_to_msf(bp, 0);
1154 				bp += 3;
1155 			} else {
1156 				*bp++ = 0;
1157 				*bp++ = 0;
1158 				*bp++ = 0;
1159 				*bp++ = 0;
1160 			}
1161 		}
1162 		*bp++ = 0;
1163 		*bp++ = 0x14;
1164 		*bp++ = 0xaa;
1165 		*bp++ = 0;
1166 		sectors = blockif_size(p->bctx) / blockif_sectsz(p->bctx);
1167 		sectors >>= 2;
1168 		if (msf) {
1169 			*bp++ = 0;
1170 			lba_to_msf(bp, sectors);
1171 			bp += 3;
1172 		} else {
1173 			be32enc(bp, sectors);
1174 			bp += 4;
1175 		}
1176 		size = bp - buf;
1177 		be16enc(buf, size - 2);
1178 		if (len > size)
1179 			len = size;
1180 		write_prdt(p, slot, cfis, buf, len);
1181 		cfis[4] = (cfis[4] & ~7) | ATA_I_CMD | ATA_I_IN;
1182 		ahci_write_fis_d2h(p, slot, cfis, ATA_S_READY | ATA_S_DSC);
1183 		break;
1184 	}
1185 	case 1:
1186 	{
1187 		uint8_t buf[12];
1188 
1189 		memset(buf, 0, sizeof(buf));
1190 		buf[1] = 0xa;
1191 		buf[2] = 0x1;
1192 		buf[3] = 0x1;
1193 		if (len > sizeof(buf))
1194 			len = sizeof(buf);
1195 		write_prdt(p, slot, cfis, buf, len);
1196 		cfis[4] = (cfis[4] & ~7) | ATA_I_CMD | ATA_I_IN;
1197 		ahci_write_fis_d2h(p, slot, cfis, ATA_S_READY | ATA_S_DSC);
1198 		break;
1199 	}
1200 	case 2:
1201 	{
1202 		int msf, size;
1203 		uint64_t sectors;
1204 		uint8_t *bp, buf[50];
1205 
1206 		msf = (acmd[1] >> 1) & 1;
1207 		bp = buf + 2;
1208 		*bp++ = 1;
1209 		*bp++ = 1;
1210 
1211 		*bp++ = 1;
1212 		*bp++ = 0x14;
1213 		*bp++ = 0;
1214 		*bp++ = 0xa0;
1215 		*bp++ = 0;
1216 		*bp++ = 0;
1217 		*bp++ = 0;
1218 		*bp++ = 0;
1219 		*bp++ = 1;
1220 		*bp++ = 0;
1221 		*bp++ = 0;
1222 
1223 		*bp++ = 1;
1224 		*bp++ = 0x14;
1225 		*bp++ = 0;
1226 		*bp++ = 0xa1;
1227 		*bp++ = 0;
1228 		*bp++ = 0;
1229 		*bp++ = 0;
1230 		*bp++ = 0;
1231 		*bp++ = 1;
1232 		*bp++ = 0;
1233 		*bp++ = 0;
1234 
1235 		*bp++ = 1;
1236 		*bp++ = 0x14;
1237 		*bp++ = 0;
1238 		*bp++ = 0xa2;
1239 		*bp++ = 0;
1240 		*bp++ = 0;
1241 		*bp++ = 0;
1242 		sectors = blockif_size(p->bctx) / blockif_sectsz(p->bctx);
1243 		sectors >>= 2;
1244 		if (msf) {
1245 			*bp++ = 0;
1246 			lba_to_msf(bp, sectors);
1247 			bp += 3;
1248 		} else {
1249 			be32enc(bp, sectors);
1250 			bp += 4;
1251 		}
1252 
1253 		*bp++ = 1;
1254 		*bp++ = 0x14;
1255 		*bp++ = 0;
1256 		*bp++ = 1;
1257 		*bp++ = 0;
1258 		*bp++ = 0;
1259 		*bp++ = 0;
1260 		if (msf) {
1261 			*bp++ = 0;
1262 			lba_to_msf(bp, 0);
1263 			bp += 3;
1264 		} else {
1265 			*bp++ = 0;
1266 			*bp++ = 0;
1267 			*bp++ = 0;
1268 			*bp++ = 0;
1269 		}
1270 
1271 		size = bp - buf;
1272 		be16enc(buf, size - 2);
1273 		if (len > size)
1274 			len = size;
1275 		write_prdt(p, slot, cfis, buf, len);
1276 		cfis[4] = (cfis[4] & ~7) | ATA_I_CMD | ATA_I_IN;
1277 		ahci_write_fis_d2h(p, slot, cfis, ATA_S_READY | ATA_S_DSC);
1278 		break;
1279 	}
1280 	default:
1281 	{
1282 		uint32_t tfd;
1283 
1284 		p->sense_key = ATA_SENSE_ILLEGAL_REQUEST;
1285 		p->asc = 0x24;
1286 		tfd = (p->sense_key << 12) | ATA_S_READY | ATA_S_ERROR;
1287 		cfis[4] = (cfis[4] & ~7) | ATA_I_CMD | ATA_I_IN;
1288 		ahci_write_fis_d2h(p, slot, cfis, tfd);
1289 		break;
1290 	}
1291 	}
1292 }
1293 
1294 static void
1295 atapi_report_luns(struct ahci_port *p, int slot, uint8_t *cfis)
1296 {
1297 	uint8_t buf[16];
1298 
1299 	memset(buf, 0, sizeof(buf));
1300 	buf[3] = 8;
1301 
1302 	cfis[4] = (cfis[4] & ~7) | ATA_I_CMD | ATA_I_IN;
1303 	write_prdt(p, slot, cfis, buf, sizeof(buf));
1304 	ahci_write_fis_d2h(p, slot, cfis, ATA_S_READY | ATA_S_DSC);
1305 }
1306 
1307 static void
1308 atapi_read(struct ahci_port *p, int slot, uint8_t *cfis, uint32_t done)
1309 {
1310 	struct ahci_ioreq *aior;
1311 	struct ahci_cmd_hdr *hdr;
1312 	struct ahci_prdt_entry *prdt;
1313 	struct blockif_req *breq;
1314 	uint8_t *acmd;
1315 	uint64_t lba;
1316 	uint32_t len;
1317 	int err;
1318 
1319 	acmd = cfis + 0x40;
1320 	hdr = (struct ahci_cmd_hdr *)(p->cmd_lst + slot * AHCI_CL_SIZE);
1321 	prdt = (struct ahci_prdt_entry *)(cfis + 0x80);
1322 
1323 	lba = be32dec(acmd + 2);
1324 	if (acmd[0] == READ_10)
1325 		len = be16dec(acmd + 7);
1326 	else
1327 		len = be32dec(acmd + 6);
1328 	if (len == 0) {
1329 		cfis[4] = (cfis[4] & ~7) | ATA_I_CMD | ATA_I_IN;
1330 		ahci_write_fis_d2h(p, slot, cfis, ATA_S_READY | ATA_S_DSC);
1331 	}
1332 	lba *= 2048;
1333 	len *= 2048;
1334 
1335 	/*
1336 	 * Pull request off free list
1337 	 */
1338 	aior = STAILQ_FIRST(&p->iofhd);
1339 	assert(aior != NULL);
1340 	STAILQ_REMOVE_HEAD(&p->iofhd, io_flist);
1341 	aior->cfis = cfis;
1342 	aior->slot = slot;
1343 	aior->len = len;
1344 	aior->done = done;
1345 	breq = &aior->io_req;
1346 	breq->br_offset = lba + done;
1347 	ahci_build_iov(p, aior, prdt, hdr->prdtl);
1348 
1349 	/* Mark this command in-flight. */
1350 	p->pending |= 1 << slot;
1351 
1352 	/* Stuff request onto busy list. */
1353 	TAILQ_INSERT_HEAD(&p->iobhd, aior, io_blist);
1354 
1355 	err = blockif_read(p->bctx, breq);
1356 	assert(err == 0);
1357 }
1358 
1359 static void
1360 atapi_request_sense(struct ahci_port *p, int slot, uint8_t *cfis)
1361 {
1362 	uint8_t buf[64];
1363 	uint8_t *acmd;
1364 	int len;
1365 
1366 	acmd = cfis + 0x40;
1367 	len = acmd[4];
1368 	if (len > sizeof(buf))
1369 		len = sizeof(buf);
1370 	memset(buf, 0, len);
1371 	buf[0] = 0x70 | (1 << 7);
1372 	buf[2] = p->sense_key;
1373 	buf[7] = 10;
1374 	buf[12] = p->asc;
1375 	write_prdt(p, slot, cfis, buf, len);
1376 	cfis[4] = (cfis[4] & ~7) | ATA_I_CMD | ATA_I_IN;
1377 	ahci_write_fis_d2h(p, slot, cfis, ATA_S_READY | ATA_S_DSC);
1378 }
1379 
1380 static void
1381 atapi_start_stop_unit(struct ahci_port *p, int slot, uint8_t *cfis)
1382 {
1383 	uint8_t *acmd = cfis + 0x40;
1384 	uint32_t tfd;
1385 
1386 	switch (acmd[4] & 3) {
1387 	case 0:
1388 	case 1:
1389 	case 3:
1390 		cfis[4] = (cfis[4] & ~7) | ATA_I_CMD | ATA_I_IN;
1391 		tfd = ATA_S_READY | ATA_S_DSC;
1392 		break;
1393 	case 2:
1394 		/* TODO eject media */
1395 		cfis[4] = (cfis[4] & ~7) | ATA_I_CMD | ATA_I_IN;
1396 		p->sense_key = ATA_SENSE_ILLEGAL_REQUEST;
1397 		p->asc = 0x53;
1398 		tfd = (p->sense_key << 12) | ATA_S_READY | ATA_S_ERROR;
1399 		break;
1400 	}
1401 	ahci_write_fis_d2h(p, slot, cfis, tfd);
1402 }
1403 
1404 static void
1405 atapi_mode_sense(struct ahci_port *p, int slot, uint8_t *cfis)
1406 {
1407 	uint8_t *acmd;
1408 	uint32_t tfd;
1409 	uint8_t pc, code;
1410 	int len;
1411 
1412 	acmd = cfis + 0x40;
1413 	len = be16dec(acmd + 7);
1414 	pc = acmd[2] >> 6;
1415 	code = acmd[2] & 0x3f;
1416 
1417 	switch (pc) {
1418 	case 0:
1419 		switch (code) {
1420 		case MODEPAGE_RW_ERROR_RECOVERY:
1421 		{
1422 			uint8_t buf[16];
1423 
1424 			if (len > sizeof(buf))
1425 				len = sizeof(buf);
1426 
1427 			memset(buf, 0, sizeof(buf));
1428 			be16enc(buf, 16 - 2);
1429 			buf[2] = 0x70;
1430 			buf[8] = 0x01;
1431 			buf[9] = 16 - 10;
1432 			buf[11] = 0x05;
1433 			write_prdt(p, slot, cfis, buf, len);
1434 			tfd = ATA_S_READY | ATA_S_DSC;
1435 			break;
1436 		}
1437 		case MODEPAGE_CD_CAPABILITIES:
1438 		{
1439 			uint8_t buf[30];
1440 
1441 			if (len > sizeof(buf))
1442 				len = sizeof(buf);
1443 
1444 			memset(buf, 0, sizeof(buf));
1445 			be16enc(buf, 30 - 2);
1446 			buf[2] = 0x70;
1447 			buf[8] = 0x2A;
1448 			buf[9] = 30 - 10;
1449 			buf[10] = 0x08;
1450 			buf[12] = 0x71;
1451 			be16enc(&buf[18], 2);
1452 			be16enc(&buf[20], 512);
1453 			write_prdt(p, slot, cfis, buf, len);
1454 			tfd = ATA_S_READY | ATA_S_DSC;
1455 			break;
1456 		}
1457 		default:
1458 			goto error;
1459 			break;
1460 		}
1461 		break;
1462 	case 3:
1463 		p->sense_key = ATA_SENSE_ILLEGAL_REQUEST;
1464 		p->asc = 0x39;
1465 		tfd = (p->sense_key << 12) | ATA_S_READY | ATA_S_ERROR;
1466 		break;
1467 error:
1468 	case 1:
1469 	case 2:
1470 		p->sense_key = ATA_SENSE_ILLEGAL_REQUEST;
1471 		p->asc = 0x24;
1472 		tfd = (p->sense_key << 12) | ATA_S_READY | ATA_S_ERROR;
1473 		break;
1474 	}
1475 	cfis[4] = (cfis[4] & ~7) | ATA_I_CMD | ATA_I_IN;
1476 	ahci_write_fis_d2h(p, slot, cfis, tfd);
1477 }
1478 
1479 static void
1480 atapi_get_event_status_notification(struct ahci_port *p, int slot,
1481     uint8_t *cfis)
1482 {
1483 	uint8_t *acmd;
1484 	uint32_t tfd;
1485 
1486 	acmd = cfis + 0x40;
1487 
1488 	/* we don't support asynchronous operation */
1489 	if (!(acmd[1] & 1)) {
1490 		p->sense_key = ATA_SENSE_ILLEGAL_REQUEST;
1491 		p->asc = 0x24;
1492 		tfd = (p->sense_key << 12) | ATA_S_READY | ATA_S_ERROR;
1493 	} else {
1494 		uint8_t buf[8];
1495 		int len;
1496 
1497 		len = be16dec(acmd + 7);
1498 		if (len > sizeof(buf))
1499 			len = sizeof(buf);
1500 
1501 		memset(buf, 0, sizeof(buf));
1502 		be16enc(buf, 8 - 2);
1503 		buf[2] = 0x04;
1504 		buf[3] = 0x10;
1505 		buf[5] = 0x02;
1506 		write_prdt(p, slot, cfis, buf, len);
1507 		tfd = ATA_S_READY | ATA_S_DSC;
1508 	}
1509 	cfis[4] = (cfis[4] & ~7) | ATA_I_CMD | ATA_I_IN;
1510 	ahci_write_fis_d2h(p, slot, cfis, tfd);
1511 }
1512 
1513 static void
1514 handle_packet_cmd(struct ahci_port *p, int slot, uint8_t *cfis)
1515 {
1516 	uint8_t *acmd;
1517 
1518 	acmd = cfis + 0x40;
1519 
1520 #ifdef AHCI_DEBUG
1521 	{
1522 		int i;
1523 		DPRINTF("ACMD:");
1524 		for (i = 0; i < 16; i++)
1525 			DPRINTF("%02x ", acmd[i]);
1526 		DPRINTF("\n");
1527 	}
1528 #endif
1529 
1530 	switch (acmd[0]) {
1531 	case TEST_UNIT_READY:
1532 		cfis[4] = (cfis[4] & ~7) | ATA_I_CMD | ATA_I_IN;
1533 		ahci_write_fis_d2h(p, slot, cfis, ATA_S_READY | ATA_S_DSC);
1534 		break;
1535 	case INQUIRY:
1536 		atapi_inquiry(p, slot, cfis);
1537 		break;
1538 	case READ_CAPACITY:
1539 		atapi_read_capacity(p, slot, cfis);
1540 		break;
1541 	case PREVENT_ALLOW:
1542 		/* TODO */
1543 		cfis[4] = (cfis[4] & ~7) | ATA_I_CMD | ATA_I_IN;
1544 		ahci_write_fis_d2h(p, slot, cfis, ATA_S_READY | ATA_S_DSC);
1545 		break;
1546 	case READ_TOC:
1547 		atapi_read_toc(p, slot, cfis);
1548 		break;
1549 	case REPORT_LUNS:
1550 		atapi_report_luns(p, slot, cfis);
1551 		break;
1552 	case READ_10:
1553 	case READ_12:
1554 		atapi_read(p, slot, cfis, 0);
1555 		break;
1556 	case REQUEST_SENSE:
1557 		atapi_request_sense(p, slot, cfis);
1558 		break;
1559 	case START_STOP_UNIT:
1560 		atapi_start_stop_unit(p, slot, cfis);
1561 		break;
1562 	case MODE_SENSE_10:
1563 		atapi_mode_sense(p, slot, cfis);
1564 		break;
1565 	case GET_EVENT_STATUS_NOTIFICATION:
1566 		atapi_get_event_status_notification(p, slot, cfis);
1567 		break;
1568 	default:
1569 		cfis[4] = (cfis[4] & ~7) | ATA_I_CMD | ATA_I_IN;
1570 		p->sense_key = ATA_SENSE_ILLEGAL_REQUEST;
1571 		p->asc = 0x20;
1572 		ahci_write_fis_d2h(p, slot, cfis, (p->sense_key << 12) |
1573 				ATA_S_READY | ATA_S_ERROR);
1574 		break;
1575 	}
1576 }
1577 
1578 static void
1579 ahci_handle_cmd(struct ahci_port *p, int slot, uint8_t *cfis)
1580 {
1581 
1582 	p->tfd |= ATA_S_BUSY;
1583 	switch (cfis[2]) {
1584 	case ATA_ATA_IDENTIFY:
1585 		handle_identify(p, slot, cfis);
1586 		break;
1587 	case ATA_SETFEATURES:
1588 	{
1589 		switch (cfis[3]) {
1590 		case ATA_SF_ENAB_SATA_SF:
1591 			switch (cfis[12]) {
1592 			case ATA_SATA_SF_AN:
1593 				p->tfd = ATA_S_DSC | ATA_S_READY;
1594 				break;
1595 			default:
1596 				p->tfd = ATA_S_ERROR | ATA_S_READY;
1597 				p->tfd |= (ATA_ERROR_ABORT << 8);
1598 				break;
1599 			}
1600 			break;
1601 		case ATA_SF_ENAB_WCACHE:
1602 		case ATA_SF_DIS_WCACHE:
1603 		case ATA_SF_ENAB_RCACHE:
1604 		case ATA_SF_DIS_RCACHE:
1605 			p->tfd = ATA_S_DSC | ATA_S_READY;
1606 			break;
1607 		case ATA_SF_SETXFER:
1608 		{
1609 			switch (cfis[12] & 0xf8) {
1610 			case ATA_PIO:
1611 			case ATA_PIO0:
1612 				break;
1613 			case ATA_WDMA0:
1614 			case ATA_UDMA0:
1615 				p->xfermode = (cfis[12] & 0x7);
1616 				break;
1617 			}
1618 			p->tfd = ATA_S_DSC | ATA_S_READY;
1619 			break;
1620 		}
1621 		default:
1622 			p->tfd = ATA_S_ERROR | ATA_S_READY;
1623 			p->tfd |= (ATA_ERROR_ABORT << 8);
1624 			break;
1625 		}
1626 		ahci_write_fis_d2h(p, slot, cfis, p->tfd);
1627 		break;
1628 	}
1629 	case ATA_SET_MULTI:
1630 		if (cfis[12] != 0 &&
1631 			(cfis[12] > 128 || (cfis[12] & (cfis[12] - 1)))) {
1632 			p->tfd = ATA_S_ERROR | ATA_S_READY;
1633 			p->tfd |= (ATA_ERROR_ABORT << 8);
1634 		} else {
1635 			p->mult_sectors = cfis[12];
1636 			p->tfd = ATA_S_DSC | ATA_S_READY;
1637 		}
1638 		ahci_write_fis_d2h(p, slot, cfis, p->tfd);
1639 		break;
1640 	case ATA_READ:
1641 	case ATA_WRITE:
1642 	case ATA_READ48:
1643 	case ATA_WRITE48:
1644 	case ATA_READ_MUL:
1645 	case ATA_WRITE_MUL:
1646 	case ATA_READ_MUL48:
1647 	case ATA_WRITE_MUL48:
1648 	case ATA_READ_DMA:
1649 	case ATA_WRITE_DMA:
1650 	case ATA_READ_DMA48:
1651 	case ATA_WRITE_DMA48:
1652 	case ATA_READ_FPDMA_QUEUED:
1653 	case ATA_WRITE_FPDMA_QUEUED:
1654 		ahci_handle_rw(p, slot, cfis, 0);
1655 		break;
1656 	case ATA_FLUSHCACHE:
1657 	case ATA_FLUSHCACHE48:
1658 		ahci_handle_flush(p, slot, cfis);
1659 		break;
1660 	case ATA_DATA_SET_MANAGEMENT:
1661 		if (cfis[11] == 0 && cfis[3] == ATA_DSM_TRIM &&
1662 		    cfis[13] == 0 && cfis[12] == 1) {
1663 			ahci_handle_dsm_trim(p, slot, cfis, 0);
1664 			break;
1665 		}
1666 		ahci_write_fis_d2h(p, slot, cfis,
1667 		    (ATA_E_ABORT << 8) | ATA_S_READY | ATA_S_ERROR);
1668 		break;
1669 	case ATA_SEND_FPDMA_QUEUED:
1670 		if ((cfis[13] & 0x1f) == ATA_SFPDMA_DSM &&
1671 		    cfis[17] == 0 && cfis[16] == ATA_DSM_TRIM &&
1672 		    cfis[11] == 0 && cfis[13] == 1) {
1673 			ahci_handle_dsm_trim(p, slot, cfis, 0);
1674 			break;
1675 		}
1676 		ahci_write_fis_d2h(p, slot, cfis,
1677 		    (ATA_E_ABORT << 8) | ATA_S_READY | ATA_S_ERROR);
1678 		break;
1679 	case ATA_READ_LOG_EXT:
1680 	case ATA_READ_LOG_DMA_EXT:
1681 		ahci_handle_read_log(p, slot, cfis);
1682 		break;
1683 	case ATA_SECURITY_FREEZE_LOCK:
1684 	case ATA_SMART_CMD:
1685 	case ATA_NOP:
1686 		ahci_write_fis_d2h(p, slot, cfis,
1687 		    (ATA_E_ABORT << 8) | ATA_S_READY | ATA_S_ERROR);
1688 		break;
1689 	case ATA_CHECK_POWER_MODE:
1690 		cfis[12] = 0xff;	/* always on */
1691 		ahci_write_fis_d2h(p, slot, cfis, ATA_S_READY | ATA_S_DSC);
1692 		break;
1693 	case ATA_STANDBY_CMD:
1694 	case ATA_STANDBY_IMMEDIATE:
1695 	case ATA_IDLE_CMD:
1696 	case ATA_IDLE_IMMEDIATE:
1697 	case ATA_SLEEP:
1698 	case ATA_READ_VERIFY:
1699 	case ATA_READ_VERIFY48:
1700 		ahci_write_fis_d2h(p, slot, cfis, ATA_S_READY | ATA_S_DSC);
1701 		break;
1702 	case ATA_ATAPI_IDENTIFY:
1703 		handle_atapi_identify(p, slot, cfis);
1704 		break;
1705 	case ATA_PACKET_CMD:
1706 		if (!p->atapi) {
1707 			ahci_write_fis_d2h(p, slot, cfis,
1708 			    (ATA_E_ABORT << 8) | ATA_S_READY | ATA_S_ERROR);
1709 		} else
1710 			handle_packet_cmd(p, slot, cfis);
1711 		break;
1712 	default:
1713 		WPRINTF("Unsupported cmd:%02x\n", cfis[2]);
1714 		ahci_write_fis_d2h(p, slot, cfis,
1715 		    (ATA_E_ABORT << 8) | ATA_S_READY | ATA_S_ERROR);
1716 		break;
1717 	}
1718 }
1719 
1720 static void
1721 ahci_handle_slot(struct ahci_port *p, int slot)
1722 {
1723 	struct ahci_cmd_hdr *hdr;
1724 	struct ahci_prdt_entry *prdt;
1725 	struct pci_ahci_softc *sc;
1726 	uint8_t *cfis;
1727 	int cfl;
1728 
1729 	sc = p->pr_sc;
1730 	hdr = (struct ahci_cmd_hdr *)(p->cmd_lst + slot * AHCI_CL_SIZE);
1731 	cfl = (hdr->flags & 0x1f) * 4;
1732 	cfis = paddr_guest2host(ahci_ctx(sc), hdr->ctba,
1733 			0x80 + hdr->prdtl * sizeof(struct ahci_prdt_entry));
1734 	prdt = (struct ahci_prdt_entry *)(cfis + 0x80);
1735 
1736 #ifdef AHCI_DEBUG
1737 	DPRINTF("\ncfis:");
1738 	for (i = 0; i < cfl; i++) {
1739 		if (i % 10 == 0)
1740 			DPRINTF("\n");
1741 		DPRINTF("%02x ", cfis[i]);
1742 	}
1743 	DPRINTF("\n");
1744 
1745 	for (i = 0; i < hdr->prdtl; i++) {
1746 		DPRINTF("%d@%08"PRIx64"\n", prdt->dbc & 0x3fffff, prdt->dba);
1747 		prdt++;
1748 	}
1749 #endif
1750 
1751 	if (cfis[0] != FIS_TYPE_REGH2D) {
1752 		WPRINTF("Not a H2D FIS:%02x\n", cfis[0]);
1753 		return;
1754 	}
1755 
1756 	if (cfis[1] & 0x80) {
1757 		ahci_handle_cmd(p, slot, cfis);
1758 	} else {
1759 		if (cfis[15] & (1 << 2))
1760 			p->reset = 1;
1761 		else if (p->reset) {
1762 			p->reset = 0;
1763 			ahci_port_reset(p);
1764 		}
1765 		p->ci &= ~(1 << slot);
1766 	}
1767 }
1768 
1769 static void
1770 ahci_handle_port(struct ahci_port *p)
1771 {
1772 
1773 	if (!(p->cmd & AHCI_P_CMD_ST))
1774 		return;
1775 
1776 	/*
1777 	 * Search for any new commands to issue ignoring those that
1778 	 * are already in-flight.  Stop if device is busy or in error.
1779 	 */
1780 	for (; (p->ci & ~p->pending) != 0; p->ccs = ((p->ccs + 1) & 31)) {
1781 		if ((p->tfd & (ATA_S_BUSY | ATA_S_DRQ)) != 0)
1782 			break;
1783 		if (p->waitforclear)
1784 			break;
1785 		if ((p->ci & ~p->pending & (1 << p->ccs)) != 0) {
1786 			p->cmd &= ~AHCI_P_CMD_CCS_MASK;
1787 			p->cmd |= p->ccs << AHCI_P_CMD_CCS_SHIFT;
1788 			ahci_handle_slot(p, p->ccs);
1789 		}
1790 	}
1791 }
1792 
1793 /*
1794  * blockif callback routine - this runs in the context of the blockif
1795  * i/o thread, so the mutex needs to be acquired.
1796  */
1797 static void
1798 ata_ioreq_cb(struct blockif_req *br, int err)
1799 {
1800 	struct ahci_cmd_hdr *hdr;
1801 	struct ahci_ioreq *aior;
1802 	struct ahci_port *p;
1803 	struct pci_ahci_softc *sc;
1804 	uint32_t tfd;
1805 	uint8_t *cfis;
1806 	int slot, ncq, dsm;
1807 
1808 	DPRINTF("%s %d\n", __func__, err);
1809 
1810 	ncq = dsm = 0;
1811 	aior = br->br_param;
1812 	p = aior->io_pr;
1813 	cfis = aior->cfis;
1814 	slot = aior->slot;
1815 	sc = p->pr_sc;
1816 	hdr = (struct ahci_cmd_hdr *)(p->cmd_lst + slot * AHCI_CL_SIZE);
1817 
1818 	if (cfis[2] == ATA_WRITE_FPDMA_QUEUED ||
1819 	    cfis[2] == ATA_READ_FPDMA_QUEUED ||
1820 	    cfis[2] == ATA_SEND_FPDMA_QUEUED)
1821 		ncq = 1;
1822 	if (cfis[2] == ATA_DATA_SET_MANAGEMENT ||
1823 	    (cfis[2] == ATA_SEND_FPDMA_QUEUED &&
1824 	     (cfis[13] & 0x1f) == ATA_SFPDMA_DSM))
1825 		dsm = 1;
1826 
1827 	pthread_mutex_lock(&sc->mtx);
1828 
1829 	/*
1830 	 * Delete the blockif request from the busy list
1831 	 */
1832 	TAILQ_REMOVE(&p->iobhd, aior, io_blist);
1833 
1834 	/*
1835 	 * Move the blockif request back to the free list
1836 	 */
1837 	STAILQ_INSERT_TAIL(&p->iofhd, aior, io_flist);
1838 
1839 	if (!err)
1840 		hdr->prdbc = aior->done;
1841 
1842 	if (!err && aior->more) {
1843 		if (dsm)
1844 			ahci_handle_dsm_trim(p, slot, cfis, aior->done);
1845 		else
1846 			ahci_handle_rw(p, slot, cfis, aior->done);
1847 		goto out;
1848 	}
1849 
1850 	if (!err)
1851 		tfd = ATA_S_READY | ATA_S_DSC;
1852 	else
1853 		tfd = (ATA_E_ABORT << 8) | ATA_S_READY | ATA_S_ERROR;
1854 	if (ncq)
1855 		ahci_write_fis_sdb(p, slot, cfis, tfd);
1856 	else
1857 		ahci_write_fis_d2h(p, slot, cfis, tfd);
1858 
1859 	/*
1860 	 * This command is now complete.
1861 	 */
1862 	p->pending &= ~(1 << slot);
1863 
1864 	ahci_check_stopped(p);
1865 	ahci_handle_port(p);
1866 out:
1867 	pthread_mutex_unlock(&sc->mtx);
1868 	DPRINTF("%s exit\n", __func__);
1869 }
1870 
1871 static void
1872 atapi_ioreq_cb(struct blockif_req *br, int err)
1873 {
1874 	struct ahci_cmd_hdr *hdr;
1875 	struct ahci_ioreq *aior;
1876 	struct ahci_port *p;
1877 	struct pci_ahci_softc *sc;
1878 	uint8_t *cfis;
1879 	uint32_t tfd;
1880 	int slot;
1881 
1882 	DPRINTF("%s %d\n", __func__, err);
1883 
1884 	aior = br->br_param;
1885 	p = aior->io_pr;
1886 	cfis = aior->cfis;
1887 	slot = aior->slot;
1888 	sc = p->pr_sc;
1889 	hdr = (struct ahci_cmd_hdr *)(p->cmd_lst + aior->slot * AHCI_CL_SIZE);
1890 
1891 	pthread_mutex_lock(&sc->mtx);
1892 
1893 	/*
1894 	 * Delete the blockif request from the busy list
1895 	 */
1896 	TAILQ_REMOVE(&p->iobhd, aior, io_blist);
1897 
1898 	/*
1899 	 * Move the blockif request back to the free list
1900 	 */
1901 	STAILQ_INSERT_TAIL(&p->iofhd, aior, io_flist);
1902 
1903 	if (!err)
1904 		hdr->prdbc = aior->done;
1905 
1906 	if (!err && aior->more) {
1907 		atapi_read(p, slot, cfis, aior->done);
1908 		goto out;
1909 	}
1910 
1911 	if (!err) {
1912 		tfd = ATA_S_READY | ATA_S_DSC;
1913 	} else {
1914 		p->sense_key = ATA_SENSE_ILLEGAL_REQUEST;
1915 		p->asc = 0x21;
1916 		tfd = (p->sense_key << 12) | ATA_S_READY | ATA_S_ERROR;
1917 	}
1918 	cfis[4] = (cfis[4] & ~7) | ATA_I_CMD | ATA_I_IN;
1919 	ahci_write_fis_d2h(p, slot, cfis, tfd);
1920 
1921 	/*
1922 	 * This command is now complete.
1923 	 */
1924 	p->pending &= ~(1 << slot);
1925 
1926 	ahci_check_stopped(p);
1927 	ahci_handle_port(p);
1928 out:
1929 	pthread_mutex_unlock(&sc->mtx);
1930 	DPRINTF("%s exit\n", __func__);
1931 }
1932 
1933 static void
1934 pci_ahci_ioreq_init(struct ahci_port *pr)
1935 {
1936 	struct ahci_ioreq *vr;
1937 	int i;
1938 
1939 	pr->ioqsz = blockif_queuesz(pr->bctx);
1940 	pr->ioreq = calloc(pr->ioqsz, sizeof(struct ahci_ioreq));
1941 	STAILQ_INIT(&pr->iofhd);
1942 
1943 	/*
1944 	 * Add all i/o request entries to the free queue
1945 	 */
1946 	for (i = 0; i < pr->ioqsz; i++) {
1947 		vr = &pr->ioreq[i];
1948 		vr->io_pr = pr;
1949 		if (!pr->atapi)
1950 			vr->io_req.br_callback = ata_ioreq_cb;
1951 		else
1952 			vr->io_req.br_callback = atapi_ioreq_cb;
1953 		vr->io_req.br_param = vr;
1954 		STAILQ_INSERT_TAIL(&pr->iofhd, vr, io_flist);
1955 	}
1956 
1957 	TAILQ_INIT(&pr->iobhd);
1958 }
1959 
1960 static void
1961 pci_ahci_port_write(struct pci_ahci_softc *sc, uint64_t offset, uint64_t value)
1962 {
1963 	int port = (offset - AHCI_OFFSET) / AHCI_STEP;
1964 	offset = (offset - AHCI_OFFSET) % AHCI_STEP;
1965 	struct ahci_port *p = &sc->port[port];
1966 
1967 	DPRINTF("pci_ahci_port %d: write offset 0x%"PRIx64" value 0x%"PRIx64"\n",
1968 		port, offset, value);
1969 
1970 	switch (offset) {
1971 	case AHCI_P_CLB:
1972 		p->clb = value;
1973 		break;
1974 	case AHCI_P_CLBU:
1975 		p->clbu = value;
1976 		break;
1977 	case AHCI_P_FB:
1978 		p->fb = value;
1979 		break;
1980 	case AHCI_P_FBU:
1981 		p->fbu = value;
1982 		break;
1983 	case AHCI_P_IS:
1984 		p->is &= ~value;
1985 		break;
1986 	case AHCI_P_IE:
1987 		p->ie = value & 0xFDC000FF;
1988 		ahci_generate_intr(sc);
1989 		break;
1990 	case AHCI_P_CMD:
1991 	{
1992 		p->cmd &= ~(AHCI_P_CMD_ST | AHCI_P_CMD_SUD | AHCI_P_CMD_POD |
1993 		    AHCI_P_CMD_CLO | AHCI_P_CMD_FRE | AHCI_P_CMD_APSTE |
1994 		    AHCI_P_CMD_ATAPI | AHCI_P_CMD_DLAE | AHCI_P_CMD_ALPE |
1995 		    AHCI_P_CMD_ASP | AHCI_P_CMD_ICC_MASK);
1996 		p->cmd |= (AHCI_P_CMD_ST | AHCI_P_CMD_SUD | AHCI_P_CMD_POD |
1997 		    AHCI_P_CMD_CLO | AHCI_P_CMD_FRE | AHCI_P_CMD_APSTE |
1998 		    AHCI_P_CMD_ATAPI | AHCI_P_CMD_DLAE | AHCI_P_CMD_ALPE |
1999 		    AHCI_P_CMD_ASP | AHCI_P_CMD_ICC_MASK) & value;
2000 
2001 		if (!(value & AHCI_P_CMD_ST)) {
2002 			ahci_port_stop(p);
2003 		} else {
2004 			uint64_t clb;
2005 
2006 			p->cmd |= AHCI_P_CMD_CR;
2007 			clb = (uint64_t)p->clbu << 32 | p->clb;
2008 			p->cmd_lst = paddr_guest2host(ahci_ctx(sc), clb,
2009 					AHCI_CL_SIZE * AHCI_MAX_SLOTS);
2010 		}
2011 
2012 		if (value & AHCI_P_CMD_FRE) {
2013 			uint64_t fb;
2014 
2015 			p->cmd |= AHCI_P_CMD_FR;
2016 			fb = (uint64_t)p->fbu << 32 | p->fb;
2017 			/* we don't support FBSCP, so rfis size is 256Bytes */
2018 			p->rfis = paddr_guest2host(ahci_ctx(sc), fb, 256);
2019 		} else {
2020 			p->cmd &= ~AHCI_P_CMD_FR;
2021 		}
2022 
2023 		if (value & AHCI_P_CMD_CLO) {
2024 			p->tfd &= ~(ATA_S_BUSY | ATA_S_DRQ);
2025 			p->cmd &= ~AHCI_P_CMD_CLO;
2026 		}
2027 
2028 		if (value & AHCI_P_CMD_ICC_MASK) {
2029 			p->cmd &= ~AHCI_P_CMD_ICC_MASK;
2030 		}
2031 
2032 		ahci_handle_port(p);
2033 		break;
2034 	}
2035 	case AHCI_P_TFD:
2036 	case AHCI_P_SIG:
2037 	case AHCI_P_SSTS:
2038 		WPRINTF("pci_ahci_port: read only registers 0x%"PRIx64"\n", offset);
2039 		break;
2040 	case AHCI_P_SCTL:
2041 		p->sctl = value;
2042 		if (!(p->cmd & AHCI_P_CMD_ST)) {
2043 			if (value & ATA_SC_DET_RESET)
2044 				ahci_port_reset(p);
2045 		}
2046 		break;
2047 	case AHCI_P_SERR:
2048 		p->serr &= ~value;
2049 		break;
2050 	case AHCI_P_SACT:
2051 		p->sact |= value;
2052 		break;
2053 	case AHCI_P_CI:
2054 		p->ci |= value;
2055 		ahci_handle_port(p);
2056 		break;
2057 	case AHCI_P_SNTF:
2058 	case AHCI_P_FBS:
2059 	default:
2060 		break;
2061 	}
2062 }
2063 
2064 static void
2065 pci_ahci_host_write(struct pci_ahci_softc *sc, uint64_t offset, uint64_t value)
2066 {
2067 	DPRINTF("pci_ahci_host: write offset 0x%"PRIx64" value 0x%"PRIx64"\n",
2068 		offset, value);
2069 
2070 	switch (offset) {
2071 	case AHCI_CAP:
2072 	case AHCI_PI:
2073 	case AHCI_VS:
2074 	case AHCI_CAP2:
2075 		DPRINTF("pci_ahci_host: read only registers 0x%"PRIx64"\n", offset);
2076 		break;
2077 	case AHCI_GHC:
2078 		if (value & AHCI_GHC_HR)
2079 			ahci_reset(sc);
2080 		else if (value & AHCI_GHC_IE) {
2081 			sc->ghc |= AHCI_GHC_IE;
2082 			ahci_generate_intr(sc);
2083 		}
2084 		break;
2085 	case AHCI_IS:
2086 		sc->is &= ~value;
2087 		ahci_generate_intr(sc);
2088 		break;
2089 	default:
2090 		break;
2091 	}
2092 }
2093 
2094 static void
2095 pci_ahci_write(struct vmctx *ctx, int vcpu, struct pci_devinst *pi,
2096 		int baridx, uint64_t offset, int size, uint64_t value)
2097 {
2098 	struct pci_ahci_softc *sc = pi->pi_arg;
2099 
2100 	assert(baridx == 5);
2101 	assert((offset % 4) == 0 && size == 4);
2102 
2103 	pthread_mutex_lock(&sc->mtx);
2104 
2105 	if (offset < AHCI_OFFSET)
2106 		pci_ahci_host_write(sc, offset, value);
2107 	else if (offset < AHCI_OFFSET + sc->ports * AHCI_STEP)
2108 		pci_ahci_port_write(sc, offset, value);
2109 	else
2110 		WPRINTF("pci_ahci: unknown i/o write offset 0x%"PRIx64"\n", offset);
2111 
2112 	pthread_mutex_unlock(&sc->mtx);
2113 }
2114 
2115 static uint64_t
2116 pci_ahci_host_read(struct pci_ahci_softc *sc, uint64_t offset)
2117 {
2118 	uint32_t value;
2119 
2120 	switch (offset) {
2121 	case AHCI_CAP:
2122 	case AHCI_GHC:
2123 	case AHCI_IS:
2124 	case AHCI_PI:
2125 	case AHCI_VS:
2126 	case AHCI_CCCC:
2127 	case AHCI_CCCP:
2128 	case AHCI_EM_LOC:
2129 	case AHCI_EM_CTL:
2130 	case AHCI_CAP2:
2131 	{
2132 		uint32_t *p = &sc->cap;
2133 		p += (offset - AHCI_CAP) / sizeof(uint32_t);
2134 		value = *p;
2135 		break;
2136 	}
2137 	default:
2138 		value = 0;
2139 		break;
2140 	}
2141 	DPRINTF("pci_ahci_host: read offset 0x%"PRIx64" value 0x%x\n",
2142 		offset, value);
2143 
2144 	return (value);
2145 }
2146 
2147 static uint64_t
2148 pci_ahci_port_read(struct pci_ahci_softc *sc, uint64_t offset)
2149 {
2150 	uint32_t value;
2151 	int port = (offset - AHCI_OFFSET) / AHCI_STEP;
2152 	offset = (offset - AHCI_OFFSET) % AHCI_STEP;
2153 
2154 	switch (offset) {
2155 	case AHCI_P_CLB:
2156 	case AHCI_P_CLBU:
2157 	case AHCI_P_FB:
2158 	case AHCI_P_FBU:
2159 	case AHCI_P_IS:
2160 	case AHCI_P_IE:
2161 	case AHCI_P_CMD:
2162 	case AHCI_P_TFD:
2163 	case AHCI_P_SIG:
2164 	case AHCI_P_SSTS:
2165 	case AHCI_P_SCTL:
2166 	case AHCI_P_SERR:
2167 	case AHCI_P_SACT:
2168 	case AHCI_P_CI:
2169 	case AHCI_P_SNTF:
2170 	case AHCI_P_FBS:
2171 	{
2172 		uint32_t *p= &sc->port[port].clb;
2173 		p += (offset - AHCI_P_CLB) / sizeof(uint32_t);
2174 		value = *p;
2175 		break;
2176 	}
2177 	default:
2178 		value = 0;
2179 		break;
2180 	}
2181 
2182 	DPRINTF("pci_ahci_port %d: read offset 0x%"PRIx64" value 0x%x\n",
2183 		port, offset, value);
2184 
2185 	return value;
2186 }
2187 
2188 static uint64_t
2189 pci_ahci_read(struct vmctx *ctx, int vcpu, struct pci_devinst *pi, int baridx,
2190     uint64_t regoff, int size)
2191 {
2192 	struct pci_ahci_softc *sc = pi->pi_arg;
2193 	uint64_t offset;
2194 	uint32_t value;
2195 
2196 	assert(baridx == 5);
2197 	assert(size == 1 || size == 2 || size == 4);
2198 	assert((regoff & (size - 1)) == 0);
2199 
2200 	pthread_mutex_lock(&sc->mtx);
2201 
2202 	offset = regoff & ~0x3;	    /* round down to a multiple of 4 bytes */
2203 	if (offset < AHCI_OFFSET)
2204 		value = pci_ahci_host_read(sc, offset);
2205 	else if (offset < AHCI_OFFSET + sc->ports * AHCI_STEP)
2206 		value = pci_ahci_port_read(sc, offset);
2207 	else {
2208 		value = 0;
2209 		WPRINTF("pci_ahci: unknown i/o read offset 0x%"PRIx64"\n",
2210 		    regoff);
2211 	}
2212 	value >>= 8 * (regoff & 0x3);
2213 
2214 	pthread_mutex_unlock(&sc->mtx);
2215 
2216 	return (value);
2217 }
2218 
2219 static int
2220 pci_ahci_init(struct vmctx *ctx, struct pci_devinst *pi, char *opts, int atapi)
2221 {
2222 	char bident[sizeof("XX:X:X")];
2223 	struct blockif_ctxt *bctxt;
2224 	struct pci_ahci_softc *sc;
2225 	int ret, slots;
2226 	MD5_CTX mdctx;
2227 	u_char digest[16];
2228 
2229 	ret = 0;
2230 
2231 	if (opts == NULL) {
2232 		fprintf(stderr, "pci_ahci: backing device required\n");
2233 		return (1);
2234 	}
2235 
2236 #ifdef AHCI_DEBUG
2237 	dbg = fopen("/tmp/log", "w+");
2238 #endif
2239 
2240 	sc = calloc(1, sizeof(struct pci_ahci_softc));
2241 	pi->pi_arg = sc;
2242 	sc->asc_pi = pi;
2243 	sc->ports = MAX_PORTS;
2244 
2245 	/*
2246 	 * Only use port 0 for a backing device. All other ports will be
2247 	 * marked as unused
2248 	 */
2249 	sc->port[0].atapi = atapi;
2250 
2251 	/*
2252 	 * Attempt to open the backing image. Use the PCI
2253 	 * slot/func for the identifier string.
2254 	 */
2255 	snprintf(bident, sizeof(bident), "%d:%d", pi->pi_slot, pi->pi_func);
2256 	bctxt = blockif_open(opts, bident);
2257 	if (bctxt == NULL) {
2258 		ret = 1;
2259 		goto open_fail;
2260 	}
2261 	sc->port[0].bctx = bctxt;
2262 	sc->port[0].pr_sc = sc;
2263 
2264 	/*
2265 	 * Create an identifier for the backing file. Use parts of the
2266 	 * md5 sum of the filename
2267 	 */
2268 	MD5Init(&mdctx);
2269 	MD5Update(&mdctx, opts, strlen(opts));
2270 	MD5Final(digest, &mdctx);
2271 	sprintf(sc->port[0].ident, "BHYVE-%02X%02X-%02X%02X-%02X%02X",
2272 	    digest[0], digest[1], digest[2], digest[3], digest[4], digest[5]);
2273 
2274 	/*
2275 	 * Allocate blockif request structures and add them
2276 	 * to the free list
2277 	 */
2278 	pci_ahci_ioreq_init(&sc->port[0]);
2279 
2280 	pthread_mutex_init(&sc->mtx, NULL);
2281 
2282 	/* Intel ICH8 AHCI */
2283 	slots = sc->port[0].ioqsz;
2284 	if (slots > 32)
2285 		slots = 32;
2286 	--slots;
2287 	sc->cap = AHCI_CAP_64BIT | AHCI_CAP_SNCQ | AHCI_CAP_SSNTF |
2288 	    AHCI_CAP_SMPS | AHCI_CAP_SSS | AHCI_CAP_SALP |
2289 	    AHCI_CAP_SAL | AHCI_CAP_SCLO | (0x3 << AHCI_CAP_ISS_SHIFT)|
2290 	    AHCI_CAP_PMD | AHCI_CAP_SSC | AHCI_CAP_PSC |
2291 	    (slots << AHCI_CAP_NCS_SHIFT) | AHCI_CAP_SXS | (sc->ports - 1);
2292 
2293 	/* Only port 0 implemented */
2294 	sc->pi = 1;
2295 	sc->vs = 0x10300;
2296 	sc->cap2 = AHCI_CAP2_APST;
2297 	ahci_reset(sc);
2298 
2299 	pci_set_cfgdata16(pi, PCIR_DEVICE, 0x2821);
2300 	pci_set_cfgdata16(pi, PCIR_VENDOR, 0x8086);
2301 	pci_set_cfgdata8(pi, PCIR_CLASS, PCIC_STORAGE);
2302 	pci_set_cfgdata8(pi, PCIR_SUBCLASS, PCIS_STORAGE_SATA);
2303 	pci_set_cfgdata8(pi, PCIR_PROGIF, PCIP_STORAGE_SATA_AHCI_1_0);
2304 	pci_emul_add_msicap(pi, 1);
2305 	pci_emul_alloc_bar(pi, 5, PCIBAR_MEM32,
2306 	    AHCI_OFFSET + sc->ports * AHCI_STEP);
2307 
2308 	pci_lintr_request(pi);
2309 
2310 open_fail:
2311 	if (ret) {
2312 		if (sc->port[0].bctx != NULL)
2313 			blockif_close(sc->port[0].bctx);
2314 		free(sc);
2315 	}
2316 
2317 	return (ret);
2318 }
2319 
2320 static int
2321 pci_ahci_hd_init(struct vmctx *ctx, struct pci_devinst *pi, char *opts)
2322 {
2323 
2324 	return (pci_ahci_init(ctx, pi, opts, 0));
2325 }
2326 
2327 static int
2328 pci_ahci_atapi_init(struct vmctx *ctx, struct pci_devinst *pi, char *opts)
2329 {
2330 
2331 	return (pci_ahci_init(ctx, pi, opts, 1));
2332 }
2333 
2334 /*
2335  * Use separate emulation names to distinguish drive and atapi devices
2336  */
2337 struct pci_devemu pci_de_ahci_hd = {
2338 	.pe_emu =	"ahci-hd",
2339 	.pe_init =	pci_ahci_hd_init,
2340 	.pe_barwrite =	pci_ahci_write,
2341 	.pe_barread =	pci_ahci_read
2342 };
2343 PCI_EMUL_SET(pci_de_ahci_hd);
2344 
2345 struct pci_devemu pci_de_ahci_cd = {
2346 	.pe_emu =	"ahci-cd",
2347 	.pe_init =	pci_ahci_atapi_init,
2348 	.pe_barwrite =	pci_ahci_write,
2349 	.pe_barread =	pci_ahci_read
2350 };
2351 PCI_EMUL_SET(pci_de_ahci_cd);
2352