Adding a unit test that will set the acknowledge shutdown count enable.

Signed-off-by: Dave Jiang <[email protected]>
---
 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);
+}

_______________________________________________
Linux-nvdimm mailing list
[email protected]
https://lists.01.org/mailman/listinfo/linux-nvdimm

Reply via email to