blob: aad703d182193806f84bb8bd1b3c07446d44f182 [file] [log] [blame]
#!/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