1 /* 2 * Copyright (C) 2017 NXP Semiconductors 3 * Copyright (C) 2017 Bin Meng <bmeng.cn@gmail.com> 4 * 5 * SPDX-License-Identifier: GPL-2.0+ 6 */ 7 8 #include <common.h> 9 #include <command.h> 10 #include <dm.h> 11 #include <nvme.h> 12 #include <part.h> 13 #include <linux/math64.h> 14 15 static int nvme_curr_device; 16 17 static int do_nvme_scan(cmd_tbl_t *cmdtp, int flag, 18 int argc, char * const argv[]) 19 { 20 int ret; 21 22 ret = nvme_scan_namespace(); 23 if (ret) 24 return CMD_RET_FAILURE; 25 26 return CMD_RET_SUCCESS; 27 } 28 29 static int do_nvme_list(cmd_tbl_t *cmdtp, int flag, 30 int argc, char * const argv[]) 31 { 32 blk_list_devices(IF_TYPE_NVME); 33 34 return CMD_RET_SUCCESS; 35 } 36 37 static int do_nvme_info(cmd_tbl_t *cmdtp, int flag, 38 int argc, char * const argv[]) 39 { 40 int devnum; 41 struct udevice *udev; 42 int ret; 43 44 if (argc > 1) 45 devnum = (int)simple_strtoul(argv[1], NULL, 10); 46 else 47 devnum = nvme_curr_device; 48 49 ret = blk_get_device(IF_TYPE_NVME, devnum, &udev); 50 if (ret < 0) 51 return CMD_RET_FAILURE; 52 53 nvme_print_info(udev); 54 55 return CMD_RET_SUCCESS; 56 } 57 58 static int do_nvme_device(cmd_tbl_t *cmdtp, int flag, 59 int argc, char * const argv[]) 60 { 61 if (argc > 1) { 62 int devnum = (int)simple_strtoul(argv[1], NULL, 10); 63 64 if (!blk_show_device(IF_TYPE_NVME, devnum)) { 65 nvme_curr_device = devnum; 66 printf("... is now current device\n"); 67 } else { 68 return CMD_RET_FAILURE; 69 } 70 } else { 71 blk_show_device(IF_TYPE_NVME, nvme_curr_device); 72 } 73 74 return CMD_RET_SUCCESS; 75 } 76 77 static int do_nvme_part(cmd_tbl_t *cmdtp, int flag, 78 int argc, char * const argv[]) 79 { 80 if (argc > 1) { 81 int devnum = (int)simple_strtoul(argv[2], NULL, 10); 82 83 if (blk_print_part_devnum(IF_TYPE_NVME, devnum)) { 84 printf("\nNVMe device %d not available\n", devnum); 85 return CMD_RET_FAILURE; 86 } 87 } else { 88 blk_print_part_devnum(IF_TYPE_NVME, nvme_curr_device); 89 } 90 91 return CMD_RET_SUCCESS; 92 } 93 94 static int do_nvme_read(cmd_tbl_t *cmdtp, int flag, int argc, 95 char * const argv[]) 96 { 97 unsigned long time; 98 if (argc != 4) 99 return CMD_RET_USAGE; 100 101 ulong addr = simple_strtoul(argv[1], NULL, 16); 102 ulong cnt = simple_strtoul(argv[3], NULL, 16); 103 ulong n; 104 lbaint_t blk = simple_strtoul(argv[2], NULL, 16); 105 106 printf("\nNVMe read: device %d block # " LBAFU " count %ld ... ", 107 nvme_curr_device, blk, cnt); 108 109 time = get_timer(0); 110 n = blk_read_devnum(IF_TYPE_NVME, nvme_curr_device, blk, 111 cnt, (ulong *)addr); 112 time = get_timer(time); 113 114 printf("read: %s\n", (n == cnt) ? "OK" : "ERROR"); 115 printf("%lu bytes read in %lu ms", cnt * 512, time); 116 if (time > 0) { 117 puts(" ("); 118 print_size(div_u64(cnt * 512, time) * 1000, "/s"); 119 puts(")"); 120 } 121 puts("\n"); 122 123 return (n == cnt) ? CMD_RET_SUCCESS : CMD_RET_FAILURE; 124 } 125 126 static int do_nvme_write(cmd_tbl_t *cmdtp, int flag, int argc, 127 char * const argv[]) 128 { 129 unsigned long time; 130 if (argc != 4) 131 return CMD_RET_USAGE; 132 133 ulong addr = simple_strtoul(argv[1], NULL, 16); 134 ulong cnt = simple_strtoul(argv[3], NULL, 16); 135 ulong n; 136 lbaint_t blk = simple_strtoul(argv[2], NULL, 16); 137 138 printf("\nNVMe write: device %d block # " LBAFU " count %ld ... ", 139 nvme_curr_device, blk, cnt); 140 141 time = get_timer(0); 142 n = blk_write_devnum(IF_TYPE_NVME, nvme_curr_device, blk, 143 cnt, (ulong *)addr); 144 time = get_timer(time); 145 146 printf("write: %s\n", (n == cnt) ? "OK" : "ERROR"); 147 printf("%lu bytes write in %lu ms", cnt * 512, time); 148 if (time > 0) { 149 puts(" ("); 150 print_size(div_u64(cnt * 512, time) * 1000, "/s"); 151 puts(")"); 152 } 153 puts("\n"); 154 155 return (n == cnt) ? CMD_RET_SUCCESS : CMD_RET_FAILURE; 156 } 157 158 static cmd_tbl_t cmd_nvme[] = { 159 U_BOOT_CMD_MKENT(scan, 1, 1, do_nvme_scan, "", ""), 160 U_BOOT_CMD_MKENT(list, 1, 1, do_nvme_list, "", ""), 161 U_BOOT_CMD_MKENT(info, 2, 1, do_nvme_info, "", ""), 162 U_BOOT_CMD_MKENT(device, 2, 1, do_nvme_device, "", ""), 163 U_BOOT_CMD_MKENT(part, 2, 1, do_nvme_part, "", ""), 164 U_BOOT_CMD_MKENT(write, 4, 0, do_nvme_write, "", ""), 165 U_BOOT_CMD_MKENT(read, 4, 0, do_nvme_read, "", "") 166 }; 167 168 static int do_nvmecops(cmd_tbl_t *cmdtp, int flag, int argc, 169 char * const argv[]) 170 { 171 cmd_tbl_t *cp; 172 173 cp = find_cmd_tbl(argv[1], cmd_nvme, ARRAY_SIZE(cmd_nvme)); 174 175 argc--; 176 argv++; 177 178 if (cp == NULL || argc > cp->maxargs) 179 return CMD_RET_USAGE; 180 181 if (flag == CMD_FLAG_REPEAT && !cp->repeatable) 182 return CMD_RET_SUCCESS; 183 184 return cp->cmd(cmdtp, flag, argc, argv); 185 } 186 187 U_BOOT_CMD( 188 nvme, 8, 1, do_nvmecops, 189 "NVM Express sub-system", 190 "\nnvme scan - scan NVMe blk devices\n" 191 "nvme list - show all available NVMe blk devices\n" 192 "nvme info [dev]- show current or a specific NVMe blk device\n" 193 "nvme device [dev] - show or set current device\n" 194 "nvme part [dev] - print partition table\n" 195 "nvme read addr blk# cnt\n" 196 "nvme write addr blk# cnt" 197 ); 198