mirror of https://github.com/F-Stack/f-stack.git
116 lines
2.3 KiB
C
116 lines
2.3 KiB
C
/* SPDX-License-Identifier: BSD-3-Clause
|
|
* Copyright (c) 2022 Marvell.
|
|
*/
|
|
|
|
#include <rte_debug.h>
|
|
#include <rte_eal.h>
|
|
#include <rte_mldev.h>
|
|
|
|
#include "ml_common.h"
|
|
#include "ml_test.h"
|
|
|
|
struct ml_options opt;
|
|
struct ml_test *test;
|
|
|
|
int
|
|
main(int argc, char **argv)
|
|
{
|
|
uint16_t mldevs;
|
|
int ret;
|
|
|
|
ret = rte_eal_init(argc, argv);
|
|
if (ret < 0)
|
|
rte_panic("invalid EAL arguments\n");
|
|
argc -= ret;
|
|
argv += ret;
|
|
|
|
mldevs = rte_ml_dev_count();
|
|
if (!mldevs) {
|
|
ml_err("no mldev devices found\n");
|
|
goto error;
|
|
}
|
|
|
|
/* set default values for options */
|
|
ml_options_default(&opt);
|
|
|
|
/* parse the command line arguments */
|
|
ret = ml_options_parse(&opt, argc, argv);
|
|
if (ret) {
|
|
ml_err("parsing one or more user options failed");
|
|
goto error;
|
|
}
|
|
|
|
/* get test struct from name */
|
|
test = ml_test_get(opt.test_name);
|
|
if (test == NULL) {
|
|
ml_err("failed to find requested test: %s", opt.test_name);
|
|
goto error;
|
|
}
|
|
|
|
if (test->ops.test_result == NULL) {
|
|
ml_err("%s: ops.test_result not found", opt.test_name);
|
|
goto error;
|
|
}
|
|
|
|
/* check test options */
|
|
if (test->ops.opt_check) {
|
|
if (test->ops.opt_check(&opt)) {
|
|
ml_err("invalid command line argument");
|
|
goto error;
|
|
}
|
|
}
|
|
|
|
/* check the device capability */
|
|
if (test->ops.cap_check) {
|
|
if (test->ops.cap_check(&opt) == false) {
|
|
ml_info("unsupported test: %s", opt.test_name);
|
|
ret = ML_TEST_UNSUPPORTED;
|
|
goto no_cap;
|
|
}
|
|
}
|
|
|
|
/* dump options */
|
|
if (opt.debug) {
|
|
if (test->ops.opt_dump)
|
|
test->ops.opt_dump(&opt);
|
|
}
|
|
|
|
/* test specific setup */
|
|
if (test->ops.test_setup) {
|
|
if (test->ops.test_setup(test, &opt)) {
|
|
ml_err("failed to setup test: %s", opt.test_name);
|
|
goto error;
|
|
}
|
|
}
|
|
|
|
/* test driver */
|
|
if (test->ops.test_driver)
|
|
test->ops.test_driver(test, &opt);
|
|
|
|
/* get result */
|
|
if (test->ops.test_result)
|
|
ret = test->ops.test_result(test, &opt);
|
|
|
|
if (test->ops.test_destroy)
|
|
test->ops.test_destroy(test, &opt);
|
|
|
|
no_cap:
|
|
if (ret == ML_TEST_SUCCESS) {
|
|
printf("Result: " CLGRN "%s" CLNRM "\n", "Success");
|
|
} else if (ret == ML_TEST_FAILED) {
|
|
printf("Result: " CLRED "%s" CLNRM "\n", "Failed");
|
|
return EXIT_FAILURE;
|
|
} else if (ret == ML_TEST_UNSUPPORTED) {
|
|
printf("Result: " CLYEL "%s" CLNRM "\n", "Unsupported");
|
|
}
|
|
|
|
rte_eal_cleanup();
|
|
|
|
return 0;
|
|
|
|
error:
|
|
rte_eal_cleanup();
|
|
|
|
return EXIT_FAILURE;
|
|
}
|