/*      $NetBSD: sdp_session.c,v 1.3 2010/11/13 19:43:56 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_session.c,v 1.3 2010/11/13 19:43:56 plunky Exp $");

#include <sys/socket.h>
#include <sys/un.h>

#include <errno.h>
#include <sdp.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include "sdp-int.h"

/*
* open session with remote Bluetooth SDP server
*/
struct sdp_session *
_sdp_open(const bdaddr_t *laddr, const bdaddr_t *raddr)
{
       struct sdp_session *    ss;
       struct sockaddr_bt      sa;
       struct linger           li;
       socklen_t               len;

       ss = calloc(1, sizeof(struct sdp_session));
       if (ss == NULL)
               goto fail;

       ss->s = socket(PF_BLUETOOTH, SOCK_SEQPACKET, BTPROTO_L2CAP);
       if (ss->s == -1)
               goto fail;

       memset(&li, 0, sizeof(li));
       li.l_onoff = 1;
       li.l_linger = 5;
       if (setsockopt(ss->s, SOL_SOCKET, SO_LINGER, &li, sizeof(li)) == -1)
               goto fail;

       if (laddr == NULL)
               laddr = BDADDR_ANY;

       memset(&sa, 0, sizeof(sa));
       sa.bt_len = sizeof(sa);
       sa.bt_family = AF_BLUETOOTH;
       bdaddr_copy(&sa.bt_bdaddr, laddr);
       if (bind(ss->s, (struct sockaddr *)&sa, sizeof(sa)) == -1)
               goto fail;

       sa.bt_psm = L2CAP_PSM_SDP;
       bdaddr_copy(&sa.bt_bdaddr, raddr);
       if (connect(ss->s, (struct sockaddr *)&sa, sizeof(sa)) == -1)
               goto fail;

       len = sizeof(ss->imtu);
       if (getsockopt(ss->s, BTPROTO_L2CAP, SO_L2CAP_IMTU, &ss->imtu, &len) == -1)
               goto fail;

       ss->ibuf = malloc((size_t)(ss->imtu));
       if (ss->ibuf == NULL)
               goto fail;

       return ss;

fail:
       _sdp_close(ss);
       return NULL;
}

/*
* open session with local SDP server
*/
struct sdp_session *
_sdp_open_local(const char *control)
{
       struct sdp_session *    ss;
       struct sockaddr_un      sa;

       ss = calloc(1, sizeof(struct sdp_session));
       if (ss == NULL)
               goto fail;

       ss->s = socket(PF_LOCAL, SOCK_STREAM, 0);
       if (ss->s == -1)
               goto fail;

       if (control == NULL)
               control = SDP_LOCAL_PATH;

       memset(&sa, 0, sizeof(sa));
       sa.sun_len = sizeof(sa);
       sa.sun_family = AF_LOCAL;
       strlcpy(sa.sun_path, control, sizeof(sa.sun_path));
       if (connect(ss->s, (struct sockaddr *)&sa, sizeof(sa)) == -1)
               goto fail;

       ss->imtu = L2CAP_MTU_DEFAULT;

       ss->ibuf = malloc((size_t)(ss->imtu));
       if (ss->ibuf == NULL)
               goto fail;

       return ss;

fail:
       _sdp_close(ss);
       return NULL;
}

/*
* close session and release all resources
*/
void
_sdp_close(struct sdp_session *ss)
{

       if (ss == NULL)
               return;

       if (ss->s != -1)
               close(ss->s);

       if (ss->ibuf != NULL)
               free(ss->ibuf);

       if (ss->rbuf != NULL)
               free(ss->rbuf);

       free(ss);
}

/*
* internal function; send a PDU on session
*
* caller provides an iovec array with an empty slot at the beginning for
* PDU header, num is total iovec count.
*/
bool
_sdp_send_pdu(struct sdp_session *ss, uint8_t pid, struct iovec *iov, int num)
{
       sdp_pdu_t       pdu;
       ssize_t         len, nw;
       int             i;

       for (len = 0, i = 1; i < num; i++)
               len += iov[i].iov_len;

       if (len > UINT16_MAX) {
               errno = EMSGSIZE;
               return false;
       }

       ss->tid += 1;

       pdu.pid = pid;
       pdu.tid = htobe16(ss->tid);
       pdu.len = htobe16(len);

       iov[0].iov_base = &pdu;
       iov[0].iov_len = sizeof(pdu);

       do {
               nw = writev(ss->s, iov, num);
       } while (nw == -1 && errno == EINTR);

       if ((size_t)nw != sizeof(pdu) + len) {
               errno = EIO;
               return false;
       }

       return true;
}

/*
* internal function; receive a PDU on session
*
* validate the PDU and transaction IDs and data length, stores
* received data in the session incoming buffer.
*/
ssize_t
_sdp_recv_pdu(struct sdp_session *ss, uint8_t pid)
{
       struct iovec    iov[2];
       sdp_pdu_t       pdu;
       ssize_t         nr;

       iov[0].iov_base = &pdu;
       iov[0].iov_len = sizeof(pdu);

       iov[1].iov_base = ss->ibuf;
       iov[1].iov_len = ss->imtu;

       do {
               nr = readv(ss->s, iov, __arraycount(iov));
       } while (nr == -1 && errno == EINTR);

       if (nr == -1)
               return -1;

       if ((size_t)nr < sizeof(pdu)) {
               errno = EIO;
               return -1;
       }

       pdu.tid = be16toh(pdu.tid);
       pdu.len = be16toh(pdu.len);

       if (pid != pdu.pid
           || ss->tid != pdu.tid
           || (size_t)nr != sizeof(pdu) + pdu.len) {
               if (pdu.pid == SDP_PDU_ERROR_RESPONSE
                   && pdu.len == sizeof(uint16_t))
                       errno = _sdp_errno(be16dec(ss->ibuf));
               else
                       errno = EIO;

               return -1;
       }

       return pdu.len;
}

/*
* translate ErrorCode to errno
*/
int
_sdp_errno(uint16_t ec)
{

       switch (ec) {
       case SDP_ERROR_CODE_INVALID_SERVICE_RECORD_HANDLE:
               return ENOATTR;

       case SDP_ERROR_CODE_INVALID_SDP_VERSION:
       case SDP_ERROR_CODE_INVALID_REQUEST_SYNTAX:
       case SDP_ERROR_CODE_INVALID_PDU_SIZE:
       case SDP_ERROR_CODE_INVALID_CONTINUATION_STATE:
       case SDP_ERROR_CODE_INSUFFICIENT_RESOURCES:
       default:
               return EIO;
       }
}