nvram: improve argument check when program start

print help message when argument count is less
than 2.

Signed-off-by: BangLang Huang <banglang.huang@foxmail.com>
This commit is contained in:
BangLang Huang 2017-02-24 10:52:52 +08:00 committed by Alexander Couzens
parent 2a253e7cdb
commit c7e2a6fe92

View file

@ -155,15 +155,16 @@ int main( int argc, const char *argv[] )
int done = 0; int done = 0;
int i; int i;
if( argc < 2 ) {
usage();
return 1;
}
/* Ugly... iterate over arguments to see whether we can expect a write */ /* Ugly... iterate over arguments to see whether we can expect a write */
for( i = 1; i < argc; i++ ) if( ( !strcmp(argv[1], "set") && 2 < argc ) ||
if( ( !strcmp(argv[i], "set") && ++i < argc ) || ( !strcmp(argv[1], "unset") && 2 < argc ) ||
( !strcmp(argv[i], "unset") && ++i < argc ) || !strcmp(argv[1], "commit") )
!strcmp(argv[i], "commit") ) write = 1;
{
write = 1;
break;
}
nvram = write ? nvram_open_staging() : nvram_open_rdonly(); nvram = write ? nvram_open_staging() : nvram_open_rdonly();