/******************************************************************************
 *
 *  Copyright (C) 2014 Google, Inc.
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at:
 *
 *  http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *
 ******************************************************************************/
#include <stdbool.h>
#include "bt_trace.h"
#include "bdaddr.h"
#include "bt_types.h"
#include "controller.h"
#include "event_mask.h"
#include "hcimsgs.h"
#include "hci_layer.h"
#include "hci_packet_factory.h"
#include "hci_packet_parser.h"
#include "btm_ble_api.h"
#include "version.h"

//#include "bluedroid_test.h" /*FOr Test Case*/
const bt_event_mask_t BLE_EVENT_MASK = { "\x00\x00\x00\x00\x00\x00\x06\x7f" };

#if (BLE_INCLUDED)
const bt_event_mask_t CLASSIC_EVENT_MASK = { HCI_DUMO_EVENT_MASK_EXT };
#else
const bt_event_mask_t CLASSIC_EVENT_MASK = { HCI_LISBON_EVENT_MASK_EXT };
#endif

// TODO(zachoverflow): factor out into common module
const uint8_t SCO_HOST_BUFFER_SIZE = 0xff;

#define HCI_SUPPORTED_COMMANDS_ARRAY_SIZE 64
#define MAX_FEATURES_CLASSIC_PAGE_COUNT 3
#define BLE_SUPPORTED_STATES_SIZE         8
#define BLE_SUPPORTED_FEATURES_SIZE       8

static const hci_t *hci;
static const hci_packet_factory_t *packet_factory;
static const hci_packet_parser_t *packet_parser;

static bt_bdaddr_t address;
static bt_version_t bt_version;

static uint8_t supported_commands[HCI_SUPPORTED_COMMANDS_ARRAY_SIZE];
static bt_device_features_t features_classic[MAX_FEATURES_CLASSIC_PAGE_COUNT];
static uint8_t last_features_classic_page_index;

static uint16_t acl_data_size_classic;
static uint16_t acl_data_size_ble;
static uint16_t acl_buffer_count_classic;
static uint8_t acl_buffer_count_ble;

static uint8_t ble_white_list_size;
static uint8_t ble_resolving_list_max_size;
static uint8_t ble_supported_states[BLE_SUPPORTED_STATES_SIZE];
static bt_device_features_t features_ble;
static uint16_t ble_suggested_default_data_length;

static bool readable;
static bool ble_supported;
static bool simple_pairing_supported;
static bool secure_connections_supported;

