code.vuplus.com Git - vuplus_dvbapp/blob - lib/python/Plugins/SystemPlugins/WirelessLan/plugin.py

   1 from enigma import eTimer
   2 from Screens.Screen import Screen
   3 from Components.ActionMap import ActionMap, NumberActionMap
   4 from Components.Pixmap import Pixmap,MultiPixmap
   5 from Components.Label import Label
   6 from Components.Sources.StaticText import StaticText
   7 from Components.Sources.List import List
   8 from Components.MenuList import MenuList
   9 from Components.config import config, getConfigListEntry, ConfigYesNo, NoSave, ConfigSubsection, ConfigText, ConfigSelection, ConfigPassword
  10 from Components.ConfigList import ConfigListScreen
  11 from Components.Network import Network, iNetwork
  12 from Components.Console import Console
  13 from Plugins.Plugin import PluginDescriptor
  14 from os import system, path as os_path, listdir
  15 from Tools.Directories import resolveFilename, SCOPE_PLUGINS, SCOPE_SKIN_IMAGE
  16 from Tools.LoadPixmap import LoadPixmap
  17 from Wlan import Wlan, wpaSupplicant, iStatus
  18
  19 plugin_path = "/usr/lib/enigma2/python/Plugins/SystemPlugins/WirelessLan"
  20
  21 list = []
  22 list.append("WEP")
  23 list.append("WPA")
  24 list.append("WPA2")
  25 list.append("WPA/WPA2")
  26
  27 weplist = []
  28 weplist.append("ASCII")
  29 weplist.append("HEX")
  30
  31 config.plugins.wlan = ConfigSubsection()
  32 config.plugins.wlan.essid = NoSave(ConfigText(default = "home", fixed_size = False))
  33 config.plugins.wlan.hiddenessid = NoSave(ConfigText(default = "home", fixed_size = False))
  34
  35 config.plugins.wlan.encryption = ConfigSubsection()
  36 config.plugins.wlan.encryption.enabled = NoSave(ConfigYesNo(default = False))
  37 config.plugins.wlan.encryption.type = NoSave(ConfigSelection(list, default = "WPA/WPA2" ))
  38 config.plugins.wlan.encryption.wepkeytype = NoSave(ConfigSelection(weplist, default = "ASCII"))
  39 config.plugins.wlan.encryption.psk = NoSave(ConfigPassword(default = "mysecurewlan", fixed_size = False))
  40
  41
  42 class WlanStatus(Screen):
  43         skin = """
  44                 <screen name="WlanStatus" position="center,center" size="560,400" title="Wireless Network State" >
  45                         <ePixmap pixmap="skin_default/buttons/red.png" position="0,0" size="140,40" alphatest="on" />
  46                         <widget source="key_red" render="Label" position="0,0" zPosition="1" size="140,40" font="Regular;20" halign="center" valign="center" backgroundColor="#9f1313" transparent="1" />
  47
  48                         <widget source="LabelBSSID" render="Label" position="10,60" size="250,25" valign="left" font="Regular;20" transparent="1" foregroundColor="#FFFFFF" />
  49                         <widget source="LabelESSID" render="Label" position="10,100" size="250,25" valign="center" font="Regular;20" transparent="1" foregroundColor="#FFFFFF" />
  50                         <widget source="LabelQuality" render="Label" position="10,140" size="250,25" valign="center" font="Regular;20" transparent="1" foregroundColor="#FFFFFF" />
  51                         <widget source="LabelSignal" render="Label" position="10,180" size="250,25" valign="center" font="Regular;20" transparent="1" foregroundColor="#FFFFFF" />
  52                         <widget source="LabelBitrate" render="Label" position="10,220" size="250,25" valign="center" font="Regular;20" transparent="1" foregroundColor="#FFFFFF" />
  53                         <widget source="LabelEnc" render="Label" position="10,260" size="250,25" valign="center" font="Regular;20" transparent="1" foregroundColor="#FFFFFF" />
  54
  55                         <widget source="BSSID" render="Label" position="320,60" size="180,25" valign="center" font="Regular;20" transparent="1" foregroundColor="#FFFFFF" />
  56                         <widget source="ESSID" render="Label" position="320,100" size="180,25" valign="center" font="Regular;20" transparent="1" foregroundColor="#FFFFFF" />
  57                         <widget source="quality" render="Label" position="320,140" size="180,25" valign="center" font="Regular;20" transparent="1" foregroundColor="#FFFFFF" />
  58                         <widget source="signal" render="Label" position="320,180" size="180,25" valign="center" font="Regular;20" transparent="1" foregroundColor="#FFFFFF" />
  59                         <widget source="bitrate" render="Label" position="320,220" size="180,25" valign="center" font="Regular;20" transparent="1" foregroundColor="#FFFFFF" />
  60                         <widget source="enc" render="Label" position="320,260" size="180,25" valign="center" font="Regular;20" transparent="1" foregroundColor="#FFFFFF" />
  61
  62                         <ePixmap pixmap="skin_default/div-h.png" position="0,350" zPosition="1" size="560,2" />
  63                         <widget source="IFtext" render="Label" position="10,355" size="120,21" zPosition="10" font="Regular;20" halign="left" backgroundColor="#25062748" transparent="1" />
  64                         <widget source="IF" render="Label" position="120,355" size="400,21" zPosition="10" font="Regular;20" halign="left" backgroundColor="#25062748" transparent="1" />
  65                         <widget source="Statustext" render="Label" position="10,375" size="115,21" zPosition="10" font="Regular;20" halign="left" backgroundColor="#25062748" transparent="1"/>
  66                         <widget name="statuspic" pixmaps="skin_default/buttons/button_green.png,skin_default/buttons/button_green_off.png" position="130,380" zPosition="10" size="15,16" transparent="1" alphatest="on"/>
  67                 </screen>"""
  68
  69         def __init__(self, session, iface):
  70                 Screen.__init__(self, session)
  71                 self.session = session
  72                 self.iface = iface
  73
  74                 self["LabelBSSID"] = StaticText(_('Accesspoint:'))
  75                 self["LabelESSID"] = StaticText(_('SSID:'))
  76                 self["LabelQuality"] = StaticText(_('Link Quality:'))
  77                 self["LabelSignal"] = StaticText(_('Signal Strength:'))
  78                 self["LabelBitrate"] = StaticText(_('Bitrate:'))
  79                 self["LabelEnc"] = StaticText(_('Encryption:'))
  80
  81                 self["BSSID"] = StaticText()
  82                 self["ESSID"] = StaticText()
  83                 self["quality"] = StaticText()
  84                 self["signal"] = StaticText()
  85                 self["bitrate"] = StaticText()
  86                 self["enc"] = StaticText()
  87
  88                 self["IFtext"] = StaticText()
  89                 self["IF"] = StaticText()
  90                 self["Statustext"] = StaticText()
  91                 self["statuspic"] = MultiPixmap()
  92                 self["statuspic"].hide()
  93                 self["key_red"] = StaticText(_("Close"))
  94
  95                 self.resetList()
  96                 self.updateStatusbar()
  97
  98                 self["actions"] = NumberActionMap(["WizardActions", "InputActions", "EPGSelectActions", "ShortcutActions"],
  99                 {
 100                         "ok": self.exit,
 101                         "back": self.exit,
 102                         "red": self.exit,
 103                 }, -1)
 104                 self.timer = eTimer()
 105                 self.timer.timeout.get().append(self.resetList)
 106                 self.onShown.append(lambda: self.timer.start(5000))
 107                 self.onLayoutFinish.append(self.layoutFinished)
 108                 self.onClose.append(self.cleanup)
 109
 110         def cleanup(self):
 111                 iStatus.stopWlanConsole()
 112
 113         def layoutFinished(self):
 114                 self.setTitle(_("Wireless Network State"))
 115
 116         def resetList(self):
 117                 iStatus.getDataForInterface(self.iface,self.getInfoCB)
 118
 119         def getInfoCB(self,data,status):
 120                 if data is not None:
 121                         if data is True:
 122                                 if status is not None:
 123                                         self["BSSID"].setText(status[self.iface]["acesspoint"])
 124                                         self["ESSID"].setText(status[self.iface]["essid"])
 125                                         self["quality"].setText(status[self.iface]["quality"]+"%")
 126                                         self["signal"].setText(status[self.iface]["signal"])
 127                                         self["bitrate"].setText(status[self.iface]["bitrate"])
 128                                         self["enc"].setText(status[self.iface]["encryption"])
 129                                         self.updateStatusLink(status)
 130
 131         def exit(self):
 132                 self.timer.stop()
 133                 self.close(True)
 134
 135         def updateStatusbar(self):
 136                 self["BSSID"].setText(_("Please wait..."))
 137                 self["ESSID"].setText(_("Please wait..."))
 138                 self["quality"].setText(_("Please wait..."))
 139                 self["signal"].setText(_("Please wait..."))
 140                 self["bitrate"].setText(_("Please wait..."))
 141                 self["enc"].setText(_("Please wait..."))
 142                 self["IFtext"].setText(_("Network:"))
 143                 self["IF"].setText(iNetwork.getFriendlyAdapterName(self.iface))
 144                 self["Statustext"].setText(_("Link:"))
 145
 146         def updateStatusLink(self,status):
 147                 if status is not None:
 148                         if status[self.iface]["acesspoint"] == "No Connection" or status[self.iface]["acesspoint"] == "Not-Associated" or status[self.iface]["acesspoint"] == False:
 149                                 self["statuspic"].setPixmapNum(1)
 150                         else:
 151                                 self["statuspic"].setPixmapNum(0)
 152                         self["statuspic"].show()
 153
 154 class WlanScan(Screen):
 155         skin = """
 156                 <screen name="WlanScan" position="center,center" size="560,400" title="Choose a Wireless Network" >
 157                         <ePixmap pixmap="skin_default/buttons/red.png" position="0,0" size="140,40" alphatest="on" />
 158                         <ePixmap pixmap="skin_default/buttons/green.png" position="140,0" size="140,40" alphatest="on" />
 159                         <ePixmap pixmap="skin_default/buttons/yellow.png" position="280,0" size="140,40" alphatest="on" />
 160                         <widget source="key_red" render="Label" position="0,0" zPosition="1" size="140,40" font="Regular;20" halign="center" valign="center" backgroundColor="#9f1313" transparent="1" />
 161                         <widget source="key_green" render="Label" position="140,0" zPosition="1" size="140,40" font="Regular;20" halign="center" valign="center" backgroundColor="#1f771f" transparent="1" />
 162                         <widget source="key_yellow" render="Label" position="280,0" zPosition="1" size="140,40" font="Regular;20" halign="center" valign="center" backgroundColor="#a08500" transparent="1" />
 163                         <widget source="list" render="Listbox" position="5,40" size="550,300" scrollbarMode="showOnDemand">
 164                                 <convert type="TemplatedMultiContent">
 165                                         {"template": [
 166                                                         MultiContentEntryText(pos = (0, 0), size = (550, 30), font=0, flags = RT_HALIGN_LEFT, text = 0), # index 0 is the essid
 167                                                         MultiContentEntryText(pos = (0, 30), size = (175, 20), font=1, flags = RT_HALIGN_LEFT, text = 5), # index 5 is the interface
 168                                                         MultiContentEntryText(pos = (175, 30), size = (175, 20), font=1, flags = RT_HALIGN_LEFT, text = 4), # index 0 is the encryption
 169                                                         MultiContentEntryText(pos = (350, 0), size = (200, 20), font=1, flags = RT_HALIGN_LEFT, text = 2), # index 0 is the signal
 170                                                         MultiContentEntryText(pos = (350, 30), size = (200, 20), font=1, flags = RT_HALIGN_LEFT, text = 3), # index 0 is the maxrate
 171                                                         MultiContentEntryPixmapAlphaTest(pos = (0, 52), size = (550, 2), png = 6), # index 6 is the div pixmap
 172                                                 ],
 173                                         "fonts": [gFont("Regular", 28),gFont("Regular", 18)],
 174                                         "itemHeight": 54
 175                                         }
 176                                 </convert>
 177                         </widget>
 178                         <ePixmap pixmap="skin_default/div-h.png" position="0,340" zPosition="1" size="560,2" />
 179                         <widget source="info" render="Label" position="0,350" size="560,50" font="Regular;24" halign="center" valign="center" backgroundColor="#25062748" transparent="1" />
 180                 </screen>"""
 181
 182         def __init__(self, session, iface):
 183                 Screen.__init__(self, session)
 184                 self.session = session
 185                 self.iface = iface
 186                 self.skin_path = plugin_path
 187                 self.oldInterfaceState = iNetwork.getAdapterAttribute(self.iface, "up")
 188                 self.APList = None
 189                 self.newAPList = None
 190                 self.WlanList = None
 191                 self.cleanList = None
 192                 self.oldlist = None
 193                 self.listLenght = None
 194                 self.rescanTimer = eTimer()
 195                 self.rescanTimer.callback.append(self.rescanTimerFired)
 196
 197                 self["info"] = StaticText()
 198
 199                 self.list = []
 200                 self["list"] = List(self.list)
 201
 202                 self["key_red"] = StaticText(_("Close"))
 203                 self["key_green"] = StaticText(_("Connect"))
 204                 self["key_yellow"] = StaticText()
 205
 206                 self["actions"] = NumberActionMap(["WizardActions", "InputActions", "EPGSelectActions"],
 207                 {
 208                         "ok": self.select,
 209                         "back": self.cancel,
 210                 }, -1)
 211
 212                 self["shortcuts"] = ActionMap(["ShortcutActions"],
 213                 {
 214                         "red": self.cancel,
 215                         "green": self.select,
 216                 })
 217                 self.onLayoutFinish.append(self.layoutFinished)
 218                 self.getAccessPoints(refresh = False)
 219
 220         def layoutFinished(self):
 221                 self.setTitle(_("Choose a wireless network"))
 222
 223         def select(self):
 224                 cur = self["list"].getCurrent()
 225                 if cur is not None:
 226                         self.rescanTimer.stop()
 227                         del self.rescanTimer
 228                         if cur[1] is not None:
 229                                 essid = cur[1]
 230                                 self.close(essid,self.getWlanList())
 231                         else:
 232                                 self.close(None,None)
 233                 else:
 234                         self.rescanTimer.stop()
 235                         del self.rescanTimer
 236                         self.close(None,None)
 237
 238         def WlanSetupClosed(self, *ret):
 239                 if ret[0] == 2:
 240                         self.rescanTimer.stop()
 241                         del self.rescanTimer
 242                         self.close(None)
 243
 244         def cancel(self):
 245                 if self.oldInterfaceState is False:
 246                         iNetwork.setAdapterAttribute(self.iface, "up", False)
 247                         iNetwork.deactivateInterface(self.iface,self.deactivateInterfaceCB)
 248                 else:
 249                         self.rescanTimer.stop()
 250                         del self.rescanTimer
 251                         self.close(None)
 252
 253         def deactivateInterfaceCB(self,data):
 254                 if data is not None:
 255                         if data is True:
 256                                 self.rescanTimer.stop()
 257                                 del self.rescanTimer
 258                                 self.close(None)
 259
 260         def rescanTimerFired(self):
 261                 self.rescanTimer.stop()
 262                 self.updateAPList()
 263
 264         def buildEntryComponent(self, essid, bssid, encrypted, iface, maxrate, signal):
 265 <<<<<<< HEAD:lib/python/Plugins/SystemPlugins/WirelessLan/plugin.py
 266                 print "buildEntryComponent",essid
 267                 print "buildEntryComponent",bssid
 268 =======
 269 >>>>>>> origin/bug_203_fix_wrong_networkstate:lib/python/Plugins/SystemPlugins/WirelessLan/plugin.py
 270                 divpng = LoadPixmap(cached=True, path=resolveFilename(SCOPE_SKIN_IMAGE, "skin_default/div-h.png"))
 271                 encryption = encrypted and _("Yes") or _("No")
 272                 if bssid == 'hidden...':
 273                         return((essid, bssid, None, None, None, None, divpng))
 274                 else:
 275                         return((essid, bssid, _("Signal: ") + str(signal), _("Max. Bitrate: ") + str(maxrate), _("Encrypted: ") + encryption, _("Interface: ") + str(iface), divpng))
 276
 277         def updateAPList(self):
 278                 self.oldlist = []
 279                 self.oldlist = self.cleanList
 280                 self.newAPList = []
 281                 newList = []
 282                 tmpList = []
 283                 newListIndex = None
 284                 currentListEntry = None
 285                 currentListIndex = None
 286                 newList = self.getAccessPoints(refresh = True)
 287
 288                 for oldentry in self.oldlist:
 289                         if oldentry not in newList:
 290                                 newList.append(oldentry)
 291
 292                 for newentry in newList:
 293                         if newentry[1] == "hidden...":
 294                                 continue
 295                         tmpList.append(newentry)
 296
 297                 if len(tmpList):
 298                         if "hidden..." not in tmpList:
 299                                 tmpList.append( ( _("enter hidden network SSID"), "hidden...", True, self.iface, _("unavailable"), "" ) )
 300
 301                         for entry in tmpList:
 302                                 self.newAPList.append(self.buildEntryComponent( entry[0], entry[1], entry[2], entry[3], entry[4], entry[5] ))
 303
 304                         currentListEntry = self["list"].getCurrent()
 305                         idx = 0
 306                         for entry in self.newAPList:
 307                                 if entry == currentListEntry:
 308                                         newListIndex = idx
 309                                 idx +=1
 310                         self['list'].setList(self.newAPList)
 311                         self["list"].setIndex(newListIndex)
 312                         self["list"].updateList(self.newAPList)
 313                         self.listLenght = len(self.newAPList)
 314                         self.buildWlanList()
 315                         self.setInfo()
 316
 317         def getAccessPoints(self, refresh = False):
 318                 self.APList = []
 319                 self.cleanList = []
 320                 self.w = Wlan(self.iface)
 321                 aps = self.w.getNetworkList()
 322                 if aps is not None:
 323                         print "[NetworkWizard.py] got Accespoints!"
 324                         tmpList = []
 325                         compList = []
 326                         for ap in aps:
 327                                 a = aps[ap]
 328                                 if a['active']:
 329                                         tmpList.append( (a['essid'], a['bssid']) )
 330                                         compList.append( (a['essid'], a['bssid'], a['encrypted'], a['iface'], a['maxrate'], a['signal']) )
 331
 332                         for entry in tmpList:
 333                                 if entry[0] == "":
 334                                         for compentry in compList:
 335                                                 if compentry[1] == entry[1]:
 336                                                         compList.remove(compentry)
 337                         for entry in compList:
 338                                 self.cleanList.append( ( entry[0], entry[1], entry[2], entry[3], entry[4], entry[5] ) )
 339
 340                 if "hidden..." not in self.cleanList:
 341                         self.cleanList.append( ( _("enter hidden network SSID"), "hidden...", True, self.iface, _("unavailable"), "" ) )
 342
 343                 for entry in self.cleanList:
 344                         self.APList.append(self.buildEntryComponent( entry[0], entry[1], entry[2], entry[3], entry[4], entry[5] ))
 345
 346                 if refresh is False:
 347                         self['list'].setList(self.APList)
 348                 self.listLenght = len(self.APList)
 349                 self.setInfo()
 350                 self.rescanTimer.start(5000)
 351                 return self.cleanList
 352
 353         def setInfo(self):
 354                 length = self.getLength()
 355                 if length == 0:
 356                         self["info"].setText(_("No wireless networks found! Please refresh."))
 357                 elif length == 1:
 358                         self["info"].setText(_("1 wireless network found!"))
 359                 else:
 360                         self["info"].setText(str(length)+_(" wireless networks found!"))
 361
 362         def buildWlanList(self):
 363                 self.WlanList = []
 364                 currList = []
 365                 currList = self['list'].list
 366                 for entry in currList:
 367                         self.WlanList.append( (entry[1], entry[0]) )
 368
 369         def getLength(self):
 370                 return self.listLenght
 371
 372         def getWlanList(self):
 373                 return self.WlanList
 374
 375
 376 def WlanStatusScreenMain(session, iface):
 377         session.open(WlanStatus, iface)
 378
 379
 380 def callFunction(iface):
 381         w = Wlan(iface)
 382         i = w.getWirelessInterfaces()
 383         if i:
 384                 if iface in i:
 385                         return WlanStatusScreenMain
 386         return None
 387
 388
 389 def configStrings(iface):
 390         driver = iNetwork.detectWlanModule()
 391         print "Found WLAN-Driver:",driver
 392         if driver  in ('ralink', 'zydas'):
 393                 return "        pre-up /usr/sbin/wpa_supplicant -i"+iface+" -c/etc/wpa_supplicant.conf -B -D"+driver+"\n        post-down wpa_cli terminate"
 394         else:
 395                 if config.plugins.wlan.essid.value == "hidden...":
 396                         return '        pre-up iwconfig '+iface+' essid "'+config.plugins.wlan.hiddenessid.value+'"\n   pre-up /usr/sbin/wpa_supplicant -i'+iface+' -c/etc/wpa_supplicant.conf -B -dd -D'+driver+'\n    post-down wpa_cli terminate'
 397                 else:
 398                         return '        pre-up iwconfig '+iface+' essid "'+config.plugins.wlan.essid.value+'"\n pre-up /usr/sbin/wpa_supplicant -i'+iface+' -c/etc/wpa_supplicant.conf -B -dd -D'+driver+'\n    post-down wpa_cli terminate'
 399
 400 def Plugins(**kwargs):
 401         return PluginDescriptor(name=_("Wireless LAN"), description=_("Connect to a Wireless Network"), where = PluginDescriptor.WHERE_NETWORKSETUP, fnc={"ifaceSupported": callFunction, "configStrings": configStrings, "WlanPluginEntry": lambda x: "Wireless Network Configuartion..."})
 402