On 12/3/20 6:53 PM, Qiaowei Ren wrote:
> This patch creates the testing module for nvm pages allocator.
> Before this module is loaded, the super block needs to be writen
> into nvdimm device (like /dev/pmemX).
>
> Signed-off-by: Jianpeng Ma <jianpeng.ma@xxxxxxxxx>
> Signed-off-by: Qiaowei Ren <qiaowei.ren@xxxxxxxxx>
The testing patch is OK for me. Although I will not submit it to
mainline kernel, but we do need it during development.
Thanks for this.
Coly Li
> ---
> drivers/md/bcache/Kconfig | 6 ++
> drivers/md/bcache/Makefile | 2 +
> drivers/md/bcache/test-nvm.c | 117 +++++++++++++++++++++++++++++++++++
> 3 files changed, 125 insertions(+)
> create mode 100644 drivers/md/bcache/test-nvm.c
>
> diff --git a/drivers/md/bcache/Kconfig b/drivers/md/bcache/Kconfig
> index 448a99ce13b2..1e4f4ea2f1a0 100644
> --- a/drivers/md/bcache/Kconfig
> +++ b/drivers/md/bcache/Kconfig
> @@ -41,3 +41,9 @@ config BCACHE_NVM_PAGES
> depends on BCACHE
> help
> nvm pages allocator for bcache.
> +
> +config BCACHE_NVM_PAGES_TEST
> + tristate "Testing for NVM pages"
> + depends on BCACHE_NVM_PAGES
> + help
> + Testing module for NVM pages allocator.
> diff --git a/drivers/md/bcache/Makefile b/drivers/md/bcache/Makefile
> index 948e5ed2ca66..7b7d3535f4ef 100644
> --- a/drivers/md/bcache/Makefile
> +++ b/drivers/md/bcache/Makefile
> @@ -5,3 +5,5 @@ obj-$(CONFIG_BCACHE) += bcache.o
> bcache-y := alloc.o bset.o btree.o closure.o debug.o extents.o\
> io.o journal.o movinggc.o request.o stats.o super.o sysfs.o trace.o\
> util.o writeback.o features.o nvm-pages.o
> +
> +obj-$(CONFIG_BCACHE_NVM_PAGES_TEST) += test-nvm.o
> diff --git a/drivers/md/bcache/test-nvm.c b/drivers/md/bcache/test-nvm.c
> new file mode 100644
> index 000000000000..28133ceaa8fd
> --- /dev/null
> +++ b/drivers/md/bcache/test-nvm.c
> @@ -0,0 +1,117 @@
> +// SPDX-License-Identifier: GPL-2.0
> +
> +#include <linux/init.h>
> +#include <linux/kernel.h>
> +#include <linux/module.h>
> +#include <linux/slab.h>
> +#include <linux/crc32.h>
> +#include <linux/uuid.h>
> +#include <linux/prandom.h>
> +#include <linux/pagemap.h>
> +#include <linux/pfn_t.h>
> +#include "nvm-pages.h"
> +
> +static char *host = "NVDIMM device name";
> +module_param(host, charp, 0444);
> +
> +#define MAX_OWNER 10
> +
> +static pgoff_t vaddr_to_nvm_pgoff(struct nvm_namespace *ns, void *kaddr)
> +{
> + return (kaddr - ns->kaddr - ns->pages_offset) / PAGE_SIZE;
> +}
> +
> +static void print_nvm_extent(struct nvm_alloced_recs *extents)
> +{
> + struct list_head *list = extents->extent_head.next;
> + struct nvm_namespace *ns = extents->ns;
> + struct extent *e;
> + pgoff_t pgoff;
> +
> + while (list != &extents->extent_head) {
> + e = container_of(list, struct extent, list);
> + pgoff = vaddr_to_nvm_pgoff(ns, e->kaddr);
> + pr_info(" [%ld ~ %u)", pgoff, e->nr);
> + list = list->next;
> + }
> + pr_info("\n");
> +}
> +
> +static void print_owner_list_info(struct nvm_set *nvm_set, bool print_extent)
> +{
> + struct owner_list *owner_list;
> + struct nvm_alloced_recs *extents;
> + int i, j;
> +
> + for (i = 0; i < nvm_set->owner_list_size; i++) {
> + owner_list = nvm_set->owner_lists[i];
> + pr_info("owner uuid=%pU\n", owner_list->owner_uuid);
> + for (j = 0; j < nvm_set->total_namespaces_nr; j++) {
> + if (owner_list->alloced_recs[j]) {
> + extents = owner_list->alloced_recs[j];
> + pr_info("\t nvm uuid=%pU, allocated extents=%u\n",
> + extents->ns->uuid, extents->size);
> + if (print_extent)
> + print_nvm_extent(extents);
> + }
> + }
> + }
> +}
> +
> +static void test_case(struct nvm_set *nvm_set, char **owner_uuids)
> +{
> + int i, order;
> + void *addr[MAX_OWNER];
> +
> + for (i = 0; i < MAX_OWNER; i++) {
> + order = prandom_u32() % MAX_ORDER;
> + addr[i] = nvm_alloc_pages(order, owner_uuids[i]);
> + }
> + print_owner_list_info(nvm_set, true);
> + for (i = 0; i < MAX_OWNER; i++) {
> + struct page *page = virt_to_page(addr[i]);
> +
> + nvm_free_pages(addr[i], page->private, owner_uuids[i]);
> + }
> + print_owner_list_info(nvm_set, true);
> +}
> +
> +static int __init test_nvm_init(void)
> +{
> + char **owner_uuids;
> + struct nvm_set *nvm_set;
> + int i, r = 0;
> + struct nvm_namespace *ns = register_namespace(host);
> +
> + pr_info("nvm pages test enter: %s\n", host);
> + if (IS_ERR(ns)) {
> + pr_info("failed to register namespace: %s\n", host);
> + r = -EINVAL;
> + goto err;
> + }
> +
> + owner_uuids = kcalloc(MAX_OWNER, sizeof(char *), GFP_KERNEL);
> + for (i = 0; i < MAX_OWNER; i++) {
> + owner_uuids[i] = kmalloc(16, GFP_KERNEL);
> + generate_random_uuid(owner_uuids[i]);
> + }
> +
> + nvm_set = ns->nvm_set;
> + test_case(nvm_set, owner_uuids);
> +
> + for (i = 0; i < MAX_OWNER; i++)
> + kfree(owner_uuids[i]);
> + kfree(owner_uuids);
> +
> +err:
> + return r;
> +}
> +module_init(test_nvm_init);
> +
> +static void __exit test_nvm_exit(void)
> +{
> + pr_info("nvm pages test exit\n");
> +}
> +module_exit(test_nvm_exit);
> +
> +MODULE_LICENSE("GPL v2");
>
