remove debug output
[enigma2.git] / lib / python / Components / Network.py
index 6fbd99945e57c2eb71f1147696c9aaf3b66698f7..cf21c6d053e5067c63efbddb0ffaa5fe8dee3d28 100644 (file)
@@ -1,32 +1,33 @@
-import os
-import re
+from os import system, popen, path as os_path, listdir
+from re import compile as re_compile
 from socket import *
+from enigma import eConsoleAppContainer
+from Components.Console import Console
 
 class Network:
        def __init__(self):
                self.ifaces = {}
+               self.configuredInterfaces = []          
                self.nameservers = []
+               self.ethtool_bin = "/usr/sbin/ethtool"
+               self.container = eConsoleAppContainer()
+               self.Console = Console()
                self.getInterfaces()
-               
-       def getInterfaces(self):
-               devicesPattern = re.compile('[a-z]+[0-9]+')
-               
+
+       def getInterfaces(self, callback = None):
+               devicesPattern = re_compile('[a-z]+[0-9]+')
+               self.configuredInterfaces = []
                fp = file('/proc/net/dev', 'r')
                result = fp.readlines()
                fp.close()
                for line in result:
                        try:
                                device = devicesPattern.search(line).group()
-                               self.ifaces[device] = self.getDataForInterface(device)
+                               if device == 'wifi0':
+                                       continue
+                               self.getDataForInterface(device, callback)
                        except AttributeError:
                                pass
-               
-               print "self.ifaces:", self.ifaces
-               self.loadNetworkConfig()
-               #self.writeNetworkConfig()
-               #print ord(' ')
-               #for line in result:
-#                      print ord(line[0])
 
        # helper function
        def regExpMatch(self, pattern, string):
@@ -36,7 +37,7 @@ class Network:
                        return pattern.search(string).group()
                except AttributeError:
                        None
-       
+
        # helper function to convert ips from a sring to a list of ints
        def convertIP(self, ip):
                strIP = ip.split('.')
@@ -45,65 +46,105 @@ class Network:
                        ip.append(int(x))
                return ip
 
-       def getDataForInterface(self, iface):
-               #ipRegexp = '[0-9]{1,2,3}\.[0-9]{1,2,3}\.[0-9]{1,2,3}\.[0-9]{1,2,3}'
+       def IPaddrFinished(self, result, retval, extra_args):
+               (iface, callback ) = extra_args
+               data = { 'up': False, 'dhcp': False, 'preup' : False, 'postdown' : False }
+               globalIPpattern = re_compile("scope global")
                ipRegexp = '[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}'
