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