/*      $NetBSD: sdp_put.c,v 1.6 2011/04/16 07:19:36 plunky Exp $       */

/*-
* Copyright (c) 2009 The NetBSD Foundation, Inc.
* All rights reserved.
*
* This code is derived from software contributed to The NetBSD Foundation
* by Iain Hibbert.
*
* 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.
*
* THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. 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 FOUNDATION 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.
*/

#include <sys/cdefs.h>
__RCSID("$NetBSD: sdp_put.c,v 1.6 2011/04/16 07:19:36 plunky Exp $");

#include <bluetooth.h>
#include <limits.h>
#include <sdp.h>
#include <string.h>

/******************************************************************************
*      sdp_put_xxxx(data, value)
*
* write a value to data space and advance data pointers,
* fail if data space is not large enough
*/

bool
sdp_put_data(sdp_data_t *data, sdp_data_t *value)
{
       ssize_t len;

       len = value->end - value->next;

       if (len > data->end - data->next)
               return false;

       memcpy(data->next, value->next, (size_t)len);
       data->next += len;
       return true;
}

bool
sdp_put_attr(sdp_data_t *data, uint16_t attr, sdp_data_t *value)
{
       sdp_data_t d = *data;

       if (!sdp_put_uint16(&d, attr)
           || sdp_data_size(value) != (value->end - value->next)
           || !sdp_put_data(&d, value))
               return false;

       *data = d;
       return true;
}

bool
sdp_put_uuid(sdp_data_t *data, const uuid_t *uuid)
{
       uuid_t u = *uuid;

       u.time_low = 0;

       if (uuid_equal(&u, &BLUETOOTH_BASE_UUID, NULL) == 0)
               return sdp_put_uuid128(data, uuid);

       if (uuid->time_low > UINT16_MAX)
               return sdp_put_uuid32(data, (uint32_t)uuid->time_low);

       return sdp_put_uuid16(data, (uint16_t)uuid->time_low);
}

bool
sdp_put_uuid16(sdp_data_t *data, uint16_t uuid)
{

       if (data->next + 3 > data->end)
               return false;

       data->next[0] = SDP_DATA_UUID16;
       be16enc(data->next + 1, uuid);
       data->next += 3;
       return true;
}

bool
sdp_put_uuid32(sdp_data_t *data, uint32_t uuid)
{

       if (data->next + 5 > data->end)
               return false;

       data->next[0] = SDP_DATA_UUID32;
       be32enc(data->next + 1, uuid);
       data->next += 5;
       return true;
}

bool
sdp_put_uuid128(sdp_data_t *data, const uuid_t *uuid)
{

       if (data->next + 17 > data->end)
               return false;

       data->next[0] = SDP_DATA_UUID128;
       uuid_enc_be(data->next + 1, uuid);
       data->next += 17;
       return true;
}

bool
sdp_put_bool(sdp_data_t *data, bool value)
{

       if (data->next + 2 > data->end)
               return false;

       data->next[0] = SDP_DATA_BOOL;
       data->next[1] = (value ? 0x01 : 0x00);
       data->next += 2;
       return true;
}

bool
sdp_put_uint(sdp_data_t *data, uintmax_t value)
{

       if (value > UINT64_MAX)
               return false;

       if (value > UINT32_MAX)
               return sdp_put_uint64(data, (uint64_t)value);

       if (value > UINT16_MAX)
               return sdp_put_uint32(data, (uint32_t)value);

       if (value > UINT8_MAX)
               return sdp_put_uint16(data, (uint16_t)value);

       return sdp_put_uint8(data, (uint8_t)value);
}

bool
sdp_put_uint8(sdp_data_t *data, uint8_t value)
{

       if (data->next + 2 > data->end)
               return false;

       data->next[0] = SDP_DATA_UINT8;
       data->next[1] = value;
       data->next += 2;
       return true;
}

bool
sdp_put_uint16(sdp_data_t *data, uint16_t value)
{

       if (data->next + 3 > data->end)
               return false;

       data->next[0] = SDP_DATA_UINT16;
       be16enc(data->next + 1, value);
       data->next += 3;
       return true;
}

