123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442 |
- #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
- #include <linux/io.h>
- #include <linux/kernel.h>
- #include <linux/mm.h>
- #include <linux/slab.h>
- #include <linux/vmalloc.h>
- #include <linux/hyperv.h>
- #include <linux/random.h>
- #include <linux/clockchips.h>
- #include <linux/delay.h>
- #include <linux/interrupt.h>
- #include <clocksource/hyperv_timer.h>
- #include <asm/mshyperv.h>
- #include "hyperv_vmbus.h"
- struct hv_context hv_context;
- int hv_init(void)
- {
- hv_context.cpu_context = alloc_percpu(struct hv_per_cpu_context);
- if (!hv_context.cpu_context)
- return -ENOMEM;
- return 0;
- }
- void *hv_alloc_hyperv_page(void)
- {
- BUILD_BUG_ON(PAGE_SIZE < HV_HYP_PAGE_SIZE);
- if (PAGE_SIZE == HV_HYP_PAGE_SIZE)
- return (void *)__get_free_page(GFP_KERNEL);
- else
- return kmalloc(HV_HYP_PAGE_SIZE, GFP_KERNEL);
- }
- void *hv_alloc_hyperv_zeroed_page(void)
- {
- if (PAGE_SIZE == HV_HYP_PAGE_SIZE)
- return (void *)__get_free_page(GFP_KERNEL | __GFP_ZERO);
- else
- return kzalloc(HV_HYP_PAGE_SIZE, GFP_KERNEL);
- }
- void hv_free_hyperv_page(unsigned long addr)
- {
- if (PAGE_SIZE == HV_HYP_PAGE_SIZE)
- free_page(addr);
- else
- kfree((void *)addr);
- }
- int hv_post_message(union hv_connection_id connection_id,
- enum hv_message_type message_type,
- void *payload, size_t payload_size)
- {
- struct hv_input_post_message *aligned_msg;
- struct hv_per_cpu_context *hv_cpu;
- u64 status;
- if (payload_size > HV_MESSAGE_PAYLOAD_BYTE_COUNT)
- return -EMSGSIZE;
- hv_cpu = get_cpu_ptr(hv_context.cpu_context);
- aligned_msg = hv_cpu->post_msg_page;
- aligned_msg->connectionid = connection_id;
- aligned_msg->reserved = 0;
- aligned_msg->message_type = message_type;
- aligned_msg->payload_size = payload_size;
- memcpy((void *)aligned_msg->payload, payload, payload_size);
- if (hv_isolation_type_snp())
- status = hv_ghcb_hypercall(HVCALL_POST_MESSAGE,
- (void *)aligned_msg, NULL,
- sizeof(*aligned_msg));
- else
- status = hv_do_hypercall(HVCALL_POST_MESSAGE,
- aligned_msg, NULL);
-
- put_cpu_ptr(hv_cpu);
- return hv_result(status);
- }
- int hv_synic_alloc(void)
- {
- int cpu;
- struct hv_per_cpu_context *hv_cpu;
-
- for_each_present_cpu(cpu) {
- hv_cpu = per_cpu_ptr(hv_context.cpu_context, cpu);
- memset(hv_cpu, 0, sizeof(*hv_cpu));
- }
- hv_context.hv_numa_map = kcalloc(nr_node_ids, sizeof(struct cpumask),
- GFP_KERNEL);
- if (hv_context.hv_numa_map == NULL) {
- pr_err("Unable to allocate NUMA map\n");
- goto err;
- }
- for_each_present_cpu(cpu) {
- hv_cpu = per_cpu_ptr(hv_context.cpu_context, cpu);
- tasklet_init(&hv_cpu->msg_dpc,
- vmbus_on_msg_dpc, (unsigned long) hv_cpu);
-
- if (!hv_isolation_type_snp()) {
- hv_cpu->synic_message_page =
- (void *)get_zeroed_page(GFP_ATOMIC);
- if (hv_cpu->synic_message_page == NULL) {
- pr_err("Unable to allocate SYNIC message page\n");
- goto err;
- }
- hv_cpu->synic_event_page =
- (void *)get_zeroed_page(GFP_ATOMIC);
- if (hv_cpu->synic_event_page == NULL) {
- pr_err("Unable to allocate SYNIC event page\n");
- goto err;
- }
- }
- hv_cpu->post_msg_page = (void *)get_zeroed_page(GFP_ATOMIC);
- if (hv_cpu->post_msg_page == NULL) {
- pr_err("Unable to allocate post msg page\n");
- goto err;
- }
- }
- return 0;
- err:
-
- return -ENOMEM;
- }
- void hv_synic_free(void)
- {
- int cpu;
- for_each_present_cpu(cpu) {
- struct hv_per_cpu_context *hv_cpu
- = per_cpu_ptr(hv_context.cpu_context, cpu);
- free_page((unsigned long)hv_cpu->synic_event_page);
- free_page((unsigned long)hv_cpu->synic_message_page);
- free_page((unsigned long)hv_cpu->post_msg_page);
- }
- kfree(hv_context.hv_numa_map);
- }
- void hv_synic_enable_regs(unsigned int cpu)
- {
- struct hv_per_cpu_context *hv_cpu
- = per_cpu_ptr(hv_context.cpu_context, cpu);
- union hv_synic_simp simp;
- union hv_synic_siefp siefp;
- union hv_synic_sint shared_sint;
- union hv_synic_scontrol sctrl;
-
- simp.as_uint64 = hv_get_register(HV_REGISTER_SIMP);
- simp.simp_enabled = 1;
- if (hv_isolation_type_snp()) {
- hv_cpu->synic_message_page
- = memremap(simp.base_simp_gpa << HV_HYP_PAGE_SHIFT,
- HV_HYP_PAGE_SIZE, MEMREMAP_WB);
- if (!hv_cpu->synic_message_page)
- pr_err("Fail to map syinc message page.\n");
- } else {
- simp.base_simp_gpa = virt_to_phys(hv_cpu->synic_message_page)
- >> HV_HYP_PAGE_SHIFT;
- }
- hv_set_register(HV_REGISTER_SIMP, simp.as_uint64);
-
- siefp.as_uint64 = hv_get_register(HV_REGISTER_SIEFP);
- siefp.siefp_enabled = 1;
- if (hv_isolation_type_snp()) {
- hv_cpu->synic_event_page =
- memremap(siefp.base_siefp_gpa << HV_HYP_PAGE_SHIFT,
- HV_HYP_PAGE_SIZE, MEMREMAP_WB);
- if (!hv_cpu->synic_event_page)
- pr_err("Fail to map syinc event page.\n");
- } else {
- siefp.base_siefp_gpa = virt_to_phys(hv_cpu->synic_event_page)
- >> HV_HYP_PAGE_SHIFT;
- }
- hv_set_register(HV_REGISTER_SIEFP, siefp.as_uint64);
-
- if (vmbus_irq != -1)
- enable_percpu_irq(vmbus_irq, 0);
- shared_sint.as_uint64 = hv_get_register(HV_REGISTER_SINT0 +
- VMBUS_MESSAGE_SINT);
- shared_sint.vector = vmbus_interrupt;
- shared_sint.masked = false;
-
- #ifdef HV_DEPRECATING_AEOI_RECOMMENDED
- shared_sint.auto_eoi =
- !(ms_hyperv.hints & HV_DEPRECATING_AEOI_RECOMMENDED);
- #else
- shared_sint.auto_eoi = 0;
- #endif
- hv_set_register(HV_REGISTER_SINT0 + VMBUS_MESSAGE_SINT,
- shared_sint.as_uint64);
-
- sctrl.as_uint64 = hv_get_register(HV_REGISTER_SCONTROL);
- sctrl.enable = 1;
- hv_set_register(HV_REGISTER_SCONTROL, sctrl.as_uint64);
- }
- int hv_synic_init(unsigned int cpu)
- {
- hv_synic_enable_regs(cpu);
- hv_stimer_legacy_init(cpu, VMBUS_MESSAGE_SINT);
- return 0;
- }
- void hv_synic_disable_regs(unsigned int cpu)
- {
- struct hv_per_cpu_context *hv_cpu
- = per_cpu_ptr(hv_context.cpu_context, cpu);
- union hv_synic_sint shared_sint;
- union hv_synic_simp simp;
- union hv_synic_siefp siefp;
- union hv_synic_scontrol sctrl;
- shared_sint.as_uint64 = hv_get_register(HV_REGISTER_SINT0 +
- VMBUS_MESSAGE_SINT);
- shared_sint.masked = 1;
-
-
- hv_set_register(HV_REGISTER_SINT0 + VMBUS_MESSAGE_SINT,
- shared_sint.as_uint64);
- simp.as_uint64 = hv_get_register(HV_REGISTER_SIMP);
-
- simp.simp_enabled = 0;
- if (hv_isolation_type_snp())
- memunmap(hv_cpu->synic_message_page);
- else
- simp.base_simp_gpa = 0;
- hv_set_register(HV_REGISTER_SIMP, simp.as_uint64);
- siefp.as_uint64 = hv_get_register(HV_REGISTER_SIEFP);
- siefp.siefp_enabled = 0;
- if (hv_isolation_type_snp())
- memunmap(hv_cpu->synic_event_page);
- else
- siefp.base_siefp_gpa = 0;
- hv_set_register(HV_REGISTER_SIEFP, siefp.as_uint64);
-
- sctrl.as_uint64 = hv_get_register(HV_REGISTER_SCONTROL);
- sctrl.enable = 0;
- hv_set_register(HV_REGISTER_SCONTROL, sctrl.as_uint64);
- if (vmbus_irq != -1)
- disable_percpu_irq(vmbus_irq);
- }
- #define HV_MAX_TRIES 3
- static bool hv_synic_event_pending(void)
- {
- struct hv_per_cpu_context *hv_cpu = this_cpu_ptr(hv_context.cpu_context);
- union hv_synic_event_flags *event =
- (union hv_synic_event_flags *)hv_cpu->synic_event_page + VMBUS_MESSAGE_SINT;
- unsigned long *recv_int_page = event->flags;
- bool pending;
- u32 relid;
- int tries = 0;
- retry:
- pending = false;
- for_each_set_bit(relid, recv_int_page, HV_EVENT_FLAGS_COUNT) {
-
- if (relid == 0)
- continue;
- pending = true;
- break;
- }
- if (pending && tries++ < HV_MAX_TRIES) {
- usleep_range(10000, 20000);
- goto retry;
- }
- return pending;
- }
- int hv_synic_cleanup(unsigned int cpu)
- {
- struct vmbus_channel *channel, *sc;
- bool channel_found = false;
- if (vmbus_connection.conn_state != CONNECTED)
- goto always_cleanup;
-
- if (cpu == VMBUS_CONNECT_CPU)
- return -EBUSY;
-
- mutex_lock(&vmbus_connection.channel_mutex);
- list_for_each_entry(channel, &vmbus_connection.chn_list, listentry) {
- if (channel->target_cpu == cpu) {
- channel_found = true;
- break;
- }
- list_for_each_entry(sc, &channel->sc_list, sc_list) {
- if (sc->target_cpu == cpu) {
- channel_found = true;
- break;
- }
- }
- if (channel_found)
- break;
- }
- mutex_unlock(&vmbus_connection.channel_mutex);
- if (channel_found)
- return -EBUSY;
-
- if (vmbus_proto_version >= VERSION_WIN10_V4_1 && hv_synic_event_pending())
- return -EBUSY;
- always_cleanup:
- hv_stimer_legacy_cleanup(cpu);
- hv_synic_disable_regs(cpu);
- return 0;
- }
|