// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (C) 2024 Loongson Technology Corporation Limited
 */

#include <linux/kvm_host.h>
#include <trace/events/kvm.h>
#include <asm/kvm_pch_pic.h>

static int kvm_set_pic_irq(struct kvm_kernel_irq_routing_entry *e,
  struct kvm *kvm, int irq_source_id, int level, bool line_status)
{
 /* PCH-PIC pin (0 ~ 64) <---> GSI (0 ~ 64) */
 pch_pic_set_irq(kvm->arch.pch_pic, e->irqchip.pin, level);

 return 0;
}

/*
 * kvm_set_msi: inject the MSI corresponding to the
 * MSI routing entry
 *
 * This is the entry point for irqfd MSI injection
 * and userspace MSI injection.
 */
int kvm_set_msi(struct kvm_kernel_irq_routing_entry *e,
  struct kvm *kvm, int irq_source_id, int level, bool line_status)
{
 if (!level)
  return -1;

 pch_msi_set_irq(kvm, e->msi.data, level);

 return 0;
}

/*
 * kvm_set_routing_entry: populate a kvm routing entry
 * from a user routing entry
 *
 * @kvm: the VM this entry is applied to
 * @e: kvm kernel routing entry handle
 * @ue: user api routing entry handle
 * return 0 on success, -EINVAL on errors.
 */
int kvm_set_routing_entry(struct kvm *kvm,
   struct kvm_kernel_irq_routing_entry *e,
   const struct kvm_irq_routing_entry *ue)
{
 switch (ue->type) {
 case KVM_IRQ_ROUTING_IRQCHIP:
  e->set = kvm_set_pic_irq;
  e->irqchip.irqchip = ue->u.irqchip.irqchip;
  e->irqchip.pin = ue->u.irqchip.pin;

  if (e->irqchip.pin >= KVM_IRQCHIP_NUM_PINS)
   return -EINVAL;

  return 0;
 case KVM_IRQ_ROUTING_MSI:
  e->set = kvm_set_msi;
  e->msi.address_lo = ue->u.msi.address_lo;
  e->msi.address_hi = ue->u.msi.address_hi;
  e->msi.data = ue->u.msi.data;
  return 0;
 default:
  return -EINVAL;
 }
}

int kvm_arch_set_irq_inatomic(struct kvm_kernel_irq_routing_entry *e,
  struct kvm *kvm, int irq_source_id, int level, bool line_status)
{
 switch (e->type) {
 case KVM_IRQ_ROUTING_IRQCHIP:
  pch_pic_set_irq(kvm->arch.pch_pic, e->irqchip.pin, level);
  return 0;
 case KVM_IRQ_ROUTING_MSI:
  pch_msi_set_irq(kvm, e->msi.data, level);
  return 0;
 default:
  return -EWOULDBLOCK;
 }
}

bool kvm_arch_intc_initialized(struct kvm *kvm)
{
 return kvm_arch_irqchip_in_kernel(kvm);
}