ata: ahci: fix memory leak
[platform/kernel/u-boot.git] / cmd / blk_common.c
1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * Handling of common block commands
4  *
5  * Copyright (c) 2017 Google, Inc
6  *
7  * (C) Copyright 2000-2011
8  * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
9  */
10
11 #include <common.h>
12 #include <blk.h>
13
14 #ifdef CONFIG_HAVE_BLOCK_DEVICE
15 int blk_common_cmd(int argc, char * const argv[], enum if_type if_type,
16                    int *cur_devnump)
17 {
18         const char *if_name = blk_get_if_type_name(if_type);
19
20         switch (argc) {
21         case 0:
22         case 1:
23                 return CMD_RET_USAGE;
24         case 2:
25                 if (strncmp(argv[1], "inf", 3) == 0) {
26                         blk_list_devices(if_type);
27                         return 0;
28                 } else if (strncmp(argv[1], "dev", 3) == 0) {
29                         if (blk_print_device_num(if_type, *cur_devnump)) {
30                                 printf("\nno %s devices available\n", if_name);
31                                 return CMD_RET_FAILURE;
32                         }
33                         return 0;
34                 } else if (strncmp(argv[1], "part", 4) == 0) {
35                         if (blk_list_part(if_type))
36                                 printf("\nno %s devices available\n", if_name);
37                         return 0;
38                 }
39                 return CMD_RET_USAGE;
40         case 3:
41                 if (strncmp(argv[1], "dev", 3) == 0) {
42                         int dev = (int)simple_strtoul(argv[2], NULL, 10);
43
44                         if (!blk_show_device(if_type, dev)) {
45                                 *cur_devnump = dev;
46                                 printf("... is now current device\n");
47                         } else {
48                                 return CMD_RET_FAILURE;
49                         }
50                         return 0;
51                 } else if (strncmp(argv[1], "part", 4) == 0) {
52                         int dev = (int)simple_strtoul(argv[2], NULL, 10);
53
54                         if (blk_print_part_devnum(if_type, dev)) {
55                                 printf("\n%s device %d not available\n",
56                                        if_name, dev);
57                                 return CMD_RET_FAILURE;
58                         }
59                         return 0;
60                 }
61                 return CMD_RET_USAGE;
62
63         default: /* at least 4 args */
64                 if (strcmp(argv[1], "read") == 0) {
65                         ulong addr = simple_strtoul(argv[2], NULL, 16);
66                         lbaint_t blk = simple_strtoul(argv[3], NULL, 16);
67                         ulong cnt = simple_strtoul(argv[4], NULL, 16);
68                         ulong n;
69
70                         printf("\n%s read: device %d block # "LBAFU", count %lu ... ",
71                                if_name, *cur_devnump, blk, cnt);
72
73                         n = blk_read_devnum(if_type, *cur_devnump, blk, cnt,
74                                             (ulong *)addr);
75
76                         printf("%ld blocks read: %s\n", n,
77                                n == cnt ? "OK" : "ERROR");
78                         return n == cnt ? 0 : 1;
79                 } else if (strcmp(argv[1], "write") == 0) {
80                         ulong addr = simple_strtoul(argv[2], NULL, 16);
81                         lbaint_t blk = simple_strtoul(argv[3], NULL, 16);
82                         ulong cnt = simple_strtoul(argv[4], NULL, 16);
83                         ulong n;
84
85                         printf("\n%s write: device %d block # "LBAFU", count %lu ... ",
86                                if_name, *cur_devnump, blk, cnt);
87
88                         n = blk_write_devnum(if_type, *cur_devnump, blk, cnt,
89                                              (ulong *)addr);
90
91                         printf("%ld blocks written: %s\n", n,
92                                n == cnt ? "OK" : "ERROR");
93                         return n == cnt ? 0 : 1;
94                 } else {
95                         return CMD_RET_USAGE;
96                 }
97         }
98 }
99 #endif