devctl_reset_callback reset_cb;
static uint8_t page_number = 0;
static void devctl_hdl_cmd_complete(BT_HDR *response, void *context) {
  BT_HDR *command = NULL;
  command_opcode_t opcode;
  uint8_t *stream = response->data + response->offset;

  STREAM_SKIP_UINT16(stream);      //skip event_code and total length field
  STREAM_SKIP_UINT8(stream);       //skip command_credits field
  STREAM_TO_UINT16(opcode, stream);

  switch (opcode) {
  case HCI_RESET:
    packet_parser->parse_generic_command_complete(response);
    command = packet_factory->make_read_buffer_size();
    break;
  case HCI_READ_BUFFER_SIZE:
    packet_parser->parse_read_buffer_size_response(
        response, &acl_data_size_classic, &acl_buffer_count_classic);
    command = packet_factory->make_host_buffer_size(
        L2CAP_MTU_SIZE, SCO_HOST_BUFFER_SIZE, L2CAP_HOST_FC_ACL_BUFS, 10);
    break;
  case HCI_HOST_BUFFER_SIZE:
    packet_parser->parse_generic_command_complete(response);
    command = packet_factory->make_read_local_version_info();
    break;
  case HCI_READ_LOCAL_VERSION_INFO:
    packet_parser->parse_read_local_version_info_response(response, &bt_version);
    command = packet_factory->make_read_bd_addr();
    break;
  case HCI_READ_BD_ADDR:
    packet_parser->parse_read_bd_addr_response(response, &address);
    command = packet_factory->make_read_local_supported_commands();
    break;
  case HCI_READ_LOCAL_SUPPORTED_CMDS:
    packet_parser->parse_read_local_supported_commands_response(
        response, supported_commands, HCI_SUPPORTED_COMMANDS_ARRAY_SIZE);
    page_number = 0;
    command = packet_factory->make_read_local_extended_features(page_number);
    break;
  case HCI_READ_LOCAL_EXT_FEATURES:
    if (response) {
      packet_parser->parse_read_local_extended_features_response(
	  response, &page_number,&last_features_classic_page_index,
	  features_classic, MAX_FEATURES_CLASSIC_PAGE_COUNT);
      response = NULL;
      page_number++;
    }
    if (1 == page_number) {
      simple_pairing_supported = HCI_SIMPLE_PAIRING_SUPPORTED(features_classic[0].as_array);
      if (simple_pairing_supported) {
	command = packet_factory->make_write_simple_pairing_mode(HCI_SP_MODE_ENABLED);
	break;
      }
      // BLOCK_BEGIN
#if (BLE_INCLUDED == TRUE)
      if (HCI_LE_SPT_SUPPORTED(features_classic[0].as_array)) {
	uint8_t simultaneous_le_host = HCI_SIMUL_LE_BREDR_SUPPORTED(features_classic[0].as_array) ? BTM_BLE_SIMULTANEOUS_HOST : 0;
	command = packet_factory->make_ble_write_host_support(BTM_BLE_HOST_SUPPORT, simultaneous_le_host);
	break;
      }
#endif
    }

    if (page_number <= last_features_classic_page_index &&
	page_number < MAX_FEATURES_CLASSIC_PAGE_COUNT) {
      command = packet_factory->make_read_local_extended_features(page_number);
      break;
    } else {
#if (SC_MODE_INCLUDED == TRUE)
      secure_connections_supported = HCI_SC_CTRLR_SUPPORTED(features_classic[2].as_array);
      if (secure_connections_supported) {
	command = packet_factory->make_write_secure_connections_host_support(HCI_SC_MODE_ENABLED);
	break;
      }
#endif
#if (BLE_INCLUDED == TRUE)
      ble_supported = last_features_classic_page_index >= 1 && HCI_LE_HOST_SUPPORTED(features_classic[1].as_array);
      if (ble_supported) {
	// Request the ble white list size next
	command = packet_factory->make_ble_read_white_list_size();
	break;
      }
#endif
      if (simple_pairing_supported) {
	command = packet_factory->make_set_event_mask(&CLASSIC_EVENT_MASK);
	break;
      }
    }
    // BLOCK_END

  case HCI_WRITE_SIMPLE_PAIRING_MODE:
    if (response) {
      packet_parser->parse_generic_command_complete(response);
      response = NULL;
    }
      // BLOCK_BEGIN
#if (BLE_INCLUDED == TRUE)
     if (HCI_LE_SPT_SUPPORTED(features_classic[0].as_array)) {
	uint8_t simultaneous_le_host = HCI_SIMUL_LE_BREDR_SUPPORTED(features_classic[0].as_array) ? BTM_BLE_SIMULTANEOUS_HOST : 0;
	command = packet_factory->make_ble_write_host_support(BTM_BLE_HOST_SUPPORT, simultaneous_le_host);
	break;
      }
#endif
    if (page_number <= last_features_classic_page_index &&
	page_number < MAX_FEATURES_CLASSIC_PAGE_COUNT) {
      command = packet_factory->make_read_local_extended_features(page_number);
      break;
    } else {
#if (SC_MODE_INCLUDED == TRUE)
      secure_connections_supported = HCI_SC_CTRLR_SUPPORTED(features_classic[2].as_array);
      if (secure_connections_supported) {
	command = packet_factory->make_write_secure_connections_host_support(HCI_SC_MODE_ENABLED);
	break;
      }
#endif
#if (BLE_INCLUDED == TRUE)
      ble_supported = last_features_classic_page_index >= 1 && HCI_LE_HOST_SUPPORTED(features_classic[1].as_array);
      if (ble_supported) {
	// Request the ble white list size next
	command = packet_factory->make_ble_read_white_list_size();
	break;
      }
#endif
      if (simple_pairing_supported) {
	command = packet_factory->make_set_event_mask(&CLASSIC_EVENT_MASK);
	break;
      }
    }

#if (SC_MODE_INCLUDED == TRUE)
    case HCI_WRITE_SECURE_CONNS_SUPPORT:
      if (response) {
        packet_parser->parse_generic_command_complete(response);
        response = NULL;
      }
#if (BLE_INCLUDED == TRUE)
      ble_supported = last_features_classic_page_index >= 1 && HCI_LE_HOST_SUPPORTED(features_classic[1].as_array);
      if (ble_supported) {
	// Request the ble white list size next
	command = packet_factory->make_ble_read_white_list_size();
	break;
      }
#endif /* (BLE_INCLUDED == TRUE) */
      if (simple_pairing_supported) {
	command = packet_factory->make_set_event_mask(&CLASSIC_EVENT_MASK);
	break;
      }
#endif /* (SC_MODE_INCLUDED == TRUE) */

#if (BLE_INCLUDED == TRUE)
  case HCI_WRITE_LE_HOST_SUPPORT:
    if (response) {
      packet_parser->parse_generic_command_complete(response);
      response = NULL;
    }
    if (page_number <= last_features_classic_page_index &&
	page_number < MAX_FEATURES_CLASSIC_PAGE_COUNT) {
      command = packet_factory->make_read_local_extended_features(page_number);
      break;
    } else {
#if (SC_MODE_INCLUDED == TRUE)
      secure_connections_supported = HCI_SC_CTRLR_SUPPORTED(features_classic[2].as_array);
      if (secure_connections_supported) {
	command = packet_factory->make_write_secure_connections_host_support(HCI_SC_MODE_ENABLED);
	break;
      }
#endif
      ble_supported = last_features_classic_page_index >= 1 && HCI_LE_HOST_SUPPORTED(features_classic[1].as_array);
      if (ble_supported) {
	// Request the ble white list size next
	command = packet_factory->make_ble_read_white_list_size();
	break;
      }
      if (simple_pairing_supported) {
	command = packet_factory->make_set_event_mask(&CLASSIC_EVENT_MASK);
	break;
      }
    }
  case HCI_BLE_READ_WHITE_LIST_SIZE:
    if (response) {
      packet_parser->parse_ble_read_white_list_size_response(response, &ble_white_list_size);
      response = NULL;
    }
    if (ble_supported) {
      // Request the ble buffer size next
      command = packet_factory->make_ble_read_buffer_size();
      break;
    }
  // Fall Through if no next command generated
  case HCI_BLE_READ_BUFFER_SIZE:
    if (response) {
      packet_parser->parse_ble_read_buffer_size_response(
          response, &acl_data_size_ble, &acl_buffer_count_ble);
      response = NULL;
    }
    // Response of 0 indicates ble has the same buffer size as classic
    if (acl_data_size_ble == 0)
      acl_data_size_ble = acl_data_size_classic;
    if (ble_supported) {
      // Request the ble supported states next
      command = packet_factory->make_ble_read_supported_states();
      break;
    }
    // Fall Through if no next command generated
  case HCI_BLE_READ_SUPPORTED_STATES:
    if (response) {
      packet_parser->parse_ble_read_supported_states_response(
            response, ble_supported_states, sizeof(ble_supported_states));
      response = NULL;
    }

    if (ble_supported) {
      // Request the ble supported features next
      command = packet_factory->make_ble_read_local_supported_features();
      break;
    }
    // Fall Through if no next command generated
  case HCI_BLE_READ_LOCAL_SPT_FEAT:
    if (response) {
      packet_parser->parse_ble_read_local_supported_features_response(
            response, &features_ble);
      response = NULL;
    }

    if (ble_supported &&
	HCI_LE_ENHANCED_PRIVACY_SUPPORTED(features_ble.as_array)) {
      command = packet_factory->make_ble_read_resolving_list_size();
      break;
    }
  case HCI_BLE_READ_RESOLVING_LIST_SIZE:
    if (response) {
      packet_parser->parse_ble_read_resolving_list_size_response(
            response, &ble_resolving_list_max_size);
      response = NULL;
    }

    if (ble_supported &&
	HCI_LE_DATA_LEN_EXT_SUPPORTED(features_ble.as_array)) {
      command = packet_factory->make_ble_read_suggested_default_data_length();
      break;
    }
  case HCI_BLE_READ_DEFAULT_DATA_LENGTH:
    if (response) {
      packet_parser->parse_ble_read_suggested_default_data_length_response(
            response, &ble_suggested_default_data_length);
      response = NULL;
    }

    if (ble_supported) {
      // Set the ble event mask next
      command = packet_factory->make_ble_set_event_mask(&BLE_EVENT_MASK);
      break;
    }
  case HCI_BLE_SET_EVENT_MASK:
    if (response) {
      packet_parser->parse_generic_command_complete(response);
      response = NULL;
    }
    if (simple_pairing_supported) {
      command = packet_factory->make_set_event_mask(&CLASSIC_EVENT_MASK);
      break;
    }
#endif
  case HCI_SET_EVENT_MASK:
    if (response) {
      packet_parser->parse_generic_command_complete(response);
      response = command = NULL;
    }
    //At this point, Reset Thread should be completed well.
    readable = true;
    page_number = 0;
    if (reset_cb)
      reset_cb();

    break;
  default:
    LOG_ERROR("%s: No available opcode matched.", __func__);
    break;
  }

  if (command)
    hci->transmit_command(command, devctl_hdl_cmd_complete, NULL, NULL);
}

