xref: /freebsd/usr.bin/bluetooth/rfcomm_sppd/rfcomm_sppd.c (revision 6b3455a7665208c366849f0b2b3bc916fb97516e)
1 /*
2  * rfcomm_sppd.c
3  *
4  * Copyright (c) 2003 Maksim Yevmenkin <m_evmenkin@yahoo.com>
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 AND CONTRIBUTORS ``AS IS'' AND
17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19  * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26  * SUCH DAMAGE.
27  *
28  * $Id: rfcomm_sppd.c,v 1.4 2003/09/07 18:15:55 max Exp $
29  * $FreeBSD$
30  */
31 
32 #include <sys/stat.h>
33 #include <bluetooth.h>
34 #include <err.h>
35 #include <errno.h>
36 #include <fcntl.h>
37 #include <grp.h>
38 #include <limits.h>
39 #include <paths.h>
40 #include <sdp.h>
41 #include <signal.h>
42 #include <stdarg.h>
43 #include <stdio.h>
44 #include <stdlib.h>
45 #include <string.h>
46 #include <syslog.h>
47 #include <termios.h>
48 #include <unistd.h>
49 
50 #define SPPD_IDENT		"rfcomm_sppd"
51 #define SPPD_BUFFER_SIZE	1024
52 #define max(a, b)		(((a) > (b))? (a) : (b))
53 
54 int		rfcomm_channel_lookup	(bdaddr_t const *local,
55 					 bdaddr_t const *remote,
56 					 int service, int *channel, int *error);
57 
58 static int	sppd_ttys_open	(char const *tty, int *amaster, int *aslave);
59 static int	sppd_read	(int fd, char *buffer, int size);
60 static int	sppd_write	(int fd, char *buffer, int size);
61 static void	sppd_sighandler	(int s);
62 static void	usage		(void);
63 
64 static int	done;	/* are we done? */
65 
66 /* Main */
67 int
68 main(int argc, char *argv[])
69 {
70 	struct sigaction	 sa;
71 	struct sockaddr_rfcomm	 ra;
72 	bdaddr_t		 addr;
73 	int			 n, background, channel, s, amaster, aslave;
74 	fd_set			 rfd;
75 	char			*tty = NULL, buf[SPPD_BUFFER_SIZE];
76 
77 	memcpy(&addr, NG_HCI_BDADDR_ANY, sizeof(addr));
78 	background = channel = 0;
79 
80 	/* Parse command line options */
81 	while ((n = getopt(argc, argv, "a:bc:t:h")) != -1) {
82 		switch (n) {
83 		case 'a': /* BDADDR */
84 			if (!bt_aton(optarg, &addr)) {
85 				struct hostent	*he = NULL;
86 
87 				if ((he = bt_gethostbyname(optarg)) == NULL)
88 					errx(1, "%s: %s", optarg, hstrerror(h_errno));
89 
90 				memcpy(&addr, he->h_addr, sizeof(addr));
91 			}
92 			break;
93 
94 		case 'c': /* RFCOMM channel */
95 			channel = atoi(optarg);
96 			break;
97 
98 		case 'b': /* Run in background */
99 			background = 1;
100 			break;
101 
102 		case 't': /* Slave TTY name */
103 			if (optarg[0] != '/')
104 				asprintf(&tty, "%s%s", _PATH_DEV, optarg);
105 			else
106 				tty = optarg;
107 			break;
108 
109 		case 'h':
110 		default:
111 			usage();
112 			/* NOT REACHED */
113 		}
114 	}
115 
116 	/* Check if we have everything we need */
117 	if (tty == NULL || memcmp(&addr, NG_HCI_BDADDR_ANY, sizeof(addr)) == 0)
118 		usage();
119 		/* NOT REACHED */
120 
121 	/* Set signal handlers */
122 	memset(&sa, 0, sizeof(sa));
123 	sa.sa_handler = sppd_sighandler;
124 
125 	if (sigaction(SIGTERM, &sa, NULL) < 0)
126 		err(1, "Could not sigaction(SIGTERM)");
127 
128 	if (sigaction(SIGHUP, &sa, NULL) < 0)
129 		err(1, "Could not sigaction(SIGHUP)");
130 
131 	if (sigaction(SIGINT, &sa, NULL) < 0)
132 		err(1, "Could not sigaction(SIGINT)");
133 
134 	sa.sa_handler = SIG_IGN;
135 	sa.sa_flags = SA_NOCLDWAIT;
136 
137 	if (sigaction(SIGCHLD, &sa, NULL) < 0)
138 		err(1, "Could not sigaction(SIGCHLD)");
139 
140 	/* Check channel, if was not set then obtain it via SDP */
141 	if (channel == 0)
142 		if (rfcomm_channel_lookup(NULL, &addr,
143 			    SDP_SERVICE_CLASS_SERIAL_PORT, &channel, &n) != 0)
144 			errc(1, n, "Could not obtain RFCOMM channel");
145 	if (channel <= 0 || channel > 30)
146 		errx(1, "Invalid RFCOMM channel number %d", channel);
147 
148 	/* Open TTYs */
149 	if (sppd_ttys_open(tty, &amaster, &aslave) < 0)
150 		exit(1);
151 
152 	/* Open RFCOMM connection */
153 	memset(&ra, 0, sizeof(ra));
154 	ra.rfcomm_len = sizeof(ra);
155 	ra.rfcomm_family = AF_BLUETOOTH;
156 
157 	s = socket(PF_BLUETOOTH, SOCK_STREAM, BLUETOOTH_PROTO_RFCOMM);
158 	if (s < 0)
159 		err(1, "Could not create socket");
160 
161 	if (bind(s, (struct sockaddr *) &ra, sizeof(ra)) < 0)
162 		err(1, "Could not bind socket");
163 
164 	memcpy(&ra.rfcomm_bdaddr, &addr, sizeof(ra.rfcomm_bdaddr));
165 	ra.rfcomm_channel = channel;
166 
167 	if (connect(s, (struct sockaddr *) &ra, sizeof(ra)) < 0)
168 		err(1, "Could not connect socket");
169 
170 	/* Became daemon if required */
171 	if (background) {
172 		switch (fork()) {
173 		case -1:
174 			err(1, "Could not fork()");
175 			/* NOT REACHED */
176 
177 		case 0:
178 			exit(0);
179 			/* NOT REACHED */
180 
181 		default:
182 			if (daemon(0, 0) < 0)
183 				err(1, "Could not daemon()");
184 			break;
185 		}
186 	}
187 
188 	openlog(SPPD_IDENT, LOG_NDELAY|LOG_PERROR|LOG_PID, LOG_DAEMON);
189 	syslog(LOG_INFO, "Starting on %s...", tty);
190 
191 	for (done = 0; !done; ) {
192 		FD_ZERO(&rfd);
193 		FD_SET(amaster, &rfd);
194 		FD_SET(s, &rfd);
195 
196 		n = select(max(amaster, s) + 1, &rfd, NULL, NULL, NULL);
197 		if (n < 0) {
198 			if (errno == EINTR)
199 				continue;
200 
201 			syslog(LOG_ERR, "Could not select(). %s",
202 					strerror(errno));
203 			exit(1);
204 		}
205 
206 		if (n == 0)
207 			continue;
208 
209 		if (FD_ISSET(amaster, &rfd)) {
210 			n = sppd_read(amaster, buf, sizeof(buf));
211 			if (n < 0) {
212 				syslog(LOG_ERR, "Could not read master pty, " \
213 					"fd=%d. %s", amaster, strerror(errno));
214 				exit(1);
215 			}
216 
217 			if (n == 0)
218 				break; /* XXX */
219 
220 			if (sppd_write(s, buf, n) < 0) {
221 				syslog(LOG_ERR, "Could not write to socket, " \
222 					"fd=%d, size=%d. %s",
223 					s, n, strerror(errno));
224 				exit(1);
225 			}
226 		}
227 
228 		if (FD_ISSET(s, &rfd)) {
229 			n = sppd_read(s, buf, sizeof(buf));
230 			if (n < 0) {
231 				syslog(LOG_ERR, "Could not read socket, " \
232 					"fd=%d. %s", s, strerror(errno));
233 				exit(1);
234 			}
235 
236 			if (n == 0)
237 				break;
238 
239 			if (sppd_write(amaster, buf, n) < 0) {
240 				syslog(LOG_ERR, "Could not write to master " \
241 					"pty, fd=%d, size=%d. %s",
242 					amaster, n, strerror(errno));
243 				exit(1);
244 			}
245 		}
246 	}
247 
248 	syslog(LOG_INFO, "Completed on %s", tty);
249 	closelog();
250 
251 	close(s);
252 	close(aslave);
253 	close(amaster);
254 
255 	return (0);
256 }
257 
258 /* Open TTYs */
259 static int
260 sppd_ttys_open(char const *tty, int *amaster, int *aslave)
261 {
262 	char		 pty[PATH_MAX], *slash;
263 	struct group	*gr = NULL;
264 	gid_t		 ttygid;
265 	struct termios	 tio;
266 
267 	/*
268 	 * Construct master PTY name. The slave tty name must be less then
269 	 * PATH_MAX characters in length, must contain '/' character and
270 	 * must not end with '/'.
271 	 */
272 
273 	if (strlen(tty) >= sizeof(pty)) {
274 		syslog(LOG_ERR, "Slave tty name is too long");
275 		return (-1);
276 	}
277 
278 	strlcpy(pty, tty, sizeof(pty));
279 	slash = strrchr(pty, '/');
280 	if (slash == NULL || slash[1] == '\0') {
281 		syslog(LOG_ERR, "Invalid slave tty name (%s)", tty);
282 		return (-1);
283 	}
284 
285 	slash[1] = 'p';
286 
287 	if (strcmp(pty, tty) == 0) {
288 		syslog(LOG_ERR, "Master and slave tty are the same (%s)", tty);
289 		return (-1);
290 	}
291 
292 	if ((*amaster = open(pty, O_RDWR, 0)) < 0) {
293 		syslog(LOG_ERR, "Could not open(%s). %s", pty, strerror(errno));
294 		return (-1);
295 	}
296 
297 	/*
298 	 * Slave TTY
299 	 */
300 
301 	if ((gr = getgrnam("tty")) != NULL)
302 		ttygid = gr->gr_gid;
303 	else
304 		ttygid = -1;
305 
306 	(void) chown(tty, getuid(), ttygid);
307 	(void) chmod(tty, S_IRUSR|S_IWUSR|S_IWGRP);
308 	(void) revoke(tty);
309 
310 	if ((*aslave = open(tty, O_RDWR, 0)) < 0) {
311 		syslog(LOG_ERR, "Could not open(%s). %s", tty, strerror(errno));
312 		close(*amaster);
313 		return (-1);
314 	}
315 
316 	/*
317 	 * Make slave TTY raw
318 	 */
319 
320 	cfmakeraw(&tio);
321 
322 	if (tcsetattr(*aslave, TCSANOW, &tio) < 0) {
323 		syslog(LOG_ERR, "Could not tcsetattr(). %s", strerror(errno));
324 		close(*aslave);
325 		close(*amaster);
326 		return (-1);
327 	}
328 
329 	return (0);
330 } /* sppd_ttys_open */
331 
332 /* Read data */
333 static int
334 sppd_read(int fd, char *buffer, int size)
335 {
336 	int	n;
337 
338 again:
339 	n = read(fd, buffer, size);
340 	if (n < 0) {
341 		if (errno == EINTR)
342 			goto again;
343 
344 		return (-1);
345 	}
346 
347 	return (n);
348 } /* sppd_read */
349 
350 /* Write data */
351 static int
352 sppd_write(int fd, char *buffer, int size)
353 {
354 	int	n, wrote;
355 
356 	for (wrote = 0; size > 0; ) {
357 		n = write(fd, buffer, size);
358 		switch (n) {
359 		case -1:
360 			if (errno != EINTR)
361 				return (-1);
362 			break;
363 
364 		case 0:
365 			/* XXX can happen? */
366 			break;
367 
368 		default:
369 			wrote += n;
370 			buffer += n;
371 			size -= n;
372 			break;
373 		}
374 	}
375 
376 	return (wrote);
377 } /* sppd_write */
378 
379 /* Signal handler */
380 static void
381 sppd_sighandler(int s)
382 {
383 	syslog(LOG_INFO, "Signal %d received. Total %d signals received\n",
384 			s, ++ done);
385 } /* sppd_sighandler */
386 
387 /* Display usage and exit */
388 static void
389 usage(void)
390 {
391 	fprintf(stdout,
392 "Usage: %s options\n" \
393 "Where options are:\n" \
394 "\t-a bdaddr  BDADDR to connect to (required)\n" \
395 "\t-b         Run in background\n" \
396 "\t-c channel RFCOMM channel to connect to\n" \
397 "\t-t tty     TTY name\n" \
398 "\t-h         Display this message\n", SPPD_IDENT);
399 
400 	exit(255);
401 } /* usage */
402 
403