/*####COPYRIGHTBEGIN####
 -------------------------------------------
 Copyright (C) 2005 Steve Karg

 This program is free software; you can redistribute it and/or
 modify it under the terms of the GNU General Public License
 as published by the Free Software Foundation; either version 2
 of the License, or (at your option) any later version.

 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with this program; if not, write to:
 The Free Software Foundation, Inc.
 59 Temple Place - Suite 330
 Boston, MA  02111-1307, USA.

 As a special exception, if other files instantiate templates or
 use macros or inline functions from this file, or you compile
 this file and link it with other works to produce a work based
 on this file, this file does not by itself cause the resulting
 work to be covered by the GNU General Public License. However
 the source code for this file must still be made available in
 accordance with section (3) of the GNU General Public License.

 This exception does not invalidate any other reasons why a work
 based on this file might be covered by the GNU General Public
 License.
 -------------------------------------------
####COPYRIGHTEND####*/
#include <stdbool.h>
#include <stdint.h>
#include <stddef.h>
#include "bits.h"
#include "apdu.h"
#include "bacdef.h"
#include "bacdcode.h"
#include "bacenum.h"
#include "tsm.h"
#include "iam.h"

// Confirmed Function Handlers
// If they are not set, they are handled by a reject message
static confirmed_function 
Confirmed_Function[MAX_BACNET_CONFIRMED_SERVICE];

void apdu_set_confirmed_handler(
  BACNET_CONFIRMED_SERVICE service_choice,
  confirmed_function pFunction)
{
  if (service_choice < MAX_BACNET_CONFIRMED_SERVICE)
    Confirmed_Function[service_choice] = pFunction;
}

// Allow the APDU handler to automatically reject
static confirmed_function Unrecognized_Service_Handler;

void apdu_set_unrecognized_service_handler_handler(
  confirmed_function pFunction)
{
    Unrecognized_Service_Handler = pFunction;
}

// Unconfirmed Function Handlers
// If they are not set, they are not handled
static unconfirmed_function 
Unconfirmed_Function[MAX_BACNET_UNCONFIRMED_SERVICE] = 
{
    iam_handler,
    NULL
};

void apdu_set_unconfirmed_handler(
  BACNET_UNCONFIRMED_SERVICE service_choice,
  unconfirmed_function pFunction)
{
  if (service_choice < MAX_BACNET_UNCONFIRMED_SERVICE)
    Unconfirmed_Function[service_choice] = pFunction;
}

// Confirmed ACK Function Handlers
static void *Confirmed_ACK_Function[MAX_BACNET_CONFIRMED_SERVICE];

void apdu_set_confirmed_simple_ack_handler(
  BACNET_CONFIRMED_SERVICE service_choice,
  confirmed_simple_ack_function pFunction)
{
  switch (service_choice) 
  {
    case SERVICE_CONFIRMED_ACKNOWLEDGE_ALARM:
    case SERVICE_CONFIRMED_COV_NOTIFICATION:
    case SERVICE_CONFIRMED_EVENT_NOTIFICATION:
    case SERVICE_CONFIRMED_SUBSCRIBE_COV:
    case SERVICE_CONFIRMED_SUBSCRIBE_COV_PROPERTY:
    case SERVICE_CONFIRMED_LIFE_SAFETY_OPERATION:
    // Object Access Services
    case SERVICE_CONFIRMED_ADD_LIST_ELEMENT:
    case SERVICE_CONFIRMED_REMOVE_LIST_ELEMENT:
    case SERVICE_CONFIRMED_DELETE_OBJECT:
    case SERVICE_CONFIRMED_WRITE_PROPERTY:
    case SERVICE_CONFIRMED_WRITE_PROPERTY_MULTIPLE:
    // Remote Device Management Services
    case SERVICE_CONFIRMED_DEVICE_COMMUNICATION_CONTROL:
    case SERVICE_CONFIRMED_TEXT_MESSAGE:
    case SERVICE_CONFIRMED_REINITIALIZE_DEVICE:
    // Virtual Terminal Services
    case SERVICE_CONFIRMED_VT_CLOSE:
    // Security Services
    case SERVICE_CONFIRMED_REQUEST_KEY:
      Confirmed_ACK_Function[service_choice] = (void *)pFunction;
      break;
    default:
      break;
  }
}

