Home | History | Annotate | Line # | Download | only in btpand
server.c revision 1.3
      1 /*	$NetBSD: server.c,v 1.3 2009/05/12 21:08:30 plunky Exp $	*/
      2 
      3 /*-
      4  * Copyright (c) 2008-2009 Iain Hibbert
      5  * All rights reserved.
      6  *
      7  * Redistribution and use in source and binary forms, with or without
      8  * modification, are permitted provided that the following conditions
      9  * are met:
     10  * 1. Redistributions of source code must retain the above copyright
     11  *    notice, this list of conditions and the following disclaimer.
     12  * 2. Redistributions in binary form must reproduce the above copyright
     13  *    notice, this list of conditions and the following disclaimer in the
     14  *    documentation and/or other materials provided with the distribution.
     15  *
     16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
     17  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
     18  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
     19  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
     20  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
     21  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     22  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
     23  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
     25  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     26  */
     27 
     28 #include <sys/cdefs.h>
     29 __RCSID("$NetBSD: server.c,v 1.3 2009/05/12 21:08:30 plunky Exp $");
     30 
     31 #include <sys/ioctl.h>
     32 
     33 #include <bluetooth.h>
     34 #include <errno.h>
     35 #include <sdp.h>
     36 #include <unistd.h>
     37 
     38 #include "btpand.h"
     39 #include "bnep.h"
     40 
     41 static struct event	server_ev;
     42 static int		server_count;
     43 
     44 static void *		server_ss;
     45 static uint32_t		server_handle;
     46 
     47 static void server_open(void);
     48 static void server_read(int, short, void *);
     49 static void server_down(channel_t *);
     50 static void server_update(void);
     51 
     52 void
     53 server_init(void)
     54 {
     55 
     56 	if (server_limit == 0)
     57 		return;
     58 
     59 	server_open();
     60 	server_update();
     61 }
     62 
     63 /*
     64  * Start listening on server socket
     65  */
     66 static void
     67 server_open(void)
     68 {
     69 	struct sockaddr_bt sa;
     70 	uint16_t mru;
     71 	int fd;
     72 
     73 	fd = socket(PF_BLUETOOTH, SOCK_SEQPACKET, BTPROTO_L2CAP);
     74 	if (fd == -1) {
     75 		log_err("Could not open L2CAP socket: %m");
     76 		exit(EXIT_FAILURE);
     77 	}
     78 
     79 	memset(&sa, 0, sizeof(sa));
     80 	sa.bt_family = AF_BLUETOOTH;
     81 	sa.bt_len = sizeof(sa);
     82 	sa.bt_psm = l2cap_psm;
     83 	bdaddr_copy(&sa.bt_bdaddr, &local_bdaddr);
     84 	if (bind(fd, (struct sockaddr *)&sa, sizeof(sa)) == -1) {
     85 		log_err("Could not bind server socket: %m");
     86 		exit(EXIT_FAILURE);
     87 	}
     88 
     89 	if (setsockopt(fd, BTPROTO_L2CAP,
     90 	    SO_L2CAP_LM, &l2cap_mode, sizeof(l2cap_mode)) == -1) {
     91 		log_err("Could not set link mode (0x%4.4x): %m", l2cap_mode);
     92 		exit(EXIT_FAILURE);
     93 	}
     94 
     95 	mru = BNEP_MTU_MIN;
     96 	if (setsockopt(fd, BTPROTO_L2CAP,
     97 	    SO_L2CAP_IMTU, &mru, sizeof(mru)) == -1) {
     98 		log_err("Could not set L2CAP IMTU (%d): %m", mru);
     99 		exit(EXIT_FAILURE);
    100 	}
    101 
    102 	if (listen(fd, 0) == -1) {
    103 		log_err("Could not listen on server socket: %m");
    104 		exit(EXIT_FAILURE);
    105 	}
    106 
    107 	event_set(&server_ev, fd, EV_READ | EV_PERSIST, server_read, NULL);
    108 	if (event_add(&server_ev, NULL) == -1) {
    109 		log_err("Could not add server event: %m");
    110 		exit(EXIT_FAILURE);
    111 	}
    112 
    113 	log_info("server socket open");
    114 }
    115 
    116 /*
    117  * handle connection request
    118  */
    119 static void
    120 server_read(int s, short ev, void *arg)
    121 {
    122 	struct sockaddr_bt ra, la;
    123 	channel_t *chan;
    124 	socklen_t len;
    125 	int fd, n;
    126 	uint16_t mru, mtu;
    127 
    128 	assert(server_count < server_limit);
    129 
    130 	len = sizeof(ra);
    131 	fd = accept(s, (struct sockaddr *)&ra, &len);
    132 	if (fd == -1)
    133 		return;
    134 
    135 	n = 1;
    136 	if (ioctl(fd, FIONBIO, &n) == -1) {
    137 		log_err("Could not set NonBlocking IO: %m");
    138 		close(fd);
    139 		return;
    140 	}
    141 
    142 	len = sizeof(mru);
    143 	if (getsockopt(fd, BTPROTO_L2CAP, SO_L2CAP_IMTU, &mru, &len) == -1) {
    144 		log_err("Could not get L2CAP IMTU: %m");
    145 		close(fd);
    146 		return;
    147 	}
    148 	if(mru < BNEP_MTU_MIN) {
    149 		log_err("L2CAP IMTU too small (%d)", mru);
    150 		close(fd);
    151 		return;
    152 	}
    153 
    154 	len = sizeof(mtu);
    155 	if (getsockopt(fd, BTPROTO_L2CAP, SO_L2CAP_OMTU, &mtu, &len) == -1) {
    156 		log_err("Could not get L2CAP OMTU: %m");
    157 		close(fd);
    158 		return;
    159 	}
    160 	if (mtu < BNEP_MTU_MIN) {
    161 		log_err("L2CAP OMTU too small (%d)", mtu);
    162 		close(fd);
    163 		return;
    164 	}
    165 
    166 	len = sizeof(n);
    167 	if (getsockopt(fd, SOL_SOCKET, SO_SNDBUF, &n, &len) == -1) {
    168 		log_err("Could not get socket send buffer size: %m");
    169 		close(fd);
    170 		return;
    171 	}
    172 
    173 	if (n < (mtu * 2)) {
    174 		n = mtu * 2;
    175 		if (setsockopt(fd, SOL_SOCKET, SO_SNDBUF, &n, sizeof(n)) == -1) {
    176 			log_err("Could not set socket send buffer size (%d): %m", n);
    177 			close(fd);
    178 			return;
    179 		}
    180 	}
    181 
    182 	n = mtu;
    183 	if (setsockopt(fd, SOL_SOCKET, SO_SNDLOWAT, &n, sizeof(n)) == -1) {
    184 		log_err("Could not set socket low water mark (%d): %m", n);
    185 		close(fd);
    186 		return;
    187 	}
    188 
    189 	len = sizeof(la);
    190 	if (getsockname(fd, (struct sockaddr *)&la, &len) == -1) {
    191 		log_err("Could not get socket address: %m");
    192 		close(fd);
    193 		return;
    194 	}
    195 
    196 	log_info("Accepted connection from %s", bt_ntoa(&ra.bt_bdaddr, NULL));
    197 
    198 	chan = channel_alloc();
    199 	if (chan == NULL) {
    200 		close(fd);
    201 		return;
    202 	}
    203 
    204 	chan->send = bnep_send;
    205 	chan->recv = bnep_recv;
    206 	chan->down = server_down;
    207 	chan->mru = mru;
    208 	chan->mtu = mtu;
    209 	b2eaddr(chan->raddr, &ra.bt_bdaddr);
    210 	b2eaddr(chan->laddr, &la.bt_bdaddr);
    211 	chan->state = CHANNEL_WAIT_CONNECT_REQ;
    212 	channel_timeout(chan, 10);
    213 	if (!channel_open(chan, fd)) {
    214 		chan->state = CHANNEL_CLOSED;
    215 		channel_free(chan);
    216 		close(fd);
    217 		return;
    218 	}
    219 
    220 	if (++server_count == server_limit) {
    221 		log_info("Server limit reached, closing server socket");
    222 		event_del(&server_ev);
    223 		close(s);
    224 	}
    225 
    226 	server_update();
    227 }
    228 
    229 /*
    230  * Shut down a server channel, we need to update the service record and
    231  * may want to restart accepting connections on the server socket
    232  */
    233 static void
    234 server_down(channel_t *chan)
    235 {
    236 
    237 	assert(server_count > 0);
    238 
    239 	channel_close(chan);
    240 
    241 	if (server_count-- == server_limit)
    242 		server_open();
    243 
    244 	server_update();
    245 }
    246 
    247 static void
    248 server_update(void)
    249 {
    250 	sdp_nap_profile_t p;
    251 	int rv;
    252 
    253 	if (service_name == NULL)
    254 		return;
    255 
    256 	if (server_ss == NULL) {
    257 		server_ss = sdp_open_local(control_path);
    258 		if (server_ss == NULL || sdp_error(server_ss) != 0) {
    259 			log_err("failed to contact SDP server");
    260 			return;
    261 		}
    262 	}
    263 
    264 	memset(&p, 0, sizeof(p));
    265 	p.psm = l2cap_psm;
    266 	p.load_factor = (UINT8_MAX - server_count * UINT8_MAX / server_limit);
    267 	p.security_description = (l2cap_mode == 0 ? 0x0000 : 0x0001);
    268 
    269 	if (server_handle)
    270 		rv = sdp_change_service(server_ss, server_handle,
    271 		    (uint8_t *)&p, sizeof(p));
    272 	else
    273 		rv = sdp_register_service(server_ss, service_class,
    274 		    &local_bdaddr, (uint8_t *)&p, sizeof(p), &server_handle);
    275 
    276 	if (rv != 0) {
    277 		errno = sdp_error(server_ss);
    278 		log_err("%s: %m", service_name);
    279 		exit(EXIT_FAILURE);
    280 	}
    281 }
    282