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 i;
if( argc < 2 ) {
usage();
return 1;
}
/* Ugly... iterate over arguments to see whether we can expect a write */
for( i = 1; i < argc; i++ )
if( ( !strcmp(argv[i], "set") && ++i < argc ) ||
( !strcmp(argv[i], "unset") && ++i < argc ) ||
!strcmp(argv[i], "commit") )
{
write = 1;
break;
}
if( ( !strcmp(argv[1], "set") && 2 < argc ) ||
( !strcmp(argv[1], "unset") && 2 < argc ) ||
!strcmp(argv[1], "commit") )
write = 1;
nvram = write ? nvram_open_staging() : nvram_open_rdonly();