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 <blk.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 22static int sata_curr_device = -1; 23 24int sata_remove(int devnum) 25{ 26#ifdef CONFIG_AHCI 27 struct udevice *dev; 28 int rc; 29 30 blk_unbind_all(UCLASS_AHCI); 31 32 rc = uclass_find_device(UCLASS_AHCI, devnum, &dev); 33 if (!rc && !dev) 34 rc = uclass_find_first_device(UCLASS_AHCI, &dev); 35 if (rc || !dev) { 36 printf("Cannot find SATA device %d (err=%d)\n", devnum, rc); 37 return CMD_RET_FAILURE; 38 } 39 40 rc = device_remove(dev, DM_REMOVE_NORMAL); 41 if (rc) { 42 printf("Cannot remove SATA device '%s' (err=%d)\n", dev->name, 43 rc); 44 return CMD_RET_FAILURE; 45 } 46 47 return 0; 48#else 49 return sata_stop(); 50#endif 51} 52 53int sata_probe(int devnum) 54{ 55#ifdef CONFIG_AHCI 56 struct udevice *dev; 57 int rc; 58 59 rc = uclass_get_device(UCLASS_AHCI, devnum, &dev); 60 if (rc) 61 rc = uclass_find_first_device(UCLASS_AHCI, &dev); 62 if (rc) { 63 printf("Cannot probe SATA device %d (err=%d)\n", devnum, rc); 64 return CMD_RET_FAILURE; 65 } 66 if (!dev) { 67 printf("No SATA device found!\n"); 68 return CMD_RET_FAILURE; 69 } 70 rc = sata_scan(dev); 71 if (rc) { 72 printf("Cannot scan SATA device %d (err=%d)\n", devnum, rc); 73 return CMD_RET_FAILURE; 74 } 75 76 return 0; 77#else 78 return sata_initialize() < 0 ? CMD_RET_FAILURE : CMD_RET_SUCCESS; 79#endif 80} 81 82static int do_sata(struct cmd_tbl *cmdtp, int flag, int argc, 83 char *const argv[]) 84{ 85 int rc = 0; 86 87 if (argc >= 2) { 88 int devnum = 0; 89 90 if (argc == 3) 91 devnum = (int)dectoul(argv[2], NULL); 92 if (!strcmp(argv[1], "stop")) 93 return sata_remove(devnum); 94 95 if (!strcmp(argv[1], "init")) { 96 if (sata_curr_device != -1) { 97 rc = sata_remove(devnum); 98 if (rc) 99 return rc; 100 } 101 102 return sata_probe(devnum); 103 } 104 } 105 106 /* If the user has not yet run `sata init`, do it now */ 107 if (sata_curr_device == -1) { 108 rc = sata_probe(0); 109 if (rc) 110 return rc; 111 sata_curr_device = 0; 112 } 113 114 return blk_common_cmd(argc, argv, UCLASS_AHCI, &sata_curr_device); 115} 116 117U_BOOT_CMD( 118 sata, 5, 1, do_sata, 119 "SATA sub system", 120 "init - init SATA sub system\n" 121 "sata stop [dev] - disable SATA sub system or device\n" 122 "sata info - show available SATA devices\n" 123 "sata device [dev] - show or set current device\n" 124 "sata part [dev] - print partition table\n" 125 "sata read addr blk# cnt\n" 126 "sata write addr blk# cnt" 127); 128