// SPDX-License-Identifier: GPL-2.0+
/*
 * virtio-snd: Virtio sound device
 * Copyright (C) 2022 OpenSynergy GmbH
 */
#include <sound/control.h>
#include <linux/virtio_config.h>

#include "virtio_card.h"

/* Map for converting VirtIO types to ALSA types. */
static const snd_ctl_elem_type_t g_v2a_type_map[] = {
 [VIRTIO_SND_CTL_TYPE_BOOLEAN] = SNDRV_CTL_ELEM_TYPE_BOOLEAN,
 [VIRTIO_SND_CTL_TYPE_INTEGER] = SNDRV_CTL_ELEM_TYPE_INTEGER,
 [VIRTIO_SND_CTL_TYPE_INTEGER64] = SNDRV_CTL_ELEM_TYPE_INTEGER64,
 [VIRTIO_SND_CTL_TYPE_ENUMERATED] = SNDRV_CTL_ELEM_TYPE_ENUMERATED,
 [VIRTIO_SND_CTL_TYPE_BYTES] = SNDRV_CTL_ELEM_TYPE_BYTES,
 [VIRTIO_SND_CTL_TYPE_IEC958] = SNDRV_CTL_ELEM_TYPE_IEC958
};

/* Map for converting VirtIO access rights to ALSA access rights. */
static const unsigned int g_v2a_access_map[] = {
 [VIRTIO_SND_CTL_ACCESS_READ] = SNDRV_CTL_ELEM_ACCESS_READ,
 [VIRTIO_SND_CTL_ACCESS_WRITE] = SNDRV_CTL_ELEM_ACCESS_WRITE,
 [VIRTIO_SND_CTL_ACCESS_VOLATILE] = SNDRV_CTL_ELEM_ACCESS_VOLATILE,
 [VIRTIO_SND_CTL_ACCESS_INACTIVE] = SNDRV_CTL_ELEM_ACCESS_INACTIVE,
 [VIRTIO_SND_CTL_ACCESS_TLV_READ] = SNDRV_CTL_ELEM_ACCESS_TLV_READ,
 [VIRTIO_SND_CTL_ACCESS_TLV_WRITE] = SNDRV_CTL_ELEM_ACCESS_TLV_WRITE,
 [VIRTIO_SND_CTL_ACCESS_TLV_COMMAND] = SNDRV_CTL_ELEM_ACCESS_TLV_COMMAND
};

/* Map for converting VirtIO event masks to ALSA event masks. */
static const unsigned int g_v2a_mask_map[] = {
 [VIRTIO_SND_CTL_EVT_MASK_VALUE] = SNDRV_CTL_EVENT_MASK_VALUE,
 [VIRTIO_SND_CTL_EVT_MASK_INFO] = SNDRV_CTL_EVENT_MASK_INFO,
 [VIRTIO_SND_CTL_EVT_MASK_TLV] = SNDRV_CTL_EVENT_MASK_TLV
};

/**
 * virtsnd_kctl_info() - Returns information about the control.
 * @kcontrol: ALSA control element.
 * @uinfo: Element information.
 *
 * Context: Process context.
 * Return: 0 on success, -errno on failure.
 */
static int virtsnd_kctl_info(struct snd_kcontrol *kcontrol,
        struct snd_ctl_elem_info *uinfo)
{
 struct virtio_snd *snd = snd_kcontrol_chip(kcontrol);
 struct virtio_kctl *kctl = &snd->kctls[kcontrol->private_value];
 struct virtio_snd_ctl_info *kinfo =
  &snd->kctl_infos[kcontrol->private_value];
 unsigned int i;

 uinfo->type = g_v2a_type_map[le32_to_cpu(kinfo->type)];
 uinfo->count = le32_to_cpu(kinfo->count);

