← Back to branch summary

~ubuntu-branches/ubuntu/trusty/grub2/trusty-updates

« back to all changes in this revision

Viewing changes to commands/iorw.c

Tags: upstream-1.99~20101122
Import upstream version 1.99~20101122

Show diffs side-by-side

added added

removed removed

Lines of Context:
commands/iorw.c
1
 
/* memrw.c - command to read / write physical memory  */
2
 
/*
3
 
 *  GRUB  --  GRand Unified Bootloader
4
 
 *  Copyright (C) 2009  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/misc.h>
22
 
#include <grub/extcmd.h>
23
 
#include <grub/env.h>
24
 
#include <grub/cpu/io.h>
25
 
 
26
 
static grub_extcmd_t cmd_read_byte, cmd_read_word, cmd_read_dword;
27
 
static grub_command_t cmd_write_byte, cmd_write_word, cmd_write_dword;
28
 
 
29
 
static const struct grub_arg_option options[] =
30
 
  {
31
 
    {0, 'v', 0, "Save read value into variable VARNAME.",
32
 
     "VARNAME", ARG_TYPE_STRING},
33
 
    {0, 0, 0, 0, 0, 0}
34
 
  };
35
 
 
36
 
 
37
 
static grub_err_t
38
 
grub_cmd_read (grub_extcmd_t cmd, int argc, char **argv)
39
 
{
40
 
  grub_target_addr_t addr;
41
 
  grub_uint32_t value = 0;
42
 
 
43
 
  if (argc != 1)
44
 
    return grub_error (GRUB_ERR_BAD_ARGUMENT, "Invalid number of arguments");
45
 
 
46
 
  addr = grub_strtoul (argv[0], 0, 0);
47
 
  switch (cmd->cmd->name[sizeof ("grub_in") - 1])
48
 
    {
49
 
    case 'l':
50
 
      value = grub_inl (addr);
51
 
      break;
52
 
 
53
 
    case 'w':
54
 
      value = grub_inw (addr);
55
 
      break;
56
 
 
57
 
    case 'b':
58
 
      value = grub_inb (addr);
59
 
      break;
60
 
    }
61
 
 
62
 
  if (cmd->state[0].set)
63
 
    {
64
 
      char buf[sizeof ("XXXXXXXX")];
65
 
      grub_snprintf (buf, sizeof (buf), "%x", value);
66
 
      grub_env_set (cmd->state[0].arg, buf);
67
 
    }
68
 
  else
69
 
    grub_printf ("0x%x\n", value);
70
 
 
71
 
  return 0;
72
 
}
73
 
 
74
 
static grub_err_t
75
 
grub_cmd_write (grub_command_t cmd, int argc, char **argv)
76
 
{
77
 
  grub_target_addr_t addr;
78
 
  grub_uint32_t value;
79
 
  grub_uint32_t mask = 0xffffffff;
80
 
 
81
 
  if (argc != 2 && argc != 3)
82
 
    return grub_error (GRUB_ERR_BAD_ARGUMENT, "Invalid number of arguments");
83
 
 
84
 
  addr = grub_strtoul (argv[0], 0, 0);
85
 
  value = grub_strtoul (argv[1], 0, 0);
86
 
  if (argc == 3)
87
 
    mask = grub_strtoul (argv[2], 0, 0);
88
 
  value &= mask;
89
 
  switch (cmd->name[sizeof ("grub_out") - 1])
90
 
    {
91
 
    case 'l':
92
 
      if (mask != 0xffffffff)
93
 
        grub_outl ((grub_inl (addr) & ~mask) | value, addr);
94
 
      else
95
 
        grub_outl (value, addr);
96
 
      break;
97
 
 
98
 
    case 'w':
99
 
      if ((mask & 0xffff) != 0xffff)
100
 
        grub_outw ((grub_inw (addr) & ~mask) | value, addr);
101
 
      else
102
 
        grub_outw (value, addr);
103
 
      break;
104
 
 
105
 
    case 'b':
106
 
      if ((mask & 0xff) != 0xff)
107
 
        grub_outb ((grub_inb (addr) & ~mask) | value, addr);
108
 
      else
109
 
        grub_outb (value, addr);
110
 
      break;
111
 
    }
112
 
 
113
 
  return 0;
114
 
}
115
 
 
116
 
GRUB_MOD_INIT(memrw)
117
 
{
118
 
  cmd_read_byte =
119
 
    grub_register_extcmd ("grub_inb", grub_cmd_read, GRUB_COMMAND_FLAG_BOTH,
120
 
                          "grub_inb PORT", "Read byte from PORT.", options);
121
 
  cmd_read_word =
122
 
    grub_register_extcmd ("grub_inw", grub_cmd_read, GRUB_COMMAND_FLAG_BOTH,
123
 
                          "grub_inw PORT", "Read word from PORT.", options);
124
 
  cmd_read_dword =
125
 
    grub_register_extcmd ("grub_inl", grub_cmd_read, GRUB_COMMAND_FLAG_BOTH,
126
 
                          "grub_inl PORT", "Read dword from PORT.", options);
127
 
  cmd_write_byte =
128
 
    grub_register_command ("grub_outb", grub_cmd_write,
129
 
                           "grub_outb PORT VALUE [MASK]", "Write byte VALUE to PORT.");
130
 
  cmd_write_word =
131
 
    grub_register_command ("grub_outw", grub_cmd_write,
132
 
                           "grub_outw PORT VALUE [MASK]", "Write word VALUE to PORT.");
133
 
  cmd_write_dword =
134
 
    grub_register_command ("grub_outl", grub_cmd_write,
135
 
                           "grub_outl ADDR VALUE [MASK]", "Write dword VALUE to PORT.");
136
 
}
137
 
 
138
 
GRUB_MOD_FINI(memrw)
139
 
{
140
 
  grub_unregister_extcmd (cmd_read_byte);
141
 
  grub_unregister_extcmd (cmd_read_word);
142
 
  grub_unregister_extcmd (cmd_read_dword);
143
 
  grub_unregister_command (cmd_write_byte);
144
 
  grub_unregister_command (cmd_write_word);
145
 
  grub_unregister_command (cmd_write_dword);
146
 
}