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>
(cherry picked from commit c7e2a6fe92)
This commit is contained in:
BangLang Huang 2017-02-24 10:52:52 +08:00 committed by Jo-Philipp Wich
parent c446ee4ad4
commit 118a2ea0bc
1 changed files with 9 additions and 8 deletions

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();