sata.c 2.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128
  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. #include <common.h>
  12. #include <ahci.h>
  13. #include <dm.h>
  14. #include <command.h>
  15. #include <part.h>
  16. #include <sata.h>
  17. #include <dm/device-internal.h>
  18. #include <dm/uclass-internal.h>
  19. static int sata_curr_device = -1;
  20. int sata_remove(int devnum)
  21. {
  22. #ifdef CONFIG_AHCI
  23. struct udevice *dev;
  24. int rc;
  25. rc = uclass_find_device(UCLASS_AHCI, devnum, &dev);
  26. if (!rc && !dev)
  27. rc = uclass_find_first_device(UCLASS_AHCI, &dev);
  28. if (rc || !dev) {
  29. printf("Cannot find SATA device %d (err=%d)\n", devnum, rc);
  30. return CMD_RET_FAILURE;
  31. }
  32. rc = device_remove(dev, DM_REMOVE_NORMAL);
  33. if (rc) {
  34. printf("Cannot remove SATA device '%s' (err=%d)\n", dev->name,
  35. rc);
  36. return CMD_RET_FAILURE;
  37. }
  38. return 0;
  39. #else
  40. return sata_stop();
  41. #endif
  42. }
  43. int sata_probe(int devnum)
  44. {
  45. #ifdef CONFIG_AHCI
  46. struct udevice *dev;
  47. struct udevice *blk;
  48. int rc;
  49. rc = uclass_get_device(UCLASS_AHCI, devnum, &dev);
  50. if (rc)
  51. rc = uclass_find_first_device(UCLASS_AHCI, &dev);
  52. if (rc) {
  53. printf("Cannot probe SATA device %d (err=%d)\n", devnum, rc);
  54. return CMD_RET_FAILURE;
  55. }
  56. rc = sata_scan(dev);
  57. if (rc) {
  58. printf("Cannot scan SATA device %d (err=%d)\n", devnum, rc);
  59. return CMD_RET_FAILURE;
  60. }
  61. rc = blk_get_from_parent(dev, &blk);
  62. if (!rc) {
  63. struct blk_desc *desc = dev_get_uclass_platdata(blk);
  64. if (desc->lba > 0 && desc->blksz > 0)
  65. part_init(desc);
  66. }
  67. return 0;
  68. #else
  69. return sata_initialize() < 0 ? CMD_RET_FAILURE : CMD_RET_SUCCESS;
  70. #endif
  71. }
  72. static int do_sata(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
  73. {
  74. int rc = 0;
  75. if (argc >= 2) {
  76. int devnum = 0;
  77. if (argc == 3)
  78. devnum = (int)simple_strtoul(argv[2], NULL, 10);
  79. if (!strcmp(argv[1], "stop"))
  80. return sata_remove(devnum);
  81. if (!strcmp(argv[1], "init")) {
  82. if (sata_curr_device != -1) {
  83. rc = sata_remove(devnum);
  84. if (rc)
  85. return rc;
  86. }
  87. return sata_probe(devnum);
  88. }
  89. }
  90. /* If the user has not yet run `sata init`, do it now */
  91. if (sata_curr_device == -1) {
  92. rc = sata_probe(0);
  93. if (rc)
  94. return rc;
  95. sata_curr_device = 0;
  96. }
  97. return blk_common_cmd(argc, argv, IF_TYPE_SATA, &sata_curr_device);
  98. }
  99. U_BOOT_CMD(
  100. sata, 5, 1, do_sata,
  101. "SATA sub system",
  102. "init - init SATA sub system\n"
  103. "sata stop [dev] - disable SATA sub system or device\n"
  104. "sata info - show available SATA devices\n"
  105. "sata device [dev] - show or set current device\n"
  106. "sata part [dev] - print partition table\n"
  107. "sata read addr blk# cnt\n"
  108. "sata write addr blk# cnt"
  109. );