=== modified file 'grub-core/Makefile.core.def' --- grub-core/Makefile.core.def 2012-04-01 19:35:18 +0000 +++ grub-core/Makefile.core.def 2012-04-29 12:10:27 +0000 @@ -1840,3 +1840,7 @@ enable = i386; }; +module = { + name = testspeed; + common = commands/testspeed.c; +}; === added file 'grub-core/commands/testspeed.c' --- grub-core/commands/testspeed.c 1970-01-01 00:00:00 +0000 +++ grub-core/commands/testspeed.c 2012-04-29 20:04:05 +0000 @@ -0,0 +1,129 @@ +/* testspeed.c - Command to test file read speed */ +/* + * GRUB -- GRand Unified Bootloader + * Copyright (C) 2012 Free Software Foundation, Inc. + * + * GRUB is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * GRUB is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with GRUB. If not, see . + */ + +#include +#include +#include +#include +#include +#include +#include + +GRUB_MOD_LICENSE ("GPLv3+"); + +#define DEFAULT_BLOCK_SIZE 65536 + +static const struct grub_arg_option options[] = + { + {"size", 's', 0, N_("Specify size for each read operation"), 0, ARG_TYPE_INT}, + {0, 0, 0, 0, 0, 0} + }; + +static const char* grub_human_sizes[] = + {N_("B"), N_("KiB"), N_("MiB"), N_("GiB"), N_("TiB")}; + +static grub_err_t +grub_cmd_testspeed (grub_extcmd_context_t ctxt, int argc, char **args) +{ + struct grub_arg_list *state = ctxt->state; + grub_uint32_t start; + grub_uint32_t end; + grub_ssize_t block_size; + grub_disk_addr_t total_size; + char *buffer; + grub_file_t file; + + if (argc == 0) + return grub_error (GRUB_ERR_BAD_ARGUMENT, N_("filename expected")); + + block_size = (state[0].set) ? + grub_strtoul (state[0].arg, 0, 0) : DEFAULT_BLOCK_SIZE; + + if (block_size <= 0) + return grub_error (GRUB_ERR_BAD_ARGUMENT, N_("invalid block size")); + + buffer = grub_malloc (block_size); + if (buffer == NULL) + return grub_errno; + + file = grub_file_open (args[0]); + if (file == NULL) + goto quit; + + total_size = 0; + start = grub_get_time_ms (); + while (1) + { + grub_ssize_t size = grub_file_read (file, buffer, block_size); + if (size <= 0) + break; + total_size += size; + } + end = grub_get_time_ms (); + grub_file_close (file); + + grub_printf_ (N_("File size: %llu B \n"), (unsigned long long) total_size); + grub_printf_ (N_("Elapsed time: %d.%03d s \n"), (end - start) / 1000, + (end - start) % 1000); + + if (end != start) + { + grub_uint64_t fraction; + grub_uint64_t whole = + grub_divmod64 (total_size * 1000ULL, end - start, &fraction); + grub_uint64_t speed = whole * 1000ULL; + int sp = whole; + int units = 0; + + while (sp / 1024) + { + speed = (speed + 512) / 1024; + sp /= 1024; + units++; + } + + if (units) + whole = grub_divmod64 (speed, 1000, &fraction); + else + fraction = grub_divmod64 (fraction * 1000ULL, end - start, NULL); + + grub_printf_ (N_("Speed: %llu.%03d %s/s \n"), + (unsigned long long) whole, (int) fraction, + grub_human_sizes[units]); + } + + quit: + grub_free (buffer); + + return grub_errno; +} + +static grub_extcmd_t cmd; + +GRUB_MOD_INIT(time) +{ + cmd = grub_register_extcmd ("testspeed", grub_cmd_testspeed, 0, N_("[-s SIZE] FILENAME"), + N_("Test file read speed."), + options); +} + +GRUB_MOD_FINI(time) +{ + grub_unregister_extcmd (cmd); +}