X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/b2c9f6632fac847577a52e23aa56a1f0816b1927..f06af856d2a0210bf7931f800ecb7ad13fbf659a:/lib/python/Components/Network.py diff --git a/lib/python/Components/Network.py b/lib/python/Components/Network.py index 3740e51b..3abfc34c 100644 --- a/lib/python/Components/Network.py +++ b/lib/python/Components/Network.py @@ -1,49 +1,375 @@ -from config import * - -import os +from os import system, popen +from re import compile as re_compile +from socket import * +from enigma import eConsoleAppContainer class Network: def __init__(self): - pass - - def setIPAddress(self, ip): - print ip - os.system("echo ifconfig eth0 %d.%d.%d.%d" % (ip[0], ip[1], ip[2], ip[3])) + self.ifaces = {} + self.configuredInterfaces = {} + self.nameservers = [] + self.getInterfaces() + self.ethtool_bin = "/usr/sbin/ethtool" + self.container = eConsoleAppContainer() - def setIPGateway(self, ip): - os.system("echo route add default gw %d.%d.%d.%d" % (ip[0], ip[1], ip[2], ip[3])) + def getInterfaces(self): + devicesPattern = re_compile('[a-z]+[0-9]+') - def setIPNameserver(self, ip): - resolvconf = file('/etc/resolv.conf', 'w') - resolvconf.write("nameserver %d.%d.%d.%d" % (ip[0], ip[1], ip[2], ip[3])) - resolvconf.close() - - def setMACAddress(self, mac): - os.system("echo ifconfig eth0 %02x:%02x:%02x:%02x:%02x:%02x" % (mac[0], mac[1], mac[2], mac[3], mac[4], mac[5])) + 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) + # Show only UP Interfaces in E2 + #if self.getAdapterAttribute(device, 'up') is False: + # del self.ifaces[device] + 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): + if string is None: + return None + try: + 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('.') + ip = [] + for x in strIP: + 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}' + 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}') -def InitNetwork(): - config.network = ConfigSubsection() - config.network.ip = configElement("config.network.ip", configSequence, [192,168,1,45], (("."), 3)) - config.network.gateway = configElement("config.network.gateway", configSequence, [192,168,1,3], (("."), 3)) - config.network.dns = configElement("config.network.dns", configSequence, [192,168,1,3], (("."), 3)) - config.network.mac = configElement("config.network.mac", configSequence, [00,11,22,33,44,55], ((":"), 2)) + fp = 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)) + 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 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 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 = popen("route -n | grep " + iface) + result = fp.readlines() + fp.close() + for line in result: + 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 - iNetwork = Network() + def writeNetworkConfig(self): + 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") + if iface['dhcp'] == True: + fp.write("iface "+ ifacename +" inet dhcp\n") + if iface['dhcp'] == False: + fp.write("iface "+ ifacename +" inet static\n") + if iface.has_key('ip'): + print tuple(iface['ip']) + fp.write(" address %d.%d.%d.%d\n" % tuple(iface['ip'])) + 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") + fp.close() + self.writeNameserverConfig() - def setIPAddress(configElement): - iNetwork.setIPAddress(configElement.value) + 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 setIPGateway(configElement): - iNetwork.setIPGateway(configElement.value) + def loadNetworkConfig(self): + self.loadNameserverConfig() + interfaces = [] + # parse the interfaces-file + try: + fp = file('/etc/network/interfaces', 'r') + interfaces = fp.readlines() + fp.close() + except: + print "[Network.py] interfaces - opening failed" + + ifaces = {} + currif = "" + for i in interfaces: + split = i.strip().split(' ') + if (split[0] == "iface"): + currif = split[1] + ifaces[currif] = {} + if (len(split) == 4 and split[3] == "dhcp"): + ifaces[currif]["dhcp"] = True + else: + ifaces[currif]["dhcp"] = False + if (currif != ""): + if (split[0] == "address"): + ifaces[currif]["address"] = map(int, split[1].split('.')) + if (split[0] == "netmask"): + ifaces[currif]["netmask"] = map(int, split[1].split('.')) + if (split[0] == "gateway"): + ifaces[currif]["gateway"] = map(int, split[1].split('.')) - def setIPNameserver(configElement): - iNetwork.setIPNameserver(configElement.value) + self.configuredInterfaces = ifaces + print "read interfaces:", ifaces + 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 + + 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) + + resolv = [] + try: + fp = file('/etc/resolv.conf', 'r') + resolv = fp.readlines() + fp.close() + 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 + + def deactivateNetworkConfig(self): + for iface in self.ifaces.keys(): + system("ip addr flush " + iface) + system("/etc/init.d/networking stop") + system("killall -9 udhcpc") + system("rm /var/run/udhcpc*") + + def activateNetworkConfig(self): + 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 setMACAddress(configElement): - iNetwork.setMACAddress(configElement.value) + def getAdapterList(self): + return self.ifaces.keys() - # this will call the "setup-val" initial - config.network.ip.addNotifier(setIPAddress) - config.network.gateway.addNotifier(setIPGateway) - config.network.dns.addNotifier(setIPNameserver) - config.network.mac.addNotifier(setMACAddress) \ No newline at end of file + 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): + self.ifaces[iface][attribute] = value + + def removeAdapterAttribute(self, iface, attribute): + if self.ifaces.has_key(iface): + if self.ifaces[iface].has_key(attribute): + del self.ifaces[iface][attribute] + + def getNameserverList(self): + 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 eth0 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): + ok_counter = 0 + ret1=system("ping -c 1 www.dream-multimedia-tv.de") + if ret1 != 0: + ok_counter = ok_counter + 1 + ret2=system("ping -c 1 www.heise.de") + if ret2 != 0: + ok_counter = ok_counter + 1 + ret3=system("ping -c 1 www.google.de") + if ret2 != 0: + ok_counter = ok_counter + 1 + if ok_counter <= 2: + return True + else: + return False + + def restartNetwork(self): + iNetwork.deactivateNetworkConfig() + iNetwork.activateNetworkConfig() + + def getLinkState(self,iface,callback): + self.dataAvail = callback + cmd = self.ethtool_bin + " " + iface + self.container.appClosed.get().append(self.cmdFinished) + self.container.dataAvail.get().append(callback) + self.container.execute(cmd) + + def cmdFinished(self,retval): + self.container.appClosed.get().remove(self.cmdFinished) + self.container.dataAvail.get().remove(self.dataAvail) + + 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): + ok_counter = 0 + ret1=system("nslookup www.dream-multimedia-tv.de") + if ret1 != 0: + ok_counter = ok_counter + 1 + ret2=system("nslookup www.heise.de") + if ret2 != 0: + ok_counter = ok_counter + 1 + ret3=system("nslookup www.google.de") + if ret2 != 0: + ok_counter = ok_counter + 1 + if ok_counter <= 2: + return True + else: + return False + + 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/" + if os_path.exists(rt73_dir): + files = [] + files = listdir(rt73_dir) + if len(files) >= 1: + self.wlanmodule = 'ralink' + if os_path.exists(zd1211b_dir): + files = [] + files = listdir(zd1211b_dir) + if len(files) != 0: + self.wlanmodule = 'zydas' + return self.wlanmodule + + +iNetwork = Network() + +def InitNetwork(): + pass