X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/ac896c99e4cdcaf8090bdb2b76763d84c2d5191c..e7b3db898270e6a6a252eee7697302fb0c1f7d38:/lib/python/Components/Network.py diff --git a/lib/python/Components/Network.py b/lib/python/Components/Network.py index ab89d3c6..37af0c98 100644 --- a/lib/python/Components/Network.py +++ b/lib/python/Components/Network.py @@ -1,49 +1,249 @@ -from config import * +from Components.config import config, ConfigYesNo, ConfigIP, NoSave, ConfigSubsection, ConfigMAC import os +import re +from socket import * class Network: def __init__(self): - pass + self.ifaces = {} + self.nameservers = [] + self.getInterfaces() - def setIPAddress(self, ip): - print ip - os.system("echo ifconfig eth0 %d.%d.%d.%d" % (ip[0], ip[1], ip[2], ip[3])) + def getInterfaces(self): + devicesPattern = re.compile('[a-z]+[0-9]+') + + 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) + except AttributeError: + pass + + print "self.ifaces:", self.ifaces + self.loadNetworkConfig() + #self.writeNetworkConfig() + #print ord(' ') + #for line in result: +# print ord(line[0]) - def setIPGateway(self, ip): - os.system("echo route add default gw %d.%d.%d.%d" % (ip[0], ip[1], ip[2], ip[3])) + # 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 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 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 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 = 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)) + 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] -def InitNetwork(): - config.network = ConfigSubsection() - config.network.ip = configElement("config.network.ip", configSequence, [192,168,1,45], (".")) - config.network.gateway = configElement("config.network.gateway", configSequence, [192,168,1,3], (".")) - config.network.dns = configElement("config.network.dns", configSequence, [192,168,1,3], (".")) - config.network.mac = configElement("config.network.mac", configSequence, [00,11,22,33,44,55], (":")) + fp = os.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.tmp', '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") + else: + 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.tmp', '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) + 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 setMACAddress(configElement): - iNetwork.setMACAddress(configElement.value) + 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(): + os.system("ip addr flush" + iface) + os.system("/etc/init.d/networking stop") + os.system("killall -9 udhcpc") + os.system("rm /var/run/udhcpc*") + + def activateNetworkConfig(self): + os.system("/etc/init.d/networking start") + + def getNumberOfAdapters(self): + return len(self.ifaces) + + 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): + self.ifaces[iface][attribute] = value - # 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 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): + 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 + +iNetwork = Network() + +def InitNetwork(): + pass