xref: /linux/drivers/ata/libahci.c (revision ae22a94997b8a03dcb3c922857c203246711f9d4)
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  *  libahci.c - Common AHCI SATA low-level routines
4  *
5  *  Maintained by:  Tejun Heo <tj@kernel.org>
6  *    		    Please ALWAYS copy linux-ide@vger.kernel.org
7  *		    on emails.
8  *
9  *  Copyright 2004-2005 Red Hat, Inc.
10  *
11  * libata documentation is available via 'make {ps|pdf}docs',
12  * as Documentation/driver-api/libata.rst
13  *
14  * AHCI hardware documentation:
15  * http://www.intel.com/technology/serialata/pdf/rev1_0.pdf
16  * http://www.intel.com/technology/serialata/pdf/rev1_1.pdf
17  */
18 
19 #include <linux/bitops.h>
20 #include <linux/kernel.h>
21 #include <linux/gfp.h>
22 #include <linux/module.h>
23 #include <linux/nospec.h>
24 #include <linux/blkdev.h>
25 #include <linux/delay.h>
26 #include <linux/interrupt.h>
27 #include <linux/dma-mapping.h>
28 #include <linux/device.h>
29 #include <scsi/scsi_host.h>
30 #include <scsi/scsi_cmnd.h>
31 #include <linux/libata.h>
32 #include <linux/pci.h>
33 #include "ahci.h"
34 #include "libata.h"
35 
36 static int ahci_skip_host_reset;
37 int ahci_ignore_sss;
38 EXPORT_SYMBOL_GPL(ahci_ignore_sss);
39 
40 module_param_named(skip_host_reset, ahci_skip_host_reset, int, 0444);
41 MODULE_PARM_DESC(skip_host_reset, "skip global host reset (0=don't skip, 1=skip)");
42 
43 module_param_named(ignore_sss, ahci_ignore_sss, int, 0444);
44 MODULE_PARM_DESC(ignore_sss, "Ignore staggered spinup flag (0=don't ignore, 1=ignore)");
45 
46 static int ahci_set_lpm(struct ata_link *link, enum ata_lpm_policy policy,
47 			unsigned hints);
48 static ssize_t ahci_led_show(struct ata_port *ap, char *buf);
49 static ssize_t ahci_led_store(struct ata_port *ap, const char *buf,
50 			      size_t size);
51 static ssize_t ahci_transmit_led_message(struct ata_port *ap, u32 state,
52 					ssize_t size);
53 
54 
55 
56 static int ahci_scr_read(struct ata_link *link, unsigned int sc_reg, u32 *val);
57 static int ahci_scr_write(struct ata_link *link, unsigned int sc_reg, u32 val);
58 static void ahci_qc_fill_rtf(struct ata_queued_cmd *qc);
59 static void ahci_qc_ncq_fill_rtf(struct ata_port *ap, u64 done_mask);
60 static int ahci_port_start(struct ata_port *ap);
61 static void ahci_port_stop(struct ata_port *ap);
62 static enum ata_completion_errors ahci_qc_prep(struct ata_queued_cmd *qc);
63 static int ahci_pmp_qc_defer(struct ata_queued_cmd *qc);
64 static void ahci_freeze(struct ata_port *ap);
65 static void ahci_thaw(struct ata_port *ap);
66 static void ahci_set_aggressive_devslp(struct ata_port *ap, bool sleep);
67 static void ahci_enable_fbs(struct ata_port *ap);
68 static void ahci_disable_fbs(struct ata_port *ap);
69 static void ahci_pmp_attach(struct ata_port *ap);
70 static void ahci_pmp_detach(struct ata_port *ap);
71 static int ahci_softreset(struct ata_link *link, unsigned int *class,
72 			  unsigned long deadline);
73 static int ahci_pmp_retry_softreset(struct ata_link *link, unsigned int *class,
74 			  unsigned long deadline);
75 static int ahci_hardreset(struct ata_link *link, unsigned int *class,
76 			  unsigned long deadline);
77 static void ahci_postreset(struct ata_link *link, unsigned int *class);
78 static void ahci_post_internal_cmd(struct ata_queued_cmd *qc);
79 static void ahci_dev_config(struct ata_device *dev);
80 #ifdef CONFIG_PM
81 static int ahci_port_suspend(struct ata_port *ap, pm_message_t mesg);
82 #endif
83 static ssize_t ahci_activity_show(struct ata_device *dev, char *buf);
84 static ssize_t ahci_activity_store(struct ata_device *dev,
85 				   enum sw_activity val);
86 static void ahci_init_sw_activity(struct ata_link *link);
87 
88 static ssize_t ahci_show_host_caps(struct device *dev,
89 				   struct device_attribute *attr, char *buf);
90 static ssize_t ahci_show_host_cap2(struct device *dev,
91 				   struct device_attribute *attr, char *buf);
92 static ssize_t ahci_show_host_version(struct device *dev,
93 				      struct device_attribute *attr, char *buf);
94 static ssize_t ahci_show_port_cmd(struct device *dev,
95 				  struct device_attribute *attr, char *buf);
96 static ssize_t ahci_read_em_buffer(struct device *dev,
97 				   struct device_attribute *attr, char *buf);
98 static ssize_t ahci_store_em_buffer(struct device *dev,
99 				    struct device_attribute *attr,
100 				    const char *buf, size_t size);
101 static ssize_t ahci_show_em_supported(struct device *dev,
102 				      struct device_attribute *attr, char *buf);
103 static irqreturn_t ahci_single_level_irq_intr(int irq, void *dev_instance);
104 
105 static DEVICE_ATTR(ahci_host_caps, S_IRUGO, ahci_show_host_caps, NULL);
106 static DEVICE_ATTR(ahci_host_cap2, S_IRUGO, ahci_show_host_cap2, NULL);
107 static DEVICE_ATTR(ahci_host_version, S_IRUGO, ahci_show_host_version, NULL);
108 static DEVICE_ATTR(ahci_port_cmd, S_IRUGO, ahci_show_port_cmd, NULL);
109 static DEVICE_ATTR(em_buffer, S_IWUSR | S_IRUGO,
110 		   ahci_read_em_buffer, ahci_store_em_buffer);
111 static DEVICE_ATTR(em_message_supported, S_IRUGO, ahci_show_em_supported, NULL);
112 
113 static struct attribute *ahci_shost_attrs[] = {
114 	&dev_attr_link_power_management_policy.attr,
115 	&dev_attr_em_message_type.attr,
116 	&dev_attr_em_message.attr,
117 	&dev_attr_ahci_host_caps.attr,
118 	&dev_attr_ahci_host_cap2.attr,
119 	&dev_attr_ahci_host_version.attr,
120 	&dev_attr_ahci_port_cmd.attr,
121 	&dev_attr_em_buffer.attr,
122 	&dev_attr_em_message_supported.attr,
123 	NULL
124 };
125 
126 static const struct attribute_group ahci_shost_attr_group = {
127 	.attrs = ahci_shost_attrs
128 };
129 
130 const struct attribute_group *ahci_shost_groups[] = {
131 	&ahci_shost_attr_group,
132 	NULL
133 };
134 EXPORT_SYMBOL_GPL(ahci_shost_groups);
135 
136 static struct attribute *ahci_sdev_attrs[] = {
137 	&dev_attr_sw_activity.attr,
138 	&dev_attr_unload_heads.attr,
139 	&dev_attr_ncq_prio_supported.attr,
140 	&dev_attr_ncq_prio_enable.attr,
141 	NULL
142 };
143 
144 static const struct attribute_group ahci_sdev_attr_group = {
145 	.attrs = ahci_sdev_attrs
146 };
147 
148 const struct attribute_group *ahci_sdev_groups[] = {
149 	&ahci_sdev_attr_group,
150 	NULL
151 };
152 EXPORT_SYMBOL_GPL(ahci_sdev_groups);
153 
154 struct ata_port_operations ahci_ops = {
155 	.inherits		= &sata_pmp_port_ops,
156 
157 	.qc_defer		= ahci_pmp_qc_defer,
158 	.qc_prep		= ahci_qc_prep,
159 	.qc_issue		= ahci_qc_issue,
160 	.qc_fill_rtf		= ahci_qc_fill_rtf,
161 	.qc_ncq_fill_rtf	= ahci_qc_ncq_fill_rtf,
162 
163 	.freeze			= ahci_freeze,
164 	.thaw			= ahci_thaw,
165 	.softreset		= ahci_softreset,
166 	.hardreset		= ahci_hardreset,
167 	.postreset		= ahci_postreset,
168 	.pmp_softreset		= ahci_softreset,
169 	.error_handler		= ahci_error_handler,
170 	.post_internal_cmd	= ahci_post_internal_cmd,
171 	.dev_config		= ahci_dev_config,
172 
173 	.scr_read		= ahci_scr_read,
174 	.scr_write		= ahci_scr_write,
175 	.pmp_attach		= ahci_pmp_attach,
176 	.pmp_detach		= ahci_pmp_detach,
177 
178 	.set_lpm		= ahci_set_lpm,
179 	.em_show		= ahci_led_show,
180 	.em_store		= ahci_led_store,
181 	.sw_activity_show	= ahci_activity_show,
182 	.sw_activity_store	= ahci_activity_store,
183 	.transmit_led_message	= ahci_transmit_led_message,
184 #ifdef CONFIG_PM
185 	.port_suspend		= ahci_port_suspend,
186 	.port_resume		= ahci_port_resume,
187 #endif
188 	.port_start		= ahci_port_start,
189 	.port_stop		= ahci_port_stop,
190 };
191 EXPORT_SYMBOL_GPL(ahci_ops);
192 
193 struct ata_port_operations ahci_pmp_retry_srst_ops = {
194 	.inherits		= &ahci_ops,
195 	.softreset		= ahci_pmp_retry_softreset,
196 };
197 EXPORT_SYMBOL_GPL(ahci_pmp_retry_srst_ops);
198 
199 static bool ahci_em_messages __read_mostly = true;
200 module_param(ahci_em_messages, bool, 0444);
201 /* add other LED protocol types when they become supported */
202 MODULE_PARM_DESC(ahci_em_messages,
203 	"AHCI Enclosure Management Message control (0 = off, 1 = on)");
204 
205 /* device sleep idle timeout in ms */
206 static int devslp_idle_timeout __read_mostly = 1000;
207 module_param(devslp_idle_timeout, int, 0644);
208 MODULE_PARM_DESC(devslp_idle_timeout, "device sleep idle timeout");
209 
210 static void ahci_enable_ahci(void __iomem *mmio)
211 {
212 	int i;
213 	u32 tmp;
214 
215 	/* turn on AHCI_EN */
216 	tmp = readl(mmio + HOST_CTL);
217 	if (tmp & HOST_AHCI_EN)
218 		return;
219 
220 	/* Some controllers need AHCI_EN to be written multiple times.
221 	 * Try a few times before giving up.
222 	 */
223 	for (i = 0; i < 5; i++) {
224 		tmp |= HOST_AHCI_EN;
225 		writel(tmp, mmio + HOST_CTL);
226 		tmp = readl(mmio + HOST_CTL);	/* flush && sanity check */
227 		if (tmp & HOST_AHCI_EN)
228 			return;
229 		msleep(10);
230 	}
231 
232 	WARN_ON(1);
233 }
234 
235 /**
236  *	ahci_rpm_get_port - Make sure the port is powered on
237  *	@ap: Port to power on
238  *
239  *	Whenever there is need to access the AHCI host registers outside of
240  *	normal execution paths, call this function to make sure the host is
241  *	actually powered on.
242  */
243 static int ahci_rpm_get_port(struct ata_port *ap)
244 {
245 	return pm_runtime_get_sync(ap->dev);
246 }
247 
248 /**
249  *	ahci_rpm_put_port - Undoes ahci_rpm_get_port()
250  *	@ap: Port to power down
251  *
252  *	Undoes ahci_rpm_get_port() and possibly powers down the AHCI host
253  *	if it has no more active users.
254  */
255 static void ahci_rpm_put_port(struct ata_port *ap)
256 {
257 	pm_runtime_put(ap->dev);
258 }
259 
260 static ssize_t ahci_show_host_caps(struct device *dev,
261 				   struct device_attribute *attr, char *buf)
262 {
263 	struct Scsi_Host *shost = class_to_shost(dev);
264 	struct ata_port *ap = ata_shost_to_port(shost);
265 	struct ahci_host_priv *hpriv = ap->host->private_data;
266 
267 	return sprintf(buf, "%x\n", hpriv->cap);
268 }
269 
270 static ssize_t ahci_show_host_cap2(struct device *dev,
271 				   struct device_attribute *attr, char *buf)
272 {
273 	struct Scsi_Host *shost = class_to_shost(dev);
274 	struct ata_port *ap = ata_shost_to_port(shost);
275 	struct ahci_host_priv *hpriv = ap->host->private_data;
276 
277 	return sprintf(buf, "%x\n", hpriv->cap2);
278 }
279 
280 static ssize_t ahci_show_host_version(struct device *dev,
281 				   struct device_attribute *attr, char *buf)
282 {
283 	struct Scsi_Host *shost = class_to_shost(dev);
284 	struct ata_port *ap = ata_shost_to_port(shost);
285 	struct ahci_host_priv *hpriv = ap->host->private_data;
286 
287 	return sprintf(buf, "%x\n", hpriv->version);
288 }
289 
290 static ssize_t ahci_show_port_cmd(struct device *dev,
291 				  struct device_attribute *attr, char *buf)
292 {
293 	struct Scsi_Host *shost = class_to_shost(dev);
294 	struct ata_port *ap = ata_shost_to_port(shost);
295 	void __iomem *port_mmio = ahci_port_base(ap);
296 	ssize_t ret;
297 
298 	ahci_rpm_get_port(ap);
299 	ret = sprintf(buf, "%x\n", readl(port_mmio + PORT_CMD));
300 	ahci_rpm_put_port(ap);
301 
302 	return ret;
303 }
304 
305 static ssize_t ahci_read_em_buffer(struct device *dev,
306 				   struct device_attribute *attr, char *buf)
307 {
308 	struct Scsi_Host *shost = class_to_shost(dev);
309 	struct ata_port *ap = ata_shost_to_port(shost);
310 	struct ahci_host_priv *hpriv = ap->host->private_data;
311 	void __iomem *mmio = hpriv->mmio;
312 	void __iomem *em_mmio = mmio + hpriv->em_loc;
313 	u32 em_ctl, msg;
314 	unsigned long flags;
315 	size_t count;
316 	int i;
317 
318 	ahci_rpm_get_port(ap);
319 	spin_lock_irqsave(ap->lock, flags);
320 
321 	em_ctl = readl(mmio + HOST_EM_CTL);
322 	if (!(ap->flags & ATA_FLAG_EM) || em_ctl & EM_CTL_XMT ||
323 	    !(hpriv->em_msg_type & EM_MSG_TYPE_SGPIO)) {
324 		spin_unlock_irqrestore(ap->lock, flags);
325 		ahci_rpm_put_port(ap);
326 		return -EINVAL;
327 	}
328 
329 	if (!(em_ctl & EM_CTL_MR)) {
330 		spin_unlock_irqrestore(ap->lock, flags);
331 		ahci_rpm_put_port(ap);
332 		return -EAGAIN;
333 	}
334 
335 	if (!(em_ctl & EM_CTL_SMB))
336 		em_mmio += hpriv->em_buf_sz;
337 
338 	count = hpriv->em_buf_sz;
339 
340 	/* the count should not be larger than PAGE_SIZE */
341 	if (count > PAGE_SIZE) {
342 		if (printk_ratelimit())
343 			ata_port_warn(ap,
344 				      "EM read buffer size too large: "
345 				      "buffer size %u, page size %lu\n",
346 				      hpriv->em_buf_sz, PAGE_SIZE);
347 		count = PAGE_SIZE;
348 	}
349 
350 	for (i = 0; i < count; i += 4) {
351 		msg = readl(em_mmio + i);
352 		buf[i] = msg & 0xff;
353 		buf[i + 1] = (msg >> 8) & 0xff;
354 		buf[i + 2] = (msg >> 16) & 0xff;
355 		buf[i + 3] = (msg >> 24) & 0xff;
356 	}
357 
358 	spin_unlock_irqrestore(ap->lock, flags);
359 	ahci_rpm_put_port(ap);
360 
361 	return i;
362 }
363 
364 static ssize_t ahci_store_em_buffer(struct device *dev,
365 				    struct device_attribute *attr,
366 				    const char *buf, size_t size)
367 {
368 	struct Scsi_Host *shost = class_to_shost(dev);
369 	struct ata_port *ap = ata_shost_to_port(shost);
370 	struct ahci_host_priv *hpriv = ap->host->private_data;
371 	void __iomem *mmio = hpriv->mmio;
372 	void __iomem *em_mmio = mmio + hpriv->em_loc;
373 	const unsigned char *msg_buf = buf;
374 	u32 em_ctl, msg;
375 	unsigned long flags;
376 	int i;
377 
378 	/* check size validity */
379 	if (!(ap->flags & ATA_FLAG_EM) ||
380 	    !(hpriv->em_msg_type & EM_MSG_TYPE_SGPIO) ||
381 	    size % 4 || size > hpriv->em_buf_sz)
382 		return -EINVAL;
383 
384 	ahci_rpm_get_port(ap);
385 	spin_lock_irqsave(ap->lock, flags);
386 
387 	em_ctl = readl(mmio + HOST_EM_CTL);
388 	if (em_ctl & EM_CTL_TM) {
389 		spin_unlock_irqrestore(ap->lock, flags);
390 		ahci_rpm_put_port(ap);
391 		return -EBUSY;
392 	}
393 
394 	for (i = 0; i < size; i += 4) {
395 		msg = msg_buf[i] | msg_buf[i + 1] << 8 |
396 		      msg_buf[i + 2] << 16 | msg_buf[i + 3] << 24;
397 		writel(msg, em_mmio + i);
398 	}
399 
400 	writel(em_ctl | EM_CTL_TM, mmio + HOST_EM_CTL);
401 
402 	spin_unlock_irqrestore(ap->lock, flags);
403 	ahci_rpm_put_port(ap);
404 
405 	return size;
406 }
407 
408 static ssize_t ahci_show_em_supported(struct device *dev,
409 				      struct device_attribute *attr, char *buf)
410 {
411 	struct Scsi_Host *shost = class_to_shost(dev);
412 	struct ata_port *ap = ata_shost_to_port(shost);
413 	struct ahci_host_priv *hpriv = ap->host->private_data;
414 	void __iomem *mmio = hpriv->mmio;
415 	u32 em_ctl;
416 
417 	ahci_rpm_get_port(ap);
418 	em_ctl = readl(mmio + HOST_EM_CTL);
419 	ahci_rpm_put_port(ap);
420 
421 	return sprintf(buf, "%s%s%s%s\n",
422 		       em_ctl & EM_CTL_LED ? "led " : "",
423 		       em_ctl & EM_CTL_SAFTE ? "saf-te " : "",
424 		       em_ctl & EM_CTL_SES ? "ses-2 " : "",
425 		       em_ctl & EM_CTL_SGPIO ? "sgpio " : "");
426 }
427 
428 /**
429  *	ahci_save_initial_config - Save and fixup initial config values
430  *	@dev: target AHCI device
431  *	@hpriv: host private area to store config values
432  *
433  *	Some registers containing configuration info might be setup by
434  *	BIOS and might be cleared on reset.  This function saves the
435  *	initial values of those registers into @hpriv such that they
436  *	can be restored after controller reset.
437  *
438  *	If inconsistent, config values are fixed up by this function.
439  *
440  *	If it is not set already this function sets hpriv->start_engine to
441  *	ahci_start_engine.
442  *
443  *	LOCKING:
444  *	None.
445  */
446 void ahci_save_initial_config(struct device *dev, struct ahci_host_priv *hpriv)
447 {
448 	void __iomem *mmio = hpriv->mmio;
449 	void __iomem *port_mmio;
450 	unsigned long port_map;
451 	u32 cap, cap2, vers;
452 	int i;
453 
454 	/* make sure AHCI mode is enabled before accessing CAP */
455 	ahci_enable_ahci(mmio);
456 
457 	/*
458 	 * Values prefixed with saved_ are written back to the HBA and ports
459 	 * registers after reset. Values without are used for driver operation.
460 	 */
461 
462 	/*
463 	 * Override HW-init HBA capability fields with the platform-specific
464 	 * values. The rest of the HBA capabilities are defined as Read-only
465 	 * and can't be modified in CSR anyway.
466 	 */
467 	cap = readl(mmio + HOST_CAP);
468 	if (hpriv->saved_cap)
469 		cap = (cap & ~(HOST_CAP_SSS | HOST_CAP_MPS)) | hpriv->saved_cap;
470 	hpriv->saved_cap = cap;
471 
472 	/* CAP2 register is only defined for AHCI 1.2 and later */
473 	vers = readl(mmio + HOST_VERSION);
474 	if ((vers >> 16) > 1 ||
475 	   ((vers >> 16) == 1 && (vers & 0xFFFF) >= 0x200))
476 		hpriv->saved_cap2 = cap2 = readl(mmio + HOST_CAP2);
477 	else
478 		hpriv->saved_cap2 = cap2 = 0;
479 
480 	/* some chips have errata preventing 64bit use */
481 	if ((cap & HOST_CAP_64) && (hpriv->flags & AHCI_HFLAG_32BIT_ONLY)) {
482 		dev_info(dev, "controller can't do 64bit DMA, forcing 32bit\n");
483 		cap &= ~HOST_CAP_64;
484 	}
485 
486 	if ((cap & HOST_CAP_NCQ) && (hpriv->flags & AHCI_HFLAG_NO_NCQ)) {
487 		dev_info(dev, "controller can't do NCQ, turning off CAP_NCQ\n");
488 		cap &= ~HOST_CAP_NCQ;
489 	}
490 
491 	if (!(cap & HOST_CAP_NCQ) && (hpriv->flags & AHCI_HFLAG_YES_NCQ)) {
492 		dev_info(dev, "controller can do NCQ, turning on CAP_NCQ\n");
493 		cap |= HOST_CAP_NCQ;
494 	}
495 
496 	if ((cap & HOST_CAP_PMP) && (hpriv->flags & AHCI_HFLAG_NO_PMP)) {
497 		dev_info(dev, "controller can't do PMP, turning off CAP_PMP\n");
498 		cap &= ~HOST_CAP_PMP;
499 	}
500 
501 	if ((cap & HOST_CAP_SNTF) && (hpriv->flags & AHCI_HFLAG_NO_SNTF)) {
502 		dev_info(dev,
503 			 "controller can't do SNTF, turning off CAP_SNTF\n");
504 		cap &= ~HOST_CAP_SNTF;
505 	}
506 
507 	if ((cap2 & HOST_CAP2_SDS) && (hpriv->flags & AHCI_HFLAG_NO_DEVSLP)) {
508 		dev_info(dev,
509 			 "controller can't do DEVSLP, turning off\n");
510 		cap2 &= ~HOST_CAP2_SDS;
511 		cap2 &= ~HOST_CAP2_SADM;
512 	}
513 
514 	if (!(cap & HOST_CAP_FBS) && (hpriv->flags & AHCI_HFLAG_YES_FBS)) {
515 		dev_info(dev, "controller can do FBS, turning on CAP_FBS\n");
516 		cap |= HOST_CAP_FBS;
517 	}
518 
519 	if ((cap & HOST_CAP_FBS) && (hpriv->flags & AHCI_HFLAG_NO_FBS)) {
520 		dev_info(dev, "controller can't do FBS, turning off CAP_FBS\n");
521 		cap &= ~HOST_CAP_FBS;
522 	}
523 
524 	if (!(cap & HOST_CAP_ALPM) && (hpriv->flags & AHCI_HFLAG_YES_ALPM)) {
525 		dev_info(dev, "controller can do ALPM, turning on CAP_ALPM\n");
526 		cap |= HOST_CAP_ALPM;
527 	}
528 
529 	if ((cap & HOST_CAP_SXS) && (hpriv->flags & AHCI_HFLAG_NO_SXS)) {
530 		dev_info(dev, "controller does not support SXS, disabling CAP_SXS\n");
531 		cap &= ~HOST_CAP_SXS;
532 	}
533 
534 	/* Override the HBA ports mapping if the platform needs it */
535 	port_map = readl(mmio + HOST_PORTS_IMPL);
536 	if (hpriv->saved_port_map && port_map != hpriv->saved_port_map) {
537 		dev_info(dev, "forcing port_map 0x%lx -> 0x%x\n",
538 			 port_map, hpriv->saved_port_map);
539 		port_map = hpriv->saved_port_map;
540 	} else {
541 		hpriv->saved_port_map = port_map;
542 	}
543 
544 	if (hpriv->mask_port_map) {
545 		dev_warn(dev, "masking port_map 0x%lx -> 0x%lx\n",
546 			port_map,
547 			port_map & hpriv->mask_port_map);
548 		port_map &= hpriv->mask_port_map;
549 	}
550 
551 	/* cross check port_map and cap.n_ports */
552 	if (port_map) {
553 		int map_ports = 0;
554 
555 		for (i = 0; i < AHCI_MAX_PORTS; i++)
556 			if (port_map & (1 << i))
557 				map_ports++;
558 
559 		/* If PI has more ports than n_ports, whine, clear
560 		 * port_map and let it be generated from n_ports.
561 		 */
562 		if (map_ports > ahci_nr_ports(cap)) {
563 			dev_warn(dev,
564 				 "implemented port map (0x%lx) contains more ports than nr_ports (%u), using nr_ports\n",
565 				 port_map, ahci_nr_ports(cap));
566 			port_map = 0;
567 		}
568 	}
569 
570 	/* fabricate port_map from cap.nr_ports for < AHCI 1.3 */
571 	if (!port_map && vers < 0x10300) {
572 		port_map = (1 << ahci_nr_ports(cap)) - 1;
573 		dev_warn(dev, "forcing PORTS_IMPL to 0x%lx\n", port_map);
574 
575 		/* write the fixed up value to the PI register */
576 		hpriv->saved_port_map = port_map;
577 	}
578 
579 	/*
580 	 * Preserve the ports capabilities defined by the platform. Note there
581 	 * is no need in storing the rest of the P#.CMD fields since they are
582 	 * volatile.
583 	 */
584 	for_each_set_bit(i, &port_map, AHCI_MAX_PORTS) {
585 		if (hpriv->saved_port_cap[i])
586 			continue;
587 
588 		port_mmio = __ahci_port_base(hpriv, i);
589 		hpriv->saved_port_cap[i] =
590 			readl(port_mmio + PORT_CMD) & PORT_CMD_CAP;
591 	}
592 
593 	/* record values to use during operation */
594 	hpriv->cap = cap;
595 	hpriv->cap2 = cap2;
596 	hpriv->version = vers;
597 	hpriv->port_map = port_map;
598 
599 	if (!hpriv->start_engine)
600 		hpriv->start_engine = ahci_start_engine;
601 
602 	if (!hpriv->stop_engine)
603 		hpriv->stop_engine = ahci_stop_engine;
604 
605 	if (!hpriv->irq_handler)
606 		hpriv->irq_handler = ahci_single_level_irq_intr;
607 }
608 EXPORT_SYMBOL_GPL(ahci_save_initial_config);
609 
610 /**
611  *	ahci_restore_initial_config - Restore initial config
612  *	@host: target ATA host
613  *
614  *	Restore initial config stored by ahci_save_initial_config().
615  *
616  *	LOCKING:
617  *	None.
618  */
619 static void ahci_restore_initial_config(struct ata_host *host)
620 {
621 	struct ahci_host_priv *hpriv = host->private_data;
622 	unsigned long port_map = hpriv->port_map;
623 	void __iomem *mmio = hpriv->mmio;
624 	void __iomem *port_mmio;
625 	int i;
626 
627 	writel(hpriv->saved_cap, mmio + HOST_CAP);
628 	if (hpriv->saved_cap2)
629 		writel(hpriv->saved_cap2, mmio + HOST_CAP2);
630 	writel(hpriv->saved_port_map, mmio + HOST_PORTS_IMPL);
631 	(void) readl(mmio + HOST_PORTS_IMPL);	/* flush */
632 
633 	for_each_set_bit(i, &port_map, AHCI_MAX_PORTS) {
634 		port_mmio = __ahci_port_base(hpriv, i);
635 		writel(hpriv->saved_port_cap[i], port_mmio + PORT_CMD);
636 	}
637 }
638 
639 static unsigned ahci_scr_offset(struct ata_port *ap, unsigned int sc_reg)
640 {
641 	static const int offset[] = {
642 		[SCR_STATUS]		= PORT_SCR_STAT,
643 		[SCR_CONTROL]		= PORT_SCR_CTL,
644 		[SCR_ERROR]		= PORT_SCR_ERR,
645 		[SCR_ACTIVE]		= PORT_SCR_ACT,
646 		[SCR_NOTIFICATION]	= PORT_SCR_NTF,
647 	};
648 	struct ahci_host_priv *hpriv = ap->host->private_data;
649 
650 	if (sc_reg < ARRAY_SIZE(offset) &&
651 	    (sc_reg != SCR_NOTIFICATION || (hpriv->cap & HOST_CAP_SNTF)))
652 		return offset[sc_reg];
653 	return 0;
654 }
655 
656 static int ahci_scr_read(struct ata_link *link, unsigned int sc_reg, u32 *val)
657 {
658 	void __iomem *port_mmio = ahci_port_base(link->ap);
659 	int offset = ahci_scr_offset(link->ap, sc_reg);
660 
661 	if (offset) {
662 		*val = readl(port_mmio + offset);
663 		return 0;
664 	}
665 	return -EINVAL;
666 }
667 
668 static int ahci_scr_write(struct ata_link *link, unsigned int sc_reg, u32 val)
669 {
670 	void __iomem *port_mmio = ahci_port_base(link->ap);
671 	int offset = ahci_scr_offset(link->ap, sc_reg);
672 
673 	if (offset) {
674 		writel(val, port_mmio + offset);
675 		return 0;
676 	}
677 	return -EINVAL;
678 }
679 
680 void ahci_start_engine(struct ata_port *ap)
681 {
682 	void __iomem *port_mmio = ahci_port_base(ap);
683 	u32 tmp;
684 
685 	/* start DMA */
686 	tmp = readl(port_mmio + PORT_CMD);
687 	tmp |= PORT_CMD_START;
688 	writel(tmp, port_mmio + PORT_CMD);
689 	readl(port_mmio + PORT_CMD); /* flush */
690 }
691 EXPORT_SYMBOL_GPL(ahci_start_engine);
692 
693 int ahci_stop_engine(struct ata_port *ap)
694 {
695 	void __iomem *port_mmio = ahci_port_base(ap);
696 	struct ahci_host_priv *hpriv = ap->host->private_data;
697 	u32 tmp;
698 
699 	/*
700 	 * On some controllers, stopping a port's DMA engine while the port
701 	 * is in ALPM state (partial or slumber) results in failures on
702 	 * subsequent DMA engine starts.  For those controllers, put the
703 	 * port back in active state before stopping its DMA engine.
704 	 */
705 	if ((hpriv->flags & AHCI_HFLAG_WAKE_BEFORE_STOP) &&
706 	    (ap->link.lpm_policy > ATA_LPM_MAX_POWER) &&
707 	    ahci_set_lpm(&ap->link, ATA_LPM_MAX_POWER, ATA_LPM_WAKE_ONLY)) {
708 		dev_err(ap->host->dev, "Failed to wake up port before engine stop\n");
709 		return -EIO;
710 	}
711 
712 	tmp = readl(port_mmio + PORT_CMD);
713 
714 	/* check if the HBA is idle */
715 	if ((tmp & (PORT_CMD_START | PORT_CMD_LIST_ON)) == 0)
716 		return 0;
717 
718 	/*
719 	 * Don't try to issue commands but return with ENODEV if the
720 	 * AHCI controller not available anymore (e.g. due to PCIe hot
721 	 * unplugging). Otherwise a 500ms delay for each port is added.
722 	 */
723 	if (tmp == 0xffffffff) {
724 		dev_err(ap->host->dev, "AHCI controller unavailable!\n");
725 		return -ENODEV;
726 	}
727 
728 	/* setting HBA to idle */
729 	tmp &= ~PORT_CMD_START;
730 	writel(tmp, port_mmio + PORT_CMD);
731 
732 	/* wait for engine to stop. This could be as long as 500 msec */
733 	tmp = ata_wait_register(ap, port_mmio + PORT_CMD,
734 				PORT_CMD_LIST_ON, PORT_CMD_LIST_ON, 1, 500);
735 	if (tmp & PORT_CMD_LIST_ON)
736 		return -EIO;
737 
738 	return 0;
739 }
740 EXPORT_SYMBOL_GPL(ahci_stop_engine);
741 
742 void ahci_start_fis_rx(struct ata_port *ap)
743 {
744 	void __iomem *port_mmio = ahci_port_base(ap);
745 	struct ahci_host_priv *hpriv = ap->host->private_data;
746 	struct ahci_port_priv *pp = ap->private_data;
747 	u32 tmp;
748 
749 	/* set FIS registers */
750 	if (hpriv->cap & HOST_CAP_64)
751 		writel((pp->cmd_slot_dma >> 16) >> 16,
752 		       port_mmio + PORT_LST_ADDR_HI);
753 	writel(pp->cmd_slot_dma & 0xffffffff, port_mmio + PORT_LST_ADDR);
754 
755 	if (hpriv->cap & HOST_CAP_64)
756 		writel((pp->rx_fis_dma >> 16) >> 16,
757 		       port_mmio + PORT_FIS_ADDR_HI);
758 	writel(pp->rx_fis_dma & 0xffffffff, port_mmio + PORT_FIS_ADDR);
759 
760 	/* enable FIS reception */
761 	tmp = readl(port_mmio + PORT_CMD);
762 	tmp |= PORT_CMD_FIS_RX;
763 	writel(tmp, port_mmio + PORT_CMD);
764 
765 	/* flush */
766 	readl(port_mmio + PORT_CMD);
767 }
768 EXPORT_SYMBOL_GPL(ahci_start_fis_rx);
769 
770 static int ahci_stop_fis_rx(struct ata_port *ap)
771 {
772 	void __iomem *port_mmio = ahci_port_base(ap);
773 	u32 tmp;
774 
775 	/* disable FIS reception */
776 	tmp = readl(port_mmio + PORT_CMD);
777 	tmp &= ~PORT_CMD_FIS_RX;
778 	writel(tmp, port_mmio + PORT_CMD);
779 
780 	/* wait for completion, spec says 500ms, give it 1000 */
781 	tmp = ata_wait_register(ap, port_mmio + PORT_CMD, PORT_CMD_FIS_ON,
782 				PORT_CMD_FIS_ON, 10, 1000);
783 	if (tmp & PORT_CMD_FIS_ON)
784 		return -EBUSY;
785 
786 	return 0;
787 }
788 
789 static void ahci_power_up(struct ata_port *ap)
790 {
791 	struct ahci_host_priv *hpriv = ap->host->private_data;
792 	void __iomem *port_mmio = ahci_port_base(ap);
793 	u32 cmd;
794 
795 	cmd = readl(port_mmio + PORT_CMD) & ~PORT_CMD_ICC_MASK;
796 
797 	/* spin up device */
798 	if (hpriv->cap & HOST_CAP_SSS) {
799 		cmd |= PORT_CMD_SPIN_UP;
800 		writel(cmd, port_mmio + PORT_CMD);
801 	}
802 
803 	/* wake up link */
804 	writel(cmd | PORT_CMD_ICC_ACTIVE, port_mmio + PORT_CMD);
805 }
806 
807 static int ahci_set_lpm(struct ata_link *link, enum ata_lpm_policy policy,
808 			unsigned int hints)
809 {
810 	struct ata_port *ap = link->ap;
811 	struct ahci_host_priv *hpriv = ap->host->private_data;
812 	struct ahci_port_priv *pp = ap->private_data;
813 	void __iomem *port_mmio = ahci_port_base(ap);
814 
815 	if (policy != ATA_LPM_MAX_POWER) {
816 		/* wakeup flag only applies to the max power policy */
817 		hints &= ~ATA_LPM_WAKE_ONLY;
818 
819 		/*
820 		 * Disable interrupts on Phy Ready. This keeps us from
821 		 * getting woken up due to spurious phy ready
822 		 * interrupts.
823 		 */
824 		pp->intr_mask &= ~PORT_IRQ_PHYRDY;
825 		writel(pp->intr_mask, port_mmio + PORT_IRQ_MASK);
826 
827 		sata_link_scr_lpm(link, policy, false);
828 	}
829 
830 	if (hpriv->cap & HOST_CAP_ALPM) {
831 		u32 cmd = readl(port_mmio + PORT_CMD);
832 
833 		if (policy == ATA_LPM_MAX_POWER || !(hints & ATA_LPM_HIPM)) {
834 			if (!(hints & ATA_LPM_WAKE_ONLY))
835 				cmd &= ~(PORT_CMD_ASP | PORT_CMD_ALPE);
836 			cmd |= PORT_CMD_ICC_ACTIVE;
837 
838 			writel(cmd, port_mmio + PORT_CMD);
839 			readl(port_mmio + PORT_CMD);
840 
841 			/* wait 10ms to be sure we've come out of LPM state */
842 			ata_msleep(ap, 10);
843 
844 			if (hints & ATA_LPM_WAKE_ONLY)
845 				return 0;
846 		} else {
847 			cmd |= PORT_CMD_ALPE;
848 			if (policy == ATA_LPM_MIN_POWER)
849 				cmd |= PORT_CMD_ASP;
850 			else if (policy == ATA_LPM_MIN_POWER_WITH_PARTIAL)
851 				cmd &= ~PORT_CMD_ASP;
852 
853 			/* write out new cmd value */
854 			writel(cmd, port_mmio + PORT_CMD);
855 		}
856 	}
857 
858 	/* set aggressive device sleep */
859 	if ((hpriv->cap2 & HOST_CAP2_SDS) &&
860 	    (hpriv->cap2 & HOST_CAP2_SADM) &&
861 	    (link->device->flags & ATA_DFLAG_DEVSLP)) {
862 		if (policy == ATA_LPM_MIN_POWER ||
863 		    policy == ATA_LPM_MIN_POWER_WITH_PARTIAL)
864 			ahci_set_aggressive_devslp(ap, true);
865 		else
866 			ahci_set_aggressive_devslp(ap, false);
867 	}
868 
869 	if (policy == ATA_LPM_MAX_POWER) {
870 		sata_link_scr_lpm(link, policy, false);
871 
872 		/* turn PHYRDY IRQ back on */
873 		pp->intr_mask |= PORT_IRQ_PHYRDY;
874 		writel(pp->intr_mask, port_mmio + PORT_IRQ_MASK);
875 	}
876 
877 	return 0;
878 }
879 
880 #ifdef CONFIG_PM
881 static void ahci_power_down(struct ata_port *ap)
882 {
883 	struct ahci_host_priv *hpriv = ap->host->private_data;
884 	void __iomem *port_mmio = ahci_port_base(ap);
885 	u32 cmd, scontrol;
886 
887 	if (!(hpriv->cap & HOST_CAP_SSS))
888 		return;
889 
890 	/* put device into listen mode, first set PxSCTL.DET to 0 */
891 	scontrol = readl(port_mmio + PORT_SCR_CTL);
892 	scontrol &= ~0xf;
893 	writel(scontrol, port_mmio + PORT_SCR_CTL);
894 
895 	/* then set PxCMD.SUD to 0 */
896 	cmd = readl(port_mmio + PORT_CMD) & ~PORT_CMD_ICC_MASK;
897 	cmd &= ~PORT_CMD_SPIN_UP;
898 	writel(cmd, port_mmio + PORT_CMD);
899 }
900 #endif
901 
902 static void ahci_start_port(struct ata_port *ap)
903 {
904 	struct ahci_host_priv *hpriv = ap->host->private_data;
905 	struct ahci_port_priv *pp = ap->private_data;
906 	struct ata_link *link;
907 	struct ahci_em_priv *emp;
908 	ssize_t rc;
909 	int i;
910 
911 	/* enable FIS reception */
912 	ahci_start_fis_rx(ap);
913 
914 	/* enable DMA */
915 	if (!(hpriv->flags & AHCI_HFLAG_DELAY_ENGINE))
916 		hpriv->start_engine(ap);
917 
918 	/* turn on LEDs */
919 	if (ap->flags & ATA_FLAG_EM) {
920 		ata_for_each_link(link, ap, EDGE) {
921 			emp = &pp->em_priv[link->pmp];
922 
923 			/* EM Transmit bit maybe busy during init */
924 			for (i = 0; i < EM_MAX_RETRY; i++) {
925 				rc = ap->ops->transmit_led_message(ap,
926 							       emp->led_state,
927 							       4);
928 				/*
929 				 * If busy, give a breather but do not
930 				 * release EH ownership by using msleep()
931 				 * instead of ata_msleep().  EM Transmit
932 				 * bit is busy for the whole host and
933 				 * releasing ownership will cause other
934 				 * ports to fail the same way.
935 				 */
936 				if (rc == -EBUSY)
937 					msleep(1);
938 				else
939 					break;
940 			}
941 		}
942 	}
943 
944 	if (ap->flags & ATA_FLAG_SW_ACTIVITY)
945 		ata_for_each_link(link, ap, EDGE)
946 			ahci_init_sw_activity(link);
947 
948 }
949 
950 static int ahci_deinit_port(struct ata_port *ap, const char **emsg)
951 {
952 	int rc;
953 	struct ahci_host_priv *hpriv = ap->host->private_data;
954 
955 	/* disable DMA */
956 	rc = hpriv->stop_engine(ap);
957 	if (rc) {
958 		*emsg = "failed to stop engine";
959 		return rc;
960 	}
961 
962 	/* disable FIS reception */
963 	rc = ahci_stop_fis_rx(ap);
964 	if (rc) {
965 		*emsg = "failed stop FIS RX";
966 		return rc;
967 	}
968 
969 	return 0;
970 }
971 
972 int ahci_reset_controller(struct ata_host *host)
973 {
974 	struct ahci_host_priv *hpriv = host->private_data;
975 	void __iomem *mmio = hpriv->mmio;
976 	u32 tmp;
977 
978 	/*
979 	 * We must be in AHCI mode, before using anything AHCI-specific, such
980 	 * as HOST_RESET.
981 	 */
982 	ahci_enable_ahci(mmio);
983 
984 	/* Global controller reset */
985 	if (ahci_skip_host_reset) {
986 		dev_info(host->dev, "Skipping global host reset\n");
987 		return 0;
988 	}
989 
990 	tmp = readl(mmio + HOST_CTL);
991 	if (!(tmp & HOST_RESET)) {
992 		writel(tmp | HOST_RESET, mmio + HOST_CTL);
993 		readl(mmio + HOST_CTL); /* flush */
994 	}
995 
996 	/*
997 	 * To perform host reset, OS should set HOST_RESET and poll until this
998 	 * bit is read to be "0". Reset must complete within 1 second, or the
999 	 * hardware should be considered fried.
1000 	 */
1001 	tmp = ata_wait_register(NULL, mmio + HOST_CTL, HOST_RESET,
1002 				HOST_RESET, 10, 1000);
1003 	if (tmp & HOST_RESET) {
1004 		dev_err(host->dev, "Controller reset failed (0x%x)\n",
1005 			tmp);
1006 		return -EIO;
1007 	}
1008 
1009 	/* Turn on AHCI mode */
1010 	ahci_enable_ahci(mmio);
1011 
1012 	/* Some registers might be cleared on reset. Restore initial values. */
1013 	if (!(hpriv->flags & AHCI_HFLAG_NO_WRITE_TO_RO))
1014 		ahci_restore_initial_config(host);
1015 
1016 	return 0;
1017 }
1018 EXPORT_SYMBOL_GPL(ahci_reset_controller);
1019 
1020 static void ahci_sw_activity(struct ata_link *link)
1021 {
1022 	struct ata_port *ap = link->ap;
1023 	struct ahci_port_priv *pp = ap->private_data;
1024 	struct ahci_em_priv *emp = &pp->em_priv[link->pmp];
1025 
1026 	if (!(link->flags & ATA_LFLAG_SW_ACTIVITY))
1027 		return;
1028 
1029 	emp->activity++;
1030 	if (!timer_pending(&emp->timer))
1031 		mod_timer(&emp->timer, jiffies + msecs_to_jiffies(10));
1032 }
1033 
1034 static void ahci_sw_activity_blink(struct timer_list *t)
1035 {
1036 	struct ahci_em_priv *emp = from_timer(emp, t, timer);
1037 	struct ata_link *link = emp->link;
1038 	struct ata_port *ap = link->ap;
1039 
1040 	unsigned long led_message = emp->led_state;
1041 	u32 activity_led_state;
1042 	unsigned long flags;
1043 
1044 	led_message &= EM_MSG_LED_VALUE;
1045 	led_message |= ap->port_no | (link->pmp << 8);
1046 
1047 	/* check to see if we've had activity.  If so,
1048 	 * toggle state of LED and reset timer.  If not,
1049 	 * turn LED to desired idle state.
1050 	 */
1051 	spin_lock_irqsave(ap->lock, flags);
1052 	if (emp->saved_activity != emp->activity) {
1053 		emp->saved_activity = emp->activity;
1054 		/* get the current LED state */
1055 		activity_led_state = led_message & EM_MSG_LED_VALUE_ON;
1056 
1057 		if (activity_led_state)
1058 			activity_led_state = 0;
1059 		else
1060 			activity_led_state = 1;
1061 
1062 		/* clear old state */
1063 		led_message &= ~EM_MSG_LED_VALUE_ACTIVITY;
1064 
1065 		/* toggle state */
1066 		led_message |= (activity_led_state << 16);
1067 		mod_timer(&emp->timer, jiffies + msecs_to_jiffies(100));
1068 	} else {
1069 		/* switch to idle */
1070 		led_message &= ~EM_MSG_LED_VALUE_ACTIVITY;
1071 		if (emp->blink_policy == BLINK_OFF)
1072 			led_message |= (1 << 16);
1073 	}
1074 	spin_unlock_irqrestore(ap->lock, flags);
1075 	ap->ops->transmit_led_message(ap, led_message, 4);
1076 }
1077 
1078 static void ahci_init_sw_activity(struct ata_link *link)
1079 {
1080 	struct ata_port *ap = link->ap;
1081 	struct ahci_port_priv *pp = ap->private_data;
1082 	struct ahci_em_priv *emp = &pp->em_priv[link->pmp];
1083 
1084 	/* init activity stats, setup timer */
1085 	emp->saved_activity = emp->activity = 0;
1086 	emp->link = link;
1087 	timer_setup(&emp->timer, ahci_sw_activity_blink, 0);
1088 
1089 	/* check our blink policy and set flag for link if it's enabled */
1090 	if (emp->blink_policy)
1091 		link->flags |= ATA_LFLAG_SW_ACTIVITY;
1092 }
1093 
1094 int ahci_reset_em(struct ata_host *host)
1095 {
1096 	struct ahci_host_priv *hpriv = host->private_data;
1097 	void __iomem *mmio = hpriv->mmio;
1098 	u32 em_ctl;
1099 
1100 	em_ctl = readl(mmio + HOST_EM_CTL);
1101 	if ((em_ctl & EM_CTL_TM) || (em_ctl & EM_CTL_RST))
1102 		return -EINVAL;
1103 
1104 	writel(em_ctl | EM_CTL_RST, mmio + HOST_EM_CTL);
1105 	return 0;
1106 }
1107 EXPORT_SYMBOL_GPL(ahci_reset_em);
1108 
1109 static ssize_t ahci_transmit_led_message(struct ata_port *ap, u32 state,
1110 					ssize_t size)
1111 {
1112 	struct ahci_host_priv *hpriv = ap->host->private_data;
1113 	struct ahci_port_priv *pp = ap->private_data;
1114 	void __iomem *mmio = hpriv->mmio;
1115 	u32 em_ctl;
1116 	u32 message[] = {0, 0};
1117 	unsigned long flags;
1118 	int pmp;
1119 	struct ahci_em_priv *emp;
1120 
1121 	/* get the slot number from the message */
1122 	pmp = (state & EM_MSG_LED_PMP_SLOT) >> 8;
1123 	if (pmp < EM_MAX_SLOTS)
1124 		emp = &pp->em_priv[pmp];
1125 	else
1126 		return -EINVAL;
1127 
1128 	ahci_rpm_get_port(ap);
1129 	spin_lock_irqsave(ap->lock, flags);
1130 
1131 	/*
1132 	 * if we are still busy transmitting a previous message,
1133 	 * do not allow
1134 	 */
1135 	em_ctl = readl(mmio + HOST_EM_CTL);
1136 	if (em_ctl & EM_CTL_TM) {
1137 		spin_unlock_irqrestore(ap->lock, flags);
1138 		ahci_rpm_put_port(ap);
1139 		return -EBUSY;
1140 	}
1141 
1142 	if (hpriv->em_msg_type & EM_MSG_TYPE_LED) {
1143 		/*
1144 		 * create message header - this is all zero except for
1145 		 * the message size, which is 4 bytes.
1146 		 */
1147 		message[0] |= (4 << 8);
1148 
1149 		/* ignore 0:4 of byte zero, fill in port info yourself */
1150 		message[1] = ((state & ~EM_MSG_LED_HBA_PORT) | ap->port_no);
1151 
1152 		/* write message to EM_LOC */
1153 		writel(message[0], mmio + hpriv->em_loc);
1154 		writel(message[1], mmio + hpriv->em_loc+4);
1155 
1156 		/*
1157 		 * tell hardware to transmit the message
1158 		 */
1159 		writel(em_ctl | EM_CTL_TM, mmio + HOST_EM_CTL);
1160 	}
1161 
1162 	/* save off new led state for port/slot */
1163 	emp->led_state = state;
1164 
1165 	spin_unlock_irqrestore(ap->lock, flags);
1166 	ahci_rpm_put_port(ap);
1167 
1168 	return size;
1169 }
1170 
1171 static ssize_t ahci_led_show(struct ata_port *ap, char *buf)
1172 {
1173 	struct ahci_port_priv *pp = ap->private_data;
1174 	struct ata_link *link;
1175 	struct ahci_em_priv *emp;
1176 	int rc = 0;
1177 
1178 	ata_for_each_link(link, ap, EDGE) {
1179 		emp = &pp->em_priv[link->pmp];
1180 		rc += sprintf(buf, "%lx\n", emp->led_state);
1181 	}
1182 	return rc;
1183 }
1184 
1185 static ssize_t ahci_led_store(struct ata_port *ap, const char *buf,
1186 				size_t size)
1187 {
1188 	unsigned int state;
1189 	int pmp;
1190 	struct ahci_port_priv *pp = ap->private_data;
1191 	struct ahci_em_priv *emp;
1192 
1193 	if (kstrtouint(buf, 0, &state) < 0)
1194 		return -EINVAL;
1195 
1196 	/* get the slot number from the message */
1197 	pmp = (state & EM_MSG_LED_PMP_SLOT) >> 8;
1198 	if (pmp < EM_MAX_SLOTS) {
1199 		pmp = array_index_nospec(pmp, EM_MAX_SLOTS);
1200 		emp = &pp->em_priv[pmp];
1201 	} else {
1202 		return -EINVAL;
1203 	}
1204 
1205 	/* mask off the activity bits if we are in sw_activity
1206 	 * mode, user should turn off sw_activity before setting
1207 	 * activity led through em_message
1208 	 */
1209 	if (emp->blink_policy)
1210 		state &= ~EM_MSG_LED_VALUE_ACTIVITY;
1211 
1212 	return ap->ops->transmit_led_message(ap, state, size);
1213 }
1214 
1215 static ssize_t ahci_activity_store(struct ata_device *dev, enum sw_activity val)
1216 {
1217 	struct ata_link *link = dev->link;
1218 	struct ata_port *ap = link->ap;
1219 	struct ahci_port_priv *pp = ap->private_data;
1220 	struct ahci_em_priv *emp = &pp->em_priv[link->pmp];
1221 	u32 port_led_state = emp->led_state;
1222 
1223 	/* save the desired Activity LED behavior */
1224 	if (val == OFF) {
1225 		/* clear LFLAG */
1226 		link->flags &= ~(ATA_LFLAG_SW_ACTIVITY);
1227 
1228 		/* set the LED to OFF */
1229 		port_led_state &= EM_MSG_LED_VALUE_OFF;
1230 		port_led_state |= (ap->port_no | (link->pmp << 8));
1231 		ap->ops->transmit_led_message(ap, port_led_state, 4);
1232 	} else {
1233 		link->flags |= ATA_LFLAG_SW_ACTIVITY;
1234 		if (val == BLINK_OFF) {
1235 			/* set LED to ON for idle */
1236 			port_led_state &= EM_MSG_LED_VALUE_OFF;
1237 			port_led_state |= (ap->port_no | (link->pmp << 8));
1238 			port_led_state |= EM_MSG_LED_VALUE_ON; /* check this */
1239 			ap->ops->transmit_led_message(ap, port_led_state, 4);
1240 		}
1241 	}
1242 	emp->blink_policy = val;
1243 	return 0;
1244 }
1245 
1246 static ssize_t ahci_activity_show(struct ata_device *dev, char *buf)
1247 {
1248 	struct ata_link *link = dev->link;
1249 	struct ata_port *ap = link->ap;
1250 	struct ahci_port_priv *pp = ap->private_data;
1251 	struct ahci_em_priv *emp = &pp->em_priv[link->pmp];
1252 
1253 	/* display the saved value of activity behavior for this
1254 	 * disk.
1255 	 */
1256 	return sprintf(buf, "%d\n", emp->blink_policy);
1257 }
1258 
1259 static void ahci_port_clear_pending_irq(struct ata_port *ap)
1260 {
1261 	struct ahci_host_priv *hpriv = ap->host->private_data;
1262 	void __iomem *port_mmio = ahci_port_base(ap);
1263 	u32 tmp;
1264 
1265 	/* clear SError */
1266 	tmp = readl(port_mmio + PORT_SCR_ERR);
1267 	dev_dbg(ap->host->dev, "PORT_SCR_ERR 0x%x\n", tmp);
1268 	writel(tmp, port_mmio + PORT_SCR_ERR);
1269 
1270 	/* clear port IRQ */
1271 	tmp = readl(port_mmio + PORT_IRQ_STAT);
1272 	dev_dbg(ap->host->dev, "PORT_IRQ_STAT 0x%x\n", tmp);
1273 	if (tmp)
1274 		writel(tmp, port_mmio + PORT_IRQ_STAT);
1275 
1276 	writel(1 << ap->port_no, hpriv->mmio + HOST_IRQ_STAT);
1277 }
1278 
1279 static void ahci_port_init(struct device *dev, struct ata_port *ap,
1280 			   int port_no, void __iomem *mmio,
1281 			   void __iomem *port_mmio)
1282 {
1283 	const char *emsg = NULL;
1284 	int rc;
1285 
1286 	/* make sure port is not active */
1287 	rc = ahci_deinit_port(ap, &emsg);
1288 	if (rc)
1289 		dev_warn(dev, "%s (%d)\n", emsg, rc);
1290 
1291 	ahci_port_clear_pending_irq(ap);
1292 }
1293 
1294 void ahci_init_controller(struct ata_host *host)
1295 {
1296 	struct ahci_host_priv *hpriv = host->private_data;
1297 	void __iomem *mmio = hpriv->mmio;
1298 	int i;
1299 	void __iomem *port_mmio;
1300 	u32 tmp;
1301 
1302 	for (i = 0; i < host->n_ports; i++) {
1303 		struct ata_port *ap = host->ports[i];
1304 
1305 		port_mmio = ahci_port_base(ap);
1306 		if (ata_port_is_dummy(ap))
1307 			continue;
1308 
1309 		ahci_port_init(host->dev, ap, i, mmio, port_mmio);
1310 	}
1311 
1312 	tmp = readl(mmio + HOST_CTL);
1313 	dev_dbg(host->dev, "HOST_CTL 0x%x\n", tmp);
1314 	writel(tmp | HOST_IRQ_EN, mmio + HOST_CTL);
1315 	tmp = readl(mmio + HOST_CTL);
1316 	dev_dbg(host->dev, "HOST_CTL 0x%x\n", tmp);
1317 }
1318 EXPORT_SYMBOL_GPL(ahci_init_controller);
1319 
1320 static void ahci_dev_config(struct ata_device *dev)
1321 {
1322 	struct ahci_host_priv *hpriv = dev->link->ap->host->private_data;
1323 
1324 	if (hpriv->flags & AHCI_HFLAG_SECT255) {
1325 		dev->max_sectors = 255;
1326 		ata_dev_info(dev,
1327 			     "SB600 AHCI: limiting to 255 sectors per cmd\n");
1328 	}
1329 }
1330 
1331 unsigned int ahci_dev_classify(struct ata_port *ap)
1332 {
1333 	void __iomem *port_mmio = ahci_port_base(ap);
1334 	struct ata_taskfile tf;
1335 	u32 tmp;
1336 
1337 	tmp = readl(port_mmio + PORT_SIG);
1338 	tf.lbah		= (tmp >> 24)	& 0xff;
1339 	tf.lbam		= (tmp >> 16)	& 0xff;
1340 	tf.lbal		= (tmp >> 8)	& 0xff;
1341 	tf.nsect	= (tmp)		& 0xff;
1342 
1343 	return ata_port_classify(ap, &tf);
1344 }
1345 EXPORT_SYMBOL_GPL(ahci_dev_classify);
1346 
1347 void ahci_fill_cmd_slot(struct ahci_port_priv *pp, unsigned int tag,
1348 			u32 opts)
1349 {
1350 	dma_addr_t cmd_tbl_dma;
1351 
1352 	cmd_tbl_dma = pp->cmd_tbl_dma + tag * AHCI_CMD_TBL_SZ;
1353 
1354 	pp->cmd_slot[tag].opts = cpu_to_le32(opts);
1355 	pp->cmd_slot[tag].status = 0;
1356 	pp->cmd_slot[tag].tbl_addr = cpu_to_le32(cmd_tbl_dma & 0xffffffff);
1357 	pp->cmd_slot[tag].tbl_addr_hi = cpu_to_le32((cmd_tbl_dma >> 16) >> 16);
1358 }
1359 EXPORT_SYMBOL_GPL(ahci_fill_cmd_slot);
1360 
1361 int ahci_kick_engine(struct ata_port *ap)
1362 {
1363 	void __iomem *port_mmio = ahci_port_base(ap);
1364 	struct ahci_host_priv *hpriv = ap->host->private_data;
1365 	u8 status = readl(port_mmio + PORT_TFDATA) & 0xFF;
1366 	u32 tmp;
1367 	int busy, rc;
1368 
1369 	/* stop engine */
1370 	rc = hpriv->stop_engine(ap);
1371 	if (rc)
1372 		goto out_restart;
1373 
1374 	/* need to do CLO?
1375 	 * always do CLO if PMP is attached (AHCI-1.3 9.2)
1376 	 */
1377 	busy = status & (ATA_BUSY | ATA_DRQ);
1378 	if (!busy && !sata_pmp_attached(ap)) {
1379 		rc = 0;
1380 		goto out_restart;
1381 	}
1382 
1383 	if (!(hpriv->cap & HOST_CAP_CLO)) {
1384 		rc = -EOPNOTSUPP;
1385 		goto out_restart;
1386 	}
1387 
1388 	/* perform CLO */
1389 	tmp = readl(port_mmio + PORT_CMD);
1390 	tmp |= PORT_CMD_CLO;
1391 	writel(tmp, port_mmio + PORT_CMD);
1392 
1393 	rc = 0;
1394 	tmp = ata_wait_register(ap, port_mmio + PORT_CMD,
1395 				PORT_CMD_CLO, PORT_CMD_CLO, 1, 500);
1396 	if (tmp & PORT_CMD_CLO)
1397 		rc = -EIO;
1398 
1399 	/* restart engine */
1400  out_restart:
1401 	hpriv->start_engine(ap);
1402 	return rc;
1403 }
1404 EXPORT_SYMBOL_GPL(ahci_kick_engine);
1405 
1406 static int ahci_exec_polled_cmd(struct ata_port *ap, int pmp,
1407 				struct ata_taskfile *tf, int is_cmd, u16 flags,
1408 				unsigned int timeout_msec)
1409 {
1410 	const u32 cmd_fis_len = 5; /* five dwords */
1411 	struct ahci_port_priv *pp = ap->private_data;
1412 	void __iomem *port_mmio = ahci_port_base(ap);
1413 	u8 *fis = pp->cmd_tbl;
1414 	u32 tmp;
1415 
1416 	/* prep the command */
1417 	ata_tf_to_fis(tf, pmp, is_cmd, fis);
1418 	ahci_fill_cmd_slot(pp, 0, cmd_fis_len | flags | (pmp << 12));
1419 
1420 	/* set port value for softreset of Port Multiplier */
1421 	if (pp->fbs_enabled && pp->fbs_last_dev != pmp) {
1422 		tmp = readl(port_mmio + PORT_FBS);
1423 		tmp &= ~(PORT_FBS_DEV_MASK | PORT_FBS_DEC);
1424 		tmp |= pmp << PORT_FBS_DEV_OFFSET;
1425 		writel(tmp, port_mmio + PORT_FBS);
1426 		pp->fbs_last_dev = pmp;
1427 	}
1428 
1429 	/* issue & wait */
1430 	writel(1, port_mmio + PORT_CMD_ISSUE);
1431 
1432 	if (timeout_msec) {
1433 		tmp = ata_wait_register(ap, port_mmio + PORT_CMD_ISSUE,
1434 					0x1, 0x1, 1, timeout_msec);
1435 		if (tmp & 0x1) {
1436 			ahci_kick_engine(ap);
1437 			return -EBUSY;
1438 		}
1439 	} else
1440 		readl(port_mmio + PORT_CMD_ISSUE);	/* flush */
1441 
1442 	return 0;
1443 }
1444 
1445 int ahci_do_softreset(struct ata_link *link, unsigned int *class,
1446 		      int pmp, unsigned long deadline,
1447 		      int (*check_ready)(struct ata_link *link))
1448 {
1449 	struct ata_port *ap = link->ap;
1450 	struct ahci_host_priv *hpriv = ap->host->private_data;
1451 	struct ahci_port_priv *pp = ap->private_data;
1452 	const char *reason = NULL;
1453 	unsigned long now;
1454 	unsigned int msecs;
1455 	struct ata_taskfile tf;
1456 	bool fbs_disabled = false;
1457 	int rc;
1458 
1459 	/* prepare for SRST (AHCI-1.1 10.4.1) */
1460 	rc = ahci_kick_engine(ap);
1461 	if (rc && rc != -EOPNOTSUPP)
1462 		ata_link_warn(link, "failed to reset engine (errno=%d)\n", rc);
1463 
1464 	/*
1465 	 * According to AHCI-1.2 9.3.9: if FBS is enable, software shall
1466 	 * clear PxFBS.EN to '0' prior to issuing software reset to devices
1467 	 * that is attached to port multiplier.
1468 	 */
1469 	if (!ata_is_host_link(link) && pp->fbs_enabled) {
1470 		ahci_disable_fbs(ap);
1471 		fbs_disabled = true;
1472 	}
1473 
1474 	ata_tf_init(link->device, &tf);
1475 
1476 	/* issue the first H2D Register FIS */
1477 	msecs = 0;
1478 	now = jiffies;
1479 	if (time_after(deadline, now))
1480 		msecs = jiffies_to_msecs(deadline - now);
1481 
1482 	tf.ctl |= ATA_SRST;
1483 	if (ahci_exec_polled_cmd(ap, pmp, &tf, 0,
1484 				 AHCI_CMD_RESET | AHCI_CMD_CLR_BUSY, msecs)) {
1485 		rc = -EIO;
1486 		reason = "1st FIS failed";
1487 		goto fail;
1488 	}
1489 
1490 	/* spec says at least 5us, but be generous and sleep for 1ms */
1491 	ata_msleep(ap, 1);
1492 
1493 	/* issue the second H2D Register FIS */
1494 	tf.ctl &= ~ATA_SRST;
1495 	ahci_exec_polled_cmd(ap, pmp, &tf, 0, 0, 0);
1496 
1497 	/* wait for link to become ready */
1498 	rc = ata_wait_after_reset(link, deadline, check_ready);
1499 	if (rc == -EBUSY && hpriv->flags & AHCI_HFLAG_SRST_TOUT_IS_OFFLINE) {
1500 		/*
1501 		 * Workaround for cases where link online status can't
1502 		 * be trusted.  Treat device readiness timeout as link
1503 		 * offline.
1504 		 */
1505 		ata_link_info(link, "device not ready, treating as offline\n");
1506 		*class = ATA_DEV_NONE;
1507 	} else if (rc) {
1508 		/* link occupied, -ENODEV too is an error */
1509 		reason = "device not ready";
1510 		goto fail;
1511 	} else
1512 		*class = ahci_dev_classify(ap);
1513 
1514 	/* re-enable FBS if disabled before */
1515 	if (fbs_disabled)
1516 		ahci_enable_fbs(ap);
1517 
1518 	return 0;
1519 
1520  fail:
1521 	ata_link_err(link, "softreset failed (%s)\n", reason);
1522 	return rc;
1523 }
1524 
1525 int ahci_check_ready(struct ata_link *link)
1526 {
1527 	void __iomem *port_mmio = ahci_port_base(link->ap);
1528 	u8 status = readl(port_mmio + PORT_TFDATA) & 0xFF;
1529 
1530 	return ata_check_ready(status);
1531 }
1532 EXPORT_SYMBOL_GPL(ahci_check_ready);
1533 
1534 static int ahci_softreset(struct ata_link *link, unsigned int *class,
1535 			  unsigned long deadline)
1536 {
1537 	int pmp = sata_srst_pmp(link);
1538 
1539 	return ahci_do_softreset(link, class, pmp, deadline, ahci_check_ready);
1540 }
1541 EXPORT_SYMBOL_GPL(ahci_do_softreset);
1542 
1543 static int ahci_bad_pmp_check_ready(struct ata_link *link)
1544 {
1545 	void __iomem *port_mmio = ahci_port_base(link->ap);
1546 	u8 status = readl(port_mmio + PORT_TFDATA) & 0xFF;
1547 	u32 irq_status = readl(port_mmio + PORT_IRQ_STAT);
1548 
1549 	/*
1550 	 * There is no need to check TFDATA if BAD PMP is found due to HW bug,
1551 	 * which can save timeout delay.
1552 	 */
1553 	if (irq_status & PORT_IRQ_BAD_PMP)
1554 		return -EIO;
1555 
1556 	return ata_check_ready(status);
1557 }
1558 
1559 static int ahci_pmp_retry_softreset(struct ata_link *link, unsigned int *class,
1560 				    unsigned long deadline)
1561 {
1562 	struct ata_port *ap = link->ap;
1563 	void __iomem *port_mmio = ahci_port_base(ap);
1564 	int pmp = sata_srst_pmp(link);
1565 	int rc;
1566 	u32 irq_sts;
1567 
1568 	rc = ahci_do_softreset(link, class, pmp, deadline,
1569 			       ahci_bad_pmp_check_ready);
1570 
1571 	/*
1572 	 * Soft reset fails with IPMS set when PMP is enabled but
1573 	 * SATA HDD/ODD is connected to SATA port, do soft reset
1574 	 * again to port 0.
1575 	 */
1576 	if (rc == -EIO) {
1577 		irq_sts = readl(port_mmio + PORT_IRQ_STAT);
1578 		if (irq_sts & PORT_IRQ_BAD_PMP) {
1579 			ata_link_warn(link,
1580 					"applying PMP SRST workaround "
1581 					"and retrying\n");
1582 			rc = ahci_do_softreset(link, class, 0, deadline,
1583 					       ahci_check_ready);
1584 		}
1585 	}
1586 
1587 	return rc;
1588 }
1589 
1590 int ahci_do_hardreset(struct ata_link *link, unsigned int *class,
1591 		      unsigned long deadline, bool *online)
1592 {
1593 	const unsigned int *timing = sata_ehc_deb_timing(&link->eh_context);
1594 	struct ata_port *ap = link->ap;
1595 	struct ahci_port_priv *pp = ap->private_data;
1596 	struct ahci_host_priv *hpriv = ap->host->private_data;
1597 	u8 *d2h_fis = pp->rx_fis + RX_FIS_D2H_REG;
1598 	struct ata_taskfile tf;
1599 	int rc;
1600 
1601 	hpriv->stop_engine(ap);
1602 
1603 	/* clear D2H reception area to properly wait for D2H FIS */
1604 	ata_tf_init(link->device, &tf);
1605 	tf.status = ATA_BUSY;
1606 	ata_tf_to_fis(&tf, 0, 0, d2h_fis);
1607 
1608 	ahci_port_clear_pending_irq(ap);
1609 
1610 	rc = sata_link_hardreset(link, timing, deadline, online,
1611 				 ahci_check_ready);
1612 
1613 	hpriv->start_engine(ap);
1614 
1615 	if (*online)
1616 		*class = ahci_dev_classify(ap);
1617 
1618 	return rc;
1619 }
1620 EXPORT_SYMBOL_GPL(ahci_do_hardreset);
1621 
1622 static int ahci_hardreset(struct ata_link *link, unsigned int *class,
1623 			  unsigned long deadline)
1624 {
1625 	bool online;
1626 
1627 	return ahci_do_hardreset(link, class, deadline, &online);
1628 }
1629 
1630 static void ahci_postreset(struct ata_link *link, unsigned int *class)
1631 {
1632 	struct ata_port *ap = link->ap;
1633 	void __iomem *port_mmio = ahci_port_base(ap);
1634 	u32 new_tmp, tmp;
1635 
1636 	ata_std_postreset(link, class);
1637 
1638 	/* Make sure port's ATAPI bit is set appropriately */
1639 	new_tmp = tmp = readl(port_mmio + PORT_CMD);
1640 	if (*class == ATA_DEV_ATAPI)
1641 		new_tmp |= PORT_CMD_ATAPI;
1642 	else
1643 		new_tmp &= ~PORT_CMD_ATAPI;
1644 	if (new_tmp != tmp) {
1645 		writel(new_tmp, port_mmio + PORT_CMD);
1646 		readl(port_mmio + PORT_CMD); /* flush */
1647 	}
1648 }
1649 
1650 static unsigned int ahci_fill_sg(struct ata_queued_cmd *qc, void *cmd_tbl)
1651 {
1652 	struct scatterlist *sg;
1653 	struct ahci_sg *ahci_sg = cmd_tbl + AHCI_CMD_TBL_HDR_SZ;
1654 	unsigned int si;
1655 
1656 	/*
1657 	 * Next, the S/G list.
1658 	 */
1659 	for_each_sg(qc->sg, sg, qc->n_elem, si) {
1660 		dma_addr_t addr = sg_dma_address(sg);
1661 		u32 sg_len = sg_dma_len(sg);
1662 
1663 		ahci_sg[si].addr = cpu_to_le32(addr & 0xffffffff);
1664 		ahci_sg[si].addr_hi = cpu_to_le32((addr >> 16) >> 16);
1665 		ahci_sg[si].flags_size = cpu_to_le32(sg_len - 1);
1666 	}
1667 
1668 	return si;
1669 }
1670 
1671 static int ahci_pmp_qc_defer(struct ata_queued_cmd *qc)
1672 {
1673 	struct ata_port *ap = qc->ap;
1674 	struct ahci_port_priv *pp = ap->private_data;
1675 
1676 	if (!sata_pmp_attached(ap) || pp->fbs_enabled)
1677 		return ata_std_qc_defer(qc);
1678 	else
1679 		return sata_pmp_qc_defer_cmd_switch(qc);
1680 }
1681 
1682 static enum ata_completion_errors ahci_qc_prep(struct ata_queued_cmd *qc)
1683 {
1684 	struct ata_port *ap = qc->ap;
1685 	struct ahci_port_priv *pp = ap->private_data;
1686 	int is_atapi = ata_is_atapi(qc->tf.protocol);
1687 	void *cmd_tbl;
1688 	u32 opts;
1689 	const u32 cmd_fis_len = 5; /* five dwords */
1690 	unsigned int n_elem;
1691 
1692 	/*
1693 	 * Fill in command table information.  First, the header,
1694 	 * a SATA Register - Host to Device command FIS.
1695 	 */
1696 	cmd_tbl = pp->cmd_tbl + qc->hw_tag * AHCI_CMD_TBL_SZ;
1697 
1698 	ata_tf_to_fis(&qc->tf, qc->dev->link->pmp, 1, cmd_tbl);
1699 	if (is_atapi) {
1700 		memset(cmd_tbl + AHCI_CMD_TBL_CDB, 0, 32);
1701 		memcpy(cmd_tbl + AHCI_CMD_TBL_CDB, qc->cdb, qc->dev->cdb_len);
1702 	}
1703 
1704 	n_elem = 0;
1705 	if (qc->flags & ATA_QCFLAG_DMAMAP)
1706 		n_elem = ahci_fill_sg(qc, cmd_tbl);
1707 
1708 	/*
1709 	 * Fill in command slot information.
1710 	 */
1711 	opts = cmd_fis_len | n_elem << 16 | (qc->dev->link->pmp << 12);
1712 	if (qc->tf.flags & ATA_TFLAG_WRITE)
1713 		opts |= AHCI_CMD_WRITE;
1714 	if (is_atapi)
1715 		opts |= AHCI_CMD_ATAPI | AHCI_CMD_PREFETCH;
1716 
1717 	ahci_fill_cmd_slot(pp, qc->hw_tag, opts);
1718 
1719 	return AC_ERR_OK;
1720 }
1721 
1722 static void ahci_fbs_dec_intr(struct ata_port *ap)
1723 {
1724 	struct ahci_port_priv *pp = ap->private_data;
1725 	void __iomem *port_mmio = ahci_port_base(ap);
1726 	u32 fbs = readl(port_mmio + PORT_FBS);
1727 	int retries = 3;
1728 
1729 	BUG_ON(!pp->fbs_enabled);
1730 
1731 	/* time to wait for DEC is not specified by AHCI spec,
1732 	 * add a retry loop for safety.
1733 	 */
1734 	writel(fbs | PORT_FBS_DEC, port_mmio + PORT_FBS);
1735 	fbs = readl(port_mmio + PORT_FBS);
1736 	while ((fbs & PORT_FBS_DEC) && retries--) {
1737 		udelay(1);
1738 		fbs = readl(port_mmio + PORT_FBS);
1739 	}
1740 
1741 	if (fbs & PORT_FBS_DEC)
1742 		dev_err(ap->host->dev, "failed to clear device error\n");
1743 }
1744 
1745 static void ahci_error_intr(struct ata_port *ap, u32 irq_stat)
1746 {
1747 	struct ahci_host_priv *hpriv = ap->host->private_data;
1748 	struct ahci_port_priv *pp = ap->private_data;
1749 	struct ata_eh_info *host_ehi = &ap->link.eh_info;
1750 	struct ata_link *link = NULL;
1751 	struct ata_queued_cmd *active_qc;
1752 	struct ata_eh_info *active_ehi;
1753 	bool fbs_need_dec = false;
1754 	u32 serror;
1755 
1756 	/* determine active link with error */
1757 	if (pp->fbs_enabled) {
1758 		void __iomem *port_mmio = ahci_port_base(ap);
1759 		u32 fbs = readl(port_mmio + PORT_FBS);
1760 		int pmp = fbs >> PORT_FBS_DWE_OFFSET;
1761 
1762 		if ((fbs & PORT_FBS_SDE) && (pmp < ap->nr_pmp_links)) {
1763 			link = &ap->pmp_link[pmp];
1764 			fbs_need_dec = true;
1765 		}
1766 
1767 	} else
1768 		ata_for_each_link(link, ap, EDGE)
1769 			if (ata_link_active(link))
1770 				break;
1771 
1772 	if (!link)
1773 		link = &ap->link;
1774 
1775 	active_qc = ata_qc_from_tag(ap, link->active_tag);
1776 	active_ehi = &link->eh_info;
1777 
1778 	/* record irq stat */
1779 	ata_ehi_clear_desc(host_ehi);
1780 	ata_ehi_push_desc(host_ehi, "irq_stat 0x%08x", irq_stat);
1781 
1782 	/* AHCI needs SError cleared; otherwise, it might lock up */
1783 	ahci_scr_read(&ap->link, SCR_ERROR, &serror);
1784 	ahci_scr_write(&ap->link, SCR_ERROR, serror);
1785 	host_ehi->serror |= serror;
1786 
1787 	/* some controllers set IRQ_IF_ERR on device errors, ignore it */
1788 	if (hpriv->flags & AHCI_HFLAG_IGN_IRQ_IF_ERR)
1789 		irq_stat &= ~PORT_IRQ_IF_ERR;
1790 
1791 	if (irq_stat & PORT_IRQ_TF_ERR) {
1792 		/* If qc is active, charge it; otherwise, the active
1793 		 * link.  There's no active qc on NCQ errors.  It will
1794 		 * be determined by EH by reading log page 10h.
1795 		 */
1796 		if (active_qc)
1797 			active_qc->err_mask |= AC_ERR_DEV;
1798 		else
1799 			active_ehi->err_mask |= AC_ERR_DEV;
1800 
1801 		if (hpriv->flags & AHCI_HFLAG_IGN_SERR_INTERNAL)
1802 			host_ehi->serror &= ~SERR_INTERNAL;
1803 	}
1804 
1805 	if (irq_stat & PORT_IRQ_UNK_FIS) {
1806 		u32 *unk = pp->rx_fis + RX_FIS_UNK;
1807 
1808 		active_ehi->err_mask |= AC_ERR_HSM;
1809 		active_ehi->action |= ATA_EH_RESET;
1810 		ata_ehi_push_desc(active_ehi,
1811 				  "unknown FIS %08x %08x %08x %08x" ,
1812 				  unk[0], unk[1], unk[2], unk[3]);
1813 	}
1814 
1815 	if (sata_pmp_attached(ap) && (irq_stat & PORT_IRQ_BAD_PMP)) {
1816 		active_ehi->err_mask |= AC_ERR_HSM;
1817 		active_ehi->action |= ATA_EH_RESET;
1818 		ata_ehi_push_desc(active_ehi, "incorrect PMP");
1819 	}
1820 
1821 	if (irq_stat & (PORT_IRQ_HBUS_ERR | PORT_IRQ_HBUS_DATA_ERR)) {
1822 		host_ehi->err_mask |= AC_ERR_HOST_BUS;
1823 		host_ehi->action |= ATA_EH_RESET;
1824 		ata_ehi_push_desc(host_ehi, "host bus error");
1825 	}
1826 
1827 	if (irq_stat & PORT_IRQ_IF_ERR) {
1828 		if (fbs_need_dec)
1829 			active_ehi->err_mask |= AC_ERR_DEV;
1830 		else {
1831 			host_ehi->err_mask |= AC_ERR_ATA_BUS;
1832 			host_ehi->action |= ATA_EH_RESET;
1833 		}
1834 
1835 		ata_ehi_push_desc(host_ehi, "interface fatal error");
1836 	}
1837 
1838 	if (irq_stat & (PORT_IRQ_CONNECT | PORT_IRQ_PHYRDY)) {
1839 		ata_ehi_hotplugged(host_ehi);
1840 		ata_ehi_push_desc(host_ehi, "%s",
1841 			irq_stat & PORT_IRQ_CONNECT ?
1842 			"connection status changed" : "PHY RDY changed");
1843 	}
1844 
1845 	/* okay, let's hand over to EH */
1846 
1847 	if (irq_stat & PORT_IRQ_FREEZE)
1848 		ata_port_freeze(ap);
1849 	else if (fbs_need_dec) {
1850 		ata_link_abort(link);
1851 		ahci_fbs_dec_intr(ap);
1852 	} else
1853 		ata_port_abort(ap);
1854 }
1855 
1856 static void ahci_qc_complete(struct ata_port *ap, void __iomem *port_mmio)
1857 {
1858 	struct ata_eh_info *ehi = &ap->link.eh_info;
1859 	struct ahci_port_priv *pp = ap->private_data;
1860 	u32 qc_active = 0;
1861 	int rc;
1862 
1863 	/*
1864 	 * pp->active_link is not reliable once FBS is enabled, both
1865 	 * PORT_SCR_ACT and PORT_CMD_ISSUE should be checked because
1866 	 * NCQ and non-NCQ commands may be in flight at the same time.
1867 	 */
1868 	if (pp->fbs_enabled) {
1869 		if (ap->qc_active) {
1870 			qc_active = readl(port_mmio + PORT_SCR_ACT);
1871 			qc_active |= readl(port_mmio + PORT_CMD_ISSUE);
1872 		}
1873 	} else {
1874 		/* pp->active_link is valid iff any command is in flight */
1875 		if (ap->qc_active && pp->active_link->sactive)
1876 			qc_active = readl(port_mmio + PORT_SCR_ACT);
1877 		else
1878 			qc_active = readl(port_mmio + PORT_CMD_ISSUE);
1879 	}
1880 
1881 	rc = ata_qc_complete_multiple(ap, qc_active);
1882 	if (unlikely(rc < 0 && !(ap->pflags & ATA_PFLAG_RESETTING))) {
1883 		ehi->err_mask |= AC_ERR_HSM;
1884 		ehi->action |= ATA_EH_RESET;
1885 		ata_port_freeze(ap);
1886 	}
1887 }
1888 
1889 static void ahci_handle_port_interrupt(struct ata_port *ap,
1890 				       void __iomem *port_mmio, u32 status)
1891 {
1892 	struct ahci_port_priv *pp = ap->private_data;
1893 	struct ahci_host_priv *hpriv = ap->host->private_data;
1894 
1895 	/* ignore BAD_PMP while resetting */
1896 	if (unlikely(ap->pflags & ATA_PFLAG_RESETTING))
1897 		status &= ~PORT_IRQ_BAD_PMP;
1898 
1899 	if (sata_lpm_ignore_phy_events(&ap->link)) {
1900 		status &= ~PORT_IRQ_PHYRDY;
1901 		ahci_scr_write(&ap->link, SCR_ERROR, SERR_PHYRDY_CHG);
1902 	}
1903 
1904 	if (unlikely(status & PORT_IRQ_ERROR)) {
1905 		/*
1906 		 * Before getting the error notification, we may have
1907 		 * received SDB FISes notifying successful completions.
1908 		 * Handle these first and then handle the error.
1909 		 */
1910 		ahci_qc_complete(ap, port_mmio);
1911 		ahci_error_intr(ap, status);
1912 		return;
1913 	}
1914 
1915 	if (status & PORT_IRQ_SDB_FIS) {
1916 		/* If SNotification is available, leave notification
1917 		 * handling to sata_async_notification().  If not,
1918 		 * emulate it by snooping SDB FIS RX area.
1919 		 *
1920 		 * Snooping FIS RX area is probably cheaper than
1921 		 * poking SNotification but some constrollers which
1922 		 * implement SNotification, ICH9 for example, don't
1923 		 * store AN SDB FIS into receive area.
1924 		 */
1925 		if (hpriv->cap & HOST_CAP_SNTF)
1926 			sata_async_notification(ap);
1927 		else {
1928 			/* If the 'N' bit in word 0 of the FIS is set,
1929 			 * we just received asynchronous notification.
1930 			 * Tell libata about it.
1931 			 *
1932 			 * Lack of SNotification should not appear in
1933 			 * ahci 1.2, so the workaround is unnecessary
1934 			 * when FBS is enabled.
1935 			 */
1936 			if (pp->fbs_enabled)
1937 				WARN_ON_ONCE(1);
1938 			else {
1939 				const __le32 *f = pp->rx_fis + RX_FIS_SDB;
1940 				u32 f0 = le32_to_cpu(f[0]);
1941 				if (f0 & (1 << 15))
1942 					sata_async_notification(ap);
1943 			}
1944 		}
1945 	}
1946 
1947 	/* Handle completed commands */
1948 	ahci_qc_complete(ap, port_mmio);
1949 }
1950 
1951 static void ahci_port_intr(struct ata_port *ap)
1952 {
1953 	void __iomem *port_mmio = ahci_port_base(ap);
1954 	u32 status;
1955 
1956 	status = readl(port_mmio + PORT_IRQ_STAT);
1957 	writel(status, port_mmio + PORT_IRQ_STAT);
1958 
1959 	ahci_handle_port_interrupt(ap, port_mmio, status);
1960 }
1961 
1962 static irqreturn_t ahci_multi_irqs_intr_hard(int irq, void *dev_instance)
1963 {
1964 	struct ata_port *ap = dev_instance;
1965 	void __iomem *port_mmio = ahci_port_base(ap);
1966 	u32 status;
1967 
1968 	status = readl(port_mmio + PORT_IRQ_STAT);
1969 	writel(status, port_mmio + PORT_IRQ_STAT);
1970 
1971 	spin_lock(ap->lock);
1972 	ahci_handle_port_interrupt(ap, port_mmio, status);
1973 	spin_unlock(ap->lock);
1974 
1975 	return IRQ_HANDLED;
1976 }
1977 
1978 u32 ahci_handle_port_intr(struct ata_host *host, u32 irq_masked)
1979 {
1980 	unsigned int i, handled = 0;
1981 
1982 	for (i = 0; i < host->n_ports; i++) {
1983 		struct ata_port *ap;
1984 
1985 		if (!(irq_masked & (1 << i)))
1986 			continue;
1987 
1988 		ap = host->ports[i];
1989 		if (ap) {
1990 			ahci_port_intr(ap);
1991 		} else {
1992 			if (ata_ratelimit())
1993 				dev_warn(host->dev,
1994 					 "interrupt on disabled port %u\n", i);
1995 		}
1996 
1997 		handled = 1;
1998 	}
1999 
2000 	return handled;
2001 }
2002 EXPORT_SYMBOL_GPL(ahci_handle_port_intr);
2003 
2004 static irqreturn_t ahci_single_level_irq_intr(int irq, void *dev_instance)
2005 {
2006 	struct ata_host *host = dev_instance;
2007 	struct ahci_host_priv *hpriv;
2008 	unsigned int rc = 0;
2009 	void __iomem *mmio;
2010 	u32 irq_stat, irq_masked;
2011 
2012 	hpriv = host->private_data;
2013 	mmio = hpriv->mmio;
2014 
2015 	/* sigh.  0xffffffff is a valid return from h/w */
2016 	irq_stat = readl(mmio + HOST_IRQ_STAT);
2017 	if (!irq_stat)
2018 		return IRQ_NONE;
2019 
2020 	irq_masked = irq_stat & hpriv->port_map;
2021 
2022 	spin_lock(&host->lock);
2023 
2024 	rc = ahci_handle_port_intr(host, irq_masked);
2025 
2026 	/* HOST_IRQ_STAT behaves as level triggered latch meaning that
2027 	 * it should be cleared after all the port events are cleared;
2028 	 * otherwise, it will raise a spurious interrupt after each
2029 	 * valid one.  Please read section 10.6.2 of ahci 1.1 for more
2030 	 * information.
2031 	 *
2032 	 * Also, use the unmasked value to clear interrupt as spurious
2033 	 * pending event on a dummy port might cause screaming IRQ.
2034 	 */
2035 	writel(irq_stat, mmio + HOST_IRQ_STAT);
2036 
2037 	spin_unlock(&host->lock);
2038 
2039 	return IRQ_RETVAL(rc);
2040 }
2041 
2042 unsigned int ahci_qc_issue(struct ata_queued_cmd *qc)
2043 {
2044 	struct ata_port *ap = qc->ap;
2045 	void __iomem *port_mmio = ahci_port_base(ap);
2046 	struct ahci_port_priv *pp = ap->private_data;
2047 
2048 	/* Keep track of the currently active link.  It will be used
2049 	 * in completion path to determine whether NCQ phase is in
2050 	 * progress.
2051 	 */
2052 	pp->active_link = qc->dev->link;
2053 
2054 	if (ata_is_ncq(qc->tf.protocol))
2055 		writel(1 << qc->hw_tag, port_mmio + PORT_SCR_ACT);
2056 
2057 	if (pp->fbs_enabled && pp->fbs_last_dev != qc->dev->link->pmp) {
2058 		u32 fbs = readl(port_mmio + PORT_FBS);
2059 		fbs &= ~(PORT_FBS_DEV_MASK | PORT_FBS_DEC);
2060 		fbs |= qc->dev->link->pmp << PORT_FBS_DEV_OFFSET;
2061 		writel(fbs, port_mmio + PORT_FBS);
2062 		pp->fbs_last_dev = qc->dev->link->pmp;
2063 	}
2064 
2065 	writel(1 << qc->hw_tag, port_mmio + PORT_CMD_ISSUE);
2066 
2067 	ahci_sw_activity(qc->dev->link);
2068 
2069 	return 0;
2070 }
2071 EXPORT_SYMBOL_GPL(ahci_qc_issue);
2072 
2073 static void ahci_qc_fill_rtf(struct ata_queued_cmd *qc)
2074 {
2075 	struct ahci_port_priv *pp = qc->ap->private_data;
2076 	u8 *rx_fis = pp->rx_fis;
2077 
2078 	/*
2079 	 * rtf may already be filled (e.g. for successful NCQ commands).
2080 	 * If that is the case, we have nothing to do.
2081 	 */
2082 	if (qc->flags & ATA_QCFLAG_RTF_FILLED)
2083 		return;
2084 
2085 	if (pp->fbs_enabled)
2086 		rx_fis += qc->dev->link->pmp * AHCI_RX_FIS_SZ;
2087 
2088 	/*
2089 	 * After a successful execution of an ATA PIO data-in command,
2090 	 * the device doesn't send D2H Reg FIS to update the TF and
2091 	 * the host should take TF and E_Status from the preceding PIO
2092 	 * Setup FIS.
2093 	 */
2094 	if (qc->tf.protocol == ATA_PROT_PIO && qc->dma_dir == DMA_FROM_DEVICE &&
2095 	    !(qc->flags & ATA_QCFLAG_EH)) {
2096 		ata_tf_from_fis(rx_fis + RX_FIS_PIO_SETUP, &qc->result_tf);
2097 		qc->result_tf.status = (rx_fis + RX_FIS_PIO_SETUP)[15];
2098 		qc->flags |= ATA_QCFLAG_RTF_FILLED;
2099 		return;
2100 	}
2101 
2102 	/*
2103 	 * For NCQ commands, we never get a D2H FIS, so reading the D2H Register
2104 	 * FIS area of the Received FIS Structure (which contains a copy of the
2105 	 * last D2H FIS received) will contain an outdated status code.
2106 	 * For NCQ commands, we instead get a SDB FIS, so read the SDB FIS area
2107 	 * instead. However, the SDB FIS does not contain the LBA, so we can't
2108 	 * use the ata_tf_from_fis() helper.
2109 	 */
2110 	if (ata_is_ncq(qc->tf.protocol)) {
2111 		const u8 *fis = rx_fis + RX_FIS_SDB;
2112 
2113 		/*
2114 		 * Successful NCQ commands have been filled already.
2115 		 * A failed NCQ command will read the status here.
2116 		 * (Note that a failed NCQ command will get a more specific
2117 		 * error when reading the NCQ Command Error log.)
2118 		 */
2119 		qc->result_tf.status = fis[2];
2120 		qc->result_tf.error = fis[3];
2121 		qc->flags |= ATA_QCFLAG_RTF_FILLED;
2122 		return;
2123 	}
2124 
2125 	ata_tf_from_fis(rx_fis + RX_FIS_D2H_REG, &qc->result_tf);
2126 	qc->flags |= ATA_QCFLAG_RTF_FILLED;
2127 }
2128 
2129 static void ahci_qc_ncq_fill_rtf(struct ata_port *ap, u64 done_mask)
2130 {
2131 	struct ahci_port_priv *pp = ap->private_data;
2132 	const u8 *fis;
2133 
2134 	/* No outstanding commands. */
2135 	if (!ap->qc_active)
2136 		return;
2137 
2138 	/*
2139 	 * FBS not enabled, so read status and error once, since they are shared
2140 	 * for all QCs.
2141 	 */
2142 	if (!pp->fbs_enabled) {
2143 		u8 status, error;
2144 
2145 		/* No outstanding NCQ commands. */
2146 		if (!pp->active_link->sactive)
2147 			return;
2148 
2149 		fis = pp->rx_fis + RX_FIS_SDB;
2150 		status = fis[2];
2151 		error = fis[3];
2152 
2153 		while (done_mask) {
2154 			struct ata_queued_cmd *qc;
2155 			unsigned int tag = __ffs64(done_mask);
2156 
2157 			qc = ata_qc_from_tag(ap, tag);
2158 			if (qc && ata_is_ncq(qc->tf.protocol)) {
2159 				qc->result_tf.status = status;
2160 				qc->result_tf.error = error;
2161 				qc->flags |= ATA_QCFLAG_RTF_FILLED;
2162 			}
2163 			done_mask &= ~(1ULL << tag);
2164 		}
2165 
2166 		return;
2167 	}
2168 
2169 	/*
2170 	 * FBS enabled, so read the status and error for each QC, since the QCs
2171 	 * can belong to different PMP links. (Each PMP link has its own FIS
2172 	 * Receive Area.)
2173 	 */
2174 	while (done_mask) {
2175 		struct ata_queued_cmd *qc;
2176 		unsigned int tag = __ffs64(done_mask);
2177 
2178 		qc = ata_qc_from_tag(ap, tag);
2179 		if (qc && ata_is_ncq(qc->tf.protocol)) {
2180 			fis = pp->rx_fis;
2181 			fis += qc->dev->link->pmp * AHCI_RX_FIS_SZ;
2182 			fis += RX_FIS_SDB;
2183 			qc->result_tf.status = fis[2];
2184 			qc->result_tf.error = fis[3];
2185 			qc->flags |= ATA_QCFLAG_RTF_FILLED;
2186 		}
2187 		done_mask &= ~(1ULL << tag);
2188 	}
2189 }
2190 
2191 static void ahci_freeze(struct ata_port *ap)
2192 {
2193 	void __iomem *port_mmio = ahci_port_base(ap);
2194 
2195 	/* turn IRQ off */
2196 	writel(0, port_mmio + PORT_IRQ_MASK);
2197 }
2198 
2199 static void ahci_thaw(struct ata_port *ap)
2200 {
2201 	struct ahci_host_priv *hpriv = ap->host->private_data;
2202 	void __iomem *mmio = hpriv->mmio;
2203 	void __iomem *port_mmio = ahci_port_base(ap);
2204 	u32 tmp;
2205 	struct ahci_port_priv *pp = ap->private_data;
2206 
2207 	/* clear IRQ */
2208 	tmp = readl(port_mmio + PORT_IRQ_STAT);
2209 	writel(tmp, port_mmio + PORT_IRQ_STAT);
2210 	writel(1 << ap->port_no, mmio + HOST_IRQ_STAT);
2211 
2212 	/* turn IRQ back on */
2213 	writel(pp->intr_mask, port_mmio + PORT_IRQ_MASK);
2214 }
2215 
2216 void ahci_error_handler(struct ata_port *ap)
2217 {
2218 	struct ahci_host_priv *hpriv = ap->host->private_data;
2219 
2220 	if (!ata_port_is_frozen(ap)) {
2221 		/* restart engine */
2222 		hpriv->stop_engine(ap);
2223 		hpriv->start_engine(ap);
2224 	}
2225 
2226 	sata_pmp_error_handler(ap);
2227 
2228 	if (!ata_dev_enabled(ap->link.device))
2229 		hpriv->stop_engine(ap);
2230 }
2231 EXPORT_SYMBOL_GPL(ahci_error_handler);
2232 
2233 static void ahci_post_internal_cmd(struct ata_queued_cmd *qc)
2234 {
2235 	struct ata_port *ap = qc->ap;
2236 
2237 	/* make DMA engine forget about the failed command */
2238 	if (qc->flags & ATA_QCFLAG_EH)
2239 		ahci_kick_engine(ap);
2240 }
2241 
2242 static void ahci_set_aggressive_devslp(struct ata_port *ap, bool sleep)
2243 {
2244 	struct ahci_host_priv *hpriv = ap->host->private_data;
2245 	void __iomem *port_mmio = ahci_port_base(ap);
2246 	struct ata_device *dev = ap->link.device;
2247 	u32 devslp, dm, dito, mdat, deto, dito_conf;
2248 	int rc;
2249 	unsigned int err_mask;
2250 
2251 	devslp = readl(port_mmio + PORT_DEVSLP);
2252 	if (!(devslp & PORT_DEVSLP_DSP)) {
2253 		dev_info(ap->host->dev, "port does not support device sleep\n");
2254 		return;
2255 	}
2256 
2257 	/* disable device sleep */
2258 	if (!sleep) {
2259 		if (devslp & PORT_DEVSLP_ADSE) {
2260 			writel(devslp & ~PORT_DEVSLP_ADSE,
2261 			       port_mmio + PORT_DEVSLP);
2262 			err_mask = ata_dev_set_feature(dev,
2263 						       SETFEATURES_SATA_DISABLE,
2264 						       SATA_DEVSLP);
2265 			if (err_mask && err_mask != AC_ERR_DEV)
2266 				ata_dev_warn(dev, "failed to disable DEVSLP\n");
2267 		}
2268 		return;
2269 	}
2270 
2271 	dm = (devslp & PORT_DEVSLP_DM_MASK) >> PORT_DEVSLP_DM_OFFSET;
2272 	dito = devslp_idle_timeout / (dm + 1);
2273 	if (dito > 0x3ff)
2274 		dito = 0x3ff;
2275 
2276 	dito_conf = (devslp >> PORT_DEVSLP_DITO_OFFSET) & 0x3FF;
2277 
2278 	/* device sleep was already enabled and same dito */
2279 	if ((devslp & PORT_DEVSLP_ADSE) && (dito_conf == dito))
2280 		return;
2281 
2282 	/* set DITO, MDAT, DETO and enable DevSlp, need to stop engine first */
2283 	rc = hpriv->stop_engine(ap);
2284 	if (rc)
2285 		return;
2286 
2287 	/* Use the nominal value 10 ms if the read MDAT is zero,
2288 	 * the nominal value of DETO is 20 ms.
2289 	 */
2290 	if (dev->devslp_timing[ATA_LOG_DEVSLP_VALID] &
2291 	    ATA_LOG_DEVSLP_VALID_MASK) {
2292 		mdat = dev->devslp_timing[ATA_LOG_DEVSLP_MDAT] &
2293 		       ATA_LOG_DEVSLP_MDAT_MASK;
2294 		if (!mdat)
2295 			mdat = 10;
2296 		deto = dev->devslp_timing[ATA_LOG_DEVSLP_DETO];
2297 		if (!deto)
2298 			deto = 20;
2299 	} else {
2300 		mdat = 10;
2301 		deto = 20;
2302 	}
2303 
2304 	/* Make dito, mdat, deto bits to 0s */
2305 	devslp &= ~GENMASK_ULL(24, 2);
2306 	devslp |= ((dito << PORT_DEVSLP_DITO_OFFSET) |
2307 		   (mdat << PORT_DEVSLP_MDAT_OFFSET) |
2308 		   (deto << PORT_DEVSLP_DETO_OFFSET) |
2309 		   PORT_DEVSLP_ADSE);
2310 	writel(devslp, port_mmio + PORT_DEVSLP);
2311 
2312 	hpriv->start_engine(ap);
2313 
2314 	/* enable device sleep feature for the drive */
2315 	err_mask = ata_dev_set_feature(dev,
2316 				       SETFEATURES_SATA_ENABLE,
2317 				       SATA_DEVSLP);
2318 	if (err_mask && err_mask != AC_ERR_DEV)
2319 		ata_dev_warn(dev, "failed to enable DEVSLP\n");
2320 }
2321 
2322 static void ahci_enable_fbs(struct ata_port *ap)
2323 {
2324 	struct ahci_host_priv *hpriv = ap->host->private_data;
2325 	struct ahci_port_priv *pp = ap->private_data;
2326 	void __iomem *port_mmio = ahci_port_base(ap);
2327 	u32 fbs;
2328 	int rc;
2329 
2330 	if (!pp->fbs_supported)
2331 		return;
2332 
2333 	fbs = readl(port_mmio + PORT_FBS);
2334 	if (fbs & PORT_FBS_EN) {
2335 		pp->fbs_enabled = true;
2336 		pp->fbs_last_dev = -1; /* initialization */
2337 		return;
2338 	}
2339 
2340 	rc = hpriv->stop_engine(ap);
2341 	if (rc)
2342 		return;
2343 
2344 	writel(fbs | PORT_FBS_EN, port_mmio + PORT_FBS);
2345 	fbs = readl(port_mmio + PORT_FBS);
2346 	if (fbs & PORT_FBS_EN) {
2347 		dev_info(ap->host->dev, "FBS is enabled\n");
2348 		pp->fbs_enabled = true;
2349 		pp->fbs_last_dev = -1; /* initialization */
2350 	} else
2351 		dev_err(ap->host->dev, "Failed to enable FBS\n");
2352 
2353 	hpriv->start_engine(ap);
2354 }
2355 
2356 static void ahci_disable_fbs(struct ata_port *ap)
2357 {
2358 	struct ahci_host_priv *hpriv = ap->host->private_data;
2359 	struct ahci_port_priv *pp = ap->private_data;
2360 	void __iomem *port_mmio = ahci_port_base(ap);
2361 	u32 fbs;
2362 	int rc;
2363 
2364 	if (!pp->fbs_supported)
2365 		return;
2366 
2367 	fbs = readl(port_mmio + PORT_FBS);
2368 	if ((fbs & PORT_FBS_EN) == 0) {
2369 		pp->fbs_enabled = false;
2370 		return;
2371 	}
2372 
2373 	rc = hpriv->stop_engine(ap);
2374 	if (rc)
2375 		return;
2376 
2377 	writel(fbs & ~PORT_FBS_EN, port_mmio + PORT_FBS);
2378 	fbs = readl(port_mmio + PORT_FBS);
2379 	if (fbs & PORT_FBS_EN)
2380 		dev_err(ap->host->dev, "Failed to disable FBS\n");
2381 	else {
2382 		dev_info(ap->host->dev, "FBS is disabled\n");
2383 		pp->fbs_enabled = false;
2384 	}
2385 
2386 	hpriv->start_engine(ap);
2387 }
2388 
2389 static void ahci_pmp_attach(struct ata_port *ap)
2390 {
2391 	void __iomem *port_mmio = ahci_port_base(ap);
2392 	struct ahci_port_priv *pp = ap->private_data;
2393 	u32 cmd;
2394 
2395 	cmd = readl(port_mmio + PORT_CMD);
2396 	cmd |= PORT_CMD_PMP;
2397 	writel(cmd, port_mmio + PORT_CMD);
2398 
2399 	ahci_enable_fbs(ap);
2400 
2401 	pp->intr_mask |= PORT_IRQ_BAD_PMP;
2402 
2403 	/*
2404 	 * We must not change the port interrupt mask register if the
2405 	 * port is marked frozen, the value in pp->intr_mask will be
2406 	 * restored later when the port is thawed.
2407 	 *
2408 	 * Note that during initialization, the port is marked as
2409 	 * frozen since the irq handler is not yet registered.
2410 	 */
2411 	if (!ata_port_is_frozen(ap))
2412 		writel(pp->intr_mask, port_mmio + PORT_IRQ_MASK);
2413 }
2414 
2415 static void ahci_pmp_detach(struct ata_port *ap)
2416 {
2417 	void __iomem *port_mmio = ahci_port_base(ap);
2418 	struct ahci_port_priv *pp = ap->private_data;
2419 	u32 cmd;
2420 
2421 	ahci_disable_fbs(ap);
2422 
2423 	cmd = readl(port_mmio + PORT_CMD);
2424 	cmd &= ~PORT_CMD_PMP;
2425 	writel(cmd, port_mmio + PORT_CMD);
2426 
2427 	pp->intr_mask &= ~PORT_IRQ_BAD_PMP;
2428 
2429 	/* see comment above in ahci_pmp_attach() */
2430 	if (!ata_port_is_frozen(ap))
2431 		writel(pp->intr_mask, port_mmio + PORT_IRQ_MASK);
2432 }
2433 
2434 int ahci_port_resume(struct ata_port *ap)
2435 {
2436 	ahci_rpm_get_port(ap);
2437 
2438 	ahci_power_up(ap);
2439 	ahci_start_port(ap);
2440 
2441 	if (sata_pmp_attached(ap))
2442 		ahci_pmp_attach(ap);
2443 	else
2444 		ahci_pmp_detach(ap);
2445 
2446 	return 0;
2447 }
2448 EXPORT_SYMBOL_GPL(ahci_port_resume);
2449 
2450 #ifdef CONFIG_PM
2451 static void ahci_handle_s2idle(struct ata_port *ap)
2452 {
2453 	void __iomem *port_mmio = ahci_port_base(ap);
2454 	u32 devslp;
2455 
2456 	if (pm_suspend_via_firmware())
2457 		return;
2458 	devslp = readl(port_mmio + PORT_DEVSLP);
2459 	if ((devslp & PORT_DEVSLP_ADSE))
2460 		ata_msleep(ap, devslp_idle_timeout);
2461 }
2462 
2463 static int ahci_port_suspend(struct ata_port *ap, pm_message_t mesg)
2464 {
2465 	const char *emsg = NULL;
2466 	int rc;
2467 
2468 	rc = ahci_deinit_port(ap, &emsg);
2469 	if (rc == 0)
2470 		ahci_power_down(ap);
2471 	else {
2472 		ata_port_err(ap, "%s (%d)\n", emsg, rc);
2473 		ata_port_freeze(ap);
2474 	}
2475 
2476 	if (acpi_storage_d3(ap->host->dev))
2477 		ahci_handle_s2idle(ap);
2478 
2479 	ahci_rpm_put_port(ap);
2480 	return rc;
2481 }
2482 #endif
2483 
2484 static int ahci_port_start(struct ata_port *ap)
2485 {
2486 	struct ahci_host_priv *hpriv = ap->host->private_data;
2487 	struct device *dev = ap->host->dev;
2488 	struct ahci_port_priv *pp;
2489 	void *mem;
2490 	dma_addr_t mem_dma;
2491 	size_t dma_sz, rx_fis_sz;
2492 
2493 	pp = devm_kzalloc(dev, sizeof(*pp), GFP_KERNEL);
2494 	if (!pp)
2495 		return -ENOMEM;
2496 
2497 	if (ap->host->n_ports > 1) {
2498 		pp->irq_desc = devm_kzalloc(dev, 8, GFP_KERNEL);
2499 		if (!pp->irq_desc) {
2500 			devm_kfree(dev, pp);
2501 			return -ENOMEM;
2502 		}
2503 		snprintf(pp->irq_desc, 8,
2504 			 "%s%d", dev_driver_string(dev), ap->port_no);
2505 	}
2506 
2507 	/* check FBS capability */
2508 	if ((hpriv->cap & HOST_CAP_FBS) && sata_pmp_supported(ap)) {
2509 		void __iomem *port_mmio = ahci_port_base(ap);
2510 		u32 cmd = readl(port_mmio + PORT_CMD);
2511 		if (cmd & PORT_CMD_FBSCP)
2512 			pp->fbs_supported = true;
2513 		else if (hpriv->flags & AHCI_HFLAG_YES_FBS) {
2514 			dev_info(dev, "port %d can do FBS, forcing FBSCP\n",
2515 				 ap->port_no);
2516 			pp->fbs_supported = true;
2517 		} else
2518 			dev_warn(dev, "port %d is not capable of FBS\n",
2519 				 ap->port_no);
2520 	}
2521 
2522 	if (pp->fbs_supported) {
2523 		dma_sz = AHCI_PORT_PRIV_FBS_DMA_SZ;
2524 		rx_fis_sz = AHCI_RX_FIS_SZ * 16;
2525 	} else {
2526 		dma_sz = AHCI_PORT_PRIV_DMA_SZ;
2527 		rx_fis_sz = AHCI_RX_FIS_SZ;
2528 	}
2529 
2530 	mem = dmam_alloc_coherent(dev, dma_sz, &mem_dma, GFP_KERNEL);
2531 	if (!mem)
2532 		return -ENOMEM;
2533 
2534 	/*
2535 	 * First item in chunk of DMA memory: 32-slot command table,
2536 	 * 32 bytes each in size
2537 	 */
2538 	pp->cmd_slot = mem;
2539 	pp->cmd_slot_dma = mem_dma;
2540 
2541 	mem += AHCI_CMD_SLOT_SZ;
2542 	mem_dma += AHCI_CMD_SLOT_SZ;
2543 
2544 	/*
2545 	 * Second item: Received-FIS area
2546 	 */
2547 	pp->rx_fis = mem;
2548 	pp->rx_fis_dma = mem_dma;
2549 
2550 	mem += rx_fis_sz;
2551 	mem_dma += rx_fis_sz;
2552 
2553 	/*
2554 	 * Third item: data area for storing a single command
2555 	 * and its scatter-gather table
2556 	 */
2557 	pp->cmd_tbl = mem;
2558 	pp->cmd_tbl_dma = mem_dma;
2559 
2560 	/*
2561 	 * Save off initial list of interrupts to be enabled.
2562 	 * This could be changed later
2563 	 */
2564 	pp->intr_mask = DEF_PORT_IRQ;
2565 
2566 	/*
2567 	 * Switch to per-port locking in case each port has its own MSI vector.
2568 	 */
2569 	if (hpriv->flags & AHCI_HFLAG_MULTI_MSI) {
2570 		spin_lock_init(&pp->lock);
2571 		ap->lock = &pp->lock;
2572 	}
2573 
2574 	ap->private_data = pp;
2575 
2576 	/* engage engines, captain */
2577 	return ahci_port_resume(ap);
2578 }
2579 
2580 static void ahci_port_stop(struct ata_port *ap)
2581 {
2582 	const char *emsg = NULL;
2583 	struct ahci_host_priv *hpriv = ap->host->private_data;
2584 	void __iomem *host_mmio = hpriv->mmio;
2585 	int rc;
2586 
2587 	/* de-initialize port */
2588 	rc = ahci_deinit_port(ap, &emsg);
2589 	if (rc)
2590 		ata_port_warn(ap, "%s (%d)\n", emsg, rc);
2591 
2592 	/*
2593 	 * Clear GHC.IS to prevent stuck INTx after disabling MSI and
2594 	 * re-enabling INTx.
2595 	 */
2596 	writel(1 << ap->port_no, host_mmio + HOST_IRQ_STAT);
2597 
2598 	ahci_rpm_put_port(ap);
2599 }
2600 
2601 void ahci_print_info(struct ata_host *host, const char *scc_s)
2602 {
2603 	struct ahci_host_priv *hpriv = host->private_data;
2604 	u32 vers, cap, cap2, impl, speed;
2605 	const char *speed_s;
2606 
2607 	vers = hpriv->version;
2608 	cap = hpriv->cap;
2609 	cap2 = hpriv->cap2;
2610 	impl = hpriv->port_map;
2611 
2612 	speed = (cap >> 20) & 0xf;
2613 	if (speed == 1)
2614 		speed_s = "1.5";
2615 	else if (speed == 2)
2616 		speed_s = "3";
2617 	else if (speed == 3)
2618 		speed_s = "6";
2619 	else
2620 		speed_s = "?";
2621 
2622 	dev_info(host->dev,
2623 		"AHCI vers %02x%02x.%02x%02x, "
2624 		"%u command slots, %s Gbps, %s mode\n"
2625 		,
2626 
2627 		(vers >> 24) & 0xff,
2628 		(vers >> 16) & 0xff,
2629 		(vers >> 8) & 0xff,
2630 		vers & 0xff,
2631 
2632 		((cap >> 8) & 0x1f) + 1,
2633 		speed_s,
2634 		scc_s);
2635 
2636 	dev_info(host->dev,
2637 		"%u/%u ports implemented (port mask 0x%x)\n"
2638 		,
2639 
2640 		hweight32(impl),
2641 		(cap & 0x1f) + 1,
2642 		impl);
2643 
2644 	dev_info(host->dev,
2645 		"flags: "
2646 		"%s%s%s%s%s%s%s"
2647 		"%s%s%s%s%s%s%s"
2648 		"%s%s%s%s%s%s%s"
2649 		"%s%s\n"
2650 		,
2651 
2652 		cap & HOST_CAP_64 ? "64bit " : "",
2653 		cap & HOST_CAP_NCQ ? "ncq " : "",
2654 		cap & HOST_CAP_SNTF ? "sntf " : "",
2655 		cap & HOST_CAP_MPS ? "ilck " : "",
2656 		cap & HOST_CAP_SSS ? "stag " : "",
2657 		cap & HOST_CAP_ALPM ? "pm " : "",
2658 		cap & HOST_CAP_LED ? "led " : "",
2659 		cap & HOST_CAP_CLO ? "clo " : "",
2660 		cap & HOST_CAP_ONLY ? "only " : "",
2661 		cap & HOST_CAP_PMP ? "pmp " : "",
2662 		cap & HOST_CAP_FBS ? "fbs " : "",
2663 		cap & HOST_CAP_PIO_MULTI ? "pio " : "",
2664 		cap & HOST_CAP_SSC ? "slum " : "",
2665 		cap & HOST_CAP_PART ? "part " : "",
2666 		cap & HOST_CAP_CCC ? "ccc " : "",
2667 		cap & HOST_CAP_EMS ? "ems " : "",
2668 		cap & HOST_CAP_SXS ? "sxs " : "",
2669 		cap2 & HOST_CAP2_DESO ? "deso " : "",
2670 		cap2 & HOST_CAP2_SADM ? "sadm " : "",
2671 		cap2 & HOST_CAP2_SDS ? "sds " : "",
2672 		cap2 & HOST_CAP2_APST ? "apst " : "",
2673 		cap2 & HOST_CAP2_NVMHCI ? "nvmp " : "",
2674 		cap2 & HOST_CAP2_BOH ? "boh " : ""
2675 		);
2676 }
2677 EXPORT_SYMBOL_GPL(ahci_print_info);
2678 
2679 void ahci_set_em_messages(struct ahci_host_priv *hpriv,
2680 			  struct ata_port_info *pi)
2681 {
2682 	u8 messages;
2683 	void __iomem *mmio = hpriv->mmio;
2684 	u32 em_loc = readl(mmio + HOST_EM_LOC);
2685 	u32 em_ctl = readl(mmio + HOST_EM_CTL);
2686 
2687 	if (!ahci_em_messages || !(hpriv->cap & HOST_CAP_EMS))
2688 		return;
2689 
2690 	messages = (em_ctl & EM_CTRL_MSG_TYPE) >> 16;
2691 
2692 	if (messages) {
2693 		/* store em_loc */
2694 		hpriv->em_loc = ((em_loc >> 16) * 4);
2695 		hpriv->em_buf_sz = ((em_loc & 0xff) * 4);
2696 		hpriv->em_msg_type = messages;
2697 		pi->flags |= ATA_FLAG_EM;
2698 		if (!(em_ctl & EM_CTL_ALHD))
2699 			pi->flags |= ATA_FLAG_SW_ACTIVITY;
2700 	}
2701 }
2702 EXPORT_SYMBOL_GPL(ahci_set_em_messages);
2703 
2704 static int ahci_host_activate_multi_irqs(struct ata_host *host,
2705 					 const struct scsi_host_template *sht)
2706 {
2707 	struct ahci_host_priv *hpriv = host->private_data;
2708 	int i, rc;
2709 
2710 	rc = ata_host_start(host);
2711 	if (rc)
2712 		return rc;
2713 	/*
2714 	 * Requests IRQs according to AHCI-1.1 when multiple MSIs were
2715 	 * allocated. That is one MSI per port, starting from @irq.
2716 	 */
2717 	for (i = 0; i < host->n_ports; i++) {
2718 		struct ahci_port_priv *pp = host->ports[i]->private_data;
2719 		int irq = hpriv->get_irq_vector(host, i);
2720 
2721 		/* Do not receive interrupts sent by dummy ports */
2722 		if (!pp) {
2723 			disable_irq(irq);
2724 			continue;
2725 		}
2726 
2727 		rc = devm_request_irq(host->dev, irq, ahci_multi_irqs_intr_hard,
2728 				0, pp->irq_desc, host->ports[i]);
2729 
2730 		if (rc)
2731 			return rc;
2732 		ata_port_desc_misc(host->ports[i], irq);
2733 	}
2734 
2735 	return ata_host_register(host, sht);
2736 }
2737 
2738 /**
2739  *	ahci_host_activate - start AHCI host, request IRQs and register it
2740  *	@host: target ATA host
2741  *	@sht: scsi_host_template to use when registering the host
2742  *
2743  *	LOCKING:
2744  *	Inherited from calling layer (may sleep).
2745  *
2746  *	RETURNS:
2747  *	0 on success, -errno otherwise.
2748  */
2749 int ahci_host_activate(struct ata_host *host, const struct scsi_host_template *sht)
2750 {
2751 	struct ahci_host_priv *hpriv = host->private_data;
2752 	int irq = hpriv->irq;
2753 	int rc;
2754 
2755 	if (hpriv->flags & AHCI_HFLAG_MULTI_MSI) {
2756 		if (hpriv->irq_handler &&
2757 		    hpriv->irq_handler != ahci_single_level_irq_intr)
2758 			dev_warn(host->dev,
2759 			         "both AHCI_HFLAG_MULTI_MSI flag set and custom irq handler implemented\n");
2760 		if (!hpriv->get_irq_vector) {
2761 			dev_err(host->dev,
2762 				"AHCI_HFLAG_MULTI_MSI requires ->get_irq_vector!\n");
2763 			return -EIO;
2764 		}
2765 
2766 		rc = ahci_host_activate_multi_irqs(host, sht);
2767 	} else {
2768 		rc = ata_host_activate(host, irq, hpriv->irq_handler,
2769 				       IRQF_SHARED, sht);
2770 	}
2771 
2772 
2773 	return rc;
2774 }
2775 EXPORT_SYMBOL_GPL(ahci_host_activate);
2776 
2777 MODULE_AUTHOR("Jeff Garzik");
2778 MODULE_DESCRIPTION("Common AHCI SATA low-level routines");
2779 MODULE_LICENSE("GPL");
2780