1*4882a593Smuzhiyun /*
2*4882a593Smuzhiyun * Copyright (C) 2000-2005, DENX Software Engineering
3*4882a593Smuzhiyun * Wolfgang Denk <wd@denx.de>
4*4882a593Smuzhiyun * Copyright (C) Procsys. All rights reserved.
5*4882a593Smuzhiyun * Mushtaq Khan <mushtaq_k@procsys.com>
6*4882a593Smuzhiyun * <mushtaqk_921@yahoo.co.in>
7*4882a593Smuzhiyun * Copyright (C) 2008 Freescale Semiconductor, Inc.
8*4882a593Smuzhiyun * Dave Liu <daveliu@freescale.com>
9*4882a593Smuzhiyun *
10*4882a593Smuzhiyun * SPDX-License-Identifier: GPL-2.0+
11*4882a593Smuzhiyun */
12*4882a593Smuzhiyun
13*4882a593Smuzhiyun #include <common.h>
14*4882a593Smuzhiyun #include <ahci.h>
15*4882a593Smuzhiyun #include <dm.h>
16*4882a593Smuzhiyun #include <command.h>
17*4882a593Smuzhiyun #include <part.h>
18*4882a593Smuzhiyun #include <sata.h>
19*4882a593Smuzhiyun #include <dm/device-internal.h>
20*4882a593Smuzhiyun #include <dm/uclass-internal.h>
21*4882a593Smuzhiyun
22*4882a593Smuzhiyun static int sata_curr_device = -1;
23*4882a593Smuzhiyun
sata_remove(int devnum)24*4882a593Smuzhiyun int sata_remove(int devnum)
25*4882a593Smuzhiyun {
26*4882a593Smuzhiyun #ifdef CONFIG_AHCI
27*4882a593Smuzhiyun struct udevice *dev;
28*4882a593Smuzhiyun int rc;
29*4882a593Smuzhiyun
30*4882a593Smuzhiyun rc = uclass_find_device(UCLASS_AHCI, devnum, &dev);
31*4882a593Smuzhiyun if (!rc && !dev)
32*4882a593Smuzhiyun rc = uclass_find_first_device(UCLASS_AHCI, &dev);
33*4882a593Smuzhiyun if (rc || !dev) {
34*4882a593Smuzhiyun printf("Cannot find SATA device %d (err=%d)\n", devnum, rc);
35*4882a593Smuzhiyun return CMD_RET_FAILURE;
36*4882a593Smuzhiyun }
37*4882a593Smuzhiyun
38*4882a593Smuzhiyun rc = device_remove(dev, DM_REMOVE_NORMAL);
39*4882a593Smuzhiyun if (rc) {
40*4882a593Smuzhiyun printf("Cannot remove SATA device '%s' (err=%d)\n", dev->name,
41*4882a593Smuzhiyun rc);
42*4882a593Smuzhiyun return CMD_RET_FAILURE;
43*4882a593Smuzhiyun }
44*4882a593Smuzhiyun
45*4882a593Smuzhiyun return 0;
46*4882a593Smuzhiyun #else
47*4882a593Smuzhiyun return sata_stop();
48*4882a593Smuzhiyun #endif
49*4882a593Smuzhiyun }
50*4882a593Smuzhiyun
sata_probe(int devnum)51*4882a593Smuzhiyun int sata_probe(int devnum)
52*4882a593Smuzhiyun {
53*4882a593Smuzhiyun #ifdef CONFIG_AHCI
54*4882a593Smuzhiyun struct udevice *dev;
55*4882a593Smuzhiyun struct udevice *blk;
56*4882a593Smuzhiyun int rc;
57*4882a593Smuzhiyun
58*4882a593Smuzhiyun rc = uclass_get_device(UCLASS_AHCI, devnum, &dev);
59*4882a593Smuzhiyun if (rc)
60*4882a593Smuzhiyun rc = uclass_find_first_device(UCLASS_AHCI, &dev);
61*4882a593Smuzhiyun if (rc) {
62*4882a593Smuzhiyun printf("Cannot probe SATA device %d (err=%d)\n", devnum, rc);
63*4882a593Smuzhiyun return CMD_RET_FAILURE;
64*4882a593Smuzhiyun }
65*4882a593Smuzhiyun rc = sata_scan(dev);
66*4882a593Smuzhiyun if (rc) {
67*4882a593Smuzhiyun printf("Cannot scan SATA device %d (err=%d)\n", devnum, rc);
68*4882a593Smuzhiyun return CMD_RET_FAILURE;
69*4882a593Smuzhiyun }
70*4882a593Smuzhiyun
71*4882a593Smuzhiyun rc = blk_get_from_parent(dev, &blk);
72*4882a593Smuzhiyun if (!rc) {
73*4882a593Smuzhiyun struct blk_desc *desc = dev_get_uclass_platdata(blk);
74*4882a593Smuzhiyun
75*4882a593Smuzhiyun if (desc->lba > 0 && desc->blksz > 0)
76*4882a593Smuzhiyun part_init(desc);
77*4882a593Smuzhiyun }
78*4882a593Smuzhiyun
79*4882a593Smuzhiyun return 0;
80*4882a593Smuzhiyun #else
81*4882a593Smuzhiyun return sata_initialize() < 0 ? CMD_RET_FAILURE : CMD_RET_SUCCESS;
82*4882a593Smuzhiyun #endif
83*4882a593Smuzhiyun }
84*4882a593Smuzhiyun
do_sata(cmd_tbl_t * cmdtp,int flag,int argc,char * const argv[])85*4882a593Smuzhiyun static int do_sata(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
86*4882a593Smuzhiyun {
87*4882a593Smuzhiyun int rc = 0;
88*4882a593Smuzhiyun
89*4882a593Smuzhiyun if (argc >= 2) {
90*4882a593Smuzhiyun int devnum = 0;
91*4882a593Smuzhiyun
92*4882a593Smuzhiyun if (argc == 3)
93*4882a593Smuzhiyun devnum = (int)simple_strtoul(argv[2], NULL, 10);
94*4882a593Smuzhiyun if (!strcmp(argv[1], "stop"))
95*4882a593Smuzhiyun return sata_remove(devnum);
96*4882a593Smuzhiyun
97*4882a593Smuzhiyun if (!strcmp(argv[1], "init")) {
98*4882a593Smuzhiyun if (sata_curr_device != -1) {
99*4882a593Smuzhiyun rc = sata_remove(devnum);
100*4882a593Smuzhiyun if (rc)
101*4882a593Smuzhiyun return rc;
102*4882a593Smuzhiyun }
103*4882a593Smuzhiyun
104*4882a593Smuzhiyun return sata_probe(devnum);
105*4882a593Smuzhiyun }
106*4882a593Smuzhiyun }
107*4882a593Smuzhiyun
108*4882a593Smuzhiyun /* If the user has not yet run `sata init`, do it now */
109*4882a593Smuzhiyun if (sata_curr_device == -1) {
110*4882a593Smuzhiyun rc = sata_probe(0);
111*4882a593Smuzhiyun if (rc < 0)
112*4882a593Smuzhiyun return CMD_RET_FAILURE;
113*4882a593Smuzhiyun sata_curr_device = 0;
114*4882a593Smuzhiyun }
115*4882a593Smuzhiyun
116*4882a593Smuzhiyun return blk_common_cmd(argc, argv, IF_TYPE_SATA, &sata_curr_device);
117*4882a593Smuzhiyun }
118*4882a593Smuzhiyun
119*4882a593Smuzhiyun U_BOOT_CMD(
120*4882a593Smuzhiyun sata, 5, 1, do_sata,
121*4882a593Smuzhiyun "SATA sub system",
122*4882a593Smuzhiyun "init - init SATA sub system\n"
123*4882a593Smuzhiyun "sata stop [dev] - disable SATA sub system or device\n"
124*4882a593Smuzhiyun "sata info - show available SATA devices\n"
125*4882a593Smuzhiyun "sata device [dev] - show or set current device\n"
126*4882a593Smuzhiyun "sata part [dev] - print partition table\n"
127*4882a593Smuzhiyun "sata read addr blk# cnt\n"
128*4882a593Smuzhiyun "sata write addr blk# cnt"
129*4882a593Smuzhiyun );
130