Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 1 | /*****************************************************************************\ |
| 2 | * opts.c |
Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 3 | ***************************************************************************** |
| 4 | * Copyright (C) 2002-2005 The Regents of the University of California. |
| 5 | * Produced at the Lawrence Livermore National Laboratory. |
| 6 | * Written by Dave Peterson <dsp@llnl.gov> <dave_peterson@pobox.com>. |
| 7 | * UCRL-CODE-2003-012 |
| 8 | * All rights reserved. |
| 9 | * |
Uwe Hermann | 6e56594 | 2008-03-01 19:06:32 +0000 | [diff] [blame] | 10 | * This file is part of nvramtool, a utility for reading/writing coreboot |
Stefan Reinauer | f527e70 | 2008-01-18 15:33:49 +0000 | [diff] [blame] | 11 | * parameters and displaying information from the coreboot table. |
Uwe Hermann | 6e56594 | 2008-03-01 19:06:32 +0000 | [diff] [blame] | 12 | * For details, see http://coreboot.org/nvramtool. |
Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 13 | * |
| 14 | * Please also read the file DISCLAIMER which is included in this software |
| 15 | * distribution. |
| 16 | * |
| 17 | * This program is free software; you can redistribute it and/or modify it |
| 18 | * under the terms of the GNU General Public License (as published by the |
| 19 | * Free Software Foundation) version 2, dated June 1991. |
| 20 | * |
| 21 | * This program is distributed in the hope that it will be useful, but |
| 22 | * WITHOUT ANY WARRANTY; without even the IMPLIED WARRANTY OF |
| 23 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the terms and |
| 24 | * conditions of the GNU General Public License for more details. |
| 25 | * |
| 26 | * You should have received a copy of the GNU General Public License along |
| 27 | * with this program; if not, write to the Free Software Foundation, Inc., |
Stefan Reinauer | ac7a2d2 | 2009-09-23 21:53:25 +0000 | [diff] [blame] | 28 | * 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 29 | \*****************************************************************************/ |
| 30 | |
| 31 | #include "common.h" |
| 32 | #include "opts.h" |
| 33 | |
Uwe Hermann | 6e56594 | 2008-03-01 19:06:32 +0000 | [diff] [blame] | 34 | nvramtool_op_info_t nvramtool_op; |
Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 35 | |
Uwe Hermann | 6e56594 | 2008-03-01 19:06:32 +0000 | [diff] [blame] | 36 | nvramtool_op_modifier_info_t nvramtool_op_modifiers[NVRAMTOOL_NUM_OP_MODIFIERS]; |
Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 37 | |
Stefan Reinauer | 90b96b6 | 2010-01-13 21:00:23 +0000 | [diff] [blame] | 38 | static char *handle_optional_arg(int argc, char *argv[]); |
| 39 | static void register_op(int *op_found, nvramtool_op_t op, char op_param[]); |
| 40 | static void register_op_modifier(nvramtool_op_modifier_t mod, char mod_param[]); |
| 41 | static void resolve_op_modifiers(void); |
| 42 | static void sanity_check_args(void); |
Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 43 | |
Patrick Georgi | 202be7b | 2011-01-21 07:29:40 +0000 | [diff] [blame^] | 44 | static const char getopt_string[] = "-ab:B:c::C:dD:e:hil::np:r:tvw:xX:y:Y"; |
Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 45 | |
| 46 | /**************************************************************************** |
Uwe Hermann | 6e56594 | 2008-03-01 19:06:32 +0000 | [diff] [blame] | 47 | * parse_nvramtool_args |
Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 48 | * |
| 49 | * Parse command line arguments. |
| 50 | ****************************************************************************/ |
Stefan Reinauer | 90b96b6 | 2010-01-13 21:00:23 +0000 | [diff] [blame] | 51 | void parse_nvramtool_args(int argc, char *argv[]) |
| 52 | { |
| 53 | nvramtool_op_modifier_info_t *mod_info; |
| 54 | int i, op_found; |
| 55 | char c; |
Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 56 | |
Stefan Reinauer | 90b96b6 | 2010-01-13 21:00:23 +0000 | [diff] [blame] | 57 | for (i = 0, mod_info = nvramtool_op_modifiers; |
| 58 | i < NVRAMTOOL_NUM_OP_MODIFIERS; i++, mod_info++) { |
| 59 | mod_info->found = FALSE; |
| 60 | mod_info->found_seq = 0; |
| 61 | mod_info->param = NULL; |
| 62 | } |
Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 63 | |
Stefan Reinauer | 90b96b6 | 2010-01-13 21:00:23 +0000 | [diff] [blame] | 64 | op_found = FALSE; |
| 65 | opterr = 0; |
Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 66 | |
Stefan Reinauer | 90b96b6 | 2010-01-13 21:00:23 +0000 | [diff] [blame] | 67 | do { |
| 68 | switch (c = getopt(argc, argv, getopt_string)) { |
| 69 | case 'a': |
| 70 | register_op(&op_found, |
| 71 | NVRAMTOOL_OP_CMOS_SHOW_ALL_PARAMS, NULL); |
| 72 | break; |
| 73 | case 'b': |
| 74 | register_op(&op_found, NVRAMTOOL_OP_WRITE_CMOS_DUMP, |
| 75 | optarg); |
| 76 | break; |
| 77 | case 'B': |
| 78 | register_op(&op_found, NVRAMTOOL_OP_READ_CMOS_DUMP, |
| 79 | optarg); |
| 80 | break; |
| 81 | case 'c': |
| 82 | register_op(&op_found, NVRAMTOOL_OP_CMOS_CHECKSUM, |
| 83 | handle_optional_arg(argc, argv)); |
| 84 | break; |
Patrick Georgi | 269e932 | 2011-01-21 07:24:08 +0000 | [diff] [blame] | 85 | case 'C': |
| 86 | register_op_modifier(NVRAMTOOL_MOD_USE_CBFS_FILE, |
| 87 | optarg); |
| 88 | break; |
Stefan Reinauer | 90b96b6 | 2010-01-13 21:00:23 +0000 | [diff] [blame] | 89 | case 'd': |
| 90 | register_op(&op_found, NVRAMTOOL_OP_LBTABLE_DUMP, NULL); |
| 91 | break; |
Patrick Georgi | 202be7b | 2011-01-21 07:29:40 +0000 | [diff] [blame^] | 92 | case 'D': |
| 93 | register_op_modifier(NVRAMTOOL_MOD_USE_CMOS_FILE, |
| 94 | optarg); |
| 95 | break; |
Stefan Reinauer | 90b96b6 | 2010-01-13 21:00:23 +0000 | [diff] [blame] | 96 | case 'e': |
| 97 | register_op(&op_found, NVRAMTOOL_OP_SHOW_PARAM_VALUES, |
| 98 | optarg); |
| 99 | break; |
| 100 | case 'h': |
| 101 | register_op(&op_found, NVRAMTOOL_OP_SHOW_USAGE, NULL); |
| 102 | break; |
| 103 | case 'i': |
| 104 | register_op(&op_found, |
| 105 | NVRAMTOOL_OP_CMOS_SET_PARAMS_STDIN, NULL); |
| 106 | break; |
| 107 | case 'l': |
| 108 | register_op(&op_found, NVRAMTOOL_OP_LBTABLE_SHOW_INFO, |
| 109 | handle_optional_arg(argc, argv)); |
| 110 | break; |
| 111 | case 'n': |
| 112 | register_op_modifier(NVRAMTOOL_MOD_SHOW_VALUE_ONLY, |
| 113 | NULL); |
| 114 | break; |
| 115 | case 'p': |
| 116 | register_op(&op_found, |
| 117 | NVRAMTOOL_OP_CMOS_SET_PARAMS_FILE, optarg); |
| 118 | break; |
| 119 | case 'r': |
| 120 | register_op(&op_found, NVRAMTOOL_OP_CMOS_SHOW_ONE_PARAM, |
| 121 | optarg); |
| 122 | break; |
| 123 | case 't': |
| 124 | register_op_modifier(NVRAMTOOL_MOD_USE_CMOS_OPT_TABLE, |
| 125 | NULL); |
| 126 | break; |
| 127 | case 'v': |
| 128 | register_op(&op_found, NVRAMTOOL_OP_SHOW_VERSION, NULL); |
| 129 | break; |
| 130 | case 'w': |
| 131 | register_op(&op_found, NVRAMTOOL_OP_CMOS_SET_ONE_PARAM, |
| 132 | optarg); |
| 133 | break; |
| 134 | case 'x': |
| 135 | register_op(&op_found, NVRAMTOOL_OP_SHOW_CMOS_HEX_DUMP, |
| 136 | NULL); |
| 137 | break; |
| 138 | case 'X': |
| 139 | register_op(&op_found, NVRAMTOOL_OP_SHOW_CMOS_DUMPFILE, |
| 140 | optarg); |
| 141 | break; |
| 142 | case 'y': |
| 143 | register_op_modifier(NVRAMTOOL_MOD_USE_CMOS_LAYOUT_FILE, |
| 144 | optarg); |
| 145 | break; |
| 146 | case 'Y': |
| 147 | register_op(&op_found, NVRAMTOOL_OP_SHOW_LAYOUT, NULL); |
| 148 | break; |
| 149 | case -1: /* no more command line args */ |
| 150 | break; |
| 151 | case '?': /* unknown option found */ |
| 152 | case 1: /* nonoption command line arg found */ |
| 153 | default: |
| 154 | usage(stderr); |
| 155 | break; |
| 156 | } |
| 157 | } while (c != -1); |
Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 158 | |
Stefan Reinauer | 90b96b6 | 2010-01-13 21:00:23 +0000 | [diff] [blame] | 159 | if (!op_found) |
| 160 | usage(stderr); |
Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 161 | |
Stefan Reinauer | 90b96b6 | 2010-01-13 21:00:23 +0000 | [diff] [blame] | 162 | resolve_op_modifiers(); |
| 163 | sanity_check_args(); |
| 164 | } |
Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 165 | |
| 166 | /**************************************************************************** |
| 167 | * handle_optional_arg |
| 168 | * |
| 169 | * Handle a command line option with an optional argument. |
| 170 | ****************************************************************************/ |
Stefan Reinauer | 90b96b6 | 2010-01-13 21:00:23 +0000 | [diff] [blame] | 171 | static char *handle_optional_arg(int argc, char *argv[]) |
| 172 | { |
| 173 | char *arg; |
Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 174 | |
Stefan Reinauer | 90b96b6 | 2010-01-13 21:00:23 +0000 | [diff] [blame] | 175 | if (optarg != NULL) { |
| 176 | /* optional arg is present and arg was specified as |
| 177 | * "-zarg" (with no whitespace between "z" and "arg"), |
| 178 | * where -z is the option and "arg" is the value of the |
| 179 | * optional arg |
| 180 | */ |
| 181 | return optarg; |
| 182 | } |
Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 183 | |
Stefan Reinauer | 90b96b6 | 2010-01-13 21:00:23 +0000 | [diff] [blame] | 184 | if ((argv[optind] == NULL) || (argv[optind][0] == '-')) |
| 185 | return NULL; |
Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 186 | |
Stefan Reinauer | 90b96b6 | 2010-01-13 21:00:23 +0000 | [diff] [blame] | 187 | arg = argv[optind]; /* optional arg is present */ |
Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 188 | |
Stefan Reinauer | 90b96b6 | 2010-01-13 21:00:23 +0000 | [diff] [blame] | 189 | /* This call to getopt yields the optional arg we just found, |
| 190 | * which we want to skip. |
| 191 | */ |
| 192 | getopt(argc, argv, getopt_string); |
Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 193 | |
Stefan Reinauer | 90b96b6 | 2010-01-13 21:00:23 +0000 | [diff] [blame] | 194 | return arg; |
| 195 | } |
Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 196 | |
| 197 | /**************************************************************************** |
| 198 | * register_op |
| 199 | * |
| 200 | * Store the user's selection of which operation this program should perform. |
| 201 | ****************************************************************************/ |
Stefan Reinauer | 90b96b6 | 2010-01-13 21:00:23 +0000 | [diff] [blame] | 202 | static void register_op(int *op_found, nvramtool_op_t op, char op_param[]) |
| 203 | { |
| 204 | if (*op_found && (op != nvramtool_op.op)) |
| 205 | usage(stderr); |
Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 206 | |
Stefan Reinauer | 90b96b6 | 2010-01-13 21:00:23 +0000 | [diff] [blame] | 207 | *op_found = TRUE; |
| 208 | nvramtool_op.op = op; |
| 209 | nvramtool_op.param = op_param; |
| 210 | } |
Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 211 | |
| 212 | /**************************************************************************** |
| 213 | * register_op_modifier |
| 214 | * |
| 215 | * Store information regarding an optional argument specified in addition to |
| 216 | * the user's selection of which operation this program should perform. |
| 217 | ****************************************************************************/ |
Stefan Reinauer | 90b96b6 | 2010-01-13 21:00:23 +0000 | [diff] [blame] | 218 | static void register_op_modifier(nvramtool_op_modifier_t mod, char mod_param[]) |
| 219 | { |
| 220 | static int found_seq = 0; |
| 221 | nvramtool_op_modifier_info_t *mod_info; |
Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 222 | |
Stefan Reinauer | 90b96b6 | 2010-01-13 21:00:23 +0000 | [diff] [blame] | 223 | mod_info = &nvramtool_op_modifiers[mod]; |
| 224 | mod_info->found = TRUE; |
| 225 | mod_info->found_seq = ++found_seq; |
| 226 | mod_info->param = mod_param; |
| 227 | } |
Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 228 | |
| 229 | /**************************************************************************** |
| 230 | * resolve_op_modifiers |
| 231 | * |
| 232 | * If the user specifies multiple arguments that conflict with each other, |
| 233 | * the last specified argument overrides previous conflicting arguments. |
| 234 | ****************************************************************************/ |
Stefan Reinauer | 90b96b6 | 2010-01-13 21:00:23 +0000 | [diff] [blame] | 235 | static void resolve_op_modifiers(void) |
| 236 | { |
| 237 | if (nvramtool_op_modifiers[NVRAMTOOL_MOD_USE_CMOS_LAYOUT_FILE].found && |
| 238 | nvramtool_op_modifiers[NVRAMTOOL_MOD_USE_CMOS_OPT_TABLE].found) { |
| 239 | if (nvramtool_op_modifiers[NVRAMTOOL_MOD_USE_CMOS_LAYOUT_FILE].found_seq > |
| 240 | nvramtool_op_modifiers[NVRAMTOOL_MOD_USE_CMOS_OPT_TABLE].found_seq) |
| 241 | nvramtool_op_modifiers |
| 242 | [NVRAMTOOL_MOD_USE_CMOS_OPT_TABLE].found = FALSE; |
| 243 | else |
| 244 | nvramtool_op_modifiers |
| 245 | [NVRAMTOOL_MOD_USE_CMOS_LAYOUT_FILE].found = FALSE; |
| 246 | } |
| 247 | } |
Stefan Reinauer | 6540ae5 | 2007-07-12 16:35:42 +0000 | [diff] [blame] | 248 | |
| 249 | /**************************************************************************** |
| 250 | * sanity_check_args |
| 251 | * |
| 252 | * Perform sanity checking on command line arguments. |
| 253 | ****************************************************************************/ |
Stefan Reinauer | 90b96b6 | 2010-01-13 21:00:23 +0000 | [diff] [blame] | 254 | static void sanity_check_args(void) |
| 255 | { |
| 256 | if ((nvramtool_op_modifiers[NVRAMTOOL_MOD_SHOW_VALUE_ONLY].found) && |
| 257 | (nvramtool_op.op != NVRAMTOOL_OP_CMOS_SHOW_ONE_PARAM)) |
| 258 | usage(stderr); |
| 259 | } |