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