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