@@ -18,30 +18,18 @@ static struct tst_test test = {
static void print_help(void)
{
- fprintf(stderr, "\nUsage: tst_device acquire [size [filename]]\n");
- fprintf(stderr, " or: tst_device release /path/to/device\n\n");
+ fprintf(stderr, "\nUsage:\n");
+ fprintf(stderr, "tst_device [-s size [-d /path/to/device]] acquire\n");
+ fprintf(stderr, "tst_device -d /path/to/device release\n");
+ fprintf(stderr, "tst_device -h\n\n");
}
-static int acquire_device(int argc, char *argv[])
+static int acquire_device(const char *device_path, unsigned int size)
{
- unsigned int size = 0;
const char *device;
- if (argc > 4)
- return 1;
-
- if (argc >= 3) {
- size = atoi(argv[2]);
-
- if (!size) {
- fprintf(stderr, "ERROR: Invalid device size '%s'",
- argv[2]);
- return 1;
- }
- }
-
- if (argc >= 4)
- device = tst_acquire_loop_device(size, argv[3]);
+ if (device_path)
+ device = tst_acquire_loop_device(size, device_path);
else
device = tst_acquire_device__(size, TST_ALL_FILESYSTEMS);
@@ -58,10 +46,12 @@ static int acquire_device(int argc, char *argv[])
return 0;
}
-static int release_device(int argc, char *argv[])
+static int release_device(const char *device_path)
{
- if (argc != 3)
+ if (!device_path) {
+ fprintf(stderr, "ERROR: Missing /path/to/device\n");
return 1;
+ }
/*
* tst_acquire_[loop_]device() was called in a different process.
@@ -69,11 +59,15 @@ static int release_device(int argc, char *argv[])
* and do nothing. Call tst_detach_device() directly to bypass
* the check.
*/
- return tst_detach_device(argv[2]);
+ return tst_detach_device(device_path);
}
int main(int argc, char *argv[])
{
+ char *device_path = NULL;
+ unsigned int size = 0;
+ int ret;
+
/*
* Force messages to be printed from the new library i.e. tst_test.c
*
@@ -85,17 +79,38 @@ int main(int argc, char *argv[])
*/
tst_test = &test;
- if (argc < 2)
+ while ((ret = getopt(argc, argv, "d:hs:"))) {
+ if (ret < 0)
+ break;
+
+ switch (ret) {
+ case 'd':
+ device_path = optarg;
+ break;
+ case 'h':
+ print_help();
+ return 0;
+ case 's':
+ size = atoi(optarg);
+ if (!size) {
+ fprintf(stderr, "ERROR: Invalid device size '%s'", optarg);
+ return 1;
+ }
+ break;
+ }
+ }
+
+ if (argc - optind < 1)
goto help;
- if (!strcmp(argv[1], "acquire")) {
- if (acquire_device(argc, argv))
+ if (!strcmp(argv[optind], "acquire")) {
+ if (acquire_device(device_path, size))
goto help;
- } else if (!strcmp(argv[1], "release")) {
- if (release_device(argc, argv))
+ } else if (!strcmp(argv[optind], "release")) {
+ if (release_device(device_path))
goto help;
} else {
- fprintf(stderr, "ERROR: Invalid COMMAND '%s'\n", argv[1]);
+ fprintf(stderr, "ERROR: Invalid COMMAND '%s'\n", argv[optind]);
goto help;
}
@@ -46,7 +46,7 @@ _tst_do_exit()
fi
if [ "$TST_NEEDS_DEVICE" = 1 -a "$TST_DEVICE_FLAG" = 1 ]; then
- if ! tst_device release "$TST_DEVICE"; then
+ if ! tst_device -d "$TST_DEVICE" release; then
tst_res TWARN "Failed to release device '$TST_DEVICE'"
fi
fi
@@ -24,7 +24,7 @@ cleanup_lvm()
ROD vgremove -y ltp_test_vg2
for devname in $DEVLIST; do
- ROD tst_device release $devname
+ ROD tst_device -d $devname release
done
rm -rf $LVM_DIR/ltp
@@ -43,17 +43,17 @@ prepare_mounts()
FSNAME1=$1
FSNAME2=$2
shift 2
- LVM_DEV1=`tst_device acquire 1040 "$LVM_IMGDIR/lvm_pv1.img"`
+ LVM_DEV1=`tst_device -s 1040 -d "$LVM_IMGDIR/lvm_pv1.img" acquire`
error_check
- LVM_DEV2=`tst_device acquire 1040 "$LVM_IMGDIR/lvm_pv2.img"`
+ LVM_DEV2=`tst_device -s 1040 -d "$LVM_IMGDIR/lvm_pv2.img" acquire`
error_check
# DEVSIZE=($# * 1GB / 2) + 16MB. The extra 16MB is for LVM physical
# volume headers
DEVSIZE=$(( $# * 512 + 16 ))
- LVM_DEV3=`tst_device acquire $DEVSIZE "$LVM_IMGDIR/lvm_pv3.img"`
+ LVM_DEV3=`tst_device -s $DEVSIZE -d "$LVM_IMGDIR/lvm_pv3.img" acquire`
error_check
- LVM_DEV4=`tst_device acquire $DEVSIZE "$LVM_IMGDIR/lvm_pv4.img"`
+ LVM_DEV4=`tst_device -s $DEVSIZE -d "$LVM_IMGDIR/lvm_pv4.img" acquire`
error_check
ROD pvcreate $LVM_DEV1 $LVM_DEV2 $LVM_DEV3 $LVM_DEV4
ROD vgcreate ltp_test_vg1 $LVM_DEV1 $LVM_DEV2
size and filesystems are passed by -s and -f flags. That will help to pass used filesystem. When it, add also -h. Signed-off-by: Petr Vorel <pvorel@suse.cz> --- testcases/lib/tst_device.c | 71 +++++++++++++++++++------------ testcases/lib/tst_test.sh | 2 +- testcases/misc/lvm/cleanup_lvm.sh | 2 +- testcases/misc/lvm/prepare_lvm.sh | 8 ++-- 4 files changed, 49 insertions(+), 34 deletions(-)