arch/all-pc/boot/grub2-aros/grub-core/commands/testload.c

   1 /* testload.c - load the same file in multiple ways */
   2 /*
   3  *  GRUB  --  GRand Unified Bootloader
   4  *  Copyright (C) 2003,2005,2006,2007,2009,2010  Free Software Foundation, Inc.
   5  *
   6  *  GRUB is free software: you can redistribute it and/or modify
   7  *  it under the terms of the GNU General Public License as published by
   8  *  the Free Software Foundation, either version 3 of the License, or
   9  *  (at your option) any later version.
  10  *
  11  *  GRUB is distributed in the hope that it will be useful,
  12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  *  GNU General Public License for more details.
  15  *
  16  *  You should have received a copy of the GNU General Public License
  17  *  along with GRUB.  If not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20 #include <grub/dl.h>
  21 #include <grub/mm.h>
  22 #include <grub/err.h>
  23 #include <grub/env.h>
  24 #include <grub/misc.h>
  25 #include <grub/file.h>
  26 #include <grub/disk.h>
  27 #include <grub/term.h>
  28 #include <grub/loader.h>
  29 #include <grub/command.h>
  30 #include <grub/i18n.h>
  31
  32 GRUB_MOD_LICENSE ("GPLv3+");
  33
  34 static grub_err_t
  35 grub_cmd_testload (struct grub_command *cmd __attribute__ ((unused)),
  36                    int argc, char *argv[])
  37 {
  38   grub_file_t file;
  39   char *buf;
  40   grub_size_t size;
  41   grub_off_t pos;
  42   auto void NESTED_FUNC_ATTR read_func (grub_disk_addr_t sector, unsigned offset, unsigned len);
  43
  44   void NESTED_FUNC_ATTR read_func (grub_disk_addr_t sector __attribute__ ((unused)),
  45                   unsigned offset __attribute__ ((unused)),
  46                   unsigned len __attribute__ ((unused)))
  47     {
  48       grub_xputs (".");
  49       grub_refresh ();
  50     }
  51
  52   if (argc < 1)
  53     return grub_error (GRUB_ERR_BAD_ARGUMENT, N_("filename expected"));
  54
  55   file = grub_file_open (argv[0]);
  56   if (! file)
  57     return grub_errno;
  58
  59   size = grub_file_size (file) & ~(GRUB_DISK_SECTOR_SIZE - 1);
  60   if (size == 0)
  61     {
  62       grub_file_close (file);
  63       return GRUB_ERR_NONE;
  64     }
  65
  66   buf = grub_malloc (size);
  67   if (! buf)
  68     goto fail;
  69
  70   grub_printf ("Reading %s sequentially", argv[0]);
  71   file->read_hook = read_func;
  72   if (grub_file_read (file, buf, size) != (grub_ssize_t) size)
  73     goto fail;
  74   grub_printf (" Done.\n");
  75
  76   /* Read sequentially again.  */
  77   grub_printf ("Reading %s sequentially again", argv[0]);
  78   grub_file_seek (file, 0);
  79
  80   for (pos = 0; pos < size;)
  81     {
  82       char sector[GRUB_DISK_SECTOR_SIZE];
  83       grub_size_t curlen = GRUB_DISK_SECTOR_SIZE;
  84
  85       if (curlen > size - pos)
  86         curlen = size - pos;
  87
  88       if (grub_file_read (file, sector, curlen)
  89           != (grub_ssize_t) curlen)
  90         goto fail;
  91
  92       if (grub_memcmp (sector, buf + pos, curlen) != 0)
  93         {
  94           grub_printf ("\nDiffers in %lld\n", (unsigned long long) pos);
  95           goto fail;
  96         }
  97       pos += curlen;
  98     }
  99   grub_printf (" Done.\n");
 100
 101   /* Read backwards and compare.  */
 102   grub_printf ("Reading %s backwards", argv[0]);
 103   pos = size;
 104   while (pos > 0)
 105     {
 106       char sector[GRUB_DISK_SECTOR_SIZE];
 107
 108       if (pos >= GRUB_DISK_SECTOR_SIZE)
 109         pos -= GRUB_DISK_SECTOR_SIZE;
 110       else
 111         pos = 0;
 112
 113       grub_file_seek (file, pos);
 114
 115       if (grub_file_read (file, sector, GRUB_DISK_SECTOR_SIZE)
 116           != GRUB_DISK_SECTOR_SIZE)
 117         goto fail;
 118
 119       if (grub_memcmp (sector, buf + pos, GRUB_DISK_SECTOR_SIZE) != 0)
 120         {
 121           int i;
 122
 123           grub_printf ("\nDiffers in %lld\n", (unsigned long long) pos);
 124
 125           for (i = 0; i < GRUB_DISK_SECTOR_SIZE; i++)
 126             {
 127               grub_printf ("%02x ", buf[pos + i]);
 128               if ((i & 15) == 15)
 129                 grub_printf ("\n");
 130             }
 131
 132           if (i)
 133             grub_refresh ();
 134
 135           goto fail;
 136         }
 137     }
 138   grub_printf (" Done.\n");
 139
 140   return GRUB_ERR_NONE;
 141
 142  fail:
 143
 144   grub_file_close (file);
 145   grub_free (buf);
 146
 147   if (!grub_errno)
 148     grub_error (GRUB_ERR_IO, "bad read");
 149   return grub_errno;
 150 }
 151
 152 static grub_command_t cmd;
 153
 154 GRUB_MOD_INIT(testload)
 155 {
 156   cmd =
 157     grub_register_command ("testload", grub_cmd_testload,
 158                            N_("FILE"),
 159                            N_("Load the same file in multiple ways."));
 160 }
 161
 162 GRUB_MOD_FINI(testload)
 163 {
 164   grub_unregister_command (cmd);
 165 }