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 #include <command.h>
  14
  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 partition table available\n",
  37                                        if_name);
  38                         return 0;
  39                 }
  40                 return CMD_RET_USAGE;
  41         case 3:
  42                 if (strncmp(argv[1], "dev", 3) == 0) {
  43                         int dev = (int)simple_strtoul(argv[2], NULL, 10);
  44
  45                         if (!blk_show_device(if_type, dev)) {
  46                                 *cur_devnump = dev;
  47                                 printf("... is now current device\n");
  48                         } else {
  49                                 return CMD_RET_FAILURE;
  50                         }
  51                         return 0;
  52                 } else if (strncmp(argv[1], "part", 4) == 0) {
  53                         int dev = (int)simple_strtoul(argv[2], NULL, 10);
  54
  55                         if (blk_print_part_devnum(if_type, dev)) {
  56                                 printf("\n%s device %d not available\n",
  57                                        if_name, dev);
  58                                 return CMD_RET_FAILURE;
  59                         }
  60                         return 0;
  61                 }
  62                 return CMD_RET_USAGE;
  63
  64         default: /* at least 4 args */
  65                 if (strcmp(argv[1], "read") == 0) {
  66                         ulong addr = simple_strtoul(argv[2], NULL, 16);
  67                         lbaint_t blk = simple_strtoul(argv[3], NULL, 16);
  68                         ulong cnt = simple_strtoul(argv[4], NULL, 16);
  69                         ulong n;
  70
  71                         printf("\n%s read: device %d block # "LBAFU", count %lu ... ",
  72                                if_name, *cur_devnump, blk, cnt);
  73
  74                         n = blk_read_devnum(if_type, *cur_devnump, blk, cnt,
  75                                             (ulong *)addr);
  76
  77                         printf("%ld blocks read: %s\n", n,
  78                                n == cnt ? "OK" : "ERROR");
  79                         return n == cnt ? 0 : 1;
  80                 } else if (strcmp(argv[1], "write") == 0) {
  81                         ulong addr = simple_strtoul(argv[2], NULL, 16);
  82                         lbaint_t blk = simple_strtoul(argv[3], NULL, 16);
  83                         ulong cnt = simple_strtoul(argv[4], NULL, 16);
  84                         ulong n;
  85
  86                         printf("\n%s write: device %d block # "LBAFU", count %lu ... ",
  87                                if_name, *cur_devnump, blk, cnt);
  88
  89                         n = blk_write_devnum(if_type, *cur_devnump, blk, cnt,
  90                                              (ulong *)addr);
  91
  92                         printf("%ld blocks written: %s\n", n,
  93                                n == cnt ? "OK" : "ERROR");
  94                         return n == cnt ? 0 : 1;
  95                 } else {
  96                         return CMD_RET_USAGE;
  97                 }
  98         }
  99 }