9:32 a.m.
Adding a unit test that will set the acknowledge shutdown count enable.
Signed-off-by: Dave Jiang <dave.jiang(a)intel.com>
---
test/Makefile.am | 12 +++
test/ack-shutdown-count-set.c | 144 +++++++++++++++++++++++++++++++++++++++++
2 files changed, 154 insertions(+), 2 deletions(-)
create mode 100644 test/ack-shutdown-count-set.c
diff --git a/test/Makefile.am b/test/Makefile.am
index a473674..749055c 100644
--- a/test/Makefile.am
+++ b/test/Makefile.am
@@ -19,7 +19,8 @@ TESTS =\
btt-errors.sh \
hugetlb \
btt-pad-compat.sh \
- firmware-update.sh
+ firmware-update.sh \
+ ack-shutdown-count-set
check_PROGRAMS =\
libndctl \
@@ -31,7 +32,8 @@ check_PROGRAMS =\
smart-notify \
smart-listen \
hugetlb \
- daxdev-errors
+ daxdev-errors \
+ ack-shutdown-count-set
if ENABLE_DESTRUCTIVE
TESTS +=\
@@ -70,6 +72,12 @@ dsm_fail_SOURCES =\
dsm_fail_LDADD = $(LIBNDCTL_LIB) $(KMOD_LIBS)
+ack_shutdown_count_set_SOURCES =\
+ ack-shutdown-count-set.c \
+ $(testcore)
+
+ack_shutdown_count_set_LDADD = $(LIBNDCTL_LIB) $(KMOD_LIBS)
+
blk_ns_SOURCES = blk_namespaces.c $(testcore)
blk_ns_LDADD = $(LIBNDCTL_LIB) $(KMOD_LIBS)
diff --git a/test/ack-shutdown-count-set.c b/test/ack-shutdown-count-set.c
new file mode 100644
index 0000000..ef38176
--- /dev/null
+++ b/test/ack-shutdown-count-set.c
@@ -0,0 +1,144 @@
+/* SPDX-License-Identifier: GPL-2.0 */
+/* Copyright(c) 2018 Intel Corporation. All rights reserved. */
+
+#include <stdio.h>
+#include <stddef.h>
+#include <stdlib.h>
+#include <string.h>
+#include <ctype.h>
+#include <errno.h>
+#include <syslog.h>
+#include <libkmod.h>
+#include <util/log.h>
+#include <util/sysfs.h>
+#include <linux/version.h>
+
+#include <ccan/array_size/array_size.h>
+#include <ndctl/libndctl.h>
+#ifdef HAVE_NDCTL_H
+#include <linux/ndctl.h>
+#else
+#include <ndctl.h>
+#endif
+#include <test.h>
+
+static int test_dimm(struct ndctl_dimm *dimm)
+{
+ struct ndctl_cmd *cmd;
+ int rc = 0;
+
+ cmd = ndctl_dimm_cmd_new_ack_shutdown_count(dimm);
+ if (!cmd)
+ return -ENOMEM;
+
+ rc = ndctl_cmd_submit(cmd);
+ if (rc < 0)
+ goto out;
+
+ rc = ndctl_cmd_get_firmware_status(cmd);
+ if (rc != 0) {
+ fprintf(stderr, "dimm %s LSS enable set failed\n",
+ ndctl_dimm_get_devname(dimm));
+ goto out;
+ }
+
+ printf("DIMM %s LSS enable set\n", ndctl_dimm_get_devname(dimm));
+
+out:
+ ndctl_cmd_unref(cmd);
+ return rc;
+}
+
+static void reset_bus(struct ndctl_bus *bus)
+{
+ struct ndctl_region *region;
+ struct ndctl_dimm *dimm;
+
+ /* disable all regions so that set_config_data commands are permitted */
+ ndctl_region_foreach(bus, region)
+ ndctl_region_disable_invalidate(region);
+
+ ndctl_dimm_foreach(bus, dimm)
+ ndctl_dimm_zero_labels(dimm);
+}
+
+static int do_test(struct ndctl_ctx *ctx, struct ndctl_test *test)
+{
+ struct ndctl_bus *bus = ndctl_bus_get_by_provider(ctx, "nfit_test.0");
+ struct ndctl_dimm *dimm;
+ struct ndctl_region *region;
+ struct log_ctx log_ctx;
+ int rc = 0;
+
+ if (!ndctl_test_attempt(test, KERNEL_VERSION(4, 15, 0)))
+ return 77;
+
+ if (!bus)
+ return -ENXIO;
+
+ log_init(&log_ctx, "test/ack-shutdown-count-set",
"NDCTL_TEST");
+
+ ndctl_bus_wait_probe(bus);
+
+ ndctl_region_foreach(bus, region)
+ ndctl_region_disable_invalidate(region);
+
+ ndctl_dimm_foreach(bus, dimm) {
+ fprintf(stderr, "Testing dimm: %s\n",
+ ndctl_dimm_get_devname(dimm));
+ rc = test_dimm(dimm);
+ if (rc < 0) {
+ fprintf(stderr, "dimm %s failed\n",
+ ndctl_dimm_get_devname(dimm));
+ goto out;
+ }
+ }
+
+out:
+ reset_bus(bus);
+ return rc;
+}
+
+static int test_ack_shutdown_count_set(int loglevel, struct ndctl_test *test,
+ struct ndctl_ctx *ctx)
+{
+ struct kmod_module *mod;
+ struct kmod_ctx *kmod_ctx;
+ int result = EXIT_FAILURE, err;
+
+ ndctl_set_log_priority(ctx, loglevel);
+ err = nfit_test_init(&kmod_ctx, &mod, NULL, loglevel, test);
+ if (err < 0) {
+ result = 77;
+ ndctl_test_skip(test);
+ fprintf(stderr, "%s unavailable skipping tests\n",
+ "nfit_test");
+ return result;
+ }
+
+ result = do_test(ctx, test);
+ kmod_module_remove_module(mod, 0);
+
+ kmod_unref(kmod_ctx);
+ return result;
+}
+
+int main(int argc, char *argv[])
+{
+ struct ndctl_test *test = ndctl_test_new(0);
+ struct ndctl_ctx *ctx;
+ int rc;
+
+ if (!test) {
+ fprintf(stderr, "failed to initialize test\n");
+ return EXIT_FAILURE;
+ }
+
+ rc = ndctl_new(&ctx);
+ if (rc)
+ return ndctl_test_result(test, rc);
+ rc = test_ack_shutdown_count_set(LOG_DEBUG, test, ctx);
+ ndctl_unref(ctx);
+
+ return ndctl_test_result(test, rc);
+}