/****************************************************************************
 * ieee802154/common/ieee802154_addrparse.c
 *
 *   Copyright (C) 2015 Sebastien Lorquet. All rights reserved.
 *   Author: Sebastien Lorquet <sebastien@lorquet.fr>
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in
 *    the documentation and/or other materials provided with the
 *    distribution.
 * 3. Neither the name NuttX nor the names of its contributors may be
 *    used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
 * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 *
 ****************************************************************************/

/****************************************************************************
 * Included Files
 ****************************************************************************/

#include <nuttx/config.h>
#include <stdint.h>
#include <string.h>
#include <errno.h>
#include <nuttx/ieee802154/ieee802154.h>
#include <apps/ieee802154/ieee802154.h>

int ieee802154_addrstore(FAR struct ieee802154_packet_s *packet,
                         FAR struct ieee802154_addr_s *dest,
                         FAR struct ieee802154_addr_s *src)
{
  int index=3; //skip fc and seq

  /* encode dest addr */

  if(dest == NULL || dest->ia_len == 0)
    {
      packet->data[1] = (packet->data[1] & ~IEEE802154_FC2_DADDR) | IEEE802154_DADDR_NONE;
    }
  else
    {
      memcpy(packet->data+index, &dest->ia_panid, 2);
      index += 2; /* skip dest pan id */
      if(dest->ia_len == 2)
        {
          memcpy(packet->data+index, &dest->ia_saddr, 2);
          index += 2; /* skip dest addr */
          packet->data[1] = (packet->data[1] & ~IEEE802154_FC2_DADDR) | IEEE802154_DADDR_SHORT;
        }
      else if(dest->ia_len == 8)
        {
          memcpy(packet->data+index, &dest->ia_panid, 2);
          index += 2; /* skip dest pan id */
          memcpy(packet->data+index, dest->ia_eaddr, 8);
          index += 8; /* skip dest addr */
          packet->data[1] = (packet->data[1] & ~IEEE802154_FC2_DADDR) | IEEE802154_DADDR_EXT;
        }
      else
        {
          return -EINVAL;
        }
    }

  packet->data[0] &= ~IEEE802154_FC1_INTRA;
  if( (dest != NULL && dest->ia_len != 0) && (src != NULL && src->ia_len != 0) )
    {
      /* we have both adresses, encode source pan id according to compression */
      if( dest->ia_panid == src->ia_panid)
        {
          packet->data[0] |= IEEE802154_FC1_INTRA;
        }
    }


  /* encode source addr */

  if(src == NULL || src->ia_len==0)
    {
    packet->data[1] = (packet->data[1] & ~IEEE802154_FC2_SADDR) | IEEE802154_SADDR_NONE;
    }
  else
    {
      /*add src pan id if it was not compressed before*/
      if(!(packet->data[0] & IEEE802154_FC1_INTRA))
        {
          memcpy(packet->data+index, &src->ia_panid, 2);
          index += 2; /*skip src pan id*/
        }
      if(src->ia_len == 2)
        {
          memcpy(packet->data+index, &src->ia_saddr, 2);
          index += 2; /* skip src addr */
          packet->data[1] = (packet->data[1] & ~IEEE802154_FC2_SADDR) | IEEE802154_SADDR_SHORT;
        }
      else if(src->ia_len == 8)
        {
          memcpy(packet->data+index, src->ia_eaddr, 8);
          index += 8; /* skip src addr */
          packet->data[1] = (packet->data[1] & ~IEEE802154_FC2_SADDR) | IEEE802154_SADDR_EXT;
        }
      else
        {
          return -EINVAL;
        }
    }

  return index;
}