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
|
#!/usr/bin/python
#
# Example Android logcat to wpa_supplicant wrapper for QR Code scans
# Copyright (c) 2017, Qualcomm Atheros, Inc.
#
# This software may be distributed under the terms of the BSD license.
# See README for more details.
import os
import sys
import argparse
import logging
import qrcode
scriptsdir = os.path.dirname(os.path.realpath(sys.modules[__name__].__file__))
sys.path.append(os.path.join(scriptsdir, '..', '..', 'wpaspy'))
import wpaspy
wpas_ctrl = '/var/run/wpa_supplicant'
def wpas_connect():
ifaces = []
if os.path.isdir(wpas_ctrl):
try:
ifaces = [os.path.join(wpas_ctrl, i) for i in os.listdir(wpas_ctrl)]
except OSError as error:
print("Could not find wpa_supplicant: ", error)
return None
if len(ifaces) < 1:
print("No wpa_supplicant control interface found")
return None
for ctrl in ifaces:
try:
wpas = wpaspy.Ctrl(ctrl)
return wpas
except Exception as e:
pass
return None
def dpp_logcat():
for line in iter(sys.stdin.readline, ''):
if "ResultHandler: Launching intent: Intent" not in line:
continue
if "act=android.intent.action.VIEW" not in line:
continue
uri = None
for val in line.split(' '):
if val.startswith('dat='):
uri = val.split('=', 1)[1]
break
if not uri:
continue
if not uri.startswith('DPP:'):
continue
print("Found DPP bootstrap info URI:")
print(uri)
wpas = wpas_connect()
if not wpas:
print("Could not connect to wpa_supplicant")
print('')
continue
res = wpas.request("DPP_QR_CODE " + uri);
try:
id = int(res)
except ValueError:
print("QR Code URI rejected")
continue
print("QR Code URI accepted - ID=%d" % id)
print(wpas.request("DPP_BOOTSTRAP_INFO %d" % id))
del wpas
def dpp_display(curve):
wpas = wpas_connect()
if not wpas:
print("Could not connect to wpa_supplicant")
return
res = wpas.request("STATUS")
addr = None
for line in res.splitlines():
if line.startswith("address="):
addr = line.split('=')[1]
break
cmd = "DPP_BOOTSTRAP_GEN type=qrcode"
cmd += " chan=81/1"
if addr:
cmd += " mac=" + addr.replace(':','')
if curve:
cmd += " curve=" + curve
res = wpas.request(cmd)
try:
id = int(res)
except ValueError:
print("Failed to generate bootstrap info URI")
return
print("Bootstrap information - ID=%d" % id)
print(wpas.request("DPP_BOOTSTRAP_INFO %d" % id))
uri = wpas.request("DPP_BOOTSTRAP_GET_URI %d" % id)
print(uri)
print("ID=%d" % id)
qr = qrcode.QRCode(error_correction=qrcode.constants.ERROR_CORRECT_M,
border=3)
qr.add_data(uri, optimize=5)
qr.print_ascii(tty=True)
print("ID=%d" % id)
del wpas
def main():
parser = argparse.ArgumentParser(description='Android logcat to wpa_supplicant integration for DPP QR Code operations')
parser.add_argument('-d', const=logging.DEBUG, default=logging.INFO,
action='store_const', dest='loglevel',
help='verbose debug output')
parser.add_argument('--curve', '-c',
help='set a specific curve (P-256, P-384, P-521, BP-256R1, BP-384R1, BP-512R1) for key generation')
parser.add_argument('command', choices=['logcat',
'display'],
nargs='?')
args = parser.parse_args()
logging.basicConfig(level=args.loglevel)
if args.command == "logcat":
dpp_logcat()
elif args.command == "display":
dpp_display(args.curve)
if __name__ == '__main__':
main()
|
