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