#!/usr/bin/env python import gobject import dbus import dbus.service import dbus.mainloop.glib import subprocess import tempfile import shutil import tarfile import os from obmc.dbuslib.bindings import get_dbus, DbusProperties, DbusObjectManager DBUS_NAME = 'org.openbmc.control.BmcFlash' OBJ_NAME = '/org/openbmc/control/flash/bmc' DOWNLOAD_INTF = 'org.openbmc.managers.Download' BMC_DBUS_NAME = 'org.openbmc.control.Bmc' BMC_OBJ_NAME = '/org/openbmc/control/bmc0' UPDATE_PATH = '/run/initramfs' def doExtract(members, files): for tarinfo in members: if tarinfo.name in files: yield tarinfo class BmcFlashControl(DbusProperties, DbusObjectManager): def __init__(self, bus, name): super(BmcFlashControl, self).__init__( conn=bus, object_path=name) self.Set(DBUS_NAME, "status", "Idle") self.Set(DBUS_NAME, "filename", "") self.Set(DBUS_NAME, "preserve_network_settings", True) self.Set(DBUS_NAME, "restore_application_defaults", False) self.Set(DBUS_NAME, "update_kernel_and_apps", False) self.Set(DBUS_NAME, "clear_persistent_files", False) self.Set(DBUS_NAME, "auto_apply", False) bus.add_signal_receiver( self.download_error_handler, signal_name="DownloadError") bus.add_signal_receiver( self.download_complete_handler, signal_name="DownloadComplete") self.update_process = None self.progress_name = None @dbus.service.method( DBUS_NAME, in_signature='ss', out_signature='') def updateViaTftp(self, ip, filename): self.Set(DBUS_NAME, "status", "Downloading") self.TftpDownload(ip, filename) @dbus.service.method( DBUS_NAME, in_signature='s', out_signature='') def update(self, filename): self.Set(DBUS_NAME, "filename", filename) self.download_complete_handler(filename, filename) @dbus.service.signal(DOWNLOAD_INTF, signature='ss') def TftpDownload(self, ip, filename): self.Set(DBUS_NAME, "filename", filename) pass ## Signal handler def download_error_handler(self, filename): if (filename == self.Get(DBUS_NAME, "filename")): self.Set(DBUS_NAME, "status", "Download Error") def download_complete_handler(self, outfile, filename): ## do update if (filename != self.Get(DBUS_NAME, "filename")): return print "Download complete. Updating..." self.Set(DBUS_NAME, "status", "Download Complete") copy_files = {} ## determine needed files if not self.Get(DBUS_NAME, "update_kernel_and_apps"): copy_files["image-bmc"] = True else: copy_files["image-kernel"] = True copy_files["image-initramfs"] = True copy_files["image-rofs"] = True if self.Get(DBUS_NAME, "restore_application_defaults"): copy_files["image-rwfs"] = True ## make sure files exist in archive try: tar = tarfile.open(outfile, "r") files = {} for f in tar.getnames(): files[f] = True tar.close() for f in copy_files.keys(): if f not in files: raise Exception( "ERROR: File not found in update archive: "+f) except Exception as e: print e self.Set(DBUS_NAME, "status", "Unpack Error") return try: tar = tarfile.open(outfile, "r") tar.extractall(UPDATE_PATH, members=doExtract(tar, copy_files)) tar.close() if self.Get(DBUS_NAME, "clear_persistent_files"): print "Removing persistent files" try: os.unlink(UPDATE_PATH+"/whitelist") except OSError as e: if (e.errno == errno.EISDIR): pass elif (e.errno == errno.ENOENT): pass else: raise try: wldir = UPDATE_PATH + "/whitelist.d" for file in os.listdir(wldir): os.unlink(os.path.join(wldir, file)) except OSError as e: if (e.errno == errno.EISDIR): pass else: raise if self.Get(DBUS_NAME, "preserve_network_settings"): print "Preserving network settings" shutil.copy2("/run/fw_env", UPDATE_PATH+"/image-u-boot-env") except Exception as e: print e self.Set(DBUS_NAME, "status", "Unpack Error") self.Verify() def Verify(self): self.Set(DBUS_NAME, "status", "Checking Image") try: subprocess.check_call([ "/run/initramfs/update", "--no-flash", "--no-save-files", "--no-restore-files", "--no-clean-saved-files"]) self.Set(DBUS_NAME, "status", "Image ready to apply.") if (self.Get(DBUS_NAME, "auto_apply")): self.Apply() except: self.Set(DBUS_NAME, "auto_apply", False) try: subprocess.check_output([ "/run/initramfs/update", "--no-flash", "--ignore-mount", "--no-save-files", "--no-restore-files", "--no-clean-saved-files"], stderr=subprocess.STDOUT) self.Set( DBUS_NAME, "status", "Deferred for mounted filesystem. reboot BMC to apply.") except subprocess.CalledProcessError as e: self.Set( DBUS_NAME, "status", "Verify error: %s" % e.output) except OSError as e: self.Set( DBUS_NAME, "status", "Verify error: problem calling update: %s" % e.strerror) def Cleanup(self): if self.progress_name: try: os.unlink(self.progress_name) self.progress_name = None except oserror as e: if e.errno == EEXIST: pass raise self.update_process = None self.Set(DBUS_NAME, "status", "Idle") @dbus.service.method( DBUS_NAME, in_signature='', out_signature='') def Abort(self): if self.update_process: try: self.update_process.kill() except: pass for file in os.listdir(UPDATE_PATH): if file.startswith('image-'): os.unlink(os.path.join(UPDATE_PATH, file)) self.Cleanup() @dbus.service.method( DBUS_NAME, in_signature='', out_signature='s') def GetUpdateProgress(self): msg = "" if self.update_process and self.update_process.returncode is None: self.update_process.poll() if (self.update_process is None): pass elif (self.update_process.returncode > 0): self.Set(DBUS_NAME, "status", "Apply failed") elif (self.update_process.returncode is None): pass else: # (self.update_process.returncode == 0) files = "" for file in os.listdir(UPDATE_PATH): if file.startswith('image-'): files = files + file if files == "": msg = "Apply Complete. Reboot to take effect." else: msg = "Apply Incomplete, Remaining:" + files self.Set(DBUS_NAME, "status", msg) msg = self.Get(DBUS_NAME, "status") + "\n" if self.progress_name: try: prog = open(self.progress_name, 'r') for line in prog: # strip off initial sets of xxx\r here # ignore crlf at the end # cr will be -1 if no '\r' is found cr = line.rfind("\r", 0, -2) msg = msg + line[cr + 1:] except OSError as e: if (e.error == EEXIST): pass raise return msg @dbus.service.method( DBUS_NAME, in_signature='', out_signature='') def Apply(self): progress = None self.Set(DBUS_NAME, "status", "Writing images to flash") try: progress = tempfile.NamedTemporaryFile( delete=False, prefix="progress.") self.progress_name = progress.name self.update_process = subprocess.Popen([ "/run/initramfs/update"], stdout=progress.file, stderr=subprocess.STDOUT) except Exception as e: try: progress.close() os.unlink(progress.name) self.progress_name = None except: pass raise try: progress.close() except: pass @dbus.service.method( DBUS_NAME, in_signature='', out_signature='') def PrepareForUpdate(self): subprocess.call([ "fw_setenv", "openbmconce", "copy-files-to-ram copy-base-filesystem-to-ram"]) self.Set(DBUS_NAME, "status", "Switch to update mode in progress") o = bus.get_object(BMC_DBUS_NAME, BMC_OBJ_NAME) intf = dbus.Interface(o, BMC_DBUS_NAME) intf.warmReset() if __name__ == '__main__': dbus.mainloop.glib.DBusGMainLoop(set_as_default=True) bus = get_dbus() obj = BmcFlashControl(bus, OBJ_NAME) mainloop = gobject.MainLoop() obj.unmask_signals() name = dbus.service.BusName(DBUS_NAME, bus) print "Running Bmc Flash Control" mainloop.run() # vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4