aboutsummaryrefslogtreecommitdiff
path: root/lib/libbe/tests/target_prog.c
diff options
context:
space:
mode:
Diffstat (limited to 'lib/libbe/tests/target_prog.c')
-rw-r--r--lib/libbe/tests/target_prog.c11
1 files changed, 7 insertions, 4 deletions
diff --git a/lib/libbe/tests/target_prog.c b/lib/libbe/tests/target_prog.c
index 4c79d1a30714..83ef68ab2890 100644
--- a/lib/libbe/tests/target_prog.c
+++ b/lib/libbe/tests/target_prog.c
@@ -27,6 +27,7 @@
#include <sys/cdefs.h>
#include <be.h>
+#include <string.h>
/*
* argv[1] = root boot environment (e.g. zroot/ROOT),
@@ -36,15 +37,17 @@
*/
int main(int argc, char *argv[]) {
- libbe_handle_t *lbh;
+ libbe_handle_t *lbh;
if (argc != 5)
- return -1;
+ return (-1);
- if ((lbh = libbe_init(argv[1])) == NULL)
- return -1;
+ if ((lbh = libbe_init(argv[1])) == NULL)
+ return (-1);
libbe_print_on_error(lbh, true);
+ if (strcmp(argv[2], "empty") == 0)
+ return (be_create_empty(lbh, argv[2]));
return (be_create_depth(lbh, argv[2], argv[3], atoi(argv[4])));
}