-               ipLinePattern = re.compile('inet addr:' + ipRegexp)
-               netmaskLinePattern = re.compile('Mask:' + ipRegexp)
-               bcastLinePattern = re.compile('Bcast:' + ipRegexp)
-               ipPattern = re.compile(ipRegexp)
-               upPattern = re.compile('UP ')
-               macPattern = re.compile('[0-9]{2}\:[0-9]{2}\:[0-9]{2}\:[0-9]{2}\:[0-9]{2}\:[0-9]{2}')
-               
-               fp = os.popen("ifconfig " + iface)
-               result = fp.readlines()
-               fp.close()
-               data = { 'up': False, 'dhcp': False }
-               for line in result:
-                       ip = self.regExpMatch(ipPattern, self.regExpMatch(ipLinePattern, line))
+               ipLinePattern = re_compile('inet ' + ipRegexp +'/')
+               ipPattern = re_compile(ipRegexp)
+
+               for line in result.splitlines():
+                       split = line.strip().split(' ',2)
+                       if (split[1] == iface):
+                               if re_search(globalIPpattern, split[2]):
+                                       ip = self.regExpMatch(ipPattern, self.regExpMatch(ipLinePattern, split[2]))
+                                       if ip is not None:
+                                               data['ip'] = self.convertIP(ip)
+               if not data.has_key('ip'):
+                       data['dhcp'] = True
+                       data['ip'] = [0, 0, 0, 0]
+                       data['netmask'] = [0, 0, 0, 0]
+                       data['gateway'] = [0, 0, 0, 0]
+
+               cmd = "ifconfig " + iface
+               self.Console.ePopen(cmd, self.ifconfigFinished, [iface, data, callback])
+
+       def getDataForInterface(self, iface,callback):
+               #get ip out of ip addr, as avahi sometimes overrides it in ifconfig.
+               cmd = "ip -o addr"
+               self.Console.ePopen(cmd, self.IPaddrFinished, [iface,callback])
+
+       def ifconfigFinished(self, result, retval, extra_args ):
+               (iface, data, callback ) = extra_args
+               ipRegexp = '[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}'
+               ipLinePattern = re_compile('inet addr:' + ipRegexp)
+               netmaskLinePattern = re_compile('Mask:' + ipRegexp)
+               bcastLinePattern = re_compile('Bcast:' + ipRegexp)
+               ipPattern = re_compile(ipRegexp)
+               upPattern = re_compile('UP ')
+               macPattern = re_compile('[0-9]{2}\:[0-9]{2}\:[0-9]{2}\:[0-9]{2}\:[0-9]{2}\:[0-9]{2}')
+
+               for line in result.splitlines():
+                       #ip = self.regExpMatch(ipPattern, self.regExpMatch(ipLinePattern, line))
                        netmask = self.regExpMatch(ipPattern, self.regExpMatch(netmaskLinePattern, line))
                        bcast = self.regExpMatch(ipPattern, self.regExpMatch(bcastLinePattern, line))
                        up = self.regExpMatch(upPattern, line)
                        mac = self.regExpMatch(macPattern, line)
-                       if ip is not None:
-                               data['ip'] = self.convertIP(ip)
+                       #if ip is not None:
+                       #       data['ip'] = self.convertIP(ip)
                        if netmask is not None:
                                data['netmask'] = self.convertIP(netmask)
                        if bcast is not None:
                                data['bcast'] = self.convertIP(bcast)
                        if up is not None:
                                data['up'] = True
+                               if iface is not 'lo':
+                                       self.configuredInterfaces.append(iface)
                        if mac is not None:
                                data['mac'] = mac
-               if not data.has_key('ip'):
-                       data['dhcp'] = True
-                       data['ip'] = [192, 168, 1, 2]
-                       data['netmask'] = [255, 255, 255, 0]
-                       data['gateway'] = [192, 168, 1, 1]
-               
-               fp = os.popen("route -n | grep  " + iface)
-               result = fp.readlines()
-               fp.close()                              
-               for line in result:
+
+               cmd = "route -n | grep  " + iface
+               self.Console.ePopen(cmd,self.routeFinished,[iface,data,callback])
+
+       def routeFinished(self, result, retval, extra_args):
+               (iface, data, callback) = extra_args
+               ipRegexp = '[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}'
+               ipPattern = re_compile(ipRegexp)
+               ipLinePattern = re_compile(ipRegexp)
+
+               for line in result.splitlines():
                        print line[0:7]
                        if line[0:7] == "0.0.0.0":
                                gateway = self.regExpMatch(ipPattern, line[16:31])
                                if gateway is not None:
                                        data['gateway'] = self.convertIP(gateway)
-               return data
+
+               for line in result.splitlines(): #get real netmask in case avahi has overridden ifconfig netmask
+                       split = line.strip().split('   ')
+                       if re_search(ipPattern, split[0]):
+                               foundip = self.convertIP(split[0])
+                               if (foundip[0] == data['ip'][0] and foundip[1] == data['ip'][1]):
+                                       if re_search(ipPattern, split[4]):
+                                               mask = self.regExpMatch(ipPattern, self.regExpMatch(ipLinePattern, split[4]))
+                                               if mask is not None:
+                                                       data['netmask'] = self.convertIP(mask)
+               self.ifaces[iface] = data
+               self.loadNetworkConfig(iface,callback)
 
        def writeNetworkConfig(self):
+               self.configuredInterfaces = []
                fp = file('/etc/network/interfaces', 'w')
                fp.write("# automatically generated by enigma 2\n# do NOT change manually!\n\n")
                fp.write("auto lo\n")
                fp.write("iface lo inet loopback\n\n")
