// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright (C) 2021 Intel Corporation
 */

#include <net/bluetooth/bluetooth.h>
#include <net/bluetooth/hci_core.h>

#include "aosp.h"

/* Command complete parameters of LE_Get_Vendor_Capabilities_Command
 * The parameters grow over time. The base version that declares the
 * version_supported field is v0.95. Refer to
 * https://cs.android.com/android/platform/superproject/+/master:system/
 *         bt/gd/hci/controller.cc;l=452?q=le_get_vendor_capabilities_handler
 */
struct aosp_rp_le_get_vendor_capa {
 /* v0.95: 15 octets */
 __u8 status;
 __u8 max_advt_instances;
 __u8 offloaded_resolution_of_private_address;
 __le16 total_scan_results_storage;
 __u8 max_irk_list_sz;
 __u8 filtering_support;
 __u8 max_filter;
 __u8 activity_energy_info_support;
 __le16 version_supported;
 __le16 total_num_of_advt_tracked;
 __u8 extended_scan_support;
 __u8 debug_logging_supported;
 /* v0.96: 16 octets */
 __u8 le_address_generation_offloading_support;
 /* v0.98: 21 octets */
 __le32 a2dp_source_offload_capability_mask;
 __u8 bluetooth_quality_report_support;
 /* v1.00: 25 octets */
 __le32 dynamic_audio_buffer_support;
} __packed;

#define VENDOR_CAPA_BASE_SIZE  15
#define VENDOR_CAPA_0_98_SIZE  21

void aosp_do_open(struct hci_dev *hdev)
{
 struct sk_buff *skb;
 struct aosp_rp_le_get_vendor_capa *rp;
 u16 version_supported;

 if (!hdev->aosp_capable)
  return;

 bt_dev_dbg(hdev, "Initialize AOSP extension");

 /* LE Get Vendor Capabilities Command */
 skb = __hci_cmd_sync(hdev, hci_opcode_pack(0x3f, 0x153), 0, NULL,
        HCI_CMD_TIMEOUT);
 if (IS_ERR_OR_NULL(skb)) {
  if (!skb)
   skb = ERR_PTR(-EIO);

  bt_dev_err(hdev, "AOSP get vendor capabilities (%ld)",
      PTR_ERR(skb));
  return;
 }

 /* A basic length check */
 if (skb->len < VENDOR_CAPA_BASE_SIZE)
  goto length_error;

 rp = (struct aosp_rp_le_get_vendor_capa *)skb->data;

 version_supported = le16_to_cpu(rp->version_supported);
 /* AOSP displays the version number like v0.98, v1.00, etc. */
 bt_dev_info(hdev, "AOSP extensions version v%u.%02u",
      version_supported >> 8, version_supported & 0xff);

 /* Do not support very old versions. */
 if (version_supported < 95) {
  bt_dev_warn(hdev, "AOSP capabilities version %u too old",
       version_supported);
  goto done;
 }

 if (version_supported < 98) {
  bt_dev_warn(hdev, "AOSP quality report is not supported");
  goto done;
 }

 if (skb->len < VENDOR_CAPA_0_98_SIZE)
  goto length_error;

 /* The bluetooth_quality_report_support is defined at version
  * v0.98. Refer to
  * https://cs.android.com/android/platform/superproject/+/
  *         master:system/bt/gd/hci/controller.cc;l=477
 */
 if (rp->bluetooth_quality_report_support) {
  hdev->aosp_quality_report = true;
  bt_dev_info(hdev, "AOSP quality report is supported");
 }

 goto done;

length_error:
 bt_dev_err(hdev, "AOSP capabilities length %d too short", skb->len);

done:
 kfree_skb(skb);
}

void aosp_do_close(struct hci_dev *hdev)
{
 if (!hdev->aosp_capable)
  return;

 bt_dev_dbg(hdev, "Cleanup of AOSP extension");
}

/* BQR command */
#define BQR_OPCODE   hci_opcode_pack(0x3f, 0x015e)

/* BQR report action */
#define REPORT_ACTION_ADD  0x00
#define REPORT_ACTION_DELETE  0x01
#define REPORT_ACTION_CLEAR  0x02

/* BQR event masks */
#define QUALITY_MONITORING  BIT(0)
#define APPRAOCHING_LSTO  BIT(1)
#define A2DP_AUDIO_CHOPPY  BIT(2)
#define SCO_VOICE_CHOPPY  BIT(3)

#define DEFAULT_BQR_EVENT_MASK (QUALITY_MONITORING | APPRAOCHING_LSTO | \
     A2DP_AUDIO_CHOPPY | SCO_VOICE_CHOPPY)

/* Reporting at milliseconds so as not to stress the controller too much.
 * Range: 0 ~ 65535 ms
 */
#define DEFALUT_REPORT_INTERVAL_MS 5000

struct aosp_bqr_cp {
 __u8 report_action;
 __u32 event_mask;
 __u16 min_report_interval;
} __packed;

static int enable_quality_report(struct hci_dev *hdev)
{
 struct sk_buff *skb;
 struct aosp_bqr_cp cp;

 cp.report_action = REPORT_ACTION_ADD;
 cp.event_mask = DEFAULT_BQR_EVENT_MASK;
 cp.min_report_interval = DEFALUT_REPORT_INTERVAL_MS;

 skb = __hci_cmd_sync(hdev, BQR_OPCODE, sizeof(cp), &cp,
        HCI_CMD_TIMEOUT);
 if (IS_ERR_OR_NULL(skb)) {
  if (!skb)
   skb = ERR_PTR(-EIO);

  bt_dev_err(hdev, "Enabling Android BQR failed (%ld)",
      PTR_ERR(skb));
  return PTR_ERR(skb);
 }

 kfree_skb(skb);
 return 0;
}

static int disable_quality_report(struct hci_dev *hdev)
{
 struct sk_buff *skb;
 struct aosp_bqr_cp cp = { 0 };

 cp.report_action = REPORT_ACTION_CLEAR;

 skb = __hci_cmd_sync(hdev, BQR_OPCODE, sizeof(cp), &cp,
        HCI_CMD_TIMEOUT);
 if (IS_ERR_OR_NULL(skb)) {
  if (!skb)
   skb = ERR_PTR(-EIO);

  bt_dev_err(hdev, "Disabling Android BQR failed (%ld)",
      PTR_ERR(skb));
  return PTR_ERR(skb);
 }

 kfree_skb(skb);
 return 0;
}

bool aosp_has_quality_report(struct hci_dev *hdev)
{
 return hdev->aosp_quality_report;
}

int aosp_set_quality_report(struct hci_dev *hdev, bool enable)
{
 if (!aosp_has_quality_report(hdev))
  return -EOPNOTSUPP;

 bt_dev_dbg(hdev, "quality report enable %d", enable);

 /* Enable or disable the quality report feature. */
 if (enable)
  return enable_quality_report(hdev);
 else
  return disable_quality_report(hdev);
}