12e192b24SSimon Glass /*
22e192b24SSimon Glass * Copyright (C) 2000-2005, DENX Software Engineering
32e192b24SSimon Glass * Wolfgang Denk <wd@denx.de>
42e192b24SSimon Glass * Copyright (C) Procsys. All rights reserved.
52e192b24SSimon Glass * Mushtaq Khan <mushtaq_k@procsys.com>
62e192b24SSimon Glass * <mushtaqk_921@yahoo.co.in>
72e192b24SSimon Glass * Copyright (C) 2008 Freescale Semiconductor, Inc.
82e192b24SSimon Glass * Dave Liu <daveliu@freescale.com>
92e192b24SSimon Glass *
102e192b24SSimon Glass * SPDX-License-Identifier: GPL-2.0+
112e192b24SSimon Glass */
122e192b24SSimon Glass
132e192b24SSimon Glass #include <common.h>
14*f19f1ecbSSimon Glass #include <ahci.h>
15*f19f1ecbSSimon Glass #include <dm.h>
162e192b24SSimon Glass #include <command.h>
172e192b24SSimon Glass #include <part.h>
182e192b24SSimon Glass #include <sata.h>
19*f19f1ecbSSimon Glass #include <dm/device-internal.h>
20*f19f1ecbSSimon Glass #include <dm/uclass-internal.h>
212e192b24SSimon Glass
222e192b24SSimon Glass static int sata_curr_device = -1;
232e192b24SSimon Glass
sata_remove(int devnum)24*f19f1ecbSSimon Glass int sata_remove(int devnum)
25*f19f1ecbSSimon Glass {
26*f19f1ecbSSimon Glass #ifdef CONFIG_AHCI
27*f19f1ecbSSimon Glass struct udevice *dev;
28*f19f1ecbSSimon Glass int rc;
29*f19f1ecbSSimon Glass
30*f19f1ecbSSimon Glass rc = uclass_find_device(UCLASS_AHCI, devnum, &dev);
31*f19f1ecbSSimon Glass if (!rc && !dev)
32*f19f1ecbSSimon Glass rc = uclass_find_first_device(UCLASS_AHCI, &dev);
33*f19f1ecbSSimon Glass if (rc || !dev) {
34*f19f1ecbSSimon Glass printf("Cannot find SATA device %d (err=%d)\n", devnum, rc);
35*f19f1ecbSSimon Glass return CMD_RET_FAILURE;
36*f19f1ecbSSimon Glass }
37*f19f1ecbSSimon Glass
38*f19f1ecbSSimon Glass rc = device_remove(dev, DM_REMOVE_NORMAL);
39*f19f1ecbSSimon Glass if (rc) {
40*f19f1ecbSSimon Glass printf("Cannot remove SATA device '%s' (err=%d)\n", dev->name,
41*f19f1ecbSSimon Glass rc);
42*f19f1ecbSSimon Glass return CMD_RET_FAILURE;
43*f19f1ecbSSimon Glass }
44*f19f1ecbSSimon Glass
45*f19f1ecbSSimon Glass return 0;
46*f19f1ecbSSimon Glass #else
47*f19f1ecbSSimon Glass return sata_stop();
48*f19f1ecbSSimon Glass #endif
49*f19f1ecbSSimon Glass }
50*f19f1ecbSSimon Glass
sata_probe(int devnum)51*f19f1ecbSSimon Glass int sata_probe(int devnum)
52*f19f1ecbSSimon Glass {
53*f19f1ecbSSimon Glass #ifdef CONFIG_AHCI
54*f19f1ecbSSimon Glass struct udevice *dev;
55*f19f1ecbSSimon Glass struct udevice *blk;
56*f19f1ecbSSimon Glass int rc;
57*f19f1ecbSSimon Glass
58*f19f1ecbSSimon Glass rc = uclass_get_device(UCLASS_AHCI, devnum, &dev);
59*f19f1ecbSSimon Glass if (rc)
60*f19f1ecbSSimon Glass rc = uclass_find_first_device(UCLASS_AHCI, &dev);
61*f19f1ecbSSimon Glass if (rc) {
62*f19f1ecbSSimon Glass printf("Cannot probe SATA device %d (err=%d)\n", devnum, rc);
63*f19f1ecbSSimon Glass return CMD_RET_FAILURE;
64*f19f1ecbSSimon Glass }
65*f19f1ecbSSimon Glass rc = sata_scan(dev);
66*f19f1ecbSSimon Glass if (rc) {
67*f19f1ecbSSimon Glass printf("Cannot scan SATA device %d (err=%d)\n", devnum, rc);
68*f19f1ecbSSimon Glass return CMD_RET_FAILURE;
69*f19f1ecbSSimon Glass }
70*f19f1ecbSSimon Glass
71*f19f1ecbSSimon Glass rc = blk_get_from_parent(dev, &blk);
72*f19f1ecbSSimon Glass if (!rc) {
73*f19f1ecbSSimon Glass struct blk_desc *desc = dev_get_uclass_platdata(blk);
74*f19f1ecbSSimon Glass
75*f19f1ecbSSimon Glass if (desc->lba > 0 && desc->blksz > 0)
76*f19f1ecbSSimon Glass part_init(desc);
77*f19f1ecbSSimon Glass }
78*f19f1ecbSSimon Glass
79*f19f1ecbSSimon Glass return 0;
80*f19f1ecbSSimon Glass #else
81*f19f1ecbSSimon Glass return sata_initialize() < 0 ? CMD_RET_FAILURE : CMD_RET_SUCCESS;
82*f19f1ecbSSimon Glass #endif
83*f19f1ecbSSimon Glass }
84*f19f1ecbSSimon Glass
do_sata(cmd_tbl_t * cmdtp,int flag,int argc,char * const argv[])852e192b24SSimon Glass static int do_sata(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
862e192b24SSimon Glass {
872e192b24SSimon Glass int rc = 0;
882e192b24SSimon Glass
89*f19f1ecbSSimon Glass if (argc >= 2) {
90*f19f1ecbSSimon Glass int devnum = 0;
912e192b24SSimon Glass
92*f19f1ecbSSimon Glass if (argc == 3)
93*f19f1ecbSSimon Glass devnum = (int)simple_strtoul(argv[2], NULL, 10);
94*f19f1ecbSSimon Glass if (!strcmp(argv[1], "stop"))
95*f19f1ecbSSimon Glass return sata_remove(devnum);
962e192b24SSimon Glass
97*f19f1ecbSSimon Glass if (!strcmp(argv[1], "init")) {
98*f19f1ecbSSimon Glass if (sata_curr_device != -1) {
99*f19f1ecbSSimon Glass rc = sata_remove(devnum);
100*f19f1ecbSSimon Glass if (rc)
101*f19f1ecbSSimon Glass return rc;
102*f19f1ecbSSimon Glass }
103*f19f1ecbSSimon Glass
104*f19f1ecbSSimon Glass return sata_probe(devnum);
105*f19f1ecbSSimon Glass }
1062e192b24SSimon Glass }
1072e192b24SSimon Glass
1082e192b24SSimon Glass /* If the user has not yet run `sata init`, do it now */
109aa6ab905STang Yuantian if (sata_curr_device == -1) {
110*f19f1ecbSSimon Glass rc = sata_probe(0);
111*f19f1ecbSSimon Glass if (rc < 0)
1128547f45bSGary Bisson return CMD_RET_FAILURE;
113*f19f1ecbSSimon Glass sata_curr_device = 0;
114aa6ab905STang Yuantian }
1152e192b24SSimon Glass
116e29e71e9SSimon Glass return blk_common_cmd(argc, argv, IF_TYPE_SATA, &sata_curr_device);
1172e192b24SSimon Glass }
1182e192b24SSimon Glass
1192e192b24SSimon Glass U_BOOT_CMD(
1202e192b24SSimon Glass sata, 5, 1, do_sata,
1212e192b24SSimon Glass "SATA sub system",
1222e192b24SSimon Glass "init - init SATA sub system\n"
123*f19f1ecbSSimon Glass "sata stop [dev] - disable SATA sub system or device\n"
1242e192b24SSimon Glass "sata info - show available SATA devices\n"
1252e192b24SSimon Glass "sata device [dev] - show or set current device\n"
1262e192b24SSimon Glass "sata part [dev] - print partition table\n"
1272e192b24SSimon Glass "sata read addr blk# cnt\n"
1282e192b24SSimon Glass "sata write addr blk# cnt"
1292e192b24SSimon Glass );
130