1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * Copyright (C) 2008,2010 Freescale Semiconductor, Inc.
4  * Copyright 2019 NXP
5  * Author: Dave Liu <daveliu@freescale.com>
6  */
7 
8 #include <common.h>
9 #include <ahci.h>
10 #include <blk.h>
11 #include <command.h>
12 #include <console.h>
13 #include <cpu_func.h>
14 #include <dm.h>
15 #include <dm/device-internal.h>
16 #include <log.h>
17 #include <asm/io.h>
18 #include <asm/processor.h>
19 #include <asm/fsl_serdes.h>
20 #include <malloc.h>
21 #include <libata.h>
22 #include <fis.h>
23 #include <sata.h>
24 #include <linux/delay.h>
25 #include "fsl_sata.h"
26 
sdelay(unsigned long sec)27 static inline void sdelay(unsigned long sec)
28 {
29 	unsigned long i;
30 	for (i = 0; i < sec; i++)
31 		mdelay(1000);
32 }
33 
fsl_sata_dump_sfis(struct sata_fis_d2h * s)34 static void fsl_sata_dump_sfis(struct sata_fis_d2h *s)
35 {
36 	printf("Status FIS dump:\n\r");
37 	printf("fis_type:		%02x\n\r", s->fis_type);
38 	printf("pm_port_i:		%02x\n\r", s->pm_port_i);
39 	printf("status:			%02x\n\r", s->status);
40 	printf("error:			%02x\n\r", s->error);
41 	printf("lba_low:		%02x\n\r", s->lba_low);
42 	printf("lba_mid:		%02x\n\r", s->lba_mid);
43 	printf("lba_high:		%02x\n\r", s->lba_high);
44 	printf("device:			%02x\n\r", s->device);
45 	printf("lba_low_exp:		%02x\n\r", s->lba_low_exp);
46 	printf("lba_mid_exp:		%02x\n\r", s->lba_mid_exp);
47 	printf("lba_high_exp:		%02x\n\r", s->lba_high_exp);
48 	printf("res1:			%02x\n\r", s->res1);
49 	printf("sector_count:		%02x\n\r", s->sector_count);
50 	printf("sector_count_exp:	%02x\n\r", s->sector_count_exp);
51 }
52 
ata_wait_register(unsigned __iomem * addr,u32 mask,u32 val,u32 timeout_msec)53 static int ata_wait_register(unsigned __iomem *addr, u32 mask,
54 			 u32 val, u32 timeout_msec)
55 {
56 	int i;
57 	u32 temp;
58 
59 	for (i = 0; (((temp = in_le32(addr)) & mask) != val)
60 			 && i < timeout_msec; i++)
61 		mdelay(1);
62 	return (i < timeout_msec) ? 0 : -1;
63 }
64 
init_sata(struct fsl_ata_priv * priv,int dev)65 static int init_sata(struct fsl_ata_priv *priv, int dev)
66 {
67 	u32 length, align;
68 	cmd_hdr_tbl_t *cmd_hdr;
69 	u32 cda;
70 	u32 val32;
71 	fsl_sata_reg_t __iomem *reg;
72 	u32 sig;
73 	int i;
74 	fsl_sata_t *sata;
75 
76 	if (dev < 0 || dev > (CONFIG_SYS_SATA_MAX_DEVICE - 1)) {
77 		printf("the sata index %d is out of ranges\n\r", dev);
78 		return -1;
79 	}
80 
81 #ifdef CONFIG_MPC85xx
82 	if ((dev == 0) && (!is_serdes_configured(SATA1))) {
83 		printf("SATA%d [dev = %d] is not enabled\n", dev+1, dev);
84 		return -1;
85 	}
86 	if ((dev == 1) && (!is_serdes_configured(SATA2))) {
87 		printf("SATA%d [dev = %d] is not enabled\n", dev+1, dev);
88 		return -1;
89 	}
90 #endif
91 
92 	/* Allocate SATA device driver struct */
93 	sata = (fsl_sata_t *)malloc(sizeof(fsl_sata_t));
94 	if (!sata) {
95 		printf("alloc the sata device struct failed\n\r");
96 		return -1;
97 	}
98 	/* Zero all of the device driver struct */
99 	memset((void *)sata, 0, sizeof(fsl_sata_t));
100 
101 	snprintf(sata->name, 12, "SATA%d:", dev);
102 
103 	/* Set the controller register base address to device struct */
104 	reg = (fsl_sata_reg_t *)(priv->base + priv->offset * dev);
105 	sata->dma_flag = priv->flag;
106 	priv->fsl_sata = sata;
107 	sata->reg_base = reg;
108 
109 	/* Allocate the command header table, 4 bytes aligned */
110 	length = sizeof(struct cmd_hdr_tbl);
111 	align = SATA_HC_CMD_HDR_TBL_ALIGN;
112 	sata->cmd_hdr_tbl_offset = (void *)malloc(length + align);
113 	if (!sata->cmd_hdr_tbl_offset) {
114 		printf("alloc the command header failed\n\r");
115 		return -1;
116 	}
117 
118 	cmd_hdr = (cmd_hdr_tbl_t *)(((u32)sata->cmd_hdr_tbl_offset + align)
119 						& ~(align - 1));
120 	sata->cmd_hdr = cmd_hdr;
121 
122 	/* Zero all of the command header table */
123 	memset((void *)sata->cmd_hdr_tbl_offset, 0, length + align);
124 
125 	/* Allocate command descriptor for all command */
126 	length = sizeof(struct cmd_desc) * SATA_HC_MAX_CMD;
127 	align = SATA_HC_CMD_DESC_ALIGN;
128 	sata->cmd_desc_offset = (void *)malloc(length + align);
129 	if (!sata->cmd_desc_offset) {
130 		printf("alloc the command descriptor failed\n\r");
131 		return -1;
132 	}
133 	sata->cmd_desc = (cmd_desc_t *)(((u32)sata->cmd_desc_offset + align)
134 						& ~(align - 1));
135 	/* Zero all of command descriptor */
136 	memset((void *)sata->cmd_desc_offset, 0, length + align);
137 
138 	/* Link the command descriptor to command header */
139 	for (i = 0; i < SATA_HC_MAX_CMD; i++) {
140 		cda = ((u32)sata->cmd_desc + SATA_HC_CMD_DESC_SIZE * i)
141 					 & ~(CMD_HDR_CDA_ALIGN - 1);
142 		cmd_hdr->cmd_slot[i].cda = cpu_to_le32(cda);
143 	}
144 
145 	/* To have safe state, force the controller offline */
146 	val32 = in_le32(&reg->hcontrol);
147 	val32 &= ~HCONTROL_ONOFF;
148 	val32 |= HCONTROL_FORCE_OFFLINE;
149 	out_le32(&reg->hcontrol, val32);
150 
151 	/* Wait the controller offline */
152 	ata_wait_register(&reg->hstatus, HSTATUS_ONOFF, 0, 1000);
153 
154 	/* Set the command header base address to CHBA register to tell DMA */
155 	out_le32(&reg->chba, (u32)cmd_hdr & ~0x3);
156 
157 	/* Snoop for the command header */
158 	val32 = in_le32(&reg->hcontrol);
159 	val32 |= HCONTROL_HDR_SNOOP;
160 	out_le32(&reg->hcontrol, val32);
161 
162 	/* Disable all of interrupts */
163 	val32 = in_le32(&reg->hcontrol);
164 	val32 &= ~HCONTROL_INT_EN_ALL;
165 	out_le32(&reg->hcontrol, val32);
166 
167 	/* Clear all of interrupts */
168 	val32 = in_le32(&reg->hstatus);
169 	out_le32(&reg->hstatus, val32);
170 
171 	/* Set the ICC, no interrupt coalescing */
172 	out_le32(&reg->icc, 0x01000000);
173 
174 	/* No PM attatched, the SATA device direct connect */
175 	out_le32(&reg->cqpmp, 0);
176 
177 	/* Clear SError register */
178 	val32 = in_le32(&reg->serror);
179 	out_le32(&reg->serror, val32);
180 
181 	/* Clear CER register */
182 	val32 = in_le32(&reg->cer);
183 	out_le32(&reg->cer, val32);
184 
185 	/* Clear DER register */
186 	val32 = in_le32(&reg->der);
187 	out_le32(&reg->der, val32);
188 
189 	/* No device detection or initialization action requested */
190 	out_le32(&reg->scontrol, 0x00000300);
191 
192 	/* Configure the transport layer, default value */
193 	out_le32(&reg->transcfg, 0x08000016);
194 
195 	/* Configure the link layer, default value */
196 	out_le32(&reg->linkcfg, 0x0000ff34);
197 
198 	/* Bring the controller online */
199 	val32 = in_le32(&reg->hcontrol);
200 	val32 |= HCONTROL_ONOFF;
201 	out_le32(&reg->hcontrol, val32);
202 
203 	mdelay(100);
204 
205 	/* print sata device name */
206 	printf("%s ", sata->name);
207 
208 	/* Wait PHY RDY signal changed for 500ms */
209 	ata_wait_register(&reg->hstatus, HSTATUS_PHY_RDY,
210 			  HSTATUS_PHY_RDY, 500);
211 
212 	/* Check PHYRDY */
213 	val32 = in_le32(&reg->hstatus);
214 	if (val32 & HSTATUS_PHY_RDY) {
215 		sata->link = 1;
216 	} else {
217 		sata->link = 0;
218 		printf("(No RDY)\n\r");
219 		return -1;
220 	}
221 
222 	/* Wait for signature updated, which is 1st D2H */
223 	ata_wait_register(&reg->hstatus, HSTATUS_SIGNATURE,
224 			  HSTATUS_SIGNATURE, 10000);
225 
226 	if (val32 & HSTATUS_SIGNATURE) {
227 		sig = in_le32(&reg->sig);
228 		debug("Signature updated, the sig =%08x\n\r", sig);
229 		sata->ata_device_type = ata_dev_classify(sig);
230 	}
231 
232 	/* Check the speed */
233 	val32 = in_le32(&reg->sstatus);
234 	if ((val32 & SSTATUS_SPD_MASK) == SSTATUS_SPD_GEN1)
235 		printf("(1.5 Gbps)\n\r");
236 	else if ((val32 & SSTATUS_SPD_MASK) == SSTATUS_SPD_GEN2)
237 		printf("(3 Gbps)\n\r");
238 
239 	return 0;
240 }
241 
reset_sata(int dev)242 int reset_sata(int dev)
243 {
244 	return 0;
245 }
246 
fsl_sata_dump_regs(fsl_sata_reg_t __iomem * reg)247 static void fsl_sata_dump_regs(fsl_sata_reg_t __iomem *reg)
248 {
249 	printf("\n\rSATA:           %08x\n\r", (u32)reg);
250 	printf("CQR:            %08x\n\r", in_le32(&reg->cqr));
251 	printf("CAR:            %08x\n\r", in_le32(&reg->car));
252 	printf("CCR:            %08x\n\r", in_le32(&reg->ccr));
253 	printf("CER:            %08x\n\r", in_le32(&reg->cer));
254 	printf("CQR:            %08x\n\r", in_le32(&reg->cqr));
255 	printf("DER:            %08x\n\r", in_le32(&reg->der));
256 	printf("CHBA:           %08x\n\r", in_le32(&reg->chba));
257 	printf("HStatus:        %08x\n\r", in_le32(&reg->hstatus));
258 	printf("HControl:       %08x\n\r", in_le32(&reg->hcontrol));
259 	printf("CQPMP:          %08x\n\r", in_le32(&reg->cqpmp));
260 	printf("SIG:            %08x\n\r", in_le32(&reg->sig));
261 	printf("ICC:            %08x\n\r", in_le32(&reg->icc));
262 	printf("SStatus:        %08x\n\r", in_le32(&reg->sstatus));
263 	printf("SError:         %08x\n\r", in_le32(&reg->serror));
264 	printf("SControl:       %08x\n\r", in_le32(&reg->scontrol));
265 	printf("SNotification:  %08x\n\r", in_le32(&reg->snotification));
266 	printf("TransCfg:       %08x\n\r", in_le32(&reg->transcfg));
267 	printf("TransStatus:    %08x\n\r", in_le32(&reg->transstatus));
268 	printf("LinkCfg:        %08x\n\r", in_le32(&reg->linkcfg));
269 	printf("LinkCfg1:       %08x\n\r", in_le32(&reg->linkcfg1));
270 	printf("LinkCfg2:       %08x\n\r", in_le32(&reg->linkcfg2));
271 	printf("LinkStatus:     %08x\n\r", in_le32(&reg->linkstatus));
272 	printf("LinkStatus1:    %08x\n\r", in_le32(&reg->linkstatus1));
273 	printf("PhyCtrlCfg:     %08x\n\r", in_le32(&reg->phyctrlcfg));
274 	printf("SYSPR:          %08x\n\r", in_be32(&reg->syspr));
275 }
276 
fsl_ata_exec_ata_cmd(struct fsl_sata * sata,struct sata_fis_h2d * cfis,int is_ncq,int tag,u8 * buffer,u32 len)277 static int fsl_ata_exec_ata_cmd(struct fsl_sata *sata, struct sata_fis_h2d *cfis,
278 				int is_ncq, int tag, u8 *buffer, u32 len)
279 {
280 	cmd_hdr_entry_t *cmd_hdr;
281 	cmd_desc_t *cmd_desc;
282 	sata_fis_h2d_t *h2d;
283 	prd_entry_t *prde;
284 	u32 ext_c_ddc;
285 	u32 prde_count;
286 	u32 val32;
287 	u32 ttl;
288 	fsl_sata_reg_t __iomem *reg = sata->reg_base;
289 	int i;
290 
291 	/* Check xfer length */
292 	if (len > SATA_HC_MAX_XFER_LEN) {
293 		printf("max transfer length is 64MB\n\r");
294 		return 0;
295 	}
296 
297 	/* Setup the command descriptor */
298 	cmd_desc = sata->cmd_desc + tag;
299 
300 	/* Get the pointer cfis of command descriptor */
301 	h2d = (sata_fis_h2d_t *)cmd_desc->cfis;
302 
303 	/* Zero the cfis of command descriptor */
304 	memset((void *)h2d, 0, SATA_HC_CMD_DESC_CFIS_SIZE);
305 
306 	/* Copy the cfis from user to command descriptor */
307 	h2d->fis_type = cfis->fis_type;
308 	h2d->pm_port_c = cfis->pm_port_c;
309 	h2d->command = cfis->command;
310 
311 	h2d->features = cfis->features;
312 	h2d->features_exp = cfis->features_exp;
313 
314 	h2d->lba_low = cfis->lba_low;
315 	h2d->lba_mid = cfis->lba_mid;
316 	h2d->lba_high = cfis->lba_high;
317 	h2d->lba_low_exp = cfis->lba_low_exp;
318 	h2d->lba_mid_exp = cfis->lba_mid_exp;
319 	h2d->lba_high_exp = cfis->lba_high_exp;
320 
321 	if (!is_ncq) {
322 		h2d->sector_count = cfis->sector_count;
323 		h2d->sector_count_exp = cfis->sector_count_exp;
324 	} else { /* NCQ */
325 		h2d->sector_count = (u8)(tag << 3);
326 	}
327 
328 	h2d->device = cfis->device;
329 	h2d->control = cfis->control;
330 
331 	/* Setup the PRD table */
332 	prde = (prd_entry_t *)cmd_desc->prdt;
333 	memset((void *)prde, 0, sizeof(struct prdt));
334 
335 	prde_count = 0;
336 	ttl = len;
337 	for (i = 0; i < SATA_HC_MAX_PRD_DIRECT; i++) {
338 		if (!len)
339 			break;
340 		prde->dba = cpu_to_le32((u32)buffer & ~0x3);
341 		debug("dba = %08x\n\r", (u32)buffer);
342 
343 		if (len < PRD_ENTRY_MAX_XFER_SZ) {
344 			ext_c_ddc = PRD_ENTRY_DATA_SNOOP | len;
345 			debug("ext_c_ddc1 = %08x, len = %08x\n\r", ext_c_ddc, len);
346 			prde->ext_c_ddc = cpu_to_le32(ext_c_ddc);
347 			prde_count++;
348 			prde++;
349 			break;
350 		} else {
351 			ext_c_ddc = PRD_ENTRY_DATA_SNOOP; /* 4M bytes */
352 			debug("ext_c_ddc2 = %08x, len = %08x\n\r", ext_c_ddc, len);
353 			prde->ext_c_ddc = cpu_to_le32(ext_c_ddc);
354 			buffer += PRD_ENTRY_MAX_XFER_SZ;
355 			len -= PRD_ENTRY_MAX_XFER_SZ;
356 			prde_count++;
357 			prde++;
358 		}
359 	}
360 
361 	/* Setup the command slot of cmd hdr */
362 	cmd_hdr = (cmd_hdr_entry_t *)&sata->cmd_hdr->cmd_slot[tag];
363 
364 	cmd_hdr->cda = cpu_to_le32((u32)cmd_desc & ~0x3);
365 
366 	val32 = prde_count << CMD_HDR_PRD_ENTRY_SHIFT;
367 	val32 |= sizeof(sata_fis_h2d_t);
368 	cmd_hdr->prde_fis_len = cpu_to_le32(val32);
369 
370 	cmd_hdr->ttl = cpu_to_le32(ttl);
371 
372 	if (!is_ncq) {
373 		val32 = CMD_HDR_ATTR_RES | CMD_HDR_ATTR_SNOOP;
374 	} else {
375 		val32 = CMD_HDR_ATTR_RES | CMD_HDR_ATTR_SNOOP | CMD_HDR_ATTR_FPDMA;
376 	}
377 
378 	tag &= CMD_HDR_ATTR_TAG;
379 	val32 |= tag;
380 
381 	debug("attribute = %08x\n\r", val32);
382 	cmd_hdr->attribute = cpu_to_le32(val32);
383 
384 	/* Make sure cmd desc and cmd slot valid before command issue */
385 	sync();
386 
387 	/* PMP*/
388 	val32 = (u32)(h2d->pm_port_c & 0x0f);
389 	out_le32(&reg->cqpmp, val32);
390 
391 	/* Wait no active */
392 	if (ata_wait_register(&reg->car, (1 << tag), 0, 10000))
393 		printf("Wait no active time out\n\r");
394 
395 	/* Issue command */
396 	if (!(in_le32(&reg->cqr) & (1 << tag))) {
397 		val32 = 1 << tag;
398 		out_le32(&reg->cqr, val32);
399 	}
400 
401 	/* Wait command completed for 10s */
402 	if (ata_wait_register(&reg->ccr, (1 << tag), (1 << tag), 10000)) {
403 		if (!is_ncq)
404 			printf("Non-NCQ command time out\n\r");
405 		else
406 			printf("NCQ command time out\n\r");
407 	}
408 
409 	val32 = in_le32(&reg->cer);
410 
411 	if (val32) {
412 		u32 der;
413 		fsl_sata_dump_sfis((struct sata_fis_d2h *)cmd_desc->sfis);
414 		printf("CE at device\n\r");
415 		fsl_sata_dump_regs(reg);
416 		der = in_le32(&reg->der);
417 		out_le32(&reg->cer, val32);
418 		out_le32(&reg->der, der);
419 	}
420 
421 	/* Clear complete flags */
422 	val32 = in_le32(&reg->ccr);
423 	out_le32(&reg->ccr, val32);
424 
425 	return len;
426 }
427 
fsl_ata_exec_reset_cmd(struct fsl_sata * sata,struct sata_fis_h2d * cfis,int tag,u8 * buffer,u32 len)428 static int fsl_ata_exec_reset_cmd(struct fsl_sata *sata, struct sata_fis_h2d *cfis,
429 				 int tag, u8 *buffer, u32 len)
430 {
431 	return 0;
432 }
433 
fsl_sata_exec_cmd(struct fsl_sata * sata,struct sata_fis_h2d * cfis,enum cmd_type command_type,int tag,u8 * buffer,u32 len)434 static int fsl_sata_exec_cmd(struct fsl_sata *sata, struct sata_fis_h2d *cfis,
435 		 enum cmd_type command_type, int tag, u8 *buffer, u32 len)
436 {
437 	int rc;
438 
439 	if (tag > SATA_HC_MAX_CMD || tag < 0) {
440 		printf("tag is out of range, tag=%d\n\r", tag);
441 		return -1;
442 	}
443 
444 	switch (command_type) {
445 	case CMD_ATA:
446 		rc = fsl_ata_exec_ata_cmd(sata, cfis, 0, tag, buffer, len);
447 		return rc;
448 	case CMD_RESET:
449 		rc = fsl_ata_exec_reset_cmd(sata, cfis, tag, buffer, len);
450 		return rc;
451 	case CMD_NCQ:
452 		rc = fsl_ata_exec_ata_cmd(sata, cfis, 1, tag, buffer, len);
453 		return rc;
454 	case CMD_ATAPI:
455 	case CMD_VENDOR_BIST:
456 	case CMD_BIST:
457 		printf("not support now\n\r");
458 		return -1;
459 	default:
460 		break;
461 	}
462 
463 	return -1;
464 }
465 
fsl_sata_xfer_mode(fsl_sata_t * sata,u16 * id)466 static void fsl_sata_xfer_mode(fsl_sata_t *sata, u16 *id)
467 {
468 	sata->pio = id[ATA_ID_PIO_MODES];
469 	sata->mwdma = id[ATA_ID_MWDMA_MODES];
470 	sata->udma = id[ATA_ID_UDMA_MODES];
471 	debug("pio %04x, mwdma %04x, udma %04x\n\r", sata->pio, sata->mwdma, sata->udma);
472 }
473 
fsl_sata_set_features(fsl_sata_t * sata)474 static void fsl_sata_set_features(fsl_sata_t *sata)
475 {
476 	struct sata_fis_h2d h2d, *cfis = &h2d;
477 	u8 udma_cap;
478 
479 	memset(cfis, 0, sizeof(struct sata_fis_h2d));
480 
481 	cfis->fis_type = SATA_FIS_TYPE_REGISTER_H2D;
482 	cfis->pm_port_c = 0x80; /* is command */
483 	cfis->command = ATA_CMD_SET_FEATURES;
484 	cfis->features = SETFEATURES_XFER;
485 
486 	/* First check the device capablity */
487 	udma_cap = (u8)(sata->udma & 0xff);
488 	debug("udma_cap %02x\n\r", udma_cap);
489 
490 	if (udma_cap == ATA_UDMA6)
491 		cfis->sector_count = XFER_UDMA_6;
492 	if (udma_cap == ATA_UDMA5)
493 		cfis->sector_count = XFER_UDMA_5;
494 	if (udma_cap == ATA_UDMA4)
495 		cfis->sector_count = XFER_UDMA_4;
496 	if (udma_cap == ATA_UDMA3)
497 		cfis->sector_count = XFER_UDMA_3;
498 
499 	fsl_sata_exec_cmd(sata, cfis, CMD_ATA, 0, NULL, 0);
500 }
501 
fsl_sata_rw_cmd(fsl_sata_t * sata,u32 start,u32 blkcnt,u8 * buffer,int is_write)502 static u32 fsl_sata_rw_cmd(fsl_sata_t *sata, u32 start, u32 blkcnt, u8 *buffer,
503 			   int is_write)
504 {
505 	struct sata_fis_h2d h2d, *cfis = &h2d;
506 	u32 block;
507 
508 	block = start;
509 
510 	memset(cfis, 0, sizeof(struct sata_fis_h2d));
511 
512 	cfis->fis_type = SATA_FIS_TYPE_REGISTER_H2D;
513 	cfis->pm_port_c = 0x80; /* is command */
514 	cfis->command = (is_write) ? ATA_CMD_WRITE : ATA_CMD_READ;
515 	cfis->device = ATA_LBA;
516 
517 	cfis->device |= (block >> 24) & 0xf;
518 	cfis->lba_high = (block >> 16) & 0xff;
519 	cfis->lba_mid = (block >> 8) & 0xff;
520 	cfis->lba_low = block & 0xff;
521 	cfis->sector_count = (u8)(blkcnt & 0xff);
522 
523 	fsl_sata_exec_cmd(sata, cfis, CMD_ATA, 0, buffer, ATA_SECT_SIZE * blkcnt);
524 	return blkcnt;
525 }
526 
fsl_sata_flush_cache(fsl_sata_t * sata)527 static void fsl_sata_flush_cache(fsl_sata_t *sata)
528 {
529 	struct sata_fis_h2d h2d, *cfis = &h2d;
530 
531 	memset(cfis, 0, sizeof(struct sata_fis_h2d));
532 
533 	cfis->fis_type = SATA_FIS_TYPE_REGISTER_H2D;
534 	cfis->pm_port_c = 0x80; /* is command */
535 	cfis->command = ATA_CMD_FLUSH;
536 
537 	fsl_sata_exec_cmd(sata, cfis, CMD_ATA, 0, NULL, 0);
538 }
539 
fsl_sata_rw_cmd_ext(fsl_sata_t * sata,u32 start,u32 blkcnt,u8 * buffer,int is_write)540 static u32 fsl_sata_rw_cmd_ext(fsl_sata_t *sata, u32 start, u32 blkcnt,
541 			       u8 *buffer, int is_write)
542 {
543 	struct sata_fis_h2d h2d, *cfis = &h2d;
544 	u64 block;
545 
546 	block = (u64)start;
547 
548 	memset(cfis, 0, sizeof(struct sata_fis_h2d));
549 
550 	cfis->fis_type = SATA_FIS_TYPE_REGISTER_H2D;
551 	cfis->pm_port_c = 0x80; /* is command */
552 
553 	cfis->command = (is_write) ? ATA_CMD_WRITE_EXT
554 				 : ATA_CMD_READ_EXT;
555 
556 	cfis->lba_high_exp = (block >> 40) & 0xff;
557 	cfis->lba_mid_exp = (block >> 32) & 0xff;
558 	cfis->lba_low_exp = (block >> 24) & 0xff;
559 	cfis->lba_high = (block >> 16) & 0xff;
560 	cfis->lba_mid = (block >> 8) & 0xff;
561 	cfis->lba_low = block & 0xff;
562 	cfis->device = ATA_LBA;
563 	cfis->sector_count_exp = (blkcnt >> 8) & 0xff;
564 	cfis->sector_count = blkcnt & 0xff;
565 
566 	fsl_sata_exec_cmd(sata, cfis, CMD_ATA, 0, buffer, ATA_SECT_SIZE * blkcnt);
567 	return blkcnt;
568 }
569 
fsl_sata_rw_ncq_cmd(fsl_sata_t * sata,u32 start,u32 blkcnt,u8 * buffer,int is_write)570 static u32 fsl_sata_rw_ncq_cmd(fsl_sata_t *sata, u32 start, u32 blkcnt,
571 			       u8 *buffer, int is_write)
572 {
573 	struct sata_fis_h2d h2d, *cfis = &h2d;
574 	int ncq_channel;
575 	u64 block;
576 
577 	if (sata->lba48 != 1) {
578 		printf("execute FPDMA command on non-LBA48 hard disk\n\r");
579 		return -1;
580 	}
581 
582 	block = (u64)start;
583 
584 	memset(cfis, 0, sizeof(struct sata_fis_h2d));
585 
586 	cfis->fis_type = SATA_FIS_TYPE_REGISTER_H2D;
587 	cfis->pm_port_c = 0x80; /* is command */
588 
589 	cfis->command = (is_write) ? ATA_CMD_FPDMA_WRITE
590 				 : ATA_CMD_FPDMA_READ;
591 
592 	cfis->lba_high_exp = (block >> 40) & 0xff;
593 	cfis->lba_mid_exp = (block >> 32) & 0xff;
594 	cfis->lba_low_exp = (block >> 24) & 0xff;
595 	cfis->lba_high = (block >> 16) & 0xff;
596 	cfis->lba_mid = (block >> 8) & 0xff;
597 	cfis->lba_low = block & 0xff;
598 
599 	cfis->device = ATA_LBA;
600 	cfis->features_exp = (blkcnt >> 8) & 0xff;
601 	cfis->features = blkcnt & 0xff;
602 
603 	if (sata->queue_depth >= SATA_HC_MAX_CMD)
604 		ncq_channel = SATA_HC_MAX_CMD - 1;
605 	else
606 		ncq_channel = sata->queue_depth - 1;
607 
608 	/* Use the latest queue */
609 	fsl_sata_exec_cmd(sata, cfis, CMD_NCQ, ncq_channel, buffer, ATA_SECT_SIZE * blkcnt);
610 	return blkcnt;
611 }
612 
fsl_sata_flush_cache_ext(fsl_sata_t * sata)613 static void fsl_sata_flush_cache_ext(fsl_sata_t *sata)
614 {
615 	struct sata_fis_h2d h2d, *cfis = &h2d;
616 
617 	memset(cfis, 0, sizeof(struct sata_fis_h2d));
618 
619 	cfis->fis_type = SATA_FIS_TYPE_REGISTER_H2D;
620 	cfis->pm_port_c = 0x80; /* is command */
621 	cfis->command = ATA_CMD_FLUSH_EXT;
622 
623 	fsl_sata_exec_cmd(sata, cfis, CMD_ATA, 0, NULL, 0);
624 }
625 
fsl_sata_init_wcache(fsl_sata_t * sata,u16 * id)626 static void fsl_sata_init_wcache(fsl_sata_t *sata, u16 *id)
627 {
628 	if (ata_id_has_wcache(id) && ata_id_wcache_enabled(id))
629 		sata->wcache = 1;
630 	if (ata_id_has_flush(id))
631 		sata->flush = 1;
632 	if (ata_id_has_flush_ext(id))
633 		sata->flush_ext = 1;
634 }
635 
ata_low_level_rw_lba48(fsl_sata_t * sata,u32 blknr,lbaint_t blkcnt,const void * buffer,int is_write)636 static u32 ata_low_level_rw_lba48(fsl_sata_t *sata, u32 blknr, lbaint_t blkcnt,
637 				  const void *buffer, int is_write)
638 {
639 	u32 start, blks;
640 	u8 *addr;
641 	int max_blks;
642 
643 	start = blknr;
644 	blks = blkcnt;
645 	addr = (u8 *)buffer;
646 
647 	max_blks = ATA_MAX_SECTORS_LBA48;
648 	do {
649 		if (blks > max_blks) {
650 			if (sata->dma_flag != FLAGS_FPDMA)
651 				fsl_sata_rw_cmd_ext(sata, start, max_blks, addr,
652 						    is_write);
653 			else
654 				fsl_sata_rw_ncq_cmd(sata, start, max_blks, addr,
655 						    is_write);
656 			start += max_blks;
657 			blks -= max_blks;
658 			addr += ATA_SECT_SIZE * max_blks;
659 		} else {
660 			if (sata->dma_flag != FLAGS_FPDMA)
661 				fsl_sata_rw_cmd_ext(sata, start, blks, addr,
662 						    is_write);
663 			else
664 				fsl_sata_rw_ncq_cmd(sata, start, blks, addr,
665 						    is_write);
666 			start += blks;
667 			blks = 0;
668 			addr += ATA_SECT_SIZE * blks;
669 		}
670 	} while (blks != 0);
671 
672 	return blkcnt;
673 }
674 
ata_low_level_rw_lba28(fsl_sata_t * sata,u32 blknr,u32 blkcnt,const void * buffer,int is_write)675 static u32 ata_low_level_rw_lba28(fsl_sata_t *sata, u32 blknr, u32 blkcnt,
676 				  const void *buffer, int is_write)
677 {
678 	u32 start, blks;
679 	u8 *addr;
680 	int max_blks;
681 
682 	start = blknr;
683 	blks = blkcnt;
684 	addr = (u8 *)buffer;
685 
686 	max_blks = ATA_MAX_SECTORS;
687 	do {
688 		if (blks > max_blks) {
689 			fsl_sata_rw_cmd(sata, start, max_blks, addr, is_write);
690 			start += max_blks;
691 			blks -= max_blks;
692 			addr += ATA_SECT_SIZE * max_blks;
693 		} else {
694 			fsl_sata_rw_cmd(sata, start, blks, addr, is_write);
695 			start += blks;
696 			blks = 0;
697 			addr += ATA_SECT_SIZE * blks;
698 		}
699 	} while (blks != 0);
700 
701 	return blkcnt;
702 }
703 
704 /*
705  * SATA interface between low level driver and command layer
706  */
sata_read(struct udevice * dev,lbaint_t blknr,lbaint_t blkcnt,void * buffer)707 static ulong sata_read(struct udevice *dev, lbaint_t blknr, lbaint_t blkcnt,
708 		       void *buffer)
709 {
710 	struct fsl_ata_priv *priv = dev_get_plat(dev);
711 	fsl_sata_t *sata = priv->fsl_sata;
712 	u32 rc;
713 
714 	if (sata->lba48)
715 		rc = ata_low_level_rw_lba48(sata, blknr, blkcnt, buffer,
716 					    READ_CMD);
717 	else
718 		rc = ata_low_level_rw_lba28(sata, blknr, blkcnt, buffer,
719 					    READ_CMD);
720 	return rc;
721 }
722 
sata_write(struct udevice * dev,lbaint_t blknr,lbaint_t blkcnt,const void * buffer)723 static ulong sata_write(struct udevice *dev, lbaint_t blknr, lbaint_t blkcnt,
724 			const void *buffer)
725 {
726 	struct fsl_ata_priv *priv = dev_get_plat(dev);
727 	fsl_sata_t *sata = priv->fsl_sata;
728 	u32 rc;
729 
730 	if (sata->lba48) {
731 		rc = ata_low_level_rw_lba48(sata, blknr, blkcnt, buffer,
732 					    WRITE_CMD);
733 		if (sata->wcache && sata->flush_ext)
734 			fsl_sata_flush_cache_ext(sata);
735 	} else {
736 		rc = ata_low_level_rw_lba28(sata, blknr, blkcnt, buffer,
737 					    WRITE_CMD);
738 		if (sata->wcache && sata->flush)
739 			fsl_sata_flush_cache(sata);
740 	}
741 	return rc;
742 }
743 
fsl_sata_identify(fsl_sata_t * sata,u16 * id)744 static void fsl_sata_identify(fsl_sata_t *sata, u16 *id)
745 {
746 	struct sata_fis_h2d h2d, *cfis = &h2d;
747 
748 	memset(cfis, 0, sizeof(struct sata_fis_h2d));
749 
750 	cfis->fis_type = SATA_FIS_TYPE_REGISTER_H2D;
751 	cfis->pm_port_c = 0x80; /* is command */
752 	cfis->command = ATA_CMD_ID_ATA;
753 
754 	fsl_sata_exec_cmd(sata, cfis, CMD_ATA, 0, (u8 *)id, ATA_ID_WORDS * 2);
755 	ata_swap_buf_le16(id, ATA_ID_WORDS);
756 }
757 
scan_sata(struct udevice * dev)758 static int scan_sata(struct udevice *dev)
759 {
760 	struct blk_desc *desc = dev_get_uclass_plat(dev);
761 	struct fsl_ata_priv *priv = dev_get_plat(dev);
762 	fsl_sata_t *sata = priv->fsl_sata;
763 
764 	unsigned char serial[ATA_ID_SERNO_LEN + 1];
765 	unsigned char firmware[ATA_ID_FW_REV_LEN + 1];
766 	unsigned char product[ATA_ID_PROD_LEN + 1];
767 	u16 *id;
768 	u64 n_sectors;
769 
770 	/* if no detected link */
771 	if (!sata->link)
772 		return -1;
773 
774 	id = (u16 *)malloc(ATA_ID_WORDS * 2);
775 	if (!id) {
776 		printf("id malloc failed\n\r");
777 		return -1;
778 	}
779 
780 	/* Identify device to get information */
781 	fsl_sata_identify(sata, id);
782 
783 	/* Serial number */
784 	ata_id_c_string(id, serial, ATA_ID_SERNO, sizeof(serial));
785 
786 	/* Firmware version */
787 	ata_id_c_string(id, firmware, ATA_ID_FW_REV, sizeof(firmware));
788 
789 	/* Product model */
790 	ata_id_c_string(id, product, ATA_ID_PROD, sizeof(product));
791 
792 	/* Totoal sectors */
793 	n_sectors = ata_id_n_sectors(id);
794 
795 #ifdef CONFIG_LBA48
796 	/* Check if support LBA48 */
797 	if (ata_id_has_lba48(id)) {
798 		sata->lba48 = 1;
799 		debug("Device support LBA48\n\r");
800 	} else
801 		debug("Device supports LBA28\n\r");
802 #endif
803 
804 	memcpy(desc->product, serial, sizeof(serial));
805 	memcpy(desc->revision, firmware, sizeof(firmware));
806 	memcpy(desc->vendor, product, sizeof(product));
807 	desc->lba = n_sectors;
808 #ifdef CONFIG_LBA48
809 	desc->lba48 = sata->lba48;
810 #endif
811 
812 	/* Get the NCQ queue depth from device */
813 	sata->queue_depth = ata_id_queue_depth(id);
814 
815 	/* Get the xfer mode from device */
816 	fsl_sata_xfer_mode(sata, id);
817 
818 	/* Get the write cache status from device */
819 	fsl_sata_init_wcache(sata, id);
820 
821 	/* Set the xfer mode to highest speed */
822 	fsl_sata_set_features(sata);
823 
824 #ifdef DEBUG
825 	ata_dump_id(id);
826 #endif
827 	free((void *)id);
828 	return 0;
829 }
830 
831 static const struct blk_ops sata_fsl_blk_ops = {
832 	.read	= sata_read,
833 	.write	= sata_write,
834 };
835 
836 U_BOOT_DRIVER(sata_fsl_driver) = {
837 	.name = "sata_fsl_blk",
838 	.id = UCLASS_BLK,
839 	.ops = &sata_fsl_blk_ops,
840 	.plat_auto	= sizeof(struct fsl_ata_priv),
841 };
842 
fsl_ata_of_to_plat(struct udevice * dev)843 static int fsl_ata_of_to_plat(struct udevice *dev)
844 {
845 	struct fsl_ata_priv *priv = dev_get_priv(dev);
846 
847 	priv->number = dev_read_u32_default(dev, "sata-number", -1);
848 	priv->flag = dev_read_u32_default(dev, "sata-fpdma", -1);
849 	priv->offset = dev_read_u32_default(dev, "sata-offset", -1);
850 
851 	priv->base = dev_read_addr(dev);
852 	if (priv->base == FDT_ADDR_T_NONE)
853 		return -EINVAL;
854 
855 	return 0;
856 }
857 
fsl_unbind_device(struct udevice * dev)858 static int fsl_unbind_device(struct udevice *dev)
859 {
860 	int ret;
861 
862 	ret = device_remove(dev, DM_REMOVE_NORMAL);
863 	if (ret)
864 		return ret;
865 
866 	ret = device_unbind(dev);
867 	if (ret)
868 		return ret;
869 
870 	return 0;
871 }
872 
fsl_ata_probe(struct udevice * dev)873 static int fsl_ata_probe(struct udevice *dev)
874 {
875 	struct fsl_ata_priv *blk_priv, *priv;
876 	struct udevice *blk;
877 	int failed_number;
878 	char sata_name[10];
879 	int nr_ports;
880 	int ret;
881 	int i;
882 
883 	failed_number = 0;
884 	priv = dev_get_priv(dev);
885 	nr_ports = priv->number;
886 	nr_ports = min(nr_ports, CONFIG_SYS_SATA_MAX_DEVICE);
887 
888 	for (i = 0; i < nr_ports; i++) {
889 		snprintf(sata_name, sizeof(sata_name), "fsl_sata%d", i);
890 		ret = blk_create_devicef(dev, "sata_fsl_blk", sata_name,
891 					 UCLASS_AHCI, -1, 512, 0, &blk);
892 		if (ret) {
893 			debug("Can't create device\n");
894 			return ret;
895 		}
896 
897 		/* Init SATA port */
898 		ret = init_sata(priv, i);
899 		if (ret) {
900 			debug("%s: Failed to init sata\n", __func__);
901 			ret = fsl_unbind_device(blk);
902 			if (ret)
903 				return ret;
904 
905 			failed_number++;
906 			continue;
907 		}
908 
909 		blk_priv = dev_get_plat(blk);
910 		blk_priv->fsl_sata = priv->fsl_sata;
911 		/* Scan SATA port */
912 		ret = scan_sata(blk);
913 		if (ret) {
914 			debug("%s: Failed to scan bus\n", __func__);
915 			ret = fsl_unbind_device(blk);
916 			if (ret)
917 				return ret;
918 
919 			failed_number++;
920 			continue;
921 		}
922 
923 		ret = device_probe(dev);
924 		if (ret < 0) {
925 			debug("Probing %s failed (%d)\n", dev->name, ret);
926 			ret = fsl_unbind_device(blk);
927 			if (ret)
928 				return ret;
929 
930 			failed_number++;
931 			continue;
932 		}
933 	}
934 
935 	if (failed_number == nr_ports)
936 		return -ENODEV;
937 	else
938 		return 0;
939 }
940 
fsl_ata_remove(struct udevice * dev)941 static int fsl_ata_remove(struct udevice *dev)
942 {
943 	fsl_sata_t *sata;
944 	struct fsl_ata_priv *priv;
945 
946 	priv = dev_get_priv(dev);
947 	sata = priv->fsl_sata;
948 
949 	free(sata->cmd_hdr_tbl_offset);
950 	free(sata->cmd_desc_offset);
951 	free(sata);
952 
953 	return 0;
954 }
955 
sata_fsl_scan(struct udevice * dev)956 static int sata_fsl_scan(struct udevice *dev)
957 {
958 	/* Nothing to do here */
959 
960 	return 0;
961 }
962 
963 struct ahci_ops sata_fsl_ahci_ops = {
964 	.scan = sata_fsl_scan,
965 };
966 
967 static const struct udevice_id fsl_ata_ids[] = {
968 	{ .compatible = "fsl,pq-sata-v2" },
969 	{ }
970 };
971 
972 U_BOOT_DRIVER(fsl_ahci) = {
973 	.name	= "fsl_ahci",
974 	.id = UCLASS_AHCI,
975 	.of_match = fsl_ata_ids,
976 	.ops = &sata_fsl_ahci_ops,
977 	.of_to_plat = fsl_ata_of_to_plat,
978 	.probe	= fsl_ata_probe,
979 	.remove = fsl_ata_remove,
980 	.priv_auto	= sizeof(struct fsl_ata_priv),
981 };
982