Kumar Gala | d058698 | 2008-02-14 20:44:42 -0600 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2008 Freescale Semiconductor, Inc. |
| 3 | * |
| 4 | * See file CREDITS for list of people who contributed to this |
| 5 | * project. |
| 6 | * |
| 7 | * This program is free software; you can redistribute it and/or |
| 8 | * modify it under the terms of the GNU General Public License as |
| 9 | * published by the Free Software Foundation; either version 2 of |
| 10 | * the License, or (at your option) any later version. |
| 11 | * |
| 12 | * This program is distributed in the hope that it will be useful, |
| 13 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 14 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 15 | * GNU General Public License for more details. |
| 16 | * |
| 17 | * You should have received a copy of the GNU General Public License |
| 18 | * along with this program; if not, write to the Free Software |
| 19 | * Foundation, Inc., 59 Temple Place, Suite 330, Boston, |
| 20 | * MA 02111-1307 USA |
| 21 | */ |
| 22 | |
| 23 | /* |
| 24 | * This file provides a shell like 'expr' function to return. |
| 25 | */ |
| 26 | |
| 27 | #include <common.h> |
| 28 | #include <config.h> |
| 29 | #include <command.h> |
| 30 | |
Frans Meulenbroeks | 47ab5ad | 2010-02-26 14:00:19 +0100 | [diff] [blame] | 31 | static ulong get_arg(char *s, int w) |
| 32 | { |
| 33 | ulong *p; |
| 34 | |
Wolfgang Denk | 482126e | 2010-06-23 20:50:54 +0200 | [diff] [blame] | 35 | /* |
Frans Meulenbroeks | 47ab5ad | 2010-02-26 14:00:19 +0100 | [diff] [blame] | 36 | * if the parameter starts with a '*' then assume |
| 37 | * it is a pointer to the value we want |
| 38 | */ |
| 39 | |
| 40 | if (s[0] == '*') { |
| 41 | p = (ulong *)simple_strtoul(&s[1], NULL, 16); |
| 42 | switch (w) { |
| 43 | case 1: return((ulong)(*(uchar *)p)); |
| 44 | case 2: return((ulong)(*(ushort *)p)); |
| 45 | case 4: |
| 46 | default: return(*p); |
| 47 | } |
| 48 | } else { |
| 49 | return simple_strtoul(s, NULL, 16); |
| 50 | } |
| 51 | } |
| 52 | |
Kim Phillips | 088f1b1 | 2012-10-29 13:34:31 +0000 | [diff] [blame] | 53 | static int do_setexpr(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]) |
Kumar Gala | d058698 | 2008-02-14 20:44:42 -0600 | [diff] [blame] | 54 | { |
| 55 | ulong a, b; |
Simon Glass | 41ef372 | 2013-02-24 17:33:22 +0000 | [diff] [blame] | 56 | ulong value; |
Frans Meulenbroeks | 47ab5ad | 2010-02-26 14:00:19 +0100 | [diff] [blame] | 57 | int w; |
Kumar Gala | d058698 | 2008-02-14 20:44:42 -0600 | [diff] [blame] | 58 | |
Wolfgang Denk | 103c94b | 2013-03-23 23:50:33 +0000 | [diff] [blame^] | 59 | if (argc < 3) |
Simon Glass | 4c12eeb | 2011-12-10 08:44:01 +0000 | [diff] [blame] | 60 | return CMD_RET_USAGE; |
Kumar Gala | d058698 | 2008-02-14 20:44:42 -0600 | [diff] [blame] | 61 | |
Frans Meulenbroeks | 47ab5ad | 2010-02-26 14:00:19 +0100 | [diff] [blame] | 62 | w = cmd_get_data_size(argv[0], 4); |
| 63 | |
| 64 | a = get_arg(argv[2], w); |
Joe Hershberger | 4823b45 | 2012-11-01 16:21:14 +0000 | [diff] [blame] | 65 | |
Wolfgang Denk | 103c94b | 2013-03-23 23:50:33 +0000 | [diff] [blame^] | 66 | /* plain assignment: "setexpr name value" */ |
Joe Hershberger | 4823b45 | 2012-11-01 16:21:14 +0000 | [diff] [blame] | 67 | if (argc == 3) { |
Simon Glass | 41ef372 | 2013-02-24 17:33:22 +0000 | [diff] [blame] | 68 | setenv_hex(argv[1], a); |
Joe Hershberger | 4823b45 | 2012-11-01 16:21:14 +0000 | [diff] [blame] | 69 | return 0; |
| 70 | } |
| 71 | |
Wolfgang Denk | 103c94b | 2013-03-23 23:50:33 +0000 | [diff] [blame^] | 72 | /* standard operators: "setexpr name val1 op val2" */ |
| 73 | if (argc != 5) |
| 74 | return CMD_RET_USAGE; |
| 75 | |
| 76 | if (strlen(argv[3]) != 1) |
| 77 | return CMD_RET_USAGE; |
| 78 | |
Frans Meulenbroeks | 47ab5ad | 2010-02-26 14:00:19 +0100 | [diff] [blame] | 79 | b = get_arg(argv[4], w); |
Kumar Gala | d058698 | 2008-02-14 20:44:42 -0600 | [diff] [blame] | 80 | |
| 81 | switch (argv[3][0]) { |
Simon Glass | 41ef372 | 2013-02-24 17:33:22 +0000 | [diff] [blame] | 82 | case '|': |
| 83 | value = a | b; |
| 84 | break; |
| 85 | case '&': |
| 86 | value = a & b; |
| 87 | break; |
| 88 | case '+': |
| 89 | value = a + b; |
| 90 | break; |
| 91 | case '^': |
| 92 | value = a ^ b; |
| 93 | break; |
| 94 | case '-': |
| 95 | value = a - b; |
| 96 | break; |
| 97 | case '*': |
| 98 | value = a * b; |
| 99 | break; |
| 100 | case '/': |
| 101 | value = a / b; |
| 102 | break; |
| 103 | case '%': |
| 104 | value = a % b; |
| 105 | break; |
Kumar Gala | d058698 | 2008-02-14 20:44:42 -0600 | [diff] [blame] | 106 | default: |
| 107 | printf("invalid op\n"); |
| 108 | return 1; |
| 109 | } |
| 110 | |
Simon Glass | 41ef372 | 2013-02-24 17:33:22 +0000 | [diff] [blame] | 111 | setenv_hex(argv[1], value); |
Kumar Gala | d058698 | 2008-02-14 20:44:42 -0600 | [diff] [blame] | 112 | |
| 113 | return 0; |
| 114 | } |
| 115 | |
| 116 | U_BOOT_CMD( |
| 117 | setexpr, 5, 0, do_setexpr, |
Peter Tyser | 2fb2604 | 2009-01-27 18:03:12 -0600 | [diff] [blame] | 118 | "set environment variable as the result of eval expression", |
Joe Hershberger | 4823b45 | 2012-11-01 16:21:14 +0000 | [diff] [blame] | 119 | "[.b, .w, .l] name [*]value1 <op> [*]value2\n" |
Kumar Gala | d058698 | 2008-02-14 20:44:42 -0600 | [diff] [blame] | 120 | " - set environment variable 'name' to the result of the evaluated\n" |
Frans Meulenbroeks | 47ab5ad | 2010-02-26 14:00:19 +0100 | [diff] [blame] | 121 | " express specified by <op>. <op> can be &, |, ^, +, -, *, /, %\n" |
Joe Hershberger | 4823b45 | 2012-11-01 16:21:14 +0000 | [diff] [blame] | 122 | " size argument is only meaningful if value1 and/or value2 are\n" |
| 123 | " memory addresses (*)\n" |
Wolfgang Denk | 103c94b | 2013-03-23 23:50:33 +0000 | [diff] [blame^] | 124 | "setexpr[.b, .w, .l] name [*]value\n" |
| 125 | " - load a value into a variable" |
Kumar Gala | d058698 | 2008-02-14 20:44:42 -0600 | [diff] [blame] | 126 | ); |