// Interface functions
static void devctl_reset(devctl_reset_callback reset_callback) {
  reset_cb = reset_callback;
  BT_HDR *command = packet_factory->make_read_buffer_size();
  //BT_HDR *command = packet_factory->make_reset();
  LOG_ERROR("Device Control Send Device Read Buffer Size Command\n");
  page_number = 0;
  if (command)
    hci->transmit_command(command, devctl_hdl_cmd_complete, NULL, NULL);
}

static void devctl_shutdown(void) {
  reset_cb = NULL;
  readable = false;
}

static bool get_is_ready(void) {
  return readable;
}

static const bt_bdaddr_t *get_address(void) {
  assert(readable);
  return &address;
}

static const bt_version_t *get_bt_version(void) {
  assert(readable);
  return &bt_version;
}

// TODO(zachoverflow): hide inside, move decoder inside too
static const bt_device_features_t *get_features_classic(int index) {
  assert(readable);
  assert(index < MAX_FEATURES_CLASSIC_PAGE_COUNT);
  return &features_classic[index];
}

static uint8_t get_last_features_classic_index(void) {
  assert(readable);
  return last_features_classic_page_index;
}

static const bt_device_features_t *get_features_ble(void) {
  assert(readable);
  assert(ble_supported);
  return &features_ble;
}

