--- embedtools/src/ube.c 2014/01/28 08:28:18 1.1.2.2 +++ embedtools/src/ube.c 2014/01/28 08:57:43 1.1.2.3 @@ -15,6 +15,7 @@ Usage() "\n" "\t-g \tSet parameter to value\n" "\t-s \tGet parameter value\n" + "\t-q\t\tQuiet mode\n" "\t-v\t\tVerbose ...\n" "\n", compiled, compiledby, compilehost); } @@ -22,21 +23,26 @@ Usage() int main(int argc, char **argv) { - char ch, mode = 0, szName[STRSIZ], szVal[STRSIZ] = { 0 }; + char ch, mode = 0, szName[STRSIZ] = { 0 }, szVal[STRSIZ] = { 0 }; + const char *str; + int ret; - while ((ch = getopt(argc, argv, "hvg:s:")) != -1) + while ((ch = getopt(argc, argv, "hvqg:s:")) != -1) switch (ch) { case 'g': - mode = 1; + mode |= 1; strlcpy(szName, optarg, sizeof szName); break; case 's': - mode = 2; + mode |= 2; strlcpy(szName, optarg, sizeof szName); break; case 'v': Verbose++; break; + case 'q': + mode |= 0x80; + break; case 'h': default: Usage(); @@ -44,15 +50,37 @@ main(int argc, char **argv) } argc -= optind; argv += optind; - if (mode == 2) { - if (!argc) { - Usage(); - return 1; - } else - strlcpy(szVal, *argv, sizeof szVal); + if ((mode & 0x7f) == 2 && argc) + strlcpy(szVal, *argv, sizeof szVal); + + VERB(1) printf("u-boot-env: mode=0x%hhx name=%s value=%s\n", mode, szName, szVal); + + if (!(mode & 0x7f)) { + VERB(2) printf("u-boot-env: list variables\n"); } - VERB(1) printf("u-boot-env: name=%s value=%s\n", szName, szVal); + if ((mode & 0x7f) & 1) { + VERB(2) printf("u-boot-env: get variable %s\n", szName); + str = ub_getenv(szName); + if (!str) { + printf("Error:: Variable %s not found!\n", szName); + return 2; + } else if (mode & 0x80) + printf("%s\n", str); + else + printf("Variable %s=%s\n", szName, str); + } + + if ((mode & 0x7f) & 2) { + VERB(2) printf("u-boot-env: set variable %s\n", szName); + ret = ub_setenv(szName, argc ? szVal : NULL); + if (ret) { + printf("Error:: Writing variable %s!\n", szName); + return 3; + } + if (!(mode & 0x80)) + printf("Done\n"); + } return 0; }