code.vuplus.com Git - vuplus_dvbapp/blob - lib/python/Components/Network.py

   1 import os
   2 import re
   3 from socket import *
   4 from enigma import eTimer, eConsoleAppContainer
   5
   6 class Network:
   7         def __init__(self):
   8                 self.ifaces = {}
   9                 self.nameservers = []
  10                 self.getInterfaces()
  11                 self.ethtool_bin = "/usr/sbin/ethtool"
  12                 self.container = eConsoleAppContainer()
  13
  14         def getInterfaces(self):
  15                 devicesPattern = re.compile('[a-z]+[0-9]+')
  16
  17                 fp = file('/proc/net/dev', 'r')
  18                 result = fp.readlines()
  19                 fp.close()
  20                 for line in result:
  21                         try:
  22                                 device = devicesPattern.search(line).group()
  23                                 self.ifaces[device] = self.getDataForInterface(device)
  24                                 # Show only UP Interfaces in E2
  25                                 if self.getAdapterAttribute(device, 'up') is False:
  26                                         del self.ifaces[device]
  27                         except AttributeError:
  28                                 pass
  29
  30                 print "self.ifaces:", self.ifaces
  31                 self.loadNetworkConfig()
  32                 #self.writeNetworkConfig()
  33                 #print ord(' ')
  34                 #for line in result:
  35                 #       print ord(line[0])
  36
  37         # helper function
  38         def regExpMatch(self, pattern, string):
  39                 if string is None:
  40                         return None
  41                 try:
  42                         return pattern.search(string).group()
  43                 except AttributeError:
  44                         None
  45
  46         # helper function to convert ips from a sring to a list of ints
  47         def convertIP(self, ip):
  48                 strIP = ip.split('.')
  49                 ip = []
  50                 for x in strIP:
  51                         ip.append(int(x))
  52                 return ip
  53
  54         def getDataForInterface(self, iface):
  55                 #ipRegexp = '[0-9]{1,2,3}\.[0-9]{1,2,3}\.[0-9]{1,2,3}\.[0-9]{1,2,3}'
  56                 ipRegexp = '[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}'
  57                 ipLinePattern = re.compile('inet addr:' + ipRegexp)
  58                 netmaskLinePattern = re.compile('Mask:' + ipRegexp)
  59                 bcastLinePattern = re.compile('Bcast:' + ipRegexp)
  60                 ipPattern = re.compile(ipRegexp)
  61                 upPattern = re.compile('UP ')
  62                 macPattern = re.compile('[0-9]{2}\:[0-9]{2}\:[0-9]{2}\:[0-9]{2}\:[0-9]{2}\:[0-9]{2}')
  63
  64                 fp = os.popen("ifconfig " + iface)
  65                 result = fp.readlines()
  66                 fp.close()
  67                 data = { 'up': False, 'dhcp': False }
  68                 for line in result:
  69                         ip = self.regExpMatch(ipPattern, self.regExpMatch(ipLinePattern, line))
  70                         netmask = self.regExpMatch(ipPattern, self.regExpMatch(netmaskLinePattern, line))
  71                         bcast = self.regExpMatch(ipPattern, self.regExpMatch(bcastLinePattern, line))
  72                         up = self.regExpMatch(upPattern, line)
  73                         mac = self.regExpMatch(macPattern, line)
  74                         if ip is not None:
  75                                 data['ip'] = self.convertIP(ip)
  76                         if netmask is not None:
  77                                 data['netmask'] = self.convertIP(netmask)
  78                         if bcast is not None:
  79                                 data['bcast'] = self.convertIP(bcast)
  80                         if up is not None:
  81                                 data['up'] = True
  82                         if mac is not None:
  83                                 data['mac'] = mac
  84                 if not data.has_key('ip'):
  85                         data['dhcp'] = True
  86                         data['ip'] = [192, 168, 1, 2]
  87                         data['netmask'] = [255, 255, 255, 0]
  88                         data['gateway'] = [192, 168, 1, 1]
  89
  90                 fp = os.popen("route -n | grep  " + iface)
  91                 result = fp.readlines()
  92                 fp.close()
  93                 for line in result:
  94                         print line[0:7]
  95                         if line[0:7] == "0.0.0.0":
  96                                 gateway = self.regExpMatch(ipPattern, line[16:31])
  97                                 if gateway is not None:
  98                                         data['gateway'] = self.convertIP(gateway)
  99                 return data
 100
 101         def writeNetworkConfig(self):
 102                 fp = file('/etc/network/interfaces', 'w')
 103                 fp.write("# automatically generated by enigma 2\n# do NOT change manually!\n\n")
 104                 fp.write("auto lo\n")
 105                 fp.write("iface lo inet loopback\n\n")
 106
 107                 for ifacename, iface in self.ifaces.items():
 108                         if iface['up'] == True:
 109                                 fp.write("auto " + ifacename + "\n")
 110                         if iface['dhcp'] == True:
 111                                 fp.write("iface "+ ifacename +" inet dhcp\n")
 112                         else:
 113                                 fp.write("iface "+ ifacename +" inet static\n")
 114                                 if iface.has_key('ip'):
 115                                         print tuple(iface['ip'])
 116                                         fp.write("      address %d.%d.%d.%d\n" % tuple(iface['ip']))
 117                                         fp.write("      netmask %d.%d.%d.%d\n" % tuple(iface['netmask']))
 118                                         if iface.has_key('gateway'):
 119                                                 fp.write("      gateway %d.%d.%d.%d\n" % tuple(iface['gateway']))
 120
 121                         if iface.has_key("configStrings"):
 122                                 fp.write("\n" + iface["configStrings"] + "\n")
 123                         fp.write("\n")
 124                 fp.close()
 125                 self.writeNameserverConfig()
 126
 127         def writeNameserverConfig(self):
 128                 fp = file('/etc/resolv.conf', 'w')
 129                 for nameserver in self.nameservers:
 130                         fp.write("nameserver %d.%d.%d.%d\n" % tuple(nameserver))
 131                 fp.close()
 132
 133         def loadNetworkConfig(self):
 134                 self.loadNameserverConfig()
 135                 interfaces = []
 136                 # parse the interfaces-file
 137                 try:
 138                         fp = file('/etc/network/interfaces', 'r')
 139                         interfaces = fp.readlines()
 140                         fp.close()
 141                 except:
 142                         print "[Network.py] interfaces - opening failed"
 143
 144                 ifaces = {}
 145                 currif = ""
 146                 for i in interfaces:
 147                         split = i.strip().split(' ')
 148                         if (split[0] == "iface"):
 149                                 currif = split[1]
 150                                 ifaces[currif] = {}
 151                                 if (len(split) == 4 and split[3] == "dhcp"):
 152                                         ifaces[currif]["dhcp"] = True
 153                                 else:
 154                                         ifaces[currif]["dhcp"] = False
 155                         if (currif != ""):
 156                                 if (split[0] == "address"):
 157                                         ifaces[currif]["address"] = map(int, split[1].split('.'))
 158                                 if (split[0] == "netmask"):
 159                                         ifaces[currif]["netmask"] = map(int, split[1].split('.'))
 160                                 if (split[0] == "gateway"):
 161                                         ifaces[currif]["gateway"] = map(int, split[1].split('.'))
 162
 163                 print "read interfaces:", ifaces
 164                 for ifacename, iface in ifaces.items():
 165                         if self.ifaces.has_key(ifacename):
 166                                 self.ifaces[ifacename]["dhcp"] = iface["dhcp"]
 167
 168                 print "self.ifaces after loading:", self.ifaces
 169
 170         def loadNameserverConfig(self):
 171                 ipRegexp = "[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}"
 172                 nameserverPattern = re.compile("nameserver +" + ipRegexp)
 173                 ipPattern = re.compile(ipRegexp)
 174
 175                 resolv = []
 176                 try:
 177                         fp = file('/etc/resolv.conf', 'r')
 178                         resolv = fp.readlines()
 179                         fp.close()
 180                         self.nameservers = []
 181                 except:
 182                         print "[Network.py] resolv.conf - opening failed"
 183
 184                 for line in resolv:
 185                         if self.regExpMatch(nameserverPattern, line) is not None:
 186                                 ip = self.regExpMatch(ipPattern, line)
 187                                 if ip is not None:
 188                                         self.nameservers.append(self.convertIP(ip))
 189
 190                 print "nameservers:", self.nameservers
 191
 192         def deactivateNetworkConfig(self):
 193                 for iface in self.ifaces.keys():
 194                         os.system("ip addr flush " + iface)
 195                 os.system("/etc/init.d/networking stop")
 196                 os.system("killall -9 udhcpc")
 197                 os.system("rm /var/run/udhcpc*")
 198
 199         def activateNetworkConfig(self):
 200                 os.system("/etc/init.d/networking start")
 201                 self.getInterfaces()
 202
 203         def getNumberOfAdapters(self):
 204                 return len(self.ifaces)
 205
 206         def getFriendlyAdapterName(self, x):
 207                 # maybe this needs to be replaced by an external list.
 208                 friendlyNames = {
 209                         "eth0": _("Integrated Ethernet"),
 210                         "wlan0": _("Wireless")
 211                 }
 212                 return friendlyNames.get(x, x) # when we have no friendly name, use adapter name
 213
 214         def getAdapterName(self, iface):
 215                 return iface
 216
 217         def getAdapterList(self):
 218                 return self.ifaces.keys()
 219
 220         def getAdapterAttribute(self, iface, attribute):
 221                 if self.ifaces.has_key(iface):
 222                         if self.ifaces[iface].has_key(attribute):
 223                                 return self.ifaces[iface][attribute]
 224                 return None
 225
 226         def setAdapterAttribute(self, iface, attribute, value):
 227                 print "setting for adapter", iface, "attribute", attribute, " to value", value
 228                 if self.ifaces.has_key(iface):
 229                         self.ifaces[iface][attribute] = value
 230
 231         def removeAdapterAttribute(self, iface, attribute):
 232                 if self.ifaces.has_key(iface):
 233                         if self.ifaces[iface].has_key(attribute):
 234                                 del self.ifaces[iface][attribute]
 235
 236         def getNameserverList(self):
 237                 if len(self.nameservers) == 0:
 238                         return [[0, 0, 0, 0], [0, 0, 0, 0]]
 239                 else:
 240                         return self.nameservers
 241
 242         def clearNameservers(self):
 243                 self.nameservers = []
 244
 245         def addNameserver(self, nameserver):
 246                 if nameserver not in self.nameservers:
 247                         self.nameservers.append(nameserver)
 248
 249         def removeNameserver(self, nameserver):
 250                 if nameserver in self.nameservers:
 251                         self.nameservers.remove(nameserver)
 252
 253         def changeNameserver(self, oldnameserver, newnameserver):
 254                 if oldnameserver in self.nameservers:
 255                         for i in range(len(self.nameservers)):
 256                                 if self.nameservers[i] == oldnameserver:
 257                                         self.nameservers[i] = newnameserver
 258
 259         def writeDefaultNetworkConfig(self,mode='lan'):
 260                 fp = file('/etc/network/interfaces', 'w')
 261                 fp.write("# automatically generated by enigma 2\n# do NOT change manually!\n\n")
 262                 fp.write("auto lo\n")
 263                 fp.write("iface lo inet loopback\n\n")
 264                 if mode == 'wlan':
 265                         fp.write("auto wlan0\n")
 266                         fp.write("iface wlan0 inet dhcp\n")
 267                 else:
 268                         fp.write("auto eth0\n")
 269                         fp.write("iface eth0 inet dhcp\n")
 270                 fp.write("\n")
 271                 fp.close()
 272
 273         def resetNetworkConfig(self,mode='lan'):
 274                 self.deactivateNetworkConfig()
 275                 self.writeDefaultNetworkConfig(mode)
 276                 if mode == 'wlan':
 277                         os.system("ifconfig eth0 down")
 278                         os.system("ifconfig wlan0 up")
 279                 else:
 280                         os.system("ifconfig eth0 up")
 281                         os.system("ifconfig wlan0 down")
 282
 283         def checkNetworkState(self):
 284                 ret=os.system("ping -c 1 www.dream-multimedia-tv.de")
 285                 if ret == 0:
 286                         return True
 287                 else:
 288                         return False
 289
 290         def restartNetwork(self):
 291                 iNetwork.deactivateNetworkConfig()
 292                 iNetwork.activateNetworkConfig()
 293
 294         def getLinkState(self,iface,callback):
 295                 self.dataAvail = callback
 296                 cmd = self.ethtool_bin + " " + iface
 297                 self.container.appClosed.get().append(self.cmdFinished)
 298                 self.container.dataAvail.get().append(callback)
 299                 self.container.execute(cmd)
 300
 301         def cmdFinished(self,retval):
 302                 self.container.appClosed.get().remove(self.cmdFinished)
 303                 self.container.dataAvail.get().remove(self.dataAvail)
 304
 305 iNetwork = Network()
 306
 307 def InitNetwork():
 308         pass