static const uint8_t *get_ble_supported_states(void) {
  assert(readable);
  assert(ble_supported);
  return ble_supported_states;
}

static bool supports_simple_pairing(void) {
  assert(readable);
  return simple_pairing_supported;
}

static bool supports_secure_connections(void) {
  assert(readable);
  return secure_connections_supported;
}

static bool supports_simultaneous_le_bredr(void) {
  assert(readable);
  return HCI_SIMUL_LE_BREDR_SUPPORTED(features_classic[0].as_array);
}

static bool supports_reading_remote_extended_features(void) {
  assert(readable);
  return HCI_READ_REMOTE_EXT_FEATURES_SUPPORTED(supported_commands);
}

static bool supports_interlaced_inquiry_scan(void) {
  assert(readable);
  return HCI_LMP_INTERLACED_INQ_SCAN_SUPPORTED(features_classic[0].as_array);
}

static bool supports_rssi_with_inquiry_results(void) {
  assert(readable);
  return HCI_LMP_INQ_RSSI_SUPPORTED(features_classic[0].as_array);
}

static bool supports_extended_inquiry_response(void) {
  assert(readable);
  return HCI_EXT_INQ_RSP_SUPPORTED(features_classic[0].as_array);
}

static bool supports_master_slave_role_switch(void) {
  assert(readable);
  return HCI_SWITCH_SUPPORTED(features_classic[0].as_array);
}

