1 /*
2  * hccontrol.c
3  *
4  * Copyright (c) 2001-2002 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: hccontrol.c,v 1.5 2003/09/05 00:38:24 max Exp $
29  * $FreeBSD$
30  */
31 
32 #define L2CAP_SOCKET_CHECKED
33 #include <bluetooth.h>
34 #include <sys/ioctl.h>
35 #include <sys/sysctl.h>
36 #include <assert.h>
37 #include <err.h>
38 #include <errno.h>
39 #include <netgraph/ng_message.h>
40 #include <stdio.h>
41 #include <stdlib.h>
42 #include <string.h>
43 #include <unistd.h>
44 #include "hccontrol.h"
45 
46 /* Prototypes */
47 static int                  do_hci_command    (char const *, int, char **);
48 static struct hci_command * find_hci_command  (char const *, struct hci_command *);
49 static int                  find_hci_nodes    (struct nodeinfo **);
50 static void                 print_hci_command (struct hci_command *);
51 static void usage                             (void);
52 
53 /* Globals */
54 int	 verbose = 0;
55 int	 timeout;
56 int	 numeric_bdaddr = 0;
57 
58 /* Main */
59 int
60 main(int argc, char *argv[])
61 {
62 	char	*node = NULL;
63 	int	 n;
64 
65 	/* Process command line arguments */
66 	while ((n = getopt(argc, argv, "n:Nvh")) != -1) {
67 		switch (n) {
68 		case 'n':
69 			node = optarg;
70 			break;
71 
72 		case 'N':
73 			numeric_bdaddr = 1;
74 			break;
75 
76 		case 'v':
77 			verbose = 1;
78 			break;
79 
80 		case 'h':
81 		default:
82 			usage();
83 		}
84 	}
85 
86 	argc -= optind;
87 	argv += optind;
88 
89 	if (*argv == NULL)
90 		usage();
91 
92 	n = do_hci_command(node, argc, argv);
93 
94 	return (n);
95 } /* main */
96 
97 /* Create socket and bind it */
98 static int
99 socket_open(char const *node)
100 {
101 	struct sockaddr_hci			 addr;
102 	struct ng_btsocket_hci_raw_filter	 filter;
103 	int					 s, mib[4], num;
104 	size_t					 size;
105 	struct nodeinfo 			*nodes;
106 
107 	num = find_hci_nodes(&nodes);
108 	if (num == 0)
109 		errx(7, "Could not find HCI nodes");
110 
111 	if (node == NULL) {
112 		node = strdup(nodes[0].name);
113 		if (num > 1)
114 			fprintf(stdout, "Using HCI node: %s\n", node);
115 	}
116 
117 	free(nodes);
118 
119 	s = socket(PF_BLUETOOTH, SOCK_RAW, BLUETOOTH_PROTO_HCI);
120 	if (s < 0)
121 		err(1, "Could not create socket");
122 
123 	memset(&addr, 0, sizeof(addr));
124 	addr.hci_len = sizeof(addr);
125 	addr.hci_family = AF_BLUETOOTH;
126 	strncpy(addr.hci_node, node, sizeof(addr.hci_node));
127 	if (bind(s, (struct sockaddr *) &addr, sizeof(addr)) < 0)
128 		err(2, "Could not bind socket, node=%s", node);
129 
130 	if (connect(s, (struct sockaddr *) &addr, sizeof(addr)) < 0)
131 		err(3, "Could not connect socket, node=%s", node);
132 
133 	memset(&filter, 0, sizeof(filter));
134 	bit_set(filter.event_mask, NG_HCI_EVENT_COMMAND_COMPL - 1);
135 	bit_set(filter.event_mask, NG_HCI_EVENT_COMMAND_STATUS - 1);
136 	bit_set(filter.event_mask, NG_HCI_EVENT_INQUIRY_COMPL - 1);
137 	bit_set(filter.event_mask, NG_HCI_EVENT_INQUIRY_RESULT - 1);
138 	bit_set(filter.event_mask, NG_HCI_EVENT_CON_COMPL - 1);
139 	bit_set(filter.event_mask, NG_HCI_EVENT_DISCON_COMPL - 1);
140 	bit_set(filter.event_mask, NG_HCI_EVENT_REMOTE_NAME_REQ_COMPL - 1);
141 	bit_set(filter.event_mask, NG_HCI_EVENT_READ_REMOTE_FEATURES_COMPL - 1);
142 	bit_set(filter.event_mask, NG_HCI_EVENT_READ_REMOTE_VER_INFO_COMPL - 1);
143 	bit_set(filter.event_mask, NG_HCI_EVENT_RETURN_LINK_KEYS - 1);
144 	bit_set(filter.event_mask, NG_HCI_EVENT_READ_CLOCK_OFFSET_COMPL - 1);
145 	bit_set(filter.event_mask, NG_HCI_EVENT_CON_PKT_TYPE_CHANGED - 1);
146 	bit_set(filter.event_mask, NG_HCI_EVENT_ROLE_CHANGE - 1);
147 	bit_set(filter.event_mask, NG_HCI_EVENT_LE -1);
148 
149 	if (setsockopt(s, SOL_HCI_RAW, SO_HCI_RAW_FILTER,
150 			(void * const) &filter, sizeof(filter)) < 0)
151 		err(4, "Could not setsockopt()");
152 
153 	size = (sizeof(mib)/sizeof(mib[0]));
154 	if (sysctlnametomib("net.bluetooth.hci.command_timeout",mib,&size) < 0)
155 		err(5, "Could not sysctlnametomib()");
156 
157 	if (sysctl(mib, sizeof(mib)/sizeof(mib[0]),
158 			(void *) &timeout, &size, NULL, 0) < 0)
159 		err(6, "Could not sysctl()");
160 
161 	timeout ++;
162 
163 	return (s);
164 } /* socket_open */
165 
166 /* Execute commands */
167 static int
168 do_hci_command(char const *node, int argc, char **argv)
169 {
170 	char			*cmd = argv[0];
171 	struct hci_command	*c = NULL;
172 	int			 s, e, help;
173 
174 	help = 0;
175 	if (strcasecmp(cmd, "help") == 0) {
176 		argc --;
177 		argv ++;
178 
179 		if (argc <= 0) {
180 			fprintf(stdout, "Supported commands:\n");
181 			print_hci_command(link_control_commands);
182 			print_hci_command(link_policy_commands);
183 			print_hci_command(host_controller_baseband_commands);
184 			print_hci_command(info_commands);
185 			print_hci_command(status_commands);
186 			print_hci_command(le_commands);
187 			print_hci_command(node_commands);
188 			fprintf(stdout, "\nFor more information use " \
189 				"'help command'\n");
190 
191 			return (OK);
192 		}
193 
194 		help = 1;
195 		cmd = argv[0];
196 	}
197 
198 	c = find_hci_command(cmd, link_control_commands);
199 	if (c != NULL)
200 		goto execute;
201 
202 	c = find_hci_command(cmd, link_policy_commands);
203 	if (c != NULL)
204 		goto execute;
205 
206 	c = find_hci_command(cmd, host_controller_baseband_commands);
207 	if (c != NULL)
208 		goto execute;
209 
210 	c = find_hci_command(cmd, info_commands);
211 	if (c != NULL)
212 		goto execute;
213 
214 	c = find_hci_command(cmd, status_commands);
215 	if (c != NULL)
216 		goto execute;
217 
218 	c = find_hci_command(cmd, le_commands);
219 	if (c != NULL)
220 		goto execute;
221 
222 
223 	c = find_hci_command(cmd, node_commands);
224 	if (c == NULL) {
225 		fprintf(stdout, "Unknown command: \"%s\"\n", cmd);
226 		return (ERROR);
227 	}
228 execute:
229 	if (!help) {
230 		s = socket_open(node);
231 		e = (c->handler)(s, -- argc, ++ argv);
232 		close(s);
233 	} else
234 		e = USAGE;
235 
236 	switch (e) {
237 	case OK:
238 	case FAILED:
239 		break;
240 
241 	case ERROR:
242 		fprintf(stdout, "Could not execute command \"%s\". %s\n",
243 			cmd, strerror(errno));
244 		break;
245 
246 	case USAGE:
247 		fprintf(stdout, "Usage: %s\n%s\n", c->command, c->description);
248 		break;
249 
250 	default: assert(0); break;
251 	}
252 
253 
254 	return (e);
255 } /* do_hci_command */
256 
257 /* Try to find command in specified category */
258 static struct hci_command *
259 find_hci_command(char const *command, struct hci_command *category)
260 {
261 	struct hci_command	*c = NULL;
262 
263 	for (c = category; c->command != NULL; c++) {
264 		char 	*c_end = strchr(c->command, ' ');
265 
266 		if (c_end != NULL) {
267 			int	len = c_end - c->command;
268 
269 			if (strncasecmp(command, c->command, len) == 0)
270 				return (c);
271 		} else if (strcasecmp(command, c->command) == 0)
272 				return (c);
273 	}
274 
275 	return (NULL);
276 } /* find_hci_command */
277 
278 /* Find all HCI nodes */
279 static int
280 find_hci_nodes(struct nodeinfo** nodes)
281 {
282 	struct ng_btsocket_hci_raw_node_list_names	r;
283 	struct sockaddr_hci				addr;
284 	int						s;
285 	const char *					node = "ubt0hci";
286 
287 	r.num_names = MAX_NODE_NUM;
288 	r.names = (struct nodeinfo*)calloc(MAX_NODE_NUM, sizeof(struct nodeinfo));
289 	if (r.names == NULL)
290 		err(8, "Could not allocate memory");
291 
292 	s = socket(PF_BLUETOOTH, SOCK_RAW, BLUETOOTH_PROTO_HCI);
293 	if (s < 0)
294 		err(9, "Could not create socket");
295 
296 	memset(&addr, 0, sizeof(addr));
297 	addr.hci_len = sizeof(addr);
298 	addr.hci_family = AF_BLUETOOTH;
299 	strncpy(addr.hci_node, node, sizeof(addr.hci_node));
300 	if (bind(s, (struct sockaddr *) &addr, sizeof(addr)) < 0)
301 		err(10, "Could not bind socket");
302 
303 	if (ioctl(s, SIOC_HCI_RAW_NODE_LIST_NAMES, &r, sizeof(r)) < 0)
304 		err(11, "Could not get list of HCI nodes");
305 
306 	close(s);
307 
308 	*nodes = r.names;
309 
310 	return (r.num_names);
311 } /* find_hci_nodes */
312 
313 /* Print commands in specified category */
314 static void
315 print_hci_command(struct hci_command *category)
316 {
317 	struct hci_command	*c = NULL;
318 
319 	for (c = category; c->command != NULL; c++)
320 		fprintf(stdout, "\t%s\n", c->command);
321 } /* print_hci_command */
322 
323 /* Usage */
324 static void
325 usage(void)
326 {
327 	fprintf(stdout, "Usage: hccontrol [-hN] [-n HCI_node_name] cmd [p1] [..]\n");
328 	exit(255);
329 } /* usage */
330 
331