lib/python/Plugins/SystemPlugins/NFIFlash/flasher.py

   1 from Screens.Screen import Screen
   2 from Screens.MessageBox import MessageBox
   3 from Screens.ChoiceBox import ChoiceBox
   4 from Screens.Standby import TryQuitMainloop
   5 from Components.ActionMap import ActionMap
   6 from Components.Sources.StaticText import StaticText
   7 from Components.Sources.Progress import Progress
   8 from Components.Sources.Boolean import Boolean
   9 from Components.Label import Label
  10 from Components.FileList import FileList
  11 from Components.Task import Task, Job, JobManager
  12 from Tools.Directories import fileExists
  13 from Tools.HardwareInfo import HardwareInfo
  14 from os import system
  15 from enigma import eConsoleAppContainer
  16 import re
  17
  18 class writeNAND(Task):
  19         def __init__(self,job,param,box):
  20                 Task.__init__(self,job, _("Writing image file to NAND Flash"))
  21                 self.setTool("/usr/lib/enigma2/python/Plugins/SystemPlugins/NFIFlash/mywritenand")
  22                 if box == "dm7025":
  23                         self.end = 256
  24                 elif box[:5] == "dm800":
  25                         self.end = 512
  26                 if box == "dm8000":
  27                         self.setTool("/usr/lib/enigma2/python/Plugins/SystemPlugins/NFIFlash/dm8000_writenand")
  28                 self.args += param
  29                 self.weighting = 1
  30
  31         def processOutput(self, data):
  32                 print "[writeNand] " + data
  33                 if data == "." or data.endswith(" ."):
  34                         self.progress += 1
  35                 elif data.find("*** done!") > 0:
  36                         print "data.found done"
  37                         self.setProgress(self.end)
  38                 else:
  39                         self.output_line = data
  40
  41 class NFISummary(Screen):
  42         skin = """
  43         <screen position="0,0" size="132,64">
  44                 <widget source="title" render="Label" position="2,0" size="120,14" valign="center" font="Regular;12" />
  45                 <widget source="content" render="Label" position="2,14" size="120,34" font="Regular;12" transparent="1" zPosition="1"  />
  46                 <widget source="job_progresslabel" render="Label" position="66,50" size="60,14" font="Regular;12" transparent="1" halign="right" zPosition="0" />
  47                 <widget source="job_progressbar" render="Progress" position="2,50" size="66,14" borderWidth="1" />
  48         </screen>"""
  49
  50         def __init__(self, session, parent):
  51                 Screen.__init__(self, session, parent)
  52                 self["title"] = StaticText(_("Image flash utility"))
  53                 self["content"] = StaticText(_("Please select .NFI flash image file from medium"))
  54                 self["job_progressbar"] = Progress()
  55                 self["job_progresslabel"] = StaticText("")
  56
  57         def setText(self, text):
  58                 self["content"].setText(text)
  59
  60 class NFIFlash(Screen):
  61         skin = """
  62                 <screen name="NFIFlash" position="90,95" size="560,420" title="Image flash utility">
  63                         <ePixmap pixmap="skin_default/buttons/green.png" position="140,0" zPosition="0" size="140,40" transparent="1" alphatest="on" />
  64                         <ePixmap pixmap="skin_default/buttons/yellow.png" position="280,0" zPosition="0" size="140,40" transparent="1" alphatest="on" />
  65                         <ePixmap pixmap="skin_default/buttons/blue.png" position="420,0" zPosition="0" size="140,40" transparent="1" alphatest="on" />
  66                         <widget source="key_green" render="Label" position="140,0" zPosition="1" size="140,40" font="Regular;20" valign="center" halign="center" backgroundColor="#1f771f" transparent="1" />
  67                         <widget source="key_yellow" render="Label" position="280,0" zPosition="1" size="140,40" font="Regular;20" valign="center" halign="center" backgroundColor="#a08500" transparent="1" />
  68                         <widget source="key_blue" render="Label" position="420,0" zPosition="1" size="140,40" font="Regular;20" valign="center" halign="center" backgroundColor="#18188b" transparent="1" />
  69                         <widget source="listlabel" render="Label" position="16,44" size="200,21" valign="center" font="Regular;18" />
  70                         <widget name="filelist" position="0,68" size="260,260" scrollbarMode="showOnDemand" />
  71                         <widget source="infolabel" render="Label" position="270,44" size="280,284" font="Regular;16" />
  72                         <widget source="job_progressbar" render="Progress" position="10,374" size="540,26" borderWidth="1" backgroundColor="#254f7497" />
  73                         <widget source="job_progresslabel" render="Label" position="180,378" zPosition="2" font="Regular;18" halign="center" transparent="1" size="200,22" foregroundColor="#000000" />
  74                         <widget source="statusbar" render="Label" position="10,404" size="540,16" font="Regular;16" foregroundColor="#cccccc" />
  75                 </screen>"""
  76
  77         def __init__(self, session, cancelable = True, close_on_finish = False):
  78                 self.skin = NFIFlash.skin
  79                 Screen.__init__(self, session)
  80
  81                 self["job_progressbar"] = Progress()
  82                 self["job_progresslabel"] = StaticText("")
  83
  84                 self["finished"] = Boolean()
  85
  86                 self["infolabel"] = StaticText("")
  87                 self["statusbar"] = StaticText(_("Please select .NFI flash image file from medium"))
  88                 self["listlabel"] = StaticText(_("select .NFI flash file")+":")
  89
  90                 self["key_green"] = StaticText()
  91                 self["key_yellow"] = StaticText()
  92                 self["key_blue"] = StaticText()
  93
  94                 self["actions"] = ActionMap(["OkCancelActions", "ColorActions", "DirectionActions"],
  95                 {
  96                         "green": self.ok,
  97                         "yellow": self.reboot,
  98                         "ok": self.ok,
  99                         "left": self.left,
 100                         "right": self.right,
 101                         "up": self.up,
 102                         "down": self.down
 103                 }, -1)
 104
 105                 currDir = "/media/usb/"
 106                 self.filelist = FileList(currDir, matchingPattern = "^.*\.(nfi|NFI)")
 107                 self["filelist"] = self.filelist
 108                 self.nfifile = ""
 109                 self.md5sum = ""
 110                 self.job = None
 111                 self.box = HardwareInfo().get_device_name()
 112
 113         def closeCB(self):
 114                 if ( self.job is None or self.job.status is not self.job.IN_PROGRESS ) and not self.no_autostart:
 115                         self.close()
 116                 #else:
 117                         #if self.cancelable:
 118                                 #self.cancel()
 119
 120         def up(self):
 121                 self["filelist"].up()
 122                 self.check_for_NFO()
 123
 124         def down(self):
 125                 self["filelist"].down()
 126                 self.check_for_NFO()
 127
 128         def right(self):
 129                 self["filelist"].pageDown()
 130                 self.check_for_NFO()
 131
 132         def left(self):
 133                 self["filelist"].pageUp()
 134                 self.check_for_NFO()
 135
 136         def check_for_NFO(self):
 137                 self.session.summary.setText(self["filelist"].getFilename())
 138                 if self["filelist"].getFilename() is None:
 139                         return
 140                 if self["filelist"].getCurrentDirectory() is not None:
 141                         self.nfifile = self["filelist"].getCurrentDirectory()+self["filelist"].getFilename()
 142
 143                 if self.nfifile.upper().endswith(".NFI"):
 144                         self["key_green"].text = _("Flash")
 145                         nfofilename = self.nfifile[0:-3]+"nfo"
 146                         if fileExists(nfofilename):
 147                                 nfocontent = open(nfofilename, "r").read()
 148                                 self["infolabel"].text = nfocontent
 149                                 pos = nfocontent.find("md5sum")
 150                                 if pos > 0:
 151                                         self.md5sum = nfofilename
 152                                 else:
 153                                         self.md5sum = ""
 154                         else:
 155                                 self["infolabel"].text = _("No details for this image file") + ":\n" + self["filelist"].getFilename()
 156                                 self.md5sum = ""
 157                 else:
 158                         self["infolabel"].text = ""
 159                         self["key_green"].text = ""
 160
 161         def ok(self):
 162                 if self.job is None or self.job.status is not self.job.IN_PROGRESS:
 163                         if self["filelist"].canDescent(): # isDir
 164                                 self["filelist"].descent()
 165                                 self.session.summary.setText(self["filelist"].getFilename())
 166                                 self.check_for_NFO()
 167                         else:
 168                                 self.queryFlash()
 169
 170         def queryFlash(self):
 171                 fd = open(self.nfifile, 'r')
 172                 print fd
 173                 sign = fd.read(11)
 174                 print sign
 175                 if sign.find("NFI1" + self.box + "\0") == 0:
 176                         if self.md5sum != "":
 177                                 self["statusbar"].text = _("Please wait for md5 signature verification...")
 178                                 self.session.summary.setText(_("Please wait for md5 signature verification..."))
 179                                 self.container = eConsoleAppContainer()
 180                                 self.container.setCWD(self["filelist"].getCurrentDirectory())
 181                                 self.container.appClosed.get().append(self.md5finished)
 182                                 self.container.execute("md5sum -cs " + self.md5sum)
 183                         else:
 184                                 self.session.openWithCallback(self.queryCB, MessageBox, _("This .NFI file does not have a md5sum signature and is not guaranteed to work. Do you really want to burn this image to flash memory?"), MessageBox.TYPE_YESNO)
 185                 else:
 186                         self.session.open(MessageBox, (_("This .NFI file does not contain a valid %s image!") % (self.box.upper())), MessageBox.TYPE_ERROR)
 187
 188         def md5finished(self, retval):
 189                 if retval==0:
 190                         self.session.openWithCallback(self.queryCB, MessageBox, _("This .NFI file has a valid md5 signature. Continue programming this image to flash memory?"), MessageBox.TYPE_YESNO)
 191                 else:
 192                         self.session.openWithCallback(self.queryCB, MessageBox, _("The md5sum validation failed, the file may be corrupted! Are you sure that you want to burn this image to flash memory? You are doing this at your own risk!"), MessageBox.TYPE_YESNO)
 193
 194         def queryCB(self, answer):
 195                 if answer == True:
 196                         self.createJob()
 197                 else:
 198                         self["statusbar"].text = _("Please select .NFI flash image file from medium")
 199
 200         def createJob(self):
 201                 self.job = Job("Image flashing job")
 202                 param = [self.nfifile]
 203                 writeNAND(self.job,param,self.box)
 204                 #writeNAND2(self.job,param)
 205                 #writeNAND3(self.job,param)
 206                 self.job.state_changed.append(self.update_job)
 207                 self.job.end = 540
 208                 self.cwd = self["filelist"].getCurrentDirectory()
 209                 self["job_progressbar"].range = self.job.end
 210                 self.startJob()
 211
 212         def startJob(self):
 213                 self["key_blue"].text = ""
 214                 self["key_yellow"].text = ""
 215                 self["key_green"].text = ""
 216                 #self["progress0"].show()
 217                 #self["progress1"].show()
 218
 219                 self.job.start(self.jobcb)
 220
 221         def update_job(self):
 222                 j = self.job
 223                 #print "[job state_changed]"
 224                 if j.status == j.IN_PROGRESS:
 225                         self.session.summary["job_progressbar"].value = j.progress
 226                         self.session.summary["job_progressbar"].range = j.end
 227                         self.session.summary["job_progresslabel"].text = "%.2f%%" % (100*j.progress/float(j.end))
 228                         self["job_progressbar"].range = j.end
 229                         self["job_progressbar"].value = j.progress
 230                         #print "[update_job] j.progress=%f, j.getProgress()=%f, j.end=%d, text=%f" % (j.progress, j.getProgress(), j.end,  (100*j.progress/float(j.end)))
 231                         self["job_progresslabel"].text = "%.2f%%" % (100*j.progress/float(j.end))
 232                         self.session.summary.setText(j.tasks[j.current_task].name)
 233                         self["statusbar"].text = (j.tasks[j.current_task].name)
 234
 235                 elif j.status == j.FINISHED:
 236                         self["statusbar"].text = _("Writing NFI image file to flash completed")
 237                         self.session.summary.setText(_("NFI image flashing completed. Press Yellow to Reboot!"))
 238                         self["key_yellow"].text = _("Reboot")
 239
 240                 elif j.status == j.FAILED:
 241                         self["statusbar"].text = j.tasks[j.current_task].name + " " + _("failed")
 242                         self.session.open(MessageBox, (_("Flashing failed") + ":\n" + j.tasks[j.current_task].name + ":\n" + j.tasks[j.current_task].output_line), MessageBox.TYPE_ERROR)
 243
 244         def jobcb(self, jobref, fasel, blubber):
 245                 print "[jobcb] %s %s %s" % (jobref, fasel, blubber)
 246                 self["key_green"].text = _("Flash")
 247
 248         def reboot(self):
 249                 if self.job.status == self.job.FINISHED:
 250                         self["statusbar"].text = _("rebooting...")
 251                         TryQuitMainloop(self.session,2)
 252
 253         def createSummary(self):
 254                 return NFISummary