-               
                for ifacename, iface in self.ifaces.items():
                        if iface['up'] == True:
                                fp.write("auto " + ifacename + "\n")
+                               self.configuredInterfaces.append(ifacename)
                        if iface['dhcp'] == True:
                                fp.write("iface "+ ifacename +" inet dhcp\n")
-                       else:
+                       if iface['dhcp'] == False:
                                fp.write("iface "+ ifacename +" inet static\n")
                                if iface.has_key('ip'):
                                        print tuple(iface['ip'])
@@ -111,24 +152,22 @@ class Network:
                                        fp.write("      netmask %d.%d.%d.%d\n" % tuple(iface['netmask']))
                                        if iface.has_key('gateway'):
                                                fp.write("      gateway %d.%d.%d.%d\n" % tuple(iface['gateway']))
-                                               
                        if iface.has_key("configStrings"):
                                fp.write("\n" + iface["configStrings"] + "\n")
-                       fp.write("\n")
+                       if iface["preup"] is not False and not iface.has_key("configStrings"):
+                               fp.write(iface["preup"])
+                               fp.write(iface["postdown"])
+                       fp.write("\n")                          
                fp.close()
                self.writeNameserverConfig()
 
-               
        def writeNameserverConfig(self):
                fp = file('/etc/resolv.conf', 'w')
                for nameserver in self.nameservers:
                        fp.write("nameserver %d.%d.%d.%d\n" % tuple(nameserver))
                fp.close()
-               
 
-       def loadNetworkConfig(self):
-               self.loadNameserverConfig()
-               
+       def loadNetworkConfig(self,iface,callback = None):
                interfaces = []
                # parse the interfaces-file
                try:
@@ -137,7 +176,7 @@ class Network:
                        fp.close()
                except:
                        print "[Network.py] interfaces - opening failed"
-                       
+
                ifaces = {}
                currif = ""
                for i in interfaces:
@@ -149,26 +188,45 @@ class Network:
                                        ifaces[currif]["dhcp"] = True
                                else:
                                        ifaces[currif]["dhcp"] = False
-                       if (currif != ""):
+                       if (currif == iface): #read information only for available interfaces
                                if (split[0] == "address"):
                                        ifaces[currif]["address"] = map(int, split[1].split('.'))
+                                       if self.ifaces[currif].has_key("ip"):
+                                               if self.ifaces[currif]["ip"] != ifaces[currif]["address"] and ifaces[currif]["dhcp"] == False:
+                                                       self.ifaces[currif]["ip"] = map(int, split[1].split('.'))
                                if (split[0] == "netmask"):
                                        ifaces[currif]["netmask"] = map(int, split[1].split('.'))
+                                       if self.ifaces[currif].has_key("netmask"):
+                                               if self.ifaces[currif]["netmask"] != ifaces[currif]["netmask"] and ifaces[currif]["dhcp"] == False:
+                                                       self.ifaces[currif]["netmask"] = map(int, split[1].split('.'))
                                if (split[0] == "gateway"):
-                                       ifaces[currif]["gateway"] = map(int, split[1].split('.'))                                                                       
-               
-               print "read interfaces:", ifaces
+                                       ifaces[currif]["gateway"] = map(int, split[1].split('.'))
+                                       if self.ifaces[currif].has_key("gateway"):
+                                               if self.ifaces[currif]["gateway"] != ifaces[currif]["gateway"] and ifaces[currif]["dhcp"] == False:
+                                                       self.ifaces[currif]["gateway"] = map(int, split[1].split('.'))
+                               if (split[0] == "pre-up"):
+                                       if self.ifaces[currif].has_key("preup"):
+                                               self.ifaces[currif]["preup"] = i
+                               if (split[0] == "post-down"):
+                                       if self.ifaces[currif].has_key("postdown"):
+                                               self.ifaces[currif]["postdown"] = i
+
                for ifacename, iface in ifaces.items():
                        if self.ifaces.has_key(ifacename):
                                self.ifaces[ifacename]["dhcp"] = iface["dhcp"]
