Lines Matching refs:vm
276 static void virtio_mem_retry(struct virtio_mem *vm);
277 static int virtio_mem_create_resource(struct virtio_mem *vm);
278 static void virtio_mem_delete_resource(struct virtio_mem *vm);
284 static int register_virtio_mem_device(struct virtio_mem *vm) in register_virtio_mem_device() argument
293 list_add_rcu(&vm->next, &virtio_mem_devices); in register_virtio_mem_device()
303 static void unregister_virtio_mem_device(struct virtio_mem *vm) in unregister_virtio_mem_device() argument
307 list_del_rcu(&vm->next); in unregister_virtio_mem_device()
334 static unsigned long virtio_mem_phys_to_bb_id(struct virtio_mem *vm, in virtio_mem_phys_to_bb_id() argument
337 return addr / vm->bbm.bb_size; in virtio_mem_phys_to_bb_id()
343 static uint64_t virtio_mem_bb_id_to_phys(struct virtio_mem *vm, in virtio_mem_bb_id_to_phys() argument
346 return bb_id * vm->bbm.bb_size; in virtio_mem_bb_id_to_phys()
352 static unsigned long virtio_mem_phys_to_sb_id(struct virtio_mem *vm, in virtio_mem_phys_to_sb_id() argument
358 return (addr - mb_addr) / vm->sbm.sb_size; in virtio_mem_phys_to_sb_id()
364 static void virtio_mem_bbm_set_bb_state(struct virtio_mem *vm, in virtio_mem_bbm_set_bb_state() argument
368 const unsigned long idx = bb_id - vm->bbm.first_bb_id; in virtio_mem_bbm_set_bb_state()
371 old_state = vm->bbm.bb_states[idx]; in virtio_mem_bbm_set_bb_state()
372 vm->bbm.bb_states[idx] = state; in virtio_mem_bbm_set_bb_state()
374 BUG_ON(vm->bbm.bb_count[old_state] == 0); in virtio_mem_bbm_set_bb_state()
375 vm->bbm.bb_count[old_state]--; in virtio_mem_bbm_set_bb_state()
376 vm->bbm.bb_count[state]++; in virtio_mem_bbm_set_bb_state()
382 static enum virtio_mem_bbm_bb_state virtio_mem_bbm_get_bb_state(struct virtio_mem *vm, in virtio_mem_bbm_get_bb_state() argument
385 return vm->bbm.bb_states[bb_id - vm->bbm.first_bb_id]; in virtio_mem_bbm_get_bb_state()
391 static int virtio_mem_bbm_bb_states_prepare_next_bb(struct virtio_mem *vm) in virtio_mem_bbm_bb_states_prepare_next_bb() argument
393 unsigned long old_bytes = vm->bbm.next_bb_id - vm->bbm.first_bb_id; in virtio_mem_bbm_bb_states_prepare_next_bb()
399 if (vm->bbm.bb_states && old_pages == new_pages) in virtio_mem_bbm_bb_states_prepare_next_bb()
406 mutex_lock(&vm->hotplug_mutex); in virtio_mem_bbm_bb_states_prepare_next_bb()
407 if (vm->bbm.bb_states) in virtio_mem_bbm_bb_states_prepare_next_bb()
408 memcpy(new_array, vm->bbm.bb_states, old_pages * PAGE_SIZE); in virtio_mem_bbm_bb_states_prepare_next_bb()
409 vfree(vm->bbm.bb_states); in virtio_mem_bbm_bb_states_prepare_next_bb()
410 vm->bbm.bb_states = new_array; in virtio_mem_bbm_bb_states_prepare_next_bb()
411 mutex_unlock(&vm->hotplug_mutex); in virtio_mem_bbm_bb_states_prepare_next_bb()
417 for (_bb_id = vm->bbm.first_bb_id; \
418 _bb_id < vm->bbm.next_bb_id && _vm->bbm.bb_count[_state]; \
423 for (_bb_id = vm->bbm.next_bb_id - 1; \
424 _bb_id >= vm->bbm.first_bb_id && _vm->bbm.bb_count[_state]; \
431 static void virtio_mem_sbm_set_mb_state(struct virtio_mem *vm, in virtio_mem_sbm_set_mb_state() argument
434 const unsigned long idx = mb_id - vm->sbm.first_mb_id; in virtio_mem_sbm_set_mb_state()
437 old_state = vm->sbm.mb_states[idx]; in virtio_mem_sbm_set_mb_state()
438 vm->sbm.mb_states[idx] = state; in virtio_mem_sbm_set_mb_state()
440 BUG_ON(vm->sbm.mb_count[old_state] == 0); in virtio_mem_sbm_set_mb_state()
441 vm->sbm.mb_count[old_state]--; in virtio_mem_sbm_set_mb_state()
442 vm->sbm.mb_count[state]++; in virtio_mem_sbm_set_mb_state()
448 static uint8_t virtio_mem_sbm_get_mb_state(struct virtio_mem *vm, in virtio_mem_sbm_get_mb_state() argument
451 const unsigned long idx = mb_id - vm->sbm.first_mb_id; in virtio_mem_sbm_get_mb_state()
453 return vm->sbm.mb_states[idx]; in virtio_mem_sbm_get_mb_state()
459 static int virtio_mem_sbm_mb_states_prepare_next_mb(struct virtio_mem *vm) in virtio_mem_sbm_mb_states_prepare_next_mb() argument
461 int old_pages = PFN_UP(vm->sbm.next_mb_id - vm->sbm.first_mb_id); in virtio_mem_sbm_mb_states_prepare_next_mb()
462 int new_pages = PFN_UP(vm->sbm.next_mb_id - vm->sbm.first_mb_id + 1); in virtio_mem_sbm_mb_states_prepare_next_mb()
465 if (vm->sbm.mb_states && old_pages == new_pages) in virtio_mem_sbm_mb_states_prepare_next_mb()
472 mutex_lock(&vm->hotplug_mutex); in virtio_mem_sbm_mb_states_prepare_next_mb()
473 if (vm->sbm.mb_states) in virtio_mem_sbm_mb_states_prepare_next_mb()
474 memcpy(new_array, vm->sbm.mb_states, old_pages * PAGE_SIZE); in virtio_mem_sbm_mb_states_prepare_next_mb()
475 vfree(vm->sbm.mb_states); in virtio_mem_sbm_mb_states_prepare_next_mb()
476 vm->sbm.mb_states = new_array; in virtio_mem_sbm_mb_states_prepare_next_mb()
477 mutex_unlock(&vm->hotplug_mutex); in virtio_mem_sbm_mb_states_prepare_next_mb()
498 static int virtio_mem_sbm_sb_state_bit_nr(struct virtio_mem *vm, in virtio_mem_sbm_sb_state_bit_nr() argument
501 return (mb_id - vm->sbm.first_mb_id) * vm->sbm.sbs_per_mb + sb_id; in virtio_mem_sbm_sb_state_bit_nr()
509 static void virtio_mem_sbm_set_sb_plugged(struct virtio_mem *vm, in virtio_mem_sbm_set_sb_plugged() argument
513 const int bit = virtio_mem_sbm_sb_state_bit_nr(vm, mb_id, sb_id); in virtio_mem_sbm_set_sb_plugged()
515 __bitmap_set(vm->sbm.sb_states, bit, count); in virtio_mem_sbm_set_sb_plugged()
523 static void virtio_mem_sbm_set_sb_unplugged(struct virtio_mem *vm, in virtio_mem_sbm_set_sb_unplugged() argument
527 const int bit = virtio_mem_sbm_sb_state_bit_nr(vm, mb_id, sb_id); in virtio_mem_sbm_set_sb_unplugged()
529 __bitmap_clear(vm->sbm.sb_states, bit, count); in virtio_mem_sbm_set_sb_unplugged()
535 static bool virtio_mem_sbm_test_sb_plugged(struct virtio_mem *vm, in virtio_mem_sbm_test_sb_plugged() argument
539 const int bit = virtio_mem_sbm_sb_state_bit_nr(vm, mb_id, sb_id); in virtio_mem_sbm_test_sb_plugged()
542 return test_bit(bit, vm->sbm.sb_states); in virtio_mem_sbm_test_sb_plugged()
545 return find_next_zero_bit(vm->sbm.sb_states, bit + count, bit) >= in virtio_mem_sbm_test_sb_plugged()
552 static bool virtio_mem_sbm_test_sb_unplugged(struct virtio_mem *vm, in virtio_mem_sbm_test_sb_unplugged() argument
556 const int bit = virtio_mem_sbm_sb_state_bit_nr(vm, mb_id, sb_id); in virtio_mem_sbm_test_sb_unplugged()
559 return find_next_bit(vm->sbm.sb_states, bit + count, bit) >= in virtio_mem_sbm_test_sb_unplugged()
567 static int virtio_mem_sbm_first_unplugged_sb(struct virtio_mem *vm, in virtio_mem_sbm_first_unplugged_sb() argument
570 const int bit = virtio_mem_sbm_sb_state_bit_nr(vm, mb_id, 0); in virtio_mem_sbm_first_unplugged_sb()
572 return find_next_zero_bit(vm->sbm.sb_states, in virtio_mem_sbm_first_unplugged_sb()
573 bit + vm->sbm.sbs_per_mb, bit) - bit; in virtio_mem_sbm_first_unplugged_sb()
579 static int virtio_mem_sbm_sb_states_prepare_next_mb(struct virtio_mem *vm) in virtio_mem_sbm_sb_states_prepare_next_mb() argument
581 const unsigned long old_nb_mb = vm->sbm.next_mb_id - vm->sbm.first_mb_id; in virtio_mem_sbm_sb_states_prepare_next_mb()
582 const unsigned long old_nb_bits = old_nb_mb * vm->sbm.sbs_per_mb; in virtio_mem_sbm_sb_states_prepare_next_mb()
583 const unsigned long new_nb_bits = (old_nb_mb + 1) * vm->sbm.sbs_per_mb; in virtio_mem_sbm_sb_states_prepare_next_mb()
588 if (vm->sbm.sb_states && old_pages == new_pages) in virtio_mem_sbm_sb_states_prepare_next_mb()
595 mutex_lock(&vm->hotplug_mutex); in virtio_mem_sbm_sb_states_prepare_next_mb()
596 if (vm->sbm.sb_states) in virtio_mem_sbm_sb_states_prepare_next_mb()
597 memcpy(new_bitmap, vm->sbm.sb_states, old_pages * PAGE_SIZE); in virtio_mem_sbm_sb_states_prepare_next_mb()
599 old_bitmap = vm->sbm.sb_states; in virtio_mem_sbm_sb_states_prepare_next_mb()
600 vm->sbm.sb_states = new_bitmap; in virtio_mem_sbm_sb_states_prepare_next_mb()
601 mutex_unlock(&vm->hotplug_mutex); in virtio_mem_sbm_sb_states_prepare_next_mb()
611 static bool virtio_mem_could_add_memory(struct virtio_mem *vm, uint64_t size) in virtio_mem_could_add_memory() argument
613 if (WARN_ON_ONCE(size > vm->offline_threshold)) in virtio_mem_could_add_memory()
616 return atomic64_read(&vm->offline_size) + size <= vm->offline_threshold; in virtio_mem_could_add_memory()
627 static int virtio_mem_add_memory(struct virtio_mem *vm, uint64_t addr, in virtio_mem_add_memory() argument
636 if (!vm->resource_name) { in virtio_mem_add_memory()
637 vm->resource_name = kstrdup_const("System RAM (virtio_mem)", in virtio_mem_add_memory()
639 if (!vm->resource_name) in virtio_mem_add_memory()
643 dev_dbg(&vm->vdev->dev, "adding memory: 0x%llx - 0x%llx\n", addr, in virtio_mem_add_memory()
646 atomic64_add(size, &vm->offline_size); in virtio_mem_add_memory()
647 rc = add_memory_driver_managed(vm->mgid, addr, size, vm->resource_name, in virtio_mem_add_memory()
650 atomic64_sub(size, &vm->offline_size); in virtio_mem_add_memory()
651 dev_warn(&vm->vdev->dev, "adding memory failed: %d\n", rc); in virtio_mem_add_memory()
663 static int virtio_mem_sbm_add_mb(struct virtio_mem *vm, unsigned long mb_id) in virtio_mem_sbm_add_mb() argument
668 return virtio_mem_add_memory(vm, addr, size); in virtio_mem_sbm_add_mb()
674 static int virtio_mem_bbm_add_bb(struct virtio_mem *vm, unsigned long bb_id) in virtio_mem_bbm_add_bb() argument
676 const uint64_t addr = virtio_mem_bb_id_to_phys(vm, bb_id); in virtio_mem_bbm_add_bb()
677 const uint64_t size = vm->bbm.bb_size; in virtio_mem_bbm_add_bb()
679 return virtio_mem_add_memory(vm, addr, size); in virtio_mem_bbm_add_bb()
691 static int virtio_mem_remove_memory(struct virtio_mem *vm, uint64_t addr, in virtio_mem_remove_memory() argument
696 dev_dbg(&vm->vdev->dev, "removing memory: 0x%llx - 0x%llx\n", addr, in virtio_mem_remove_memory()
700 atomic64_sub(size, &vm->offline_size); in virtio_mem_remove_memory()
705 virtio_mem_retry(vm); in virtio_mem_remove_memory()
707 dev_dbg(&vm->vdev->dev, "removing memory failed: %d\n", rc); in virtio_mem_remove_memory()
715 static int virtio_mem_sbm_remove_mb(struct virtio_mem *vm, unsigned long mb_id) in virtio_mem_sbm_remove_mb() argument
720 return virtio_mem_remove_memory(vm, addr, size); in virtio_mem_sbm_remove_mb()
731 static int virtio_mem_offline_and_remove_memory(struct virtio_mem *vm, in virtio_mem_offline_and_remove_memory() argument
737 dev_dbg(&vm->vdev->dev, in virtio_mem_offline_and_remove_memory()
743 atomic64_sub(size, &vm->offline_size); in virtio_mem_offline_and_remove_memory()
748 virtio_mem_retry(vm); in virtio_mem_offline_and_remove_memory()
750 dev_dbg(&vm->vdev->dev, in virtio_mem_offline_and_remove_memory()
760 static int virtio_mem_sbm_offline_and_remove_mb(struct virtio_mem *vm, in virtio_mem_sbm_offline_and_remove_mb() argument
766 return virtio_mem_offline_and_remove_memory(vm, addr, size); in virtio_mem_sbm_offline_and_remove_mb()
773 static int virtio_mem_bbm_offline_and_remove_bb(struct virtio_mem *vm, in virtio_mem_bbm_offline_and_remove_bb() argument
776 const uint64_t addr = virtio_mem_bb_id_to_phys(vm, bb_id); in virtio_mem_bbm_offline_and_remove_bb()
777 const uint64_t size = vm->bbm.bb_size; in virtio_mem_bbm_offline_and_remove_bb()
779 return virtio_mem_offline_and_remove_memory(vm, addr, size); in virtio_mem_bbm_offline_and_remove_bb()
785 static void virtio_mem_retry(struct virtio_mem *vm) in virtio_mem_retry() argument
789 spin_lock_irqsave(&vm->removal_lock, flags); in virtio_mem_retry()
790 if (!vm->removing) in virtio_mem_retry()
791 queue_work(system_freezable_wq, &vm->wq); in virtio_mem_retry()
792 spin_unlock_irqrestore(&vm->removal_lock, flags); in virtio_mem_retry()
795 static int virtio_mem_translate_node_id(struct virtio_mem *vm, uint16_t node_id) in virtio_mem_translate_node_id() argument
800 if (virtio_has_feature(vm->vdev, VIRTIO_MEM_F_ACPI_PXM)) in virtio_mem_translate_node_id()
810 static bool virtio_mem_overlaps_range(struct virtio_mem *vm, uint64_t start, in virtio_mem_overlaps_range() argument
813 return start < vm->addr + vm->region_size && vm->addr < start + size; in virtio_mem_overlaps_range()
820 static bool virtio_mem_contains_range(struct virtio_mem *vm, uint64_t start, in virtio_mem_contains_range() argument
823 return start >= vm->addr && start + size <= vm->addr + vm->region_size; in virtio_mem_contains_range()
826 static int virtio_mem_sbm_notify_going_online(struct virtio_mem *vm, in virtio_mem_sbm_notify_going_online() argument
829 switch (virtio_mem_sbm_get_mb_state(vm, mb_id)) { in virtio_mem_sbm_notify_going_online()
836 dev_warn_ratelimited(&vm->vdev->dev, in virtio_mem_sbm_notify_going_online()
841 static void virtio_mem_sbm_notify_offline(struct virtio_mem *vm, in virtio_mem_sbm_notify_offline() argument
844 switch (virtio_mem_sbm_get_mb_state(vm, mb_id)) { in virtio_mem_sbm_notify_offline()
847 virtio_mem_sbm_set_mb_state(vm, mb_id, in virtio_mem_sbm_notify_offline()
852 virtio_mem_sbm_set_mb_state(vm, mb_id, in virtio_mem_sbm_notify_offline()
861 static void virtio_mem_sbm_notify_online(struct virtio_mem *vm, in virtio_mem_sbm_notify_online() argument
868 switch (virtio_mem_sbm_get_mb_state(vm, mb_id)) { in virtio_mem_sbm_notify_online()
883 virtio_mem_sbm_set_mb_state(vm, mb_id, new_state); in virtio_mem_sbm_notify_online()
886 static void virtio_mem_sbm_notify_going_offline(struct virtio_mem *vm, in virtio_mem_sbm_notify_going_offline() argument
889 const unsigned long nr_pages = PFN_DOWN(vm->sbm.sb_size); in virtio_mem_sbm_notify_going_offline()
893 for (sb_id = 0; sb_id < vm->sbm.sbs_per_mb; sb_id++) { in virtio_mem_sbm_notify_going_offline()
894 if (virtio_mem_sbm_test_sb_plugged(vm, mb_id, sb_id, 1)) in virtio_mem_sbm_notify_going_offline()
897 sb_id * vm->sbm.sb_size); in virtio_mem_sbm_notify_going_offline()
902 static void virtio_mem_sbm_notify_cancel_offline(struct virtio_mem *vm, in virtio_mem_sbm_notify_cancel_offline() argument
905 const unsigned long nr_pages = PFN_DOWN(vm->sbm.sb_size); in virtio_mem_sbm_notify_cancel_offline()
909 for (sb_id = 0; sb_id < vm->sbm.sbs_per_mb; sb_id++) { in virtio_mem_sbm_notify_cancel_offline()
910 if (virtio_mem_sbm_test_sb_plugged(vm, mb_id, sb_id, 1)) in virtio_mem_sbm_notify_cancel_offline()
913 sb_id * vm->sbm.sb_size); in virtio_mem_sbm_notify_cancel_offline()
918 static void virtio_mem_bbm_notify_going_offline(struct virtio_mem *vm, in virtio_mem_bbm_notify_going_offline() argument
927 if (virtio_mem_bbm_get_bb_state(vm, bb_id) != in virtio_mem_bbm_notify_going_offline()
933 static void virtio_mem_bbm_notify_cancel_offline(struct virtio_mem *vm, in virtio_mem_bbm_notify_cancel_offline() argument
938 if (virtio_mem_bbm_get_bb_state(vm, bb_id) != in virtio_mem_bbm_notify_cancel_offline()
952 struct virtio_mem *vm = container_of(nb, struct virtio_mem, in virtio_mem_memory_notifier_cb() local
960 if (!virtio_mem_overlaps_range(vm, start, size)) in virtio_mem_memory_notifier_cb()
963 if (vm->in_sbm) { in virtio_mem_memory_notifier_cb()
974 id = virtio_mem_phys_to_bb_id(vm, start); in virtio_mem_memory_notifier_cb()
981 if (WARN_ON_ONCE(id != virtio_mem_phys_to_bb_id(vm, start + size - 1))) in virtio_mem_memory_notifier_cb()
995 mutex_lock(&vm->hotplug_mutex); in virtio_mem_memory_notifier_cb()
996 if (vm->removing) { in virtio_mem_memory_notifier_cb()
998 mutex_unlock(&vm->hotplug_mutex); in virtio_mem_memory_notifier_cb()
1001 vm->hotplug_active = true; in virtio_mem_memory_notifier_cb()
1002 if (vm->in_sbm) in virtio_mem_memory_notifier_cb()
1003 virtio_mem_sbm_notify_going_offline(vm, id); in virtio_mem_memory_notifier_cb()
1005 virtio_mem_bbm_notify_going_offline(vm, id, in virtio_mem_memory_notifier_cb()
1010 mutex_lock(&vm->hotplug_mutex); in virtio_mem_memory_notifier_cb()
1011 if (vm->removing) { in virtio_mem_memory_notifier_cb()
1013 mutex_unlock(&vm->hotplug_mutex); in virtio_mem_memory_notifier_cb()
1016 vm->hotplug_active = true; in virtio_mem_memory_notifier_cb()
1017 if (vm->in_sbm) in virtio_mem_memory_notifier_cb()
1018 rc = virtio_mem_sbm_notify_going_online(vm, id); in virtio_mem_memory_notifier_cb()
1021 if (vm->in_sbm) in virtio_mem_memory_notifier_cb()
1022 virtio_mem_sbm_notify_offline(vm, id); in virtio_mem_memory_notifier_cb()
1024 atomic64_add(size, &vm->offline_size); in virtio_mem_memory_notifier_cb()
1030 virtio_mem_retry(vm); in virtio_mem_memory_notifier_cb()
1032 vm->hotplug_active = false; in virtio_mem_memory_notifier_cb()
1033 mutex_unlock(&vm->hotplug_mutex); in virtio_mem_memory_notifier_cb()
1036 if (vm->in_sbm) in virtio_mem_memory_notifier_cb()
1037 virtio_mem_sbm_notify_online(vm, id, mhp->start_pfn); in virtio_mem_memory_notifier_cb()
1039 atomic64_sub(size, &vm->offline_size); in virtio_mem_memory_notifier_cb()
1046 if (!atomic_read(&vm->wq_active) && in virtio_mem_memory_notifier_cb()
1047 virtio_mem_could_add_memory(vm, vm->offline_threshold / 2)) in virtio_mem_memory_notifier_cb()
1048 virtio_mem_retry(vm); in virtio_mem_memory_notifier_cb()
1050 vm->hotplug_active = false; in virtio_mem_memory_notifier_cb()
1051 mutex_unlock(&vm->hotplug_mutex); in virtio_mem_memory_notifier_cb()
1054 if (!vm->hotplug_active) in virtio_mem_memory_notifier_cb()
1056 if (vm->in_sbm) in virtio_mem_memory_notifier_cb()
1057 virtio_mem_sbm_notify_cancel_offline(vm, id); in virtio_mem_memory_notifier_cb()
1059 virtio_mem_bbm_notify_cancel_offline(vm, id, in virtio_mem_memory_notifier_cb()
1062 vm->hotplug_active = false; in virtio_mem_memory_notifier_cb()
1063 mutex_unlock(&vm->hotplug_mutex); in virtio_mem_memory_notifier_cb()
1066 if (!vm->hotplug_active) in virtio_mem_memory_notifier_cb()
1068 vm->hotplug_active = false; in virtio_mem_memory_notifier_cb()
1069 mutex_unlock(&vm->hotplug_mutex); in virtio_mem_memory_notifier_cb()
1231 static void virtio_mem_online_page(struct virtio_mem *vm, in virtio_mem_online_page() argument
1249 if (vm->in_sbm) { in virtio_mem_online_page()
1251 sb_id = virtio_mem_phys_to_sb_id(vm, addr); in virtio_mem_online_page()
1252 count = virtio_mem_phys_to_sb_id(vm, next - 1) - sb_id + 1; in virtio_mem_online_page()
1254 if (virtio_mem_sbm_test_sb_plugged(vm, id, sb_id, count)) { in virtio_mem_online_page()
1258 virtio_mem_sbm_test_sb_unplugged(vm, id, sb_id, count)) { in virtio_mem_online_page()
1267 order = ilog2(vm->sbm.sb_size) - PAGE_SHIFT; in virtio_mem_online_page()
1268 do_online = virtio_mem_sbm_test_sb_plugged(vm, id, sb_id, 1); in virtio_mem_online_page()
1276 id = virtio_mem_phys_to_bb_id(vm, addr); in virtio_mem_online_page()
1277 do_online = virtio_mem_bbm_get_bb_state(vm, id) != in virtio_mem_online_page()
1293 struct virtio_mem *vm; in virtio_mem_online_page_cb() local
1296 list_for_each_entry_rcu(vm, &virtio_mem_devices, next) { in virtio_mem_online_page_cb()
1301 if (!virtio_mem_contains_range(vm, addr, PFN_PHYS(1 << order))) in virtio_mem_online_page_cb()
1312 virtio_mem_online_page(vm, page, order); in virtio_mem_online_page_cb()
1321 static uint64_t virtio_mem_send_request(struct virtio_mem *vm, in virtio_mem_send_request() argument
1329 vm->req = *req; in virtio_mem_send_request()
1332 sg_init_one(&sg_req, &vm->req, sizeof(vm->req)); in virtio_mem_send_request()
1336 sg_init_one(&sg_resp, &vm->resp, sizeof(vm->resp)); in virtio_mem_send_request()
1339 rc = virtqueue_add_sgs(vm->vq, sgs, 1, 1, vm, GFP_KERNEL); in virtio_mem_send_request()
1343 virtqueue_kick(vm->vq); in virtio_mem_send_request()
1346 wait_event(vm->host_resp, virtqueue_get_buf(vm->vq, &len)); in virtio_mem_send_request()
1348 return virtio16_to_cpu(vm->vdev, vm->resp.type); in virtio_mem_send_request()
1351 static int virtio_mem_send_plug_request(struct virtio_mem *vm, uint64_t addr, in virtio_mem_send_plug_request() argument
1354 const uint64_t nb_vm_blocks = size / vm->device_block_size; in virtio_mem_send_plug_request()
1356 .type = cpu_to_virtio16(vm->vdev, VIRTIO_MEM_REQ_PLUG), in virtio_mem_send_plug_request()
1357 .u.plug.addr = cpu_to_virtio64(vm->vdev, addr), in virtio_mem_send_plug_request()
1358 .u.plug.nb_blocks = cpu_to_virtio16(vm->vdev, nb_vm_blocks), in virtio_mem_send_plug_request()
1362 if (atomic_read(&vm->config_changed)) in virtio_mem_send_plug_request()
1365 dev_dbg(&vm->vdev->dev, "plugging memory: 0x%llx - 0x%llx\n", addr, in virtio_mem_send_plug_request()
1368 switch (virtio_mem_send_request(vm, &req)) { in virtio_mem_send_plug_request()
1370 vm->plugged_size += size; in virtio_mem_send_plug_request()
1385 dev_dbg(&vm->vdev->dev, "plugging memory failed: %d\n", rc); in virtio_mem_send_plug_request()
1389 static int virtio_mem_send_unplug_request(struct virtio_mem *vm, uint64_t addr, in virtio_mem_send_unplug_request() argument
1392 const uint64_t nb_vm_blocks = size / vm->device_block_size; in virtio_mem_send_unplug_request()
1394 .type = cpu_to_virtio16(vm->vdev, VIRTIO_MEM_REQ_UNPLUG), in virtio_mem_send_unplug_request()
1395 .u.unplug.addr = cpu_to_virtio64(vm->vdev, addr), in virtio_mem_send_unplug_request()
1396 .u.unplug.nb_blocks = cpu_to_virtio16(vm->vdev, nb_vm_blocks), in virtio_mem_send_unplug_request()
1400 if (atomic_read(&vm->config_changed)) in virtio_mem_send_unplug_request()
1403 dev_dbg(&vm->vdev->dev, "unplugging memory: 0x%llx - 0x%llx\n", addr, in virtio_mem_send_unplug_request()
1406 switch (virtio_mem_send_request(vm, &req)) { in virtio_mem_send_unplug_request()
1408 vm->plugged_size -= size; in virtio_mem_send_unplug_request()
1420 dev_dbg(&vm->vdev->dev, "unplugging memory failed: %d\n", rc); in virtio_mem_send_unplug_request()
1424 static int virtio_mem_send_unplug_all_request(struct virtio_mem *vm) in virtio_mem_send_unplug_all_request() argument
1427 .type = cpu_to_virtio16(vm->vdev, VIRTIO_MEM_REQ_UNPLUG_ALL), in virtio_mem_send_unplug_all_request()
1431 dev_dbg(&vm->vdev->dev, "unplugging all memory"); in virtio_mem_send_unplug_all_request()
1433 switch (virtio_mem_send_request(vm, &req)) { in virtio_mem_send_unplug_all_request()
1435 vm->unplug_all_required = false; in virtio_mem_send_unplug_all_request()
1436 vm->plugged_size = 0; in virtio_mem_send_unplug_all_request()
1438 atomic_set(&vm->config_changed, 1); in virtio_mem_send_unplug_all_request()
1447 dev_dbg(&vm->vdev->dev, "unplugging all memory failed: %d\n", rc); in virtio_mem_send_unplug_all_request()
1455 static int virtio_mem_sbm_plug_sb(struct virtio_mem *vm, unsigned long mb_id, in virtio_mem_sbm_plug_sb() argument
1459 sb_id * vm->sbm.sb_size; in virtio_mem_sbm_plug_sb()
1460 const uint64_t size = count * vm->sbm.sb_size; in virtio_mem_sbm_plug_sb()
1463 rc = virtio_mem_send_plug_request(vm, addr, size); in virtio_mem_sbm_plug_sb()
1465 virtio_mem_sbm_set_sb_plugged(vm, mb_id, sb_id, count); in virtio_mem_sbm_plug_sb()
1473 static int virtio_mem_sbm_unplug_sb(struct virtio_mem *vm, unsigned long mb_id, in virtio_mem_sbm_unplug_sb() argument
1477 sb_id * vm->sbm.sb_size; in virtio_mem_sbm_unplug_sb()
1478 const uint64_t size = count * vm->sbm.sb_size; in virtio_mem_sbm_unplug_sb()
1481 rc = virtio_mem_send_unplug_request(vm, addr, size); in virtio_mem_sbm_unplug_sb()
1483 virtio_mem_sbm_set_sb_unplugged(vm, mb_id, sb_id, count); in virtio_mem_sbm_unplug_sb()
1492 static int virtio_mem_bbm_unplug_bb(struct virtio_mem *vm, unsigned long bb_id) in virtio_mem_bbm_unplug_bb() argument
1494 const uint64_t addr = virtio_mem_bb_id_to_phys(vm, bb_id); in virtio_mem_bbm_unplug_bb()
1495 const uint64_t size = vm->bbm.bb_size; in virtio_mem_bbm_unplug_bb()
1497 return virtio_mem_send_unplug_request(vm, addr, size); in virtio_mem_bbm_unplug_bb()
1505 static int virtio_mem_bbm_plug_bb(struct virtio_mem *vm, unsigned long bb_id) in virtio_mem_bbm_plug_bb() argument
1507 const uint64_t addr = virtio_mem_bb_id_to_phys(vm, bb_id); in virtio_mem_bbm_plug_bb()
1508 const uint64_t size = vm->bbm.bb_size; in virtio_mem_bbm_plug_bb()
1510 return virtio_mem_send_plug_request(vm, addr, size); in virtio_mem_bbm_plug_bb()
1522 static int virtio_mem_sbm_unplug_any_sb_raw(struct virtio_mem *vm, in virtio_mem_sbm_unplug_any_sb_raw() argument
1528 sb_id = vm->sbm.sbs_per_mb - 1; in virtio_mem_sbm_unplug_any_sb_raw()
1532 virtio_mem_sbm_test_sb_unplugged(vm, mb_id, sb_id, 1)) in virtio_mem_sbm_unplug_any_sb_raw()
1539 virtio_mem_sbm_test_sb_plugged(vm, mb_id, sb_id - 1, 1)) { in virtio_mem_sbm_unplug_any_sb_raw()
1544 rc = virtio_mem_sbm_unplug_sb(vm, mb_id, sb_id, count); in virtio_mem_sbm_unplug_any_sb_raw()
1561 static int virtio_mem_sbm_unplug_mb(struct virtio_mem *vm, unsigned long mb_id) in virtio_mem_sbm_unplug_mb() argument
1563 uint64_t nb_sb = vm->sbm.sbs_per_mb; in virtio_mem_sbm_unplug_mb()
1565 return virtio_mem_sbm_unplug_any_sb_raw(vm, mb_id, &nb_sb); in virtio_mem_sbm_unplug_mb()
1571 static int virtio_mem_sbm_prepare_next_mb(struct virtio_mem *vm, in virtio_mem_sbm_prepare_next_mb() argument
1576 if (vm->sbm.next_mb_id > vm->sbm.last_usable_mb_id) in virtio_mem_sbm_prepare_next_mb()
1580 rc = virtio_mem_sbm_mb_states_prepare_next_mb(vm); in virtio_mem_sbm_prepare_next_mb()
1585 rc = virtio_mem_sbm_sb_states_prepare_next_mb(vm); in virtio_mem_sbm_prepare_next_mb()
1589 vm->sbm.mb_count[VIRTIO_MEM_SBM_MB_UNUSED]++; in virtio_mem_sbm_prepare_next_mb()
1590 *mb_id = vm->sbm.next_mb_id++; in virtio_mem_sbm_prepare_next_mb()
1600 static int virtio_mem_sbm_plug_and_add_mb(struct virtio_mem *vm, in virtio_mem_sbm_plug_and_add_mb() argument
1603 const int count = min_t(int, *nb_sb, vm->sbm.sbs_per_mb); in virtio_mem_sbm_plug_and_add_mb()
1613 rc = virtio_mem_sbm_plug_sb(vm, mb_id, 0, count); in virtio_mem_sbm_plug_and_add_mb()
1621 if (count == vm->sbm.sbs_per_mb) in virtio_mem_sbm_plug_and_add_mb()
1622 virtio_mem_sbm_set_mb_state(vm, mb_id, in virtio_mem_sbm_plug_and_add_mb()
1625 virtio_mem_sbm_set_mb_state(vm, mb_id, in virtio_mem_sbm_plug_and_add_mb()
1629 rc = virtio_mem_sbm_add_mb(vm, mb_id); in virtio_mem_sbm_plug_and_add_mb()
1633 if (virtio_mem_sbm_unplug_sb(vm, mb_id, 0, count)) in virtio_mem_sbm_plug_and_add_mb()
1635 virtio_mem_sbm_set_mb_state(vm, mb_id, new_state); in virtio_mem_sbm_plug_and_add_mb()
1651 static int virtio_mem_sbm_plug_any_sb(struct virtio_mem *vm, in virtio_mem_sbm_plug_any_sb() argument
1654 const int old_state = virtio_mem_sbm_get_mb_state(vm, mb_id); in virtio_mem_sbm_plug_any_sb()
1663 sb_id = virtio_mem_sbm_first_unplugged_sb(vm, mb_id); in virtio_mem_sbm_plug_any_sb()
1664 if (sb_id >= vm->sbm.sbs_per_mb) in virtio_mem_sbm_plug_any_sb()
1668 sb_id + count < vm->sbm.sbs_per_mb && in virtio_mem_sbm_plug_any_sb()
1669 !virtio_mem_sbm_test_sb_plugged(vm, mb_id, sb_id + count, 1)) in virtio_mem_sbm_plug_any_sb()
1672 rc = virtio_mem_sbm_plug_sb(vm, mb_id, sb_id, count); in virtio_mem_sbm_plug_any_sb()
1681 sb_id * vm->sbm.sb_size); in virtio_mem_sbm_plug_any_sb()
1682 nr_pages = PFN_DOWN(count * vm->sbm.sb_size); in virtio_mem_sbm_plug_any_sb()
1686 if (virtio_mem_sbm_test_sb_plugged(vm, mb_id, 0, vm->sbm.sbs_per_mb)) in virtio_mem_sbm_plug_any_sb()
1687 virtio_mem_sbm_set_mb_state(vm, mb_id, old_state - 1); in virtio_mem_sbm_plug_any_sb()
1692 static int virtio_mem_sbm_plug_request(struct virtio_mem *vm, uint64_t diff) in virtio_mem_sbm_plug_request() argument
1699 uint64_t nb_sb = diff / vm->sbm.sb_size; in virtio_mem_sbm_plug_request()
1707 mutex_lock(&vm->hotplug_mutex); in virtio_mem_sbm_plug_request()
1710 virtio_mem_sbm_for_each_mb(vm, mb_id, mb_states[i]) { in virtio_mem_sbm_plug_request()
1711 rc = virtio_mem_sbm_plug_any_sb(vm, mb_id, &nb_sb); in virtio_mem_sbm_plug_request()
1722 mutex_unlock(&vm->hotplug_mutex); in virtio_mem_sbm_plug_request()
1725 virtio_mem_sbm_for_each_mb(vm, mb_id, VIRTIO_MEM_SBM_MB_UNUSED) { in virtio_mem_sbm_plug_request()
1726 if (!virtio_mem_could_add_memory(vm, memory_block_size_bytes())) in virtio_mem_sbm_plug_request()
1729 rc = virtio_mem_sbm_plug_and_add_mb(vm, mb_id, &nb_sb); in virtio_mem_sbm_plug_request()
1737 if (!virtio_mem_could_add_memory(vm, memory_block_size_bytes())) in virtio_mem_sbm_plug_request()
1740 rc = virtio_mem_sbm_prepare_next_mb(vm, &mb_id); in virtio_mem_sbm_plug_request()
1743 rc = virtio_mem_sbm_plug_and_add_mb(vm, mb_id, &nb_sb); in virtio_mem_sbm_plug_request()
1751 mutex_unlock(&vm->hotplug_mutex); in virtio_mem_sbm_plug_request()
1760 static int virtio_mem_bbm_plug_and_add_bb(struct virtio_mem *vm, in virtio_mem_bbm_plug_and_add_bb() argument
1765 if (WARN_ON_ONCE(virtio_mem_bbm_get_bb_state(vm, bb_id) != in virtio_mem_bbm_plug_and_add_bb()
1769 rc = virtio_mem_bbm_plug_bb(vm, bb_id); in virtio_mem_bbm_plug_and_add_bb()
1772 virtio_mem_bbm_set_bb_state(vm, bb_id, VIRTIO_MEM_BBM_BB_ADDED); in virtio_mem_bbm_plug_and_add_bb()
1774 rc = virtio_mem_bbm_add_bb(vm, bb_id); in virtio_mem_bbm_plug_and_add_bb()
1776 if (!virtio_mem_bbm_unplug_bb(vm, bb_id)) in virtio_mem_bbm_plug_and_add_bb()
1777 virtio_mem_bbm_set_bb_state(vm, bb_id, in virtio_mem_bbm_plug_and_add_bb()
1781 virtio_mem_bbm_set_bb_state(vm, bb_id, in virtio_mem_bbm_plug_and_add_bb()
1791 static int virtio_mem_bbm_prepare_next_bb(struct virtio_mem *vm, in virtio_mem_bbm_prepare_next_bb() argument
1796 if (vm->bbm.next_bb_id > vm->bbm.last_usable_bb_id) in virtio_mem_bbm_prepare_next_bb()
1800 rc = virtio_mem_bbm_bb_states_prepare_next_bb(vm); in virtio_mem_bbm_prepare_next_bb()
1804 vm->bbm.bb_count[VIRTIO_MEM_BBM_BB_UNUSED]++; in virtio_mem_bbm_prepare_next_bb()
1805 *bb_id = vm->bbm.next_bb_id; in virtio_mem_bbm_prepare_next_bb()
1806 vm->bbm.next_bb_id++; in virtio_mem_bbm_prepare_next_bb()
1810 static int virtio_mem_bbm_plug_request(struct virtio_mem *vm, uint64_t diff) in virtio_mem_bbm_plug_request() argument
1812 uint64_t nb_bb = diff / vm->bbm.bb_size; in virtio_mem_bbm_plug_request()
1820 virtio_mem_bbm_for_each_bb(vm, bb_id, VIRTIO_MEM_BBM_BB_UNUSED) { in virtio_mem_bbm_plug_request()
1821 if (!virtio_mem_could_add_memory(vm, vm->bbm.bb_size)) in virtio_mem_bbm_plug_request()
1824 rc = virtio_mem_bbm_plug_and_add_bb(vm, bb_id); in virtio_mem_bbm_plug_request()
1834 if (!virtio_mem_could_add_memory(vm, vm->bbm.bb_size)) in virtio_mem_bbm_plug_request()
1837 rc = virtio_mem_bbm_prepare_next_bb(vm, &bb_id); in virtio_mem_bbm_plug_request()
1840 rc = virtio_mem_bbm_plug_and_add_bb(vm, bb_id); in virtio_mem_bbm_plug_request()
1854 static int virtio_mem_plug_request(struct virtio_mem *vm, uint64_t diff) in virtio_mem_plug_request() argument
1856 if (vm->in_sbm) in virtio_mem_plug_request()
1857 return virtio_mem_sbm_plug_request(vm, diff); in virtio_mem_plug_request()
1858 return virtio_mem_bbm_plug_request(vm, diff); in virtio_mem_plug_request()
1870 static int virtio_mem_sbm_unplug_any_sb_offline(struct virtio_mem *vm, in virtio_mem_sbm_unplug_any_sb_offline() argument
1876 rc = virtio_mem_sbm_unplug_any_sb_raw(vm, mb_id, nb_sb); in virtio_mem_sbm_unplug_any_sb_offline()
1879 if (!virtio_mem_sbm_test_sb_plugged(vm, mb_id, 0, vm->sbm.sbs_per_mb)) in virtio_mem_sbm_unplug_any_sb_offline()
1880 virtio_mem_sbm_set_mb_state(vm, mb_id, in virtio_mem_sbm_unplug_any_sb_offline()
1885 if (virtio_mem_sbm_test_sb_unplugged(vm, mb_id, 0, vm->sbm.sbs_per_mb)) { in virtio_mem_sbm_unplug_any_sb_offline()
1892 virtio_mem_sbm_set_mb_state(vm, mb_id, in virtio_mem_sbm_unplug_any_sb_offline()
1895 mutex_unlock(&vm->hotplug_mutex); in virtio_mem_sbm_unplug_any_sb_offline()
1896 rc = virtio_mem_sbm_remove_mb(vm, mb_id); in virtio_mem_sbm_unplug_any_sb_offline()
1898 mutex_lock(&vm->hotplug_mutex); in virtio_mem_sbm_unplug_any_sb_offline()
1908 static int virtio_mem_sbm_unplug_sb_online(struct virtio_mem *vm, in virtio_mem_sbm_unplug_sb_online() argument
1912 const unsigned long nr_pages = PFN_DOWN(vm->sbm.sb_size) * count; in virtio_mem_sbm_unplug_sb_online()
1913 const int old_state = virtio_mem_sbm_get_mb_state(vm, mb_id); in virtio_mem_sbm_unplug_sb_online()
1918 sb_id * vm->sbm.sb_size); in virtio_mem_sbm_unplug_sb_online()
1925 rc = virtio_mem_sbm_unplug_sb(vm, mb_id, sb_id, count); in virtio_mem_sbm_unplug_sb_online()
1934 virtio_mem_sbm_set_mb_state(vm, mb_id, in virtio_mem_sbm_unplug_sb_online()
1938 virtio_mem_sbm_set_mb_state(vm, mb_id, in virtio_mem_sbm_unplug_sb_online()
1956 static int virtio_mem_sbm_unplug_any_sb_online(struct virtio_mem *vm, in virtio_mem_sbm_unplug_any_sb_online() argument
1963 if (*nb_sb >= vm->sbm.sbs_per_mb && in virtio_mem_sbm_unplug_any_sb_online()
1964 virtio_mem_sbm_test_sb_plugged(vm, mb_id, 0, vm->sbm.sbs_per_mb)) { in virtio_mem_sbm_unplug_any_sb_online()
1965 rc = virtio_mem_sbm_unplug_sb_online(vm, mb_id, 0, in virtio_mem_sbm_unplug_any_sb_online()
1966 vm->sbm.sbs_per_mb); in virtio_mem_sbm_unplug_any_sb_online()
1968 *nb_sb -= vm->sbm.sbs_per_mb; in virtio_mem_sbm_unplug_any_sb_online()
1975 for (sb_id = vm->sbm.sbs_per_mb - 1; sb_id >= 0 && *nb_sb; sb_id--) { in virtio_mem_sbm_unplug_any_sb_online()
1978 !virtio_mem_sbm_test_sb_plugged(vm, mb_id, sb_id, 1)) in virtio_mem_sbm_unplug_any_sb_online()
1983 rc = virtio_mem_sbm_unplug_sb_online(vm, mb_id, sb_id, 1); in virtio_mem_sbm_unplug_any_sb_online()
1997 if (virtio_mem_sbm_test_sb_unplugged(vm, mb_id, 0, vm->sbm.sbs_per_mb)) { in virtio_mem_sbm_unplug_any_sb_online()
1998 mutex_unlock(&vm->hotplug_mutex); in virtio_mem_sbm_unplug_any_sb_online()
1999 rc = virtio_mem_sbm_offline_and_remove_mb(vm, mb_id); in virtio_mem_sbm_unplug_any_sb_online()
2000 mutex_lock(&vm->hotplug_mutex); in virtio_mem_sbm_unplug_any_sb_online()
2002 virtio_mem_sbm_set_mb_state(vm, mb_id, in virtio_mem_sbm_unplug_any_sb_online()
2021 static int virtio_mem_sbm_unplug_any_sb(struct virtio_mem *vm, in virtio_mem_sbm_unplug_any_sb() argument
2025 const int old_state = virtio_mem_sbm_get_mb_state(vm, mb_id); in virtio_mem_sbm_unplug_any_sb()
2032 return virtio_mem_sbm_unplug_any_sb_online(vm, mb_id, nb_sb); in virtio_mem_sbm_unplug_any_sb()
2035 return virtio_mem_sbm_unplug_any_sb_offline(vm, mb_id, nb_sb); in virtio_mem_sbm_unplug_any_sb()
2040 static int virtio_mem_sbm_unplug_request(struct virtio_mem *vm, uint64_t diff) in virtio_mem_sbm_unplug_request() argument
2050 uint64_t nb_sb = diff / vm->sbm.sb_size; in virtio_mem_sbm_unplug_request()
2062 mutex_lock(&vm->hotplug_mutex); in virtio_mem_sbm_unplug_request()
2072 virtio_mem_sbm_for_each_mb_rev(vm, mb_id, mb_states[i]) { in virtio_mem_sbm_unplug_request()
2073 rc = virtio_mem_sbm_unplug_any_sb(vm, mb_id, &nb_sb); in virtio_mem_sbm_unplug_request()
2076 mutex_unlock(&vm->hotplug_mutex); in virtio_mem_sbm_unplug_request()
2078 mutex_lock(&vm->hotplug_mutex); in virtio_mem_sbm_unplug_request()
2081 mutex_unlock(&vm->hotplug_mutex); in virtio_mem_sbm_unplug_request()
2086 mutex_unlock(&vm->hotplug_mutex); in virtio_mem_sbm_unplug_request()
2089 mutex_unlock(&vm->hotplug_mutex); in virtio_mem_sbm_unplug_request()
2100 static int virtio_mem_bbm_offline_remove_and_unplug_bb(struct virtio_mem *vm, in virtio_mem_bbm_offline_remove_and_unplug_bb() argument
2103 const unsigned long start_pfn = PFN_DOWN(virtio_mem_bb_id_to_phys(vm, bb_id)); in virtio_mem_bbm_offline_remove_and_unplug_bb()
2104 const unsigned long nr_pages = PFN_DOWN(vm->bbm.bb_size); in virtio_mem_bbm_offline_remove_and_unplug_bb()
2110 if (WARN_ON_ONCE(virtio_mem_bbm_get_bb_state(vm, bb_id) != in virtio_mem_bbm_offline_remove_and_unplug_bb()
2121 mutex_lock(&vm->hotplug_mutex); in virtio_mem_bbm_offline_remove_and_unplug_bb()
2122 virtio_mem_bbm_set_bb_state(vm, bb_id, in virtio_mem_bbm_offline_remove_and_unplug_bb()
2136 mutex_unlock(&vm->hotplug_mutex); in virtio_mem_bbm_offline_remove_and_unplug_bb()
2139 rc = virtio_mem_bbm_offline_and_remove_bb(vm, bb_id); in virtio_mem_bbm_offline_remove_and_unplug_bb()
2142 mutex_lock(&vm->hotplug_mutex); in virtio_mem_bbm_offline_remove_and_unplug_bb()
2148 rc = virtio_mem_bbm_unplug_bb(vm, bb_id); in virtio_mem_bbm_offline_remove_and_unplug_bb()
2150 virtio_mem_bbm_set_bb_state(vm, bb_id, in virtio_mem_bbm_offline_remove_and_unplug_bb()
2153 virtio_mem_bbm_set_bb_state(vm, bb_id, in virtio_mem_bbm_offline_remove_and_unplug_bb()
2164 virtio_mem_bbm_set_bb_state(vm, bb_id, VIRTIO_MEM_BBM_BB_ADDED); in virtio_mem_bbm_offline_remove_and_unplug_bb()
2165 mutex_unlock(&vm->hotplug_mutex); in virtio_mem_bbm_offline_remove_and_unplug_bb()
2172 static bool virtio_mem_bbm_bb_is_offline(struct virtio_mem *vm, in virtio_mem_bbm_bb_is_offline() argument
2175 const unsigned long start_pfn = PFN_DOWN(virtio_mem_bb_id_to_phys(vm, bb_id)); in virtio_mem_bbm_bb_is_offline()
2176 const unsigned long nr_pages = PFN_DOWN(vm->bbm.bb_size); in virtio_mem_bbm_bb_is_offline()
2191 static bool virtio_mem_bbm_bb_is_movable(struct virtio_mem *vm, in virtio_mem_bbm_bb_is_movable() argument
2194 const unsigned long start_pfn = PFN_DOWN(virtio_mem_bb_id_to_phys(vm, bb_id)); in virtio_mem_bbm_bb_is_movable()
2195 const unsigned long nr_pages = PFN_DOWN(vm->bbm.bb_size); in virtio_mem_bbm_bb_is_movable()
2211 static int virtio_mem_bbm_unplug_request(struct virtio_mem *vm, uint64_t diff) in virtio_mem_bbm_unplug_request() argument
2213 uint64_t nb_bb = diff / vm->bbm.bb_size; in virtio_mem_bbm_unplug_request()
2225 virtio_mem_bbm_for_each_bb_rev(vm, bb_id, VIRTIO_MEM_BBM_BB_ADDED) { in virtio_mem_bbm_unplug_request()
2232 if (i == 0 && !virtio_mem_bbm_bb_is_offline(vm, bb_id)) in virtio_mem_bbm_unplug_request()
2234 if (i == 1 && !virtio_mem_bbm_bb_is_movable(vm, bb_id)) in virtio_mem_bbm_unplug_request()
2236 rc = virtio_mem_bbm_offline_remove_and_unplug_bb(vm, bb_id); in virtio_mem_bbm_unplug_request()
2254 static int virtio_mem_unplug_request(struct virtio_mem *vm, uint64_t diff) in virtio_mem_unplug_request() argument
2256 if (vm->in_sbm) in virtio_mem_unplug_request()
2257 return virtio_mem_sbm_unplug_request(vm, diff); in virtio_mem_unplug_request()
2258 return virtio_mem_bbm_unplug_request(vm, diff); in virtio_mem_unplug_request()
2265 static int virtio_mem_unplug_pending_mb(struct virtio_mem *vm) in virtio_mem_unplug_pending_mb() argument
2270 if (!vm->in_sbm) { in virtio_mem_unplug_pending_mb()
2271 virtio_mem_bbm_for_each_bb(vm, id, in virtio_mem_unplug_pending_mb()
2273 rc = virtio_mem_bbm_unplug_bb(vm, id); in virtio_mem_unplug_pending_mb()
2276 virtio_mem_bbm_set_bb_state(vm, id, in virtio_mem_unplug_pending_mb()
2282 virtio_mem_sbm_for_each_mb(vm, id, VIRTIO_MEM_SBM_MB_PLUGGED) { in virtio_mem_unplug_pending_mb()
2283 rc = virtio_mem_sbm_unplug_mb(vm, id); in virtio_mem_unplug_pending_mb()
2286 virtio_mem_sbm_set_mb_state(vm, id, in virtio_mem_unplug_pending_mb()
2296 static void virtio_mem_refresh_config(struct virtio_mem *vm) in virtio_mem_refresh_config() argument
2302 virtio_cread_le(vm->vdev, struct virtio_mem_config, plugged_size, in virtio_mem_refresh_config()
2304 if (WARN_ON_ONCE(new_plugged_size != vm->plugged_size)) in virtio_mem_refresh_config()
2305 vm->plugged_size = new_plugged_size; in virtio_mem_refresh_config()
2308 virtio_cread_le(vm->vdev, struct virtio_mem_config, in virtio_mem_refresh_config()
2310 end_addr = min(vm->addr + usable_region_size - 1, in virtio_mem_refresh_config()
2313 if (vm->in_sbm) { in virtio_mem_refresh_config()
2314 vm->sbm.last_usable_mb_id = virtio_mem_phys_to_mb_id(end_addr); in virtio_mem_refresh_config()
2316 vm->sbm.last_usable_mb_id--; in virtio_mem_refresh_config()
2318 vm->bbm.last_usable_bb_id = virtio_mem_phys_to_bb_id(vm, in virtio_mem_refresh_config()
2320 if (!IS_ALIGNED(end_addr + 1, vm->bbm.bb_size)) in virtio_mem_refresh_config()
2321 vm->bbm.last_usable_bb_id--; in virtio_mem_refresh_config()
2331 virtio_cread_le(vm->vdev, struct virtio_mem_config, requested_size, in virtio_mem_refresh_config()
2332 &vm->requested_size); in virtio_mem_refresh_config()
2334 dev_info(&vm->vdev->dev, "plugged size: 0x%llx", vm->plugged_size); in virtio_mem_refresh_config()
2335 dev_info(&vm->vdev->dev, "requested size: 0x%llx", vm->requested_size); in virtio_mem_refresh_config()
2343 struct virtio_mem *vm = container_of(work, struct virtio_mem, wq); in virtio_mem_run_wq() local
2347 if (unlikely(vm->in_kdump)) { in virtio_mem_run_wq()
2348 dev_warn_once(&vm->vdev->dev, in virtio_mem_run_wq()
2353 hrtimer_cancel(&vm->retry_timer); in virtio_mem_run_wq()
2355 if (vm->broken) in virtio_mem_run_wq()
2358 atomic_set(&vm->wq_active, 1); in virtio_mem_run_wq()
2363 if (unlikely(vm->unplug_all_required)) in virtio_mem_run_wq()
2364 rc = virtio_mem_send_unplug_all_request(vm); in virtio_mem_run_wq()
2366 if (atomic_read(&vm->config_changed)) { in virtio_mem_run_wq()
2367 atomic_set(&vm->config_changed, 0); in virtio_mem_run_wq()
2368 virtio_mem_refresh_config(vm); in virtio_mem_run_wq()
2373 rc = virtio_mem_unplug_pending_mb(vm); in virtio_mem_run_wq()
2375 if (!rc && vm->requested_size != vm->plugged_size) { in virtio_mem_run_wq()
2376 if (vm->requested_size > vm->plugged_size) { in virtio_mem_run_wq()
2377 diff = vm->requested_size - vm->plugged_size; in virtio_mem_run_wq()
2378 rc = virtio_mem_plug_request(vm, diff); in virtio_mem_run_wq()
2380 diff = vm->plugged_size - vm->requested_size; in virtio_mem_run_wq()
2381 rc = virtio_mem_unplug_request(vm, diff); in virtio_mem_run_wq()
2387 vm->retry_timer_ms = VIRTIO_MEM_RETRY_TIMER_MIN_MS; in virtio_mem_run_wq()
2407 hrtimer_start(&vm->retry_timer, ms_to_ktime(vm->retry_timer_ms), in virtio_mem_run_wq()
2415 dev_err(&vm->vdev->dev, in virtio_mem_run_wq()
2417 vm->broken = true; in virtio_mem_run_wq()
2420 atomic_set(&vm->wq_active, 0); in virtio_mem_run_wq()
2425 struct virtio_mem *vm = container_of(timer, struct virtio_mem, in virtio_mem_timer_expired() local
2428 virtio_mem_retry(vm); in virtio_mem_timer_expired()
2429 vm->retry_timer_ms = min_t(unsigned int, vm->retry_timer_ms * 2, in virtio_mem_timer_expired()
2436 struct virtio_mem *vm = vq->vdev->priv; in virtio_mem_handle_response() local
2438 wake_up(&vm->host_resp); in virtio_mem_handle_response()
2441 static int virtio_mem_init_vq(struct virtio_mem *vm) in virtio_mem_init_vq() argument
2445 vq = virtio_find_single_vq(vm->vdev, virtio_mem_handle_response, in virtio_mem_init_vq()
2449 vm->vq = vq; in virtio_mem_init_vq()
2454 static int virtio_mem_init_hotplug(struct virtio_mem *vm) in virtio_mem_init_hotplug() argument
2461 if (!IS_ALIGNED(vm->addr, memory_block_size_bytes())) in virtio_mem_init_hotplug()
2462 dev_warn(&vm->vdev->dev, in virtio_mem_init_hotplug()
2464 if (!IS_ALIGNED(vm->addr + vm->region_size, memory_block_size_bytes())) in virtio_mem_init_hotplug()
2465 dev_warn(&vm->vdev->dev, in virtio_mem_init_hotplug()
2467 if (vm->addr < pluggable_range.start || in virtio_mem_init_hotplug()
2468 vm->addr + vm->region_size - 1 > pluggable_range.end) in virtio_mem_init_hotplug()
2469 dev_warn(&vm->vdev->dev, in virtio_mem_init_hotplug()
2473 vm->offline_threshold = max_t(uint64_t, 2 * memory_block_size_bytes(), in virtio_mem_init_hotplug()
2481 sb_size = max_t(uint64_t, vm->device_block_size, sb_size); in virtio_mem_init_hotplug()
2485 vm->in_sbm = true; in virtio_mem_init_hotplug()
2486 vm->sbm.sb_size = sb_size; in virtio_mem_init_hotplug()
2487 vm->sbm.sbs_per_mb = memory_block_size_bytes() / in virtio_mem_init_hotplug()
2488 vm->sbm.sb_size; in virtio_mem_init_hotplug()
2491 addr = max_t(uint64_t, vm->addr, pluggable_range.start) + in virtio_mem_init_hotplug()
2493 vm->sbm.first_mb_id = virtio_mem_phys_to_mb_id(addr); in virtio_mem_init_hotplug()
2494 vm->sbm.next_mb_id = vm->sbm.first_mb_id; in virtio_mem_init_hotplug()
2497 vm->bbm.bb_size = max_t(uint64_t, vm->device_block_size, in virtio_mem_init_hotplug()
2502 dev_warn(&vm->vdev->dev, in virtio_mem_init_hotplug()
2504 } else if (bbm_block_size < vm->bbm.bb_size) { in virtio_mem_init_hotplug()
2505 dev_warn(&vm->vdev->dev, in virtio_mem_init_hotplug()
2508 vm->bbm.bb_size = bbm_block_size; in virtio_mem_init_hotplug()
2513 addr = max_t(uint64_t, vm->addr, pluggable_range.start) + in virtio_mem_init_hotplug()
2514 vm->bbm.bb_size - 1; in virtio_mem_init_hotplug()
2515 vm->bbm.first_bb_id = virtio_mem_phys_to_bb_id(vm, addr); in virtio_mem_init_hotplug()
2516 vm->bbm.next_bb_id = vm->bbm.first_bb_id; in virtio_mem_init_hotplug()
2519 vm->offline_threshold = max_t(uint64_t, 2 * vm->bbm.bb_size, in virtio_mem_init_hotplug()
2520 vm->offline_threshold); in virtio_mem_init_hotplug()
2523 dev_info(&vm->vdev->dev, "memory block size: 0x%lx", in virtio_mem_init_hotplug()
2525 if (vm->in_sbm) in virtio_mem_init_hotplug()
2526 dev_info(&vm->vdev->dev, "subblock size: 0x%llx", in virtio_mem_init_hotplug()
2527 (unsigned long long)vm->sbm.sb_size); in virtio_mem_init_hotplug()
2529 dev_info(&vm->vdev->dev, "big block size: 0x%llx", in virtio_mem_init_hotplug()
2530 (unsigned long long)vm->bbm.bb_size); in virtio_mem_init_hotplug()
2533 rc = virtio_mem_create_resource(vm); in virtio_mem_init_hotplug()
2538 if (vm->in_sbm) in virtio_mem_init_hotplug()
2541 unit_pages = PHYS_PFN(vm->bbm.bb_size); in virtio_mem_init_hotplug()
2542 rc = memory_group_register_dynamic(vm->nid, unit_pages); in virtio_mem_init_hotplug()
2545 vm->mgid = rc; in virtio_mem_init_hotplug()
2552 if (vm->plugged_size) { in virtio_mem_init_hotplug()
2553 vm->unplug_all_required = true; in virtio_mem_init_hotplug()
2554 dev_info(&vm->vdev->dev, "unplugging all memory is required\n"); in virtio_mem_init_hotplug()
2558 vm->memory_notifier.notifier_call = virtio_mem_memory_notifier_cb; in virtio_mem_init_hotplug()
2559 rc = register_memory_notifier(&vm->memory_notifier); in virtio_mem_init_hotplug()
2562 rc = register_virtio_mem_device(vm); in virtio_mem_init_hotplug()
2568 unregister_memory_notifier(&vm->memory_notifier); in virtio_mem_init_hotplug()
2570 memory_group_unregister(vm->mgid); in virtio_mem_init_hotplug()
2572 virtio_mem_delete_resource(vm); in virtio_mem_init_hotplug()
2577 static int virtio_mem_send_state_request(struct virtio_mem *vm, uint64_t addr, in virtio_mem_send_state_request() argument
2580 const uint64_t nb_vm_blocks = size / vm->device_block_size; in virtio_mem_send_state_request()
2582 .type = cpu_to_virtio16(vm->vdev, VIRTIO_MEM_REQ_STATE), in virtio_mem_send_state_request()
2583 .u.state.addr = cpu_to_virtio64(vm->vdev, addr), in virtio_mem_send_state_request()
2584 .u.state.nb_blocks = cpu_to_virtio16(vm->vdev, nb_vm_blocks), in virtio_mem_send_state_request()
2588 dev_dbg(&vm->vdev->dev, "requesting state: 0x%llx - 0x%llx\n", addr, in virtio_mem_send_state_request()
2591 switch (virtio_mem_send_request(vm, &req)) { in virtio_mem_send_state_request()
2593 return virtio16_to_cpu(vm->vdev, vm->resp.u.state.state); in virtio_mem_send_state_request()
2601 dev_dbg(&vm->vdev->dev, "requesting state failed: %d\n", rc); in virtio_mem_send_state_request()
2608 struct virtio_mem *vm = container_of(cb, struct virtio_mem, in virtio_mem_vmcore_pfn_is_ram() local
2614 if (!virtio_mem_contains_range(vm, addr, PAGE_SIZE)) in virtio_mem_vmcore_pfn_is_ram()
2616 if (!vm->plugged_size) in virtio_mem_vmcore_pfn_is_ram()
2623 mutex_lock(&vm->hotplug_mutex); in virtio_mem_vmcore_pfn_is_ram()
2625 addr = ALIGN_DOWN(addr, vm->device_block_size); in virtio_mem_vmcore_pfn_is_ram()
2626 if (addr != vm->last_block_addr) { in virtio_mem_vmcore_pfn_is_ram()
2627 rc = virtio_mem_send_state_request(vm, addr, in virtio_mem_vmcore_pfn_is_ram()
2628 vm->device_block_size); in virtio_mem_vmcore_pfn_is_ram()
2631 vm->last_block_plugged = true; in virtio_mem_vmcore_pfn_is_ram()
2633 vm->last_block_plugged = false; in virtio_mem_vmcore_pfn_is_ram()
2634 vm->last_block_addr = addr; in virtio_mem_vmcore_pfn_is_ram()
2637 is_ram = vm->last_block_plugged; in virtio_mem_vmcore_pfn_is_ram()
2638 mutex_unlock(&vm->hotplug_mutex); in virtio_mem_vmcore_pfn_is_ram()
2643 static int virtio_mem_init_kdump(struct virtio_mem *vm) in virtio_mem_init_kdump() argument
2646 dev_info(&vm->vdev->dev, "memory hot(un)plug disabled in kdump kernel\n"); in virtio_mem_init_kdump()
2647 vm->vmcore_cb.pfn_is_ram = virtio_mem_vmcore_pfn_is_ram; in virtio_mem_init_kdump()
2648 register_vmcore_cb(&vm->vmcore_cb); in virtio_mem_init_kdump()
2651 dev_warn(&vm->vdev->dev, "disabled in kdump kernel without vmcore\n"); in virtio_mem_init_kdump()
2656 static int virtio_mem_init(struct virtio_mem *vm) in virtio_mem_init() argument
2660 if (!vm->vdev->config->get) { in virtio_mem_init()
2661 dev_err(&vm->vdev->dev, "config access disabled\n"); in virtio_mem_init()
2666 virtio_cread_le(vm->vdev, struct virtio_mem_config, plugged_size, in virtio_mem_init()
2667 &vm->plugged_size); in virtio_mem_init()
2668 virtio_cread_le(vm->vdev, struct virtio_mem_config, block_size, in virtio_mem_init()
2669 &vm->device_block_size); in virtio_mem_init()
2670 virtio_cread_le(vm->vdev, struct virtio_mem_config, node_id, in virtio_mem_init()
2672 vm->nid = virtio_mem_translate_node_id(vm, node_id); in virtio_mem_init()
2673 virtio_cread_le(vm->vdev, struct virtio_mem_config, addr, &vm->addr); in virtio_mem_init()
2674 virtio_cread_le(vm->vdev, struct virtio_mem_config, region_size, in virtio_mem_init()
2675 &vm->region_size); in virtio_mem_init()
2678 if (vm->nid == NUMA_NO_NODE) in virtio_mem_init()
2679 vm->nid = memory_add_physaddr_to_nid(vm->addr); in virtio_mem_init()
2681 dev_info(&vm->vdev->dev, "start address: 0x%llx", vm->addr); in virtio_mem_init()
2682 dev_info(&vm->vdev->dev, "region size: 0x%llx", vm->region_size); in virtio_mem_init()
2683 dev_info(&vm->vdev->dev, "device block size: 0x%llx", in virtio_mem_init()
2684 (unsigned long long)vm->device_block_size); in virtio_mem_init()
2685 if (vm->nid != NUMA_NO_NODE && IS_ENABLED(CONFIG_NUMA)) in virtio_mem_init()
2686 dev_info(&vm->vdev->dev, "nid: %d", vm->nid); in virtio_mem_init()
2692 if (vm->in_kdump) in virtio_mem_init()
2693 return virtio_mem_init_kdump(vm); in virtio_mem_init()
2694 return virtio_mem_init_hotplug(vm); in virtio_mem_init()
2697 static int virtio_mem_create_resource(struct virtio_mem *vm) in virtio_mem_create_resource() argument
2703 const char *name = kstrdup(dev_name(&vm->vdev->dev), GFP_KERNEL); in virtio_mem_create_resource()
2709 vm->parent_resource = __request_mem_region(vm->addr, vm->region_size, in virtio_mem_create_resource()
2712 if (!vm->parent_resource) { in virtio_mem_create_resource()
2714 dev_warn(&vm->vdev->dev, "could not reserve device region\n"); in virtio_mem_create_resource()
2715 dev_info(&vm->vdev->dev, in virtio_mem_create_resource()
2721 vm->parent_resource->flags &= ~IORESOURCE_BUSY; in virtio_mem_create_resource()
2725 static void virtio_mem_delete_resource(struct virtio_mem *vm) in virtio_mem_delete_resource() argument
2729 if (!vm->parent_resource) in virtio_mem_delete_resource()
2732 name = vm->parent_resource->name; in virtio_mem_delete_resource()
2733 release_resource(vm->parent_resource); in virtio_mem_delete_resource()
2734 kfree(vm->parent_resource); in virtio_mem_delete_resource()
2736 vm->parent_resource = NULL; in virtio_mem_delete_resource()
2744 static bool virtio_mem_has_memory_added(struct virtio_mem *vm) in virtio_mem_has_memory_added() argument
2748 return walk_iomem_res_desc(IORES_DESC_NONE, flags, vm->addr, in virtio_mem_has_memory_added()
2749 vm->addr + vm->region_size, NULL, in virtio_mem_has_memory_added()
2755 struct virtio_mem *vm; in virtio_mem_probe() local
2761 vdev->priv = vm = kzalloc(sizeof(*vm), GFP_KERNEL); in virtio_mem_probe()
2762 if (!vm) in virtio_mem_probe()
2765 init_waitqueue_head(&vm->host_resp); in virtio_mem_probe()
2766 vm->vdev = vdev; in virtio_mem_probe()
2767 INIT_WORK(&vm->wq, virtio_mem_run_wq); in virtio_mem_probe()
2768 mutex_init(&vm->hotplug_mutex); in virtio_mem_probe()
2769 INIT_LIST_HEAD(&vm->next); in virtio_mem_probe()
2770 spin_lock_init(&vm->removal_lock); in virtio_mem_probe()
2771 hrtimer_init(&vm->retry_timer, CLOCK_MONOTONIC, HRTIMER_MODE_REL); in virtio_mem_probe()
2772 vm->retry_timer.function = virtio_mem_timer_expired; in virtio_mem_probe()
2773 vm->retry_timer_ms = VIRTIO_MEM_RETRY_TIMER_MIN_MS; in virtio_mem_probe()
2774 vm->in_kdump = is_kdump_kernel(); in virtio_mem_probe()
2777 rc = virtio_mem_init_vq(vm); in virtio_mem_probe()
2782 rc = virtio_mem_init(vm); in virtio_mem_probe()
2789 if (!vm->in_kdump) { in virtio_mem_probe()
2790 atomic_set(&vm->config_changed, 1); in virtio_mem_probe()
2791 queue_work(system_freezable_wq, &vm->wq); in virtio_mem_probe()
2798 kfree(vm); in virtio_mem_probe()
2804 static void virtio_mem_deinit_hotplug(struct virtio_mem *vm) in virtio_mem_deinit_hotplug() argument
2813 mutex_lock(&vm->hotplug_mutex); in virtio_mem_deinit_hotplug()
2814 spin_lock_irq(&vm->removal_lock); in virtio_mem_deinit_hotplug()
2815 vm->removing = true; in virtio_mem_deinit_hotplug()
2816 spin_unlock_irq(&vm->removal_lock); in virtio_mem_deinit_hotplug()
2817 mutex_unlock(&vm->hotplug_mutex); in virtio_mem_deinit_hotplug()
2820 cancel_work_sync(&vm->wq); in virtio_mem_deinit_hotplug()
2821 hrtimer_cancel(&vm->retry_timer); in virtio_mem_deinit_hotplug()
2823 if (vm->in_sbm) { in virtio_mem_deinit_hotplug()
2828 virtio_mem_sbm_for_each_mb(vm, mb_id, in virtio_mem_deinit_hotplug()
2830 rc = virtio_mem_sbm_remove_mb(vm, mb_id); in virtio_mem_deinit_hotplug()
2832 virtio_mem_sbm_set_mb_state(vm, mb_id, in virtio_mem_deinit_hotplug()
2843 unregister_virtio_mem_device(vm); in virtio_mem_deinit_hotplug()
2844 unregister_memory_notifier(&vm->memory_notifier); in virtio_mem_deinit_hotplug()
2851 if (virtio_mem_has_memory_added(vm)) { in virtio_mem_deinit_hotplug()
2852 dev_warn(&vm->vdev->dev, in virtio_mem_deinit_hotplug()
2855 virtio_mem_delete_resource(vm); in virtio_mem_deinit_hotplug()
2856 kfree_const(vm->resource_name); in virtio_mem_deinit_hotplug()
2857 memory_group_unregister(vm->mgid); in virtio_mem_deinit_hotplug()
2861 if (vm->in_sbm) { in virtio_mem_deinit_hotplug()
2862 vfree(vm->sbm.mb_states); in virtio_mem_deinit_hotplug()
2863 vfree(vm->sbm.sb_states); in virtio_mem_deinit_hotplug()
2865 vfree(vm->bbm.bb_states); in virtio_mem_deinit_hotplug()
2869 static void virtio_mem_deinit_kdump(struct virtio_mem *vm) in virtio_mem_deinit_kdump() argument
2872 unregister_vmcore_cb(&vm->vmcore_cb); in virtio_mem_deinit_kdump()
2878 struct virtio_mem *vm = vdev->priv; in virtio_mem_remove() local
2880 if (vm->in_kdump) in virtio_mem_remove()
2881 virtio_mem_deinit_kdump(vm); in virtio_mem_remove()
2883 virtio_mem_deinit_hotplug(vm); in virtio_mem_remove()
2889 kfree(vm); in virtio_mem_remove()
2895 struct virtio_mem *vm = vdev->priv; in virtio_mem_config_changed() local
2897 if (unlikely(vm->in_kdump)) in virtio_mem_config_changed()
2900 atomic_set(&vm->config_changed, 1); in virtio_mem_config_changed()
2901 virtio_mem_retry(vm); in virtio_mem_config_changed()