 switch (uinfo->type) {
 case SNDRV_CTL_ELEM_TYPE_INTEGER:
  uinfo->value.integer.min =
   le32_to_cpu(kinfo->value.integer.min);
  uinfo->value.integer.max =
   le32_to_cpu(kinfo->value.integer.max);
  uinfo->value.integer.step =
   le32_to_cpu(kinfo->value.integer.step);

  break;
 case SNDRV_CTL_ELEM_TYPE_INTEGER64:
  uinfo->value.integer64.min =
   le64_to_cpu(kinfo->value.integer64.min);
  uinfo->value.integer64.max =
   le64_to_cpu(kinfo->value.integer64.max);
  uinfo->value.integer64.step =
   le64_to_cpu(kinfo->value.integer64.step);

  break;
 case SNDRV_CTL_ELEM_TYPE_ENUMERATED:
  uinfo->value.enumerated.items =
   le32_to_cpu(kinfo->value.enumerated.items);
  i = uinfo->value.enumerated.item;
  if (i >= uinfo->value.enumerated.items)
   return -EINVAL;

  strscpy(uinfo->value.enumerated.name, kctl->items[i].item,
   sizeof(uinfo->value.enumerated.name));

  break;
 }

 return 0;
}

/**
 * virtsnd_kctl_get() - Read the value from the control.
 * @kcontrol: ALSA control element.
 * @uvalue: Element value.
 *
 * Context: Process context.
 * Return: 0 on success, -errno on failure.
 */
static int virtsnd_kctl_get(struct snd_kcontrol *kcontrol,
       struct snd_ctl_elem_value *uvalue)
{
 struct virtio_snd *snd = snd_kcontrol_chip(kcontrol);
 struct virtio_snd_ctl_info *kinfo =
  &snd->kctl_infos[kcontrol->private_value];
 unsigned int type = le32_to_cpu(kinfo->type);
 unsigned int count = le32_to_cpu(kinfo->count);
 struct virtio_snd_msg *msg;
 struct virtio_snd_ctl_hdr *hdr;
 struct virtio_snd_ctl_value *kvalue;
 size_t request_size = sizeof(*hdr);
 size_t response_size = sizeof(struct virtio_snd_hdr) + sizeof(*kvalue);
 unsigned int i;
 int rc;

 msg = virtsnd_ctl_msg_alloc(request_size, response_size, GFP_KERNEL);
 if (!msg)
  return -ENOMEM;

 virtsnd_ctl_msg_ref(msg);

 hdr = virtsnd_ctl_msg_request(msg);
 hdr->hdr.code = cpu_to_le32(VIRTIO_SND_R_CTL_READ);
 hdr->control_id = cpu_to_le32(kcontrol->private_value);

 rc = virtsnd_ctl_msg_send_sync(snd, msg);
 if (rc)
  goto on_failure;

 kvalue = (void *)((u8 *)virtsnd_ctl_msg_response(msg) +
     sizeof(struct virtio_snd_hdr));

 switch (type) {
 case VIRTIO_SND_CTL_TYPE_BOOLEAN:
 case VIRTIO_SND_CTL_TYPE_INTEGER:
  for (i = 0; i < count; ++i)
   uvalue->value.integer.value[i] =
    le32_to_cpu(kvalue->value.integer[i]);
  break;
 case VIRTIO_SND_CTL_TYPE_INTEGER64:
  for (i = 0; i < count; ++i)
   uvalue->value.integer64.value[i] =
    le64_to_cpu(kvalue->value.integer64[i]);
  break;
 case VIRTIO_SND_CTL_TYPE_ENUMERATED:
  for (i = 0; i < count; ++i)
   uvalue->value.enumerated.item[i] =
    le32_to_cpu(kvalue->value.enumerated[i]);
  break;
 case VIRTIO_SND_CTL_TYPE_BYTES:
  memcpy(uvalue->value.bytes.data, kvalue->value.bytes, count);
  break;
 case VIRTIO_SND_CTL_TYPE_IEC958:
  memcpy(&uvalue->value.iec958, &kvalue->value.iec958,
         sizeof(uvalue->value.iec958));
  break;
 }

on_failure:
 virtsnd_ctl_msg_unref(msg);

