diff mbox series

[RFC,v2,1/3] main: Check libuboot_set_env() result.

Message ID 20210706180018.342553-2-francis.laniel@amarulasolutions.com
State Accepted
Headers show
Series [RFC,v2,1/3] main: Check libuboot_set_env() result. | expand

Commit Message

Francis Laniel July 6, 2021, 6 p.m. UTC 
If function failed, an error message is printed with returned error code.

Signed-off-by: Francis Laniel <francis.laniel@amarulasolutions.com>
---
 src/fw_printenv.c | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)
diff mbox series

Patch

diff --git a/src/fw_printenv.c b/src/fw_printenv.c
index bf9951b..436b8fb 100644
--- a/src/fw_printenv.c
+++ b/src/fw_printenv.c
@@ -167,12 +167,26 @@  int main (int argc, char **argv) {
 				value = libuboot_get_env(ctx, argv[i]);
 				if (i + 1 == argc) {
 					if (value != NULL) {
-						libuboot_set_env(ctx, argv[i], NULL);
+						int ret;
+
+						ret = libuboot_set_env(ctx, argv[i], NULL);
+						if (ret) {
+							fprintf(stderr, "libuboot_set_env failed: %d\n", ret);
+							exit(-ret);
+						}
+
 						need_store = true;
 					}
 				} else {
 					if (value == NULL || strcmp(value, argv[i+1]) != 0) {
-						libuboot_set_env(ctx, argv[i], argv[i+1]);
+						int ret;
+
+						ret = libuboot_set_env(ctx, argv[i], argv[i+1]);
+						if (ret) {
+							fprintf(stderr, "libuboot_set_env failed: %d\n", ret);
+							exit(-ret);
+						}
+
 						need_store = true;
 					}
 				}