• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Copyright 2021-2022 Google LLC
2#
3# Licensed under the Apache License, Version 2.0 (the "License");
4# you may not use this file except in compliance with the License.
5# You may obtain a copy of the License at
6#
7#      https://www.apache.org/licenses/LICENSE-2.0
8#
9# Unless required by applicable law or agreed to in writing, software
10# distributed under the License is distributed on an "AS IS" BASIS,
11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12# See the License for the specific language governing permissions and
13# limitations under the License.
14
15# -----------------------------------------------------------------------------
16# Imports
17# -----------------------------------------------------------------------------
18import asyncio
19import sys
20import os
21import logging
22from colors import color
23
24from bumble.device import Device
25from bumble.transport import open_transport_or_link
26
27
28# -----------------------------------------------------------------------------
29async def main():
30    if len(sys.argv) < 2:
31        print('Usage: run_scanner.py <transport-spec> [filter]')
32        print('example: run_scanner.py usb:0')
33        return
34
35    print('<<< connecting to HCI...')
36    async with await open_transport_or_link(sys.argv[1]) as (hci_source, hci_sink):
37        print('<<< connected')
38        filter_duplicates = (len(sys.argv) == 3 and sys.argv[2] == 'filter')
39
40        device = Device.with_hci('Bumble', 'F0:F1:F2:F3:F4:F5', hci_source, hci_sink)
41
42        @device.on('advertisement')
43        def _(address, ad_data, rssi, connectable):
44            address_type_string = ('PUBLIC', 'RANDOM', 'PUBLIC_ID', 'RANDOM_ID')[address.address_type]
45            address_color = 'yellow' if connectable else 'red'
46            address_qualifier = ''
47            if address_type_string.startswith('P'):
48                type_color = 'cyan'
49            else:
50                if address.is_static:
51                    type_color = 'green'
52                    address_qualifier = '(static)'
53                elif address.is_resolvable:
54                    type_color = 'magenta'
55                    address_qualifier = '(resolvable)'
56                else:
57                    type_color = 'white'
58
59            separator = '\n  '
60            print(f'>>> {color(address, address_color)} [{color(address_type_string, type_color)}]{address_qualifier}:{separator}RSSI:{rssi}{separator}{ad_data.to_string(separator)}')
61
62        await device.power_on()
63        await device.start_scanning(filter_duplicates=filter_duplicates)
64
65        await hci_source.wait_for_termination()
66
67# -----------------------------------------------------------------------------
68logging.basicConfig(level = os.environ.get('BUMBLE_LOGLEVEL', 'DEBUG').upper())
69asyncio.run(main())
70