 return rc;
}

/**
 * virtsnd_kctl_put() - Write the value to the control.
 * @kcontrol: ALSA control element.
 * @uvalue: Element value.
 *
 * Context: Process context.
 * Return: 0 on success, -errno on failure.
 */
static int virtsnd_kctl_put(struct snd_kcontrol *kcontrol,
       struct snd_ctl_elem_value *uvalue)
{
 struct virtio_snd *snd = snd_kcontrol_chip(kcontrol);
 struct virtio_snd_ctl_info *kinfo =
  &snd->kctl_infos[kcontrol->private_value];
 unsigned int type = le32_to_cpu(kinfo->type);
 unsigned int count = le32_to_cpu(kinfo->count);
 struct virtio_snd_msg *msg;
 struct virtio_snd_ctl_hdr *hdr;
 struct virtio_snd_ctl_value *kvalue;
 size_t request_size = sizeof(*hdr) + sizeof(*kvalue);
 size_t response_size = sizeof(struct virtio_snd_hdr);
 unsigned int i;

 msg = virtsnd_ctl_msg_alloc(request_size, response_size, GFP_KERNEL);
 if (!msg)
  return -ENOMEM;

 hdr = virtsnd_ctl_msg_request(msg);
 hdr->hdr.code = cpu_to_le32(VIRTIO_SND_R_CTL_WRITE);
 hdr->control_id = cpu_to_le32(kcontrol->private_value);

 kvalue = (void *)((u8 *)hdr + sizeof(*hdr));

 switch (type) {
 case VIRTIO_SND_CTL_TYPE_BOOLEAN:
 case VIRTIO_SND_CTL_TYPE_INTEGER:
  for (i = 0; i < count; ++i)
   kvalue->value.integer[i] =
    cpu_to_le32(uvalue->value.integer.value[i]);
  break;
 case VIRTIO_SND_CTL_TYPE_INTEGER64:
  for (i = 0; i < count; ++i)
   kvalue->value.integer64[i] =
    cpu_to_le64(uvalue->value.integer64.value[i]);
  break;
 case VIRTIO_SND_CTL_TYPE_ENUMERATED:
  for (i = 0; i < count; ++i)
   kvalue->value.enumerated[i] =
    cpu_to_le32(uvalue->value.enumerated.item[i]);
  break;
 case VIRTIO_SND_CTL_TYPE_BYTES:
  memcpy(kvalue->value.bytes, uvalue->value.bytes.data, count);
  break;
 case VIRTIO_SND_CTL_TYPE_IEC958:
  memcpy(&kvalue->value.iec958, &uvalue->value.iec958,
         sizeof(kvalue->value.iec958));
  break;
 }

 return virtsnd_ctl_msg_send_sync(snd, msg);
}

/**
 * virtsnd_kctl_tlv_op() - Perform an operation on the control's metadata.
 * @kcontrol: ALSA control element.
 * @op_flag: Operation code (SNDRV_CTL_TLV_OP_XXX).
 * @size: Size of the TLV data in bytes.
 * @utlv: TLV data.
 *
 * Context: Process context.
 * Return: 0 on success, -errno on failure.
 */
