@@ -60,14 +60,14 @@ static int cmd_get_state(uint8_t part)
6060{
6161 uint8_t state ;
6262 int ret ;
63-
63+
6464 ret = wolfBoot_get_partition_state (part , & state );
6565 if (ret != 0 ) {
6666 wolfBoot_printf ("Error: Failed to get state for %s partition (error: %d)\n" ,
6767 partition_name (part ), ret );
6868 return -1 ;
6969 }
70-
70+
7171 wolfBoot_printf ("%s partition state: %s (0x%02X)\n" ,
7272 partition_name (part ), state_name (state ), state );
7373 return 0 ;
@@ -77,15 +77,15 @@ static int cmd_get_state(uint8_t part)
7777static int cmd_get_all_states (void )
7878{
7979 int ret = 0 ;
80-
80+
8181 wolfBoot_printf ("=== Partition States ===\n" );
82-
82+
8383 if (cmd_get_state (PART_BOOT ) != 0 )
8484 ret = -1 ;
85-
85+
8686 if (cmd_get_state (PART_UPDATE ) != 0 )
8787 ret = -1 ;
88-
88+
8989 return ret ;
9090}
9191
@@ -164,15 +164,21 @@ static int cmd_verify(uint8_t part)
164164int main (int argc , const char * argv [])
165165{
166166 int ret = 0 ;
167-
167+ const char * prog_name = "lib-fs" ;
168+ const char * command ;
169+
170+ if (argc >= 1 ) {
171+ prog_name = argv [0 ];
172+ }
173+
168174 /* Check for argument count */
169175 if (argc != 2 ) {
170- print_usage (argv [ 0 ] );
176+ print_usage (prog_name );
171177 return 1 ;
172178 }
173-
174- const char * command = argv [1 ];
175-
179+
180+ command = argv [1 ];
181+
176182 /* Process commands */
177183 if (strcmp (command , "status" ) == 0 ) {
178184 ret = cmd_get_all_states ();
@@ -195,7 +201,7 @@ int main(int argc, const char* argv[])
195201 else if (strcmp (command , "verify-update" ) == 0 ) {
196202 ret = cmd_verify (PART_UPDATE );
197203 }
198- else if (strcmp (command , "help" ) == 0 || strcmp (command , "--help" ) == 0 ||
204+ else if (strcmp (command , "help" ) == 0 || strcmp (command , "--help" ) == 0 ||
199205 strcmp (command , "-h" ) == 0 ) {
200206 print_usage (argv [0 ]);
201207 ret = 0 ;
@@ -205,6 +211,6 @@ int main(int argc, const char* argv[])
205211 print_usage (argv [0 ]);
206212 ret = 1 ;
207213 }
208-
214+
209215 return ret ;
210216}
0 commit comments