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 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 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 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