Blame view
cmd/binop.c
2.87 KB
83d290c56 SPDX: Convert all... |
1 |
// SPDX-License-Identifier: GPL-2.0+ |
55b255611 cmd: Add command ... |
2 3 4 |
#include <common.h> #include <command.h> |
9fb625ce0 env: Move env_set... |
5 |
#include <env.h> |
32dbe8c58 cmd: binop: Use h... |
6 |
#include <hexdump.h> |
55b255611 cmd: Add command ... |
7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
#include <malloc.h> #include <mapmem.h> #include <linux/ctype.h> enum { OP_ID_XOR, OP_ID_AND, OP_ID_OR, }; void write_to_env_var(char *varname, u8 *result, ulong len) { char *str_output; char *str_ptr; int i; str_output = malloc(len * 2 + 1); str_ptr = str_output; for (i = 0; i < len; i++) { sprintf(str_ptr, "%02x", result[i]); str_ptr += 2; } *str_ptr = '\0'; |
29dada9c2 cmd: binop: Use n... |
31 |
env_set(varname, str_output); |
55b255611 cmd: Add command ... |
32 33 34 |
free(str_output); } |
55b255611 cmd: Add command ... |
35 36 37 |
void read_from_env_var(char *varname, u8 *result) { char *str_value; |
29dada9c2 cmd: binop: Use n... |
38 |
str_value = env_get(varname); |
55b255611 cmd: Add command ... |
39 |
if (str_value) |
32dbe8c58 cmd: binop: Use h... |
40 |
hex2bin(result, str_value, strlen(str_value) / 2); |
55b255611 cmd: Add command ... |
41 |
else |
32dbe8c58 cmd: binop: Use h... |
42 |
hex2bin(result, varname, strlen(varname) / 2); |
55b255611 cmd: Add command ... |
43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 |
} void read_from_mem(ulong addr, u8 *result, ulong len) { u8 *src; src = map_sysmem(addr, len); memcpy(result, src, len); unmap_sysmem(src); } void write_to_mem(char *varname, u8 *result, ulong len) { ulong addr; u8 *buf; addr = simple_strtoul(varname, NULL, 16); buf = map_sysmem(addr, len); memcpy(buf, result, len); unmap_sysmem(buf); } static int do_binop(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]) { ulong len; u8 *result, *src1, *src2; char *oparg, *lenarg, *src1arg, *src2arg, *destarg; int i, op; if (argc < 5) return CMD_RET_USAGE; oparg = argv[1]; lenarg = argv[2]; src1arg = argv[3]; src2arg = argv[4]; if (!strcmp(oparg, "xor")) op = OP_ID_XOR; else if (!strcmp(oparg, "or")) op = OP_ID_OR; else if (!strcmp(oparg, "and")) op = OP_ID_AND; else return CMD_RET_USAGE; len = simple_strtoul(lenarg, NULL, 10); src1 = malloc(len); src2 = malloc(len); if (*src1arg == '*') read_from_mem(simple_strtoul(src1arg + 1, NULL, 16), src1, len); else read_from_env_var(src1arg, src1); if (*src2arg == '*') read_from_mem(simple_strtoul(src2arg + 1, NULL, 16), src2, len); else read_from_env_var(src2arg, src2); result = malloc(len); switch (op) { case OP_ID_XOR: for (i = 0; i < len; i++) result[i] = src1[i] ^ src2[i]; break; case OP_ID_OR: for (i = 0; i < len; i++) result[i] = src1[i] | src2[i]; break; case OP_ID_AND: for (i = 0; i < len; i++) result[i] = src1[i] & src2[i]; break; } if (argc == 5) { for (i = 0; i < len; i++) { printf("%02x ", result[i]); if (i % 16 == 15) puts(" "); } puts(" "); goto exit; } destarg = argv[5]; if (*destarg == '*') write_to_mem(destarg + 1, result, len); /* Skip asterisk */ else write_to_env_var(destarg, result, len); exit: free(result); free(src2); free(src1); return 0; } U_BOOT_CMD( binop, 6, 1, do_binop, "compute binary operation", "op count [*]src1 [*]src2 [[*]dest] " " - compute binary operation of data at/in src1 and src2 (either *memaddr, env var name or hex string) and store result in/at dest, where op is one of xor, or, and." ); |