108 lines
2.0 KiB
C
108 lines
2.0 KiB
C
// SPDX-License-Identifier: GPL-2.0-or-later
|
|
/*
|
|
* Copyright (c) 2017 Cyril Hrubis <chrubis@suse.cz>
|
|
*/
|
|
|
|
#include <errno.h>
|
|
#include <limits.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#define TST_NO_DEFAULT_MAIN
|
|
#include "tst_test.h"
|
|
#include "old/old_device.h"
|
|
|
|
extern struct tst_test *tst_test;
|
|
|
|
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");
|
|
}
|
|
|
|
static int acquire_device(int argc, char *argv[])
|
|
{
|
|
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]);
|
|
} else {
|
|
device = tst_acquire_device__(size);
|
|
}
|
|
|
|
if (!device)
|
|
return 1;
|
|
|
|
if (tst_clear_device(device)) {
|
|
tst_release_device(device);
|
|
return 1;
|
|
}
|
|
|
|
printf("%s", device);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int release_device(int argc, char *argv[])
|
|
{
|
|
if (argc != 3)
|
|
return 1;
|
|
|
|
/*
|
|
* tst_acquire_[loop_]device() was called in a different process.
|
|
* tst_release_device() would think that no device was acquired yet
|
|
* and do nothing. Call tst_detach_device() directly to bypass
|
|
* the check.
|
|
*/
|
|
return tst_detach_device(argv[2]);
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
/*
|
|
* Force messages to be printed from the new library i.e. tst_test.c
|
|
*
|
|
* The new library prints messages into stderr while the old one prints
|
|
* them into stdout. When messages are printed into stderr we can
|
|
* safely do:
|
|
*
|
|
* DEV=$(tst_device acquire)
|
|
*/
|
|
tst_test = &test;
|
|
|
|
if (argc < 2)
|
|
goto help;
|
|
|
|
if (!strcmp(argv[1], "acquire")) {
|
|
if (acquire_device(argc, argv))
|
|
goto help;
|
|
} else if (!strcmp(argv[1], "release")) {
|
|
if (release_device(argc, argv))
|
|
goto help;
|
|
} else {
|
|
fprintf(stderr, "ERROR: Invalid COMMAND '%s'\n", argv[1]);
|
|
goto help;
|
|
}
|
|
|
|
return 0;
|
|
help:
|
|
print_help();
|
|
return 1;
|
|
}
|