u64 ndr_start;
        int id, num_lanes, ro, numa_node;
        void *provider_data;
+       struct badblocks bb;
        struct nd_interleave_set *nd_set;
        struct nd_percpu_lane __percpu *lane;
        struct nd_mapping mapping[0];
 
 #include <linux/module.h>
 #include <linux/device.h>
 #include <linux/nd.h>
+#include "nd-core.h"
 #include "nd.h"
 
 static int nd_region_probe(struct device *dev)
        if (rc && err && rc == err)
                return -ENODEV;
 
+       if (is_nd_pmem(&nd_region->dev)) {
+               struct resource ndr_res;
+
+               if (devm_init_badblocks(dev, &nd_region->bb))
+                       return -ENODEV;
+               ndr_res.start = nd_region->ndr_start;
+               ndr_res.end = nd_region->ndr_start + nd_region->ndr_size - 1;
+               nvdimm_badblocks_populate(nd_region,
+                               &nd_region->bb, &ndr_res);
+       }
+
        nd_region->btt_seed = nd_btt_create(nd_region);
        nd_region->pfn_seed = nd_pfn_create(nd_region);
        nd_region->dax_seed = nd_dax_create(nd_region);
 
 static void nd_region_notify(struct device *dev, enum nvdimm_event event)
 {
+       if (event == NVDIMM_REVALIDATE_POISON) {
+               struct nd_region *nd_region = to_nd_region(dev);
+               struct resource res;
+
+               if (is_nd_pmem(&nd_region->dev)) {
+                       res.start = nd_region->ndr_start;
+                       res.end = nd_region->ndr_start +
+                               nd_region->ndr_size - 1;
+                       nvdimm_badblocks_populate(nd_region,
+                                       &nd_region->bb, &res);
+               }
+       }
        device_for_each_child(dev, &event, child_notify);
 }
 
 
 }
 static DEVICE_ATTR_RW(read_only);
 
+static ssize_t nd_badblocks_show(struct device *dev,
+               struct device_attribute *attr, char *buf)
+{
+       struct nd_region *nd_region = to_nd_region(dev);
+
+       return badblocks_show(&nd_region->bb, buf, 0);
+}
+static struct device_attribute dev_attr_nd_badblocks = {
+       .attr = {
+               .name = "badblocks",
+               .mode = S_IRUGO
+       },
+       .show = nd_badblocks_show,
+};
+
 static struct attribute *nd_region_attributes[] = {
        &dev_attr_size.attr,
        &dev_attr_nstype.attr,
        &dev_attr_available_size.attr,
        &dev_attr_namespace_seed.attr,
        &dev_attr_init_namespaces.attr,
+       &dev_attr_nd_badblocks.attr,
        NULL,
 };
 
        if (!is_nd_pmem(dev) && a == &dev_attr_dax_seed.attr)
                return 0;
 
+       if (!is_nd_pmem(dev) && a == &dev_attr_nd_badblocks.attr)
+               return 0;
+
        if (a != &dev_attr_set_cookie.attr
                        && a != &dev_attr_available_size.attr)
                return a->mode;