On 25.01.21 03:58, Anshuman Khandual wrote: > From: David Hildenbrand <david@xxxxxxxxxx> > > Right now, we only check against MAX_PHYSMEM_BITS - but turns out there > are more restrictions of which memory we can actually hotplug, especially > om arm64 or s390x once we support them: we might receive something like > -E2BIG or -ERANGE from add_memory_driver_managed(), stopping device > operation. > > So, check right when initializing the device which memory we can add, > warning the user. Try only adding actually pluggable ranges: in the worst > case, no memory provided by our device is pluggable. > > In the usual case, we expect all device memory to be pluggable, and in > corner cases only some memory at the end of the device-managed memory > region to not be pluggable. > > Cc: "Michael S. Tsirkin" <mst@xxxxxxxxxx> > Cc: Jason Wang <jasowang@xxxxxxxxxx> > Cc: Pankaj Gupta <pankaj.gupta.linux@xxxxxxxxx> > Cc: Oscar Salvador <osalvador@xxxxxxx> > Cc: Wei Yang <richard.weiyang@xxxxxxxxxxxxxxxxx> > Cc: Andrew Morton <akpm@xxxxxxxxxxxxxxxxxxxx> > Cc: catalin.marinas@xxxxxxx > Cc: teawater <teawaterz@xxxxxxxxxxxxxxxxx> > Cc: Anshuman Khandual <anshuman.khandual@xxxxxxx> > Cc: Pankaj Gupta <pankaj.gupta@xxxxxxxxxxxxxxx> > Cc: Jonathan Cameron <Jonathan.Cameron@xxxxxxxxxx> > Cc: hca@xxxxxxxxxxxxx > Cc: Vasily Gorbik <gor@xxxxxxxxxxxxx> > Cc: Will Deacon <will@xxxxxxxxxx> > Cc: Ard Biesheuvel <ardb@xxxxxxxxxx> > Cc: Mark Rutland <mark.rutland@xxxxxxx> > Cc: Heiko Carstens <hca@xxxxxxxxxxxxx> > Cc: Michal Hocko <mhocko@xxxxxxxxxx> > Signed-off-by: David Hildenbrand <david@xxxxxxxxxx> > Signed-off-by: Anshuman Khandual <anshuman.khandual@xxxxxxx> > --- > drivers/virtio/virtio_mem.c | 40 +++++++++++++++++++++++++------------ > 1 file changed, 27 insertions(+), 13 deletions(-) > > diff --git a/drivers/virtio/virtio_mem.c b/drivers/virtio/virtio_mem.c > index 9fc9ec4a25f5..14c17c5c1695 100644 > --- a/drivers/virtio/virtio_mem.c > +++ b/drivers/virtio/virtio_mem.c > @@ -2222,7 +2222,7 @@ static int virtio_mem_unplug_pending_mb(struct virtio_mem *vm) > */ > static void virtio_mem_refresh_config(struct virtio_mem *vm) > { > - const uint64_t phys_limit = 1UL << MAX_PHYSMEM_BITS; > + const struct range pluggable_range = mhp_get_pluggable_range(true); > uint64_t new_plugged_size, usable_region_size, end_addr; > > /* the plugged_size is just a reflection of what _we_ did previously */ > @@ -2234,15 +2234,25 @@ static void virtio_mem_refresh_config(struct virtio_mem *vm) > /* calculate the last usable memory block id */ > virtio_cread_le(vm->vdev, struct virtio_mem_config, > usable_region_size, &usable_region_size); > - end_addr = vm->addr + usable_region_size; > - end_addr = min(end_addr, phys_limit); > + end_addr = min(vm->addr + usable_region_size - 1, > + pluggable_range.end); > > - if (vm->in_sbm) > - vm->sbm.last_usable_mb_id = > - virtio_mem_phys_to_mb_id(end_addr) - 1; > - else > - vm->bbm.last_usable_bb_id = > - virtio_mem_phys_to_bb_id(vm, end_addr) - 1; > + if (vm->in_sbm) { > + vm->sbm.last_usable_mb_id = virtio_mem_phys_to_mb_id(end_addr); > + if (!IS_ALIGNED(end_addr + 1, memory_block_size_bytes())) > + vm->sbm.last_usable_mb_id--; > + } else { > + vm->bbm.last_usable_bb_id = virtio_mem_phys_to_bb_id(vm, > + end_addr); > + if (!IS_ALIGNED(end_addr + 1, vm->bbm.bb_size)) > + vm->bbm.last_usable_bb_id--; > + } > + /* > + * If we cannot plug any of our device memory (e.g., nothing in the > + * usable region is addressable), the last usable memory block id will > + * be smaller than the first usable memory block id. We'll stop > + * attempting to add memory with -ENOSPC from our main loop. > + */ > > /* see if there is a request to change the size */ > virtio_cread_le(vm->vdev, struct virtio_mem_config, requested_size, > @@ -2364,6 +2374,7 @@ static int virtio_mem_init_vq(struct virtio_mem *vm) > > static int virtio_mem_init(struct virtio_mem *vm) > { > + const struct range pluggable_range = mhp_get_pluggable_range(true); > const uint64_t phys_limit = 1UL << MAX_PHYSMEM_BITS; Sorry, forgot to drop ^ (phys_limit), otherwise ther is a friendly warning from the compiler. We have to drop that line. Apart from that, at least on x86-64 it does what it's supposed to do. I temporarily changed MAX_PHYSMEM_BITS to 35 bits and added a virtio-mem device that crosses the 32 GiB address limit. [ 0.572084] virtio_mem virtio1: Some device memory is not addressable/pluggable. This can make some memory unusable. [ 0.573013] virtio_mem virtio1: start address: 0x740000000 [ 0.573497] virtio_mem virtio1: region size: 0x500000000 And virtio-mem won't add any memory exceeding that: (qemu) qom-set vmem0 requested-size 20G (qemu) info memory-devices Memory device [virtio-mem]: "vmem0" [...] Memory device [virtio-mem]: "vmem1" memaddr: 0x740000000 node: 1 requested-size: 21474836480 size: 3221225472 max-size: 21474836480 block-size: 2097152 memdev: /objects/mem1 I adds all memory up to the 32GiB address limit (35 bits) and stops. LGTM (arm64 to be tested in the future once supported). -- Thanks, David / dhildenb