static int virtsnd_kctl_tlv_op(struct snd_kcontrol *kcontrol, int op_flag,
          unsigned int size, unsigned int __user *utlv)
{
 struct virtio_snd *snd = snd_kcontrol_chip(kcontrol);
 struct virtio_snd_msg *msg;
 struct virtio_snd_ctl_hdr *hdr;
 unsigned int *tlv;
 struct scatterlist sg;
 int rc;

 msg = virtsnd_ctl_msg_alloc(sizeof(*hdr), sizeof(struct virtio_snd_hdr),
        GFP_KERNEL);
 if (!msg)
  return -ENOMEM;

 tlv = kzalloc(size, GFP_KERNEL);
 if (!tlv) {
  rc = -ENOMEM;
  goto on_msg_unref;
 }

 sg_init_one(&sg, tlv, size);

 hdr = virtsnd_ctl_msg_request(msg);
 hdr->control_id = cpu_to_le32(kcontrol->private_value);

 switch (op_flag) {
 case SNDRV_CTL_TLV_OP_READ:
  hdr->hdr.code = cpu_to_le32(VIRTIO_SND_R_CTL_TLV_READ);

  rc = virtsnd_ctl_msg_send(snd, msg, NULL, &sg, false);
  if (!rc) {
   if (copy_to_user(utlv, tlv, size))
    rc = -EFAULT;
  }

  break;
 case SNDRV_CTL_TLV_OP_WRITE:
 case SNDRV_CTL_TLV_OP_CMD:
  if (op_flag == SNDRV_CTL_TLV_OP_WRITE)
   hdr->hdr.code = cpu_to_le32(VIRTIO_SND_R_CTL_TLV_WRITE);
  else
   hdr->hdr.code =
    cpu_to_le32(VIRTIO_SND_R_CTL_TLV_COMMAND);

  if (copy_from_user(tlv, utlv, size)) {
   rc = -EFAULT;
   goto on_msg_unref;
  } else {
   rc = virtsnd_ctl_msg_send(snd, msg, &sg, NULL, false);
  }

  break;
 default:
  rc = -EINVAL;
  /* We never get here - we listed all values for op_flag */
  WARN_ON(1);
  goto on_msg_unref;
 }
 kfree(tlv);
 return rc;

on_msg_unref:
 virtsnd_ctl_msg_unref(msg);
 kfree(tlv);

 return rc;
}

/**
 * virtsnd_kctl_get_enum_items() - Query items for the ENUMERATED element type.
 * @snd: VirtIO sound device.
 * @cid: Control element ID.
 *
 * This function is called during initial device initialization.
 *
 * Context: Any context that permits to sleep.
 * Return: 0 on success, -errno on failure.
 */
static int virtsnd_kctl_get_enum_items(struct virtio_snd *snd, unsigned int cid)
{
 struct virtio_device *vdev = snd->vdev;
 struct virtio_snd_ctl_info *kinfo = &snd->kctl_infos[cid];
 struct virtio_kctl *kctl = &snd->kctls[cid];
 struct virtio_snd_msg *msg;
 struct virtio_snd_ctl_hdr *hdr;
 unsigned int n = le32_to_cpu(kinfo->value.enumerated.items);
 struct scatterlist sg;

 msg = virtsnd_ctl_msg_alloc(sizeof(*hdr),
        sizeof(struct virtio_snd_hdr), GFP_KERNEL);
 if (!msg)
  return -ENOMEM;

 kctl->items = devm_kcalloc(&vdev->dev, n, sizeof(*kctl->items),
       GFP_KERNEL);
 if (!kctl->items) {
  virtsnd_ctl_msg_unref(msg);
  return -ENOMEM;
 }

 sg_init_one(&sg, kctl->items, n * sizeof(*kctl->items));

 hdr = virtsnd_ctl_msg_request(msg);
 hdr->hdr.code = cpu_to_le32(VIRTIO_SND_R_CTL_ENUM_ITEMS);
 hdr->control_id = cpu_to_le32(cid);

 return virtsnd_ctl_msg_send(snd, msg, NULL, &sg, false);
}

/**
 * virtsnd_kctl_parse_cfg() - Parse the control element configuration.
 * @snd: VirtIO sound device.
 *
 * This function is called during initial device initialization.
 *
 * Context: Any context that permits to sleep.
 * Return: 0 on success, -errno on failure.
 */