-               
-               print "self.ifaces after loading:", self.ifaces
+               if len(self.Console.appContainers) == 0:
+                       # load ns only once
+                       self.loadNameserverConfig()
+                       print "read configured interfac:", ifaces
+                       print "self.ifaces after loading:", self.ifaces
+                       if callback is not None:
+                               callback(True)
 
        def loadNameserverConfig(self):
                ipRegexp = "[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}"
-               nameserverPattern = re.compile("nameserver +" + ipRegexp)
-               ipPattern = re.compile(ipRegexp)
-               
+               nameserverPattern = re_compile("nameserver +" + ipRegexp)
+               ipPattern = re_compile(ipRegexp)
+
                resolv = []
                try:
                        fp = file('/etc/resolv.conf', 'r')
@@ -177,40 +235,50 @@ class Network:
                        self.nameservers = []
                except:
                        print "[Network.py] resolv.conf - opening failed"
-               
+
                for line in resolv:
                        if self.regExpMatch(nameserverPattern, line) is not None:
                                ip = self.regExpMatch(ipPattern, line)
                                if ip is not None:
                                        self.nameservers.append(self.convertIP(ip))
-                                       
-               print "nameservers:", self.nameservers          
+
+               print "nameservers:", self.nameservers
 
        def deactivateNetworkConfig(self):
                for iface in self.ifaces.keys():
-                       os.system("ip addr flush " + iface)
-               os.system("/etc/init.d/networking stop")
-               os.system("killall -9 udhcpc")
-               os.system("rm /var/run/udhcpc*")
+                       system("ip addr flush " + iface)
+               system("/etc/init.d/networking stop")
+               system("killall -9 udhcpc")
+               system("rm /var/run/udhcpc*")
 
        def activateNetworkConfig(self):
-               os.system("/etc/init.d/networking start")
-               
+               system("/etc/init.d/networking start")
+               self.getInterfaces()
+
        def getNumberOfAdapters(self):
                return len(self.ifaces)
-       
+
+       def getFriendlyAdapterName(self, x):
+               # maybe this needs to be replaced by an external list.
+               friendlyNames = {
+                       "eth0": _("Integrated Ethernet"),
+                       "wlan0": _("Wireless"),
+                       "ath0": _("Integrated Wireless")
+               }
+               return friendlyNames.get(x, x) # when we have no friendly name, use adapter name
+
        def getAdapterName(self, iface):
                return iface
-       
+
        def getAdapterList(self):
                return self.ifaces.keys()
-       
+
        def getAdapterAttribute(self, iface, attribute):
                if self.ifaces.has_key(iface):
                        if self.ifaces[iface].has_key(attribute):
                                return self.ifaces[iface][attribute]
                return None
-       
+
        def setAdapterAttribute(self, iface, attribute, value):
                print "setting for adapter", iface, "attribute", attribute, " to value", value
                if self.ifaces.has_key(iface):
@@ -220,27 +288,130 @@ class Network:
                if self.ifaces.has_key(iface):
                        if self.ifaces[iface].has_key(attribute):
                                del self.ifaces[iface][attribute]
-                               
+
        def getNameserverList(self):
-               return self.nameservers
-       
+               if len(self.nameservers) == 0:
+                       return [[0, 0, 0, 0], [0, 0, 0, 0]]
+               else: 
+                       return self.nameservers
+
        def clearNameservers(self):
                self.nameservers = []
-       
+
        def addNameserver(self, nameserver):
                if nameserver not in self.nameservers:
                        self.nameservers.append(nameserver)
-               
+
        def removeNameserver(self, nameserver):
                if nameserver in self.nameservers:
                        self.nameservers.remove(nameserver)
-                       
+
        def changeNameserver(self, oldnameserver, newnameserver):
                if oldnameserver in self.nameservers:
                        for i in range(len(self.nameservers)):
                                if self.nameservers[i] == oldnameserver:
                                        self.nameservers[i] = newnameserver
 