bool
sdp_put_uint32(sdp_data_t *data, uint32_t value)
{

       if (data->next + 5 > data->end)
               return false;

       data->next[0] = SDP_DATA_UINT32;
       be32enc(data->next + 1, value);
       data->next += 5;
       return true;
}

bool
sdp_put_uint64(sdp_data_t *data, uint64_t value)
{

       if (data->next + 9 > data->end)
               return false;

       data->next[0] = SDP_DATA_UINT64;
       be64enc(data->next + 1, value);
       data->next += 9;
       return true;
}

bool
sdp_put_int(sdp_data_t *data, intmax_t value)
{

       if (value > INT64_MAX || value < INT64_MIN)
               return false;

       if (value > INT32_MAX || value < INT32_MIN)
               return sdp_put_int64(data, (int64_t)value);

       if (value > INT16_MAX || value < INT16_MIN)
               return sdp_put_int32(data, (int32_t)value);

       if (value > INT8_MAX || value < INT8_MIN)
               return sdp_put_int16(data, (int16_t)value);

       return sdp_put_int8(data, (int8_t)value);
}

bool
sdp_put_int8(sdp_data_t *data, int8_t value)
{

       if (data->next + 2 > data->end)
               return false;

       data->next[0] = SDP_DATA_INT8;
       data->next[1] = (uint8_t)value;
       data->next += 2;
       return true;
}

bool
sdp_put_int16(sdp_data_t *data, int16_t value)
{

       if (data->next + 3 > data->end)
               return false;

       data->next[0] = SDP_DATA_INT16;
       be16enc(data->next + 1, (uint16_t)value);
       data->next += 3;
       return true;
}

bool
sdp_put_int32(sdp_data_t *data, int32_t value)
{

       if (data->next + 5 > data->end)
               return false;

       data->next[0] = SDP_DATA_INT32;
       be32enc(data->next + 1, (uint32_t)value);
       data->next += 5;
       return true;
}

bool
sdp_put_int64(sdp_data_t *data, int64_t value)
{

       if (data->next + 9 > data->end)
               return false;

       data->next[0] = SDP_DATA_INT64;
       be64enc(data->next + 1, (uint64_t)value);
       data->next += 9;
       return true;
}

static bool
_sdp_put_ext(uint8_t type, sdp_data_t *data, ssize_t len)
{
       uint8_t *p = data->next;

       if (len == -1) {
               if (p + 2 > data->end)
                       return false;

               len = data->end - p - 2;

               if (len > UINT8_MAX)
                       len -= 1;

               if (len > UINT16_MAX)
                       len -= 2;
       }

       if ((size_t)len > UINT32_MAX)
               return false;

       if ((size_t)len > UINT16_MAX) {
               if (len > data->end - 5 - p)
                       return false;

               p[0] = type | SDP_DATA_EXT32;
               be32enc(p + 1, (uint32_t)len);
               p += 5;
       } else if ((size_t)len > UINT8_MAX) {
               if (len > data->end - 3 - p)
                       return false;

               p[0] = type | SDP_DATA_EXT16;
               be16enc(p + 1, (uint16_t)len);
               p += 3;
       } else {
               if (len > data->end - 2 - p)
                       return false;

               p[0] = type | SDP_DATA_EXT8;
               p[1] = (uint8_t)len;
               p += 2;
       }

       data->next = p;
       return true;
}

bool
sdp_put_seq(sdp_data_t *data, ssize_t len)
{

       return _sdp_put_ext(SDP_DATA_SEQ, data, len);
}

bool
sdp_put_alt(sdp_data_t *data, ssize_t len)
{

       return _sdp_put_ext(SDP_DATA_ALT, data, len);
}

bool
sdp_put_str(sdp_data_t *data, const char *str, ssize_t len)
{

       if (len == -1)
               len = strlen(str);

       if (!_sdp_put_ext(SDP_DATA_STR, data, len))
               return false;

       memcpy(data->next, str, (size_t)len);
       data->next += len;
       return true;
}

bool
sdp_put_url(sdp_data_t *data, const char *url, ssize_t len)
{

       if (len == -1)
               len = strlen(url);

       if (!_sdp_put_ext(SDP_DATA_URL, data, len))
               return false;

       memcpy(data->next, url, (size_t)len);
       data->next += len;
       return true;
}