Merge branch 'bug_514_new_proc_oled_brightness_path'
[enigma2.git] / lib / python / Plugins / SystemPlugins / WirelessLan / Wlan.py
1 from Components.config import config, ConfigYesNo, NoSave, ConfigSubsection, ConfigText, ConfigSelection, ConfigPassword
2 from Components.Console import Console
3
4 from os import system
5 from string import maketrans, strip
6 import sys
7 import types
8 from re import compile as re_compile, search as re_search
9 from pythonwifi.iwlibs import getNICnames, Wireless, Iwfreq, getWNICnames
10 from pythonwifi import flags as wififlags
11
12 list = []
13 list.append("WEP")
14 list.append("WPA")
15 list.append("WPA2")
16 list.append("WPA/WPA2")
17
18 weplist = []
19 weplist.append("ASCII")
20 weplist.append("HEX")
21
22 config.plugins.wlan = ConfigSubsection()
23 config.plugins.wlan.essid = NoSave(ConfigText(default = "home", fixed_size = False))
24 config.plugins.wlan.hiddenessid = NoSave(ConfigText(default = "home", fixed_size = False))
25
26 config.plugins.wlan.encryption = ConfigSubsection()
27 config.plugins.wlan.encryption.enabled = NoSave(ConfigYesNo(default = True))
28 config.plugins.wlan.encryption.type = NoSave(ConfigSelection(list, default = "WPA/WPA2"))
29 config.plugins.wlan.encryption.wepkeytype = NoSave(ConfigSelection(weplist, default = "ASCII"))
30 config.plugins.wlan.encryption.psk = NoSave(ConfigPassword(default = "mysecurewlan", fixed_size = False))
31
32 class Wlan:
33         def __init__(self, iface):
34                 a = ''; b = ''
35                 for i in range(0, 255):
36                         a = a + chr(i)
37                         if i < 32 or i > 127:
38                                 b = b + ' '
39                         else:
40                                 b = b + chr(i)
41                 
42                 self.iface = iface
43                 self.wlaniface = {}
44                 self.WlanConsole = Console()
45                 self.asciitrans = maketrans(a, b)
46
47         def stopWlanConsole(self):
48                 if self.WlanConsole is not None:
49                         print "killing self.WlanConsole"
50                         self.WlanConsole = None
51                         del self.WlanConsole
52                         
53         def getDataForInterface(self, callback = None):
54                 #get ip out of ip addr, as avahi sometimes overrides it in ifconfig.
55                 print "self.iface im getDataForInterface",self.iface
56                 if len(self.WlanConsole.appContainers) == 0:
57                         self.WlanConsole = Console()
58                         cmd = "iwconfig " + self.iface
59                         self.WlanConsole.ePopen(cmd, self.iwconfigFinished, callback)
60
61         def iwconfigFinished(self, result, retval, extra_args):
62                 print "self.iface im iwconfigFinished",self.iface
63                 callback = extra_args
64                 data = { 'essid': False, 'frequency': False, 'acesspoint': False, 'bitrate': False, 'encryption': False, 'quality': False, 'signal': False }
65                 
66                 for line in result.splitlines():
67                         line = line.strip()
68                         if "ESSID" in line:
69                                 if "off/any" in line:
70                                         ssid = _("No Connection")
71                                 else:
72                                         if "Nickname" in line:
73                                                 tmpssid=(line[line.index('ESSID')+7:line.index('"  Nickname')])
74                                                 if tmpssid == '':
75                                                         ssid = _("Hidden networkname")
76                                                 elif tmpssid ==' ':
77                                                         ssid = _("Hidden networkname")
78                                                 else:
79                                                         ssid = tmpssid
80                                         else:
81                                                 tmpssid=(line[line.index('ESSID')+7:len(line)-1])
82                                                 if tmpssid == '':
83                                                         ssid = _("Hidden networkname")
84                                                 elif tmpssid ==' ':
85                                                         ssid = _("Hidden networkname")
86                                                 else:
87                                                         ssid = tmpssid                                          
88
89                                 if ssid is not None:
90                                         data['essid'] = ssid
91                         if 'Frequency' in line:
92                                 frequency = line[line.index('Frequency')+10 :line.index(' GHz')]
93                                 if frequency is not None:
94                                         data['frequency'] = frequency
95                         if "Access Point" in line:
96                                 ap=line[line.index('Access Point')+14:len(line)-1]
97                                 if ap is not None:
98                                         data['acesspoint'] = ap
99                         if "Bit Rate" in line:
100                                 br = line[line.index('Bit Rate')+9 :line.index(' Mb/s')]
101                                 if br is not None:
102                                         data['bitrate'] = br
103                         if 'Encryption key' in line:
104                                 if ":off" in line:
105                                     enc = _("Disabled")
106                                 else:
107                                     enc = line[line.index('Encryption key')+15 :line.index('   Security')]
108                                 if enc is not None:
109                                         data['encryption'] = _("Enabled")
110                         if 'Quality' in line:
111                                 if "/100" in line:
112                                         qual = line[line.index('Quality')+8:line.index('/100')]
113                                 else:
114                                         qual = line[line.index('Quality')+8:line.index('Sig')]
115                                 if qual is not None:
116                                         data['quality'] = qual
117                         if 'Signal level' in line:
118                                 signal = line[line.index('Signal level')+13 :line.index(' dBm')]
119                                 if signal is not None:
120                                         data['signal'] = signal
121
122                 self.wlaniface[self.iface] = data
123                 
124                 if len(self.WlanConsole.appContainers) == 0:
125                         print "self.wlaniface after loading:", self.wlaniface
126                         self.WlanConsole = None
127                         if callback is not None:
128                                 callback(True,self.wlaniface)
129
130         def getAdapterAttribute(self, attribute):
131                 if self.wlaniface.has_key(self.iface):
132                         print "self.wlaniface.has_key",self.iface
133                         if self.wlaniface[self.iface].has_key(attribute):
134                                 return self.wlaniface[self.iface][attribute]
135                 return None
136                 
137         def asciify(self, str):
138                 return str.translate(self.asciitrans)
139
140         
141         def getWirelessInterfaces(self):
142                 device = re_compile('[a-z]{2,}[0-9]*:')
143                 ifnames = []
144
145                 fp = open('/proc/net/wireless', 'r')
146                 for line in fp:
147                         try:
148                                 # append matching pattern, without the trailing colon
149                                 ifnames.append(device.search(line).group()[:-1])
150                         except AttributeError:
151                                 pass
152                 return ifnames
153
154         
155         def getNetworkList(self):
156                 system("ifconfig "+self.iface+" up")
157                 ifobj = Wireless(self.iface) # a Wireless NIC Object
158                 
159                 #Association mappings
160                 #stats, quality, discard, missed_beacon = ifobj.getStatistics()
161                 #snr = quality.signallevel - quality.noiselevel
162
163                 try:
164                         scanresults = ifobj.scan()
165                 except:
166                         scanresults = None
167                         print "[Wlan.py] No Wireless Networks could be found"
168                 
169                 if scanresults is not None:
170                         aps = {}
171                         (num_channels, frequencies) = ifobj.getChannelInfo()
172                         index = 1
173                         for result in scanresults:
174                                 bssid = result.bssid
175
176                                 if result.encode.flags & wififlags.IW_ENCODE_DISABLED > 0:
177                                         encryption = False
178                                 elif result.encode.flags & wififlags.IW_ENCODE_NOKEY > 0:
179                                         encryption = True
180                                 else:
181                                         encryption = None
182                                 
183                                 signal = str(result.quality.siglevel-0x100) + " dBm"
184                                 quality = "%s/%s" % (result.quality.quality,ifobj.getQualityMax().quality)
185                                 
186                                 extra = []
187                                 for element in result.custom:
188                                         element = element.encode()
189                                         extra.append( strip(self.asciify(element)) )
190                                 for element in extra:
191                                         print element
192                                         if 'SignalStrength' in element:
193                                                 signal = element[element.index('SignalStrength')+15:element.index(',L')]                                        
194                                         if 'LinkQuality' in element:
195                                                 quality = element[element.index('LinkQuality')+12:len(element)]                         
196
197                                 aps[bssid] = {
198                                         'active' : True,
199                                         'bssid': result.bssid,
200                                         'channel': frequencies.index(ifobj._formatFrequency(result.frequency.getFrequency())) + 1,
201                                         'encrypted': encryption,
202                                         'essid': strip(self.asciify(result.essid)),
203                                         'iface': self.iface,
204                                         'maxrate' : ifobj._formatBitrate(result.rate[-1][-1]),
205                                         'noise' : '',#result.quality.nlevel-0x100,
206                                         'quality' : str(quality),
207                                         'signal' : str(signal),
208                                         'custom' : extra,
209                                 }
210                                 #print "GOT APS ENTRY:",aps[bssid]
211                                 index = index + 1
212                         return aps
213
214                 
215         def getStatus(self):
216                 ifobj = Wireless(self.iface)
217                 fq = Iwfreq()
218                 try:
219                         self.channel = str(fq.getChannel(str(ifobj.getFrequency()[0:-3])))
220                 except:
221                         self.channel = 0
222                 status = {
223                                   'BSSID': str(ifobj.getAPaddr()), #ifobj.getStatistics()
224                                   'ESSID': str(ifobj.getEssid()),
225                                   'quality': "%s/%s" % (ifobj.getStatistics()[1].quality,ifobj.getQualityMax().quality),
226                                   'signal': str(ifobj.getStatistics()[1].siglevel-0x100) + " dBm",
227                                   'bitrate': str(ifobj.getBitrate()),
228                                   'channel': str(self.channel),
229                                   #'channel': str(fq.getChannel(str(ifobj.getFrequency()[0:-3]))),
230                 }
231                 
232                 for (key, item) in status.items():
233                         if item is "None" or item is "":
234                                         status[key] = _("N/A")
235                                 
236                 return status
237
238
239 class wpaSupplicant:
240         def __init__(self):
241                 pass
242         
243                 
244         def writeConfig(self):  
245                         
246                         essid = config.plugins.wlan.essid.value
247                         hiddenessid = config.plugins.wlan.hiddenessid.value
248                         encrypted = config.plugins.wlan.encryption.enabled.value
249                         encryption = config.plugins.wlan.encryption.type.value
250                         wepkeytype = config.plugins.wlan.encryption.wepkeytype.value
251                         psk = config.plugins.wlan.encryption.psk.value
252                         fp = file('/etc/wpa_supplicant.conf', 'w')
253                         fp.write('#WPA Supplicant Configuration by enigma2\n')
254                         fp.write('ctrl_interface=/var/run/wpa_supplicant\n')
255                         fp.write('eapol_version=1\n')
256                         fp.write('fast_reauth=1\n')     
257                         if essid == 'hidden...':
258                                 fp.write('ap_scan=2\n')
259                         else:
260                                 fp.write('ap_scan=1\n')
261                         fp.write('network={\n')
262                         if essid == 'hidden...':
263                                 fp.write('\tssid="'+hiddenessid+'"\n')
264                         else:
265                                 fp.write('\tssid="'+essid+'"\n')
266                         fp.write('\tscan_ssid=0\n')                     
267                         if encrypted:
268                                 if encryption == 'WPA' or encryption == 'WPA2' or encryption == 'WPA/WPA2' :
269                                         fp.write('\tkey_mgmt=WPA-PSK\n')
270                                         
271                                         if encryption == 'WPA':
272                                                 fp.write('\tproto=WPA\n')
273                                                 fp.write('\tpairwise=TKIP\n')
274                                                 fp.write('\tgroup=TKIP\n')
275                                         elif encryption == 'WPA2':
276                                                 fp.write('\tproto=WPA RSN\n')
277                                                 fp.write('\tpairwise=CCMP TKIP\n')
278                                                 fp.write('\tgroup=CCMP TKIP\n')                                         
279                                         else:
280                                                 fp.write('\tproto=WPA WPA2\n')
281                                                 fp.write('\tpairwise=CCMP\n')
282                                                 fp.write('\tgroup=TKIP\n')                                      
283                                         fp.write('\tpsk="'+psk+'"\n')
284                                                 
285                                 elif encryption == 'WEP':
286                                         fp.write('\tkey_mgmt=NONE\n')
287                                         if wepkeytype == 'ASCII':
288                                                 fp.write('\twep_key0="'+psk+'"\n')
289                                         else:
290                                                 fp.write('\twep_key0='+psk+'\n')
291                         else:
292                                 fp.write('\tkey_mgmt=NONE\n')                   
293                         fp.write('}')
294                         fp.write('\n')
295                         fp.close()
296                         system("cat /etc/wpa_supplicant.conf")
297                 
298         def loadConfig(self):
299                 try:
300                         #parse the wpasupplicant configfile
301                         fp = file('/etc/wpa_supplicant.conf', 'r')
302                         supplicant = fp.readlines()
303                         fp.close()
304                         ap_scan = False
305                         essid = None
306
307                         for s in supplicant:
308                                 split = s.strip().split('=',1)
309                                 if split[0] == 'ap_scan':
310                                         print "[Wlan.py] Got Hidden SSID Scan  Value "+split[1]
311                                         if split[1] == '2':
312                                                 ap_scan = True
313                                         else:
314                                                 ap_scan = False
315                                                 
316                                 elif split[0] == 'ssid':
317                                         print "[Wlan.py] Got SSID "+split[1][1:-1]
318                                         essid = split[1][1:-1]
319                                         
320                                 elif split[0] == 'proto':
321                                         config.plugins.wlan.encryption.enabled.value = True
322                                         if split[1] == "WPA" :
323                                                 mode = 'WPA'
324                                         if split[1] == "WPA WPA2" :
325                                                 mode = 'WPA/WPA2'
326                                         if split[1] == "WPA RSN" :
327                                                 mode = 'WPA2'
328                                         config.plugins.wlan.encryption.type.value = mode
329                                         print "[Wlan.py] Got Encryption: "+mode
330                                         
331                                 #currently unused !
332                                 #elif split[0] == 'key_mgmt':
333                                 #       print "split[1]",split[1]
334                                 #       if split[1] == "WPA-PSK" :
335                                 #               config.plugins.wlan.encryption.enabled.value = True
336                                 #               config.plugins.wlan.encryption.type.value = "WPA/WPA2"
337                                 #       print "[Wlan.py] Got Encryption: "+ config.plugins.wlan.encryption.type.value
338                                         
339                                 elif split[0] == 'wep_key0':
340                                         config.plugins.wlan.encryption.enabled.value = True
341                                         config.plugins.wlan.encryption.type.value = 'WEP'
342                                         if split[1].startswith('"') and split[1].endswith('"'):
343                                                 config.plugins.wlan.encryption.wepkeytype.value = 'ASCII'
344                                                 config.plugins.wlan.encryption.psk.value = split[1][1:-1]
345                                         else:
346                                                 config.plugins.wlan.encryption.wepkeytype.value = 'HEX'
347                                                 config.plugins.wlan.encryption.psk.value = split[1]                                             
348                                         
349                                 elif split[0] == 'psk':
350                                         config.plugins.wlan.encryption.psk.value = split[1][1:-1]
351                                 else:
352                                         pass
353                                 
354                         if ap_scan is True:
355                                 config.plugins.wlan.hiddenessid.value = essid
356                                 config.plugins.wlan.essid.value = 'hidden...'
357                         else:
358                                 config.plugins.wlan.hiddenessid.value = essid
359                                 config.plugins.wlan.essid.value = essid
360                         wsconfig = {
361                                         'hiddenessid': config.plugins.wlan.hiddenessid.value,
362                                         'ssid': config.plugins.wlan.essid.value,
363                                         'encryption': config.plugins.wlan.encryption.enabled.value,
364                                         'encryption_type': config.plugins.wlan.encryption.type.value,
365                                         'encryption_wepkeytype': config.plugins.wlan.encryption.wepkeytype.value,
366                                         'key': config.plugins.wlan.encryption.psk.value,
367                                 }
368                 
369                         for (key, item) in wsconfig.items():
370                                 if item is "None" or item is "":
371                                         if key == 'hiddenessid':
372                                                 wsconfig['hiddenessid'] = "home"
373                                         if key == 'ssid':
374                                                 wsconfig['ssid'] = "home"
375                                         if key == 'encryption':
376                                                 wsconfig['encryption'] = True                           
377                                         if key == 'encryption':
378                                                 wsconfig['encryption_type'] = "WPA/WPA2"
379                                         if key == 'encryption':
380                                                 wsconfig['encryption_wepkeytype'] = "ASCII"
381                                         if key == 'encryption':
382                                                 wsconfig['key'] = "mysecurewlan"
383
384                 except:
385                         print "[Wlan.py] Error parsing /etc/wpa_supplicant.conf"
386                         wsconfig = {
387                                         'hiddenessid': "home",
388                                         'ssid': "home",
389                                         'encryption': True,
390                                         'encryption_type': "WPA/WPA2",
391                                         'encryption_wepkeytype': "ASCII",
392                                         'key': "mysecurewlan",
393                                 }
394                 print "[Wlan.py] WS-CONFIG-->",wsconfig
395                 return wsconfig
396
397         
398         def restart(self, iface):
399                 system("start-stop-daemon -K -x /usr/sbin/wpa_supplicant")
400                 system("start-stop-daemon -S -x /usr/sbin/wpa_supplicant -- -B -i"+iface+" -c/etc/wpa_supplicant.conf")
401
402 class Status:
403         def __init__(self):
404                 self.wlaniface = {}
405                 self.backupwlaniface = {}
406                 self.WlanConsole = Console()
407
408         def stopWlanConsole(self):
409                 if self.WlanConsole is not None:
410                         print "killing self.WlanConsole"
411                         self.WlanConsole = None
412                         
413         def getDataForInterface(self, iface, callback = None):
414                 self.WlanConsole = Console()
415                 cmd = "iwconfig " + iface
416                 self.WlanConsole.ePopen(cmd, self.iwconfigFinished, [iface, callback])
417
418         def iwconfigFinished(self, result, retval, extra_args):
419                 (iface, callback) = extra_args
420                 data = { 'essid': False, 'frequency': False, 'acesspoint': False, 'bitrate': False, 'encryption': False, 'quality': False, 'signal': False }
421                 for line in result.splitlines():
422                         line = line.strip()
423                         if "ESSID" in line:
424                                 if "off/any" in line:
425                                         ssid = _("No Connection")
426                                 else:
427                                         if "Nickname" in line:
428                                                 tmpssid=(line[line.index('ESSID')+7:line.index('"  Nickname')])
429                                                 if tmpssid == '':
430                                                         ssid = _("Hidden networkname")
431                                                 elif tmpssid ==' ':
432                                                         ssid = _("Hidden networkname")
433                                                 else:
434                                                         ssid = tmpssid
435                                         else:
436                                                 tmpssid=(line[line.index('ESSID')+7:len(line)-1])
437                                                 if tmpssid == '':
438                                                         ssid = _("Hidden networkname")
439                                                 elif tmpssid ==' ':
440                                                         ssid = _("Hidden networkname")
441                                                 else:
442                                                         ssid = tmpssid                                          
443                                 if ssid is not None:
444                                         data['essid'] = ssid
445                         if 'Frequency' in line:
446                                 frequency = line[line.index('Frequency')+10 :line.index(' GHz')]
447                                 if frequency is not None:
448                                         data['frequency'] = frequency
449                         if "Access Point" in line:
450                                 ap=line[line.index('Access Point')+14:len(line)]
451                                 if ap is not None:
452                                         data['acesspoint'] = ap
453                                         if ap == "Not-Associated":
454                                                 data['essid'] = _("No Connection")
455                         if "Bit Rate" in line:
456                                 if "kb" in line:
457                                         br = line[line.index('Bit Rate')+9 :line.index(' kb/s')]
458                                         if br == '0':
459                                                 br = _("Unsupported")
460                                         else:
461                                                 br += " Mb/s"
462                                 else:
463                                         br = line[line.index('Bit Rate')+9 :line.index(' Mb/s')] + " Mb/s"
464                                 if br is not None:
465                                         data['bitrate'] = br
466                         if 'Encryption key' in line:
467                                 if ":off" in line:
468                                         if data['acesspoint'] is not "Not-Associated":
469                                                 enc = _("Unsupported")
470                                         else:
471                                                 enc = _("Disabled")
472                                 else:
473                                         enc = line[line.index('Encryption key')+15 :line.index('   Security')]
474                                         if enc is not None:
475                                                 enc = _("Enabled")
476                                 if enc is not None:
477                                         data['encryption'] = enc
478                         if 'Quality' in line:
479                                 if "/100" in line:
480                                         #qual = line[line.index('Quality')+8:line.index('/100')]
481                                         qual = line[line.index('Quality')+8:line.index('  Signal')]
482                                 else:
483                                         qual = line[line.index('Quality')+8:line.index('Sig')]
484                                 if qual is not None:
485                                         data['quality'] = qual
486                         if 'Signal level' in line:
487                                 if "dBm" in line:
488                                         signal = line[line.index('Signal level')+13 :line.index(' dBm')]
489                                         signal += " dBm"
490                                 elif "/100" in line:
491                                         if "Noise" in line:
492                                                 signal = line[line.index('Signal level')+13:line.index('  Noise')]
493                                         else:
494                                                 signal = line[line.index('Signal level')+13:len(line)]
495                                 else:
496                                         if "Noise" in line:
497                                                 signal = line[line.index('Signal level')+13:line.index('  Noise')]
498                                         else:
499                                                 signal = line[line.index('Signal level')+13:len(line)]                                          
500                                 if signal is not None:
501                                         data['signal'] = signal
502
503                 self.wlaniface[iface] = data
504                 self.backupwlaniface = self.wlaniface
505                 
506                 if self.WlanConsole is not None:
507                         if len(self.WlanConsole.appContainers) == 0:
508                                 print "self.wlaniface after loading:", self.wlaniface
509                                 if callback is not None:
510                                         callback(True,self.wlaniface)
511
512         def getAdapterAttribute(self, iface, attribute):
513                 self.iface = iface
514                 if self.wlaniface.has_key(self.iface):
515                         if self.wlaniface[self.iface].has_key(attribute):
516                                 return self.wlaniface[self.iface][attribute]
517                 return None
518         
519 iStatus = Status()