]> git.sur5r.net Git - u-boot/blob - drivers/block/ahci.c
rockchip: Add basic support for kylin board
[u-boot] / drivers / block / ahci.c
1 /*
2  * Copyright (C) Freescale Semiconductor, Inc. 2006.
3  * Author: Jason Jin<Jason.jin@freescale.com>
4  *         Zhang Wei<wei.zhang@freescale.com>
5  *
6  * SPDX-License-Identifier:     GPL-2.0+
7  *
8  * with the reference on libata and ahci drvier in kernel
9  */
10 #include <common.h>
11
12 #include <command.h>
13 #include <pci.h>
14 #include <asm/processor.h>
15 #include <asm/errno.h>
16 #include <asm/io.h>
17 #include <malloc.h>
18 #include <memalign.h>
19 #include <scsi.h>
20 #include <libata.h>
21 #include <linux/ctype.h>
22 #include <ahci.h>
23
24 static int ata_io_flush(u8 port);
25
26 struct ahci_probe_ent *probe_ent = NULL;
27 u16 *ataid[AHCI_MAX_PORTS];
28
29 #define writel_with_flush(a,b)  do { writel(a,b); readl(b); } while (0)
30
31 /*
32  * Some controllers limit number of blocks they can read/write at once.
33  * Contemporary SSD devices work much faster if the read/write size is aligned
34  * to a power of 2.  Let's set default to 128 and allowing to be overwritten if
35  * needed.
36  */
37 #ifndef MAX_SATA_BLOCKS_READ_WRITE
38 #define MAX_SATA_BLOCKS_READ_WRITE      0x80
39 #endif
40
41 /* Maximum timeouts for each event */
42 #define WAIT_MS_SPINUP  20000
43 #define WAIT_MS_DATAIO  10000
44 #define WAIT_MS_FLUSH   5000
45 #define WAIT_MS_LINKUP  200
46
47 static inline void __iomem *ahci_port_base(void __iomem *base, u32 port)
48 {
49         return base + 0x100 + (port * 0x80);
50 }
51
52
53 static void ahci_setup_port(struct ahci_ioports *port, void __iomem *base,
54                             unsigned int port_idx)
55 {
56         base = ahci_port_base(base, port_idx);
57
58         port->cmd_addr = base;
59         port->scr_addr = base + PORT_SCR;
60 }
61
62
63 #define msleep(a) udelay(a * 1000)
64
65 static void ahci_dcache_flush_range(unsigned long begin, unsigned long len)
66 {
67         const unsigned long start = begin;
68         const unsigned long end = start + len;
69
70         debug("%s: flush dcache: [%#lx, %#lx)\n", __func__, start, end);
71         flush_dcache_range(start, end);
72 }
73
74 /*
75  * SATA controller DMAs to physical RAM.  Ensure data from the
76  * controller is invalidated from dcache; next access comes from
77  * physical RAM.
78  */
79 static void ahci_dcache_invalidate_range(unsigned long begin, unsigned long len)
80 {
81         const unsigned long start = begin;
82         const unsigned long end = start + len;
83
84         debug("%s: invalidate dcache: [%#lx, %#lx)\n", __func__, start, end);
85         invalidate_dcache_range(start, end);
86 }
87
88 /*
89  * Ensure data for SATA controller is flushed out of dcache and
90  * written to physical memory.
91  */
92 static void ahci_dcache_flush_sata_cmd(struct ahci_ioports *pp)
93 {
94         ahci_dcache_flush_range((unsigned long)pp->cmd_slot,
95                                 AHCI_PORT_PRIV_DMA_SZ);
96 }
97
98 static int waiting_for_cmd_completed(void __iomem *offset,
99                                      int timeout_msec,
100                                      u32 sign)
101 {
102         int i;
103         u32 status;
104
105         for (i = 0; ((status = readl(offset)) & sign) && i < timeout_msec; i++)
106                 msleep(1);
107
108         return (i < timeout_msec) ? 0 : -1;
109 }
110
111 int __weak ahci_link_up(struct ahci_probe_ent *probe_ent, u8 port)
112 {
113         u32 tmp;
114         int j = 0;
115         void __iomem *port_mmio = probe_ent->port[port].port_mmio;
116
117         /*
118          * Bring up SATA link.
119          * SATA link bringup time is usually less than 1 ms; only very
120          * rarely has it taken between 1-2 ms. Never seen it above 2 ms.
121          */
122         while (j < WAIT_MS_LINKUP) {
123                 tmp = readl(port_mmio + PORT_SCR_STAT);
124                 tmp &= PORT_SCR_STAT_DET_MASK;
125                 if (tmp == PORT_SCR_STAT_DET_PHYRDY)
126                         return 0;
127                 udelay(1000);
128                 j++;
129         }
130         return 1;
131 }
132
133 #ifdef CONFIG_SUNXI_AHCI
134 /* The sunxi AHCI controller requires this undocumented setup */
135 static void sunxi_dma_init(void __iomem *port_mmio)
136 {
137         clrsetbits_le32(port_mmio + PORT_P0DMACR, 0x0000ff00, 0x00004400);
138 }
139 #endif
140
141 int ahci_reset(void __iomem *base)
142 {
143         int i = 1000;
144         u32 __iomem *host_ctl_reg = base + HOST_CTL;
145         u32 tmp = readl(host_ctl_reg); /* global controller reset */
146
147         if ((tmp & HOST_RESET) == 0)
148                 writel_with_flush(tmp | HOST_RESET, host_ctl_reg);
149
150         /*
151          * reset must complete within 1 second, or
152          * the hardware should be considered fried.
153          */
154         do {
155                 udelay(1000);
156                 tmp = readl(host_ctl_reg);
157                 i--;
158         } while ((i > 0) && (tmp & HOST_RESET));
159
160         if (i == 0) {
161                 printf("controller reset failed (0x%x)\n", tmp);
162                 return -1;
163         }
164
165         return 0;
166 }
167
168 static int ahci_host_init(struct ahci_probe_ent *probe_ent)
169 {
170 #ifndef CONFIG_SCSI_AHCI_PLAT
171         pci_dev_t pdev = probe_ent->dev;
172         u16 tmp16;
173         unsigned short vendor;
174 #endif
175         void __iomem *mmio = probe_ent->mmio_base;
176         u32 tmp, cap_save, cmd;
177         int i, j, ret;
178         void __iomem *port_mmio;
179         u32 port_map;
180
181         debug("ahci_host_init: start\n");
182
183         cap_save = readl(mmio + HOST_CAP);
184         cap_save &= ((1 << 28) | (1 << 17));
185         cap_save |= (1 << 27);  /* Staggered Spin-up. Not needed. */
186
187         ret = ahci_reset(probe_ent->mmio_base);
188         if (ret)
189                 return ret;
190
191         writel_with_flush(HOST_AHCI_EN, mmio + HOST_CTL);
192         writel(cap_save, mmio + HOST_CAP);
193         writel_with_flush(0xf, mmio + HOST_PORTS_IMPL);
194
195 #ifndef CONFIG_SCSI_AHCI_PLAT
196         pci_read_config_word(pdev, PCI_VENDOR_ID, &vendor);
197
198         if (vendor == PCI_VENDOR_ID_INTEL) {
199                 u16 tmp16;
200                 pci_read_config_word(pdev, 0x92, &tmp16);
201                 tmp16 |= 0xf;
202                 pci_write_config_word(pdev, 0x92, tmp16);
203         }
204 #endif
205         probe_ent->cap = readl(mmio + HOST_CAP);
206         probe_ent->port_map = readl(mmio + HOST_PORTS_IMPL);
207         port_map = probe_ent->port_map;
208         probe_ent->n_ports = (probe_ent->cap & 0x1f) + 1;
209
210         debug("cap 0x%x  port_map 0x%x  n_ports %d\n",
211               probe_ent->cap, probe_ent->port_map, probe_ent->n_ports);
212
213         if (probe_ent->n_ports > CONFIG_SYS_SCSI_MAX_SCSI_ID)
214                 probe_ent->n_ports = CONFIG_SYS_SCSI_MAX_SCSI_ID;
215
216         for (i = 0; i < probe_ent->n_ports; i++) {
217                 if (!(port_map & (1 << i)))
218                         continue;
219                 probe_ent->port[i].port_mmio = ahci_port_base(mmio, i);
220                 port_mmio = (u8 *) probe_ent->port[i].port_mmio;
221                 ahci_setup_port(&probe_ent->port[i], mmio, i);
222
223                 /* make sure port is not active */
224                 tmp = readl(port_mmio + PORT_CMD);
225                 if (tmp & (PORT_CMD_LIST_ON | PORT_CMD_FIS_ON |
226                            PORT_CMD_FIS_RX | PORT_CMD_START)) {
227                         debug("Port %d is active. Deactivating.\n", i);
228                         tmp &= ~(PORT_CMD_LIST_ON | PORT_CMD_FIS_ON |
229                                  PORT_CMD_FIS_RX | PORT_CMD_START);
230                         writel_with_flush(tmp, port_mmio + PORT_CMD);
231
232                         /* spec says 500 msecs for each bit, so
233                          * this is slightly incorrect.
234                          */
235                         msleep(500);
236                 }
237
238 #ifdef CONFIG_SUNXI_AHCI
239                 sunxi_dma_init(port_mmio);
240 #endif
241
242                 /* Add the spinup command to whatever mode bits may
243                  * already be on in the command register.
244                  */
245                 cmd = readl(port_mmio + PORT_CMD);
246                 cmd |= PORT_CMD_SPIN_UP;
247                 writel_with_flush(cmd, port_mmio + PORT_CMD);
248
249                 /* Bring up SATA link. */
250                 ret = ahci_link_up(probe_ent, i);
251                 if (ret) {
252                         printf("SATA link %d timeout.\n", i);
253                         continue;
254                 } else {
255                         debug("SATA link ok.\n");
256                 }
257
258                 /* Clear error status */
259                 tmp = readl(port_mmio + PORT_SCR_ERR);
260                 if (tmp)
261                         writel(tmp, port_mmio + PORT_SCR_ERR);
262
263                 debug("Spinning up device on SATA port %d... ", i);
264
265                 j = 0;
266                 while (j < WAIT_MS_SPINUP) {
267                         tmp = readl(port_mmio + PORT_TFDATA);
268                         if (!(tmp & (ATA_BUSY | ATA_DRQ)))
269                                 break;
270                         udelay(1000);
271                         tmp = readl(port_mmio + PORT_SCR_STAT);
272                         tmp &= PORT_SCR_STAT_DET_MASK;
273                         if (tmp == PORT_SCR_STAT_DET_PHYRDY)
274                                 break;
275                         j++;
276                 }
277
278                 tmp = readl(port_mmio + PORT_SCR_STAT) & PORT_SCR_STAT_DET_MASK;
279                 if (tmp == PORT_SCR_STAT_DET_COMINIT) {
280                         debug("SATA link %d down (COMINIT received), retrying...\n", i);
281                         i--;
282                         continue;
283                 }
284
285                 printf("Target spinup took %d ms.\n", j);
286                 if (j == WAIT_MS_SPINUP)
287                         debug("timeout.\n");
288                 else
289                         debug("ok.\n");
290
291                 tmp = readl(port_mmio + PORT_SCR_ERR);
292                 debug("PORT_SCR_ERR 0x%x\n", tmp);
293                 writel(tmp, port_mmio + PORT_SCR_ERR);
294
295                 /* ack any pending irq events for this port */
296                 tmp = readl(port_mmio + PORT_IRQ_STAT);
297                 debug("PORT_IRQ_STAT 0x%x\n", tmp);
298                 if (tmp)
299                         writel(tmp, port_mmio + PORT_IRQ_STAT);
300
301                 writel(1 << i, mmio + HOST_IRQ_STAT);
302
303                 /* register linkup ports */
304                 tmp = readl(port_mmio + PORT_SCR_STAT);
305                 debug("SATA port %d status: 0x%x\n", i, tmp);
306                 if ((tmp & PORT_SCR_STAT_DET_MASK) == PORT_SCR_STAT_DET_PHYRDY)
307                         probe_ent->link_port_map |= (0x01 << i);
308         }
309
310         tmp = readl(mmio + HOST_CTL);
311         debug("HOST_CTL 0x%x\n", tmp);
312         writel(tmp | HOST_IRQ_EN, mmio + HOST_CTL);
313         tmp = readl(mmio + HOST_CTL);
314         debug("HOST_CTL 0x%x\n", tmp);
315 #ifndef CONFIG_SCSI_AHCI_PLAT
316         pci_read_config_word(pdev, PCI_COMMAND, &tmp16);
317         tmp |= PCI_COMMAND_MASTER;
318         pci_write_config_word(pdev, PCI_COMMAND, tmp16);
319 #endif
320         return 0;
321 }
322
323
324 static void ahci_print_info(struct ahci_probe_ent *probe_ent)
325 {
326 #ifndef CONFIG_SCSI_AHCI_PLAT
327         pci_dev_t pdev = probe_ent->dev;
328         u16 cc;
329 #endif
330         void __iomem *mmio = probe_ent->mmio_base;
331         u32 vers, cap, cap2, impl, speed;
332         const char *speed_s;
333         const char *scc_s;
334
335         vers = readl(mmio + HOST_VERSION);
336         cap = probe_ent->cap;
337         cap2 = readl(mmio + HOST_CAP2);
338         impl = probe_ent->port_map;
339
340         speed = (cap >> 20) & 0xf;
341         if (speed == 1)
342                 speed_s = "1.5";
343         else if (speed == 2)
344                 speed_s = "3";
345         else if (speed == 3)
346                 speed_s = "6";
347         else
348                 speed_s = "?";
349
350 #ifdef CONFIG_SCSI_AHCI_PLAT
351         scc_s = "SATA";
352 #else
353         pci_read_config_word(pdev, 0x0a, &cc);
354         if (cc == 0x0101)
355                 scc_s = "IDE";
356         else if (cc == 0x0106)
357                 scc_s = "SATA";
358         else if (cc == 0x0104)
359                 scc_s = "RAID";
360         else
361                 scc_s = "unknown";
362 #endif
363         printf("AHCI %02x%02x.%02x%02x "
364                "%u slots %u ports %s Gbps 0x%x impl %s mode\n",
365                (vers >> 24) & 0xff,
366                (vers >> 16) & 0xff,
367                (vers >> 8) & 0xff,
368                vers & 0xff,
369                ((cap >> 8) & 0x1f) + 1, (cap & 0x1f) + 1, speed_s, impl, scc_s);
370
371         printf("flags: "
372                "%s%s%s%s%s%s%s"
373                "%s%s%s%s%s%s%s"
374                "%s%s%s%s%s%s\n",
375                cap & (1 << 31) ? "64bit " : "",
376                cap & (1 << 30) ? "ncq " : "",
377                cap & (1 << 28) ? "ilck " : "",
378                cap & (1 << 27) ? "stag " : "",
379                cap & (1 << 26) ? "pm " : "",
380                cap & (1 << 25) ? "led " : "",
381                cap & (1 << 24) ? "clo " : "",
382                cap & (1 << 19) ? "nz " : "",
383                cap & (1 << 18) ? "only " : "",
384                cap & (1 << 17) ? "pmp " : "",
385                cap & (1 << 16) ? "fbss " : "",
386                cap & (1 << 15) ? "pio " : "",
387                cap & (1 << 14) ? "slum " : "",
388                cap & (1 << 13) ? "part " : "",
389                cap & (1 << 7) ? "ccc " : "",
390                cap & (1 << 6) ? "ems " : "",
391                cap & (1 << 5) ? "sxs " : "",
392                cap2 & (1 << 2) ? "apst " : "",
393                cap2 & (1 << 1) ? "nvmp " : "",
394                cap2 & (1 << 0) ? "boh " : "");
395 }
396
397 #ifndef CONFIG_SCSI_AHCI_PLAT
398 static int ahci_init_one(pci_dev_t pdev)
399 {
400         u16 vendor;
401         int rc;
402
403         probe_ent = malloc(sizeof(struct ahci_probe_ent));
404         if (!probe_ent) {
405                 printf("%s: No memory for probe_ent\n", __func__);
406                 return -ENOMEM;
407         }
408
409         memset(probe_ent, 0, sizeof(struct ahci_probe_ent));
410         probe_ent->dev = pdev;
411
412         probe_ent->host_flags = ATA_FLAG_SATA
413                                 | ATA_FLAG_NO_LEGACY
414                                 | ATA_FLAG_MMIO
415                                 | ATA_FLAG_PIO_DMA
416                                 | ATA_FLAG_NO_ATAPI;
417         probe_ent->pio_mask = 0x1f;
418         probe_ent->udma_mask = 0x7f;    /*Fixme,assume to support UDMA6 */
419
420         probe_ent->mmio_base = pci_map_bar(pdev, PCI_BASE_ADDRESS_5,
421                                            PCI_REGION_MEM);
422         debug("ahci mmio_base=0x%p\n", probe_ent->mmio_base);
423
424         /* Take from kernel:
425          * JMicron-specific fixup:
426          * make sure we're in AHCI mode
427          */
428         pci_read_config_word(pdev, PCI_VENDOR_ID, &vendor);
429         if (vendor == 0x197b)
430                 pci_write_config_byte(pdev, 0x41, 0xa1);
431
432         /* initialize adapter */
433         rc = ahci_host_init(probe_ent);
434         if (rc)
435                 goto err_out;
436
437         ahci_print_info(probe_ent);
438
439         return 0;
440
441       err_out:
442         return rc;
443 }
444 #endif
445
446 #define MAX_DATA_BYTE_COUNT  (4*1024*1024)
447
448 static int ahci_fill_sg(u8 port, unsigned char *buf, int buf_len)
449 {
450         struct ahci_ioports *pp = &(probe_ent->port[port]);
451         struct ahci_sg *ahci_sg = pp->cmd_tbl_sg;
452         u32 sg_count;
453         int i;
454
455         sg_count = ((buf_len - 1) / MAX_DATA_BYTE_COUNT) + 1;
456         if (sg_count > AHCI_MAX_SG) {
457                 printf("Error:Too much sg!\n");
458                 return -1;
459         }
460
461         for (i = 0; i < sg_count; i++) {
462                 ahci_sg->addr =
463                     cpu_to_le32((unsigned long) buf + i * MAX_DATA_BYTE_COUNT);
464                 ahci_sg->addr_hi = 0;
465                 ahci_sg->flags_size = cpu_to_le32(0x3fffff &
466                                           (buf_len < MAX_DATA_BYTE_COUNT
467                                            ? (buf_len - 1)
468                                            : (MAX_DATA_BYTE_COUNT - 1)));
469                 ahci_sg++;
470                 buf_len -= MAX_DATA_BYTE_COUNT;
471         }
472
473         return sg_count;
474 }
475
476
477 static void ahci_fill_cmd_slot(struct ahci_ioports *pp, u32 opts)
478 {
479         pp->cmd_slot->opts = cpu_to_le32(opts);
480         pp->cmd_slot->status = 0;
481         pp->cmd_slot->tbl_addr = cpu_to_le32((u32)pp->cmd_tbl & 0xffffffff);
482 #ifdef CONFIG_PHYS_64BIT
483         pp->cmd_slot->tbl_addr_hi =
484             cpu_to_le32((u32)(((pp->cmd_tbl) >> 16) >> 16));
485 #endif
486 }
487
488 static int wait_spinup(void __iomem *port_mmio)
489 {
490         ulong start;
491         u32 tf_data;
492
493         start = get_timer(0);
494         do {
495                 tf_data = readl(port_mmio + PORT_TFDATA);
496                 if (!(tf_data & ATA_BUSY))
497                         return 0;
498         } while (get_timer(start) < WAIT_MS_SPINUP);
499
500         return -ETIMEDOUT;
501 }
502
503 static int ahci_port_start(u8 port)
504 {
505         struct ahci_ioports *pp = &(probe_ent->port[port]);
506         void __iomem *port_mmio = pp->port_mmio;
507         u32 port_status;
508         void __iomem *mem;
509
510         debug("Enter start port: %d\n", port);
511         port_status = readl(port_mmio + PORT_SCR_STAT);
512         debug("Port %d status: %x\n", port, port_status);
513         if ((port_status & 0xf) != 0x03) {
514                 printf("No Link on this port!\n");
515                 return -1;
516         }
517
518         mem = malloc(AHCI_PORT_PRIV_DMA_SZ + 2048);
519         if (!mem) {
520                 free(pp);
521                 printf("%s: No mem for table!\n", __func__);
522                 return -ENOMEM;
523         }
524
525         /* Aligned to 2048-bytes */
526         mem = memalign(2048, AHCI_PORT_PRIV_DMA_SZ);
527         memset(mem, 0, AHCI_PORT_PRIV_DMA_SZ);
528
529         /*
530          * First item in chunk of DMA memory: 32-slot command table,
531          * 32 bytes each in size
532          */
533         pp->cmd_slot =
534                 (struct ahci_cmd_hdr *)(uintptr_t)virt_to_phys((void *)mem);
535         debug("cmd_slot = %p\n", pp->cmd_slot);
536         mem += (AHCI_CMD_SLOT_SZ + 224);
537
538         /*
539          * Second item: Received-FIS area
540          */
541         pp->rx_fis = virt_to_phys((void *)mem);
542         mem += AHCI_RX_FIS_SZ;
543
544         /*
545          * Third item: data area for storing a single command
546          * and its scatter-gather table
547          */
548         pp->cmd_tbl = virt_to_phys((void *)mem);
549         debug("cmd_tbl_dma = %lx\n", pp->cmd_tbl);
550
551         mem += AHCI_CMD_TBL_HDR;
552         pp->cmd_tbl_sg =
553                         (struct ahci_sg *)(uintptr_t)virt_to_phys((void *)mem);
554
555         writel_with_flush((unsigned long)pp->cmd_slot,
556                           port_mmio + PORT_LST_ADDR);
557
558         writel_with_flush(pp->rx_fis, port_mmio + PORT_FIS_ADDR);
559
560 #ifdef CONFIG_SUNXI_AHCI
561         sunxi_dma_init(port_mmio);
562 #endif
563
564         writel_with_flush(PORT_CMD_ICC_ACTIVE | PORT_CMD_FIS_RX |
565                           PORT_CMD_POWER_ON | PORT_CMD_SPIN_UP |
566                           PORT_CMD_START, port_mmio + PORT_CMD);
567
568         debug("Exit start port %d\n", port);
569
570         /*
571          * Make sure interface is not busy based on error and status
572          * information from task file data register before proceeding
573          */
574         return wait_spinup(port_mmio);
575 }
576
577
578 static int ahci_device_data_io(u8 port, u8 *fis, int fis_len, u8 *buf,
579                                 int buf_len, u8 is_write)
580 {
581
582         struct ahci_ioports *pp = &(probe_ent->port[port]);
583         void __iomem *port_mmio = pp->port_mmio;
584         u32 opts;
585         u32 port_status;
586         int sg_count;
587
588         debug("Enter %s: for port %d\n", __func__, port);
589
590         if (port > probe_ent->n_ports) {
591                 printf("Invalid port number %d\n", port);
592                 return -1;
593         }
594
595         port_status = readl(port_mmio + PORT_SCR_STAT);
596         if ((port_status & 0xf) != 0x03) {
597                 debug("No Link on port %d!\n", port);
598                 return -1;
599         }
600
601         memcpy((unsigned char *)pp->cmd_tbl, fis, fis_len);
602
603         sg_count = ahci_fill_sg(port, buf, buf_len);
604         opts = (fis_len >> 2) | (sg_count << 16) | (is_write << 6);
605         ahci_fill_cmd_slot(pp, opts);
606
607         ahci_dcache_flush_sata_cmd(pp);
608         ahci_dcache_flush_range((unsigned long)buf, (unsigned long)buf_len);
609
610         writel_with_flush(1, port_mmio + PORT_CMD_ISSUE);
611
612         if (waiting_for_cmd_completed(port_mmio + PORT_CMD_ISSUE,
613                                 WAIT_MS_DATAIO, 0x1)) {
614                 printf("timeout exit!\n");
615                 return -1;
616         }
617
618         ahci_dcache_invalidate_range((unsigned long)buf,
619                                      (unsigned long)buf_len);
620         debug("%s: %d byte transferred.\n", __func__, pp->cmd_slot->status);
621
622         return 0;
623 }
624
625
626 static char *ata_id_strcpy(u16 *target, u16 *src, int len)
627 {
628         int i;
629         for (i = 0; i < len / 2; i++)
630                 target[i] = swab16(src[i]);
631         return (char *)target;
632 }
633
634 /*
635  * SCSI INQUIRY command operation.
636  */
637 static int ata_scsiop_inquiry(ccb *pccb)
638 {
639         static const u8 hdr[] = {
640                 0,
641                 0,
642                 0x5,            /* claim SPC-3 version compatibility */
643                 2,
644                 95 - 4,
645         };
646         u8 fis[20];
647         u16 *idbuf;
648         ALLOC_CACHE_ALIGN_BUFFER(u16, tmpid, ATA_ID_WORDS);
649         u8 port;
650
651         /* Clean ccb data buffer */
652         memset(pccb->pdata, 0, pccb->datalen);
653
654         memcpy(pccb->pdata, hdr, sizeof(hdr));
655
656         if (pccb->datalen <= 35)
657                 return 0;
658
659         memset(fis, 0, sizeof(fis));
660         /* Construct the FIS */
661         fis[0] = 0x27;          /* Host to device FIS. */
662         fis[1] = 1 << 7;        /* Command FIS. */
663         fis[2] = ATA_CMD_ID_ATA; /* Command byte. */
664
665         /* Read id from sata */
666         port = pccb->target;
667
668         if (ahci_device_data_io(port, (u8 *) &fis, sizeof(fis), (u8 *)tmpid,
669                                 ATA_ID_WORDS * 2, 0)) {
670                 debug("scsi_ahci: SCSI inquiry command failure.\n");
671                 return -EIO;
672         }
673
674         if (!ataid[port]) {
675                 ataid[port] = malloc(ATA_ID_WORDS * 2);
676                 if (!ataid[port]) {
677                         printf("%s: No memory for ataid[port]\n", __func__);
678                         return -ENOMEM;
679                 }
680         }
681
682         idbuf = ataid[port];
683
684         memcpy(idbuf, tmpid, ATA_ID_WORDS * 2);
685         ata_swap_buf_le16(idbuf, ATA_ID_WORDS);
686
687         memcpy(&pccb->pdata[8], "ATA     ", 8);
688         ata_id_strcpy((u16 *)&pccb->pdata[16], &idbuf[ATA_ID_PROD], 16);
689         ata_id_strcpy((u16 *)&pccb->pdata[32], &idbuf[ATA_ID_FW_REV], 4);
690
691 #ifdef DEBUG
692         ata_dump_id(idbuf);
693 #endif
694         return 0;
695 }
696
697
698 /*
699  * SCSI READ10/WRITE10 command operation.
700  */
701 static int ata_scsiop_read_write(ccb *pccb, u8 is_write)
702 {
703         lbaint_t lba = 0;
704         u16 blocks = 0;
705         u8 fis[20];
706         u8 *user_buffer = pccb->pdata;
707         u32 user_buffer_size = pccb->datalen;
708
709         /* Retrieve the base LBA number from the ccb structure. */
710         if (pccb->cmd[0] == SCSI_READ16) {
711                 memcpy(&lba, pccb->cmd + 2, 8);
712                 lba = be64_to_cpu(lba);
713         } else {
714                 u32 temp;
715                 memcpy(&temp, pccb->cmd + 2, 4);
716                 lba = be32_to_cpu(temp);
717         }
718
719         /*
720          * Retrieve the base LBA number and the block count from
721          * the ccb structure.
722          *
723          * For 10-byte and 16-byte SCSI R/W commands, transfer
724          * length 0 means transfer 0 block of data.
725          * However, for ATA R/W commands, sector count 0 means
726          * 256 or 65536 sectors, not 0 sectors as in SCSI.
727          *
728          * WARNING: one or two older ATA drives treat 0 as 0...
729          */
730         if (pccb->cmd[0] == SCSI_READ16)
731                 blocks = (((u16)pccb->cmd[13]) << 8) | ((u16) pccb->cmd[14]);
732         else
733                 blocks = (((u16)pccb->cmd[7]) << 8) | ((u16) pccb->cmd[8]);
734
735         debug("scsi_ahci: %s %u blocks starting from lba 0x" LBAFU "\n",
736               is_write ?  "write" : "read", blocks, lba);
737
738         /* Preset the FIS */
739         memset(fis, 0, sizeof(fis));
740         fis[0] = 0x27;           /* Host to device FIS. */
741         fis[1] = 1 << 7;         /* Command FIS. */
742         /* Command byte (read/write). */
743         fis[2] = is_write ? ATA_CMD_WRITE_EXT : ATA_CMD_READ_EXT;
744
745         while (blocks) {
746                 u16 now_blocks; /* number of blocks per iteration */
747                 u32 transfer_size; /* number of bytes per iteration */
748
749                 now_blocks = min((u16)MAX_SATA_BLOCKS_READ_WRITE, blocks);
750
751                 transfer_size = ATA_SECT_SIZE * now_blocks;
752                 if (transfer_size > user_buffer_size) {
753                         printf("scsi_ahci: Error: buffer too small.\n");
754                         return -EIO;
755                 }
756
757                 /*
758                  * LBA48 SATA command but only use 32bit address range within
759                  * that (unless we've enabled 64bit LBA support). The next
760                  * smaller command range (28bit) is too small.
761                  */
762                 fis[4] = (lba >> 0) & 0xff;
763                 fis[5] = (lba >> 8) & 0xff;
764                 fis[6] = (lba >> 16) & 0xff;
765                 fis[7] = 1 << 6; /* device reg: set LBA mode */
766                 fis[8] = ((lba >> 24) & 0xff);
767 #ifdef CONFIG_SYS_64BIT_LBA
768                 if (pccb->cmd[0] == SCSI_READ16) {
769                         fis[9] = ((lba >> 32) & 0xff);
770                         fis[10] = ((lba >> 40) & 0xff);
771                 }
772 #endif
773
774                 fis[3] = 0xe0; /* features */
775
776                 /* Block (sector) count */
777                 fis[12] = (now_blocks >> 0) & 0xff;
778                 fis[13] = (now_blocks >> 8) & 0xff;
779
780                 /* Read/Write from ahci */
781                 if (ahci_device_data_io(pccb->target, (u8 *) &fis, sizeof(fis),
782                                         user_buffer, transfer_size,
783                                         is_write)) {
784                         debug("scsi_ahci: SCSI %s10 command failure.\n",
785                               is_write ? "WRITE" : "READ");
786                         return -EIO;
787                 }
788
789                 /* If this transaction is a write, do a following flush.
790                  * Writes in u-boot are so rare, and the logic to know when is
791                  * the last write and do a flush only there is sufficiently
792                  * difficult. Just do a flush after every write. This incurs,
793                  * usually, one extra flush when the rare writes do happen.
794                  */
795                 if (is_write) {
796                         if (-EIO == ata_io_flush(pccb->target))
797                                 return -EIO;
798                 }
799                 user_buffer += transfer_size;
800                 user_buffer_size -= transfer_size;
801                 blocks -= now_blocks;
802                 lba += now_blocks;
803         }
804
805         return 0;
806 }
807
808
809 /*
810  * SCSI READ CAPACITY10 command operation.
811  */
812 static int ata_scsiop_read_capacity10(ccb *pccb)
813 {
814         u32 cap;
815         u64 cap64;
816         u32 block_size;
817
818         if (!ataid[pccb->target]) {
819                 printf("scsi_ahci: SCSI READ CAPACITY10 command failure. "
820                        "\tNo ATA info!\n"
821                        "\tPlease run SCSI command INQUIRY first!\n");
822                 return -EPERM;
823         }
824
825         cap64 = ata_id_n_sectors(ataid[pccb->target]);
826         if (cap64 > 0x100000000ULL)
827                 cap64 = 0xffffffff;
828
829         cap = cpu_to_be32(cap64);
830         memcpy(pccb->pdata, &cap, sizeof(cap));
831
832         block_size = cpu_to_be32((u32)512);
833         memcpy(&pccb->pdata[4], &block_size, 4);
834
835         return 0;
836 }
837
838
839 /*
840  * SCSI READ CAPACITY16 command operation.
841  */
842 static int ata_scsiop_read_capacity16(ccb *pccb)
843 {
844         u64 cap;
845         u64 block_size;
846
847         if (!ataid[pccb->target]) {
848                 printf("scsi_ahci: SCSI READ CAPACITY16 command failure. "
849                        "\tNo ATA info!\n"
850                        "\tPlease run SCSI command INQUIRY first!\n");
851                 return -EPERM;
852         }
853
854         cap = ata_id_n_sectors(ataid[pccb->target]);
855         cap = cpu_to_be64(cap);
856         memcpy(pccb->pdata, &cap, sizeof(cap));
857
858         block_size = cpu_to_be64((u64)512);
859         memcpy(&pccb->pdata[8], &block_size, 8);
860
861         return 0;
862 }
863
864
865 /*
866  * SCSI TEST UNIT READY command operation.
867  */
868 static int ata_scsiop_test_unit_ready(ccb *pccb)
869 {
870         return (ataid[pccb->target]) ? 0 : -EPERM;
871 }
872
873
874 int scsi_exec(ccb *pccb)
875 {
876         int ret;
877
878         switch (pccb->cmd[0]) {
879         case SCSI_READ16:
880         case SCSI_READ10:
881                 ret = ata_scsiop_read_write(pccb, 0);
882                 break;
883         case SCSI_WRITE10:
884                 ret = ata_scsiop_read_write(pccb, 1);
885                 break;
886         case SCSI_RD_CAPAC10:
887                 ret = ata_scsiop_read_capacity10(pccb);
888                 break;
889         case SCSI_RD_CAPAC16:
890                 ret = ata_scsiop_read_capacity16(pccb);
891                 break;
892         case SCSI_TST_U_RDY:
893                 ret = ata_scsiop_test_unit_ready(pccb);
894                 break;
895         case SCSI_INQUIRY:
896                 ret = ata_scsiop_inquiry(pccb);
897                 break;
898         default:
899                 printf("Unsupport SCSI command 0x%02x\n", pccb->cmd[0]);
900                 return false;
901         }
902
903         if (ret) {
904                 debug("SCSI command 0x%02x ret errno %d\n", pccb->cmd[0], ret);
905                 return false;
906         }
907         return true;
908
909 }
910
911
912 void scsi_low_level_init(int busdevfunc)
913 {
914         int i;
915         u32 linkmap;
916
917 #ifndef CONFIG_SCSI_AHCI_PLAT
918         ahci_init_one(busdevfunc);
919 #endif
920
921         linkmap = probe_ent->link_port_map;
922
923         for (i = 0; i < CONFIG_SYS_SCSI_MAX_SCSI_ID; i++) {
924                 if (((linkmap >> i) & 0x01)) {
925                         if (ahci_port_start((u8) i)) {
926                                 printf("Can not start port %d\n", i);
927                                 continue;
928                         }
929                 }
930         }
931 }
932
933 #ifdef CONFIG_SCSI_AHCI_PLAT
934 int ahci_init(void __iomem *base)
935 {
936         int i, rc = 0;
937         u32 linkmap;
938
939         probe_ent = malloc(sizeof(struct ahci_probe_ent));
940         if (!probe_ent) {
941                 printf("%s: No memory for probe_ent\n", __func__);
942                 return -ENOMEM;
943         }
944
945         memset(probe_ent, 0, sizeof(struct ahci_probe_ent));
946
947         probe_ent->host_flags = ATA_FLAG_SATA
948                                 | ATA_FLAG_NO_LEGACY
949                                 | ATA_FLAG_MMIO
950                                 | ATA_FLAG_PIO_DMA
951                                 | ATA_FLAG_NO_ATAPI;
952         probe_ent->pio_mask = 0x1f;
953         probe_ent->udma_mask = 0x7f;    /*Fixme,assume to support UDMA6 */
954
955         probe_ent->mmio_base = base;
956
957         /* initialize adapter */
958         rc = ahci_host_init(probe_ent);
959         if (rc)
960                 goto err_out;
961
962         ahci_print_info(probe_ent);
963
964         linkmap = probe_ent->link_port_map;
965
966         for (i = 0; i < CONFIG_SYS_SCSI_MAX_SCSI_ID; i++) {
967                 if (((linkmap >> i) & 0x01)) {
968                         if (ahci_port_start((u8) i)) {
969                                 printf("Can not start port %d\n", i);
970                                 continue;
971                         }
972                 }
973         }
974 err_out:
975         return rc;
976 }
977
978 void __weak scsi_init(void)
979 {
980 }
981
982 #endif
983
984 /*
985  * In the general case of generic rotating media it makes sense to have a
986  * flush capability. It probably even makes sense in the case of SSDs because
987  * one cannot always know for sure what kind of internal cache/flush mechanism
988  * is embodied therein. At first it was planned to invoke this after the last
989  * write to disk and before rebooting. In practice, knowing, a priori, which
990  * is the last write is difficult. Because writing to the disk in u-boot is
991  * very rare, this flush command will be invoked after every block write.
992  */
993 static int ata_io_flush(u8 port)
994 {
995         u8 fis[20];
996         struct ahci_ioports *pp = &(probe_ent->port[port]);
997         void __iomem *port_mmio = pp->port_mmio;
998         u32 cmd_fis_len = 5;    /* five dwords */
999
1000         /* Preset the FIS */
1001         memset(fis, 0, 20);
1002         fis[0] = 0x27;           /* Host to device FIS. */
1003         fis[1] = 1 << 7;         /* Command FIS. */
1004         fis[2] = ATA_CMD_FLUSH_EXT;
1005
1006         memcpy((unsigned char *)pp->cmd_tbl, fis, 20);
1007         ahci_fill_cmd_slot(pp, cmd_fis_len);
1008         writel_with_flush(1, port_mmio + PORT_CMD_ISSUE);
1009
1010         if (waiting_for_cmd_completed(port_mmio + PORT_CMD_ISSUE,
1011                         WAIT_MS_FLUSH, 0x1)) {
1012                 debug("scsi_ahci: flush command timeout on port %d.\n", port);
1013                 return -EIO;
1014         }
1015
1016         return 0;
1017 }
1018
1019
1020 __weak void scsi_bus_reset(void)
1021 {
1022         /*Not implement*/
1023 }
1024
1025 void scsi_print_error(ccb * pccb)
1026 {
1027         /*The ahci error info can be read in the ahci driver*/
1028 }