1/*
2 * rfcomm_sppd.c
3 */
4
5/*-
6 * Copyright (c) 2003 Maksim Yevmenkin <m_evmenkin@yahoo.com>
7 * All rights reserved.
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions
11 * are met:
12 * 1. Redistributions of source code must retain the above copyright
13 *    notice, this list of conditions and the following disclaimer.
14 * 2. Redistributions in binary form must reproduce the above copyright
15 *    notice, this list of conditions and the following disclaimer in the
16 *    documentation and/or other materials provided with the distribution.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
19 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
22 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
23 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
24 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
25 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
26 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
27 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
28 * SUCH DAMAGE.
29 *
30 * $Id: rfcomm_sppd.c,v 1.4 2003/09/07 18:15:55 max Exp $
31 * $FreeBSD$
32 */
33
34#include <sys/stat.h>
35#include <bluetooth.h>
36#include <ctype.h>
37#include <err.h>
38#include <errno.h>
39#include <fcntl.h>
40#include <grp.h>
41#include <limits.h>
42#include <paths.h>
43#include <sdp.h>
44#include <signal.h>
45#include <stdarg.h>
46#include <stdio.h>
47#include <stdlib.h>
48#include <string.h>
49#include <syslog.h>
50#include <termios.h>
51#include <unistd.h>
52
53#define SPPD_IDENT		"rfcomm_sppd"
54#define SPPD_BUFFER_SIZE	1024
55#define max(a, b)		(((a) > (b))? (a) : (b))
56
57int		rfcomm_channel_lookup	(bdaddr_t const *local,
58					 bdaddr_t const *remote,
59					 int service, int *channel, int *error);
60
61static int	sppd_ttys_open	(char const *tty, int *amaster, int *aslave);
62static int	sppd_read	(int fd, char *buffer, int size);
63static int	sppd_write	(int fd, char *buffer, int size);
64static void	sppd_sighandler	(int s);
65static void	usage		(void);
66
67static int	done;	/* are we done? */
68
69/* Main */
70int
71main(int argc, char *argv[])
72{
73	struct sigaction	 sa;
74	struct sockaddr_rfcomm	 ra;
75	bdaddr_t		 addr;
76	int			 n, background, channel, service,
77				 s, amaster, aslave, fd, doserver;
78	fd_set			 rfd;
79	char			*tty = NULL, *ep = NULL, buf[SPPD_BUFFER_SIZE];
80
81	memcpy(&addr, NG_HCI_BDADDR_ANY, sizeof(addr));
82	background = channel = 0;
83	service = SDP_SERVICE_CLASS_SERIAL_PORT;
84	doserver = 0;
85
86	/* Parse command line options */
87	while ((n = getopt(argc, argv, "a:bc:t:hS")) != -1) {
88		switch (n) {
89		case 'a': /* BDADDR */
90			if (!bt_aton(optarg, &addr)) {
91				struct hostent	*he = NULL;
92
93				if ((he = bt_gethostbyname(optarg)) == NULL)
94					errx(1, "%s: %s", optarg, hstrerror(h_errno));
95
96				memcpy(&addr, he->h_addr, sizeof(addr));
97			}
98			break;
99
100		case 'c': /* RFCOMM channel */
101			channel = strtoul(optarg, &ep, 10);
102			if (*ep != '\0') {
103				channel = 0;
104				switch (tolower(optarg[0])) {
105				case 'd': /* DialUp Networking */
106					service = SDP_SERVICE_CLASS_DIALUP_NETWORKING;
107					break;
108
109				case 'f': /* Fax */
110					service = SDP_SERVICE_CLASS_FAX;
111					break;
112
113				case 'l': /* LAN */
114					service = SDP_SERVICE_CLASS_LAN_ACCESS_USING_PPP;
115					break;
116
117				case 's': /* Serial Port */
118					service = SDP_SERVICE_CLASS_SERIAL_PORT;
119					break;
120
121				default:
122					errx(1, "Unknown service name: %s",
123						optarg);
124					/* NOT REACHED */
125				}
126			}
127			break;
128
129		case 'b': /* Run in background */
130			background = 1;
131			break;
132
133		case 't': /* Slave TTY name */
134			if (optarg[0] != '/')
135				asprintf(&tty, "%s%s", _PATH_DEV, optarg);
136			else
137				tty = optarg;
138			break;
139
140		case 'S':
141			doserver = 1;
142			break;
143
144		case 'h':
145		default:
146			usage();
147			/* NOT REACHED */
148		}
149	}
150
151	/* Check if we have everything we need */
152	if (!doserver && memcmp(&addr, NG_HCI_BDADDR_ANY, sizeof(addr)) == 0)
153		usage();
154		/* NOT REACHED */
155
156	/* Set signal handlers */
157	memset(&sa, 0, sizeof(sa));
158	sa.sa_handler = sppd_sighandler;
159
160	if (sigaction(SIGTERM, &sa, NULL) < 0)
161		err(1, "Could not sigaction(SIGTERM)");
162
163	if (sigaction(SIGHUP, &sa, NULL) < 0)
164		err(1, "Could not sigaction(SIGHUP)");
165
166	if (sigaction(SIGINT, &sa, NULL) < 0)
167		err(1, "Could not sigaction(SIGINT)");
168
169	sa.sa_handler = SIG_IGN;
170	sa.sa_flags = SA_NOCLDWAIT;
171
172	if (sigaction(SIGCHLD, &sa, NULL) < 0)
173		err(1, "Could not sigaction(SIGCHLD)");
174
175	/* Open TTYs */
176	if (tty == NULL) {
177		if (background)
178			usage();
179
180		amaster = STDIN_FILENO;
181		fd = STDOUT_FILENO;
182	} else {
183		if (sppd_ttys_open(tty, &amaster, &aslave) < 0)
184			exit(1);
185
186		fd = amaster;
187	}
188
189	/* Open RFCOMM connection */
190
191	if (doserver) {
192		struct sockaddr_rfcomm	 ma;
193		bdaddr_t		 bt_addr_any;
194		sdp_sp_profile_t	 sp;
195		void			*ss;
196		uint32_t		 sdp_handle;
197		int			 acceptsock, aaddrlen;
198
199		acceptsock = socket(PF_BLUETOOTH, SOCK_STREAM,
200					BLUETOOTH_PROTO_RFCOMM);
201		if (acceptsock < 0)
202			err(1, "Could not create socket");
203
204		memcpy(&bt_addr_any, NG_HCI_BDADDR_ANY, sizeof(bt_addr_any));
205
206		memset(&ma, 0, sizeof(ma));
207		ma.rfcomm_len = sizeof(ma);
208		ma.rfcomm_family = AF_BLUETOOTH;
209		memcpy(&ma.rfcomm_bdaddr, &bt_addr_any, sizeof(bt_addr_any));
210		ma.rfcomm_channel = channel;
211
212		if (bind(acceptsock, (struct sockaddr *)&ma, sizeof(ma)) < 0)
213			err(1, "Could not bind socket on channel %d", channel);
214		if (listen(acceptsock, 10) != 0)
215			err(1, "Could not listen on socket");
216
217		aaddrlen = sizeof(ma);
218		if (getsockname(acceptsock, (struct sockaddr *)&ma, &aaddrlen) < 0)
219			err(1, "Could not get socket name");
220		channel = ma.rfcomm_channel;
221
222		ss = sdp_open_local(NULL);
223		if (ss == NULL)
224			errx(1, "Unable to create local SDP session");
225		if (sdp_error(ss) != 0)
226			errx(1, "Unable to open local SDP session. %s (%d)",
227			    strerror(sdp_error(ss)), sdp_error(ss));
228		memset(&sp, 0, sizeof(sp));
229		sp.server_channel = channel;
230
231		if (sdp_register_service(ss, SDP_SERVICE_CLASS_SERIAL_PORT,
232				&bt_addr_any, (void *)&sp, sizeof(sp),
233				&sdp_handle) != 0) {
234			errx(1, "Unable to register LAN service with "
235			    "local SDP daemon. %s (%d)",
236			    strerror(sdp_error(ss)), sdp_error(ss));
237		}
238
239		s = -1;
240		while (s < 0) {
241			aaddrlen = sizeof(ra);
242			s = accept(acceptsock, (struct sockaddr *)&ra,
243			    &aaddrlen);
244			if (s < 0)
245				err(1, "Unable to accept()");
246			if (memcmp(&addr, NG_HCI_BDADDR_ANY, sizeof(addr)) &&
247			    memcmp(&addr, &ra.rfcomm_bdaddr, sizeof(addr))) {
248				warnx("Connect from wrong client");
249				close(s);
250				s = -1;
251			}
252		}
253		sdp_unregister_service(ss, sdp_handle);
254		sdp_close(ss);
255		close(acceptsock);
256	} else {
257		/* Check channel, if was not set then obtain it via SDP */
258		if (channel == 0 && service != 0)
259			if (rfcomm_channel_lookup(NULL, &addr,
260				    service, &channel, &n) != 0)
261				errc(1, n, "Could not obtain RFCOMM channel");
262		if (channel <= 0 || channel > 30)
263			errx(1, "Invalid RFCOMM channel number %d", channel);
264
265		s = socket(PF_BLUETOOTH, SOCK_STREAM, BLUETOOTH_PROTO_RFCOMM);
266		if (s < 0)
267			err(1, "Could not create socket");
268
269		memset(&ra, 0, sizeof(ra));
270		ra.rfcomm_len = sizeof(ra);
271		ra.rfcomm_family = AF_BLUETOOTH;
272
273		if (bind(s, (struct sockaddr *) &ra, sizeof(ra)) < 0)
274			err(1, "Could not bind socket");
275
276		memcpy(&ra.rfcomm_bdaddr, &addr, sizeof(ra.rfcomm_bdaddr));
277		ra.rfcomm_channel = channel;
278
279		if (connect(s, (struct sockaddr *) &ra, sizeof(ra)) < 0)
280			err(1, "Could not connect socket");
281	}
282
283	/* Became daemon if required */
284	if (background && daemon(0, 0) < 0)
285		err(1, "Could not daemon()");
286
287	openlog(SPPD_IDENT, LOG_NDELAY|LOG_PERROR|LOG_PID, LOG_DAEMON);
288	syslog(LOG_INFO, "Starting on %s...", (tty != NULL)? tty : "stdin/stdout");
289
290	for (done = 0; !done; ) {
291		FD_ZERO(&rfd);
292		FD_SET(amaster, &rfd);
293		FD_SET(s, &rfd);
294
295		n = select(max(amaster, s) + 1, &rfd, NULL, NULL, NULL);
296		if (n < 0) {
297			if (errno == EINTR)
298				continue;
299
300			syslog(LOG_ERR, "Could not select(). %s",
301					strerror(errno));
302			exit(1);
303		}
304
305		if (n == 0)
306			continue;
307
308		if (FD_ISSET(amaster, &rfd)) {
309			n = sppd_read(amaster, buf, sizeof(buf));
310			if (n < 0) {
311				syslog(LOG_ERR, "Could not read master pty, " \
312					"fd=%d. %s", amaster, strerror(errno));
313				exit(1);
314			}
315
316			if (n == 0)
317				break; /* XXX */
318
319			if (sppd_write(s, buf, n) < 0) {
320				syslog(LOG_ERR, "Could not write to socket, " \
321					"fd=%d, size=%d. %s",
322					s, n, strerror(errno));
323				exit(1);
324			}
325		}
326
327		if (FD_ISSET(s, &rfd)) {
328			n = sppd_read(s, buf, sizeof(buf));
329			if (n < 0) {
330				syslog(LOG_ERR, "Could not read socket, " \
331					"fd=%d. %s", s, strerror(errno));
332				exit(1);
333			}
334
335			if (n == 0)
336				break;
337
338			if (sppd_write(fd, buf, n) < 0) {
339				syslog(LOG_ERR, "Could not write to master " \
340					"pty, fd=%d, size=%d. %s",
341					fd, n, strerror(errno));
342				exit(1);
343			}
344		}
345	}
346
347	syslog(LOG_INFO, "Completed on %s", (tty != NULL)? tty : "stdin/stdout");
348	closelog();
349
350	close(s);
351
352	if (tty != NULL) {
353		close(aslave);
354		close(amaster);
355	}
356
357	return (0);
358}
359
360/* Open TTYs */
361static int
362sppd_ttys_open(char const *tty, int *amaster, int *aslave)
363{
364	char		 pty[PATH_MAX], *slash;
365	struct group	*gr = NULL;
366	gid_t		 ttygid;
367	struct termios	 tio;
368
369	/*
370	 * Construct master PTY name. The slave tty name must be less then
371	 * PATH_MAX characters in length, must contain '/' character and
372	 * must not end with '/'.
373	 */
374
375	if (strlen(tty) >= sizeof(pty)) {
376		syslog(LOG_ERR, "Slave tty name is too long");
377		return (-1);
378	}
379
380	strlcpy(pty, tty, sizeof(pty));
381	slash = strrchr(pty, '/');
382	if (slash == NULL || slash[1] == '\0') {
383		syslog(LOG_ERR, "Invalid slave tty name (%s)", tty);
384		return (-1);
385	}
386
387	slash[1] = 'p';
388
389	if (strcmp(pty, tty) == 0) {
390		syslog(LOG_ERR, "Master and slave tty are the same (%s)", tty);
391		return (-1);
392	}
393
394	if ((*amaster = open(pty, O_RDWR, 0)) < 0) {
395		syslog(LOG_ERR, "Could not open(%s). %s", pty, strerror(errno));
396		return (-1);
397	}
398
399	/*
400	 * Slave TTY
401	 */
402
403	if ((gr = getgrnam("tty")) != NULL)
404		ttygid = gr->gr_gid;
405	else
406		ttygid = -1;
407
408	(void) chown(tty, getuid(), ttygid);
409	(void) chmod(tty, S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP);
410	(void) revoke(tty);
411
412	if ((*aslave = open(tty, O_RDWR, 0)) < 0) {
413		syslog(LOG_ERR, "Could not open(%s). %s", tty, strerror(errno));
414		close(*amaster);
415		return (-1);
416	}
417
418	/*
419	 * Make slave TTY raw
420	 */
421
422	cfmakeraw(&tio);
423
424	if (tcsetattr(*aslave, TCSANOW, &tio) < 0) {
425		syslog(LOG_ERR, "Could not tcsetattr(). %s", strerror(errno));
426		close(*aslave);
427		close(*amaster);
428		return (-1);
429	}
430
431	return (0);
432} /* sppd_ttys_open */
433
434/* Read data */
435static int
436sppd_read(int fd, char *buffer, int size)
437{
438	int	n;
439
440again:
441	n = read(fd, buffer, size);
442	if (n < 0) {
443		if (errno == EINTR)
444			goto again;
445
446		return (-1);
447	}
448
449	return (n);
450} /* sppd_read */
451
452/* Write data */
453static int
454sppd_write(int fd, char *buffer, int size)
455{
456	int	n, wrote;
457
458	for (wrote = 0; size > 0; ) {
459		n = write(fd, buffer, size);
460		switch (n) {
461		case -1:
462			if (errno != EINTR)
463				return (-1);
464			break;
465
466		case 0:
467			/* XXX can happen? */
468			break;
469
470		default:
471			wrote += n;
472			buffer += n;
473			size -= n;
474			break;
475		}
476	}
477
478	return (wrote);
479} /* sppd_write */
480
481/* Signal handler */
482static void
483sppd_sighandler(int s)
484{
485	syslog(LOG_INFO, "Signal %d received. Total %d signals received\n",
486			s, ++ done);
487} /* sppd_sighandler */
488
489/* Display usage and exit */
490static void
491usage(void)
492{
493	fprintf(stdout,
494"Usage: %s options\n" \
495"Where options are:\n" \
496"\t-a address Peer address (required in client mode)\n" \
497"\t-b         Run in background\n" \
498"\t-c channel RFCOMM channel to connect to or listen on\n" \
499"\t-t tty     TTY name (required in background mode)\n" \
500"\t-S         Server mode\n" \
501"\t-h         Display this message\n", SPPD_IDENT);
502	exit(255);
503} /* usage */
504
505