庖丁解牛fqrouter - 整体构成

Test

1.      def list_existing_networks(iface, control_socket_dir):
   
2.     output = shell_execute('%s -p %s -i %s list_network' % (P2P_CLI_PATH, control_socket_dir, iface))
   
3.     LOGGER.info('existing networks: %s' % output)
   
4.     existing_networks = {}
   
5.     for line in output.splitlines(False)[1:]:
   
6.     index, ssid, bssid, status = line.split('\t')
   
7.     existing_networks[int(index)] = {
   
8.     'ssid': ssid,
   
9.     'bssid': bssid,
   
10.     'status': status
    
11.     }
    
12.     return existing_networks
    
13. 
    
14. 
    
15.     def get_p2p_supplicant_control_socket_dir():
    
16.     try:
    
17.     return get_wpa_supplicant_control_socket_dir(P2P_SUPPLICANT_CONF_PATH)
    
18.     except:
    
19.     LOGGER.exception('failed to get p2p supplicant control socket dir')
    
20.     return get_wpa_supplicant_control_socket_dir()
    
21. 
    
22. 
    
23.     def get_wpa_supplicant_control_socket_dir(conf_path=WPA_SUPPLICANT_CONF_PATH):
    
24.     if shell.USE_SU:
    
25.     content = shell_execute('/data/data/fq.router2/busybox cat %s' % conf_path)
    
26.     return parse_wpa_supplicant_conf(content)
    
27.     try:
    
28.     if not os.path.exists(conf_path):
    
29.     raise Exception('can not find wpa_supplicant.conf')
    
30.     with open(conf_path) as f:
    
31.     content = f.read()
    
32.     control_socket_dir = parse_wpa_supplicant_conf(content)
    
33.     if WPA_SUPPLICANT_CONF_PATH == conf_path:
    
34.     control_socket_dir = fix_wrong_control_socket_dir(control_socket_dir)
    
35.     if control_socket_dir:
    
36.     return control_socket_dir
    
37.     else:
    
38.     raise Exception('can not find ctrl_interface dir from wpa_supplicant.conf')
    
39.     except:
    
40.     LOGGER.exception('failed to get wpa_supplicant control socket dir')
    
41.     return '/data/misc/wifi/wpa_supplicant'
    
42. 
    
43. 
    
44.     def fix_wrong_control_socket_dir(control_socket_dir):
    
45.     if os.path.exists('/data/system/wpa_supplicant/%s' % WIFI_INTERFACE):
    
46.     return '/data/system/wpa_supplicant'
    
47.     if os.path.exists('/data/misc/wifi/wpa_supplicant/%s' % WIFI_INTERFACE):
    
48.     return '/data/misc/wifi/wpa_supplicant'
    
49.     if control_socket_dir == WIFI_INTERFACE:
    
50.     return WIFI_INTERFACE
    
51.     if control_socket_dir:
    
52.     control_socket_not_exists = not os.path.exists(
    
53.     os.path.join(control_socket_dir, WIFI_INTERFACE))
    
54.     else:
    
55.     control_socket_not_exists = True
    
56.     dev_socket_exists = os.path.exists('/dev/socket/wpa_%s' % WIFI_INTERFACE)
    
57.     if dev_socket_exists and control_socket_not_exists:
    
58.     return WIFI_INTERFACE # any valid dir will cause wpa_cli fail
    
59.     return control_socket_dir
    
60. 
    
61. 
    
62.     def parse_wpa_supplicant_conf(content):
    
63.     for line in content.splitlines():
    
64.     line = line.strip()
    
65.     if not line:
    
66.     continue
    
67.     if line.startswith('ctrl_interface='):
    
68.     line = line.replace('ctrl_interface=', '')
    
69.     parts = line.split(' ')
    
70.     for part in parts:
    
71.     if part.startswith('DIR='): # if there is DIR=
    
72.     return part.replace('DIR=', '')
    
73.     return line # otherwise just return the ctrl_interface=
    
74.     return None
    
75. 
    
76. 
    
77.     def netd_execute(command):
    
78.     if shell.USE_SU:
    
79.     return shell.launch_python('main', ['netd-execute', '"%s"' % command])
    
80.     global netd_sequence_number
    
81.     netd_socket = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
    
82.     try:
    
83.     netd_socket.connect('/dev/socket/netd')
    
84.     if netd_sequence_number:
    
85.     netd_sequence_number += 1
    
86.     LOGGER.info('send: %s %s' % (netd_sequence_number, command))
    
87.     netd_socket.send('%s %s\0' % (netd_sequence_number, command))
    
88.     else:
    
89.     LOGGER.info('send: %s' % command)
    
90.     netd_socket.send('%s\0' % command)
    
91.     output = netd_socket.recv(1024)
    
92.     LOGGER.info('received: %s' % output)
    
93.     if not netd_sequence_number and 'Invalid sequence number' in output:
    
94.     LOGGER.info('resend command to netd with sequence number')
    
95.     netd_sequence_number = 1
    
96.     netd_execute(command)
    
97.     finally:
    
98.     netd_socket.close()
    
99. 
    
100. 
     
101.     def shell_execute(command):
     
102.     LOGGER.info('execute: %s' % command)
     
103.     try:
     
104.     output = shell.check_output(shlex.split(command) if isinstance(command, basestring) else command)
     
105.     LOGGER.info('succeed, output: %s' % output)
     
106.     except subprocess.CalledProcessError, e:
     
107.     LOGGER.error('failed, output: %s' % e.output)
     
108.     raise
     
109.     return output

复制代码

Test

Pick & Play：