xref: /freebsd/usr.bin/bluetooth/rfcomm_sppd/rfcomm_sppd.c (revision 6af83ee0d2941d18880b6aaa2b4facd1d30c6106)
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, fd;
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 (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 (tty == NULL) {
150 		if (background)
151 			usage();
152 
153 		amaster = STDIN_FILENO;
154 		fd = STDOUT_FILENO;
155 	} else {
156 		if (sppd_ttys_open(tty, &amaster, &aslave) < 0)
157 			exit(1);
158 
159 		fd = amaster;
160 	}
161 
162 
163 	/* Open RFCOMM connection */
164 	memset(&ra, 0, sizeof(ra));
165 	ra.rfcomm_len = sizeof(ra);
166 	ra.rfcomm_family = AF_BLUETOOTH;
167 
168 	s = socket(PF_BLUETOOTH, SOCK_STREAM, BLUETOOTH_PROTO_RFCOMM);
169 	if (s < 0)
170 		err(1, "Could not create socket");
171 
172 	if (bind(s, (struct sockaddr *) &ra, sizeof(ra)) < 0)
173 		err(1, "Could not bind socket");
174 
175 	memcpy(&ra.rfcomm_bdaddr, &addr, sizeof(ra.rfcomm_bdaddr));
176 	ra.rfcomm_channel = channel;
177 
178 	if (connect(s, (struct sockaddr *) &ra, sizeof(ra)) < 0)
179 		err(1, "Could not connect socket");
180 
181 	/* Became daemon if required */
182 	if (background) {
183 		switch (fork()) {
184 		case -1:
185 			err(1, "Could not fork()");
186 			/* NOT REACHED */
187 
188 		case 0:
189 			exit(0);
190 			/* NOT REACHED */
191 
192 		default:
193 			if (daemon(0, 0) < 0)
194 				err(1, "Could not daemon()");
195 			break;
196 		}
197 	}
198 
199 	openlog(SPPD_IDENT, LOG_NDELAY|LOG_PERROR|LOG_PID, LOG_DAEMON);
200 	syslog(LOG_INFO, "Starting on %s...", (tty != NULL)? tty : "stdin/stdout");
201 
202 	for (done = 0; !done; ) {
203 		FD_ZERO(&rfd);
204 		FD_SET(amaster, &rfd);
205 		FD_SET(s, &rfd);
206 
207 		n = select(max(amaster, s) + 1, &rfd, NULL, NULL, NULL);
208 		if (n < 0) {
209 			if (errno == EINTR)
210 				continue;
211 
212 			syslog(LOG_ERR, "Could not select(). %s",
213 					strerror(errno));
214 			exit(1);
215 		}
216 
217 		if (n == 0)
218 			continue;
219 
220 		if (FD_ISSET(amaster, &rfd)) {
221 			n = sppd_read(amaster, buf, sizeof(buf));
222 			if (n < 0) {
223 				syslog(LOG_ERR, "Could not read master pty, " \
224 					"fd=%d. %s", amaster, strerror(errno));
225 				exit(1);
226 			}
227 
228 			if (n == 0)
229 				break; /* XXX */
230 
231 			if (sppd_write(s, buf, n) < 0) {
232 				syslog(LOG_ERR, "Could not write to socket, " \
233 					"fd=%d, size=%d. %s",
234 					s, n, strerror(errno));
235 				exit(1);
236 			}
237 		}
238 
239 		if (FD_ISSET(s, &rfd)) {
240 			n = sppd_read(s, buf, sizeof(buf));
241 			if (n < 0) {
242 				syslog(LOG_ERR, "Could not read socket, " \
243 					"fd=%d. %s", s, strerror(errno));
244 				exit(1);
245 			}
246 
247 			if (n == 0)
248 				break;
249 
250 			if (sppd_write(fd, buf, n) < 0) {
251 				syslog(LOG_ERR, "Could not write to master " \
252 					"pty, fd=%d, size=%d. %s",
253 					fd, n, strerror(errno));
254 				exit(1);
255 			}
256 		}
257 	}
258 
259 	syslog(LOG_INFO, "Completed on %s", (tty != NULL)? tty : "stdin/stdout");
260 	closelog();
261 
262 	close(s);
263 
264 	if (tty != NULL) {
265 		close(aslave);
266 		close(amaster);
267 	}
268 
269 	return (0);
270 }
271 
272 /* Open TTYs */
273 static int
274 sppd_ttys_open(char const *tty, int *amaster, int *aslave)
275 {
276 	char		 pty[PATH_MAX], *slash;
277 	struct group	*gr = NULL;
278 	gid_t		 ttygid;
279 	struct termios	 tio;
280 
281 	/*
282 	 * Construct master PTY name. The slave tty name must be less then
283 	 * PATH_MAX characters in length, must contain '/' character and
284 	 * must not end with '/'.
285 	 */
286 
287 	if (strlen(tty) >= sizeof(pty)) {
288 		syslog(LOG_ERR, "Slave tty name is too long");
289 		return (-1);
290 	}
291 
292 	strlcpy(pty, tty, sizeof(pty));
293 	slash = strrchr(pty, '/');
294 	if (slash == NULL || slash[1] == '\0') {
295 		syslog(LOG_ERR, "Invalid slave tty name (%s)", tty);
296 		return (-1);
297 	}
298 
299 	slash[1] = 'p';
300 
301 	if (strcmp(pty, tty) == 0) {
302 		syslog(LOG_ERR, "Master and slave tty are the same (%s)", tty);
303 		return (-1);
304 	}
305 
306 	if ((*amaster = open(pty, O_RDWR, 0)) < 0) {
307 		syslog(LOG_ERR, "Could not open(%s). %s", pty, strerror(errno));
308 		return (-1);
309 	}
310 
311 	/*
312 	 * Slave TTY
313 	 */
314 
315 	if ((gr = getgrnam("tty")) != NULL)
316 		ttygid = gr->gr_gid;
317 	else
318 		ttygid = -1;
319 
320 	(void) chown(tty, getuid(), ttygid);
321 	(void) chmod(tty, S_IRUSR|S_IWUSR|S_IWGRP);
322 	(void) revoke(tty);
323 
324 	if ((*aslave = open(tty, O_RDWR, 0)) < 0) {
325 		syslog(LOG_ERR, "Could not open(%s). %s", tty, strerror(errno));
326 		close(*amaster);
327 		return (-1);
328 	}
329 
330 	/*
331 	 * Make slave TTY raw
332 	 */
333 
334 	cfmakeraw(&tio);
335 
336 	if (tcsetattr(*aslave, TCSANOW, &tio) < 0) {
337 		syslog(LOG_ERR, "Could not tcsetattr(). %s", strerror(errno));
338 		close(*aslave);
339 		close(*amaster);
340 		return (-1);
341 	}
342 
343 	return (0);
344 } /* sppd_ttys_open */
345 
346 /* Read data */
347 static int
348 sppd_read(int fd, char *buffer, int size)
349 {
350 	int	n;
351 
352 again:
353 	n = read(fd, buffer, size);
354 	if (n < 0) {
355 		if (errno == EINTR)
356 			goto again;
357 
358 		return (-1);
359 	}
360 
361 	return (n);
362 } /* sppd_read */
363 
364 /* Write data */
365 static int
366 sppd_write(int fd, char *buffer, int size)
367 {
368 	int	n, wrote;
369 
370 	for (wrote = 0; size > 0; ) {
371 		n = write(fd, buffer, size);
372 		switch (n) {
373 		case -1:
374 			if (errno != EINTR)
375 				return (-1);
376 			break;
377 
378 		case 0:
379 			/* XXX can happen? */
380 			break;
381 
382 		default:
383 			wrote += n;
384 			buffer += n;
385 			size -= n;
386 			break;
387 		}
388 	}
389 
390 	return (wrote);
391 } /* sppd_write */
392 
393 /* Signal handler */
394 static void
395 sppd_sighandler(int s)
396 {
397 	syslog(LOG_INFO, "Signal %d received. Total %d signals received\n",
398 			s, ++ done);
399 } /* sppd_sighandler */
400 
401 /* Display usage and exit */
402 static void
403 usage(void)
404 {
405 	fprintf(stdout,
406 "Usage: %s options\n" \
407 "Where options are:\n" \
408 "\t-a address Address to connect to (required)\n" \
409 "\t-b         Run in background\n" \
410 "\t-c channel RFCOMM channel to connect to\n" \
411 "\t-t tty     TTY name (required in background mode)\n" \
412 "\t-h         Display this message\n", SPPD_IDENT);
413 
414 	exit(255);
415 } /* usage */
416 
417