+       def writeDefaultNetworkConfig(self,mode='lan'):
+               fp = file('/etc/network/interfaces', 'w')
+               fp.write("# automatically generated by enigma 2\n# do NOT change manually!\n\n")
+               fp.write("auto lo\n")
+               fp.write("iface lo inet loopback\n\n")
+               if mode == 'wlan':
+                       fp.write("auto wlan0\n")
+                       fp.write("iface wlan0 inet dhcp\n")
+               if mode == 'wlan-mpci':
+                       fp.write("auto ath0\n")
+                       fp.write("iface ath0 inet dhcp\n")
+               if mode == 'lan':
+                       fp.write("auto eth0\n")
+                       fp.write("iface eth0 inet dhcp\n")
+               fp.write("\n")
+               fp.close()
+
+       def resetNetworkConfig(self,mode='lan'):
+               self.deactivateNetworkConfig()
+               self.writeDefaultNetworkConfig(mode)
+               if mode == 'wlan':
+                       system("ifconfig eth0 down")
+                       system("ifconfig ath0 down")
+                       system("ifconfig wlan0 up")
+               if mode == 'wlan-mpci':
+                       system("ifconfig eth0 down")
+                       system("ifconfig wlan0 down")
+                       system("ifconfig ath0 up")              
+               if mode == 'lan':                       
+                       system("ifconfig eth0 up")
+                       system("ifconfig wlan0 down")
+                       system("ifconfig ath0 down")
+               self.getInterfaces()    
+
+       def checkNetworkState(self):
+                # www.dream-multimedia-tv.de, www.heise.de, www.google.de
+               return system("ping -c 1 82.149.226.170") == 0 or \
+                       system("ping -c 1 193.99.144.85") == 0 or \
+                       system("ping -c 1 209.85.135.103") == 0
+
+       def restartNetwork(self):
+               iNetwork.deactivateNetworkConfig()
+               iNetwork.activateNetworkConfig()
+
+       def getLinkState(self,iface,callback):
+               self.dataAvail = callback
+               cmd = self.ethtool_bin + " " + iface
+               self.container.appClosed.append(self.cmdFinished)
+               self.container.dataAvail.append(callback)
+               self.container.execute(cmd)
+
+       def cmdFinished(self,retval):
+               self.container.appClosed.remove(self.cmdFinished)
+               self.container.dataAvail.remove(self.dataAvail)
+
+       def stopContainer(self):
+               self.container.kill()
+               
+       def ContainerRunning(self):
+               return self.container.running()
+
+       def checkforInterface(self,iface):
+               if self.getAdapterAttribute(iface, 'up') is True:
+                       return True
+               else:
+                       ret=system("ifconfig " + iface + " up")
+                       system("ifconfig " + iface + " down")
+                       if ret == 0:
+                               return True
+                       else:
+                               return False
+
+       def checkDNSLookup(self):
+               return system("nslookup www.dream-multimedia-tv.de") == 0 or \
+                       system("nslookup www.heise.de") == 0 or \
+                       system("nslookup www.google.de")
+
+       def deactivateInterface(self,iface):
+               system("ifconfig " + iface + " down")
+
+       def detectWlanModule(self):
+               self.wlanmodule = None
+               rt73_dir = "/sys/bus/usb/drivers/rt73/"
+               zd1211b_dir = "/sys/bus/usb/drivers/zd1211b/"
+               madwifi_dir = "/sys/bus/pci/drivers/ath_pci/"
+               if os_path.exists(madwifi_dir):
+                       files = listdir(madwifi_dir)
+                       if len(files) >= 1:
+                               self.wlanmodule = 'madwifi'
+               if os_path.exists(rt73_dir):
+                       rtfiles = listdir(rt73_dir)
+                       if len(rtfiles) == 2:
+                               self.wlanmodule = 'ralink'
+               if os_path.exists(zd1211b_dir):
+                       zdfiles = listdir(zd1211b_dir)
+                       if len(zdfiles) == 1:
+                               self.wlanmodule = 'zydas'
+               return self.wlanmodule
+       
+       
 iNetwork = Network()
 
 def InitNetwork():