wirelesslan/src/Wlan.py

   1 from enigma import eListboxPythonMultiContent, eListbox, gFont, RT_HALIGN_LEFT, RT_HALIGN_RIGHT, RT_HALIGN_CENTER
   2
   3 from Components.MultiContent import MultiContentEntryText
   4 from Components.GUIComponent import GUIComponent
   5 from Components.HTMLComponent import HTMLComponent
   6 from Components.config import config, ConfigYesNo, ConfigIP, NoSave, ConfigSubsection, ConfigMAC, ConfigEnableDisable, ConfigText, ConfigSelection
   7
   8 from pythonwifi import iwlibs
   9
  10 import os, string
  11
  12
  13 list = []
  14 list.append(_("WEP"))
  15 list.append(_("WPA"))
  16 list.append(_("WPA2"))
  17
  18 config.plugins.wlan = ConfigSubsection()
  19 config.plugins.wlan.essid = NoSave(ConfigText(default = "home", fixed_size = False))
  20
  21 config.plugins.wlan.encryption = ConfigSubsection()
  22 config.plugins.wlan.encryption.enabled = NoSave(ConfigYesNo(default = False))
  23 config.plugins.wlan.encryption.type = NoSave(ConfigSelection(list, default = _("WPA")))
  24 config.plugins.wlan.encryption.psk = NoSave(ConfigText(default = "mysecurewlan", fixed_size = False))
  25
  26 class Wlan:
  27         def __init__(self):
  28                 a = ''; b = ''
  29
  30                 for i in range(0, 255):
  31                     a = a + chr(i)
  32                     if i < 32 or i > 127:
  33                         b = b + ' '
  34                     else:
  35                         b = b + chr(i)
  36
  37                 self.asciitrans = string.maketrans(a, b)
  38
  39         def asciify(self, str):
  40                 return str.translate(self.asciitrans)
  41
  42         def getWirelessInterfaces(self):
  43                 iwifaces = None
  44                 try:
  45                         iwifaces = iwlibs.getNICnames()
  46                 except:
  47                         iwifaces = None
  48                         "[Wlan.py] No Wireless Networkcards could be found"
  49
  50                 return iwifaces
  51
  52         def getNetworkList(self, iface):
  53
  54                 ifobj = iwlibs.Wireless(iface) # a Wireless NIC Object
  55                 print "ifobj.getStatistics(): ", ifobj.getStatistics()
  56
  57                 #Association mappings
  58                 stats, quality, discard, missed_beacon = ifobj.getStatistics()
  59                 snr = quality.signallevel - quality.noiselevel
  60                 os.system("ifconfig "+iface+" up")
  61
  62                 try:
  63                         scanresults = ifobj.scan()
  64                 except:
  65                         scanresults = None
  66                         print "[Wlan.py] No Wireless Networks could be found"
  67
  68                 if scanresults is not None:
  69                         aps = {}
  70                         for result in scanresults:
  71
  72                                 bssid = result.bssid
  73
  74                                 encryption = map(lambda x: hex(ord(x)), result.encode)
  75
  76                                 if encryption[-1] == "0x8":
  77                                         encryption = True
  78                                 else:
  79                                         encryption = False
  80
  81                                 extra = []
  82                                 for element in result.custom:
  83                                         element = element.encode()
  84                                         extra.append( string.strip(self.asciify(element)) )
  85
  86                                 aps[bssid] = {
  87                                         'active' : True,
  88                                         'bssid': result.bssid,
  89                                         'channel': result.frequency.getChannel(result.frequency.getFrequency(), result.range),
  90                                         'encrypted': encryption,
  91                                         'essid': string.strip(self.asciify(result.essid)),
  92                                         'iface': iface,
  93                                         'maxrate' : result.rate[-1],
  94                                         'noise' : result.quality.getNoiselevel(),
  95                                         'quality' : result.quality.quality,
  96                                         'signal' : result.quality.getSignallevel(),
  97                                         'custom' : extra,
  98                                 }
  99
 100                         return aps
 101
 102
 103 class WlanList(HTMLComponent, GUIComponent):
 104
 105         def __init__(self, session, iface):
 106
 107                 GUIComponent.__init__(self)
 108                 self.w = Wlan()
 109                 self.iface = iface
 110
 111                 self.l = None
 112                 self.l = eListboxPythonMultiContent()
 113
 114                 self.l.setFont(0, gFont("Regular", 32))
 115                 self.l.setFont(1, gFont("Regular", 18))
 116                 self.l.setFont(2, gFont("Regular", 16))
 117                 self.l.setBuildFunc(self.buildWlanListEntry)
 118
 119                 self.reload()
 120
 121         def buildWlanListEntry(self, essid, bssid, encrypted, iface, maxrate):
 122
 123                 res = [ (essid, encrypted, iface) ]
 124                 e = encrypted and _("Yes") or _("No")
 125                 res.append( MultiContentEntryText(pos=(0, 0), size=(570, 35), font=0, flags=RT_HALIGN_LEFT, text=essid) )
 126                 res.append( MultiContentEntryText(pos=(0, 40), size=(180, 20), font=1, flags=RT_HALIGN_LEFT, text=_("Max. Bitrate: ")+maxrate) )
 127                 res.append( MultiContentEntryText(pos=(190, 40), size=(180, 20), font=1, flags=RT_HALIGN_CENTER, text=_("Encrypted: ")+e) )
 128                 res.append( MultiContentEntryText(pos=(380, 40), size=(190, 20), font=1, flags=RT_HALIGN_RIGHT, text=_("Interface: ")+iface) )
 129                 return res
 130
 131         def reload(self):
 132                 aps = self.w.getNetworkList(self.iface)
 133                 list = []
 134                 if aps is not None:
 135                         print "[Wlan.py] got Accespoints!"
 136                         for ap in aps:
 137                                 a = aps[ap]
 138                                 if a['active']:
 139                                         list.append((a['essid'], a['bssid'], a['encrypted'], a['iface'], a['maxrate']))
 140
 141                 self.l.setList([])
 142                 self.l.setList(list)
 143
 144         GUI_WIDGET = eListbox
 145
 146         def getCurrent(self):
 147                 return self.l.getCurrentSelection()
 148
 149         def postWidgetCreate(self, instance):
 150                 instance.setContent(self.l)
 151                 instance.setItemHeight(60)
 152
 153
 154 class wpaSupplicant:
 155         def __init__(self):
 156                 pass
 157
 158         def writeConfig(self):
 159
 160                         essid = config.plugins.wlan.essid.value
 161                         encrypted = config.plugins.wlan.encryption.enabled.value
 162                         encryption = config.plugins.wlan.encryption.type.value
 163                         psk = config.plugins.wlan.encryption.psk.value
 164
 165
 166                         fp = file('/etc/wpa_supplicant.conf', 'w')
 167                         fp.write('#WPA Supplicant Configuration by enigma2\n\n')
 168                         fp.write('ctrl_interface=/var/run/wpa_supplicant\n')
 169                         fp.write('ctrl_interface_group=0\n')
 170                         fp.write('network={\n')
 171                         fp.write('\tssid="'+essid+'"\n')
 172                         fp.write('\tscan_ssid=1\n')
 173
 174                         if encrypted:
 175
 176                                 if encryption == 'WPA' or encryption == 'WPA2':
 177                                         fp.write('\tkey_mgmt=WPA-PSK\n')
 178
 179                                         if encryption == 'WPA':
 180                                                 fp.write('\tproto=WPA\n')
 181                                                 fp.write('\tpairwise=TKIP\n')
 182                                         else:
 183                                                 fp.write('\tproto=WPA RSN\n')
 184                                                 fp.write('\tpairwise=CCMP TKIP\n')
 185
 186                                         fp.write('\tpsk="'+psk+'"\n')
 187
 188                                 elif encryption == 'WEP':
 189                                         fp.write('\tkey_mgmt=NONE\n')
 190                                         fp.write('\twep_key0="'+psk+'"\n')
 191
 192                         fp.write('}')
 193                         fp.close()
 194
 195
 196         def loadConfig(self):
 197
 198                 try:
 199                         #parse the wpasupplicant configfile
 200                         fp = file('/etc/wpa_supplicant.conf', 'r')
 201                         supplicant = fp.readlines()
 202                         fp.close()
 203
 204                         for s in supplicant:
 205
 206                                 split = s.strip().split('=')
 207                                 if split[0] == 'ssid':
 208                                         print "[Wlan.py] Got SSID "+split[1][1:-1]
 209                                         config.plugins.wlan.essid.value = split[1][1:-1]
 210
 211                                 elif split[0] == 'proto':
 212                                         config.plugins.wlan.encryption.enabled.value = True
 213                                         if split[1] == "WPA RSN" : split[1] = 'WPA2'
 214                                         config.plugins.wlan.encryption.type.value = split[1]
 215                                         print "[Wlan.py] Got Encryption "+split[1]
 216
 217                                 elif split[0] == 'psk':
 218                                         config.plugins.wlan.encryption.psk.value = split[1][1:-1]
 219                                         print "[Wlan.py] Got PSK "+split[1][1:-1]
 220                                 else:
 221                                         pass
 222
 223                 except:
 224                         print "[Wlan.py] Error parsing /etc/wpa_supplicant.conf"
 225
 226         def restart(self, iface):
 227                 import os
 228                 os.system("start-stop-daemon -K -x /usr/sbin/wpa_supplicant")
 229                 os.system("start-stop-daemon -S -x /usr/sbin/wpa_supplicant -- -B -i"+iface+" -c/etc/wpa_supplicant.conf")