void apdu_set_confirmed_ack_handler(
  BACNET_CONFIRMED_SERVICE service_choice,
  confirmed_ack_function pFunction)
{
  switch (service_choice) 
  {
    case SERVICE_CONFIRMED_GET_ALARM_SUMMARY:
    case SERVICE_CONFIRMED_GET_ENROLLMENT_SUMMARY:
    case SERVICE_CONFIRMED_GET_EVENT_INFORMATION:
    // File Access Services
    case SERVICE_CONFIRMED_ATOMIC_READ_FILE:
    case SERVICE_CONFIRMED_ATOMIC_WRITE_FILE:
    // Object Access Services
    case SERVICE_CONFIRMED_CREATE_OBJECT:
    case SERVICE_CONFIRMED_READ_PROPERTY:
    case SERVICE_CONFIRMED_READ_PROPERTY_CONDITIONAL:
    case SERVICE_CONFIRMED_READ_PROPERTY_MULTIPLE:
    case SERVICE_CONFIRMED_READ_RANGE:
    // Remote Device Management Services
    case SERVICE_CONFIRMED_PRIVATE_TRANSFER:
    // Virtual Terminal Services
    case SERVICE_CONFIRMED_VT_OPEN:
    case SERVICE_CONFIRMED_VT_DATA:
    // Security Services
    case SERVICE_CONFIRMED_AUTHENTICATE:
      Confirmed_ACK_Function[service_choice] = (void *)pFunction;
      break;
    default:
      break;
  }
}

