xref: /aosp_15_r20/external/ltp/testcases/lib/tst_device.c (revision 49cdfc7efb34551c7342be41a7384b9c40d7cab7)

// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * Copyright (c) 2017 Cyril Hrubis <[email protected]>
 */

#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:\n");
	fprintf(stderr, "tst_device acquire [size [filename]]\n");
	fprintf(stderr, "tst_device release /path/to/device\n");
	fprintf(stderr, "tst_device clear /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]);
}

static int clear_device(int argc, char *argv[])
{
	if (argc != 3)
		return 1;

	if (tst_clear_device(argv[2]))
		return 1;

	return 0;
}

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 if (!strcmp(argv[1], "clear")) {
		if (clear_device(argc, argv))
			goto help;
	} else {
		fprintf(stderr, "ERROR: Invalid COMMAND '%s'\n", argv[1]);
		goto help;
	}

	return 0;
help:
	print_help();
	return 1;
}