int virtsnd_kctl_parse_cfg(struct virtio_snd *snd)
{
 struct virtio_device *vdev = snd->vdev;
 u32 i;
 int rc;

 virtio_cread_le(vdev, struct virtio_snd_config, controls,
   &snd->nkctls);
 if (!snd->nkctls)
  return 0;

 snd->kctl_infos = devm_kcalloc(&vdev->dev, snd->nkctls,
           sizeof(*snd->kctl_infos), GFP_KERNEL);
 if (!snd->kctl_infos)
  return -ENOMEM;

 snd->kctls = devm_kcalloc(&vdev->dev, snd->nkctls, sizeof(*snd->kctls),
      GFP_KERNEL);
 if (!snd->kctls)
  return -ENOMEM;

 rc = virtsnd_ctl_query_info(snd, VIRTIO_SND_R_CTL_INFO, 0, snd->nkctls,
        sizeof(*snd->kctl_infos), snd->kctl_infos);
 if (rc)
  return rc;

 for (i = 0; i < snd->nkctls; ++i) {
  struct virtio_snd_ctl_info *kinfo = &snd->kctl_infos[i];
  unsigned int type = le32_to_cpu(kinfo->type);

  if (type == VIRTIO_SND_CTL_TYPE_ENUMERATED) {
   rc = virtsnd_kctl_get_enum_items(snd, i);
   if (rc)
    return rc;
  }
 }

 return 0;
}

/**
 * virtsnd_kctl_build_devs() - Build ALSA control elements.
 * @snd: VirtIO sound device.
 *
 * Context: Any context that permits to sleep.
 * Return: 0 on success, -errno on failure.
 */
int virtsnd_kctl_build_devs(struct virtio_snd *snd)
{
 unsigned int cid;

 for (cid = 0; cid < snd->nkctls; ++cid) {
  struct virtio_snd_ctl_info *kinfo = &snd->kctl_infos[cid];
  struct virtio_kctl *kctl = &snd->kctls[cid];
  struct snd_kcontrol_new kctl_new;
  unsigned int i;
  int rc;

  memset(&kctl_new, 0, sizeof(kctl_new));

  kctl_new.iface = SNDRV_CTL_ELEM_IFACE_MIXER;
  kctl_new.name = kinfo->name;
  kctl_new.index = le32_to_cpu(kinfo->index);

  for (i = 0; i < ARRAY_SIZE(g_v2a_access_map); ++i)
   if (le32_to_cpu(kinfo->access) & (1 << i))
    kctl_new.access |= g_v2a_access_map[i];

  if (kctl_new.access & (SNDRV_CTL_ELEM_ACCESS_TLV_READ |
           SNDRV_CTL_ELEM_ACCESS_TLV_WRITE |
           SNDRV_CTL_ELEM_ACCESS_TLV_COMMAND)) {
   kctl_new.access |= SNDRV_CTL_ELEM_ACCESS_TLV_CALLBACK;
   kctl_new.tlv.c = virtsnd_kctl_tlv_op;
  }

  kctl_new.info = virtsnd_kctl_info;
  kctl_new.get = virtsnd_kctl_get;
  kctl_new.put = virtsnd_kctl_put;
  kctl_new.private_value = cid;

  kctl->kctl = snd_ctl_new1(&kctl_new, snd);
  if (!kctl->kctl)
   return -ENOMEM;

  rc = snd_ctl_add(snd->card, kctl->kctl);
  if (rc)
   return rc;
 }

 return 0;
}

/**
 * virtsnd_kctl_event() - Handle the control element event notification.
 * @snd: VirtIO sound device.
 * @event: VirtIO sound event.
 *
 * Context: Interrupt context.
 */
void virtsnd_kctl_event(struct virtio_snd *snd, struct virtio_snd_event *event)
{
 struct virtio_snd_ctl_event *kevent =
  (struct virtio_snd_ctl_event *)event;
 struct virtio_kctl *kctl;
 unsigned int cid = le16_to_cpu(kevent->control_id);
 unsigned int mask = 0;
 unsigned int i;

 if (cid >= snd->nkctls)
  return;

 for (i = 0; i < ARRAY_SIZE(g_v2a_mask_map); ++i)
  if (le16_to_cpu(kevent->mask) & (1 << i))
   mask |= g_v2a_mask_map[i];


 kctl = &snd->kctls[cid];

 snd_ctl_notify(snd->card, mask, &kctl->kctl->id);
}