void apdu_handler(
  BACNET_ADDRESS *src,  // source address
  bool data_expecting_reply,
  uint8_t *apdu, // APDU data
  uint16_t apdu_len)
{
  BACNET_CONFIRMED_SERVICE_DATA service_data = {0};
  BACNET_CONFIRMED_SERVICE_ACK_DATA service_ack_data = {0};
  uint8_t invoke_id = 0;
  uint8_t service_choice = 0;
  uint8_t *service_request = NULL;
  uint16_t service_request_len = 0;
  uint16_t len = 0; // counts where we are in PDU

  (void)data_expecting_reply;
  if (apdu)
  {
    // PDU Type
    switch (apdu[0] & 0xF0)
    {
      case PDU_TYPE_CONFIRMED_SERVICE_REQUEST:
        service_data.segmented_message = (apdu[0] & BIT3) ? true : false;
        service_data.more_follows = (apdu[0] & BIT2) ? true : false;
        service_data.segmented_response_accepted = (apdu[0] & BIT1) ? true : false;
        service_data.max_segs = decode_max_segs(apdu[1]);
        service_data.max_resp = decode_max_apdu(apdu[1]);
        service_data.invoke_id = apdu[2];
        len = 3;
        if (service_data.segmented_message)
        {
          service_data.sequence_number = apdu[len++];
          service_data.proposed_window_number = apdu[len++];
        }
        service_choice = apdu[len++];
        service_request = &apdu[len++];
        service_request_len = apdu_len - len;
        if (service_choice < MAX_BACNET_CONFIRMED_SERVICE)
        {
          if (Confirmed_Function[service_choice])
            Confirmed_Function[service_choice](
              service_request,
              service_request_len,
              src,
              &service_data);
          else
          {
            if (Unrecognized_Service_Handler)
              Unrecognized_Service_Handler(
                service_request,
                service_request_len,
                src,
                &service_data);
          }
        }
        break;
      case PDU_TYPE_UNCONFIRMED_SERVICE_REQUEST:
        service_choice = apdu[1];
        service_request = &apdu[2];
        service_request_len = apdu_len - 2;
        if (service_choice < MAX_BACNET_UNCONFIRMED_SERVICE)
        {
          if (Unconfirmed_Function[service_choice])
            Unconfirmed_Function[service_choice](
                service_request,
                service_request_len,
                src);
        }
        break;
      case PDU_TYPE_SIMPLE_ACK:
        invoke_id = apdu[1];
        service_choice = apdu[2];
        switch (service_choice) 
        {
          case SERVICE_CONFIRMED_ACKNOWLEDGE_ALARM:
          case SERVICE_CONFIRMED_COV_NOTIFICATION:
          case SERVICE_CONFIRMED_EVENT_NOTIFICATION:
          case SERVICE_CONFIRMED_SUBSCRIBE_COV:
          case SERVICE_CONFIRMED_SUBSCRIBE_COV_PROPERTY:
          case SERVICE_CONFIRMED_LIFE_SAFETY_OPERATION:
          // Object Access Services
          case SERVICE_CONFIRMED_ADD_LIST_ELEMENT:
          case SERVICE_CONFIRMED_REMOVE_LIST_ELEMENT:
          case SERVICE_CONFIRMED_DELETE_OBJECT:
          case SERVICE_CONFIRMED_WRITE_PROPERTY:
          case SERVICE_CONFIRMED_WRITE_PROPERTY_MULTIPLE:
          // Remote Device Management Services
          case SERVICE_CONFIRMED_DEVICE_COMMUNICATION_CONTROL:
          case SERVICE_CONFIRMED_REINITIALIZE_DEVICE:
          case SERVICE_CONFIRMED_TEXT_MESSAGE:
          // Virtual Terminal Services
          case SERVICE_CONFIRMED_VT_CLOSE:
          // Security Services
          case SERVICE_CONFIRMED_REQUEST_KEY:
            if (Confirmed_ACK_Function[service_choice])
            {
              ((confirmed_simple_ack_function)
              Confirmed_ACK_Function[service_choice])(
                src,
                invoke_id); 
            }
            else
              tsm_free_invoke_id(invoke_id);
            break;
          default:
            break;
        }
        break;
      case PDU_TYPE_COMPLEX_ACK:
        service_ack_data.segmented_message = (apdu[0] & BIT3) ? true : false;
        service_ack_data.more_follows = (apdu[0] & BIT2) ? true : false;
        service_ack_data.invoke_id = apdu[1];
        len = 2;
        if (service_ack_data.segmented_message)
        {
          service_ack_data.sequence_number = apdu[len++];
          service_ack_data.proposed_window_number = apdu[len++];
        }
        service_choice = apdu[len++];
        service_request = &apdu[len++];
        service_request_len = apdu_len - len;
        invoke_id = apdu[1];
        service_choice = apdu[2];
        switch (service_choice) 
        {
          case SERVICE_CONFIRMED_GET_ALARM_SUMMARY:
          case SERVICE_CONFIRMED_GET_ENROLLMENT_SUMMARY:
          case SERVICE_CONFIRMED_GET_EVENT_INFORMATION:
          // File Access Services
          case SERVICE_CONFIRMED_ATOMIC_READ_FILE:
          case SERVICE_CONFIRMED_ATOMIC_WRITE_FILE:
          // Object Access Services
          case SERVICE_CONFIRMED_CREATE_OBJECT:
          case SERVICE_CONFIRMED_READ_PROPERTY:
          case SERVICE_CONFIRMED_READ_PROPERTY_CONDITIONAL:
          case SERVICE_CONFIRMED_READ_PROPERTY_MULTIPLE:
          case SERVICE_CONFIRMED_READ_RANGE:
          case SERVICE_CONFIRMED_PRIVATE_TRANSFER:
          // Virtual Terminal Services
          case SERVICE_CONFIRMED_VT_OPEN:
          case SERVICE_CONFIRMED_VT_DATA:
          // Security Services
          case SERVICE_CONFIRMED_AUTHENTICATE:
            if (Confirmed_ACK_Function[service_choice])
            {
              ((confirmed_ack_function)
              Confirmed_ACK_Function[service_choice])(
                service_request,
                service_request_len,
                src,
                &service_ack_data); 
            }
            else
              tsm_free_invoke_id(invoke_id);
            break;
          default:
            break;
        }
        break;
      case PDU_TYPE_SEGMENT_ACK:
      case PDU_TYPE_ERROR:
      case PDU_TYPE_REJECT:
      case PDU_TYPE_ABORT:
        invoke_id = apdu[1];
        tsm_free_invoke_id(invoke_id);
        break;
      default:
        break;
    }
  }
  return;
}