/*      $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;
}