static bool supports_ble(void) {
  assert(readable);
  return ble_supported;
}

static bool supports_ble_privacy(void) {
  assert(readable);
  assert(ble_supported);
  return HCI_LE_ENHANCED_PRIVACY_SUPPORTED(features_ble.as_array);
}

static bool supports_ble_packet_extension(void) {
  assert(readable);
  assert(ble_supported);
  return HCI_LE_DATA_LEN_EXT_SUPPORTED(features_ble.as_array);
}

static bool supports_ble_connection_parameters_request(void) {
  assert(readable);
  assert(ble_supported);
  return HCI_LE_CONN_PARAM_REQ_SUPPORTED(features_ble.as_array);
}

static uint16_t get_acl_data_size_classic(void) {
  assert(readable);
  return acl_data_size_classic;
}

static uint16_t get_acl_data_size_ble(void) {
  assert(readable);
  assert(ble_supported);
  return acl_data_size_ble;
}

static uint16_t get_acl_packet_size_classic(void) {
  assert(readable);
  return acl_data_size_classic + HCI_DATA_PREAMBLE_SIZE;
}

static uint16_t get_acl_packet_size_ble(void) {
  assert(readable);
  return acl_data_size_ble + HCI_DATA_PREAMBLE_SIZE;
}

static uint16_t get_ble_suggested_default_data_length(void) {
  assert(readable);
  assert(ble_supported);
  return ble_suggested_default_data_length;
}

static uint16_t get_acl_buffer_count_classic(void) {
  assert(readable);
  return acl_buffer_count_classic;
}

static uint8_t get_acl_buffer_count_ble(void) {
  assert(readable);
  assert(ble_supported);
  return acl_buffer_count_ble;
}

static uint8_t get_ble_white_list_size(void) {
  assert(readable);
  assert(ble_supported);
  return ble_white_list_size;
}

static uint8_t get_ble_resolving_list_max_size(void) {
  assert(readable);
  assert(ble_supported);
  return ble_resolving_list_max_size;
}

static void set_ble_resolving_list_max_size(int resolving_list_max_size) {
  assert(readable);
  assert(ble_supported);
  ble_resolving_list_max_size = resolving_list_max_size;
}

static const controller_t interface = {
  devctl_reset,
  devctl_shutdown,
  get_is_ready,

  get_address,
  get_bt_version,

  get_features_classic,
  get_last_features_classic_index,

  get_features_ble,
  get_ble_supported_states,

  supports_simple_pairing,
  supports_secure_connections,
  supports_simultaneous_le_bredr,
  supports_reading_remote_extended_features,
  supports_interlaced_inquiry_scan,
  supports_rssi_with_inquiry_results,
  supports_extended_inquiry_response,
  supports_master_slave_role_switch,

  supports_ble,
  supports_ble_packet_extension,
  supports_ble_connection_parameters_request,
  supports_ble_privacy,

  get_acl_data_size_classic,
  get_acl_data_size_ble,

  get_acl_packet_size_classic,
  get_acl_packet_size_ble,
  get_ble_suggested_default_data_length,

  get_acl_buffer_count_classic,
  get_acl_buffer_count_ble,

  get_ble_white_list_size,

  get_ble_resolving_list_max_size,
  set_ble_resolving_list_max_size
};

const controller_t *controller_get_interface() {
  static bool loaded = false;
  if (!loaded) {
    loaded = true;

    hci = hci_layer_get_interface();
    packet_factory = hci_packet_factory_get_interface();
    packet_parser = hci_packet_parser_get_interface();
  }

  return &interface;
}