module_param_named(zone_size, g_zone_size, ulong, S_IRUGO);
 MODULE_PARM_DESC(zone_size, "Zone size in MB when block device is zoned. Must be power-of-two: Default: 256");
 
+static unsigned long g_zone_capacity;
+module_param_named(zone_capacity, g_zone_capacity, ulong, 0444);
+MODULE_PARM_DESC(zone_capacity, "Zone capacity in MB when block device is zoned. Can be less than or equal to zone size. Default: Zone size");
+
 static unsigned int g_zone_nr_conv;
 module_param_named(zone_nr_conv, g_zone_nr_conv, uint, 0444);
 MODULE_PARM_DESC(zone_nr_conv, "Number of conventional zones when block device is zoned. Default: 0");
 NULLB_DEVICE_ATTR(cache_size, ulong, NULL);
 NULLB_DEVICE_ATTR(zoned, bool, NULL);
 NULLB_DEVICE_ATTR(zone_size, ulong, NULL);
+NULLB_DEVICE_ATTR(zone_capacity, ulong, NULL);
 NULLB_DEVICE_ATTR(zone_nr_conv, uint, NULL);
 
 static ssize_t nullb_device_power_show(struct config_item *item, char *page)
        &nullb_device_attr_badblocks,
        &nullb_device_attr_zoned,
        &nullb_device_attr_zone_size,
+       &nullb_device_attr_zone_capacity,
        &nullb_device_attr_zone_nr_conv,
        NULL,
 };
 
 static ssize_t memb_group_features_show(struct config_item *item, char *page)
 {
-       return snprintf(page, PAGE_SIZE, "memory_backed,discard,bandwidth,cache,badblocks,zoned,zone_size,zone_nr_conv\n");
+       return snprintf(page, PAGE_SIZE,
+                       "memory_backed,discard,bandwidth,cache,badblocks,zoned,zone_size,zone_capacity,zone_nr_conv\n");
 }
 
 CONFIGFS_ATTR_RO(memb_group_, features);
        dev->use_per_node_hctx = g_use_per_node_hctx;
        dev->zoned = g_zoned;
        dev->zone_size = g_zone_size;
+       dev->zone_capacity = g_zone_capacity;
        dev->zone_nr_conv = g_zone_nr_conv;
        return dev;
 }
 
                return -EINVAL;
        }
 
+       if (!dev->zone_capacity)
+               dev->zone_capacity = dev->zone_size;
+
+       if (dev->zone_capacity > dev->zone_size) {
+               pr_err("null_blk: zone capacity (%lu MB) larger than zone size (%lu MB)\n",
+                                       dev->zone_capacity, dev->zone_size);
+               return -EINVAL;
+       }
+
        dev->zone_size_sects = dev->zone_size << ZONE_SIZE_SHIFT;
        dev->nr_zones = dev_size >>
                                (SECTOR_SHIFT + ilog2(dev->zone_size_sects));
 
                zone->start = zone->wp = sector;
                zone->len = dev->zone_size_sects;
-               zone->capacity = zone->len;
+               zone->capacity = dev->zone_capacity << ZONE_SIZE_SHIFT;
                zone->type = BLK_ZONE_TYPE_SEQWRITE_REQ;
                zone->cond = BLK_ZONE_COND_EMPTY;
 
                        return BLK_STS_IOERR;
                }
 
+               if (zone->wp + nr_sectors > zone->start + zone->capacity)
+                       return BLK_STS_IOERR;
+
                if (zone->cond != BLK_ZONE_COND_EXP_OPEN)
                        zone->cond = BLK_ZONE_COND_IMP_OPEN;
 
                        return ret;
 
                zone->wp += nr_sectors;
-               if (zone->wp == zone->start + zone->len)
+               if (zone->wp == zone->start + zone->capacity)
                        zone->cond = BLK_ZONE_COND_FULL;
                return BLK_STS_OK;
        default: