2N/A/* tr.c -- The tr command. */
2N/A/*
2N/A * GRUB -- GRand Unified Bootloader
2N/A * Copyright (C) 2010 Free Software Foundation, Inc.
2N/A *
2N/A * GRUB is free software: you can redistribute it and/or modify
2N/A * it under the terms of the GNU General Public License as published by
2N/A * the Free Software Foundation, either version 3 of the License, or
2N/A * (at your option) any later version.
2N/A *
2N/A * GRUB is distributed in the hope that it will be useful,
2N/A * but WITHOUT ANY WARRANTY; without even the implied warranty of
2N/A * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
2N/A * GNU General Public License for more details.
2N/A *
2N/A * You should have received a copy of the GNU General Public License
2N/A * along with GRUB. If not, see <http://www.gnu.org/licenses/>.
2N/A */
2N/A
2N/A#include <grub/dl.h>
2N/A#include <grub/mm.h>
2N/A#include <grub/err.h>
2N/A#include <grub/env.h>
2N/A#include <grub/i18n.h>
2N/A#include <grub/misc.h>
2N/A#include <grub/extcmd.h>
2N/A
2N/AGRUB_MOD_LICENSE ("GPLv3+");
2N/A
2N/Astatic const struct grub_arg_option options[] =
2N/A {
2N/A { "set", 's', 0, N_("Variable name to update."), N_("VARNAME"), ARG_TYPE_STRING },
2N/A { "upcase", 'U', 0, N_("Translate to upper case."), 0, 0 },
2N/A { "downcase", 'D', 0, N_("Translate to lower case."), 0, 0 },
2N/A { 0, 0, 0, 0, 0, 0 }
2N/A };
2N/A
2N/Astatic grub_err_t
2N/Agrub_cmd_tr (grub_extcmd_context_t ctxt, int argc, char **args)
2N/A{
2N/A int i;
2N/A char *p;
2N/A char *var = 0;
2N/A char *input = 0;
2N/A char *output = 0;
2N/A const char *s1 = 0;
2N/A const char *s2 = 0;
2N/A
2N/A /* Select the defaults from options. */
2N/A if (ctxt->state[0].set) {
2N/A var = ctxt->state[0].arg;
2N/A input = (char *)grub_env_get (var);
2N/A }
2N/A
2N/A /* Check for arguments and update the defaults. */
2N/A if (argc == 1)
2N/A input = args[0];
2N/A
2N/A else if (argc == 2) {
2N/A s1 = args[0];
2N/A s2 = args[1];
2N/A
2N/A } else if (argc == 3) {
2N/A s1 = args[0];
2N/A s2 = args[1];
2N/A input = args[2];
2N/A
2N/A } else if (argc > 3)
2N/A return grub_error (GRUB_ERR_BAD_ARGUMENT, "too many parameters");
2N/A
2N/A if (argc <= 0 && !input)
2N/A return grub_error (GRUB_ERR_BAD_ARGUMENT, "missing string argument");
2N/A
2N/A if (s1 && s2 && grub_strlen (s1) != grub_strlen (s2))
2N/A return grub_error (GRUB_ERR_BAD_ARGUMENT, "set sizes did not match");
2N/A
2N/A else if ((!s1 || !s2) && !ctxt->state[1].set && !ctxt->state[2].set)
2N/A return grub_error (GRUB_ERR_BAD_ARGUMENT, "set parameters missing");
2N/A
2N/A /* Translate input into output buffer. */
2N/A
2N/A output = grub_malloc (grub_strlen (input) + 1);
2N/A if (! output)
2N/A return grub_errno;
2N/A
2N/A for (i = 0; input[i]; i++) {
2N/A if (ctxt->state[1].set)
2N/A output[i] = grub_toupper (input[i]);
2N/A
2N/A else if (ctxt->state[2].set)
2N/A output[i] = grub_tolower (input[i]);
2N/A
2N/A else if ((p = grub_strchr (s1, input[i])) != NULL)
2N/A output[i] = s2[p - s1];
2N/A
2N/A else
2N/A output[i] = input[i];
2N/A }
2N/A output[i] = '\0';
2N/A
2N/A if (ctxt->state[0].set)
2N/A grub_env_set (var, output);
2N/A else
2N/A grub_printf ("%s\n", output);
2N/A
2N/A grub_free (output);
2N/A return GRUB_ERR_NONE;
2N/A}
2N/A
2N/Astatic grub_extcmd_t cmd;
2N/A
2N/AGRUB_MOD_INIT(tr)
2N/A{
2N/A cmd = grub_register_extcmd ("tr", grub_cmd_tr, 0, N_("[OPTIONS] [SET1] [SET2] [STRING]"),
2N/A N_("Translate SET1 characters to SET2 in STRING."), options);
2N/A}
2N/A
2N/AGRUB_MOD_FINI(tr)
2N/A{
2N/A grub_unregister_extcmd (cmd);
2N/A}