xref: /rk3399_ARM-atf/drivers/ufs/ufs.c (revision f2de48cb143c20ccd7a9c141df3d34cae74049de)
1 /*
2  * Copyright (c) 2017-2021, ARM Limited and Contributors. All rights reserved.
3  *
4  * SPDX-License-Identifier: BSD-3-Clause
5  */
6 
7 #include <assert.h>
8 #include <endian.h>
9 #include <errno.h>
10 #include <stdint.h>
11 #include <string.h>
12 
13 #include <platform_def.h>
14 
15 #include <arch_helpers.h>
16 #include <common/debug.h>
17 #include <drivers/delay_timer.h>
18 #include <drivers/ufs.h>
19 #include <lib/mmio.h>
20 
21 #define CDB_ADDR_MASK			127
22 #define ALIGN_CDB(x)			(((x) + CDB_ADDR_MASK) & ~CDB_ADDR_MASK)
23 #define ALIGN_8(x)			(((x) + 7) & ~7)
24 
25 #define UFS_DESC_SIZE			0x400
26 #define MAX_UFS_DESC_SIZE		0x8000		/* 32 descriptors */
27 
28 #define MAX_PRDT_SIZE			0x40000		/* 256KB */
29 
30 static ufs_params_t ufs_params;
31 static int nutrs;	/* Number of UTP Transfer Request Slots */
32 
33 int ufshc_send_uic_cmd(uintptr_t base, uic_cmd_t *cmd)
34 {
35 	unsigned int data;
36 
37 	if (base == 0 || cmd == NULL)
38 		return -EINVAL;
39 
40 	data = mmio_read_32(base + HCS);
41 	if ((data & HCS_UCRDY) == 0)
42 		return -EBUSY;
43 	mmio_write_32(base + IS, ~0);
44 	mmio_write_32(base + UCMDARG1, cmd->arg1);
45 	mmio_write_32(base + UCMDARG2, cmd->arg2);
46 	mmio_write_32(base + UCMDARG3, cmd->arg3);
47 	mmio_write_32(base + UICCMD, cmd->op);
48 
49 	do {
50 		data = mmio_read_32(base + IS);
51 	} while ((data & UFS_INT_UCCS) == 0);
52 	mmio_write_32(base + IS, UFS_INT_UCCS);
53 	return mmio_read_32(base + UCMDARG2) & CONFIG_RESULT_CODE_MASK;
54 }
55 
56 int ufshc_dme_get(unsigned int attr, unsigned int idx, unsigned int *val)
57 {
58 	uintptr_t base;
59 	unsigned int data;
60 	int result, retries;
61 	uic_cmd_t cmd;
62 
63 	assert(ufs_params.reg_base != 0);
64 
65 	if (val == NULL)
66 		return -EINVAL;
67 
68 	base = ufs_params.reg_base;
69 	for (retries = 0; retries < 100; retries++) {
70 		data = mmio_read_32(base + HCS);
71 		if ((data & HCS_UCRDY) != 0)
72 			break;
73 		mdelay(1);
74 	}
75 	if (retries >= 100)
76 		return -EBUSY;
77 
78 	cmd.arg1 = (attr << 16) | GEN_SELECTOR_IDX(idx);
79 	cmd.arg2 = 0;
80 	cmd.arg3 = 0;
81 	cmd.op = DME_GET;
82 	for (retries = 0; retries < UFS_UIC_COMMAND_RETRIES; ++retries) {
83 		result = ufshc_send_uic_cmd(base, &cmd);
84 		if (result == 0)
85 			break;
86 		data = mmio_read_32(base + IS);
87 		if (data & UFS_INT_UE)
88 			return -EINVAL;
89 	}
90 	if (retries >= UFS_UIC_COMMAND_RETRIES)
91 		return -EIO;
92 
93 	*val = mmio_read_32(base + UCMDARG3);
94 	return 0;
95 }
96 
97 int ufshc_dme_set(unsigned int attr, unsigned int idx, unsigned int val)
98 {
99 	uintptr_t base;
100 	unsigned int data;
101 	int result, retries;
102 	uic_cmd_t cmd;
103 
104 	assert((ufs_params.reg_base != 0));
105 
106 	base = ufs_params.reg_base;
107 	cmd.arg1 = (attr << 16) | GEN_SELECTOR_IDX(idx);
108 	cmd.arg2 = 0;
109 	cmd.arg3 = val;
110 	cmd.op = DME_SET;
111 
112 	for (retries = 0; retries < UFS_UIC_COMMAND_RETRIES; ++retries) {
113 		result = ufshc_send_uic_cmd(base, &cmd);
114 		if (result == 0)
115 			break;
116 		data = mmio_read_32(base + IS);
117 		if (data & UFS_INT_UE)
118 			return -EINVAL;
119 	}
120 	if (retries >= UFS_UIC_COMMAND_RETRIES)
121 		return -EIO;
122 
123 	return 0;
124 }
125 
126 static int ufshc_hce_enable(uintptr_t base)
127 {
128 	unsigned int data;
129 	int retries;
130 
131 	/* Enable Host Controller */
132 	mmio_write_32(base + HCE, HCE_ENABLE);
133 
134 	/* Wait until basic initialization sequence completed */
135 	for (retries = 0; retries < HCE_ENABLE_INNER_RETRIES; ++retries) {
136 		data = mmio_read_32(base + HCE);
137 		if (data & HCE_ENABLE) {
138 			break;
139 		}
140 		udelay(HCE_ENABLE_TIMEOUT_US);
141 	}
142 	if (retries >= HCE_ENABLE_INNER_RETRIES) {
143 		return -ETIMEDOUT;
144 	}
145 
146 	return 0;
147 }
148 
149 static int ufshc_reset(uintptr_t base)
150 {
151 	unsigned int data;
152 	int retries, result;
153 
154 	for (retries = 0; retries < HCE_ENABLE_OUTER_RETRIES; ++retries) {
155 		result = ufshc_hce_enable(base);
156 		if (result == 0) {
157 			break;
158 		}
159 	}
160 	if (retries >= HCE_ENABLE_OUTER_RETRIES) {
161 		return -EIO;
162 	}
163 
164 	/* Enable Interrupts */
165 	data = UFS_INT_UCCS | UFS_INT_ULSS | UFS_INT_UE | UFS_INT_UTPES |
166 	       UFS_INT_DFES | UFS_INT_HCFES | UFS_INT_SBFES;
167 	mmio_write_32(base + IE, data);
168 
169 	return 0;
170 }
171 
172 static int ufshc_dme_link_startup(uintptr_t base)
173 {
174 	uic_cmd_t cmd;
175 
176 	memset(&cmd, 0, sizeof(cmd));
177 	cmd.op = DME_LINKSTARTUP;
178 	return ufshc_send_uic_cmd(base, &cmd);
179 }
180 
181 static int ufshc_link_startup(uintptr_t base)
182 {
183 	int data, result;
184 	int retries;
185 
186 	for (retries = DME_LINKSTARTUP_RETRIES; retries > 0; retries--) {
187 		result = ufshc_dme_link_startup(base);
188 		if (result != 0) {
189 			/* Reset controller before trying again */
190 			result = ufshc_reset(base);
191 			if (result != 0) {
192 				return result;
193 			}
194 			continue;
195 		}
196 		while ((mmio_read_32(base + HCS) & HCS_DP) == 0)
197 			;
198 		data = mmio_read_32(base + IS);
199 		if (data & UFS_INT_ULSS)
200 			mmio_write_32(base + IS, UFS_INT_ULSS);
201 		return 0;
202 	}
203 	return -EIO;
204 }
205 
206 /* Check Door Bell register to get an empty slot */
207 static int get_empty_slot(int *slot)
208 {
209 	unsigned int data;
210 	int i;
211 
212 	data = mmio_read_32(ufs_params.reg_base + UTRLDBR);
213 	for (i = 0; i < nutrs; i++) {
214 		if ((data & 1) == 0)
215 			break;
216 		data = data >> 1;
217 	}
218 	if (i >= nutrs)
219 		return -EBUSY;
220 	*slot = i;
221 	return 0;
222 }
223 
224 static void get_utrd(utp_utrd_t *utrd)
225 {
226 	uintptr_t base;
227 	int slot = 0, result;
228 	utrd_header_t *hd;
229 
230 	assert(utrd != NULL);
231 	result = get_empty_slot(&slot);
232 	assert(result == 0);
233 
234 	/* clear utrd */
235 	memset((void *)utrd, 0, sizeof(utp_utrd_t));
236 	base = ufs_params.desc_base + (slot * UFS_DESC_SIZE);
237 	/* clear the descriptor */
238 	memset((void *)base, 0, UFS_DESC_SIZE);
239 
240 	utrd->header = base;
241 	utrd->task_tag = slot + 1;
242 	/* CDB address should be aligned with 128 bytes */
243 	utrd->upiu = ALIGN_CDB(utrd->header + sizeof(utrd_header_t));
244 	utrd->resp_upiu = ALIGN_8(utrd->upiu + sizeof(cmd_upiu_t));
245 	utrd->size_upiu = utrd->resp_upiu - utrd->upiu;
246 	utrd->size_resp_upiu = ALIGN_8(sizeof(resp_upiu_t));
247 	utrd->prdt = utrd->resp_upiu + utrd->size_resp_upiu;
248 
249 	hd = (utrd_header_t *)utrd->header;
250 	hd->ucdba = utrd->upiu & UINT32_MAX;
251 	hd->ucdbau = (utrd->upiu >> 32) & UINT32_MAX;
252 	/* Both RUL and RUO is based on DWORD */
253 	hd->rul = utrd->size_resp_upiu >> 2;
254 	hd->ruo = utrd->size_upiu >> 2;
255 	(void)result;
256 }
257 
258 /*
259  * Prepare UTRD, Command UPIU, Response UPIU.
260  */
261 static int ufs_prepare_cmd(utp_utrd_t *utrd, uint8_t op, uint8_t lun,
262 			   int lba, uintptr_t buf, size_t length)
263 {
264 	utrd_header_t *hd;
265 	cmd_upiu_t *upiu;
266 	prdt_t *prdt;
267 	unsigned int ulba;
268 	unsigned int lba_cnt;
269 	int prdt_size;
270 
271 
272 	mmio_write_32(ufs_params.reg_base + UTRLBA,
273 		      utrd->header & UINT32_MAX);
274 	mmio_write_32(ufs_params.reg_base + UTRLBAU,
275 		      (utrd->upiu >> 32) & UINT32_MAX);
276 
277 	hd = (utrd_header_t *)utrd->header;
278 	upiu = (cmd_upiu_t *)utrd->upiu;
279 
280 	hd->i = 1;
281 	hd->ct = CT_UFS_STORAGE;
282 	hd->ocs = OCS_MASK;
283 
284 	upiu->trans_type = CMD_UPIU;
285 	upiu->task_tag = utrd->task_tag;
286 	upiu->cdb[0] = op;
287 	ulba = (unsigned int)lba;
288 	lba_cnt = (unsigned int)(length >> UFS_BLOCK_SHIFT);
289 	switch (op) {
290 	case CDBCMD_TEST_UNIT_READY:
291 		break;
292 	case CDBCMD_READ_CAPACITY_10:
293 		hd->dd = DD_OUT;
294 		upiu->flags = UPIU_FLAGS_R | UPIU_FLAGS_ATTR_S;
295 		upiu->lun = lun;
296 		break;
297 	case CDBCMD_READ_10:
298 		hd->dd = DD_OUT;
299 		upiu->flags = UPIU_FLAGS_R | UPIU_FLAGS_ATTR_S;
300 		upiu->lun = lun;
301 		upiu->cdb[1] = RW_WITHOUT_CACHE;
302 		/* set logical block address */
303 		upiu->cdb[2] = (ulba >> 24) & 0xff;
304 		upiu->cdb[3] = (ulba >> 16) & 0xff;
305 		upiu->cdb[4] = (ulba >> 8) & 0xff;
306 		upiu->cdb[5] = ulba & 0xff;
307 		/* set transfer length */
308 		upiu->cdb[7] = (lba_cnt >> 8) & 0xff;
309 		upiu->cdb[8] = lba_cnt & 0xff;
310 		break;
311 	case CDBCMD_WRITE_10:
312 		hd->dd = DD_IN;
313 		upiu->flags = UPIU_FLAGS_W | UPIU_FLAGS_ATTR_S;
314 		upiu->lun = lun;
315 		upiu->cdb[1] = RW_WITHOUT_CACHE;
316 		/* set logical block address */
317 		upiu->cdb[2] = (ulba >> 24) & 0xff;
318 		upiu->cdb[3] = (ulba >> 16) & 0xff;
319 		upiu->cdb[4] = (ulba >> 8) & 0xff;
320 		upiu->cdb[5] = ulba & 0xff;
321 		/* set transfer length */
322 		upiu->cdb[7] = (lba_cnt >> 8) & 0xff;
323 		upiu->cdb[8] = lba_cnt & 0xff;
324 		break;
325 	default:
326 		assert(0);
327 		break;
328 	}
329 	if (hd->dd == DD_IN)
330 		flush_dcache_range(buf, length);
331 	else if (hd->dd == DD_OUT)
332 		inv_dcache_range(buf, length);
333 	if (length) {
334 		upiu->exp_data_trans_len = htobe32(length);
335 		assert(lba_cnt <= UINT16_MAX);
336 		prdt = (prdt_t *)utrd->prdt;
337 
338 		prdt_size = 0;
339 		while (length > 0) {
340 			prdt->dba = (unsigned int)(buf & UINT32_MAX);
341 			prdt->dbau = (unsigned int)((buf >> 32) & UINT32_MAX);
342 			/* prdt->dbc counts from 0 */
343 			if (length > MAX_PRDT_SIZE) {
344 				prdt->dbc = MAX_PRDT_SIZE - 1;
345 				length = length - MAX_PRDT_SIZE;
346 			} else {
347 				prdt->dbc = length - 1;
348 				length = 0;
349 			}
350 			buf += MAX_PRDT_SIZE;
351 			prdt++;
352 			prdt_size += sizeof(prdt_t);
353 		}
354 		utrd->size_prdt = ALIGN_8(prdt_size);
355 		hd->prdtl = utrd->size_prdt >> 2;
356 		hd->prdto = (utrd->size_upiu + utrd->size_resp_upiu) >> 2;
357 	}
358 
359 	flush_dcache_range((uintptr_t)utrd, sizeof(utp_utrd_t));
360 	flush_dcache_range((uintptr_t)utrd->header, UFS_DESC_SIZE);
361 	return 0;
362 }
363 
364 static int ufs_prepare_query(utp_utrd_t *utrd, uint8_t op, uint8_t idn,
365 			     uint8_t index, uint8_t sel,
366 			     uintptr_t buf, size_t length)
367 {
368 	utrd_header_t *hd;
369 	query_upiu_t *query_upiu;
370 
371 
372 	hd = (utrd_header_t *)utrd->header;
373 	query_upiu = (query_upiu_t *)utrd->upiu;
374 
375 	mmio_write_32(ufs_params.reg_base + UTRLBA,
376 		      utrd->header & UINT32_MAX);
377 	mmio_write_32(ufs_params.reg_base + UTRLBAU,
378 		      (utrd->header >> 32) & UINT32_MAX);
379 
380 
381 	hd->i = 1;
382 	hd->ct = CT_UFS_STORAGE;
383 	hd->ocs = OCS_MASK;
384 
385 	query_upiu->trans_type = QUERY_REQUEST_UPIU;
386 	query_upiu->task_tag = utrd->task_tag;
387 	query_upiu->ts.desc.opcode = op;
388 	query_upiu->ts.desc.idn = idn;
389 	query_upiu->ts.desc.index = index;
390 	query_upiu->ts.desc.selector = sel;
391 	switch (op) {
392 	case QUERY_READ_DESC:
393 		query_upiu->query_func = QUERY_FUNC_STD_READ;
394 		query_upiu->ts.desc.length = htobe16(length);
395 		break;
396 	case QUERY_WRITE_DESC:
397 		query_upiu->query_func = QUERY_FUNC_STD_WRITE;
398 		query_upiu->ts.desc.length = htobe16(length);
399 		memcpy((void *)(utrd->upiu + sizeof(query_upiu_t)),
400 		       (void *)buf, length);
401 		break;
402 	case QUERY_READ_ATTR:
403 	case QUERY_READ_FLAG:
404 		query_upiu->query_func = QUERY_FUNC_STD_READ;
405 		break;
406 	case QUERY_CLEAR_FLAG:
407 	case QUERY_SET_FLAG:
408 		query_upiu->query_func = QUERY_FUNC_STD_WRITE;
409 		break;
410 	case QUERY_WRITE_ATTR:
411 		query_upiu->query_func = QUERY_FUNC_STD_WRITE;
412 		memcpy((void *)&query_upiu->ts.attr.value, (void *)buf, length);
413 		break;
414 	default:
415 		assert(0);
416 		break;
417 	}
418 	flush_dcache_range((uintptr_t)utrd, sizeof(utp_utrd_t));
419 	flush_dcache_range((uintptr_t)utrd->header, UFS_DESC_SIZE);
420 	return 0;
421 }
422 
423 static void ufs_prepare_nop_out(utp_utrd_t *utrd)
424 {
425 	utrd_header_t *hd;
426 	nop_out_upiu_t *nop_out;
427 
428 	mmio_write_32(ufs_params.reg_base + UTRLBA,
429 		      utrd->header & UINT32_MAX);
430 	mmio_write_32(ufs_params.reg_base + UTRLBAU,
431 		      (utrd->header >> 32) & UINT32_MAX);
432 
433 	hd = (utrd_header_t *)utrd->header;
434 	nop_out = (nop_out_upiu_t *)utrd->upiu;
435 
436 	hd->i = 1;
437 	hd->ct = CT_UFS_STORAGE;
438 	hd->ocs = OCS_MASK;
439 
440 	nop_out->trans_type = 0;
441 	nop_out->task_tag = utrd->task_tag;
442 	flush_dcache_range((uintptr_t)utrd, sizeof(utp_utrd_t));
443 	flush_dcache_range((uintptr_t)utrd->header, UFS_DESC_SIZE);
444 }
445 
446 static void ufs_send_request(int task_tag)
447 {
448 	unsigned int data;
449 	int slot;
450 
451 	slot = task_tag - 1;
452 	/* clear all interrupts */
453 	mmio_write_32(ufs_params.reg_base + IS, ~0);
454 
455 	mmio_write_32(ufs_params.reg_base + UTRLRSR, 1);
456 	do {
457 		data = mmio_read_32(ufs_params.reg_base + UTRLRSR);
458 	} while (data == 0);
459 
460 	data = UTRIACR_IAEN | UTRIACR_CTR | UTRIACR_IACTH(0x1F) |
461 	       UTRIACR_IATOVAL(0xFF);
462 	mmio_write_32(ufs_params.reg_base + UTRIACR, data);
463 	/* send request */
464 	mmio_setbits_32(ufs_params.reg_base + UTRLDBR, 1 << slot);
465 }
466 
467 static int ufs_check_resp(utp_utrd_t *utrd, int trans_type)
468 {
469 	utrd_header_t *hd;
470 	resp_upiu_t *resp;
471 	unsigned int data;
472 	int slot;
473 
474 	hd = (utrd_header_t *)utrd->header;
475 	resp = (resp_upiu_t *)utrd->resp_upiu;
476 	inv_dcache_range((uintptr_t)hd, UFS_DESC_SIZE);
477 	do {
478 		data = mmio_read_32(ufs_params.reg_base + IS);
479 		if ((data & ~(UFS_INT_UCCS | UFS_INT_UTRCS)) != 0)
480 			return -EIO;
481 	} while ((data & UFS_INT_UTRCS) == 0);
482 	slot = utrd->task_tag - 1;
483 
484 	data = mmio_read_32(ufs_params.reg_base + UTRLDBR);
485 	assert((data & (1 << slot)) == 0);
486 	assert(hd->ocs == OCS_SUCCESS);
487 	assert((resp->trans_type & TRANS_TYPE_CODE_MASK) == trans_type);
488 	(void)resp;
489 	(void)slot;
490 	return 0;
491 }
492 
493 static void ufs_send_cmd(utp_utrd_t *utrd, uint8_t cmd_op, uint8_t lun, int lba, uintptr_t buf,
494 			 size_t length)
495 {
496 	int result;
497 
498 	get_utrd(utrd);
499 
500 	result = ufs_prepare_cmd(utrd, cmd_op, lun, lba, buf, length);
501 	assert(result == 0);
502 	ufs_send_request(utrd->task_tag);
503 	result = ufs_check_resp(utrd, RESPONSE_UPIU);
504 	assert(result == 0);
505 	(void)result;
506 }
507 
508 #ifdef UFS_RESP_DEBUG
509 static void dump_upiu(utp_utrd_t *utrd)
510 {
511 	utrd_header_t *hd;
512 	int i;
513 
514 	hd = (utrd_header_t *)utrd->header;
515 	INFO("utrd:0x%x, ruo:0x%x, rul:0x%x, ocs:0x%x, UTRLDBR:0x%x\n",
516 		(unsigned int)(uintptr_t)utrd, hd->ruo, hd->rul, hd->ocs,
517 		mmio_read_32(ufs_params.reg_base + UTRLDBR));
518 	for (i = 0; i < sizeof(utrd_header_t); i += 4) {
519 		INFO("[%lx]:0x%x\n",
520 			(uintptr_t)utrd->header + i,
521 			*(unsigned int *)((uintptr_t)utrd->header + i));
522 	}
523 
524 	for (i = 0; i < sizeof(cmd_upiu_t); i += 4) {
525 		INFO("cmd[%lx]:0x%x\n",
526 			utrd->upiu + i,
527 			*(unsigned int *)(utrd->upiu + i));
528 	}
529 	for (i = 0; i < sizeof(resp_upiu_t); i += 4) {
530 		INFO("resp[%lx]:0x%x\n",
531 			utrd->resp_upiu + i,
532 			*(unsigned int *)(utrd->resp_upiu + i));
533 	}
534 	for (i = 0; i < sizeof(prdt_t); i += 4) {
535 		INFO("prdt[%lx]:0x%x\n",
536 			utrd->prdt + i,
537 			*(unsigned int *)(utrd->prdt + i));
538 	}
539 }
540 #endif
541 
542 static void ufs_verify_init(void)
543 {
544 	utp_utrd_t utrd;
545 	int result;
546 
547 	get_utrd(&utrd);
548 	ufs_prepare_nop_out(&utrd);
549 	ufs_send_request(utrd.task_tag);
550 	result = ufs_check_resp(&utrd, NOP_IN_UPIU);
551 	assert(result == 0);
552 	(void)result;
553 }
554 
555 static void ufs_verify_ready(void)
556 {
557 	utp_utrd_t utrd;
558 	ufs_send_cmd(&utrd, CDBCMD_TEST_UNIT_READY, 0, 0, 0, 0);
559 }
560 
561 static void ufs_query(uint8_t op, uint8_t idn, uint8_t index, uint8_t sel,
562 		      uintptr_t buf, size_t size)
563 {
564 	utp_utrd_t utrd;
565 	query_resp_upiu_t *resp;
566 	int result;
567 
568 	switch (op) {
569 	case QUERY_READ_FLAG:
570 	case QUERY_READ_ATTR:
571 	case QUERY_READ_DESC:
572 	case QUERY_WRITE_DESC:
573 	case QUERY_WRITE_ATTR:
574 		assert(((buf & 3) == 0) && (size != 0));
575 		break;
576 	default:
577 		/* Do nothing in default case */
578 		break;
579 	}
580 	get_utrd(&utrd);
581 	ufs_prepare_query(&utrd, op, idn, index, sel, buf, size);
582 	ufs_send_request(utrd.task_tag);
583 	result = ufs_check_resp(&utrd, QUERY_RESPONSE_UPIU);
584 	assert(result == 0);
585 	resp = (query_resp_upiu_t *)utrd.resp_upiu;
586 #ifdef UFS_RESP_DEBUG
587 	dump_upiu(&utrd);
588 #endif
589 	assert(resp->query_resp == QUERY_RESP_SUCCESS);
590 
591 	switch (op) {
592 	case QUERY_READ_FLAG:
593 		*(uint32_t *)buf = (uint32_t)resp->ts.flag.value;
594 		break;
595 	case QUERY_READ_ATTR:
596 	case QUERY_READ_DESC:
597 		memcpy((void *)buf,
598 		       (void *)(utrd.resp_upiu + sizeof(query_resp_upiu_t)),
599 		       size);
600 		break;
601 	default:
602 		/* Do nothing in default case */
603 		break;
604 	}
605 	(void)result;
606 }
607 
608 unsigned int ufs_read_attr(int idn)
609 {
610 	unsigned int value;
611 
612 	ufs_query(QUERY_READ_ATTR, idn, 0, 0,
613 		  (uintptr_t)&value, sizeof(value));
614 	return value;
615 }
616 
617 void ufs_write_attr(int idn, unsigned int value)
618 {
619 	ufs_query(QUERY_WRITE_ATTR, idn, 0, 0,
620 		  (uintptr_t)&value, sizeof(value));
621 }
622 
623 unsigned int ufs_read_flag(int idn)
624 {
625 	unsigned int value;
626 
627 	ufs_query(QUERY_READ_FLAG, idn, 0, 0,
628 		  (uintptr_t)&value, sizeof(value));
629 	return value;
630 }
631 
632 void ufs_set_flag(int idn)
633 {
634 	ufs_query(QUERY_SET_FLAG, idn, 0, 0, 0, 0);
635 }
636 
637 void ufs_clear_flag(int idn)
638 {
639 	ufs_query(QUERY_CLEAR_FLAG, idn, 0, 0, 0, 0);
640 }
641 
642 void ufs_read_desc(int idn, int index, uintptr_t buf, size_t size)
643 {
644 	ufs_query(QUERY_READ_DESC, idn, index, 0, buf, size);
645 }
646 
647 void ufs_write_desc(int idn, int index, uintptr_t buf, size_t size)
648 {
649 	ufs_query(QUERY_WRITE_DESC, idn, index, 0, buf, size);
650 }
651 
652 static void ufs_read_capacity(int lun, unsigned int *num, unsigned int *size)
653 {
654 	utp_utrd_t utrd;
655 	resp_upiu_t *resp;
656 	sense_data_t *sense;
657 	unsigned char data[CACHE_WRITEBACK_GRANULE << 1];
658 	uintptr_t buf;
659 	int result;
660 	int retry;
661 
662 	assert((ufs_params.reg_base != 0) &&
663 	       (ufs_params.desc_base != 0) &&
664 	       (ufs_params.desc_size >= UFS_DESC_SIZE) &&
665 	       (num != NULL) && (size != NULL));
666 
667 	/* align buf address */
668 	buf = (uintptr_t)data;
669 	buf = (buf + CACHE_WRITEBACK_GRANULE - 1) &
670 	      ~(CACHE_WRITEBACK_GRANULE - 1);
671 	memset((void *)buf, 0, CACHE_WRITEBACK_GRANULE);
672 	flush_dcache_range(buf, CACHE_WRITEBACK_GRANULE);
673 	do {
674 		ufs_send_cmd(&utrd, CDBCMD_READ_CAPACITY_10, lun, 0,
675 			    buf, READ_CAPACITY_LENGTH);
676 #ifdef UFS_RESP_DEBUG
677 		dump_upiu(&utrd);
678 #endif
679 		resp = (resp_upiu_t *)utrd.resp_upiu;
680 		retry = 0;
681 		sense = &resp->sd.sense;
682 		if (sense->resp_code == SENSE_DATA_VALID) {
683 			if ((sense->sense_key == SENSE_KEY_UNIT_ATTENTION) &&
684 			    (sense->asc == 0x29) && (sense->ascq == 0)) {
685 				retry = 1;
686 			}
687 		}
688 		inv_dcache_range(buf, CACHE_WRITEBACK_GRANULE);
689 		/* last logical block address */
690 		*num = be32toh(*(unsigned int *)buf);
691 		if (*num)
692 			*num += 1;
693 		/* logical block length in bytes */
694 		*size = be32toh(*(unsigned int *)(buf + 4));
695 	} while (retry);
696 	(void)result;
697 }
698 
699 size_t ufs_read_blocks(int lun, int lba, uintptr_t buf, size_t size)
700 {
701 	utp_utrd_t utrd;
702 	resp_upiu_t *resp;
703 	int result;
704 
705 	assert((ufs_params.reg_base != 0) &&
706 	       (ufs_params.desc_base != 0) &&
707 	       (ufs_params.desc_size >= UFS_DESC_SIZE));
708 
709 	ufs_send_cmd(&utrd, CDBCMD_READ_10, lun, lba, buf, size);
710 #ifdef UFS_RESP_DEBUG
711 	dump_upiu(&utrd);
712 #endif
713 	resp = (resp_upiu_t *)utrd.resp_upiu;
714 	(void)result;
715 	return size - resp->res_trans_cnt;
716 }
717 
718 size_t ufs_write_blocks(int lun, int lba, const uintptr_t buf, size_t size)
719 {
720 	utp_utrd_t utrd;
721 	resp_upiu_t *resp;
722 	int result;
723 
724 	assert((ufs_params.reg_base != 0) &&
725 	       (ufs_params.desc_base != 0) &&
726 	       (ufs_params.desc_size >= UFS_DESC_SIZE));
727 
728 	ufs_send_cmd(&utrd, CDBCMD_WRITE_10, lun, lba, buf, size);
729 #ifdef UFS_RESP_DEBUG
730 	dump_upiu(&utrd);
731 #endif
732 	resp = (resp_upiu_t *)utrd.resp_upiu;
733 	(void)result;
734 	return size - resp->res_trans_cnt;
735 }
736 
737 static void ufs_enum(void)
738 {
739 	unsigned int blk_num, blk_size;
740 	int i;
741 
742 	ufs_verify_init();
743 	ufs_verify_ready();
744 
745 	ufs_set_flag(FLAG_DEVICE_INIT);
746 	mdelay(200);
747 	/* dump available LUNs */
748 	for (i = 0; i < UFS_MAX_LUNS; i++) {
749 		ufs_read_capacity(i, &blk_num, &blk_size);
750 		if (blk_num && blk_size) {
751 			INFO("UFS LUN%d contains %d blocks with %d-byte size\n",
752 			     i, blk_num, blk_size);
753 		}
754 	}
755 }
756 
757 static void ufs_get_device_info(struct ufs_dev_desc *card_data)
758 {
759 	uint8_t desc_buf[DESC_DEVICE_MAX_SIZE];
760 
761 	ufs_query(QUERY_READ_DESC, DESC_TYPE_DEVICE, 0, 0,
762 				(uintptr_t)desc_buf, DESC_DEVICE_MAX_SIZE);
763 
764 	/*
765 	 * getting vendor (manufacturerID) and Bank Index in big endian
766 	 * format
767 	 */
768 	card_data->wmanufacturerid = (uint16_t)((desc_buf[DEVICE_DESC_PARAM_MANF_ID] << 8) |
769 				     (desc_buf[DEVICE_DESC_PARAM_MANF_ID + 1]));
770 }
771 
772 int ufs_init(const ufs_ops_t *ops, ufs_params_t *params)
773 {
774 	int result;
775 	unsigned int data;
776 	uic_cmd_t cmd;
777 	struct ufs_dev_desc card = {0};
778 
779 	assert((params != NULL) &&
780 	       (params->reg_base != 0) &&
781 	       (params->desc_base != 0) &&
782 	       (params->desc_size >= UFS_DESC_SIZE));
783 
784 	memcpy(&ufs_params, params, sizeof(ufs_params_t));
785 
786 	/* 0 means 1 slot */
787 	nutrs = (mmio_read_32(ufs_params.reg_base + CAP) & CAP_NUTRS_MASK) + 1;
788 	if (nutrs > (ufs_params.desc_size / UFS_DESC_SIZE)) {
789 		nutrs = ufs_params.desc_size / UFS_DESC_SIZE;
790 	}
791 
792 
793 	if (ufs_params.flags & UFS_FLAGS_SKIPINIT) {
794 		result = ufshc_dme_get(0x1571, 0, &data);
795 		assert(result == 0);
796 		result = ufshc_dme_get(0x41, 0, &data);
797 		assert(result == 0);
798 		if (data == 1) {
799 			/* prepare to exit hibernate mode */
800 			memset(&cmd, 0, sizeof(uic_cmd_t));
801 			cmd.op = DME_HIBERNATE_EXIT;
802 			result = ufshc_send_uic_cmd(ufs_params.reg_base,
803 						    &cmd);
804 			assert(result == 0);
805 			data = mmio_read_32(ufs_params.reg_base + UCMDARG2);
806 			assert(data == 0);
807 			do {
808 				data = mmio_read_32(ufs_params.reg_base + IS);
809 			} while ((data & UFS_INT_UHXS) == 0);
810 			mmio_write_32(ufs_params.reg_base + IS, UFS_INT_UHXS);
811 			data = mmio_read_32(ufs_params.reg_base + HCS);
812 			assert((data & HCS_UPMCRS_MASK) == HCS_PWR_LOCAL);
813 		}
814 		result = ufshc_dme_get(0x1568, 0, &data);
815 		assert(result == 0);
816 		assert((data > 0) && (data <= 3));
817 	} else {
818 		assert((ops != NULL) && (ops->phy_init != NULL) &&
819 		       (ops->phy_set_pwr_mode != NULL));
820 
821 		result = ufshc_reset(ufs_params.reg_base);
822 		assert(result == 0);
823 		ops->phy_init(&ufs_params);
824 		result = ufshc_link_startup(ufs_params.reg_base);
825 		assert(result == 0);
826 
827 		ufs_enum();
828 
829 		ufs_get_device_info(&card);
830 		if (card.wmanufacturerid == UFS_VENDOR_SKHYNIX) {
831 			ufs_params.flags |= UFS_FLAGS_VENDOR_SKHYNIX;
832 		}
833 
834 		ops->phy_set_pwr_mode(&ufs_params);
835 	}
836 
837 	(void)result;
838 	return 0;
839 }
840