tools/testing/nvdimm: stricter bounds checking for error injection commands

[mirror_ubuntu-bionic-kernel.git] / tools / testing / nvdimm / test / nfit.c

diff --git a/tools/testing/nvdimm/test/nfit.c b/tools/testing/nvdimm/test/nfit.c
index 79c5cb23693f33b1701bc8048c76c24688836243..edab68c2e2215724966215607a434a6e6baf1786 100644 (file)
--- a/tools/testing/nvdimm/test/nfit.c
+++ b/tools/testing/nvdimm/test/nfit.c
@@ -488,7 +488,7 @@ static int nfit_test_cmd_ars_error_inject(struct nfit_test *t,
 {
        int rc;
 
-       if (buf_len < sizeof(*err_inj)) {
+       if (buf_len != sizeof(*err_inj)) {
                rc = -EINVAL;
                goto err;
        }
@@ -519,7 +519,7 @@ static int nfit_test_cmd_ars_inject_clear(struct nfit_test *t,
 {
        int rc;
 
-       if (buf_len < sizeof(*err_clr)) {
+       if (buf_len != sizeof(*err_clr)) {
                rc = -EINVAL;
                goto err;
        }